-
Notifications
You must be signed in to change notification settings - Fork 1
/
bcat.c
74 lines (67 loc) · 2.3 KB
/
bcat.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
#include "util.h"
#include "argh.h"
#include "load.h"
#include "write_simple.h"
#define DESCRIPTION "cat some bsv files to csv\n\n"
#define USAGE "bcat [-l|--lz4] [-p|--prefix] [-h N|--head N] FILE1 ... FILEN\n\n"
#define EXAMPLE \
">> for char in a a b b c c; do\n" \
" echo $char | bsv >> /tmp/$char\n" \
" done\n\n" \
">> bcat --head 1 --prefix /tmp/{a,b,c}\n" \
"/tmp/a:a\n" \
"/tmp/b:b\n" \
"/tmp/c:c\n"
int main(int argc, char **argv) {
// setup bsv
SETUP();
// setup state
i32 ran = 0;
i64 line;
// parse args
bool prefix = false;
bool lz4 = false;
i64 head = 0;
ARGH_PARSE {
ARGH_NEXT();
if ARGH_BOOL("-p", "--prefix") { prefix = true; }
else if ARGH_BOOL("-l", "--lz4") { lz4 = true; }
else if ARGH_FLAG("-h", "--head") { ASSERT(isdigits(ARGH_VAL()), "fatal: should have been `--head INT`, not `--head %s`\n", ARGH_VAL());
head = atol(ARGH_VAL());}
}
// setup input
ASSERT(ARGH_ARGC > 0, "usage: %s", USAGE);
FILE *files[ARGH_ARGC];
for (i32 i = 0; i < ARGH_ARGC; i++)
FOPEN(files[i], ARGH_ARGV[i], "rb");
readbuf_t rbuf = rbuf_init(files, ARGH_ARGC, lz4);
row_t row;
// setup output
writebuf_t wbuf = wbuf_init((FILE*[]){stdout}, 1);
// process input row by row
for (i32 i = 0; i < ARGH_ARGC; i++) {
line = 0;
while (1) {
line++;
load_next(&rbuf, &row, i);
if (row.stop)
break;
if (head != 0 && line > head)
break;
if (prefix) {
write_bytes(&wbuf, ARGH_ARGV[i], strlen(ARGH_ARGV[i]), 0);
write_bytes(&wbuf, ":", 1, 0);
}
for (i32 j = 0; j <= row.max; j++) {
write_bytes(&wbuf, row.columns[j], row.sizes[j], 0);
if (j != row.max)
write_bytes(&wbuf, ",", 1, 0);
}
write_bytes(&wbuf, "\n", 1, 0);
ran = 1;
}
}
if (ran == 0)
write_bytes(&wbuf, "\n", 1, 0);
write_flush(&wbuf, 0);
}