Changes in src/SymTab/Validate.cc [2bf9c37:b128d3e]

File:

• src/SymTab/Validate.cc (modified) (7 diffs)

src/SymTab/Validate.cc

@@ -127 +127 @@
           public:
                 LinkReferenceToTypes( bool doDebug, const Indexer *indexer );
-                using Parent::visit;
-                void visit( TypeInstType *typeInst ) final;
-
+                using Parent::visit;
                 void visit( EnumInstType *enumInst ) final;
                 void visit( StructInstType *structInst ) final;
                 void visit( UnionInstType *unionInst ) final;
-                void visit( TraitInstType *traitInst ) final;
-
+                void visit( TraitInstType *contextInst ) final;
                 void visit( EnumDecl *enumDecl ) final;
                 void visit( StructDecl *structDecl ) final;
                 void visit( UnionDecl *unionDecl ) final;
-                void visit( TraitDecl * traitDecl ) final;
-
+                void visit( TypeInstType *typeInst ) final;
           private:
                 const Indexer *indexer;
@@ -291 +287 @@
 
         HoistStruct::HoistStruct() : inStruct( false ) {
+        }
+
+        void filter( std::list< Declaration * > &declList, bool (*pred)( Declaration * ), bool doDelete ) {
+                std::list< Declaration * >::iterator i = declList.begin();
+                while ( i != declList.end() ) {
+                        std::list< Declaration * >::iterator next = i;
+                        ++next;
+                        if ( pred( *i ) ) {
+                                if ( doDelete ) {
+                                        delete *i;
+                                } // if
+                                declList.erase( i );
+                        } // if
+                        i = next;
+                } // while
         }
 
@@ -442 +453 @@
         }
 
