1 //===-- SPIRVInstPrinter.h - Output SPIR-V MCInsts as ASM -------*- 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 class prints a SPIR-V MCInst to a .s file.
10 //
11 //===----------------------------------------------------------------------===//
12 
13 #ifndef LLVM_LIB_TARGET_SPIRV_INSTPRINTER_SPIRVINSTPRINTER_H
14 #define LLVM_LIB_TARGET_SPIRV_INSTPRINTER_SPIRVINSTPRINTER_H
15 
16 #include "MCTargetDesc/SPIRVBaseInfo.h"
17 #include "llvm/ADT/DenseSet.h"
18 #include "llvm/MC/MCInstPrinter.h"
19 
20 namespace llvm {
21 class SPIRVInstPrinter : public MCInstPrinter {
22 private:
23   SmallDenseMap<unsigned, SPIRV::InstructionSet::InstructionSet> ExtInstSetIDs;
24   void recordOpExtInstImport(const MCInst *MI);
25 
26 public:
27   using MCInstPrinter::MCInstPrinter;
28 
29   void printInst(const MCInst *MI, uint64_t Address, StringRef Annot,
30                  const MCSubtargetInfo &STI, raw_ostream &OS) override;
31   void printOperand(const MCInst *MI, unsigned OpNo, raw_ostream &O,
32                     const char *Modifier = nullptr);
33 
34   void printStringImm(const MCInst *MI, unsigned OpNo, raw_ostream &O);
35 
36   void printOpDecorate(const MCInst *MI, raw_ostream &O);
37   void printOpExtInst(const MCInst *MI, raw_ostream &O);
38   void printRemainingVariableOps(const MCInst *MI, unsigned StartIndex,
39                                  raw_ostream &O, bool SkipFirstSpace = false,
40                                  bool SkipImmediates = false);
41   void printOpConstantVarOps(const MCInst *MI, unsigned StartIndex,
42                              raw_ostream &O);
43 
44   void printExtension(const MCInst *MI, unsigned OpNo, raw_ostream &O);
45   template <SPIRV::OperandCategory::OperandCategory category>
46   void printSymbolicOperand(const MCInst *MI, unsigned OpNo, raw_ostream &O);
47 
48   // Autogenerated by tblgen.
49   std::pair<const char *, uint64_t> getMnemonic(const MCInst *MI) override;
50   void printInstruction(const MCInst *MI, uint64_t Address, raw_ostream &O);
51   static const char *getRegisterName(MCRegister Reg);
52 };
53 } // namespace llvm
54 
55 #endif // LLVM_LIB_TARGET_SPIRV_INSTPRINTER_SPIRVINSTPRINTER_H
56