SIL: add getCalleeFunction utility also in ApplySite

danilodorgam · Nov 16, 2015 · ea5c51a · ea5c51a
1 parent 3e894f3
commit ea5c51a
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/include/swift/SIL/SILInstruction.h b/include/swift/SIL/SILInstruction.h
@@ -4238,6 +4238,11 @@ class ApplySite {
     FOREACH_IMPL_RETURN(getCallee());
   }
 
+  // Return the referenced function if the callee is a function_ref instruction.
+  SILFunction *getCalleeFunction() const {
+    FOREACH_IMPL_RETURN(getCalleeFunction());
+  }
+
   /// Return the type.
   SILType getType() const {
     FOREACH_IMPL_RETURN(getSubstCalleeType()->getResult().getSILType());