source: src/CodeGen/CodeGenerator.cc@ 893256d

ADT aaron-thesis arm-eh ast-experimental cleanup-dtors ctor deferred_resn demangler enum forall-pointer-decay gc_noraii jacob/cs343-translation jenkins-sandbox memory 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 893256d was 2794fff, checked in by Rob Schluntz <rschlunt@…>, 10 years ago

started adding some of the basic code for constructors and destructors
  • Property mode set to 100644
File size: 17.9 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 : Rob Schluntz
12// Last Modified On : Wed Jul 15 14:47:42 2015
13// Update Count : 177
14//
15
16#include <algorithm>
17#include <iostream>
18#include <cassert>
19#include <list>
20
21#include "Parser/ParseNode.h"
22
23#include "SynTree/Type.h"
24#include "SynTree/Expression.h"
25#include "SynTree/Initializer.h"
26#include "SynTree/Statement.h"
27
28#include "utility.h"
29#include "UnimplementedError.h"
30
31#include "CodeGenerator.h"
32#include "OperatorTable.h"
33#include "GenType.h"
34
35using namespace std;
36
37namespace CodeGen {
38 int CodeGenerator::tabsize = 4;
39
40 // the kinds of statements that would ideally be followed by whitespace
41 bool wantSpacing( Statement * stmt) {
42 return dynamic_cast< IfStmt * >( stmt ) || dynamic_cast< CompoundStmt * >( stmt ) ||
43 dynamic_cast< WhileStmt * >( stmt ) || dynamic_cast< ForStmt * > ( stmt ) || dynamic_cast< SwitchStmt *>( stmt );
44 }
45
46 ostream & CodeGenerator::Indenter::operator()( ostream & output ) {
47 return output << string( cg.cur_indent, ' ' );
48 }
49
50 ostream & operator<<( ostream & output, CodeGenerator::Indenter &indent ) {
51 return indent( output );
52 }
53
54 CodeGenerator::CodeGenerator( std::ostream &os ) : indent(*this), cur_indent( 0 ), insideFunction( false ), output( os ) { }
55
56 CodeGenerator::CodeGenerator( std::ostream &os, std::string init, int indentation, bool infunp )
57 : indent(*this), cur_indent( indentation ), insideFunction( infunp ), output( os ) {
58 //output << std::string( init );
59 }
60
61 CodeGenerator::CodeGenerator( std::ostream &os, char *init, int indentation, bool infunp )
62 : indent(*this), cur_indent( indentation ), insideFunction( infunp ), output( os ) {
63 //output << std::string( init );
64 }
65
66 string mangleName( DeclarationWithType *decl ) {
67 if ( decl->get_mangleName() != "" ) {
68 return decl->get_mangleName();
69 } else {
70 return decl->get_name();
71 } // if
72 }
73
74 //*** Declarations
75 void CodeGenerator::visit( FunctionDecl *functionDecl ) {
76 handleStorageClass( functionDecl );
77 if ( functionDecl->get_isInline() ) {
78 output << "inline ";
79 } // if
80 if ( functionDecl->get_isNoreturn() ) {
81 output << "_Noreturn ";
82 } // if
83 output << genType( functionDecl->get_functionType(), mangleName( functionDecl ) );
84
85 // how to get this to the Functype?
86 std::list< Declaration * > olds = functionDecl->get_oldDecls();
87 if ( ! olds.empty() ) {
88 output << " /* function has old declaration */";
89 } // if
90
91 // acceptAll( functionDecl->get_oldDecls(), *this );
92 if ( functionDecl->get_statements() ) {
93 functionDecl->get_statements()->accept(*this );
94 } // if
95 }
96
97 void CodeGenerator::visit( ObjectDecl *objectDecl ) {
98 handleStorageClass( objectDecl );
99 output << genType( objectDecl->get_type(), mangleName( objectDecl ) );
100
101 if ( objectDecl->get_init() ) {
102 output << " = ";
103 objectDecl->get_init()->accept( *this );
104 } // if
105 if ( objectDecl->get_bitfieldWidth() ) {
106 output << ":";
107 objectDecl->get_bitfieldWidth()->accept( *this );
108 } // if
109 }
110
111 void CodeGenerator::handleAggregate( AggregateDecl *aggDecl ) {
112 if ( aggDecl->get_name() != "" )
113 output << aggDecl->get_name();
114
115 std::list< Declaration * > &memb = aggDecl->get_members();
116
117 if ( ! memb.empty() ) {
118 output << " {" << endl;
119
120 cur_indent += CodeGenerator::tabsize;
121 for ( std::list< Declaration* >::iterator i = memb.begin(); i != memb.end(); i++) {
122 output << indent;
123 (*i)->accept(*this );
124 output << ";" << endl;
125 }
126
127 cur_indent -= CodeGenerator::tabsize;
128
129 output << indent << "}";
130 } // if
131 }
132
133 void CodeGenerator::visit( StructDecl *structDecl ) {
134 output << "struct ";
135 handleAggregate( structDecl );
136 }
137
138 void CodeGenerator::visit( UnionDecl *aggregateDecl ) {
139 output << "union ";
140 handleAggregate( aggregateDecl );
141 }
142
143 void CodeGenerator::visit( EnumDecl *aggDecl ) {
144 output << "enum ";
145
146 if ( aggDecl->get_name() != "" )
147 output << aggDecl->get_name();
148
149 std::list< Declaration* > &memb = aggDecl->get_members();
150
151 if ( ! memb.empty() ) {
152 output << " {" << endl;
153
154 cur_indent += CodeGenerator::tabsize;
155 for ( std::list< Declaration* >::iterator i = memb.begin(); i != memb.end(); i++) {
156 ObjectDecl *obj = dynamic_cast< ObjectDecl* >( *i );
157 assert( obj );
158 output << indent << mangleName( obj );
159 if ( obj->get_init() ) {
160 output << " = ";
161 obj->get_init()->accept(*this );
162 } // if
163 output << "," << endl;
164 } // for
165
166 cur_indent -= CodeGenerator::tabsize;
167
168 output << indent << "}";
169 } // if
170 }
171
172 void CodeGenerator::visit( ContextDecl *aggregateDecl ) {}
173
174 void CodeGenerator::visit( TypedefDecl *typeDecl ) {
175 output << "typedef ";
176 output << genType( typeDecl->get_base(), typeDecl->get_name() );
177 }
178
179 void CodeGenerator::visit( TypeDecl *typeDecl ) {
180 // really, we should mutate this into something that isn't a TypeDecl but that requires large-scale changes,
181 // still to be done
182 output << "extern unsigned long " << typeDecl->get_name();
183 if ( typeDecl->get_base() ) {
184 output << " = sizeof( " << genType( typeDecl->get_base(), "" ) << " )";
185 } // if
186 }
187
188 void CodeGenerator::visit( SingleInit *init ) {
189 init->get_value()->accept( *this );
190 }
191
192 void CodeGenerator::visit( ListInit *init ) {
193 output << "{ ";
194 genCommaList( init->begin_initializers(), init->end_initializers() );
195 output << " }";
196 }
197
198 void CodeGenerator::visit( Constant *constant ) {
199 output << constant->get_value() ;
200 }
201
202 //*** Expressions
203 void CodeGenerator::visit( ApplicationExpr *applicationExpr ) {
204 if ( VariableExpr *varExpr = dynamic_cast< VariableExpr* >( applicationExpr->get_function() ) ) {
205 OperatorInfo opInfo;
206 if ( varExpr->get_var()->get_linkage() == LinkageSpec::Intrinsic && operatorLookup( varExpr->get_var()->get_name(), opInfo ) ) {
207 std::list< Expression* >::iterator arg = applicationExpr->get_args().begin();
208 switch ( opInfo.type ) {
209 case OT_PREFIXASSIGN:
210 case OT_POSTFIXASSIGN:
211 case OT_INFIXASSIGN:
212 {
213 assert( arg != applicationExpr->get_args().end() );
214 if ( AddressExpr *addrExpr = dynamic_cast< AddressExpr * >( *arg ) ) {
215
216 *arg = addrExpr->get_arg();
217 } else {
218 UntypedExpr *newExpr = new UntypedExpr( new NameExpr( "*?" ) );
219 newExpr->get_args().push_back( *arg );
220 *arg = newExpr;
221 } // if
222 break;
223 }
224
225 default:
226 // do nothing
227 ;
228 }
229
230 switch ( opInfo.type ) {
231 case OT_INDEX:
232 assert( applicationExpr->get_args().size() == 2 );
233 (*arg++)->accept( *this );
234 output << "[";
235 (*arg)->accept( *this );
236 output << "]";
237 break;
238
239 case OT_CALL:
240 case OT_CTOR:
241 case OT_DTOR:
242 // there are no intrinsic definitions of the function call operator or constructors or destructors
243 assert( false );
244 break;
245
246 case OT_PREFIX:
247 case OT_PREFIXASSIGN:
248 assert( applicationExpr->get_args().size() == 1 );
249 output << "(";
250 output << opInfo.symbol;
251 (*arg)->accept( *this );
252 output << ")";
253 break;
254
255 case OT_POSTFIX:
256 case OT_POSTFIXASSIGN:
257 assert( applicationExpr->get_args().size() == 1 );
258 (*arg)->accept( *this );
259 output << opInfo.symbol;
260 break;
261
262 case OT_INFIX:
263 case OT_INFIXASSIGN:
264 assert( applicationExpr->get_args().size() == 2 );
265 output << "(";
266 (*arg++)->accept( *this );
267 output << opInfo.symbol;
268 (*arg)->accept( *this );
269 output << ")";
270 break;
271
272 case OT_CONSTANT:
273 case OT_LABELADDRESS:
274 // there are no intrinsic definitions of 0/1 or label addresses as functions
275 assert( false );
276 }
277 } else {
278 varExpr->accept( *this );
279 output << "(";
280 genCommaList( applicationExpr->get_args().begin(), applicationExpr->get_args().end() );
281 output << ")";
282 } // if
283 } else {
284 applicationExpr->get_function()->accept( *this );
285 output << "(";
286 genCommaList( applicationExpr->get_args().begin(), applicationExpr->get_args().end() );
287 output << ")";
288 } // if
289 }
290
291 void CodeGenerator::visit( UntypedExpr *untypedExpr ) {
292 if ( NameExpr *nameExpr = dynamic_cast< NameExpr* >( untypedExpr->get_function() ) ) {
293 OperatorInfo opInfo;
294 if ( operatorLookup( nameExpr->get_name(), opInfo ) ) {
295 std::list< Expression* >::iterator arg = untypedExpr->get_args().begin();
296 switch ( opInfo.type ) {
297 case OT_INDEX:
298 assert( untypedExpr->get_args().size() == 2 );
299 (*arg++)->accept( *this );
300 output << "[";
301 (*arg)->accept( *this );
302 output << "]";
303 break;
304
305 case OT_CALL:
306 assert( false );
307 break;
308
309 case OT_PREFIX:
310 case OT_PREFIXASSIGN:
311 case OT_LABELADDRESS:
312 assert( untypedExpr->get_args().size() == 1 );
313 output << "(";
314 output << opInfo.symbol;
315 (*arg)->accept( *this );
316 output << ")";
317 break;
318
319 case OT_POSTFIX:
320 case OT_POSTFIXASSIGN:
321 assert( untypedExpr->get_args().size() == 1 );
322 (*arg)->accept( *this );
323 output << opInfo.symbol;
324 break;
325
326 case OT_INFIX:
327 case OT_INFIXASSIGN:
328 assert( untypedExpr->get_args().size() == 2 );
329 output << "(";
330 (*arg++)->accept( *this );
331 output << opInfo.symbol;
332 (*arg)->accept( *this );
333 output << ")";
334 break;
335
336 case OT_CONSTANT:
337 // there are no intrinsic definitions of 0 or 1 as functions
338 assert( false );
339 }
340 } else {
341 nameExpr->accept( *this );
342 output << "(";
343 genCommaList( untypedExpr->get_args().begin(), untypedExpr->get_args().end() );
344 output << ")";
345 } // if
346 } else {
347 untypedExpr->get_function()->accept( *this );
348 output << "(";
349 genCommaList( untypedExpr->get_args().begin(), untypedExpr->get_args().end() );
350 output << ")";
351 } // if
352 }
353
354 void CodeGenerator::visit( NameExpr *nameExpr ) {
355 OperatorInfo opInfo;
356 if ( operatorLookup( nameExpr->get_name(), opInfo ) ) {
357 assert( opInfo.type == OT_CONSTANT );
358 output << opInfo.symbol;
359 } else {
360 output << nameExpr->get_name();
361 } // if
362 }
363
364 void CodeGenerator::visit( AddressExpr *addressExpr ) {
365 output << "(&";
366 // this hack makes sure that we don't convert "constant_zero" to "0" if we're taking its address
367 if ( VariableExpr *variableExpr = dynamic_cast< VariableExpr* >( addressExpr->get_arg() ) ) {
368 output << mangleName( variableExpr->get_var() );
369 } else {
370 addressExpr->get_arg()->accept( *this );
371 } // if
372 output << ")";
373 }
374
375 void CodeGenerator::visit( CastExpr *castExpr ) {
376 output << "((";
377 if ( castExpr->get_results().empty() ) {
378 output << "void" ;
379 } else {
380 output << genType( castExpr->get_results().front(), "" );
381 } // if
382 output << ")";
383 castExpr->get_arg()->accept( *this );
384 output << ")";
385 }
386
387 void CodeGenerator::visit( UntypedMemberExpr *memberExpr ) {
388 assert( false );
389 }
390
391 void CodeGenerator::visit( MemberExpr *memberExpr ) {
392 memberExpr->get_aggregate()->accept( *this );
393 output << "." << mangleName( memberExpr->get_member() );
394 }
395
396 void CodeGenerator::visit( VariableExpr *variableExpr ) {
397 OperatorInfo opInfo;
398 if ( variableExpr->get_var()->get_linkage() == LinkageSpec::Intrinsic && operatorLookup( variableExpr->get_var()->get_name(), opInfo ) && opInfo.type == OT_CONSTANT ) {
399 output << opInfo.symbol;
400 } else {
401 output << mangleName( variableExpr->get_var() );
402 } // if
403 }
404
405 void CodeGenerator::visit( ConstantExpr *constantExpr ) {
406 assert( constantExpr->get_constant() );
407 constantExpr->get_constant()->accept( *this );
408 }
409
410 void CodeGenerator::visit( SizeofExpr *sizeofExpr ) {
411 output << "sizeof(";
412 if ( sizeofExpr->get_isType() ) {
413 output << genType( sizeofExpr->get_type(), "" );
414 } else {
415 sizeofExpr->get_expr()->accept( *this );
416 } // if
417 output << ")";
418 }
419
420 void CodeGenerator::visit( LogicalExpr *logicalExpr ) {
421 output << "(";
422 logicalExpr->get_arg1()->accept( *this );
423 if ( logicalExpr->get_isAnd() ) {
424 output << " && ";
425 } else {
426 output << " || ";
427 } // if
428 logicalExpr->get_arg2()->accept( *this );
429 output << ")";
430 }
431
432 void CodeGenerator::visit( ConditionalExpr *conditionalExpr ) {
433 output << "(";
434 conditionalExpr->get_arg1()->accept( *this );
435 output << " ? ";
436 conditionalExpr->get_arg2()->accept( *this );
437 output << " : ";
438 conditionalExpr->get_arg3()->accept( *this );
439 output << ")";
440 }
441
442 void CodeGenerator::visit( CommaExpr *commaExpr ) {
443 output << "(";
444 commaExpr->get_arg1()->accept( *this );
445 output << " , ";
446 commaExpr->get_arg2()->accept( *this );
447 output << ")";
448 }
449
450 void CodeGenerator::visit( TupleExpr *tupleExpr ) {}
451
452 void CodeGenerator::visit( TypeExpr *typeExpr ) {}
453
454 //*** Statements
455 void CodeGenerator::visit( CompoundStmt *compoundStmt ) {
456 std::list<Statement*> ks = compoundStmt->get_kids();
457 output << "{" << endl;
458
459 cur_indent += CodeGenerator::tabsize;
460
461 for ( std::list<Statement *>::iterator i = ks.begin(); i != ks.end(); i++) {
462 output << indent << printLabels( (*i)->get_labels() );
463 (*i)->accept(*this );
464
465 output << endl;
466 if ( wantSpacing( *i ) ) {
467 output << endl;
468 }
469 }
470 cur_indent -= CodeGenerator::tabsize;
471
472 output << indent << "}";
473 }
474
475 void CodeGenerator::visit( ExprStmt *exprStmt ) {
476 // I don't see why this check is necessary.
477 // If this starts to cause problems then put it back in,
478 // with an explanation
479 assert( exprStmt );
480
481 // if ( exprStmt != 0 ) {
482 exprStmt->get_expr()->accept( *this );
483 output << ";" ;
484 // } // if
485 }
486
487 void CodeGenerator::visit( IfStmt *ifStmt ) {
488 output << "if (";
489 ifStmt->get_condition()->accept(*this );
490 output << ") ";
491
492 ifStmt->get_thenPart()->accept(*this );
493
494 if ( ifStmt->get_elsePart() != 0) {
495 output << " else ";
496 ifStmt->get_elsePart()->accept(*this );
497 } // if
498 }
499
500 void CodeGenerator::visit( SwitchStmt *switchStmt ) {
501 output << "switch (" ;
502 switchStmt->get_condition()->accept(*this );
503 output << ") ";
504
505 output << "{" << std::endl;
506 cur_indent += CodeGenerator::tabsize;
507
508 acceptAll( switchStmt->get_branches(), *this );
509
510 cur_indent -= CodeGenerator::tabsize;
511
512 output << indent << "}";
513 }
514
515 void CodeGenerator::visit( CaseStmt *caseStmt ) {
516 output << indent;
517 if ( caseStmt->isDefault()) {
518 output << "default";
519 } else {
520 output << "case ";
521 caseStmt->get_condition()->accept(*this );
522 } // if
523 output << ":\n";
524
525 std::list<Statement *> sts = caseStmt->get_statements();
526
527 cur_indent += CodeGenerator::tabsize;
528 for ( std::list<Statement *>::iterator i = sts.begin(); i != sts.end(); i++) {
529 output << indent << printLabels( (*i)->get_labels() ) ;
530 (*i)->accept(*this );
531 output << endl;
532 }
533 cur_indent -= CodeGenerator::tabsize;
534 }
535
536 void CodeGenerator::visit( BranchStmt *branchStmt ) {
537 switch ( branchStmt->get_type()) {
538 case BranchStmt::Goto:
539 if ( ! branchStmt->get_target().empty() )
540 output << "goto " << branchStmt->get_target();
541 else {
542 if ( branchStmt->get_computedTarget() != 0 ) {
543 output << "goto *";
544 branchStmt->get_computedTarget()->accept( *this );
545 } // if
546 } // if
547 break;
548 case BranchStmt::Break:
549 output << "break";
550 break;
551 case BranchStmt::Continue:
552 output << "continue";
553 break;
554 }
555 output << ";";
556 }
557
558
559 void CodeGenerator::visit( ReturnStmt *returnStmt ) {
560 output << "return ";
561
562 // xxx -- check for null expression;
563 if ( returnStmt->get_expr() ) {
564 returnStmt->get_expr()->accept( *this );
565 } // if
566 output << ";";
567 }
568
569 void CodeGenerator::visit( WhileStmt *whileStmt ) {
570 if ( whileStmt->get_isDoWhile() )
571 output << "do" ;
572 else {
573 output << "while (" ;
574 whileStmt->get_condition()->accept(*this );
575 output << ")";
576 } // if
577 output << " ";
578
579 output << CodeGenerator::printLabels( whileStmt->get_body()->get_labels() );
580 whileStmt->get_body()->accept( *this );
581
582 output << indent;
583
584 if ( whileStmt->get_isDoWhile() ) {
585 output << " while (" ;
586 whileStmt->get_condition()->accept(*this );
587 output << ");";
588 } // if
589 }
590
591 void CodeGenerator::visit( ForStmt *forStmt ) {
592 // initialization is always hoisted, so don't
593 // bother doing anything with that
594 output << "for (;";
595
596 if ( forStmt->get_condition() != 0 )
597 forStmt->get_condition()->accept( *this );
598 output << ";";
599
600 if ( forStmt->get_increment() != 0 )
601 forStmt->get_increment()->accept( *this );
602 output << ") ";
603
604 if ( forStmt->get_body() != 0 ) {
605 output << CodeGenerator::printLabels( forStmt->get_body()->get_labels() );
606 forStmt->get_body()->accept( *this );
607 } // if
608 }
609
610 void CodeGenerator::visit( NullStmt *nullStmt ) {
611 //output << indent << CodeGenerator::printLabels( nullStmt->get_labels() );
612 output << "/* null statement */ ;";
613 }
614
615 void CodeGenerator::visit( DeclStmt *declStmt ) {
616 declStmt->get_decl()->accept( *this );
617
618 if ( doSemicolon( declStmt->get_decl() ) ) {
619 output << ";";
620 } // if
621 }
622
623 std::string CodeGenerator::printLabels( std::list< Label > &l ) {
624 std::string str( "" );
625 l.unique(); // assumes a sorted list. Why not use set?
626
627 for ( std::list< Label >::iterator i = l.begin(); i != l.end(); i++ )
628 str += *i + ": ";
629
630 return str;
631 }
632
633 void CodeGenerator::handleStorageClass( Declaration *decl ) {
634 switch ( decl->get_storageClass() ) {
635 case DeclarationNode::Extern:
636 output << "extern ";
637 break;
638 case DeclarationNode::Static:
639 output << "static ";
640 break;
641 case DeclarationNode::Auto:
642 // silently drop storage class
643 break;
644 case DeclarationNode::Register:
645 output << "register ";
646 break;
647 case DeclarationNode::Inline:
648 output << "inline ";
649 break;
650 case DeclarationNode::Fortran:
651 output << "fortran ";
652 break;
653 case DeclarationNode::Noreturn:
654 output << "_Noreturn ";
655 break;
656 case DeclarationNode::Threadlocal:
657 output << "_Thread_local ";
658 break;
659 case DeclarationNode::NoStorageClass:
660 break;
661 } // switch
662 }
663} // namespace CodeGen
664
665// Local Variables: //
666// tab-width: 4 //
667// mode: c++ //
668// compile-command: "make install" //
669// End: //
Note: See TracBrowser for help on using the repository browser.