source: translator/ResolvExpr/RenameVars.cc@ fe3b61b

/*
 * This file is part of the Cforall project
 *
 * $Id: RenameVars.cc,v 1.4 2005/08/29 20:14:16 rcbilson Exp $
 *
 */

#include <strstream>

#include "RenameVars.h"
#include "SynTree/Visitor.h"
#include "SynTree/Type.h"
#include "SynTree/Declaration.h"
#include "SynTree/Expression.h"

namespace ResolvExpr {

RenameVars global_renamer;

RenameVars::RenameVars()
  : level( 0 )
{
  mapStack.push_front( std::map< std::string, std::string >() );
}

void 
RenameVars::reset()
{
  level = 0;
}

void 
RenameVars::visit( VoidType *voidType )
{
  typeBefore( voidType );
  typeAfter( voidType );
}

void 
RenameVars::visit( BasicType *basicType )
{
  typeBefore( basicType );
  typeAfter( basicType );
}

void 
RenameVars::visit( PointerType *pointerType )
{
  typeBefore( pointerType );
///   std::cout << "do pointer" << std::endl;
  maybeAccept( pointerType->get_base(), *this );
///   std::cout << "done pointer" << std::endl;
  typeAfter( pointerType );
}

void 
RenameVars::visit( ArrayType *arrayType )
{
  typeBefore( arrayType );
  maybeAccept( arrayType->get_dimension(), *this );
  maybeAccept( arrayType->get_base(), *this );
  typeAfter( arrayType );
}

void 
RenameVars::visit( FunctionType *functionType )
{
  typeBefore( functionType );
///   std::cout << "return vals" << std::endl;
  acceptAll( functionType->get_returnVals(), *this );
///   std::cout << functionType->get_parameters().size() << " parameters" << std::endl;
  acceptAll( functionType->get_parameters(), *this );
///   std::cout << "done function" << std::endl;
  typeAfter( functionType );
}

void 
RenameVars::visit( StructInstType *aggregateUseType )
{
  typeBefore( aggregateUseType );
  acceptAll( aggregateUseType->get_parameters(), *this );
  typeAfter( aggregateUseType );
}

void 
RenameVars::visit( UnionInstType *aggregateUseType )
{
  typeBefore( aggregateUseType );
  acceptAll( aggregateUseType->get_parameters(), *this );
  typeAfter( aggregateUseType );
}

void 
RenameVars::visit( EnumInstType *aggregateUseType )
{
  typeBefore( aggregateUseType );
  acceptAll( aggregateUseType->get_parameters(), *this );
  typeAfter( aggregateUseType );
}

void 
RenameVars::visit( ContextInstType *aggregateUseType )
{
  typeBefore( aggregateUseType );
  acceptAll( aggregateUseType->get_parameters(), *this );
  acceptAll( aggregateUseType->get_members(), *this );
  typeAfter( aggregateUseType );
}

void 
RenameVars::visit( TypeInstType *instType )
{
  typeBefore( instType );
///   std::cout << "instance of type " << instType->get_name() << std::endl;
  std::map< std::string, std::string >::const_iterator i = mapStack.front().find( instType->get_name() );
  if( i != mapStack.front().end() ) {
///     std::cout << "found name " << i->second << std::endl;
    instType->set_name( i->second );
  } else {
///     std::cout << "no name found" << std::endl;
  }
  acceptAll( instType->get_parameters(), *this );
  typeAfter( instType );
}

void 
RenameVars::visit( TupleType *tupleType )
{
  typeBefore( tupleType );
  acceptAll( tupleType->get_types(), *this );
  typeAfter( tupleType );
}

void 
RenameVars::typeBefore( Type *type )
{
  if( !type->get_forall().empty() ) {
///     std::cout << "type with forall: ";
///     type->print( std::cout );
///     std::cout << std::endl;
    mapStack.push_front( mapStack.front() );
    for( std::list< TypeDecl* >::iterator i = type->get_forall().begin(); i != type->get_forall().end(); ++i ) {
      std::ostrstream output;
      output << "_" << level << "_" << (*i)->get_name();
      std::string newname( output.str(), output.pcount() );
      mapStack.front()[ (*i)->get_name() ] = newname;
      (*i)->set_name( newname );
      level++;
      acceptAll( (*i)->get_assertions(), *this );
    }
  }
}

void 
RenameVars::typeAfter( Type *type )
{
  if( !type->get_forall().empty() ) {
    mapStack.pop_front();
  }
}

} // namespace ResolvExpr