SIL: add some DeclRef APIs

* `var decl`
* `func ==`
* `func calleesAreStaticallyKnowable`

Author: eeckstein

SwiftCompilerSources/Sources/Optimizer/PassManager/Context.swift

@@ -791,3 +791,9 @@ extension Function {
     return context._bridged.appendBlock(bridged).block
   }
 }
+
+extension DeclRef {
+  func calleesAreStaticallyKnowable(_ context: some Context) -> Bool {
+    context._bridged.calleesAreStaticallyKnowable(bridged)
+  }
+}

SwiftCompilerSources/Sources/SIL/DeclRef.swift

@@ -11,11 +11,23 @@
 //===----------------------------------------------------------------------===//
 
 import SILBridging
+import AST
 
+/// A key for referencing an AST declaration in SIL.
+///
+/// In addition to the AST reference, there are discriminators for referencing different
+/// implementation-level entities associated with a single language-level declaration,
+/// such as the allocating and initializing entry points of a constructor, etc.
 public struct DeclRef: CustomStringConvertible, NoReflectionChildren {
   public let bridged: BridgedDeclRef
 
   public var location: Location { Location(bridged: bridged.getLocation()) }
 
   public var description: String { String(taking: bridged.getDebugDescription()) }
+
+  public var decl: Decl { bridged.getDecl().decl }
+
+  public static func ==(lhs: DeclRef, rhs: DeclRef) -> Bool {
+    lhs.bridged.isEqualTo(rhs.bridged)
+  }
 }

include/swift/SIL/SILBridging.h

@@ -927,8 +927,10 @@ struct BridgedDeclRef {
   BRIDGED_INLINE BridgedDeclRef(swift::SILDeclRef declRef);
   BRIDGED_INLINE swift::SILDeclRef unbridged() const;
 
+  BRIDGED_INLINE bool isEqualTo(BridgedDeclRef rhs) const;
   BridgedOwnedString getDebugDescription() const;
   SWIFT_IMPORT_UNSAFE BRIDGED_INLINE BridgedLocation getLocation() const;
+  SWIFT_IMPORT_UNSAFE BRIDGED_INLINE BridgedDeclObj getDecl() const;
   SWIFT_IMPORT_UNSAFE BRIDGED_INLINE BridgedDiagnosticArgument asDiagnosticArgument() const;
 };
 

include/swift/SIL/SILBridgingImpl.h

@@ -1726,10 +1726,18 @@ swift::SILDeclRef BridgedDeclRef::unbridged() const {
   return *reinterpret_cast<const swift::SILDeclRef *>(&storage);
 }
 
+bool BridgedDeclRef::isEqualTo(BridgedDeclRef rhs) const {
+  return unbridged() == rhs.unbridged();
+}
+
 BridgedLocation BridgedDeclRef::getLocation() const {
   return swift::SILDebugLocation(unbridged().getDecl(), nullptr);
 }
 
+BridgedDeclObj BridgedDeclRef::getDecl() const {
+  return {unbridged().getDecl()};
+}
+
 BridgedDiagnosticArgument BridgedDeclRef::asDiagnosticArgument() const {
   return swift::DiagnosticArgument(unbridged().getDecl()->getName());
 }

include/swift/SILOptimizer/OptimizerBridging.h

@@ -339,6 +339,7 @@ struct BridgedPassContext {
   SWIFT_IMPORT_UNSAFE OptionalBridgedFunction lookUpNominalDeinitFunction(BridgedDeclObj nominal) const;
   SWIFT_IMPORT_UNSAFE BRIDGED_INLINE BridgedSubstitutionMap getContextSubstitutionMap(BridgedType type) const;
   SWIFT_IMPORT_UNSAFE BRIDGED_INLINE BridgedType getBuiltinIntegerType(SwiftInt bitWidth) const;
+  BRIDGED_INLINE bool calleesAreStaticallyKnowable(BridgedDeclRef method) const;
   SWIFT_IMPORT_UNSAFE BridgedFunction createEmptyFunction(BridgedStringRef name,
                                                           const BridgedParameterInfo * _Nullable bridgedParams,
                                                           SwiftInt paramCount,

include/swift/SILOptimizer/OptimizerBridgingImpl.h

@@ -488,6 +488,11 @@ BridgedType BridgedPassContext::getBuiltinIntegerType(SwiftInt bitWidth) const {
   return swift::SILType::getBuiltinIntegerType(bitWidth, ctxt);
 }
 
+bool BridgedPassContext::calleesAreStaticallyKnowable(BridgedDeclRef method) const {
+  swift::SILModule *mod = invocation->getPassManager()->getModule();
+  return swift::calleesAreStaticallyKnowable(*mod, method.unbridged());
+}
+
 void BridgedPassContext::beginTransformFunction(BridgedFunction function) const {
   invocation->beginTransformFunction(function.getFunction());
 }