source: proto/parabix2/pablo_template_avx.cpp @ 4355

Last change on this file since 4355 was 2547, checked in by ksherdy, 7 years ago

Updated templates for new perfsec.h

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