1234353Sdim//===-- 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" 19249423Sdim#include "llvm/CodeGen/MachineFrameInfo.h" 20204642Srdivacky#include "llvm/CodeGen/MachineFunction.h" 21204642Srdivacky 22204642Srdivackynamespace llvm { 23204642Srdivacky 24204642Srdivacky/// MBlazeFunctionInfo - This class is derived from MachineFunction private 25204642Srdivacky/// MBlaze target-specific information for each MachineFunction. 26204642Srdivackyclass MBlazeFunctionInfo : public MachineFunctionInfo { 27234353Sdim virtual void anchor(); 28204642Srdivacky 29218893Sdim /// Holds for each function where on the stack the Frame Pointer must be 30204642Srdivacky /// saved. This is used on Prologue and Epilogue to emit FP save/restore 31204642Srdivacky int FPStackOffset; 32204642Srdivacky 33218893Sdim /// Holds for each function where on the stack the Return Address must be 34204642Srdivacky /// saved. This is used on Prologue and Epilogue to emit RA save/restore 35204642Srdivacky int RAStackOffset; 36204642Srdivacky 37204642Srdivacky /// MBlazeFIHolder - Holds a FrameIndex and it's Stack Pointer Offset 38204642Srdivacky struct MBlazeFIHolder { 39204642Srdivacky 40204642Srdivacky int FI; 41204642Srdivacky int SPOffset; 42204642Srdivacky 43204642Srdivacky MBlazeFIHolder(int FrameIndex, int StackPointerOffset) 44204642Srdivacky : FI(FrameIndex), SPOffset(StackPointerOffset) {} 45204642Srdivacky }; 46204642Srdivacky 47218893Sdim /// When PIC is used the GP must be saved on the stack on the function 48218893Sdim /// prologue and must be reloaded from this stack location after every 49218893Sdim /// call. A reference to its stack location and frame index must be kept 50204642Srdivacky /// to be used on emitPrologue and processFunctionBeforeFrameFinalized. 51204642Srdivacky MBlazeFIHolder GPHolder; 52204642Srdivacky 53204642Srdivacky /// On LowerFormalArguments the stack size is unknown, so the Stack 54218893Sdim /// Pointer Offset calculation of "not in register arguments" must be 55218893Sdim /// postponed to emitPrologue. 56204642Srdivacky SmallVector<MBlazeFIHolder, 16> FnLoadArgs; 57204642Srdivacky bool HasLoadArgs; 58204642Srdivacky 59218893Sdim // When VarArgs, we must write registers back to caller stack, preserving 60218893Sdim // on register arguments. Since the stack size is unknown on 61204642Srdivacky // LowerFormalArguments, the Stack Pointer Offset calculation must be 62218893Sdim // postponed to emitPrologue. 63204642Srdivacky SmallVector<MBlazeFIHolder, 4> FnStoreVarArgs; 64204642Srdivacky bool HasStoreVarArgs; 65204642Srdivacky 66218893Sdim // When determining the final stack layout some of the frame indexes may 67218893Sdim // be replaced by new frame indexes that reside in the caller's stack 68218893Sdim // frame. The replacements are recorded in this structure. 69218893Sdim DenseMap<int,int> FIReplacements; 70218893Sdim 71204642Srdivacky /// SRetReturnReg - Some subtargets require that sret lowering includes 72204642Srdivacky /// returning the value of the returned struct in a register. This field 73204642Srdivacky /// holds the virtual register into which the sret argument is passed. 74204642Srdivacky unsigned SRetReturnReg; 75204642Srdivacky 76204642Srdivacky /// GlobalBaseReg - keeps track of the virtual register initialized for 77204642Srdivacky /// use as the global base register. This is used for PIC in some PIC 78204642Srdivacky /// relocation models. 79204642Srdivacky unsigned GlobalBaseReg; 80204642Srdivacky 81207618Srdivacky // VarArgsFrameIndex - FrameIndex for start of varargs area. 82207618Srdivacky int VarArgsFrameIndex; 83207618Srdivacky 84218893Sdim /// LiveInFI - keeps track of the frame indexes in a callers stack 85218893Sdim /// frame that are live into a function. 86218893Sdim SmallVector<int, 16> LiveInFI; 87218893Sdim 88204642Srdivackypublic: 89218893Sdim MBlazeFunctionInfo(MachineFunction& MF) 90218893Sdim : FPStackOffset(0), RAStackOffset(0), GPHolder(-1,-1), HasLoadArgs(false), 91218893Sdim HasStoreVarArgs(false), SRetReturnReg(0), GlobalBaseReg(0), 92218893Sdim VarArgsFrameIndex(0), LiveInFI() 93204642Srdivacky {} 94204642Srdivacky 95204642Srdivacky int getFPStackOffset() const { return FPStackOffset; } 96204642Srdivacky void setFPStackOffset(int Off) { FPStackOffset = Off; } 97204642Srdivacky 98204642Srdivacky int getRAStackOffset() const { return RAStackOffset; } 99204642Srdivacky void setRAStackOffset(int Off) { RAStackOffset = Off; } 100204642Srdivacky 101204642Srdivacky int getGPStackOffset() const { return GPHolder.SPOffset; } 102204642Srdivacky int getGPFI() const { return GPHolder.FI; } 103204642Srdivacky void setGPStackOffset(int Off) { GPHolder.SPOffset = Off; } 104204642Srdivacky void setGPFI(int FI) { GPHolder.FI = FI; } 105204642Srdivacky bool needGPSaveRestore() const { return GPHolder.SPOffset != -1; } 106204642Srdivacky 107204642Srdivacky bool hasLoadArgs() const { return HasLoadArgs; } 108218893Sdim bool hasStoreVarArgs() const { return HasStoreVarArgs; } 109204642Srdivacky 110218893Sdim void recordLiveIn(int FI) { 111218893Sdim LiveInFI.push_back(FI); 112218893Sdim } 113218893Sdim 114218893Sdim bool isLiveIn(int FI) { 115218893Sdim for (unsigned i = 0, e = LiveInFI.size(); i < e; ++i) 116218893Sdim if (FI == LiveInFI[i]) return true; 117218893Sdim 118218893Sdim return false; 119218893Sdim } 120218893Sdim 121218893Sdim const SmallVector<int, 16>& getLiveIn() const { return LiveInFI; } 122218893Sdim 123218893Sdim void recordReplacement(int OFI, int NFI) { 124218893Sdim FIReplacements.insert(std::make_pair(OFI,NFI)); 125218893Sdim } 126218893Sdim 127218893Sdim bool hasReplacement(int OFI) const { 128218893Sdim return FIReplacements.find(OFI) != FIReplacements.end(); 129218893Sdim } 130218893Sdim 131218893Sdim int getReplacement(int OFI) const { 132218893Sdim return FIReplacements.lookup(OFI); 133218893Sdim } 134218893Sdim 135204642Srdivacky void recordLoadArgsFI(int FI, int SPOffset) { 136204642Srdivacky if (!HasLoadArgs) HasLoadArgs=true; 137204642Srdivacky FnLoadArgs.push_back(MBlazeFIHolder(FI, SPOffset)); 138204642Srdivacky } 139218893Sdim 140204642Srdivacky void recordStoreVarArgsFI(int FI, int SPOffset) { 141204642Srdivacky if (!HasStoreVarArgs) HasStoreVarArgs=true; 142204642Srdivacky FnStoreVarArgs.push_back(MBlazeFIHolder(FI, SPOffset)); 143204642Srdivacky } 144204642Srdivacky 145204642Srdivacky void adjustLoadArgsFI(MachineFrameInfo *MFI) const { 146204642Srdivacky if (!hasLoadArgs()) return; 147218893Sdim for (unsigned i = 0, e = FnLoadArgs.size(); i != e; ++i) 148218893Sdim MFI->setObjectOffset(FnLoadArgs[i].FI, FnLoadArgs[i].SPOffset); 149204642Srdivacky } 150218893Sdim 151204642Srdivacky void adjustStoreVarArgsFI(MachineFrameInfo *MFI) const { 152218893Sdim if (!hasStoreVarArgs()) return; 153218893Sdim for (unsigned i = 0, e = FnStoreVarArgs.size(); i != e; ++i) 154218893Sdim MFI->setObjectOffset(FnStoreVarArgs[i].FI, FnStoreVarArgs[i].SPOffset); 155204642Srdivacky } 156204642Srdivacky 157204642Srdivacky unsigned getSRetReturnReg() const { return SRetReturnReg; } 158204642Srdivacky void setSRetReturnReg(unsigned Reg) { SRetReturnReg = Reg; } 159204642Srdivacky 160204642Srdivacky unsigned getGlobalBaseReg() const { return GlobalBaseReg; } 161204642Srdivacky void setGlobalBaseReg(unsigned Reg) { GlobalBaseReg = Reg; } 162207618Srdivacky 163207618Srdivacky int getVarArgsFrameIndex() const { return VarArgsFrameIndex; } 164207618Srdivacky void setVarArgsFrameIndex(int Index) { VarArgsFrameIndex = Index; } 165204642Srdivacky}; 166204642Srdivacky 167204642Srdivacky} // end of namespace llvm 168204642Srdivacky 169204642Srdivacky#endif // MBLAZE_MACHINE_FUNCTION_INFO_H 170