-        template< typename Decl >
-        void normalizeAssertions( std::list< Decl * > & assertions ) {
-                // ensure no duplicate trait members after the clone
-                auto pred = [](Decl * d1, Decl * d2) {
-                        // only care if they're equal
-                        DeclarationWithType * dwt1 = dynamic_cast<DeclarationWithType *>( d1 );
-                        DeclarationWithType * dwt2 = dynamic_cast<DeclarationWithType *>( d2 );
-                        if ( dwt1 && dwt2 ) {
-                                if ( dwt1->get_name() == dwt2->get_name() && ResolvExpr::typesCompatible( dwt1->get_type(), dwt2->get_type(), SymTab::Indexer() ) ) {
-                                        // std::cerr << "=========== equal:" << std::endl;
-                                        // std::cerr << "d1: " << d1 << std::endl;
-                                        // std::cerr << "d2: " << d2 << std::endl;
-                                        return false;
-                                }
+        void LinkReferenceToTypes::visit( TraitInstType *traitInst ) {
+                Parent::visit( traitInst );
+                if ( traitInst->get_name() == "sized" ) {
+                        // "sized" is a special trait with no members - just flick the sized status on for the type variable
+                        if ( traitInst->get_parameters().size() != 1 ) {
+                                throw SemanticError( "incorrect number of trait parameters: ", traitInst );
                         }
-                        return d1 < d2;
-                };
-                std::set<Decl *, decltype(pred)> unique_members( assertions.begin(), assertions.end(), pred );
-                // if ( unique_members.size() != assertions.size() ) {
-                //      std::cerr << "============different" << std::endl;
-                //      std::cerr << unique_members.size() << " " << assertions.size() << std::endl;
-                // }
-
-                std::list< Decl * > order;
-                order.splice( order.end(), assertions );
-                std::copy_if( order.begin(), order.end(), back_inserter( assertions ), [&]( Decl * decl ) {
-                        return unique_members.count( decl );
-                });
-        }
-
-        // expand assertions from trait instance, performing the appropriate type variable substitutions
-        template< typename Iterator >
-        void expandAssertions( TraitInstType * inst, Iterator out ) {
-                assertf( inst->baseTrait, "Trait instance not linked to base trait: %s", toString( inst ).c_str() );
-                std::list< DeclarationWithType * > asserts;
-                for ( Declaration * decl : inst->baseTrait->members ) {
-                        asserts.push_back( safe_dynamic_cast<DeclarationWithType *>( decl->clone() ) );
-                }
-                // substitute trait decl parameters for instance parameters
-                applySubstitution( inst->baseTrait->parameters.begin(), inst->baseTrait->parameters.end(), inst->parameters.begin(), asserts.begin(), asserts.end(), out );
-        }
-
-        void LinkReferenceToTypes::visit( TraitDecl * traitDecl ) {
-                Parent::visit( traitDecl );
-
-                if ( traitDecl->name == "sized" ) {
-                        // "sized" is a special trait - flick the sized status on for the type variable
-                        assertf( traitDecl->parameters.size() == 1, "Built-in trait 'sized' has incorrect number of parameters: %zd", traitDecl->parameters.size() );
-                        TypeDecl * td = traitDecl->parameters.front();
-                        td->set_sized( true );
-                }
-
-                // move assertions from type parameters into the body of the trait
-                for ( TypeDecl * td : traitDecl->parameters ) {
-                        for ( DeclarationWithType * assert : td->assertions ) {
-                                if ( TraitInstType * inst = dynamic_cast< TraitInstType * >( assert->get_type() ) ) {
-                                        expandAssertions( inst, back_inserter( traitDecl->members ) );
-                                } else {
-                                        traitDecl->members.push_back( assert->clone() );
-                                }
-                        }
-                        deleteAll( td->assertions );
-                        td->assertions.clear();
-                } // for
-        }
-
-        void LinkReferenceToTypes::visit( TraitInstType * traitInst ) {
-                Parent::visit( traitInst );
+                        TypeExpr * param = safe_dynamic_cast< TypeExpr * > ( traitInst->get_parameters().front() );
+                        TypeInstType * inst = safe_dynamic_cast< TypeInstType * > ( param->get_type() );
+                        TypeDecl * decl = inst->get_baseType();
+                        decl->set_sized( true );
+                        // since "sized" is special, the next few steps don't apply
+                        return;
+                }
+
                 // handle other traits
-                TraitDecl *traitDecl = indexer->lookupTrait( traitInst->name );
+                TraitDecl *traitDecl = indexer->lookupTrait( traitInst->get_name() );
                 if ( ! traitDecl ) {
-                        throw SemanticError( "use of undeclared trait " + traitInst->name );
+                        throw SemanticError( "use of undeclared trait " + traitInst->get_name() );
                 } // if
                 if ( traitDecl->get_parameters().size() != traitInst->get_parameters().size() ) {
                         throw SemanticError( "incorrect number of trait parameters: ", traitInst );
                 } // if
-                traitInst->baseTrait = traitDecl;
+
+                for ( TypeDecl * td : traitDecl->get_parameters() ) {
+                        for ( DeclarationWithType * assert : td->get_assertions() ) {
+                                traitInst->get_members().push_back( assert->clone() );
+                        } // for
+                } // for
+
+                // need to clone members of the trait for ownership purposes
+                std::list< Declaration * > members;
+                std::transform( traitDecl->get_members().begin(), traitDecl->get_members().end(), back_inserter( members ), [](Declaration * dwt) { return dwt->clone(); } );
+
+                applySubstitution( traitDecl->get_parameters().begin(), traitDecl->get_parameters().end(), traitInst->get_parameters().begin(), members.begin(), members.end(), back_inserter( traitInst->get_members() ) );
 
                 // need to carry over the 'sized' status of each decl in the instance
@@ -529 +498 @@
                         }
                 }
-                // normalizeAssertions( traitInst->members );
         }
 
