Changes in src/GenPoly/InstantiateGeneric.cc [5bf4712:ed94eac]

File:

• src/GenPoly/InstantiateGeneric.cc (modified) (13 diffs)

src/GenPoly/InstantiateGeneric.cc

@@ -5 +5 @@
 // file "LICENCE" distributed with Cforall.
 //
-// InstantiateGeneric.cc --
+// InstantiateGeneric.h --
 //
 // Author           : Aaron B. Moss
@@ -21 +21 @@
 
 #include "InstantiateGeneric.h"
-#include "DeclMutator.h"
+#include "PolyMutator.h"
 
 #include "ResolvExpr/typeops.h"
@@ -44 +44 @@
                 ConcreteType(const ConcreteType& that) : base(that.base), params() { cloneAll(that.params, params); }
 
-                /// Extracts types from a list of TypeExpr*
-                ConcreteType(AggregateDecl *_base, const std::list< TypeExpr* >& _params) : base(_base), params() {
-                        for ( std::list< TypeExpr* >::const_iterator param = _params.begin(); param != _params.end(); ++param ) {
-                                params.push_back( (*param)->get_type()->clone() );
+                /// Extracts types from a list of Expression* (which should be TypeExpr*)
+                ConcreteType(AggregateDecl *_base, const std::list< Expression* >& _params) : base(_base), params() {
+                        for ( std::list< Expression* >::const_iterator it = _params.begin(); it != _params.end(); ++it ) {
+                                TypeExpr *param = dynamic_cast< TypeExpr* >(*it);
+                                assert(param && "Aggregate parameters should be type expressions");
+                                params.push_back( param->get_type()->clone() );
                         }
                 }
@@ -64 +66 @@
 
                 bool operator== (const ConcreteType& that) const {
+                        SymTab::Indexer dummy;
+
                         if ( base != that.base ) return false;
-
-                        SymTab::Indexer dummy;
                         if ( params.size() != that.params.size() ) return false;
                         for ( std::list< Type* >::const_iterator it = params.begin(), jt = that.params.begin(); it != params.end(); ++it, ++jt ) {
@@ -80 +82 @@
         /// Maps a concrete type to the instantiated struct type, accounting for scope
         class InstantiationMap {
-                /// Instantiation of a generic type, with key information to find it
-                struct Instantiation {
-                        ConcreteType key;     ///< Instantiation parameters for this type
-                        AggregateDecl *decl;  ///< Declaration of the instantiated generic type
-
-                        Instantiation() : key(), decl(0) {}
-                        Instantiation(const ConcreteType &_key, AggregateDecl *_decl) : key(_key), decl(_decl) {}
-                };
+                /// Pair of concrete type and declaration that instantiates it
+                typedef std::pair< ConcreteType, AggregateDecl* > Instantiation;
                 /// Map of generic types to instantiations of them
                 typedef std::map< AggregateDecl*, std::vector< Instantiation > > Scope;
@@ -111 +107 @@
                 /// Gets the declaration for the concrete instantiation of this type, assuming it has already been instantiated in the current scope.
                 /// Returns NULL on none such.
-                AggregateDecl* lookup( AggregateDecl *generic, const std::list< TypeExpr* >& params ) {
+                AggregateDecl* lookup( AggregateDecl *generic, std::list< Expression* >& params ) {
                         ConcreteType key(generic, params);
                         // scan scopes from innermost out
@@ -120 +116 @@
                                 // look through instantiations for matches to concrete type
                                 for ( std::vector< Instantiation >::const_iterator inst = insts->second.begin(); inst != insts->second.end(); ++inst ) {
-                                        if ( inst->key == key ) return inst->decl;
+                                        if ( inst->first == key ) return inst->second;
                                 }
                         }
@@ -127 +123 @@
                 }
         public:
-                StructDecl* lookup( StructInstType *inst, const std::list< TypeExpr* > &typeSubs ) { return (StructDecl*)lookup( inst->get_baseStruct(), typeSubs ); }
-                UnionDecl* lookup( UnionInstType *inst, const std::list< TypeExpr* > &typeSubs ) { return (UnionDecl*)lookup( inst->get_baseUnion(), typeSubs ); }
+                StructDecl* lookup( StructInstType *inst ) { return (StructDecl*)lookup( inst->get_baseStruct(), inst->get_parameters() ); }
+                UnionDecl* lookup( UnionInstType *inst ) { return (UnionDecl*)lookup( inst->get_baseUnion(), inst->get_parameters() ); }
 
         private:
                 /// Adds a declaration for a concrete type to the current scope
-                void insert( AggregateDecl *generic, const std::list< TypeExpr* > &params, AggregateDecl *decl ) {
+                void insert( AggregateDecl *generic, std::list< Expression* >& params, AggregateDecl *decl ) {
                         ConcreteType key(generic, params);
-                        scopes.back()[generic].push_back( Instantiation( key, decl ) );
-                }
-        public:
-                void insert( StructInstType *inst, const std::list< TypeExpr* > &typeSubs, StructDecl *decl ) { insert( inst->get_baseStruct(), typeSubs, decl ); }
-                void insert( UnionInstType *inst, const std::list< TypeExpr* > &typeSubs, UnionDecl *decl ) { insert( inst->get_baseUnion(), typeSubs, decl ); }
+                        scopes.back()[generic].push_back( std::make_pair( key, decl ) );
+                }
+        public:
+                void insert( StructInstType *inst, StructDecl *decl ) { insert( inst->get_baseStruct(), inst->get_parameters(), decl ); }
+                void insert( UnionInstType *inst, UnionDecl *decl ) { insert( inst->get_baseUnion(), inst->get_parameters(), decl ); }
         };
 
         /// Mutator pass that replaces concrete instantiations of generic types with actual struct declarations, scoped appropriately
-        class Instantiate : public DeclMutator {
+        class Instantiate : public PolyMutator {
                 InstantiationMap instantiations;
                 UniqueName typeNamer;
 
         public:
-                Instantiate() : DeclMutator(), instantiations(), typeNamer("_conc_") {}
-
-//              virtual Declaration* mutate( StructDecl *aggregateDecl );
-//              virtual Declaration* mutate( UnionDecl *aggregateDecl );
-
+                Instantiate() : instantiations(), typeNamer("_conc_") {}
+
+                /// Mutates the whole translation unit, inserting new struct declarations as appropriate
+                void mutateAll( std::list< Declaration* >& translationUnit );
+                
                 virtual Type* mutate( StructInstType *inst );
                 virtual Type* mutate( UnionInstType *inst );
@@ -157 +153 @@
                 virtual void doBeginScope();
                 virtual void doEndScope();
+
+        private:
+                /// Adds a declaration to the current environment and the statements to add
+                void addDeclaration( AggregateDecl *decl ) {
+                        std::list< Label > nolabels;
+                        DeclStmt *stmt = new DeclStmt( nolabels, decl );
+                        PolyMutator::stmtsToAdd.push_back( stmt );
+                }
         };
         
         void instantiateGeneric( std::list< Declaration* >& translationUnit ) {
                 Instantiate instantiator;
-                instantiator.mutateDeclarationList( translationUnit );
-        }
-
-        /// Makes substitutions of params into baseParams; returns true if all parameters substituted for a concrete type
-        bool makeSubstitutions( const std::list< TypeDecl* >& baseParams, const std::list< Expression* >& params, std::list< TypeExpr* >& out ) {
-                bool allConcrete = true;  // will finish the substitution list even if they're not all concrete
-
-                // substitute concrete types for given parameters, and incomplete types for placeholders
-                std::list< TypeDecl* >::const_iterator baseParam = baseParams.begin();
-                std::list< Expression* >::const_iterator param = params.begin();
-                for ( ; baseParam != baseParams.end() && param != params.end(); ++baseParam, ++param ) {
-                        switch ( (*baseParam)->get_kind() ) {
-                        case TypeDecl::Any: {   // any type is a valid substitution here; complete types can be used to instantiate generics
-                                TypeExpr *paramType = dynamic_cast< TypeExpr* >( *param );
-                                assert(paramType && "Aggregate parameters should be type expressions");
-                                out.push_back( paramType->clone() );
-                                // check that the substituted type isn't a type variable itself
-                                if ( dynamic_cast< TypeInstType* >( paramType->get_type() ) ) {
-                                        allConcrete = false;
-                                }
-                                break;
-                        }
-                        case TypeDecl::Dtype:  // dtype can be consistently replaced with void [only pointers, which become void*]
-                                out.push_back( new TypeExpr( new VoidType( Type::Qualifiers() ) ) );
-                                break;
-                        case TypeDecl::Ftype:  // pointer-to-ftype can be consistently replaced with void (*)(void) [similar to dtype]
-                                out.push_back( new TypeExpr( new FunctionType( Type::Qualifiers(), false ) ) );
-                                break;
-                        }
-                }
-
-                // if not enough parameters given, substitute remaining incomplete types for placeholders
-                for ( ; baseParam != baseParams.end(); ++baseParam ) {
-                        switch ( (*baseParam)->get_kind() ) {
-                        case TypeDecl::Any:    // no more substitutions here, fail early
-                                return false;
-                        case TypeDecl::Dtype:  // dtype can be consistently replaced with void [only pointers, which become void*]
-                                out.push_back( new TypeExpr( new VoidType( Type::Qualifiers() ) ) );
-                                break;
-                        case TypeDecl::Ftype:  // pointer-to-ftype can be consistently replaced with void (*)(void) [similar to dtype]
-                                out.push_back( new TypeExpr( new FunctionType( Type::Qualifiers(), false ) ) );
-                                break;
-                        }
-                }
-
-                return allConcrete;
-        }
-        
-        /// Substitutes types of members of in according to baseParams => typeSubs, appending the result to out
-        void substituteMembers( const std::list< Declaration* >& in, const std::list< TypeDecl* >& baseParams, const std::list< TypeExpr* >& typeSubs, 
+//              mutateAll( translationUnit, instantiator );
+                instantiator.mutateAll( translationUnit );
+        }
+
+        void Instantiate::mutateAll( std::list< Declaration* >& translationUnit ) {
+                // below copied and modified from Mutator.h:mutateAll()
+                SemanticError errors;
+                for ( std::list< Declaration* >::iterator decl = translationUnit.begin(); decl != translationUnit.end(); ++decl ) {
+                        try {
+                                if ( *decl ) {
+                                        *decl = dynamic_cast< Declaration* >( (*decl)->acceptMutator( *this ) );
+                                        assert( *decl );
+                                        // account for missing top-level declarations
+                                        for ( std::list< Statement* >::const_iterator stmt = PolyMutator::stmtsToAdd.begin(); stmt != PolyMutator::stmtsToAdd.end(); ++stmt ) {
+                                                DeclStmt *declStmt = dynamic_cast< DeclStmt* >( *stmt );
+                                                assert( declStmt );
+                                                translationUnit.insert( decl, declStmt->get_decl() );
+                                        }
+                                        PolyMutator::stmtsToAdd.clear();
+                                } // if
+                        } catch( SemanticError &e ) {
+                                errors.append( e );
+                        } // try
+                } // for
+                if ( ! errors.isEmpty() ) {
+                        throw errors;
+                } // if
+        }
+
+        /// Substitutes types of members of in according to baseParams => params, appending the result to out
+        void substituteMembers( const std::list< Declaration* >& in,
+                                                        const std::list< TypeDecl * >& baseParams, const std::list< Expression* >& params,
                                                     std::list< Declaration* >& out ) {
                 // substitute types into new members
-                TypeSubstitution subs( baseParams.begin(), baseParams.end(), typeSubs.begin() );
+                TypeSubstitution subs( baseParams.begin(), baseParams.end(), params.begin() );
                 for ( std::list< Declaration* >::const_iterator member = in.begin(); member != in.end(); ++member ) {
                         Declaration *newMember = (*member)->clone();
@@ -229 +215 @@
                 // exit early if no need for further mutation
                 if ( inst->get_parameters().empty() ) return inst;
-                assert( inst->get_baseParameters() && "Base struct has parameters" );
-
-                // check if type can be concretely instantiated; put substitutions into typeSubs
-                std::list< TypeExpr* > typeSubs;
-                if ( ! makeSubstitutions( *inst->get_baseParameters(), inst->get_parameters(), typeSubs ) ) {
-                        deleteAll( typeSubs );
-                        return inst;
-                }
                 
                 // make concrete instantiation of generic type
-                StructDecl *concDecl = instantiations.lookup( inst, typeSubs );
+                StructDecl *concDecl = instantiations.lookup( inst );
                 if ( ! concDecl ) {
+                        assert( inst->get_baseParameters() && "Base struct has parameters" );
                         // set concDecl to new type, insert type declaration into statements to add
                         concDecl = new StructDecl( typeNamer.newName( inst->get_name() ) );
-                        substituteMembers( inst->get_baseStruct()->get_members(), *inst->get_baseParameters(), typeSubs,        concDecl->get_members() );
-                        DeclMutator::addDeclaration( concDecl );
-                        instantiations.insert( inst, typeSubs, concDecl );
+                        substituteMembers( inst->get_baseStruct()->get_members(),
+                                                                *inst->get_baseParameters(), inst->get_parameters(),
+                                                                concDecl->get_members() );
+                        addDeclaration( concDecl );
+                        instantiations.insert( inst, concDecl );
                 }
                 StructInstType *newInst = new StructInstType( inst->get_qualifiers(), concDecl->get_name() );
                 newInst->set_baseStruct( concDecl );
-
-                deleteAll( typeSubs );
                 delete inst;
                 return newInst;
@@ -263 +242 @@
                 // exit early if no need for further mutation
                 if ( inst->get_parameters().empty() ) return inst;
-                assert( inst->get_baseParameters() && "Base union has parameters" );
-
-                // check if type can be concretely instantiated; put substitutions into typeSubs
-                std::list< TypeExpr* > typeSubs;
-                if ( ! makeSubstitutions( *inst->get_baseParameters(), inst->get_parameters(), typeSubs ) ) {
-                        deleteAll( typeSubs );
-                        return inst;
-                }
-                
+
                 // make concrete instantiation of generic type
-                UnionDecl *concDecl = instantiations.lookup( inst, typeSubs );
+                UnionDecl *concDecl = instantiations.lookup( inst );
                 if ( ! concDecl ) {
                         // set concDecl to new type, insert type declaration into statements to add
+                        assert( inst->get_baseParameters() && "Base union has parameters" );
                         concDecl = new UnionDecl( typeNamer.newName( inst->get_name() ) );
-                        substituteMembers( inst->get_baseUnion()->get_members(), *inst->get_baseParameters(), typeSubs, concDecl->get_members() );
-                        DeclMutator::addDeclaration( concDecl );
-                        instantiations.insert( inst, typeSubs, concDecl );
+                        substituteMembers( inst->get_baseUnion()->get_members(),
+                                                                *inst->get_baseParameters(), inst->get_parameters(),
+                                                                concDecl->get_members() );
+                        addDeclaration( concDecl );
+                        instantiations.insert( inst, concDecl );
                 }
                 UnionInstType *newInst = new UnionInstType( inst->get_qualifiers(), concDecl->get_name() );
                 newInst->set_baseUnion( concDecl );
-
-                deleteAll( typeSubs );
                 delete inst;
                 return newInst;
@@ -290 +262 @@
         
         void Instantiate::doBeginScope() {
-                DeclMutator::doBeginScope();
                 // push a new concrete type scope
                 instantiations.beginScope();
@@ -296 +267 @@
 
         void Instantiate::doEndScope() {
-                DeclMutator::doEndScope();
                 // pop the last concrete type scope
                 instantiations.endScope();