[llvm-commits] CVS: llvm/include/llvm/Support/CallSite.h

[Chris Lattner]

Changes in directory llvm/include/llvm/Support:

CallSite.h updated: 1.4 -> 1.5

---
Log message:

* Add new CallSite::get factory method
* add new setCalledFunction method
* FIX arg_end method which was horribly broken!



---
Diffs of the changes:

Index: llvm/include/llvm/Support/CallSite.h
diff -u llvm/include/llvm/Support/CallSite.h:1.4 llvm/include/llvm/Support/CallSite.h:1.5
--- llvm/include/llvm/Support/CallSite.h:1.4	Tue Jun 17 16:44:51 2003
+++ llvm/include/llvm/Support/CallSite.h	Tue Jun 17 17:16:59 2003
@@ -22,6 +22,21 @@
   CallSite(const CallSite &CS) : I(CS.I) {}
   CallSite &operator=(const CallSite &CS) { I = CS.I; return *this; }
 
+  /// CallSite::get - This static method is sort of like a constructor.  It will
+  /// create an appropriate call site for a Call or Invoke instruction, but it
+  /// can also create a null initialized CallSite object for something which is
+  /// NOT a call site.
+  ///
+  static CallSite get(Value *V) {
+    if (Instruction *I = dyn_cast<Instruction>(V)) {
+      if (I->getOpcode() == Instruction::Call)
+        return CallSite((CallInst*)I);
+      else if (I->getOpcode() == Instruction::Invoke)
+        return CallSite((InvokeInst*)I);
+    }
+    return CallSite();
+  }
+
   /// getInstruction - Return the instruction this call site corresponds to
   ///
   Instruction *getInstruction() const { return I; }
@@ -37,6 +52,12 @@
     return dyn_cast<Function>(getCalledValue());
   }
 
+  /// setCalledFunction - Set the callee to the specied value...
+  ///
+  void setCalledFunction(Value *V) {
+    I->setOperand(0, V);
+  }
+
   /// arg_iterator - The type of iterator to use when looping over actual
   /// arguments at this call site...
   typedef User::op_iterator arg_iterator;
@@ -50,7 +71,7 @@
     else
       return I->op_begin()+3; // Skip Function, BB, BB
   }
-  arg_iterator arg_end() const { return I->op_begin(); }
+  arg_iterator arg_end() const { return I->op_end(); }
 };
 
 #endif