averello/swift-mirror
1
0
Fork 0
mirror of https://github.com/apple/swift.git synced 2025-12-14 20:36:38 +01:00
Code Issues Packages Projects Releases Wiki Activity

[GenericEnvironment] Add a method to retrieve the bindings from pack element

Browse Source 
archetypes to their originating pack archetypes.
This commit is contained in:
Holly Borla
2022-11-23 12:52:54 -05:00
parent 8ec5405a30
commit 8db8b7c29a
2 changed files with 37 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
lib/AST/GenericEnvironment.cpp
View File

@@ -125,6 +125,33 @@ UUID GenericEnvironment::getOpenedElementUUID() const {
return getTrailingObjects<OpenedElementEnvironmentData>()->uuid;
}
void GenericEnvironment::getPackElementBindings(
SmallVectorImpl<PackElementBinding> &bindings) const {
auto packElements = getGenericSignature().getInnermostGenericParams();
auto packElementDepth = packElements.front()->getDepth();
auto elementIt = packElements.begin();
// Each parameter pack in the outer generic parameters has
// a corresponding pack element parameter at the innermost
// depth.
for (auto *genericParam : getGenericParams()) {
if (genericParam->getDepth() == packElementDepth)
break;
if (!genericParam->isParameterPack())
continue;
assert(elementIt != packElements.end());
auto *elementArchetype =
mapTypeIntoContext(*elementIt++)->getAs<ElementArchetypeType>();
auto *packArchetype =
mapTypeIntoContext(genericParam)->getAs<PackArchetypeType>();
assert(elementArchetype && packArchetype);
bindings.emplace_back(elementArchetype, packArchetype);
}
}
GenericEnvironment::GenericEnvironment(GenericSignature signature)
: SignatureAndKind(signature, Kind::Primary)
{