source: proto/parabix2/pablo_template.cpp @ 911

Last change on this file since 911 was 911, checked in by lindanl, 9 years ago

Change post_process to check_streams in parabix2_pablo.py. Add postprocess_do_block function. Change tag_macher to TagMatcher? and make the matcher global.

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