1 //===- Attributes.cpp - Generate attributes -------------------------------===// 2 // 3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 4 // See https://llvm.org/LICENSE.txt for license information. 5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 6 // 7 //===----------------------------------------------------------------------===// 8 9 #include "llvm/Support/MemoryBuffer.h" 10 #include "llvm/TableGen/Record.h" 11 #include <algorithm> 12 #include <string> 13 #include <vector> 14 using namespace llvm; 15 16 #define DEBUG_TYPE "attr-enum" 17 18 namespace { 19 20 class Attributes { 21 public: 22 Attributes(RecordKeeper &R) : Records(R) {} 23 void emit(raw_ostream &OS); 24 25 private: 26 void emitTargetIndependentEnums(raw_ostream &OS); 27 void emitConversionFn(raw_ostream &OS); 28 void emitFnAttrCompatCheck(raw_ostream &OS, bool IsStringAttr); 29 30 void printEnumAttrClasses(raw_ostream &OS, 31 const std::vector<Record *> &Records); 32 void printStrBoolAttrClasses(raw_ostream &OS, 33 const std::vector<Record *> &Records); 34 35 RecordKeeper &Records; 36 }; 37 38 } // End anonymous namespace. 39 40 void Attributes::emitTargetIndependentEnums(raw_ostream &OS) { 41 OS << "#ifdef GET_ATTR_ENUM\n"; 42 OS << "#undef GET_ATTR_ENUM\n"; 43 44 std::vector<Record*> Attrs = 45 Records.getAllDerivedDefinitions("EnumAttr"); 46 47 for (auto A : Attrs) 48 OS << A->getName() << ",\n"; 49 50 OS << "#endif\n"; 51 } 52 53 void Attributes::emitConversionFn(raw_ostream &OS) { 54 OS << "#ifdef GET_ATTR_KIND_FROM_NAME\n"; 55 OS << "#undef GET_ATTR_KIND_FROM_NAME\n"; 56 57 std::vector<Record*> Attrs = 58 Records.getAllDerivedDefinitions("EnumAttr"); 59 60 OS << "static Attribute::AttrKind getAttrKindFromName(StringRef AttrName) {\n"; 61 OS << " return StringSwitch<Attribute::AttrKind>(AttrName)\n"; 62 63 for (auto A : Attrs) { 64 OS << " .Case(\"" << A->getValueAsString("AttrString"); 65 OS << "\", Attribute::" << A->getName() << ")\n"; 66 } 67 68 OS << " .Default(Attribute::None);\n"; 69 OS << "}\n\n"; 70 71 OS << "#endif\n"; 72 } 73 74 void Attributes::emitFnAttrCompatCheck(raw_ostream &OS, bool IsStringAttr) { 75 OS << "#ifdef GET_ATTR_COMPAT_FUNC\n"; 76 OS << "#undef GET_ATTR_COMPAT_FUNC\n"; 77 78 OS << "struct EnumAttr {\n"; 79 OS << " static bool isSet(const Function &Fn,\n"; 80 OS << " Attribute::AttrKind Kind) {\n"; 81 OS << " return Fn.hasFnAttribute(Kind);\n"; 82 OS << " }\n\n"; 83 OS << " static void set(Function &Fn,\n"; 84 OS << " Attribute::AttrKind Kind, bool Val) {\n"; 85 OS << " if (Val)\n"; 86 OS << " Fn.addFnAttr(Kind);\n"; 87 OS << " else\n"; 88 OS << " Fn.removeFnAttr(Kind);\n"; 89 OS << " }\n"; 90 OS << "};\n\n"; 91 92 OS << "struct StrBoolAttr {\n"; 93 OS << " static bool isSet(const Function &Fn,\n"; 94 OS << " StringRef Kind) {\n"; 95 OS << " auto A = Fn.getFnAttribute(Kind);\n"; 96 OS << " return A.getValueAsString().equals(\"true\");\n"; 97 OS << " }\n\n"; 98 OS << " static void set(Function &Fn,\n"; 99 OS << " StringRef Kind, bool Val) {\n"; 100 OS << " Fn.addFnAttr(Kind, Val ? \"true\" : \"false\");\n"; 101 OS << " }\n"; 102 OS << "};\n\n"; 103 104 printEnumAttrClasses(OS ,Records.getAllDerivedDefinitions("EnumAttr")); 105 printStrBoolAttrClasses(OS , Records.getAllDerivedDefinitions("StrBoolAttr")); 106 107 OS << "static inline bool hasCompatibleFnAttrs(const Function &Caller,\n" 108 << " const Function &Callee) {\n"; 109 OS << " bool Ret = true;\n\n"; 110 111 std::vector<Record *> CompatRules = 112 Records.getAllDerivedDefinitions("CompatRule"); 113 114 for (auto *Rule : CompatRules) { 115 StringRef FuncName = Rule->getValueAsString("CompatFunc"); 116 OS << " Ret &= " << FuncName << "(Caller, Callee);\n"; 117 } 118 119 OS << "\n"; 120 OS << " return Ret;\n"; 121 OS << "}\n\n"; 122 123 std::vector<Record *> MergeRules = 124 Records.getAllDerivedDefinitions("MergeRule"); 125 OS << "static inline void mergeFnAttrs(Function &Caller,\n" 126 << " const Function &Callee) {\n"; 127 128 for (auto *Rule : MergeRules) { 129 StringRef FuncName = Rule->getValueAsString("MergeFunc"); 130 OS << " " << FuncName << "(Caller, Callee);\n"; 131 } 132 133 OS << "}\n\n"; 134 135 OS << "#endif\n"; 136 } 137 138 void Attributes::printEnumAttrClasses(raw_ostream &OS, 139 const std::vector<Record *> &Records) { 140 OS << "// EnumAttr classes\n"; 141 for (const auto *R : Records) { 142 OS << "struct " << R->getName() << "Attr : EnumAttr {\n"; 143 OS << " static enum Attribute::AttrKind getKind() {\n"; 144 OS << " return llvm::Attribute::" << R->getName() << ";\n"; 145 OS << " }\n"; 146 OS << "};\n"; 147 } 148 OS << "\n"; 149 } 150 151 void Attributes::printStrBoolAttrClasses(raw_ostream &OS, 152 const std::vector<Record *> &Records) { 153 OS << "// StrBoolAttr classes\n"; 154 for (const auto *R : Records) { 155 OS << "struct " << R->getName() << "Attr : StrBoolAttr {\n"; 156 OS << " static StringRef getKind() {\n"; 157 OS << " return \"" << R->getValueAsString("AttrString") << "\";\n"; 158 OS << " }\n"; 159 OS << "};\n"; 160 } 161 OS << "\n"; 162 } 163 164 void Attributes::emit(raw_ostream &OS) { 165 emitTargetIndependentEnums(OS); 166 emitConversionFn(OS); 167 emitFnAttrCompatCheck(OS, false); 168 } 169 170 namespace llvm { 171 172 void EmitAttributes(RecordKeeper &RK, raw_ostream &OS) { 173 Attributes(RK).emit(OS); 174 } 175 176 } // End llvm namespace. 177