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 emitFnAttrCompatCheck(raw_ostream &OS, bool IsStringAttr);
  31
  32   void printEnumAttrClasses(raw_ostream &OS,
  33                             const std::vector<Record *> &Records);
  34   void printStrBoolAttrClasses(raw_ostream &OS,
  35                                const std::vector<Record *> &Records);
  36
  37   RecordKeeper &Records;
  38 };
  39
  40 } // End anonymous namespace.
  41
  42 void Attributes::emitTargetIndependentEnums(raw_ostream &OS) {
  43   OS << "#ifdef GET_ATTR_ENUM\n";
  44   OS << "#undef GET_ATTR_ENUM\n";
  45
  46   const std::vector<Record*> &Attrs =
  47       Records.getAllDerivedDefinitions("EnumAttr");
  48
  49   for (auto A : Attrs)
  50     OS << A->getName() << ",\n";
  51
  52   OS << "#endif\n";
  53 }
  54
  55 void Attributes::emitFnAttrCompatCheck(raw_ostream &OS, bool IsStringAttr) {
  56   OS << "#ifdef GET_ATTR_COMPAT_FUNC\n";
  57   OS << "#undef GET_ATTR_COMPAT_FUNC\n";
  58
  59   OS << "struct EnumAttr {\n";
  60   OS << "  static bool isSet(const Function &Fn,\n";
  61   OS << "                    Attribute::AttrKind Kind) {\n";
  62   OS << "    return Fn.hasFnAttribute(Kind);\n";
  63   OS << "  }\n\n";
  64   OS << "  static void set(Function &Fn,\n";
  65   OS << "                  Attribute::AttrKind Kind, bool Val) {\n";
  66   OS << "    if (Val)\n";
  67   OS << "      Fn.addFnAttr(Kind);\n";
  68   OS << "    else\n";
  69   OS << "      Fn.removeFnAttr(Kind);\n";
  70   OS << "  }\n";
  71   OS << "};\n\n";
  72
  73   OS << "struct StrBoolAttr {\n";
  74   OS << "  static bool isSet(const Function &Fn,\n";
  75   OS << "                    StringRef Kind) {\n";
  76   OS << "    auto A = Fn.getFnAttribute(Kind);\n";
  77   OS << "    return A.getValueAsString().equals(\"true\");\n";
  78   OS << "  }\n\n";
  79   OS << "  static void set(Function &Fn,\n";
  80   OS << "                  StringRef Kind, bool Val) {\n";
  81   OS << "    Fn.addFnAttr(Kind, Val ? \"true\" : \"false\");\n";
  82   OS << "  }\n";
  83   OS << "};\n\n";
  84
  85   printEnumAttrClasses(OS ,Records.getAllDerivedDefinitions("EnumAttr"));
  86   printStrBoolAttrClasses(OS , Records.getAllDerivedDefinitions("StrBoolAttr"));
  87
  88   OS << "static inline bool hasCompatibleFnAttrs(const Function &Caller,\n"
  89      << "                                        const Function &Callee) {\n";
  90   OS << "  bool Ret = true;\n\n";
  91
  92   std::vector<Record *> CompatRules =
  93       Records.getAllDerivedDefinitions("CompatRule");
  94
  95   for (auto *Rule : CompatRules) {
  96     StringRef FuncName = Rule->getValueAsString("CompatFunc");
  97     OS << "  Ret &= " << FuncName << "(Caller, Callee);\n";
  98   }
  99
 100   OS << "\n";
 101   OS << "  return Ret;\n";
 102   OS << "}\n\n";
 103
 104   std::vector<Record *> MergeRules =
 105       Records.getAllDerivedDefinitions("MergeRule");
 106   OS << "static inline void mergeFnAttrs(Function &Caller,\n"
 107      << "                                const Function &Callee) {\n";
 108
 109   for (auto *Rule : MergeRules) {
 110     StringRef FuncName = Rule->getValueAsString("MergeFunc");
 111     OS << "  " << FuncName << "(Caller, Callee);\n";
 112   }
 113
 114   OS << "}\n\n";
 115
 116   OS << "#endif\n";
 117 }
 118
 119 void Attributes::printEnumAttrClasses(raw_ostream &OS,
 120                                       const std::vector<Record *> &Records) {
 121   OS << "// EnumAttr classes\n";
 122   for (const auto *R : Records) {
 123     OS << "struct " << R->getName() << "Attr : EnumAttr {\n";
 124     OS << "  static enum Attribute::AttrKind getKind() {\n";
 125     OS << "    return llvm::Attribute::" << R->getName() << ";\n";
 126     OS << "  }\n";
 127     OS << "};\n";
 128   }
 129   OS << "\n";
 130 }
 131
 132 void Attributes::printStrBoolAttrClasses(raw_ostream &OS,
 133                                          const std::vector<Record *> &Records) {
 134   OS << "// StrBoolAttr classes\n";
 135   for (const auto *R : Records) {
 136     OS << "struct " << R->getName() << "Attr : StrBoolAttr {\n";
 137     OS << "  static const char *getKind() {\n";
 138     OS << "    return \"" << R->getValueAsString("AttrString") << "\";\n";
 139     OS << "  }\n";
 140     OS << "};\n";
 141   }
 142   OS << "\n";
 143 }
 144
 145 void Attributes::emit(raw_ostream &OS) {
 146   emitTargetIndependentEnums(OS);
 147   emitFnAttrCompatCheck(OS, false);
 148 }
 149
 150 namespace llvm {
 151
 152 void EmitAttributes(RecordKeeper &RK, raw_ostream &OS) {
 153   Attributes(RK).emit(OS);
 154 }
 155
 156 } // End llvm namespace.