Home | History | Annotate | Line # | Download | only in CodeGen
      1 //===- CodeGen/MachineInstrBuilder.h - Simplify creation of MIs --*- C++ -*-===//
      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 exposes a function named BuildMI, which is useful for dramatically
     10 // simplifying how MachineInstr's are created.  It allows use of code like this:
     11 //
     12 //   M = BuildMI(MBB, MI, DL, TII.get(X86::ADD8rr), Dst)
     13 //           .addReg(argVal1)
     14 //           .addReg(argVal2);
     15 //
     16 //===----------------------------------------------------------------------===//
     17 
     18 #ifndef LLVM_CODEGEN_MACHINEINSTRBUILDER_H
     19 #define LLVM_CODEGEN_MACHINEINSTRBUILDER_H
     20 
     21 #include "llvm/ADT/ArrayRef.h"
     22 #include "llvm/CodeGen/GlobalISel/Utils.h"
     23 #include "llvm/CodeGen/MachineBasicBlock.h"
     24 #include "llvm/CodeGen/MachineFunction.h"
     25 #include "llvm/CodeGen/MachineInstr.h"
     26 #include "llvm/CodeGen/MachineInstrBundle.h"
     27 #include "llvm/CodeGen/MachineOperand.h"
     28 #include "llvm/CodeGen/TargetRegisterInfo.h"
     29 #include "llvm/IR/InstrTypes.h"
     30 #include "llvm/IR/Intrinsics.h"
     31 #include "llvm/Support/ErrorHandling.h"
     32 #include <cassert>
     33 #include <cstdint>
     34 
     35 namespace llvm {
     36 
     37 class MCInstrDesc;
     38 class MDNode;
     39 
     40 namespace RegState {
     41 
     42 enum {
     43   /// Register definition.
     44   Define = 0x2,
     45   /// Not emitted register (e.g. carry, or temporary result).
     46   Implicit = 0x4,
     47   /// The last use of a register.
     48   Kill = 0x8,
     49   /// Unused definition.
     50   Dead = 0x10,
     51   /// Value of the register doesn't matter.
     52   Undef = 0x20,
     53   /// Register definition happens before uses.
     54   EarlyClobber = 0x40,
     55   /// Register 'use' is for debugging purpose.
     56   Debug = 0x80,
     57   /// Register reads a value that is defined inside the same instruction or
     58   /// bundle.
     59   InternalRead = 0x100,
     60   /// Register that may be renamed.
     61   Renamable = 0x200,
     62   DefineNoRead = Define | Undef,
     63   ImplicitDefine = Implicit | Define,
     64   ImplicitKill = Implicit | Kill
     65 };
     66 
     67 } // end namespace RegState
     68 
     69 class MachineInstrBuilder {
     70   MachineFunction *MF = nullptr;
     71   MachineInstr *MI = nullptr;
     72 
     73 public:
     74   MachineInstrBuilder() = default;
     75 
     76   /// Create a MachineInstrBuilder for manipulating an existing instruction.
     77   /// F must be the machine function that was used to allocate I.
     78   MachineInstrBuilder(MachineFunction &F, MachineInstr *I) : MF(&F), MI(I) {}
     79   MachineInstrBuilder(MachineFunction &F, MachineBasicBlock::iterator I)
     80       : MF(&F), MI(&*I) {}
     81 
     82   /// Allow automatic conversion to the machine instruction we are working on.
     83   operator MachineInstr*() const { return MI; }
     84   MachineInstr *operator->() const { return MI; }
     85   operator MachineBasicBlock::iterator() const { return MI; }
     86 
     87   /// If conversion operators fail, use this method to get the MachineInstr
     88   /// explicitly.
     89   MachineInstr *getInstr() const { return MI; }
     90 
     91   /// Get the register for the operand index.
     92   /// The operand at the index should be a register (asserted by
     93   /// MachineOperand).
     94   Register getReg(unsigned Idx) const { return MI->getOperand(Idx).getReg(); }
     95 
     96   /// Add a new virtual register operand.
     97   const MachineInstrBuilder &addReg(Register RegNo, unsigned flags = 0,
     98                                     unsigned SubReg = 0) const {
     99     assert((flags & 0x1) == 0 &&
    100            "Passing in 'true' to addReg is forbidden! Use enums instead.");
    101     MI->addOperand(*MF, MachineOperand::CreateReg(RegNo,
    102                                                flags & RegState::Define,
    103                                                flags & RegState::Implicit,
    104                                                flags & RegState::Kill,
    105                                                flags & RegState::Dead,
    106                                                flags & RegState::Undef,
    107                                                flags & RegState::EarlyClobber,
    108                                                SubReg,
    109                                                flags & RegState::Debug,
    110                                                flags & RegState::InternalRead,
    111                                                flags & RegState::Renamable));
    112     return *this;
    113   }
    114 
    115   /// Add a virtual register definition operand.
    116   const MachineInstrBuilder &addDef(Register RegNo, unsigned Flags = 0,
    117                                     unsigned SubReg = 0) const {
    118     return addReg(RegNo, Flags | RegState::Define, SubReg);
    119   }
    120 
    121   /// Add a virtual register use operand. It is an error for Flags to contain
    122   /// `RegState::Define` when calling this function.
    123   const MachineInstrBuilder &addUse(Register RegNo, unsigned Flags = 0,
    124                                     unsigned SubReg = 0) const {
    125     assert(!(Flags & RegState::Define) &&
    126            "Misleading addUse defines register, use addReg instead.");
    127     return addReg(RegNo, Flags, SubReg);
    128   }
    129 
    130   /// Add a new immediate operand.
    131   const MachineInstrBuilder &addImm(int64_t Val) const {
    132     MI->addOperand(*MF, MachineOperand::CreateImm(Val));
    133     return *this;
    134   }
    135 
    136   const MachineInstrBuilder &addCImm(const ConstantInt *Val) const {
    137     MI->addOperand(*MF, MachineOperand::CreateCImm(Val));
    138     return *this;
    139   }
    140 
    141   const MachineInstrBuilder &addFPImm(const ConstantFP *Val) const {
    142     MI->addOperand(*MF, MachineOperand::CreateFPImm(Val));
    143     return *this;
    144   }
    145 
    146   const MachineInstrBuilder &addMBB(MachineBasicBlock *MBB,
    147                                     unsigned TargetFlags = 0) const {
    148     MI->addOperand(*MF, MachineOperand::CreateMBB(MBB, TargetFlags));
    149     return *this;
    150   }
    151 
    152   const MachineInstrBuilder &addFrameIndex(int Idx) const {
    153     MI->addOperand(*MF, MachineOperand::CreateFI(Idx));
    154     return *this;
    155   }
    156 
    157   const MachineInstrBuilder &
    158   addConstantPoolIndex(unsigned Idx, int Offset = 0,
    159                        unsigned TargetFlags = 0) const {
    160     MI->addOperand(*MF, MachineOperand::CreateCPI(Idx, Offset, TargetFlags));
    161     return *this;
    162   }
    163 
    164   const MachineInstrBuilder &addTargetIndex(unsigned Idx, int64_t Offset = 0,
    165                                           unsigned TargetFlags = 0) const {
    166     MI->addOperand(*MF, MachineOperand::CreateTargetIndex(Idx, Offset,
    167                                                           TargetFlags));
    168     return *this;
    169   }
    170 
    171   const MachineInstrBuilder &addJumpTableIndex(unsigned Idx,
    172                                                unsigned TargetFlags = 0) const {
    173     MI->addOperand(*MF, MachineOperand::CreateJTI(Idx, TargetFlags));
    174     return *this;
    175   }
    176 
    177   const MachineInstrBuilder &addGlobalAddress(const GlobalValue *GV,
    178                                               int64_t Offset = 0,
    179                                               unsigned TargetFlags = 0) const {
    180     MI->addOperand(*MF, MachineOperand::CreateGA(GV, Offset, TargetFlags));
    181     return *this;
    182   }
    183 
    184   const MachineInstrBuilder &addExternalSymbol(const char *FnName,
    185                                                unsigned TargetFlags = 0) const {
    186     MI->addOperand(*MF, MachineOperand::CreateES(FnName, TargetFlags));
    187     return *this;
    188   }
    189 
    190   const MachineInstrBuilder &addBlockAddress(const BlockAddress *BA,
    191                                              int64_t Offset = 0,
    192                                              unsigned TargetFlags = 0) const {
    193     MI->addOperand(*MF, MachineOperand::CreateBA(BA, Offset, TargetFlags));
    194     return *this;
    195   }
    196 
    197   const MachineInstrBuilder &addRegMask(const uint32_t *Mask) const {
    198     MI->addOperand(*MF, MachineOperand::CreateRegMask(Mask));
    199     return *this;
    200   }
    201 
    202   const MachineInstrBuilder &addMemOperand(MachineMemOperand *MMO) const {
    203     MI->addMemOperand(*MF, MMO);
    204     return *this;
    205   }
    206 
    207   const MachineInstrBuilder &
    208   setMemRefs(ArrayRef<MachineMemOperand *> MMOs) const {
    209     MI->setMemRefs(*MF, MMOs);
    210     return *this;
    211   }
    212 
    213   const MachineInstrBuilder &cloneMemRefs(const MachineInstr &OtherMI) const {
    214     MI->cloneMemRefs(*MF, OtherMI);
    215     return *this;
    216   }
    217 
    218   const MachineInstrBuilder &
    219   cloneMergedMemRefs(ArrayRef<const MachineInstr *> OtherMIs) const {
    220     MI->cloneMergedMemRefs(*MF, OtherMIs);
    221     return *this;
    222   }
    223 
    224   const MachineInstrBuilder &add(const MachineOperand &MO) const {
    225     MI->addOperand(*MF, MO);
    226     return *this;
    227   }
    228 
    229   const MachineInstrBuilder &add(ArrayRef<MachineOperand> MOs) const {
    230     for (const MachineOperand &MO : MOs) {
    231       MI->addOperand(*MF, MO);
    232     }
    233     return *this;
    234   }
    235 
    236   const MachineInstrBuilder &addMetadata(const MDNode *MD) const {
    237     MI->addOperand(*MF, MachineOperand::CreateMetadata(MD));
    238     assert((MI->isDebugValue() ? static_cast<bool>(MI->getDebugVariable())
    239                                : true) &&
    240            "first MDNode argument of a DBG_VALUE not a variable");
    241     assert((MI->isDebugLabel() ? static_cast<bool>(MI->getDebugLabel())
    242                                : true) &&
    243            "first MDNode argument of a DBG_LABEL not a label");
    244     return *this;
    245   }
    246 
    247   const MachineInstrBuilder &addCFIIndex(unsigned CFIIndex) const {
    248     MI->addOperand(*MF, MachineOperand::CreateCFIIndex(CFIIndex));
    249     return *this;
    250   }
    251 
    252   const MachineInstrBuilder &addIntrinsicID(Intrinsic::ID ID) const {
    253     MI->addOperand(*MF, MachineOperand::CreateIntrinsicID(ID));
    254     return *this;
    255   }
    256 
    257   const MachineInstrBuilder &addPredicate(CmpInst::Predicate Pred) const {
    258     MI->addOperand(*MF, MachineOperand::CreatePredicate(Pred));
    259     return *this;
    260   }
    261 
    262   const MachineInstrBuilder &addShuffleMask(ArrayRef<int> Val) const {
    263     MI->addOperand(*MF, MachineOperand::CreateShuffleMask(Val));
    264     return *this;
    265   }
    266 
    267   const MachineInstrBuilder &addSym(MCSymbol *Sym,
    268                                     unsigned char TargetFlags = 0) const {
    269     MI->addOperand(*MF, MachineOperand::CreateMCSymbol(Sym, TargetFlags));
    270     return *this;
    271   }
    272 
    273   const MachineInstrBuilder &setMIFlags(unsigned Flags) const {
    274     MI->setFlags(Flags);
    275     return *this;
    276   }
    277 
    278   const MachineInstrBuilder &setMIFlag(MachineInstr::MIFlag Flag) const {
    279     MI->setFlag(Flag);
    280     return *this;
    281   }
    282 
    283   // Add a displacement from an existing MachineOperand with an added offset.
    284   const MachineInstrBuilder &addDisp(const MachineOperand &Disp, int64_t off,
    285                                      unsigned char TargetFlags = 0) const {
    286     // If caller specifies new TargetFlags then use it, otherwise the
    287     // default behavior is to copy the target flags from the existing
    288     // MachineOperand. This means if the caller wants to clear the
    289     // target flags it needs to do so explicitly.
    290     if (0 == TargetFlags)
    291       TargetFlags = Disp.getTargetFlags();
    292 
    293     switch (Disp.getType()) {
    294       default:
    295         llvm_unreachable("Unhandled operand type in addDisp()");
    296       case MachineOperand::MO_Immediate:
    297         return addImm(Disp.getImm() + off);
    298       case MachineOperand::MO_ConstantPoolIndex:
    299         return addConstantPoolIndex(Disp.getIndex(), Disp.getOffset() + off,
    300                                     TargetFlags);
    301       case MachineOperand::MO_GlobalAddress:
    302         return addGlobalAddress(Disp.getGlobal(), Disp.getOffset() + off,
    303                                 TargetFlags);
    304       case MachineOperand::MO_BlockAddress:
    305         return addBlockAddress(Disp.getBlockAddress(), Disp.getOffset() + off,
    306                                TargetFlags);
    307       case MachineOperand::MO_JumpTableIndex:
    308         assert(off == 0 && "cannot create offset into jump tables");
    309         return addJumpTableIndex(Disp.getIndex(), TargetFlags);
    310     }
    311   }
    312 
    313   /// Copy all the implicit operands from OtherMI onto this one.
    314   const MachineInstrBuilder &
    315   copyImplicitOps(const MachineInstr &OtherMI) const {
    316     MI->copyImplicitOps(*MF, OtherMI);
    317     return *this;
    318   }
    319 
    320   bool constrainAllUses(const TargetInstrInfo &TII,
    321                         const TargetRegisterInfo &TRI,
    322                         const RegisterBankInfo &RBI) const {
    323     return constrainSelectedInstRegOperands(*MI, TII, TRI, RBI);
    324   }
    325 };
    326 
    327 /// Builder interface. Specify how to create the initial instruction itself.
    328 inline MachineInstrBuilder BuildMI(MachineFunction &MF, const DebugLoc &DL,
    329                                    const MCInstrDesc &MCID) {
    330   return MachineInstrBuilder(MF, MF.CreateMachineInstr(MCID, DL));
    331 }
    332 
    333 /// This version of the builder sets up the first operand as a
    334 /// destination virtual register.
    335 inline MachineInstrBuilder BuildMI(MachineFunction &MF, const DebugLoc &DL,
    336                                    const MCInstrDesc &MCID, Register DestReg) {
    337   return MachineInstrBuilder(MF, MF.CreateMachineInstr(MCID, DL))
    338            .addReg(DestReg, RegState::Define);
    339 }
    340 
    341 /// This version of the builder inserts the newly-built instruction before
    342 /// the given position in the given MachineBasicBlock, and sets up the first
    343 /// operand as a destination virtual register.
    344 inline MachineInstrBuilder BuildMI(MachineBasicBlock &BB,
    345                                    MachineBasicBlock::iterator I,
    346                                    const DebugLoc &DL, const MCInstrDesc &MCID,
    347                                    Register DestReg) {
    348   MachineFunction &MF = *BB.getParent();
    349   MachineInstr *MI = MF.CreateMachineInstr(MCID, DL);
    350   BB.insert(I, MI);
    351   return MachineInstrBuilder(MF, MI).addReg(DestReg, RegState::Define);
    352 }
    353 
    354 /// This version of the builder inserts the newly-built instruction before
    355 /// the given position in the given MachineBasicBlock, and sets up the first
    356 /// operand as a destination virtual register.
    357 ///
    358 /// If \c I is inside a bundle, then the newly inserted \a MachineInstr is
    359 /// added to the same bundle.
    360 inline MachineInstrBuilder BuildMI(MachineBasicBlock &BB,
    361                                    MachineBasicBlock::instr_iterator I,
    362                                    const DebugLoc &DL, const MCInstrDesc &MCID,
    363                                    Register DestReg) {
    364   MachineFunction &MF = *BB.getParent();
    365   MachineInstr *MI = MF.CreateMachineInstr(MCID, DL);
    366   BB.insert(I, MI);
    367   return MachineInstrBuilder(MF, MI).addReg(DestReg, RegState::Define);
    368 }
    369 
    370 inline MachineInstrBuilder BuildMI(MachineBasicBlock &BB, MachineInstr &I,
    371                                    const DebugLoc &DL, const MCInstrDesc &MCID,
    372                                    Register DestReg) {
    373   // Calling the overload for instr_iterator is always correct.  However, the
    374   // definition is not available in headers, so inline the check.
    375   if (I.isInsideBundle())
    376     return BuildMI(BB, MachineBasicBlock::instr_iterator(I), DL, MCID, DestReg);
    377   return BuildMI(BB, MachineBasicBlock::iterator(I), DL, MCID, DestReg);
    378 }
    379 
    380 inline MachineInstrBuilder BuildMI(MachineBasicBlock &BB, MachineInstr *I,
    381                                    const DebugLoc &DL, const MCInstrDesc &MCID,
    382                                    Register DestReg) {
    383   return BuildMI(BB, *I, DL, MCID, DestReg);
    384 }
    385 
    386 /// This version of the builder inserts the newly-built instruction before the
    387 /// given position in the given MachineBasicBlock, and does NOT take a
    388 /// destination register.
    389 inline MachineInstrBuilder BuildMI(MachineBasicBlock &BB,
    390                                    MachineBasicBlock::iterator I,
    391                                    const DebugLoc &DL,
    392                                    const MCInstrDesc &MCID) {
    393   MachineFunction &MF = *BB.getParent();
    394   MachineInstr *MI = MF.CreateMachineInstr(MCID, DL);
    395   BB.insert(I, MI);
    396   return MachineInstrBuilder(MF, MI);
    397 }
    398 
    399 inline MachineInstrBuilder BuildMI(MachineBasicBlock &BB,
    400                                    MachineBasicBlock::instr_iterator I,
    401                                    const DebugLoc &DL,
    402                                    const MCInstrDesc &MCID) {
    403   MachineFunction &MF = *BB.getParent();
    404   MachineInstr *MI = MF.CreateMachineInstr(MCID, DL);
    405   BB.insert(I, MI);
    406   return MachineInstrBuilder(MF, MI);
    407 }
    408 
    409 inline MachineInstrBuilder BuildMI(MachineBasicBlock &BB, MachineInstr &I,
    410                                    const DebugLoc &DL,
    411                                    const MCInstrDesc &MCID) {
    412   // Calling the overload for instr_iterator is always correct.  However, the
    413   // definition is not available in headers, so inline the check.
    414   if (I.isInsideBundle())
    415     return BuildMI(BB, MachineBasicBlock::instr_iterator(I), DL, MCID);
    416   return BuildMI(BB, MachineBasicBlock::iterator(I), DL, MCID);
    417 }
    418 
    419 inline MachineInstrBuilder BuildMI(MachineBasicBlock &BB, MachineInstr *I,
    420                                    const DebugLoc &DL,
    421                                    const MCInstrDesc &MCID) {
    422   return BuildMI(BB, *I, DL, MCID);
    423 }
    424 
    425 /// This version of the builder inserts the newly-built instruction at the end
    426 /// of the given MachineBasicBlock, and does NOT take a destination register.
    427 inline MachineInstrBuilder BuildMI(MachineBasicBlock *BB, const DebugLoc &DL,
    428                                    const MCInstrDesc &MCID) {
    429   return BuildMI(*BB, BB->end(), DL, MCID);
    430 }
    431 
    432 /// This version of the builder inserts the newly-built instruction at the
    433 /// end of the given MachineBasicBlock, and sets up the first operand as a
    434 /// destination virtual register.
    435 inline MachineInstrBuilder BuildMI(MachineBasicBlock *BB, const DebugLoc &DL,
    436                                    const MCInstrDesc &MCID, Register DestReg) {
    437   return BuildMI(*BB, BB->end(), DL, MCID, DestReg);
    438 }
    439 
    440 /// This version of the builder builds a DBG_VALUE intrinsic
    441 /// for either a value in a register or a register-indirect
    442 /// address.  The convention is that a DBG_VALUE is indirect iff the
    443 /// second operand is an immediate.
    444 MachineInstrBuilder BuildMI(MachineFunction &MF, const DebugLoc &DL,
    445                             const MCInstrDesc &MCID, bool IsIndirect,
    446                             Register Reg, const MDNode *Variable,
    447                             const MDNode *Expr);
    448 
    449 /// This version of the builder builds a DBG_VALUE intrinsic
    450 /// for a MachineOperand.
    451 MachineInstrBuilder BuildMI(MachineFunction &MF, const DebugLoc &DL,
    452                             const MCInstrDesc &MCID, bool IsIndirect,
    453                             const MachineOperand &MO, const MDNode *Variable,
    454                             const MDNode *Expr);
    455 
    456 /// This version of the builder builds a DBG_VALUE or DBG_VALUE_LIST intrinsic
    457 /// for a MachineOperand.
    458 MachineInstrBuilder BuildMI(MachineFunction &MF, const DebugLoc &DL,
    459                             const MCInstrDesc &MCID, bool IsIndirect,
    460                             ArrayRef<MachineOperand> MOs,
    461                             const MDNode *Variable, const MDNode *Expr);
    462 
    463 /// This version of the builder builds a DBG_VALUE intrinsic
    464 /// for either a value in a register or a register-indirect
    465 /// address and inserts it at position I.
    466 MachineInstrBuilder BuildMI(MachineBasicBlock &BB,
    467                             MachineBasicBlock::iterator I, const DebugLoc &DL,
    468                             const MCInstrDesc &MCID, bool IsIndirect,
    469                             Register Reg, const MDNode *Variable,
    470                             const MDNode *Expr);
    471 
    472 /// This version of the builder builds a DBG_VALUE intrinsic
    473 /// for a machine operand and inserts it at position I.
    474 MachineInstrBuilder BuildMI(MachineBasicBlock &BB,
    475                             MachineBasicBlock::iterator I, const DebugLoc &DL,
    476                             const MCInstrDesc &MCID, bool IsIndirect,
    477                             MachineOperand &MO, const MDNode *Variable,
    478                             const MDNode *Expr);
    479 
    480 /// This version of the builder builds a DBG_VALUE or DBG_VALUE_LIST intrinsic
    481 /// for a machine operand and inserts it at position I.
    482 MachineInstrBuilder BuildMI(MachineBasicBlock &BB,
    483                             MachineBasicBlock::iterator I, const DebugLoc &DL,
    484                             const MCInstrDesc &MCID, bool IsIndirect,
    485                             ArrayRef<MachineOperand> MOs,
    486                             const MDNode *Variable, const MDNode *Expr);
    487 
    488 /// Clone a DBG_VALUE whose value has been spilled to FrameIndex.
    489 MachineInstr *buildDbgValueForSpill(MachineBasicBlock &BB,
    490                                     MachineBasicBlock::iterator I,
    491                                     const MachineInstr &Orig, int FrameIndex,
    492                                     Register SpillReg);
    493 MachineInstr *
    494 buildDbgValueForSpill(MachineBasicBlock &BB, MachineBasicBlock::iterator I,
    495                       const MachineInstr &Orig, int FrameIndex,
    496                       SmallVectorImpl<const MachineOperand *> &SpilledOperands);
    497 
    498 /// Update a DBG_VALUE whose value has been spilled to FrameIndex. Useful when
    499 /// modifying an instruction in place while iterating over a basic block.
    500 void updateDbgValueForSpill(MachineInstr &Orig, int FrameIndex, Register Reg);
    501 
    502 inline unsigned getDefRegState(bool B) {
    503   return B ? RegState::Define : 0;
    504 }
    505 inline unsigned getImplRegState(bool B) {
    506   return B ? RegState::Implicit : 0;
    507 }
    508 inline unsigned getKillRegState(bool B) {
    509   return B ? RegState::Kill : 0;
    510 }
    511 inline unsigned getDeadRegState(bool B) {
    512   return B ? RegState::Dead : 0;
    513 }
    514 inline unsigned getUndefRegState(bool B) {
    515   return B ? RegState::Undef : 0;
    516 }
    517 inline unsigned getInternalReadRegState(bool B) {
    518   return B ? RegState::InternalRead : 0;
    519 }
    520 inline unsigned getDebugRegState(bool B) {
    521   return B ? RegState::Debug : 0;
    522 }
    523 inline unsigned getRenamableRegState(bool B) {
    524   return B ? RegState::Renamable : 0;
    525 }
    526 
    527 /// Get all register state flags from machine operand \p RegOp.
    528 inline unsigned getRegState(const MachineOperand &RegOp) {
    529   assert(RegOp.isReg() && "Not a register operand");
    530   return getDefRegState(RegOp.isDef()) | getImplRegState(RegOp.isImplicit()) |
    531          getKillRegState(RegOp.isKill()) | getDeadRegState(RegOp.isDead()) |
    532          getUndefRegState(RegOp.isUndef()) |
    533          getInternalReadRegState(RegOp.isInternalRead()) |
    534          getDebugRegState(RegOp.isDebug()) |
    535          getRenamableRegState(Register::isPhysicalRegister(RegOp.getReg()) &&
    536                               RegOp.isRenamable());
    537 }
    538 
    539 /// Helper class for constructing bundles of MachineInstrs.
    540 ///
    541 /// MIBundleBuilder can create a bundle from scratch by inserting new
    542 /// MachineInstrs one at a time, or it can create a bundle from a sequence of
    543 /// existing MachineInstrs in a basic block.
    544 class MIBundleBuilder {
    545   MachineBasicBlock &MBB;
    546   MachineBasicBlock::instr_iterator Begin;
    547   MachineBasicBlock::instr_iterator End;
    548 
    549 public:
    550   /// Create an MIBundleBuilder that inserts instructions into a new bundle in
    551   /// BB above the bundle or instruction at Pos.
    552   MIBundleBuilder(MachineBasicBlock &BB, MachineBasicBlock::iterator Pos)
    553       : MBB(BB), Begin(Pos.getInstrIterator()), End(Begin) {}
    554 
    555   /// Create a bundle from the sequence of instructions between B and E.
    556   MIBundleBuilder(MachineBasicBlock &BB, MachineBasicBlock::iterator B,
    557                   MachineBasicBlock::iterator E)
    558       : MBB(BB), Begin(B.getInstrIterator()), End(E.getInstrIterator()) {
    559     assert(B != E && "No instructions to bundle");
    560     ++B;
    561     while (B != E) {
    562       MachineInstr &MI = *B;
    563       ++B;
    564       MI.bundleWithPred();
    565     }
    566   }
    567 
    568   /// Create an MIBundleBuilder representing an existing instruction or bundle
    569   /// that has MI as its head.
    570   explicit MIBundleBuilder(MachineInstr *MI)
    571       : MBB(*MI->getParent()), Begin(MI),
    572         End(getBundleEnd(MI->getIterator())) {}
    573 
    574   /// Return a reference to the basic block containing this bundle.
    575   MachineBasicBlock &getMBB() const { return MBB; }
    576 
    577   /// Return true if no instructions have been inserted in this bundle yet.
    578   /// Empty bundles aren't representable in a MachineBasicBlock.
    579   bool empty() const { return Begin == End; }
    580 
    581   /// Return an iterator to the first bundled instruction.
    582   MachineBasicBlock::instr_iterator begin() const { return Begin; }
    583 
    584   /// Return an iterator beyond the last bundled instruction.
    585   MachineBasicBlock::instr_iterator end() const { return End; }
    586 
    587   /// Insert MI into this bundle before I which must point to an instruction in
    588   /// the bundle, or end().
    589   MIBundleBuilder &insert(MachineBasicBlock::instr_iterator I,
    590                           MachineInstr *MI) {
    591     MBB.insert(I, MI);
    592     if (I == Begin) {
    593       if (!empty())
    594         MI->bundleWithSucc();
    595       Begin = MI->getIterator();
    596       return *this;
    597     }
    598     if (I == End) {
    599       MI->bundleWithPred();
    600       return *this;
    601     }
    602     // MI was inserted in the middle of the bundle, so its neighbors' flags are
    603     // already fine. Update MI's bundle flags manually.
    604     MI->setFlag(MachineInstr::BundledPred);
    605     MI->setFlag(MachineInstr::BundledSucc);
    606     return *this;
    607   }
    608 
    609   /// Insert MI into MBB by prepending it to the instructions in the bundle.
    610   /// MI will become the first instruction in the bundle.
    611   MIBundleBuilder &prepend(MachineInstr *MI) {
    612     return insert(begin(), MI);
    613   }
    614 
    615   /// Insert MI into MBB by appending it to the instructions in the bundle.
    616   /// MI will become the last instruction in the bundle.
    617   MIBundleBuilder &append(MachineInstr *MI) {
    618     return insert(end(), MI);
    619   }
    620 };
    621 
    622 } // end namespace llvm
    623 
    624 #endif // LLVM_CODEGEN_MACHINEINSTRBUILDER_H
    625