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