-//===-- MachineCodeForMethod.cpp --------------------------------------------=//
+//===-- MachineFunction.cpp -----------------------------------------------===//
//
-// Purpose:
-// Collect native machine code information for a method.
-// This allows target-specific information about the generated code
-// to be stored with each method.
+// 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/MachineCodeForMethod.h"
#include "llvm/CodeGen/MachineInstr.h" // For debug output
+#include "llvm/CodeGen/MachineFunction.h"
+#include "llvm/CodeGen/MachineCodeForInstruction.h"
#include "llvm/Target/TargetMachine.h"
#include "llvm/Target/MachineFrameInfo.h"
#include "llvm/Target/MachineCacheInfo.h"
-#include "llvm/Method.h"
-#include "llvm/BasicBlock.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(
- AnnotationManager::getID("CodeGen::MachineCodeForMethod"));
+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).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& target)
+ : Annotation(MF_AID),
+ Fn(F), Target(target), staticStackSize(0),
+ automaticVarsSize(0), regSpillsSize(0),
+ maxOptionalArgsSize(0), maxOptionalNumArgs(0),
+ currentTmpValuesSize(0), maxTmpValuesSize(0), compiledAsLeaf(false),
+ spillsAreaFrozen(false), automaticVarsAreaFrozen(false)
+{
+}
+
+void MachineFunction::dump() const { print(std::cerr); }
+
+void MachineFunction::print(std::ostream &OS) const {
+ OS << "\n" << *(Value*)Fn->getReturnType() << " \"" << Fn->getName()<< "\"\n";
+
+ for (const_iterator BB = begin(); BB != end(); ++BB) {
+ BasicBlock *LBB = BB->getBasicBlock();
+ OS << "\n" << LBB->getName() << " ("
+ << (const void*)BB->getBasicBlock() << "):\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 MachineCodeForMethod object for the given method.
-// construct() -- Allocates and initializes for a given method and target
+// the MachineCodeForFunction object for the given function.
+// construct() -- Allocates and initializes for a given function and target
// get() -- Returns a handle to the object.
// This should not be called before "construct()"
-// for a given Method.
+// for a given Function.
//
-MachineCodeForMethod &MachineCodeForMethod::construct(const Method *M,
- const TargetMachine &Tar){
- assert(M->getAnnotation(MCFM_AID) == 0 &&
- "Object already exists for this method!");
- MachineCodeForMethod* mcInfo = new MachineCodeForMethod(M, Tar);
- M->addAnnotation(mcInfo);
+MachineFunction&
+MachineFunction::construct(const Function *Fn, const TargetMachine &Tar)
+{
+ assert(Fn->getAnnotation(MF_AID) == 0 &&
+ "Object already exists for this function!");
+ MachineFunction* mcInfo = new MachineFunction(Fn, Tar);
+ Fn->addAnnotation(mcInfo);
return *mcInfo;
}
-void MachineCodeForMethod::destruct(const Method *M) {
- bool Deleted = M->deleteAnnotation(MCFM_AID);
- assert(Deleted && "Machine code did not exist for method!");
+void
+MachineFunction::destruct(const Function *Fn)
+{
+ bool Deleted = Fn->deleteAnnotation(MF_AID);
+ assert(Deleted && "Machine code did not exist for function!");
}
-
-MachineCodeForMethod &MachineCodeForMethod::get(const Method* method) {
- MachineCodeForMethod* mc = (MachineCodeForMethod*)
- method->getAnnotation(MCFM_AID);
+MachineFunction& MachineFunction::get(const Function *F)
+{
+ MachineFunction *mc = (MachineFunction*)F->getAnnotation(MF_AID);
assert(mc && "Call construct() method first to allocate the object");
return *mc;
}
static unsigned
-ComputeMaxOptionalArgsSize(const TargetMachine& target, const Method* method)
+ComputeMaxOptionalArgsSize(const TargetMachine& target, const Function *F,
+ unsigned &maxOptionalNumArgs)
{
const MachineFrameInfo& frameInfo = target.getFrameInfo();
- unsigned int maxSize = 0;
+ unsigned maxSize = 0;
- for (Method::const_iterator MI=method->begin(), ME=method->end();
- MI != ME; ++MI) {
- const BasicBlock *BB = *MI;
+ 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 (CallInst *callInst = dyn_cast<CallInst>(*I)) {
- unsigned int numOperands = callInst->getNumOperands() - 1;
- int numExtra = (int) numOperands - frameInfo.getNumFixedOutgoingArgs();
- if (numExtra <= 0)
- continue;
-
- unsigned int sizeForThisCall;
- if (frameInfo.argsOnStackHaveFixedSize())
- {
- int argSize = frameInfo.getSizeOfEachArgOnStack();
- sizeForThisCall = numExtra * (unsigned) argSize;
- }
- else
- {
- assert(0 && "UNTESTED CODE: Size per stack argument is not fixed on this architecture: use actual arg sizes to compute MaxOptionalArgsSize");
- sizeForThisCall = 0;
- for (unsigned i=0; i < numOperands; ++i)
- sizeForThisCall += target.findOptimalStorageSize(callInst->
- getOperand(i)->getType());
- }
-
- if (maxSize < sizeForThisCall)
- maxSize = sizeForThisCall;
- }
- }
+ 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;
+ if (frameInfo.argsOnStackHaveFixedSize())
+ {
+ int argSize = frameInfo.getSizeOfEachArgOnStack();
+ sizeForThisCall = numExtra * (unsigned) argSize;
+ }
+ else
+ {
+ assert(0 && "UNTESTED CODE: Size per stack argument is not "
+ "fixed on this architecture: use actual arg sizes to "
+ "compute MaxOptionalArgsSize");
+ sizeForThisCall = 0;
+ for (unsigned i = 0; i < numOperands; ++i)
+ sizeForThisCall += target.DataLayout.getTypeSize(callInst->
+ getOperand(i)->getType());
+ }
+
+ if (maxSize < sizeForThisCall)
+ maxSize = sizeForThisCall;
+
+ if ((int)maxOptionalNumArgs < numExtra)
+ maxOptionalNumArgs = (unsigned) numExtra;
+ }
+
return maxSize;
}
// Align data larger than one L1 cache line on L1 cache line boundaries.
-// Align all smaller data on the next higher 2^x boundary (4, 8, ...).
+// Align all smaller data on the next higher 2^x boundary (4, 8, ...),
+// but not higher than the alignment of the largest type we support
+// (currently a double word). -- see class TargetData).
//
-// THIS FUNCTION HAS BEEN COPIED FROM EMITASSEMBLY.CPP AND
-// SHOULD BE USED DIRECTLY THERE
+// This function is similar to the corresponding function in EmitAssembly.cpp
+// 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)
return cacheLineSize;
else
for (unsigned sz=1; /*no condition*/; sz *= 2)
- if (sz >= size)
+ if (sz >= size || sz >= target.DataLayout.getDoubleAlignment())
return sz;
}
+void MachineFunction::CalculateArgSize() {
+ maxOptionalArgsSize = ComputeMaxOptionalArgsSize(Target, Fn,
+ maxOptionalNumArgs);
+ staticStackSize = maxOptionalArgsSize
+ + Target.getFrameInfo().getMinStackFrameSize();
+}
-/*ctor*/
-MachineCodeForMethod::MachineCodeForMethod(const Method* _M,
- const TargetMachine& target)
- : Annotation(MCFM_AID),
- method(_M), compiledAsLeaf(false), staticStackSize(0),
- automaticVarsSize(0), regSpillsSize(0),
- currentOptionalArgsSize(0), maxOptionalArgsSize(0),
- currentTmpValuesSize(0)
+int
+MachineFunction::computeOffsetforLocalVar(const TargetMachine& target,
+ const Value* val,
+ unsigned int& getPaddedSize,
+ unsigned int sizeToUse)
{
- maxOptionalArgsSize = ComputeMaxOptionalArgsSize(target, method);
- staticStackSize = maxOptionalArgsSize +
- target.getFrameInfo().getMinStackFrameSize();
+ if (sizeToUse == 0)
+ sizeToUse = target.findOptimalStorageSize(val->getType());
+ unsigned int align = SizeToAlignment(sizeToUse, target);
+
+ bool growUp;
+ int firstOffset = target.getFrameInfo().getFirstAutomaticVarOffset(*this,
+ growUp);
+ int offset = growUp? firstOffset + getAutomaticVarsSize()
+ : firstOffset - (getAutomaticVarsSize() + sizeToUse);
+
+ int aligned = target.getFrameInfo().adjustAlignment(offset, growUp, align);
+ getPaddedSize = sizeToUse + abs(aligned - offset);
+
+ return aligned;
}
int
-MachineCodeForMethod::allocateLocalVar(const TargetMachine& target,
+MachineFunction::allocateLocalVar(const TargetMachine& target,
const Value* val,
- unsigned int size)
+ unsigned int sizeToUse)
{
+ assert(! automaticVarsAreaFrozen &&
+ "Size of auto vars area has been used to compute an offset so "
+ "no more automatic vars should be allocated!");
+
// Check if we've allocated a stack slot for this value already
//
int offset = getOffset(val);
if (offset == INVALID_FRAME_OFFSET)
{
- bool growUp;
- int firstOffset =target.getFrameInfo().getFirstAutomaticVarOffset(*this,
- growUp);
- unsigned char align;
- if (size == 0)
- {
- size = target.findOptimalStorageSize(val->getType());
- // align = target.DataLayout.getTypeAlignment(val->getType());
- }
-
- align = SizeToAlignment(size, target);
-
- offset = getAutomaticVarsSize();
- if (! growUp)
- offset += size;
-
- if (unsigned int mod = offset % align)
- {
- offset += align - mod;
- size += align - mod;
- }
-
- offset = growUp? firstOffset + offset
- : firstOffset - offset;
-
+ unsigned int getPaddedSize;
+ offset = computeOffsetforLocalVar(target, val, getPaddedSize, sizeToUse);
offsets[val] = offset;
-
- incrementAutomaticVarsSize(size);
+ incrementAutomaticVarsSize(getPaddedSize);
}
return offset;
}
int
-MachineCodeForMethod::allocateSpilledValue(const TargetMachine& target,
+MachineFunction::allocateSpilledValue(const TargetMachine& target,
const Type* type)
{
- unsigned int size = target.findOptimalStorageSize(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);
bool growUp;
int firstOffset = target.getFrameInfo().getRegSpillAreaOffset(*this, growUp);
- int offset = getRegSpillsSize();
- if (! growUp)
- offset += size;
-
- if (unsigned int mod = offset % align)
- {
- offset += align - mod;
- size += align - mod;
- }
-
- offset = growUp? firstOffset + offset
- : firstOffset - offset;
-
- incrementRegSpillsSize(size);
-
- return offset;
-}
+ int offset = growUp? firstOffset + getRegSpillsSize()
+ : firstOffset - (getRegSpillsSize() + size);
-int
-MachineCodeForMethod::allocateOptionalArg(const TargetMachine& target,
- const Type* type)
-{
- const MachineFrameInfo& frameInfo = target.getFrameInfo();
-
- int size = INT_MAX;
- if (frameInfo.argsOnStackHaveFixedSize())
- size = frameInfo.getSizeOfEachArgOnStack();
- else
- {
- size = target.findOptimalStorageSize(type);
- assert(0 && "UNTESTED CODE: Size per stack argument is not fixed on this architecture: use actual argument sizes for computing optional arg offsets");
- }
- unsigned char align = target.DataLayout.getTypeAlignment(type);
-
- bool growUp;
- int firstOffset = frameInfo.getFirstOptionalOutgoingArgOffset(*this, growUp);
-
- int offset = getCurrentOptionalArgsSize();
- if (! growUp)
- offset += size;
+ int aligned = target.getFrameInfo().adjustAlignment(offset, growUp, align);
+ size += abs(aligned - offset); // include alignment padding in size
- if (unsigned int mod = offset % align)
- {
- offset += align - mod;
- size += align - mod;
- }
-
- offset = growUp? firstOffset + offset
- : firstOffset - offset;
-
- incrementCurrentOptionalArgsSize(size);
-
- return offset;
-}
+ incrementRegSpillsSize(size); // update size of reg. spills area
-void
-MachineCodeForMethod::resetOptionalArgs(const TargetMachine& target)
-{
- currentOptionalArgsSize = 0;
+ return aligned;
}
int
-MachineCodeForMethod::pushTempValue(const TargetMachine& target,
+MachineFunction::pushTempValue(const TargetMachine& target,
unsigned int size)
{
- // Compute a power-of-2 alignment according to the possible sizes,
- // but not greater than the alignment of the largest type we support
- // (currently a double word -- see class TargetData).
- unsigned char align = 1;
- for (; align < size && align < target.DataLayout.getDoubleAlignment();
- align = 2*align)
- ;
-
+ unsigned int align = SizeToAlignment(size, target);
+
bool growUp;
- int firstTmpOffset = target.getFrameInfo().getTmpAreaOffset(*this, growUp);
-
- int offset = currentTmpValuesSize;
- if (! growUp)
- offset += size;
-
- if (unsigned int mod = offset % align)
- {
- offset += align - mod;
- size += align - mod;
- }
-
- offset = growUp ? firstTmpOffset + offset : firstTmpOffset - offset;
-
- currentTmpValuesSize += size;
- return offset;
+ int firstOffset = target.getFrameInfo().getTmpAreaOffset(*this, growUp);
+
+ int offset = growUp? firstOffset + currentTmpValuesSize
+ : firstOffset - (currentTmpValuesSize + size);
+
+ int aligned = target.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
-MachineCodeForMethod::popAllTempValues(const TargetMachine& target)
+MachineFunction::popAllTempValues(const TargetMachine& target)
{
- currentTmpValuesSize = 0;
+ resetTmpAreaSize(); // clear tmp area to reuse
}
int
-MachineCodeForMethod::getOffset(const Value* val) const
+MachineFunction::getOffset(const Value* val) const
{
- std::hash_map<const Value*, int>::const_iterator pair = offsets.find(val);
- return (pair == offsets.end())? INVALID_FRAME_OFFSET : pair->second;
-}
-
-void
-MachineCodeForMethod::dump() const
-{
- cerr << "\n" << method->getReturnType()
- << " \"" << method->getName() << "\"\n";
-
- for (Method::const_iterator BI = method->begin(); BI != method->end(); ++BI)
- {
- BasicBlock* bb = *BI;
- cerr << "\n"
- << (bb->hasName()? bb->getName() : "Label")
- << " (" << bb << ")" << ":\n";
-
- MachineCodeForBasicBlock& mvec = bb->getMachineInstrVec();
- for (unsigned i=0; i < mvec.size(); i++)
- cerr << "\t" << *mvec[i];
- }
- cerr << "\nEnd method \"" << method->getName() << "\"\n\n";
+ hash_map<const Value*, int>::const_iterator pair = offsets.find(val);
+ return (pair == offsets.end()) ? INVALID_FRAME_OFFSET : pair->second;
}
projects / oota-llvm.git / blobdiff