mirror of
https://github.com/c64scene-ar/llvm-6502.git
synced 2025-01-12 17:32:19 +00:00
* Add new CallSite::get factory method
* add new setCalledFunction method * FIX arg_end method which was horribly broken! git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@6758 91177308-0d34-0410-b5e6-96231b3b80d8
This commit is contained in:
parent
a599f871af
commit
697a00fbf0
@ -22,6 +22,21 @@ public:
|
||||
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 @@ public:
|
||||
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 @@ public:
|
||||
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
|
||||
|
Loading…
x
Reference in New Issue
Block a user