llvm/lib/CodeGen/MachineSSAContext.cpp

   1 //===- MachineSSAContext.cpp ------------------------------------*- 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 /// \file
   9 ///
  10 /// This file defines a specialization of the GenericSSAContext<X>
  11 /// template class for Machine IR.
  12 ///
  13 //===----------------------------------------------------------------------===//
  14
  15 #include "llvm/CodeGen/MachineSSAContext.h"
  16 #include "llvm/CodeGen/MachineBasicBlock.h"
  17 #include "llvm/CodeGen/MachineInstr.h"
  18 #include "llvm/Support/raw_ostream.h"
  19
  20 using namespace llvm;
  21
  22 MachineBasicBlock *MachineSSAContext::getEntryBlock(MachineFunction &F) {
  23   return &F.front();
  24 }
  25
  26 void MachineSSAContext::setFunction(MachineFunction &Fn) {
  27   MF = &Fn;
  28   RegInfo = &MF->getRegInfo();
  29 }
  30
  31 Printable MachineSSAContext::print(MachineBasicBlock *Block) const {
  32   return Printable([Block](raw_ostream &Out) { Block->printName(Out); });
  33 }
  34
  35 Printable MachineSSAContext::print(MachineInstr *I) const {
  36   return Printable([I](raw_ostream &Out) { I->print(Out); });
  37 }
  38
  39 Printable MachineSSAContext::print(Register Value) const {
  40   auto *MRI = RegInfo;
  41   return Printable([MRI, Value](raw_ostream &Out) {
  42     Out << printReg(Value, MRI->getTargetRegisterInfo(), 0, MRI);
  43
  44     if (Value) {
  45       // Try to print the definition.
  46       if (auto *Instr = MRI->getUniqueVRegDef(Value)) {
  47         Out << ": ";
  48         Instr->print(Out);
  49       }
  50     }
  51   });
  52 }