12cab237bSDimitry Andric //===-- RISCVInstPrinter.cpp - Convert RISCV MCInst to asm syntax ---------===//
22cab237bSDimitry Andric //
32cab237bSDimitry Andric //                     The LLVM Compiler Infrastructure
42cab237bSDimitry Andric //
52cab237bSDimitry Andric // This file is distributed under the University of Illinois Open Source
62cab237bSDimitry Andric // License. See LICENSE.TXT for details.
72cab237bSDimitry Andric //
82cab237bSDimitry Andric //===----------------------------------------------------------------------===//
92cab237bSDimitry Andric //
102cab237bSDimitry Andric // This class prints an RISCV MCInst to a .s file.
112cab237bSDimitry Andric //
122cab237bSDimitry Andric //===----------------------------------------------------------------------===//
132cab237bSDimitry Andric 
142cab237bSDimitry Andric #include "RISCVInstPrinter.h"
154ba319b5SDimitry Andric #include "MCTargetDesc/RISCVMCExpr.h"
16*b5893f02SDimitry Andric #include "Utils/RISCVBaseInfo.h"
172cab237bSDimitry Andric #include "llvm/MC/MCAsmInfo.h"
182cab237bSDimitry Andric #include "llvm/MC/MCExpr.h"
192cab237bSDimitry Andric #include "llvm/MC/MCInst.h"
202cab237bSDimitry Andric #include "llvm/MC/MCRegisterInfo.h"
214ba319b5SDimitry Andric #include "llvm/MC/MCSubtargetInfo.h"
222cab237bSDimitry Andric #include "llvm/MC/MCSymbol.h"
232cab237bSDimitry Andric #include "llvm/Support/CommandLine.h"
242cab237bSDimitry Andric #include "llvm/Support/ErrorHandling.h"
252cab237bSDimitry Andric #include "llvm/Support/FormattedStream.h"
262cab237bSDimitry Andric using namespace llvm;
272cab237bSDimitry Andric 
282cab237bSDimitry Andric #define DEBUG_TYPE "asm-printer"
292cab237bSDimitry Andric 
302cab237bSDimitry Andric // Include the auto-generated portion of the assembly writer.
312cab237bSDimitry Andric #define PRINT_ALIAS_INSTR
322cab237bSDimitry Andric #include "RISCVGenAsmWriter.inc"
332cab237bSDimitry Andric 
344ba319b5SDimitry Andric // Include the auto-generated portion of the compress emitter.
354ba319b5SDimitry Andric #define GEN_UNCOMPRESS_INSTR
364ba319b5SDimitry Andric #include "RISCVGenCompressInstEmitter.inc"
374ba319b5SDimitry Andric 
382cab237bSDimitry Andric static cl::opt<bool>
392cab237bSDimitry Andric     NoAliases("riscv-no-aliases",
402cab237bSDimitry Andric               cl::desc("Disable the emission of assembler pseudo instructions"),
41*b5893f02SDimitry Andric               cl::init(false), cl::Hidden);
422cab237bSDimitry Andric 
printInst(const MCInst * MI,raw_ostream & O,StringRef Annot,const MCSubtargetInfo & STI)432cab237bSDimitry Andric void RISCVInstPrinter::printInst(const MCInst *MI, raw_ostream &O,
442cab237bSDimitry Andric                                  StringRef Annot, const MCSubtargetInfo &STI) {
454ba319b5SDimitry Andric   bool Res = false;
464ba319b5SDimitry Andric   const MCInst *NewMI = MI;
474ba319b5SDimitry Andric   MCInst UncompressedMI;
484ba319b5SDimitry Andric   if (!NoAliases)
494ba319b5SDimitry Andric     Res = uncompressInst(UncompressedMI, *MI, MRI, STI);
504ba319b5SDimitry Andric   if (Res)
514ba319b5SDimitry Andric     NewMI = const_cast<MCInst *>(&UncompressedMI);
524ba319b5SDimitry Andric   if (NoAliases || !printAliasInstr(NewMI, STI, O))
534ba319b5SDimitry Andric     printInstruction(NewMI, STI, O);
542cab237bSDimitry Andric   printAnnotation(O, Annot);
552cab237bSDimitry Andric }
562cab237bSDimitry Andric 
printRegName(raw_ostream & O,unsigned RegNo) const572cab237bSDimitry Andric void RISCVInstPrinter::printRegName(raw_ostream &O, unsigned RegNo) const {
582cab237bSDimitry Andric   O << getRegisterName(RegNo);
592cab237bSDimitry Andric }
602cab237bSDimitry Andric 
printOperand(const MCInst * MI,unsigned OpNo,const MCSubtargetInfo & STI,raw_ostream & O,const char * Modifier)612cab237bSDimitry Andric void RISCVInstPrinter::printOperand(const MCInst *MI, unsigned OpNo,
62*b5893f02SDimitry Andric                                     const MCSubtargetInfo &STI, raw_ostream &O,
63*b5893f02SDimitry Andric                                     const char *Modifier) {
642cab237bSDimitry Andric   assert((Modifier == 0 || Modifier[0] == 0) && "No modifiers supported");
652cab237bSDimitry Andric   const MCOperand &MO = MI->getOperand(OpNo);
662cab237bSDimitry Andric 
672cab237bSDimitry Andric   if (MO.isReg()) {
682cab237bSDimitry Andric     printRegName(O, MO.getReg());
692cab237bSDimitry Andric     return;
702cab237bSDimitry Andric   }
712cab237bSDimitry Andric 
722cab237bSDimitry Andric   if (MO.isImm()) {
732cab237bSDimitry Andric     O << MO.getImm();
742cab237bSDimitry Andric     return;
752cab237bSDimitry Andric   }
762cab237bSDimitry Andric 
772cab237bSDimitry Andric   assert(MO.isExpr() && "Unknown operand kind in printOperand");
782cab237bSDimitry Andric   MO.getExpr()->print(O, &MAI);
792cab237bSDimitry Andric }
802cab237bSDimitry Andric 
printCSRSystemRegister(const MCInst * MI,unsigned OpNo,const MCSubtargetInfo & STI,raw_ostream & O)81*b5893f02SDimitry Andric void RISCVInstPrinter::printCSRSystemRegister(const MCInst *MI, unsigned OpNo,
82*b5893f02SDimitry Andric                                               const MCSubtargetInfo &STI,
83*b5893f02SDimitry Andric                                               raw_ostream &O) {
84*b5893f02SDimitry Andric   unsigned Imm = MI->getOperand(OpNo).getImm();
85*b5893f02SDimitry Andric   auto SysReg = RISCVSysReg::lookupSysRegByEncoding(Imm);
86*b5893f02SDimitry Andric   if (SysReg && SysReg->haveRequiredFeatures(STI.getFeatureBits()))
87*b5893f02SDimitry Andric     O << SysReg->Name;
88*b5893f02SDimitry Andric   else
89*b5893f02SDimitry Andric     O << Imm;
90*b5893f02SDimitry Andric }
91*b5893f02SDimitry Andric 
printFenceArg(const MCInst * MI,unsigned OpNo,const MCSubtargetInfo & STI,raw_ostream & O)922cab237bSDimitry Andric void RISCVInstPrinter::printFenceArg(const MCInst *MI, unsigned OpNo,
934ba319b5SDimitry Andric                                      const MCSubtargetInfo &STI,
942cab237bSDimitry Andric                                      raw_ostream &O) {
952cab237bSDimitry Andric   unsigned FenceArg = MI->getOperand(OpNo).getImm();
96*b5893f02SDimitry Andric   assert (((FenceArg >> 4) == 0) && "Invalid immediate in printFenceArg");
97*b5893f02SDimitry Andric 
982cab237bSDimitry Andric   if ((FenceArg & RISCVFenceField::I) != 0)
992cab237bSDimitry Andric     O << 'i';
1002cab237bSDimitry Andric   if ((FenceArg & RISCVFenceField::O) != 0)
1012cab237bSDimitry Andric     O << 'o';
1022cab237bSDimitry Andric   if ((FenceArg & RISCVFenceField::R) != 0)
1032cab237bSDimitry Andric     O << 'r';
1042cab237bSDimitry Andric   if ((FenceArg & RISCVFenceField::W) != 0)
1052cab237bSDimitry Andric     O << 'w';
106*b5893f02SDimitry Andric   if (FenceArg == 0)
107*b5893f02SDimitry Andric     O << "unknown";
1082cab237bSDimitry Andric }
1092cab237bSDimitry Andric 
printFRMArg(const MCInst * MI,unsigned OpNo,const MCSubtargetInfo & STI,raw_ostream & O)1102cab237bSDimitry Andric void RISCVInstPrinter::printFRMArg(const MCInst *MI, unsigned OpNo,
111*b5893f02SDimitry Andric                                    const MCSubtargetInfo &STI, raw_ostream &O) {
1122cab237bSDimitry Andric   auto FRMArg =
1132cab237bSDimitry Andric       static_cast<RISCVFPRndMode::RoundingMode>(MI->getOperand(OpNo).getImm());
1142cab237bSDimitry Andric   O << RISCVFPRndMode::roundingModeToString(FRMArg);
1152cab237bSDimitry Andric }
116