SIL: add some APIs to SubstitutionMap

Author: eeckstein

SwiftCompilerSources/Sources/SIL/SubstitutionMap.swift

@@ -27,8 +27,32 @@ public struct SubstitutionMap {
 
   public var hasAnySubstitutableParams: Bool { bridged.hasAnySubstitutableParams() }
 
+  public var conformances: ConformanceArray { ConformanceArray(substitutionMap: self) }
+
   public var replacementTypes: OptionalTypeArray {
     let types = BridgedTypeArray.fromReplacementTypes(bridged)
     return OptionalTypeArray(bridged: types)
   }
+
+  public func getMethodSubstitutions(for method: Function) -> SubstitutionMap {
+    return SubstitutionMap(bridged: bridged.getMethodSubstitutions(method.bridged))
+  }
+
+  public struct ConformanceArray : BridgedRandomAccessCollection {
+    fileprivate let bridgedSubs: BridgedSubstitutionMap
+    public let count: Int
+
+    init(substitutionMap: SubstitutionMap) {
+      self.bridgedSubs = substitutionMap.bridged
+      self.count = substitutionMap.bridged.getNumConformances()
+    }
+
+    public var startIndex: Int { return 0 }
+    public var endIndex: Int { return count }
+
+    public subscript(_ index: Int) -> ProtocolConformance {
+      assert(index >= startIndex && index < endIndex)
+      return ProtocolConformance(bridged: bridgedSubs.getConformance(index))
+    }
+  }
 }

include/swift/SIL/SILBridging.h

@@ -652,6 +652,9 @@ struct BridgedSubstitutionMap {
   BRIDGED_INLINE BridgedSubstitutionMap();
   BRIDGED_INLINE bool isEmpty() const;
   BRIDGED_INLINE bool hasAnySubstitutableParams() const;
+  SWIFT_IMPORT_UNSAFE BridgedSubstitutionMap getMethodSubstitutions(BridgedFunction method) const;
+  BRIDGED_INLINE SwiftInt getNumConformances() const;
+  SWIFT_IMPORT_UNSAFE BRIDGED_INLINE BridgedProtocolConformance getConformance(SwiftInt index) const;
 };
 
 struct BridgedTypeArray {

include/swift/SIL/SILBridgingImpl.h

@@ -710,6 +710,14 @@ bool BridgedSubstitutionMap::hasAnySubstitutableParams() const {
   return unbridged().hasAnySubstitutableParams();
 }
 
+SwiftInt BridgedSubstitutionMap::getNumConformances() const {
+  return (SwiftInt)unbridged().getConformances().size();
+}
+
+BridgedProtocolConformance BridgedSubstitutionMap::getConformance(SwiftInt index) const {
+  return unbridged().getConformances()[index];
+}
+
 //===----------------------------------------------------------------------===//
 //                                BridgedLocation
 //===----------------------------------------------------------------------===//

lib/SIL/Utils/SILBridging.cpp

@@ -381,6 +381,17 @@ BridgedOwnedString BridgedDefaultWitnessTable::getDebugDescription() const {
 static_assert(sizeof(BridgedSubstitutionMap) >= sizeof(swift::SubstitutionMap),
               "BridgedSubstitutionMap has wrong size");
 
+BridgedSubstitutionMap BridgedSubstitutionMap::getMethodSubstitutions(BridgedFunction method) const {
+  swift::SILFunction *f = method.getFunction();
+  swift::GenericSignature genericSig = f->getLoweredFunctionType()->getInvocationGenericSignature();
+
+  if (!genericSig || genericSig->areAllParamsConcrete())
+    return swift::SubstitutionMap();
+
+  return swift::SubstitutionMap::get(genericSig,
+                                     swift::QuerySubstitutionMap{unbridged()},
+                                     swift::LookUpConformanceInModule());
+}
 
 //===----------------------------------------------------------------------===//
 //                               SILDebugLocation