source: src/CodeGen/CodeGenerator.cc @ 396fd72

ADTaaron-thesisarm-ehast-experimentalcleanup-dtorsdeferred_resndemanglerenumforall-pointer-decayjacob/cs343-translationjenkins-sandboxnew-astnew-ast-unique-exprnew-envno_listpersistent-indexerpthread-emulationqualifiedEnumresolv-newwith_gc
Last change on this file since 396fd72 was 55d6e8de, checked in by Rob Schluntz <rschlunt@…>, 7 years ago

Add CodeGen? for WithStmt?

  • Property mode set to 100644
File size: 33.8 KB
Line 
1//
2// Cforall Version 1.0.0 Copyright (C) 2015 University of Waterloo
3//
4// The contents of this file are covered under the licence agreement in the
5// file "LICENCE" distributed with Cforall.
6//
7// CodeGenerator.cc --
8//
9// Author           : Richard C. Bilson
10// Created On       : Mon May 18 07:44:20 2015
11// Last Modified By : Peter A. Buhr
12// Last Modified On : Sun Sep  3 20:42:52 2017
13// Update Count     : 490
14//
15#include "CodeGenerator.h"
16
17#include <cassert>                   // for assert, assertf
18#include <list>                      // for _List_iterator, list, list<>::it...
19
20#include "Common/SemanticError.h"    // for SemanticError
21#include "Common/UniqueName.h"       // for UniqueName
22#include "Common/utility.h"          // for CodeLocation, toString
23#include "GenType.h"                 // for genType
24#include "InitTweak/InitTweak.h"     // for getPointerBase
25#include "OperatorTable.h"           // for OperatorInfo, operatorLookup
26#include "Parser/LinkageSpec.h"      // for Spec, Intrinsic
27#include "SynTree/Attribute.h"       // for Attribute
28#include "SynTree/BaseSyntaxNode.h"  // for BaseSyntaxNode
29#include "SynTree/Constant.h"        // for Constant
30#include "SynTree/Declaration.h"     // for DeclarationWithType, TypeDecl
31#include "SynTree/Expression.h"      // for Expression, UntypedExpr, Applica...
32#include "SynTree/Initializer.h"     // for Initializer, ListInit, Designation
33#include "SynTree/Label.h"           // for Label, operator<<
34#include "SynTree/Statement.h"       // for Statement, AsmStmt, BranchStmt
35#include "SynTree/Type.h"            // for Type, Type::StorageClasses, Func...
36
37using namespace std;
38
39namespace CodeGen {
40        int CodeGenerator::tabsize = 4;
41
42        // the kinds of statements that would ideally be followed by whitespace
43        bool wantSpacing( Statement * stmt) {
44                return dynamic_cast< IfStmt * >( stmt ) || dynamic_cast< CompoundStmt * >( stmt ) ||
45                        dynamic_cast< WhileStmt * >( stmt ) || dynamic_cast< ForStmt * >( stmt ) || dynamic_cast< SwitchStmt *>( stmt );
46        }
47
48        void CodeGenerator::extension( Expression * expr ) {
49                if ( expr->get_extension() ) {
50                        output << "__extension__ ";
51                } // if
52        } // extension
53
54        void CodeGenerator::extension( Declaration * decl ) {
55                if ( decl->get_extension() ) {
56                        output << "__extension__ ";
57                } // if
58        } // extension
59
60        void CodeGenerator::asmName( DeclarationWithType * decl ) {
61                if ( ConstantExpr * asmName = dynamic_cast<ConstantExpr *>(decl->get_asmName()) ) {
62                        output << " asm ( " << asmName->get_constant()->get_value() << " )";
63                } // if
64        } // extension
65
66        CodeGenerator::LabelPrinter & CodeGenerator::LabelPrinter::operator()( std::list< Label > & l ) {
67                labels = &l;
68                return *this;
69        }
70
71        ostream & operator<<( ostream & output, CodeGenerator::LabelPrinter & printLabels ) {
72                std::list< Label > & labs = *printLabels.labels;
73                // l.unique(); // assumes a sorted list. Why not use set? Does order matter?
74                for ( Label & l : labs ) {
75                        output << l.get_name() + ": ";
76                        printLabels.cg.genAttributes( l.get_attributes() );
77                } // for
78                return output;
79        }
80
81        /* Using updateLocation at the beginning of a node and endl
82         * within a node should become the method of formating.
83         */
84        void CodeGenerator::updateLocation( CodeLocation const & to ) {
85                // skip if linemarks shouldn't appear or if codelocation is unset
86                if ( !lineMarks || to.isUnset() ) return;
87
88                if ( currentLocation.followedBy( to, 0 ) ) {
89                        return;
90                } else if ( currentLocation.followedBy( to, 1 ) ) {
91                        output << "\n" << indent;
92                        currentLocation.first_line += 1;
93                } else if ( currentLocation.followedBy( to, 2 ) ) {
94                        output << "\n\n" << indent;
95                        currentLocation.first_line += 2;
96                } else {
97                        output << "\n# " << to.first_line << " \"" << to.filename
98                               << "\"\n" << indent;
99                        currentLocation = to;
100                }
101                output << std::flush;
102        }
103
104        void CodeGenerator::updateLocation( BaseSyntaxNode const * to ) {
105                updateLocation( to->location );
106        }
107
108        // replace endl
109        ostream & CodeGenerator::LineEnder::operator()( ostream & os ) const {
110                // if ( !cg.lineMarks ) {
111                //      os << "\n" << cg.indent << std::flush;
112                // }
113                os << "\n" << std::flush;
114                cg.currentLocation.first_line++;
115                // os << "/* did endl; current loc is: " << cg.currentLocation.first_line << "*/";
116                return os;
117        }
118
119        CodeGenerator::CodeGenerator( std::ostream & os, bool pretty, bool genC, bool lineMarks ) : indent( CodeGenerator::tabsize ), output( os ), printLabels( *this ), pretty( pretty ), genC( genC ), lineMarks( lineMarks ), endl( *this ) {}
120
121        string CodeGenerator::mangleName( DeclarationWithType * decl ) {
122                if ( pretty ) return decl->get_name();
123                if ( decl->get_mangleName() != "" ) {
124                        // need to incorporate scope level in order to differentiate names for destructors
125                        return decl->get_scopedMangleName();
126                } else {
127                        return decl->get_name();
128                } // if
129        }
130
131        void CodeGenerator::genAttributes( list< Attribute * > & attributes ) {
132          if ( attributes.empty() ) return;
133                output << "__attribute__ ((";
134                for ( list< Attribute * >::iterator attr( attributes.begin() );; ) {
135                        output << (*attr)->get_name();
136                        if ( ! (*attr)->get_parameters().empty() ) {
137                                output << "(";
138                                genCommaList( (*attr)->get_parameters().begin(), (*attr)->get_parameters().end() );
139                                output << ")";
140                        } // if
141                  if ( ++attr == attributes.end() ) break;
142                        output << ",";                                                          // separator
143                } // for
144                output << ")) ";
145        } // CodeGenerator::genAttributes
146
147        // *** BaseSyntaxNode
148        void CodeGenerator::previsit( BaseSyntaxNode * node ) {
149                // turn off automatic recursion for all nodes, to allow each visitor to
150                // precisely control the order in which its children are visited.
151                visit_children = false;
152                updateLocation( node );
153        }
154
155        // *** BaseSyntaxNode
156        void CodeGenerator::postvisit( BaseSyntaxNode * node ) {
157                std::stringstream ss;
158                node->print( ss );
159                assertf( false, "Unhandled node reached in CodeGenerator: %s", ss.str().c_str() );
160        }
161
162        // *** Declarations
163        void CodeGenerator::postvisit( FunctionDecl * functionDecl ) {
164                extension( functionDecl );
165                genAttributes( functionDecl->get_attributes() );
166
167                handleStorageClass( functionDecl );
168                functionDecl->get_funcSpec().print( output );
169
170                output << genType( functionDecl->get_functionType(), mangleName( functionDecl ), pretty, genC );
171
172                asmName( functionDecl );
173
174                if ( functionDecl->get_statements() ) {
175                        functionDecl->get_statements()->accept( *visitor );
176                } // if
177        }
178
179        void CodeGenerator::postvisit( ObjectDecl * objectDecl ) {
180                if (objectDecl->get_name().empty() && genC ) {
181                        // only generate an anonymous name when generating C code, otherwise it clutters the output too much
182                        static UniqueName name = { "__anonymous_object" };
183                        objectDecl->set_name( name.newName() );
184                }
185
186                extension( objectDecl );
187                genAttributes( objectDecl->get_attributes() );
188
189                handleStorageClass( objectDecl );
190                output << genType( objectDecl->get_type(), mangleName( objectDecl ), pretty, genC );
191
192                asmName( objectDecl );
193
194                if ( objectDecl->get_init() ) {
195                        output << " = ";
196                        objectDecl->get_init()->accept( *visitor );
197                } // if
198
199                if ( objectDecl->get_bitfieldWidth() ) {
200                        output << ":";
201                        objectDecl->get_bitfieldWidth()->accept( *visitor );
202                } // if
203        }
204
205        void CodeGenerator::handleAggregate( AggregateDecl * aggDecl, const std::string & kind ) {
206                genAttributes( aggDecl->get_attributes() );
207
208                if( ! aggDecl->get_parameters().empty() && ! genC ) {
209                        // assertf( ! genC, "Aggregate type parameters should not reach code generation." );
210                        output << "forall(";
211                        genCommaList( aggDecl->get_parameters().begin(), aggDecl->get_parameters().end() );
212                        output << ")" << endl;
213                        output << indent;
214                }
215
216                output << kind << aggDecl->get_name();
217
218                if ( aggDecl->has_body() ) {
219                        std::list< Declaration * > & memb = aggDecl->get_members();
220                        output << " {" << endl;
221
222                        ++indent;
223                        for ( std::list< Declaration* >::iterator i = memb.begin(); i != memb.end(); i++ ) {
224                                output << indent;
225                                (*i)->accept( *visitor );
226                                output << ";" << endl;
227                        } // for
228
229                        --indent;
230
231                        output << indent << "}";
232                } // if
233        }
234
235        void CodeGenerator::postvisit( StructDecl * structDecl ) {
236                extension( structDecl );
237                handleAggregate( structDecl, "struct " );
238        }
239
240        void CodeGenerator::postvisit( UnionDecl * unionDecl ) {
241                extension( unionDecl );
242                handleAggregate( unionDecl, "union " );
243        }
244
245        void CodeGenerator::postvisit( EnumDecl * enumDecl ) {
246                extension( enumDecl );
247                output << "enum ";
248                genAttributes( enumDecl->get_attributes() );
249
250                output << enumDecl->get_name();
251
252                std::list< Declaration* > &memb = enumDecl->get_members();
253
254                if ( ! memb.empty() ) {
255                        output << " {" << endl;
256
257                        ++indent;
258                        for ( std::list< Declaration* >::iterator i = memb.begin(); i != memb.end();  i++) {
259                                ObjectDecl * obj = dynamic_cast< ObjectDecl* >( *i );
260                                assert( obj );
261                                output << indent << mangleName( obj );
262                                if ( obj->get_init() ) {
263                                        output << " = ";
264                                        obj->get_init()->accept( *visitor );
265                                } // if
266                                output << "," << endl;
267                        } // for
268
269                        --indent;
270
271                        output << indent << "}";
272                } // if
273        }
274
275        void CodeGenerator::postvisit( TraitDecl * traitDecl ) {
276                assertf( ! genC, "TraitDecls should not reach code generation." );
277                extension( traitDecl );
278                handleAggregate( traitDecl, "trait " );
279        }
280
281        void CodeGenerator::postvisit( TypedefDecl * typeDecl ) {
282                assertf( ! genC, "Typedefs are removed and substituted in earlier passes." );
283                output << "typedef ";
284                output << genType( typeDecl->get_base(), typeDecl->get_name(), pretty, genC ) << endl;
285        }
286
287        void CodeGenerator::postvisit( TypeDecl * typeDecl ) {
288                assertf( ! genC, "TypeDecls should not reach code generation." );
289                output << typeDecl->genTypeString() << " " << typeDecl->name;
290                if ( typeDecl->sized ) {
291                        output << " | sized(" << typeDecl->name << ")";
292                }
293                if ( ! typeDecl->assertions.empty() ) {
294                        output << " | { ";
295                        for ( DeclarationWithType * assert :  typeDecl->assertions ) {
296                                assert->accept( *visitor );
297                                output << "; ";
298                        }
299                        output << " }";
300                }
301        }
302
303        void CodeGenerator::postvisit( Designation * designation ) {
304                std::list< Expression * > designators = designation->get_designators();
305                if ( designators.size() == 0 ) return;
306                for ( Expression * des : designators ) {
307                        if ( dynamic_cast< NameExpr * >( des ) || dynamic_cast< VariableExpr * >( des ) ) {
308                                // if expression is a NameExpr or VariableExpr, then initializing aggregate member
309                                output << ".";
310                                des->accept( *visitor );
311                        } else {
312                                // otherwise, it has to be a ConstantExpr or CastExpr, initializing array eleemnt
313                                output << "[";
314                                des->accept( *visitor );
315                                output << "]";
316                        } // if
317                } // for
318                output << " = ";
319        }
320
321        void CodeGenerator::postvisit( SingleInit * init ) {
322                init->get_value()->accept( *visitor );
323        }
324
325        void CodeGenerator::postvisit( ListInit * init ) {
326                auto initBegin = init->begin();
327                auto initEnd = init->end();
328                auto desigBegin = init->get_designations().begin();
329                auto desigEnd = init->get_designations().end();
330
331                output << "{ ";
332                for ( ; initBegin != initEnd && desigBegin != desigEnd; ) {
333                        (*desigBegin)->accept( *visitor );
334                        (*initBegin)->accept( *visitor );
335                        ++initBegin, ++desigBegin;
336                        if ( initBegin != initEnd ) {
337                                output << ", ";
338                        }
339                }
340                output << " }";
341                assertf( initBegin == initEnd && desigBegin == desigEnd, "Initializers and designators not the same length. %s", toString( init ).c_str() );
342        }
343
344        void CodeGenerator::postvisit( ConstructorInit * init ){
345                assertf( ! genC, "ConstructorInit nodes should not reach code generation." );
346                // pseudo-output for constructor/destructor pairs
347                output << "<ctorinit>{" << endl << ++indent << "ctor: ";
348                maybeAccept( init->get_ctor(), *visitor );
349                output << ", " << endl << indent << "dtor: ";
350                maybeAccept( init->get_dtor(), *visitor );
351                output << endl << --indent << "}";
352        }
353
354        void CodeGenerator::postvisit( Constant * constant ) {
355                output << constant->get_value() ;
356        }
357
358        // *** Expressions
359        void CodeGenerator::postvisit( ApplicationExpr * applicationExpr ) {
360                extension( applicationExpr );
361                if ( VariableExpr * varExpr = dynamic_cast< VariableExpr* >( applicationExpr->get_function() ) ) {
362                        OperatorInfo opInfo;
363                        if ( varExpr->get_var()->get_linkage() == LinkageSpec::Intrinsic && operatorLookup( varExpr->get_var()->get_name(), opInfo ) ) {
364                                std::list< Expression* >::iterator arg = applicationExpr->get_args().begin();
365                                switch ( opInfo.type ) {
366                                  case OT_INDEX:
367                                        assert( applicationExpr->get_args().size() == 2 );
368                                        (*arg++)->accept( *visitor );
369                                        output << "[";
370                                        (*arg)->accept( *visitor );
371                                        output << "]";
372                                        break;
373
374                                  case OT_CALL:
375                                        // there are no intrinsic definitions of the function call operator
376                                        assert( false );
377                                        break;
378
379                                  case OT_CTOR:
380                                  case OT_DTOR:
381                                        if ( applicationExpr->get_args().size() == 1 ) {
382                                                // the expression fed into a single parameter constructor or destructor may contain side
383                                                // effects, so must still output this expression
384                                                output << "(";
385                                                (*arg++)->accept( *visitor );
386                                                output << ") /* " << opInfo.inputName << " */";
387                                        } else if ( applicationExpr->get_args().size() == 2 ) {
388                                                // intrinsic two parameter constructors are essentially bitwise assignment
389                                                output << "(";
390                                                (*arg++)->accept( *visitor );
391                                                output << opInfo.symbol;
392                                                (*arg)->accept( *visitor );
393                                                output << ") /* " << opInfo.inputName << " */";
394                                        } else {
395                                                // no constructors with 0 or more than 2 parameters
396                                                assert( false );
397                                        } // if
398                                        break;
399
400                                  case OT_PREFIX:
401                                  case OT_PREFIXASSIGN:
402                                        assert( applicationExpr->get_args().size() == 1 );
403                                        output << "(";
404                                        output << opInfo.symbol;
405                                        (*arg)->accept( *visitor );
406                                        output << ")";
407                                        break;
408
409                                  case OT_POSTFIX:
410                                  case OT_POSTFIXASSIGN:
411                                        assert( applicationExpr->get_args().size() == 1 );
412                                        (*arg)->accept( *visitor );
413                                        output << opInfo.symbol;
414                                        break;
415
416
417                                  case OT_INFIX:
418                                  case OT_INFIXASSIGN:
419                                        assert( applicationExpr->get_args().size() == 2 );
420                                        output << "(";
421                                        (*arg++)->accept( *visitor );
422                                        output << opInfo.symbol;
423                                        (*arg)->accept( *visitor );
424                                        output << ")";
425                                        break;
426
427                                  case OT_CONSTANT:
428                                  case OT_LABELADDRESS:
429                                        // there are no intrinsic definitions of 0/1 or label addresses as functions
430                                        assert( false );
431                                } // switch
432                        } else {
433                                varExpr->accept( *visitor );
434                                output << "(";
435                                genCommaList( applicationExpr->get_args().begin(), applicationExpr->get_args().end() );
436                                output << ")";
437                        } // if
438                } else {
439                        applicationExpr->get_function()->accept( *visitor );
440                        output << "(";
441                        genCommaList( applicationExpr->get_args().begin(), applicationExpr->get_args().end() );
442                        output << ")";
443                } // if
444        }
445
446        void CodeGenerator::postvisit( UntypedExpr * untypedExpr ) {
447                extension( untypedExpr );
448                if ( NameExpr * nameExpr = dynamic_cast< NameExpr* >( untypedExpr->function ) ) {
449                        OperatorInfo opInfo;
450                        if ( operatorLookup( nameExpr->name, opInfo ) ) {
451                                std::list< Expression* >::iterator arg = untypedExpr->args.begin();
452                                switch ( opInfo.type ) {
453                                  case OT_INDEX:
454                                        assert( untypedExpr->args.size() == 2 );
455                                        (*arg++)->accept( *visitor );
456                                        output << "[";
457                                        (*arg)->accept( *visitor );
458                                        output << "]";
459                                        break;
460
461                                  case OT_CALL:
462                                        assert( false );
463
464                                  case OT_CTOR:
465                                  case OT_DTOR:
466                                        if ( untypedExpr->args.size() == 1 ) {
467                                                // the expression fed into a single parameter constructor or destructor may contain side
468                                                // effects, so must still output this expression
469                                                output << "(";
470                                                (*arg++)->accept( *visitor );
471                                                output << ") /* " << opInfo.inputName << " */";
472                                        } else if ( untypedExpr->get_args().size() == 2 ) {
473                                                // intrinsic two parameter constructors are essentially bitwise assignment
474                                                output << "(";
475                                                (*arg++)->accept( *visitor );
476                                                output << opInfo.symbol;
477                                                (*arg)->accept( *visitor );
478                                                output << ") /* " << opInfo.inputName << " */";
479                                        } else {
480                                                // no constructors with 0 or more than 2 parameters
481                                                assertf( ! genC, "UntypedExpr constructor/destructor with 0 or more than 2 parameters." );
482                                                output << "(";
483                                                (*arg++)->accept( *visitor );
484                                                output << opInfo.symbol << "{ ";
485                                                genCommaList( arg, untypedExpr->args.end() );
486                                                output << "}) /* " << opInfo.inputName << " */";
487                                        } // if
488                                        break;
489
490                                  case OT_PREFIX:
491                                  case OT_PREFIXASSIGN:
492                                  case OT_LABELADDRESS:
493                                        assert( untypedExpr->args.size() == 1 );
494                                        output << "(";
495                                        output << opInfo.symbol;
496                                        (*arg)->accept( *visitor );
497                                        output << ")";
498                                        break;
499
500                                  case OT_POSTFIX:
501                                  case OT_POSTFIXASSIGN:
502                                        assert( untypedExpr->args.size() == 1 );
503                                        (*arg)->accept( *visitor );
504                                        output << opInfo.symbol;
505                                        break;
506
507                                  case OT_INFIX:
508                                  case OT_INFIXASSIGN:
509                                        assert( untypedExpr->args.size() == 2 );
510                                        output << "(";
511                                        (*arg++)->accept( *visitor );
512                                        output << opInfo.symbol;
513                                        (*arg)->accept( *visitor );
514                                        output << ")";
515                                        break;
516
517                                  case OT_CONSTANT:
518                                        // there are no intrinsic definitions of 0 or 1 as functions
519                                        assert( false );
520                                } // switch
521                        } else {
522                                // builtin routines
523                                nameExpr->accept( *visitor );
524                                output << "(";
525                                genCommaList( untypedExpr->args.begin(), untypedExpr->args.end() );
526                                output << ")";
527                        } // if
528                } else {
529                        untypedExpr->function->accept( *visitor );
530                        output << "(";
531                        genCommaList( untypedExpr->args.begin(), untypedExpr->args.end() );
532                        output << ")";
533                } // if
534        }
535
536        void CodeGenerator::postvisit( RangeExpr * rangeExpr ) {
537                rangeExpr->low->accept( *visitor );
538                output << " ... ";
539                rangeExpr->high->accept( *visitor );
540        }
541
542        void CodeGenerator::postvisit( NameExpr * nameExpr ) {
543                extension( nameExpr );
544                OperatorInfo opInfo;
545                if ( operatorLookup( nameExpr->name, opInfo ) ) {
546                        if ( opInfo.type == OT_CONSTANT ) {
547                                output << opInfo.symbol;
548                        } else {
549                                output << opInfo.outputName;
550                        }
551                } else {
552                        output << nameExpr->get_name();
553                } // if
554        }
555
556        void CodeGenerator::postvisit( AddressExpr * addressExpr ) {
557                extension( addressExpr );
558                output << "(&";
559                addressExpr->arg->accept( *visitor );
560                output << ")";
561        }
562
563        void CodeGenerator::postvisit( LabelAddressExpr *addressExpr ) {
564                extension( addressExpr );
565                output << "(&&" << addressExpr->arg << ")";
566        }
567
568        void CodeGenerator::postvisit( CastExpr * castExpr ) {
569                extension( castExpr );
570                output << "(";
571                if ( castExpr->get_result()->isVoid() ) {
572                        output << "(void)" ;
573                } else {
574                        // at least one result type of cast.
575                        // Note: previously, lvalue casts were skipped. Since it's now impossible for the user to write
576                        // an lvalue cast, this has been taken out.
577                        output << "(";
578                        output << genType( castExpr->get_result(), "", pretty, genC );
579                        output << ")";
580                } // if
581                castExpr->get_arg()->accept( *visitor );
582                output << ")";
583        }
584
585        void CodeGenerator::postvisit( VirtualCastExpr * castExpr ) {
586                assertf( ! genC, "VirtualCastExpr should not reach code generation." );
587                extension( castExpr );
588                output << "(virtual ";
589                castExpr->get_arg()->accept( *visitor );
590                output << ")";
591        }
592
593        void CodeGenerator::postvisit( UntypedMemberExpr * memberExpr ) {
594                assertf( ! genC, "UntypedMemberExpr should not reach code generation." );
595                extension( memberExpr );
596                memberExpr->get_aggregate()->accept( *visitor );
597                output << ".";
598                memberExpr->get_member()->accept( *visitor );
599        }
600
601        void CodeGenerator::postvisit( MemberExpr * memberExpr ) {
602                extension( memberExpr );
603                memberExpr->get_aggregate()->accept( *visitor );
604                output << "." << mangleName( memberExpr->get_member() );
605        }
606
607        void CodeGenerator::postvisit( VariableExpr * variableExpr ) {
608                extension( variableExpr );
609                OperatorInfo opInfo;
610                if ( variableExpr->get_var()->get_linkage() == LinkageSpec::Intrinsic && operatorLookup( variableExpr->get_var()->get_name(), opInfo ) && opInfo.type == OT_CONSTANT ) {
611                        output << opInfo.symbol;
612                } else {
613                        output << mangleName( variableExpr->get_var() );
614                } // if
615        }
616
617        void CodeGenerator::postvisit( ConstantExpr * constantExpr ) {
618                assert( constantExpr->get_constant() );
619                extension( constantExpr );
620                constantExpr->get_constant()->accept( *visitor );
621        }
622
623        void CodeGenerator::postvisit( SizeofExpr * sizeofExpr ) {
624                extension( sizeofExpr );
625                output << "sizeof(";
626                if ( sizeofExpr->get_isType() ) {
627                        output << genType( sizeofExpr->get_type(), "", pretty, genC );
628                } else {
629                        sizeofExpr->get_expr()->accept( *visitor );
630                } // if
631                output << ")";
632        }
633
634        void CodeGenerator::postvisit( AlignofExpr * alignofExpr ) {
635                // use GCC extension to avoid bumping std to C11
636                extension( alignofExpr );
637                output << "__alignof__(";
638                if ( alignofExpr->get_isType() ) {
639                        output << genType( alignofExpr->get_type(), "", pretty, genC );
640                } else {
641                        alignofExpr->get_expr()->accept( *visitor );
642                } // if
643                output << ")";
644        }
645
646        void CodeGenerator::postvisit( UntypedOffsetofExpr * offsetofExpr ) {
647                assertf( ! genC, "UntypedOffsetofExpr should not reach code generation." );
648                output << "offsetof(";
649                output << genType( offsetofExpr->get_type(), "", pretty, genC );
650                output << ", " << offsetofExpr->get_member();
651                output << ")";
652        }
653
654        void CodeGenerator::postvisit( OffsetofExpr * offsetofExpr ) {
655                // use GCC builtin
656                output << "__builtin_offsetof(";
657                output << genType( offsetofExpr->get_type(), "", pretty, genC );
658                output << ", " << mangleName( offsetofExpr->get_member() );
659                output << ")";
660        }
661
662        void CodeGenerator::postvisit( OffsetPackExpr * offsetPackExpr ) {
663                assertf( ! genC, "OffsetPackExpr should not reach code generation." );
664                output << "__CFA_offsetpack(" << genType( offsetPackExpr->get_type(), "", pretty, genC ) << ")";
665        }
666
667        void CodeGenerator::postvisit( LogicalExpr * logicalExpr ) {
668                extension( logicalExpr );
669                output << "(";
670                logicalExpr->get_arg1()->accept( *visitor );
671                if ( logicalExpr->get_isAnd() ) {
672                        output << " && ";
673                } else {
674                        output << " || ";
675                } // if
676                logicalExpr->get_arg2()->accept( *visitor );
677                output << ")";
678        }
679
680        void CodeGenerator::postvisit( ConditionalExpr * conditionalExpr ) {
681                extension( conditionalExpr );
682                output << "(";
683                conditionalExpr->get_arg1()->accept( *visitor );
684                output << " ? ";
685                conditionalExpr->get_arg2()->accept( *visitor );
686                output << " : ";
687                conditionalExpr->get_arg3()->accept( *visitor );
688                output << ")";
689        }
690
691        void CodeGenerator::postvisit( CommaExpr * commaExpr ) {
692                extension( commaExpr );
693                output << "(";
694                if ( genC ) {
695                        // arg1 of a CommaExpr is never used, so it can be safely cast to void to reduce gcc warnings.
696                        commaExpr->set_arg1( new CastExpr( commaExpr->get_arg1() ) );
697                }
698                commaExpr->get_arg1()->accept( *visitor );
699                output << " , ";
700                commaExpr->get_arg2()->accept( *visitor );
701                output << ")";
702        }
703
704        void CodeGenerator::postvisit( TupleAssignExpr * tupleExpr ) {
705                assertf( ! genC, "TupleAssignExpr should not reach code generation." );
706                tupleExpr->stmtExpr->accept( *visitor );
707        }
708
709        void CodeGenerator::postvisit( UntypedTupleExpr * tupleExpr ) {
710                assertf( ! genC, "UntypedTupleExpr should not reach code generation." );
711                extension( tupleExpr );
712                output << "[";
713                genCommaList( tupleExpr->get_exprs().begin(), tupleExpr->get_exprs().end() );
714                output << "]";
715        }
716
717        void CodeGenerator::postvisit( TupleExpr * tupleExpr ) {
718                assertf( ! genC, "TupleExpr should not reach code generation." );
719                extension( tupleExpr );
720                output << "[";
721                genCommaList( tupleExpr->get_exprs().begin(), tupleExpr->get_exprs().end() );
722                output << "]";
723        }
724
725        void CodeGenerator::postvisit( TupleIndexExpr * tupleExpr ) {
726                assertf( ! genC, "TupleIndexExpr should not reach code generation." );
727                extension( tupleExpr );
728                tupleExpr->get_tuple()->accept( *visitor );
729                output << "." << tupleExpr->get_index();
730        }
731
732        void CodeGenerator::postvisit( TypeExpr * typeExpr ) {
733                // if ( genC ) std::cerr << "typeexpr still exists: " << typeExpr << std::endl;
734                // assertf( ! genC, "TypeExpr should not reach code generation." );
735                if ( ! genC ) {
736                        output<< genType( typeExpr->get_type(), "", pretty, genC );
737                }
738        }
739
740        void CodeGenerator::postvisit( AsmExpr * asmExpr ) {
741                if ( asmExpr->get_inout() ) {
742                        output << "[ ";
743                        asmExpr->get_inout()->accept( *visitor );
744                        output << " ] ";
745                } // if
746                asmExpr->get_constraint()->accept( *visitor );
747                output << " ( ";
748                asmExpr->get_operand()->accept( *visitor );
749                output << " )";
750        }
751
752        void CodeGenerator::postvisit( CompoundLiteralExpr *compLitExpr ) {
753                assert( compLitExpr->get_result() && dynamic_cast< ListInit * > ( compLitExpr->get_initializer() ) );
754                output << "(" << genType( compLitExpr->get_result(), "", pretty, genC ) << ")";
755                compLitExpr->get_initializer()->accept( *visitor );
756        }
757
758        void CodeGenerator::postvisit( UniqueExpr * unqExpr ) {
759                assertf( ! genC, "Unique expressions should not reach code generation." );
760                output << "unq<" << unqExpr->get_id() << ">{ ";
761                unqExpr->get_expr()->accept( *visitor );
762                output << " }";
763        }
764
765        void CodeGenerator::postvisit( StmtExpr * stmtExpr ) {
766                std::list< Statement * > & stmts = stmtExpr->get_statements()->get_kids();
767                output << "({" << endl;
768                ++indent;
769                unsigned int numStmts = stmts.size();
770                unsigned int i = 0;
771                for ( Statement * stmt : stmts ) {
772                        output << indent << printLabels( stmt->get_labels() );
773                        if ( i+1 == numStmts ) {
774                                // last statement in a statement expression needs to be handled specially -
775                                // cannot cast to void, otherwise the expression statement has no value
776                                if ( ExprStmt * exprStmt = dynamic_cast< ExprStmt * >( stmt ) ) {
777                                        exprStmt->get_expr()->accept( *visitor );
778                                        output << ";" << endl;
779                                        ++i;
780                                        break;
781                                }
782                        }
783                        stmt->accept( *visitor );
784                        output << endl;
785                        if ( wantSpacing( stmt ) ) {
786                                output << endl;
787                        } // if
788                        ++i;
789                }
790                --indent;
791                output << indent << "})";
792        }
793
794        void CodeGenerator::postvisit( ConstructorExpr * expr ) {
795                assertf( ! genC, "Unique expressions should not reach code generation." );
796                expr->callExpr->accept( *visitor );
797        }
798
799        // *** Statements
800        void CodeGenerator::postvisit( CompoundStmt * compoundStmt ) {
801                std::list<Statement*> ks = compoundStmt->get_kids();
802                output << "{" << endl;
803
804                ++indent;
805
806                for ( std::list<Statement *>::iterator i = ks.begin(); i != ks.end();  i++ ) {
807                        output << indent << printLabels( (*i)->get_labels() );
808                        (*i)->accept( *visitor );
809
810                        output << endl;
811                        if ( wantSpacing( *i ) ) {
812                                output << endl;
813                        } // if
814                } // for
815                --indent;
816
817                output << indent << "}";
818        }
819
820        void CodeGenerator::postvisit( ExprStmt * exprStmt ) {
821                assert( exprStmt );
822                if ( genC ) {
823                        // cast the top-level expression to void to reduce gcc warnings.
824                        exprStmt->set_expr( new CastExpr( exprStmt->get_expr() ) );
825                }
826                exprStmt->get_expr()->accept( *visitor );
827                output << ";";
828        }
829
830        void CodeGenerator::postvisit( AsmStmt * asmStmt ) {
831                output << "asm ";
832                if ( asmStmt->get_voltile() ) output << "volatile ";
833                if ( ! asmStmt->get_gotolabels().empty()  ) output << "goto ";
834                output << "( ";
835                if ( asmStmt->get_instruction() ) asmStmt->get_instruction()->accept( *visitor );
836                output << " : ";
837                genCommaList( asmStmt->get_output().begin(), asmStmt->get_output().end() );
838                output << " : ";
839                genCommaList( asmStmt->get_input().begin(), asmStmt->get_input().end() );
840                output << " : ";
841                genCommaList( asmStmt->get_clobber().begin(), asmStmt->get_clobber().end() );
842                if ( ! asmStmt->get_gotolabels().empty() ) {
843                        output << " : ";
844                        for ( std::list<Label>::iterator begin = asmStmt->get_gotolabels().begin();; ) {
845                                output << *begin++;
846                                if ( begin == asmStmt->get_gotolabels().end() ) break;
847                                output << ", ";
848                        } // for
849                } // if
850                output << " );" ;
851        }
852
853        void CodeGenerator::postvisit( AsmDecl * asmDecl ) {
854                output << "asm ";
855                AsmStmt * asmStmt = asmDecl->get_stmt();
856                output << "( ";
857                if ( asmStmt->get_instruction() ) asmStmt->get_instruction()->accept( *visitor );
858                output << " )" ;
859        }
860
861        void CodeGenerator::postvisit( IfStmt * ifStmt ) {
862                output << "if ( ";
863                ifStmt->get_condition()->accept( *visitor );
864                output << " ) ";
865
866                ifStmt->get_thenPart()->accept( *visitor );
867
868                if ( ifStmt->get_elsePart() != 0) {
869                        output << " else ";
870                        ifStmt->get_elsePart()->accept( *visitor );
871                } // if
872        }
873
874        void CodeGenerator::postvisit( SwitchStmt * switchStmt ) {
875                output << "switch ( " ;
876                switchStmt->get_condition()->accept( *visitor );
877                output << " ) ";
878
879                output << "{" << endl;
880                ++indent;
881                acceptAll( switchStmt->get_statements(), *visitor );
882                --indent;
883                output << indent << "}";
884        }
885
886        void CodeGenerator::postvisit( CaseStmt * caseStmt ) {
887                updateLocation( caseStmt );
888                output << indent;
889                if ( caseStmt->isDefault()) {
890                        output << "default";
891                } else {
892                        output << "case ";
893                        caseStmt->get_condition()->accept( *visitor );
894                } // if
895                output << ":" << endl;
896
897                std::list<Statement *> sts = caseStmt->get_statements();
898
899                ++indent;
900                for ( std::list<Statement *>::iterator i = sts.begin(); i != sts.end();  i++) {
901                        output << indent << printLabels( (*i)->get_labels() )  ;
902                        (*i)->accept( *visitor );
903                        output << endl;
904                } // for
905                --indent;
906        }
907
908        void CodeGenerator::postvisit( BranchStmt * branchStmt ) {
909                switch ( branchStmt->get_type()) {
910                  case BranchStmt::Goto:
911                        if ( ! branchStmt->get_target().empty() )
912                                output << "goto " << branchStmt->get_target();
913                        else {
914                                if ( branchStmt->get_computedTarget() != 0 ) {
915                                        output << "goto *";
916                                        branchStmt->get_computedTarget()->accept( *visitor );
917                                } // if
918                        } // if
919                        break;
920                  case BranchStmt::Break:
921                        output << "break";
922                        break;
923                  case BranchStmt::Continue:
924                        output << "continue";
925                        break;
926                } // switch
927                output << ";";
928        }
929
930        void CodeGenerator::postvisit( ReturnStmt * returnStmt ) {
931                output << "return ";
932                maybeAccept( returnStmt->get_expr(), *visitor );
933                output << ";";
934        }
935
936        void CodeGenerator::postvisit( ThrowStmt * throwStmt ) {
937                assertf( ! genC, "Throw statements should not reach code generation." );
938
939                output << ((throwStmt->get_kind() == ThrowStmt::Terminate) ?
940                           "throw" : "throwResume");
941                if (throwStmt->get_expr()) {
942                        output << " ";
943                        throwStmt->get_expr()->accept( *visitor );
944                }
945                if (throwStmt->get_target()) {
946                        output << " _At ";
947                        throwStmt->get_target()->accept( *visitor );
948                }
949                output << ";";
950        }
951        void CodeGenerator::postvisit( CatchStmt * stmt ) {
952                assertf( ! genC, "Catch statements should not reach code generation." );
953
954                output << ((stmt->get_kind() == CatchStmt::Terminate) ?
955                "catch" : "catchResume");
956                output << "( ";
957                stmt->decl->accept( *visitor );
958                output << " ) ";
959
960                if( stmt->cond ) {
961                        output << "if/when(?) (";
962                        stmt->cond->accept( *visitor );
963                        output << ") ";
964                }
965                stmt->body->accept( *visitor );
966        }
967
968        void CodeGenerator::postvisit( WaitForStmt * stmt ) {
969                assertf( ! genC, "Waitfor statements should not reach code generation." );
970
971                bool first = true;
972                for( auto & clause : stmt->clauses ) {
973                        if(first) { output << "or "; first = false; }
974                        if( clause.condition ) {
975                                output << "when(";
976                                stmt->timeout.condition->accept( *visitor );
977                                output << ") ";
978                        }
979                        output << "waitfor(";
980                        clause.target.function->accept( *visitor );
981                        for( Expression * expr : clause.target.arguments ) {
982                                output << ",";
983                                expr->accept( *visitor );
984                        }
985                        output << ") ";
986                        clause.statement->accept( *visitor );
987                }
988
989                if( stmt->timeout.statement ) {
990                        output << "or ";
991                        if( stmt->timeout.condition ) {
992                                output << "when(";
993                                stmt->timeout.condition->accept( *visitor );
994                                output << ") ";
995                        }
996                        output << "timeout(";
997                        stmt->timeout.time->accept( *visitor );
998                        output << ") ";
999                        stmt->timeout.statement->accept( *visitor );
1000                }
1001
1002                if( stmt->orelse.statement ) {
1003                        output << "or ";
1004                        if( stmt->orelse.condition ) {
1005                                output << "when(";
1006                                stmt->orelse.condition->accept( *visitor );
1007                                output << ")";
1008                        }
1009                        output << "else ";
1010                        stmt->orelse.statement->accept( *visitor );
1011                }
1012        }
1013
1014        void CodeGenerator::postvisit( WithStmt * with ) {
1015                if ( ! genC ) {
1016                        output << "with ( ";
1017                        genCommaList( with->exprs.begin(), with->exprs.end() );
1018                        output << " ) ";
1019                }
1020                with->stmt->accept( *visitor );
1021        }
1022
1023        void CodeGenerator::postvisit( WhileStmt * whileStmt ) {
1024                if ( whileStmt->get_isDoWhile() ) {
1025                        output << "do" ;
1026                } else {
1027                        output << "while (" ;
1028                        whileStmt->get_condition()->accept( *visitor );
1029                        output << ")";
1030                } // if
1031                output << " ";
1032
1033                output << CodeGenerator::printLabels( whileStmt->get_body()->get_labels() );
1034                whileStmt->get_body()->accept( *visitor );
1035
1036                output << indent;
1037
1038                if ( whileStmt->get_isDoWhile() ) {
1039                        output << " while (" ;
1040                        whileStmt->get_condition()->accept( *visitor );
1041                        output << ");";
1042                } // if
1043        }
1044
1045        void CodeGenerator::postvisit( ForStmt * forStmt ) {
1046                // initialization is always hoisted, so don't bother doing anything with that
1047                output << "for (;";
1048
1049                if ( forStmt->get_condition() != 0 ) {
1050                        forStmt->get_condition()->accept( *visitor );
1051                } // if
1052                output << ";";
1053
1054                if ( forStmt->get_increment() != 0 ) {
1055                        // cast the top-level expression to void to reduce gcc warnings.
1056                        Expression * expr = new CastExpr( forStmt->get_increment() );
1057                        expr->accept( *visitor );
1058                } // if
1059                output << ") ";
1060
1061                if ( forStmt->get_body() != 0 ) {
1062                        output << CodeGenerator::printLabels( forStmt->get_body()->get_labels() );
1063                        forStmt->get_body()->accept( *visitor );
1064                } // if
1065        }
1066
1067        void CodeGenerator::postvisit( __attribute__((unused)) NullStmt * nullStmt ) {
1068                //output << indent << CodeGenerator::printLabels( nullStmt->get_labels() );
1069                output << "/* null statement */ ;";
1070        }
1071
1072        void CodeGenerator::postvisit( DeclStmt * declStmt ) {
1073                declStmt->get_decl()->accept( *visitor );
1074
1075                if ( doSemicolon( declStmt->get_decl() ) ) {
1076                        output << ";";
1077                } // if
1078        }
1079
1080        void CodeGenerator::postvisit( ImplicitCtorDtorStmt * stmt ) {
1081                assertf( ! genC, "ImplicitCtorDtorStmts should not reach code generation." );
1082                stmt->callStmt->accept( *visitor );
1083        }
1084
1085        void CodeGenerator::handleStorageClass( DeclarationWithType * decl ) {
1086                if ( decl->get_storageClasses().any() ) {
1087                        decl->get_storageClasses().print( output );
1088                } // if
1089        } // CodeGenerator::handleStorageClass
1090
1091        std::string genName( DeclarationWithType * decl ) {
1092                CodeGen::OperatorInfo opInfo;
1093                if ( operatorLookup( decl->get_name(), opInfo ) ) {
1094                        return opInfo.outputName;
1095                } else {
1096                        return decl->get_name();
1097                } // if
1098        }
1099} // namespace CodeGen
1100
1101
1102unsigned Indenter::tabsize = 2;
1103
1104std::ostream & operator<<( std::ostream & out, const BaseSyntaxNode * node ) {
1105        if ( node ) {
1106                node->print( out );
1107        } else {
1108                out << "nullptr";
1109        }
1110        return out;
1111}
1112
1113// Local Variables: //
1114// tab-width: 4 //
1115// mode: c++ //
1116// compile-command: "make install" //
1117// End: //
Note: See TracBrowser for help on using the repository browser.