llvm/lib/Target/LoongArch/LoongArchMachineFunctionInfo.h

   1 //=- LoongArchMachineFunctionInfo.h - LoongArch machine function info -----===//
   2 //
   3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
   4 // See https://llvm.org/LICENSE.txt for license information.
   5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
   6 //
   7 //===----------------------------------------------------------------------===//
   8 //
   9 // This file declares LoongArch-specific per-machine-function information.
  10 //
  11 //===----------------------------------------------------------------------===//
  12
  13 #ifndef LLVM_LIB_TARGET_LOONGARCH_LOONGARCHMACHINEFUNCTIONINFO_H
  14 #define LLVM_LIB_TARGET_LOONGARCH_LOONGARCHMACHINEFUNCTIONINFO_H
  15
  16 #include "LoongArchSubtarget.h"
  17 #include "llvm/CodeGen/MachineFrameInfo.h"
  18 #include "llvm/CodeGen/MachineFunction.h"
  19
  20 namespace llvm {
  21
  22 /// LoongArchMachineFunctionInfo - This class is derived from
  23 /// MachineFunctionInfo and contains private LoongArch-specific information for
  24 /// each MachineFunction.
  25 class LoongArchMachineFunctionInfo : public MachineFunctionInfo {
  26 private:
  27   /// FrameIndex for start of varargs area
  28   int VarArgsFrameIndex = 0;
  29   /// Size of the save area used for varargs
  30   int VarArgsSaveSize = 0;
  31
  32   /// Size of stack frame to save callee saved registers
  33   unsigned CalleeSavedStackSize = 0;
  34
  35   /// FrameIndex of the spill slot when there is no scavenged register in
  36   /// insertIndirectBranch.
  37   int BranchRelaxationSpillFrameIndex = -1;
  38
  39   /// Registers that have been sign extended from i32.
  40   SmallVector<Register, 8> SExt32Registers;
  41
  42   /// Pairs of `jr` instructions and corresponding JTI operands, used for the
  43   /// `annotate-tablejump` option.
  44   SmallVector<std::pair<MachineInstr *, MachineOperand *>, 4> JumpInfos;
  45
  46 public:
  47   LoongArchMachineFunctionInfo(const Function &F,
  48                                const TargetSubtargetInfo *STI) {}
  49
  50   MachineFunctionInfo *
  51   clone(BumpPtrAllocator &Allocator, MachineFunction &DestMF,
  52         const DenseMap<MachineBasicBlock *, MachineBasicBlock *> &Src2DstMBB)
  53       const override {
  54     return DestMF.cloneInfo<LoongArchMachineFunctionInfo>(*this);
  55   }
  56
  57   int getVarArgsFrameIndex() const { return VarArgsFrameIndex; }
  58   void setVarArgsFrameIndex(int Index) { VarArgsFrameIndex = Index; }
  59
  60   unsigned getVarArgsSaveSize() const { return VarArgsSaveSize; }
  61   void setVarArgsSaveSize(int Size) { VarArgsSaveSize = Size; }
  62
  63   unsigned getCalleeSavedStackSize() const { return CalleeSavedStackSize; }
  64   void setCalleeSavedStackSize(unsigned Size) { CalleeSavedStackSize = Size; }
  65
  66   int getBranchRelaxationSpillFrameIndex() {
  67     return BranchRelaxationSpillFrameIndex;
  68   }
  69   void setBranchRelaxationSpillFrameIndex(int Index) {
  70     BranchRelaxationSpillFrameIndex = Index;
  71   }
  72
  73   void addSExt32Register(Register Reg) { SExt32Registers.push_back(Reg); }
  74
  75   bool isSExt32Register(Register Reg) const {
  76     return is_contained(SExt32Registers, Reg);
  77   }
  78
  79   void setJumpInfo(MachineInstr *JrMI, MachineOperand *JTIMO) {
  80     JumpInfos.push_back(std::make_pair(JrMI, JTIMO));
  81   }
  82   unsigned getJumpInfoSize() { return JumpInfos.size(); }
  83   MachineInstr *getJumpInfoJrMI(unsigned Idx) { return JumpInfos[Idx].first; }
  84   MachineOperand *getJumpInfoJTIMO(unsigned Idx) {
  85     return JumpInfos[Idx].second;
  86   }
  87 };
  88
  89 } // end namespace llvm
  90
  91 #endif // LLVM_LIB_TARGET_LOONGARCH_LOONGARCHMACHINEFUNCTIONINFO_H