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

Add a helper function for checking if a tuple AP contains expansions

Browse Source
This commit is contained in:
John McCall
2023-03-09 02:16:09 -05:00
parent 2d72da3942
commit a15a2e86fe
2 changed files with 36 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

34
lib/SIL/IR/AbstractionPattern.cpp
View File

@@ -430,6 +430,40 @@ AbstractionPattern::getTupleElementType(unsigned index) const {
llvm_unreachable("bad kind");
}
bool AbstractionPattern::doesTupleContainPackExpansionType() const {
switch (getKind()) {
case Kind::Invalid:
llvm_unreachable("querying invalid abstraction pattern!");
case Kind::Opaque:
case Kind::PartialCurriedObjCMethodType:
case Kind::CurriedObjCMethodType:
case Kind::CFunctionAsMethodType:
case Kind::CurriedCFunctionAsMethodType:
case Kind::PartialCurriedCFunctionAsMethodType:
case Kind::ObjCMethodType:
case Kind::CXXMethodType:
case Kind::CurriedCXXMethodType:
case Kind::PartialCurriedCXXMethodType:
case Kind::OpaqueFunction:
case Kind::OpaqueDerivativeFunction:
llvm_unreachable("pattern is not a tuple");
case Kind::Tuple: {
for (auto &elt : llvm::makeArrayRef(OrigTupleElements,
getNumTupleElements_Stored())) {
if (elt.isPackExpansion())
return true;
}
return true;
}
case Kind::ObjCCompletionHandlerArgumentsType:
case Kind::Type:
case Kind::Discard:
case Kind::ClangType:
return cast<TupleType>(getType()).containsPackExpansionType();
}
llvm_unreachable("bad kind");
}
static CanType getCanPackElementType(CanType type, unsigned index) {
return cast<PackType>(type).getElementType(index);
}