source: src/ResolvExpr/Resolver.cc @ 94b4364

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 94b4364 was 94b4364, checked in by Rob Schluntz <rschlunt@…>, 9 years ago

implemented resolver methods for aggregate initialization

  • Property mode set to 100644
File size: 15.2 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// Resolver.cc --
8//
9// Author           : Richard C. Bilson
10// Created On       : Sun May 17 12:17:01 2015
11// Last Modified By : Rob Schluntz
12// Last Modified On : Tue Jun 16 14:50:11 2015
13// Update Count     : 154
14//
15
16#include "Resolver.h"
17#include "AlternativeFinder.h"
18#include "Alternative.h"
19#include "RenameVars.h"
20#include "ResolveTypeof.h"
21#include "SynTree/Statement.h"
22#include "SynTree/Type.h"
23#include "SynTree/Expression.h"
24#include "SynTree/Initializer.h"
25#include "SymTab/Indexer.h"
26#include "utility.h"
27
28#include <iostream>
29using namespace std;
30
31namespace ResolvExpr {
32        class Resolver : public SymTab::Indexer {
33          public:
34                Resolver() : SymTab::Indexer( false ), switchType( 0 ) {}
35 
36                virtual void visit( FunctionDecl *functionDecl );
37                virtual void visit( ObjectDecl *functionDecl );
38                virtual void visit( TypeDecl *typeDecl );
39
40                virtual void visit( ArrayType * at );
41
42                virtual void visit( ExprStmt *exprStmt );
43                virtual void visit( IfStmt *ifStmt );
44                virtual void visit( WhileStmt *whileStmt );
45                virtual void visit( ForStmt *forStmt );
46                virtual void visit( SwitchStmt *switchStmt );
47                virtual void visit( ChooseStmt *switchStmt );
48                virtual void visit( CaseStmt *caseStmt );
49                virtual void visit( ReturnStmt *returnStmt );
50
51                virtual void visit( SingleInit *singleInit );
52                virtual void visit( ListInit *listInit );
53          private:
54        typedef std::list< Initializer * >::iterator InitIterator;
55
56          void resolveAggrInit( AggregateDecl *, InitIterator &, InitIterator & );
57          void resolveSingleAggrInit( Declaration *, InitIterator &, InitIterator & );
58
59                std::list< Type * > functionReturn;
60                Type *initContext;
61                Type *switchType;
62        };
63
64        void resolve( std::list< Declaration * > translationUnit ) {
65                Resolver resolver;
66                acceptAll( translationUnit, resolver );
67#if 0
68                resolver.print( cerr );
69                for ( std::list< Declaration * >::iterator i = translationUnit.begin(); i != translationUnit.end(); ++i ) {
70                        (*i)->print( std::cerr );
71                        (*i)->accept( resolver );
72                } // for
73#endif
74        }
75
76        Expression *resolveInVoidContext( Expression *expr, const SymTab::Indexer &indexer ) {
77                TypeEnvironment env;
78                return resolveInVoidContext( expr, indexer, env );
79        }
80
81        namespace {
82                void finishExpr( Expression *expr, const TypeEnvironment &env ) {
83                        expr->set_env( new TypeSubstitution );
84                        env.makeSubstitution( *expr->get_env() );
85                }
86
87                Expression *findVoidExpression( Expression *untyped, const SymTab::Indexer &indexer ) {
88                        global_renamer.reset();
89                        TypeEnvironment env;
90                        Expression *newExpr = resolveInVoidContext( untyped, indexer, env );
91                        finishExpr( newExpr, env );
92                        return newExpr;
93                }
94 
95                Expression *findSingleExpression( Expression *untyped, const SymTab::Indexer &indexer ) {
96                        TypeEnvironment env;
97                        AlternativeFinder finder( indexer, env );
98                        finder.find( untyped );
99#if 0
100                        if ( finder.get_alternatives().size() != 1 ) {
101                                std::cout << "untyped expr is ";
102                                untyped->print( std::cout );
103                                std::cout << std::endl << "alternatives are:";
104                                for ( std::list< Alternative >::const_iterator i = finder.get_alternatives().begin(); i != finder.get_alternatives().end(); ++i ) {
105                                        i->print( std::cout );
106                                } // for
107                        } // if
108#endif
109                        assert( finder.get_alternatives().size() == 1 );
110                        Alternative &choice = finder.get_alternatives().front();
111                        Expression *newExpr = choice.expr->clone();
112                        finishExpr( newExpr, choice.env );
113                        return newExpr;
114                }
115
116                bool isIntegralType( Type *type ) {
117                        if ( dynamic_cast< EnumInstType * >( type ) ) {
118                                return true;
119                        } else if ( BasicType *bt = dynamic_cast< BasicType * >( type ) ) {
120                                return bt->isInteger();
121                        } else {
122                                return false;
123                        } // if
124                }
125 
126                Expression *findIntegralExpression( Expression *untyped, const SymTab::Indexer &indexer ) {
127                        TypeEnvironment env;
128                        AlternativeFinder finder( indexer, env );
129                        finder.find( untyped );
130#if 0
131                        if ( finder.get_alternatives().size() != 1 ) {
132                                std::cout << "untyped expr is ";
133                                untyped->print( std::cout );
134                                std::cout << std::endl << "alternatives are:";
135                                for ( std::list< Alternative >::const_iterator i = finder.get_alternatives().begin(); i != finder.get_alternatives().end(); ++i ) {
136                                        i->print( std::cout );
137                                } // for
138                        } // if
139#endif
140                        Expression *newExpr = 0;
141                        const TypeEnvironment *newEnv = 0;
142                        for ( AltList::const_iterator i = finder.get_alternatives().begin(); i != finder.get_alternatives().end(); ++i ) {
143                                if ( i->expr->get_results().size() == 1 && isIntegralType( i->expr->get_results().front() ) ) {
144                                        if ( newExpr ) {
145                                                throw SemanticError( "Too many interpretations for case control expression", untyped );
146                                        } else {
147                                                newExpr = i->expr->clone();
148                                                newEnv = &i->env;
149                                        } // if
150                                } // if
151                        } // for
152                        if ( ! newExpr ) {
153                                throw SemanticError( "No interpretations for case control expression", untyped );
154                        } // if
155                        finishExpr( newExpr, *newEnv );
156                        return newExpr;
157                }
158 
159        }
160 
161        void Resolver::visit( ObjectDecl *objectDecl ) {
162                Type *new_type = resolveTypeof( objectDecl->get_type(), *this );
163                objectDecl->set_type( new_type );
164                initContext = new_type;
165                SymTab::Indexer::visit( objectDecl );
166        }
167
168        void Resolver::visit( ArrayType * at ) {
169                if ( at->get_dimension() ) {
170                        BasicType arrayLenType = BasicType( Type::Qualifiers(), BasicType::LongUnsignedInt );
171                        CastExpr *castExpr = new CastExpr( at->get_dimension(), arrayLenType.clone() );
172                        Expression *newExpr = findSingleExpression( castExpr, *this );
173                        delete at->get_dimension();
174                        at->set_dimension( newExpr );
175                }
176                Visitor::visit( at );
177        }
178
179        void Resolver::visit( TypeDecl *typeDecl ) {
180                if ( typeDecl->get_base() ) {
181                        Type *new_type = resolveTypeof( typeDecl->get_base(), *this );
182                        typeDecl->set_base( new_type );
183                } // if
184                SymTab::Indexer::visit( typeDecl );
185        }
186
187        void Resolver::visit( FunctionDecl *functionDecl ) {
188#if 0
189                std::cout << "resolver visiting functiondecl ";
190                functionDecl->print( std::cout );
191                std::cout << std::endl;
192#endif
193                Type *new_type = resolveTypeof( functionDecl->get_type(), *this );
194                functionDecl->set_type( new_type );
195                std::list< Type * > oldFunctionReturn = functionReturn;
196                functionReturn.clear();
197                for ( std::list< DeclarationWithType * >::const_iterator i = functionDecl->get_functionType()->get_returnVals().begin(); i != functionDecl->get_functionType()->get_returnVals().end(); ++i ) {
198                        functionReturn.push_back( (*i)->get_type() );
199                } // for
200                SymTab::Indexer::visit( functionDecl );
201                functionReturn = oldFunctionReturn;
202        }
203
204        void Resolver::visit( ExprStmt *exprStmt ) {
205                if ( exprStmt->get_expr() ) {
206                        Expression *newExpr = findVoidExpression( exprStmt->get_expr(), *this );
207                        delete exprStmt->get_expr();
208                        exprStmt->set_expr( newExpr );
209                } // if
210        }
211
212        void Resolver::visit( IfStmt *ifStmt ) {
213                Expression *newExpr = findSingleExpression( ifStmt->get_condition(), *this );
214                delete ifStmt->get_condition();
215                ifStmt->set_condition( newExpr );
216                Visitor::visit( ifStmt );
217        }
218
219        void Resolver::visit( WhileStmt *whileStmt ) {
220                Expression *newExpr = findSingleExpression( whileStmt->get_condition(), *this );
221                delete whileStmt->get_condition();
222                whileStmt->set_condition( newExpr );
223                Visitor::visit( whileStmt );
224        }
225
226        void Resolver::visit( ForStmt *forStmt ) {
227            // SymTab::Indexer::visit( forStmt );
228                Expression *newExpr;
229            // for statements introduce a level of scope
230            enterScope();
231            maybeAccept( forStmt->get_initialization(), *this );
232                if ( forStmt->get_condition() ) {
233                        newExpr = findSingleExpression( forStmt->get_condition(), *this );
234                        delete forStmt->get_condition();
235                        forStmt->set_condition( newExpr );
236                } // if
237 
238                if ( forStmt->get_increment() ) {
239                        newExpr = findVoidExpression( forStmt->get_increment(), *this );
240                        delete forStmt->get_increment();
241                        forStmt->set_increment( newExpr );
242                } // if
243
244            maybeAccept( forStmt->get_condition(), *this );
245            maybeAccept( forStmt->get_increment(), *this );
246            maybeAccept( forStmt->get_body(), *this );
247            leaveScope();
248        }
249
250        template< typename SwitchClass >
251        void handleSwitchStmt( SwitchClass *switchStmt, SymTab::Indexer &visitor ) {
252                Expression *newExpr;
253                newExpr = findIntegralExpression( switchStmt->get_condition(), visitor );
254                delete switchStmt->get_condition();
255                switchStmt->set_condition( newExpr );
256 
257                visitor.Visitor::visit( switchStmt );
258        }
259
260        void Resolver::visit( SwitchStmt *switchStmt ) {
261                handleSwitchStmt( switchStmt, *this );
262        }
263
264        void Resolver::visit( ChooseStmt *switchStmt ) {
265                handleSwitchStmt( switchStmt, *this );
266        }
267
268        void Resolver::visit( CaseStmt *caseStmt ) {
269                Visitor::visit( caseStmt );
270        }
271
272        void Resolver::visit( ReturnStmt *returnStmt ) {
273                if ( returnStmt->get_expr() ) {
274                        CastExpr *castExpr = new CastExpr( returnStmt->get_expr() );
275                        cloneAll( functionReturn, castExpr->get_results() );
276                        Expression *newExpr = findSingleExpression( castExpr, *this );
277                        delete castExpr;
278                        returnStmt->set_expr( newExpr );
279                } // if
280        }
281
282        template< typename T >
283        bool isCharType( T t ) {
284                if ( BasicType * bt = dynamic_cast< BasicType * >( t ) ) {
285                        return bt->get_kind() == BasicType::Char || bt->get_kind() == BasicType::SignedChar || 
286                                bt->get_kind() == BasicType::UnsignedChar;
287                }
288                return false;
289        }
290
291        void Resolver::visit( SingleInit *singleInit ) {
292                if ( singleInit->get_value() ) {
293#if 0
294                        if (NameExpr * ne = dynamic_cast<NameExpr*>(singleInit->get_value())) {
295                                string n = ne->get_name();
296                                if (n == "0") {
297                                        initContext = new BasicType(Type::Qualifiers(),
298                                                                                                BasicType::SignedInt);
299                                } else {
300                                        DeclarationWithType * decl = lookupId(n);
301                                        initContext = decl->get_type();
302                                }
303                        } else if (ConstantExpr * e =
304                                           dynamic_cast<ConstantExpr*>(singleInit->get_value())) {
305                                Constant *c = e->get_constant();
306                                initContext = c->get_type();
307                        } else {
308                                assert(0);
309                        }
310#endif
311                        CastExpr *castExpr = new CastExpr( singleInit->get_value(), initContext->clone() );
312                        Expression *newExpr = findSingleExpression( castExpr, *this );
313                        delete castExpr;
314                        singleInit->set_value( newExpr );
315
316                        // check if initializing type is char[]
317                        if ( ArrayType * at = dynamic_cast< ArrayType * >( initContext ) ) {
318                                if ( isCharType( at->get_base() ) ) {
319                                        // check if the resolved type is char *
320                                        if ( PointerType * pt = dynamic_cast< PointerType *>( newExpr->get_results().front() ) ) {
321                                                if ( isCharType( pt->get_base() ) ) {
322                                                        // strip cast if we're initializing a char[] with a char *, e.g.
323                                                        // char x[] = "hello";
324                                                        CastExpr *ce = dynamic_cast< CastExpr * >( newExpr );
325                                                        singleInit->set_value( ce->get_arg() );
326                                                        ce->set_arg( NULL );
327                                                        delete ce;                                                                     
328                                                }
329                                        }
330                                }
331                        }
332                } // if
333//      singleInit->get_value()->accept( *this );
334        }
335
336        void Resolver::resolveSingleAggrInit( Declaration * dcl, InitIterator & init, InitIterator & initEnd ) {
337                DeclarationWithType * dt = dynamic_cast< DeclarationWithType * >( dcl );
338                assert( dt );
339                initContext = dt->get_type();
340                try {
341                        if ( init == initEnd ) return; // stop when there are no more initializers
342                        (*init)->accept( *this );
343                        ++init; // made it past an initializer
344                } catch( SemanticError & ) {
345                        // need to delve deeper, if you can
346                        if ( StructInstType * sit = dynamic_cast< StructInstType * >( dt->get_type() ) ) {
347                                resolveAggrInit( sit->get_baseStruct(), init, initEnd );
348                        } else if ( UnionInstType * uit = dynamic_cast< UnionInstType * >( dt->get_type() ) ) {
349                                resolveAggrInit( uit->get_baseUnion(), init, initEnd );
350                        } else {
351                                // might need to rethink what is being thrown
352                                throw;
353                        } // if
354                }
355        }
356
357        void Resolver::resolveAggrInit( AggregateDecl * aggr, InitIterator & init, InitIterator & initEnd ) {
358                if ( StructDecl * st = dynamic_cast< StructDecl * >( aggr ) ) {
359                        // want to resolve each initializer to the members of the struct,
360                        // but if there are more initializers than members we should stop
361                        list< Declaration * >::iterator it = st->get_members().begin();
362                        for ( ; it != st->get_members().end(); ++it) {
363                                resolveSingleAggrInit( *it, init, initEnd );
364                        }
365                } else if ( UnionDecl * un = dynamic_cast< UnionDecl * >( aggr ) ) {
366                        // only resolve to the first member of a union
367                        resolveSingleAggrInit( *un->get_members().begin(), init, initEnd );
368                } // if
369        }
370
371        void Resolver::visit( ListInit * listInit ) {
372                InitIterator iter = listInit->begin_initializers();
373                InitIterator end = listInit->end_initializers();
374
375                if ( ArrayType * at = dynamic_cast< ArrayType * >( initContext ) ) {
376                        // resolve each member to the base type of the array
377                        for ( ; iter != end; ++iter ) {
378                                initContext = at->get_base();
379                                (*iter)->accept( *this );
380                        } // for
381                } else if ( StructInstType * st = dynamic_cast< StructInstType * >( initContext ) ) {
382                        resolveAggrInit( st->get_baseStruct(), iter, end );
383                } else if ( UnionInstType *st = dynamic_cast< UnionInstType * >( initContext ) ) {
384                        resolveAggrInit( st->get_baseUnion(), iter, end );
385                } else {
386                        // basic types are handled here
387                        Visitor::visit( listInit );
388                }
389
390#if 0
391                if ( ArrayType *at = dynamic_cast<ArrayType*>(initContext) ) {
392                        std::list<Initializer *>::iterator iter( listInit->begin_initializers() );
393                        for ( ; iter != listInit->end_initializers(); ++iter ) {
394                                initContext = at->get_base();
395                                (*iter)->accept( *this );
396                        } // for
397                } else if ( StructInstType *st = dynamic_cast<StructInstType*>(initContext) ) {
398                        StructDecl *baseStruct = st->get_baseStruct();
399                        std::list<Declaration *>::iterator iter1( baseStruct->get_members().begin() );
400                        std::list<Initializer *>::iterator iter2( listInit->begin_initializers() );
401                        for ( ; iter1 != baseStruct->get_members().end() && iter2 != listInit->end_initializers(); ++iter2 ) {
402                                if ( (*iter2)->get_designators().empty() ) {
403                                        DeclarationWithType *dt = dynamic_cast<DeclarationWithType *>( *iter1 );
404                                        initContext = dt->get_type();
405                                        (*iter2)->accept( *this );
406                                        ++iter1;
407                                } else {
408                                        StructDecl *st = baseStruct;
409                                        iter1 = st->get_members().begin();
410                                        std::list<Expression *>::iterator iter3( (*iter2)->get_designators().begin() );
411                                        for ( ; iter3 != (*iter2)->get_designators().end(); ++iter3 ) {
412                                                NameExpr *key = dynamic_cast<NameExpr *>( *iter3 );
413                                                assert( key );
414                                                for ( ; iter1 != st->get_members().end(); ++iter1 ) {
415                                                        if ( key->get_name() == (*iter1)->get_name() ) {
416                                                                (*iter1)->print( cout );
417                                                                cout << key->get_name() << endl;
418                                                                ObjectDecl *fred = dynamic_cast<ObjectDecl *>( *iter1 );
419                                                                assert( fred );
420                                                                StructInstType *mary = dynamic_cast<StructInstType*>( fred->get_type() );
421                                                                assert( mary );
422                                                                st = mary->get_baseStruct();
423                                                                iter1 = st->get_members().begin();
424                                                                break;
425                                                        } // if
426                                                }  // for
427                                        } // for
428                                        ObjectDecl *fred = dynamic_cast<ObjectDecl *>( *iter1 );
429                                        assert( fred );
430                                        initContext = fred->get_type();
431                                        (*listInit->begin_initializers())->accept( *this );
432                                } // if
433                        } // for
434                } else if ( UnionInstType *st = dynamic_cast<UnionInstType*>(initContext) ) {
435                        DeclarationWithType *dt = dynamic_cast<DeclarationWithType *>( *st->get_baseUnion()->get_members().begin() );
436                        initContext = dt->get_type();
437                        (*listInit->begin_initializers())->accept( *this );
438                } // if
439#endif
440        }
441} // namespace ResolvExpr
442
443// Local Variables: //
444// tab-width: 4 //
445// mode: c++ //
446// compile-command: "make install" //
447// End: //
Note: See TracBrowser for help on using the repository browser.