Changes in src/ResolvExpr/PolyCost.cc [00ac42e:5aa4656]

File:

• src/ResolvExpr/PolyCost.cc (modified) (2 diffs)

src/ResolvExpr/PolyCost.cc

@@ -9 +9 @@
 // Author           : Richard C. Bilson
 // Created On       : Sun May 17 09:50:12 2015
-// Last Modified By : Peter A. Buhr
-// Last Modified On : Sun May 17 09:52:02 2015
-// Update Count     : 3
+// Last Modified By : Andrew Beach
+// Last Modified On : Wed Jun 19 10:45:00 2019
+// Update Count     : 4
 //
 
+#include "AST/SymbolTable.hpp"
+#include "AST/Type.hpp"
+#include "AST/TypeEnvironment.hpp"
 #include "Common/PassVisitor.h"
 #include "SymTab/Indexer.h"   // for Indexer
@@ -54 +57 @@
         }
 
+// TODO: When the old PolyCost is torn out get rid of the _new suffix.
+struct PolyCost_new {
+        int result;
+        const ast::SymbolTable &symtab;
+        const ast::TypeEnvironment &env_;
+
+        PolyCost_new( const ast::SymbolTable & symtab, const ast::TypeEnvironment & env ) :
+                result( 0 ), symtab( symtab ), env_( env ) {}
+
+        void previsit( const ast::TypeInstType * type ) {
+                if ( const ast::EqvClass * eqv = env_.lookup( type->name ) ) /* && */ if ( eqv->bound ) {
+                        if ( const ast::TypeInstType * otherType = eqv->bound.as< ast::TypeInstType >() ) {
+                                if ( symtab.lookupType( otherType->name ) ) {
+                                        // Bound to opaque type.
+                                        result += 1;
+                                }
+                        } else {
+                                // Bound to concrete type.
+                                result += 1;
+                        }
+                }
+        }
+};
+
+int polyCost(
+        const ast::Type * type, const ast::SymbolTable & symtab, const ast::TypeEnvironment & env
+) {
+        ast::Pass<PolyCost_new> costing( symtab, env );
+        type->accept( costing );
+        return costing.pass.result;
+}
+
 } // namespace ResolvExpr