source: translator/SymTab/Mangler.cc @ a32b204

#include <cassert>
#include <string>
#include <algorithm>
#include <iterator>
#include <functional>
#include <set>

#include "SynTree/Declaration.h"
#include "SynTree/Type.h"
#include "SynTree/Expression.h"
#include "SynTree/Initializer.h"
#include "SynTree/Statement.h"
#include "Mangler.h"
#include "CodeGen/OperatorTable.h"

namespace SymTab {
    Mangler::Mangler() : nextVarNum( 0 ), isTopLevel( true )
    {}

//Mangler::Mangler( const Mangler & )
//  : mangleName(), varNums( varNums ), nextVarNum( nextVarNum ), isTopLevel( isTopLevel )
//{
//}
    Mangler::Mangler( const Mangler &rhs ) : mangleName() {
        varNums = rhs.varNums;
        nextVarNum = rhs.nextVarNum;
        isTopLevel = rhs.isTopLevel;
    }

    void Mangler::mangleDecl( DeclarationWithType *declaration ) {
        bool wasTopLevel = isTopLevel;
        if ( isTopLevel ) {
            varNums.clear();
            nextVarNum = 0;
            isTopLevel = false;
        }
        mangleName << "__";
        CodeGen::OperatorInfo opInfo;
        if ( operatorLookup( declaration->get_name(), opInfo ) ) {
            mangleName << opInfo.outputName;
        } else {
            mangleName << declaration->get_name();
        }
        mangleName << "__";
        maybeAccept( declaration->get_type(), *this );
        isTopLevel = wasTopLevel;
    }

    void Mangler::visit( ObjectDecl *declaration ) {
        mangleDecl( declaration );
    }

    void Mangler::visit( FunctionDecl *declaration ) {
        mangleDecl( declaration );
    }

    void Mangler::visit( VoidType *voidType ) {
        printQualifiers( voidType );
        mangleName << "v";
    }

    void Mangler::visit( BasicType *basicType ) {
        static const char *btLetter[] = {
            "b",        // Bool
            "c",        // Char
            "Sc",       // SignedChar
            "Uc",       // UnsignedChar
            "s",        // ShortSignedInt
            "Us",       // ShortUnsignedInt
            "i",        // SignedInt
            "Ui",       // UnsignedInt
            "l",        // LongSignedInt
            "Ul",       // LongUnsignedInt
            "q",        // LongLongSignedInt
            "Uq",       // LongLongUnsignedInt
            "f",        // Float
            "d",        // Double
            "r",        // LongDouble
            "Xf",       // FloatComplex
            "Xd",       // DoubleComplex
            "Xr",       // LongDoubleComplex
            "If",       // FloatImaginary
            "Id",       // DoubleImaginary
            "Ir",       // LongDoubleImaginary
        };
  
        printQualifiers( basicType );
        mangleName << btLetter[ basicType->get_kind() ];
    }

    void Mangler::visit( PointerType *pointerType ) {
        printQualifiers( pointerType );
        mangleName << "P";
        maybeAccept( pointerType->get_base(), *this );
    }

    void Mangler::visit( ArrayType *arrayType ) {
        // TODO: encode dimension
        printQualifiers( arrayType );
        mangleName << "A0";
        maybeAccept( arrayType->get_base(), *this );
    }

    namespace {
        inline std::list< Type* >
        getTypes( const std::list< DeclarationWithType* > decls )
        {
            std::list< Type* > ret;
            std::transform( decls.begin(), decls.end(), std::back_inserter( ret ),
                            std::mem_fun( &DeclarationWithType::get_type ) );
            return ret;
        }
    }

    void Mangler::visit( FunctionType *functionType ) {
        printQualifiers( functionType );
        mangleName << "F";
        std::list< Type* > returnTypes = getTypes( functionType->get_returnVals() );
        acceptAll( returnTypes, *this );
        mangleName << "_";
        std::list< Type* > paramTypes = getTypes( functionType->get_parameters() );
        acceptAll( paramTypes, *this );
        mangleName << "_";
    }

