source: proto/JSON/json_pablo_template.c @ 869

Last change on this file since 869 was 869, checked in by cameron, 8 years ago

Update for new pablo with @global structs.

File size: 7.4 KB
Line 
1#include <string.h>
2#include <stdio.h>
3#include <stdlib.h>
4#include <errno.h>
5#include <sys/types.h>
6#include <sys/stat.h>
7
8#include "../lib/lib_simd.h"
9#include "../lib/perflib/perfsec.h"
10#include "../lib/s2p.h"
11#include "../lib/carryQ.h"
12
13typedef SIMD_type BytePack;
14typedef SIMD_type BitBlock;
15
16#define BLOCK_SIZE (sizeof(SIMD_type) * 8)      /* BLOCK_SIZE           - SIMD register in bits */
17#define BUFFER_SIZE (BLOCK_SIZE * 16)           /* BUFFER_SIZE          - multiple of BLOCK_SIZE */
18#define OVERLAP_BUFSIZE BLOCK_SIZE              /* OVERLAP_BUFSIZE      - SIMD register width in bytes overlap */
19
20#define DEBUG 1
21
22#ifdef BUFFER_PROFILING
23        BOM_Table * perf_timer;
24#elif CODE_CLOCKER
25        #define NUM_EVENTS 1
26        int Events[NUM_EVENTS] = {PAPI_TOT_CYC};
27        //int Events[NUM_EVENTS] = {PAPI_L2_DCM};
28        //int Events[NUM_EVENTS] = {PAPI_TOT_CYC, PAPI_BR_MSP};
29        int cal_size = 20;
30        CC * perf_timer = new CC(Events,NUM_EVENTS,cal_size);
31#else
32        void * perf_timer;
33#endif
34
35@global
36
37void do_process(FILE *infile, FILE *outfile) {
38
39        /* Compiler generated stream variable declarations. */
40        @decl
41
42        // BitBlock error_mask;  // PyBit compiler auto declares this variable
43        BitBlock bit[8];
44        BitBlock EOF_mask = simd_const_1(1);
45
46        /*extra*/
47        /* parse_escape, do not declare, since the PyBit compiler does so by default*/
48        parity.Odd = simd_const_2(1);
49        parity.Even = simd_const_2(2);
50
51        BitBlock high_bit_mask = sisd_high_bit_mask;
52        BitBlock high_bit_is_set;
53        // BitBlock ParityMask = simd_const_1(0); // PyBit compiler auto declares this variable
54        BitBlock bitblock_parity_mask;
55
56        /* 8 * sizeof(SIMD_type) bytes. */
57        BytePack U8[8];
58
59        /* Source file data is broken into buffers, buffers are broken into blocks, blocks are SIMD register width. */
60        int buf_pos = 0;
61        int block_pos = 0;
62        int err_col = 0;
63
64        BytePack buf[(OVERLAP_BUFSIZE+BUFFER_SIZE+OVERLAP_BUFSIZE)];
65        char * srcbuf = ((char *) buf) + OVERLAP_BUFSIZE;
66        memset((unsigned char *)srcbuf,0,OVERLAP_BUFSIZE);
67        memset(((unsigned char *)srcbuf)+OVERLAP_BUFSIZE+BUFFER_SIZE,0,OVERLAP_BUFSIZE); /* Zero out first and last sizeof(SIMD_type) bytes. */
68
69        /* File IO */
70        int chars_avail = 0;
71        int chars_read = 0;
72
73        chars_read = fread((void *)srcbuf, 1, BUFFER_SIZE, infile);
74        chars_avail = chars_read;
75        if(ferror(infile)) { fprintf(stderr, "Error: Failed to read input file.\n"); return; }
76        if(chars_avail==0 && feof(infile)) { return; }
77
78        @stream_stmts
79
80        while(1) {
81
82                PERF_SEC_START(perf_timer);
83
84                while (chars_avail >= BLOCK_SIZE) { /* process full blocks */
85
86                        BytePack * U8 = (BytePack *) &srcbuf[block_pos];
87
88                        s2p(U8[0], U8[1], U8[2], U8[3], U8[4], U8[5], U8[6], U8[7],
89                        bit[0], bit[1], bit[2], bit[3], bit[4], bit[5], bit[6], bit[7]);
90
91                        @block_stmts
92
93                        /* Generate Parity Mask */
94                        high_bit_mask = sisd_high_bit_mask;
95                        high_bit_is_set = simd_and(high_bit_mask, bitblock_parity_mask);
96                        bitblock_parity_mask = sisd_add( simd_not(sisd_srl(high_bit_is_set, sisd_from_int(BLOCK_SIZE-1))) , sisd_low_bit_mask);
97                        parity.ParityMask = bitblock_parallel_prefix_parity(UnescapedDQuotes, bitblock_parity_mask);
98
99                        #ifdef DEBUG
100
101                        print_array_le("Data", &srcbuf[block_pos],BLOCK_SIZE);
102                        print_simd_register("Escape", Escape);
103                        print_simd_register("Unescaped DQuote",UnescapedDQuotes);
104                        /*
105                        print_simd_register("high_bit_is_set", high_bit_is_set);
106                        print_simd_register("high_bit_mask", high_bit_mask);
107                        */
108                        print_simd_register("parity.ParityMask", parity.ParityMask);
109                        print_simd_register("StringMask", StringMask);
110                        print_simd_register("StringSpans", StringSpans);
111
112                        print_simd_register("AtomStarts", AtomStarts);
113                        print_simd_register("StringStarts", StringStarts);
114                        print_simd_register("NumberStarts", NumberStarts);
115                        print_simd_register("TrueStarts", TrueStarts);
116                        print_simd_register("FalseStarts", FalseStarts);
117                        print_simd_register("NullStarts", NullStarts);
118                        print_simd_register("AtomSpans", AtomSpans);
119                        print_simd_register("ErrorMask", error_mask);
120
121                        #endif
122
123                        if (bitblock_has_bit(error_mask)){
124                                err_col = buf_pos + block_pos + count_forward_zeroes(error_mask) + 1;
125                                fprintf(stderr, "Initial error at column %i.\n", err_col);
126                                exit(-1);
127                        }
128
129                        chars_avail -= BLOCK_SIZE;
130                        block_pos += BLOCK_SIZE;
131                }
132
133                if(chars_avail > 0 || @any_carry ) { /* process final partial block */
134                        /* extra */
135                        EOF_mask = sisd_srl(simd_const_1(1),sisd_from_int(BLOCK_SIZE-chars_avail));
136
137                        BytePack * U8 = (BytePack *) &srcbuf[block_pos];
138
139                        s2p(U8[0], U8[1], U8[2], U8[3], U8[4], U8[5], U8[6], U8[7],
140                        bit[0], bit[1], bit[2], bit[3], bit[4], bit[5], bit[6], bit[7]);
141
142                        bit[0] = simd_and(bit[0], EOF_mask);
143                        bit[1] = simd_and(bit[1], EOF_mask);
144                        bit[2] = simd_and(bit[2], EOF_mask);
145                        bit[3] = simd_and(bit[3], EOF_mask);
146                        bit[4] = simd_and(bit[4], EOF_mask);
147                        bit[5] = simd_and(bit[5], EOF_mask);
148                        bit[6] = simd_and(bit[6], EOF_mask);
149                        bit[7] = simd_and(bit[7], EOF_mask);
150
151                        @block_stmts
152
153                        /* Generate Parity Mask */
154                        high_bit_mask = sisd_srl(sisd_high_bit_mask, sisd_from_int(BLOCK_SIZE-chars_avail));
155                        high_bit_is_set = simd_and(high_bit_mask, bitblock_parity_mask);
156                        bitblock_parity_mask = sisd_add(simd_not(sisd_srl(high_bit_is_set, sisd_from_int(chars_avail-1))) , sisd_low_bit_mask);
157                        parity.ParityMask = bitblock_parallel_prefix_parity(UnescapedDQuotes, bitblock_parity_mask);
158                        parity.ParityMask = simd_and(parity.ParityMask, EOF_mask);
159
160                        #ifdef DEBUG
161
162                        print_array_le("Data", &srcbuf[block_pos],BLOCK_SIZE);
163                        print_simd_register("Escape", Escape);
164                        print_simd_register("Unescaped DQuote",UnescapedDQuotes);
165                        /*
166                        print_simd_register("high_bit_is_set", high_bit_is_set);
167                        print_simd_register("high_bit_mask", high_bit_mask);
168                        */
169                        print_simd_register("parity.ParityMask", parity.ParityMask);
170                        print_simd_register("StringMask", StringMask);
171                        print_simd_register("StringSpans", StringSpans);
172
173                        print_simd_register("AtomStarts", AtomStarts);
174                        print_simd_register("StringStarts", StringStarts);
175                        print_simd_register("NumberStarts", NumberStarts);
176                        print_simd_register("TrueStarts", TrueStarts);
177                        print_simd_register("FalseStarts", FalseStarts);
178                        print_simd_register("NullStarts", NullStarts);
179                        print_simd_register("AtomSpans", AtomSpans);
180                        print_simd_register("ErrorMask", error_mask);
181
182                        #endif
183
184                        if(bitblock_has_bit(error_mask)) {
185                                err_col = buf_pos + block_pos + count_forward_zeroes(error_mask) + 1;
186                                fprintf(stderr, "Initial error at position %i.\n", err_col);
187                                exit(-1);
188                        }
189                }
190
191                PERF_SEC_END(perf_timer, chars_avail);
192
193                buf_pos += chars_read;
194                block_pos = 0;
195
196                chars_read = fread((void *)srcbuf, 1, BUFFER_SIZE, infile);
197                chars_avail = chars_read;
198                if(ferror(infile)) { fprintf(stderr, "Failed to read input file.\n"); return; }
199                if(chars_avail==0 && feof(infile)) { return; }
200        }
201}
202
203int main(int argc, char * argv[]) {
204        char * infilename, * outfilename;
205        FILE *infile, *outfile;
206        struct stat fileinfo;
207
208        if (argc < 2) {
209                printf("Usage: %s <filename> [<outputfile>]\n", argv[0]);
210                exit(-1);
211        }
212
213        infilename = argv[1];
214        stat(infilename, &fileinfo);
215        infile = fopen(infilename, "rb");
216        if (!infile) {
217                fprintf(stderr, "Error: Cannot open %s for reading.\n", infilename);
218                exit(-1);
219        }
220
221        if (argc < 3) {
222                outfile = stdout;
223        }       else {
224                outfilename = argv[2];
225                outfile = fopen(outfilename, "wb");
226                if (!outfile) {
227                        fprintf(stderr, "Error: Cannot open %s for writing.\n", outfilename);
228                        exit(-1);
229                }
230        }
231
232        // PERF_SEC_BIND(1);
233
234        PERF_SEC_INIT(perf_timer);
235
236        do_process(infile, outfile);
237
238        PERF_SEC_DUMP(perf_timer);
239
240        PERF_SEC_DESTROY(perf_timer);
241
242        fclose(infile);
243        fclose(outfile);
244
245        return(0);
246}
Note: See TracBrowser for help on using the repository browser.