source: src/CodeGen/CodeGenerator.cc@ 6545adff

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 new-env no_list persistent-indexer pthread-emulation qualifiedEnum resolv-new with_gc
Last change on this file since 6545adff was 50377a4, checked in by Rob Schluntz <rschlunt@…>, 8 years ago

Refactor tree print code to use Indenter
  • Property mode set to 100644
File size: 33.6 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->get_name();
290 if ( typeDecl->get_kind() != TypeDecl::Any && typeDecl->get_sized() ) {
291 output << " | sized(" << typeDecl->get_name() << ")";
292 }
293 if ( ! typeDecl->get_assertions().empty() ) {
294 output << " | { ";
295 genCommaList( typeDecl->get_assertions().begin(), typeDecl->get_assertions().end() );
296 output << " }";
297 }
298 }
299
300 void CodeGenerator::postvisit( Designation * designation ) {
301 std::list< Expression * > designators = designation->get_designators();
302 if ( designators.size() == 0 ) return;
303 for ( Expression * des : designators ) {
304 if ( dynamic_cast< NameExpr * >( des ) || dynamic_cast< VariableExpr * >( des ) ) {
305 // if expression is a NameExpr or VariableExpr, then initializing aggregate member
306 output << ".";
307 des->accept( *visitor );
308 } else {
309 // otherwise, it has to be a ConstantExpr or CastExpr, initializing array eleemnt
310 output << "[";
311 des->accept( *visitor );
312 output << "]";
313 } // if
314 } // for
315 output << " = ";
316 }
317
318 void CodeGenerator::postvisit( SingleInit * init ) {
319 init->get_value()->accept( *visitor );
320 }
321
322 void CodeGenerator::postvisit( ListInit * init ) {
323 auto initBegin = init->begin();
324 auto initEnd = init->end();
325 auto desigBegin = init->get_designations().begin();
326 auto desigEnd = init->get_designations().end();
327
328 output << "{ ";
329 for ( ; initBegin != initEnd && desigBegin != desigEnd; ) {
330 (*desigBegin)->accept( *visitor );
331 (*initBegin)->accept( *visitor );
332 ++initBegin, ++desigBegin;
333 if ( initBegin != initEnd ) {
334 output << ", ";
335 }
336 }
337 output << " }";
338 assertf( initBegin == initEnd && desigBegin == desigEnd, "Initializers and designators not the same length. %s", toString( init ).c_str() );
339 }
340
341 void CodeGenerator::postvisit( ConstructorInit * init ){
342 assertf( ! genC, "ConstructorInit nodes should not reach code generation." );
343 // pseudo-output for constructor/destructor pairs
344 output << "<ctorinit>{" << endl << ++indent << "ctor: ";
345 maybeAccept( init->get_ctor(), *visitor );
346 output << ", " << endl << indent << "dtor: ";
347 maybeAccept( init->get_dtor(), *visitor );
348 output << endl << --indent << "}";
349 }
350
351 void CodeGenerator::postvisit( Constant * constant ) {
352 output << constant->get_value() ;
353 }
354
355 // *** Expressions
356 void CodeGenerator::postvisit( ApplicationExpr * applicationExpr ) {
357 extension( applicationExpr );
358 if ( VariableExpr * varExpr = dynamic_cast< VariableExpr* >( applicationExpr->get_function() ) ) {
359 OperatorInfo opInfo;
360 if ( varExpr->get_var()->get_linkage() == LinkageSpec::Intrinsic && operatorLookup( varExpr->get_var()->get_name(), opInfo ) ) {
361 std::list< Expression* >::iterator arg = applicationExpr->get_args().begin();
362 switch ( opInfo.type ) {
363 case OT_INDEX:
364 assert( applicationExpr->get_args().size() == 2 );
365 (*arg++)->accept( *visitor );
366 output << "[";
367 (*arg)->accept( *visitor );
368 output << "]";
369 break;
370
371 case OT_CALL:
372 // there are no intrinsic definitions of the function call operator
373 assert( false );
374 break;
375
376 case OT_CTOR:
377 case OT_DTOR:
378 if ( applicationExpr->get_args().size() == 1 ) {
379 // the expression fed into a single parameter constructor or destructor may contain side
380 // effects, so must still output this expression
381 output << "(";
382 (*arg++)->accept( *visitor );
383 output << ") /* " << opInfo.inputName << " */";
384 } else if ( applicationExpr->get_args().size() == 2 ) {
385 // intrinsic two parameter constructors are essentially bitwise assignment
386 output << "(";
387 (*arg++)->accept( *visitor );
388 output << opInfo.symbol;
389 (*arg)->accept( *visitor );
390 output << ") /* " << opInfo.inputName << " */";
391 } else {
392 // no constructors with 0 or more than 2 parameters
393 assert( false );
394 } // if
395 break;
396
397 case OT_PREFIX:
398 case OT_PREFIXASSIGN:
399 assert( applicationExpr->get_args().size() == 1 );
400 output << "(";
401 output << opInfo.symbol;
402 (*arg)->accept( *visitor );
403 output << ")";
404 break;
405
406 case OT_POSTFIX:
407 case OT_POSTFIXASSIGN:
408 assert( applicationExpr->get_args().size() == 1 );
409 (*arg)->accept( *visitor );
410 output << opInfo.symbol;
411 break;
412
413
414 case OT_INFIX:
415 case OT_INFIXASSIGN:
416 assert( applicationExpr->get_args().size() == 2 );
417 output << "(";
418 (*arg++)->accept( *visitor );
419 output << opInfo.symbol;
420 (*arg)->accept( *visitor );
421 output << ")";
422 break;
423
424 case OT_CONSTANT:
425 case OT_LABELADDRESS:
426 // there are no intrinsic definitions of 0/1 or label addresses as functions
427 assert( false );
428 } // switch
429 } else {
430 varExpr->accept( *visitor );
431 output << "(";
432 genCommaList( applicationExpr->get_args().begin(), applicationExpr->get_args().end() );
433 output << ")";
434 } // if
435 } else {
436 applicationExpr->get_function()->accept( *visitor );
437 output << "(";
438 genCommaList( applicationExpr->get_args().begin(), applicationExpr->get_args().end() );
439 output << ")";
440 } // if
441 }
442
443 void CodeGenerator::postvisit( UntypedExpr * untypedExpr ) {
444 extension( untypedExpr );
445 if ( NameExpr * nameExpr = dynamic_cast< NameExpr* >( untypedExpr->function ) ) {
446 OperatorInfo opInfo;
447 if ( operatorLookup( nameExpr->name, opInfo ) ) {
448 std::list< Expression* >::iterator arg = untypedExpr->args.begin();
449 switch ( opInfo.type ) {
450 case OT_INDEX:
451 assert( untypedExpr->args.size() == 2 );
452 (*arg++)->accept( *visitor );
453 output << "[";
454 (*arg)->accept( *visitor );
455 output << "]";
456 break;
457
458 case OT_CALL:
459 assert( false );
460
461 case OT_CTOR:
462 case OT_DTOR:
463 if ( untypedExpr->args.size() == 1 ) {
464 // the expression fed into a single parameter constructor or destructor may contain side
465 // effects, so must still output this expression
466 output << "(";
467 (*arg++)->accept( *visitor );
468 output << ") /* " << opInfo.inputName << " */";
469 } else if ( untypedExpr->get_args().size() == 2 ) {
470 // intrinsic two parameter constructors are essentially bitwise assignment
471 output << "(";
472 (*arg++)->accept( *visitor );
473 output << opInfo.symbol;
474 (*arg)->accept( *visitor );
475 output << ") /* " << opInfo.inputName << " */";
476 } else {
477 // no constructors with 0 or more than 2 parameters
478 assertf( ! genC, "UntypedExpr constructor/destructor with 0 or more than 2 parameters." );
479 output << "(";
480 (*arg++)->accept( *visitor );
481 output << opInfo.symbol << "{ ";
482 genCommaList( arg, untypedExpr->args.end() );
483 output << "}) /* " << opInfo.inputName << " */";
484 } // if
485 break;
486
487 case OT_PREFIX:
488 case OT_PREFIXASSIGN:
489 case OT_LABELADDRESS:
490 assert( untypedExpr->args.size() == 1 );
491 output << "(";
492 output << opInfo.symbol;
493 (*arg)->accept( *visitor );
494 output << ")";
495 break;
496
497 case OT_POSTFIX:
498 case OT_POSTFIXASSIGN:
499 assert( untypedExpr->args.size() == 1 );
500 (*arg)->accept( *visitor );
501 output << opInfo.symbol;
502 break;
503
504 case OT_INFIX:
505 case OT_INFIXASSIGN:
506 assert( untypedExpr->args.size() == 2 );
507 output << "(";
508 (*arg++)->accept( *visitor );
509 output << opInfo.symbol;
510 (*arg)->accept( *visitor );
511 output << ")";
512 break;
513
514 case OT_CONSTANT:
515 // there are no intrinsic definitions of 0 or 1 as functions
516 assert( false );
517 } // switch
518 } else {
519 // builtin routines
520 nameExpr->accept( *visitor );
521 output << "(";
522 genCommaList( untypedExpr->args.begin(), untypedExpr->args.end() );
523 output << ")";
524 } // if
525 } else {
526 untypedExpr->function->accept( *visitor );
527 output << "(";
528 genCommaList( untypedExpr->args.begin(), untypedExpr->args.end() );
529 output << ")";
530 } // if
531 }
532
533 void CodeGenerator::postvisit( RangeExpr * rangeExpr ) {
534 rangeExpr->low->accept( *visitor );
535 output << " ... ";
536 rangeExpr->high->accept( *visitor );
537 }
538
539 void CodeGenerator::postvisit( NameExpr * nameExpr ) {
540 extension( nameExpr );
541 OperatorInfo opInfo;
542 if ( operatorLookup( nameExpr->name, opInfo ) ) {
543 if ( opInfo.type == OT_CONSTANT ) {
544 output << opInfo.symbol;
545 } else {
546 output << opInfo.outputName;
547 }
548 } else {
549 output << nameExpr->get_name();
550 } // if
551 }
552
553 void CodeGenerator::postvisit( AddressExpr * addressExpr ) {
554 extension( addressExpr );
555 output << "(&";
556 addressExpr->arg->accept( *visitor );
557 output << ")";
558 }
559
560 void CodeGenerator::postvisit( LabelAddressExpr *addressExpr ) {
561 extension( addressExpr );
562 output << "(&&" << addressExpr->arg << ")";
563 }
564
565 void CodeGenerator::postvisit( CastExpr * castExpr ) {
566 extension( castExpr );
567 output << "(";
568 if ( castExpr->get_result()->isVoid() ) {
569 output << "(void)" ;
570 } else {
571 // at least one result type of cast.
572 // Note: previously, lvalue casts were skipped. Since it's now impossible for the user to write
573 // an lvalue cast, this has been taken out.
574 output << "(";
575 output << genType( castExpr->get_result(), "", pretty, genC );
576 output << ")";
577 } // if
578 castExpr->get_arg()->accept( *visitor );
579 output << ")";
580 }
581
582 void CodeGenerator::postvisit( VirtualCastExpr * castExpr ) {
583 assertf( ! genC, "VirtualCastExpr should not reach code generation." );
584 extension( castExpr );
585 output << "(virtual ";
586 castExpr->get_arg()->accept( *visitor );
587 output << ")";
588 }
589
590 void CodeGenerator::postvisit( UntypedMemberExpr * memberExpr ) {
591 assertf( ! genC, "UntypedMemberExpr should not reach code generation." );
592 extension( memberExpr );
593 memberExpr->get_aggregate()->accept( *visitor );
594 output << ".";
595 memberExpr->get_member()->accept( *visitor );
596 }
597
598 void CodeGenerator::postvisit( MemberExpr * memberExpr ) {
599 extension( memberExpr );
600 memberExpr->get_aggregate()->accept( *visitor );
601 output << "." << mangleName( memberExpr->get_member() );
602 }
603
604 void CodeGenerator::postvisit( VariableExpr * variableExpr ) {
605 extension( variableExpr );
606 OperatorInfo opInfo;
607 if ( variableExpr->get_var()->get_linkage() == LinkageSpec::Intrinsic && operatorLookup( variableExpr->get_var()->get_name(), opInfo ) && opInfo.type == OT_CONSTANT ) {
608 output << opInfo.symbol;
609 } else {
610 output << mangleName( variableExpr->get_var() );
611 } // if
612 }
613
614 void CodeGenerator::postvisit( ConstantExpr * constantExpr ) {
615 assert( constantExpr->get_constant() );
616 extension( constantExpr );
617 constantExpr->get_constant()->accept( *visitor );
618 }
619
620 void CodeGenerator::postvisit( SizeofExpr * sizeofExpr ) {
621 extension( sizeofExpr );
622 output << "sizeof(";
623 if ( sizeofExpr->get_isType() ) {
624 output << genType( sizeofExpr->get_type(), "", pretty, genC );
625 } else {
626 sizeofExpr->get_expr()->accept( *visitor );
627 } // if
628 output << ")";
629 }
630
631 void CodeGenerator::postvisit( AlignofExpr * alignofExpr ) {
632 // use GCC extension to avoid bumping std to C11
633 extension( alignofExpr );
634 output << "__alignof__(";
635 if ( alignofExpr->get_isType() ) {
636 output << genType( alignofExpr->get_type(), "", pretty, genC );
637 } else {
638 alignofExpr->get_expr()->accept( *visitor );
639 } // if
640 output << ")";
641 }
642
643 void CodeGenerator::postvisit( UntypedOffsetofExpr * offsetofExpr ) {
644 assertf( ! genC, "UntypedOffsetofExpr should not reach code generation." );
645 output << "offsetof(";
646 output << genType( offsetofExpr->get_type(), "", pretty, genC );
647 output << ", " << offsetofExpr->get_member();
648 output << ")";
649 }
650
651 void CodeGenerator::postvisit( OffsetofExpr * offsetofExpr ) {
652 // use GCC builtin
653 output << "__builtin_offsetof(";
654 output << genType( offsetofExpr->get_type(), "", pretty, genC );
655 output << ", " << mangleName( offsetofExpr->get_member() );
656 output << ")";
657 }
658
659 void CodeGenerator::postvisit( OffsetPackExpr * offsetPackExpr ) {
660 assertf( ! genC, "OffsetPackExpr should not reach code generation." );
661 output << "__CFA_offsetpack(" << genType( offsetPackExpr->get_type(), "", pretty, genC ) << ")";
662 }
663
664 void CodeGenerator::postvisit( LogicalExpr * logicalExpr ) {
665 extension( logicalExpr );
666 output << "(";
667 logicalExpr->get_arg1()->accept( *visitor );
668 if ( logicalExpr->get_isAnd() ) {
669 output << " && ";
670 } else {
671 output << " || ";
672 } // if
673 logicalExpr->get_arg2()->accept( *visitor );
674 output << ")";
675 }
676
677 void CodeGenerator::postvisit( ConditionalExpr * conditionalExpr ) {
678 extension( conditionalExpr );
679 output << "(";
680 conditionalExpr->get_arg1()->accept( *visitor );
681 output << " ? ";
682 conditionalExpr->get_arg2()->accept( *visitor );
683 output << " : ";
684 conditionalExpr->get_arg3()->accept( *visitor );
685 output << ")";
686 }
687
688 void CodeGenerator::postvisit( CommaExpr * commaExpr ) {
689 extension( commaExpr );
690 output << "(";
691 if ( genC ) {
692 // arg1 of a CommaExpr is never used, so it can be safely cast to void to reduce gcc warnings.
693 commaExpr->set_arg1( new CastExpr( commaExpr->get_arg1() ) );
694 }
695 commaExpr->get_arg1()->accept( *visitor );
696 output << " , ";
697 commaExpr->get_arg2()->accept( *visitor );
698 output << ")";
699 }
700
701 void CodeGenerator::postvisit( TupleAssignExpr * tupleExpr ) {
702 assertf( ! genC, "TupleAssignExpr should not reach code generation." );
703 tupleExpr->stmtExpr->accept( *visitor );
704 }
705
706 void CodeGenerator::postvisit( UntypedTupleExpr * tupleExpr ) {
707 assertf( ! genC, "UntypedTupleExpr should not reach code generation." );
708 extension( tupleExpr );
709 output << "[";
710 genCommaList( tupleExpr->get_exprs().begin(), tupleExpr->get_exprs().end() );
711 output << "]";
712 }
713
714 void CodeGenerator::postvisit( TupleExpr * tupleExpr ) {
715 assertf( ! genC, "TupleExpr should not reach code generation." );
716 extension( tupleExpr );
717 output << "[";
718 genCommaList( tupleExpr->get_exprs().begin(), tupleExpr->get_exprs().end() );
719 output << "]";
720 }
721
722 void CodeGenerator::postvisit( TupleIndexExpr * tupleExpr ) {
723 assertf( ! genC, "TupleIndexExpr should not reach code generation." );
724 extension( tupleExpr );
725 tupleExpr->get_tuple()->accept( *visitor );
726 output << "." << tupleExpr->get_index();
727 }
728
729 void CodeGenerator::postvisit( TypeExpr * typeExpr ) {
730 // if ( genC ) std::cerr << "typeexpr still exists: " << typeExpr << std::endl;
731 // assertf( ! genC, "TypeExpr should not reach code generation." );
732 if ( ! genC ) {
733 output<< genType( typeExpr->get_type(), "", pretty, genC );
734 }
735 }
736
737 void CodeGenerator::postvisit( AsmExpr * asmExpr ) {
738 if ( asmExpr->get_inout() ) {
739 output << "[ ";
740 asmExpr->get_inout()->accept( *visitor );
741 output << " ] ";
742 } // if
743 asmExpr->get_constraint()->accept( *visitor );
744 output << " ( ";
745 asmExpr->get_operand()->accept( *visitor );
746 output << " )";
747 }
748
749 void CodeGenerator::postvisit( CompoundLiteralExpr *compLitExpr ) {
750 assert( compLitExpr->get_result() && dynamic_cast< ListInit * > ( compLitExpr->get_initializer() ) );
751 output << "(" << genType( compLitExpr->get_result(), "", pretty, genC ) << ")";
752 compLitExpr->get_initializer()->accept( *visitor );
753 }
754
755 void CodeGenerator::postvisit( UniqueExpr * unqExpr ) {
756 assertf( ! genC, "Unique expressions should not reach code generation." );
757 output << "unq<" << unqExpr->get_id() << ">{ ";
758 unqExpr->get_expr()->accept( *visitor );
759 output << " }";
760 }
761
762 void CodeGenerator::postvisit( StmtExpr * stmtExpr ) {
763 std::list< Statement * > & stmts = stmtExpr->get_statements()->get_kids();
764 output << "({" << endl;
765 ++indent;
766 unsigned int numStmts = stmts.size();
767 unsigned int i = 0;
768 for ( Statement * stmt : stmts ) {
769 output << indent << printLabels( stmt->get_labels() );
770 if ( i+1 == numStmts ) {
771 // last statement in a statement expression needs to be handled specially -
772 // cannot cast to void, otherwise the expression statement has no value
773 if ( ExprStmt * exprStmt = dynamic_cast< ExprStmt * >( stmt ) ) {
774 exprStmt->get_expr()->accept( *visitor );
775 output << ";" << endl;
776 ++i;
777 break;
778 }
779 }
780 stmt->accept( *visitor );
781 output << endl;
782 if ( wantSpacing( stmt ) ) {
783 output << endl;
784 } // if
785 ++i;
786 }
787 --indent;
788 output << indent << "})";
789 }
790
791 void CodeGenerator::postvisit( ConstructorExpr * expr ) {
792 assertf( ! genC, "Unique expressions should not reach code generation." );
793 expr->callExpr->accept( *visitor );
794 }
795
796 // *** Statements
797 void CodeGenerator::postvisit( CompoundStmt * compoundStmt ) {
798 std::list<Statement*> ks = compoundStmt->get_kids();
799 output << "{" << endl;
800
801 ++indent;
802
803 for ( std::list<Statement *>::iterator i = ks.begin(); i != ks.end(); i++ ) {
804 output << indent << printLabels( (*i)->get_labels() );
805 (*i)->accept( *visitor );
806
807 output << endl;
808 if ( wantSpacing( *i ) ) {
809 output << endl;
810 } // if
811 } // for
812 --indent;
813
814 output << indent << "}";
815 }
816
817 void CodeGenerator::postvisit( ExprStmt * exprStmt ) {
818 assert( exprStmt );
819 if ( genC ) {
820 // cast the top-level expression to void to reduce gcc warnings.
821 exprStmt->set_expr( new CastExpr( exprStmt->get_expr() ) );
822 }
823 exprStmt->get_expr()->accept( *visitor );
824 output << ";";
825 }
826
827 void CodeGenerator::postvisit( AsmStmt * asmStmt ) {
828 output << "asm ";
829 if ( asmStmt->get_voltile() ) output << "volatile ";
830 if ( ! asmStmt->get_gotolabels().empty() ) output << "goto ";
831 output << "( ";
832 if ( asmStmt->get_instruction() ) asmStmt->get_instruction()->accept( *visitor );
833 output << " : ";
834 genCommaList( asmStmt->get_output().begin(), asmStmt->get_output().end() );
835 output << " : ";
836 genCommaList( asmStmt->get_input().begin(), asmStmt->get_input().end() );
837 output << " : ";
838 genCommaList( asmStmt->get_clobber().begin(), asmStmt->get_clobber().end() );
839 if ( ! asmStmt->get_gotolabels().empty() ) {
840 output << " : ";
841 for ( std::list<Label>::iterator begin = asmStmt->get_gotolabels().begin();; ) {
842 output << *begin++;
843 if ( begin == asmStmt->get_gotolabels().end() ) break;
844 output << ", ";
845 } // for
846 } // if
847 output << " );" ;
848 }
849
850 void CodeGenerator::postvisit( AsmDecl * asmDecl ) {
851 output << "asm ";
852 AsmStmt * asmStmt = asmDecl->get_stmt();
853 output << "( ";
854 if ( asmStmt->get_instruction() ) asmStmt->get_instruction()->accept( *visitor );
855 output << " )" ;
856 }
857
858 void CodeGenerator::postvisit( IfStmt * ifStmt ) {
859 output << "if ( ";
860 ifStmt->get_condition()->accept( *visitor );
861 output << " ) ";
862
863 ifStmt->get_thenPart()->accept( *visitor );
864
865 if ( ifStmt->get_elsePart() != 0) {
866 output << " else ";
867 ifStmt->get_elsePart()->accept( *visitor );
868 } // if
869 }
870
871 void CodeGenerator::postvisit( SwitchStmt * switchStmt ) {
872 output << "switch ( " ;
873 switchStmt->get_condition()->accept( *visitor );
874 output << " ) ";
875
876 output << "{" << endl;
877 ++indent;
878 acceptAll( switchStmt->get_statements(), *visitor );
879 --indent;
880 output << indent << "}";
881 }
882
883 void CodeGenerator::postvisit( CaseStmt * caseStmt ) {
884 updateLocation( caseStmt );
885 output << indent;
886 if ( caseStmt->isDefault()) {
887 output << "default";
888 } else {
889 output << "case ";
890 caseStmt->get_condition()->accept( *visitor );
891 } // if
892 output << ":" << endl;
893
894 std::list<Statement *> sts = caseStmt->get_statements();
895
896 ++indent;
897 for ( std::list<Statement *>::iterator i = sts.begin(); i != sts.end(); i++) {
898 output << indent << printLabels( (*i)->get_labels() ) ;
899 (*i)->accept( *visitor );
900 output << endl;
901 } // for
902 --indent;
903 }
904
905 void CodeGenerator::postvisit( BranchStmt * branchStmt ) {
906 switch ( branchStmt->get_type()) {
907 case BranchStmt::Goto:
908 if ( ! branchStmt->get_target().empty() )
909 output << "goto " << branchStmt->get_target();
910 else {
911 if ( branchStmt->get_computedTarget() != 0 ) {
912 output << "goto *";
913 branchStmt->get_computedTarget()->accept( *visitor );
914 } // if
915 } // if
916 break;
917 case BranchStmt::Break:
918 output << "break";
919 break;
920 case BranchStmt::Continue:
921 output << "continue";
922 break;
923 } // switch
924 output << ";";
925 }
926
927 void CodeGenerator::postvisit( ReturnStmt * returnStmt ) {
928 output << "return ";
929 maybeAccept( returnStmt->get_expr(), *visitor );
930 output << ";";
931 }
932
933 void CodeGenerator::postvisit( ThrowStmt * throwStmt ) {
934 assertf( ! genC, "Throw statements should not reach code generation." );
935
936 output << ((throwStmt->get_kind() == ThrowStmt::Terminate) ?
937 "throw" : "throwResume");
938 if (throwStmt->get_expr()) {
939 output << " ";
940 throwStmt->get_expr()->accept( *visitor );
941 }
942 if (throwStmt->get_target()) {
943 output << " _At ";
944 throwStmt->get_target()->accept( *visitor );
945 }
946 output << ";";
947 }
948 void CodeGenerator::postvisit( CatchStmt * stmt ) {
949 assertf( ! genC, "Catch statements should not reach code generation." );
950
951 output << ((stmt->get_kind() == CatchStmt::Terminate) ?
952 "catch" : "catchResume");
953 output << "( ";
954 stmt->decl->accept( *visitor );
955 output << " ) ";
956
957 if( stmt->cond ) {
958 output << "if/when(?) (";
959 stmt->cond->accept( *visitor );
960 output << ") ";
961 }
962 stmt->body->accept( *visitor );
963 }
964
965 void CodeGenerator::postvisit( WaitForStmt * stmt ) {
966 assertf( ! genC, "Waitfor statements should not reach code generation." );
967
968 bool first = true;
969 for( auto & clause : stmt->clauses ) {
970 if(first) { output << "or "; first = false; }
971 if( clause.condition ) {
972 output << "when(";
973 stmt->timeout.condition->accept( *visitor );
974 output << ") ";
975 }
976 output << "waitfor(";
977 clause.target.function->accept( *visitor );
978 for( Expression * expr : clause.target.arguments ) {
979 output << ",";
980 expr->accept( *visitor );
981 }
982 output << ") ";
983 clause.statement->accept( *visitor );
984 }
985
986 if( stmt->timeout.statement ) {
987 output << "or ";
988 if( stmt->timeout.condition ) {
989 output << "when(";
990 stmt->timeout.condition->accept( *visitor );
991 output << ") ";
992 }
993 output << "timeout(";
994 stmt->timeout.time->accept( *visitor );
995 output << ") ";
996 stmt->timeout.statement->accept( *visitor );
997 }
998
999 if( stmt->orelse.statement ) {
1000 output << "or ";
1001 if( stmt->orelse.condition ) {
1002 output << "when(";
1003 stmt->orelse.condition->accept( *visitor );
1004 output << ")";
1005 }
1006 output << "else ";
1007 stmt->orelse.statement->accept( *visitor );
1008 }
1009 }
1010
1011
1012 void CodeGenerator::postvisit( WhileStmt * whileStmt ) {
1013 if ( whileStmt->get_isDoWhile() ) {
1014 output << "do" ;
1015 } else {
1016 output << "while (" ;
1017 whileStmt->get_condition()->accept( *visitor );
1018 output << ")";
1019 } // if
1020 output << " ";
1021
1022 output << CodeGenerator::printLabels( whileStmt->get_body()->get_labels() );
1023 whileStmt->get_body()->accept( *visitor );
1024
1025 output << indent;
1026
1027 if ( whileStmt->get_isDoWhile() ) {
1028 output << " while (" ;
1029 whileStmt->get_condition()->accept( *visitor );
1030 output << ");";
1031 } // if
1032 }
1033
1034 void CodeGenerator::postvisit( ForStmt * forStmt ) {
1035 // initialization is always hoisted, so don't bother doing anything with that
1036 output << "for (;";
1037
1038 if ( forStmt->get_condition() != 0 ) {
1039 forStmt->get_condition()->accept( *visitor );
1040 } // if
1041 output << ";";
1042
1043 if ( forStmt->get_increment() != 0 ) {
1044 // cast the top-level expression to void to reduce gcc warnings.
1045 Expression * expr = new CastExpr( forStmt->get_increment() );
1046 expr->accept( *visitor );
1047 } // if
1048 output << ") ";
1049
1050 if ( forStmt->get_body() != 0 ) {
1051 output << CodeGenerator::printLabels( forStmt->get_body()->get_labels() );
1052 forStmt->get_body()->accept( *visitor );
1053 } // if
1054 }
1055
1056 void CodeGenerator::postvisit( __attribute__((unused)) NullStmt * nullStmt ) {
1057 //output << indent << CodeGenerator::printLabels( nullStmt->get_labels() );
1058 output << "/* null statement */ ;";
1059 }
1060
1061 void CodeGenerator::postvisit( DeclStmt * declStmt ) {
1062 declStmt->get_decl()->accept( *visitor );
1063
1064 if ( doSemicolon( declStmt->get_decl() ) ) {
1065 output << ";";
1066 } // if
1067 }
1068
1069 void CodeGenerator::postvisit( ImplicitCtorDtorStmt * stmt ) {
1070 assertf( ! genC, "ImplicitCtorDtorStmts should not reach code generation." );
1071 stmt->callStmt->accept( *visitor );
1072 }
1073
1074 void CodeGenerator::handleStorageClass( DeclarationWithType * decl ) {
1075 if ( decl->get_storageClasses().any() ) {
1076 decl->get_storageClasses().print( output );
1077 } // if
1078 } // CodeGenerator::handleStorageClass
1079
1080 std::string genName( DeclarationWithType * decl ) {
1081 CodeGen::OperatorInfo opInfo;
1082 if ( operatorLookup( decl->get_name(), opInfo ) ) {
1083 return opInfo.outputName;
1084 } else {
1085 return decl->get_name();
1086 } // if
1087 }
1088} // namespace CodeGen
1089
1090
1091unsigned Indenter::tabsize = 2;
1092
1093std::ostream & operator<<( std::ostream & out, const BaseSyntaxNode * node ) {
1094 if ( node ) {
1095 node->print( out );
1096 } else {
1097 out << "nullptr";
1098 }
1099 return out;
1100}
1101
1102// Local Variables: //
1103// tab-width: 4 //
1104// mode: c++ //
1105// compile-command: "make install" //
1106// End: //
Note: See TracBrowser for help on using the repository browser.