    void Mangler::mangleRef( ReferenceToType *refType, std::string prefix ) {
        printQualifiers( refType );
        mangleName << ( refType->get_name().length() + prefix.length() ) << prefix << refType->get_name();
    }

    void Mangler::visit( StructInstType *aggregateUseType ) {
        mangleRef( aggregateUseType, "s" );
    }

    void Mangler::visit( UnionInstType *aggregateUseType ) {
        mangleRef( aggregateUseType, "u" );
    }

    void Mangler::visit( EnumInstType *aggregateUseType ) {
        mangleRef( aggregateUseType, "e" );
    }

    void Mangler::visit( TypeInstType *typeInst ) {
        VarMapType::iterator varNum = varNums.find( typeInst->get_name() );
        if ( varNum == varNums.end() ) {
            mangleRef( typeInst, "t" );
        } else {
            printQualifiers( typeInst );
            std::ostrstream numStream;
            numStream << varNum->second.first;
            mangleName << (numStream.pcount() + 1);
            switch ( (TypeDecl::Kind )varNum->second.second ) {
              case TypeDecl::Any:
                mangleName << "t";
                break;
              case TypeDecl::Dtype:
                mangleName << "d";
                break;
              case TypeDecl::Ftype:
                mangleName << "f";
                break;
            }
            mangleName << std::string( numStream.str(), numStream.pcount() );
        }
    }

    void Mangler::visit( TupleType *tupleType ) {
        printQualifiers( tupleType );
        mangleName << "T";
        acceptAll( tupleType->get_types(), *this );
        mangleName << "_";
    }

    void Mangler::visit( TypeDecl *decl ) {
        static const char *typePrefix[] = { "BT", "BD", "BF" };
        mangleName << typePrefix[ decl->get_kind() ] << ( decl->get_name().length() + 1 ) << decl->get_name();
    }

    void printVarMap( const std::map< std::string, std::pair< int, int > > &varMap, std::ostream &os ) {
        for ( std::map< std::string, std::pair< int, int > >::const_iterator i = varMap.begin(); i != varMap.end(); ++i ) {
            os << i->first << "(" << i->second.first << "/" << i->second.second << ")" << std::endl;
        }
    }

    void Mangler::printQualifiers( Type *type ) {
        if ( ! type->get_forall().empty() ) {
            std::list< std::string > assertionNames;
            int tcount = 0, dcount = 0, fcount = 0;
            mangleName << "A";
            for ( std::list< TypeDecl* >::iterator i = type->get_forall().begin(); i != type->get_forall().end(); ++i ) {
                switch ( (*i)->get_kind() ) {
                  case TypeDecl::Any:
                    tcount++;
                    break;
                  case TypeDecl::Dtype:
                    dcount++;
                    break;
                  case TypeDecl::Ftype:
                    fcount++;
                    break;
                }
                varNums[ (*i )->get_name() ] = std::pair< int, int >( nextVarNum++, (int )(*i )->get_kind() );
                for ( std::list< DeclarationWithType* >::iterator assert = (*i )->get_assertions().begin(); assert != (*i )->get_assertions().end(); ++assert ) {
                    Mangler sub_mangler;
                    sub_mangler.nextVarNum = nextVarNum;
                    sub_mangler.isTopLevel = false;
                    sub_mangler.varNums = varNums;
                    (*assert)->accept( sub_mangler );
                    assertionNames.push_back( std::string( sub_mangler.mangleName.str(), sub_mangler.mangleName.pcount() ) );
                }
            }
            mangleName << tcount << "_" << dcount << "_" << fcount << "_";
            std::copy( assertionNames.begin(), assertionNames.end(), std::ostream_iterator< std::string >( mangleName, "" ) );
            mangleName << "_";
        }
        if ( type->get_isConst() ) {
            mangleName << "C";
        }
        if ( type->get_isVolatile() ) {
            mangleName << "V";
        }
        if ( type->get_isRestrict() ) {
            mangleName << "R";
        }
        if ( type->get_isLvalue() ) {
            mangleName << "L";
        }
        if ( type->get_isAtomic() ) {
            mangleName << "A";
        }
    }
} // SymTab