source: src/GenPoly/ScrubTyVars.h @ f1a4ccb

//
// Cforall Version 1.0.0 Copyright (C) 2015 University of Waterloo
//
// The contents of this file are covered under the licence agreement in the
// file "LICENCE" distributed with Cforall.
//
// ScrubTyVars.h -- 
//
// Author           : Richard C. Bilson
// Created On       : Mon May 18 07:44:20 2015
// Last Modified By : Peter A. Buhr
// Last Modified On : Sat Jul 22 09:21:47 2017
// Update Count     : 2
//

#pragma once

#include <string>

#include "GenPoly.h"

#include "SynTree/SynTree.h"
#include "SynTree/Mutator.h"

namespace GenPoly {
        class ScrubTyVars : public Mutator {
                /// Whether to scrub all type variables from the provided map, dynamic type variables from the provided map, or all type variables
                enum ScrubMode { FromMap, DynamicFromMap, All };

                ScrubTyVars() : tyVars(nullptr), mode( All ) {}

                ScrubTyVars( const TyVarMap &tyVars, ScrubMode mode = FromMap ): tyVars( &tyVars ), mode( mode ) {}

        public:
                /// For all polymorphic types with type variables in `tyVars`, replaces generic types, dtypes, and ftypes with the appropriate void type,
                /// and sizeof/alignof expressions with the proper variable
                template< typename SynTreeClass >
                static SynTreeClass *scrub( SynTreeClass *target, const TyVarMap &tyVars );

                /// For all dynamic-layout types with type variables in `tyVars`, replaces generic types, dtypes, and ftypes with the appropriate void type,
                /// and sizeof/alignof expressions with the proper variable
                template< typename SynTreeClass >
                static SynTreeClass *scrubDynamic( SynTreeClass *target, const TyVarMap &tyVars );

                /// For all polymorphic types, replaces generic types, dtypes, and ftypes with the appropriate void type,
                /// and sizeof/alignof expressions with the proper variable
                template< typename SynTreeClass >
                static SynTreeClass *scrubAll( SynTreeClass *target );

                virtual Type* mutate( TypeInstType *typeInst );
                virtual Type* mutate( StructInstType *structInst );
                virtual Type* mutate( UnionInstType *unionInst );
                virtual Expression* mutate( SizeofExpr *szeof );
                virtual Expression* mutate( AlignofExpr *algnof );
                virtual Type* mutate( PointerType *pointer );

          private:
                /// Returns the type if it should be scrubbed, NULL otherwise.
                Type* shouldScrub( Type *ty ) {
                        switch ( mode ) {
                        case FromMap: return isPolyType( ty, *tyVars );
                        case DynamicFromMap: return isDynType( ty, *tyVars );
                        case All: return isPolyType( ty );
                        }
                        assert(false); return nullptr; // unreachable
                        // return dynamicOnly ? isDynType( ty, tyVars ) : isPolyType( ty, tyVars );
                }
                
                /// Mutates (possibly generic) aggregate types appropriately
                Type* mutateAggregateType( Type *ty );
                
                const TyVarMap *tyVars;  ///< Type variables to scrub
                ScrubMode mode;          ///< which type variables to scrub? [FromMap]
        };

        template< typename SynTreeClass >
        SynTreeClass * ScrubTyVars::scrub( SynTreeClass *target, const TyVarMap &tyVars ) {
                ScrubTyVars scrubber( tyVars );
                return static_cast< SynTreeClass * >( target->acceptMutator( scrubber ) );
        }

        template< typename SynTreeClass >
        SynTreeClass * ScrubTyVars::scrubDynamic( SynTreeClass *target, const TyVarMap &tyVars ) {
                ScrubTyVars scrubber( tyVars, ScrubTyVars::DynamicFromMap );
                return static_cast< SynTreeClass * >( target->acceptMutator( scrubber ) );
        }

        template< typename SynTreeClass >
        SynTreeClass * ScrubTyVars::scrubAll( SynTreeClass *target ) {
                ScrubTyVars scrubber;
                return static_cast< SynTreeClass * >( target->acceptMutator( scrubber ) );
        }

} // namespace GenPoly

// Local Variables: //
// tab-width: 4 //
// mode: c++ //
// compile-command: "make install" //
// End: //