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 }
53