Changes in / [bead1cf:c018b85]

Location:

src

Files:

• ResolvExpr/ResolveAssertions.cc (modified) (13 diffs)
• SymTab/Mangler.cc (modified) (6 diffs)
• SymTab/Mangler.h (modified) (2 diffs)

src/ResolvExpr/ResolveAssertions.cc

@@ -20 +20 @@
 #include <list>                     // for list
 #include <memory>                   // for unique_ptr
+#include <string>
 #include <unordered_map>            // for unordered_map, unordered_multimap
 #include <utility>                  // for move
@@ -55 +56 @@
         using CandidateList = std::vector<AssnCandidate>;
 
+        /// Unique identifier for a yet-to-be-resolved assertion
+        struct AssnId {
+                DeclarationWithType* decl;  ///< Declaration of assertion
+                AssertionSetValue info;     ///< Information about assertion
+
+                AssnId(DeclarationWithType* decl, const AssertionSetValue& info) : decl(decl), info(info) {}
+        };
+
+        /// Cached assertion items
+        struct AssnCacheItem {
+                CandidateList matches;         ///< Possible matches for this assertion
+                std::vector<AssnId> deferIds;  ///< Deferred assertions which resolve to this item
+
+                AssnCacheItem( CandidateList&& m ) : matches(std::move(m)), deferIds() {}
+        };
+
+        /// Cache of resolved assertions
+        using AssnCache = std::unordered_map<std::string, AssnCacheItem>;
+
         /// Reference to single deferred item
         struct DeferRef {
-                const DeclarationWithType* decl;
-                const AssertionSetValue& info;
+                const AssnCacheItem& item;
                 const AssnCandidate& match;
         };
@@ -65 +84 @@
         /// Acts like indexed list of DeferRef
         struct DeferItem {
-                DeclarationWithType* decl;
-                AssertionSetValue info;
-                CandidateList matches;
-
-                DeferItem( DeclarationWithType* decl, const AssertionSetValue& info, 
-                        CandidateList&& matches )
-                : decl(decl), info(info), matches(std::move(matches)) {}
-
-                bool empty() const { return matches.empty(); }
-
-                CandidateList::size_type size() const { return matches.size(); }
-
-                DeferRef operator[] ( unsigned i ) const { return { decl, info, matches[i] }; }
+                const AssnCache* cache;     ///< Cache storing assertion item
+                std::string key;            ///< Key into cache
+                
+                DeferItem( const AssnCache& cache, const std::string& key ) : cache(&cache), key(key) {}
+
+                bool empty() const { return cache->at(key).matches.empty(); }
+
+                CandidateList::size_type size() const { return cache->at(key).matches.size(); }
+
+                DeferRef operator[] ( unsigned i ) const {
+                        const AssnCacheItem& item = cache->at(key);
+                        return { item, item.matches[i] };
+                }
+
+                // sortable by key
+                // TODO look into optimizing combination process with other sort orders (e.g. by number 
+                // of matches in candidate)
+                bool operator< ( const DeferItem& o ) const { return key < o.key; }
+                bool operator== ( const DeferItem& o ) const { return key == o.key; }
         };
 
@@ -152 +177 @@
                                 for ( const auto& assn : x.assns ) {
                                         k += computeConversionCost( 
-                                                assn.match.adjType, assn.decl->get_type(), indexer, x.env );
+                                                assn.match.adjType, assn.item.deferIds[0].decl->get_type(), indexer, 
+                                                x.env );
                                 }
                                 it = cache.emplace_hint( it, &x, k );
@@ -208 +234 @@
                                 candidate->get_uniqueId(), match.adjType->clone(), decl->get_type()->clone(), 
                                 varExpr };
