Changes in src/GenPoly/ScrubTyVars.cc [2f61765:0026d67]

File:

• src/GenPoly/ScrubTyVars.cc (modified) (5 diffs)

src/GenPoly/ScrubTyVars.cc

@@ -10 +10 @@
 // Created On       : Mon May 18 07:44:20 2015
 // Last Modified By : Andrew Beach
-// Last Modified On : Wed Dec  7 17:01:00 2022
-// Update Count     : 6
+// Last Modified On : Fri Oct  7 15:42:00 2022
+// Update Count     : 5
 //
 
@@ -117 +117 @@
 namespace {
 
+enum class ScrubMode {
+        FromMap,
+        DynamicFromMap,
+        All,
+};
+
 struct ScrubTypeVars :
         public ast::WithGuards,
@@ -178 +184 @@
 
 ast::Type const * ScrubTypeVars::postvisit( ast::TypeInstType const * type ) {
-        ast::TypeDecl::Kind kind;
         // This implies that mode == ScrubMode::All.
         if ( !typeVars ) {
-                kind = type->kind;
-        } else {
-                // Otherwise, only scrub the type var if it is in map.
-                auto typeVar = typeVars->find( *type );
-                if ( typeVar == typeVars->end() ) {
-                        return type;
-                }
-                kind = typeVar->second.kind;
-        }
-
-        switch ( kind ) {
-        case ast::TypeDecl::Dtype:
-        case ast::TypeDecl::Ttype:
+                if ( ast::TypeDecl::Ftype == type->kind ) {
+                        return new ast::PointerType(
+                                new ast::FunctionType( ast::FixedArgs ) );
+                } else {
+                        return new ast::PointerType(
+                                new ast::VoidType( type->qualifiers ) );
+                }
+        }
+
+        auto typeVar = typeVars->find( *type );
+        if ( typeVar == typeVars->end() ) {
+                return type;
+        }
+
+        switch ( typeVar->second.kind ) {
+        case ::TypeDecl::Dtype:
+        case ::TypeDecl::Ttype:
                 return new ast::PointerType(
                         new ast::VoidType( type->qualifiers ) );
-        case ast::TypeDecl::Ftype:
+        case ::TypeDecl::Ftype:
                 return new ast::PointerType(
                         new ast::FunctionType( ast::VariableArgs ) );
         default:
-                assertf( false, "Unhandled type variable kind: %d", kind );
+                assertf( false,
+                        "Unhandled type variable kind: %d", typeVar->second.kind );
                 throw; // Just in case the assert is removed, stop here.
         }
@@ -243 +253 @@
 }
 
-} // namespace
-
 const ast::Node * scrubTypeVarsBase(
-                const ast::Node * node, const TypeVarMap * typeVars, ScrubMode mode ) {
+                const ast::Node * target,
+                ScrubMode mode, const TypeVarMap * typeVars ) {
         if ( ScrubMode::All == mode ) {
                 assert( nullptr == typeVars );
@@ -253 +262 @@
         }
         ast::Pass<ScrubTypeVars> visitor( mode, typeVars );
-        return node->accept( visitor );
+        return target->accept( visitor );
+}
+
+} // namespace
+
+template<>
+ast::Node const * scrubTypeVars<ast::Node>(
+        const ast::Node * target, const TypeVarMap & typeVars ) {
+        return scrubTypeVarsBase( target, ScrubMode::FromMap, &typeVars );
+}
+
+template<>
+ast::Node const * scrubTypeVarsDynamic<ast::Node>(
+        ast::Node const * target, const TypeVarMap & typeVars ) {
+        return scrubTypeVarsBase( target, ScrubMode::DynamicFromMap, &typeVars );
+}
+
+template<>
+ast::Node const * scrubAllTypeVars<ast::Node>( const ast::Node * target ) {
+        return scrubTypeVarsBase( target, ScrubMode::All, nullptr );
 }