MBlazeMachineFunction.h revision 218893
1204642Srdivacky//===-- MBlazeMachineFunctionInfo.h - Private data ----------------*- C++ -*-=// 2204642Srdivacky// 3204642Srdivacky// The LLVM Compiler Infrastructure 4204642Srdivacky// 5204642Srdivacky// This file is distributed under the University of Illinois Open Source 6204642Srdivacky// License. See LICENSE.TXT for details. 7204642Srdivacky// 8204642Srdivacky//===----------------------------------------------------------------------===// 9204642Srdivacky// 10204642Srdivacky// This file declares the MBlaze specific subclass of MachineFunctionInfo. 11204642Srdivacky// 12204642Srdivacky//===----------------------------------------------------------------------===// 13204642Srdivacky 14204642Srdivacky#ifndef MBLAZE_MACHINE_FUNCTION_INFO_H 15204642Srdivacky#define MBLAZE_MACHINE_FUNCTION_INFO_H 16204642Srdivacky 17218893Sdim#include "llvm/ADT/DenseMap.h" 18204642Srdivacky#include "llvm/ADT/SmallVector.h" 19204642Srdivacky#include "llvm/ADT/VectorExtras.h" 20204642Srdivacky#include "llvm/CodeGen/MachineFunction.h" 21204642Srdivacky#include "llvm/CodeGen/MachineFrameInfo.h" 22204642Srdivacky 23204642Srdivackynamespace llvm { 24204642Srdivacky 25204642Srdivacky/// MBlazeFunctionInfo - This class is derived from MachineFunction private 26204642Srdivacky/// MBlaze target-specific information for each MachineFunction. 27204642Srdivackyclass MBlazeFunctionInfo : public MachineFunctionInfo { 28204642Srdivacky 29204642Srdivackyprivate: 30218893Sdim /// Holds for each function where on the stack the Frame Pointer must be 31204642Srdivacky /// saved. This is used on Prologue and Epilogue to emit FP save/restore 32204642Srdivacky int FPStackOffset; 33204642Srdivacky 34218893Sdim /// Holds for each function where on the stack the Return Address must be 35204642Srdivacky /// saved. This is used on Prologue and Epilogue to emit RA save/restore 36204642Srdivacky int RAStackOffset; 37204642Srdivacky 38204642Srdivacky /// MBlazeFIHolder - Holds a FrameIndex and it's Stack Pointer Offset 39204642Srdivacky struct MBlazeFIHolder { 40204642Srdivacky 41204642Srdivacky int FI; 42204642Srdivacky int SPOffset; 43204642Srdivacky 44204642Srdivacky MBlazeFIHolder(int FrameIndex, int StackPointerOffset) 45204642Srdivacky : FI(FrameIndex), SPOffset(StackPointerOffset) {} 46204642Srdivacky }; 47204642Srdivacky 48218893Sdim /// When PIC is used the GP must be saved on the stack on the function 49218893Sdim /// prologue and must be reloaded from this stack location after every 50218893Sdim /// call. A reference to its stack location and frame index must be kept 51204642Srdivacky /// to be used on emitPrologue and processFunctionBeforeFrameFinalized. 52204642Srdivacky MBlazeFIHolder GPHolder; 53204642Srdivacky 54204642Srdivacky /// On LowerFormalArguments the stack size is unknown, so the Stack 55218893Sdim /// Pointer Offset calculation of "not in register arguments" must be 56218893Sdim /// postponed to emitPrologue. 57204642Srdivacky SmallVector<MBlazeFIHolder, 16> FnLoadArgs; 58204642Srdivacky bool HasLoadArgs; 59204642Srdivacky 60218893Sdim // When VarArgs, we must write registers back to caller stack, preserving 61218893Sdim // on register arguments. Since the stack size is unknown on 62204642Srdivacky // LowerFormalArguments, the Stack Pointer Offset calculation must be 63218893Sdim // postponed to emitPrologue. 64204642Srdivacky SmallVector<MBlazeFIHolder, 4> FnStoreVarArgs; 65204642Srdivacky bool HasStoreVarArgs; 66204642Srdivacky 67218893Sdim // When determining the final stack layout some of the frame indexes may 68218893Sdim // be replaced by new frame indexes that reside in the caller's stack 69218893Sdim // frame. The replacements are recorded in this structure. 70218893Sdim DenseMap<int,int> FIReplacements; 71218893Sdim 72204642Srdivacky /// SRetReturnReg - Some subtargets require that sret lowering includes 73204642Srdivacky /// returning the value of the returned struct in a register. This field 74204642Srdivacky /// holds the virtual register into which the sret argument is passed. 75204642Srdivacky unsigned SRetReturnReg; 76204642Srdivacky 77204642Srdivacky /// GlobalBaseReg - keeps track of the virtual register initialized for 78204642Srdivacky /// use as the global base register. This is used for PIC in some PIC 79204642Srdivacky /// relocation models. 80204642Srdivacky unsigned GlobalBaseReg; 81204642Srdivacky 82207618Srdivacky // VarArgsFrameIndex - FrameIndex for start of varargs area. 83207618Srdivacky int VarArgsFrameIndex; 84207618Srdivacky 85218893Sdim /// LiveInFI - keeps track of the frame indexes in a callers stack 86218893Sdim /// frame that are live into a function. 87218893Sdim SmallVector<int, 16> LiveInFI; 88218893Sdim 89204642Srdivackypublic: 90218893Sdim MBlazeFunctionInfo(MachineFunction& MF) 91218893Sdim : FPStackOffset(0), RAStackOffset(0), GPHolder(-1,-1), HasLoadArgs(false), 92218893Sdim HasStoreVarArgs(false), SRetReturnReg(0), GlobalBaseReg(0), 93218893Sdim VarArgsFrameIndex(0), LiveInFI() 94204642Srdivacky {} 95204642Srdivacky 96204642Srdivacky int getFPStackOffset() const { return FPStackOffset; } 97204642Srdivacky void setFPStackOffset(int Off) { FPStackOffset = Off; } 98204642Srdivacky 99204642Srdivacky int getRAStackOffset() const { return RAStackOffset; } 100204642Srdivacky void setRAStackOffset(int Off) { RAStackOffset = Off; } 101204642Srdivacky 102204642Srdivacky int getGPStackOffset() const { return GPHolder.SPOffset; } 103204642Srdivacky int getGPFI() const { return GPHolder.FI; } 104204642Srdivacky void setGPStackOffset(int Off) { GPHolder.SPOffset = Off; } 105204642Srdivacky void setGPFI(int FI) { GPHolder.FI = FI; } 106204642Srdivacky bool needGPSaveRestore() const { return GPHolder.SPOffset != -1; } 107204642Srdivacky 108204642Srdivacky bool hasLoadArgs() const { return HasLoadArgs; } 109218893Sdim bool hasStoreVarArgs() const { return HasStoreVarArgs; } 110204642Srdivacky 111218893Sdim void recordLiveIn(int FI) { 112218893Sdim LiveInFI.push_back(FI); 113218893Sdim } 114218893Sdim 115218893Sdim bool isLiveIn(int FI) { 116218893Sdim for (unsigned i = 0, e = LiveInFI.size(); i < e; ++i) 117218893Sdim if (FI == LiveInFI[i]) return true; 118218893Sdim 119218893Sdim return false; 120218893Sdim } 121218893Sdim 122218893Sdim const SmallVector<int, 16>& getLiveIn() const { return LiveInFI; } 123218893Sdim 124218893Sdim void recordReplacement(int OFI, int NFI) { 125218893Sdim FIReplacements.insert(std::make_pair(OFI,NFI)); 126218893Sdim } 127218893Sdim 128218893Sdim bool hasReplacement(int OFI) const { 129218893Sdim return FIReplacements.find(OFI) != FIReplacements.end(); 130218893Sdim } 131218893Sdim 132218893Sdim int getReplacement(int OFI) const { 133218893Sdim return FIReplacements.lookup(OFI); 134218893Sdim } 135218893Sdim 136204642Srdivacky void recordLoadArgsFI(int FI, int SPOffset) { 137204642Srdivacky if (!HasLoadArgs) HasLoadArgs=true; 138204642Srdivacky FnLoadArgs.push_back(MBlazeFIHolder(FI, SPOffset)); 139204642Srdivacky } 140218893Sdim 141204642Srdivacky void recordStoreVarArgsFI(int FI, int SPOffset) { 142204642Srdivacky if (!HasStoreVarArgs) HasStoreVarArgs=true; 143204642Srdivacky FnStoreVarArgs.push_back(MBlazeFIHolder(FI, SPOffset)); 144204642Srdivacky } 145204642Srdivacky 146204642Srdivacky void adjustLoadArgsFI(MachineFrameInfo *MFI) const { 147204642Srdivacky if (!hasLoadArgs()) return; 148218893Sdim for (unsigned i = 0, e = FnLoadArgs.size(); i != e; ++i) 149218893Sdim MFI->setObjectOffset(FnLoadArgs[i].FI, FnLoadArgs[i].SPOffset); 150204642Srdivacky } 151218893Sdim 152204642Srdivacky void adjustStoreVarArgsFI(MachineFrameInfo *MFI) const { 153218893Sdim if (!hasStoreVarArgs()) return; 154218893Sdim for (unsigned i = 0, e = FnStoreVarArgs.size(); i != e; ++i) 155218893Sdim MFI->setObjectOffset(FnStoreVarArgs[i].FI, FnStoreVarArgs[i].SPOffset); 156204642Srdivacky } 157204642Srdivacky 158204642Srdivacky unsigned getSRetReturnReg() const { return SRetReturnReg; } 159204642Srdivacky void setSRetReturnReg(unsigned Reg) { SRetReturnReg = Reg; } 160204642Srdivacky 161204642Srdivacky unsigned getGlobalBaseReg() const { return GlobalBaseReg; } 162204642Srdivacky void setGlobalBaseReg(unsigned Reg) { GlobalBaseReg = Reg; } 163207618Srdivacky 164207618Srdivacky int getVarArgsFrameIndex() const { return VarArgsFrameIndex; } 165207618Srdivacky void setVarArgsFrameIndex(int Index) { VarArgsFrameIndex = Index; } 166204642Srdivacky}; 167204642Srdivacky 168204642Srdivacky} // end of namespace llvm 169204642Srdivacky 170204642Srdivacky#endif // MBLAZE_MACHINE_FUNCTION_INFO_H 171