Bridging: Implement bridges required for ongoing AutoDiff changes

In swiftlang#83926, part of the changes resolving swiftlang#68944 is submitted. The
AutoDiff closure specialization optimizer pass relies on several
AST-level bridges not implemented yet.

This patch introduces these missing bridges.

Author: kovdan01

SwiftCompilerSources/Sources/AST/Declarations.swift

@@ -56,6 +56,10 @@ public class NominalTypeDecl: GenericTypeDecl {
   final public var valueTypeDestructor: DestructorDecl? {
     bridged.NominalType_getValueTypeDestructor().getAs(DestructorDecl.self)
   }
+
+  public var declaredInterfaceType : AST.`Type` {
+    AST.`Type`(bridged: bridged.NominalType_getDeclaredInterfaceType())
+  }
 }
 
 final public class EnumDecl: NominalTypeDecl {
@@ -118,6 +122,8 @@ final public class MacroDecl: ValueDecl {}
 
 final public class EnumElementDecl: ValueDecl {
   public var hasAssociatedValues: Bool { bridged.EnumElementDecl_hasAssociatedValues() }
+  public var parameterList: BridgedParameterList { bridged.EnumElementDecl_getParameterList() }
+  public var name: StringRef { StringRef(bridged: bridged.EnumElementDecl_getNameStr()) }
 }
 
 final public class ExtensionDecl: Decl {}

SwiftCompilerSources/Sources/AST/GenericSignature.swift

@@ -35,4 +35,18 @@ public struct GenericSignature: CustomStringConvertible, NoReflectionChildren {
   }
 
   public var isEmpty: Bool { bridged.impl == nil }
+
+  public var canonicalSignature: CanGenericSignature { CanGenericSignature(bridged: bridged.getCanonicalSignature()) }
+}
+
+public struct CanGenericSignature {
+  public let bridged: BridgedCanGenericSignature
+
+  public init(bridged: BridgedCanGenericSignature) {
+    self.bridged = bridged
+  }
+
+  public var isEmpty: Bool { bridged.impl == nil }
+
+  public var genericSignature: GenericSignature { GenericSignature(bridged: bridged.getGenericSignature()) }
 }

SwiftCompilerSources/Sources/AST/Type.swift

@@ -66,6 +66,14 @@ public struct Type: TypeProperties, CustomStringConvertible, NoReflectionChildre
   public func subst(with substitutionMap: SubstitutionMap) -> Type {
     return Type(bridged: bridged.subst(substitutionMap.bridged))
   }
+
+  public func mapTypeOutOfContext() -> Type {
+    return Type(bridged: bridged.mapTypeOutOfContext())
+  }
+
+  public func getReducedType(sig: GenericSignature) -> CanonicalType {
+    CanonicalType(bridged: bridged.getReducedType(sig.bridged))
+  }
 }
 
 /// A Type that is statically known to be canonical.
@@ -86,6 +94,10 @@ public struct CanonicalType: TypeProperties, CustomStringConvertible, NoReflecti
   public func subst(with substitutionMap: SubstitutionMap) -> CanonicalType {
     return rawType.subst(with: substitutionMap).canonical
   }
+
+  public func SILFunctionType_getSubstGenericSignature() -> CanGenericSignature {
+    CanGenericSignature(bridged: bridged.SILFunctionType_getSubstGenericSignature())
+  }
 }
 
 /// Implements the common members of `AST.Type`, `AST.CanonicalType` and `SIL.Type`.

SwiftCompilerSources/Sources/SIL/ASTExtensions.swift

@@ -36,6 +36,10 @@ extension CanonicalType {
     precondition(isBox)
     return BoxFieldsArray(boxType: self, function: function)
   }
+
+  public func loweredType(in function: Function) -> Type {
+    function.bridged.getLoweredType(bridged).type
+  }
 }
 
 extension Decl {

SwiftCompilerSources/Sources/SIL/Context.swift

@@ -55,6 +55,20 @@ extension Context {
     }
   }
 
+  public func getTupleType(elements: [AST.`Type`]) -> AST.`Type` {
+    let bridgedElements = elements.map { $0.bridged }
+    return bridgedElements.withBridgedArrayRef {
+      AST.`Type`(bridged: _bridged.getTupleType($0))
+    }
+  }
+
+  public func getTupleTypeWithLabels(elements: [AST.`Type`], labels: [swift.Identifier]) -> AST.`Type` {
+    assert(elements.count == labels.count)
+    return elements.withBridgedArrayRef{
+      eltArr in labels.withBridgedArrayRef{labelsArr in
+      AST.`Type`(bridged: _bridged.getTupleTypeWithLabels(eltArr, labelsArr))}}
+  }
+
   public var swiftArrayDecl: NominalTypeDecl {
     _bridged.getSwiftArrayDecl().getAs(NominalTypeDecl.self)
   }

