Merge pull request swiftlang#87301 from gottesmm/pr-fd56e8499232cd2e6686191b9453ad1cda0cbca9

[sil] Add support for ApplySite::getCalleeDeclRef.

Author: gottesmm

include/swift/SIL/ApplySite.h

@@ -170,6 +170,12 @@ class ApplySite {
     FOREACH_IMPL_RETURN(getCalleeFunction());
   }
 
+  /// Gets the referenced function by looking through partial apply,
+  /// convert_function, and thin to thick function until we find a function_ref.
+  SILDeclRef getCalleeDeclRef() const {
+    FOREACH_IMPL_RETURN(getCalleeDeclRef());
+  }
+
   bool isCalleeDynamicallyReplaceable() const {
     FOREACH_IMPL_RETURN(isCalleeDynamicallyReplaceable());
   }

include/swift/SIL/SILInstruction.h

@@ -2802,6 +2802,18 @@ class ApplyInstBase<Impl, Base, false> : public Base {
   /// issues. It is at the bottom of the file.
   SILFunction *getCalleeFunction() const;
 
+  /// Returns the SILDeclRef for a referenced function after looking through
+  /// partial_apply, convert_function, and thin_to_thick_function.
+  ///
+  /// If we do not find something that we know how to handle, we return an empty
+  /// SILDeclRef.
+  ///
+  /// This is based off of getCalleeOrigin.
+  ///
+  /// This is defined out of line to work around incomplete definition
+  /// issues. It is next to getCalleeFunction at the bottom of the file.
+  SILDeclRef getCalleeDeclRef() const;
+
   bool isCalleeDynamicallyReplaceable() const;
 
   /// Gets the referenced function if the callee is a function_ref instruction.

lib/SIL/IR/SILInstruction.cpp

@@ -2179,6 +2179,31 @@ SILModule &SILInstructionContext::getModule() {
   return storage.get<SILFunction *>()->getModule();
 }
 
+template <class Impl, class Base>
+SILDeclRef ApplyInstBase<Impl, Base, false>::getCalleeDeclRef() const {
+  SILValue origin = getCalleeOrigin();
+  if (!origin)
+    return {};
+
+  if (auto *fri = dyn_cast<FunctionRefInst>(origin))
+    return fri->getReferencedFunction()->getDeclRef();
+
+  if (auto *m = dyn_cast<MethodInst>(origin))
+    return m->getMember();
+
+  return {};
+}
+
+template SILDeclRef
+ApplyInstBase<ApplyInst, SingleValueInstruction, false>::getCalleeDeclRef()
+    const;
+template SILDeclRef ApplyInstBase<PartialApplyInst, SingleValueInstruction,
+                                  false>::getCalleeDeclRef() const;
+template SILDeclRef ApplyInstBase<BeginApplyInst, MultipleValueInstruction,
+                                  false>::getCalleeDeclRef() const;
+template SILDeclRef
+ApplyInstBase<TryApplyInst, TryApplyInstBase, false>::getCalleeDeclRef() const;
+
 #ifndef NDEBUG
 
 //---