[51587aa] | 1 | // |
---|
| 2 | // Cforall Version 1.0.0 Copyright (C) 2015 University of Waterloo |
---|
| 3 | // |
---|
| 4 | // The contents of this file are covered under the licence agreement in the |
---|
| 5 | // file "LICENCE" distributed with Cforall. |
---|
| 6 | // |
---|
| 7 | // XXX.cc -- |
---|
| 8 | // |
---|
| 9 | // Author : Richard C. Bilson |
---|
| 10 | // Created On : Mon May 18 07:44:20 2015 |
---|
| 11 | // Last Modified By : |
---|
| 12 | // Last Modified On : |
---|
| 13 | // Update Count : 0 |
---|
| 14 | // |
---|
[51b7345] | 15 | #include <set> |
---|
| 16 | #include <map> |
---|
| 17 | #include <cassert> |
---|
| 18 | |
---|
| 19 | #include "SynTree/Declaration.h" |
---|
| 20 | #include "SynTree/Type.h" |
---|
| 21 | #include "SynTree/Expression.h" |
---|
| 22 | #include "SynTree/Statement.h" |
---|
| 23 | #include "SynTree/Visitor.h" |
---|
| 24 | #include "UniqueName.h" |
---|
| 25 | |
---|
| 26 | |
---|
| 27 | namespace GenPoly { |
---|
[17cd4eb] | 28 | class CopyParams : public Visitor { |
---|
| 29 | public: |
---|
| 30 | CopyParams(); |
---|
[51b7345] | 31 | |
---|
[17cd4eb] | 32 | virtual void visit( FunctionDecl *funcDecl ); |
---|
| 33 | virtual void visit( AddressExpr *addrExpr ); |
---|
[51b7345] | 34 | |
---|
[17cd4eb] | 35 | private: |
---|
| 36 | std::set< UniqueId > modVars; |
---|
| 37 | UniqueName namer; |
---|
| 38 | }; |
---|
[51b7345] | 39 | |
---|
[17cd4eb] | 40 | void copyParams( std::list< Declaration* > &translationUnit ) { |
---|
| 41 | CopyParams copier; |
---|
| 42 | acceptAll( translationUnit, copier ); |
---|
| 43 | } |
---|
[51b7345] | 44 | |
---|
[17cd4eb] | 45 | CopyParams::CopyParams() : namer( "_cp" ) {} |
---|
[51b7345] | 46 | |
---|
[17cd4eb] | 47 | static const std::list< Label > noLabels; |
---|
[51b7345] | 48 | |
---|
[17cd4eb] | 49 | void CopyParams::visit( FunctionDecl *funcDecl ) { |
---|
| 50 | if ( funcDecl->get_statements() ) { |
---|
| 51 | funcDecl->get_statements()->accept( *this ); |
---|
[51b7345] | 52 | |
---|
[17cd4eb] | 53 | if ( ! modVars.empty() ) { |
---|
| 54 | std::map< std::string, DeclarationWithType* > assignOps; |
---|
| 55 | // assume the assignment operator is the first assert param after any "type" parameter |
---|
| 56 | for ( std::list< TypeDecl* >::const_iterator tyVar = funcDecl->get_functionType()->get_forall().begin(); tyVar != funcDecl->get_functionType()->get_forall().end(); ++tyVar ) { |
---|
| 57 | if ( (*tyVar)->get_kind() == TypeDecl::Any ) { |
---|
| 58 | assert( !(*tyVar)->get_assertions().empty() ); |
---|
| 59 | assignOps[ (*tyVar)->get_name() ] = (*tyVar)->get_assertions().front(); |
---|
| 60 | } // if |
---|
| 61 | } // for |
---|
[a32b204] | 62 | for ( std::list< DeclarationWithType* >::iterator param = funcDecl->get_functionType()->get_parameters().begin(); param != funcDecl->get_functionType()->get_parameters().end(); ++param ) { |
---|
[17cd4eb] | 63 | std::set< UniqueId >::const_iterator var = modVars.find( (*param)->get_uniqueId() ); |
---|
| 64 | if ( var != modVars.end() ) { |
---|
| 65 | TypeInstType *typeInst = dynamic_cast< TypeInstType* >( (*param)->get_type() ); |
---|
| 66 | assert( typeInst ); |
---|
| 67 | std::map< std::string, DeclarationWithType* >::const_iterator assignOp = assignOps.find( typeInst->get_name() ); |
---|
| 68 | if ( assignOp != assignOps.end() ) { |
---|
| 69 | DeclarationWithType *oldParam = *param; |
---|
| 70 | *param = (*param)->clone(); |
---|
| 71 | (*param)->set_mangleName( namer.newName( (*param)->get_mangleName() ) ); |
---|
| 72 | ApplicationExpr *assign = new ApplicationExpr( new VariableExpr( assignOp->second ) ); |
---|
| 73 | assign->get_args().push_back( new VariableExpr( oldParam ) ); |
---|
| 74 | assign->get_args().push_back( new VariableExpr( *param ) ); |
---|
| 75 | funcDecl->get_statements()->get_kids().push_front( new ExprStmt( noLabels, assign ) ); |
---|
| 76 | funcDecl->get_statements()->get_kids().push_front( new DeclStmt( noLabels, oldParam ) ); |
---|
| 77 | } // if |
---|
| 78 | modVars.erase( var ); |
---|
| 79 | } // if |
---|
| 80 | } // for |
---|
| 81 | } // if |
---|
| 82 | } // if |
---|
[51b7345] | 83 | } |
---|
| 84 | |
---|
[17cd4eb] | 85 | // this test is insufficient because it is possible for values to be modified by being passed to other polymorphic |
---|
| 86 | // routines (e.g., assignment operators) without having their addresses explicitly taken. Some thought is needed to |
---|
| 87 | // make sure that all of the correct cases are identified where copies are necessary. |
---|
| 88 | // |
---|
| 89 | // As a temporary measure, for correctness at the expense of performance, ignore the modVars list entirely and copy |
---|
| 90 | // every parameter of TypeInstType* when visiting the FunctionDecl. |
---|
| 91 | void CopyParams::visit( AddressExpr *addrExpr ) { |
---|
| 92 | if ( VariableExpr *varExpr = dynamic_cast< VariableExpr* >( addrExpr->get_arg() ) ) { |
---|
| 93 | if ( dynamic_cast< TypeInstType* >( varExpr->get_var()->get_type() ) ) { |
---|
| 94 | modVars.insert( varExpr->get_var()->get_uniqueId() ); |
---|
| 95 | } // if |
---|
| 96 | } // if |
---|
[51b7345] | 97 | } |
---|
| 98 | } // namespace GenPoly |
---|
[51587aa] | 99 | // Local Variables: // |
---|
| 100 | // tab-width: 4 // |
---|
| 101 | // mode: c++ // |
---|
| 102 | // compile-command: "make install" // |
---|
| 103 | // End: // |
---|