Context Navigation

source: src/CodeGen/CodeGenerator2.cc @ 79551331

ADTaaron-thesisarm-ehast-experimentalcleanup-dtorsctordeferred_resndemanglerenumforall-pointer-decaygc_noraiijacob/cs343-translationjenkins-sandboxmemorynew-astnew-ast-unique-exprnew-envno_listpersistent-indexerpthread-emulationqualifiedEnumresolv-newstringwith_gc

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

Note: See TracBrowser for help on using the repository browser.

Download in other formats: