Merge pull request #8297 from gottesmm/add_getsubst_methods_to_argume…

…ntsource [silgen] Add helper methods to compute ArgumentSource::getSubst{RValue,}Type() as SILTypes.
swiftlang · Mar 24, 2017 · 73bc62c · 73bc62c
2 parents ae5c682 + 001345e
commit 73bc62c
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/lib/SILGen/ArgumentSource.cpp b/lib/SILGen/ArgumentSource.cpp
@@ -217,3 +217,17 @@ void ArgumentSource::forwardInto(SILGenFunction &SGF,
   auto substLoweredType = destTL.getLoweredType().getSwiftRValueType();
   RValue(SGF, loc, substLoweredType, outputValue).forwardInto(SGF, loc, dest);
 }
+
+SILType ArgumentSource::getSILSubstRValueType(SILGenFunction &SGF) const & {
+  CanSILFunctionType funcType = SGF.F.getLoweredFunctionType();
+  CanType substType = getSubstType();
+  AbstractionPattern origType(funcType->getGenericSignature(), substType);
+  return SGF.getLoweredType(origType, substType);
+}
+
+SILType ArgumentSource::getSILSubstType(SILGenFunction &SGF) const & {
+  CanSILFunctionType funcType = SGF.F.getLoweredFunctionType();
+  CanType substType = getSubstType();
+  AbstractionPattern origType(funcType->getGenericSignature(), substType);
+  return SGF.getLoweredType(origType, substType);
+}
diff --git a/lib/SILGen/ArgumentSource.h b/lib/SILGen/ArgumentSource.h
@@ -178,6 +178,8 @@ class ArgumentSource {
     llvm_unreachable("bad kind");
   }
 
+  SILType getSILSubstType(SILGenFunction &SGF) const &;
+
   CanType getSubstRValueType() const & {
     switch (StoredKind) {
     case Kind::RValue:
@@ -190,6 +192,8 @@ class ArgumentSource {
     llvm_unreachable("bad kind");
   }
 
+  SILType getSILSubstRValueType(SILGenFunction &SGF) const &;
+
   bool hasLValueType() const & {
     switch (StoredKind) {
     case Kind::RValue: return false;