-//===-- MachineCodeForMethod.cpp -------------------------------------------=//
+//===-- MachineFunction.cpp -----------------------------------------------===//
//
-// Purpose:
-// Collect native machine code information for a function.
-// This allows target-specific information about the generated code
-// to be stored with each function.
-//===---------------------------------------------------------------------===//
+// Collect native machine code information for a function. This allows
+// target-specific information about the generated code to be stored with each
+// function.
+//
+//===----------------------------------------------------------------------===//
#include "llvm/CodeGen/MachineFunction.h"
-#include "llvm/CodeGen/MachineInstr.h" // For debug output
-#include "llvm/CodeGen/MachineCodeForBasicBlock.h"
+#include "llvm/CodeGen/MachineInstr.h"
+#include "llvm/CodeGen/MachineCodeForInstruction.h"
+#include "llvm/CodeGen/SSARegMap.h"
+#include "llvm/CodeGen/MachineFunctionInfo.h"
+#include "llvm/CodeGen/MachineFrameInfo.h"
+#include "llvm/CodeGen/MachineConstantPool.h"
#include "llvm/Target/TargetMachine.h"
-#include "llvm/Target/MachineFrameInfo.h"
-#include "llvm/Target/MachineCacheInfo.h"
+#include "llvm/Target/TargetFrameInfo.h"
+#include "llvm/Target/TargetCacheInfo.h"
#include "llvm/Function.h"
#include "llvm/iOther.h"
+#include "llvm/Pass.h"
#include <limits.h>
const int INVALID_FRAME_OFFSET = INT_MAX; // std::numeric_limits<int>::max();
-static AnnotationID MCFM_AID(
+static AnnotationID MF_AID(
AnnotationManager::getID("CodeGen::MachineCodeForFunction"));
+
+//===---------------------------------------------------------------------===//
+// Code generation/destruction passes
+//===---------------------------------------------------------------------===//
+
+namespace {
+ class ConstructMachineFunction : public FunctionPass {
+ TargetMachine &Target;
+ public:
+ ConstructMachineFunction(TargetMachine &T) : Target(T) {}
+
+ const char *getPassName() const {
+ return "ConstructMachineFunction";
+ }
+
+ bool runOnFunction(Function &F) {
+ MachineFunction::construct(&F, Target).getInfo()->CalculateArgSize();
+ return false;
+ }
+ };
+
+ struct DestroyMachineFunction : public FunctionPass {
+ const char *getPassName() const { return "FreeMachineFunction"; }
+
+ static void freeMachineCode(Instruction &I) {
+ MachineCodeForInstruction::destroy(&I);
+ }
+
+ bool runOnFunction(Function &F) {
+ for (Function::iterator FI = F.begin(), FE = F.end(); FI != FE; ++FI)
+ for (BasicBlock::iterator I = FI->begin(), E = FI->end(); I != E; ++I)
+ MachineCodeForInstruction::get(I).dropAllReferences();
+
+ for (Function::iterator FI = F.begin(), FE = F.end(); FI != FE; ++FI)
+ for_each(FI->begin(), FI->end(), freeMachineCode);
+
+ return false;
+ }
+ };
+
+ struct Printer : public FunctionPass {
+ const char *getPassName() const { return "MachineFunction Printer"; }
+
+ virtual void getAnalysisUsage(AnalysisUsage &AU) const {
+ AU.setPreservesAll();
+ }
+
+ bool runOnFunction(Function &F) {
+ MachineFunction::get(&F).dump();
+ return false;
+ }
+ };
+}
+
+Pass *createMachineCodeConstructionPass(TargetMachine &Target) {
+ return new ConstructMachineFunction(Target);
+}
+
+Pass *createMachineCodeDestructionPass() {
+ return new DestroyMachineFunction();
+}
+
+Pass *createMachineFunctionPrinterPass() {
+ return new Printer();
+}
+
+
+//===---------------------------------------------------------------------===//
+// MachineFunction implementation
+//===---------------------------------------------------------------------===//
+
+MachineFunction::MachineFunction(const Function *F,
+ const TargetMachine &TM)
+ : Annotation(MF_AID), Fn(F), Target(TM) {
+ SSARegMapping = new SSARegMap();
+ MFInfo = new MachineFunctionInfo(*this);
+ FrameInfo = new MachineFrameInfo();
+ ConstantPool = new MachineConstantPool();
+}
+
+MachineFunction::~MachineFunction() {
+ delete SSARegMapping;
+ delete MFInfo;
+ delete FrameInfo;
+ delete ConstantPool;
+}
+
+void MachineFunction::dump() const { print(std::cerr); }
+
+void MachineFunction::print(std::ostream &OS) const {
+ OS << "\n" << *(Value*)Fn->getFunctionType() << " \"" << Fn->getName()
+ << "\"\n";
+
+ // Print Frame Information
+ getFrameInfo()->print(*this, OS);
+
+ // Print Constant Pool
+ getConstantPool()->print(OS);
+
+ for (const_iterator BB = begin(); BB != end(); ++BB) {
+ BasicBlock *LBB = BB->getBasicBlock();
+ OS << "\n" << LBB->getName() << " (" << (const void*)LBB << "):\n";
+ for (MachineBasicBlock::const_iterator I = BB->begin(); I != BB->end();++I){
+ OS << "\t";
+ (*I)->print(OS, Target);
+ }
+ }
+ OS << "\nEnd function \"" << Fn->getName() << "\"\n\n";
+}
+
+
// The next two methods are used to construct and to retrieve
// the MachineCodeForFunction object for the given function.
// construct() -- Allocates and initializes for a given function and target
// for a given Function.
//
MachineFunction&
-MachineFunction::construct(const Function *M, const TargetMachine &Tar)
+MachineFunction::construct(const Function *Fn, const TargetMachine &Tar)
{
- assert(M->getAnnotation(MCFM_AID) == 0 &&
+ assert(Fn->getAnnotation(MF_AID) == 0 &&
"Object already exists for this function!");
- MachineFunction* mcInfo = new MachineFunction(M, Tar);
- M->addAnnotation(mcInfo);
+ MachineFunction* mcInfo = new MachineFunction(Fn, Tar);
+ Fn->addAnnotation(mcInfo);
return *mcInfo;
}
void
-MachineFunction::destruct(const Function *M)
+MachineFunction::destruct(const Function *Fn)
{
- bool Deleted = M->deleteAnnotation(MCFM_AID);
+ bool Deleted = Fn->deleteAnnotation(MF_AID);
assert(Deleted && "Machine code did not exist for function!");
}
-MachineFunction&
-MachineFunction::get(const Function *F)
+MachineFunction& MachineFunction::get(const Function *F)
{
- MachineFunction *mc = (MachineFunction*)F->getAnnotation(MCFM_AID);
+ MachineFunction *mc = (MachineFunction*)F->getAnnotation(MF_AID);
assert(mc && "Call construct() method first to allocate the object");
return *mc;
}
+void MachineFunction::clearSSARegMap() {
+ delete SSARegMapping;
+ SSARegMapping = 0;
+}
+
+//===----------------------------------------------------------------------===//
+// MachineFrameInfo implementation
+//===----------------------------------------------------------------------===//
+
+/// CreateStackObject - Create a stack object for a value of the specified type.
+///
+int MachineFrameInfo::CreateStackObject(const Type *Ty, const TargetData &TD) {
+ return CreateStackObject(TD.getTypeSize(Ty), TD.getTypeAlignment(Ty));
+}
+
+int MachineFrameInfo::CreateStackObject(const TargetRegisterClass *RC) {
+ return CreateStackObject(RC->getSize(), RC->getAlignment());
+}
+
+
+void MachineFrameInfo::print(const MachineFunction &MF, std::ostream &OS) const{
+ int ValOffset = MF.getTarget().getFrameInfo().getOffsetOfLocalArea();
+
+ for (unsigned i = 0, e = Objects.size(); i != e; ++i) {
+ const StackObject &SO = Objects[i];
+ OS << " <fi #" << (int)(i-NumFixedObjects) << "> is ";
+ if (SO.Size == 0)
+ OS << "variable sized";
+ else
+ OS << SO.Size << " byte" << (SO.Size != 1 ? "s" : " ");
+
+ if (i < NumFixedObjects)
+ OS << " fixed";
+ if (i < NumFixedObjects || SO.SPOffset != -1) {
+ int Off = SO.SPOffset + ValOffset;
+ OS << " at location [SP";
+ if (Off > 0)
+ OS << "+" << Off;
+ else if (Off < 0)
+ OS << Off;
+ OS << "]";
+ }
+ OS << "\n";
+ }
+
+ if (HasVarSizedObjects)
+ OS << " Stack frame contains variable sized objects\n";
+}
+
+void MachineFrameInfo::dump(const MachineFunction &MF) const {
+ print(MF, std::cerr);
+}
+
+
+//===----------------------------------------------------------------------===//
+// MachineConstantPool implementation
+//===----------------------------------------------------------------------===//
+
+void MachineConstantPool::print(std::ostream &OS) const {
+ for (unsigned i = 0, e = Constants.size(); i != e; ++i)
+ OS << " <cp #" << i << "> is" << *(Value*)Constants[i] << "\n";
+}
+
+void MachineConstantPool::dump() const { print(std::cerr); }
+
+//===----------------------------------------------------------------------===//
+// MachineFunctionInfo implementation
+//===----------------------------------------------------------------------===//
+
static unsigned
ComputeMaxOptionalArgsSize(const TargetMachine& target, const Function *F,
unsigned &maxOptionalNumArgs)
{
- const MachineFrameInfo& frameInfo = target.getFrameInfo();
+ const TargetFrameInfo &frameInfo = target.getFrameInfo();
unsigned maxSize = 0;
for (Function::const_iterator BB = F->begin(), BBE = F->end(); BB !=BBE; ++BB)
for (BasicBlock::const_iterator I = BB->begin(), E = BB->end(); I != E; ++I)
- if (const CallInst *callInst = dyn_cast<CallInst>(&*I))
+ if (const CallInst *callInst = dyn_cast<CallInst>(I))
{
unsigned numOperands = callInst->getNumOperands() - 1;
int numExtra = (int)numOperands-frameInfo.getNumFixedOutgoingArgs();
if (numExtra <= 0)
continue;
- unsigned int sizeForThisCall;
+ unsigned sizeForThisCall;
if (frameInfo.argsOnStackHaveFixedSize())
{
int argSize = frameInfo.getSizeOfEachArgOnStack();
"compute MaxOptionalArgsSize");
sizeForThisCall = 0;
for (unsigned i = 0; i < numOperands; ++i)
- sizeForThisCall += target.DataLayout.getTypeSize(callInst->
+ sizeForThisCall += target.getTargetData().getTypeSize(callInst->
getOperand(i)->getType());
}
// but they are unrelated. This one does not align at more than a
// double-word boundary whereas that one might.
//
-inline unsigned int
-SizeToAlignment(unsigned int size, const TargetMachine& target)
+inline unsigned
+SizeToAlignment(unsigned size, const TargetMachine& target)
{
unsigned short cacheLineSize = target.getCacheInfo().getCacheLineSize(1);
if (size > (unsigned) cacheLineSize / 2)
return cacheLineSize;
else
for (unsigned sz=1; /*no condition*/; sz *= 2)
- if (sz >= size || sz >= target.DataLayout.getDoubleAlignment())
+ if (sz >= size || sz >= target.getTargetData().getDoubleAlignment())
return sz;
}
-/*ctor*/
-MachineFunction::MachineFunction(const Function *F,
- const TargetMachine& target)
- : Annotation(MCFM_AID),
- method(F), staticStackSize(0),
- automaticVarsSize(0), regSpillsSize(0),
- maxOptionalArgsSize(0), maxOptionalNumArgs(0),
- currentTmpValuesSize(0), maxTmpValuesSize(0), compiledAsLeaf(false),
- spillsAreaFrozen(false), automaticVarsAreaFrozen(false)
-{
- maxOptionalArgsSize = ComputeMaxOptionalArgsSize(target, method,
+void MachineFunctionInfo::CalculateArgSize() {
+ maxOptionalArgsSize = ComputeMaxOptionalArgsSize(MF.getTarget(),
+ MF.getFunction(),
maxOptionalNumArgs);
staticStackSize = maxOptionalArgsSize
- + target.getFrameInfo().getMinStackFrameSize();
+ + MF.getTarget().getFrameInfo().getMinStackFrameSize();
}
int
-MachineFunction::computeOffsetforLocalVar(const TargetMachine& target,
- const Value* val,
- unsigned int& getPaddedSize,
- unsigned int sizeToUse)
+MachineFunctionInfo::computeOffsetforLocalVar(const Value* val,
+ unsigned &getPaddedSize,
+ unsigned sizeToUse)
{
if (sizeToUse == 0)
- sizeToUse = target.findOptimalStorageSize(val->getType());
- unsigned int align = SizeToAlignment(sizeToUse, target);
+ sizeToUse = MF.getTarget().findOptimalStorageSize(val->getType());
+ unsigned align = SizeToAlignment(sizeToUse, MF.getTarget());
bool growUp;
- int firstOffset = target.getFrameInfo().getFirstAutomaticVarOffset(*this,
- growUp);
+ int firstOffset = MF.getTarget().getFrameInfo().getFirstAutomaticVarOffset(MF,
+ growUp);
int offset = growUp? firstOffset + getAutomaticVarsSize()
: firstOffset - (getAutomaticVarsSize() + sizeToUse);
- int aligned = target.getFrameInfo().adjustAlignment(offset, growUp, align);
+ int aligned = MF.getTarget().getFrameInfo().adjustAlignment(offset, growUp, align);
getPaddedSize = sizeToUse + abs(aligned - offset);
return aligned;
}
int
-MachineFunction::allocateLocalVar(const TargetMachine& target,
- const Value* val,
- unsigned int sizeToUse)
+MachineFunctionInfo::allocateLocalVar(const Value* val,
+ unsigned sizeToUse)
{
assert(! automaticVarsAreaFrozen &&
"Size of auto vars area has been used to compute an offset so "
int offset = getOffset(val);
if (offset == INVALID_FRAME_OFFSET)
{
- unsigned int getPaddedSize;
- offset = this->computeOffsetforLocalVar(target, val, getPaddedSize,
- sizeToUse);
+ unsigned getPaddedSize;
+ offset = computeOffsetforLocalVar(val, getPaddedSize, sizeToUse);
offsets[val] = offset;
incrementAutomaticVarsSize(getPaddedSize);
}
}
int
-MachineFunction::allocateSpilledValue(const TargetMachine& target,
- const Type* type)
+MachineFunctionInfo::allocateSpilledValue(const Type* type)
{
assert(! spillsAreaFrozen &&
"Size of reg spills area has been used to compute an offset so "
"no more register spill slots should be allocated!");
- unsigned int size = target.DataLayout.getTypeSize(type);
- unsigned char align = target.DataLayout.getTypeAlignment(type);
+ unsigned size = MF.getTarget().getTargetData().getTypeSize(type);
+ unsigned char align = MF.getTarget().getTargetData().getTypeAlignment(type);
bool growUp;
- int firstOffset = target.getFrameInfo().getRegSpillAreaOffset(*this, growUp);
+ int firstOffset = MF.getTarget().getFrameInfo().getRegSpillAreaOffset(MF, growUp);
int offset = growUp? firstOffset + getRegSpillsSize()
: firstOffset - (getRegSpillsSize() + size);
- int aligned = target.getFrameInfo().adjustAlignment(offset, growUp, align);
+ int aligned = MF.getTarget().getFrameInfo().adjustAlignment(offset, growUp, align);
size += abs(aligned - offset); // include alignment padding in size
incrementRegSpillsSize(size); // update size of reg. spills area
}
int
-MachineFunction::pushTempValue(const TargetMachine& target,
- unsigned int size)
+MachineFunctionInfo::pushTempValue(unsigned size)
{
- unsigned int align = SizeToAlignment(size, target);
+ unsigned align = SizeToAlignment(size, MF.getTarget());
bool growUp;
- int firstOffset = target.getFrameInfo().getTmpAreaOffset(*this, growUp);
+ int firstOffset = MF.getTarget().getFrameInfo().getTmpAreaOffset(MF, growUp);
int offset = growUp? firstOffset + currentTmpValuesSize
: firstOffset - (currentTmpValuesSize + size);
- int aligned = target.getFrameInfo().adjustAlignment(offset, growUp, align);
+ int aligned = MF.getTarget().getFrameInfo().adjustAlignment(offset, growUp,
+ align);
size += abs(aligned - offset); // include alignment padding in size
incrementTmpAreaSize(size); // update "current" size of tmp area
return aligned;
}
-void
-MachineFunction::popAllTempValues(const TargetMachine& target)
-{
+void MachineFunctionInfo::popAllTempValues() {
resetTmpAreaSize(); // clear tmp area to reuse
}
int
-MachineFunction::getOffset(const Value* val) const
+MachineFunctionInfo::getOffset(const Value* val) const
{
hash_map<const Value*, int>::const_iterator pair = offsets.find(val);
- return (pair == offsets.end())? INVALID_FRAME_OFFSET : pair->second;
-}
-
-void
-MachineFunction::dump() const
-{
- std::cerr << "\n" << method->getReturnType()
- << " \"" << method->getName() << "\"\n";
-
- for (Function::const_iterator BB = method->begin(); BB != method->end(); ++BB)
- {
- std::cerr << std::endl << (*BB).getName() << " (" << (const void*) BB << ")" << ":" << std::endl;
- MachineCodeForBasicBlock& mvec = MachineCodeForBasicBlock::get(BB);
- for (unsigned i=0; i < mvec.size(); i++)
- std::cerr << "\t" << *mvec[i];
- }
- std::cerr << "\nEnd function \"" << method->getName() << "\"\n\n";
+ return (pair == offsets.end()) ? INVALID_FRAME_OFFSET : pair->second;
}