//= NVPTXInstPrinter.h - Convert NVPTX MCInst to assembly syntax --*- C++ -*-=//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
//
// This class prints an NVPTX MCInst to .ptx file syntax.
//
//===----------------------------------------------------------------------===//

#ifndef LLVM_LIB_TARGET_NVPTX_MCTARGETDESC_NVPTXINSTPRINTER_H
#define LLVM_LIB_TARGET_NVPTX_MCTARGETDESC_NVPTXINSTPRINTER_H

#include "llvm/MC/MCInstPrinter.h"

namespace llvm {

class MCSubtargetInfo;

class NVPTXInstPrinter : public MCInstPrinter {
public:
  NVPTXInstPrinter(const MCAsmInfo &MAI, const MCInstrInfo &MII,
                   const MCRegisterInfo &MRI);

  void printRegName(raw_ostream &OS, MCRegister Reg) override;
  void printInst(const MCInst *MI, uint64_t Address, StringRef Annot,
                 const MCSubtargetInfo &STI, raw_ostream &OS) override;

  // Autogenerated by tblgen.
  std::pair<const char *, uint64_t>
  getMnemonic(const MCInst &MI) const override;
  void printInstruction(const MCInst *MI, uint64_t Address, raw_ostream &O);
  static const char *getRegisterName(MCRegister Reg);
  // End

  void printOperand(const MCInst *MI, unsigned OpNo, raw_ostream &O);
  void printCvtMode(const MCInst *MI, int OpNum, raw_ostream &O,
                    StringRef Modifier = {});
  void printCmpMode(const MCInst *MI, int OpNum, raw_ostream &O,
                    StringRef Modifier = {});
  void printAtomicCode(const MCInst *MI, int OpNum, raw_ostream &O,
                       StringRef Modifier = {});
  void printMmaCode(const MCInst *MI, int OpNum, raw_ostream &O,
                    StringRef Modifier = {});
  void printMemOperand(const MCInst *MI, int OpNum, raw_ostream &O,
                       StringRef Modifier = {});
  void printHexu32imm(const MCInst *MI, int OpNum, raw_ostream &O);
  void printProtoIdent(const MCInst *MI, int OpNum, raw_ostream &O);
  void printPrmtMode(const MCInst *MI, int OpNum, raw_ostream &O);
  void printTmaReductionMode(const MCInst *MI, int OpNum, raw_ostream &O);
  void printCTAGroup(const MCInst *MI, int OpNum, raw_ostream &O);
  void printCallOperand(const MCInst *MI, int OpNum, raw_ostream &O,
                        StringRef Modifier = {});
  void printFTZFlag(const MCInst *MI, int OpNum, raw_ostream &O);
};

}

#endif