source: proto/SymbolTable/wcd_identity_template.cpp @ 1791

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

Added some macro for easy PAPI performance testing

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