source: proto/parabix2/pablo_template_avx.cpp @ 1496

Last change on this file since 1496 was 1468, checked in by ksherdy, 8 years ago

Centralized typedefs.

File size: 12.3 KB
Line 
1#include <stdio.h>
2#include <stdlib.h>
3#include <errno.h>
4#include <sys/types.h>
5#include <sys/stat.h>
6#include "../lib/lib_simd_avx.h"
7
8#define BLOCK_SIZE (sizeof(SIMD_type) * 8)
9#define SEGMENT_BLOCKS 12
10#define BUFFER_SIZE (BLOCK_SIZE * SEGMENT_BLOCKS)
11#define OVERLAP_BUFSIZE (sizeof(SIMD_type))
12
13#include "../lib/carryQ_avx.h"
14#include "xmldecl.h"
15#include "xml_error.c"
16#include "xmldecl.c"
17#include "namechars.h"
18
19#include "../lib/perflib/perfsec.h"
20#include "../lib/s2p.h"
21
22#include "TagMatcher_avx.h"
23#include "LineColTracker.h"
24
25#ifdef BUFFER_PROFILING
26        BOM_Table * parser_timer;
27
28#elif CODE_CLOCKER
29        #define NUM_EVENTS 1
30        int Events[NUM_EVENTS] = {PAPI_TOT_CYC};
31        //int Events[NUM_EVENTS] = {PAPI_L2_DCM};
32        //int Events[NUM_EVENTS] = {PAPI_TOT_CYC, PAPI_BR_MSP};
33        int cal_size = 20;
34        CC * parser_timer = new CC(Events,NUM_EVENTS,cal_size);
35#else
36        void * parser_timer;
37#endif
38
39int block_base=0;
40int buffer_base=0;
41int buffer_last;
42char * source;
43LineColTracker tracker;
44TagMatcher matcher;
45BitBlock EOF_mask = simd_const_1(1);
46
47
48static inline int StreamScan(ScanBlock * stream, int blk_count, int ProcessPos(int)) {
49        int blk;
50        int block_pos = 0;
51
52        for (blk = 0; blk < blk_count; blk++) {
53                ScanBlock s = stream[blk];
54                while(s) {
55                        int code = (ProcessPos(cfzl(s) + block_pos));
56                        if (code) return code;
57                        s = s & (s-1);  // clear rightmost bit.
58                }
59                block_pos += 8 * sizeof(ScanBlock);
60        }
61        return 0;
62}
63
64
65static inline void ReportError(const char * error_msg, int error_pos_in_block) {
66  int error_line, error_column;
67  tracker.get_Line_and_Column(error_pos_in_block, error_line, error_column);
68  fprintf(stderr, "%s at line %i, column %i\n", error_msg, error_line, error_column);
69}
70
71
72static inline int NameStrt_check(int pos) {
73        int block_pos = block_base + pos;
74        if(XML_10_UTF8_NameStrt_bytes((unsigned char*)&source[block_pos]) == 0){
75              ReportError("name start error", pos);
76              exit(-1);
77        }
78        return 0;
79}
80
81static inline int Name_check(int pos) {
82        int block_pos = block_base + pos;
83        if(XML_10_UTF8_NameChar_bytes((unsigned char*)&source[block_pos]) == 0){
84              ReportError("name error", pos);
85              exit(-1);
86        }
87        return 0;
88}
89
90static inline int PIName_check(int pos) {
91        int block_pos = block_base + pos;
92        int file_pos = block_pos+buffer_base;
93        if (at_XxMmLll<ASCII>((unsigned char*)&source[block_pos]) && (source[block_pos+3]=='?' || source[block_pos+3]<= ' ')) {
94              // "<?xml" legal at start of file.
95              if ((file_pos == 2) && at_XmlDecl_start<ASCII>((unsigned char*)&source[0])) return 0;
96              ReportError("[Xx][Mm][Ll] illegal as PI name", pos);
97              exit(-1);
98        }
99        return 0;
100}
101
102static inline int CD_check(int pos) {
103        int block_pos = block_base + pos;
104        if (!at_CDATA1<ASCII>((unsigned char*)&source[block_pos])){
105              ReportError("CDATA error", pos);
106              exit(-1);
107        }
108        return 0;
109}
110
111static inline int GenRef_check(int pos) {
112        int block_pos = block_base + pos;
113        unsigned char* s = (unsigned char*)&source[block_pos];
114        if (!(at_Ref_gt<ASCII>(s)||at_Ref_lt<ASCII>(s)||at_Ref_amp<ASCII>(s)||at_Ref_quot<ASCII>(s)||at_Ref_apos<ASCII>(s))){
115              ReportError("Undefined reference", pos);
116              exit(-1);
117        }
118        return 0;
119}
120
121static inline int HexRef_check(int pos) {
122        int block_pos = block_base + pos;
123        unsigned char* s = (unsigned char*)&source[block_pos];
124        int ch_val = 0;
125        while(at_HexDigit<ASCII>(s)){
126          ch_val = HexVal<ASCII>(s[0]) + (ch_val<<4);
127          if (ch_val> 0x10FFFF ){
128            ReportError("Illegal character reference", pos);
129            exit(-1);
130          }
131          s++;
132        }
133        if ((ch_val == 0x0) || ((ch_val | 0x7FF) == 0xDFFF)|| ((ch_val | 0x1) == 0xFFFF)){
134          ReportError("Illegal character reference", pos);
135          exit(-1);
136        }
137        else if (((ch_val < 0x20) && (ch_val != 0x9) && (ch_val != 0xD) && (ch_val != 0xA))){
138          ReportError("Illegal XML 1.0 character reference", pos);
139          exit(-1);
140        }
141        return 0;
142}
143
144static inline int DecRef_check(int pos) {
145        int block_pos = block_base + pos;
146        unsigned char* s = (unsigned char*)&source[block_pos];
147        int ch_val = 0;
148        while(at_HexDigit<ASCII>(s)){
149          ch_val = DigitVal<ASCII>(s[0]) + ch_val*10;
150          if (ch_val> 0x10FFFF ){
151            ReportError("Illegal character reference", pos);
152            exit(-1);
153          }
154          s++;
155        }
156        if ((ch_val == 0x0) || ((ch_val | 0x7FF) == 0xDFFF)|| ((ch_val | 0x1) == 0xFFFF)){
157          ReportError("Illegal character reference", pos);
158          exit(-1);
159        }
160        else if (((ch_val < 0x20) && (ch_val != 0x9) && (ch_val != 0xD) && (ch_val != 0xA))){
161          ReportError("Illegal XML 1.0 character reference", pos);
162          exit(-1);
163        }
164        return 0;
165}
166
167static inline int AttRef_check(int pos) {
168        int block_pos = block_base + pos;
169        unsigned char* s = (unsigned char*)&source[block_pos];
170        int ch_val = 0;
171        if(s[0]=='#'){
172          s++;
173          if(s[0]=='x' || s[0]=='X'){
174            s++;
175            while(at_HexDigit<ASCII>(s)){
176              ch_val = HexVal<ASCII>(s[0]) + (ch_val<<4);
177              s++;
178            }
179          }
180          else{
181            while(at_HexDigit<ASCII>(s)){
182              ch_val = DigitVal<ASCII>(s[0]) + ch_val*10;
183              s++;
184            }
185          }
186          if (ch_val==60){
187            ReportError("Attribute values contain '<' characters after reference expansion", pos);
188            exit(-1);
189          }
190        }
191        else if(at_Ref_lt<ASCII>(s)){
192          ReportError("Attribute values contain '<' characters after reference expansion", pos);
193          exit(-1);
194        }
195        return 0;
196}
197
198
199
200@global
201
202static inline void s2p_do_block(BytePack U8[], Basis_bits & basis_bits) {
203  s2p(U8[0], U8[1], U8[2], U8[3], U8[4], U8[5], U8[6], U8[7],
204        basis_bits.bit_0, basis_bits.bit_1, basis_bits.bit_2, basis_bits.bit_3, basis_bits.bit_4, basis_bits.bit_5, basis_bits.bit_6, basis_bits.bit_7);
205}
206
207static inline void s2p_do_final_block(BytePack U8[], Basis_bits & basis_bits, BitBlock EOF_mask) {
208  s2p_do_block(U8, basis_bits);
209  basis_bits.bit_0 = simd_and(basis_bits.bit_0, EOF_mask);
210  basis_bits.bit_1 = simd_and(basis_bits.bit_1, EOF_mask);
211  basis_bits.bit_2 = simd_and(basis_bits.bit_2, EOF_mask);
212  basis_bits.bit_3 = simd_and(basis_bits.bit_3, EOF_mask);
213  basis_bits.bit_4 = simd_and(basis_bits.bit_4, EOF_mask);
214  basis_bits.bit_5 = simd_and(basis_bits.bit_5, EOF_mask);
215  basis_bits.bit_6 = simd_and(basis_bits.bit_6, EOF_mask);
216  basis_bits.bit_7 = simd_and(basis_bits.bit_7, EOF_mask);
217}
218
219static inline void postprocess_do_block(Lex & lex, CtCDPI_Callouts & ctCDPI_Callouts, Ref_Callouts & ref_Callouts, Check_streams & check_streams, int chars_avail){
220            tracker.StoreNewlines(lex.LF);
221
222                if (bitblock_has_bit(simd_or(check_streams.non_ascii_name_starts, check_streams.non_ascii_names))) {
223                  StreamScan((ScanBlock *) &check_streams.non_ascii_name_starts, sizeof(BitBlock)/sizeof(ScanBlock), NameStrt_check);
224                  StreamScan((ScanBlock *) &check_streams.non_ascii_names, sizeof(BitBlock)/sizeof(ScanBlock), Name_check);
225                }
226
227                if (bitblock_has_bit(ctCDPI_Callouts.PI_name_starts)){
228                  StreamScan((ScanBlock *) &ctCDPI_Callouts.PI_name_starts, sizeof(BitBlock)/sizeof(ScanBlock), PIName_check);
229                }
230
231                if (bitblock_has_bit(ctCDPI_Callouts.CD_starts)){
232                  StreamScan((ScanBlock *) &ctCDPI_Callouts.CD_starts, sizeof(BitBlock)/sizeof(ScanBlock), CD_check);
233                }
234
235                if (bitblock_has_bit(ref_Callouts.GenRef_starts)){
236                  StreamScan((ScanBlock *) &ref_Callouts.GenRef_starts, sizeof(BitBlock)/sizeof(ScanBlock), GenRef_check);
237                }
238
239                if (bitblock_has_bit(ref_Callouts.DecRef_starts)){
240                  StreamScan((ScanBlock *) &ref_Callouts.DecRef_starts, sizeof(BitBlock)/sizeof(ScanBlock), DecRef_check);
241                }
242
243                if (bitblock_has_bit(ref_Callouts.HexRef_starts)){
244                  StreamScan((ScanBlock *) &ref_Callouts.HexRef_starts, sizeof(BitBlock)/sizeof(ScanBlock), HexRef_check);
245                }
246
247                if (bitblock_has_bit(check_streams.att_refs)){
248                  StreamScan((ScanBlock *) &check_streams.att_refs, sizeof(BitBlock)/sizeof(ScanBlock), AttRef_check);
249                }
250
251                if (bitblock_has_bit(check_streams.error_mask)) {
252                  int errpos = count_forward_zeroes(check_streams.error_mask);
253                  ReportError("error found", errpos);
254              exit(-1);
255                }
256
257                matcher.store_streams(check_streams.tag_marks, check_streams.name_follows, check_streams.misc_mask, chars_avail);
258                tracker.AdvanceBlock();
259}
260
261void do_process(FILE *infile, FILE *outfile) {
262
263@decl
264
265  int buf_pos = 0;
266  int block_pos = 0;
267  int errpos = 0;
268  int chars_avail = 0;
269  int check_pos = 0;
270  int chars_read = 0;
271  BytePack buf[(BUFFER_SIZE+BLOCK_SIZE+OVERLAP_BUFSIZE*2)/sizeof(SIMD_type)];
272
273  char * srcbuf = ((char *) buf) + OVERLAP_BUFSIZE;
274  buffer_base = buf_pos;
275  source = srcbuf;
276
277  chars_read = fread((void *)srcbuf, 1, BUFFER_SIZE + OVERLAP_BUFSIZE, infile);
278  chars_avail = chars_read;
279  if (chars_avail > BUFFER_SIZE) chars_avail = BUFFER_SIZE;
280
281  matcher.setSrc(srcbuf);
282
283  if(chars_read<4){
284    fprintf(stderr,"File is too short. Not well formed.\n");
285    exit(-1);
286  }
287
288  Entity_Info * e = new Entity_Info;
289  e->AnalyzeSignature((unsigned char *)srcbuf);
290
291  if (e->code_unit_base == ASCII) {
292
293    XML_Decl_Parser<ASCII> decl_parser((unsigned char *)srcbuf);
294
295    decl_parser.ReadXMLInfo(*e);
296
297    if (e->code_unit_size != SingleByte || (e->has_encoding_decl && (!at_UTF_8(e->encoding)))){
298        fprintf(stderr,"Sorry, this xmlwf demo only works for UTF-8.\n");
299        exit(-1);
300    }
301  }
302  else {
303    fprintf(stderr,"Sorry, this xmlwf demo does not process EBCDIC.\n");
304        exit(-1);
305  }
306
307  if (e->content_start != 0) {
308        memmove(&srcbuf[0], &srcbuf[e->content_start], chars_read - e->content_start);
309        buf_pos = e->content_start;
310        if (chars_avail == BUFFER_SIZE) {
311                chars_read = chars_read - e->content_start +
312                             fread(&srcbuf[chars_read-e->content_start], 1, e->content_start, infile);
313                chars_avail = chars_read;
314                if (chars_avail > BUFFER_SIZE) chars_avail = BUFFER_SIZE;
315        }
316        else {
317          chars_read -=e->content_start;
318          chars_avail -=e->content_start;
319        }
320  }
321
322@stream_stmts
323
324/* Full Buffers */
325
326    while (chars_avail == BUFFER_SIZE) {
327      PERF_SEC_START(parser_timer);
328      for (int blk = 0; blk < SEGMENT_BLOCKS; blk++) {
329          block_base = blk*BLOCK_SIZE;
330          s2p_do_block((BytePack *) &srcbuf[block_base], basis_bits);
331          @block_stmts
332          postprocess_do_block(lex, ctCDPI_Callouts, ref_Callouts, check_streams, chars_avail);
333      }
334      matcher.StreamScan(chars_avail);
335      matcher.Advance_buffer();
336      tracker.Advance_buffer();
337      PERF_SEC_END(parser_timer, chars_avail);
338           
339      int bytes_left = chars_read - chars_avail;
340      memmove(srcbuf, &srcbuf[BUFFER_SIZE], bytes_left);
341      chars_read = fread(&srcbuf[bytes_left],1, BUFFER_SIZE + OVERLAP_BUFSIZE - bytes_left, infile) + bytes_left;
342      chars_avail = chars_read;
343      if (chars_avail > BUFFER_SIZE) chars_avail = BUFFER_SIZE;
344      buf_pos += chars_avail;
345      buffer_base = buf_pos;
346    }
347/* Final Partial Buffer */
348    PERF_SEC_START(parser_timer);
349
350    block_pos = 0;
351    int remaining = chars_avail;
352/* Full Blocks */
353    while (remaining >= BLOCK_SIZE) {
354          block_base = block_pos;
355          s2p_do_block((BytePack *) &srcbuf[block_pos], basis_bits);
356          @block_stmts
357          postprocess_do_block(lex, ctCDPI_Callouts, ref_Callouts, check_streams, chars_avail);
358          block_pos += BLOCK_SIZE;
359          remaining -= BLOCK_SIZE;
360    }
361    block_base = block_pos;
362    if (remaining > 0 || @any_carry) {
363          EOF_mask = sisd_srl(simd_const_1(1),sisd_from_int(BLOCK_SIZE-remaining));
364          s2p_do_final_block((BytePack *) &srcbuf[block_pos], basis_bits, EOF_mask);
365          @final_block_stmts
366          postprocess_do_block(lex, ctCDPI_Callouts, ref_Callouts, check_streams, chars_avail);
367    }
368    buf_pos += chars_avail;
369    buffer_base = buf_pos;
370
371    matcher.StreamScan(chars_avail);
372    matcher.Advance_buffer();
373    tracker.Advance_buffer();
374
375
376    PERF_SEC_END(parser_timer, chars_avail);
377    if (matcher.depth != 0) {
378      fprintf(stderr, "tag matching error (depth %i) at position %i\n", matcher.depth, buffer_base);
379      exit(-1);
380    }
381}
382
383
384
385int
386main(int argc, char * argv[]) {
387        char * infilename, * outfilename;
388        FILE *infile, *outfile;
389        struct stat fileinfo;
390
391        if (argc < 2) {
392                printf("Usage: %s <filename> [<outputfile>]\n", argv[0]);
393                exit(-1);
394        }
395
396        infilename = argv[1];
397        stat(infilename, &fileinfo);
398        infile = fopen(infilename, "rb");
399        if (!infile) {
400                fprintf(stderr, "Error: cannot open %s for input.\n", infilename);
401                exit(-1);
402        }
403
404        if (argc < 3) outfile = stdout;
405        else {
406                outfilename = argv[2];
407                outfile = fopen(outfilename, "wb");
408                if (!outfile) {
409                        fprintf(stderr, "Error: cannot open %s for writing.\n", outfilename);
410                        exit(-1);
411                }
412        }
413
414//      PERF_SEC_BIND(1);
415
416        PERF_SEC_INIT(parser_timer);
417
418        do_process(infile, outfile);
419
420        PERF_SEC_DUMP(parser_timer);
421
422        PERF_SEC_DESTROY(parser_timer);
423
424        fclose(infile);
425        fclose(outfile);
426        return(0);
427}
Note: See TracBrowser for help on using the repository browser.