source: src/CodeGen/CodeGenerator.cc@ 1d386a7

ADT aaron-thesis arm-eh ast-experimental cleanup-dtors deferred_resn demangler enum forall-pointer-decay jacob/cs343-translation jenkins-sandbox new-ast new-ast-unique-expr no_list persistent-indexer pthread-emulation qualifiedEnum
Last change on this file since 1d386a7 was 85b2300, checked in by Rob Schluntz <rschlunt@…>, 7 years ago

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