IRGen: Conditional conformances for variadic types

slavapestov · slavapestov · commit c84a14fdc178 · 2023-06-21T13:42:57.000-04:00
diff --git a/lib/IRGen/Fulfillment.cpp b/lib/IRGen/Fulfillment.cpp
@@ -228,8 +228,22 @@ bool FulfillmentMap::searchConformance(
 
   SILWitnessTable::enumerateWitnessTableConditionalConformances(
       conformance, [&](unsigned index, CanType type, ProtocolDecl *protocol) {
+        Optional<unsigned> packExpansionComponent;
+
+        if (auto packType = dyn_cast<PackType>(type)) {
+          auto param =
+              getSingletonPackExpansionParameter(packType, interestingKeys,
+                                                 packExpansionComponent);
+          if (!param)
+            return /*finished?*/ false;
+          type = param;
+        }
+
         MetadataPath conditionalPath = path;
         conditionalPath.addConditionalConformanceComponent(index);
+        if (packExpansionComponent)
+          conditionalPath.addPackExpansionPatternComponent(*packExpansionComponent);
+
         hadFulfillment |=
             searchWitnessTable(IGM, type, protocol, sourceIndex,
                                std::move(conditionalPath), interestingKeys);
diff --git a/lib/IRGen/GenMeta.cpp b/lib/IRGen/GenMeta.cpp
@@ -566,21 +566,7 @@ namespace {
       B.addInt(IGM.Int16Ty, shapes.size());
 
       // Emit each GenericPackShapeDescriptor collected previously.
-      for (const auto &packArg : packArgs) {
-        // Kind
-        B.addInt(IGM.Int16Ty, uint16_t(packArg.Kind));
-
-        // Index
-        B.addInt(IGM.Int16Ty, packArg.Index);
-
-        // ShapeClass
-        auto found = std::find(shapes.begin(), shapes.end(), packArg.ReducedShape);
-        assert(found != shapes.end());
-        B.addInt(IGM.Int16Ty, found - shapes.begin());
-
-        // Unused
-        B.addInt(IGM.Int16Ty, 0);
-      }
+      irgen::addGenericPackShapeDescriptors(IGM, B, shapes, packArgs);
     }
 
     uint8_t getVersion() {
@@ -6580,6 +6566,27 @@ GenericArgumentMetadata irgen::addGenericRequirements(
   return metadata;
 }
 
+void irgen::addGenericPackShapeDescriptors(IRGenModule &IGM,
+                                           ConstantStructBuilder &B,
+                                           ArrayRef<CanType> shapes,
+                                           ArrayRef<GenericPackArgument> packArgs) {
+  for (const auto &packArg : packArgs) {
+    // Kind
+    B.addInt(IGM.Int16Ty, uint16_t(packArg.Kind));
+
+    // Index
+    B.addInt(IGM.Int16Ty, packArg.Index);
+
+    // ShapeClass
+    auto found = std::find(shapes.begin(), shapes.end(), packArg.ReducedShape);
+    assert(found != shapes.end());
+    B.addInt(IGM.Int16Ty, found - shapes.begin());
+
+    // Unused
+    B.addInt(IGM.Int16Ty, 0);
+  }
+}
+
 //===----------------------------------------------------------------------===//
 // Other metadata.
 //===----------------------------------------------------------------------===//
diff --git a/lib/IRGen/GenMeta.h b/lib/IRGen/GenMeta.h
@@ -239,6 +239,15 @@ namespace irgen {
                                           GenericSignature sig,
                                           ArrayRef<Requirement> requirements);
 
+  /// Add generic pack shape descriptors to the given constant struct builder.
+  ///
+  /// These appear in generic type metadata, and conformance descriptors with
+  /// conditional pack requirements.
+  void addGenericPackShapeDescriptors(IRGenModule &IGM,
+                                      ConstantStructBuilder &B,
+                                      ArrayRef<CanType> shapes,
+                                      ArrayRef<GenericPackArgument> packArgs);
+
   llvm::GlobalValue *emitAsyncFunctionPointer(IRGenModule &IGM,
                                               llvm::Function *function,
                                               LinkEntity entity,
diff --git a/lib/IRGen/GenProto.cpp b/lib/IRGen/GenProto.cpp
@@ -1942,6 +1942,10 @@ namespace {
     ConformanceDescription Description;
     ConformanceFlags Flags;
 
+    using PlaceholderPosition =
+      ConstantAggregateBuilderBase::PlaceholderPosition;
+    Optional<PlaceholderPosition> FlagsPP;
+
   public:
     ProtocolConformanceDescriptorBuilder(
                                  IRGenModule &IGM,
@@ -1960,6 +1964,11 @@ namespace {
       addResilientWitnesses();
       addGenericWitnessTable();
 
+      // We fill the flags last, since we continue filling them in
+      // after the call to addFlags() deposits the placeholder.
+      B.fillPlaceholderWithInt(*FlagsPP, IGM.Int32Ty,
+                               Flags.getIntValue());
+
       B.suggestType(IGM.ProtocolConformanceDescriptorTy);
     }
 
@@ -1981,13 +1990,6 @@ namespace {
     }
 
     void addWitnessTable() {
-      // Note the number of conditional requirements.
-      unsigned numConditional = 0;
-      if (auto normal = dyn_cast<NormalProtocolConformance>(Conformance)) {
-        numConditional = normal->getConditionalRequirements().size();
-      }
-      Flags = Flags.withNumConditionalRequirements(numConditional);
-
       // Relative reference to the witness table.
       B.addRelativeAddressOrNull(Description.pattern);
     }
@@ -2001,13 +2003,9 @@ namespace {
         Flags = Flags.withIsRetroactive(false)
                      .withIsSynthesizedNonUnique(false);
       }
-      Flags = Flags.withHasResilientWitnesses(
-                                      !Description.resilientWitnesses.empty());
-      Flags =
-        Flags.withHasGenericWitnessTable(Description.requiresSpecialization);
 
-      // Add the flags.
-      B.addInt32(Flags.getIntValue());
+      // Add a placeholder for the flags.
+      FlagsPP = B.addPlaceholderWithSize(IGM.Int32Ty);
     }
 
     void addContext() {
@@ -2025,19 +2023,42 @@ namespace {
 
     void addConditionalRequirements() {
       auto normal = dyn_cast<NormalProtocolConformance>(Conformance);
-      if (!normal || normal->getConditionalRequirements().empty())
+      if (!normal)
         return;
 
-      auto nominal = normal->getType()->getAnyNominal();
-      irgen::addGenericRequirements(IGM, B,
-        nominal->getGenericSignatureOfContext(),
-        normal->getConditionalRequirements());
+      auto condReqs = normal->getConditionalRequirements();
+      if (condReqs.empty())
+        return;
+
+      Flags = Flags.withNumConditionalRequirements(condReqs.size());
+
+      auto nominal = normal->getDeclContext()->getSelfNominalTypeDecl();
+      auto sig = nominal->getGenericSignatureOfContext();
+      auto metadata = irgen::addGenericRequirements(IGM, B, sig, condReqs);
+
+      Flags = Flags.withNumConditionalPackDescriptors(
+          metadata.GenericPackArguments.size());
+
+      // Collect the shape classes from the nominal type's generic signature.
+      sig->forEachParam([&](GenericTypeParamType *param, bool canonical) {
+        if (canonical && param->isParameterPack()) {
+          auto reducedShape = sig->getReducedShape(param)->getCanonicalType();
+          if (reducedShape->isEqual(param))
+            metadata.ShapeClasses.push_back(reducedShape);
+        }
+      });
+
+      irgen::addGenericPackShapeDescriptors(
+          IGM, B, metadata.ShapeClasses,
+          metadata.GenericPackArguments);
     }
 
     void addResilientWitnesses() {
       if (Description.resilientWitnesses.empty())
         return;
 
+      Flags = Flags.withHasResilientWitnesses(true);
+
       // TargetResilientWitnessesHeader
       ArrayRef<llvm::Constant *> witnesses = Description.resilientWitnesses;
       B.addInt32(witnesses.size());
@@ -2098,6 +2119,8 @@ namespace {
       if (!Description.requiresSpecialization)
         return;
 
+      Flags = Flags.withHasGenericWitnessTable(true);
+
       // WitnessTableSizeInWords
       B.addInt(IGM.Int16Ty, Description.witnessTableSize);
       // WitnessTablePrivateSizeInWordsAndRequiresInstantiation
diff --git a/lib/IRGen/LocalTypeData.cpp b/lib/IRGen/LocalTypeData.cpp
@@ -473,18 +473,24 @@ void IRGenFunction::bindLocalTypeDataFromSelfWitnessTable(
   SILWitnessTable::enumerateWitnessTableConditionalConformances(
       conformance,
       [&](unsigned index, CanType type, ProtocolDecl *proto) {
-        auto archetype = getTypeInContext(type);
-        if (isa<ArchetypeType>(archetype)) {
+        if (auto packType = dyn_cast<PackType>(type)) {
+          if (auto expansion = packType.unwrapSingletonPackExpansion())
+            type = expansion.getPatternType();
+        }
+
+        type = getTypeInContext(type);
+
+        if (isa<ArchetypeType>(type)) {
           WitnessIndex wIndex(privateWitnessTableIndexToTableOffset(index),
                               /*prefix*/ false);
 
           auto table = loadConditionalConformance(*this ,selfTable,
                                                   wIndex.forProtocolWitnessTable());
           table = Builder.CreateBitCast(table, IGM.WitnessTablePtrTy);
-          setProtocolWitnessTableName(IGM, table, archetype, proto);
+          setProtocolWitnessTableName(IGM, table, type, proto);
 
           setUnscopedLocalTypeData(
-              archetype,
+              type,
               LocalTypeDataKind::forAbstractProtocolWitnessTable(proto),
               table);
         }
diff --git a/test/IRGen/variadic_generic_conditional_conformances.swift b/test/IRGen/variadic_generic_conditional_conformances.swift
@@ -0,0 +1,13 @@
+// RUN: %target-swift-frontend -emit-ir %s -disable-availability-checking | %FileCheck %s
+
+protocol P {
+  static func foobar()
+}
+
+struct G<each T> {}
+
+extension G: P where repeat each T: P {
+  static func foobar() {}
+}
+
+// CHECK-LABEL: define internal swiftcc void @"$s41variadic_generic_conditional_conformances1GVyxxQp_QPGAA1PA2aEP6foobaryyFZTW"(ptr swiftself %0, ptr %Self, ptr %SelfWitnessTable)
