Changeset d3e4d6c for src/GenPoly/InstantiateGeneric.cc

Timestamp:

Aug 23, 2017, 6:22:07 PM (8 years ago)

Author:

Peter A. Buhr <pabuhr@…>

Branches:

ADT, aaron-thesis, arm-eh, ast-experimental, cleanup-dtors, deferred_resn, demangler, enum, forall-pointer-decay, jacob/cs343-translation, jenkins-sandbox, master, new-ast, new-ast-unique-expr, new-env, no_list, persistent-indexer, pthread-emulation, qualifiedEnum, resolv-new, with_gc

Children:

87e08e24, cb811ac

Parents:

9f07232 (diff), bd37119 (diff)
Note: this is a merge changeset, the changes displayed below correspond to the merge itself.
Use the (diff) links above to see all the changes relative to each parent.

Message:

Merge branch 'master' of plg2:software/cfa/cfa-cc

File:

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

src/GenPoly/InstantiateGeneric.cc

@@ -162 +162 @@
                 /// Should not make use of type environment to replace types of function parameter and return values.
                 bool inFunctionType = false;
+                /// Index of current member, used to recreate MemberExprs with the member from an instantiation
+                int memberIndex = -1;
                 GenericInstantiator() : instantiations(), dtypeStatics(), typeNamer("_conc_") {}
 
@@ -167 +169 @@
                 Type* postmutate( UnionInstType *inst );
 
-                void premutate( __attribute__((unused)) FunctionType * ftype ) {
+                // fix MemberExprs to use the member from the instantiation
+                void premutate( MemberExpr * memberExpr );
+                Expression * postmutate( MemberExpr * memberExpr );
+
+                void premutate( FunctionType * ) {
                         GuardValue( inFunctionType );
                         inFunctionType = true;
@@ -414 +420 @@
         }
 
+        namespace {
+                bool isGenericType( Type * t ) {
+                        if ( StructInstType * inst = dynamic_cast< StructInstType * >( t ) ) {
+                                return ! inst->parameters.empty();
+                        } else if ( UnionInstType * inst = dynamic_cast< UnionInstType * >( t ) ) {
+                                return ! inst->parameters.empty();
+                        }
+                        return false;
+                }
+
+                AggregateDecl * getAggr( Type * t ) {
+                        if ( StructInstType * inst = dynamic_cast< StructInstType * >( t ) ) {
+                                return inst->baseStruct;
+                        } else if ( UnionInstType * inst = dynamic_cast< UnionInstType * >( t ) ) {
+                                return inst->baseUnion;
+                        }
+                        assertf( false, "Non-aggregate type: %s", toString( t ).c_str() );
+                }
+        }
+
+        void GenericInstantiator::premutate( MemberExpr * memberExpr ) {
+                GuardValue( memberIndex );
+                memberIndex = -1;
+                if ( isGenericType( memberExpr->aggregate->result ) ) {
+                        // find the location of the member
+                        AggregateDecl * aggr = getAggr( memberExpr->aggregate->result );
+                        std::list< Declaration * > & members = aggr->members;
+                        memberIndex = std::distance( members.begin(), std::find( members.begin(), members.end(), memberExpr->member ) );
+                        assertf( memberIndex < (int)members.size(), "Could not find member %s in generic type %s", toString( memberExpr->member ).c_str(), toString( memberExpr->aggregate ).c_str() );
+                }
+        }
+
+        Expression * GenericInstantiator::postmutate( MemberExpr * memberExpr ) {
+                if ( memberIndex != -1 ) {
+                        // using the location from the generic type, find the member in the instantiation and rebuild the member expression
+                        AggregateDecl * aggr = getAggr( memberExpr->aggregate->result );
+                        assertf( memberIndex < (int)aggr->members.size(), "Instantiation somehow has fewer members than the generic type." );
+                        Declaration * member = *std::next( aggr->members.begin(), memberIndex );
+                        assertf( member->name == memberExpr->member->name, "Instantiation has different member order than the generic type. %s / %s", toString( member ).c_str(), toString( memberExpr->member ).c_str() );
+                        DeclarationWithType * field = safe_dynamic_cast< DeclarationWithType * >( member );
+                        MemberExpr * ret = new MemberExpr( field, memberExpr->aggregate->clone() );
+                        std::swap( ret->env, memberExpr->env );
+                        delete memberExpr;
+                        return ret;
+                }
+                return memberExpr;
+        }
+
         void GenericInstantiator::beginScope() {
                 instantiations.beginScope();