lib/Target/ARM/ARMRegisterInfo.cpp

   1 //===- ARMRegisterInfo.cpp - ARM Register Information -----------*- C++ -*-===//
   2 //
   3 //                     The LLVM Compiler Infrastructure
   4 //
   5 // This file was developed by the "Instituto Nokia de Tecnologia" and
   6 // is distributed under the University of Illinois Open Source
   7 // License. See LICENSE.TXT for details.
   8 //
   9 //===----------------------------------------------------------------------===//
  10 //
  11 // This file contains the ARM implementation of the MRegisterInfo class.
  12 //
  13 //===----------------------------------------------------------------------===//
  14
  15 #include "ARM.h"
  16 #include "ARMRegisterInfo.h"
  17 #include "llvm/CodeGen/MachineInstrBuilder.h"
  18 #include "llvm/CodeGen/MachineFunction.h"
  19 #include "llvm/CodeGen/MachineFrameInfo.h"
  20 #include "llvm/CodeGen/MachineLocation.h"
  21 #include "llvm/Type.h"
  22 #include "llvm/ADT/STLExtras.h"
  23 #include <iostream>
  24 using namespace llvm;
  25
  26 ARMRegisterInfo::ARMRegisterInfo()
  27   : ARMGenRegisterInfo(ARM::ADJCALLSTACKDOWN, ARM::ADJCALLSTACKUP) {
  28 }
  29
  30 void ARMRegisterInfo::
  31 storeRegToStackSlot(MachineBasicBlock &MBB, MachineBasicBlock::iterator I,
  32                     unsigned SrcReg, int FI,
  33                     const TargetRegisterClass *RC) const {
  34   // On the order of operands here: think "[FI + 0] = SrcReg".
  35   assert (RC == ARM::IntRegsRegisterClass);
  36   BuildMI(MBB, I, ARM::str, 3).addFrameIndex(FI).addImm(0).addReg(SrcReg);
  37 }
  38
  39 void ARMRegisterInfo::
  40 loadRegFromStackSlot(MachineBasicBlock &MBB, MachineBasicBlock::iterator I,
  41                      unsigned DestReg, int FI,
  42                      const TargetRegisterClass *RC) const {
  43   assert (RC == ARM::IntRegsRegisterClass);
  44   BuildMI(MBB, I, ARM::ldr, 2, DestReg).addFrameIndex(FI).addImm(0);
  45 }
  46
  47 void ARMRegisterInfo::copyRegToReg(MachineBasicBlock &MBB,
  48                                      MachineBasicBlock::iterator I,
  49                                      unsigned DestReg, unsigned SrcReg,
  50                                      const TargetRegisterClass *RC) const {
  51   assert (RC == ARM::IntRegsRegisterClass);
  52   BuildMI(MBB, I, ARM::movrr, 1, DestReg).addReg(SrcReg);
  53 }
  54
  55 MachineInstr *ARMRegisterInfo::foldMemoryOperand(MachineInstr* MI,
  56                                                    unsigned OpNum,
  57                                                    int FI) const {
  58   return NULL;
  59 }
  60
  61 const unsigned* ARMRegisterInfo::getCalleeSaveRegs() const {
  62   static const unsigned CalleeSaveRegs[] = { 0 };
  63   return CalleeSaveRegs;
  64 }
  65
  66 const TargetRegisterClass* const *
  67 ARMRegisterInfo::getCalleeSaveRegClasses() const {
  68   static const TargetRegisterClass * const CalleeSaveRegClasses[] = { 0 };
  69   return CalleeSaveRegClasses;
  70 }
  71
  72 void ARMRegisterInfo::
  73 eliminateCallFramePseudoInstr(MachineFunction &MF, MachineBasicBlock &MBB,
  74                               MachineBasicBlock::iterator I) const {
  75   MBB.erase(I);
  76 }
  77
  78 void
  79 ARMRegisterInfo::eliminateFrameIndex(MachineBasicBlock::iterator II) const {
  80   MachineInstr &MI = *II;
  81   MachineBasicBlock &MBB = *MI.getParent();
  82   MachineFunction &MF = *MBB.getParent();
  83
  84   assert (MI.getOpcode() == ARM::ldr);
  85
  86   unsigned FrameIdx = 2;
  87   unsigned OffIdx = 1;
  88
  89   int FrameIndex = MI.getOperand(FrameIdx).getFrameIndex();
  90
  91   int Offset = MF.getFrameInfo()->getObjectOffset(FrameIndex);
  92   assert (MI.getOperand(OffIdx).getImmedValue() == 0);
  93
  94   unsigned StackSize = MF.getFrameInfo()->getStackSize();
  95
  96   Offset += StackSize;
  97
  98   assert (Offset >= 0);
  99   if (Offset < 4096) {
 100     // Replace the FrameIndex with r13
 101     MI.getOperand(FrameIdx).ChangeToRegister(ARM::R13);
 102     // Replace the ldr offset with Offset
 103     MI.getOperand(OffIdx).ChangeToImmediate(Offset);
 104   } else {
 105     // Insert a set of r12 with the full address
 106     // r12 = r13 + offset
 107     MachineBasicBlock *MBB2 = MI.getParent();
 108     BuildMI(*MBB2, II, ARM::addri, 2, ARM::R12).addReg(ARM::R13).addImm(Offset);
 109
 110     // Replace the FrameIndex with r12
 111     MI.getOperand(FrameIdx).ChangeToRegister(ARM::R12);
 112   }
 113 }
 114
 115 void ARMRegisterInfo::
 116 processFunctionBeforeFrameFinalized(MachineFunction &MF) const {}
 117
 118 void ARMRegisterInfo::emitPrologue(MachineFunction &MF) const {
 119   MachineBasicBlock &MBB = MF.front();
 120   MachineBasicBlock::iterator MBBI = MBB.begin();
 121   MachineFrameInfo  *MFI = MF.getFrameInfo();
 122   int           NumBytes = (int) MFI->getStackSize();
 123
 124   //hack
 125   assert(NumBytes == 0);
 126
 127   if (MFI->hasCalls()) {
 128     // We reserve argument space for call sites in the function immediately on
 129     // entry to the current function.  This eliminates the need for add/sub
 130     // brackets around call sites.
 131     NumBytes += MFI->getMaxCallFrameSize();
 132   }
 133
 134   MFI->setStackSize(NumBytes);
 135
 136   //sub sp, sp, #NumBytes
 137   BuildMI(MBB, MBBI, ARM::subri, 2, ARM::R13).addReg(ARM::R13).addImm(NumBytes);
 138   //str lr, [sp, #NumBytes - 4]
 139   BuildMI(MBB, MBBI, ARM::str, 2, ARM::R14).addImm(NumBytes - 4).addReg(ARM::R13);
 140 }
 141
 142 void ARMRegisterInfo::emitEpilogue(MachineFunction &MF,
 143                                    MachineBasicBlock &MBB) const {
 144   MachineBasicBlock::iterator MBBI = prior(MBB.end());
 145   assert(MBBI->getOpcode() == ARM::bx &&
 146          "Can only insert epilog into returning blocks");
 147
 148   MachineFrameInfo *MFI = MF.getFrameInfo();
 149   int          NumBytes = (int) MFI->getStackSize();
 150
 151   //ldr lr, [sp]
 152   BuildMI(MBB, MBBI, ARM::ldr, 2, ARM::R14).addImm(0).addReg(ARM::R13);
 153   //add sp, sp, #NumBytes
 154   BuildMI(MBB, MBBI, ARM::addri, 2, ARM::R13).addReg(ARM::R13).addImm(NumBytes);
 155 }
 156
 157 unsigned ARMRegisterInfo::getRARegister() const {
 158   return ARM::R14;
 159 }
 160
 161 unsigned ARMRegisterInfo::getFrameRegister(MachineFunction &MF) const {
 162   return ARM::R13;
 163 }
 164
 165 #include "ARMGenRegisterInfo.inc"
 166