source: proto/parabix2/pablo_template_avx.cpp @ 1534

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

AVX template update

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