-
-                // // follow the current assertion's ID chain to find the correct set of inferred parameters 
-                // // to add the candidate o (i.e. the set of inferred parameters belonging to the entity 
-                // // which requested the assertion parameter)
-                // InferredParams* inferParams = &alt.expr->inferParams;
-                // for ( UniqueId id : info.idChain ) {
-                //      inferParams = (*inferParams)[ id ].inferParams.get();
-                // }
-
-                // (*inferParams)[ decl->get_uniqueId() ] = ParamEntry{
-                //              candidate->get_uniqueId(), match.adjType, decl->get_type()->clone(), varExpr };
         }
 
         /// Adds a captured assertion to the symbol table
         void addToIndexer( AssertionSet &assertSet, SymTab::Indexer &indexer ) {
-                for ( AssertionSet::iterator i = assertSet.begin(); i != assertSet.end(); ++i ) {
-                        if ( i->second.isUsed ) {
-                                indexer.addId( i->first );
+                for ( auto&  i : assertSet ) {
+                        if ( i.second.isUsed ) {
+                                indexer.addId( i.first );
                         }
                 }
@@ -234 +249 @@
 
         /// Resolve a single assertion, in context
-        bool resolveAssertion( AssertionItem& assn, ResnState& resn ) {
+        bool resolveAssertion( AssertionItem& assn, ResnState& resn, AssnCache& cache ) {
                 // skip unused assertions
                 if ( ! assn.info.isUsed ) return true;
 
-                // lookup candidates for this assertion
-                std::list< SymTab::Indexer::IdData > candidates;
-                resn.indexer.lookupId( assn.decl->name, candidates );
-
-                // find the candidates that unify with the desired type
-                CandidateList matches;
-                for ( const auto& cdata : candidates ) {
-                        DeclarationWithType* candidate = cdata.id;
-
-                        // build independent unification context for candidate
-                        AssertionSet have, newNeed;
-                        TypeEnvironment newEnv{ resn.alt.env };
-                        OpenVarSet newOpenVars{ resn.alt.openVars };
-                        Type* adjType = candidate->get_type()->clone();
-                        adjustExprType( adjType, newEnv, resn.indexer );
-                        renameTyVars( adjType );
-
-                        // keep unifying candidates
-                        if ( unify( assn.decl->get_type(), adjType, newEnv, newNeed, have, newOpenVars, 
-                                        resn.indexer ) ) {
-                                // set up binding slot for recursive assertions
-                                UniqueId crntResnSlot = 0;
-                                if ( ! newNeed.empty() ) {
-                                        crntResnSlot = ++globalResnSlot;
-                                        for ( auto& a : newNeed ) {
-                                                a.second.resnSlot = crntResnSlot;
+                // check cache for this assertion
+                std::string assnKey = SymTab::Mangler::mangleAssnKey( assn.decl, resn.alt.env );
+                auto it = cache.find( assnKey );
+
+                // attempt to resolve assertion if this is the first time seen
+                if ( it == cache.end() ) {
+                        // lookup candidates for this assertion
+                        std::list< SymTab::Indexer::IdData > candidates;
+                        resn.indexer.lookupId( assn.decl->name, candidates );
+
+                        // find the candidates that unify with the desired type
+                        CandidateList matches;
+                        for ( const auto& cdata : candidates ) {
+                                DeclarationWithType* candidate = cdata.id;
+
+                                // build independent unification context for candidate
+                                AssertionSet have, newNeed;
+                                TypeEnvironment newEnv{ resn.alt.env };
+                                OpenVarSet newOpenVars{ resn.alt.openVars };
+                                Type* adjType = candidate->get_type()->clone();
+                                adjustExprType( adjType, newEnv, resn.indexer );
+                                renameTyVars( adjType );
+
+                                // keep unifying candidates
+                                if ( unify( assn.decl->get_type(), adjType, newEnv, newNeed, have, newOpenVars, 
+                                                resn.indexer ) ) {
+                                        // set up binding slot for recursive assertions
+                                        UniqueId crntResnSlot = 0;
+                                        if ( ! newNeed.empty() ) {
+                                                crntResnSlot = ++globalResnSlot;
+                                                for ( auto& a : newNeed ) {
+                                                        a.second.resnSlot = crntResnSlot;
+                                                }
                                         }
-                                }
-                                // // set up idChain on new assertions
-                                // for ( auto& a : newNeed ) {
-                                //      a.second.idChain = assn.info.idChain;
-                                //      a.second.idChain.push_back( assn.decl->get_uniqueId() );
-                                // }
-
-                                matches.emplace_back( cdata, adjType, std::move(newEnv), std::move(have), 
-                                        std::move(newNeed), std::move(newOpenVars), crntResnSlot );
-                        } else {
-                                delete adjType;
+
+                                        matches.emplace_back( cdata, adjType, std::move(newEnv), std::move(have), 
+                                                std::move(newNeed), std::move(newOpenVars), crntResnSlot );
+                                } else {
+                                        delete adjType;
+                                }
                         }
-                }
+
+                        it = cache.emplace_hint( it, assnKey, AssnCacheItem{ std::move(matches) } );
+                }
+
+                CandidateList& matches = it->second.matches;
 
                 // break if no suitable assertion
@@ -284 +305 @@
                 // defer if too many suitable assertions
                 if ( matches.size() > 1 ) {
-                        resn.deferred.emplace_back( assn.decl, assn.info, std::move(matches) );
+                        it->second.deferIds.emplace_back( assn.decl, assn.info );
+                        resn.deferred.emplace_back( cache, assnKey );
                         return true;
                 }
@@ -292 +314 @@
                 addToIndexer( match.have, resn.indexer );
                 resn.newNeed.insert( match.need.begin(), match.need.end() );
-                resn.alt.env = std::move(match.env);
-                resn.alt.openVars = std::move(match.openVars);
+                resn.alt.env = match.env;
+                resn.alt.openVars = match.openVars;
 
                 bindAssertion( assn.decl, assn.info, resn.alt, match, resn.inferred );
@@ -354 +376 @@
                 ResnList resns{ ResnState{ alt, root_indexer } };
                 ResnList new_resns{};
+                AssnCache assnCache;
 
                 // resolve assertions in breadth-first-order up to a limited number of levels deep
@@ -362 +385 @@
                                 for ( auto& assn : resn.need ) {
                                         // fail early if any assertion is not resolvable
-                                        if ( ! resolveAssertion( assn, resn ) ) goto nextResn;
+                                        if ( ! resolveAssertion( assn, resn, assnCache ) ) goto nextResn;
                                 }
 
@@ -373 +396 @@
                                         }
                                 } else {
+                                        // only resolve each deferred assertion once
+                                        std::sort( resn.deferred.begin(), resn.deferred.end() );
+                                        auto last = std::unique( resn.deferred.begin(), resn.deferred.end() );
+                                        resn.deferred.erase( last, resn.deferred.end() );
                                         // resolve deferred assertions by mutual compatibility
                                         std::vector<CandidateEnvMerger::OutType> compatible = filterCombos(
@@ -380 +407 @@
                                         CandidateCost coster{ resn.indexer };
                                         std::sort( compatible.begin(), compatible.end(), coster );
-                                        // // sort by cost if pruning
-                                        // if ( pruneAssertions ) {
-                                        //      auto lmin = sort_mins( compatible.begin(), compatible.end(), 
-                                        //              CandidateCost{resn.indexer} );
-                                        //      compatible.erase( lmin, compatible.end() );
-                                        // }
 
                                         // keep map of detected options
@@ -408 +429 @@
                                                         new_resn.newNeed.insert( match.need.begin(), match.need.end() );
 
-                                                        bindAssertion( r.decl, r.info, new_resn.alt, match, new_resn.inferred );
+                                                        // for each deferred assertion with the same form
+                                                        for ( AssnId id : r.item.deferIds ) {
+                                                                bindAssertion( 
+                                                                        id.decl, id.info, new_resn.alt, match, new_resn.inferred );
+                                                        }
                                                 }
 

src/SymTab/Mangler.cc

@@ -15 +15 @@
 #include "Mangler.h"
 
-#include <algorithm>                // for copy, transform
-#include <cassert>                  // for assert, assertf
-#include <functional>               // for const_mem_fun_t, mem_fun
-#include <iterator>                 // for ostream_iterator, back_insert_ite...
-#include <list>                     // for _List_iterator, list, _List_const...
-#include <string>                   // for string, char_traits, operator<<
-
-#include "CodeGen/OperatorTable.h"  // for OperatorInfo, operatorLookup
+#include <algorithm>                     // for copy, transform
+#include <cassert>                       // for assert, assertf
+#include <functional>                    // for const_mem_fun_t, mem_fun
+#include <iterator>                      // for ostream_iterator, back_insert_ite...
+#include <list>                          // for _List_iterator, list, _List_const...
+#include <string>                        // for string, char_traits, operator<<
+
+#include "CodeGen/OperatorTable.h"       // for OperatorInfo, operatorLookup
 #include "Common/PassVisitor.h"
-#include "Common/SemanticError.h"   // for SemanticError
-#include "Common/utility.h"         // for toString
-#include "Parser/LinkageSpec.h"     // for Spec, isOverridable, AutoGen, Int...
-#include "SynTree/Declaration.h"    // for TypeDecl, DeclarationWithType
-#include "SynTree/Expression.h"     // for TypeExpr, Expression, operator<<
-#include "SynTree/Type.h"           // for Type, ReferenceToType, Type::Fora...
+#include "Common/SemanticError.h"        // for SemanticError
+#include "Common/utility.h"              // for toString
+#include "Parser/LinkageSpec.h"          // for Spec, isOverridable, AutoGen, Int...
+#include "ResolvExpr/TypeEnvironment.h"  // for TypeEnvironment
+#include "SynTree/Declaration.h"         // for TypeDecl, DeclarationWithType
+#include "SynTree/Expression.h"          // for TypeExpr, Expression, operator<<
+#include "SynTree/Type.h"                // for Type, ReferenceToType, Type::Fora...
 
 namespace SymTab {
@@ -37 +38 @@
                         struct Mangler : public WithShortCircuiting, public WithVisitorRef<Mangler>, public WithGuards {
                                 Mangler( bool mangleOverridable, bool typeMode, bool mangleGenericParams );
+                                Mangler( const ResolvExpr::TypeEnvironment& env );
                                 Mangler( const Mangler & ) = delete;
 
@@ -65 +67 @@
                           private:
                                 std::ostringstream mangleName;  ///< Mangled name being constructed
-                                typedef std::map< std::string, std::pair< int, int > > VarMapType;
+                                typedef std::map< std::string, std::pair< std::string, int > > VarMapType;
                                 VarMapType varNums;             ///< Map of type variables to indices
                                 int nextVarNum;                 ///< Next type variable index
+                                const ResolvExpr::TypeEnvironment* env;  ///< optional environment for substitutions
                                 bool isTopLevel;                ///< Is the Mangler at the top level
                                 bool mangleOverridable;         ///< Specially mangle overridable built-in methods
@@ -75 +78 @@
                                 bool inQualifiedType = false;   ///< Add start/end delimiters around qualified type
 
+                          public:
+                                Mangler( bool mangleOverridable, bool typeMode, bool mangleGenericParams, 
+                                        int nextVarNum, const ResolvExpr::TypeEnvironment* env, 
+                                        const VarMapType& varNums );
+
+                          private:
                                 void mangleDecl( DeclarationWithType *declaration );
                                 void mangleRef( ReferenceToType *refType, std::string prefix );
@@ -100 +109 @@
                 }
 
+                std::string mangleAssnKey( DeclarationWithType* decl, 
+                                const ResolvExpr::TypeEnvironment& env ) {
+                        PassVisitor<Mangler> mangler( env );
+                        maybeAccept( decl, mangler );
+                        return mangler.pass.get_mangleName();
+                }
+
                 namespace {
                         Mangler::Mangler( bool mangleOverridable, bool typeMode, bool mangleGenericParams )
-                                : nextVarNum( 0 ), isTopLevel( true ), mangleOverridable( mangleOverridable ), typeMode( typeMode ), mangleGenericParams( mangleGenericParams ) {}
+                                : nextVarNum( 0 ), env(nullptr), isTopLevel( true ), 
+                                mangleOverridable( mangleOverridable ), typeMode( typeMode ), 
+                                mangleGenericParams( mangleGenericParams ) {}
+                        
+                        Mangler::Mangler( const ResolvExpr::TypeEnvironment& env )
+                                : nextVarNum( 0 ), env( &env ), isTopLevel( true ), mangleOverridable( false ),
+                                typeMode( false ), mangleGenericParams( true ) {}
+                        
+                        Mangler::Mangler( bool mangleOverridable, bool typeMode, bool mangleGenericParams, 
+                                int nextVarNum, const ResolvExpr::TypeEnvironment* env, 
+                                const VarMapType& varNums )
+                                : varNums( varNums ), nextVarNum( nextVarNum ), env( env ), isTopLevel( false ), 
+                                mangleOverridable( mangleOverridable ), typeMode( typeMode ), 
+                                mangleGenericParams( mangleGenericParams ) {}
 
                         void Mangler::mangleDecl( DeclarationWithType * declaration ) {
@@ -329 +358 @@
                                                         assert( false );
                                                 } // switch
-                                                varNums[ (*i)->name ] = std::pair< int, int >( nextVarNum++, (int)(*i)->get_kind() );
+                                                std::string varName;
+                                                // replace type with substitution name if environment is available and bound
+                                                if ( env ) {
+                                                        const ResolvExpr::EqvClass* varClass = env->lookup( (*i)->name );
+                                                        if ( varClass && varClass->type ) {
+                                                                PassVisitor<Mangler> sub_mangler(
+                                                                        mangleOverridable, typeMode, mangleGenericParams, nextVarNum, 
+                                                                        env, varNums );
+                                                                varClass->type->accept( sub_mangler );
+                                                                varName = std::string{"%"} + sub_mangler.pass.get_mangleName();
+                                                        }
+                                                }
+                                                // otherwise just give type numeric name
+                                                if ( varName.empty() ) {
+                                                        varName = std::to_string( nextVarNum++ );
+                                                }
+                                                varNums[ (*i)->name ] = std::make_pair( varName, (int)(*i)->get_kind() );
                                                 for ( std::list< DeclarationWithType* >::iterator assert = (*i)->assertions.begin(); assert != (*i)->assertions.end(); ++assert ) {
-                                                        PassVisitor<Mangler> sub_mangler( mangleOverridable, typeMode, mangleGenericParams );
-                                                        sub_mangler.pass.nextVarNum = nextVarNum;
-                                                        sub_mangler.pass.isTopLevel = false;
-                                                        sub_mangler.pass.varNums = varNums;
+                                                        PassVisitor<Mangler> sub_mangler( 
+                                                                mangleOverridable, typeMode, mangleGenericParams, nextVarNum, env, 
+                                                                varNums );
                                                         (*assert)->accept( sub_mangler );
-                                                        assertionNames.push_back( sub_mangler.pass.mangleName.str() );
+                                                        assertionNames.push_back( sub_mangler.pass.get_mangleName() );
                                                         acount++;
                                                 } // for

src/SymTab/Mangler.h

@@ -31 +31 @@
 // * Currently name compression is not implemented.
 
+namespace ResolvExpr {
+        class TypeEnvironment;
+}
+
 namespace SymTab {
         namespace Mangler {
@@ -40 +44 @@
                 /// Mangle ignoring generic type parameters
                 std::string mangleConcrete( Type* ty );
+                /// Mangle for assertion key
+                std::string mangleAssnKey( DeclarationWithType* decl, 
+                        const ResolvExpr::TypeEnvironment& env );
 
                 namespace Encoding {