Changes in src/InitTweak/InitTweak.cc [207c7e1d:b7b8674]

File:

• src/InitTweak/InitTweak.cc (modified) (1 diff)

src/InitTweak/InitTweak.cc

@@ -332 +332 @@
                         return nullptr;
                 }
+        }
+
+        DeclarationWithType * getFunction( Expression * expr ) {
+                if ( ApplicationExpr * appExpr = dynamic_cast< ApplicationExpr * >( expr ) ) {
+                        return getCalledFunction( appExpr->get_function() );
+                } else if ( UntypedExpr * untyped = dynamic_cast< UntypedExpr * > ( expr ) ) {
+                        return getCalledFunction( untyped->get_function() );
+                }
+                assertf( false, "getFunction received unknown expression: %s", toString( expr ).c_str() );
         }