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

[John Criswell]

Changes in directory llvm/include/llvm/Support:

CallSite.h updated: 1.1 -> 1.1.2.1

---
Log message:

Merged with mainline on Thursday, June 26, 2003.
Kept includes of "Config/assert.h" to avoid implicit dependencies on header
files.


---
Diffs of the changes:

Index: llvm/include/llvm/Support/CallSite.h
diff -u llvm/include/llvm/Support/CallSite.h:1.1 llvm/include/llvm/Support/CallSite.h:1.1.2.1
--- llvm/include/llvm/Support/CallSite.h:1.1	Mon Feb 24 14:35:45 2003
+++ llvm/include/llvm/Support/CallSite.h	Thu Jun 26 16:34:53 2003
@@ -16,8 +16,30 @@
 class CallSite {
   Instruction *I;
 public:
+  CallSite() : I(0) {}
   CallSite(CallInst *CI) : I((Instruction*)CI) {}
   CallSite(InvokeInst *II) : I((Instruction*)II) {}
+  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; }
 
   /// getCalledValue - Return the pointer to function that is being called...
   ///
@@ -30,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;
@@ -43,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