include/llvm/CodeGen/MachineInstrBuilder.h

   1 //===-- CodeGen/MachineInstBuilder.h - Simplify creation of MIs -*- 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 exposes a function named BuildMI, which is useful for dramatically
  11 // simplifying how MachineInstr's are created.  It allows use of code like this:
  12 //
  13 //   M = BuildMI(X86::ADDrr8, 2).addReg(argVal1).addReg(argVal2);
  14 //
  15 //===----------------------------------------------------------------------===//
  16
  17 #ifndef LLVM_CODEGEN_MACHINEINSTRBUILDER_H
  18 #define LLVM_CODEGEN_MACHINEINSTRBUILDER_H
  19
  20 #include "llvm/CodeGen/MachineFunction.h"
  21
  22 namespace llvm {
  23
  24 class TargetInstrDesc;
  25
  26 namespace RegState {
  27   enum {
  28     Define         = 0x2,
  29     Implicit       = 0x4,
  30     Kill           = 0x8,
  31     Dead           = 0x10,
  32     Undef          = 0x20,
  33     EarlyClobber   = 0x40,
  34     ImplicitDefine = Implicit | Define,
  35     ImplicitKill   = Implicit | Kill
  36   };
  37 }
  38
  39 class MachineInstrBuilder {
  40   MachineInstr *MI;
  41 public:
  42   MachineInstrBuilder() : MI(0) {}
  43   explicit MachineInstrBuilder(MachineInstr *mi) : MI(mi) {}
  44
  45   /// Allow automatic conversion to the machine instruction we are working on.
  46   ///
  47   operator MachineInstr*() const { return MI; }
  48   operator MachineBasicBlock::iterator() const { return MI; }
  49
  50   /// addReg - Add a new virtual register operand...
  51   ///
  52   const
  53   MachineInstrBuilder &addReg(unsigned RegNo, unsigned flags = 0,
  54                               unsigned SubReg = 0) const {
  55     assert((flags & 0x1) == 0 &&
  56            "Passing in 'true' to addReg is forbidden! Use enums instead.");
  57     MI->addOperand(MachineOperand::CreateReg(RegNo,
  58                                              flags & RegState::Define,
  59                                              flags & RegState::Implicit,
  60                                              flags & RegState::Kill,
  61                                              flags & RegState::Dead,
  62                                              flags & RegState::Undef,
  63                                              flags & RegState::EarlyClobber,
  64                                              SubReg));
  65     return *this;
  66   }
  67
  68   /// addImm - Add a new immediate operand.
  69   ///
  70   const MachineInstrBuilder &addImm(int64_t Val) const {
  71     MI->addOperand(MachineOperand::CreateImm(Val));
  72     return *this;
  73   }
  74
  75   const MachineInstrBuilder &addFPImm(const ConstantFP *Val) const {
  76     MI->addOperand(MachineOperand::CreateFPImm(Val));
  77     return *this;
  78   }
  79
  80   const MachineInstrBuilder &addMBB(MachineBasicBlock *MBB,
  81                                     unsigned char TargetFlags = 0) const {
  82     MI->addOperand(MachineOperand::CreateMBB(MBB, TargetFlags));
  83     return *this;
  84   }
  85
  86   const MachineInstrBuilder &addFrameIndex(unsigned Idx) const {
  87     MI->addOperand(MachineOperand::CreateFI(Idx));
  88     return *this;
  89   }
  90
  91   const MachineInstrBuilder &addConstantPoolIndex(unsigned Idx,
  92                                                   int Offset = 0,
  93                                           unsigned char TargetFlags = 0) const {
  94     MI->addOperand(MachineOperand::CreateCPI(Idx, Offset, TargetFlags));
  95     return *this;
  96   }
  97
  98   const MachineInstrBuilder &addJumpTableIndex(unsigned Idx,
  99                                           unsigned char TargetFlags = 0) const {
 100     MI->addOperand(MachineOperand::CreateJTI(Idx, TargetFlags));
 101     return *this;
 102   }
 103
 104   const MachineInstrBuilder &addGlobalAddress(GlobalValue *GV,
 105                                               int64_t Offset = 0,
 106                                           unsigned char TargetFlags = 0) const {
 107     MI->addOperand(MachineOperand::CreateGA(GV, Offset, TargetFlags));
 108     return *this;
 109   }
 110
 111   const MachineInstrBuilder &addExternalSymbol(const char *FnName,
 112                                           unsigned char TargetFlags = 0) const {
 113     MI->addOperand(MachineOperand::CreateES(FnName, TargetFlags));
 114     return *this;
 115   }
 116
 117   const MachineInstrBuilder &addMemOperand(MachineMemOperand *MMO) const {
 118     MI->addMemOperand(*MI->getParent()->getParent(), MMO);
 119     return *this;
 120   }
 121
 122   const MachineInstrBuilder &addOperand(const MachineOperand &MO) const {
 123     MI->addOperand(MO);
 124     return *this;
 125   }
 126 };
 127
 128 /// BuildMI - Builder interface.  Specify how to create the initial instruction
 129 /// itself.
 130 ///
 131 inline MachineInstrBuilder BuildMI(MachineFunction &MF,
 132                                    DebugLoc DL,
 133                                    const TargetInstrDesc &TID) {
 134   return MachineInstrBuilder(MF.CreateMachineInstr(TID, DL));
 135 }
 136
 137 /// BuildMI - This version of the builder sets up the first operand as a
 138 /// destination virtual register.
 139 ///
 140 inline MachineInstrBuilder BuildMI(MachineFunction &MF,
 141                                    DebugLoc DL,
 142                                    const TargetInstrDesc &TID,
 143                                    unsigned DestReg) {
 144   return MachineInstrBuilder(MF.CreateMachineInstr(TID, DL))
 145            .addReg(DestReg, RegState::Define);
 146 }
 147
 148 /// BuildMI - This version of the builder inserts the newly-built
 149 /// instruction before the given position in the given MachineBasicBlock, and
 150 /// sets up the first operand as a destination virtual register.
 151 ///
 152 inline MachineInstrBuilder BuildMI(MachineBasicBlock &BB,
 153                                    MachineBasicBlock::iterator I,
 154                                    DebugLoc DL,
 155                                    const TargetInstrDesc &TID,
 156                                    unsigned DestReg) {
 157   MachineInstr *MI = BB.getParent()->CreateMachineInstr(TID, DL);
 158   BB.insert(I, MI);
 159   return MachineInstrBuilder(MI).addReg(DestReg, RegState::Define);
 160 }
 161
 162 /// BuildMI - This version of the builder inserts the newly-built
 163 /// instruction before the given position in the given MachineBasicBlock, and
 164 /// does NOT take a destination register.
 165 ///
 166 inline MachineInstrBuilder BuildMI(MachineBasicBlock &BB,
 167                                    MachineBasicBlock::iterator I,
 168                                    DebugLoc DL,
 169                                    const TargetInstrDesc &TID) {
 170   MachineInstr *MI = BB.getParent()->CreateMachineInstr(TID, DL);
 171   BB.insert(I, MI);
 172   return MachineInstrBuilder(MI);
 173 }
 174
 175 /// BuildMI - This version of the builder inserts the newly-built
 176 /// instruction at the end of the given MachineBasicBlock, and does NOT take a
 177 /// destination register.
 178 ///
 179 inline MachineInstrBuilder BuildMI(MachineBasicBlock *BB,
 180                                    DebugLoc DL,
 181                                    const TargetInstrDesc &TID) {
 182   return BuildMI(*BB, BB->end(), DL, TID);
 183 }
 184
 185 /// BuildMI - This version of the builder inserts the newly-built
 186 /// instruction at the end of the given MachineBasicBlock, and sets up the first
 187 /// operand as a destination virtual register.
 188 ///
 189 inline MachineInstrBuilder BuildMI(MachineBasicBlock *BB,
 190                                    DebugLoc DL,
 191                                    const TargetInstrDesc &TID,
 192                                    unsigned DestReg) {
 193   return BuildMI(*BB, BB->end(), DL, TID, DestReg);
 194 }
 195
 196 inline unsigned getDefRegState(bool B) {
 197   return B ? RegState::Define : 0;
 198 }
 199 inline unsigned getImplRegState(bool B) {
 200   return B ? RegState::Implicit : 0;
 201 }
 202 inline unsigned getKillRegState(bool B) {
 203   return B ? RegState::Kill : 0;
 204 }
 205 inline unsigned getDeadRegState(bool B) {
 206   return B ? RegState::Dead : 0;
 207 }
 208 inline unsigned getUndefRegState(bool B) {
 209   return B ? RegState::Undef : 0;
 210 }
 211
 212 } // End llvm namespace
 213
 214 #endif