@@ -593 +561 @@
         void forallFixer( Type * func ) {
                 for ( TypeDecl * type : func->get_forall() ) {
-                        std::list< DeclarationWithType * > asserts;
-                        asserts.splice( asserts.end(), type->assertions );
-                        // expand trait instances into their members
-                        for ( DeclarationWithType * assertion : asserts ) {
-                                if ( TraitInstType *traitInst = dynamic_cast< TraitInstType * >( assertion->get_type() ) ) {
-                                        // expand trait instance into all of its members
-                                        expandAssertions( traitInst, back_inserter( type->assertions ) );
-                                        delete traitInst;
-                                } else {
-                                        // pass other assertions through
-                                        type->assertions.push_back( assertion );
-                                } // if
-                        } // for
-                        // apply FixFunction to every assertion to check for invalid void type
-                        for ( DeclarationWithType *& assertion : type->assertions ) {
-                                FixFunction fixer;
-                                assertion = assertion->acceptMutator( fixer );
-                                if ( fixer.get_isVoid() ) {
-                                        throw SemanticError( "invalid type void in assertion of function ", func );
-                                } // if
-                        } // for
-                        // normalizeAssertions( type->assertions );
+                        std::list< DeclarationWithType * > toBeDone, nextRound;
+                        toBeDone.splice( toBeDone.end(), type->get_assertions() );
+                        while ( ! toBeDone.empty() ) {
+                                for ( DeclarationWithType * assertion : toBeDone ) {
+                                        if ( TraitInstType *traitInst = dynamic_cast< TraitInstType * >( assertion->get_type() ) ) {
+                                                // expand trait instance into all of its members
+                                                for ( Declaration * member : traitInst->get_members() ) {
+                                                        DeclarationWithType *dwt = safe_dynamic_cast< DeclarationWithType * >( member );
+                                                        nextRound.push_back( dwt->clone() );
+                                                }
+                                                delete traitInst;
+                                        } else {
+                                                // pass assertion through
+                                                FixFunction fixer;
+                                                assertion = assertion->acceptMutator( fixer );
+                                                if ( fixer.get_isVoid() ) {
+                                                        throw SemanticError( "invalid type void in assertion of function ", func );
+                                                }
+                                                type->get_assertions().push_back( assertion );
+                                        } // if
+                                } // for
+                                toBeDone.clear();
+                                toBeDone.splice( toBeDone.end(), nextRound );
+                        } // while
                 } // for
         }
@@ -782 +752 @@
                 CompoundStmt *ret = Mutator::mutate( compoundStmt );
                 scopeLevel -= 1;
-                // remove and delete decl stmts
-                filter( compoundStmt->kids, [](Statement * stmt) {
-                        if ( DeclStmt *declStmt = dynamic_cast< DeclStmt * >( stmt ) ) {
+                std::list< Statement * >::iterator i = compoundStmt->get_kids().begin();
+                while ( i != compoundStmt->get_kids().end() ) {
+                        std::list< Statement * >::iterator next = i+1;
+                        if ( DeclStmt *declStmt = dynamic_cast< DeclStmt * >( *i ) ) {
                                 if ( dynamic_cast< TypedefDecl * >( declStmt->get_decl() ) ) {
-                                        return true;
+                                        delete *i;
+                                        compoundStmt->get_kids().erase( i );
                                 } // if
                         } // if
-                        return false;
-                }, true);
+                        i = next;
+                } // while
                 typedefNames.endScope();
                 return ret;
@@ -799 +771 @@
         template<typename AggDecl>
         AggDecl *EliminateTypedef::handleAggregate( AggDecl * aggDecl ) {
-                filter( aggDecl->members, isTypedef, true );
+                std::list<Declaration *>::iterator it = aggDecl->get_members().begin();
+                for ( ; it != aggDecl->get_members().end(); ) {
+                        std::list< Declaration * >::iterator next = it+1;
+                        if ( dynamic_cast< TypedefDecl * >( *it ) ) {
+                                delete *it;
+                                aggDecl->get_members().erase( it );
+                        } // if
+                        it = next;
+                }
                 return aggDecl;
         }