Add new setCondition member

git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@6603 91177308-0d34-0410-b5e6-96231b3b80d8

diff --git a/include/llvm/iTerminators.h b/include/llvm/iTerminators.h
index 0101f85d4bf5ea00633f79bf59faba44da52dd9c..35b07171ff13d5784813129eb0e9570393521855 100644 (file)
--- a/include/llvm/iTerminators.h
+++ b/include/llvm/iTerminators.h
@@ -78,11 +78,13 @@ public:
   inline bool isUnconditional() const { return Operands.size() == 1; }
   inline bool isConditional()   const { return Operands.size() == 3; }
 
-  inline const Value *getCondition() const {
-    return isUnconditional() ? 0 : Operands[2].get();
+  inline Value *getCondition() const {
+    return isUnconditional() ? 0 : (Value*)Operands[2].get();
   }
-  Value *getCondition() {
-    return isUnconditional() ? 0 : Operands[2].get();
+
+  void setCondition(Value *V) {
+    assert(isConditional() && "Cannot set condition of unconditional branch!");
+    setOperand(2, V);
   }
 
   // setUnconditionalDest - Change the current branch to an unconditional branch