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