source: translator/GenPoly/ScrubTyVars.cc @ b87a5ed

#include "GenPoly.h"
#include "ScrubTyVars.h"

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


namespace GenPoly {
    Type * ScrubTyVars::mutate( TypeInstType *typeInst ) {
        TyVarMap::const_iterator tyVar = tyVars.find( typeInst->get_name() );
        if ( doAll || tyVar != tyVars.end() ) {
            switch( tyVar->second ) {
              case TypeDecl::Any:
              case TypeDecl::Dtype:
                {
                    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 ) );
            }
        }
        return typeInst;
    }

    Expression * ScrubTyVars::mutate( SizeofExpr *szeof ) {
        // sizeof( T ) => T parameter, which is the size of T
        if ( TypeInstType *typeInst = dynamic_cast< TypeInstType * >( szeof->get_type() ) ) {
            Expression *expr = new NameExpr( typeInst->get_name() );
            return expr;
        } else {
            return Mutator::mutate( szeof );
        }
    }

    Type * ScrubTyVars::mutate( PointerType *pointer ) {
        if ( TypeInstType *typeInst = dynamic_cast< TypeInstType * >( pointer->get_base() ) ) {
            if ( doAll || tyVars.find( typeInst->get_name() ) != tyVars.end() ) {
                Type *ret = mutate( typeInst );
                ret->get_qualifiers() += pointer->get_qualifiers();
                pointer->set_base( 0 );
                delete pointer;
                return ret;
            }
        }
        return Mutator::mutate( pointer );
    }
} // namespace GenPoly