| 1 | #include <set>
|
|---|
| 2 | #include <map>
|
|---|
| 3 | #include <cassert>
|
|---|
| 4 |
|
|---|
| 5 | #include "SynTree/Declaration.h"
|
|---|
| 6 | #include "SynTree/Type.h"
|
|---|
| 7 | #include "SynTree/Expression.h"
|
|---|
| 8 | #include "SynTree/Statement.h"
|
|---|
| 9 | #include "SynTree/Visitor.h"
|
|---|
| 10 | #include "UniqueName.h"
|
|---|
| 11 |
|
|---|
| 12 |
|
|---|
| 13 | namespace GenPoly {
|
|---|
| 14 | class CopyParams : public Visitor {
|
|---|
| 15 | public:
|
|---|
| 16 | CopyParams();
|
|---|
| 17 |
|
|---|
| 18 | virtual void visit( FunctionDecl *funcDecl );
|
|---|
| 19 | virtual void visit( AddressExpr *addrExpr );
|
|---|
| 20 |
|
|---|
| 21 | private:
|
|---|
| 22 | std::set< UniqueId > modVars;
|
|---|
| 23 | UniqueName namer;
|
|---|
| 24 | };
|
|---|
| 25 |
|
|---|
| 26 | void copyParams( std::list< Declaration* > &translationUnit ) {
|
|---|
| 27 | CopyParams copier;
|
|---|
| 28 | acceptAll( translationUnit, copier );
|
|---|
| 29 | }
|
|---|
| 30 |
|
|---|
| 31 | CopyParams::CopyParams() : namer( "_cp" ) {}
|
|---|
| 32 |
|
|---|
| 33 | static const std::list< Label > noLabels;
|
|---|
| 34 |
|
|---|
| 35 | void CopyParams::visit( FunctionDecl *funcDecl ) {
|
|---|
| 36 | if ( funcDecl->get_statements() ) {
|
|---|
| 37 | funcDecl->get_statements()->accept( *this );
|
|---|
| 38 |
|
|---|
| 39 | if ( ! modVars.empty() ) {
|
|---|
| 40 | std::map< std::string, DeclarationWithType* > assignOps;
|
|---|
| 41 | // assume the assignment operator is the first assert param after any "type" parameter
|
|---|
| 42 | for ( std::list< TypeDecl* >::const_iterator tyVar = funcDecl->get_functionType()->get_forall().begin(); tyVar != funcDecl->get_functionType()->get_forall().end(); ++tyVar ) {
|
|---|
| 43 | if ( (*tyVar)->get_kind() == TypeDecl::Any ) {
|
|---|
| 44 | assert( !(*tyVar)->get_assertions().empty() );
|
|---|
| 45 | assignOps[ (*tyVar)->get_name() ] = (*tyVar)->get_assertions().front();
|
|---|
| 46 | } // if
|
|---|
| 47 | } // for
|
|---|
| 48 | for( std::list< DeclarationWithType* >::iterator param = funcDecl->get_functionType()->get_parameters().begin(); param != funcDecl->get_functionType()->get_parameters().end(); ++param ) {
|
|---|
| 49 | std::set< UniqueId >::const_iterator var = modVars.find( (*param)->get_uniqueId() );
|
|---|
| 50 | if ( var != modVars.end() ) {
|
|---|
| 51 | TypeInstType *typeInst = dynamic_cast< TypeInstType* >( (*param)->get_type() );
|
|---|
| 52 | assert( typeInst );
|
|---|
| 53 | std::map< std::string, DeclarationWithType* >::const_iterator assignOp = assignOps.find( typeInst->get_name() );
|
|---|
| 54 | if ( assignOp != assignOps.end() ) {
|
|---|
| 55 | DeclarationWithType *oldParam = *param;
|
|---|
| 56 | *param = (*param)->clone();
|
|---|
| 57 | (*param)->set_mangleName( namer.newName( (*param)->get_mangleName() ) );
|
|---|
| 58 | ApplicationExpr *assign = new ApplicationExpr( new VariableExpr( assignOp->second ) );
|
|---|
| 59 | assign->get_args().push_back( new VariableExpr( oldParam ) );
|
|---|
| 60 | assign->get_args().push_back( new VariableExpr( *param ) );
|
|---|
| 61 | funcDecl->get_statements()->get_kids().push_front( new ExprStmt( noLabels, assign ) );
|
|---|
| 62 | funcDecl->get_statements()->get_kids().push_front( new DeclStmt( noLabels, oldParam ) );
|
|---|
| 63 | } // if
|
|---|
| 64 | modVars.erase( var );
|
|---|
| 65 | } // if
|
|---|
| 66 | } // for
|
|---|
| 67 | } // if
|
|---|
| 68 | } // if
|
|---|
| 69 | }
|
|---|
| 70 |
|
|---|
| 71 | // this test is insufficient because it is possible for values to be modified by being passed to other polymorphic
|
|---|
| 72 | // routines (e.g., assignment operators) without having their addresses explicitly taken. Some thought is needed to
|
|---|
| 73 | // make sure that all of the correct cases are identified where copies are necessary.
|
|---|
| 74 | //
|
|---|
| 75 | // As a temporary measure, for correctness at the expense of performance, ignore the modVars list entirely and copy
|
|---|
| 76 | // every parameter of TypeInstType* when visiting the FunctionDecl.
|
|---|
| 77 | void CopyParams::visit( AddressExpr *addrExpr ) {
|
|---|
| 78 | if ( VariableExpr *varExpr = dynamic_cast< VariableExpr* >( addrExpr->get_arg() ) ) {
|
|---|
| 79 | if ( dynamic_cast< TypeInstType* >( varExpr->get_var()->get_type() ) ) {
|
|---|
| 80 | modVars.insert( varExpr->get_var()->get_uniqueId() );
|
|---|
| 81 | } // if
|
|---|
| 82 | } // if
|
|---|
| 83 | }
|
|---|
| 84 | } // namespace GenPoly
|
|---|
