lib/Target/MBlaze/MBlazeMachineFunction.h

   1 //===-- MBlazeMachineFunctionInfo.h - Private data ----------------*- C++ -*-=//
   2 //
   3 //                     The LLVM Compiler Infrastructure
   4 //
   5 // This file is distributed under the University of Illinois Open Source
   6 // License. See LICENSE.TXT for details.
   7 //
   8 //===----------------------------------------------------------------------===//
   9 //
  10 // This file declares the MBlaze specific subclass of MachineFunctionInfo.
  11 //
  12 //===----------------------------------------------------------------------===//
  13
  14 #ifndef MBLAZE_MACHINE_FUNCTION_INFO_H
  15 #define MBLAZE_MACHINE_FUNCTION_INFO_H
  16
  17 #include "llvm/ADT/SmallVector.h"
  18 #include "llvm/ADT/VectorExtras.h"
  19 #include "llvm/CodeGen/MachineFunction.h"
  20 #include "llvm/CodeGen/MachineFrameInfo.h"
  21
  22 namespace llvm {
  23
  24 /// MBlazeFunctionInfo - This class is derived from MachineFunction private
  25 /// MBlaze target-specific information for each MachineFunction.
  26 class MBlazeFunctionInfo : public MachineFunctionInfo {
  27
  28 private:
  29   /// Holds for each function where on the stack the Frame Pointer must be
  30   /// saved. This is used on Prologue and Epilogue to emit FP save/restore
  31   int FPStackOffset;
  32
  33   /// Holds for each function where on the stack the Return Address must be
  34   /// saved. This is used on Prologue and Epilogue to emit RA save/restore
  35   int RAStackOffset;
  36
  37   /// MBlazeFIHolder - Holds a FrameIndex and it's Stack Pointer Offset
  38   struct MBlazeFIHolder {
  39
  40     int FI;
  41     int SPOffset;
  42
  43     MBlazeFIHolder(int FrameIndex, int StackPointerOffset)
  44       : FI(FrameIndex), SPOffset(StackPointerOffset) {}
  45   };
  46
  47   /// When PIC is used the GP must be saved on the stack on the function
  48   /// prologue and must be reloaded from this stack location after every
  49   /// call. A reference to its stack location and frame index must be kept
  50   /// to be used on emitPrologue and processFunctionBeforeFrameFinalized.
  51   MBlazeFIHolder GPHolder;
  52
  53   /// On LowerFormalArguments the stack size is unknown, so the Stack
  54   /// Pointer Offset calculation of "not in register arguments" must be
  55   /// postponed to emitPrologue.
  56   SmallVector<MBlazeFIHolder, 16> FnLoadArgs;
  57   bool HasLoadArgs;
  58
  59   // When VarArgs, we must write registers back to caller stack, preserving
  60   // on register arguments. Since the stack size is unknown on
  61   // LowerFormalArguments, the Stack Pointer Offset calculation must be
  62   // postponed to emitPrologue.
  63   SmallVector<MBlazeFIHolder, 4> FnStoreVarArgs;
  64   bool HasStoreVarArgs;
  65
  66   /// SRetReturnReg - Some subtargets require that sret lowering includes
  67   /// returning the value of the returned struct in a register. This field
  68   /// holds the virtual register into which the sret argument is passed.
  69   unsigned SRetReturnReg;
  70
  71   /// GlobalBaseReg - keeps track of the virtual register initialized for
  72   /// use as the global base register. This is used for PIC in some PIC
  73   /// relocation models.
  74   unsigned GlobalBaseReg;
  75
  76   // VarArgsFrameIndex - FrameIndex for start of varargs area.
  77   int VarArgsFrameIndex;
  78
  79 public:
  80   MBlazeFunctionInfo(MachineFunction& MF)
  81   : FPStackOffset(0), RAStackOffset(0), GPHolder(-1,-1),
  82     HasLoadArgs(false), HasStoreVarArgs(false), SRetReturnReg(0),
  83     GlobalBaseReg(0), VarArgsFrameIndex(0)
  84   {}
  85
  86   int getFPStackOffset() const { return FPStackOffset; }
  87   void setFPStackOffset(int Off) { FPStackOffset = Off; }
  88
  89   int getRAStackOffset() const { return RAStackOffset; }
  90   void setRAStackOffset(int Off) { RAStackOffset = Off; }
  91
  92   int getGPStackOffset() const { return GPHolder.SPOffset; }
  93   int getGPFI() const { return GPHolder.FI; }
  94   void setGPStackOffset(int Off) { GPHolder.SPOffset = Off; }
  95   void setGPFI(int FI) { GPHolder.FI = FI; }
  96   bool needGPSaveRestore() const { return GPHolder.SPOffset != -1; }
  97
  98   bool hasLoadArgs() const { return HasLoadArgs; }
  99   bool hasStoreVarArgs() const { return HasStoreVarArgs; }
 100
 101   void recordLoadArgsFI(int FI, int SPOffset) {
 102     if (!HasLoadArgs) HasLoadArgs=true;
 103     FnLoadArgs.push_back(MBlazeFIHolder(FI, SPOffset));
 104   }
 105   void recordStoreVarArgsFI(int FI, int SPOffset) {
 106     if (!HasStoreVarArgs) HasStoreVarArgs=true;
 107     FnStoreVarArgs.push_back(MBlazeFIHolder(FI, SPOffset));
 108   }
 109
 110   void adjustLoadArgsFI(MachineFrameInfo *MFI) const {
 111     if (!hasLoadArgs()) return;
 112     for (unsigned i = 0, e = FnLoadArgs.size(); i != e; ++i)
 113       MFI->setObjectOffset(FnLoadArgs[i].FI, FnLoadArgs[i].SPOffset);
 114   }
 115   void adjustStoreVarArgsFI(MachineFrameInfo *MFI) const {
 116     if (!hasStoreVarArgs()) return;
 117     for (unsigned i = 0, e = FnStoreVarArgs.size(); i != e; ++i)
 118       MFI->setObjectOffset(FnStoreVarArgs[i].FI, FnStoreVarArgs[i].SPOffset);
 119   }
 120
 121   unsigned getSRetReturnReg() const { return SRetReturnReg; }
 122   void setSRetReturnReg(unsigned Reg) { SRetReturnReg = Reg; }
 123
 124   unsigned getGlobalBaseReg() const { return GlobalBaseReg; }
 125   void setGlobalBaseReg(unsigned Reg) { GlobalBaseReg = Reg; }
 126
 127   int getVarArgsFrameIndex() const { return VarArgsFrameIndex; }
 128   void setVarArgsFrameIndex(int Index) { VarArgsFrameIndex = Index; }
 129 };
 130
 131 } // end of namespace llvm
 132
 133 #endif // MBLAZE_MACHINE_FUNCTION_INFO_H