source: proto/SymbolTable/wcd_identity_template.cpp @ 1787

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

SymbolTable?: completed dictionary implementation and refactored templates

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