[2b46a13] | 1 | #include "InitTweak.h" |
---|
| 2 | #include "SynTree/Visitor.h" |
---|
| 3 | #include "SynTree/Statement.h" |
---|
| 4 | #include "SynTree/Initializer.h" |
---|
| 5 | #include "SynTree/Expression.h" |
---|
| 6 | #include "GenPoly/GenPoly.h" |
---|
| 7 | |
---|
| 8 | namespace InitTweak { |
---|
[64071c2] | 9 | namespace { |
---|
| 10 | class HasDesignations : public Visitor { |
---|
| 11 | public: |
---|
| 12 | bool hasDesignations = false; |
---|
| 13 | template<typename Init> |
---|
| 14 | void handleInit( Init * init ) { |
---|
| 15 | if ( ! init->get_designators().empty() ) hasDesignations = true; |
---|
| 16 | else Visitor::visit( init ); |
---|
| 17 | } |
---|
| 18 | virtual void visit( SingleInit * singleInit ) { handleInit( singleInit); } |
---|
| 19 | virtual void visit( ListInit * listInit ) { handleInit( listInit); } |
---|
| 20 | }; |
---|
[2b46a13] | 21 | |
---|
[64071c2] | 22 | class InitExpander : public Visitor { |
---|
| 23 | public: |
---|
| 24 | InitExpander() {} |
---|
| 25 | virtual void visit( SingleInit * singleInit ); |
---|
| 26 | virtual void visit( ListInit * listInit ); |
---|
| 27 | std::list< Expression * > argList; |
---|
| 28 | }; |
---|
[2b46a13] | 29 | |
---|
[64071c2] | 30 | void InitExpander::visit( SingleInit * singleInit ) { |
---|
| 31 | argList.push_back( singleInit->get_value()->clone() ); |
---|
| 32 | } |
---|
[2b46a13] | 33 | |
---|
[64071c2] | 34 | void InitExpander::visit( ListInit * listInit ) { |
---|
| 35 | // xxx - for now, assume no nested list inits |
---|
| 36 | std::list<Initializer*>::iterator it = listInit->begin_initializers(); |
---|
| 37 | for ( ; it != listInit->end_initializers(); ++it ) { |
---|
| 38 | (*it)->accept( *this ); |
---|
| 39 | } |
---|
| 40 | } |
---|
| 41 | } |
---|
[2b46a13] | 42 | |
---|
[64071c2] | 43 | std::list< Expression * > makeInitList( Initializer * init ) { |
---|
| 44 | InitExpander expander; |
---|
| 45 | maybeAccept( init, expander ); |
---|
| 46 | return expander.argList; |
---|
| 47 | } |
---|
[2b46a13] | 48 | |
---|
[64071c2] | 49 | bool isDesignated( Initializer * init ) { |
---|
| 50 | HasDesignations finder; |
---|
| 51 | maybeAccept( init, finder ); |
---|
| 52 | return finder.hasDesignations; |
---|
| 53 | } |
---|
[2b46a13] | 54 | |
---|
[64071c2] | 55 | bool tryConstruct( ObjectDecl * objDecl ) { |
---|
| 56 | return ! LinkageSpec::isBuiltin( objDecl->get_linkage() ) && |
---|
| 57 | (objDecl->get_init() == NULL || |
---|
| 58 | ( objDecl->get_init() != NULL && objDecl->get_init()->get_maybeConstructed() )) && |
---|
| 59 | ! isDesignated( objDecl->get_init() ); |
---|
| 60 | } |
---|
[2b46a13] | 61 | |
---|
[64071c2] | 62 | Expression * getCtorDtorCall( Statement * stmt ) { |
---|
| 63 | if ( stmt == NULL ) return NULL; |
---|
| 64 | if ( ExprStmt * exprStmt = dynamic_cast< ExprStmt * >( stmt ) ) { |
---|
| 65 | return exprStmt->get_expr(); |
---|
| 66 | } else if ( CompoundStmt * compoundStmt = dynamic_cast< CompoundStmt * >( stmt ) ) { |
---|
| 67 | // could also be a compound statement with a loop, in the case of an array |
---|
[cad355a] | 68 | if( compoundStmt->get_kids().size() == 2 ) { |
---|
| 69 | // loop variable and loop |
---|
| 70 | ForStmt * forStmt = dynamic_cast< ForStmt * >( compoundStmt->get_kids().back() ); |
---|
| 71 | assert( forStmt && forStmt->get_body() ); |
---|
| 72 | return getCtorDtorCall( forStmt->get_body() ); |
---|
| 73 | } else if ( compoundStmt->get_kids().size() == 1 ) { |
---|
| 74 | // should be the call statement, but in any case there's only one option |
---|
| 75 | return getCtorDtorCall( compoundStmt->get_kids().front() ); |
---|
| 76 | } else { |
---|
| 77 | assert( false && "too many statements in compoundStmt for getCtorDtorCall" ); |
---|
| 78 | } |
---|
[64071c2] | 79 | } if ( ImplicitCtorDtorStmt * impCtorDtorStmt = dynamic_cast< ImplicitCtorDtorStmt * > ( stmt ) ) { |
---|
| 80 | return getCtorDtorCall( impCtorDtorStmt->get_callStmt() ); |
---|
| 81 | } else { |
---|
| 82 | // should never get here |
---|
| 83 | assert( false && "encountered unknown call statement" ); |
---|
| 84 | } |
---|
| 85 | } |
---|
[70f89d00] | 86 | |
---|
[64071c2] | 87 | bool isInstrinsicSingleArgCallStmt( Statement * stmt ) { |
---|
| 88 | Expression * callExpr = getCtorDtorCall( stmt ); |
---|
| 89 | if ( ! callExpr ) return false; |
---|
| 90 | ApplicationExpr * appExpr = dynamic_cast< ApplicationExpr * >( callExpr ); |
---|
| 91 | assert( appExpr ); |
---|
| 92 | VariableExpr * function = dynamic_cast< VariableExpr * >( appExpr->get_function() ); |
---|
| 93 | assert( function ); |
---|
| 94 | // check for Intrinsic only - don't want to remove all overridable ctor/dtors because autogenerated ctor/dtor |
---|
| 95 | // will call all member dtors, and some members may have a user defined dtor. |
---|
| 96 | FunctionType * funcType = GenPoly::getFunctionType( function->get_var()->get_type() ); |
---|
| 97 | assert( funcType ); |
---|
| 98 | return function->get_var()->get_linkage() == LinkageSpec::Intrinsic && funcType->get_parameters().size() == 1; |
---|
| 99 | } |
---|
[f1b1e4c] | 100 | |
---|
[64071c2] | 101 | namespace { |
---|
| 102 | template<typename CallExpr> |
---|
| 103 | Expression *& callArg( CallExpr * callExpr, unsigned int pos ) { |
---|
| 104 | if ( pos >= callExpr->get_args().size() ) assert( false && "asking for argument that doesn't exist. Return NULL/throw exception?" ); |
---|
| 105 | for ( Expression *& arg : callExpr->get_args() ) { |
---|
| 106 | if ( pos == 0 ) return arg; |
---|
| 107 | pos--; |
---|
| 108 | } |
---|
| 109 | assert( false ); |
---|
| 110 | } |
---|
| 111 | } |
---|
[f1b1e4c] | 112 | |
---|
[64071c2] | 113 | Expression *& getCallArg( Expression * callExpr, unsigned int pos ) { |
---|
| 114 | if ( ApplicationExpr * appExpr = dynamic_cast< ApplicationExpr * >( callExpr ) ) { |
---|
| 115 | return callArg( appExpr, pos ); |
---|
| 116 | } else if ( UntypedExpr * untypedExpr = dynamic_cast< UntypedExpr * >( callExpr ) ) { |
---|
| 117 | return callArg( untypedExpr, pos ); |
---|
| 118 | } else { |
---|
| 119 | assert( false && "Unexpected expression type passed to getCallArg" ); |
---|
| 120 | } |
---|
| 121 | } |
---|
[f1b1e4c] | 122 | |
---|
[64071c2] | 123 | namespace { |
---|
[c738ca4] | 124 | std::string funcName( Expression * func ) { |
---|
[64071c2] | 125 | if ( NameExpr * nameExpr = dynamic_cast< NameExpr * >( func ) ) { |
---|
| 126 | return nameExpr->get_name(); |
---|
| 127 | } else if ( VariableExpr * varExpr = dynamic_cast< VariableExpr * >( func ) ) { |
---|
| 128 | return varExpr->get_var()->get_name(); |
---|
[c738ca4] | 129 | } else if ( CastExpr * castExpr = dynamic_cast< CastExpr * >( func ) ) { |
---|
| 130 | return funcName( castExpr->get_arg() ); |
---|
[64071c2] | 131 | } else { |
---|
| 132 | assert( false && "Unexpected expression type being called as a function in call expression" ); |
---|
| 133 | } |
---|
| 134 | } |
---|
| 135 | } |
---|
[70f89d00] | 136 | |
---|
[64071c2] | 137 | std::string getFunctionName( Expression * expr ) { |
---|
| 138 | if ( ApplicationExpr * appExpr = dynamic_cast< ApplicationExpr * >( expr ) ) { |
---|
[c738ca4] | 139 | return funcName( appExpr->get_function() ); |
---|
[64071c2] | 140 | } else if ( UntypedExpr * untypedExpr = dynamic_cast< UntypedExpr * > ( expr ) ) { |
---|
[c738ca4] | 141 | return funcName( untypedExpr->get_function() ); |
---|
[64071c2] | 142 | } else { |
---|
[c738ca4] | 143 | std::cerr << expr << std::endl; |
---|
[64071c2] | 144 | assert( false && "Unexpected expression type passed to getFunctionName" ); |
---|
| 145 | } |
---|
| 146 | } |
---|
[10a7775] | 147 | |
---|
[64071c2] | 148 | Type * getPointerBase( Type * type ) { |
---|
| 149 | if ( PointerType * ptrType = dynamic_cast< PointerType * >( type ) ) { |
---|
| 150 | return ptrType->get_base(); |
---|
| 151 | } else if ( ArrayType * arrayType = dynamic_cast< ArrayType * >( type ) ) { |
---|
| 152 | return arrayType->get_base(); |
---|
| 153 | } else { |
---|
| 154 | return NULL; |
---|
| 155 | } |
---|
| 156 | } |
---|
[10a7775] | 157 | |
---|
[64071c2] | 158 | Type * isPointerType( Type * type ) { |
---|
| 159 | if ( getPointerBase( type ) ) return type; |
---|
| 160 | else return NULL; |
---|
| 161 | } |
---|
[2b46a13] | 162 | } |
---|