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

AST: Checking of pack requirements

Browse Source
This commit is contained in:
Slava Pestov
2023-03-17 17:48:35 -04:00
parent d27231a91c
commit b6b51cf4cf
7 changed files with 146 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

44
lib/AST/Requirement.cpp
View File

@@ -82,12 +82,36 @@ CheckRequirementResult Requirement::checkRequirement(
if (hasError())
return CheckRequirementResult::SubstitutionFailure;
auto firstType = getFirstType();
auto expandPackRequirement = [&](PackType *packType) {
for (auto eltType : packType->getElementTypes()) {
// FIXME: Doesn't seem right
if (auto *expansionType = eltType->getAs<PackExpansionType>())
eltType = expansionType->getPatternType();
auto kind = getKind();
if (kind == RequirementKind::Layout) {
subReqs.emplace_back(kind, eltType,
getLayoutConstraint());
} else {
subReqs.emplace_back(kind, eltType,
getSecondType());
}
}
return CheckRequirementResult::PackRequirement;
};
switch (getKind()) {
case RequirementKind::Conformance: {
if (auto packType = firstType->getAs<PackType>()) {
return expandPackRequirement(packType);
}
auto *proto = getProtocolDecl();
auto *module = proto->getParentModule();
auto conformance = module->lookupConformance(
getFirstType(), proto, allowMissing);
firstType, proto, allowMissing);
if (!conformance)
return CheckRequirementResult::RequirementFailure;
@@ -99,7 +123,11 @@ CheckRequirementResult Requirement::checkRequirement(
}
case RequirementKind::Layout: {
if (auto *archetypeType = getFirstType()->getAs<ArchetypeType>()) {
if (auto packType = firstType->getAs<PackType>()) {
return expandPackRequirement(packType);
}
if (auto *archetypeType = firstType->getAs<ArchetypeType>()) {
auto layout = archetypeType->getLayoutConstraint();
if (layout && layout.merge(getLayoutConstraint()))
return CheckRequirementResult::Success;
@@ -108,7 +136,7 @@ CheckRequirementResult Requirement::checkRequirement(
}
if (getLayoutConstraint()->isClass()) {
if (getFirstType()->satisfiesClassConstraint())
if (firstType->satisfiesClassConstraint())
return CheckRequirementResult::Success;
return CheckRequirementResult::RequirementFailure;
@@ -120,19 +148,23 @@ CheckRequirementResult Requirement::checkRequirement(
}
case RequirementKind::Superclass:
if (getSecondType()->isExactSuperclassOf(getFirstType()))
if (auto packType = firstType->getAs<PackType>()) {
return expandPackRequirement(packType);
}
if (getSecondType()->isExactSuperclassOf(firstType))
return CheckRequirementResult::Success;
return CheckRequirementResult::RequirementFailure;
case RequirementKind::SameType:
if (getFirstType()->isEqual(getSecondType()))
if (firstType->isEqual(getSecondType()))
return CheckRequirementResult::Success;
return CheckRequirementResult::RequirementFailure;
case RequirementKind::SameShape:
if (getFirstType()->getReducedShape() ==
if (firstType->getReducedShape() ==
getSecondType()->getReducedShape())
return CheckRequirementResult::Success;