All constant-evaluation code now unified into
[oota-llvm.git] / lib / CodeGen / MachineFunction.cpp
index 23b8040c440806b78d82a4b8b578382ce1b32577..85bf4b6a272702e3aba13f4f0a40f7530aedb4b6 100644 (file)
@@ -6,21 +6,24 @@
 //
 //===----------------------------------------------------------------------===//
 
-#include "llvm/CodeGen/MachineInstr.h"  // For debug output
 #include "llvm/CodeGen/MachineFunction.h"
-#include "llvm/CodeGen/MachineBasicBlock.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>
+#include "Config/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"));
 
 
@@ -39,7 +42,7 @@ namespace {
     }
     
     bool runOnFunction(Function &F) {
-      MachineFunction::construct(&F, Target);
+      MachineFunction::construct(&F, Target).getInfo()->CalculateArgSize();
       return false;
     }
   };
@@ -62,6 +65,19 @@ namespace {
       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) {
@@ -72,11 +88,55 @@ 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) {
+    const 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
@@ -85,48 +145,116 @@ Pass *createMachineCodeDestructionPass() {
 //                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(); 
@@ -139,7 +267,7 @@ ComputeMaxOptionalArgsSize(const TargetMachine& target, const Function *F,
                      "compute MaxOptionalArgsSize");
               sizeForThisCall = 0;
               for (unsigned i = 0; i < numOperands; ++i)
-                sizeForThisCall += target.DataLayout.getTypeSize(callInst->
+                sizeForThisCall += target.getTargetData().getTypeSize(callInst->
                                               getOperand(i)->getType());
             }
           
@@ -162,61 +290,51 @@ ComputeMaxOptionalArgsSize(const TargetMachine& target, const Function *F,
 // 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 "
@@ -227,8 +345,8 @@ MachineFunction::allocateLocalVar(const TargetMachine& target,
   int offset = getOffset(val);
   if (offset == INVALID_FRAME_OFFSET)
     {
-      unsigned int getPaddedSize;
-      offset = computeOffsetforLocalVar(target, val, getPaddedSize, sizeToUse);
+      unsigned getPaddedSize;
+      offset = computeOffsetforLocalVar(val, getPaddedSize, sizeToUse);
       offsets[val] = offset;
       incrementAutomaticVarsSize(getPaddedSize);
     }
@@ -236,23 +354,22 @@ MachineFunction::allocateLocalVar(const TargetMachine& target,
 }
 
 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
@@ -261,18 +378,18 @@ MachineFunction::allocateSpilledValue(const TargetMachine& target,
 }
 
 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
@@ -280,32 +397,13 @@ MachineFunction::pushTempValue(const TargetMachine& target,
   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 << "\n" << BB->getName() << " (" << (const void*)BB
-                << ")" << ":" << "\n";
-      MachineBasicBlock& mvec = MachineBasicBlock::get(BB);
-      for (unsigned i=0; i < mvec.size(); i++)
-       std::cerr << "\t" << *mvec[i];
-    } 
-  std::cerr << "\nEnd function \"" << method->getName() << "\"\n\n";
-}