contrib/llvm/utils/TableGen/Attributes.cpp

   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:
  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
  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
  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
  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
 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
 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
 167 void Attributes::emit(raw_ostream &OS) {
 168   emitTargetIndependentEnums(OS);
 169   emitConversionFn(OS);
 170   emitFnAttrCompatCheck(OS, false);
 171 }
 172
 173 namespace llvm {
 174
 175 void EmitAttributes(RecordKeeper &RK, raw_ostream &OS) {
 176   Attributes(RK).emit(OS);
 177 }
 178
 179 } // End llvm namespace.