source: proto/SymbolTable/wcd_stl_template.cpp @ 1721

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

SymbolTable?: completed dictionary implementation and refactored templates

File size: 6.3 KB
Line 
1#include "../common_definitions.h"
2#include <symtab.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;
24char * source;
25
26queue <size_t> elem_starts_buf;
27queue <size_t> elem_ends_buf;
28vector <int> gids;
29SymbolTable symbol_table;
30
31template<bool allow_performance_check> void do_process(FILE *infile, FILE *outfile);
32static inline void postprocess_do_block(Dictionary& dictionary);
33static inline void do_symbol_table_lookup();
34
35int main(int argc, char * argv[]) {
36    char * dictionaryfilename, * infilename, * outfilename;
37    FILE * dictionaryfile, *infile, *outfile;
38
39    getFilenames(argc, argv, dictionaryfilename, infilename, outfilename);
40    openInputOutputFiles(dictionaryfilename, infilename, outfilename,
41                         dictionaryfile, infile, outfile);
42
43    int greatest_GID_in_dictionary;
44    populateDictionary(dictionaryfile, outfile, gids, greatest_GID_in_dictionary);
45
46//      PERF_SEC_BIND(1);
47
48    PERF_SEC_INIT(parser_timer);
49
50    // store symbols form text to Symbol Table
51    do_process<true>(infile, outfile);
52
53    PERF_SEC_DUMP(parser_timer);
54
55    PERF_SEC_DESTROY(parser_timer);
56
57    // gather dictionary statistics
58    int totalKnownWordsInDictionary, totalUnknownWordsInDictionary;
59    wordCountInDictionary(greatest_GID_in_dictionary, gids, totalUnknownWordsInDictionary, totalKnownWordsInDictionary);
60    printWordCountInDictionary(totalUnknownWordsInDictionary, totalKnownWordsInDictionary);
61
62    fclose(dictionaryfile);
63    fclose(infile);
64    fclose(outfile);
65
66#if PRINT_SYMBOL_DISTRIBUTION
67    print_GIDS();
68#endif
69
70    return(0);
71}
72
73static inline int ElemStrt_check(int pos) {
74        elem_starts_buf.push(buffer_base + pos);
75        return 0;
76}
77
78static inline int ElemEnd_check(int pos) {
79        elem_ends_buf.push(buffer_base + pos);
80        return 0;
81}
82
83static inline void do_symbol_table_lookup()
84{
85    while( !elem_starts_buf.empty() && !elem_ends_buf.empty() )
86    {
87        int start = elem_starts_buf.front();
88        int end = elem_ends_buf.front();
89        elem_starts_buf.pop();
90        elem_ends_buf.pop();
91        int length = end - start;
92
93        //lookup or insert to symbol table
94#if DEBUG
95        char* symbol = new char[length+1];
96        strncpy ( symbol, source + start - buffer_base, length );
97        symbol[length] ='\0';
98        printf ("start: %i[%i] | end: %i[%i] | length: %i | symbol: %s\n", start, start-buffer_base, end, end-buffer_base, length, symbol );
99
100        delete symbol; symbol = 0;
101#endif
102
103        int gid = symbol_table.Lookup_or_Insert_Name(source + start - buffer_base, length);
104        gids.push_back(gid);
105    }
106}
107
108static inline void postprocess_do_block(Dictionary& dictionary){
109
110    if ( bitblock_has_bit(dictionary.Word_starts))
111    {
112        BitBlockForwardIterator iter_length_grouping_starts(&dictionary.Word_starts);
113        validate_block(iter_length_grouping_starts, block_base, ElemStrt_check);
114    }
115
116    if ( bitblock_has_bit(dictionary.Word_ends) )
117    {
118        BitBlockForwardIterator iter_length_grouping_ends(&dictionary.Word_ends);
119        validate_block(iter_length_grouping_ends, block_base, ElemEnd_check);
120    }
121
122    do_symbol_table_lookup();
123}
124
125template<bool allow_performance_check> void do_process(FILE *infile, FILE *outfile) {
126
127@decl
128
129  int buf_pos = 0;
130  int block_pos = 0;
131  int errpos = 0;
132  int chars_avail = 0;
133  int check_pos = 0;
134  int chars_read = 0;
135  BytePack buf[(BUFFER_SIZE+BLOCK_SIZE+OVERLAP_BUFSIZE*2)/sizeof(SIMD_type)];
136
137  char * srcbuf = ((char *) buf) + OVERLAP_BUFSIZE;
138  buffer_base = buf_pos;
139  source = srcbuf;
140
141  chars_read = fread((void *)srcbuf, 1, BUFFER_SIZE + OVERLAP_BUFSIZE, infile);
142  chars_avail = chars_read;
143  if (chars_avail > BUFFER_SIZE) chars_avail = BUFFER_SIZE;
144
145  Entity_Info * e = new Entity_Info;
146  e->AnalyzeSignature((unsigned char *)srcbuf);
147
148  if (e->content_start != 0) {
149        memmove(&srcbuf[0], &srcbuf[e->content_start], chars_read - e->content_start);
150        buf_pos = e->content_start;
151        buffer_base = buf_pos;
152        if (chars_avail == BUFFER_SIZE) {
153                chars_read = chars_read - e->content_start +
154                             fread(&srcbuf[chars_read-e->content_start], 1, e->content_start, infile);
155                chars_avail = chars_read;
156                if (chars_avail > BUFFER_SIZE) chars_avail = BUFFER_SIZE;
157        }
158        else {
159          chars_read -=e->content_start;
160          chars_avail -=e->content_start;
161        }
162  }
163
164@stream_stmts
165
166/* Full Buffers */
167
168    while (chars_avail == BUFFER_SIZE) {
169      if (allow_performance_check)
170      {
171        PERF_SEC_START(parser_timer);
172      }
173      for (int blk = 0; blk < SEGMENT_BLOCKS; blk++) {
174          block_base = blk*BLOCK_SIZE;
175          s2p_do_block((BytePack *) &srcbuf[block_base], basis_bits);
176          @block_stmts
177          postprocess_do_block(dictionary);
178      }
179      if (allow_performance_check)
180      {
181        PERF_SEC_END(parser_timer, chars_avail);
182      }
183
184      int bytes_left = chars_read - chars_avail;
185      memmove(buf, &srcbuf[BUFFER_SIZE - OVERLAP_BUFSIZE], bytes_left + OVERLAP_BUFSIZE);
186      chars_read = fread(&srcbuf[bytes_left],1, BUFFER_SIZE + OVERLAP_BUFSIZE - bytes_left, infile) + bytes_left;
187      chars_avail = chars_read;
188      if (chars_avail > BUFFER_SIZE) chars_avail = BUFFER_SIZE;
189      buf_pos += chars_avail;
190      buffer_base = buf_pos;
191    }
192/* Final Partial Buffer */
193    if (allow_performance_check)
194    {
195      PERF_SEC_START(parser_timer);
196    }
197
198    block_pos = 0;
199    int remaining = chars_avail;
200/* Full Blocks */
201    while (remaining >= BLOCK_SIZE) {
202          block_base = block_pos;
203          s2p_do_block((BytePack *) &srcbuf[block_pos], basis_bits);
204          @block_stmts
205          postprocess_do_block(dictionary);
206          block_pos += BLOCK_SIZE;
207          remaining -= BLOCK_SIZE;
208    }
209    block_base = block_pos;
210    if (remaining > 0 || @any_carry) {
211          EOF_mask = bitblock::srl(simd<1>::constant<1>(), convert(BLOCK_SIZE-remaining));
212          s2p_do_final_block((BytePack *) &srcbuf[block_pos], basis_bits, EOF_mask);
213          @final_block_stmts
214          postprocess_do_block(dictionary);
215    }
216    buf_pos += chars_avail;
217    buffer_base = buf_pos;
218
219    if (allow_performance_check)
220    {
221      PERF_SEC_END(parser_timer, chars_avail);
222    }
223}
Note: See TracBrowser for help on using the repository browser.