25f4d2429f4f59e0c0cf3b8662ad1001239cced3
[external/ragel.git] / rlcodegen / fflatcodegen.cpp
1 /*
2  *  Copyright 2004-2006 Adrian Thurston <thurston@cs.queensu.ca>
3  *            2004 Eric Ocean <eric.ocean@ampede.com>
4  *            2005 Alan West <alan@alanz.com>
5  */
6
7 /*  This file is part of Ragel.
8  *
9  *  Ragel is free software; you can redistribute it and/or modify
10  *  it under the terms of the GNU General Public License as published by
11  *  the Free Software Foundation; either version 2 of the License, or
12  *  (at your option) any later version.
13  * 
14  *  Ragel is distributed in the hope that it will be useful,
15  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
16  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
17  *  GNU General Public License for more details.
18  * 
19  *  You should have received a copy of the GNU General Public License
20  *  along with Ragel; if not, write to the Free Software
21  *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA 
22  */
23
24 #include "rlcodegen.h"
25 #include "fflatcodegen.h"
26 #include "redfsm.h"
27 #include "gendata.h"
28
29 std::ostream &FFlatCodeGen::TO_STATE_ACTION( RedStateAp *state )
30 {
31         int act = 0;
32         if ( state->toStateAction != 0 )
33                 act = state->toStateAction->actListId+1;
34         out << act;
35         return out;
36 }
37
38 std::ostream &FFlatCodeGen::FROM_STATE_ACTION( RedStateAp *state )
39 {
40         int act = 0;
41         if ( state->fromStateAction != 0 )
42                 act = state->fromStateAction->actListId+1;
43         out << act;
44         return out;
45 }
46
47 std::ostream &FFlatCodeGen::EOF_ACTION( RedStateAp *state )
48 {
49         int act = 0;
50         if ( state->eofAction != 0 )
51                 act = state->eofAction->actListId+1;
52         out << act;
53         return out;
54 }
55
56 /* Write out the function for a transition. */
57 std::ostream &FFlatCodeGen::TRANS_ACTION( RedTransAp *trans )
58 {
59         int action = 0;
60         if ( trans->action != 0 )
61                 action = trans->action->actListId+1;
62         out << action;
63         return out;
64 }
65
66 /* Write out the function switch. This switch is keyed on the values
67  * of the func index. */
68 std::ostream &FFlatCodeGen::TO_STATE_ACTION_SWITCH()
69 {
70         /* Loop the actions. */
71         for ( ActionTableMap::Iter redAct = redFsm->actionMap; redAct.lte(); redAct++ ) {
72                 if ( redAct->numToStateRefs > 0 ) {
73                         /* Write the entry label. */
74                         out << "\tcase " << redAct->actListId+1 << ":\n";
75
76                         /* Write each action in the list of action items. */
77                         for ( ActionTable::Iter item = redAct->key; item.lte(); item++ )
78                                 ACTION( out, item->value, 0, false );
79
80                         out << "\tbreak;\n";
81                 }
82         }
83
84         genLineDirective( out );
85         return out;
86 }
87
88 /* Write out the function switch. This switch is keyed on the values
89  * of the func index. */
90 std::ostream &FFlatCodeGen::FROM_STATE_ACTION_SWITCH()
91 {
92         /* Loop the actions. */
93         for ( ActionTableMap::Iter redAct = redFsm->actionMap; redAct.lte(); redAct++ ) {
94                 if ( redAct->numFromStateRefs > 0 ) {
95                         /* Write the entry label. */
96                         out << "\tcase " << redAct->actListId+1 << ":\n";
97
98                         /* Write each action in the list of action items. */
99                         for ( ActionTable::Iter item = redAct->key; item.lte(); item++ )
100                                 ACTION( out, item->value, 0, false );
101
102                         out << "\tbreak;\n";
103                 }
104         }
105
106         genLineDirective( out );
107         return out;
108 }
109
110 std::ostream &FFlatCodeGen::EOF_ACTION_SWITCH()
111 {
112         /* Loop the actions. */
113         for ( ActionTableMap::Iter redAct = redFsm->actionMap; redAct.lte(); redAct++ ) {
114                 if ( redAct->numEofRefs > 0 ) {
115                         /* Write the entry label. */
116                         out << "\tcase " << redAct->actListId+1 << ":\n";
117
118                         /* Write each action in the list of action items. */
119                         for ( ActionTable::Iter item = redAct->key; item.lte(); item++ )
120                                 ACTION( out, item->value, 0, true );
121
122                         out << "\tbreak;\n";
123                 }
124         }
125
126         genLineDirective( out );
127         return out;
128 }
129
130 /* Write out the function switch. This switch is keyed on the values
131  * of the func index. */
132 std::ostream &FFlatCodeGen::ACTION_SWITCH()
133 {
134         /* Loop the actions. */
135         for ( ActionTableMap::Iter redAct = redFsm->actionMap; redAct.lte(); redAct++ ) {
136                 if ( redAct->numTransRefs > 0 ) {
137                         /* Write the entry label. */
138                         out << "\tcase " << redAct->actListId+1 << ":\n";
139
140                         /* Write each action in the list of action items. */
141                         for ( ActionTable::Iter item = redAct->key; item.lte(); item++ )
142                                 ACTION( out, item->value, 0, false );
143
144                         out << "\tbreak;\n";
145                 }
146         }
147
148         genLineDirective( out );
149         return out;
150 }
151
152 void FFlatCodeGen::writeOutData()
153 {
154         if ( anyConditions() ) {
155                 OPEN_ARRAY( WIDE_ALPH_TYPE(), CK() );
156                 COND_KEYS();
157                 CLOSE_ARRAY() <<
158                 "\n";
159
160                 OPEN_ARRAY( ARRAY_TYPE(maxCondSpan), CSP() );
161                 COND_KEY_SPANS();
162                 CLOSE_ARRAY() <<
163                 "\n";
164
165                 OPEN_ARRAY( ARRAY_TYPE(maxCond), C() );
166                 CONDS();
167                 CLOSE_ARRAY() <<
168                 "\n";
169
170                 OPEN_ARRAY( ARRAY_TYPE(maxCondIndexOffset), CO() );
171                 COND_INDEX_OFFSET();
172                 CLOSE_ARRAY() <<
173                 "\n";
174         }
175
176         OPEN_ARRAY( WIDE_ALPH_TYPE(), K() );
177         KEYS();
178         CLOSE_ARRAY() <<
179         "\n";
180
181         OPEN_ARRAY( ARRAY_TYPE(maxSpan), SP() );
182         KEY_SPANS();
183         CLOSE_ARRAY() <<
184         "\n";
185
186         OPEN_ARRAY( ARRAY_TYPE(maxFlatIndexOffset), IO() );
187         FLAT_INDEX_OFFSET();
188         CLOSE_ARRAY() <<
189         "\n";
190
191         OPEN_ARRAY( ARRAY_TYPE(maxIndex), I() );
192         INDICIES();
193         CLOSE_ARRAY() <<
194         "\n";
195
196         OPEN_ARRAY( ARRAY_TYPE(maxState), TT() );
197         TRANS_TARGS();
198         CLOSE_ARRAY() <<
199         "\n";
200
201         if ( anyActions() ) {
202                 OPEN_ARRAY( ARRAY_TYPE(maxActListId), TA() );
203                 TRANS_ACTIONS();
204                 CLOSE_ARRAY() <<
205                 "\n";
206         }
207
208         if ( anyToStateActions() ) {
209                 OPEN_ARRAY( ARRAY_TYPE(maxActionLoc),  TSA() );
210                 TO_STATE_ACTIONS();
211                 CLOSE_ARRAY() <<
212                 "\n";
213         }
214
215         if ( anyFromStateActions() ) {
216                 OPEN_ARRAY( ARRAY_TYPE(maxActionLoc), FSA() );
217                 FROM_STATE_ACTIONS();
218                 CLOSE_ARRAY() <<
219                 "\n";
220         }
221
222         if ( anyEofActions() ) {
223                 OPEN_ARRAY( ARRAY_TYPE(maxActListId), EA() );
224                 EOF_ACTIONS();
225                 CLOSE_ARRAY() <<
226                 "\n";
227         }
228
229         out <<
230                 "static const int " << START() << " = " << START_STATE_ID() << ";\n"
231                 "\n";
232
233         if ( cgd->writeFirstFinal ) {
234                 out <<
235                         "static const int " << FIRST_FINAL() << " = " << FIRST_FINAL_STATE() << ";\n"
236                         "\n";
237         }
238
239         if ( cgd->writeErr ) {
240                 out <<
241                         "static const int " << ERROR() << " = " << ERROR_STATE() << ";\n"
242                         "\n";
243         }
244 }
245
246 void FFlatCodeGen::writeOutExec()
247 {
248         outLabelUsed = false;
249
250         out << 
251                 "       {\n"
252                 "       int _slen";
253
254         if ( anyRegCurStateRef() )
255                 out << ", _ps";
256         
257         out << ";\n";
258         out << "        int _trans";
259
260         if ( anyConditions() )
261                 out << ", _cond";
262
263         out << ";\n";
264
265         out <<
266                 "       " << PTR_CONST() << WIDE_ALPH_TYPE() << POINTER() << "_keys;\n"
267                 "       " << PTR_CONST() << ARRAY_TYPE(maxIndex) << POINTER() << "_inds;\n";
268
269         if ( anyConditions() ) {
270                 out << 
271                         "       " << PTR_CONST() << ARRAY_TYPE(maxCond) << POINTER() << "_conds;\n"
272                         "       " << WIDE_ALPH_TYPE() << " _widec;\n";
273         }
274
275         if ( cgd->hasEnd ) {
276                 outLabelUsed = true;
277                 out << 
278                         "       if ( " << P() << " == " << PE() << " )\n"
279                         "               goto _out;\n";
280         }
281
282         out << "_resume:\n";
283
284         if ( redFsm->errState != 0 ) {
285                 outLabelUsed = true;
286                 out << 
287                         "       if ( " << CS() << " == " << redFsm->errState->id << " )\n"
288                         "               goto _out;\n";
289         }
290
291         if ( anyFromStateActions() ) {
292                 out <<
293                         "       switch ( " << FSA() << "[" << CS() << "] ) {\n";
294                         FROM_STATE_ACTION_SWITCH();
295                         SWITCH_DEFAULT() <<
296                         "       }\n"
297                         "\n";
298         }
299
300         if ( anyConditions() )
301                 COND_TRANSLATE();
302
303         LOCATE_TRANS();
304         
305         if ( anyRegCurStateRef() )
306                 out << "        _ps = " << CS() << ";\n";
307
308         out << 
309                 "       " << CS() << " = " << TT() << "[_trans];\n\n";
310
311         if ( anyRegActions() ) {
312                 out << 
313                         "       if ( " << TA() << "[_trans] == 0 )\n"
314                         "               goto _again;\n"
315                         "\n"
316                         "       switch ( " << TA() << "[_trans] ) {\n";
317                         ACTION_SWITCH();
318                         SWITCH_DEFAULT() <<
319                         "       }\n"
320                         "\n";
321         }
322
323         if ( anyRegActions() || anyActionGotos() || anyActionCalls() || anyActionRets() )
324                 out << "_again:\n";
325
326         if ( anyToStateActions() ) {
327                 out <<
328                         "       switch ( " << TSA() << "[" << CS() << "] ) {\n";
329                         TO_STATE_ACTION_SWITCH();
330                         SWITCH_DEFAULT() <<
331                         "       }\n"
332                         "\n";
333         }
334
335         if ( cgd->hasEnd ) {
336                 out << 
337                         "       if ( ++" << P() << " != " << PE() << " )\n"
338                         "               goto _resume;\n";
339         }
340         else {
341                 out << 
342                         "       " << P() << " += 1;\n"
343                         "       goto _resume;\n";
344         }
345
346         if ( outLabelUsed )
347                 out << "        _out: {}\n";
348
349         out << "        }\n";
350 }
351
352 void FFlatCodeGen::writeOutEOF()
353 {
354         if ( anyEofActions() ) {
355                 out <<
356                         "       {\n"
357                         "       switch ( " << EA() << "[" << CS() << "] ) {\n";
358                         EOF_ACTION_SWITCH();
359                         SWITCH_DEFAULT() <<
360                         "       }\n"
361                         "       }\n"
362                         "\n";
363         }
364 }