Emit formats enum and supplemental info for LoongArch

Author: jiegec

llvm/utils/TableGen/PrinterCapstone.cpp

@@ -2662,13 +2662,59 @@ std::string getArchSupplInfoPPC(StringRef const &TargetName,
  return "{{ 0 }}";
 }
 
+std::string getArchSupplInfoLoongArch(StringRef const &TargetName,
+ CodeGenInstruction const *CGI,
+ raw_string_ostream &LoongArchFormatEnum) {
+ static std::set<std::string> Formats;
+ // Get instruction format
+ ArrayRef<std::pair<Record *, SMRange>> SCs = CGI->TheDef->getSuperClasses();
+ if (SCs.empty()) {
+ llvm_unreachable("A CGI without superclass should not exist.");
+ }
+
+ // Compute memory access type
+ std::string MemoryAccess;
+ if (CGI->mayLoad && CGI->mayStore) {
+ MemoryAccess = "CS_AC_READ_WRTE";
+ } else if (CGI->mayLoad && !CGI->mayStore) {
+ MemoryAccess = "CS_AC_READ";
+ } else if (!CGI->mayLoad && CGI->mayStore) {
+ MemoryAccess = "CS_AC_WRITE";
+ } else {
+ MemoryAccess = "CS_AC_INVALID";
+ }
+
+ // Get base instruction format class "LAInst"
+ const Record *PrevSC = nullptr;
+ // Superclasses are in post-order. So we go through them backwards.
+ // The class before the "LAInst" class is the format class.
+ for (int I = SCs.size() - 1; I >= 0; --I) {
+ const Record *SC = SCs[I].first;
+ if (SC->getName() == "LAInst") {
+ if (!PrevSC)
+ llvm_unreachable("I class has no predecessor.");
+ std::string Format = "LoongArch_INSN_FORM_" + PrevSC->getName().upper();
+ if (Formats.find(Format) == Formats.end()) {
+ LoongArchFormatEnum << Format + ",\n";
+ }
+ Formats.emplace(Format);
+ return "{ .loongarch = { " + Format + ", " + MemoryAccess + " }}";
+ }
+ PrevSC = SC;
+ }
+ // Pseudo instructions
+ return "{ .loongarch = { 0, " + MemoryAccess + " }}";
+}
+
 std::string getArchSupplInfo(StringRef const &TargetName,
  CodeGenInstruction const *CGI,
- raw_string_ostream &PPCFormatEnum) {
+ raw_string_ostream &FormatEnum) {
  if (TargetName == "PPC")
- return getArchSupplInfoPPC(TargetName, CGI, PPCFormatEnum);
+ return getArchSupplInfoPPC(TargetName, CGI, FormatEnum);
  else if (TargetName == "AArch64") {
  return getArchSupplInfoAArch64(CGI);
+ } else if (TargetName == "LoongArch") {
+ return getArchSupplInfoLoongArch(TargetName, CGI, FormatEnum);
  }
  return "{{ 0 }}";
 }
@@ -2936,7 +2982,7 @@ void printInsnMapEntry(StringRef const &TargetName, AsmMatcherInfo &AMI,
  std::unique_ptr<MatchableInfo> const &MI, bool UseMI,
  CodeGenInstruction const *CGI,
  raw_string_ostream &InsnMap, unsigned InsnNum,
- raw_string_ostream &PPCFormatEnum) {
+ raw_string_ostream &FormatEnum) {
  InsnMap << "{\n";
  InsnMap.indent(2) << "/* "
  << (CGI->AsmString != "" ? CGI->AsmString
@@ -2955,7 +3001,7 @@ void printInsnMapEntry(StringRef const &TargetName, AsmMatcherInfo &AMI,
  InsnMap << getReqFeatures(TargetName, AMI, MI, UseMI, CGI) << ", ";
  InsnMap << ((CGI->isBranch || CGI->isReturn) ? "1" : "0") << ", ";
  InsnMap << (CGI->isIndirectBranch ? "1" : "0") << ", ";
- InsnMap << getArchSupplInfo(TargetName, CGI, PPCFormatEnum) << "\n";
+ InsnMap << getArchSupplInfo(TargetName, CGI, FormatEnum) << "\n";
  } else {
  InsnMap.indent(4) << "{ 0 }, { 0 }, { 0 }, 0, 0, {{ 0 }}";
  }
@@ -3390,7 +3436,7 @@ void PrinterCapstone::asmMatcherEmitMatchTable(CodeGenTarget const &Target,
  std::string FeatureEnumStr;
  std::string FeatureNameArrayStr;
  std::string OpGroupStr;
- std::string PPCFormatEnumStr;
+ std::string FormatEnumStr;
  std::string AliasEnumStr;
  std::string AliasMnemMapStr;
  raw_string_ostream InsnMap(InsnMapStr);
@@ -3400,7 +3446,7 @@ void PrinterCapstone::asmMatcherEmitMatchTable(CodeGenTarget const &Target,
  raw_string_ostream FeatureEnum(FeatureEnumStr);
  raw_string_ostream FeatureNameArray(FeatureNameArrayStr);
  raw_string_ostream OpGroups(OpGroupStr);
- raw_string_ostream PPCFormatEnum(PPCFormatEnumStr);
+ raw_string_ostream FormatEnum(FormatEnumStr);
  raw_string_ostream AliasEnum(AliasEnumStr);
  raw_string_ostream AliasMnemMap(AliasMnemMapStr);
  emitDefaultSourceFileHeader(InsnMap);
@@ -3410,7 +3456,7 @@ void PrinterCapstone::asmMatcherEmitMatchTable(CodeGenTarget const &Target,
  emitDefaultSourceFileHeader(FeatureEnum);
  emitDefaultSourceFileHeader(FeatureNameArray);
  emitDefaultSourceFileHeader(OpGroups);
- emitDefaultSourceFileHeader(PPCFormatEnum);
+ emitDefaultSourceFileHeader(FormatEnum);
  emitDefaultSourceFileHeader(AliasEnum);
  emitDefaultSourceFileHeader(AliasMnemMap);
 
@@ -3460,7 +3506,7 @@ void PrinterCapstone::asmMatcherEmitMatchTable(CodeGenTarget const &Target,
  printInsnOpMapEntry(Target, MI, UseMI, CGI, InsnOpMap, InsnNum,
  InsnPatternMap);
  printInsnMapEntry(Target.getName(), Info, MI, UseMI, CGI, InsnMap, InsnNum,
- PPCFormatEnum);
+ FormatEnum);
 
  ++InsnNum;
  }
@@ -3485,9 +3531,9 @@ void PrinterCapstone::asmMatcherEmitMatchTable(CodeGenTarget const &Target,
  writeFile(InsnMapFilename, AliasEnumStr);
  InsnMapFilename = TName + "GenCSAliasMnemMap.inc";
  writeFile(InsnMapFilename, AliasMnemMapStr);
- if (TName == "PPC") {
- InsnMapFilename = "PPCGenCSInsnFormatsEnum.inc";
- writeFile(InsnMapFilename, PPCFormatEnumStr);
+ if (TName == "PPC" || TName == "LoongArch") {
+ InsnMapFilename = TName + "GenCSInsnFormatsEnum.inc";
+ writeFile(InsnMapFilename, FormatEnumStr);
  }
 }