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>
17using namespace llvm;
18
19#define DEBUG_TYPE "attr-enum"
20
21namespace {
22
23class Attributes {
24public:
25  Attributes(RecordKeeper &R) : Records(R) {}
26  void emit(raw_ostream &OS);
27
28private:
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
43void 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
56void 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
77void 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
141void 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
154void 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
167void Attributes::emit(raw_ostream &OS) {
168  emitTargetIndependentEnums(OS);
169  emitConversionFn(OS);
170  emitFnAttrCompatCheck(OS, false);
171}
172
173namespace llvm {
174
175void EmitAttributes(RecordKeeper &RK, raw_ostream &OS) {
176  Attributes(RK).emit(OS);
177}
178
179} // End llvm namespace.
180