diff options
| -rw-r--r-- | include/llvm/InstrTypes.h | 1 | ||||
| -rw-r--r-- | include/llvm/iTerminators.h | 18 |
2 files changed, 19 insertions, 0 deletions
diff --git a/include/llvm/InstrTypes.h b/include/llvm/InstrTypes.h index 9852a32fda..f7d63d6a90 100644 --- a/include/llvm/InstrTypes.h +++ b/include/llvm/InstrTypes.h @@ -35,6 +35,7 @@ public: // virtual const BasicBlock *getSuccessor(unsigned idx) const = 0; virtual unsigned getNumSuccessors() const = 0; + virtual void setSuccessor(unsigned idx, BasicBlock *NewSucc) = 0; inline BasicBlock *getSuccessor(unsigned idx) { return (BasicBlock*)((const TerminatorInst *)this)->getSuccessor(idx); diff --git a/include/llvm/iTerminators.h b/include/llvm/iTerminators.h index c5047bc6c3..482684aa8a 100644 --- a/include/llvm/iTerminators.h +++ b/include/llvm/iTerminators.h @@ -52,6 +52,9 @@ public: assert(0 && "ReturnInst has no successors!"); abort(); } + virtual void setSuccessor(unsigned idx, BasicBlock *NewSucc) { + assert(0 && "ReturnInst has no successors!"); + } virtual unsigned getNumSuccessors() const { return 0; } // Methods for support type inquiry through isa, cast, and dyn_cast: @@ -105,6 +108,11 @@ public: return (BasicBlock*)((const BranchInst *)this)->getSuccessor(idx); } + virtual void setSuccessor(unsigned idx, BasicBlock *NewSucc) { + assert(idx < getNumSuccessors() && "Successor # out of range for Branch!"); + Operands[idx] = (Value*)NewSucc; + } + virtual unsigned getNumSuccessors() const { return 1+isConditional(); } // Methods for support type inquiry through isa, cast, and dyn_cast: @@ -156,6 +164,11 @@ public: return cast<BasicBlock>(Operands[idx*2+1].get()); } + virtual void setSuccessor(unsigned idx, BasicBlock *NewSucc) { + assert(idx < getNumSuccessors() && "Successor # out of range for switch!"); + Operands[idx*2+1] = (Value*)NewSucc; + } + // getSuccessorValue - Return the value associated with the specified // successor. inline const Constant *getSuccessorValue(unsigned idx) const { @@ -231,6 +244,11 @@ public: return i == 0 ? getNormalDest() : getExceptionalDest(); } + virtual void setSuccessor(unsigned idx, BasicBlock *NewSucc) { + assert(idx < 2 && "Successor # out of range for invoke!"); + Operands[idx+1] = (Value*)NewSucc; + } + virtual unsigned getNumSuccessors() const { return 2; } // Methods for support type inquiry through isa, cast, and dyn_cast: |