source: src/Tuples/MultRet.cc@ ac633d0

ADT aaron-thesis arm-eh ast-experimental cleanup-dtors ctor deferred_resn demangler enum forall-pointer-decay gc_noraii jacob/cs343-translation jenkins-sandbox memory new-ast new-ast-unique-expr new-env no_list persistent-indexer pthread-emulation qualifiedEnum resolv-new string with_gc
Last change on this file since ac633d0 was 5f2f2d7, checked in by Peter A. Buhr <pabuhr@…>, 10 years ago

fix constant types, remove unnecessary string copying, work on regression testing, fix several memory leaks
  • Property mode set to 100644
File size: 4.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// MultRet.cc --
8//
9// Author : Rodolfo G. Esteves
10// Created On : Mon May 18 07:44:20 2015
11// Last Modified By : Peter A. Buhr
12// Last Modified On : Mon Jun 8 14:54:44 2015
13// Update Count : 2
14//
15
16#include <list>
17#include <vector>
18#include <cassert>
19#include <algorithm>
20
21#include "MultRet.h"
22#include "SynTree/Statement.h"
23#include "SynTree/Expression.h"
24#include "SynTree/Declaration.h"
25#include "SynTree/Type.h"
26
27namespace Tuples {
28 MVRMutator::MVRMutator() : newVars( 0 ), newCode( 0 ) {
29 }
30
31 MVRMutator::~MVRMutator() {
32 }
33
34 int MVRMutator::curVal = 0;
35
36 Statement *MVRMutator::mutate( ExprStmt *exprStmt ) {
37 MVRMutator toplevel;
38 exprStmt->set_expr( maybeMutate( exprStmt->get_expr(), toplevel ) );
39
40 if ( toplevel.hasCode() ) {
41 assert ( toplevel.getVars() != 0 );
42
43 typedef std::list<Statement *> Statements;
44
45 CompoundStmt *code = new CompoundStmt( std::list< Label >() );
46
47 // copy variables
48 Statements &vars = toplevel.getVars()->get_kids();
49 for ( Statements::iterator i = vars.begin(); i != vars.end(); i++ )
50 code->get_kids().push_back( *i );
51
52 // copy statements
53 Statements &block = toplevel.getCode()->get_kids();
54 for ( Statements::iterator i = block.begin(); i != block.end(); i++ )
55 code->get_kids().push_back( *i );
56
57 return code;
58 } else
59 return exprStmt;
60 }
61
62 Expression *MVRMutator::mutate( ApplicationExpr *appExpr ) {
63 // appExpr->set_function( maybeMutate( appExpr->get_function(), *this ) );
64 bool mulretp = false;
65 VariableExpr *funname;
66 if ( (funname = dynamic_cast<VariableExpr *>(appExpr->get_function())) == 0 ) goto DoArgs;
67
68 FunctionDecl *fundecl;
69 if ((fundecl = dynamic_cast<FunctionDecl *>(funname->get_var())) == 0) goto DoArgs;
70 {
71 typedef std::list<DeclarationWithType*> RetType;
72
73 RetType &rets = fundecl->get_functionType()->get_returnVals();
74 if ( rets.size() <= 1 ) goto DoArgs;
75 mulretp = true;
76
77 if ( newVars == 0 )
78 newVars = new CompoundStmt( std::list<Label>(0) );
79
80 for (RetType::iterator i = rets.begin() ; i != rets.end(); i++) {
81 DeclStmt *arg = newVar( *i );
82 newVars->get_kids().push_back( arg );
83 add_pending( arg->get_decl() );
84 } // for
85 }
86
87 DoArgs:
88 // mutate the argument list
89 typedef std::list< Expression *> Exprs;
90 Exprs &args = appExpr->get_args();
91 std::list< Expression * > newArgs;
92 for ( Exprs::iterator i = args.begin(); i != args.end(); i++ ) {
93 MVRMutator next;
94 Expression *mutated = (*i)->acceptMutator( next );
95
96 if ( next.hasCode() ) {
97 // merge new vars and bodies
98 typedef std::list< Statement * > Stmts;
99 Stmts &vars = next.getVars()->get_kids();
100 Stmts &block = next.getCode()->get_kids();
101
102 if (newVars == 0)
103 newVars = new CompoundStmt( std::list< Label >() );
104 for ( Stmts::iterator i = vars.begin(); i != vars.end(); i++ ) // std::splice? -- need to append lists
105 newVars->get_kids().push_back( *i );
106
107 if (newCode == 0)
108 newCode = new CompoundStmt( std::list< Label >() );
109 for ( Stmts::iterator i = block.begin(); i != block.end(); i++ )
110 newCode->get_kids().push_back( *i );
111 } // if
112
113 if ( next.hasResults() ) {
114 Exprs &res = next.get_results();
115 for ( Exprs::iterator i = res.begin(); i != res.end(); i++ )
116 newArgs.push_back( *i );
117 } else
118 newArgs.push_back( mutated );
119 }
120
121 appExpr->get_args() = newArgs; // new argument list
122
123 if ( mulretp ) {
124 // add 'out' parameters
125 if ( ! argsToAdd.empty() )
126 for (std::list< Expression *>::iterator i = argsToAdd.begin(); i != argsToAdd.end(); i++)
127 (appExpr->get_args()).push_back( *i );
128 // clear 'out' parameters ( so that the list can be reused -- substitute by auto_ptr later? )
129
130 if (newCode == 0)
131 newCode = new CompoundStmt( std::list<Label>(0) );
132 } // if
133
134 // add to block of code
135 if ( newCode != 0 )
136 newCode->get_kids().push_back( new ExprStmt( std::list<Label>(), appExpr ) );
137
138 return appExpr;
139 }
140
141 // Auxiliary function to generate new names for the `output' parameters
142 DeclStmt *MVRMutator::newVar( DeclarationWithType *reqDecl ) {
143 // std::ostringstream os;
144 // os << "__" << curVal++ << "__";// << std::ends;
145
146 ObjectDecl *decl;
147 if ((decl = dynamic_cast<ObjectDecl *>( reqDecl )) != 0)
148 // return new DeclStmt( new ObjectDecl( std::string (os.str(), os.pcount()), );
149 return new DeclStmt( std::list<Label>(), decl );
150 else
151 return 0;
152 }
153
154 void MVRMutator::add_pending( Declaration *decl ) {
155 ObjectDecl *obj;
156 if ( (obj = dynamic_cast< ObjectDecl * >( decl )) == 0 ) return;
157
158 VariableExpr *var = new VariableExpr(obj, 0 );
159 results.push_back( var ); // probably change this name to newResults or something
160 argsToAdd.push_back( new AddressExpr( var ) );
161 return;
162 }
163}
164
165// Local Variables: //
166// tab-width: 4 //
167// mode: c++ //
168// compile-command: "make install" //
169// End: //
Note: See TracBrowser for help on using the repository browser.