llvm/utils/TableGen/Attributes.cpp

   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/TableGen/Record.h"
  10 #include "llvm/TableGen/TableGenBackend.h"
  11 #include <vector>
  12 using namespace llvm;
  13
  14 #define DEBUG_TYPE "attr-enum"
  15
  16 namespace {
  17
  18 class Attributes {
  19 public:
  20   Attributes(RecordKeeper &R) : Records(R) {}
  21   void run(raw_ostream &OS);
  22
  23 private:
  24   void emitTargetIndependentNames(raw_ostream &OS);
  25   void emitFnAttrCompatCheck(raw_ostream &OS, bool IsStringAttr);
  26   void emitAttributeProperties(raw_ostream &OF);
  27
  28   RecordKeeper &Records;
  29 };
  30
  31 } // End anonymous namespace.
  32
  33 void Attributes::emitTargetIndependentNames(raw_ostream &OS) {
  34   OS << "#ifdef GET_ATTR_NAMES\n";
  35   OS << "#undef GET_ATTR_NAMES\n";
  36
  37   OS << "#ifndef ATTRIBUTE_ALL\n";
  38   OS << "#define ATTRIBUTE_ALL(FIRST, SECOND)\n";
  39   OS << "#endif\n\n";
  40
  41   auto Emit = [&](ArrayRef<StringRef> KindNames, StringRef MacroName) {
  42     OS << "#ifndef " << MacroName << "\n";
  43     OS << "#define " << MacroName
  44        << "(FIRST, SECOND) ATTRIBUTE_ALL(FIRST, SECOND)\n";
  45     OS << "#endif\n\n";
  46     for (StringRef KindName : KindNames) {
  47       for (auto *A : Records.getAllDerivedDefinitions(KindName)) {
  48         OS << MacroName << "(" << A->getName() << ","
  49            << A->getValueAsString("AttrString") << ")\n";
  50       }
  51     }
  52     OS << "#undef " << MacroName << "\n\n";
  53   };
  54
  55   // Emit attribute enums in the same order llvm::Attribute::operator< expects.
  56   Emit({"EnumAttr", "TypeAttr", "IntAttr"}, "ATTRIBUTE_ENUM");
  57   Emit({"StrBoolAttr"}, "ATTRIBUTE_STRBOOL");
  58   Emit({"ComplexStrAttr"}, "ATTRIBUTE_COMPLEXSTR");
  59
  60   OS << "#undef ATTRIBUTE_ALL\n";
  61   OS << "#endif\n\n";
  62
  63   OS << "#ifdef GET_ATTR_ENUM\n";
  64   OS << "#undef GET_ATTR_ENUM\n";
  65   unsigned Value = 1; // Leave zero for AttrKind::None.
  66   for (StringRef KindName : {"EnumAttr", "TypeAttr", "IntAttr"}) {
  67     OS << "First" << KindName << " = " << Value << ",\n";
  68     for (auto *A : Records.getAllDerivedDefinitions(KindName)) {
  69       OS << A->getName() << " = " << Value << ",\n";
  70       Value++;
  71     }
  72     OS << "Last" << KindName << " = " << (Value - 1) << ",\n";
  73   }
  74   OS << "#endif\n\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 << "static inline bool hasCompatibleFnAttrs(const Function &Caller,\n"
  82      << "                                        const Function &Callee) {\n";
  83   OS << "  bool Ret = true;\n\n";
  84
  85   std::vector<Record *> CompatRules =
  86       Records.getAllDerivedDefinitions("CompatRule");
  87
  88   for (auto *Rule : CompatRules) {
  89     StringRef FuncName = Rule->getValueAsString("CompatFunc");
  90     OS << "  Ret &= " << FuncName << "(Caller, Callee);\n";
  91   }
  92
  93   OS << "\n";
  94   OS << "  return Ret;\n";
  95   OS << "}\n\n";
  96
  97   std::vector<Record *> MergeRules =
  98       Records.getAllDerivedDefinitions("MergeRule");
  99   OS << "static inline void mergeFnAttrs(Function &Caller,\n"
 100      << "                                const Function &Callee) {\n";
 101
 102   for (auto *Rule : MergeRules) {
 103     StringRef FuncName = Rule->getValueAsString("MergeFunc");
 104     OS << "  " << FuncName << "(Caller, Callee);\n";
 105   }
 106
 107   OS << "}\n\n";
 108
 109   OS << "#endif\n";
 110 }
 111
 112 void Attributes::emitAttributeProperties(raw_ostream &OS) {
 113   OS << "#ifdef GET_ATTR_PROP_TABLE\n";
 114   OS << "#undef GET_ATTR_PROP_TABLE\n";
 115   OS << "static const uint8_t AttrPropTable[] = {\n";
 116   for (StringRef KindName : {"EnumAttr", "TypeAttr", "IntAttr"}) {
 117     for (auto *A : Records.getAllDerivedDefinitions(KindName)) {
 118       OS << "0";
 119       for (Init *P : *A->getValueAsListInit("Properties"))
 120         OS << " | AttributeProperty::" << cast<DefInit>(P)->getDef()->getName();
 121       OS << ",\n";
 122     }
 123   }
 124   OS << "};\n";
 125   OS << "#endif\n";
 126 }
 127
 128 void Attributes::run(raw_ostream &OS) {
 129   emitTargetIndependentNames(OS);
 130   emitFnAttrCompatCheck(OS, false);
 131   emitAttributeProperties(OS);
 132 }
 133
 134 static TableGen::Emitter::OptClass<Attributes> X("gen-attrs",
 135                                                  "Generate attributes");