Changes in src/InitTweak/InitTweak.cc [dcd73d1:ee1635c8]

File:

• src/InitTweak/InitTweak.cc (modified) (5 diffs)

src/InitTweak/InitTweak.cc

@@ -304 +304 @@
 
         namespace {
-                VariableExpr * getCalledFunction( ApplicationExpr * appExpr ) {
-                        assert( appExpr );
-                        // xxx - it's possible this can be other things, e.g. MemberExpr, so this is insufficient
-                        return dynamic_cast< VariableExpr * >( appExpr->get_function() );
+                DeclarationWithType * getCalledFunction( Expression * expr ) {
+                        assert( expr );
+                        if ( VariableExpr * varExpr = dynamic_cast< VariableExpr * >( expr ) ) {
+                                return varExpr->get_var();
+                        } else if ( MemberExpr * memberExpr = dynamic_cast< MemberExpr * >( expr ) ) {
+                                return memberExpr->get_member();
+                        } else if ( CastExpr * castExpr = dynamic_cast< CastExpr * >( expr ) ) {
+                                return getCalledFunction( castExpr->get_arg() );
+                        }
+                        return nullptr;
                 }
         }
@@ -314 +320 @@
                 ApplicationExpr * appExpr = dynamic_cast< ApplicationExpr * >( expr );
                 if ( ! appExpr ) return NULL;
-                VariableExpr * function = getCalledFunction( appExpr );
+                DeclarationWithType * function = getCalledFunction( appExpr->get_function() );
                 assert( function );
                 // check for Intrinsic only - don't want to remove all overridable ctor/dtors because autogenerated ctor/dtor
                 // will call all member dtors, and some members may have a user defined dtor.
-                return function->get_var()->get_linkage() == LinkageSpec::Intrinsic ? appExpr : NULL;
+                return function->get_linkage() == LinkageSpec::Intrinsic ? appExpr : NULL;
         }
 
@@ -379 +385 @@
                         }       else if ( CastExpr * castExpr = dynamic_cast< CastExpr * >( func ) ) {
                                 return funcName( castExpr->get_arg() );
+                        } else if ( MemberExpr * memberExpr = dynamic_cast< MemberExpr * >( func ) ) {
+                                return memberExpr->get_member()->get_name();
                         } else {
                                 assert( false && "Unexpected expression type being called as a function in call expression" );
@@ -472 +480 @@
         bool isConstructor( const std::string & str ) { return str == "?{}"; }
         bool isDestructor( const std::string & str ) { return str == "^?{}"; }
+        bool isAssignment( const std::string & str ) { return str == "?=?"; }
         bool isCtorDtor( const std::string & str ) { return isConstructor( str ) || isDestructor( str ); }
-
-        FunctionDecl * isCopyConstructor( Declaration * decl ) {
+        bool isCtorDtorAssign( const std::string & str ) { return isCtorDtor( str ) || isAssignment( str ); }
+
+        FunctionDecl * isCopyFunction( Declaration * decl, const std::string & fname ) {
                 FunctionDecl * function = dynamic_cast< FunctionDecl * >( decl );
                 if ( ! function ) return 0;
-                if ( ! isConstructor( function->get_name() ) ) return 0;
+                if ( function->get_name() != fname ) return 0;
                 FunctionType * ftype = function->get_functionType();
                 if ( ftype->get_parameters().size() != 2 ) return 0;
@@ -486 +496 @@
                 assert( ptrType );
 
-                if ( ResolvExpr::typesCompatible( ptrType->get_base(), t2, SymTab::Indexer() ) ) {
+                if ( ResolvExpr::typesCompatibleIgnoreQualifiers( ptrType->get_base(), t2, SymTab::Indexer() ) ) {
                         return function;
                 } else {
-                        return 0;
-                }
+                        return nullptr;
+                }
+        }
+
+        FunctionDecl * isCopyConstructor( Declaration * decl ) {
+                return isCopyFunction( decl, "?{}" );
         }
 }