From 91da70b398c221883806395d451d84909bbb1357 Mon Sep 17 00:00:00 2001
From: Jiajie Chen <c@jia.je>
Date: Fri, 3 May 2024 17:38:58 +0800
Subject: [PATCH] Handle multiple template arguments in handleDefaultArg

---
 llvm/utils/TableGen/PrinterCapstone.cpp | 61 +++++++++++++++++++------
 1 file changed, 47 insertions(+), 14 deletions(-)

diff --git a/llvm/utils/TableGen/PrinterCapstone.cpp b/llvm/utils/TableGen/PrinterCapstone.cpp
index aca29b0fe4a1..c969be1a16d4 100644
--- a/llvm/utils/TableGen/PrinterCapstone.cpp
+++ b/llvm/utils/TableGen/PrinterCapstone.cpp
@@ -669,33 +669,66 @@ void patchIsGetImmReg(std::string &Code) {
 static std::string handleDefaultArg(const std::string &TargetName,
                                     std::string &Code) {
   // Default values of the template function arguments.
-  static SmallVector<std::pair<std::string, std::string>>
+  // Tuple is (function name, default argument, number of argumetns)
+  static SmallVector<std::tuple<std::string, std::string, int>>
       AArch64TemplFuncWithDefaults = {// Default is 1
-                                      {"printVectorIndex", "1"},
+                                      {"printVectorIndex", "1", 1},
                                       // Default is false == 0
-                                      {"printPrefetchOp", "0"},
+                                      {"printPrefetchOp", "0", 1},
                                       // Default is 0
-                                      {"printSVERegOp", "0"},
-                                      {"printMatrixIndex", "1"}
+                                      {"printSVERegOp", "0", 1},
+                                      {"printMatrixIndex", "1", 1}
                                       };
-  SmallVector<std::pair<std::string, std::string>> *TemplFuncWithDefaults;
+  static SmallVector<std::tuple<std::string, std::string, int>>
+      LoongArchTemplFuncWithDefaults = {// Default is 0
+                                      {"decodeSImmOperand", "0", 2},
+                                      {"decodeUImmOperand", "0", 2},
+                                      };
+  SmallVector<std::tuple<std::string, std::string, int>> *TemplFuncWithDefaults;
   if (TargetName == "AArch64")
     TemplFuncWithDefaults = &AArch64TemplFuncWithDefaults;
+  else if (TargetName == "LoongArch")
+    TemplFuncWithDefaults = &LoongArchTemplFuncWithDefaults;
   else
     return Code;
 
-  for (std::pair Func : *TemplFuncWithDefaults) {
-    // Search for function name without "<>" and replace it with <name>_<default-arg>
-    // Note: There can be multiple calls to such a function in the code.
-    // We only replace one here.
-    auto func = Func.first;
+  for (std::tuple Func : *TemplFuncWithDefaults) {
+    // Search for function where default argument is not passed
+    // e.g. printVectorIndex -> printVectorIndex_1
+    // e.g. decodeSImmOperand<1> -> decodeSImmOperand_1_0
+    auto Name = std::get<0>(Func);
+    auto DefaultArg = std::get<1>(Func);
+    auto ExpectedArgCount = std::get<2>(Func);
     SmallVector<StringRef> Matches;
-    while (Regex(func + "(<>)?($|\\()").match(Code, &Matches)) {
+    while (Regex(Name + "(<[0-9a-zA-Z,]*>)?($|\\()").match(Code, &Matches)) {
+      StringRef Arg = Matches[1];
+      // Count the number of passed arguments
+      int ActualArgCount = 0;
+      if (!Arg.empty() && Arg != "<>") {
+        ActualArgCount = Arg.count(',') + 1;
+      }
+
+      std::string NewArg;
+      NewArg = Regex("<").sub("", Arg);
+      NewArg = Regex(">").sub("", NewArg);
+      NewArg = Regex(",").sub("_", NewArg);
+      if (ActualArgCount != ExpectedArgCount) {
+        // Add default argument
+        if (Arg.empty()) {
+          // e.g. printVectorIndex -> printVectorIndex_1
+          NewArg += DefaultArg;
+        } else {
+          // e.g. decodeSImmOperand<1> -> decodeSimmOperand_1_0
+          NewArg += "_";
+          NewArg += DefaultArg;
+        }
+      }
+
       StringRef Match = Matches[0];
       if (Match.ends_with("(")) {
-        Code = Regex(func + "(<>)?\\(").sub(func + "_" + Func.second + "(", Code);
+        Code = Regex(Name + "(<[0-9a-zA-Z,]*>)?\\(").sub(Name + "_" + NewArg + "(", Code);
       } else {
-        Code = Regex(func + "(<>)?$").sub(func + "_" + Func.second, Code);
+        Code = Regex(Name + "(<[0-9a-zA-Z,]*>)?$").sub(Name + "_" + NewArg, Code);
       }
     }
   }