source: translator/GenPoly/CopyParams.cc @ 51587aa

aaron-thesisarm-ehcleanup-dtorsctordeferred_resndemanglerenumforall-pointer-decaygc_noraiijacob/cs343-translationjenkins-sandboxmemorynew-astnew-ast-unique-exprnew-envno_listpersistent-indexerpthread-emulationqualifiedEnumresolv-newstringwith_gc
Last change on this file since 51587aa was 51587aa, checked in by Peter A. Buhr <pabuhr@…>, 8 years ago

licencing: fourth groups of files

  • Property mode set to 100644
File size: 3.9 KB
Line 
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//
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
27namespace GenPoly {
28    class CopyParams : public Visitor {
29      public:
30        CopyParams();
31 
32        virtual void visit( FunctionDecl *funcDecl );
33        virtual void visit( AddressExpr *addrExpr );
34
35      private:
36        std::set< UniqueId > modVars;
37        UniqueName namer;
38    };
39
40    void copyParams( std::list< Declaration* > &translationUnit ) {
41        CopyParams copier;
42        acceptAll( translationUnit, copier );
43    }
44
45    CopyParams::CopyParams() : namer( "_cp" ) {}
46
47    static const std::list< Label > noLabels;
48
49    void CopyParams::visit( FunctionDecl *funcDecl ) {
50        if ( funcDecl->get_statements() ) {
51            funcDecl->get_statements()->accept( *this );
52   
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
62                for ( std::list< DeclarationWithType* >::iterator param = funcDecl->get_functionType()->get_parameters().begin(); param != funcDecl->get_functionType()->get_parameters().end(); ++param ) {
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
83    }
84
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
97    }
98} // namespace GenPoly
99// Local Variables: //
100// tab-width: 4 //
101// mode: c++ //
102// compile-command: "make install" //
103// End: //
Note: See TracBrowser for help on using the repository browser.