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

File:

• src/GenPoly/ScrubTyVars.cc (modified) (1 diff)

src/GenPoly/ScrubTyVars.cc

@@ -21 +21 @@
 #include "ScrubTyVars.h"
 #include "SymTab/Mangler.h"             // for mangleType
+#include "SynTree/Declaration.h"        // for TypeDecl, TypeDecl::Data, Typ...
+#include "SynTree/Expression.h"         // for Expression (ptr only), NameExpr
+#include "SynTree/Mutator.h"            // for Mutator
+#include "SynTree/Type.h"               // for PointerType, TypeInstType, Type
 
 namespace GenPoly {
+        Type * ScrubTyVars::postmutate( TypeInstType * typeInst ) {
+                if ( ! tyVars ) {
+                        if ( typeInst->get_isFtype() ) {
+                                delete typeInst;
+                                return new PointerType( Type::Qualifiers(), new FunctionType( Type::Qualifiers(), true ) );
+                        } else {
+                                PointerType * ret = new PointerType( Type::Qualifiers(), new VoidType( typeInst->get_qualifiers() ) );
+                                delete typeInst;
+                                return ret;
+                        }
+                }
+
+                TyVarMap::const_iterator tyVar = tyVars->find( typeInst->name );
+                if ( tyVar != tyVars->end() ) {
+                        switch ( tyVar->second.kind ) {
+                          case TypeDecl::Dtype:
+                          case TypeDecl::Ttype:
+                                {
+                                        PointerType * ret = new PointerType( Type::Qualifiers(), new VoidType( typeInst->get_qualifiers() ) );
+                                        delete typeInst;
+                                        return ret;
+                                }
+                          case TypeDecl::Ftype:
+                                delete typeInst;
+                                return new PointerType( Type::Qualifiers(), new FunctionType( Type::Qualifiers(), true ) );
+                          default:
+                                assertf(false, "Unhandled tyvar kind: %d", tyVar->second.kind);
+                        } // switch
+                } // if
+                return typeInst;
+        }
+
+        Type * ScrubTyVars::mutateAggregateType( Type * ty ) {
+                if ( shouldScrub( ty ) ) {
+                        PointerType * ret = new PointerType( Type::Qualifiers(), new VoidType( ty->get_qualifiers() ) );
+                        delete ty;
+                        return ret;
+                }
+                return ty;
+        }
+
+        Type * ScrubTyVars::postmutate( StructInstType * structInst ) {
+                return mutateAggregateType( structInst );
+        }
+
+        Type * ScrubTyVars::postmutate( UnionInstType * unionInst ) {
+                return mutateAggregateType( unionInst );
+        }
+
+        void ScrubTyVars::primeBaseScrub( Type * type ) {
+                // need to determine whether type needs to be scrubbed to determine whether
+                // automatic recursion is necessary
+                if ( Type * t = shouldScrub( type ) ) {
+                        visit_children = false;
+                        GuardValue( dynType );
+                        dynType = t;
+                }
+        }
+
+        Expression * ScrubTyVars::postmutate( SizeofExpr * szeof ) {
+                // sizeof( T ) => _sizeof_T parameter, which is the size of T
+                if ( dynType ) {
+                        Expression *expr = new NameExpr( sizeofName( mangleType( dynType ) ) );
+                        return expr;
+                } // if
+                return szeof;
+        }
+
+        Expression * ScrubTyVars::postmutate( AlignofExpr * algnof ) {
+                // alignof( T ) => _alignof_T parameter, which is the alignment of T
+                if ( dynType ) {
+                        Expression *expr = new NameExpr( alignofName( mangleType( dynType ) ) );
+                        return expr;
+                } // if
+                return algnof;
+        }
+
+        Type * ScrubTyVars::postmutate( PointerType * pointer ) {
+                if ( dynType ) {
+                        Type * ret = dynType->acceptMutator( *visitor );
+                        ret->get_qualifiers() |= pointer->get_qualifiers();
+                        pointer->base = nullptr;
+                        delete pointer;
+                        return ret;
+                }
+                return pointer;
+        }
 
 namespace {