source: proto/SymbolTable/wcd_identity_template.cpp @ 1688

Last change on this file since 1688 was 1688, checked in by vla24, 8 years ago

SymbolTable?: renaming dictionary prefixed files into wcd

File size: 10.0 KB
Line 
1//#define USE_ITER
2
3#include "../symtab_global.h"
4#include <pbgs_identity_symbol_table.h>
5
6#ifdef BUFFER_PROFILING
7        BOM_Table * parser_timer;
8
9#elif CODE_CLOCKER
10        #define NUM_EVENTS 1
11        int Events[NUM_EVENTS] = {PAPI_TOT_CYC};
12        //int Events[NUM_EVENTS] = {PAPI_L2_DCM};
13        //int Events[NUM_EVENTS] = {PAPI_TOT_CYC, PAPI_BR_MSP};
14        int cal_size = 20;
15        CC * parser_timer = new CC(Events,NUM_EVENTS,cal_size);
16#else
17        void * parser_timer;
18#endif
19
20int block_base=0;
21int buffer_base=0;
22int buffer_last;
23char * source;
24BitBlock EOF_mask = simd<1>::constant<1>();
25
26BitBlock elem_ends;
27int last_elem_start;
28bool block_boundary_case = false;
29BytePack hashvalues[2];
30
31vector <int> gids;
32PBGSIdentitySymbolTable pbgs_symbol_table;
33
34@global
35
36static inline void s2p_do_block(BytePack U8[], Basis_bits & basis_bits);
37static inline void s2p_do_final_block(BytePack U8[], Basis_bits & basis_bits, BitBlock EOF_mask);
38static inline void postprocess_do_block(Dictionary& dictionary, Hash_data hash_data);
39
40void do_process(FILE *infile, FILE *outfile);
41
42static inline void validate_block_length_grouping(BitBlockForwardIterator & start, int block_base);
43
44static inline int ElemStart_grouping(int start_pos, int L) ;
45static inline int StreamScanLengthGrouping(ScanBlock * stream, int blk_count);
46static inline int ScanForwardPos(BitBlock * block, int pos);
47static inline int compute_hash_value (int lgth, int start);
48
49int main(int argc, char * argv[]) {
50        char * infilename, * outfilename;
51        FILE *infile, *outfile;
52        struct stat fileinfo;
53
54        if (argc < 2) {
55                printf("Usage: %s <filename> [<outputfile>]\n", argv[0]);
56                exit(-1);
57        }
58
59        infilename = argv[1];
60        stat(infilename, &fileinfo);
61        infile = fopen(infilename, "rb");
62        if (!infile) {
63                fprintf(stderr, "Error: cannot open %s for input.\n", infilename);
64                exit(-1);
65        }
66
67        if (argc < 3) outfile = stdout;
68        else {
69                outfilename = argv[2];
70                outfile = fopen(outfilename, "wb");
71                if (!outfile) {
72                        fprintf(stderr, "Error: cannot open %s for writing.\n", outfilename);
73                        exit(-1);
74                }
75        }
76
77//      PERF_SEC_BIND(1);
78
79        PERF_SEC_INIT(parser_timer);
80
81        do_process(infile, outfile);
82
83        PERF_SEC_DUMP(parser_timer);
84
85        PERF_SEC_DESTROY(parser_timer);
86
87        fclose(infile);
88        fclose(outfile);
89
90        return(0);
91}
92
93/* s2p Definitions */
94static inline void s2p_do_block(BytePack U8[], Basis_bits & basis_bits) {
95  s2p(U8[0], U8[1], U8[2], U8[3], U8[4], U8[5], U8[6], U8[7],
96        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);
97}
98
99static inline void s2p_do_final_block(BytePack U8[], Basis_bits & basis_bits, BitBlock EOF_mask) {
100  s2p_do_block(U8, basis_bits);
101  basis_bits.bit_0 = simd_and(basis_bits.bit_0, EOF_mask);
102  basis_bits.bit_1 = simd_and(basis_bits.bit_1, EOF_mask);
103  basis_bits.bit_2 = simd_and(basis_bits.bit_2, EOF_mask);
104  basis_bits.bit_3 = simd_and(basis_bits.bit_3, EOF_mask);
105  basis_bits.bit_4 = simd_and(basis_bits.bit_4, EOF_mask);
106  basis_bits.bit_5 = simd_and(basis_bits.bit_5, EOF_mask);
107  basis_bits.bit_6 = simd_and(basis_bits.bit_6, EOF_mask);
108  basis_bits.bit_7 = simd_and(basis_bits.bit_7, EOF_mask);
109}
110
111static inline int ScanForwardPos(BitBlock * block, int pos)
112{
113    BitBlock s = block[0];
114    BitBlock temp = simd_and(s, simd<128>::sll(simd<2>::constant<3>(), convert(pos)));
115
116    if (bitblock_has_bit(temp))
117    {
118        return count_forward_zeroes (temp);
119    }
120    else
121    {
122        //handle boundary case
123        block_boundary_case = true;
124        last_elem_start = pos - BLOCK_SIZE;
125        return 0;
126    }
127}
128
129static inline int compute_hash_value (int lgth, int start)
130{
131    unsigned int offset_bit = start + 128;
132    uint64_t stream = *((uint64_t*)(((uint32_t*)hashvalues)+(offset_bit>>5)));
133    return stream >> (offset_bit & 0x1F) & ~(~0 << lgth);
134}
135
136static inline int ElemStart_grouping(int start_pos, int L) {
137    int hashvalue = compute_hash_value(L, start_pos - block_base);
138    int gid = 0;
139
140    switch (L)
141    {
142    case 1:
143        gid = pbgs_symbol_table.Lookup_or_Insert_Name<1>(source + start_pos, hashvalue);
144        break;
145    case 2:
146        gid = pbgs_symbol_table.Lookup_or_Insert_Name<2>(source + start_pos, hashvalue);
147        break;
148    case 3:
149        gid = pbgs_symbol_table.Lookup_or_Insert_Name<3>(source + start_pos, hashvalue);
150        break;
151    case 4:
152        gid = pbgs_symbol_table.Lookup_or_Insert_Name<4>(source + start_pos, hashvalue);
153        break;
154    case 5:
155        gid = pbgs_symbol_table.Lookup_or_Insert_Name<5>(source + start_pos, hashvalue);
156        break;
157    case 6:
158        gid = pbgs_symbol_table.Lookup_or_Insert_Name<6>(source + start_pos, hashvalue);
159        break;
160    case 7:
161        gid = pbgs_symbol_table.Lookup_or_Insert_Name<7>(source + start_pos, hashvalue);
162        break;
163    case 8:
164        gid = pbgs_symbol_table.Lookup_or_Insert_Name<8>(source + start_pos, hashvalue);
165        break;
166    case 9:
167        gid = pbgs_symbol_table.Lookup_or_Insert_Name<9>(source + start_pos, hashvalue);
168        break;
169    case 10:
170        gid = pbgs_symbol_table.Lookup_or_Insert_Name<10>(source + start_pos, hashvalue);
171        break;
172    case 11:
173        gid = pbgs_symbol_table.Lookup_or_Insert_Name<11>(source + start_pos, hashvalue);
174        break;
175    case 12:
176        gid = pbgs_symbol_table.Lookup_or_Insert_Name<12>(source + start_pos, hashvalue);
177        break;
178    case 13:
179        gid = pbgs_symbol_table.Lookup_or_Insert_Name<13>(source + start_pos, hashvalue);
180        break;
181    case 14:
182        gid = pbgs_symbol_table.Lookup_or_Insert_Name<14>(source + start_pos, hashvalue);
183        break;
184    case 15:
185        gid = pbgs_symbol_table.Lookup_or_Insert_Name<15>(source + start_pos, hashvalue);
186        break;
187    case 16:
188        gid = pbgs_symbol_table.Lookup_or_Insert_Name<16>(source + start_pos, hashvalue);
189        break;
190    default:
191        gid = pbgs_symbol_table.Lookup_or_Insert_Name(source + start_pos, hashvalue, L);
192        break;
193    }
194
195    gids.push_back(gid);
196#if DEBUG
197    int end = start_pos + L;
198    char* symbol = new char[L+1];
199    strncpy ( symbol, source + start_pos, L );
200    symbol[L] ='\0';
201    printf ("%s | L: %i | start: %i[%i] | end: %i[%i] | gid: %i | hashvalue: %i | symbol: %s\n", __FUNCTION__, L, start_pos, start_pos-buffer_base, end, end-buffer_base, gid, hashvalue, symbol );
202    delete symbol; symbol = 0;
203#endif
204    return 0;
205}
206
207inline void validate_block_length_grouping(BitBlockForwardIterator & start, int block_base) {
208
209    BitBlockForwardIterator end;
210    int start_pos, end_pos, lgth;
211
212    while(start != end) {
213        start_pos = *start;
214        end_pos = ScanForwardPos (&elem_ends, start_pos);
215        if (end_pos)
216        {
217            lgth = end_pos - start_pos;
218            start_pos += block_base;
219            ElemStart_grouping(start_pos, lgth);
220        }
221        start++;
222    }
223}
224
225static inline void postprocess_do_block(Dictionary& dictionary, Hash_data hash_data){
226
227    elem_ends = dictionary.Word_ends;
228    hashvalues[1] = hash_data.Hash_value;
229
230    if (block_boundary_case)
231    {
232        int end_pos = count_forward_zeroes(elem_ends);
233        ElemStart_grouping (last_elem_start + block_base, end_pos-last_elem_start);
234        block_boundary_case = false;
235    }
236
237    if ( bitblock_has_bit(dictionary.Word_starts) )
238    {
239        BitBlockForwardIterator iter_length_grouping(&dictionary.Word_starts);
240        validate_block_length_grouping(iter_length_grouping, block_base);
241    }
242
243    //copy current hash value data as previous one.
244    memmove (&hashvalues[0], &hashvalues[1], 16);
245}
246
247static inline void print_GIDS()
248{
249    int span_count = gids.size();
250    for(int i=0;i<span_count;i++) {
251             cout << gids[i] << " ";
252    }
253    cout << endl;
254}
255
256void do_process(FILE *infile, FILE *outfile) {
257
258@decl
259
260  int buf_pos = 0;
261  int block_pos = 0;
262  int errpos = 0;
263  int chars_avail = 0;
264  int check_pos = 0;
265  int chars_read = 0;
266  BytePack buf[(BUFFER_SIZE+BLOCK_SIZE+OVERLAP_BUFSIZE*2)/sizeof(SIMD_type)];
267
268  char * srcbuf = ((char *) buf) + OVERLAP_BUFSIZE;
269  buffer_base = buf_pos;
270  source = srcbuf;
271
272  chars_read = fread((void *)srcbuf, 1, BUFFER_SIZE + OVERLAP_BUFSIZE, infile);
273  chars_avail = chars_read;
274  if (chars_avail > BUFFER_SIZE) chars_avail = BUFFER_SIZE;
275
276  Entity_Info * e = new Entity_Info;
277  e->AnalyzeSignature((unsigned char *)srcbuf);
278
279  if (e->content_start != 0) {
280        memmove(&srcbuf[0], &srcbuf[e->content_start], chars_read - e->content_start);
281        buf_pos = e->content_start;
282        buffer_base = buf_pos;
283        if (chars_avail == BUFFER_SIZE) {
284                chars_read = chars_read - e->content_start +
285                             fread(&srcbuf[chars_read-e->content_start], 1, e->content_start, infile);
286                chars_avail = chars_read;
287                if (chars_avail > BUFFER_SIZE) chars_avail = BUFFER_SIZE;
288        }
289        else {
290          chars_read -=e->content_start;
291          chars_avail -=e->content_start;
292        }
293  }
294
295@stream_stmts
296
297/* Full Buffers */
298
299    while (chars_avail == BUFFER_SIZE) {
300      PERF_SEC_START(parser_timer);
301      for (int blk = 0; blk < SEGMENT_BLOCKS; blk++) {
302          block_base = blk*BLOCK_SIZE;
303          s2p_do_block((BytePack *) &srcbuf[block_base], basis_bits);
304          @block_stmts
305          postprocess_do_block(dictionary, hash_data);
306      }
307      PERF_SEC_END(parser_timer, chars_avail);
308           
309      int bytes_left = chars_read - chars_avail;
310      memmove(buf, &srcbuf[BUFFER_SIZE - OVERLAP_BUFSIZE], bytes_left + OVERLAP_BUFSIZE);
311      chars_read = fread(&srcbuf[bytes_left],1, BUFFER_SIZE + OVERLAP_BUFSIZE - bytes_left, infile) + bytes_left;
312      chars_avail = chars_read;
313      if (chars_avail > BUFFER_SIZE) chars_avail = BUFFER_SIZE;
314      buf_pos += chars_avail;
315      buffer_base = buf_pos;
316    }
317/* Final Partial Buffer */
318    PERF_SEC_START(parser_timer);
319
320    block_pos = 0;
321    int remaining = chars_avail;
322/* Full Blocks */
323    while (remaining >= BLOCK_SIZE) {
324          block_base = block_pos;
325          s2p_do_block((BytePack *) &srcbuf[block_pos], basis_bits);
326          @block_stmts
327          postprocess_do_block(dictionary, hash_data);
328          block_pos += BLOCK_SIZE;
329          remaining -= BLOCK_SIZE;
330    }
331    block_base = block_pos;
332    if (remaining > 0 || @any_carry) {
333          EOF_mask = bitblock::srl(simd<1>::constant<1>(), convert(BLOCK_SIZE-remaining));
334          s2p_do_final_block((BytePack *) &srcbuf[block_pos], basis_bits, EOF_mask);
335          @final_block_stmts
336          postprocess_do_block(dictionary, hash_data);
337    }
338    buf_pos += chars_avail;
339    buffer_base = buf_pos;
340
341    PERF_SEC_END(parser_timer, chars_avail);
342
343#if DEBUG
344//    print_GIDS();
345    pbgs_symbol_table.Print_Symbol_Table_Distribution();
346#endif
347}
Note: See TracBrowser for help on using the repository browser.