-
Notifications
You must be signed in to change notification settings - Fork 4
/
translate-data.c
107 lines (92 loc) · 2.53 KB
/
translate-data.c
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
/* This file is part of the Theseus distribution.
* Copyright 2020 Joshua E. Hill <[email protected]>
*
* Licensed under the 3-clause BSD license. For details, see the LICENSE file.
*
* Author(s)
* Joshua E. Hill, UL VS LLC.
* Joshua E. Hill, KeyPair Consulting, Inc. <[email protected]>
*/
#include <assert.h>
#include <getopt.h> // for getopt, optind
#include <limits.h>
#include <stdbool.h>
#include <stdint.h> // for uint32_t
#include <stdio.h>
#include <stdlib.h>
#include <stdnoreturn.h>
#include <sysexits.h>
#include <time.h>
#include "binio.h"
#include "entlib.h"
#include "globals-inst.h"
#include "precision.h"
#include "translate.h"
noreturn static void useageExit(void) {
fprintf(stderr, "Usage:\n");
fprintf(stderr, "translate-data [-v] <inputfile>\n");
fprintf(stderr, "Perform an order-preserving map to arrange the input symbols to (0, ..., k-1)\n");
fprintf(stderr, "inputfile is presumed to consist of " STATDATA_STRING " in machine format\n");
fprintf(stderr, "output is sent to stdout, and is " STATDATA_STRING " in machine format\n");
fprintf(stderr, "-v\tVerbose mode (can be used up to 3 times for increased verbosity).\n");
fprintf(stderr, "-n\tNo data output. Report number of symbols on stdout.\n");
exit(EX_USAGE);
}
int main(int argc, char *argv[]) {
FILE *infp;
size_t datalen;
statData_t *data;
size_t k;
int opt;
bool configSymbolsOnly;
double median;
configVerbose = 0;
configSymbolsOnly = false;
data = NULL;
assert(PRECISION(UINT_MAX) >= 32);
while ((opt = getopt(argc, argv, "vn")) != -1) {
switch (opt) {
case 'v':
configVerbose++;
break;
case 'n':
configSymbolsOnly = true;
break;
default: /* ? */
useageExit();
}
}
argc -= optind;
argv += optind;
if (argc != 1) {
useageExit();
}
if ((infp = fopen(argv[0], "rb")) == NULL) {
perror("Can't open file");
exit(EX_NOINPUT);
}
datalen = readuintfile(infp, &data);
assert(data != NULL);
if (configVerbose > 0) {
fprintf(stderr, "Read in %zu integers\n", datalen);
}
if (fclose(infp) != 0) {
perror("Couldn't close input data file");
exit(EX_OSERR);
}
assert(datalen > 0);
translate(data, datalen, &k, &median);
if (configVerbose > 0) {
fprintf(stderr, "Found %zu symbols\n", k);
}
if (configSymbolsOnly) {
printf("%zu\n", k);
} else {
if (fwrite(data, sizeof(statData_t), datalen, stdout) != datalen) {
perror("Can't write to output");
exit(EX_OSERR);
}
}
free(data);
return EX_OK;
}