1 | #include <list> |
---|
2 | #include <vector> |
---|
3 | #include <cassert> |
---|
4 | #include <algorithm> |
---|
5 | |
---|
6 | #include "FunctionFixer.h" |
---|
7 | |
---|
8 | namespace Tuples { |
---|
9 | DeclarationWithType *FunctionFixer::mutate( FunctionDecl *functionDecl ) { |
---|
10 | functionDecl->set_functionType( maybeMutate( functionDecl->get_functionType(), *this ) ); |
---|
11 | mutateAll( functionDecl->get_oldDecls(), *this ); |
---|
12 | functionDecl->set_statements( maybeMutate( functionDecl->get_statements(), *this ) ); |
---|
13 | index.visit( functionDecl ); |
---|
14 | rets.clear(); |
---|
15 | return functionDecl; |
---|
16 | } |
---|
17 | |
---|
18 | Type *FunctionFixer::mutate( FunctionType *functionType ) |
---|
19 | { |
---|
20 | typedef std::list< DeclarationWithType * > Decls; |
---|
21 | |
---|
22 | if ( functionType->get_returnVals().size() <= 1 ) return functionType; |
---|
23 | std::copy( functionType->get_returnVals().begin(), functionType->get_returnVals().end(), back_inserter(rets) ); |
---|
24 | |
---|
25 | Type::Qualifiers qual; |
---|
26 | for ( Decls::iterator i = rets.begin(); i != rets.end(); i++ ) { |
---|
27 | (*i)->set_type( new PointerType( qual, (*i)->get_type() ) ); |
---|
28 | functionType->get_parameters().push_back( *i ); |
---|
29 | } |
---|
30 | |
---|
31 | functionType->get_returnVals() = *(new std::list< DeclarationWithType * >()); |
---|
32 | |
---|
33 | functionType->set_isVarArgs( false ); |
---|
34 | return functionType; |
---|
35 | } |
---|
36 | |
---|
37 | Statement *FunctionFixer::mutate( ReturnStmt *retStmt ) |
---|
38 | { |
---|
39 | bool tupleReturn = false; |
---|
40 | Expression *rhs = 0; |
---|
41 | // also check if returning multiple values |
---|
42 | if ( CastExpr *cst = dynamic_cast<CastExpr *>( retStmt->get_expr() ) ) { |
---|
43 | if ( ApplicationExpr *app = dynamic_cast<ApplicationExpr *>( cst->get_arg() ) ) { |
---|
44 | if ( app->get_results().size() > 1 ) { // doesn't need to be ApplicationExpr |
---|
45 | tupleReturn = true; |
---|
46 | rhs = app; |
---|
47 | } |
---|
48 | } else if ( TupleExpr *t = dynamic_cast<TupleExpr *>( cst->get_arg() ) ) { |
---|
49 | tupleReturn = true; |
---|
50 | assert( rets.size() == t->get_exprs().size() ); // stupid check, resolve expression |
---|
51 | rhs = t; |
---|
52 | } |
---|
53 | |
---|
54 | if ( tupleReturn ) { |
---|
55 | assert ( rhs != 0 ); |
---|
56 | std::list< Expression * > lhs; |
---|
57 | for ( std::list< DeclarationWithType * >::iterator d = rets.begin(); d != rets.end(); ++d ) { |
---|
58 | std::list<Expression *> largs; |
---|
59 | largs.push_back(new VariableExpr( *d )); |
---|
60 | Expression *exp = ResolvExpr::resolveInVoidContext( new CastExpr( new UntypedExpr(new NameExpr("*?"), largs), (*d)->get_type()), |
---|
61 | index ); |
---|
62 | lhs.push_back(exp); |
---|
63 | } |
---|
64 | std::list< Expression * > args; |
---|
65 | TupleExpr *tlhs = new TupleExpr; tlhs->set_exprs( lhs ); |
---|
66 | args.push_back( new AddressExpr(tlhs) ); |
---|
67 | args.push_back(rhs); |
---|
68 | |
---|
69 | return new ExprStmt( std::list< Label>(), new UntypedExpr( new NameExpr("?=?"), args ) ); |
---|
70 | } |
---|
71 | } |
---|
72 | /* |
---|
73 | else |
---|
74 | std::cerr << "Empty return statement" << std::endl; |
---|
75 | */ |
---|
76 | |
---|
77 | return retStmt; |
---|
78 | } |
---|
79 | |
---|
80 | Expression* FunctionFixer::mutate( VariableExpr *variableExpr ) { |
---|
81 | if ( rets.empty() ) return variableExpr; |
---|
82 | mutateAll( variableExpr->get_results(), *this ); |
---|
83 | if ( std::find( rets.begin(), rets.end(), variableExpr->get_var() ) != rets.end() ) |
---|
84 | // if ( PointerType *ptr = dynamic_cast<PointerType *>(variableExpr->get_var()->get_type()) ) { |
---|
85 | if ( dynamic_cast<PointerType *>(variableExpr->get_var()->get_type()) != 0 ) { |
---|
86 | std::list<Expression *> largs; |
---|
87 | largs.push_back( new AddressExpr(variableExpr) ); |
---|
88 | Expression *expr = ResolvExpr::resolveInVoidContext( /*new CastExpr(*/new UntypedExpr( new NameExpr( "*?" ), largs )/*, |
---|
89 | ptr->get_base()),*/, index); |
---|
90 | if ( ApplicationExpr *app = dynamic_cast< ApplicationExpr * >( expr ) ) { |
---|
91 | assert( app->get_args().size() == 1 ); |
---|
92 | app->get_args().pop_front(); |
---|
93 | app->get_args().push_back( variableExpr ); |
---|
94 | return app; |
---|
95 | } |
---|
96 | } |
---|
97 | return variableExpr; |
---|
98 | } |
---|
99 | |
---|
100 | /* |
---|
101 | Expression* FunctionFixer::mutate(ApplicationExpr *applicationExpr) { |
---|
102 | std::cerr << "In Application Expression" << std::endl; |
---|
103 | mutateAll( applicationExpr->get_results(), *this ); |
---|
104 | applicationExpr->set_function( maybeMutate( applicationExpr->get_function(), *this ) ); |
---|
105 | mutateAll( applicationExpr->get_args(), *this ); |
---|
106 | return applicationExpr; |
---|
107 | } |
---|
108 | */ |
---|
109 | } // namespace Tuples |
---|