mirror of
https://github.com/c64scene-ar/llvm-6502.git
synced 2025-03-03 14:31:10 +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(const CallSite &CS) : I(CS.I) {}
|
||||||
CallSite &operator=(const CallSite &CS) { I = CS.I; return *this; }
|
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
|
/// getInstruction - Return the instruction this call site corresponds to
|
||||||
///
|
///
|
||||||
Instruction *getInstruction() const { return I; }
|
Instruction *getInstruction() const { return I; }
|
||||||
@ -37,6 +52,12 @@ public:
|
|||||||
return dyn_cast<Function>(getCalledValue());
|
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
|
/// arg_iterator - The type of iterator to use when looping over actual
|
||||||
/// arguments at this call site...
|
/// arguments at this call site...
|
||||||
typedef User::op_iterator arg_iterator;
|
typedef User::op_iterator arg_iterator;
|
||||||
@ -50,7 +71,7 @@ public:
|
|||||||
else
|
else
|
||||||
return I->op_begin()+3; // Skip Function, BB, BB
|
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
|
#endif
|
||||||
|
Loading…
x
Reference in New Issue
Block a user