Magic-Mirror/llvm-capstone
1
0
Fork 0
mirror of https://github.com/capstone-engine/llvm-capstone.git synced 2024-11-23 05:40:09 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Handle multiple template arguments in handleDefaultArg

Browse Source
This commit is contained in:
Jiajie Chen 2024-05-03 17:38:58 +08:00 committed by Rot127
parent 1ac920cc21
commit 91da70b398
1 changed files with 47 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

61
llvm/utils/TableGen/PrinterCapstone.cpp
View File

@ -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);
}
}
}