SwiftCompilerSources/Sources/SIL/Type.swift

@@ -27,6 +27,7 @@ public struct Type : TypeProperties, CustomStringConvertible, NoReflectionChildr
 
   public var isAddress: Bool { bridged.isAddress() }
   public var isObject: Bool { !isAddress }
+  public var category: ValueCategory { ValueCategory(bridged: bridged.getCategory()) }
 
   public var addressType: Type { bridged.getAddressType().type }
   public var objectType: Type { bridged.getObjectType().type }
@@ -214,6 +215,12 @@ public struct Type : TypeProperties, CustomStringConvertible, NoReflectionChildr
     }
     return false
   }
+
+  public func mapTypeOutOfContext() -> Type { bridged.mapTypeOutOfContext().type }
+
+  public static func getPrimitiveType(canType: CanonicalType, silValueCategory: ValueCategory) -> Type {
+    BridgedType.getPrimitiveType(canType: canType.bridged, silValueCategory: silValueCategory._bridged).type
+  }
 }
 
 extension Type: Equatable {
@@ -266,6 +273,7 @@ public struct NominalFieldsArray : RandomAccessCollection, FormattedLikeArray {
 }
 
 public struct EnumCase {
+  public let enumElementDecl : EnumElementDecl
   public let payload: Type?
   public let index: Int
 }
@@ -288,7 +296,8 @@ public struct EnumCases : CollectionLikeSequence, IteratorProtocol {
         caseIterator = caseIterator.getNext()
         caseIndex += 1
       }
-      return EnumCase(payload: enumType.bridged.getEnumCasePayload(caseIterator, function.bridged).typeOrNil,
+      return EnumCase(enumElementDecl: enumType.bridged.getEnumElementDecl(caseIterator).getAs(EnumElementDecl.self),
+                      payload: enumType.bridged.getEnumCasePayload(caseIterator, function.bridged).typeOrNil,
                       index: caseIndex)
     }
     return nil
@@ -304,6 +313,10 @@ public struct TupleElementArray : RandomAccessCollection, FormattedLikeArray {
   public subscript(_ index: Int) -> Type {
     type.bridged.getTupleElementType(index).type
   }
+
+  public func label(at index: Int) -> swift.Identifier {
+    type.bridged.getTupleElementLabel(index)
+  }
 }
 
 public struct BoxFieldsArray : RandomAccessCollection, FormattedLikeArray {

SwiftCompilerSources/Sources/SIL/Value.swift

@@ -41,6 +41,27 @@ public protocol Value : AnyObject, CustomStringConvertible {
   var isLexical: Bool { get }
 }
 
+public enum ValueCategory {
+  case address
+  case object
+
+  public init(bridged: BridgedValueCategory) {
+    switch bridged {
+    case .Address: self = .address
+    case .Object:  self = .object
+    default:
+      fatalError("unsupported value category")
+    }
+  }
+
+  public var _bridged: BridgedValueCategory {
+    switch self {
+      case .address: return BridgedValueCategory.Address
+      case .object:  return BridgedValueCategory.Object
+    }
+  }
+}
+
 public enum Ownership {
   /// A Value with `unowned` ownership kind is an independent value that
   /// has a lifetime that is only guaranteed to last until the next program

include/swift/AST/ASTBridging.h

@@ -84,6 +84,7 @@ class BridgedCanType;
 class BridgedASTContext;
 struct BridgedSubstitutionMap;
 struct BridgedGenericSignature;
+struct BridgedCanGenericSignature;
 struct BridgedConformance;
 class BridgedParameterList;
 
@@ -369,6 +370,7 @@ struct BridgedDeclObj {
   BRIDGED_INLINE bool AbstractStorage_isConst() const;
   BRIDGED_INLINE bool GenericType_isGenericAtAnyLevel() const;
   BRIDGED_INLINE bool NominalType_isGlobalActor() const;
+  SWIFT_IMPORT_UNSAFE BRIDGED_INLINE BridgedASTType NominalType_getDeclaredInterfaceType() const;
   SWIFT_IMPORT_UNSAFE BRIDGED_INLINE OptionalBridgedDeclObj NominalType_getValueTypeDestructor() const;
   BRIDGED_INLINE bool Enum_hasRawType() const;
   BRIDGED_INLINE bool Struct_hasUnreferenceableStorage() const;
@@ -378,6 +380,10 @@ struct BridgedDeclObj {
   BRIDGED_INLINE bool AbstractFunction_isOverridden() const;
   BRIDGED_INLINE bool Destructor_isIsolated() const;
   BRIDGED_INLINE bool EnumElementDecl_hasAssociatedValues() const;
+  SWIFT_IMPORT_UNSAFE BRIDGED_INLINE BridgedParameterList
+  EnumElementDecl_getParameterList() const;
+  SWIFT_IMPORT_UNSAFE BRIDGED_INLINE BridgedStringRef
+  EnumElementDecl_getNameStr() const;
   SWIFT_IMPORT_UNSAFE BRIDGED_INLINE BridgedStringRef AccessorDecl_getKindName() const;
 };
 
@@ -431,6 +437,23 @@ class BridgedASTNode {
 #define ABSTRACT_DECL(Id, Parent) DECL(Id, Parent)
 #include "swift/AST/DeclNodes.def"
 
+// Declare `.asValueDecl` on each BridgedXXXDecl type that's also a
+// ValueDecl.
+#define DECL(Id, Parent)
+#define VALUE_DECL(Id, Parent)                                                 \
+  SWIFT_NAME("getter:Bridged" #Id "Decl.asValueDecl(self:)")                   \
+  BridgedValueDecl Bridged##Id##Decl_asValueDecl(Bridged##Id##Decl decl);
+#include "swift/AST/DeclNodes.def"
+
+// Declare `.asNominalTypeDecl` on each BridgedXXXDecl type that's also a
+// NominalTypeDecl.
+#define DECL(Id, Parent)
+#define NOMINAL_TYPE_DECL(Id, Parent)                                          \
+  SWIFT_NAME("getter:Bridged" #Id "Decl.asNominalTypeDecl(self:)")             \
+  BridgedNominalTypeDecl Bridged##Id##Decl_asNominalTypeDecl(                  \
+      Bridged##Id##Decl decl);
+#include "swift/AST/DeclNodes.def"
+
 // Declare `.asDeclContext` on each BridgedXXXDecl type that's also a
 // DeclContext.
 #define DECL(Id, Parent)
@@ -440,6 +463,16 @@ class BridgedASTNode {
 #define ABSTRACT_CONTEXT_DECL(Id, Parent) CONTEXT_DECL(Id, Parent)
 #include "swift/AST/DeclNodes.def"
 
+// Declare `.asGenericContext` on each BridgedXXXDecl type that's also a
+// GenericContext.
+#define DECL(Id, Parent)
+#define GENERIC_DECL(Id, Parent)                                               \
+  SWIFT_NAME("getter:Bridged" #Id "Decl.asGenericContext(self:)")              \
+  BridgedGenericContext Bridged##Id##Decl_asGenericContext(                    \
+      Bridged##Id##Decl decl);
+#define ITERABLE_GENERIC_DECL(Id, Parent) GENERIC_DECL(Id, Parent)
+#include "swift/AST/DeclNodes.def"
+
 // Declare `.asStmt` on each BridgedXXXStmt type, which upcasts a wrapper for
 // a Stmt subclass to a BridgedStmt.
 #define STMT(Id, Parent)                                                       \
@@ -617,6 +650,10 @@ BridgedDeclContext_getParentSourceFile(BridgedDeclContext dc);
 SWIFT_NAME("getter:BridgedSourceFile.isScriptMode(self:)")
 BRIDGED_INLINE bool BridgedSourceFile_isScriptMode(BridgedSourceFile sf);
 
+SWIFT_NAME("BridgedSourceFile.addTopLevelDecl(self:_:)")
+BRIDGED_INLINE void BridgedSourceFile_addTopLevelDecl(BridgedSourceFile sf,
+                                                      BridgedDecl decl);
+
 SWIFT_NAME("BridgedPatternBindingInitializer.create(declContext:)")
 BridgedPatternBindingInitializer
 BridgedPatternBindingInitializer_create(BridgedDeclContext cDeclContext);
@@ -1337,6 +1374,9 @@ SWIFT_NAME("BridgedDecl.forEachDeclToHoist(self:_:)")
 void BridgedDecl_forEachDeclToHoist(BridgedDecl decl,
                                     BridgedSwiftClosure closure);
 
+SWIFT_NAME("BridgedDecl.getDeclContext(self:)")
+BridgedDeclContext BridgedDecl_getDeclContext(BridgedDecl decl);
+
 enum ENUM_EXTENSIBILITY_ATTR(closed) BridgedStaticSpelling {
   BridgedStaticSpellingNone,
   BridgedStaticSpellingStatic,
@@ -1384,10 +1424,18 @@ BridgedParamDecl BridgedParamDecl_createParsed(
     swift::SourceLoc paramNameLoc, BridgedNullableExpr defaultValue,
     BridgedNullableDefaultArgumentInitializer cDefaultArgumentInitContext);
 
+SWIFT_NAME("BridgedParamDecl.cloneWithoutType(self:)")
+BRIDGED_INLINE BridgedParamDecl
+BridgedParamDecl_cloneWithoutType(BridgedParamDecl cDecl);
+
 SWIFT_NAME("BridgedParamDecl.setTypeRepr(self:_:)")
 BRIDGED_INLINE void BridgedParamDecl_setTypeRepr(BridgedParamDecl cDecl,
                                                  BridgedTypeRepr cType);
 
+SWIFT_NAME("BridgedParamDecl.setInterfaceType(self:_:)")
+BRIDGED_INLINE void BridgedParamDecl_setInterfaceType(BridgedParamDecl cDecl,
+                                                      BridgedASTType cType);
+
 /// The various spellings of ownership modifier that can be used in source.
 enum ENUM_EXTENSIBILITY_ATTR(closed) BridgedParamSpecifier {
   BridgedParamSpecifierDefault,
@@ -1406,8 +1454,13 @@ BRIDGED_INLINE void
 BridgedParamDecl_setSpecifier(BridgedParamDecl cDecl,
                               BridgedParamSpecifier cSpecifier);
 
-SWIFT_NAME("BridgedParamDecl.setImplicit(self:)")
-BRIDGED_INLINE void BridgedParamDecl_setImplicit(BridgedParamDecl cDecl);
+SWIFT_NAME("BridgedDecl.setImplicit(self:)")
+BRIDGED_INLINE void BridgedDecl_setImplicit(BridgedDecl cDecl);
+
+SWIFT_NAME("BridgedGenericContext.setGenericSignature(self:_:)")
+BRIDGED_INLINE void
+BridgedGenericContext_setGenericSignature(BridgedGenericContext cDecl,
+                                          BridgedGenericSignature cGenSig);
 
 SWIFT_NAME("BridgedConstructorDecl.setParsedBody(self:_:)")
 void BridgedConstructorDecl_setParsedBody(BridgedConstructorDecl decl,
@@ -1471,14 +1524,20 @@ void BridgedExtensionDecl_setParsedMembers(BridgedExtensionDecl decl,
 SWIFT_NAME(
     "BridgedEnumDecl.createParsed(_:declContext:enumKeywordLoc:name:nameLoc:"
     "genericParamList:inheritedTypes:genericWhereClause:braceRange:)")
-BridgedNominalTypeDecl BridgedEnumDecl_createParsed(
+BridgedEnumDecl BridgedEnumDecl_createParsed(
     BridgedASTContext cContext, BridgedDeclContext cDeclContext,
     swift::SourceLoc enumKeywordLoc, swift::Identifier name,
     swift::SourceLoc nameLoc, BridgedNullableGenericParamList genericParamList,
     BridgedArrayRef cInheritedTypes,
     BridgedNullableTrailingWhereClause genericWhereClause,
     swift::SourceRange braceRange);
 
+SWIFT_NAME("BridgedEnumDecl.create(_:declContext:name:genericParamList:)")
+BridgedEnumDecl
+BridgedEnumDecl_create(BridgedASTContext cContext,
+                       BridgedDeclContext cDeclContext, BridgedStringRef name,
+                       BridgedNullableGenericParamList genericParamList);
+
 SWIFT_NAME(
     "BridgedEnumCaseDecl.createParsed(declContext:caseKeywordLoc:elements:)")
 BridgedEnumCaseDecl
@@ -1494,6 +1553,12 @@ BridgedEnumElementDecl BridgedEnumElementDecl_createParsed(
     BridgedNullableParameterList parameterList, swift::SourceLoc equalsLoc,
     BridgedNullableExpr opaqueRawValue);
 
+SWIFT_NAME("BridgedEnumElementDecl.create(_:declContext:name:"
+           "parameterList:)")
+BridgedEnumElementDecl BridgedEnumElementDecl_create(
+    BridgedASTContext cContext, BridgedDeclContext cDeclContext,
+    BridgedStringRef name, BridgedParameterList parameterList);
+
 SWIFT_NAME("BridgedStructDecl.createParsed(_:declContext:structKeywordLoc:name:"
            "nameLoc:genericParamList:inheritedTypes:genericWhereClause:"
            "braceRange:)")
@@ -1662,6 +1727,9 @@ void BridgedTopLevelCodeDecl_dump(BridgedTopLevelCodeDecl decl);
 SWIFT_NAME("BridgedDecl.dump(self:)")
 void BridgedDecl_dump(BridgedDecl decl);
 
+SWIFT_NAME("BridgedValueDecl.setAccess(self:_:)")
+void BridgedValueDecl_setAccess(BridgedValueDecl decl, swift::AccessLevel accessLevel);
+
 //===----------------------------------------------------------------------===//
 // MARK: AbstractStorageDecl
 //===----------------------------------------------------------------------===//
@@ -1711,6 +1779,11 @@ SWIFT_NAME("BridgedNominalTypeDecl.getSourceLocation(self:)")
 BRIDGED_INLINE swift::SourceLoc
 BridgedNominalTypeDecl_getSourceLocation(BridgedNominalTypeDecl decl);
 
+SWIFT_NAME("BridgedNominalTypeDecl.addMember(self:_:)")
+BRIDGED_INLINE void
+BridgedNominalTypeDecl_addMember(BridgedNominalTypeDecl cDecl,
+                                 BridgedDecl member);
+
 //===----------------------------------------------------------------------===//
 // MARK: SubscriptDecl
 //===----------------------------------------------------------------------===//
@@ -2913,6 +2986,10 @@ BridgedParameterList BridgedParameterList_createParsed(
     BridgedASTContext cContext, swift::SourceLoc leftParenLoc,
     BridgedArrayRef cParameters, swift::SourceLoc rightParenLoc);
 
+SWIFT_NAME("BridgedParameterList.create(_:parameters:)")
+BridgedParameterList BridgedParameterList_create(BridgedASTContext cContext,
+                                                 BridgedArrayRef cParameters);
+
 SWIFT_NAME("getter:BridgedParameterList.size(self:)")
 size_t BridgedParameterList_size(BridgedParameterList cParameterList);
 
@@ -3017,6 +3094,8 @@ struct BridgedASTType {
   SWIFT_IMPORT_UNSAFE BRIDGED_INLINE BridgedGenericSignature getInvocationGenericSignatureOfFunctionType() const;
   SWIFT_IMPORT_UNSAFE BRIDGED_INLINE BridgedASTType subst(BridgedSubstitutionMap substMap) const;
   SWIFT_IMPORT_UNSAFE BRIDGED_INLINE BridgedConformance checkConformance(BridgedDeclObj proto) const;  
+  SWIFT_IMPORT_UNSAFE BRIDGED_INLINE BridgedASTType mapTypeOutOfContext() const;
+  SWIFT_IMPORT_UNSAFE BRIDGED_INLINE BridgedCanType getReducedType(BridgedGenericSignature sig) const;
 };
 
 class BridgedCanType {
@@ -3027,6 +3106,7 @@ class BridgedCanType {
   BRIDGED_INLINE BridgedCanType(swift::CanType ty);
   BRIDGED_INLINE swift::CanType unbridged() const;
   SWIFT_IMPORT_UNSAFE BRIDGED_INLINE BridgedASTType getRawType() const;
+  SWIFT_IMPORT_UNSAFE BRIDGED_INLINE BridgedCanGenericSignature SILFunctionType_getSubstGenericSignature() const;
 };
 
 struct BridgedASTTypeArray {
@@ -3091,6 +3171,14 @@ struct BridgedGenericSignature {
   BridgedOwnedString getDebugDescription() const;
   SWIFT_IMPORT_UNSAFE BRIDGED_INLINE BridgedASTTypeArray getGenericParams() const;
   SWIFT_IMPORT_UNSAFE BRIDGED_INLINE BridgedASTType mapTypeIntoContext(BridgedASTType type) const;
+  BRIDGED_INLINE BridgedCanGenericSignature getCanonicalSignature() const;
+};
+
+struct BridgedCanGenericSignature {
+  const swift::GenericSignatureImpl *_Nullable impl;
+
+  BRIDGED_INLINE swift::CanGenericSignature unbridged() const;
+  BRIDGED_INLINE BridgedGenericSignature getGenericSignature() const;
 };
 
 struct BridgedFingerprint {