Added setSuccessor() method to terminator instructions

llvm-svn: 2784

2 files changed, 11 insertions, 1 deletions

diff --git a/llvm/include/llvm/InstrTypes.h b/llvm/include/llvm/InstrTypes.h
index f7d63d6a900..8b5b39091af 100644
--- a/llvm/include/llvm/InstrTypes.h
+++ b/llvm/include/llvm/InstrTypes.h
@@ -35,7 +35,9 @@ public:
   //
   virtual const BasicBlock *getSuccessor(unsigned idx) const = 0;
   virtual unsigned getNumSuccessors() const = 0;
-  virtual void setSuccessor(unsigned idx, BasicBlock *NewSucc) = 0;
+  
+  //set a successor at a given index
+  virtual void setSuccessor(unsigned idx, BasicBlock *B) = 0;
 
   inline BasicBlock *getSuccessor(unsigned idx) {
     return (BasicBlock*)((const TerminatorInst *)this)->getSuccessor(idx);
diff --git a/llvm/include/llvm/iTerminators.h b/llvm/include/llvm/iTerminators.h
index 482684aa8ad..b17029be750 100644
--- a/llvm/include/llvm/iTerminators.h
+++ b/llvm/include/llvm/iTerminators.h
@@ -233,6 +233,14 @@ public:
     return cast<BasicBlock>(Operands[2].get());
   }
 
+  inline void setNormalDest(BasicBlock *B){
+    Operands[1] = (Value*)B;
+  }
+
+  inline void setExceptionalDest(BasicBlock *B){
+    Operands[2] = (Value*)B;
+  }
+
   virtual const char *getOpcodeName() const { return "invoke"; }
 
   virtual const BasicBlock *getSuccessor(unsigned i) const {