Changes in / [a3dc4cc:c66254e]

File:

• src/ResolvExpr/ResolveAssertions.cc (modified) (1 diff)

src/ResolvExpr/ResolveAssertions.cc

@@ -155 +155 @@
                                         k += computeConversionCost( 
                                                 assn.match.adjType, assn.decl->get_type(), indexer, x.env );
+                                        
+                                        // mark vars+specialization cost on function-type assertions
+                                        PointerType* ptr = dynamic_cast< PointerType* >( assn.decl->get_type() );
+                                        if ( ! ptr ) continue;
+                                        FunctionType* func = dynamic_cast< FunctionType* >( ptr->base );
+                                        if ( ! func ) continue;
+                                        
+                                        for ( DeclarationWithType* formal : func->parameters ) {
+                                                k.decSpec( specCost( formal->get_type() ) );
+                                        }
+                                        k.incVar( func->forall.size() );
+                                        for ( TypeDecl* td : func->forall ) {
+                                                k.decSpec( td->assertions.size() );
+                                        }
                                 }
                                 it = cache.emplace_hint( it, &x, k );