Merge pull request #5414 from gottesmm/quick_utility_method

Author: swift-ci

include/swift/SIL/SILValue.h

@@ -29,6 +29,8 @@ namespace swift {
   class ValueBaseUseIterator;
   class ValueUseIterator;
   class SILBasicBlock;
+  class SILFunction;
+  class SILModule;
   class SILInstruction;
   class SILLocation;
   class DominanceInfo;
@@ -119,6 +121,14 @@ class alignas(8) ValueBase : public SILAllocated<ValueBase> {
   /// If this is a SILArgument or a SILInstruction get its parent basic block,
   /// otherwise return null.
   SILBasicBlock *getParentBB();
+
+  /// If this is a SILArgument or a SILInstruction get its parent function,
+  /// otherwise return null.
+  SILFunction *getFunction();
+
+  /// If this is a SILArgument or a SILInstruction get its parent module,
+  /// otherwise return null.
+  SILModule *getModule();
 };
 
 inline llvm::raw_ostream &operator<<(llvm::raw_ostream &OS,

lib/SIL/SILValue.cpp

@@ -38,3 +38,19 @@ SILBasicBlock *ValueBase::getParentBB() {
     return Arg->getParent();
   return nullptr;
 }
+
+SILFunction *ValueBase::getFunction() {
+  if (auto Inst = dyn_cast<SILInstruction>(this))
+    return Inst->getFunction();
+  if (auto Arg = dyn_cast<SILArgument>(this))
+    return Arg->getFunction();
+  return nullptr;
+}
+
+SILModule *ValueBase::getModule() {
+  if (auto Inst = dyn_cast<SILInstruction>(this))
+    return &Inst->getModule();
+  if (auto Arg = dyn_cast<SILArgument>(this))
+    return &Arg->getModule();
+  return nullptr;
+}