xref: /openbsd-src/gnu/llvm/clang/utils/TableGen/ClangAttrEmitter.cpp (revision 12c855180aad702bbcca06e0398d774beeafb155)
1e5dd7070Spatrick //===- ClangAttrEmitter.cpp - Generate Clang attribute handling =-*- C++ -*--=//
2e5dd7070Spatrick //
3e5dd7070Spatrick // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4e5dd7070Spatrick // See https://llvm.org/LICENSE.txt for license information.
5e5dd7070Spatrick // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6e5dd7070Spatrick //
7e5dd7070Spatrick //===----------------------------------------------------------------------===//
8e5dd7070Spatrick //
9e5dd7070Spatrick // These tablegen backends emit Clang attribute processing code
10e5dd7070Spatrick //
11e5dd7070Spatrick //===----------------------------------------------------------------------===//
12e5dd7070Spatrick 
13e5dd7070Spatrick #include "TableGenBackends.h"
14e5dd7070Spatrick #include "ASTTableGen.h"
15e5dd7070Spatrick 
16e5dd7070Spatrick #include "llvm/ADT/ArrayRef.h"
17e5dd7070Spatrick #include "llvm/ADT/DenseMap.h"
18e5dd7070Spatrick #include "llvm/ADT/DenseSet.h"
19*12c85518Srobert #include "llvm/ADT/MapVector.h"
20e5dd7070Spatrick #include "llvm/ADT/STLExtras.h"
21e5dd7070Spatrick #include "llvm/ADT/SmallString.h"
22e5dd7070Spatrick #include "llvm/ADT/StringExtras.h"
23e5dd7070Spatrick #include "llvm/ADT/StringRef.h"
24e5dd7070Spatrick #include "llvm/ADT/StringSet.h"
25e5dd7070Spatrick #include "llvm/ADT/StringSwitch.h"
26e5dd7070Spatrick #include "llvm/ADT/iterator_range.h"
27e5dd7070Spatrick #include "llvm/Support/ErrorHandling.h"
28e5dd7070Spatrick #include "llvm/Support/raw_ostream.h"
29e5dd7070Spatrick #include "llvm/TableGen/Error.h"
30e5dd7070Spatrick #include "llvm/TableGen/Record.h"
31e5dd7070Spatrick #include "llvm/TableGen/StringMatcher.h"
32e5dd7070Spatrick #include "llvm/TableGen/TableGenBackend.h"
33e5dd7070Spatrick #include <algorithm>
34e5dd7070Spatrick #include <cassert>
35e5dd7070Spatrick #include <cctype>
36e5dd7070Spatrick #include <cstddef>
37e5dd7070Spatrick #include <cstdint>
38e5dd7070Spatrick #include <map>
39e5dd7070Spatrick #include <memory>
40*12c85518Srobert #include <optional>
41e5dd7070Spatrick #include <set>
42e5dd7070Spatrick #include <sstream>
43e5dd7070Spatrick #include <string>
44e5dd7070Spatrick #include <utility>
45e5dd7070Spatrick #include <vector>
46e5dd7070Spatrick 
47e5dd7070Spatrick using namespace llvm;
48e5dd7070Spatrick 
49e5dd7070Spatrick namespace {
50e5dd7070Spatrick 
51e5dd7070Spatrick class FlattenedSpelling {
52e5dd7070Spatrick   std::string V, N, NS;
53ec727ea7Spatrick   bool K = false;
54e5dd7070Spatrick 
55e5dd7070Spatrick public:
FlattenedSpelling(const std::string & Variety,const std::string & Name,const std::string & Namespace,bool KnownToGCC)56e5dd7070Spatrick   FlattenedSpelling(const std::string &Variety, const std::string &Name,
57e5dd7070Spatrick                     const std::string &Namespace, bool KnownToGCC) :
58e5dd7070Spatrick     V(Variety), N(Name), NS(Namespace), K(KnownToGCC) {}
FlattenedSpelling(const Record & Spelling)59ec727ea7Spatrick   explicit FlattenedSpelling(const Record &Spelling)
60ec727ea7Spatrick       : V(std::string(Spelling.getValueAsString("Variety"))),
61ec727ea7Spatrick         N(std::string(Spelling.getValueAsString("Name"))) {
62e5dd7070Spatrick     assert(V != "GCC" && V != "Clang" &&
63e5dd7070Spatrick            "Given a GCC spelling, which means this hasn't been flattened!");
64e5dd7070Spatrick     if (V == "CXX11" || V == "C2x" || V == "Pragma")
65ec727ea7Spatrick       NS = std::string(Spelling.getValueAsString("Namespace"));
66e5dd7070Spatrick   }
67e5dd7070Spatrick 
variety() const68e5dd7070Spatrick   const std::string &variety() const { return V; }
name() const69e5dd7070Spatrick   const std::string &name() const { return N; }
nameSpace() const70e5dd7070Spatrick   const std::string &nameSpace() const { return NS; }
knownToGCC() const71e5dd7070Spatrick   bool knownToGCC() const { return K; }
72e5dd7070Spatrick };
73e5dd7070Spatrick 
74e5dd7070Spatrick } // end anonymous namespace
75e5dd7070Spatrick 
76e5dd7070Spatrick static std::vector<FlattenedSpelling>
GetFlattenedSpellings(const Record & Attr)77e5dd7070Spatrick GetFlattenedSpellings(const Record &Attr) {
78e5dd7070Spatrick   std::vector<Record *> Spellings = Attr.getValueAsListOfDefs("Spellings");
79e5dd7070Spatrick   std::vector<FlattenedSpelling> Ret;
80e5dd7070Spatrick 
81e5dd7070Spatrick   for (const auto &Spelling : Spellings) {
82e5dd7070Spatrick     StringRef Variety = Spelling->getValueAsString("Variety");
83e5dd7070Spatrick     StringRef Name = Spelling->getValueAsString("Name");
84e5dd7070Spatrick     if (Variety == "GCC") {
85ec727ea7Spatrick       Ret.emplace_back("GNU", std::string(Name), "", true);
86ec727ea7Spatrick       Ret.emplace_back("CXX11", std::string(Name), "gnu", true);
87e5dd7070Spatrick       if (Spelling->getValueAsBit("AllowInC"))
88ec727ea7Spatrick         Ret.emplace_back("C2x", std::string(Name), "gnu", true);
89ec727ea7Spatrick     } else if (Variety == "Clang") {
90ec727ea7Spatrick       Ret.emplace_back("GNU", std::string(Name), "", false);
91ec727ea7Spatrick       Ret.emplace_back("CXX11", std::string(Name), "clang", false);
92ec727ea7Spatrick       if (Spelling->getValueAsBit("AllowInC"))
93ec727ea7Spatrick         Ret.emplace_back("C2x", std::string(Name), "clang", false);
94e5dd7070Spatrick     } else
95e5dd7070Spatrick       Ret.push_back(FlattenedSpelling(*Spelling));
96e5dd7070Spatrick   }
97e5dd7070Spatrick 
98e5dd7070Spatrick   return Ret;
99e5dd7070Spatrick }
100e5dd7070Spatrick 
ReadPCHRecord(StringRef type)101e5dd7070Spatrick static std::string ReadPCHRecord(StringRef type) {
102e5dd7070Spatrick   return StringSwitch<std::string>(type)
103ec727ea7Spatrick       .EndsWith("Decl *", "Record.GetLocalDeclAs<" +
104ec727ea7Spatrick                               std::string(type.data(), 0, type.size() - 1) +
105ec727ea7Spatrick                               ">(Record.readInt())")
106e5dd7070Spatrick       .Case("TypeSourceInfo *", "Record.readTypeSourceInfo()")
107e5dd7070Spatrick       .Case("Expr *", "Record.readExpr()")
108e5dd7070Spatrick       .Case("IdentifierInfo *", "Record.readIdentifier()")
109e5dd7070Spatrick       .Case("StringRef", "Record.readString()")
110e5dd7070Spatrick       .Case("ParamIdx", "ParamIdx::deserialize(Record.readInt())")
111ec727ea7Spatrick       .Case("OMPTraitInfo *", "Record.readOMPTraitInfo()")
112e5dd7070Spatrick       .Default("Record.readInt()");
113e5dd7070Spatrick }
114e5dd7070Spatrick 
115e5dd7070Spatrick // Get a type that is suitable for storing an object of the specified type.
getStorageType(StringRef type)116e5dd7070Spatrick static StringRef getStorageType(StringRef type) {
117e5dd7070Spatrick   return StringSwitch<StringRef>(type)
118e5dd7070Spatrick     .Case("StringRef", "std::string")
119e5dd7070Spatrick     .Default(type);
120e5dd7070Spatrick }
121e5dd7070Spatrick 
122e5dd7070Spatrick // Assumes that the way to get the value is SA->getname()
WritePCHRecord(StringRef type,StringRef name)123e5dd7070Spatrick static std::string WritePCHRecord(StringRef type, StringRef name) {
124ec727ea7Spatrick   return "Record." +
125ec727ea7Spatrick          StringSwitch<std::string>(type)
126e5dd7070Spatrick              .EndsWith("Decl *", "AddDeclRef(" + std::string(name) + ");\n")
127ec727ea7Spatrick              .Case("TypeSourceInfo *",
128ec727ea7Spatrick                    "AddTypeSourceInfo(" + std::string(name) + ");\n")
129e5dd7070Spatrick              .Case("Expr *", "AddStmt(" + std::string(name) + ");\n")
130ec727ea7Spatrick              .Case("IdentifierInfo *",
131ec727ea7Spatrick                    "AddIdentifierRef(" + std::string(name) + ");\n")
132e5dd7070Spatrick              .Case("StringRef", "AddString(" + std::string(name) + ");\n")
133ec727ea7Spatrick              .Case("ParamIdx",
134ec727ea7Spatrick                    "push_back(" + std::string(name) + ".serialize());\n")
135ec727ea7Spatrick              .Case("OMPTraitInfo *",
136ec727ea7Spatrick                    "writeOMPTraitInfo(" + std::string(name) + ");\n")
137e5dd7070Spatrick              .Default("push_back(" + std::string(name) + ");\n");
138e5dd7070Spatrick }
139e5dd7070Spatrick 
140e5dd7070Spatrick // Normalize attribute name by removing leading and trailing
141e5dd7070Spatrick // underscores. For example, __foo, foo__, __foo__ would
142e5dd7070Spatrick // become foo.
NormalizeAttrName(StringRef AttrName)143e5dd7070Spatrick static StringRef NormalizeAttrName(StringRef AttrName) {
144e5dd7070Spatrick   AttrName.consume_front("__");
145e5dd7070Spatrick   AttrName.consume_back("__");
146e5dd7070Spatrick   return AttrName;
147e5dd7070Spatrick }
148e5dd7070Spatrick 
149e5dd7070Spatrick // Normalize the name by removing any and all leading and trailing underscores.
150e5dd7070Spatrick // This is different from NormalizeAttrName in that it also handles names like
151e5dd7070Spatrick // _pascal and __pascal.
NormalizeNameForSpellingComparison(StringRef Name)152e5dd7070Spatrick static StringRef NormalizeNameForSpellingComparison(StringRef Name) {
153e5dd7070Spatrick   return Name.trim("_");
154e5dd7070Spatrick }
155e5dd7070Spatrick 
156e5dd7070Spatrick // Normalize the spelling of a GNU attribute (i.e. "x" in "__attribute__((x))"),
157e5dd7070Spatrick // removing "__" if it appears at the beginning and end of the attribute's name.
NormalizeGNUAttrSpelling(StringRef AttrSpelling)158e5dd7070Spatrick static StringRef NormalizeGNUAttrSpelling(StringRef AttrSpelling) {
159e5dd7070Spatrick   if (AttrSpelling.startswith("__") && AttrSpelling.endswith("__")) {
160e5dd7070Spatrick     AttrSpelling = AttrSpelling.substr(2, AttrSpelling.size() - 4);
161e5dd7070Spatrick   }
162e5dd7070Spatrick 
163e5dd7070Spatrick   return AttrSpelling;
164e5dd7070Spatrick }
165e5dd7070Spatrick 
166e5dd7070Spatrick typedef std::vector<std::pair<std::string, const Record *>> ParsedAttrMap;
167e5dd7070Spatrick 
getParsedAttrList(const RecordKeeper & Records,ParsedAttrMap * Dupes=nullptr)168e5dd7070Spatrick static ParsedAttrMap getParsedAttrList(const RecordKeeper &Records,
169e5dd7070Spatrick                                        ParsedAttrMap *Dupes = nullptr) {
170e5dd7070Spatrick   std::vector<Record *> Attrs = Records.getAllDerivedDefinitions("Attr");
171e5dd7070Spatrick   std::set<std::string> Seen;
172e5dd7070Spatrick   ParsedAttrMap R;
173e5dd7070Spatrick   for (const auto *Attr : Attrs) {
174e5dd7070Spatrick     if (Attr->getValueAsBit("SemaHandler")) {
175e5dd7070Spatrick       std::string AN;
176e5dd7070Spatrick       if (Attr->isSubClassOf("TargetSpecificAttr") &&
177e5dd7070Spatrick           !Attr->isValueUnset("ParseKind")) {
178ec727ea7Spatrick         AN = std::string(Attr->getValueAsString("ParseKind"));
179e5dd7070Spatrick 
180e5dd7070Spatrick         // If this attribute has already been handled, it does not need to be
181e5dd7070Spatrick         // handled again.
182e5dd7070Spatrick         if (Seen.find(AN) != Seen.end()) {
183e5dd7070Spatrick           if (Dupes)
184e5dd7070Spatrick             Dupes->push_back(std::make_pair(AN, Attr));
185e5dd7070Spatrick           continue;
186e5dd7070Spatrick         }
187e5dd7070Spatrick         Seen.insert(AN);
188e5dd7070Spatrick       } else
189e5dd7070Spatrick         AN = NormalizeAttrName(Attr->getName()).str();
190e5dd7070Spatrick 
191e5dd7070Spatrick       R.push_back(std::make_pair(AN, Attr));
192e5dd7070Spatrick     }
193e5dd7070Spatrick   }
194e5dd7070Spatrick   return R;
195e5dd7070Spatrick }
196e5dd7070Spatrick 
197e5dd7070Spatrick namespace {
198e5dd7070Spatrick 
199e5dd7070Spatrick   class Argument {
200e5dd7070Spatrick     std::string lowerName, upperName;
201e5dd7070Spatrick     StringRef attrName;
202e5dd7070Spatrick     bool isOpt;
203e5dd7070Spatrick     bool Fake;
204e5dd7070Spatrick 
205e5dd7070Spatrick   public:
Argument(StringRef Arg,StringRef Attr)206*12c85518Srobert     Argument(StringRef Arg, StringRef Attr)
207*12c85518Srobert         : lowerName(std::string(Arg)), upperName(lowerName), attrName(Attr),
208*12c85518Srobert           isOpt(false), Fake(false) {
209e5dd7070Spatrick       if (!lowerName.empty()) {
210e5dd7070Spatrick         lowerName[0] = std::tolower(lowerName[0]);
211e5dd7070Spatrick         upperName[0] = std::toupper(upperName[0]);
212e5dd7070Spatrick       }
213e5dd7070Spatrick       // Work around MinGW's macro definition of 'interface' to 'struct'. We
214e5dd7070Spatrick       // have an attribute argument called 'Interface', so only the lower case
215e5dd7070Spatrick       // name conflicts with the macro definition.
216e5dd7070Spatrick       if (lowerName == "interface")
217e5dd7070Spatrick         lowerName = "interface_";
218e5dd7070Spatrick     }
Argument(const Record & Arg,StringRef Attr)219*12c85518Srobert     Argument(const Record &Arg, StringRef Attr)
220*12c85518Srobert         : Argument(Arg.getValueAsString("Name"), Attr) {}
221e5dd7070Spatrick     virtual ~Argument() = default;
222e5dd7070Spatrick 
getLowerName() const223e5dd7070Spatrick     StringRef getLowerName() const { return lowerName; }
getUpperName() const224e5dd7070Spatrick     StringRef getUpperName() const { return upperName; }
getAttrName() const225e5dd7070Spatrick     StringRef getAttrName() const { return attrName; }
226e5dd7070Spatrick 
isOptional() const227e5dd7070Spatrick     bool isOptional() const { return isOpt; }
setOptional(bool set)228e5dd7070Spatrick     void setOptional(bool set) { isOpt = set; }
229e5dd7070Spatrick 
isFake() const230e5dd7070Spatrick     bool isFake() const { return Fake; }
setFake(bool fake)231e5dd7070Spatrick     void setFake(bool fake) { Fake = fake; }
232e5dd7070Spatrick 
233e5dd7070Spatrick     // These functions print the argument contents formatted in different ways.
234e5dd7070Spatrick     virtual void writeAccessors(raw_ostream &OS) const = 0;
writeAccessorDefinitions(raw_ostream & OS) const235e5dd7070Spatrick     virtual void writeAccessorDefinitions(raw_ostream &OS) const {}
writeASTVisitorTraversal(raw_ostream & OS) const236e5dd7070Spatrick     virtual void writeASTVisitorTraversal(raw_ostream &OS) const {}
237e5dd7070Spatrick     virtual void writeCloneArgs(raw_ostream &OS) const = 0;
238e5dd7070Spatrick     virtual void writeTemplateInstantiationArgs(raw_ostream &OS) const = 0;
writeTemplateInstantiation(raw_ostream & OS) const239e5dd7070Spatrick     virtual void writeTemplateInstantiation(raw_ostream &OS) const {}
writeCtorBody(raw_ostream & OS) const240e5dd7070Spatrick     virtual void writeCtorBody(raw_ostream &OS) const {}
241e5dd7070Spatrick     virtual void writeCtorInitializers(raw_ostream &OS) const = 0;
242e5dd7070Spatrick     virtual void writeCtorDefaultInitializers(raw_ostream &OS) const = 0;
243e5dd7070Spatrick     virtual void writeCtorParameters(raw_ostream &OS) const = 0;
244e5dd7070Spatrick     virtual void writeDeclarations(raw_ostream &OS) const = 0;
245e5dd7070Spatrick     virtual void writePCHReadArgs(raw_ostream &OS) const = 0;
246e5dd7070Spatrick     virtual void writePCHReadDecls(raw_ostream &OS) const = 0;
247e5dd7070Spatrick     virtual void writePCHWrite(raw_ostream &OS) const = 0;
getIsOmitted() const248e5dd7070Spatrick     virtual std::string getIsOmitted() const { return "false"; }
249e5dd7070Spatrick     virtual void writeValue(raw_ostream &OS) const = 0;
250e5dd7070Spatrick     virtual void writeDump(raw_ostream &OS) const = 0;
writeDumpChildren(raw_ostream & OS) const251e5dd7070Spatrick     virtual void writeDumpChildren(raw_ostream &OS) const {}
writeHasChildren(raw_ostream & OS) const252e5dd7070Spatrick     virtual void writeHasChildren(raw_ostream &OS) const { OS << "false"; }
253e5dd7070Spatrick 
isEnumArg() const254e5dd7070Spatrick     virtual bool isEnumArg() const { return false; }
isVariadicEnumArg() const255e5dd7070Spatrick     virtual bool isVariadicEnumArg() const { return false; }
isVariadic() const256e5dd7070Spatrick     virtual bool isVariadic() const { return false; }
257e5dd7070Spatrick 
writeImplicitCtorArgs(raw_ostream & OS) const258e5dd7070Spatrick     virtual void writeImplicitCtorArgs(raw_ostream &OS) const {
259e5dd7070Spatrick       OS << getUpperName();
260e5dd7070Spatrick     }
261e5dd7070Spatrick   };
262e5dd7070Spatrick 
263e5dd7070Spatrick   class SimpleArgument : public Argument {
264e5dd7070Spatrick     std::string type;
265e5dd7070Spatrick 
266e5dd7070Spatrick   public:
SimpleArgument(const Record & Arg,StringRef Attr,std::string T)267e5dd7070Spatrick     SimpleArgument(const Record &Arg, StringRef Attr, std::string T)
268e5dd7070Spatrick         : Argument(Arg, Attr), type(std::move(T)) {}
269e5dd7070Spatrick 
getType() const270e5dd7070Spatrick     std::string getType() const { return type; }
271e5dd7070Spatrick 
writeAccessors(raw_ostream & OS) const272e5dd7070Spatrick     void writeAccessors(raw_ostream &OS) const override {
273e5dd7070Spatrick       OS << "  " << type << " get" << getUpperName() << "() const {\n";
274e5dd7070Spatrick       OS << "    return " << getLowerName() << ";\n";
275e5dd7070Spatrick       OS << "  }";
276e5dd7070Spatrick     }
277e5dd7070Spatrick 
writeCloneArgs(raw_ostream & OS) const278e5dd7070Spatrick     void writeCloneArgs(raw_ostream &OS) const override {
279e5dd7070Spatrick       OS << getLowerName();
280e5dd7070Spatrick     }
281e5dd7070Spatrick 
writeTemplateInstantiationArgs(raw_ostream & OS) const282e5dd7070Spatrick     void writeTemplateInstantiationArgs(raw_ostream &OS) const override {
283e5dd7070Spatrick       OS << "A->get" << getUpperName() << "()";
284e5dd7070Spatrick     }
285e5dd7070Spatrick 
writeCtorInitializers(raw_ostream & OS) const286e5dd7070Spatrick     void writeCtorInitializers(raw_ostream &OS) const override {
287e5dd7070Spatrick       OS << getLowerName() << "(" << getUpperName() << ")";
288e5dd7070Spatrick     }
289e5dd7070Spatrick 
writeCtorDefaultInitializers(raw_ostream & OS) const290e5dd7070Spatrick     void writeCtorDefaultInitializers(raw_ostream &OS) const override {
291e5dd7070Spatrick       OS << getLowerName() << "()";
292e5dd7070Spatrick     }
293e5dd7070Spatrick 
writeCtorParameters(raw_ostream & OS) const294e5dd7070Spatrick     void writeCtorParameters(raw_ostream &OS) const override {
295e5dd7070Spatrick       OS << type << " " << getUpperName();
296e5dd7070Spatrick     }
297e5dd7070Spatrick 
writeDeclarations(raw_ostream & OS) const298e5dd7070Spatrick     void writeDeclarations(raw_ostream &OS) const override {
299e5dd7070Spatrick       OS << type << " " << getLowerName() << ";";
300e5dd7070Spatrick     }
301e5dd7070Spatrick 
writePCHReadDecls(raw_ostream & OS) const302e5dd7070Spatrick     void writePCHReadDecls(raw_ostream &OS) const override {
303e5dd7070Spatrick       std::string read = ReadPCHRecord(type);
304e5dd7070Spatrick       OS << "    " << type << " " << getLowerName() << " = " << read << ";\n";
305e5dd7070Spatrick     }
306e5dd7070Spatrick 
writePCHReadArgs(raw_ostream & OS) const307e5dd7070Spatrick     void writePCHReadArgs(raw_ostream &OS) const override {
308e5dd7070Spatrick       OS << getLowerName();
309e5dd7070Spatrick     }
310e5dd7070Spatrick 
writePCHWrite(raw_ostream & OS) const311e5dd7070Spatrick     void writePCHWrite(raw_ostream &OS) const override {
312ec727ea7Spatrick       OS << "    "
313ec727ea7Spatrick          << WritePCHRecord(type,
314ec727ea7Spatrick                            "SA->get" + std::string(getUpperName()) + "()");
315e5dd7070Spatrick     }
316e5dd7070Spatrick 
getIsOmitted() const317e5dd7070Spatrick     std::string getIsOmitted() const override {
318e5dd7070Spatrick       if (type == "IdentifierInfo *")
319e5dd7070Spatrick         return "!get" + getUpperName().str() + "()";
320e5dd7070Spatrick       if (type == "TypeSourceInfo *")
321e5dd7070Spatrick         return "!get" + getUpperName().str() + "Loc()";
322e5dd7070Spatrick       if (type == "ParamIdx")
323e5dd7070Spatrick         return "!get" + getUpperName().str() + "().isValid()";
324e5dd7070Spatrick       return "false";
325e5dd7070Spatrick     }
326e5dd7070Spatrick 
writeValue(raw_ostream & OS) const327e5dd7070Spatrick     void writeValue(raw_ostream &OS) const override {
328e5dd7070Spatrick       if (type == "FunctionDecl *")
329e5dd7070Spatrick         OS << "\" << get" << getUpperName()
330e5dd7070Spatrick            << "()->getNameInfo().getAsString() << \"";
331e5dd7070Spatrick       else if (type == "IdentifierInfo *")
332e5dd7070Spatrick         // Some non-optional (comma required) identifier arguments can be the
333e5dd7070Spatrick         // empty string but are then recorded as a nullptr.
334e5dd7070Spatrick         OS << "\" << (get" << getUpperName() << "() ? get" << getUpperName()
335e5dd7070Spatrick            << "()->getName() : \"\") << \"";
336a9ac8606Spatrick       else if (type == "VarDecl *")
337a9ac8606Spatrick         OS << "\" << get" << getUpperName() << "()->getName() << \"";
338e5dd7070Spatrick       else if (type == "TypeSourceInfo *")
339e5dd7070Spatrick         OS << "\" << get" << getUpperName() << "().getAsString() << \"";
340e5dd7070Spatrick       else if (type == "ParamIdx")
341e5dd7070Spatrick         OS << "\" << get" << getUpperName() << "().getSourceIndex() << \"";
342e5dd7070Spatrick       else
343e5dd7070Spatrick         OS << "\" << get" << getUpperName() << "() << \"";
344e5dd7070Spatrick     }
345e5dd7070Spatrick 
writeDump(raw_ostream & OS) const346e5dd7070Spatrick     void writeDump(raw_ostream &OS) const override {
347ec727ea7Spatrick       if (StringRef(type).endswith("Decl *")) {
348e5dd7070Spatrick         OS << "    OS << \" \";\n";
349e5dd7070Spatrick         OS << "    dumpBareDeclRef(SA->get" << getUpperName() << "());\n";
350e5dd7070Spatrick       } else if (type == "IdentifierInfo *") {
351e5dd7070Spatrick         // Some non-optional (comma required) identifier arguments can be the
352e5dd7070Spatrick         // empty string but are then recorded as a nullptr.
353e5dd7070Spatrick         OS << "    if (SA->get" << getUpperName() << "())\n"
354e5dd7070Spatrick            << "      OS << \" \" << SA->get" << getUpperName()
355e5dd7070Spatrick            << "()->getName();\n";
356e5dd7070Spatrick       } else if (type == "TypeSourceInfo *") {
357e5dd7070Spatrick         if (isOptional())
358e5dd7070Spatrick           OS << "    if (SA->get" << getUpperName() << "Loc())";
359e5dd7070Spatrick         OS << "    OS << \" \" << SA->get" << getUpperName()
360e5dd7070Spatrick            << "().getAsString();\n";
361e5dd7070Spatrick       } else if (type == "bool") {
362e5dd7070Spatrick         OS << "    if (SA->get" << getUpperName() << "()) OS << \" "
363e5dd7070Spatrick            << getUpperName() << "\";\n";
364e5dd7070Spatrick       } else if (type == "int" || type == "unsigned") {
365e5dd7070Spatrick         OS << "    OS << \" \" << SA->get" << getUpperName() << "();\n";
366e5dd7070Spatrick       } else if (type == "ParamIdx") {
367e5dd7070Spatrick         if (isOptional())
368e5dd7070Spatrick           OS << "    if (SA->get" << getUpperName() << "().isValid())\n  ";
369e5dd7070Spatrick         OS << "    OS << \" \" << SA->get" << getUpperName()
370e5dd7070Spatrick            << "().getSourceIndex();\n";
371ec727ea7Spatrick       } else if (type == "OMPTraitInfo *") {
372ec727ea7Spatrick         OS << "    OS << \" \" << SA->get" << getUpperName() << "();\n";
373e5dd7070Spatrick       } else {
374e5dd7070Spatrick         llvm_unreachable("Unknown SimpleArgument type!");
375e5dd7070Spatrick       }
376e5dd7070Spatrick     }
377e5dd7070Spatrick   };
378e5dd7070Spatrick 
379e5dd7070Spatrick   class DefaultSimpleArgument : public SimpleArgument {
380e5dd7070Spatrick     int64_t Default;
381e5dd7070Spatrick 
382e5dd7070Spatrick   public:
DefaultSimpleArgument(const Record & Arg,StringRef Attr,std::string T,int64_t Default)383e5dd7070Spatrick     DefaultSimpleArgument(const Record &Arg, StringRef Attr,
384e5dd7070Spatrick                           std::string T, int64_t Default)
385e5dd7070Spatrick       : SimpleArgument(Arg, Attr, T), Default(Default) {}
386e5dd7070Spatrick 
writeAccessors(raw_ostream & OS) const387e5dd7070Spatrick     void writeAccessors(raw_ostream &OS) const override {
388e5dd7070Spatrick       SimpleArgument::writeAccessors(OS);
389e5dd7070Spatrick 
390e5dd7070Spatrick       OS << "\n\n  static const " << getType() << " Default" << getUpperName()
391e5dd7070Spatrick          << " = ";
392e5dd7070Spatrick       if (getType() == "bool")
393e5dd7070Spatrick         OS << (Default != 0 ? "true" : "false");
394e5dd7070Spatrick       else
395e5dd7070Spatrick         OS << Default;
396e5dd7070Spatrick       OS << ";";
397e5dd7070Spatrick     }
398e5dd7070Spatrick   };
399e5dd7070Spatrick 
400e5dd7070Spatrick   class StringArgument : public Argument {
401e5dd7070Spatrick   public:
StringArgument(const Record & Arg,StringRef Attr)402e5dd7070Spatrick     StringArgument(const Record &Arg, StringRef Attr)
403e5dd7070Spatrick       : Argument(Arg, Attr)
404e5dd7070Spatrick     {}
405e5dd7070Spatrick 
writeAccessors(raw_ostream & OS) const406e5dd7070Spatrick     void writeAccessors(raw_ostream &OS) const override {
407e5dd7070Spatrick       OS << "  llvm::StringRef get" << getUpperName() << "() const {\n";
408e5dd7070Spatrick       OS << "    return llvm::StringRef(" << getLowerName() << ", "
409e5dd7070Spatrick          << getLowerName() << "Length);\n";
410e5dd7070Spatrick       OS << "  }\n";
411e5dd7070Spatrick       OS << "  unsigned get" << getUpperName() << "Length() const {\n";
412e5dd7070Spatrick       OS << "    return " << getLowerName() << "Length;\n";
413e5dd7070Spatrick       OS << "  }\n";
414e5dd7070Spatrick       OS << "  void set" << getUpperName()
415e5dd7070Spatrick          << "(ASTContext &C, llvm::StringRef S) {\n";
416e5dd7070Spatrick       OS << "    " << getLowerName() << "Length = S.size();\n";
417e5dd7070Spatrick       OS << "    this->" << getLowerName() << " = new (C, 1) char ["
418e5dd7070Spatrick          << getLowerName() << "Length];\n";
419e5dd7070Spatrick       OS << "    if (!S.empty())\n";
420e5dd7070Spatrick       OS << "      std::memcpy(this->" << getLowerName() << ", S.data(), "
421e5dd7070Spatrick          << getLowerName() << "Length);\n";
422e5dd7070Spatrick       OS << "  }";
423e5dd7070Spatrick     }
424e5dd7070Spatrick 
writeCloneArgs(raw_ostream & OS) const425e5dd7070Spatrick     void writeCloneArgs(raw_ostream &OS) const override {
426e5dd7070Spatrick       OS << "get" << getUpperName() << "()";
427e5dd7070Spatrick     }
428e5dd7070Spatrick 
writeTemplateInstantiationArgs(raw_ostream & OS) const429e5dd7070Spatrick     void writeTemplateInstantiationArgs(raw_ostream &OS) const override {
430e5dd7070Spatrick       OS << "A->get" << getUpperName() << "()";
431e5dd7070Spatrick     }
432e5dd7070Spatrick 
writeCtorBody(raw_ostream & OS) const433e5dd7070Spatrick     void writeCtorBody(raw_ostream &OS) const override {
434e5dd7070Spatrick       OS << "    if (!" << getUpperName() << ".empty())\n";
435e5dd7070Spatrick       OS << "      std::memcpy(" << getLowerName() << ", " << getUpperName()
436e5dd7070Spatrick          << ".data(), " << getLowerName() << "Length);\n";
437e5dd7070Spatrick     }
438e5dd7070Spatrick 
writeCtorInitializers(raw_ostream & OS) const439e5dd7070Spatrick     void writeCtorInitializers(raw_ostream &OS) const override {
440e5dd7070Spatrick       OS << getLowerName() << "Length(" << getUpperName() << ".size()),"
441e5dd7070Spatrick          << getLowerName() << "(new (Ctx, 1) char[" << getLowerName()
442e5dd7070Spatrick          << "Length])";
443e5dd7070Spatrick     }
444e5dd7070Spatrick 
writeCtorDefaultInitializers(raw_ostream & OS) const445e5dd7070Spatrick     void writeCtorDefaultInitializers(raw_ostream &OS) const override {
446e5dd7070Spatrick       OS << getLowerName() << "Length(0)," << getLowerName() << "(nullptr)";
447e5dd7070Spatrick     }
448e5dd7070Spatrick 
writeCtorParameters(raw_ostream & OS) const449e5dd7070Spatrick     void writeCtorParameters(raw_ostream &OS) const override {
450e5dd7070Spatrick       OS << "llvm::StringRef " << getUpperName();
451e5dd7070Spatrick     }
452e5dd7070Spatrick 
writeDeclarations(raw_ostream & OS) const453e5dd7070Spatrick     void writeDeclarations(raw_ostream &OS) const override {
454e5dd7070Spatrick       OS << "unsigned " << getLowerName() << "Length;\n";
455e5dd7070Spatrick       OS << "char *" << getLowerName() << ";";
456e5dd7070Spatrick     }
457e5dd7070Spatrick 
writePCHReadDecls(raw_ostream & OS) const458e5dd7070Spatrick     void writePCHReadDecls(raw_ostream &OS) const override {
459e5dd7070Spatrick       OS << "    std::string " << getLowerName()
460e5dd7070Spatrick          << "= Record.readString();\n";
461e5dd7070Spatrick     }
462e5dd7070Spatrick 
writePCHReadArgs(raw_ostream & OS) const463e5dd7070Spatrick     void writePCHReadArgs(raw_ostream &OS) const override {
464e5dd7070Spatrick       OS << getLowerName();
465e5dd7070Spatrick     }
466e5dd7070Spatrick 
writePCHWrite(raw_ostream & OS) const467e5dd7070Spatrick     void writePCHWrite(raw_ostream &OS) const override {
468e5dd7070Spatrick       OS << "    Record.AddString(SA->get" << getUpperName() << "());\n";
469e5dd7070Spatrick     }
470e5dd7070Spatrick 
writeValue(raw_ostream & OS) const471e5dd7070Spatrick     void writeValue(raw_ostream &OS) const override {
472e5dd7070Spatrick       OS << "\\\"\" << get" << getUpperName() << "() << \"\\\"";
473e5dd7070Spatrick     }
474e5dd7070Spatrick 
writeDump(raw_ostream & OS) const475e5dd7070Spatrick     void writeDump(raw_ostream &OS) const override {
476e5dd7070Spatrick       OS << "    OS << \" \\\"\" << SA->get" << getUpperName()
477e5dd7070Spatrick          << "() << \"\\\"\";\n";
478e5dd7070Spatrick     }
479e5dd7070Spatrick   };
480e5dd7070Spatrick 
481e5dd7070Spatrick   class AlignedArgument : public Argument {
482e5dd7070Spatrick   public:
AlignedArgument(const Record & Arg,StringRef Attr)483e5dd7070Spatrick     AlignedArgument(const Record &Arg, StringRef Attr)
484e5dd7070Spatrick       : Argument(Arg, Attr)
485e5dd7070Spatrick     {}
486e5dd7070Spatrick 
writeAccessors(raw_ostream & OS) const487e5dd7070Spatrick     void writeAccessors(raw_ostream &OS) const override {
488e5dd7070Spatrick       OS << "  bool is" << getUpperName() << "Dependent() const;\n";
489ec727ea7Spatrick       OS << "  bool is" << getUpperName() << "ErrorDependent() const;\n";
490e5dd7070Spatrick 
491e5dd7070Spatrick       OS << "  unsigned get" << getUpperName() << "(ASTContext &Ctx) const;\n";
492e5dd7070Spatrick 
493e5dd7070Spatrick       OS << "  bool is" << getUpperName() << "Expr() const {\n";
494e5dd7070Spatrick       OS << "    return is" << getLowerName() << "Expr;\n";
495e5dd7070Spatrick       OS << "  }\n";
496e5dd7070Spatrick 
497e5dd7070Spatrick       OS << "  Expr *get" << getUpperName() << "Expr() const {\n";
498e5dd7070Spatrick       OS << "    assert(is" << getLowerName() << "Expr);\n";
499e5dd7070Spatrick       OS << "    return " << getLowerName() << "Expr;\n";
500e5dd7070Spatrick       OS << "  }\n";
501e5dd7070Spatrick 
502e5dd7070Spatrick       OS << "  TypeSourceInfo *get" << getUpperName() << "Type() const {\n";
503e5dd7070Spatrick       OS << "    assert(!is" << getLowerName() << "Expr);\n";
504e5dd7070Spatrick       OS << "    return " << getLowerName() << "Type;\n";
505e5dd7070Spatrick       OS << "  }";
506e5dd7070Spatrick     }
507e5dd7070Spatrick 
writeAccessorDefinitions(raw_ostream & OS) const508e5dd7070Spatrick     void writeAccessorDefinitions(raw_ostream &OS) const override {
509e5dd7070Spatrick       OS << "bool " << getAttrName() << "Attr::is" << getUpperName()
510e5dd7070Spatrick          << "Dependent() const {\n";
511e5dd7070Spatrick       OS << "  if (is" << getLowerName() << "Expr)\n";
512e5dd7070Spatrick       OS << "    return " << getLowerName() << "Expr && (" << getLowerName()
513e5dd7070Spatrick          << "Expr->isValueDependent() || " << getLowerName()
514e5dd7070Spatrick          << "Expr->isTypeDependent());\n";
515e5dd7070Spatrick       OS << "  else\n";
516e5dd7070Spatrick       OS << "    return " << getLowerName()
517e5dd7070Spatrick          << "Type->getType()->isDependentType();\n";
518e5dd7070Spatrick       OS << "}\n";
519e5dd7070Spatrick 
520ec727ea7Spatrick       OS << "bool " << getAttrName() << "Attr::is" << getUpperName()
521ec727ea7Spatrick          << "ErrorDependent() const {\n";
522ec727ea7Spatrick       OS << "  if (is" << getLowerName() << "Expr)\n";
523ec727ea7Spatrick       OS << "    return " << getLowerName() << "Expr && " << getLowerName()
524ec727ea7Spatrick          << "Expr->containsErrors();\n";
525ec727ea7Spatrick       OS << "  return " << getLowerName()
526ec727ea7Spatrick          << "Type->getType()->containsErrors();\n";
527ec727ea7Spatrick       OS << "}\n";
528ec727ea7Spatrick 
529e5dd7070Spatrick       // FIXME: Do not do the calculation here
530e5dd7070Spatrick       // FIXME: Handle types correctly
531e5dd7070Spatrick       // A null pointer means maximum alignment
532e5dd7070Spatrick       OS << "unsigned " << getAttrName() << "Attr::get" << getUpperName()
533e5dd7070Spatrick          << "(ASTContext &Ctx) const {\n";
534e5dd7070Spatrick       OS << "  assert(!is" << getUpperName() << "Dependent());\n";
535e5dd7070Spatrick       OS << "  if (is" << getLowerName() << "Expr)\n";
536e5dd7070Spatrick       OS << "    return " << getLowerName() << "Expr ? " << getLowerName()
537e5dd7070Spatrick          << "Expr->EvaluateKnownConstInt(Ctx).getZExtValue()"
538e5dd7070Spatrick          << " * Ctx.getCharWidth() : "
539e5dd7070Spatrick          << "Ctx.getTargetDefaultAlignForAttributeAligned();\n";
540e5dd7070Spatrick       OS << "  else\n";
541e5dd7070Spatrick       OS << "    return 0; // FIXME\n";
542e5dd7070Spatrick       OS << "}\n";
543e5dd7070Spatrick     }
544e5dd7070Spatrick 
writeASTVisitorTraversal(raw_ostream & OS) const545e5dd7070Spatrick     void writeASTVisitorTraversal(raw_ostream &OS) const override {
546e5dd7070Spatrick       StringRef Name = getUpperName();
547e5dd7070Spatrick       OS << "  if (A->is" << Name << "Expr()) {\n"
548e5dd7070Spatrick          << "    if (!getDerived().TraverseStmt(A->get" << Name << "Expr()))\n"
549e5dd7070Spatrick          << "      return false;\n"
550e5dd7070Spatrick          << "  } else if (auto *TSI = A->get" << Name << "Type()) {\n"
551e5dd7070Spatrick          << "    if (!getDerived().TraverseTypeLoc(TSI->getTypeLoc()))\n"
552e5dd7070Spatrick          << "      return false;\n"
553e5dd7070Spatrick          << "  }\n";
554e5dd7070Spatrick     }
555e5dd7070Spatrick 
writeCloneArgs(raw_ostream & OS) const556e5dd7070Spatrick     void writeCloneArgs(raw_ostream &OS) const override {
557e5dd7070Spatrick       OS << "is" << getLowerName() << "Expr, is" << getLowerName()
558e5dd7070Spatrick          << "Expr ? static_cast<void*>(" << getLowerName()
559e5dd7070Spatrick          << "Expr) : " << getLowerName()
560e5dd7070Spatrick          << "Type";
561e5dd7070Spatrick     }
562e5dd7070Spatrick 
writeTemplateInstantiationArgs(raw_ostream & OS) const563e5dd7070Spatrick     void writeTemplateInstantiationArgs(raw_ostream &OS) const override {
564e5dd7070Spatrick       // FIXME: move the definition in Sema::InstantiateAttrs to here.
565e5dd7070Spatrick       // In the meantime, aligned attributes are cloned.
566e5dd7070Spatrick     }
567e5dd7070Spatrick 
writeCtorBody(raw_ostream & OS) const568e5dd7070Spatrick     void writeCtorBody(raw_ostream &OS) const override {
569e5dd7070Spatrick       OS << "    if (is" << getLowerName() << "Expr)\n";
570e5dd7070Spatrick       OS << "       " << getLowerName() << "Expr = reinterpret_cast<Expr *>("
571e5dd7070Spatrick          << getUpperName() << ");\n";
572e5dd7070Spatrick       OS << "    else\n";
573e5dd7070Spatrick       OS << "       " << getLowerName()
574e5dd7070Spatrick          << "Type = reinterpret_cast<TypeSourceInfo *>(" << getUpperName()
575e5dd7070Spatrick          << ");\n";
576e5dd7070Spatrick     }
577e5dd7070Spatrick 
writeCtorInitializers(raw_ostream & OS) const578e5dd7070Spatrick     void writeCtorInitializers(raw_ostream &OS) const override {
579e5dd7070Spatrick       OS << "is" << getLowerName() << "Expr(Is" << getUpperName() << "Expr)";
580e5dd7070Spatrick     }
581e5dd7070Spatrick 
writeCtorDefaultInitializers(raw_ostream & OS) const582e5dd7070Spatrick     void writeCtorDefaultInitializers(raw_ostream &OS) const override {
583e5dd7070Spatrick       OS << "is" << getLowerName() << "Expr(false)";
584e5dd7070Spatrick     }
585e5dd7070Spatrick 
writeCtorParameters(raw_ostream & OS) const586e5dd7070Spatrick     void writeCtorParameters(raw_ostream &OS) const override {
587e5dd7070Spatrick       OS << "bool Is" << getUpperName() << "Expr, void *" << getUpperName();
588e5dd7070Spatrick     }
589e5dd7070Spatrick 
writeImplicitCtorArgs(raw_ostream & OS) const590e5dd7070Spatrick     void writeImplicitCtorArgs(raw_ostream &OS) const override {
591e5dd7070Spatrick       OS << "Is" << getUpperName() << "Expr, " << getUpperName();
592e5dd7070Spatrick     }
593e5dd7070Spatrick 
writeDeclarations(raw_ostream & OS) const594e5dd7070Spatrick     void writeDeclarations(raw_ostream &OS) const override {
595e5dd7070Spatrick       OS << "bool is" << getLowerName() << "Expr;\n";
596e5dd7070Spatrick       OS << "union {\n";
597e5dd7070Spatrick       OS << "Expr *" << getLowerName() << "Expr;\n";
598e5dd7070Spatrick       OS << "TypeSourceInfo *" << getLowerName() << "Type;\n";
599e5dd7070Spatrick       OS << "};";
600e5dd7070Spatrick     }
601e5dd7070Spatrick 
writePCHReadArgs(raw_ostream & OS) const602e5dd7070Spatrick     void writePCHReadArgs(raw_ostream &OS) const override {
603e5dd7070Spatrick       OS << "is" << getLowerName() << "Expr, " << getLowerName() << "Ptr";
604e5dd7070Spatrick     }
605e5dd7070Spatrick 
writePCHReadDecls(raw_ostream & OS) const606e5dd7070Spatrick     void writePCHReadDecls(raw_ostream &OS) const override {
607e5dd7070Spatrick       OS << "    bool is" << getLowerName() << "Expr = Record.readInt();\n";
608e5dd7070Spatrick       OS << "    void *" << getLowerName() << "Ptr;\n";
609e5dd7070Spatrick       OS << "    if (is" << getLowerName() << "Expr)\n";
610e5dd7070Spatrick       OS << "      " << getLowerName() << "Ptr = Record.readExpr();\n";
611e5dd7070Spatrick       OS << "    else\n";
612e5dd7070Spatrick       OS << "      " << getLowerName()
613e5dd7070Spatrick          << "Ptr = Record.readTypeSourceInfo();\n";
614e5dd7070Spatrick     }
615e5dd7070Spatrick 
writePCHWrite(raw_ostream & OS) const616e5dd7070Spatrick     void writePCHWrite(raw_ostream &OS) const override {
617e5dd7070Spatrick       OS << "    Record.push_back(SA->is" << getUpperName() << "Expr());\n";
618e5dd7070Spatrick       OS << "    if (SA->is" << getUpperName() << "Expr())\n";
619e5dd7070Spatrick       OS << "      Record.AddStmt(SA->get" << getUpperName() << "Expr());\n";
620e5dd7070Spatrick       OS << "    else\n";
621e5dd7070Spatrick       OS << "      Record.AddTypeSourceInfo(SA->get" << getUpperName()
622e5dd7070Spatrick          << "Type());\n";
623e5dd7070Spatrick     }
624e5dd7070Spatrick 
getIsOmitted() const625e5dd7070Spatrick     std::string getIsOmitted() const override {
626e5dd7070Spatrick       return "!is" + getLowerName().str() + "Expr || !" + getLowerName().str()
627e5dd7070Spatrick              + "Expr";
628e5dd7070Spatrick     }
629e5dd7070Spatrick 
writeValue(raw_ostream & OS) const630e5dd7070Spatrick     void writeValue(raw_ostream &OS) const override {
631e5dd7070Spatrick       OS << "\";\n";
632e5dd7070Spatrick       OS << "    " << getLowerName()
633e5dd7070Spatrick          << "Expr->printPretty(OS, nullptr, Policy);\n";
634e5dd7070Spatrick       OS << "    OS << \"";
635e5dd7070Spatrick     }
636e5dd7070Spatrick 
writeDump(raw_ostream & OS) const637e5dd7070Spatrick     void writeDump(raw_ostream &OS) const override {
638e5dd7070Spatrick       OS << "    if (!SA->is" << getUpperName() << "Expr())\n";
639e5dd7070Spatrick       OS << "      dumpType(SA->get" << getUpperName()
640e5dd7070Spatrick          << "Type()->getType());\n";
641e5dd7070Spatrick     }
642e5dd7070Spatrick 
writeDumpChildren(raw_ostream & OS) const643e5dd7070Spatrick     void writeDumpChildren(raw_ostream &OS) const override {
644e5dd7070Spatrick       OS << "    if (SA->is" << getUpperName() << "Expr())\n";
645e5dd7070Spatrick       OS << "      Visit(SA->get" << getUpperName() << "Expr());\n";
646e5dd7070Spatrick     }
647e5dd7070Spatrick 
writeHasChildren(raw_ostream & OS) const648e5dd7070Spatrick     void writeHasChildren(raw_ostream &OS) const override {
649e5dd7070Spatrick       OS << "SA->is" << getUpperName() << "Expr()";
650e5dd7070Spatrick     }
651e5dd7070Spatrick   };
652e5dd7070Spatrick 
653e5dd7070Spatrick   class VariadicArgument : public Argument {
654e5dd7070Spatrick     std::string Type, ArgName, ArgSizeName, RangeName;
655e5dd7070Spatrick 
656e5dd7070Spatrick   protected:
657e5dd7070Spatrick     // Assumed to receive a parameter: raw_ostream OS.
writeValueImpl(raw_ostream & OS) const658e5dd7070Spatrick     virtual void writeValueImpl(raw_ostream &OS) const {
659e5dd7070Spatrick       OS << "    OS << Val;\n";
660e5dd7070Spatrick     }
661e5dd7070Spatrick     // Assumed to receive a parameter: raw_ostream OS.
writeDumpImpl(raw_ostream & OS) const662e5dd7070Spatrick     virtual void writeDumpImpl(raw_ostream &OS) const {
663e5dd7070Spatrick       OS << "      OS << \" \" << Val;\n";
664e5dd7070Spatrick     }
665e5dd7070Spatrick 
666e5dd7070Spatrick   public:
VariadicArgument(const Record & Arg,StringRef Attr,std::string T)667e5dd7070Spatrick     VariadicArgument(const Record &Arg, StringRef Attr, std::string T)
668e5dd7070Spatrick         : Argument(Arg, Attr), Type(std::move(T)),
669e5dd7070Spatrick           ArgName(getLowerName().str() + "_"), ArgSizeName(ArgName + "Size"),
670ec727ea7Spatrick           RangeName(std::string(getLowerName())) {}
671e5dd7070Spatrick 
VariadicArgument(StringRef Arg,StringRef Attr,std::string T)672*12c85518Srobert     VariadicArgument(StringRef Arg, StringRef Attr, std::string T)
673*12c85518Srobert         : Argument(Arg, Attr), Type(std::move(T)),
674*12c85518Srobert           ArgName(getLowerName().str() + "_"), ArgSizeName(ArgName + "Size"),
675*12c85518Srobert           RangeName(std::string(getLowerName())) {}
676*12c85518Srobert 
getType() const677e5dd7070Spatrick     const std::string &getType() const { return Type; }
getArgName() const678e5dd7070Spatrick     const std::string &getArgName() const { return ArgName; }
getArgSizeName() const679e5dd7070Spatrick     const std::string &getArgSizeName() const { return ArgSizeName; }
isVariadic() const680e5dd7070Spatrick     bool isVariadic() const override { return true; }
681e5dd7070Spatrick 
writeAccessors(raw_ostream & OS) const682e5dd7070Spatrick     void writeAccessors(raw_ostream &OS) const override {
683e5dd7070Spatrick       std::string IteratorType = getLowerName().str() + "_iterator";
684e5dd7070Spatrick       std::string BeginFn = getLowerName().str() + "_begin()";
685e5dd7070Spatrick       std::string EndFn = getLowerName().str() + "_end()";
686e5dd7070Spatrick 
687e5dd7070Spatrick       OS << "  typedef " << Type << "* " << IteratorType << ";\n";
688e5dd7070Spatrick       OS << "  " << IteratorType << " " << BeginFn << " const {"
689e5dd7070Spatrick          << " return " << ArgName << "; }\n";
690e5dd7070Spatrick       OS << "  " << IteratorType << " " << EndFn << " const {"
691e5dd7070Spatrick          << " return " << ArgName << " + " << ArgSizeName << "; }\n";
692e5dd7070Spatrick       OS << "  unsigned " << getLowerName() << "_size() const {"
693e5dd7070Spatrick          << " return " << ArgSizeName << "; }\n";
694e5dd7070Spatrick       OS << "  llvm::iterator_range<" << IteratorType << "> " << RangeName
695e5dd7070Spatrick          << "() const { return llvm::make_range(" << BeginFn << ", " << EndFn
696e5dd7070Spatrick          << "); }\n";
697e5dd7070Spatrick     }
698e5dd7070Spatrick 
writeSetter(raw_ostream & OS) const699*12c85518Srobert     void writeSetter(raw_ostream &OS) const {
700*12c85518Srobert       OS << "  void set" << getUpperName() << "(ASTContext &Ctx, ";
701*12c85518Srobert       writeCtorParameters(OS);
702*12c85518Srobert       OS << ") {\n";
703*12c85518Srobert       OS << "    " << ArgSizeName << " = " << getUpperName() << "Size;\n";
704*12c85518Srobert       OS << "    " << ArgName << " = new (Ctx, 16) " << getType() << "["
705*12c85518Srobert          << ArgSizeName << "];\n";
706*12c85518Srobert       OS << "  ";
707*12c85518Srobert       writeCtorBody(OS);
708*12c85518Srobert       OS << "  }\n";
709*12c85518Srobert     }
710*12c85518Srobert 
writeCloneArgs(raw_ostream & OS) const711e5dd7070Spatrick     void writeCloneArgs(raw_ostream &OS) const override {
712e5dd7070Spatrick       OS << ArgName << ", " << ArgSizeName;
713e5dd7070Spatrick     }
714e5dd7070Spatrick 
writeTemplateInstantiationArgs(raw_ostream & OS) const715e5dd7070Spatrick     void writeTemplateInstantiationArgs(raw_ostream &OS) const override {
716e5dd7070Spatrick       // This isn't elegant, but we have to go through public methods...
717e5dd7070Spatrick       OS << "A->" << getLowerName() << "_begin(), "
718e5dd7070Spatrick          << "A->" << getLowerName() << "_size()";
719e5dd7070Spatrick     }
720e5dd7070Spatrick 
writeASTVisitorTraversal(raw_ostream & OS) const721e5dd7070Spatrick     void writeASTVisitorTraversal(raw_ostream &OS) const override {
722e5dd7070Spatrick       // FIXME: Traverse the elements.
723e5dd7070Spatrick     }
724e5dd7070Spatrick 
writeCtorBody(raw_ostream & OS) const725e5dd7070Spatrick     void writeCtorBody(raw_ostream &OS) const override {
726ec727ea7Spatrick       OS << "  std::copy(" << getUpperName() << ", " << getUpperName() << " + "
727ec727ea7Spatrick          << ArgSizeName << ", " << ArgName << ");\n";
728e5dd7070Spatrick     }
729e5dd7070Spatrick 
writeCtorInitializers(raw_ostream & OS) const730e5dd7070Spatrick     void writeCtorInitializers(raw_ostream &OS) const override {
731e5dd7070Spatrick       OS << ArgSizeName << "(" << getUpperName() << "Size), "
732e5dd7070Spatrick          << ArgName << "(new (Ctx, 16) " << getType() << "["
733e5dd7070Spatrick          << ArgSizeName << "])";
734e5dd7070Spatrick     }
735e5dd7070Spatrick 
writeCtorDefaultInitializers(raw_ostream & OS) const736e5dd7070Spatrick     void writeCtorDefaultInitializers(raw_ostream &OS) const override {
737e5dd7070Spatrick       OS << ArgSizeName << "(0), " << ArgName << "(nullptr)";
738e5dd7070Spatrick     }
739e5dd7070Spatrick 
writeCtorParameters(raw_ostream & OS) const740e5dd7070Spatrick     void writeCtorParameters(raw_ostream &OS) const override {
741e5dd7070Spatrick       OS << getType() << " *" << getUpperName() << ", unsigned "
742e5dd7070Spatrick          << getUpperName() << "Size";
743e5dd7070Spatrick     }
744e5dd7070Spatrick 
writeImplicitCtorArgs(raw_ostream & OS) const745e5dd7070Spatrick     void writeImplicitCtorArgs(raw_ostream &OS) const override {
746e5dd7070Spatrick       OS << getUpperName() << ", " << getUpperName() << "Size";
747e5dd7070Spatrick     }
748e5dd7070Spatrick 
writeDeclarations(raw_ostream & OS) const749e5dd7070Spatrick     void writeDeclarations(raw_ostream &OS) const override {
750e5dd7070Spatrick       OS << "  unsigned " << ArgSizeName << ";\n";
751e5dd7070Spatrick       OS << "  " << getType() << " *" << ArgName << ";";
752e5dd7070Spatrick     }
753e5dd7070Spatrick 
writePCHReadDecls(raw_ostream & OS) const754e5dd7070Spatrick     void writePCHReadDecls(raw_ostream &OS) const override {
755e5dd7070Spatrick       OS << "    unsigned " << getLowerName() << "Size = Record.readInt();\n";
756e5dd7070Spatrick       OS << "    SmallVector<" << getType() << ", 4> "
757e5dd7070Spatrick          << getLowerName() << ";\n";
758e5dd7070Spatrick       OS << "    " << getLowerName() << ".reserve(" << getLowerName()
759e5dd7070Spatrick          << "Size);\n";
760e5dd7070Spatrick 
761e5dd7070Spatrick       // If we can't store the values in the current type (if it's something
762e5dd7070Spatrick       // like StringRef), store them in a different type and convert the
763e5dd7070Spatrick       // container afterwards.
764ec727ea7Spatrick       std::string StorageType = std::string(getStorageType(getType()));
765ec727ea7Spatrick       std::string StorageName = std::string(getLowerName());
766e5dd7070Spatrick       if (StorageType != getType()) {
767e5dd7070Spatrick         StorageName += "Storage";
768e5dd7070Spatrick         OS << "    SmallVector<" << StorageType << ", 4> "
769e5dd7070Spatrick            << StorageName << ";\n";
770e5dd7070Spatrick         OS << "    " << StorageName << ".reserve(" << getLowerName()
771e5dd7070Spatrick            << "Size);\n";
772e5dd7070Spatrick       }
773e5dd7070Spatrick 
774e5dd7070Spatrick       OS << "    for (unsigned i = 0; i != " << getLowerName() << "Size; ++i)\n";
775e5dd7070Spatrick       std::string read = ReadPCHRecord(Type);
776e5dd7070Spatrick       OS << "      " << StorageName << ".push_back(" << read << ");\n";
777e5dd7070Spatrick 
778e5dd7070Spatrick       if (StorageType != getType()) {
779e5dd7070Spatrick         OS << "    for (unsigned i = 0; i != " << getLowerName() << "Size; ++i)\n";
780e5dd7070Spatrick         OS << "      " << getLowerName() << ".push_back("
781e5dd7070Spatrick            << StorageName << "[i]);\n";
782e5dd7070Spatrick       }
783e5dd7070Spatrick     }
784e5dd7070Spatrick 
writePCHReadArgs(raw_ostream & OS) const785e5dd7070Spatrick     void writePCHReadArgs(raw_ostream &OS) const override {
786e5dd7070Spatrick       OS << getLowerName() << ".data(), " << getLowerName() << "Size";
787e5dd7070Spatrick     }
788e5dd7070Spatrick 
writePCHWrite(raw_ostream & OS) const789e5dd7070Spatrick     void writePCHWrite(raw_ostream &OS) const override {
790e5dd7070Spatrick       OS << "    Record.push_back(SA->" << getLowerName() << "_size());\n";
791e5dd7070Spatrick       OS << "    for (auto &Val : SA->" << RangeName << "())\n";
792e5dd7070Spatrick       OS << "      " << WritePCHRecord(Type, "Val");
793e5dd7070Spatrick     }
794e5dd7070Spatrick 
writeValue(raw_ostream & OS) const795e5dd7070Spatrick     void writeValue(raw_ostream &OS) const override {
796e5dd7070Spatrick       OS << "\";\n";
797a9ac8606Spatrick       OS << "  for (const auto &Val : " << RangeName << "()) {\n"
798a9ac8606Spatrick          << "    DelimitAttributeArgument(OS, IsFirstArgument);\n";
799e5dd7070Spatrick       writeValueImpl(OS);
800e5dd7070Spatrick       OS << "  }\n";
801e5dd7070Spatrick       OS << "  OS << \"";
802e5dd7070Spatrick     }
803e5dd7070Spatrick 
writeDump(raw_ostream & OS) const804e5dd7070Spatrick     void writeDump(raw_ostream &OS) const override {
805e5dd7070Spatrick       OS << "    for (const auto &Val : SA->" << RangeName << "())\n";
806e5dd7070Spatrick       writeDumpImpl(OS);
807e5dd7070Spatrick     }
808e5dd7070Spatrick   };
809e5dd7070Spatrick 
810*12c85518Srobert   class VariadicOMPInteropInfoArgument : public VariadicArgument {
811*12c85518Srobert   public:
VariadicOMPInteropInfoArgument(const Record & Arg,StringRef Attr)812*12c85518Srobert     VariadicOMPInteropInfoArgument(const Record &Arg, StringRef Attr)
813*12c85518Srobert         : VariadicArgument(Arg, Attr, "OMPInteropInfo") {}
814*12c85518Srobert 
writeDump(raw_ostream & OS) const815*12c85518Srobert     void writeDump(raw_ostream &OS) const override {
816*12c85518Srobert       OS << "    for (" << getAttrName() << "Attr::" << getLowerName()
817*12c85518Srobert          << "_iterator I = SA->" << getLowerName() << "_begin(), E = SA->"
818*12c85518Srobert          << getLowerName() << "_end(); I != E; ++I) {\n";
819*12c85518Srobert       OS << "      if (I->IsTarget && I->IsTargetSync)\n";
820*12c85518Srobert       OS << "        OS << \" Target_TargetSync\";\n";
821*12c85518Srobert       OS << "      else if (I->IsTarget)\n";
822*12c85518Srobert       OS << "        OS << \" Target\";\n";
823*12c85518Srobert       OS << "      else\n";
824*12c85518Srobert       OS << "        OS << \" TargetSync\";\n";
825*12c85518Srobert       OS << "    }\n";
826*12c85518Srobert     }
827*12c85518Srobert 
writePCHReadDecls(raw_ostream & OS) const828*12c85518Srobert     void writePCHReadDecls(raw_ostream &OS) const override {
829*12c85518Srobert       OS << "    unsigned " << getLowerName() << "Size = Record.readInt();\n";
830*12c85518Srobert       OS << "    SmallVector<OMPInteropInfo, 4> " << getLowerName() << ";\n";
831*12c85518Srobert       OS << "    " << getLowerName() << ".reserve(" << getLowerName()
832*12c85518Srobert          << "Size);\n";
833*12c85518Srobert       OS << "    for (unsigned I = 0, E = " << getLowerName() << "Size; ";
834*12c85518Srobert       OS << "I != E; ++I) {\n";
835*12c85518Srobert       OS << "      bool IsTarget = Record.readBool();\n";
836*12c85518Srobert       OS << "      bool IsTargetSync = Record.readBool();\n";
837*12c85518Srobert       OS << "      " << getLowerName()
838*12c85518Srobert          << ".emplace_back(IsTarget, IsTargetSync);\n";
839*12c85518Srobert       OS << "    }\n";
840*12c85518Srobert     }
841*12c85518Srobert 
writePCHWrite(raw_ostream & OS) const842*12c85518Srobert     void writePCHWrite(raw_ostream &OS) const override {
843*12c85518Srobert       OS << "    Record.push_back(SA->" << getLowerName() << "_size());\n";
844*12c85518Srobert       OS << "    for (" << getAttrName() << "Attr::" << getLowerName()
845*12c85518Srobert          << "_iterator I = SA->" << getLowerName() << "_begin(), E = SA->"
846*12c85518Srobert          << getLowerName() << "_end(); I != E; ++I) {\n";
847*12c85518Srobert       OS << "      Record.writeBool(I->IsTarget);\n";
848*12c85518Srobert       OS << "      Record.writeBool(I->IsTargetSync);\n";
849*12c85518Srobert       OS << "    }\n";
850*12c85518Srobert     }
851*12c85518Srobert   };
852*12c85518Srobert 
853e5dd7070Spatrick   class VariadicParamIdxArgument : public VariadicArgument {
854e5dd7070Spatrick   public:
VariadicParamIdxArgument(const Record & Arg,StringRef Attr)855e5dd7070Spatrick     VariadicParamIdxArgument(const Record &Arg, StringRef Attr)
856e5dd7070Spatrick         : VariadicArgument(Arg, Attr, "ParamIdx") {}
857e5dd7070Spatrick 
858e5dd7070Spatrick   public:
writeValueImpl(raw_ostream & OS) const859e5dd7070Spatrick     void writeValueImpl(raw_ostream &OS) const override {
860e5dd7070Spatrick       OS << "    OS << Val.getSourceIndex();\n";
861e5dd7070Spatrick     }
862e5dd7070Spatrick 
writeDumpImpl(raw_ostream & OS) const863e5dd7070Spatrick     void writeDumpImpl(raw_ostream &OS) const override {
864e5dd7070Spatrick       OS << "      OS << \" \" << Val.getSourceIndex();\n";
865e5dd7070Spatrick     }
866e5dd7070Spatrick   };
867e5dd7070Spatrick 
868e5dd7070Spatrick   struct VariadicParamOrParamIdxArgument : public VariadicArgument {
VariadicParamOrParamIdxArgument__anone86b54960211::VariadicParamOrParamIdxArgument869e5dd7070Spatrick     VariadicParamOrParamIdxArgument(const Record &Arg, StringRef Attr)
870e5dd7070Spatrick         : VariadicArgument(Arg, Attr, "int") {}
871e5dd7070Spatrick   };
872e5dd7070Spatrick 
873e5dd7070Spatrick   // Unique the enums, but maintain the original declaration ordering.
874e5dd7070Spatrick   std::vector<StringRef>
uniqueEnumsInOrder(const std::vector<StringRef> & enums)875e5dd7070Spatrick   uniqueEnumsInOrder(const std::vector<StringRef> &enums) {
876e5dd7070Spatrick     std::vector<StringRef> uniques;
877e5dd7070Spatrick     SmallDenseSet<StringRef, 8> unique_set;
878e5dd7070Spatrick     for (const auto &i : enums) {
879e5dd7070Spatrick       if (unique_set.insert(i).second)
880e5dd7070Spatrick         uniques.push_back(i);
881e5dd7070Spatrick     }
882e5dd7070Spatrick     return uniques;
883e5dd7070Spatrick   }
884e5dd7070Spatrick 
885e5dd7070Spatrick   class EnumArgument : public Argument {
886e5dd7070Spatrick     std::string type;
887e5dd7070Spatrick     std::vector<StringRef> values, enums, uniques;
888e5dd7070Spatrick 
889e5dd7070Spatrick   public:
EnumArgument(const Record & Arg,StringRef Attr)890e5dd7070Spatrick     EnumArgument(const Record &Arg, StringRef Attr)
891ec727ea7Spatrick         : Argument(Arg, Attr), type(std::string(Arg.getValueAsString("Type"))),
892e5dd7070Spatrick           values(Arg.getValueAsListOfStrings("Values")),
893e5dd7070Spatrick           enums(Arg.getValueAsListOfStrings("Enums")),
894ec727ea7Spatrick           uniques(uniqueEnumsInOrder(enums)) {
895e5dd7070Spatrick       // FIXME: Emit a proper error
896e5dd7070Spatrick       assert(!uniques.empty());
897e5dd7070Spatrick     }
898e5dd7070Spatrick 
isEnumArg() const899e5dd7070Spatrick     bool isEnumArg() const override { return true; }
900e5dd7070Spatrick 
writeAccessors(raw_ostream & OS) const901e5dd7070Spatrick     void writeAccessors(raw_ostream &OS) const override {
902e5dd7070Spatrick       OS << "  " << type << " get" << getUpperName() << "() const {\n";
903e5dd7070Spatrick       OS << "    return " << getLowerName() << ";\n";
904e5dd7070Spatrick       OS << "  }";
905e5dd7070Spatrick     }
906e5dd7070Spatrick 
writeCloneArgs(raw_ostream & OS) const907e5dd7070Spatrick     void writeCloneArgs(raw_ostream &OS) const override {
908e5dd7070Spatrick       OS << getLowerName();
909e5dd7070Spatrick     }
910e5dd7070Spatrick 
writeTemplateInstantiationArgs(raw_ostream & OS) const911e5dd7070Spatrick     void writeTemplateInstantiationArgs(raw_ostream &OS) const override {
912e5dd7070Spatrick       OS << "A->get" << getUpperName() << "()";
913e5dd7070Spatrick     }
writeCtorInitializers(raw_ostream & OS) const914e5dd7070Spatrick     void writeCtorInitializers(raw_ostream &OS) const override {
915e5dd7070Spatrick       OS << getLowerName() << "(" << getUpperName() << ")";
916e5dd7070Spatrick     }
writeCtorDefaultInitializers(raw_ostream & OS) const917e5dd7070Spatrick     void writeCtorDefaultInitializers(raw_ostream &OS) const override {
918e5dd7070Spatrick       OS << getLowerName() << "(" << type << "(0))";
919e5dd7070Spatrick     }
writeCtorParameters(raw_ostream & OS) const920e5dd7070Spatrick     void writeCtorParameters(raw_ostream &OS) const override {
921e5dd7070Spatrick       OS << type << " " << getUpperName();
922e5dd7070Spatrick     }
writeDeclarations(raw_ostream & OS) const923e5dd7070Spatrick     void writeDeclarations(raw_ostream &OS) const override {
924e5dd7070Spatrick       auto i = uniques.cbegin(), e = uniques.cend();
925e5dd7070Spatrick       // The last one needs to not have a comma.
926e5dd7070Spatrick       --e;
927e5dd7070Spatrick 
928e5dd7070Spatrick       OS << "public:\n";
929e5dd7070Spatrick       OS << "  enum " << type << " {\n";
930e5dd7070Spatrick       for (; i != e; ++i)
931e5dd7070Spatrick         OS << "    " << *i << ",\n";
932e5dd7070Spatrick       OS << "    " << *e << "\n";
933e5dd7070Spatrick       OS << "  };\n";
934e5dd7070Spatrick       OS << "private:\n";
935e5dd7070Spatrick       OS << "  " << type << " " << getLowerName() << ";";
936e5dd7070Spatrick     }
937e5dd7070Spatrick 
writePCHReadDecls(raw_ostream & OS) const938e5dd7070Spatrick     void writePCHReadDecls(raw_ostream &OS) const override {
939e5dd7070Spatrick       OS << "    " << getAttrName() << "Attr::" << type << " " << getLowerName()
940e5dd7070Spatrick          << "(static_cast<" << getAttrName() << "Attr::" << type
941e5dd7070Spatrick          << ">(Record.readInt()));\n";
942e5dd7070Spatrick     }
943e5dd7070Spatrick 
writePCHReadArgs(raw_ostream & OS) const944e5dd7070Spatrick     void writePCHReadArgs(raw_ostream &OS) const override {
945e5dd7070Spatrick       OS << getLowerName();
946e5dd7070Spatrick     }
947e5dd7070Spatrick 
writePCHWrite(raw_ostream & OS) const948e5dd7070Spatrick     void writePCHWrite(raw_ostream &OS) const override {
949e5dd7070Spatrick       OS << "Record.push_back(SA->get" << getUpperName() << "());\n";
950e5dd7070Spatrick     }
951e5dd7070Spatrick 
writeValue(raw_ostream & OS) const952e5dd7070Spatrick     void writeValue(raw_ostream &OS) const override {
953e5dd7070Spatrick       // FIXME: this isn't 100% correct -- some enum arguments require printing
954e5dd7070Spatrick       // as a string literal, while others require printing as an identifier.
955e5dd7070Spatrick       // Tablegen currently does not distinguish between the two forms.
956e5dd7070Spatrick       OS << "\\\"\" << " << getAttrName() << "Attr::Convert" << type << "ToStr(get"
957e5dd7070Spatrick          << getUpperName() << "()) << \"\\\"";
958e5dd7070Spatrick     }
959e5dd7070Spatrick 
writeDump(raw_ostream & OS) const960e5dd7070Spatrick     void writeDump(raw_ostream &OS) const override {
961e5dd7070Spatrick       OS << "    switch(SA->get" << getUpperName() << "()) {\n";
962e5dd7070Spatrick       for (const auto &I : uniques) {
963e5dd7070Spatrick         OS << "    case " << getAttrName() << "Attr::" << I << ":\n";
964e5dd7070Spatrick         OS << "      OS << \" " << I << "\";\n";
965e5dd7070Spatrick         OS << "      break;\n";
966e5dd7070Spatrick       }
967e5dd7070Spatrick       OS << "    }\n";
968e5dd7070Spatrick     }
969e5dd7070Spatrick 
writeConversion(raw_ostream & OS,bool Header) const970ec727ea7Spatrick     void writeConversion(raw_ostream &OS, bool Header) const {
971ec727ea7Spatrick       if (Header) {
972ec727ea7Spatrick         OS << "  static bool ConvertStrTo" << type << "(StringRef Val, " << type
973ec727ea7Spatrick            << " &Out);\n";
974ec727ea7Spatrick         OS << "  static const char *Convert" << type << "ToStr(" << type
975ec727ea7Spatrick            << " Val);\n";
976ec727ea7Spatrick         return;
977ec727ea7Spatrick       }
978ec727ea7Spatrick 
979ec727ea7Spatrick       OS << "bool " << getAttrName() << "Attr::ConvertStrTo" << type
980ec727ea7Spatrick          << "(StringRef Val, " << type << " &Out) {\n";
981*12c85518Srobert       OS << "  std::optional<" << type
982*12c85518Srobert          << "> R = llvm::StringSwitch<std::optional<";
983e5dd7070Spatrick       OS << type << ">>(Val)\n";
984e5dd7070Spatrick       for (size_t I = 0; I < enums.size(); ++I) {
985e5dd7070Spatrick         OS << "    .Case(\"" << values[I] << "\", ";
986e5dd7070Spatrick         OS << getAttrName() << "Attr::" << enums[I] << ")\n";
987e5dd7070Spatrick       }
988*12c85518Srobert       OS << "    .Default(std::optional<" << type << ">());\n";
989e5dd7070Spatrick       OS << "  if (R) {\n";
990e5dd7070Spatrick       OS << "    Out = *R;\n      return true;\n    }\n";
991e5dd7070Spatrick       OS << "  return false;\n";
992e5dd7070Spatrick       OS << "}\n\n";
993e5dd7070Spatrick 
994e5dd7070Spatrick       // Mapping from enumeration values back to enumeration strings isn't
995e5dd7070Spatrick       // trivial because some enumeration values have multiple named
996e5dd7070Spatrick       // enumerators, such as type_visibility(internal) and
997e5dd7070Spatrick       // type_visibility(hidden) both mapping to TypeVisibilityAttr::Hidden.
998ec727ea7Spatrick       OS << "const char *" << getAttrName() << "Attr::Convert" << type
999ec727ea7Spatrick          << "ToStr(" << type << " Val) {\n"
1000e5dd7070Spatrick          << "  switch(Val) {\n";
1001e5dd7070Spatrick       SmallDenseSet<StringRef, 8> Uniques;
1002e5dd7070Spatrick       for (size_t I = 0; I < enums.size(); ++I) {
1003e5dd7070Spatrick         if (Uniques.insert(enums[I]).second)
1004e5dd7070Spatrick           OS << "  case " << getAttrName() << "Attr::" << enums[I]
1005e5dd7070Spatrick              << ": return \"" << values[I] << "\";\n";
1006e5dd7070Spatrick       }
1007e5dd7070Spatrick       OS << "  }\n"
1008e5dd7070Spatrick          << "  llvm_unreachable(\"No enumerator with that value\");\n"
1009e5dd7070Spatrick          << "}\n";
1010e5dd7070Spatrick     }
1011e5dd7070Spatrick   };
1012e5dd7070Spatrick 
1013e5dd7070Spatrick   class VariadicEnumArgument: public VariadicArgument {
1014e5dd7070Spatrick     std::string type, QualifiedTypeName;
1015e5dd7070Spatrick     std::vector<StringRef> values, enums, uniques;
1016e5dd7070Spatrick 
1017e5dd7070Spatrick   protected:
writeValueImpl(raw_ostream & OS) const1018e5dd7070Spatrick     void writeValueImpl(raw_ostream &OS) const override {
1019e5dd7070Spatrick       // FIXME: this isn't 100% correct -- some enum arguments require printing
1020e5dd7070Spatrick       // as a string literal, while others require printing as an identifier.
1021e5dd7070Spatrick       // Tablegen currently does not distinguish between the two forms.
1022e5dd7070Spatrick       OS << "    OS << \"\\\"\" << " << getAttrName() << "Attr::Convert" << type
1023e5dd7070Spatrick          << "ToStr(Val)" << "<< \"\\\"\";\n";
1024e5dd7070Spatrick     }
1025e5dd7070Spatrick 
1026e5dd7070Spatrick   public:
VariadicEnumArgument(const Record & Arg,StringRef Attr)1027e5dd7070Spatrick     VariadicEnumArgument(const Record &Arg, StringRef Attr)
1028ec727ea7Spatrick         : VariadicArgument(Arg, Attr,
1029ec727ea7Spatrick                            std::string(Arg.getValueAsString("Type"))),
1030ec727ea7Spatrick           type(std::string(Arg.getValueAsString("Type"))),
1031e5dd7070Spatrick           values(Arg.getValueAsListOfStrings("Values")),
1032e5dd7070Spatrick           enums(Arg.getValueAsListOfStrings("Enums")),
1033ec727ea7Spatrick           uniques(uniqueEnumsInOrder(enums)) {
1034e5dd7070Spatrick       QualifiedTypeName = getAttrName().str() + "Attr::" + type;
1035e5dd7070Spatrick 
1036e5dd7070Spatrick       // FIXME: Emit a proper error
1037e5dd7070Spatrick       assert(!uniques.empty());
1038e5dd7070Spatrick     }
1039e5dd7070Spatrick 
isVariadicEnumArg() const1040e5dd7070Spatrick     bool isVariadicEnumArg() const override { return true; }
1041e5dd7070Spatrick 
writeDeclarations(raw_ostream & OS) const1042e5dd7070Spatrick     void writeDeclarations(raw_ostream &OS) const override {
1043e5dd7070Spatrick       auto i = uniques.cbegin(), e = uniques.cend();
1044e5dd7070Spatrick       // The last one needs to not have a comma.
1045e5dd7070Spatrick       --e;
1046e5dd7070Spatrick 
1047e5dd7070Spatrick       OS << "public:\n";
1048e5dd7070Spatrick       OS << "  enum " << type << " {\n";
1049e5dd7070Spatrick       for (; i != e; ++i)
1050e5dd7070Spatrick         OS << "    " << *i << ",\n";
1051e5dd7070Spatrick       OS << "    " << *e << "\n";
1052e5dd7070Spatrick       OS << "  };\n";
1053e5dd7070Spatrick       OS << "private:\n";
1054e5dd7070Spatrick 
1055e5dd7070Spatrick       VariadicArgument::writeDeclarations(OS);
1056e5dd7070Spatrick     }
1057e5dd7070Spatrick 
writeDump(raw_ostream & OS) const1058e5dd7070Spatrick     void writeDump(raw_ostream &OS) const override {
1059e5dd7070Spatrick       OS << "    for (" << getAttrName() << "Attr::" << getLowerName()
1060e5dd7070Spatrick          << "_iterator I = SA->" << getLowerName() << "_begin(), E = SA->"
1061e5dd7070Spatrick          << getLowerName() << "_end(); I != E; ++I) {\n";
1062e5dd7070Spatrick       OS << "      switch(*I) {\n";
1063e5dd7070Spatrick       for (const auto &UI : uniques) {
1064e5dd7070Spatrick         OS << "    case " << getAttrName() << "Attr::" << UI << ":\n";
1065e5dd7070Spatrick         OS << "      OS << \" " << UI << "\";\n";
1066e5dd7070Spatrick         OS << "      break;\n";
1067e5dd7070Spatrick       }
1068e5dd7070Spatrick       OS << "      }\n";
1069e5dd7070Spatrick       OS << "    }\n";
1070e5dd7070Spatrick     }
1071e5dd7070Spatrick 
writePCHReadDecls(raw_ostream & OS) const1072e5dd7070Spatrick     void writePCHReadDecls(raw_ostream &OS) const override {
1073e5dd7070Spatrick       OS << "    unsigned " << getLowerName() << "Size = Record.readInt();\n";
1074e5dd7070Spatrick       OS << "    SmallVector<" << QualifiedTypeName << ", 4> " << getLowerName()
1075e5dd7070Spatrick          << ";\n";
1076e5dd7070Spatrick       OS << "    " << getLowerName() << ".reserve(" << getLowerName()
1077e5dd7070Spatrick          << "Size);\n";
1078e5dd7070Spatrick       OS << "    for (unsigned i = " << getLowerName() << "Size; i; --i)\n";
1079e5dd7070Spatrick       OS << "      " << getLowerName() << ".push_back(" << "static_cast<"
1080e5dd7070Spatrick          << QualifiedTypeName << ">(Record.readInt()));\n";
1081e5dd7070Spatrick     }
1082e5dd7070Spatrick 
writePCHWrite(raw_ostream & OS) const1083e5dd7070Spatrick     void writePCHWrite(raw_ostream &OS) const override {
1084e5dd7070Spatrick       OS << "    Record.push_back(SA->" << getLowerName() << "_size());\n";
1085e5dd7070Spatrick       OS << "    for (" << getAttrName() << "Attr::" << getLowerName()
1086e5dd7070Spatrick          << "_iterator i = SA->" << getLowerName() << "_begin(), e = SA->"
1087e5dd7070Spatrick          << getLowerName() << "_end(); i != e; ++i)\n";
1088e5dd7070Spatrick       OS << "      " << WritePCHRecord(QualifiedTypeName, "(*i)");
1089e5dd7070Spatrick     }
1090e5dd7070Spatrick 
writeConversion(raw_ostream & OS,bool Header) const1091ec727ea7Spatrick     void writeConversion(raw_ostream &OS, bool Header) const {
1092ec727ea7Spatrick       if (Header) {
1093ec727ea7Spatrick         OS << "  static bool ConvertStrTo" << type << "(StringRef Val, " << type
1094ec727ea7Spatrick            << " &Out);\n";
1095ec727ea7Spatrick         OS << "  static const char *Convert" << type << "ToStr(" << type
1096ec727ea7Spatrick            << " Val);\n";
1097ec727ea7Spatrick         return;
1098ec727ea7Spatrick       }
1099ec727ea7Spatrick 
1100ec727ea7Spatrick       OS << "bool " << getAttrName() << "Attr::ConvertStrTo" << type
1101ec727ea7Spatrick          << "(StringRef Val, ";
1102e5dd7070Spatrick       OS << type << " &Out) {\n";
1103*12c85518Srobert       OS << "  std::optional<" << type
1104*12c85518Srobert          << "> R = llvm::StringSwitch<std::optional<";
1105e5dd7070Spatrick       OS << type << ">>(Val)\n";
1106e5dd7070Spatrick       for (size_t I = 0; I < enums.size(); ++I) {
1107e5dd7070Spatrick         OS << "    .Case(\"" << values[I] << "\", ";
1108e5dd7070Spatrick         OS << getAttrName() << "Attr::" << enums[I] << ")\n";
1109e5dd7070Spatrick       }
1110*12c85518Srobert       OS << "    .Default(std::optional<" << type << ">());\n";
1111e5dd7070Spatrick       OS << "  if (R) {\n";
1112e5dd7070Spatrick       OS << "    Out = *R;\n      return true;\n    }\n";
1113e5dd7070Spatrick       OS << "  return false;\n";
1114e5dd7070Spatrick       OS << "}\n\n";
1115e5dd7070Spatrick 
1116ec727ea7Spatrick       OS << "const char *" << getAttrName() << "Attr::Convert" << type
1117ec727ea7Spatrick          << "ToStr(" << type << " Val) {\n"
1118e5dd7070Spatrick          << "  switch(Val) {\n";
1119e5dd7070Spatrick       SmallDenseSet<StringRef, 8> Uniques;
1120e5dd7070Spatrick       for (size_t I = 0; I < enums.size(); ++I) {
1121e5dd7070Spatrick         if (Uniques.insert(enums[I]).second)
1122e5dd7070Spatrick           OS << "  case " << getAttrName() << "Attr::" << enums[I]
1123e5dd7070Spatrick              << ": return \"" << values[I] << "\";\n";
1124e5dd7070Spatrick       }
1125e5dd7070Spatrick       OS << "  }\n"
1126e5dd7070Spatrick          << "  llvm_unreachable(\"No enumerator with that value\");\n"
1127e5dd7070Spatrick          << "}\n";
1128e5dd7070Spatrick     }
1129e5dd7070Spatrick   };
1130e5dd7070Spatrick 
1131e5dd7070Spatrick   class VersionArgument : public Argument {
1132e5dd7070Spatrick   public:
VersionArgument(const Record & Arg,StringRef Attr)1133e5dd7070Spatrick     VersionArgument(const Record &Arg, StringRef Attr)
1134e5dd7070Spatrick       : Argument(Arg, Attr)
1135e5dd7070Spatrick     {}
1136e5dd7070Spatrick 
writeAccessors(raw_ostream & OS) const1137e5dd7070Spatrick     void writeAccessors(raw_ostream &OS) const override {
1138e5dd7070Spatrick       OS << "  VersionTuple get" << getUpperName() << "() const {\n";
1139e5dd7070Spatrick       OS << "    return " << getLowerName() << ";\n";
1140e5dd7070Spatrick       OS << "  }\n";
1141e5dd7070Spatrick       OS << "  void set" << getUpperName()
1142e5dd7070Spatrick          << "(ASTContext &C, VersionTuple V) {\n";
1143e5dd7070Spatrick       OS << "    " << getLowerName() << " = V;\n";
1144e5dd7070Spatrick       OS << "  }";
1145e5dd7070Spatrick     }
1146e5dd7070Spatrick 
writeCloneArgs(raw_ostream & OS) const1147e5dd7070Spatrick     void writeCloneArgs(raw_ostream &OS) const override {
1148e5dd7070Spatrick       OS << "get" << getUpperName() << "()";
1149e5dd7070Spatrick     }
1150e5dd7070Spatrick 
writeTemplateInstantiationArgs(raw_ostream & OS) const1151e5dd7070Spatrick     void writeTemplateInstantiationArgs(raw_ostream &OS) const override {
1152e5dd7070Spatrick       OS << "A->get" << getUpperName() << "()";
1153e5dd7070Spatrick     }
1154e5dd7070Spatrick 
writeCtorInitializers(raw_ostream & OS) const1155e5dd7070Spatrick     void writeCtorInitializers(raw_ostream &OS) const override {
1156e5dd7070Spatrick       OS << getLowerName() << "(" << getUpperName() << ")";
1157e5dd7070Spatrick     }
1158e5dd7070Spatrick 
writeCtorDefaultInitializers(raw_ostream & OS) const1159e5dd7070Spatrick     void writeCtorDefaultInitializers(raw_ostream &OS) const override {
1160e5dd7070Spatrick       OS << getLowerName() << "()";
1161e5dd7070Spatrick     }
1162e5dd7070Spatrick 
writeCtorParameters(raw_ostream & OS) const1163e5dd7070Spatrick     void writeCtorParameters(raw_ostream &OS) const override {
1164e5dd7070Spatrick       OS << "VersionTuple " << getUpperName();
1165e5dd7070Spatrick     }
1166e5dd7070Spatrick 
writeDeclarations(raw_ostream & OS) const1167e5dd7070Spatrick     void writeDeclarations(raw_ostream &OS) const override {
1168e5dd7070Spatrick       OS << "VersionTuple " << getLowerName() << ";\n";
1169e5dd7070Spatrick     }
1170e5dd7070Spatrick 
writePCHReadDecls(raw_ostream & OS) const1171e5dd7070Spatrick     void writePCHReadDecls(raw_ostream &OS) const override {
1172e5dd7070Spatrick       OS << "    VersionTuple " << getLowerName()
1173e5dd7070Spatrick          << "= Record.readVersionTuple();\n";
1174e5dd7070Spatrick     }
1175e5dd7070Spatrick 
writePCHReadArgs(raw_ostream & OS) const1176e5dd7070Spatrick     void writePCHReadArgs(raw_ostream &OS) const override {
1177e5dd7070Spatrick       OS << getLowerName();
1178e5dd7070Spatrick     }
1179e5dd7070Spatrick 
writePCHWrite(raw_ostream & OS) const1180e5dd7070Spatrick     void writePCHWrite(raw_ostream &OS) const override {
1181e5dd7070Spatrick       OS << "    Record.AddVersionTuple(SA->get" << getUpperName() << "());\n";
1182e5dd7070Spatrick     }
1183e5dd7070Spatrick 
writeValue(raw_ostream & OS) const1184e5dd7070Spatrick     void writeValue(raw_ostream &OS) const override {
1185e5dd7070Spatrick       OS << getLowerName() << "=\" << get" << getUpperName() << "() << \"";
1186e5dd7070Spatrick     }
1187e5dd7070Spatrick 
writeDump(raw_ostream & OS) const1188e5dd7070Spatrick     void writeDump(raw_ostream &OS) const override {
1189e5dd7070Spatrick       OS << "    OS << \" \" << SA->get" << getUpperName() << "();\n";
1190e5dd7070Spatrick     }
1191e5dd7070Spatrick   };
1192e5dd7070Spatrick 
1193e5dd7070Spatrick   class ExprArgument : public SimpleArgument {
1194e5dd7070Spatrick   public:
ExprArgument(const Record & Arg,StringRef Attr)1195e5dd7070Spatrick     ExprArgument(const Record &Arg, StringRef Attr)
1196e5dd7070Spatrick       : SimpleArgument(Arg, Attr, "Expr *")
1197e5dd7070Spatrick     {}
1198e5dd7070Spatrick 
writeASTVisitorTraversal(raw_ostream & OS) const1199e5dd7070Spatrick     void writeASTVisitorTraversal(raw_ostream &OS) const override {
1200e5dd7070Spatrick       OS << "  if (!"
1201e5dd7070Spatrick          << "getDerived().TraverseStmt(A->get" << getUpperName() << "()))\n";
1202e5dd7070Spatrick       OS << "    return false;\n";
1203e5dd7070Spatrick     }
1204e5dd7070Spatrick 
writeTemplateInstantiationArgs(raw_ostream & OS) const1205e5dd7070Spatrick     void writeTemplateInstantiationArgs(raw_ostream &OS) const override {
1206e5dd7070Spatrick       OS << "tempInst" << getUpperName();
1207e5dd7070Spatrick     }
1208e5dd7070Spatrick 
writeTemplateInstantiation(raw_ostream & OS) const1209e5dd7070Spatrick     void writeTemplateInstantiation(raw_ostream &OS) const override {
1210e5dd7070Spatrick       OS << "      " << getType() << " tempInst" << getUpperName() << ";\n";
1211e5dd7070Spatrick       OS << "      {\n";
1212e5dd7070Spatrick       OS << "        EnterExpressionEvaluationContext "
1213e5dd7070Spatrick          << "Unevaluated(S, Sema::ExpressionEvaluationContext::Unevaluated);\n";
1214e5dd7070Spatrick       OS << "        ExprResult " << "Result = S.SubstExpr("
1215e5dd7070Spatrick          << "A->get" << getUpperName() << "(), TemplateArgs);\n";
1216a9ac8606Spatrick       OS << "        if (Result.isInvalid())\n";
1217a9ac8606Spatrick       OS << "          return nullptr;\n";
1218a9ac8606Spatrick       OS << "        tempInst" << getUpperName() << " = Result.get();\n";
1219e5dd7070Spatrick       OS << "      }\n";
1220e5dd7070Spatrick     }
1221e5dd7070Spatrick 
writeValue(raw_ostream & OS) const1222*12c85518Srobert     void writeValue(raw_ostream &OS) const override {
1223*12c85518Srobert       OS << "\";\n";
1224*12c85518Srobert       OS << "    get" << getUpperName()
1225*12c85518Srobert          << "()->printPretty(OS, nullptr, Policy);\n";
1226*12c85518Srobert       OS << "    OS << \"";
1227*12c85518Srobert     }
1228*12c85518Srobert 
writeDump(raw_ostream & OS) const1229e5dd7070Spatrick     void writeDump(raw_ostream &OS) const override {}
1230e5dd7070Spatrick 
writeDumpChildren(raw_ostream & OS) const1231e5dd7070Spatrick     void writeDumpChildren(raw_ostream &OS) const override {
1232e5dd7070Spatrick       OS << "    Visit(SA->get" << getUpperName() << "());\n";
1233e5dd7070Spatrick     }
1234e5dd7070Spatrick 
writeHasChildren(raw_ostream & OS) const1235e5dd7070Spatrick     void writeHasChildren(raw_ostream &OS) const override { OS << "true"; }
1236e5dd7070Spatrick   };
1237e5dd7070Spatrick 
1238e5dd7070Spatrick   class VariadicExprArgument : public VariadicArgument {
1239e5dd7070Spatrick   public:
VariadicExprArgument(const Record & Arg,StringRef Attr)1240e5dd7070Spatrick     VariadicExprArgument(const Record &Arg, StringRef Attr)
1241e5dd7070Spatrick       : VariadicArgument(Arg, Attr, "Expr *")
1242e5dd7070Spatrick     {}
1243e5dd7070Spatrick 
VariadicExprArgument(StringRef ArgName,StringRef Attr)1244*12c85518Srobert     VariadicExprArgument(StringRef ArgName, StringRef Attr)
1245*12c85518Srobert         : VariadicArgument(ArgName, Attr, "Expr *") {}
1246*12c85518Srobert 
writeASTVisitorTraversal(raw_ostream & OS) const1247e5dd7070Spatrick     void writeASTVisitorTraversal(raw_ostream &OS) const override {
1248e5dd7070Spatrick       OS << "  {\n";
1249e5dd7070Spatrick       OS << "    " << getType() << " *I = A->" << getLowerName()
1250e5dd7070Spatrick          << "_begin();\n";
1251e5dd7070Spatrick       OS << "    " << getType() << " *E = A->" << getLowerName()
1252e5dd7070Spatrick          << "_end();\n";
1253e5dd7070Spatrick       OS << "    for (; I != E; ++I) {\n";
1254e5dd7070Spatrick       OS << "      if (!getDerived().TraverseStmt(*I))\n";
1255e5dd7070Spatrick       OS << "        return false;\n";
1256e5dd7070Spatrick       OS << "    }\n";
1257e5dd7070Spatrick       OS << "  }\n";
1258e5dd7070Spatrick     }
1259e5dd7070Spatrick 
writeTemplateInstantiationArgs(raw_ostream & OS) const1260e5dd7070Spatrick     void writeTemplateInstantiationArgs(raw_ostream &OS) const override {
1261e5dd7070Spatrick       OS << "tempInst" << getUpperName() << ", "
1262e5dd7070Spatrick          << "A->" << getLowerName() << "_size()";
1263e5dd7070Spatrick     }
1264e5dd7070Spatrick 
writeTemplateInstantiation(raw_ostream & OS) const1265e5dd7070Spatrick     void writeTemplateInstantiation(raw_ostream &OS) const override {
1266e5dd7070Spatrick       OS << "      auto *tempInst" << getUpperName()
1267e5dd7070Spatrick          << " = new (C, 16) " << getType()
1268e5dd7070Spatrick          << "[A->" << getLowerName() << "_size()];\n";
1269e5dd7070Spatrick       OS << "      {\n";
1270e5dd7070Spatrick       OS << "        EnterExpressionEvaluationContext "
1271e5dd7070Spatrick          << "Unevaluated(S, Sema::ExpressionEvaluationContext::Unevaluated);\n";
1272e5dd7070Spatrick       OS << "        " << getType() << " *TI = tempInst" << getUpperName()
1273e5dd7070Spatrick          << ";\n";
1274e5dd7070Spatrick       OS << "        " << getType() << " *I = A->" << getLowerName()
1275e5dd7070Spatrick          << "_begin();\n";
1276e5dd7070Spatrick       OS << "        " << getType() << " *E = A->" << getLowerName()
1277e5dd7070Spatrick          << "_end();\n";
1278e5dd7070Spatrick       OS << "        for (; I != E; ++I, ++TI) {\n";
1279e5dd7070Spatrick       OS << "          ExprResult Result = S.SubstExpr(*I, TemplateArgs);\n";
1280a9ac8606Spatrick       OS << "          if (Result.isInvalid())\n";
1281a9ac8606Spatrick       OS << "            return nullptr;\n";
1282a9ac8606Spatrick       OS << "          *TI = Result.get();\n";
1283e5dd7070Spatrick       OS << "        }\n";
1284e5dd7070Spatrick       OS << "      }\n";
1285e5dd7070Spatrick     }
1286e5dd7070Spatrick 
writeDump(raw_ostream & OS) const1287e5dd7070Spatrick     void writeDump(raw_ostream &OS) const override {}
1288e5dd7070Spatrick 
writeDumpChildren(raw_ostream & OS) const1289e5dd7070Spatrick     void writeDumpChildren(raw_ostream &OS) const override {
1290e5dd7070Spatrick       OS << "    for (" << getAttrName() << "Attr::" << getLowerName()
1291e5dd7070Spatrick          << "_iterator I = SA->" << getLowerName() << "_begin(), E = SA->"
1292e5dd7070Spatrick          << getLowerName() << "_end(); I != E; ++I)\n";
1293e5dd7070Spatrick       OS << "      Visit(*I);\n";
1294e5dd7070Spatrick     }
1295e5dd7070Spatrick 
writeHasChildren(raw_ostream & OS) const1296e5dd7070Spatrick     void writeHasChildren(raw_ostream &OS) const override {
1297e5dd7070Spatrick       OS << "SA->" << getLowerName() << "_begin() != "
1298e5dd7070Spatrick          << "SA->" << getLowerName() << "_end()";
1299e5dd7070Spatrick     }
1300e5dd7070Spatrick   };
1301e5dd7070Spatrick 
1302e5dd7070Spatrick   class VariadicIdentifierArgument : public VariadicArgument {
1303e5dd7070Spatrick   public:
VariadicIdentifierArgument(const Record & Arg,StringRef Attr)1304e5dd7070Spatrick     VariadicIdentifierArgument(const Record &Arg, StringRef Attr)
1305e5dd7070Spatrick       : VariadicArgument(Arg, Attr, "IdentifierInfo *")
1306e5dd7070Spatrick     {}
1307e5dd7070Spatrick   };
1308e5dd7070Spatrick 
1309e5dd7070Spatrick   class VariadicStringArgument : public VariadicArgument {
1310e5dd7070Spatrick   public:
VariadicStringArgument(const Record & Arg,StringRef Attr)1311e5dd7070Spatrick     VariadicStringArgument(const Record &Arg, StringRef Attr)
1312e5dd7070Spatrick       : VariadicArgument(Arg, Attr, "StringRef")
1313e5dd7070Spatrick     {}
1314e5dd7070Spatrick 
writeCtorBody(raw_ostream & OS) const1315e5dd7070Spatrick     void writeCtorBody(raw_ostream &OS) const override {
1316e5dd7070Spatrick       OS << "  for (size_t I = 0, E = " << getArgSizeName() << "; I != E;\n"
1317e5dd7070Spatrick             "       ++I) {\n"
1318e5dd7070Spatrick             "    StringRef Ref = " << getUpperName() << "[I];\n"
1319e5dd7070Spatrick             "    if (!Ref.empty()) {\n"
1320e5dd7070Spatrick             "      char *Mem = new (Ctx, 1) char[Ref.size()];\n"
1321e5dd7070Spatrick             "      std::memcpy(Mem, Ref.data(), Ref.size());\n"
1322e5dd7070Spatrick             "      " << getArgName() << "[I] = StringRef(Mem, Ref.size());\n"
1323e5dd7070Spatrick             "    }\n"
1324e5dd7070Spatrick             "  }\n";
1325e5dd7070Spatrick     }
1326e5dd7070Spatrick 
writeValueImpl(raw_ostream & OS) const1327e5dd7070Spatrick     void writeValueImpl(raw_ostream &OS) const override {
1328e5dd7070Spatrick       OS << "    OS << \"\\\"\" << Val << \"\\\"\";\n";
1329e5dd7070Spatrick     }
1330e5dd7070Spatrick   };
1331e5dd7070Spatrick 
1332e5dd7070Spatrick   class TypeArgument : public SimpleArgument {
1333e5dd7070Spatrick   public:
TypeArgument(const Record & Arg,StringRef Attr)1334e5dd7070Spatrick     TypeArgument(const Record &Arg, StringRef Attr)
1335e5dd7070Spatrick       : SimpleArgument(Arg, Attr, "TypeSourceInfo *")
1336e5dd7070Spatrick     {}
1337e5dd7070Spatrick 
writeAccessors(raw_ostream & OS) const1338e5dd7070Spatrick     void writeAccessors(raw_ostream &OS) const override {
1339e5dd7070Spatrick       OS << "  QualType get" << getUpperName() << "() const {\n";
1340e5dd7070Spatrick       OS << "    return " << getLowerName() << "->getType();\n";
1341e5dd7070Spatrick       OS << "  }";
1342e5dd7070Spatrick       OS << "  " << getType() << " get" << getUpperName() << "Loc() const {\n";
1343e5dd7070Spatrick       OS << "    return " << getLowerName() << ";\n";
1344e5dd7070Spatrick       OS << "  }";
1345e5dd7070Spatrick     }
1346e5dd7070Spatrick 
writeASTVisitorTraversal(raw_ostream & OS) const1347e5dd7070Spatrick     void writeASTVisitorTraversal(raw_ostream &OS) const override {
1348e5dd7070Spatrick       OS << "  if (auto *TSI = A->get" << getUpperName() << "Loc())\n";
1349e5dd7070Spatrick       OS << "    if (!getDerived().TraverseTypeLoc(TSI->getTypeLoc()))\n";
1350e5dd7070Spatrick       OS << "      return false;\n";
1351e5dd7070Spatrick     }
1352e5dd7070Spatrick 
writeTemplateInstantiation(raw_ostream & OS) const1353a9ac8606Spatrick     void writeTemplateInstantiation(raw_ostream &OS) const override {
1354a9ac8606Spatrick       OS << "      " << getType() << " tempInst" << getUpperName() << " =\n";
1355a9ac8606Spatrick       OS << "        S.SubstType(A->get" << getUpperName() << "Loc(), "
1356a9ac8606Spatrick          << "TemplateArgs, A->getLoc(), A->getAttrName());\n";
1357a9ac8606Spatrick       OS << "      if (!tempInst" << getUpperName() << ")\n";
1358a9ac8606Spatrick       OS << "        return nullptr;\n";
1359a9ac8606Spatrick     }
1360a9ac8606Spatrick 
writeTemplateInstantiationArgs(raw_ostream & OS) const1361e5dd7070Spatrick     void writeTemplateInstantiationArgs(raw_ostream &OS) const override {
1362a9ac8606Spatrick       OS << "tempInst" << getUpperName();
1363e5dd7070Spatrick     }
1364e5dd7070Spatrick 
writePCHWrite(raw_ostream & OS) const1365e5dd7070Spatrick     void writePCHWrite(raw_ostream &OS) const override {
1366ec727ea7Spatrick       OS << "    "
1367ec727ea7Spatrick          << WritePCHRecord(getType(),
1368ec727ea7Spatrick                            "SA->get" + std::string(getUpperName()) + "Loc()");
1369e5dd7070Spatrick     }
1370e5dd7070Spatrick   };
1371e5dd7070Spatrick 
1372e5dd7070Spatrick } // end anonymous namespace
1373e5dd7070Spatrick 
1374e5dd7070Spatrick static std::unique_ptr<Argument>
createArgument(const Record & Arg,StringRef Attr,const Record * Search=nullptr)1375e5dd7070Spatrick createArgument(const Record &Arg, StringRef Attr,
1376e5dd7070Spatrick                const Record *Search = nullptr) {
1377e5dd7070Spatrick   if (!Search)
1378e5dd7070Spatrick     Search = &Arg;
1379e5dd7070Spatrick 
1380e5dd7070Spatrick   std::unique_ptr<Argument> Ptr;
1381e5dd7070Spatrick   llvm::StringRef ArgName = Search->getName();
1382e5dd7070Spatrick 
1383e5dd7070Spatrick   if (ArgName == "AlignedArgument")
1384e5dd7070Spatrick     Ptr = std::make_unique<AlignedArgument>(Arg, Attr);
1385e5dd7070Spatrick   else if (ArgName == "EnumArgument")
1386e5dd7070Spatrick     Ptr = std::make_unique<EnumArgument>(Arg, Attr);
1387e5dd7070Spatrick   else if (ArgName == "ExprArgument")
1388e5dd7070Spatrick     Ptr = std::make_unique<ExprArgument>(Arg, Attr);
1389ec727ea7Spatrick   else if (ArgName == "DeclArgument")
1390ec727ea7Spatrick     Ptr = std::make_unique<SimpleArgument>(
1391ec727ea7Spatrick         Arg, Attr, (Arg.getValueAsDef("Kind")->getName() + "Decl *").str());
1392e5dd7070Spatrick   else if (ArgName == "IdentifierArgument")
1393e5dd7070Spatrick     Ptr = std::make_unique<SimpleArgument>(Arg, Attr, "IdentifierInfo *");
1394e5dd7070Spatrick   else if (ArgName == "DefaultBoolArgument")
1395e5dd7070Spatrick     Ptr = std::make_unique<DefaultSimpleArgument>(
1396e5dd7070Spatrick         Arg, Attr, "bool", Arg.getValueAsBit("Default"));
1397e5dd7070Spatrick   else if (ArgName == "BoolArgument")
1398e5dd7070Spatrick     Ptr = std::make_unique<SimpleArgument>(Arg, Attr, "bool");
1399e5dd7070Spatrick   else if (ArgName == "DefaultIntArgument")
1400e5dd7070Spatrick     Ptr = std::make_unique<DefaultSimpleArgument>(
1401e5dd7070Spatrick         Arg, Attr, "int", Arg.getValueAsInt("Default"));
1402e5dd7070Spatrick   else if (ArgName == "IntArgument")
1403e5dd7070Spatrick     Ptr = std::make_unique<SimpleArgument>(Arg, Attr, "int");
1404e5dd7070Spatrick   else if (ArgName == "StringArgument")
1405e5dd7070Spatrick     Ptr = std::make_unique<StringArgument>(Arg, Attr);
1406e5dd7070Spatrick   else if (ArgName == "TypeArgument")
1407e5dd7070Spatrick     Ptr = std::make_unique<TypeArgument>(Arg, Attr);
1408e5dd7070Spatrick   else if (ArgName == "UnsignedArgument")
1409e5dd7070Spatrick     Ptr = std::make_unique<SimpleArgument>(Arg, Attr, "unsigned");
1410e5dd7070Spatrick   else if (ArgName == "VariadicUnsignedArgument")
1411e5dd7070Spatrick     Ptr = std::make_unique<VariadicArgument>(Arg, Attr, "unsigned");
1412e5dd7070Spatrick   else if (ArgName == "VariadicStringArgument")
1413e5dd7070Spatrick     Ptr = std::make_unique<VariadicStringArgument>(Arg, Attr);
1414e5dd7070Spatrick   else if (ArgName == "VariadicEnumArgument")
1415e5dd7070Spatrick     Ptr = std::make_unique<VariadicEnumArgument>(Arg, Attr);
1416e5dd7070Spatrick   else if (ArgName == "VariadicExprArgument")
1417e5dd7070Spatrick     Ptr = std::make_unique<VariadicExprArgument>(Arg, Attr);
1418e5dd7070Spatrick   else if (ArgName == "VariadicParamIdxArgument")
1419e5dd7070Spatrick     Ptr = std::make_unique<VariadicParamIdxArgument>(Arg, Attr);
1420e5dd7070Spatrick   else if (ArgName == "VariadicParamOrParamIdxArgument")
1421e5dd7070Spatrick     Ptr = std::make_unique<VariadicParamOrParamIdxArgument>(Arg, Attr);
1422e5dd7070Spatrick   else if (ArgName == "ParamIdxArgument")
1423e5dd7070Spatrick     Ptr = std::make_unique<SimpleArgument>(Arg, Attr, "ParamIdx");
1424e5dd7070Spatrick   else if (ArgName == "VariadicIdentifierArgument")
1425e5dd7070Spatrick     Ptr = std::make_unique<VariadicIdentifierArgument>(Arg, Attr);
1426e5dd7070Spatrick   else if (ArgName == "VersionArgument")
1427e5dd7070Spatrick     Ptr = std::make_unique<VersionArgument>(Arg, Attr);
1428ec727ea7Spatrick   else if (ArgName == "OMPTraitInfoArgument")
1429ec727ea7Spatrick     Ptr = std::make_unique<SimpleArgument>(Arg, Attr, "OMPTraitInfo *");
1430*12c85518Srobert   else if (ArgName == "VariadicOMPInteropInfoArgument")
1431*12c85518Srobert     Ptr = std::make_unique<VariadicOMPInteropInfoArgument>(Arg, Attr);
1432e5dd7070Spatrick 
1433e5dd7070Spatrick   if (!Ptr) {
1434e5dd7070Spatrick     // Search in reverse order so that the most-derived type is handled first.
1435e5dd7070Spatrick     ArrayRef<std::pair<Record*, SMRange>> Bases = Search->getSuperClasses();
1436e5dd7070Spatrick     for (const auto &Base : llvm::reverse(Bases)) {
1437e5dd7070Spatrick       if ((Ptr = createArgument(Arg, Attr, Base.first)))
1438e5dd7070Spatrick         break;
1439e5dd7070Spatrick     }
1440e5dd7070Spatrick   }
1441e5dd7070Spatrick 
1442e5dd7070Spatrick   if (Ptr && Arg.getValueAsBit("Optional"))
1443e5dd7070Spatrick     Ptr->setOptional(true);
1444e5dd7070Spatrick 
1445e5dd7070Spatrick   if (Ptr && Arg.getValueAsBit("Fake"))
1446e5dd7070Spatrick     Ptr->setFake(true);
1447e5dd7070Spatrick 
1448e5dd7070Spatrick   return Ptr;
1449e5dd7070Spatrick }
1450e5dd7070Spatrick 
writeAvailabilityValue(raw_ostream & OS)1451e5dd7070Spatrick static void writeAvailabilityValue(raw_ostream &OS) {
1452e5dd7070Spatrick   OS << "\" << getPlatform()->getName();\n"
1453e5dd7070Spatrick      << "  if (getStrict()) OS << \", strict\";\n"
1454e5dd7070Spatrick      << "  if (!getIntroduced().empty()) OS << \", introduced=\" << getIntroduced();\n"
1455e5dd7070Spatrick      << "  if (!getDeprecated().empty()) OS << \", deprecated=\" << getDeprecated();\n"
1456e5dd7070Spatrick      << "  if (!getObsoleted().empty()) OS << \", obsoleted=\" << getObsoleted();\n"
1457e5dd7070Spatrick      << "  if (getUnavailable()) OS << \", unavailable\";\n"
1458e5dd7070Spatrick      << "  OS << \"";
1459e5dd7070Spatrick }
1460e5dd7070Spatrick 
writeDeprecatedAttrValue(raw_ostream & OS,std::string & Variety)1461e5dd7070Spatrick static void writeDeprecatedAttrValue(raw_ostream &OS, std::string &Variety) {
1462e5dd7070Spatrick   OS << "\\\"\" << getMessage() << \"\\\"\";\n";
1463e5dd7070Spatrick   // Only GNU deprecated has an optional fixit argument at the second position.
1464e5dd7070Spatrick   if (Variety == "GNU")
1465e5dd7070Spatrick      OS << "    if (!getReplacement().empty()) OS << \", \\\"\""
1466e5dd7070Spatrick            " << getReplacement() << \"\\\"\";\n";
1467e5dd7070Spatrick   OS << "    OS << \"";
1468e5dd7070Spatrick }
1469e5dd7070Spatrick 
writeGetSpellingFunction(const Record & R,raw_ostream & OS)1470ec727ea7Spatrick static void writeGetSpellingFunction(const Record &R, raw_ostream &OS) {
1471e5dd7070Spatrick   std::vector<FlattenedSpelling> Spellings = GetFlattenedSpellings(R);
1472e5dd7070Spatrick 
1473e5dd7070Spatrick   OS << "const char *" << R.getName() << "Attr::getSpelling() const {\n";
1474e5dd7070Spatrick   if (Spellings.empty()) {
1475e5dd7070Spatrick     OS << "  return \"(No spelling)\";\n}\n\n";
1476e5dd7070Spatrick     return;
1477e5dd7070Spatrick   }
1478e5dd7070Spatrick 
1479e5dd7070Spatrick   OS << "  switch (getAttributeSpellingListIndex()) {\n"
1480e5dd7070Spatrick         "  default:\n"
1481e5dd7070Spatrick         "    llvm_unreachable(\"Unknown attribute spelling!\");\n"
1482e5dd7070Spatrick         "    return \"(No spelling)\";\n";
1483e5dd7070Spatrick 
1484e5dd7070Spatrick   for (unsigned I = 0; I < Spellings.size(); ++I)
1485e5dd7070Spatrick     OS << "  case " << I << ":\n"
1486e5dd7070Spatrick           "    return \"" << Spellings[I].name() << "\";\n";
1487e5dd7070Spatrick   // End of the switch statement.
1488e5dd7070Spatrick   OS << "  }\n";
1489e5dd7070Spatrick   // End of the getSpelling function.
1490e5dd7070Spatrick   OS << "}\n\n";
1491e5dd7070Spatrick }
1492e5dd7070Spatrick 
1493e5dd7070Spatrick static void
writePrettyPrintFunction(const Record & R,const std::vector<std::unique_ptr<Argument>> & Args,raw_ostream & OS)1494ec727ea7Spatrick writePrettyPrintFunction(const Record &R,
1495e5dd7070Spatrick                          const std::vector<std::unique_ptr<Argument>> &Args,
1496e5dd7070Spatrick                          raw_ostream &OS) {
1497e5dd7070Spatrick   std::vector<FlattenedSpelling> Spellings = GetFlattenedSpellings(R);
1498e5dd7070Spatrick 
1499e5dd7070Spatrick   OS << "void " << R.getName() << "Attr::printPretty("
1500e5dd7070Spatrick     << "raw_ostream &OS, const PrintingPolicy &Policy) const {\n";
1501e5dd7070Spatrick 
1502e5dd7070Spatrick   if (Spellings.empty()) {
1503e5dd7070Spatrick     OS << "}\n\n";
1504e5dd7070Spatrick     return;
1505e5dd7070Spatrick   }
1506e5dd7070Spatrick 
1507a9ac8606Spatrick   OS << "  bool IsFirstArgument = true; (void)IsFirstArgument;\n"
1508a9ac8606Spatrick      << "  unsigned TrailingOmittedArgs = 0; (void)TrailingOmittedArgs;\n"
1509a9ac8606Spatrick      << "  switch (getAttributeSpellingListIndex()) {\n"
1510a9ac8606Spatrick      << "  default:\n"
1511a9ac8606Spatrick      << "    llvm_unreachable(\"Unknown attribute spelling!\");\n"
1512a9ac8606Spatrick      << "    break;\n";
1513e5dd7070Spatrick 
1514e5dd7070Spatrick   for (unsigned I = 0; I < Spellings.size(); ++ I) {
1515e5dd7070Spatrick     llvm::SmallString<16> Prefix;
1516e5dd7070Spatrick     llvm::SmallString<8> Suffix;
1517e5dd7070Spatrick     // The actual spelling of the name and namespace (if applicable)
1518e5dd7070Spatrick     // of an attribute without considering prefix and suffix.
1519e5dd7070Spatrick     llvm::SmallString<64> Spelling;
1520e5dd7070Spatrick     std::string Name = Spellings[I].name();
1521e5dd7070Spatrick     std::string Variety = Spellings[I].variety();
1522e5dd7070Spatrick 
1523e5dd7070Spatrick     if (Variety == "GNU") {
1524e5dd7070Spatrick       Prefix = " __attribute__((";
1525e5dd7070Spatrick       Suffix = "))";
1526e5dd7070Spatrick     } else if (Variety == "CXX11" || Variety == "C2x") {
1527e5dd7070Spatrick       Prefix = " [[";
1528e5dd7070Spatrick       Suffix = "]]";
1529e5dd7070Spatrick       std::string Namespace = Spellings[I].nameSpace();
1530e5dd7070Spatrick       if (!Namespace.empty()) {
1531e5dd7070Spatrick         Spelling += Namespace;
1532e5dd7070Spatrick         Spelling += "::";
1533e5dd7070Spatrick       }
1534e5dd7070Spatrick     } else if (Variety == "Declspec") {
1535e5dd7070Spatrick       Prefix = " __declspec(";
1536e5dd7070Spatrick       Suffix = ")";
1537e5dd7070Spatrick     } else if (Variety == "Microsoft") {
1538e5dd7070Spatrick       Prefix = "[";
1539e5dd7070Spatrick       Suffix = "]";
1540e5dd7070Spatrick     } else if (Variety == "Keyword") {
1541e5dd7070Spatrick       Prefix = " ";
1542e5dd7070Spatrick       Suffix = "";
1543e5dd7070Spatrick     } else if (Variety == "Pragma") {
1544e5dd7070Spatrick       Prefix = "#pragma ";
1545e5dd7070Spatrick       Suffix = "\n";
1546e5dd7070Spatrick       std::string Namespace = Spellings[I].nameSpace();
1547e5dd7070Spatrick       if (!Namespace.empty()) {
1548e5dd7070Spatrick         Spelling += Namespace;
1549e5dd7070Spatrick         Spelling += " ";
1550e5dd7070Spatrick       }
1551*12c85518Srobert     } else if (Variety == "HLSLSemantic") {
1552*12c85518Srobert       Prefix = ":";
1553*12c85518Srobert       Suffix = "";
1554e5dd7070Spatrick     } else {
1555e5dd7070Spatrick       llvm_unreachable("Unknown attribute syntax variety!");
1556e5dd7070Spatrick     }
1557e5dd7070Spatrick 
1558e5dd7070Spatrick     Spelling += Name;
1559e5dd7070Spatrick 
1560a9ac8606Spatrick     OS << "  case " << I << " : {\n"
1561a9ac8606Spatrick        << "    OS << \"" << Prefix << Spelling << "\";\n";
1562e5dd7070Spatrick 
1563e5dd7070Spatrick     if (Variety == "Pragma") {
1564e5dd7070Spatrick       OS << "    printPrettyPragma(OS, Policy);\n";
1565e5dd7070Spatrick       OS << "    OS << \"\\n\";";
1566e5dd7070Spatrick       OS << "    break;\n";
1567e5dd7070Spatrick       OS << "  }\n";
1568e5dd7070Spatrick       continue;
1569e5dd7070Spatrick     }
1570e5dd7070Spatrick 
1571e5dd7070Spatrick     if (Spelling == "availability") {
1572a9ac8606Spatrick       OS << "    OS << \"(";
1573e5dd7070Spatrick       writeAvailabilityValue(OS);
1574a9ac8606Spatrick       OS << ")\";\n";
1575e5dd7070Spatrick     } else if (Spelling == "deprecated" || Spelling == "gnu::deprecated") {
1576a9ac8606Spatrick       OS << "    OS << \"(";
1577e5dd7070Spatrick       writeDeprecatedAttrValue(OS, Variety);
1578a9ac8606Spatrick       OS << ")\";\n";
1579e5dd7070Spatrick     } else {
1580e5dd7070Spatrick       // To avoid printing parentheses around an empty argument list or
1581e5dd7070Spatrick       // printing spurious commas at the end of an argument list, we need to
1582e5dd7070Spatrick       // determine where the last provided non-fake argument is.
1583e5dd7070Spatrick       bool FoundNonOptArg = false;
1584e5dd7070Spatrick       for (const auto &arg : llvm::reverse(Args)) {
1585e5dd7070Spatrick         if (arg->isFake())
1586e5dd7070Spatrick           continue;
1587e5dd7070Spatrick         if (FoundNonOptArg)
1588e5dd7070Spatrick           continue;
1589e5dd7070Spatrick         // FIXME: arg->getIsOmitted() == "false" means we haven't implemented
1590e5dd7070Spatrick         // any way to detect whether the argument was omitted.
1591e5dd7070Spatrick         if (!arg->isOptional() || arg->getIsOmitted() == "false") {
1592e5dd7070Spatrick           FoundNonOptArg = true;
1593e5dd7070Spatrick           continue;
1594e5dd7070Spatrick         }
1595e5dd7070Spatrick         OS << "    if (" << arg->getIsOmitted() << ")\n"
1596e5dd7070Spatrick            << "      ++TrailingOmittedArgs;\n";
1597e5dd7070Spatrick       }
1598e5dd7070Spatrick       unsigned ArgIndex = 0;
1599e5dd7070Spatrick       for (const auto &arg : Args) {
1600e5dd7070Spatrick         if (arg->isFake())
1601e5dd7070Spatrick           continue;
1602e5dd7070Spatrick         std::string IsOmitted = arg->getIsOmitted();
1603e5dd7070Spatrick         if (arg->isOptional() && IsOmitted != "false")
1604a9ac8606Spatrick           OS << "    if (!(" << IsOmitted << ")) {\n";
1605a9ac8606Spatrick         // Variadic arguments print their own leading comma.
1606a9ac8606Spatrick         if (!arg->isVariadic())
1607a9ac8606Spatrick           OS << "    DelimitAttributeArgument(OS, IsFirstArgument);\n";
1608a9ac8606Spatrick         OS << "    OS << \"";
1609e5dd7070Spatrick         arg->writeValue(OS);
1610a9ac8606Spatrick         OS << "\";\n";
1611e5dd7070Spatrick         if (arg->isOptional() && IsOmitted != "false")
1612a9ac8606Spatrick           OS << "    }\n";
1613e5dd7070Spatrick         ++ArgIndex;
1614e5dd7070Spatrick       }
1615a9ac8606Spatrick       if (ArgIndex != 0)
1616a9ac8606Spatrick         OS << "    if (!IsFirstArgument)\n"
1617a9ac8606Spatrick            << "      OS << \")\";\n";
1618e5dd7070Spatrick     }
1619a9ac8606Spatrick     OS << "    OS << \"" << Suffix << "\";\n"
1620a9ac8606Spatrick        << "    break;\n"
1621a9ac8606Spatrick        << "  }\n";
1622e5dd7070Spatrick   }
1623e5dd7070Spatrick 
1624e5dd7070Spatrick   // End of the switch statement.
1625e5dd7070Spatrick   OS << "}\n";
1626e5dd7070Spatrick   // End of the print function.
1627e5dd7070Spatrick   OS << "}\n\n";
1628e5dd7070Spatrick }
1629e5dd7070Spatrick 
1630e5dd7070Spatrick /// Return the index of a spelling in a spelling list.
1631e5dd7070Spatrick static unsigned
getSpellingListIndex(const std::vector<FlattenedSpelling> & SpellingList,const FlattenedSpelling & Spelling)1632e5dd7070Spatrick getSpellingListIndex(const std::vector<FlattenedSpelling> &SpellingList,
1633e5dd7070Spatrick                      const FlattenedSpelling &Spelling) {
1634e5dd7070Spatrick   assert(!SpellingList.empty() && "Spelling list is empty!");
1635e5dd7070Spatrick 
1636e5dd7070Spatrick   for (unsigned Index = 0; Index < SpellingList.size(); ++Index) {
1637e5dd7070Spatrick     const FlattenedSpelling &S = SpellingList[Index];
1638e5dd7070Spatrick     if (S.variety() != Spelling.variety())
1639e5dd7070Spatrick       continue;
1640e5dd7070Spatrick     if (S.nameSpace() != Spelling.nameSpace())
1641e5dd7070Spatrick       continue;
1642e5dd7070Spatrick     if (S.name() != Spelling.name())
1643e5dd7070Spatrick       continue;
1644e5dd7070Spatrick 
1645e5dd7070Spatrick     return Index;
1646e5dd7070Spatrick   }
1647e5dd7070Spatrick 
1648e5dd7070Spatrick   llvm_unreachable("Unknown spelling!");
1649e5dd7070Spatrick }
1650e5dd7070Spatrick 
writeAttrAccessorDefinition(const Record & R,raw_ostream & OS)1651e5dd7070Spatrick static void writeAttrAccessorDefinition(const Record &R, raw_ostream &OS) {
1652e5dd7070Spatrick   std::vector<Record*> Accessors = R.getValueAsListOfDefs("Accessors");
1653e5dd7070Spatrick   if (Accessors.empty())
1654e5dd7070Spatrick     return;
1655e5dd7070Spatrick 
1656e5dd7070Spatrick   const std::vector<FlattenedSpelling> SpellingList = GetFlattenedSpellings(R);
1657e5dd7070Spatrick   assert(!SpellingList.empty() &&
1658e5dd7070Spatrick          "Attribute with empty spelling list can't have accessors!");
1659e5dd7070Spatrick   for (const auto *Accessor : Accessors) {
1660e5dd7070Spatrick     const StringRef Name = Accessor->getValueAsString("Name");
1661e5dd7070Spatrick     std::vector<FlattenedSpelling> Spellings = GetFlattenedSpellings(*Accessor);
1662e5dd7070Spatrick 
1663e5dd7070Spatrick     OS << "  bool " << Name
1664e5dd7070Spatrick        << "() const { return getAttributeSpellingListIndex() == ";
1665e5dd7070Spatrick     for (unsigned Index = 0; Index < Spellings.size(); ++Index) {
1666e5dd7070Spatrick       OS << getSpellingListIndex(SpellingList, Spellings[Index]);
1667e5dd7070Spatrick       if (Index != Spellings.size() - 1)
1668e5dd7070Spatrick         OS << " ||\n    getAttributeSpellingListIndex() == ";
1669e5dd7070Spatrick       else
1670e5dd7070Spatrick         OS << "; }\n";
1671e5dd7070Spatrick     }
1672e5dd7070Spatrick   }
1673e5dd7070Spatrick }
1674e5dd7070Spatrick 
1675e5dd7070Spatrick static bool
SpellingNamesAreCommon(const std::vector<FlattenedSpelling> & Spellings)1676e5dd7070Spatrick SpellingNamesAreCommon(const std::vector<FlattenedSpelling>& Spellings) {
1677e5dd7070Spatrick   assert(!Spellings.empty() && "An empty list of spellings was provided");
1678ec727ea7Spatrick   std::string FirstName =
1679ec727ea7Spatrick       std::string(NormalizeNameForSpellingComparison(Spellings.front().name()));
1680e5dd7070Spatrick   for (const auto &Spelling :
1681e5dd7070Spatrick        llvm::make_range(std::next(Spellings.begin()), Spellings.end())) {
1682ec727ea7Spatrick     std::string Name =
1683ec727ea7Spatrick         std::string(NormalizeNameForSpellingComparison(Spelling.name()));
1684e5dd7070Spatrick     if (Name != FirstName)
1685e5dd7070Spatrick       return false;
1686e5dd7070Spatrick   }
1687e5dd7070Spatrick   return true;
1688e5dd7070Spatrick }
1689e5dd7070Spatrick 
1690e5dd7070Spatrick typedef std::map<unsigned, std::string> SemanticSpellingMap;
1691e5dd7070Spatrick static std::string
CreateSemanticSpellings(const std::vector<FlattenedSpelling> & Spellings,SemanticSpellingMap & Map)1692e5dd7070Spatrick CreateSemanticSpellings(const std::vector<FlattenedSpelling> &Spellings,
1693e5dd7070Spatrick                         SemanticSpellingMap &Map) {
1694e5dd7070Spatrick   // The enumerants are automatically generated based on the variety,
1695e5dd7070Spatrick   // namespace (if present) and name for each attribute spelling. However,
1696e5dd7070Spatrick   // care is taken to avoid trampling on the reserved namespace due to
1697e5dd7070Spatrick   // underscores.
1698e5dd7070Spatrick   std::string Ret("  enum Spelling {\n");
1699e5dd7070Spatrick   std::set<std::string> Uniques;
1700e5dd7070Spatrick   unsigned Idx = 0;
1701e5dd7070Spatrick 
1702e5dd7070Spatrick   // If we have a need to have this many spellings we likely need to add an
1703e5dd7070Spatrick   // extra bit to the SpellingIndex in AttributeCommonInfo, then increase the
1704e5dd7070Spatrick   // value of SpellingNotCalculated there and here.
1705e5dd7070Spatrick   assert(Spellings.size() < 15 &&
1706e5dd7070Spatrick          "Too many spellings, would step on SpellingNotCalculated in "
1707e5dd7070Spatrick          "AttributeCommonInfo");
1708e5dd7070Spatrick   for (auto I = Spellings.begin(), E = Spellings.end(); I != E; ++I, ++Idx) {
1709e5dd7070Spatrick     const FlattenedSpelling &S = *I;
1710e5dd7070Spatrick     const std::string &Variety = S.variety();
1711e5dd7070Spatrick     const std::string &Spelling = S.name();
1712e5dd7070Spatrick     const std::string &Namespace = S.nameSpace();
1713e5dd7070Spatrick     std::string EnumName;
1714e5dd7070Spatrick 
1715e5dd7070Spatrick     EnumName += (Variety + "_");
1716e5dd7070Spatrick     if (!Namespace.empty())
1717e5dd7070Spatrick       EnumName += (NormalizeNameForSpellingComparison(Namespace).str() +
1718e5dd7070Spatrick       "_");
1719e5dd7070Spatrick     EnumName += NormalizeNameForSpellingComparison(Spelling);
1720e5dd7070Spatrick 
1721e5dd7070Spatrick     // Even if the name is not unique, this spelling index corresponds to a
1722e5dd7070Spatrick     // particular enumerant name that we've calculated.
1723e5dd7070Spatrick     Map[Idx] = EnumName;
1724e5dd7070Spatrick 
1725e5dd7070Spatrick     // Since we have been stripping underscores to avoid trampling on the
1726e5dd7070Spatrick     // reserved namespace, we may have inadvertently created duplicate
1727e5dd7070Spatrick     // enumerant names. These duplicates are not considered part of the
1728e5dd7070Spatrick     // semantic spelling, and can be elided.
1729e5dd7070Spatrick     if (Uniques.find(EnumName) != Uniques.end())
1730e5dd7070Spatrick       continue;
1731e5dd7070Spatrick 
1732e5dd7070Spatrick     Uniques.insert(EnumName);
1733e5dd7070Spatrick     if (I != Spellings.begin())
1734e5dd7070Spatrick       Ret += ",\n";
1735e5dd7070Spatrick     // Duplicate spellings are not considered part of the semantic spelling
1736e5dd7070Spatrick     // enumeration, but the spelling index and semantic spelling values are
1737e5dd7070Spatrick     // meant to be equivalent, so we must specify a concrete value for each
1738e5dd7070Spatrick     // enumerator.
1739e5dd7070Spatrick     Ret += "    " + EnumName + " = " + llvm::utostr(Idx);
1740e5dd7070Spatrick   }
1741e5dd7070Spatrick   Ret += ",\n  SpellingNotCalculated = 15\n";
1742e5dd7070Spatrick   Ret += "\n  };\n\n";
1743e5dd7070Spatrick   return Ret;
1744e5dd7070Spatrick }
1745e5dd7070Spatrick 
WriteSemanticSpellingSwitch(const std::string & VarName,const SemanticSpellingMap & Map,raw_ostream & OS)1746e5dd7070Spatrick void WriteSemanticSpellingSwitch(const std::string &VarName,
1747e5dd7070Spatrick                                  const SemanticSpellingMap &Map,
1748e5dd7070Spatrick                                  raw_ostream &OS) {
1749e5dd7070Spatrick   OS << "  switch (" << VarName << ") {\n    default: "
1750e5dd7070Spatrick     << "llvm_unreachable(\"Unknown spelling list index\");\n";
1751e5dd7070Spatrick   for (const auto &I : Map)
1752e5dd7070Spatrick     OS << "    case " << I.first << ": return " << I.second << ";\n";
1753e5dd7070Spatrick   OS << "  }\n";
1754e5dd7070Spatrick }
1755e5dd7070Spatrick 
1756e5dd7070Spatrick // Emits the LateParsed property for attributes.
emitClangAttrLateParsedList(RecordKeeper & Records,raw_ostream & OS)1757e5dd7070Spatrick static void emitClangAttrLateParsedList(RecordKeeper &Records, raw_ostream &OS) {
1758e5dd7070Spatrick   OS << "#if defined(CLANG_ATTR_LATE_PARSED_LIST)\n";
1759e5dd7070Spatrick   std::vector<Record*> Attrs = Records.getAllDerivedDefinitions("Attr");
1760e5dd7070Spatrick 
1761e5dd7070Spatrick   for (const auto *Attr : Attrs) {
1762e5dd7070Spatrick     bool LateParsed = Attr->getValueAsBit("LateParsed");
1763e5dd7070Spatrick 
1764e5dd7070Spatrick     if (LateParsed) {
1765e5dd7070Spatrick       std::vector<FlattenedSpelling> Spellings = GetFlattenedSpellings(*Attr);
1766e5dd7070Spatrick 
1767e5dd7070Spatrick       // FIXME: Handle non-GNU attributes
1768e5dd7070Spatrick       for (const auto &I : Spellings) {
1769e5dd7070Spatrick         if (I.variety() != "GNU")
1770e5dd7070Spatrick           continue;
1771e5dd7070Spatrick         OS << ".Case(\"" << I.name() << "\", " << LateParsed << ")\n";
1772e5dd7070Spatrick       }
1773e5dd7070Spatrick     }
1774e5dd7070Spatrick   }
1775e5dd7070Spatrick   OS << "#endif // CLANG_ATTR_LATE_PARSED_LIST\n\n";
1776e5dd7070Spatrick }
1777e5dd7070Spatrick 
hasGNUorCXX11Spelling(const Record & Attribute)1778e5dd7070Spatrick static bool hasGNUorCXX11Spelling(const Record &Attribute) {
1779e5dd7070Spatrick   std::vector<FlattenedSpelling> Spellings = GetFlattenedSpellings(Attribute);
1780e5dd7070Spatrick   for (const auto &I : Spellings) {
1781e5dd7070Spatrick     if (I.variety() == "GNU" || I.variety() == "CXX11")
1782e5dd7070Spatrick       return true;
1783e5dd7070Spatrick   }
1784e5dd7070Spatrick   return false;
1785e5dd7070Spatrick }
1786e5dd7070Spatrick 
1787e5dd7070Spatrick namespace {
1788e5dd7070Spatrick 
1789e5dd7070Spatrick struct AttributeSubjectMatchRule {
1790e5dd7070Spatrick   const Record *MetaSubject;
1791e5dd7070Spatrick   const Record *Constraint;
1792e5dd7070Spatrick 
AttributeSubjectMatchRule__anone86b54960311::AttributeSubjectMatchRule1793e5dd7070Spatrick   AttributeSubjectMatchRule(const Record *MetaSubject, const Record *Constraint)
1794e5dd7070Spatrick       : MetaSubject(MetaSubject), Constraint(Constraint) {
1795e5dd7070Spatrick     assert(MetaSubject && "Missing subject");
1796e5dd7070Spatrick   }
1797e5dd7070Spatrick 
isSubRule__anone86b54960311::AttributeSubjectMatchRule1798e5dd7070Spatrick   bool isSubRule() const { return Constraint != nullptr; }
1799e5dd7070Spatrick 
getSubjects__anone86b54960311::AttributeSubjectMatchRule1800e5dd7070Spatrick   std::vector<Record *> getSubjects() const {
1801e5dd7070Spatrick     return (Constraint ? Constraint : MetaSubject)
1802e5dd7070Spatrick         ->getValueAsListOfDefs("Subjects");
1803e5dd7070Spatrick   }
1804e5dd7070Spatrick 
getLangOpts__anone86b54960311::AttributeSubjectMatchRule1805e5dd7070Spatrick   std::vector<Record *> getLangOpts() const {
1806e5dd7070Spatrick     if (Constraint) {
1807e5dd7070Spatrick       // Lookup the options in the sub-rule first, in case the sub-rule
1808e5dd7070Spatrick       // overrides the rules options.
1809e5dd7070Spatrick       std::vector<Record *> Opts = Constraint->getValueAsListOfDefs("LangOpts");
1810e5dd7070Spatrick       if (!Opts.empty())
1811e5dd7070Spatrick         return Opts;
1812e5dd7070Spatrick     }
1813e5dd7070Spatrick     return MetaSubject->getValueAsListOfDefs("LangOpts");
1814e5dd7070Spatrick   }
1815e5dd7070Spatrick 
1816e5dd7070Spatrick   // Abstract rules are used only for sub-rules
isAbstractRule__anone86b54960311::AttributeSubjectMatchRule1817e5dd7070Spatrick   bool isAbstractRule() const { return getSubjects().empty(); }
1818e5dd7070Spatrick 
getName__anone86b54960311::AttributeSubjectMatchRule1819e5dd7070Spatrick   StringRef getName() const {
1820e5dd7070Spatrick     return (Constraint ? Constraint : MetaSubject)->getValueAsString("Name");
1821e5dd7070Spatrick   }
1822e5dd7070Spatrick 
isNegatedSubRule__anone86b54960311::AttributeSubjectMatchRule1823e5dd7070Spatrick   bool isNegatedSubRule() const {
1824e5dd7070Spatrick     assert(isSubRule() && "Not a sub-rule");
1825e5dd7070Spatrick     return Constraint->getValueAsBit("Negated");
1826e5dd7070Spatrick   }
1827e5dd7070Spatrick 
getSpelling__anone86b54960311::AttributeSubjectMatchRule1828e5dd7070Spatrick   std::string getSpelling() const {
1829ec727ea7Spatrick     std::string Result = std::string(MetaSubject->getValueAsString("Name"));
1830e5dd7070Spatrick     if (isSubRule()) {
1831e5dd7070Spatrick       Result += '(';
1832e5dd7070Spatrick       if (isNegatedSubRule())
1833e5dd7070Spatrick         Result += "unless(";
1834e5dd7070Spatrick       Result += getName();
1835e5dd7070Spatrick       if (isNegatedSubRule())
1836e5dd7070Spatrick         Result += ')';
1837e5dd7070Spatrick       Result += ')';
1838e5dd7070Spatrick     }
1839e5dd7070Spatrick     return Result;
1840e5dd7070Spatrick   }
1841e5dd7070Spatrick 
getEnumValueName__anone86b54960311::AttributeSubjectMatchRule1842e5dd7070Spatrick   std::string getEnumValueName() const {
1843e5dd7070Spatrick     SmallString<128> Result;
1844e5dd7070Spatrick     Result += "SubjectMatchRule_";
1845e5dd7070Spatrick     Result += MetaSubject->getValueAsString("Name");
1846e5dd7070Spatrick     if (isSubRule()) {
1847e5dd7070Spatrick       Result += "_";
1848e5dd7070Spatrick       if (isNegatedSubRule())
1849e5dd7070Spatrick         Result += "not_";
1850e5dd7070Spatrick       Result += Constraint->getValueAsString("Name");
1851e5dd7070Spatrick     }
1852e5dd7070Spatrick     if (isAbstractRule())
1853e5dd7070Spatrick       Result += "_abstract";
1854ec727ea7Spatrick     return std::string(Result.str());
1855e5dd7070Spatrick   }
1856e5dd7070Spatrick 
getEnumValue__anone86b54960311::AttributeSubjectMatchRule1857e5dd7070Spatrick   std::string getEnumValue() const { return "attr::" + getEnumValueName(); }
1858e5dd7070Spatrick 
1859e5dd7070Spatrick   static const char *EnumName;
1860e5dd7070Spatrick };
1861e5dd7070Spatrick 
1862e5dd7070Spatrick const char *AttributeSubjectMatchRule::EnumName = "attr::SubjectMatchRule";
1863e5dd7070Spatrick 
1864e5dd7070Spatrick struct PragmaClangAttributeSupport {
1865e5dd7070Spatrick   std::vector<AttributeSubjectMatchRule> Rules;
1866e5dd7070Spatrick 
1867e5dd7070Spatrick   class RuleOrAggregateRuleSet {
1868e5dd7070Spatrick     std::vector<AttributeSubjectMatchRule> Rules;
1869e5dd7070Spatrick     bool IsRule;
RuleOrAggregateRuleSet(ArrayRef<AttributeSubjectMatchRule> Rules,bool IsRule)1870e5dd7070Spatrick     RuleOrAggregateRuleSet(ArrayRef<AttributeSubjectMatchRule> Rules,
1871e5dd7070Spatrick                            bool IsRule)
1872e5dd7070Spatrick         : Rules(Rules), IsRule(IsRule) {}
1873e5dd7070Spatrick 
1874e5dd7070Spatrick   public:
isRule() const1875e5dd7070Spatrick     bool isRule() const { return IsRule; }
1876e5dd7070Spatrick 
getRule() const1877e5dd7070Spatrick     const AttributeSubjectMatchRule &getRule() const {
1878e5dd7070Spatrick       assert(IsRule && "not a rule!");
1879e5dd7070Spatrick       return Rules[0];
1880e5dd7070Spatrick     }
1881e5dd7070Spatrick 
getAggregateRuleSet() const1882e5dd7070Spatrick     ArrayRef<AttributeSubjectMatchRule> getAggregateRuleSet() const {
1883e5dd7070Spatrick       return Rules;
1884e5dd7070Spatrick     }
1885e5dd7070Spatrick 
1886e5dd7070Spatrick     static RuleOrAggregateRuleSet
getRule(const AttributeSubjectMatchRule & Rule)1887e5dd7070Spatrick     getRule(const AttributeSubjectMatchRule &Rule) {
1888e5dd7070Spatrick       return RuleOrAggregateRuleSet(Rule, /*IsRule=*/true);
1889e5dd7070Spatrick     }
1890e5dd7070Spatrick     static RuleOrAggregateRuleSet
getAggregateRuleSet(ArrayRef<AttributeSubjectMatchRule> Rules)1891e5dd7070Spatrick     getAggregateRuleSet(ArrayRef<AttributeSubjectMatchRule> Rules) {
1892e5dd7070Spatrick       return RuleOrAggregateRuleSet(Rules, /*IsRule=*/false);
1893e5dd7070Spatrick     }
1894e5dd7070Spatrick   };
1895e5dd7070Spatrick   llvm::DenseMap<const Record *, RuleOrAggregateRuleSet> SubjectsToRules;
1896e5dd7070Spatrick 
1897e5dd7070Spatrick   PragmaClangAttributeSupport(RecordKeeper &Records);
1898e5dd7070Spatrick 
1899e5dd7070Spatrick   bool isAttributedSupported(const Record &Attribute);
1900e5dd7070Spatrick 
1901e5dd7070Spatrick   void emitMatchRuleList(raw_ostream &OS);
1902e5dd7070Spatrick 
1903ec727ea7Spatrick   void generateStrictConformsTo(const Record &Attr, raw_ostream &OS);
1904e5dd7070Spatrick 
1905e5dd7070Spatrick   void generateParsingHelpers(raw_ostream &OS);
1906e5dd7070Spatrick };
1907e5dd7070Spatrick 
1908e5dd7070Spatrick } // end anonymous namespace
1909e5dd7070Spatrick 
isSupportedPragmaClangAttributeSubject(const Record & Subject)1910a9ac8606Spatrick static bool isSupportedPragmaClangAttributeSubject(const Record &Subject) {
1911a9ac8606Spatrick   // FIXME: #pragma clang attribute does not currently support statement
1912a9ac8606Spatrick   // attributes, so test whether the subject is one that appertains to a
1913a9ac8606Spatrick   // declaration node. However, it may be reasonable for support for statement
1914a9ac8606Spatrick   // attributes to be added.
1915a9ac8606Spatrick   if (Subject.isSubClassOf("DeclNode") || Subject.isSubClassOf("DeclBase") ||
1916a9ac8606Spatrick       Subject.getName() == "DeclBase")
1917a9ac8606Spatrick     return true;
1918a9ac8606Spatrick 
1919a9ac8606Spatrick   if (Subject.isSubClassOf("SubsetSubject"))
1920a9ac8606Spatrick     return isSupportedPragmaClangAttributeSubject(
1921a9ac8606Spatrick         *Subject.getValueAsDef("Base"));
1922a9ac8606Spatrick 
1923a9ac8606Spatrick   return false;
1924a9ac8606Spatrick }
1925a9ac8606Spatrick 
doesDeclDeriveFrom(const Record * D,const Record * Base)1926e5dd7070Spatrick static bool doesDeclDeriveFrom(const Record *D, const Record *Base) {
1927e5dd7070Spatrick   const Record *CurrentBase = D->getValueAsOptionalDef(BaseFieldName);
1928e5dd7070Spatrick   if (!CurrentBase)
1929e5dd7070Spatrick     return false;
1930e5dd7070Spatrick   if (CurrentBase == Base)
1931e5dd7070Spatrick     return true;
1932e5dd7070Spatrick   return doesDeclDeriveFrom(CurrentBase, Base);
1933e5dd7070Spatrick }
1934e5dd7070Spatrick 
PragmaClangAttributeSupport(RecordKeeper & Records)1935e5dd7070Spatrick PragmaClangAttributeSupport::PragmaClangAttributeSupport(
1936e5dd7070Spatrick     RecordKeeper &Records) {
1937e5dd7070Spatrick   std::vector<Record *> MetaSubjects =
1938e5dd7070Spatrick       Records.getAllDerivedDefinitions("AttrSubjectMatcherRule");
1939e5dd7070Spatrick   auto MapFromSubjectsToRules = [this](const Record *SubjectContainer,
1940e5dd7070Spatrick                                        const Record *MetaSubject,
1941e5dd7070Spatrick                                        const Record *Constraint) {
1942e5dd7070Spatrick     Rules.emplace_back(MetaSubject, Constraint);
1943e5dd7070Spatrick     std::vector<Record *> ApplicableSubjects =
1944e5dd7070Spatrick         SubjectContainer->getValueAsListOfDefs("Subjects");
1945e5dd7070Spatrick     for (const auto *Subject : ApplicableSubjects) {
1946e5dd7070Spatrick       bool Inserted =
1947e5dd7070Spatrick           SubjectsToRules
1948e5dd7070Spatrick               .try_emplace(Subject, RuleOrAggregateRuleSet::getRule(
1949e5dd7070Spatrick                                         AttributeSubjectMatchRule(MetaSubject,
1950e5dd7070Spatrick                                                                   Constraint)))
1951e5dd7070Spatrick               .second;
1952e5dd7070Spatrick       if (!Inserted) {
1953e5dd7070Spatrick         PrintFatalError("Attribute subject match rules should not represent"
1954e5dd7070Spatrick                         "same attribute subjects.");
1955e5dd7070Spatrick       }
1956e5dd7070Spatrick     }
1957e5dd7070Spatrick   };
1958e5dd7070Spatrick   for (const auto *MetaSubject : MetaSubjects) {
1959e5dd7070Spatrick     MapFromSubjectsToRules(MetaSubject, MetaSubject, /*Constraints=*/nullptr);
1960e5dd7070Spatrick     std::vector<Record *> Constraints =
1961e5dd7070Spatrick         MetaSubject->getValueAsListOfDefs("Constraints");
1962e5dd7070Spatrick     for (const auto *Constraint : Constraints)
1963e5dd7070Spatrick       MapFromSubjectsToRules(Constraint, MetaSubject, Constraint);
1964e5dd7070Spatrick   }
1965e5dd7070Spatrick 
1966e5dd7070Spatrick   std::vector<Record *> Aggregates =
1967e5dd7070Spatrick       Records.getAllDerivedDefinitions("AttrSubjectMatcherAggregateRule");
1968e5dd7070Spatrick   std::vector<Record *> DeclNodes =
1969e5dd7070Spatrick     Records.getAllDerivedDefinitions(DeclNodeClassName);
1970e5dd7070Spatrick   for (const auto *Aggregate : Aggregates) {
1971e5dd7070Spatrick     Record *SubjectDecl = Aggregate->getValueAsDef("Subject");
1972e5dd7070Spatrick 
1973e5dd7070Spatrick     // Gather sub-classes of the aggregate subject that act as attribute
1974e5dd7070Spatrick     // subject rules.
1975e5dd7070Spatrick     std::vector<AttributeSubjectMatchRule> Rules;
1976e5dd7070Spatrick     for (const auto *D : DeclNodes) {
1977e5dd7070Spatrick       if (doesDeclDeriveFrom(D, SubjectDecl)) {
1978e5dd7070Spatrick         auto It = SubjectsToRules.find(D);
1979e5dd7070Spatrick         if (It == SubjectsToRules.end())
1980e5dd7070Spatrick           continue;
1981e5dd7070Spatrick         if (!It->second.isRule() || It->second.getRule().isSubRule())
1982e5dd7070Spatrick           continue; // Assume that the rule will be included as well.
1983e5dd7070Spatrick         Rules.push_back(It->second.getRule());
1984e5dd7070Spatrick       }
1985e5dd7070Spatrick     }
1986e5dd7070Spatrick 
1987e5dd7070Spatrick     bool Inserted =
1988e5dd7070Spatrick         SubjectsToRules
1989e5dd7070Spatrick             .try_emplace(SubjectDecl,
1990e5dd7070Spatrick                          RuleOrAggregateRuleSet::getAggregateRuleSet(Rules))
1991e5dd7070Spatrick             .second;
1992e5dd7070Spatrick     if (!Inserted) {
1993e5dd7070Spatrick       PrintFatalError("Attribute subject match rules should not represent"
1994e5dd7070Spatrick                       "same attribute subjects.");
1995e5dd7070Spatrick     }
1996e5dd7070Spatrick   }
1997e5dd7070Spatrick }
1998e5dd7070Spatrick 
1999e5dd7070Spatrick static PragmaClangAttributeSupport &
getPragmaAttributeSupport(RecordKeeper & Records)2000e5dd7070Spatrick getPragmaAttributeSupport(RecordKeeper &Records) {
2001e5dd7070Spatrick   static PragmaClangAttributeSupport Instance(Records);
2002e5dd7070Spatrick   return Instance;
2003e5dd7070Spatrick }
2004e5dd7070Spatrick 
emitMatchRuleList(raw_ostream & OS)2005e5dd7070Spatrick void PragmaClangAttributeSupport::emitMatchRuleList(raw_ostream &OS) {
2006e5dd7070Spatrick   OS << "#ifndef ATTR_MATCH_SUB_RULE\n";
2007e5dd7070Spatrick   OS << "#define ATTR_MATCH_SUB_RULE(Value, Spelling, IsAbstract, Parent, "
2008e5dd7070Spatrick         "IsNegated) "
2009e5dd7070Spatrick      << "ATTR_MATCH_RULE(Value, Spelling, IsAbstract)\n";
2010e5dd7070Spatrick   OS << "#endif\n";
2011e5dd7070Spatrick   for (const auto &Rule : Rules) {
2012e5dd7070Spatrick     OS << (Rule.isSubRule() ? "ATTR_MATCH_SUB_RULE" : "ATTR_MATCH_RULE") << '(';
2013e5dd7070Spatrick     OS << Rule.getEnumValueName() << ", \"" << Rule.getSpelling() << "\", "
2014e5dd7070Spatrick        << Rule.isAbstractRule();
2015e5dd7070Spatrick     if (Rule.isSubRule())
2016e5dd7070Spatrick       OS << ", "
2017e5dd7070Spatrick          << AttributeSubjectMatchRule(Rule.MetaSubject, nullptr).getEnumValue()
2018e5dd7070Spatrick          << ", " << Rule.isNegatedSubRule();
2019e5dd7070Spatrick     OS << ")\n";
2020e5dd7070Spatrick   }
2021e5dd7070Spatrick   OS << "#undef ATTR_MATCH_SUB_RULE\n";
2022e5dd7070Spatrick }
2023e5dd7070Spatrick 
isAttributedSupported(const Record & Attribute)2024e5dd7070Spatrick bool PragmaClangAttributeSupport::isAttributedSupported(
2025e5dd7070Spatrick     const Record &Attribute) {
2026e5dd7070Spatrick   // If the attribute explicitly specified whether to support #pragma clang
2027e5dd7070Spatrick   // attribute, use that setting.
2028e5dd7070Spatrick   bool Unset;
2029e5dd7070Spatrick   bool SpecifiedResult =
2030e5dd7070Spatrick     Attribute.getValueAsBitOrUnset("PragmaAttributeSupport", Unset);
2031e5dd7070Spatrick   if (!Unset)
2032e5dd7070Spatrick     return SpecifiedResult;
2033e5dd7070Spatrick 
2034e5dd7070Spatrick   // Opt-out rules:
2035e5dd7070Spatrick   // An attribute requires delayed parsing (LateParsed is on)
2036e5dd7070Spatrick   if (Attribute.getValueAsBit("LateParsed"))
2037e5dd7070Spatrick     return false;
2038e5dd7070Spatrick   // An attribute has no GNU/CXX11 spelling
2039e5dd7070Spatrick   if (!hasGNUorCXX11Spelling(Attribute))
2040e5dd7070Spatrick     return false;
2041e5dd7070Spatrick   // An attribute subject list has a subject that isn't covered by one of the
2042e5dd7070Spatrick   // subject match rules or has no subjects at all.
2043e5dd7070Spatrick   if (Attribute.isValueUnset("Subjects"))
2044e5dd7070Spatrick     return false;
2045e5dd7070Spatrick   const Record *SubjectObj = Attribute.getValueAsDef("Subjects");
2046e5dd7070Spatrick   std::vector<Record *> Subjects = SubjectObj->getValueAsListOfDefs("Subjects");
2047a9ac8606Spatrick   bool HasAtLeastOneValidSubject = false;
2048e5dd7070Spatrick   for (const auto *Subject : Subjects) {
2049a9ac8606Spatrick     if (!isSupportedPragmaClangAttributeSubject(*Subject))
2050a9ac8606Spatrick       continue;
2051e5dd7070Spatrick     if (SubjectsToRules.find(Subject) == SubjectsToRules.end())
2052e5dd7070Spatrick       return false;
2053a9ac8606Spatrick     HasAtLeastOneValidSubject = true;
2054e5dd7070Spatrick   }
2055a9ac8606Spatrick   return HasAtLeastOneValidSubject;
2056e5dd7070Spatrick }
2057e5dd7070Spatrick 
GenerateTestExpression(ArrayRef<Record * > LangOpts)2058e5dd7070Spatrick static std::string GenerateTestExpression(ArrayRef<Record *> LangOpts) {
2059e5dd7070Spatrick   std::string Test;
2060e5dd7070Spatrick 
2061e5dd7070Spatrick   for (auto *E : LangOpts) {
2062e5dd7070Spatrick     if (!Test.empty())
2063e5dd7070Spatrick       Test += " || ";
2064e5dd7070Spatrick 
2065e5dd7070Spatrick     const StringRef Code = E->getValueAsString("CustomCode");
2066e5dd7070Spatrick     if (!Code.empty()) {
2067e5dd7070Spatrick       Test += "(";
2068e5dd7070Spatrick       Test += Code;
2069e5dd7070Spatrick       Test += ")";
2070ec727ea7Spatrick       if (!E->getValueAsString("Name").empty()) {
2071ec727ea7Spatrick         PrintWarning(
2072ec727ea7Spatrick             E->getLoc(),
2073ec727ea7Spatrick             "non-empty 'Name' field ignored because 'CustomCode' was supplied");
2074ec727ea7Spatrick       }
2075e5dd7070Spatrick     } else {
2076e5dd7070Spatrick       Test += "LangOpts.";
2077e5dd7070Spatrick       Test += E->getValueAsString("Name");
2078e5dd7070Spatrick     }
2079e5dd7070Spatrick   }
2080e5dd7070Spatrick 
2081e5dd7070Spatrick   if (Test.empty())
2082e5dd7070Spatrick     return "true";
2083e5dd7070Spatrick 
2084e5dd7070Spatrick   return Test;
2085e5dd7070Spatrick }
2086e5dd7070Spatrick 
2087ec727ea7Spatrick void
generateStrictConformsTo(const Record & Attr,raw_ostream & OS)2088e5dd7070Spatrick PragmaClangAttributeSupport::generateStrictConformsTo(const Record &Attr,
2089e5dd7070Spatrick                                                       raw_ostream &OS) {
2090ec727ea7Spatrick   if (!isAttributedSupported(Attr) || Attr.isValueUnset("Subjects"))
2091ec727ea7Spatrick     return;
2092e5dd7070Spatrick   // Generate a function that constructs a set of matching rules that describe
2093e5dd7070Spatrick   // to which declarations the attribute should apply to.
2094ec727ea7Spatrick   OS << "void getPragmaAttributeMatchRules("
2095ec727ea7Spatrick      << "llvm::SmallVectorImpl<std::pair<"
2096e5dd7070Spatrick      << AttributeSubjectMatchRule::EnumName
2097ec727ea7Spatrick      << ", bool>> &MatchRules, const LangOptions &LangOpts) const override {\n";
2098e5dd7070Spatrick   const Record *SubjectObj = Attr.getValueAsDef("Subjects");
2099e5dd7070Spatrick   std::vector<Record *> Subjects = SubjectObj->getValueAsListOfDefs("Subjects");
2100e5dd7070Spatrick   for (const auto *Subject : Subjects) {
2101a9ac8606Spatrick     if (!isSupportedPragmaClangAttributeSubject(*Subject))
2102a9ac8606Spatrick       continue;
2103e5dd7070Spatrick     auto It = SubjectsToRules.find(Subject);
2104e5dd7070Spatrick     assert(It != SubjectsToRules.end() &&
2105e5dd7070Spatrick            "This attribute is unsupported by #pragma clang attribute");
2106e5dd7070Spatrick     for (const auto &Rule : It->getSecond().getAggregateRuleSet()) {
2107e5dd7070Spatrick       // The rule might be language specific, so only subtract it from the given
2108e5dd7070Spatrick       // rules if the specific language options are specified.
2109e5dd7070Spatrick       std::vector<Record *> LangOpts = Rule.getLangOpts();
2110e5dd7070Spatrick       OS << "  MatchRules.push_back(std::make_pair(" << Rule.getEnumValue()
2111e5dd7070Spatrick          << ", /*IsSupported=*/" << GenerateTestExpression(LangOpts)
2112e5dd7070Spatrick          << "));\n";
2113e5dd7070Spatrick     }
2114e5dd7070Spatrick   }
2115e5dd7070Spatrick   OS << "}\n\n";
2116e5dd7070Spatrick }
2117e5dd7070Spatrick 
generateParsingHelpers(raw_ostream & OS)2118e5dd7070Spatrick void PragmaClangAttributeSupport::generateParsingHelpers(raw_ostream &OS) {
2119e5dd7070Spatrick   // Generate routines that check the names of sub-rules.
2120*12c85518Srobert   OS << "std::optional<attr::SubjectMatchRule> "
2121e5dd7070Spatrick         "defaultIsAttributeSubjectMatchSubRuleFor(StringRef, bool) {\n";
2122*12c85518Srobert   OS << "  return std::nullopt;\n";
2123e5dd7070Spatrick   OS << "}\n\n";
2124e5dd7070Spatrick 
2125*12c85518Srobert   llvm::MapVector<const Record *, std::vector<AttributeSubjectMatchRule>>
2126e5dd7070Spatrick       SubMatchRules;
2127e5dd7070Spatrick   for (const auto &Rule : Rules) {
2128e5dd7070Spatrick     if (!Rule.isSubRule())
2129e5dd7070Spatrick       continue;
2130e5dd7070Spatrick     SubMatchRules[Rule.MetaSubject].push_back(Rule);
2131e5dd7070Spatrick   }
2132e5dd7070Spatrick 
2133e5dd7070Spatrick   for (const auto &SubMatchRule : SubMatchRules) {
2134*12c85518Srobert     OS << "std::optional<attr::SubjectMatchRule> "
2135*12c85518Srobert           "isAttributeSubjectMatchSubRuleFor_"
2136e5dd7070Spatrick        << SubMatchRule.first->getValueAsString("Name")
2137e5dd7070Spatrick        << "(StringRef Name, bool IsUnless) {\n";
2138e5dd7070Spatrick     OS << "  if (IsUnless)\n";
2139e5dd7070Spatrick     OS << "    return "
2140*12c85518Srobert           "llvm::StringSwitch<std::optional<attr::SubjectMatchRule>>(Name).\n";
2141e5dd7070Spatrick     for (const auto &Rule : SubMatchRule.second) {
2142e5dd7070Spatrick       if (Rule.isNegatedSubRule())
2143e5dd7070Spatrick         OS << "    Case(\"" << Rule.getName() << "\", " << Rule.getEnumValue()
2144e5dd7070Spatrick            << ").\n";
2145e5dd7070Spatrick     }
2146*12c85518Srobert     OS << "    Default(std::nullopt);\n";
2147e5dd7070Spatrick     OS << "  return "
2148*12c85518Srobert           "llvm::StringSwitch<std::optional<attr::SubjectMatchRule>>(Name).\n";
2149e5dd7070Spatrick     for (const auto &Rule : SubMatchRule.second) {
2150e5dd7070Spatrick       if (!Rule.isNegatedSubRule())
2151e5dd7070Spatrick         OS << "  Case(\"" << Rule.getName() << "\", " << Rule.getEnumValue()
2152e5dd7070Spatrick            << ").\n";
2153e5dd7070Spatrick     }
2154*12c85518Srobert     OS << "  Default(std::nullopt);\n";
2155e5dd7070Spatrick     OS << "}\n\n";
2156e5dd7070Spatrick   }
2157e5dd7070Spatrick 
2158e5dd7070Spatrick   // Generate the function that checks for the top-level rules.
2159*12c85518Srobert   OS << "std::pair<std::optional<attr::SubjectMatchRule>, "
2160*12c85518Srobert         "std::optional<attr::SubjectMatchRule> (*)(StringRef, "
2161e5dd7070Spatrick         "bool)> isAttributeSubjectMatchRule(StringRef Name) {\n";
2162e5dd7070Spatrick   OS << "  return "
2163*12c85518Srobert         "llvm::StringSwitch<std::pair<std::optional<attr::SubjectMatchRule>, "
2164*12c85518Srobert         "std::optional<attr::SubjectMatchRule> (*) (StringRef, "
2165e5dd7070Spatrick         "bool)>>(Name).\n";
2166e5dd7070Spatrick   for (const auto &Rule : Rules) {
2167e5dd7070Spatrick     if (Rule.isSubRule())
2168e5dd7070Spatrick       continue;
2169e5dd7070Spatrick     std::string SubRuleFunction;
2170e5dd7070Spatrick     if (SubMatchRules.count(Rule.MetaSubject))
2171e5dd7070Spatrick       SubRuleFunction =
2172e5dd7070Spatrick           ("isAttributeSubjectMatchSubRuleFor_" + Rule.getName()).str();
2173e5dd7070Spatrick     else
2174e5dd7070Spatrick       SubRuleFunction = "defaultIsAttributeSubjectMatchSubRuleFor";
2175e5dd7070Spatrick     OS << "  Case(\"" << Rule.getName() << "\", std::make_pair("
2176e5dd7070Spatrick        << Rule.getEnumValue() << ", " << SubRuleFunction << ")).\n";
2177e5dd7070Spatrick   }
2178*12c85518Srobert   OS << "  Default(std::make_pair(std::nullopt, "
2179e5dd7070Spatrick         "defaultIsAttributeSubjectMatchSubRuleFor));\n";
2180e5dd7070Spatrick   OS << "}\n\n";
2181e5dd7070Spatrick 
2182e5dd7070Spatrick   // Generate the function that checks for the submatch rules.
2183e5dd7070Spatrick   OS << "const char *validAttributeSubjectMatchSubRules("
2184e5dd7070Spatrick      << AttributeSubjectMatchRule::EnumName << " Rule) {\n";
2185e5dd7070Spatrick   OS << "  switch (Rule) {\n";
2186e5dd7070Spatrick   for (const auto &SubMatchRule : SubMatchRules) {
2187e5dd7070Spatrick     OS << "  case "
2188e5dd7070Spatrick        << AttributeSubjectMatchRule(SubMatchRule.first, nullptr).getEnumValue()
2189e5dd7070Spatrick        << ":\n";
2190e5dd7070Spatrick     OS << "  return \"'";
2191e5dd7070Spatrick     bool IsFirst = true;
2192e5dd7070Spatrick     for (const auto &Rule : SubMatchRule.second) {
2193e5dd7070Spatrick       if (!IsFirst)
2194e5dd7070Spatrick         OS << ", '";
2195e5dd7070Spatrick       IsFirst = false;
2196e5dd7070Spatrick       if (Rule.isNegatedSubRule())
2197e5dd7070Spatrick         OS << "unless(";
2198e5dd7070Spatrick       OS << Rule.getName();
2199e5dd7070Spatrick       if (Rule.isNegatedSubRule())
2200e5dd7070Spatrick         OS << ')';
2201e5dd7070Spatrick       OS << "'";
2202e5dd7070Spatrick     }
2203e5dd7070Spatrick     OS << "\";\n";
2204e5dd7070Spatrick   }
2205e5dd7070Spatrick   OS << "  default: return nullptr;\n";
2206e5dd7070Spatrick   OS << "  }\n";
2207e5dd7070Spatrick   OS << "}\n\n";
2208e5dd7070Spatrick }
2209e5dd7070Spatrick 
2210e5dd7070Spatrick template <typename Fn>
forEachUniqueSpelling(const Record & Attr,Fn && F)2211e5dd7070Spatrick static void forEachUniqueSpelling(const Record &Attr, Fn &&F) {
2212e5dd7070Spatrick   std::vector<FlattenedSpelling> Spellings = GetFlattenedSpellings(Attr);
2213e5dd7070Spatrick   SmallDenseSet<StringRef, 8> Seen;
2214e5dd7070Spatrick   for (const FlattenedSpelling &S : Spellings) {
2215e5dd7070Spatrick     if (Seen.insert(S.name()).second)
2216e5dd7070Spatrick       F(S);
2217e5dd7070Spatrick   }
2218e5dd7070Spatrick }
2219e5dd7070Spatrick 
isTypeArgument(const Record * Arg)2220*12c85518Srobert static bool isTypeArgument(const Record *Arg) {
2221*12c85518Srobert   return !Arg->getSuperClasses().empty() &&
2222*12c85518Srobert          Arg->getSuperClasses().back().first->getName() == "TypeArgument";
2223*12c85518Srobert }
2224*12c85518Srobert 
2225e5dd7070Spatrick /// Emits the first-argument-is-type property for attributes.
emitClangAttrTypeArgList(RecordKeeper & Records,raw_ostream & OS)2226e5dd7070Spatrick static void emitClangAttrTypeArgList(RecordKeeper &Records, raw_ostream &OS) {
2227e5dd7070Spatrick   OS << "#if defined(CLANG_ATTR_TYPE_ARG_LIST)\n";
2228e5dd7070Spatrick   std::vector<Record *> Attrs = Records.getAllDerivedDefinitions("Attr");
2229e5dd7070Spatrick 
2230e5dd7070Spatrick   for (const auto *Attr : Attrs) {
2231e5dd7070Spatrick     // Determine whether the first argument is a type.
2232e5dd7070Spatrick     std::vector<Record *> Args = Attr->getValueAsListOfDefs("Args");
2233e5dd7070Spatrick     if (Args.empty())
2234e5dd7070Spatrick       continue;
2235e5dd7070Spatrick 
2236*12c85518Srobert     if (!isTypeArgument(Args[0]))
2237e5dd7070Spatrick       continue;
2238e5dd7070Spatrick 
2239e5dd7070Spatrick     // All these spellings take a single type argument.
2240e5dd7070Spatrick     forEachUniqueSpelling(*Attr, [&](const FlattenedSpelling &S) {
2241e5dd7070Spatrick       OS << ".Case(\"" << S.name() << "\", " << "true" << ")\n";
2242e5dd7070Spatrick     });
2243e5dd7070Spatrick   }
2244e5dd7070Spatrick   OS << "#endif // CLANG_ATTR_TYPE_ARG_LIST\n\n";
2245e5dd7070Spatrick }
2246e5dd7070Spatrick 
2247e5dd7070Spatrick /// Emits the parse-arguments-in-unevaluated-context property for
2248e5dd7070Spatrick /// attributes.
emitClangAttrArgContextList(RecordKeeper & Records,raw_ostream & OS)2249e5dd7070Spatrick static void emitClangAttrArgContextList(RecordKeeper &Records, raw_ostream &OS) {
2250e5dd7070Spatrick   OS << "#if defined(CLANG_ATTR_ARG_CONTEXT_LIST)\n";
2251e5dd7070Spatrick   ParsedAttrMap Attrs = getParsedAttrList(Records);
2252e5dd7070Spatrick   for (const auto &I : Attrs) {
2253e5dd7070Spatrick     const Record &Attr = *I.second;
2254e5dd7070Spatrick 
2255e5dd7070Spatrick     if (!Attr.getValueAsBit("ParseArgumentsAsUnevaluated"))
2256e5dd7070Spatrick       continue;
2257e5dd7070Spatrick 
2258e5dd7070Spatrick     // All these spellings take are parsed unevaluated.
2259e5dd7070Spatrick     forEachUniqueSpelling(Attr, [&](const FlattenedSpelling &S) {
2260e5dd7070Spatrick       OS << ".Case(\"" << S.name() << "\", " << "true" << ")\n";
2261e5dd7070Spatrick     });
2262e5dd7070Spatrick   }
2263e5dd7070Spatrick   OS << "#endif // CLANG_ATTR_ARG_CONTEXT_LIST\n\n";
2264e5dd7070Spatrick }
2265e5dd7070Spatrick 
isIdentifierArgument(const Record * Arg)2266*12c85518Srobert static bool isIdentifierArgument(const Record *Arg) {
2267e5dd7070Spatrick   return !Arg->getSuperClasses().empty() &&
2268e5dd7070Spatrick     llvm::StringSwitch<bool>(Arg->getSuperClasses().back().first->getName())
2269e5dd7070Spatrick     .Case("IdentifierArgument", true)
2270e5dd7070Spatrick     .Case("EnumArgument", true)
2271e5dd7070Spatrick     .Case("VariadicEnumArgument", true)
2272e5dd7070Spatrick     .Default(false);
2273e5dd7070Spatrick }
2274e5dd7070Spatrick 
isVariadicIdentifierArgument(const Record * Arg)2275*12c85518Srobert static bool isVariadicIdentifierArgument(const Record *Arg) {
2276e5dd7070Spatrick   return !Arg->getSuperClasses().empty() &&
2277e5dd7070Spatrick          llvm::StringSwitch<bool>(
2278e5dd7070Spatrick              Arg->getSuperClasses().back().first->getName())
2279e5dd7070Spatrick              .Case("VariadicIdentifierArgument", true)
2280e5dd7070Spatrick              .Case("VariadicParamOrParamIdxArgument", true)
2281e5dd7070Spatrick              .Default(false);
2282e5dd7070Spatrick }
2283e5dd7070Spatrick 
isVariadicExprArgument(const Record * Arg)2284*12c85518Srobert static bool isVariadicExprArgument(const Record *Arg) {
2285*12c85518Srobert   return !Arg->getSuperClasses().empty() &&
2286*12c85518Srobert          llvm::StringSwitch<bool>(
2287*12c85518Srobert              Arg->getSuperClasses().back().first->getName())
2288*12c85518Srobert              .Case("VariadicExprArgument", true)
2289*12c85518Srobert              .Default(false);
2290*12c85518Srobert }
2291*12c85518Srobert 
emitClangAttrVariadicIdentifierArgList(RecordKeeper & Records,raw_ostream & OS)2292e5dd7070Spatrick static void emitClangAttrVariadicIdentifierArgList(RecordKeeper &Records,
2293e5dd7070Spatrick                                                    raw_ostream &OS) {
2294e5dd7070Spatrick   OS << "#if defined(CLANG_ATTR_VARIADIC_IDENTIFIER_ARG_LIST)\n";
2295e5dd7070Spatrick   std::vector<Record *> Attrs = Records.getAllDerivedDefinitions("Attr");
2296e5dd7070Spatrick   for (const auto *A : Attrs) {
2297e5dd7070Spatrick     // Determine whether the first argument is a variadic identifier.
2298e5dd7070Spatrick     std::vector<Record *> Args = A->getValueAsListOfDefs("Args");
2299e5dd7070Spatrick     if (Args.empty() || !isVariadicIdentifierArgument(Args[0]))
2300e5dd7070Spatrick       continue;
2301e5dd7070Spatrick 
2302e5dd7070Spatrick     // All these spellings take an identifier argument.
2303e5dd7070Spatrick     forEachUniqueSpelling(*A, [&](const FlattenedSpelling &S) {
2304e5dd7070Spatrick       OS << ".Case(\"" << S.name() << "\", "
2305e5dd7070Spatrick          << "true"
2306e5dd7070Spatrick          << ")\n";
2307e5dd7070Spatrick     });
2308e5dd7070Spatrick   }
2309e5dd7070Spatrick   OS << "#endif // CLANG_ATTR_VARIADIC_IDENTIFIER_ARG_LIST\n\n";
2310e5dd7070Spatrick }
2311e5dd7070Spatrick 
2312e5dd7070Spatrick // Emits the first-argument-is-identifier property for attributes.
emitClangAttrIdentifierArgList(RecordKeeper & Records,raw_ostream & OS)2313e5dd7070Spatrick static void emitClangAttrIdentifierArgList(RecordKeeper &Records, raw_ostream &OS) {
2314e5dd7070Spatrick   OS << "#if defined(CLANG_ATTR_IDENTIFIER_ARG_LIST)\n";
2315e5dd7070Spatrick   std::vector<Record*> Attrs = Records.getAllDerivedDefinitions("Attr");
2316e5dd7070Spatrick 
2317e5dd7070Spatrick   for (const auto *Attr : Attrs) {
2318e5dd7070Spatrick     // Determine whether the first argument is an identifier.
2319e5dd7070Spatrick     std::vector<Record *> Args = Attr->getValueAsListOfDefs("Args");
2320e5dd7070Spatrick     if (Args.empty() || !isIdentifierArgument(Args[0]))
2321e5dd7070Spatrick       continue;
2322e5dd7070Spatrick 
2323e5dd7070Spatrick     // All these spellings take an identifier argument.
2324e5dd7070Spatrick     forEachUniqueSpelling(*Attr, [&](const FlattenedSpelling &S) {
2325e5dd7070Spatrick       OS << ".Case(\"" << S.name() << "\", " << "true" << ")\n";
2326e5dd7070Spatrick     });
2327e5dd7070Spatrick   }
2328e5dd7070Spatrick   OS << "#endif // CLANG_ATTR_IDENTIFIER_ARG_LIST\n\n";
2329e5dd7070Spatrick }
2330e5dd7070Spatrick 
keywordThisIsaIdentifierInArgument(const Record * Arg)2331e5dd7070Spatrick static bool keywordThisIsaIdentifierInArgument(const Record *Arg) {
2332e5dd7070Spatrick   return !Arg->getSuperClasses().empty() &&
2333e5dd7070Spatrick          llvm::StringSwitch<bool>(
2334e5dd7070Spatrick              Arg->getSuperClasses().back().first->getName())
2335e5dd7070Spatrick              .Case("VariadicParamOrParamIdxArgument", true)
2336e5dd7070Spatrick              .Default(false);
2337e5dd7070Spatrick }
2338e5dd7070Spatrick 
emitClangAttrThisIsaIdentifierArgList(RecordKeeper & Records,raw_ostream & OS)2339e5dd7070Spatrick static void emitClangAttrThisIsaIdentifierArgList(RecordKeeper &Records,
2340e5dd7070Spatrick                                                   raw_ostream &OS) {
2341e5dd7070Spatrick   OS << "#if defined(CLANG_ATTR_THIS_ISA_IDENTIFIER_ARG_LIST)\n";
2342e5dd7070Spatrick   std::vector<Record *> Attrs = Records.getAllDerivedDefinitions("Attr");
2343e5dd7070Spatrick   for (const auto *A : Attrs) {
2344e5dd7070Spatrick     // Determine whether the first argument is a variadic identifier.
2345e5dd7070Spatrick     std::vector<Record *> Args = A->getValueAsListOfDefs("Args");
2346e5dd7070Spatrick     if (Args.empty() || !keywordThisIsaIdentifierInArgument(Args[0]))
2347e5dd7070Spatrick       continue;
2348e5dd7070Spatrick 
2349e5dd7070Spatrick     // All these spellings take an identifier argument.
2350e5dd7070Spatrick     forEachUniqueSpelling(*A, [&](const FlattenedSpelling &S) {
2351e5dd7070Spatrick       OS << ".Case(\"" << S.name() << "\", "
2352e5dd7070Spatrick          << "true"
2353e5dd7070Spatrick          << ")\n";
2354e5dd7070Spatrick     });
2355e5dd7070Spatrick   }
2356e5dd7070Spatrick   OS << "#endif // CLANG_ATTR_THIS_ISA_IDENTIFIER_ARG_LIST\n\n";
2357e5dd7070Spatrick }
2358e5dd7070Spatrick 
emitClangAttrAcceptsExprPack(RecordKeeper & Records,raw_ostream & OS)2359*12c85518Srobert static void emitClangAttrAcceptsExprPack(RecordKeeper &Records,
2360*12c85518Srobert                                          raw_ostream &OS) {
2361*12c85518Srobert   OS << "#if defined(CLANG_ATTR_ACCEPTS_EXPR_PACK)\n";
2362*12c85518Srobert   ParsedAttrMap Attrs = getParsedAttrList(Records);
2363*12c85518Srobert   for (const auto &I : Attrs) {
2364*12c85518Srobert     const Record &Attr = *I.second;
2365*12c85518Srobert 
2366*12c85518Srobert     if (!Attr.getValueAsBit("AcceptsExprPack"))
2367*12c85518Srobert       continue;
2368*12c85518Srobert 
2369*12c85518Srobert     forEachUniqueSpelling(Attr, [&](const FlattenedSpelling &S) {
2370*12c85518Srobert       OS << ".Case(\"" << S.name() << "\", true)\n";
2371*12c85518Srobert     });
2372*12c85518Srobert   }
2373*12c85518Srobert   OS << "#endif // CLANG_ATTR_ACCEPTS_EXPR_PACK\n\n";
2374*12c85518Srobert }
2375*12c85518Srobert 
emitAttributes(RecordKeeper & Records,raw_ostream & OS,bool Header)2376ec727ea7Spatrick static void emitAttributes(RecordKeeper &Records, raw_ostream &OS,
2377ec727ea7Spatrick                            bool Header) {
2378e5dd7070Spatrick   std::vector<Record*> Attrs = Records.getAllDerivedDefinitions("Attr");
2379e5dd7070Spatrick   ParsedAttrMap AttrMap = getParsedAttrList(Records);
2380e5dd7070Spatrick 
2381a9ac8606Spatrick   // Helper to print the starting character of an attribute argument. If there
2382a9ac8606Spatrick   // hasn't been an argument yet, it prints an opening parenthese; otherwise it
2383a9ac8606Spatrick   // prints a comma.
2384a9ac8606Spatrick   OS << "static inline void DelimitAttributeArgument("
2385a9ac8606Spatrick      << "raw_ostream& OS, bool& IsFirst) {\n"
2386a9ac8606Spatrick      << "  if (IsFirst) {\n"
2387a9ac8606Spatrick      << "    IsFirst = false;\n"
2388a9ac8606Spatrick      << "    OS << \"(\";\n"
2389a9ac8606Spatrick      << "  } else\n"
2390a9ac8606Spatrick      << "    OS << \", \";\n"
2391a9ac8606Spatrick      << "}\n";
2392a9ac8606Spatrick 
2393e5dd7070Spatrick   for (const auto *Attr : Attrs) {
2394e5dd7070Spatrick     const Record &R = *Attr;
2395e5dd7070Spatrick 
2396e5dd7070Spatrick     // FIXME: Currently, documentation is generated as-needed due to the fact
2397e5dd7070Spatrick     // that there is no way to allow a generated project "reach into" the docs
2398e5dd7070Spatrick     // directory (for instance, it may be an out-of-tree build). However, we want
2399e5dd7070Spatrick     // to ensure that every attribute has a Documentation field, and produce an
2400e5dd7070Spatrick     // error if it has been neglected. Otherwise, the on-demand generation which
2401e5dd7070Spatrick     // happens server-side will fail. This code is ensuring that functionality,
2402e5dd7070Spatrick     // even though this Emitter doesn't technically need the documentation.
2403e5dd7070Spatrick     // When attribute documentation can be generated as part of the build
2404e5dd7070Spatrick     // itself, this code can be removed.
2405e5dd7070Spatrick     (void)R.getValueAsListOfDefs("Documentation");
2406e5dd7070Spatrick 
2407e5dd7070Spatrick     if (!R.getValueAsBit("ASTNode"))
2408e5dd7070Spatrick       continue;
2409e5dd7070Spatrick 
2410e5dd7070Spatrick     ArrayRef<std::pair<Record *, SMRange>> Supers = R.getSuperClasses();
2411e5dd7070Spatrick     assert(!Supers.empty() && "Forgot to specify a superclass for the attr");
2412e5dd7070Spatrick     std::string SuperName;
2413e5dd7070Spatrick     bool Inheritable = false;
2414e5dd7070Spatrick     for (const auto &Super : llvm::reverse(Supers)) {
2415e5dd7070Spatrick       const Record *R = Super.first;
2416e5dd7070Spatrick       if (R->getName() != "TargetSpecificAttr" &&
2417e5dd7070Spatrick           R->getName() != "DeclOrTypeAttr" && SuperName.empty())
2418ec727ea7Spatrick         SuperName = std::string(R->getName());
2419e5dd7070Spatrick       if (R->getName() == "InheritableAttr")
2420e5dd7070Spatrick         Inheritable = true;
2421e5dd7070Spatrick     }
2422e5dd7070Spatrick 
2423ec727ea7Spatrick     if (Header)
2424e5dd7070Spatrick       OS << "class " << R.getName() << "Attr : public " << SuperName << " {\n";
2425ec727ea7Spatrick     else
2426ec727ea7Spatrick       OS << "\n// " << R.getName() << "Attr implementation\n\n";
2427e5dd7070Spatrick 
2428e5dd7070Spatrick     std::vector<Record*> ArgRecords = R.getValueAsListOfDefs("Args");
2429e5dd7070Spatrick     std::vector<std::unique_ptr<Argument>> Args;
2430e5dd7070Spatrick     Args.reserve(ArgRecords.size());
2431e5dd7070Spatrick 
2432*12c85518Srobert     bool AttrAcceptsExprPack = Attr->getValueAsBit("AcceptsExprPack");
2433*12c85518Srobert     if (AttrAcceptsExprPack) {
2434*12c85518Srobert       for (size_t I = 0; I < ArgRecords.size(); ++I) {
2435*12c85518Srobert         const Record *ArgR = ArgRecords[I];
2436*12c85518Srobert         if (isIdentifierArgument(ArgR) || isVariadicIdentifierArgument(ArgR) ||
2437*12c85518Srobert             isTypeArgument(ArgR))
2438*12c85518Srobert           PrintFatalError(Attr->getLoc(),
2439*12c85518Srobert                           "Attributes accepting packs cannot also "
2440*12c85518Srobert                           "have identifier or type arguments.");
2441*12c85518Srobert         // When trying to determine if value-dependent expressions can populate
2442*12c85518Srobert         // the attribute without prior instantiation, the decision is made based
2443*12c85518Srobert         // on the assumption that only the last argument is ever variadic.
2444*12c85518Srobert         if (I < (ArgRecords.size() - 1) && isVariadicExprArgument(ArgR))
2445*12c85518Srobert           PrintFatalError(Attr->getLoc(),
2446*12c85518Srobert                           "Attributes accepting packs can only have the last "
2447*12c85518Srobert                           "argument be variadic.");
2448*12c85518Srobert       }
2449*12c85518Srobert     }
2450*12c85518Srobert 
2451e5dd7070Spatrick     bool HasOptArg = false;
2452e5dd7070Spatrick     bool HasFakeArg = false;
2453e5dd7070Spatrick     for (const auto *ArgRecord : ArgRecords) {
2454e5dd7070Spatrick       Args.emplace_back(createArgument(*ArgRecord, R.getName()));
2455ec727ea7Spatrick       if (Header) {
2456e5dd7070Spatrick         Args.back()->writeDeclarations(OS);
2457e5dd7070Spatrick         OS << "\n\n";
2458ec727ea7Spatrick       }
2459e5dd7070Spatrick 
2460e5dd7070Spatrick       // For these purposes, fake takes priority over optional.
2461e5dd7070Spatrick       if (Args.back()->isFake()) {
2462e5dd7070Spatrick         HasFakeArg = true;
2463e5dd7070Spatrick       } else if (Args.back()->isOptional()) {
2464e5dd7070Spatrick         HasOptArg = true;
2465e5dd7070Spatrick       }
2466e5dd7070Spatrick     }
2467e5dd7070Spatrick 
2468*12c85518Srobert     std::unique_ptr<VariadicExprArgument> DelayedArgs = nullptr;
2469*12c85518Srobert     if (AttrAcceptsExprPack) {
2470*12c85518Srobert       DelayedArgs =
2471*12c85518Srobert           std::make_unique<VariadicExprArgument>("DelayedArgs", R.getName());
2472*12c85518Srobert       if (Header) {
2473*12c85518Srobert         DelayedArgs->writeDeclarations(OS);
2474*12c85518Srobert         OS << "\n\n";
2475*12c85518Srobert       }
2476*12c85518Srobert     }
2477*12c85518Srobert 
2478ec727ea7Spatrick     if (Header)
2479e5dd7070Spatrick       OS << "public:\n";
2480e5dd7070Spatrick 
2481e5dd7070Spatrick     std::vector<FlattenedSpelling> Spellings = GetFlattenedSpellings(R);
2482e5dd7070Spatrick 
2483e5dd7070Spatrick     // If there are zero or one spellings, all spelling-related functionality
2484e5dd7070Spatrick     // can be elided. If all of the spellings share the same name, the spelling
2485e5dd7070Spatrick     // functionality can also be elided.
2486e5dd7070Spatrick     bool ElideSpelling = (Spellings.size() <= 1) ||
2487e5dd7070Spatrick                          SpellingNamesAreCommon(Spellings);
2488e5dd7070Spatrick 
2489e5dd7070Spatrick     // This maps spelling index values to semantic Spelling enumerants.
2490e5dd7070Spatrick     SemanticSpellingMap SemanticToSyntacticMap;
2491e5dd7070Spatrick 
2492ec727ea7Spatrick     std::string SpellingEnum;
2493ec727ea7Spatrick     if (Spellings.size() > 1)
2494ec727ea7Spatrick       SpellingEnum = CreateSemanticSpellings(Spellings, SemanticToSyntacticMap);
2495ec727ea7Spatrick     if (Header)
2496ec727ea7Spatrick       OS << SpellingEnum;
2497e5dd7070Spatrick 
2498e5dd7070Spatrick     const auto &ParsedAttrSpellingItr = llvm::find_if(
2499e5dd7070Spatrick         AttrMap, [R](const std::pair<std::string, const Record *> &P) {
2500e5dd7070Spatrick           return &R == P.second;
2501e5dd7070Spatrick         });
2502e5dd7070Spatrick 
2503e5dd7070Spatrick     // Emit CreateImplicit factory methods.
2504*12c85518Srobert     auto emitCreate = [&](bool Implicit, bool DelayedArgsOnly, bool emitFake) {
2505ec727ea7Spatrick       if (Header)
2506ec727ea7Spatrick         OS << "  static ";
2507ec727ea7Spatrick       OS << R.getName() << "Attr *";
2508ec727ea7Spatrick       if (!Header)
2509ec727ea7Spatrick         OS << R.getName() << "Attr::";
2510ec727ea7Spatrick       OS << "Create";
2511e5dd7070Spatrick       if (Implicit)
2512e5dd7070Spatrick         OS << "Implicit";
2513*12c85518Srobert       if (DelayedArgsOnly)
2514*12c85518Srobert         OS << "WithDelayedArgs";
2515e5dd7070Spatrick       OS << "(";
2516e5dd7070Spatrick       OS << "ASTContext &Ctx";
2517*12c85518Srobert       if (!DelayedArgsOnly) {
2518e5dd7070Spatrick         for (auto const &ai : Args) {
2519*12c85518Srobert           if (ai->isFake() && !emitFake)
2520*12c85518Srobert             continue;
2521e5dd7070Spatrick           OS << ", ";
2522e5dd7070Spatrick           ai->writeCtorParameters(OS);
2523e5dd7070Spatrick         }
2524*12c85518Srobert       } else {
2525*12c85518Srobert         OS << ", ";
2526*12c85518Srobert         DelayedArgs->writeCtorParameters(OS);
2527*12c85518Srobert       }
2528ec727ea7Spatrick       OS << ", const AttributeCommonInfo &CommonInfo";
2529a9ac8606Spatrick       if (Header && Implicit)
2530ec727ea7Spatrick         OS << " = {SourceRange{}}";
2531ec727ea7Spatrick       OS << ")";
2532ec727ea7Spatrick       if (Header) {
2533ec727ea7Spatrick         OS << ";\n";
2534ec727ea7Spatrick         return;
2535ec727ea7Spatrick       }
2536ec727ea7Spatrick 
2537ec727ea7Spatrick       OS << " {\n";
2538e5dd7070Spatrick       OS << "  auto *A = new (Ctx) " << R.getName();
2539e5dd7070Spatrick       OS << "Attr(Ctx, CommonInfo";
2540*12c85518Srobert       if (!DelayedArgsOnly) {
2541e5dd7070Spatrick         for (auto const &ai : Args) {
2542*12c85518Srobert           if (ai->isFake() && !emitFake)
2543*12c85518Srobert             continue;
2544e5dd7070Spatrick           OS << ", ";
2545e5dd7070Spatrick           ai->writeImplicitCtorArgs(OS);
2546e5dd7070Spatrick         }
2547*12c85518Srobert       }
2548e5dd7070Spatrick       OS << ");\n";
2549e5dd7070Spatrick       if (Implicit) {
2550e5dd7070Spatrick         OS << "  A->setImplicit(true);\n";
2551e5dd7070Spatrick       }
2552e5dd7070Spatrick       if (Implicit || ElideSpelling) {
2553e5dd7070Spatrick         OS << "  if (!A->isAttributeSpellingListCalculated() && "
2554e5dd7070Spatrick               "!A->getAttrName())\n";
2555e5dd7070Spatrick         OS << "    A->setAttributeSpellingListIndex(0);\n";
2556e5dd7070Spatrick       }
2557*12c85518Srobert       if (DelayedArgsOnly) {
2558*12c85518Srobert         OS << "  A->setDelayedArgs(Ctx, ";
2559*12c85518Srobert         DelayedArgs->writeImplicitCtorArgs(OS);
2560*12c85518Srobert         OS << ");\n";
2561*12c85518Srobert       }
2562e5dd7070Spatrick       OS << "  return A;\n}\n\n";
2563e5dd7070Spatrick     };
2564e5dd7070Spatrick 
2565*12c85518Srobert     auto emitCreateNoCI = [&](bool Implicit, bool DelayedArgsOnly,
2566*12c85518Srobert                               bool emitFake) {
2567ec727ea7Spatrick       if (Header)
2568ec727ea7Spatrick         OS << "  static ";
2569ec727ea7Spatrick       OS << R.getName() << "Attr *";
2570ec727ea7Spatrick       if (!Header)
2571ec727ea7Spatrick         OS << R.getName() << "Attr::";
2572ec727ea7Spatrick       OS << "Create";
2573e5dd7070Spatrick       if (Implicit)
2574e5dd7070Spatrick         OS << "Implicit";
2575*12c85518Srobert       if (DelayedArgsOnly)
2576*12c85518Srobert         OS << "WithDelayedArgs";
2577e5dd7070Spatrick       OS << "(";
2578e5dd7070Spatrick       OS << "ASTContext &Ctx";
2579*12c85518Srobert       if (!DelayedArgsOnly) {
2580e5dd7070Spatrick         for (auto const &ai : Args) {
2581*12c85518Srobert           if (ai->isFake() && !emitFake)
2582*12c85518Srobert             continue;
2583e5dd7070Spatrick           OS << ", ";
2584e5dd7070Spatrick           ai->writeCtorParameters(OS);
2585e5dd7070Spatrick         }
2586*12c85518Srobert       } else {
2587*12c85518Srobert         OS << ", ";
2588*12c85518Srobert         DelayedArgs->writeCtorParameters(OS);
2589*12c85518Srobert       }
2590e5dd7070Spatrick       OS << ", SourceRange Range, AttributeCommonInfo::Syntax Syntax";
2591ec727ea7Spatrick       if (!ElideSpelling) {
2592ec727ea7Spatrick         OS << ", " << R.getName() << "Attr::Spelling S";
2593ec727ea7Spatrick         if (Header)
2594ec727ea7Spatrick           OS << " = static_cast<Spelling>(SpellingNotCalculated)";
2595ec727ea7Spatrick       }
2596ec727ea7Spatrick       OS << ")";
2597ec727ea7Spatrick       if (Header) {
2598ec727ea7Spatrick         OS << ";\n";
2599ec727ea7Spatrick         return;
2600ec727ea7Spatrick       }
2601ec727ea7Spatrick 
2602ec727ea7Spatrick       OS << " {\n";
2603e5dd7070Spatrick       OS << "  AttributeCommonInfo I(Range, ";
2604e5dd7070Spatrick 
2605e5dd7070Spatrick       if (ParsedAttrSpellingItr != std::end(AttrMap))
2606e5dd7070Spatrick         OS << "AT_" << ParsedAttrSpellingItr->first;
2607e5dd7070Spatrick       else
2608e5dd7070Spatrick         OS << "NoSemaHandlerAttribute";
2609e5dd7070Spatrick 
2610e5dd7070Spatrick       OS << ", Syntax";
2611e5dd7070Spatrick       if (!ElideSpelling)
2612e5dd7070Spatrick         OS << ", S";
2613e5dd7070Spatrick       OS << ");\n";
2614e5dd7070Spatrick       OS << "  return Create";
2615e5dd7070Spatrick       if (Implicit)
2616e5dd7070Spatrick         OS << "Implicit";
2617*12c85518Srobert       if (DelayedArgsOnly)
2618*12c85518Srobert         OS << "WithDelayedArgs";
2619e5dd7070Spatrick       OS << "(Ctx";
2620*12c85518Srobert       if (!DelayedArgsOnly) {
2621e5dd7070Spatrick         for (auto const &ai : Args) {
2622*12c85518Srobert           if (ai->isFake() && !emitFake)
2623*12c85518Srobert             continue;
2624e5dd7070Spatrick           OS << ", ";
2625e5dd7070Spatrick           ai->writeImplicitCtorArgs(OS);
2626e5dd7070Spatrick         }
2627*12c85518Srobert       } else {
2628*12c85518Srobert         OS << ", ";
2629*12c85518Srobert         DelayedArgs->writeImplicitCtorArgs(OS);
2630*12c85518Srobert       }
2631e5dd7070Spatrick       OS << ", I);\n";
2632ec727ea7Spatrick       OS << "}\n\n";
2633e5dd7070Spatrick     };
2634e5dd7070Spatrick 
2635*12c85518Srobert     auto emitCreates = [&](bool DelayedArgsOnly, bool emitFake) {
2636*12c85518Srobert       emitCreate(true, DelayedArgsOnly, emitFake);
2637*12c85518Srobert       emitCreate(false, DelayedArgsOnly, emitFake);
2638*12c85518Srobert       emitCreateNoCI(true, DelayedArgsOnly, emitFake);
2639*12c85518Srobert       emitCreateNoCI(false, DelayedArgsOnly, emitFake);
2640e5dd7070Spatrick     };
2641e5dd7070Spatrick 
2642ec727ea7Spatrick     if (Header)
2643ec727ea7Spatrick       OS << "  // Factory methods\n";
2644ec727ea7Spatrick 
2645e5dd7070Spatrick     // Emit a CreateImplicit that takes all the arguments.
2646*12c85518Srobert     emitCreates(false, true);
2647e5dd7070Spatrick 
2648e5dd7070Spatrick     // Emit a CreateImplicit that takes all the non-fake arguments.
2649e5dd7070Spatrick     if (HasFakeArg)
2650*12c85518Srobert       emitCreates(false, false);
2651*12c85518Srobert 
2652*12c85518Srobert     // Emit a CreateWithDelayedArgs that takes only the dependent argument
2653*12c85518Srobert     // expressions.
2654*12c85518Srobert     if (DelayedArgs)
2655*12c85518Srobert       emitCreates(true, false);
2656e5dd7070Spatrick 
2657e5dd7070Spatrick     // Emit constructors.
2658*12c85518Srobert     auto emitCtor = [&](bool emitOpt, bool emitFake, bool emitNoArgs) {
2659e5dd7070Spatrick       auto shouldEmitArg = [=](const std::unique_ptr<Argument> &arg) {
2660*12c85518Srobert         if (emitNoArgs)
2661*12c85518Srobert           return false;
2662*12c85518Srobert         if (arg->isFake())
2663*12c85518Srobert           return emitFake;
2664*12c85518Srobert         if (arg->isOptional())
2665*12c85518Srobert           return emitOpt;
2666e5dd7070Spatrick         return true;
2667e5dd7070Spatrick       };
2668ec727ea7Spatrick       if (Header)
2669ec727ea7Spatrick         OS << "  ";
2670ec727ea7Spatrick       else
2671ec727ea7Spatrick         OS << R.getName() << "Attr::";
2672ec727ea7Spatrick       OS << R.getName()
2673e5dd7070Spatrick          << "Attr(ASTContext &Ctx, const AttributeCommonInfo &CommonInfo";
2674e5dd7070Spatrick       OS << '\n';
2675e5dd7070Spatrick       for (auto const &ai : Args) {
2676*12c85518Srobert         if (!shouldEmitArg(ai))
2677*12c85518Srobert           continue;
2678e5dd7070Spatrick         OS << "              , ";
2679e5dd7070Spatrick         ai->writeCtorParameters(OS);
2680e5dd7070Spatrick         OS << "\n";
2681e5dd7070Spatrick       }
2682e5dd7070Spatrick 
2683ec727ea7Spatrick       OS << "             )";
2684ec727ea7Spatrick       if (Header) {
2685ec727ea7Spatrick         OS << ";\n";
2686ec727ea7Spatrick         return;
2687ec727ea7Spatrick       }
2688ec727ea7Spatrick       OS << "\n  : " << SuperName << "(Ctx, CommonInfo, ";
2689e5dd7070Spatrick       OS << "attr::" << R.getName() << ", "
2690e5dd7070Spatrick          << (R.getValueAsBit("LateParsed") ? "true" : "false");
2691e5dd7070Spatrick       if (Inheritable) {
2692e5dd7070Spatrick         OS << ", "
2693e5dd7070Spatrick            << (R.getValueAsBit("InheritEvenIfAlreadyPresent") ? "true"
2694e5dd7070Spatrick                                                               : "false");
2695e5dd7070Spatrick       }
2696e5dd7070Spatrick       OS << ")\n";
2697e5dd7070Spatrick 
2698e5dd7070Spatrick       for (auto const &ai : Args) {
2699e5dd7070Spatrick         OS << "              , ";
2700e5dd7070Spatrick         if (!shouldEmitArg(ai)) {
2701e5dd7070Spatrick           ai->writeCtorDefaultInitializers(OS);
2702e5dd7070Spatrick         } else {
2703e5dd7070Spatrick           ai->writeCtorInitializers(OS);
2704e5dd7070Spatrick         }
2705e5dd7070Spatrick         OS << "\n";
2706e5dd7070Spatrick       }
2707*12c85518Srobert       if (DelayedArgs) {
2708*12c85518Srobert         OS << "              , ";
2709*12c85518Srobert         DelayedArgs->writeCtorDefaultInitializers(OS);
2710*12c85518Srobert         OS << "\n";
2711*12c85518Srobert       }
2712e5dd7070Spatrick 
2713e5dd7070Spatrick       OS << "  {\n";
2714e5dd7070Spatrick 
2715e5dd7070Spatrick       for (auto const &ai : Args) {
2716*12c85518Srobert         if (!shouldEmitArg(ai))
2717*12c85518Srobert           continue;
2718e5dd7070Spatrick         ai->writeCtorBody(OS);
2719e5dd7070Spatrick       }
2720e5dd7070Spatrick       OS << "}\n\n";
2721e5dd7070Spatrick     };
2722e5dd7070Spatrick 
2723ec727ea7Spatrick     if (Header)
2724ec727ea7Spatrick       OS << "\n  // Constructors\n";
2725ec727ea7Spatrick 
2726e5dd7070Spatrick     // Emit a constructor that includes all the arguments.
2727e5dd7070Spatrick     // This is necessary for cloning.
2728*12c85518Srobert     emitCtor(true, true, false);
2729e5dd7070Spatrick 
2730e5dd7070Spatrick     // Emit a constructor that takes all the non-fake arguments.
2731e5dd7070Spatrick     if (HasFakeArg)
2732*12c85518Srobert       emitCtor(true, false, false);
2733e5dd7070Spatrick 
2734e5dd7070Spatrick     // Emit a constructor that takes all the non-fake, non-optional arguments.
2735e5dd7070Spatrick     if (HasOptArg)
2736*12c85518Srobert       emitCtor(false, false, false);
2737*12c85518Srobert 
2738*12c85518Srobert     // Emit constructors that takes no arguments if none already exists.
2739*12c85518Srobert     // This is used for delaying arguments.
2740*12c85518Srobert     bool HasRequiredArgs =
2741*12c85518Srobert         llvm::count_if(Args, [=](const std::unique_ptr<Argument> &arg) {
2742*12c85518Srobert           return !arg->isFake() && !arg->isOptional();
2743*12c85518Srobert         });
2744*12c85518Srobert     if (DelayedArgs && HasRequiredArgs)
2745*12c85518Srobert       emitCtor(false, false, true);
2746e5dd7070Spatrick 
2747ec727ea7Spatrick     if (Header) {
2748ec727ea7Spatrick       OS << '\n';
2749e5dd7070Spatrick       OS << "  " << R.getName() << "Attr *clone(ASTContext &C) const;\n";
2750e5dd7070Spatrick       OS << "  void printPretty(raw_ostream &OS,\n"
2751e5dd7070Spatrick          << "                   const PrintingPolicy &Policy) const;\n";
2752e5dd7070Spatrick       OS << "  const char *getSpelling() const;\n";
2753ec727ea7Spatrick     }
2754e5dd7070Spatrick 
2755e5dd7070Spatrick     if (!ElideSpelling) {
2756e5dd7070Spatrick       assert(!SemanticToSyntacticMap.empty() && "Empty semantic mapping list");
2757ec727ea7Spatrick       if (Header)
2758ec727ea7Spatrick         OS << "  Spelling getSemanticSpelling() const;\n";
2759ec727ea7Spatrick       else {
2760ec727ea7Spatrick         OS << R.getName() << "Attr::Spelling " << R.getName()
2761ec727ea7Spatrick            << "Attr::getSemanticSpelling() const {\n";
2762e5dd7070Spatrick         WriteSemanticSpellingSwitch("getAttributeSpellingListIndex()",
2763e5dd7070Spatrick                                     SemanticToSyntacticMap, OS);
2764e5dd7070Spatrick         OS << "}\n";
2765e5dd7070Spatrick       }
2766ec727ea7Spatrick     }
2767e5dd7070Spatrick 
2768ec727ea7Spatrick     if (Header)
2769e5dd7070Spatrick       writeAttrAccessorDefinition(R, OS);
2770e5dd7070Spatrick 
2771e5dd7070Spatrick     for (auto const &ai : Args) {
2772ec727ea7Spatrick       if (Header) {
2773e5dd7070Spatrick         ai->writeAccessors(OS);
2774ec727ea7Spatrick       } else {
2775ec727ea7Spatrick         ai->writeAccessorDefinitions(OS);
2776ec727ea7Spatrick       }
2777e5dd7070Spatrick       OS << "\n\n";
2778e5dd7070Spatrick 
2779e5dd7070Spatrick       // Don't write conversion routines for fake arguments.
2780e5dd7070Spatrick       if (ai->isFake()) continue;
2781e5dd7070Spatrick 
2782e5dd7070Spatrick       if (ai->isEnumArg())
2783ec727ea7Spatrick         static_cast<const EnumArgument *>(ai.get())->writeConversion(OS,
2784ec727ea7Spatrick                                                                      Header);
2785e5dd7070Spatrick       else if (ai->isVariadicEnumArg())
2786ec727ea7Spatrick         static_cast<const VariadicEnumArgument *>(ai.get())->writeConversion(
2787ec727ea7Spatrick             OS, Header);
2788e5dd7070Spatrick     }
2789e5dd7070Spatrick 
2790ec727ea7Spatrick     if (Header) {
2791*12c85518Srobert       if (DelayedArgs) {
2792*12c85518Srobert         DelayedArgs->writeAccessors(OS);
2793*12c85518Srobert         DelayedArgs->writeSetter(OS);
2794*12c85518Srobert       }
2795*12c85518Srobert 
2796e5dd7070Spatrick       OS << R.getValueAsString("AdditionalMembers");
2797e5dd7070Spatrick       OS << "\n\n";
2798e5dd7070Spatrick 
2799e5dd7070Spatrick       OS << "  static bool classof(const Attr *A) { return A->getKind() == "
2800e5dd7070Spatrick          << "attr::" << R.getName() << "; }\n";
2801e5dd7070Spatrick 
2802e5dd7070Spatrick       OS << "};\n\n";
2803ec727ea7Spatrick     } else {
2804*12c85518Srobert       if (DelayedArgs)
2805*12c85518Srobert         DelayedArgs->writeAccessorDefinitions(OS);
2806*12c85518Srobert 
2807e5dd7070Spatrick       OS << R.getName() << "Attr *" << R.getName()
2808e5dd7070Spatrick          << "Attr::clone(ASTContext &C) const {\n";
2809e5dd7070Spatrick       OS << "  auto *A = new (C) " << R.getName() << "Attr(C, *this";
2810e5dd7070Spatrick       for (auto const &ai : Args) {
2811e5dd7070Spatrick         OS << ", ";
2812e5dd7070Spatrick         ai->writeCloneArgs(OS);
2813e5dd7070Spatrick       }
2814e5dd7070Spatrick       OS << ");\n";
2815e5dd7070Spatrick       OS << "  A->Inherited = Inherited;\n";
2816e5dd7070Spatrick       OS << "  A->IsPackExpansion = IsPackExpansion;\n";
2817e5dd7070Spatrick       OS << "  A->setImplicit(Implicit);\n";
2818*12c85518Srobert       if (DelayedArgs) {
2819*12c85518Srobert         OS << "  A->setDelayedArgs(C, ";
2820*12c85518Srobert         DelayedArgs->writeCloneArgs(OS);
2821*12c85518Srobert         OS << ");\n";
2822*12c85518Srobert       }
2823e5dd7070Spatrick       OS << "  return A;\n}\n\n";
2824e5dd7070Spatrick 
2825e5dd7070Spatrick       writePrettyPrintFunction(R, Args, OS);
2826e5dd7070Spatrick       writeGetSpellingFunction(R, OS);
2827e5dd7070Spatrick     }
2828ec727ea7Spatrick   }
2829ec727ea7Spatrick }
2830ec727ea7Spatrick // Emits the class definitions for attributes.
EmitClangAttrClass(RecordKeeper & Records,raw_ostream & OS)2831ec727ea7Spatrick void clang::EmitClangAttrClass(RecordKeeper &Records, raw_ostream &OS) {
2832ec727ea7Spatrick   emitSourceFileHeader("Attribute classes' definitions", OS);
2833ec727ea7Spatrick 
2834ec727ea7Spatrick   OS << "#ifndef LLVM_CLANG_ATTR_CLASSES_INC\n";
2835ec727ea7Spatrick   OS << "#define LLVM_CLANG_ATTR_CLASSES_INC\n\n";
2836ec727ea7Spatrick 
2837ec727ea7Spatrick   emitAttributes(Records, OS, true);
2838ec727ea7Spatrick 
2839ec727ea7Spatrick   OS << "#endif // LLVM_CLANG_ATTR_CLASSES_INC\n";
2840ec727ea7Spatrick }
2841ec727ea7Spatrick 
2842ec727ea7Spatrick // Emits the class method definitions for attributes.
EmitClangAttrImpl(RecordKeeper & Records,raw_ostream & OS)2843ec727ea7Spatrick void clang::EmitClangAttrImpl(RecordKeeper &Records, raw_ostream &OS) {
2844ec727ea7Spatrick   emitSourceFileHeader("Attribute classes' member function definitions", OS);
2845ec727ea7Spatrick 
2846ec727ea7Spatrick   emitAttributes(Records, OS, false);
2847ec727ea7Spatrick 
2848ec727ea7Spatrick   std::vector<Record *> Attrs = Records.getAllDerivedDefinitions("Attr");
2849e5dd7070Spatrick 
2850e5dd7070Spatrick   // Instead of relying on virtual dispatch we just create a huge dispatch
2851e5dd7070Spatrick   // switch. This is both smaller and faster than virtual functions.
2852e5dd7070Spatrick   auto EmitFunc = [&](const char *Method) {
2853e5dd7070Spatrick     OS << "  switch (getKind()) {\n";
2854e5dd7070Spatrick     for (const auto *Attr : Attrs) {
2855e5dd7070Spatrick       const Record &R = *Attr;
2856e5dd7070Spatrick       if (!R.getValueAsBit("ASTNode"))
2857e5dd7070Spatrick         continue;
2858e5dd7070Spatrick 
2859e5dd7070Spatrick       OS << "  case attr::" << R.getName() << ":\n";
2860e5dd7070Spatrick       OS << "    return cast<" << R.getName() << "Attr>(this)->" << Method
2861e5dd7070Spatrick          << ";\n";
2862e5dd7070Spatrick     }
2863e5dd7070Spatrick     OS << "  }\n";
2864e5dd7070Spatrick     OS << "  llvm_unreachable(\"Unexpected attribute kind!\");\n";
2865e5dd7070Spatrick     OS << "}\n\n";
2866e5dd7070Spatrick   };
2867e5dd7070Spatrick 
2868e5dd7070Spatrick   OS << "const char *Attr::getSpelling() const {\n";
2869e5dd7070Spatrick   EmitFunc("getSpelling()");
2870e5dd7070Spatrick 
2871e5dd7070Spatrick   OS << "Attr *Attr::clone(ASTContext &C) const {\n";
2872e5dd7070Spatrick   EmitFunc("clone(C)");
2873e5dd7070Spatrick 
2874e5dd7070Spatrick   OS << "void Attr::printPretty(raw_ostream &OS, "
2875e5dd7070Spatrick         "const PrintingPolicy &Policy) const {\n";
2876e5dd7070Spatrick   EmitFunc("printPretty(OS, Policy)");
2877e5dd7070Spatrick }
2878e5dd7070Spatrick 
emitAttrList(raw_ostream & OS,StringRef Class,const std::vector<Record * > & AttrList)2879e5dd7070Spatrick static void emitAttrList(raw_ostream &OS, StringRef Class,
2880e5dd7070Spatrick                          const std::vector<Record*> &AttrList) {
2881e5dd7070Spatrick   for (auto Cur : AttrList) {
2882e5dd7070Spatrick     OS << Class << "(" << Cur->getName() << ")\n";
2883e5dd7070Spatrick   }
2884e5dd7070Spatrick }
2885e5dd7070Spatrick 
2886e5dd7070Spatrick // Determines if an attribute has a Pragma spelling.
AttrHasPragmaSpelling(const Record * R)2887e5dd7070Spatrick static bool AttrHasPragmaSpelling(const Record *R) {
2888e5dd7070Spatrick   std::vector<FlattenedSpelling> Spellings = GetFlattenedSpellings(*R);
2889*12c85518Srobert   return llvm::any_of(Spellings, [](const FlattenedSpelling &S) {
2890e5dd7070Spatrick     return S.variety() == "Pragma";
2891*12c85518Srobert   });
2892e5dd7070Spatrick }
2893e5dd7070Spatrick 
2894e5dd7070Spatrick namespace {
2895e5dd7070Spatrick 
2896e5dd7070Spatrick   struct AttrClassDescriptor {
2897e5dd7070Spatrick     const char * const MacroName;
2898e5dd7070Spatrick     const char * const TableGenName;
2899e5dd7070Spatrick   };
2900e5dd7070Spatrick 
2901e5dd7070Spatrick } // end anonymous namespace
2902e5dd7070Spatrick 
2903e5dd7070Spatrick static const AttrClassDescriptor AttrClassDescriptors[] = {
2904e5dd7070Spatrick   { "ATTR", "Attr" },
2905e5dd7070Spatrick   { "TYPE_ATTR", "TypeAttr" },
2906e5dd7070Spatrick   { "STMT_ATTR", "StmtAttr" },
2907a9ac8606Spatrick   { "DECL_OR_STMT_ATTR", "DeclOrStmtAttr" },
2908e5dd7070Spatrick   { "INHERITABLE_ATTR", "InheritableAttr" },
2909e5dd7070Spatrick   { "DECL_OR_TYPE_ATTR", "DeclOrTypeAttr" },
2910e5dd7070Spatrick   { "INHERITABLE_PARAM_ATTR", "InheritableParamAttr" },
2911*12c85518Srobert   { "PARAMETER_ABI_ATTR", "ParameterABIAttr" },
2912*12c85518Srobert   { "HLSL_ANNOTATION_ATTR", "HLSLAnnotationAttr"}
2913e5dd7070Spatrick };
2914e5dd7070Spatrick 
emitDefaultDefine(raw_ostream & OS,StringRef name,const char * superName)2915e5dd7070Spatrick static void emitDefaultDefine(raw_ostream &OS, StringRef name,
2916e5dd7070Spatrick                               const char *superName) {
2917e5dd7070Spatrick   OS << "#ifndef " << name << "\n";
2918e5dd7070Spatrick   OS << "#define " << name << "(NAME) ";
2919e5dd7070Spatrick   if (superName) OS << superName << "(NAME)";
2920e5dd7070Spatrick   OS << "\n#endif\n\n";
2921e5dd7070Spatrick }
2922e5dd7070Spatrick 
2923e5dd7070Spatrick namespace {
2924e5dd7070Spatrick 
2925e5dd7070Spatrick   /// A class of attributes.
2926e5dd7070Spatrick   struct AttrClass {
2927e5dd7070Spatrick     const AttrClassDescriptor &Descriptor;
2928e5dd7070Spatrick     Record *TheRecord;
2929e5dd7070Spatrick     AttrClass *SuperClass = nullptr;
2930e5dd7070Spatrick     std::vector<AttrClass*> SubClasses;
2931e5dd7070Spatrick     std::vector<Record*> Attrs;
2932e5dd7070Spatrick 
AttrClass__anone86b54961511::AttrClass2933e5dd7070Spatrick     AttrClass(const AttrClassDescriptor &Descriptor, Record *R)
2934e5dd7070Spatrick       : Descriptor(Descriptor), TheRecord(R) {}
2935e5dd7070Spatrick 
emitDefaultDefines__anone86b54961511::AttrClass2936e5dd7070Spatrick     void emitDefaultDefines(raw_ostream &OS) const {
2937e5dd7070Spatrick       // Default the macro unless this is a root class (i.e. Attr).
2938e5dd7070Spatrick       if (SuperClass) {
2939e5dd7070Spatrick         emitDefaultDefine(OS, Descriptor.MacroName,
2940e5dd7070Spatrick                           SuperClass->Descriptor.MacroName);
2941e5dd7070Spatrick       }
2942e5dd7070Spatrick     }
2943e5dd7070Spatrick 
emitUndefs__anone86b54961511::AttrClass2944e5dd7070Spatrick     void emitUndefs(raw_ostream &OS) const {
2945e5dd7070Spatrick       OS << "#undef " << Descriptor.MacroName << "\n";
2946e5dd7070Spatrick     }
2947e5dd7070Spatrick 
emitAttrList__anone86b54961511::AttrClass2948e5dd7070Spatrick     void emitAttrList(raw_ostream &OS) const {
2949e5dd7070Spatrick       for (auto SubClass : SubClasses) {
2950e5dd7070Spatrick         SubClass->emitAttrList(OS);
2951e5dd7070Spatrick       }
2952e5dd7070Spatrick 
2953e5dd7070Spatrick       ::emitAttrList(OS, Descriptor.MacroName, Attrs);
2954e5dd7070Spatrick     }
2955e5dd7070Spatrick 
classifyAttrOnRoot__anone86b54961511::AttrClass2956e5dd7070Spatrick     void classifyAttrOnRoot(Record *Attr) {
2957e5dd7070Spatrick       bool result = classifyAttr(Attr);
2958e5dd7070Spatrick       assert(result && "failed to classify on root"); (void) result;
2959e5dd7070Spatrick     }
2960e5dd7070Spatrick 
emitAttrRange__anone86b54961511::AttrClass2961e5dd7070Spatrick     void emitAttrRange(raw_ostream &OS) const {
2962e5dd7070Spatrick       OS << "ATTR_RANGE(" << Descriptor.TableGenName
2963e5dd7070Spatrick          << ", " << getFirstAttr()->getName()
2964e5dd7070Spatrick          << ", " << getLastAttr()->getName() << ")\n";
2965e5dd7070Spatrick     }
2966e5dd7070Spatrick 
2967e5dd7070Spatrick   private:
classifyAttr__anone86b54961511::AttrClass2968e5dd7070Spatrick     bool classifyAttr(Record *Attr) {
2969e5dd7070Spatrick       // Check all the subclasses.
2970e5dd7070Spatrick       for (auto SubClass : SubClasses) {
2971e5dd7070Spatrick         if (SubClass->classifyAttr(Attr))
2972e5dd7070Spatrick           return true;
2973e5dd7070Spatrick       }
2974e5dd7070Spatrick 
2975e5dd7070Spatrick       // It's not more specific than this class, but it might still belong here.
2976e5dd7070Spatrick       if (Attr->isSubClassOf(TheRecord)) {
2977e5dd7070Spatrick         Attrs.push_back(Attr);
2978e5dd7070Spatrick         return true;
2979e5dd7070Spatrick       }
2980e5dd7070Spatrick 
2981e5dd7070Spatrick       return false;
2982e5dd7070Spatrick     }
2983e5dd7070Spatrick 
getFirstAttr__anone86b54961511::AttrClass2984e5dd7070Spatrick     Record *getFirstAttr() const {
2985e5dd7070Spatrick       if (!SubClasses.empty())
2986e5dd7070Spatrick         return SubClasses.front()->getFirstAttr();
2987e5dd7070Spatrick       return Attrs.front();
2988e5dd7070Spatrick     }
2989e5dd7070Spatrick 
getLastAttr__anone86b54961511::AttrClass2990e5dd7070Spatrick     Record *getLastAttr() const {
2991e5dd7070Spatrick       if (!Attrs.empty())
2992e5dd7070Spatrick         return Attrs.back();
2993e5dd7070Spatrick       return SubClasses.back()->getLastAttr();
2994e5dd7070Spatrick     }
2995e5dd7070Spatrick   };
2996e5dd7070Spatrick 
2997e5dd7070Spatrick   /// The entire hierarchy of attribute classes.
2998e5dd7070Spatrick   class AttrClassHierarchy {
2999e5dd7070Spatrick     std::vector<std::unique_ptr<AttrClass>> Classes;
3000e5dd7070Spatrick 
3001e5dd7070Spatrick   public:
AttrClassHierarchy(RecordKeeper & Records)3002e5dd7070Spatrick     AttrClassHierarchy(RecordKeeper &Records) {
3003e5dd7070Spatrick       // Find records for all the classes.
3004e5dd7070Spatrick       for (auto &Descriptor : AttrClassDescriptors) {
3005e5dd7070Spatrick         Record *ClassRecord = Records.getClass(Descriptor.TableGenName);
3006e5dd7070Spatrick         AttrClass *Class = new AttrClass(Descriptor, ClassRecord);
3007e5dd7070Spatrick         Classes.emplace_back(Class);
3008e5dd7070Spatrick       }
3009e5dd7070Spatrick 
3010e5dd7070Spatrick       // Link up the hierarchy.
3011e5dd7070Spatrick       for (auto &Class : Classes) {
3012e5dd7070Spatrick         if (AttrClass *SuperClass = findSuperClass(Class->TheRecord)) {
3013e5dd7070Spatrick           Class->SuperClass = SuperClass;
3014e5dd7070Spatrick           SuperClass->SubClasses.push_back(Class.get());
3015e5dd7070Spatrick         }
3016e5dd7070Spatrick       }
3017e5dd7070Spatrick 
3018e5dd7070Spatrick #ifndef NDEBUG
3019e5dd7070Spatrick       for (auto i = Classes.begin(), e = Classes.end(); i != e; ++i) {
3020e5dd7070Spatrick         assert((i == Classes.begin()) == ((*i)->SuperClass == nullptr) &&
3021e5dd7070Spatrick                "only the first class should be a root class!");
3022e5dd7070Spatrick       }
3023e5dd7070Spatrick #endif
3024e5dd7070Spatrick     }
3025e5dd7070Spatrick 
emitDefaultDefines(raw_ostream & OS) const3026e5dd7070Spatrick     void emitDefaultDefines(raw_ostream &OS) const {
3027e5dd7070Spatrick       for (auto &Class : Classes) {
3028e5dd7070Spatrick         Class->emitDefaultDefines(OS);
3029e5dd7070Spatrick       }
3030e5dd7070Spatrick     }
3031e5dd7070Spatrick 
emitUndefs(raw_ostream & OS) const3032e5dd7070Spatrick     void emitUndefs(raw_ostream &OS) const {
3033e5dd7070Spatrick       for (auto &Class : Classes) {
3034e5dd7070Spatrick         Class->emitUndefs(OS);
3035e5dd7070Spatrick       }
3036e5dd7070Spatrick     }
3037e5dd7070Spatrick 
emitAttrLists(raw_ostream & OS) const3038e5dd7070Spatrick     void emitAttrLists(raw_ostream &OS) const {
3039e5dd7070Spatrick       // Just start from the root class.
3040e5dd7070Spatrick       Classes[0]->emitAttrList(OS);
3041e5dd7070Spatrick     }
3042e5dd7070Spatrick 
emitAttrRanges(raw_ostream & OS) const3043e5dd7070Spatrick     void emitAttrRanges(raw_ostream &OS) const {
3044e5dd7070Spatrick       for (auto &Class : Classes)
3045e5dd7070Spatrick         Class->emitAttrRange(OS);
3046e5dd7070Spatrick     }
3047e5dd7070Spatrick 
classifyAttr(Record * Attr)3048e5dd7070Spatrick     void classifyAttr(Record *Attr) {
3049e5dd7070Spatrick       // Add the attribute to the root class.
3050e5dd7070Spatrick       Classes[0]->classifyAttrOnRoot(Attr);
3051e5dd7070Spatrick     }
3052e5dd7070Spatrick 
3053e5dd7070Spatrick   private:
findClassByRecord(Record * R) const3054e5dd7070Spatrick     AttrClass *findClassByRecord(Record *R) const {
3055e5dd7070Spatrick       for (auto &Class : Classes) {
3056e5dd7070Spatrick         if (Class->TheRecord == R)
3057e5dd7070Spatrick           return Class.get();
3058e5dd7070Spatrick       }
3059e5dd7070Spatrick       return nullptr;
3060e5dd7070Spatrick     }
3061e5dd7070Spatrick 
findSuperClass(Record * R) const3062e5dd7070Spatrick     AttrClass *findSuperClass(Record *R) const {
3063e5dd7070Spatrick       // TableGen flattens the superclass list, so we just need to walk it
3064e5dd7070Spatrick       // in reverse.
3065e5dd7070Spatrick       auto SuperClasses = R->getSuperClasses();
3066e5dd7070Spatrick       for (signed i = 0, e = SuperClasses.size(); i != e; ++i) {
3067e5dd7070Spatrick         auto SuperClass = findClassByRecord(SuperClasses[e - i - 1].first);
3068e5dd7070Spatrick         if (SuperClass) return SuperClass;
3069e5dd7070Spatrick       }
3070e5dd7070Spatrick       return nullptr;
3071e5dd7070Spatrick     }
3072e5dd7070Spatrick   };
3073e5dd7070Spatrick 
3074e5dd7070Spatrick } // end anonymous namespace
3075e5dd7070Spatrick 
3076e5dd7070Spatrick namespace clang {
3077e5dd7070Spatrick 
3078e5dd7070Spatrick // Emits the enumeration list for attributes.
EmitClangAttrList(RecordKeeper & Records,raw_ostream & OS)3079e5dd7070Spatrick void EmitClangAttrList(RecordKeeper &Records, raw_ostream &OS) {
3080e5dd7070Spatrick   emitSourceFileHeader("List of all attributes that Clang recognizes", OS);
3081e5dd7070Spatrick 
3082e5dd7070Spatrick   AttrClassHierarchy Hierarchy(Records);
3083e5dd7070Spatrick 
3084e5dd7070Spatrick   // Add defaulting macro definitions.
3085e5dd7070Spatrick   Hierarchy.emitDefaultDefines(OS);
3086e5dd7070Spatrick   emitDefaultDefine(OS, "PRAGMA_SPELLING_ATTR", nullptr);
3087e5dd7070Spatrick 
3088e5dd7070Spatrick   std::vector<Record *> Attrs = Records.getAllDerivedDefinitions("Attr");
3089e5dd7070Spatrick   std::vector<Record *> PragmaAttrs;
3090e5dd7070Spatrick   for (auto *Attr : Attrs) {
3091e5dd7070Spatrick     if (!Attr->getValueAsBit("ASTNode"))
3092e5dd7070Spatrick       continue;
3093e5dd7070Spatrick 
3094e5dd7070Spatrick     // Add the attribute to the ad-hoc groups.
3095e5dd7070Spatrick     if (AttrHasPragmaSpelling(Attr))
3096e5dd7070Spatrick       PragmaAttrs.push_back(Attr);
3097e5dd7070Spatrick 
3098e5dd7070Spatrick     // Place it in the hierarchy.
3099e5dd7070Spatrick     Hierarchy.classifyAttr(Attr);
3100e5dd7070Spatrick   }
3101e5dd7070Spatrick 
3102e5dd7070Spatrick   // Emit the main attribute list.
3103e5dd7070Spatrick   Hierarchy.emitAttrLists(OS);
3104e5dd7070Spatrick 
3105e5dd7070Spatrick   // Emit the ad hoc groups.
3106e5dd7070Spatrick   emitAttrList(OS, "PRAGMA_SPELLING_ATTR", PragmaAttrs);
3107e5dd7070Spatrick 
3108e5dd7070Spatrick   // Emit the attribute ranges.
3109e5dd7070Spatrick   OS << "#ifdef ATTR_RANGE\n";
3110e5dd7070Spatrick   Hierarchy.emitAttrRanges(OS);
3111e5dd7070Spatrick   OS << "#undef ATTR_RANGE\n";
3112e5dd7070Spatrick   OS << "#endif\n";
3113e5dd7070Spatrick 
3114e5dd7070Spatrick   Hierarchy.emitUndefs(OS);
3115e5dd7070Spatrick   OS << "#undef PRAGMA_SPELLING_ATTR\n";
3116e5dd7070Spatrick }
3117e5dd7070Spatrick 
3118e5dd7070Spatrick // Emits the enumeration list for attributes.
EmitClangAttrSubjectMatchRuleList(RecordKeeper & Records,raw_ostream & OS)3119e5dd7070Spatrick void EmitClangAttrSubjectMatchRuleList(RecordKeeper &Records, raw_ostream &OS) {
3120e5dd7070Spatrick   emitSourceFileHeader(
3121e5dd7070Spatrick       "List of all attribute subject matching rules that Clang recognizes", OS);
3122e5dd7070Spatrick   PragmaClangAttributeSupport &PragmaAttributeSupport =
3123e5dd7070Spatrick       getPragmaAttributeSupport(Records);
3124e5dd7070Spatrick   emitDefaultDefine(OS, "ATTR_MATCH_RULE", nullptr);
3125e5dd7070Spatrick   PragmaAttributeSupport.emitMatchRuleList(OS);
3126e5dd7070Spatrick   OS << "#undef ATTR_MATCH_RULE\n";
3127e5dd7070Spatrick }
3128e5dd7070Spatrick 
3129e5dd7070Spatrick // Emits the code to read an attribute from a precompiled header.
EmitClangAttrPCHRead(RecordKeeper & Records,raw_ostream & OS)3130e5dd7070Spatrick void EmitClangAttrPCHRead(RecordKeeper &Records, raw_ostream &OS) {
3131e5dd7070Spatrick   emitSourceFileHeader("Attribute deserialization code", OS);
3132e5dd7070Spatrick 
3133e5dd7070Spatrick   Record *InhClass = Records.getClass("InheritableAttr");
3134e5dd7070Spatrick   std::vector<Record*> Attrs = Records.getAllDerivedDefinitions("Attr"),
3135e5dd7070Spatrick                        ArgRecords;
3136e5dd7070Spatrick   std::vector<std::unique_ptr<Argument>> Args;
3137*12c85518Srobert   std::unique_ptr<VariadicExprArgument> DelayedArgs;
3138e5dd7070Spatrick 
3139e5dd7070Spatrick   OS << "  switch (Kind) {\n";
3140e5dd7070Spatrick   for (const auto *Attr : Attrs) {
3141e5dd7070Spatrick     const Record &R = *Attr;
3142e5dd7070Spatrick     if (!R.getValueAsBit("ASTNode"))
3143e5dd7070Spatrick       continue;
3144e5dd7070Spatrick 
3145e5dd7070Spatrick     OS << "  case attr::" << R.getName() << ": {\n";
3146e5dd7070Spatrick     if (R.isSubClassOf(InhClass))
3147e5dd7070Spatrick       OS << "    bool isInherited = Record.readInt();\n";
3148e5dd7070Spatrick     OS << "    bool isImplicit = Record.readInt();\n";
3149389bb291Spatrick     OS << "    bool isPackExpansion = Record.readInt();\n";
3150*12c85518Srobert     DelayedArgs = nullptr;
3151*12c85518Srobert     if (Attr->getValueAsBit("AcceptsExprPack")) {
3152*12c85518Srobert       DelayedArgs =
3153*12c85518Srobert           std::make_unique<VariadicExprArgument>("DelayedArgs", R.getName());
3154*12c85518Srobert       DelayedArgs->writePCHReadDecls(OS);
3155*12c85518Srobert     }
3156e5dd7070Spatrick     ArgRecords = R.getValueAsListOfDefs("Args");
3157e5dd7070Spatrick     Args.clear();
3158e5dd7070Spatrick     for (const auto *Arg : ArgRecords) {
3159e5dd7070Spatrick       Args.emplace_back(createArgument(*Arg, R.getName()));
3160e5dd7070Spatrick       Args.back()->writePCHReadDecls(OS);
3161e5dd7070Spatrick     }
3162e5dd7070Spatrick     OS << "    New = new (Context) " << R.getName() << "Attr(Context, Info";
3163e5dd7070Spatrick     for (auto const &ri : Args) {
3164e5dd7070Spatrick       OS << ", ";
3165e5dd7070Spatrick       ri->writePCHReadArgs(OS);
3166e5dd7070Spatrick     }
3167e5dd7070Spatrick     OS << ");\n";
3168e5dd7070Spatrick     if (R.isSubClassOf(InhClass))
3169e5dd7070Spatrick       OS << "    cast<InheritableAttr>(New)->setInherited(isInherited);\n";
3170e5dd7070Spatrick     OS << "    New->setImplicit(isImplicit);\n";
3171389bb291Spatrick     OS << "    New->setPackExpansion(isPackExpansion);\n";
3172*12c85518Srobert     if (DelayedArgs) {
3173*12c85518Srobert       OS << "    cast<" << R.getName()
3174*12c85518Srobert          << "Attr>(New)->setDelayedArgs(Context, ";
3175*12c85518Srobert       DelayedArgs->writePCHReadArgs(OS);
3176*12c85518Srobert       OS << ");\n";
3177*12c85518Srobert     }
3178e5dd7070Spatrick     OS << "    break;\n";
3179e5dd7070Spatrick     OS << "  }\n";
3180e5dd7070Spatrick   }
3181e5dd7070Spatrick   OS << "  }\n";
3182e5dd7070Spatrick }
3183e5dd7070Spatrick 
3184e5dd7070Spatrick // Emits the code to write an attribute to a precompiled header.
EmitClangAttrPCHWrite(RecordKeeper & Records,raw_ostream & OS)3185e5dd7070Spatrick void EmitClangAttrPCHWrite(RecordKeeper &Records, raw_ostream &OS) {
3186e5dd7070Spatrick   emitSourceFileHeader("Attribute serialization code", OS);
3187e5dd7070Spatrick 
3188e5dd7070Spatrick   Record *InhClass = Records.getClass("InheritableAttr");
3189e5dd7070Spatrick   std::vector<Record*> Attrs = Records.getAllDerivedDefinitions("Attr"), Args;
3190e5dd7070Spatrick 
3191e5dd7070Spatrick   OS << "  switch (A->getKind()) {\n";
3192e5dd7070Spatrick   for (const auto *Attr : Attrs) {
3193e5dd7070Spatrick     const Record &R = *Attr;
3194e5dd7070Spatrick     if (!R.getValueAsBit("ASTNode"))
3195e5dd7070Spatrick       continue;
3196e5dd7070Spatrick     OS << "  case attr::" << R.getName() << ": {\n";
3197e5dd7070Spatrick     Args = R.getValueAsListOfDefs("Args");
3198e5dd7070Spatrick     if (R.isSubClassOf(InhClass) || !Args.empty())
3199e5dd7070Spatrick       OS << "    const auto *SA = cast<" << R.getName()
3200e5dd7070Spatrick          << "Attr>(A);\n";
3201e5dd7070Spatrick     if (R.isSubClassOf(InhClass))
3202e5dd7070Spatrick       OS << "    Record.push_back(SA->isInherited());\n";
3203e5dd7070Spatrick     OS << "    Record.push_back(A->isImplicit());\n";
3204389bb291Spatrick     OS << "    Record.push_back(A->isPackExpansion());\n";
3205*12c85518Srobert     if (Attr->getValueAsBit("AcceptsExprPack"))
3206*12c85518Srobert       VariadicExprArgument("DelayedArgs", R.getName()).writePCHWrite(OS);
3207e5dd7070Spatrick 
3208e5dd7070Spatrick     for (const auto *Arg : Args)
3209e5dd7070Spatrick       createArgument(*Arg, R.getName())->writePCHWrite(OS);
3210e5dd7070Spatrick     OS << "    break;\n";
3211e5dd7070Spatrick     OS << "  }\n";
3212e5dd7070Spatrick   }
3213e5dd7070Spatrick   OS << "  }\n";
3214e5dd7070Spatrick }
3215e5dd7070Spatrick 
3216e5dd7070Spatrick // Helper function for GenerateTargetSpecificAttrChecks that alters the 'Test'
3217e5dd7070Spatrick // parameter with only a single check type, if applicable.
GenerateTargetSpecificAttrCheck(const Record * R,std::string & Test,std::string * FnName,StringRef ListName,StringRef CheckAgainst,StringRef Scope)3218e5dd7070Spatrick static bool GenerateTargetSpecificAttrCheck(const Record *R, std::string &Test,
3219e5dd7070Spatrick                                             std::string *FnName,
3220e5dd7070Spatrick                                             StringRef ListName,
3221e5dd7070Spatrick                                             StringRef CheckAgainst,
3222e5dd7070Spatrick                                             StringRef Scope) {
3223e5dd7070Spatrick   if (!R->isValueUnset(ListName)) {
3224e5dd7070Spatrick     Test += " && (";
3225e5dd7070Spatrick     std::vector<StringRef> Items = R->getValueAsListOfStrings(ListName);
3226e5dd7070Spatrick     for (auto I = Items.begin(), E = Items.end(); I != E; ++I) {
3227e5dd7070Spatrick       StringRef Part = *I;
3228e5dd7070Spatrick       Test += CheckAgainst;
3229e5dd7070Spatrick       Test += " == ";
3230e5dd7070Spatrick       Test += Scope;
3231e5dd7070Spatrick       Test += Part;
3232e5dd7070Spatrick       if (I + 1 != E)
3233e5dd7070Spatrick         Test += " || ";
3234e5dd7070Spatrick       if (FnName)
3235e5dd7070Spatrick         *FnName += Part;
3236e5dd7070Spatrick     }
3237e5dd7070Spatrick     Test += ")";
3238e5dd7070Spatrick     return true;
3239e5dd7070Spatrick   }
3240e5dd7070Spatrick   return false;
3241e5dd7070Spatrick }
3242e5dd7070Spatrick 
3243e5dd7070Spatrick // Generate a conditional expression to check if the current target satisfies
3244e5dd7070Spatrick // the conditions for a TargetSpecificAttr record, and append the code for
3245e5dd7070Spatrick // those checks to the Test string. If the FnName string pointer is non-null,
3246e5dd7070Spatrick // append a unique suffix to distinguish this set of target checks from other
3247e5dd7070Spatrick // TargetSpecificAttr records.
GenerateTargetSpecificAttrChecks(const Record * R,std::vector<StringRef> & Arches,std::string & Test,std::string * FnName)3248e5dd7070Spatrick static bool GenerateTargetSpecificAttrChecks(const Record *R,
3249e5dd7070Spatrick                                              std::vector<StringRef> &Arches,
3250e5dd7070Spatrick                                              std::string &Test,
3251e5dd7070Spatrick                                              std::string *FnName) {
3252e5dd7070Spatrick   bool AnyTargetChecks = false;
3253e5dd7070Spatrick 
3254e5dd7070Spatrick   // It is assumed that there will be an llvm::Triple object
3255e5dd7070Spatrick   // named "T" and a TargetInfo object named "Target" within
3256e5dd7070Spatrick   // scope that can be used to determine whether the attribute exists in
3257e5dd7070Spatrick   // a given target.
3258e5dd7070Spatrick   Test += "true";
3259e5dd7070Spatrick   // If one or more architectures is specified, check those.  Arches are handled
3260e5dd7070Spatrick   // differently because GenerateTargetRequirements needs to combine the list
3261e5dd7070Spatrick   // with ParseKind.
3262e5dd7070Spatrick   if (!Arches.empty()) {
3263e5dd7070Spatrick     AnyTargetChecks = true;
3264e5dd7070Spatrick     Test += " && (";
3265e5dd7070Spatrick     for (auto I = Arches.begin(), E = Arches.end(); I != E; ++I) {
3266e5dd7070Spatrick       StringRef Part = *I;
3267e5dd7070Spatrick       Test += "T.getArch() == llvm::Triple::";
3268e5dd7070Spatrick       Test += Part;
3269e5dd7070Spatrick       if (I + 1 != E)
3270e5dd7070Spatrick         Test += " || ";
3271e5dd7070Spatrick       if (FnName)
3272e5dd7070Spatrick         *FnName += Part;
3273e5dd7070Spatrick     }
3274e5dd7070Spatrick     Test += ")";
3275e5dd7070Spatrick   }
3276e5dd7070Spatrick 
3277e5dd7070Spatrick   // If the attribute is specific to particular OSes, check those.
3278e5dd7070Spatrick   AnyTargetChecks |= GenerateTargetSpecificAttrCheck(
3279e5dd7070Spatrick       R, Test, FnName, "OSes", "T.getOS()", "llvm::Triple::");
3280e5dd7070Spatrick 
3281e5dd7070Spatrick   // If one or more object formats is specified, check those.
3282e5dd7070Spatrick   AnyTargetChecks |=
3283e5dd7070Spatrick       GenerateTargetSpecificAttrCheck(R, Test, FnName, "ObjectFormats",
3284e5dd7070Spatrick                                       "T.getObjectFormat()", "llvm::Triple::");
3285e5dd7070Spatrick 
3286e5dd7070Spatrick   // If custom code is specified, emit it.
3287e5dd7070Spatrick   StringRef Code = R->getValueAsString("CustomCode");
3288e5dd7070Spatrick   if (!Code.empty()) {
3289e5dd7070Spatrick     AnyTargetChecks = true;
3290e5dd7070Spatrick     Test += " && (";
3291e5dd7070Spatrick     Test += Code;
3292e5dd7070Spatrick     Test += ")";
3293e5dd7070Spatrick   }
3294e5dd7070Spatrick 
3295e5dd7070Spatrick   return AnyTargetChecks;
3296e5dd7070Spatrick }
3297e5dd7070Spatrick 
GenerateHasAttrSpellingStringSwitch(const std::vector<Record * > & Attrs,raw_ostream & OS,const std::string & Variety="",const std::string & Scope="")3298e5dd7070Spatrick static void GenerateHasAttrSpellingStringSwitch(
3299e5dd7070Spatrick     const std::vector<Record *> &Attrs, raw_ostream &OS,
3300e5dd7070Spatrick     const std::string &Variety = "", const std::string &Scope = "") {
3301e5dd7070Spatrick   for (const auto *Attr : Attrs) {
3302e5dd7070Spatrick     // C++11-style attributes have specific version information associated with
3303e5dd7070Spatrick     // them. If the attribute has no scope, the version information must not
3304e5dd7070Spatrick     // have the default value (1), as that's incorrect. Instead, the unscoped
3305e5dd7070Spatrick     // attribute version information should be taken from the SD-6 standing
3306e5dd7070Spatrick     // document, which can be found at:
3307e5dd7070Spatrick     // https://isocpp.org/std/standing-documents/sd-6-sg10-feature-test-recommendations
3308a9ac8606Spatrick     //
3309a9ac8606Spatrick     // C2x-style attributes have the same kind of version information
3310a9ac8606Spatrick     // associated with them. The unscoped attribute version information should
3311a9ac8606Spatrick     // be taken from the specification of the attribute in the C Standard.
3312e5dd7070Spatrick     int Version = 1;
3313e5dd7070Spatrick 
3314a9ac8606Spatrick     if (Variety == "CXX11" || Variety == "C2x") {
3315e5dd7070Spatrick       std::vector<Record *> Spellings = Attr->getValueAsListOfDefs("Spellings");
3316e5dd7070Spatrick       for (const auto &Spelling : Spellings) {
3317a9ac8606Spatrick         if (Spelling->getValueAsString("Variety") == Variety) {
3318e5dd7070Spatrick           Version = static_cast<int>(Spelling->getValueAsInt("Version"));
3319e5dd7070Spatrick           if (Scope.empty() && Version == 1)
3320a9ac8606Spatrick             PrintError(Spelling->getLoc(), "Standard attributes must have "
3321a9ac8606Spatrick                                            "valid version information.");
3322e5dd7070Spatrick           break;
3323e5dd7070Spatrick         }
3324e5dd7070Spatrick       }
3325e5dd7070Spatrick     }
3326e5dd7070Spatrick 
3327e5dd7070Spatrick     std::string Test;
3328e5dd7070Spatrick     if (Attr->isSubClassOf("TargetSpecificAttr")) {
3329e5dd7070Spatrick       const Record *R = Attr->getValueAsDef("Target");
3330e5dd7070Spatrick       std::vector<StringRef> Arches = R->getValueAsListOfStrings("Arches");
3331e5dd7070Spatrick       GenerateTargetSpecificAttrChecks(R, Arches, Test, nullptr);
3332e5dd7070Spatrick 
3333e5dd7070Spatrick       // If this is the C++11 variety, also add in the LangOpts test.
3334e5dd7070Spatrick       if (Variety == "CXX11")
3335e5dd7070Spatrick         Test += " && LangOpts.CPlusPlus11";
3336e5dd7070Spatrick       else if (Variety == "C2x")
3337e5dd7070Spatrick         Test += " && LangOpts.DoubleSquareBracketAttributes";
3338e5dd7070Spatrick     } else if (Variety == "CXX11")
3339e5dd7070Spatrick       // C++11 mode should be checked against LangOpts, which is presumed to be
3340e5dd7070Spatrick       // present in the caller.
3341e5dd7070Spatrick       Test = "LangOpts.CPlusPlus11";
3342e5dd7070Spatrick     else if (Variety == "C2x")
3343e5dd7070Spatrick       Test = "LangOpts.DoubleSquareBracketAttributes";
3344e5dd7070Spatrick 
3345e5dd7070Spatrick     std::string TestStr =
3346e5dd7070Spatrick         !Test.empty() ? Test + " ? " + llvm::itostr(Version) + " : 0" : "1";
3347e5dd7070Spatrick     std::vector<FlattenedSpelling> Spellings = GetFlattenedSpellings(*Attr);
3348e5dd7070Spatrick     for (const auto &S : Spellings)
3349e5dd7070Spatrick       if (Variety.empty() || (Variety == S.variety() &&
3350e5dd7070Spatrick                               (Scope.empty() || Scope == S.nameSpace())))
3351e5dd7070Spatrick         OS << "    .Case(\"" << S.name() << "\", " << TestStr << ")\n";
3352e5dd7070Spatrick   }
3353e5dd7070Spatrick   OS << "    .Default(0);\n";
3354e5dd7070Spatrick }
3355e5dd7070Spatrick 
3356e5dd7070Spatrick // Emits the list of spellings for attributes.
EmitClangAttrHasAttrImpl(RecordKeeper & Records,raw_ostream & OS)3357e5dd7070Spatrick void EmitClangAttrHasAttrImpl(RecordKeeper &Records, raw_ostream &OS) {
3358e5dd7070Spatrick   emitSourceFileHeader("Code to implement the __has_attribute logic", OS);
3359e5dd7070Spatrick 
3360e5dd7070Spatrick   // Separate all of the attributes out into four group: generic, C++11, GNU,
3361e5dd7070Spatrick   // and declspecs. Then generate a big switch statement for each of them.
3362e5dd7070Spatrick   std::vector<Record *> Attrs = Records.getAllDerivedDefinitions("Attr");
3363*12c85518Srobert   std::vector<Record *> Declspec, Microsoft, GNU, Pragma, HLSLSemantic;
3364e5dd7070Spatrick   std::map<std::string, std::vector<Record *>> CXX, C2x;
3365e5dd7070Spatrick 
3366e5dd7070Spatrick   // Walk over the list of all attributes, and split them out based on the
3367e5dd7070Spatrick   // spelling variety.
3368e5dd7070Spatrick   for (auto *R : Attrs) {
3369e5dd7070Spatrick     std::vector<FlattenedSpelling> Spellings = GetFlattenedSpellings(*R);
3370e5dd7070Spatrick     for (const auto &SI : Spellings) {
3371e5dd7070Spatrick       const std::string &Variety = SI.variety();
3372e5dd7070Spatrick       if (Variety == "GNU")
3373e5dd7070Spatrick         GNU.push_back(R);
3374e5dd7070Spatrick       else if (Variety == "Declspec")
3375e5dd7070Spatrick         Declspec.push_back(R);
3376e5dd7070Spatrick       else if (Variety == "Microsoft")
3377e5dd7070Spatrick         Microsoft.push_back(R);
3378e5dd7070Spatrick       else if (Variety == "CXX11")
3379e5dd7070Spatrick         CXX[SI.nameSpace()].push_back(R);
3380e5dd7070Spatrick       else if (Variety == "C2x")
3381e5dd7070Spatrick         C2x[SI.nameSpace()].push_back(R);
3382e5dd7070Spatrick       else if (Variety == "Pragma")
3383e5dd7070Spatrick         Pragma.push_back(R);
3384*12c85518Srobert       else if (Variety == "HLSLSemantic")
3385*12c85518Srobert         HLSLSemantic.push_back(R);
3386e5dd7070Spatrick     }
3387e5dd7070Spatrick   }
3388e5dd7070Spatrick 
3389e5dd7070Spatrick   OS << "const llvm::Triple &T = Target.getTriple();\n";
3390e5dd7070Spatrick   OS << "switch (Syntax) {\n";
3391*12c85518Srobert   OS << "case AttributeCommonInfo::Syntax::AS_GNU:\n";
3392e5dd7070Spatrick   OS << "  return llvm::StringSwitch<int>(Name)\n";
3393e5dd7070Spatrick   GenerateHasAttrSpellingStringSwitch(GNU, OS, "GNU");
3394*12c85518Srobert   OS << "case AttributeCommonInfo::Syntax::AS_Declspec:\n";
3395e5dd7070Spatrick   OS << "  return llvm::StringSwitch<int>(Name)\n";
3396e5dd7070Spatrick   GenerateHasAttrSpellingStringSwitch(Declspec, OS, "Declspec");
3397*12c85518Srobert   OS << "case AttributeCommonInfo::Syntax::AS_Microsoft:\n";
3398e5dd7070Spatrick   OS << "  return llvm::StringSwitch<int>(Name)\n";
3399e5dd7070Spatrick   GenerateHasAttrSpellingStringSwitch(Microsoft, OS, "Microsoft");
3400*12c85518Srobert   OS << "case AttributeCommonInfo::Syntax::AS_Pragma:\n";
3401e5dd7070Spatrick   OS << "  return llvm::StringSwitch<int>(Name)\n";
3402e5dd7070Spatrick   GenerateHasAttrSpellingStringSwitch(Pragma, OS, "Pragma");
3403*12c85518Srobert   OS << "case AttributeCommonInfo::Syntax::AS_HLSLSemantic:\n";
3404*12c85518Srobert   OS << "  return llvm::StringSwitch<int>(Name)\n";
3405*12c85518Srobert   GenerateHasAttrSpellingStringSwitch(HLSLSemantic, OS, "HLSLSemantic");
3406*12c85518Srobert   auto fn = [&OS](const char *Spelling,
3407e5dd7070Spatrick                   const std::map<std::string, std::vector<Record *>> &List) {
3408*12c85518Srobert     OS << "case AttributeCommonInfo::Syntax::AS_" << Spelling << ": {\n";
3409e5dd7070Spatrick     // C++11-style attributes are further split out based on the Scope.
3410e5dd7070Spatrick     for (auto I = List.cbegin(), E = List.cend(); I != E; ++I) {
3411e5dd7070Spatrick       if (I != List.cbegin())
3412e5dd7070Spatrick         OS << " else ";
3413e5dd7070Spatrick       if (I->first.empty())
3414e5dd7070Spatrick         OS << "if (ScopeName == \"\") {\n";
3415e5dd7070Spatrick       else
3416e5dd7070Spatrick         OS << "if (ScopeName == \"" << I->first << "\") {\n";
3417e5dd7070Spatrick       OS << "  return llvm::StringSwitch<int>(Name)\n";
3418e5dd7070Spatrick       GenerateHasAttrSpellingStringSwitch(I->second, OS, Spelling, I->first);
3419e5dd7070Spatrick       OS << "}";
3420e5dd7070Spatrick     }
3421e5dd7070Spatrick     OS << "\n} break;\n";
3422e5dd7070Spatrick   };
3423*12c85518Srobert   fn("CXX11", CXX);
3424*12c85518Srobert   fn("C2x", C2x);
3425*12c85518Srobert   OS << "case AttributeCommonInfo::Syntax::AS_Keyword:\n";
3426*12c85518Srobert   OS << "case AttributeCommonInfo::Syntax::AS_ContextSensitiveKeyword:\n";
3427*12c85518Srobert   OS << "  llvm_unreachable(\"hasAttribute not supported for keyword\");\n";
3428*12c85518Srobert   OS << "  return 0;\n";
3429*12c85518Srobert 
3430e5dd7070Spatrick   OS << "}\n";
3431e5dd7070Spatrick }
3432e5dd7070Spatrick 
EmitClangAttrSpellingListIndex(RecordKeeper & Records,raw_ostream & OS)3433e5dd7070Spatrick void EmitClangAttrSpellingListIndex(RecordKeeper &Records, raw_ostream &OS) {
3434e5dd7070Spatrick   emitSourceFileHeader("Code to translate different attribute spellings "
3435e5dd7070Spatrick                        "into internal identifiers", OS);
3436e5dd7070Spatrick 
3437e5dd7070Spatrick   OS << "  switch (getParsedKind()) {\n";
3438e5dd7070Spatrick   OS << "    case IgnoredAttribute:\n";
3439e5dd7070Spatrick   OS << "    case UnknownAttribute:\n";
3440e5dd7070Spatrick   OS << "    case NoSemaHandlerAttribute:\n";
3441e5dd7070Spatrick   OS << "      llvm_unreachable(\"Ignored/unknown shouldn't get here\");\n";
3442e5dd7070Spatrick 
3443e5dd7070Spatrick   ParsedAttrMap Attrs = getParsedAttrList(Records);
3444e5dd7070Spatrick   for (const auto &I : Attrs) {
3445e5dd7070Spatrick     const Record &R = *I.second;
3446e5dd7070Spatrick     std::vector<FlattenedSpelling> Spellings = GetFlattenedSpellings(R);
3447e5dd7070Spatrick     OS << "  case AT_" << I.first << ": {\n";
3448e5dd7070Spatrick     for (unsigned I = 0; I < Spellings.size(); ++ I) {
3449e5dd7070Spatrick       OS << "    if (Name == \"" << Spellings[I].name() << "\" && "
3450e5dd7070Spatrick          << "getSyntax() == AttributeCommonInfo::AS_" << Spellings[I].variety()
3451e5dd7070Spatrick          << " && Scope == \"" << Spellings[I].nameSpace() << "\")\n"
3452e5dd7070Spatrick          << "        return " << I << ";\n";
3453e5dd7070Spatrick     }
3454e5dd7070Spatrick 
3455e5dd7070Spatrick     OS << "    break;\n";
3456e5dd7070Spatrick     OS << "  }\n";
3457e5dd7070Spatrick   }
3458e5dd7070Spatrick 
3459e5dd7070Spatrick   OS << "  }\n";
3460e5dd7070Spatrick   OS << "  return 0;\n";
3461e5dd7070Spatrick }
3462e5dd7070Spatrick 
3463e5dd7070Spatrick // Emits code used by RecursiveASTVisitor to visit attributes
EmitClangAttrASTVisitor(RecordKeeper & Records,raw_ostream & OS)3464e5dd7070Spatrick void EmitClangAttrASTVisitor(RecordKeeper &Records, raw_ostream &OS) {
3465e5dd7070Spatrick   emitSourceFileHeader("Used by RecursiveASTVisitor to visit attributes.", OS);
3466e5dd7070Spatrick 
3467e5dd7070Spatrick   std::vector<Record*> Attrs = Records.getAllDerivedDefinitions("Attr");
3468e5dd7070Spatrick 
3469e5dd7070Spatrick   // Write method declarations for Traverse* methods.
3470e5dd7070Spatrick   // We emit this here because we only generate methods for attributes that
3471e5dd7070Spatrick   // are declared as ASTNodes.
3472e5dd7070Spatrick   OS << "#ifdef ATTR_VISITOR_DECLS_ONLY\n\n";
3473e5dd7070Spatrick   for (const auto *Attr : Attrs) {
3474e5dd7070Spatrick     const Record &R = *Attr;
3475e5dd7070Spatrick     if (!R.getValueAsBit("ASTNode"))
3476e5dd7070Spatrick       continue;
3477e5dd7070Spatrick     OS << "  bool Traverse"
3478e5dd7070Spatrick        << R.getName() << "Attr(" << R.getName() << "Attr *A);\n";
3479e5dd7070Spatrick     OS << "  bool Visit"
3480e5dd7070Spatrick        << R.getName() << "Attr(" << R.getName() << "Attr *A) {\n"
3481e5dd7070Spatrick        << "    return true; \n"
3482e5dd7070Spatrick        << "  }\n";
3483e5dd7070Spatrick   }
3484e5dd7070Spatrick   OS << "\n#else // ATTR_VISITOR_DECLS_ONLY\n\n";
3485e5dd7070Spatrick 
3486e5dd7070Spatrick   // Write individual Traverse* methods for each attribute class.
3487e5dd7070Spatrick   for (const auto *Attr : Attrs) {
3488e5dd7070Spatrick     const Record &R = *Attr;
3489e5dd7070Spatrick     if (!R.getValueAsBit("ASTNode"))
3490e5dd7070Spatrick       continue;
3491e5dd7070Spatrick 
3492e5dd7070Spatrick     OS << "template <typename Derived>\n"
3493e5dd7070Spatrick        << "bool VISITORCLASS<Derived>::Traverse"
3494e5dd7070Spatrick        << R.getName() << "Attr(" << R.getName() << "Attr *A) {\n"
3495e5dd7070Spatrick        << "  if (!getDerived().VisitAttr(A))\n"
3496e5dd7070Spatrick        << "    return false;\n"
3497e5dd7070Spatrick        << "  if (!getDerived().Visit" << R.getName() << "Attr(A))\n"
3498e5dd7070Spatrick        << "    return false;\n";
3499e5dd7070Spatrick 
3500e5dd7070Spatrick     std::vector<Record*> ArgRecords = R.getValueAsListOfDefs("Args");
3501e5dd7070Spatrick     for (const auto *Arg : ArgRecords)
3502e5dd7070Spatrick       createArgument(*Arg, R.getName())->writeASTVisitorTraversal(OS);
3503e5dd7070Spatrick 
3504*12c85518Srobert     if (Attr->getValueAsBit("AcceptsExprPack"))
3505*12c85518Srobert       VariadicExprArgument("DelayedArgs", R.getName())
3506*12c85518Srobert           .writeASTVisitorTraversal(OS);
3507*12c85518Srobert 
3508e5dd7070Spatrick     OS << "  return true;\n";
3509e5dd7070Spatrick     OS << "}\n\n";
3510e5dd7070Spatrick   }
3511e5dd7070Spatrick 
3512e5dd7070Spatrick   // Write generic Traverse routine
3513e5dd7070Spatrick   OS << "template <typename Derived>\n"
3514e5dd7070Spatrick      << "bool VISITORCLASS<Derived>::TraverseAttr(Attr *A) {\n"
3515e5dd7070Spatrick      << "  if (!A)\n"
3516e5dd7070Spatrick      << "    return true;\n"
3517e5dd7070Spatrick      << "\n"
3518e5dd7070Spatrick      << "  switch (A->getKind()) {\n";
3519e5dd7070Spatrick 
3520e5dd7070Spatrick   for (const auto *Attr : Attrs) {
3521e5dd7070Spatrick     const Record &R = *Attr;
3522e5dd7070Spatrick     if (!R.getValueAsBit("ASTNode"))
3523e5dd7070Spatrick       continue;
3524e5dd7070Spatrick 
3525e5dd7070Spatrick     OS << "    case attr::" << R.getName() << ":\n"
3526e5dd7070Spatrick        << "      return getDerived().Traverse" << R.getName() << "Attr("
3527e5dd7070Spatrick        << "cast<" << R.getName() << "Attr>(A));\n";
3528e5dd7070Spatrick   }
3529e5dd7070Spatrick   OS << "  }\n";  // end switch
3530e5dd7070Spatrick   OS << "  llvm_unreachable(\"bad attribute kind\");\n";
3531e5dd7070Spatrick   OS << "}\n";  // end function
3532e5dd7070Spatrick   OS << "#endif  // ATTR_VISITOR_DECLS_ONLY\n";
3533e5dd7070Spatrick }
3534e5dd7070Spatrick 
EmitClangAttrTemplateInstantiateHelper(const std::vector<Record * > & Attrs,raw_ostream & OS,bool AppliesToDecl)3535e5dd7070Spatrick void EmitClangAttrTemplateInstantiateHelper(const std::vector<Record *> &Attrs,
3536e5dd7070Spatrick                                             raw_ostream &OS,
3537e5dd7070Spatrick                                             bool AppliesToDecl) {
3538e5dd7070Spatrick 
3539e5dd7070Spatrick   OS << "  switch (At->getKind()) {\n";
3540e5dd7070Spatrick   for (const auto *Attr : Attrs) {
3541e5dd7070Spatrick     const Record &R = *Attr;
3542e5dd7070Spatrick     if (!R.getValueAsBit("ASTNode"))
3543e5dd7070Spatrick       continue;
3544e5dd7070Spatrick     OS << "    case attr::" << R.getName() << ": {\n";
3545e5dd7070Spatrick     bool ShouldClone = R.getValueAsBit("Clone") &&
3546e5dd7070Spatrick                        (!AppliesToDecl ||
3547e5dd7070Spatrick                         R.getValueAsBit("MeaningfulToClassTemplateDefinition"));
3548e5dd7070Spatrick 
3549e5dd7070Spatrick     if (!ShouldClone) {
3550e5dd7070Spatrick       OS << "      return nullptr;\n";
3551e5dd7070Spatrick       OS << "    }\n";
3552e5dd7070Spatrick       continue;
3553e5dd7070Spatrick     }
3554e5dd7070Spatrick 
3555e5dd7070Spatrick     OS << "      const auto *A = cast<"
3556e5dd7070Spatrick        << R.getName() << "Attr>(At);\n";
3557e5dd7070Spatrick     bool TDependent = R.getValueAsBit("TemplateDependent");
3558e5dd7070Spatrick 
3559e5dd7070Spatrick     if (!TDependent) {
3560e5dd7070Spatrick       OS << "      return A->clone(C);\n";
3561e5dd7070Spatrick       OS << "    }\n";
3562e5dd7070Spatrick       continue;
3563e5dd7070Spatrick     }
3564e5dd7070Spatrick 
3565e5dd7070Spatrick     std::vector<Record*> ArgRecords = R.getValueAsListOfDefs("Args");
3566e5dd7070Spatrick     std::vector<std::unique_ptr<Argument>> Args;
3567e5dd7070Spatrick     Args.reserve(ArgRecords.size());
3568e5dd7070Spatrick 
3569e5dd7070Spatrick     for (const auto *ArgRecord : ArgRecords)
3570e5dd7070Spatrick       Args.emplace_back(createArgument(*ArgRecord, R.getName()));
3571e5dd7070Spatrick 
3572e5dd7070Spatrick     for (auto const &ai : Args)
3573e5dd7070Spatrick       ai->writeTemplateInstantiation(OS);
3574e5dd7070Spatrick 
3575e5dd7070Spatrick     OS << "      return new (C) " << R.getName() << "Attr(C, *A";
3576e5dd7070Spatrick     for (auto const &ai : Args) {
3577e5dd7070Spatrick       OS << ", ";
3578e5dd7070Spatrick       ai->writeTemplateInstantiationArgs(OS);
3579e5dd7070Spatrick     }
3580a9ac8606Spatrick     OS << ");\n"
3581a9ac8606Spatrick        << "    }\n";
3582e5dd7070Spatrick   }
3583e5dd7070Spatrick   OS << "  } // end switch\n"
3584e5dd7070Spatrick      << "  llvm_unreachable(\"Unknown attribute!\");\n"
3585e5dd7070Spatrick      << "  return nullptr;\n";
3586e5dd7070Spatrick }
3587e5dd7070Spatrick 
3588e5dd7070Spatrick // Emits code to instantiate dependent attributes on templates.
EmitClangAttrTemplateInstantiate(RecordKeeper & Records,raw_ostream & OS)3589e5dd7070Spatrick void EmitClangAttrTemplateInstantiate(RecordKeeper &Records, raw_ostream &OS) {
3590e5dd7070Spatrick   emitSourceFileHeader("Template instantiation code for attributes", OS);
3591e5dd7070Spatrick 
3592e5dd7070Spatrick   std::vector<Record*> Attrs = Records.getAllDerivedDefinitions("Attr");
3593e5dd7070Spatrick 
3594e5dd7070Spatrick   OS << "namespace clang {\n"
3595e5dd7070Spatrick      << "namespace sema {\n\n"
3596e5dd7070Spatrick      << "Attr *instantiateTemplateAttribute(const Attr *At, ASTContext &C, "
3597e5dd7070Spatrick      << "Sema &S,\n"
3598e5dd7070Spatrick      << "        const MultiLevelTemplateArgumentList &TemplateArgs) {\n";
3599e5dd7070Spatrick   EmitClangAttrTemplateInstantiateHelper(Attrs, OS, /*AppliesToDecl*/false);
3600e5dd7070Spatrick   OS << "}\n\n"
3601e5dd7070Spatrick      << "Attr *instantiateTemplateAttributeForDecl(const Attr *At,\n"
3602e5dd7070Spatrick      << " ASTContext &C, Sema &S,\n"
3603e5dd7070Spatrick      << "        const MultiLevelTemplateArgumentList &TemplateArgs) {\n";
3604e5dd7070Spatrick   EmitClangAttrTemplateInstantiateHelper(Attrs, OS, /*AppliesToDecl*/true);
3605e5dd7070Spatrick   OS << "}\n\n"
3606e5dd7070Spatrick      << "} // end namespace sema\n"
3607e5dd7070Spatrick      << "} // end namespace clang\n";
3608e5dd7070Spatrick }
3609e5dd7070Spatrick 
3610e5dd7070Spatrick // Emits the list of parsed attributes.
EmitClangAttrParsedAttrList(RecordKeeper & Records,raw_ostream & OS)3611e5dd7070Spatrick void EmitClangAttrParsedAttrList(RecordKeeper &Records, raw_ostream &OS) {
3612e5dd7070Spatrick   emitSourceFileHeader("List of all attributes that Clang recognizes", OS);
3613e5dd7070Spatrick 
3614e5dd7070Spatrick   OS << "#ifndef PARSED_ATTR\n";
3615e5dd7070Spatrick   OS << "#define PARSED_ATTR(NAME) NAME\n";
3616e5dd7070Spatrick   OS << "#endif\n\n";
3617e5dd7070Spatrick 
3618e5dd7070Spatrick   ParsedAttrMap Names = getParsedAttrList(Records);
3619e5dd7070Spatrick   for (const auto &I : Names) {
3620e5dd7070Spatrick     OS << "PARSED_ATTR(" << I.first << ")\n";
3621e5dd7070Spatrick   }
3622e5dd7070Spatrick }
3623e5dd7070Spatrick 
isArgVariadic(const Record & R,StringRef AttrName)3624e5dd7070Spatrick static bool isArgVariadic(const Record &R, StringRef AttrName) {
3625e5dd7070Spatrick   return createArgument(R, AttrName)->isVariadic();
3626e5dd7070Spatrick }
3627e5dd7070Spatrick 
emitArgInfo(const Record & R,raw_ostream & OS)3628e5dd7070Spatrick static void emitArgInfo(const Record &R, raw_ostream &OS) {
3629e5dd7070Spatrick   // This function will count the number of arguments specified for the
3630e5dd7070Spatrick   // attribute and emit the number of required arguments followed by the
3631e5dd7070Spatrick   // number of optional arguments.
3632e5dd7070Spatrick   std::vector<Record *> Args = R.getValueAsListOfDefs("Args");
3633*12c85518Srobert   unsigned ArgCount = 0, OptCount = 0, ArgMemberCount = 0;
3634e5dd7070Spatrick   bool HasVariadic = false;
3635e5dd7070Spatrick   for (const auto *Arg : Args) {
3636e5dd7070Spatrick     // If the arg is fake, it's the user's job to supply it: general parsing
3637e5dd7070Spatrick     // logic shouldn't need to know anything about it.
3638e5dd7070Spatrick     if (Arg->getValueAsBit("Fake"))
3639e5dd7070Spatrick       continue;
3640e5dd7070Spatrick     Arg->getValueAsBit("Optional") ? ++OptCount : ++ArgCount;
3641*12c85518Srobert     ++ArgMemberCount;
3642e5dd7070Spatrick     if (!HasVariadic && isArgVariadic(*Arg, R.getName()))
3643e5dd7070Spatrick       HasVariadic = true;
3644e5dd7070Spatrick   }
3645e5dd7070Spatrick 
3646e5dd7070Spatrick   // If there is a variadic argument, we will set the optional argument count
3647e5dd7070Spatrick   // to its largest value. Since it's currently a 4-bit number, we set it to 15.
3648*12c85518Srobert   OS << "    /*NumArgs=*/" << ArgCount << ",\n";
3649*12c85518Srobert   OS << "    /*OptArgs=*/" << (HasVariadic ? 15 : OptCount) << ",\n";
3650*12c85518Srobert   OS << "    /*NumArgMembers=*/" << ArgMemberCount << ",\n";
3651e5dd7070Spatrick }
3652e5dd7070Spatrick 
GetDiagnosticSpelling(const Record & R)3653e5dd7070Spatrick static std::string GetDiagnosticSpelling(const Record &R) {
3654ec727ea7Spatrick   std::string Ret = std::string(R.getValueAsString("DiagSpelling"));
3655e5dd7070Spatrick   if (!Ret.empty())
3656e5dd7070Spatrick     return Ret;
3657e5dd7070Spatrick 
3658e5dd7070Spatrick   // If we couldn't find the DiagSpelling in this object, we can check to see
3659e5dd7070Spatrick   // if the object is one that has a base, and if it is, loop up to the Base
3660e5dd7070Spatrick   // member recursively.
3661e5dd7070Spatrick   if (auto Base = R.getValueAsOptionalDef(BaseFieldName))
3662e5dd7070Spatrick     return GetDiagnosticSpelling(*Base);
3663e5dd7070Spatrick 
3664e5dd7070Spatrick   return "";
3665e5dd7070Spatrick }
3666e5dd7070Spatrick 
CalculateDiagnostic(const Record & S)3667e5dd7070Spatrick static std::string CalculateDiagnostic(const Record &S) {
3668e5dd7070Spatrick   // If the SubjectList object has a custom diagnostic associated with it,
3669e5dd7070Spatrick   // return that directly.
3670e5dd7070Spatrick   const StringRef CustomDiag = S.getValueAsString("CustomDiag");
3671e5dd7070Spatrick   if (!CustomDiag.empty())
3672e5dd7070Spatrick     return ("\"" + Twine(CustomDiag) + "\"").str();
3673e5dd7070Spatrick 
3674e5dd7070Spatrick   std::vector<std::string> DiagList;
3675e5dd7070Spatrick   std::vector<Record *> Subjects = S.getValueAsListOfDefs("Subjects");
3676e5dd7070Spatrick   for (const auto *Subject : Subjects) {
3677e5dd7070Spatrick     const Record &R = *Subject;
3678e5dd7070Spatrick     // Get the diagnostic text from the Decl or Stmt node given.
3679e5dd7070Spatrick     std::string V = GetDiagnosticSpelling(R);
3680e5dd7070Spatrick     if (V.empty()) {
3681e5dd7070Spatrick       PrintError(R.getLoc(),
3682e5dd7070Spatrick                  "Could not determine diagnostic spelling for the node: " +
3683e5dd7070Spatrick                      R.getName() + "; please add one to DeclNodes.td");
3684e5dd7070Spatrick     } else {
3685e5dd7070Spatrick       // The node may contain a list of elements itself, so split the elements
3686e5dd7070Spatrick       // by a comma, and trim any whitespace.
3687e5dd7070Spatrick       SmallVector<StringRef, 2> Frags;
3688e5dd7070Spatrick       llvm::SplitString(V, Frags, ",");
3689e5dd7070Spatrick       for (auto Str : Frags) {
3690ec727ea7Spatrick         DiagList.push_back(std::string(Str.trim()));
3691e5dd7070Spatrick       }
3692e5dd7070Spatrick     }
3693e5dd7070Spatrick   }
3694e5dd7070Spatrick 
3695e5dd7070Spatrick   if (DiagList.empty()) {
3696e5dd7070Spatrick     PrintFatalError(S.getLoc(),
3697e5dd7070Spatrick                     "Could not deduce diagnostic argument for Attr subjects");
3698e5dd7070Spatrick     return "";
3699e5dd7070Spatrick   }
3700e5dd7070Spatrick 
3701e5dd7070Spatrick   // FIXME: this is not particularly good for localization purposes and ideally
3702e5dd7070Spatrick   // should be part of the diagnostics engine itself with some sort of list
3703e5dd7070Spatrick   // specifier.
3704e5dd7070Spatrick 
3705e5dd7070Spatrick   // A single member of the list can be returned directly.
3706e5dd7070Spatrick   if (DiagList.size() == 1)
3707e5dd7070Spatrick     return '"' + DiagList.front() + '"';
3708e5dd7070Spatrick 
3709e5dd7070Spatrick   if (DiagList.size() == 2)
3710e5dd7070Spatrick     return '"' + DiagList[0] + " and " + DiagList[1] + '"';
3711e5dd7070Spatrick 
3712e5dd7070Spatrick   // If there are more than two in the list, we serialize the first N - 1
3713e5dd7070Spatrick   // elements with a comma. This leaves the string in the state: foo, bar,
3714e5dd7070Spatrick   // baz (but misses quux). We can then add ", and " for the last element
3715e5dd7070Spatrick   // manually.
3716e5dd7070Spatrick   std::string Diag = llvm::join(DiagList.begin(), DiagList.end() - 1, ", ");
3717e5dd7070Spatrick   return '"' + Diag + ", and " + *(DiagList.end() - 1) + '"';
3718e5dd7070Spatrick }
3719e5dd7070Spatrick 
GetSubjectWithSuffix(const Record * R)3720e5dd7070Spatrick static std::string GetSubjectWithSuffix(const Record *R) {
3721ec727ea7Spatrick   const std::string &B = std::string(R->getName());
3722e5dd7070Spatrick   if (B == "DeclBase")
3723e5dd7070Spatrick     return "Decl";
3724e5dd7070Spatrick   return B + "Decl";
3725e5dd7070Spatrick }
3726e5dd7070Spatrick 
functionNameForCustomAppertainsTo(const Record & Subject)3727e5dd7070Spatrick static std::string functionNameForCustomAppertainsTo(const Record &Subject) {
3728e5dd7070Spatrick   return "is" + Subject.getName().str();
3729e5dd7070Spatrick }
3730e5dd7070Spatrick 
GenerateCustomAppertainsTo(const Record & Subject,raw_ostream & OS)3731ec727ea7Spatrick static void GenerateCustomAppertainsTo(const Record &Subject, raw_ostream &OS) {
3732e5dd7070Spatrick   std::string FnName = functionNameForCustomAppertainsTo(Subject);
3733e5dd7070Spatrick 
3734ec727ea7Spatrick   // If this code has already been generated, we don't need to do anything.
3735e5dd7070Spatrick   static std::set<std::string> CustomSubjectSet;
3736e5dd7070Spatrick   auto I = CustomSubjectSet.find(FnName);
3737e5dd7070Spatrick   if (I != CustomSubjectSet.end())
3738ec727ea7Spatrick     return;
3739e5dd7070Spatrick 
3740e5dd7070Spatrick   // This only works with non-root Decls.
3741e5dd7070Spatrick   Record *Base = Subject.getValueAsDef(BaseFieldName);
3742e5dd7070Spatrick 
3743e5dd7070Spatrick   // Not currently support custom subjects within custom subjects.
3744e5dd7070Spatrick   if (Base->isSubClassOf("SubsetSubject")) {
3745e5dd7070Spatrick     PrintFatalError(Subject.getLoc(),
3746e5dd7070Spatrick                     "SubsetSubjects within SubsetSubjects is not supported");
3747ec727ea7Spatrick     return;
3748e5dd7070Spatrick   }
3749e5dd7070Spatrick 
3750e5dd7070Spatrick   OS << "static bool " << FnName << "(const Decl *D) {\n";
3751e5dd7070Spatrick   OS << "  if (const auto *S = dyn_cast<";
3752e5dd7070Spatrick   OS << GetSubjectWithSuffix(Base);
3753e5dd7070Spatrick   OS << ">(D))\n";
3754e5dd7070Spatrick   OS << "    return " << Subject.getValueAsString("CheckCode") << ";\n";
3755e5dd7070Spatrick   OS << "  return false;\n";
3756e5dd7070Spatrick   OS << "}\n\n";
3757e5dd7070Spatrick 
3758e5dd7070Spatrick   CustomSubjectSet.insert(FnName);
3759e5dd7070Spatrick }
3760e5dd7070Spatrick 
GenerateAppertainsTo(const Record & Attr,raw_ostream & OS)3761ec727ea7Spatrick static void GenerateAppertainsTo(const Record &Attr, raw_ostream &OS) {
3762e5dd7070Spatrick   // If the attribute does not contain a Subjects definition, then use the
3763e5dd7070Spatrick   // default appertainsTo logic.
3764e5dd7070Spatrick   if (Attr.isValueUnset("Subjects"))
3765ec727ea7Spatrick     return;
3766e5dd7070Spatrick 
3767e5dd7070Spatrick   const Record *SubjectObj = Attr.getValueAsDef("Subjects");
3768e5dd7070Spatrick   std::vector<Record *> Subjects = SubjectObj->getValueAsListOfDefs("Subjects");
3769e5dd7070Spatrick 
3770e5dd7070Spatrick   // If the list of subjects is empty, it is assumed that the attribute
3771e5dd7070Spatrick   // appertains to everything.
3772e5dd7070Spatrick   if (Subjects.empty())
3773ec727ea7Spatrick     return;
3774e5dd7070Spatrick 
3775e5dd7070Spatrick   bool Warn = SubjectObj->getValueAsDef("Diag")->getValueAsBit("Warn");
3776e5dd7070Spatrick 
3777a9ac8606Spatrick   // Split the subjects into declaration subjects and statement subjects.
3778a9ac8606Spatrick   // FIXME: subset subjects are added to the declaration list until there are
3779a9ac8606Spatrick   // enough statement attributes with custom subject needs to warrant
3780a9ac8606Spatrick   // the implementation effort.
3781a9ac8606Spatrick   std::vector<Record *> DeclSubjects, StmtSubjects;
3782a9ac8606Spatrick   llvm::copy_if(
3783a9ac8606Spatrick       Subjects, std::back_inserter(DeclSubjects), [](const Record *R) {
3784a9ac8606Spatrick         return R->isSubClassOf("SubsetSubject") || !R->isSubClassOf("StmtNode");
3785a9ac8606Spatrick       });
3786a9ac8606Spatrick   llvm::copy_if(Subjects, std::back_inserter(StmtSubjects),
3787a9ac8606Spatrick                 [](const Record *R) { return R->isSubClassOf("StmtNode"); });
3788a9ac8606Spatrick 
3789a9ac8606Spatrick   // We should have sorted all of the subjects into two lists.
3790a9ac8606Spatrick   // FIXME: this assertion will be wrong if we ever add type attribute subjects.
3791a9ac8606Spatrick   assert(DeclSubjects.size() + StmtSubjects.size() == Subjects.size());
3792a9ac8606Spatrick 
3793a9ac8606Spatrick   if (DeclSubjects.empty()) {
3794a9ac8606Spatrick     // If there are no decl subjects but there are stmt subjects, diagnose
3795a9ac8606Spatrick     // trying to apply a statement attribute to a declaration.
3796a9ac8606Spatrick     if (!StmtSubjects.empty()) {
3797a9ac8606Spatrick       OS << "bool diagAppertainsToDecl(Sema &S, const ParsedAttr &AL, ";
3798a9ac8606Spatrick       OS << "const Decl *D) const override {\n";
3799*12c85518Srobert       OS << "  S.Diag(AL.getLoc(), diag::err_attribute_invalid_on_decl)\n";
3800a9ac8606Spatrick       OS << "    << AL << D->getLocation();\n";
3801a9ac8606Spatrick       OS << "  return false;\n";
3802a9ac8606Spatrick       OS << "}\n\n";
3803a9ac8606Spatrick     }
3804a9ac8606Spatrick   } else {
3805a9ac8606Spatrick     // Otherwise, generate an appertainsTo check specific to this attribute
3806a9ac8606Spatrick     // which checks all of the given subjects against the Decl passed in.
3807ec727ea7Spatrick     OS << "bool diagAppertainsToDecl(Sema &S, ";
3808ec727ea7Spatrick     OS << "const ParsedAttr &Attr, const Decl *D) const override {\n";
3809ec727ea7Spatrick     OS << "  if (";
3810a9ac8606Spatrick     for (auto I = DeclSubjects.begin(), E = DeclSubjects.end(); I != E; ++I) {
3811ec727ea7Spatrick       // If the subject has custom code associated with it, use the generated
3812ec727ea7Spatrick       // function for it. The function cannot be inlined into this check (yet)
3813ec727ea7Spatrick       // because it requires the subject to be of a specific type, and were that
3814a9ac8606Spatrick       // information inlined here, it would not support an attribute with
3815a9ac8606Spatrick       // multiple custom subjects.
3816a9ac8606Spatrick       if ((*I)->isSubClassOf("SubsetSubject"))
3817ec727ea7Spatrick         OS << "!" << functionNameForCustomAppertainsTo(**I) << "(D)";
3818a9ac8606Spatrick       else
3819ec727ea7Spatrick         OS << "!isa<" << GetSubjectWithSuffix(*I) << ">(D)";
3820e5dd7070Spatrick 
3821e5dd7070Spatrick       if (I + 1 != E)
3822ec727ea7Spatrick         OS << " && ";
3823e5dd7070Spatrick     }
3824ec727ea7Spatrick     OS << ") {\n";
3825ec727ea7Spatrick     OS << "    S.Diag(Attr.getLoc(), diag::";
3826a9ac8606Spatrick     OS << (Warn ? "warn_attribute_wrong_decl_type_str"
3827a9ac8606Spatrick                 : "err_attribute_wrong_decl_type_str");
3828ec727ea7Spatrick     OS << ")\n";
3829ec727ea7Spatrick     OS << "      << Attr << ";
3830ec727ea7Spatrick     OS << CalculateDiagnostic(*SubjectObj) << ";\n";
3831ec727ea7Spatrick     OS << "    return false;\n";
3832ec727ea7Spatrick     OS << "  }\n";
3833ec727ea7Spatrick     OS << "  return true;\n";
3834ec727ea7Spatrick     OS << "}\n\n";
3835e5dd7070Spatrick   }
3836e5dd7070Spatrick 
3837a9ac8606Spatrick   if (StmtSubjects.empty()) {
3838a9ac8606Spatrick     // If there are no stmt subjects but there are decl subjects, diagnose
3839a9ac8606Spatrick     // trying to apply a declaration attribute to a statement.
3840a9ac8606Spatrick     if (!DeclSubjects.empty()) {
3841a9ac8606Spatrick       OS << "bool diagAppertainsToStmt(Sema &S, const ParsedAttr &AL, ";
3842a9ac8606Spatrick       OS << "const Stmt *St) const override {\n";
3843a9ac8606Spatrick       OS << "  S.Diag(AL.getLoc(), diag::err_decl_attribute_invalid_on_stmt)\n";
3844a9ac8606Spatrick       OS << "    << AL << St->getBeginLoc();\n";
3845a9ac8606Spatrick       OS << "  return false;\n";
3846a9ac8606Spatrick       OS << "}\n\n";
3847a9ac8606Spatrick     }
3848a9ac8606Spatrick   } else {
3849a9ac8606Spatrick     // Now, do the same for statements.
3850a9ac8606Spatrick     OS << "bool diagAppertainsToStmt(Sema &S, ";
3851a9ac8606Spatrick     OS << "const ParsedAttr &Attr, const Stmt *St) const override {\n";
3852a9ac8606Spatrick     OS << "  if (";
3853a9ac8606Spatrick     for (auto I = StmtSubjects.begin(), E = StmtSubjects.end(); I != E; ++I) {
3854a9ac8606Spatrick       OS << "!isa<" << (*I)->getName() << ">(St)";
3855a9ac8606Spatrick       if (I + 1 != E)
3856a9ac8606Spatrick         OS << " && ";
3857a9ac8606Spatrick     }
3858a9ac8606Spatrick     OS << ") {\n";
3859a9ac8606Spatrick     OS << "    S.Diag(Attr.getLoc(), diag::";
3860a9ac8606Spatrick     OS << (Warn ? "warn_attribute_wrong_decl_type_str"
3861a9ac8606Spatrick                 : "err_attribute_wrong_decl_type_str");
3862a9ac8606Spatrick     OS << ")\n";
3863a9ac8606Spatrick     OS << "      << Attr << ";
3864a9ac8606Spatrick     OS << CalculateDiagnostic(*SubjectObj) << ";\n";
3865a9ac8606Spatrick     OS << "    return false;\n";
3866a9ac8606Spatrick     OS << "  }\n";
3867a9ac8606Spatrick     OS << "  return true;\n";
3868a9ac8606Spatrick     OS << "}\n\n";
3869a9ac8606Spatrick   }
3870a9ac8606Spatrick }
3871a9ac8606Spatrick 
3872a9ac8606Spatrick // Generates the mutual exclusion checks. The checks for parsed attributes are
3873a9ac8606Spatrick // written into OS and the checks for merging declaration attributes are
3874a9ac8606Spatrick // written into MergeOS.
GenerateMutualExclusionsChecks(const Record & Attr,const RecordKeeper & Records,raw_ostream & OS,raw_ostream & MergeDeclOS,raw_ostream & MergeStmtOS)3875a9ac8606Spatrick static void GenerateMutualExclusionsChecks(const Record &Attr,
3876a9ac8606Spatrick                                            const RecordKeeper &Records,
3877a9ac8606Spatrick                                            raw_ostream &OS,
3878a9ac8606Spatrick                                            raw_ostream &MergeDeclOS,
3879a9ac8606Spatrick                                            raw_ostream &MergeStmtOS) {
3880a9ac8606Spatrick   // Find all of the definitions that inherit from MutualExclusions and include
3881a9ac8606Spatrick   // the given attribute in the list of exclusions to generate the
3882a9ac8606Spatrick   // diagMutualExclusion() check.
3883a9ac8606Spatrick   std::vector<Record *> ExclusionsList =
3884a9ac8606Spatrick       Records.getAllDerivedDefinitions("MutualExclusions");
3885a9ac8606Spatrick 
3886a9ac8606Spatrick   // We don't do any of this magic for type attributes yet.
3887a9ac8606Spatrick   if (Attr.isSubClassOf("TypeAttr"))
3888a9ac8606Spatrick     return;
3889a9ac8606Spatrick 
3890a9ac8606Spatrick   // This means the attribute is either a statement attribute, a decl
3891a9ac8606Spatrick   // attribute, or both; find out which.
3892a9ac8606Spatrick   bool CurAttrIsStmtAttr =
3893a9ac8606Spatrick       Attr.isSubClassOf("StmtAttr") || Attr.isSubClassOf("DeclOrStmtAttr");
3894a9ac8606Spatrick   bool CurAttrIsDeclAttr =
3895a9ac8606Spatrick       !CurAttrIsStmtAttr || Attr.isSubClassOf("DeclOrStmtAttr");
3896a9ac8606Spatrick 
3897a9ac8606Spatrick   std::vector<std::string> DeclAttrs, StmtAttrs;
3898a9ac8606Spatrick 
3899a9ac8606Spatrick   for (const Record *Exclusion : ExclusionsList) {
3900a9ac8606Spatrick     std::vector<Record *> MutuallyExclusiveAttrs =
3901a9ac8606Spatrick         Exclusion->getValueAsListOfDefs("Exclusions");
3902a9ac8606Spatrick     auto IsCurAttr = [Attr](const Record *R) {
3903a9ac8606Spatrick       return R->getName() == Attr.getName();
3904a9ac8606Spatrick     };
3905a9ac8606Spatrick     if (llvm::any_of(MutuallyExclusiveAttrs, IsCurAttr)) {
3906a9ac8606Spatrick       // This list of exclusions includes the attribute we're looking for, so
3907a9ac8606Spatrick       // add the exclusive attributes to the proper list for checking.
3908a9ac8606Spatrick       for (const Record *AttrToExclude : MutuallyExclusiveAttrs) {
3909a9ac8606Spatrick         if (IsCurAttr(AttrToExclude))
3910a9ac8606Spatrick           continue;
3911a9ac8606Spatrick 
3912a9ac8606Spatrick         if (CurAttrIsStmtAttr)
3913a9ac8606Spatrick           StmtAttrs.push_back((AttrToExclude->getName() + "Attr").str());
3914a9ac8606Spatrick         if (CurAttrIsDeclAttr)
3915a9ac8606Spatrick           DeclAttrs.push_back((AttrToExclude->getName() + "Attr").str());
3916a9ac8606Spatrick       }
3917a9ac8606Spatrick     }
3918a9ac8606Spatrick   }
3919a9ac8606Spatrick 
3920a9ac8606Spatrick   // If there are any decl or stmt attributes, silence -Woverloaded-virtual
3921a9ac8606Spatrick   // warnings for them both.
3922a9ac8606Spatrick   if (!DeclAttrs.empty() || !StmtAttrs.empty())
3923a9ac8606Spatrick     OS << "  using ParsedAttrInfo::diagMutualExclusion;\n\n";
3924a9ac8606Spatrick 
3925a9ac8606Spatrick   // If we discovered any decl or stmt attributes to test for, generate the
3926a9ac8606Spatrick   // predicates for them now.
3927a9ac8606Spatrick   if (!DeclAttrs.empty()) {
3928a9ac8606Spatrick     // Generate the ParsedAttrInfo subclass logic for declarations.
3929a9ac8606Spatrick     OS << "  bool diagMutualExclusion(Sema &S, const ParsedAttr &AL, "
3930a9ac8606Spatrick        << "const Decl *D) const override {\n";
3931a9ac8606Spatrick     for (const std::string &A : DeclAttrs) {
3932a9ac8606Spatrick       OS << "    if (const auto *A = D->getAttr<" << A << ">()) {\n";
3933a9ac8606Spatrick       OS << "      S.Diag(AL.getLoc(), diag::err_attributes_are_not_compatible)"
3934a9ac8606Spatrick          << " << AL << A;\n";
3935a9ac8606Spatrick       OS << "      S.Diag(A->getLocation(), diag::note_conflicting_attribute);";
3936a9ac8606Spatrick       OS << "      \nreturn false;\n";
3937a9ac8606Spatrick       OS << "    }\n";
3938a9ac8606Spatrick     }
3939a9ac8606Spatrick     OS << "    return true;\n";
3940a9ac8606Spatrick     OS << "  }\n\n";
3941a9ac8606Spatrick 
3942a9ac8606Spatrick     // Also generate the declaration attribute merging logic if the current
3943a9ac8606Spatrick     // attribute is one that can be inheritted on a declaration. It is assumed
3944a9ac8606Spatrick     // this code will be executed in the context of a function with parameters:
3945a9ac8606Spatrick     // Sema &S, Decl *D, Attr *A and that returns a bool (false on diagnostic,
3946a9ac8606Spatrick     // true on success).
3947a9ac8606Spatrick     if (Attr.isSubClassOf("InheritableAttr")) {
3948a9ac8606Spatrick       MergeDeclOS << "  if (const auto *Second = dyn_cast<"
3949a9ac8606Spatrick                   << (Attr.getName() + "Attr").str() << ">(A)) {\n";
3950a9ac8606Spatrick       for (const std::string &A : DeclAttrs) {
3951a9ac8606Spatrick         MergeDeclOS << "    if (const auto *First = D->getAttr<" << A
3952a9ac8606Spatrick                     << ">()) {\n";
3953a9ac8606Spatrick         MergeDeclOS << "      S.Diag(First->getLocation(), "
3954a9ac8606Spatrick                     << "diag::err_attributes_are_not_compatible) << First << "
3955a9ac8606Spatrick                     << "Second;\n";
3956a9ac8606Spatrick         MergeDeclOS << "      S.Diag(Second->getLocation(), "
3957a9ac8606Spatrick                     << "diag::note_conflicting_attribute);\n";
3958a9ac8606Spatrick         MergeDeclOS << "      return false;\n";
3959a9ac8606Spatrick         MergeDeclOS << "    }\n";
3960a9ac8606Spatrick       }
3961a9ac8606Spatrick       MergeDeclOS << "    return true;\n";
3962a9ac8606Spatrick       MergeDeclOS << "  }\n";
3963a9ac8606Spatrick     }
3964a9ac8606Spatrick   }
3965a9ac8606Spatrick 
3966a9ac8606Spatrick   // Statement attributes are a bit different from declarations. With
3967a9ac8606Spatrick   // declarations, each attribute is added to the declaration as it is
3968a9ac8606Spatrick   // processed, and so you can look on the Decl * itself to see if there is a
3969a9ac8606Spatrick   // conflicting attribute. Statement attributes are processed as a group
3970a9ac8606Spatrick   // because AttributedStmt needs to tail-allocate all of the attribute nodes
3971a9ac8606Spatrick   // at once. This means we cannot check whether the statement already contains
3972a9ac8606Spatrick   // an attribute to check for the conflict. Instead, we need to check whether
3973a9ac8606Spatrick   // the given list of semantic attributes contain any conflicts. It is assumed
3974a9ac8606Spatrick   // this code will be executed in the context of a function with parameters:
3975a9ac8606Spatrick   // Sema &S, const SmallVectorImpl<const Attr *> &C. The code will be within a
3976a9ac8606Spatrick   // loop which loops over the container C with a loop variable named A to
3977a9ac8606Spatrick   // represent the current attribute to check for conflicts.
3978a9ac8606Spatrick   //
3979a9ac8606Spatrick   // FIXME: it would be nice not to walk over the list of potential attributes
3980a9ac8606Spatrick   // to apply to the statement more than once, but statements typically don't
3981a9ac8606Spatrick   // have long lists of attributes on them, so re-walking the list should not
3982a9ac8606Spatrick   // be an expensive operation.
3983a9ac8606Spatrick   if (!StmtAttrs.empty()) {
3984a9ac8606Spatrick     MergeStmtOS << "    if (const auto *Second = dyn_cast<"
3985a9ac8606Spatrick                 << (Attr.getName() + "Attr").str() << ">(A)) {\n";
3986a9ac8606Spatrick     MergeStmtOS << "      auto Iter = llvm::find_if(C, [](const Attr *Check) "
3987a9ac8606Spatrick                 << "{ return isa<";
3988a9ac8606Spatrick     interleave(
3989a9ac8606Spatrick         StmtAttrs, [&](const std::string &Name) { MergeStmtOS << Name; },
3990a9ac8606Spatrick         [&] { MergeStmtOS << ", "; });
3991a9ac8606Spatrick     MergeStmtOS << ">(Check); });\n";
3992a9ac8606Spatrick     MergeStmtOS << "      if (Iter != C.end()) {\n";
3993a9ac8606Spatrick     MergeStmtOS << "        S.Diag((*Iter)->getLocation(), "
3994a9ac8606Spatrick                 << "diag::err_attributes_are_not_compatible) << *Iter << "
3995a9ac8606Spatrick                 << "Second;\n";
3996a9ac8606Spatrick     MergeStmtOS << "        S.Diag(Second->getLocation(), "
3997a9ac8606Spatrick                 << "diag::note_conflicting_attribute);\n";
3998a9ac8606Spatrick     MergeStmtOS << "        return false;\n";
3999a9ac8606Spatrick     MergeStmtOS << "      }\n";
4000a9ac8606Spatrick     MergeStmtOS << "    }\n";
4001a9ac8606Spatrick   }
4002a9ac8606Spatrick }
4003a9ac8606Spatrick 
4004e5dd7070Spatrick static void
emitAttributeMatchRules(PragmaClangAttributeSupport & PragmaAttributeSupport,raw_ostream & OS)4005e5dd7070Spatrick emitAttributeMatchRules(PragmaClangAttributeSupport &PragmaAttributeSupport,
4006e5dd7070Spatrick                         raw_ostream &OS) {
4007e5dd7070Spatrick   OS << "static bool checkAttributeMatchRuleAppliesTo(const Decl *D, "
4008e5dd7070Spatrick      << AttributeSubjectMatchRule::EnumName << " rule) {\n";
4009e5dd7070Spatrick   OS << "  switch (rule) {\n";
4010e5dd7070Spatrick   for (const auto &Rule : PragmaAttributeSupport.Rules) {
4011e5dd7070Spatrick     if (Rule.isAbstractRule()) {
4012e5dd7070Spatrick       OS << "  case " << Rule.getEnumValue() << ":\n";
4013e5dd7070Spatrick       OS << "    assert(false && \"Abstract matcher rule isn't allowed\");\n";
4014e5dd7070Spatrick       OS << "    return false;\n";
4015e5dd7070Spatrick       continue;
4016e5dd7070Spatrick     }
4017e5dd7070Spatrick     std::vector<Record *> Subjects = Rule.getSubjects();
4018e5dd7070Spatrick     assert(!Subjects.empty() && "Missing subjects");
4019e5dd7070Spatrick     OS << "  case " << Rule.getEnumValue() << ":\n";
4020e5dd7070Spatrick     OS << "    return ";
4021e5dd7070Spatrick     for (auto I = Subjects.begin(), E = Subjects.end(); I != E; ++I) {
4022e5dd7070Spatrick       // If the subject has custom code associated with it, use the function
4023e5dd7070Spatrick       // that was generated for GenerateAppertainsTo to check if the declaration
4024e5dd7070Spatrick       // is valid.
4025e5dd7070Spatrick       if ((*I)->isSubClassOf("SubsetSubject"))
4026e5dd7070Spatrick         OS << functionNameForCustomAppertainsTo(**I) << "(D)";
4027e5dd7070Spatrick       else
4028e5dd7070Spatrick         OS << "isa<" << GetSubjectWithSuffix(*I) << ">(D)";
4029e5dd7070Spatrick 
4030e5dd7070Spatrick       if (I + 1 != E)
4031e5dd7070Spatrick         OS << " || ";
4032e5dd7070Spatrick     }
4033e5dd7070Spatrick     OS << ";\n";
4034e5dd7070Spatrick   }
4035e5dd7070Spatrick   OS << "  }\n";
4036e5dd7070Spatrick   OS << "  llvm_unreachable(\"Invalid match rule\");\nreturn false;\n";
4037e5dd7070Spatrick   OS << "}\n\n";
4038e5dd7070Spatrick }
4039e5dd7070Spatrick 
GenerateLangOptRequirements(const Record & R,raw_ostream & OS)4040ec727ea7Spatrick static void GenerateLangOptRequirements(const Record &R,
4041e5dd7070Spatrick                                         raw_ostream &OS) {
4042e5dd7070Spatrick   // If the attribute has an empty or unset list of language requirements,
4043ec727ea7Spatrick   // use the default handler.
4044e5dd7070Spatrick   std::vector<Record *> LangOpts = R.getValueAsListOfDefs("LangOpts");
4045e5dd7070Spatrick   if (LangOpts.empty())
4046ec727ea7Spatrick     return;
4047e5dd7070Spatrick 
4048*12c85518Srobert   OS << "bool acceptsLangOpts(const LangOptions &LangOpts) const override {\n";
4049*12c85518Srobert   OS << "  return " << GenerateTestExpression(LangOpts) << ";\n";
4050e5dd7070Spatrick   OS << "}\n\n";
4051e5dd7070Spatrick }
4052e5dd7070Spatrick 
GenerateTargetRequirements(const Record & Attr,const ParsedAttrMap & Dupes,raw_ostream & OS)4053ec727ea7Spatrick static void GenerateTargetRequirements(const Record &Attr,
4054e5dd7070Spatrick                                        const ParsedAttrMap &Dupes,
4055e5dd7070Spatrick                                        raw_ostream &OS) {
4056ec727ea7Spatrick   // If the attribute is not a target specific attribute, use the default
4057e5dd7070Spatrick   // target handler.
4058e5dd7070Spatrick   if (!Attr.isSubClassOf("TargetSpecificAttr"))
4059ec727ea7Spatrick     return;
4060e5dd7070Spatrick 
4061e5dd7070Spatrick   // Get the list of architectures to be tested for.
4062e5dd7070Spatrick   const Record *R = Attr.getValueAsDef("Target");
4063e5dd7070Spatrick   std::vector<StringRef> Arches = R->getValueAsListOfStrings("Arches");
4064e5dd7070Spatrick 
4065e5dd7070Spatrick   // If there are other attributes which share the same parsed attribute kind,
4066e5dd7070Spatrick   // such as target-specific attributes with a shared spelling, collapse the
4067e5dd7070Spatrick   // duplicate architectures. This is required because a shared target-specific
4068e5dd7070Spatrick   // attribute has only one ParsedAttr::Kind enumeration value, but it
4069e5dd7070Spatrick   // applies to multiple target architectures. In order for the attribute to be
4070e5dd7070Spatrick   // considered valid, all of its architectures need to be included.
4071e5dd7070Spatrick   if (!Attr.isValueUnset("ParseKind")) {
4072e5dd7070Spatrick     const StringRef APK = Attr.getValueAsString("ParseKind");
4073e5dd7070Spatrick     for (const auto &I : Dupes) {
4074e5dd7070Spatrick       if (I.first == APK) {
4075e5dd7070Spatrick         std::vector<StringRef> DA =
4076e5dd7070Spatrick             I.second->getValueAsDef("Target")->getValueAsListOfStrings(
4077e5dd7070Spatrick                 "Arches");
4078e5dd7070Spatrick         Arches.insert(Arches.end(), DA.begin(), DA.end());
4079e5dd7070Spatrick       }
4080e5dd7070Spatrick     }
4081e5dd7070Spatrick   }
4082e5dd7070Spatrick 
4083e5dd7070Spatrick   std::string FnName = "isTarget";
4084e5dd7070Spatrick   std::string Test;
4085e5dd7070Spatrick   bool UsesT = GenerateTargetSpecificAttrChecks(R, Arches, Test, &FnName);
4086e5dd7070Spatrick 
4087ec727ea7Spatrick   OS << "bool existsInTarget(const TargetInfo &Target) const override {\n";
4088e5dd7070Spatrick   if (UsesT)
4089e5dd7070Spatrick     OS << "  const llvm::Triple &T = Target.getTriple(); (void)T;\n";
4090e5dd7070Spatrick   OS << "  return " << Test << ";\n";
4091e5dd7070Spatrick   OS << "}\n\n";
4092e5dd7070Spatrick }
4093e5dd7070Spatrick 
GenerateSpellingIndexToSemanticSpelling(const Record & Attr,raw_ostream & OS)4094ec727ea7Spatrick static void GenerateSpellingIndexToSemanticSpelling(const Record &Attr,
4095e5dd7070Spatrick                                                     raw_ostream &OS) {
4096e5dd7070Spatrick   // If the attribute does not have a semantic form, we can bail out early.
4097e5dd7070Spatrick   if (!Attr.getValueAsBit("ASTNode"))
4098ec727ea7Spatrick     return;
4099e5dd7070Spatrick 
4100e5dd7070Spatrick   std::vector<FlattenedSpelling> Spellings = GetFlattenedSpellings(Attr);
4101e5dd7070Spatrick 
4102e5dd7070Spatrick   // If there are zero or one spellings, or all of the spellings share the same
4103e5dd7070Spatrick   // name, we can also bail out early.
4104e5dd7070Spatrick   if (Spellings.size() <= 1 || SpellingNamesAreCommon(Spellings))
4105ec727ea7Spatrick     return;
4106e5dd7070Spatrick 
4107e5dd7070Spatrick   // Generate the enumeration we will use for the mapping.
4108e5dd7070Spatrick   SemanticSpellingMap SemanticToSyntacticMap;
4109e5dd7070Spatrick   std::string Enum = CreateSemanticSpellings(Spellings, SemanticToSyntacticMap);
4110e5dd7070Spatrick   std::string Name = Attr.getName().str() + "AttrSpellingMap";
4111e5dd7070Spatrick 
4112ec727ea7Spatrick   OS << "unsigned spellingIndexToSemanticSpelling(";
4113ec727ea7Spatrick   OS << "const ParsedAttr &Attr) const override {\n";
4114e5dd7070Spatrick   OS << Enum;
4115e5dd7070Spatrick   OS << "  unsigned Idx = Attr.getAttributeSpellingListIndex();\n";
4116e5dd7070Spatrick   WriteSemanticSpellingSwitch("Idx", SemanticToSyntacticMap, OS);
4117e5dd7070Spatrick   OS << "}\n\n";
4118ec727ea7Spatrick }
4119e5dd7070Spatrick 
GenerateHandleDeclAttribute(const Record & Attr,raw_ostream & OS)4120ec727ea7Spatrick static void GenerateHandleDeclAttribute(const Record &Attr, raw_ostream &OS) {
4121ec727ea7Spatrick   // Only generate if Attr can be handled simply.
4122ec727ea7Spatrick   if (!Attr.getValueAsBit("SimpleHandler"))
4123ec727ea7Spatrick     return;
4124ec727ea7Spatrick 
4125ec727ea7Spatrick   // Generate a function which just converts from ParsedAttr to the Attr type.
4126ec727ea7Spatrick   OS << "AttrHandling handleDeclAttribute(Sema &S, Decl *D,";
4127ec727ea7Spatrick   OS << "const ParsedAttr &Attr) const override {\n";
4128ec727ea7Spatrick   OS << "  D->addAttr(::new (S.Context) " << Attr.getName();
4129ec727ea7Spatrick   OS << "Attr(S.Context, Attr));\n";
4130ec727ea7Spatrick   OS << "  return AttributeApplied;\n";
4131ec727ea7Spatrick   OS << "}\n\n";
4132e5dd7070Spatrick }
4133e5dd7070Spatrick 
isParamExpr(const Record * Arg)4134*12c85518Srobert static bool isParamExpr(const Record *Arg) {
4135*12c85518Srobert   return !Arg->getSuperClasses().empty() &&
4136*12c85518Srobert          llvm::StringSwitch<bool>(
4137*12c85518Srobert              Arg->getSuperClasses().back().first->getName())
4138*12c85518Srobert              .Case("ExprArgument", true)
4139*12c85518Srobert              .Case("VariadicExprArgument", true)
4140*12c85518Srobert              .Default(false);
4141*12c85518Srobert }
4142*12c85518Srobert 
GenerateIsParamExpr(const Record & Attr,raw_ostream & OS)4143*12c85518Srobert void GenerateIsParamExpr(const Record &Attr, raw_ostream &OS) {
4144*12c85518Srobert   OS << "bool isParamExpr(size_t N) const override {\n";
4145*12c85518Srobert   OS << "  return ";
4146*12c85518Srobert   auto Args = Attr.getValueAsListOfDefs("Args");
4147*12c85518Srobert   for (size_t I = 0; I < Args.size(); ++I)
4148*12c85518Srobert     if (isParamExpr(Args[I]))
4149*12c85518Srobert       OS << "(N == " << I << ") || ";
4150*12c85518Srobert   OS << "false;\n";
4151*12c85518Srobert   OS << "}\n\n";
4152*12c85518Srobert }
4153*12c85518Srobert 
GenerateHandleAttrWithDelayedArgs(RecordKeeper & Records,raw_ostream & OS)4154*12c85518Srobert void GenerateHandleAttrWithDelayedArgs(RecordKeeper &Records, raw_ostream &OS) {
4155*12c85518Srobert   OS << "static void handleAttrWithDelayedArgs(Sema &S, Decl *D, ";
4156*12c85518Srobert   OS << "const ParsedAttr &Attr) {\n";
4157*12c85518Srobert   OS << "  SmallVector<Expr *, 4> ArgExprs;\n";
4158*12c85518Srobert   OS << "  ArgExprs.reserve(Attr.getNumArgs());\n";
4159*12c85518Srobert   OS << "  for (unsigned I = 0; I < Attr.getNumArgs(); ++I) {\n";
4160*12c85518Srobert   OS << "    assert(!Attr.isArgIdent(I));\n";
4161*12c85518Srobert   OS << "    ArgExprs.push_back(Attr.getArgAsExpr(I));\n";
4162*12c85518Srobert   OS << "  }\n";
4163*12c85518Srobert   OS << "  clang::Attr *CreatedAttr = nullptr;\n";
4164*12c85518Srobert   OS << "  switch (Attr.getKind()) {\n";
4165*12c85518Srobert   OS << "  default:\n";
4166*12c85518Srobert   OS << "    llvm_unreachable(\"Attribute cannot hold delayed arguments.\");\n";
4167*12c85518Srobert   ParsedAttrMap Attrs = getParsedAttrList(Records);
4168*12c85518Srobert   for (const auto &I : Attrs) {
4169*12c85518Srobert     const Record &R = *I.second;
4170*12c85518Srobert     if (!R.getValueAsBit("AcceptsExprPack"))
4171*12c85518Srobert       continue;
4172*12c85518Srobert     OS << "  case ParsedAttr::AT_" << I.first << ": {\n";
4173*12c85518Srobert     OS << "    CreatedAttr = " << R.getName() << "Attr::CreateWithDelayedArgs";
4174*12c85518Srobert     OS << "(S.Context, ArgExprs.data(), ArgExprs.size(), Attr);\n";
4175*12c85518Srobert     OS << "    break;\n";
4176*12c85518Srobert     OS << "  }\n";
4177*12c85518Srobert   }
4178*12c85518Srobert   OS << "  }\n";
4179*12c85518Srobert   OS << "  D->addAttr(CreatedAttr);\n";
4180*12c85518Srobert   OS << "}\n\n";
4181*12c85518Srobert }
4182*12c85518Srobert 
IsKnownToGCC(const Record & Attr)4183e5dd7070Spatrick static bool IsKnownToGCC(const Record &Attr) {
4184e5dd7070Spatrick   // Look at the spellings for this subject; if there are any spellings which
4185e5dd7070Spatrick   // claim to be known to GCC, the attribute is known to GCC.
4186e5dd7070Spatrick   return llvm::any_of(
4187e5dd7070Spatrick       GetFlattenedSpellings(Attr),
4188e5dd7070Spatrick       [](const FlattenedSpelling &S) { return S.knownToGCC(); });
4189e5dd7070Spatrick }
4190e5dd7070Spatrick 
4191e5dd7070Spatrick /// Emits the parsed attribute helpers
EmitClangAttrParsedAttrImpl(RecordKeeper & Records,raw_ostream & OS)4192e5dd7070Spatrick void EmitClangAttrParsedAttrImpl(RecordKeeper &Records, raw_ostream &OS) {
4193e5dd7070Spatrick   emitSourceFileHeader("Parsed attribute helpers", OS);
4194e5dd7070Spatrick 
4195a9ac8606Spatrick   OS << "#if !defined(WANT_DECL_MERGE_LOGIC) && "
4196a9ac8606Spatrick      << "!defined(WANT_STMT_MERGE_LOGIC)\n";
4197e5dd7070Spatrick   PragmaClangAttributeSupport &PragmaAttributeSupport =
4198e5dd7070Spatrick       getPragmaAttributeSupport(Records);
4199e5dd7070Spatrick 
4200e5dd7070Spatrick   // Get the list of parsed attributes, and accept the optional list of
4201e5dd7070Spatrick   // duplicates due to the ParseKind.
4202e5dd7070Spatrick   ParsedAttrMap Dupes;
4203e5dd7070Spatrick   ParsedAttrMap Attrs = getParsedAttrList(Records, &Dupes);
4204e5dd7070Spatrick 
4205ec727ea7Spatrick   // Generate all of the custom appertainsTo functions that the attributes
4206ec727ea7Spatrick   // will be using.
4207ec727ea7Spatrick   for (auto I : Attrs) {
4208ec727ea7Spatrick     const Record &Attr = *I.second;
4209ec727ea7Spatrick     if (Attr.isValueUnset("Subjects"))
4210ec727ea7Spatrick       continue;
4211ec727ea7Spatrick     const Record *SubjectObj = Attr.getValueAsDef("Subjects");
4212ec727ea7Spatrick     for (auto Subject : SubjectObj->getValueAsListOfDefs("Subjects"))
4213ec727ea7Spatrick       if (Subject->isSubClassOf("SubsetSubject"))
4214ec727ea7Spatrick         GenerateCustomAppertainsTo(*Subject, OS);
4215ec727ea7Spatrick   }
4216e5dd7070Spatrick 
4217a9ac8606Spatrick   // This stream is used to collect all of the declaration attribute merging
4218a9ac8606Spatrick   // logic for performing mutual exclusion checks. This gets emitted at the
4219a9ac8606Spatrick   // end of the file in a helper function of its own.
4220a9ac8606Spatrick   std::string DeclMergeChecks, StmtMergeChecks;
4221a9ac8606Spatrick   raw_string_ostream MergeDeclOS(DeclMergeChecks), MergeStmtOS(StmtMergeChecks);
4222a9ac8606Spatrick 
4223ec727ea7Spatrick   // Generate a ParsedAttrInfo struct for each of the attributes.
4224e5dd7070Spatrick   for (auto I = Attrs.begin(), E = Attrs.end(); I != E; ++I) {
4225e5dd7070Spatrick     // TODO: If the attribute's kind appears in the list of duplicates, that is
4226e5dd7070Spatrick     // because it is a target-specific attribute that appears multiple times.
4227e5dd7070Spatrick     // It would be beneficial to test whether the duplicates are "similar
4228e5dd7070Spatrick     // enough" to each other to not cause problems. For instance, check that
4229e5dd7070Spatrick     // the spellings are identical, and custom parsing rules match, etc.
4230e5dd7070Spatrick 
4231e5dd7070Spatrick     // We need to generate struct instances based off ParsedAttrInfo from
4232e5dd7070Spatrick     // ParsedAttr.cpp.
4233ec727ea7Spatrick     const std::string &AttrName = I->first;
4234ec727ea7Spatrick     const Record &Attr = *I->second;
4235ec727ea7Spatrick     auto Spellings = GetFlattenedSpellings(Attr);
4236ec727ea7Spatrick     if (!Spellings.empty()) {
4237ec727ea7Spatrick       OS << "static constexpr ParsedAttrInfo::Spelling " << I->first
4238ec727ea7Spatrick          << "Spellings[] = {\n";
4239ec727ea7Spatrick       for (const auto &S : Spellings) {
4240ec727ea7Spatrick         const std::string &RawSpelling = S.name();
4241ec727ea7Spatrick         std::string Spelling;
4242ec727ea7Spatrick         if (!S.nameSpace().empty())
4243ec727ea7Spatrick           Spelling += S.nameSpace() + "::";
4244ec727ea7Spatrick         if (S.variety() == "GNU")
4245ec727ea7Spatrick           Spelling += NormalizeGNUAttrSpelling(RawSpelling);
4246ec727ea7Spatrick         else
4247ec727ea7Spatrick           Spelling += RawSpelling;
4248ec727ea7Spatrick         OS << "  {AttributeCommonInfo::AS_" << S.variety();
4249ec727ea7Spatrick         OS << ", \"" << Spelling << "\"},\n";
4250ec727ea7Spatrick       }
4251ec727ea7Spatrick       OS << "};\n";
4252ec727ea7Spatrick     }
4253*12c85518Srobert 
4254*12c85518Srobert     std::vector<std::string> ArgNames;
4255*12c85518Srobert     for (const auto &Arg : Attr.getValueAsListOfDefs("Args")) {
4256*12c85518Srobert       bool UnusedUnset;
4257*12c85518Srobert       if (Arg->getValueAsBitOrUnset("Fake", UnusedUnset))
4258*12c85518Srobert         continue;
4259*12c85518Srobert       ArgNames.push_back(Arg->getValueAsString("Name").str());
4260*12c85518Srobert       for (const auto &Class : Arg->getSuperClasses()) {
4261*12c85518Srobert         if (Class.first->getName().startswith("Variadic")) {
4262*12c85518Srobert           ArgNames.back().append("...");
4263*12c85518Srobert           break;
4264*12c85518Srobert         }
4265*12c85518Srobert       }
4266*12c85518Srobert     }
4267*12c85518Srobert     if (!ArgNames.empty()) {
4268*12c85518Srobert       OS << "static constexpr const char *" << I->first << "ArgNames[] = {\n";
4269*12c85518Srobert       for (const auto &N : ArgNames)
4270*12c85518Srobert         OS << '"' << N << "\",";
4271*12c85518Srobert       OS << "};\n";
4272*12c85518Srobert     }
4273*12c85518Srobert 
4274ec727ea7Spatrick     OS << "struct ParsedAttrInfo" << I->first
4275ec727ea7Spatrick        << " final : public ParsedAttrInfo {\n";
4276*12c85518Srobert     OS << "  constexpr ParsedAttrInfo" << I->first << "() : ParsedAttrInfo(\n";
4277*12c85518Srobert     OS << "    /*AttrKind=*/ParsedAttr::AT_" << AttrName << ",\n";
4278ec727ea7Spatrick     emitArgInfo(Attr, OS);
4279*12c85518Srobert     OS << "    /*HasCustomParsing=*/";
4280*12c85518Srobert     OS << Attr.getValueAsBit("HasCustomParsing") << ",\n";
4281*12c85518Srobert     OS << "    /*AcceptsExprPack=*/";
4282*12c85518Srobert     OS << Attr.getValueAsBit("AcceptsExprPack") << ",\n";
4283*12c85518Srobert     OS << "    /*IsTargetSpecific=*/";
4284*12c85518Srobert     OS << Attr.isSubClassOf("TargetSpecificAttr") << ",\n";
4285*12c85518Srobert     OS << "    /*IsType=*/";
4286*12c85518Srobert     OS << (Attr.isSubClassOf("TypeAttr") || Attr.isSubClassOf("DeclOrTypeAttr"))
4287*12c85518Srobert        << ",\n";
4288*12c85518Srobert     OS << "    /*IsStmt=*/";
4289a9ac8606Spatrick     OS << (Attr.isSubClassOf("StmtAttr") || Attr.isSubClassOf("DeclOrStmtAttr"))
4290*12c85518Srobert        << ",\n";
4291*12c85518Srobert     OS << "    /*IsKnownToGCC=*/";
4292*12c85518Srobert     OS << IsKnownToGCC(Attr) << ",\n";
4293*12c85518Srobert     OS << "    /*IsSupportedByPragmaAttribute=*/";
4294*12c85518Srobert     OS << PragmaAttributeSupport.isAttributedSupported(*I->second) << ",\n";
4295ec727ea7Spatrick     if (!Spellings.empty())
4296*12c85518Srobert       OS << "    /*Spellings=*/" << I->first << "Spellings,\n";
4297*12c85518Srobert     else
4298*12c85518Srobert       OS << "    /*Spellings=*/{},\n";
4299*12c85518Srobert     if (!ArgNames.empty())
4300*12c85518Srobert       OS << "    /*ArgNames=*/" << I->first << "ArgNames";
4301*12c85518Srobert     else
4302*12c85518Srobert       OS << "    /*ArgNames=*/{}";
4303*12c85518Srobert     OS << ") {}\n";
4304ec727ea7Spatrick     GenerateAppertainsTo(Attr, OS);
4305a9ac8606Spatrick     GenerateMutualExclusionsChecks(Attr, Records, OS, MergeDeclOS, MergeStmtOS);
4306ec727ea7Spatrick     GenerateLangOptRequirements(Attr, OS);
4307ec727ea7Spatrick     GenerateTargetRequirements(Attr, Dupes, OS);
4308ec727ea7Spatrick     GenerateSpellingIndexToSemanticSpelling(Attr, OS);
4309ec727ea7Spatrick     PragmaAttributeSupport.generateStrictConformsTo(*I->second, OS);
4310ec727ea7Spatrick     GenerateHandleDeclAttribute(Attr, OS);
4311*12c85518Srobert     GenerateIsParamExpr(Attr, OS);
4312ec727ea7Spatrick     OS << "static const ParsedAttrInfo" << I->first << " Instance;\n";
4313ec727ea7Spatrick     OS << "};\n";
4314ec727ea7Spatrick     OS << "const ParsedAttrInfo" << I->first << " ParsedAttrInfo" << I->first
4315ec727ea7Spatrick        << "::Instance;\n";
4316e5dd7070Spatrick   }
4317e5dd7070Spatrick 
4318ec727ea7Spatrick   OS << "static const ParsedAttrInfo *AttrInfoMap[] = {\n";
4319ec727ea7Spatrick   for (auto I = Attrs.begin(), E = Attrs.end(); I != E; ++I) {
4320ec727ea7Spatrick     OS << "&ParsedAttrInfo" << I->first << "::Instance,\n";
4321ec727ea7Spatrick   }
4322e5dd7070Spatrick   OS << "};\n\n";
4323e5dd7070Spatrick 
4324*12c85518Srobert   // Generate function for handling attributes with delayed arguments
4325*12c85518Srobert   GenerateHandleAttrWithDelayedArgs(Records, OS);
4326*12c85518Srobert 
4327e5dd7070Spatrick   // Generate the attribute match rules.
4328e5dd7070Spatrick   emitAttributeMatchRules(PragmaAttributeSupport, OS);
4329a9ac8606Spatrick 
4330a9ac8606Spatrick   OS << "#elif defined(WANT_DECL_MERGE_LOGIC)\n\n";
4331a9ac8606Spatrick 
4332a9ac8606Spatrick   // Write out the declaration merging check logic.
4333a9ac8606Spatrick   OS << "static bool DiagnoseMutualExclusions(Sema &S, const NamedDecl *D, "
4334a9ac8606Spatrick      << "const Attr *A) {\n";
4335a9ac8606Spatrick   OS << MergeDeclOS.str();
4336a9ac8606Spatrick   OS << "  return true;\n";
4337a9ac8606Spatrick   OS << "}\n\n";
4338a9ac8606Spatrick 
4339a9ac8606Spatrick   OS << "#elif defined(WANT_STMT_MERGE_LOGIC)\n\n";
4340a9ac8606Spatrick 
4341a9ac8606Spatrick   // Write out the statement merging check logic.
4342a9ac8606Spatrick   OS << "static bool DiagnoseMutualExclusions(Sema &S, "
4343a9ac8606Spatrick      << "const SmallVectorImpl<const Attr *> &C) {\n";
4344a9ac8606Spatrick   OS << "  for (const Attr *A : C) {\n";
4345a9ac8606Spatrick   OS << MergeStmtOS.str();
4346a9ac8606Spatrick   OS << "  }\n";
4347a9ac8606Spatrick   OS << "  return true;\n";
4348a9ac8606Spatrick   OS << "}\n\n";
4349a9ac8606Spatrick 
4350a9ac8606Spatrick   OS << "#endif\n";
4351e5dd7070Spatrick }
4352e5dd7070Spatrick 
4353e5dd7070Spatrick // Emits the kind list of parsed attributes
EmitClangAttrParsedAttrKinds(RecordKeeper & Records,raw_ostream & OS)4354e5dd7070Spatrick void EmitClangAttrParsedAttrKinds(RecordKeeper &Records, raw_ostream &OS) {
4355e5dd7070Spatrick   emitSourceFileHeader("Attribute name matcher", OS);
4356e5dd7070Spatrick 
4357e5dd7070Spatrick   std::vector<Record *> Attrs = Records.getAllDerivedDefinitions("Attr");
4358e5dd7070Spatrick   std::vector<StringMatcher::StringPair> GNU, Declspec, Microsoft, CXX11,
4359*12c85518Srobert       Keywords, Pragma, C2x, HLSLSemantic;
4360e5dd7070Spatrick   std::set<std::string> Seen;
4361e5dd7070Spatrick   for (const auto *A : Attrs) {
4362e5dd7070Spatrick     const Record &Attr = *A;
4363e5dd7070Spatrick 
4364e5dd7070Spatrick     bool SemaHandler = Attr.getValueAsBit("SemaHandler");
4365e5dd7070Spatrick     bool Ignored = Attr.getValueAsBit("Ignored");
4366e5dd7070Spatrick     if (SemaHandler || Ignored) {
4367e5dd7070Spatrick       // Attribute spellings can be shared between target-specific attributes,
4368e5dd7070Spatrick       // and can be shared between syntaxes for the same attribute. For
4369e5dd7070Spatrick       // instance, an attribute can be spelled GNU<"interrupt"> for an ARM-
4370e5dd7070Spatrick       // specific attribute, or MSP430-specific attribute. Additionally, an
4371e5dd7070Spatrick       // attribute can be spelled GNU<"dllexport"> and Declspec<"dllexport">
4372e5dd7070Spatrick       // for the same semantic attribute. Ultimately, we need to map each of
4373e5dd7070Spatrick       // these to a single AttributeCommonInfo::Kind value, but the
4374e5dd7070Spatrick       // StringMatcher class cannot handle duplicate match strings. So we
4375e5dd7070Spatrick       // generate a list of string to match based on the syntax, and emit
4376e5dd7070Spatrick       // multiple string matchers depending on the syntax used.
4377e5dd7070Spatrick       std::string AttrName;
4378e5dd7070Spatrick       if (Attr.isSubClassOf("TargetSpecificAttr") &&
4379e5dd7070Spatrick           !Attr.isValueUnset("ParseKind")) {
4380ec727ea7Spatrick         AttrName = std::string(Attr.getValueAsString("ParseKind"));
4381*12c85518Srobert         if (!Seen.insert(AttrName).second)
4382e5dd7070Spatrick           continue;
4383e5dd7070Spatrick       } else
4384e5dd7070Spatrick         AttrName = NormalizeAttrName(StringRef(Attr.getName())).str();
4385e5dd7070Spatrick 
4386e5dd7070Spatrick       std::vector<FlattenedSpelling> Spellings = GetFlattenedSpellings(Attr);
4387e5dd7070Spatrick       for (const auto &S : Spellings) {
4388e5dd7070Spatrick         const std::string &RawSpelling = S.name();
4389e5dd7070Spatrick         std::vector<StringMatcher::StringPair> *Matches = nullptr;
4390e5dd7070Spatrick         std::string Spelling;
4391e5dd7070Spatrick         const std::string &Variety = S.variety();
4392e5dd7070Spatrick         if (Variety == "CXX11") {
4393e5dd7070Spatrick           Matches = &CXX11;
4394ec727ea7Spatrick           if (!S.nameSpace().empty())
4395ec727ea7Spatrick             Spelling += S.nameSpace() + "::";
4396e5dd7070Spatrick         } else if (Variety == "C2x") {
4397e5dd7070Spatrick           Matches = &C2x;
4398ec727ea7Spatrick           if (!S.nameSpace().empty())
4399ec727ea7Spatrick             Spelling += S.nameSpace() + "::";
4400e5dd7070Spatrick         } else if (Variety == "GNU")
4401e5dd7070Spatrick           Matches = &GNU;
4402e5dd7070Spatrick         else if (Variety == "Declspec")
4403e5dd7070Spatrick           Matches = &Declspec;
4404e5dd7070Spatrick         else if (Variety == "Microsoft")
4405e5dd7070Spatrick           Matches = &Microsoft;
4406e5dd7070Spatrick         else if (Variety == "Keyword")
4407e5dd7070Spatrick           Matches = &Keywords;
4408e5dd7070Spatrick         else if (Variety == "Pragma")
4409e5dd7070Spatrick           Matches = &Pragma;
4410*12c85518Srobert         else if (Variety == "HLSLSemantic")
4411*12c85518Srobert           Matches = &HLSLSemantic;
4412e5dd7070Spatrick 
4413e5dd7070Spatrick         assert(Matches && "Unsupported spelling variety found");
4414e5dd7070Spatrick 
4415e5dd7070Spatrick         if (Variety == "GNU")
4416e5dd7070Spatrick           Spelling += NormalizeGNUAttrSpelling(RawSpelling);
4417e5dd7070Spatrick         else
4418e5dd7070Spatrick           Spelling += RawSpelling;
4419e5dd7070Spatrick 
4420e5dd7070Spatrick         if (SemaHandler)
4421e5dd7070Spatrick           Matches->push_back(StringMatcher::StringPair(
4422e5dd7070Spatrick               Spelling, "return AttributeCommonInfo::AT_" + AttrName + ";"));
4423e5dd7070Spatrick         else
4424e5dd7070Spatrick           Matches->push_back(StringMatcher::StringPair(
4425e5dd7070Spatrick               Spelling, "return AttributeCommonInfo::IgnoredAttribute;"));
4426e5dd7070Spatrick       }
4427e5dd7070Spatrick     }
4428e5dd7070Spatrick   }
4429e5dd7070Spatrick 
4430e5dd7070Spatrick   OS << "static AttributeCommonInfo::Kind getAttrKind(StringRef Name, ";
4431e5dd7070Spatrick   OS << "AttributeCommonInfo::Syntax Syntax) {\n";
4432e5dd7070Spatrick   OS << "  if (AttributeCommonInfo::AS_GNU == Syntax) {\n";
4433e5dd7070Spatrick   StringMatcher("Name", GNU, OS).Emit();
4434e5dd7070Spatrick   OS << "  } else if (AttributeCommonInfo::AS_Declspec == Syntax) {\n";
4435e5dd7070Spatrick   StringMatcher("Name", Declspec, OS).Emit();
4436e5dd7070Spatrick   OS << "  } else if (AttributeCommonInfo::AS_Microsoft == Syntax) {\n";
4437e5dd7070Spatrick   StringMatcher("Name", Microsoft, OS).Emit();
4438e5dd7070Spatrick   OS << "  } else if (AttributeCommonInfo::AS_CXX11 == Syntax) {\n";
4439e5dd7070Spatrick   StringMatcher("Name", CXX11, OS).Emit();
4440e5dd7070Spatrick   OS << "  } else if (AttributeCommonInfo::AS_C2x == Syntax) {\n";
4441e5dd7070Spatrick   StringMatcher("Name", C2x, OS).Emit();
4442e5dd7070Spatrick   OS << "  } else if (AttributeCommonInfo::AS_Keyword == Syntax || ";
4443e5dd7070Spatrick   OS << "AttributeCommonInfo::AS_ContextSensitiveKeyword == Syntax) {\n";
4444e5dd7070Spatrick   StringMatcher("Name", Keywords, OS).Emit();
4445e5dd7070Spatrick   OS << "  } else if (AttributeCommonInfo::AS_Pragma == Syntax) {\n";
4446e5dd7070Spatrick   StringMatcher("Name", Pragma, OS).Emit();
4447*12c85518Srobert   OS << "  } else if (AttributeCommonInfo::AS_HLSLSemantic == Syntax) {\n";
4448*12c85518Srobert   StringMatcher("Name", HLSLSemantic, OS).Emit();
4449e5dd7070Spatrick   OS << "  }\n";
4450e5dd7070Spatrick   OS << "  return AttributeCommonInfo::UnknownAttribute;\n"
4451e5dd7070Spatrick      << "}\n";
4452e5dd7070Spatrick }
4453e5dd7070Spatrick 
4454e5dd7070Spatrick // Emits the code to dump an attribute.
EmitClangAttrTextNodeDump(RecordKeeper & Records,raw_ostream & OS)4455e5dd7070Spatrick void EmitClangAttrTextNodeDump(RecordKeeper &Records, raw_ostream &OS) {
4456e5dd7070Spatrick   emitSourceFileHeader("Attribute text node dumper", OS);
4457e5dd7070Spatrick 
4458e5dd7070Spatrick   std::vector<Record*> Attrs = Records.getAllDerivedDefinitions("Attr"), Args;
4459e5dd7070Spatrick   for (const auto *Attr : Attrs) {
4460e5dd7070Spatrick     const Record &R = *Attr;
4461e5dd7070Spatrick     if (!R.getValueAsBit("ASTNode"))
4462e5dd7070Spatrick       continue;
4463e5dd7070Spatrick 
4464e5dd7070Spatrick     // If the attribute has a semantically-meaningful name (which is determined
4465e5dd7070Spatrick     // by whether there is a Spelling enumeration for it), then write out the
4466e5dd7070Spatrick     // spelling used for the attribute.
4467e5dd7070Spatrick 
4468e5dd7070Spatrick     std::string FunctionContent;
4469e5dd7070Spatrick     llvm::raw_string_ostream SS(FunctionContent);
4470e5dd7070Spatrick 
4471e5dd7070Spatrick     std::vector<FlattenedSpelling> Spellings = GetFlattenedSpellings(R);
4472e5dd7070Spatrick     if (Spellings.size() > 1 && !SpellingNamesAreCommon(Spellings))
4473e5dd7070Spatrick       SS << "    OS << \" \" << A->getSpelling();\n";
4474e5dd7070Spatrick 
4475e5dd7070Spatrick     Args = R.getValueAsListOfDefs("Args");
4476e5dd7070Spatrick     for (const auto *Arg : Args)
4477e5dd7070Spatrick       createArgument(*Arg, R.getName())->writeDump(SS);
4478e5dd7070Spatrick 
4479*12c85518Srobert     if (Attr->getValueAsBit("AcceptsExprPack"))
4480*12c85518Srobert       VariadicExprArgument("DelayedArgs", R.getName()).writeDump(OS);
4481*12c85518Srobert 
4482e5dd7070Spatrick     if (SS.tell()) {
4483e5dd7070Spatrick       OS << "  void Visit" << R.getName() << "Attr(const " << R.getName()
4484e5dd7070Spatrick          << "Attr *A) {\n";
4485e5dd7070Spatrick       if (!Args.empty())
4486e5dd7070Spatrick         OS << "    const auto *SA = cast<" << R.getName()
4487e5dd7070Spatrick            << "Attr>(A); (void)SA;\n";
4488e5dd7070Spatrick       OS << SS.str();
4489e5dd7070Spatrick       OS << "  }\n";
4490e5dd7070Spatrick     }
4491e5dd7070Spatrick   }
4492e5dd7070Spatrick }
4493e5dd7070Spatrick 
EmitClangAttrNodeTraverse(RecordKeeper & Records,raw_ostream & OS)4494e5dd7070Spatrick void EmitClangAttrNodeTraverse(RecordKeeper &Records, raw_ostream &OS) {
4495e5dd7070Spatrick   emitSourceFileHeader("Attribute text node traverser", OS);
4496e5dd7070Spatrick 
4497e5dd7070Spatrick   std::vector<Record *> Attrs = Records.getAllDerivedDefinitions("Attr"), Args;
4498e5dd7070Spatrick   for (const auto *Attr : Attrs) {
4499e5dd7070Spatrick     const Record &R = *Attr;
4500e5dd7070Spatrick     if (!R.getValueAsBit("ASTNode"))
4501e5dd7070Spatrick       continue;
4502e5dd7070Spatrick 
4503e5dd7070Spatrick     std::string FunctionContent;
4504e5dd7070Spatrick     llvm::raw_string_ostream SS(FunctionContent);
4505e5dd7070Spatrick 
4506e5dd7070Spatrick     Args = R.getValueAsListOfDefs("Args");
4507e5dd7070Spatrick     for (const auto *Arg : Args)
4508e5dd7070Spatrick       createArgument(*Arg, R.getName())->writeDumpChildren(SS);
4509*12c85518Srobert     if (Attr->getValueAsBit("AcceptsExprPack"))
4510*12c85518Srobert       VariadicExprArgument("DelayedArgs", R.getName()).writeDumpChildren(SS);
4511e5dd7070Spatrick     if (SS.tell()) {
4512e5dd7070Spatrick       OS << "  void Visit" << R.getName() << "Attr(const " << R.getName()
4513e5dd7070Spatrick          << "Attr *A) {\n";
4514e5dd7070Spatrick       if (!Args.empty())
4515e5dd7070Spatrick         OS << "    const auto *SA = cast<" << R.getName()
4516e5dd7070Spatrick            << "Attr>(A); (void)SA;\n";
4517e5dd7070Spatrick       OS << SS.str();
4518e5dd7070Spatrick       OS << "  }\n";
4519e5dd7070Spatrick     }
4520e5dd7070Spatrick   }
4521e5dd7070Spatrick }
4522e5dd7070Spatrick 
EmitClangAttrParserStringSwitches(RecordKeeper & Records,raw_ostream & OS)4523e5dd7070Spatrick void EmitClangAttrParserStringSwitches(RecordKeeper &Records,
4524e5dd7070Spatrick                                        raw_ostream &OS) {
4525e5dd7070Spatrick   emitSourceFileHeader("Parser-related llvm::StringSwitch cases", OS);
4526e5dd7070Spatrick   emitClangAttrArgContextList(Records, OS);
4527e5dd7070Spatrick   emitClangAttrIdentifierArgList(Records, OS);
4528e5dd7070Spatrick   emitClangAttrVariadicIdentifierArgList(Records, OS);
4529e5dd7070Spatrick   emitClangAttrThisIsaIdentifierArgList(Records, OS);
4530*12c85518Srobert   emitClangAttrAcceptsExprPack(Records, OS);
4531e5dd7070Spatrick   emitClangAttrTypeArgList(Records, OS);
4532e5dd7070Spatrick   emitClangAttrLateParsedList(Records, OS);
4533e5dd7070Spatrick }
4534e5dd7070Spatrick 
EmitClangAttrSubjectMatchRulesParserStringSwitches(RecordKeeper & Records,raw_ostream & OS)4535e5dd7070Spatrick void EmitClangAttrSubjectMatchRulesParserStringSwitches(RecordKeeper &Records,
4536e5dd7070Spatrick                                                         raw_ostream &OS) {
4537e5dd7070Spatrick   getPragmaAttributeSupport(Records).generateParsingHelpers(OS);
4538e5dd7070Spatrick }
4539e5dd7070Spatrick 
EmitClangAttrDocTable(RecordKeeper & Records,raw_ostream & OS)4540*12c85518Srobert void EmitClangAttrDocTable(RecordKeeper &Records, raw_ostream &OS) {
4541*12c85518Srobert   emitSourceFileHeader("Clang attribute documentation", OS);
4542*12c85518Srobert 
4543*12c85518Srobert   std::vector<Record *> Attrs = Records.getAllDerivedDefinitions("Attr");
4544*12c85518Srobert   for (const auto *A : Attrs) {
4545*12c85518Srobert     if (!A->getValueAsBit("ASTNode"))
4546*12c85518Srobert       continue;
4547*12c85518Srobert     std::vector<Record *> Docs = A->getValueAsListOfDefs("Documentation");
4548*12c85518Srobert     assert(!Docs.empty());
4549*12c85518Srobert     // Only look at the first documentation if there are several.
4550*12c85518Srobert     // (Currently there's only one such attr, revisit if this becomes common).
4551*12c85518Srobert     StringRef Text =
4552*12c85518Srobert         Docs.front()->getValueAsOptionalString("Content").value_or("");
4553*12c85518Srobert     OS << "\nstatic const char AttrDoc_" << A->getName() << "[] = "
4554*12c85518Srobert        << "R\"reST(" << Text.trim() << ")reST\";\n";
4555*12c85518Srobert   }
4556*12c85518Srobert }
4557*12c85518Srobert 
4558*12c85518Srobert enum class SpellingKind : size_t {
4559e5dd7070Spatrick   GNU,
4560e5dd7070Spatrick   CXX11,
4561e5dd7070Spatrick   C2x,
4562e5dd7070Spatrick   Declspec,
4563e5dd7070Spatrick   Microsoft,
4564e5dd7070Spatrick   Keyword,
4565e5dd7070Spatrick   Pragma,
4566*12c85518Srobert   HLSLSemantic,
4567*12c85518Srobert   NumSpellingKinds
4568e5dd7070Spatrick };
4569*12c85518Srobert static const size_t NumSpellingKinds = (size_t)SpellingKind::NumSpellingKinds;
4570e5dd7070Spatrick 
4571e5dd7070Spatrick class SpellingList {
4572e5dd7070Spatrick   std::vector<std::string> Spellings[NumSpellingKinds];
4573e5dd7070Spatrick 
4574e5dd7070Spatrick public:
operator [](SpellingKind K) const4575e5dd7070Spatrick   ArrayRef<std::string> operator[](SpellingKind K) const {
4576e5dd7070Spatrick     return Spellings[(size_t)K];
4577e5dd7070Spatrick   }
4578e5dd7070Spatrick 
add(const Record & Attr,FlattenedSpelling Spelling)4579e5dd7070Spatrick   void add(const Record &Attr, FlattenedSpelling Spelling) {
4580e5dd7070Spatrick     SpellingKind Kind = StringSwitch<SpellingKind>(Spelling.variety())
4581e5dd7070Spatrick                             .Case("GNU", SpellingKind::GNU)
4582e5dd7070Spatrick                             .Case("CXX11", SpellingKind::CXX11)
4583e5dd7070Spatrick                             .Case("C2x", SpellingKind::C2x)
4584e5dd7070Spatrick                             .Case("Declspec", SpellingKind::Declspec)
4585e5dd7070Spatrick                             .Case("Microsoft", SpellingKind::Microsoft)
4586e5dd7070Spatrick                             .Case("Keyword", SpellingKind::Keyword)
4587*12c85518Srobert                             .Case("Pragma", SpellingKind::Pragma)
4588*12c85518Srobert                             .Case("HLSLSemantic", SpellingKind::HLSLSemantic);
4589e5dd7070Spatrick     std::string Name;
4590e5dd7070Spatrick     if (!Spelling.nameSpace().empty()) {
4591e5dd7070Spatrick       switch (Kind) {
4592e5dd7070Spatrick       case SpellingKind::CXX11:
4593e5dd7070Spatrick       case SpellingKind::C2x:
4594e5dd7070Spatrick         Name = Spelling.nameSpace() + "::";
4595e5dd7070Spatrick         break;
4596e5dd7070Spatrick       case SpellingKind::Pragma:
4597e5dd7070Spatrick         Name = Spelling.nameSpace() + " ";
4598e5dd7070Spatrick         break;
4599e5dd7070Spatrick       default:
4600e5dd7070Spatrick         PrintFatalError(Attr.getLoc(), "Unexpected namespace in spelling");
4601e5dd7070Spatrick       }
4602e5dd7070Spatrick     }
4603e5dd7070Spatrick     Name += Spelling.name();
4604e5dd7070Spatrick 
4605e5dd7070Spatrick     Spellings[(size_t)Kind].push_back(Name);
4606e5dd7070Spatrick   }
4607e5dd7070Spatrick };
4608e5dd7070Spatrick 
4609e5dd7070Spatrick class DocumentationData {
4610e5dd7070Spatrick public:
4611e5dd7070Spatrick   const Record *Documentation;
4612e5dd7070Spatrick   const Record *Attribute;
4613e5dd7070Spatrick   std::string Heading;
4614e5dd7070Spatrick   SpellingList SupportedSpellings;
4615e5dd7070Spatrick 
DocumentationData(const Record & Documentation,const Record & Attribute,std::pair<std::string,SpellingList> HeadingAndSpellings)4616e5dd7070Spatrick   DocumentationData(const Record &Documentation, const Record &Attribute,
4617e5dd7070Spatrick                     std::pair<std::string, SpellingList> HeadingAndSpellings)
4618e5dd7070Spatrick       : Documentation(&Documentation), Attribute(&Attribute),
4619e5dd7070Spatrick         Heading(std::move(HeadingAndSpellings.first)),
4620e5dd7070Spatrick         SupportedSpellings(std::move(HeadingAndSpellings.second)) {}
4621e5dd7070Spatrick };
4622e5dd7070Spatrick 
WriteCategoryHeader(const Record * DocCategory,raw_ostream & OS)4623e5dd7070Spatrick static void WriteCategoryHeader(const Record *DocCategory,
4624e5dd7070Spatrick                                 raw_ostream &OS) {
4625e5dd7070Spatrick   const StringRef Name = DocCategory->getValueAsString("Name");
4626e5dd7070Spatrick   OS << Name << "\n" << std::string(Name.size(), '=') << "\n";
4627e5dd7070Spatrick 
4628e5dd7070Spatrick   // If there is content, print that as well.
4629e5dd7070Spatrick   const StringRef ContentStr = DocCategory->getValueAsString("Content");
4630e5dd7070Spatrick   // Trim leading and trailing newlines and spaces.
4631e5dd7070Spatrick   OS << ContentStr.trim();
4632e5dd7070Spatrick 
4633e5dd7070Spatrick   OS << "\n\n";
4634e5dd7070Spatrick }
4635e5dd7070Spatrick 
4636e5dd7070Spatrick static std::pair<std::string, SpellingList>
GetAttributeHeadingAndSpellings(const Record & Documentation,const Record & Attribute,StringRef Cat)4637e5dd7070Spatrick GetAttributeHeadingAndSpellings(const Record &Documentation,
4638*12c85518Srobert                                 const Record &Attribute,
4639*12c85518Srobert                                 StringRef Cat) {
4640e5dd7070Spatrick   // FIXME: there is no way to have a per-spelling category for the attribute
4641e5dd7070Spatrick   // documentation. This may not be a limiting factor since the spellings
4642e5dd7070Spatrick   // should generally be consistently applied across the category.
4643e5dd7070Spatrick 
4644e5dd7070Spatrick   std::vector<FlattenedSpelling> Spellings = GetFlattenedSpellings(Attribute);
4645e5dd7070Spatrick   if (Spellings.empty())
4646e5dd7070Spatrick     PrintFatalError(Attribute.getLoc(),
4647e5dd7070Spatrick                     "Attribute has no supported spellings; cannot be "
4648e5dd7070Spatrick                     "documented");
4649e5dd7070Spatrick 
4650e5dd7070Spatrick   // Determine the heading to be used for this attribute.
4651ec727ea7Spatrick   std::string Heading = std::string(Documentation.getValueAsString("Heading"));
4652e5dd7070Spatrick   if (Heading.empty()) {
4653e5dd7070Spatrick     // If there's only one spelling, we can simply use that.
4654e5dd7070Spatrick     if (Spellings.size() == 1)
4655e5dd7070Spatrick       Heading = Spellings.begin()->name();
4656e5dd7070Spatrick     else {
4657e5dd7070Spatrick       std::set<std::string> Uniques;
4658e5dd7070Spatrick       for (auto I = Spellings.begin(), E = Spellings.end();
4659*12c85518Srobert            I != E; ++I) {
4660ec727ea7Spatrick         std::string Spelling =
4661ec727ea7Spatrick             std::string(NormalizeNameForSpellingComparison(I->name()));
4662e5dd7070Spatrick         Uniques.insert(Spelling);
4663e5dd7070Spatrick       }
4664e5dd7070Spatrick       // If the semantic map has only one spelling, that is sufficient for our
4665e5dd7070Spatrick       // needs.
4666e5dd7070Spatrick       if (Uniques.size() == 1)
4667e5dd7070Spatrick         Heading = *Uniques.begin();
4668*12c85518Srobert       // If it's in the undocumented category, just construct a header by
4669*12c85518Srobert       // concatenating all the spellings. Might not be great, but better than
4670*12c85518Srobert       // nothing.
4671*12c85518Srobert       else if (Cat == "Undocumented")
4672*12c85518Srobert         Heading = llvm::join(Uniques.begin(), Uniques.end(), ", ");
4673e5dd7070Spatrick     }
4674e5dd7070Spatrick   }
4675e5dd7070Spatrick 
4676e5dd7070Spatrick   // If the heading is still empty, it is an error.
4677e5dd7070Spatrick   if (Heading.empty())
4678e5dd7070Spatrick     PrintFatalError(Attribute.getLoc(),
4679e5dd7070Spatrick                     "This attribute requires a heading to be specified");
4680e5dd7070Spatrick 
4681e5dd7070Spatrick   SpellingList SupportedSpellings;
4682e5dd7070Spatrick   for (const auto &I : Spellings)
4683e5dd7070Spatrick     SupportedSpellings.add(Attribute, I);
4684e5dd7070Spatrick 
4685e5dd7070Spatrick   return std::make_pair(std::move(Heading), std::move(SupportedSpellings));
4686e5dd7070Spatrick }
4687e5dd7070Spatrick 
WriteDocumentation(RecordKeeper & Records,const DocumentationData & Doc,raw_ostream & OS)4688e5dd7070Spatrick static void WriteDocumentation(RecordKeeper &Records,
4689e5dd7070Spatrick                                const DocumentationData &Doc, raw_ostream &OS) {
4690e5dd7070Spatrick   OS << Doc.Heading << "\n" << std::string(Doc.Heading.length(), '-') << "\n";
4691e5dd7070Spatrick 
4692e5dd7070Spatrick   // List what spelling syntaxes the attribute supports.
4693*12c85518Srobert   // Note: "#pragma clang attribute" is handled outside the spelling kinds loop
4694*12c85518Srobert   // so it must be last.
4695e5dd7070Spatrick   OS << ".. csv-table:: Supported Syntaxes\n";
4696e5dd7070Spatrick   OS << "   :header: \"GNU\", \"C++11\", \"C2x\", \"``__declspec``\",";
4697*12c85518Srobert   OS << " \"Keyword\", \"``#pragma``\", \"HLSL Semantic\", \"``#pragma clang ";
4698*12c85518Srobert   OS << "attribute``\"\n\n   \"";
4699e5dd7070Spatrick   for (size_t Kind = 0; Kind != NumSpellingKinds; ++Kind) {
4700e5dd7070Spatrick     SpellingKind K = (SpellingKind)Kind;
4701e5dd7070Spatrick     // TODO: List Microsoft (IDL-style attribute) spellings once we fully
4702e5dd7070Spatrick     // support them.
4703e5dd7070Spatrick     if (K == SpellingKind::Microsoft)
4704e5dd7070Spatrick       continue;
4705e5dd7070Spatrick 
4706e5dd7070Spatrick     bool PrintedAny = false;
4707e5dd7070Spatrick     for (StringRef Spelling : Doc.SupportedSpellings[K]) {
4708e5dd7070Spatrick       if (PrintedAny)
4709e5dd7070Spatrick         OS << " |br| ";
4710e5dd7070Spatrick       OS << "``" << Spelling << "``";
4711e5dd7070Spatrick       PrintedAny = true;
4712e5dd7070Spatrick     }
4713e5dd7070Spatrick 
4714e5dd7070Spatrick     OS << "\",\"";
4715e5dd7070Spatrick   }
4716e5dd7070Spatrick 
4717e5dd7070Spatrick   if (getPragmaAttributeSupport(Records).isAttributedSupported(
4718e5dd7070Spatrick           *Doc.Attribute))
4719e5dd7070Spatrick     OS << "Yes";
4720e5dd7070Spatrick   OS << "\"\n\n";
4721e5dd7070Spatrick 
4722e5dd7070Spatrick   // If the attribute is deprecated, print a message about it, and possibly
4723e5dd7070Spatrick   // provide a replacement attribute.
4724e5dd7070Spatrick   if (!Doc.Documentation->isValueUnset("Deprecated")) {
4725e5dd7070Spatrick     OS << "This attribute has been deprecated, and may be removed in a future "
4726e5dd7070Spatrick        << "version of Clang.";
4727e5dd7070Spatrick     const Record &Deprecated = *Doc.Documentation->getValueAsDef("Deprecated");
4728e5dd7070Spatrick     const StringRef Replacement = Deprecated.getValueAsString("Replacement");
4729e5dd7070Spatrick     if (!Replacement.empty())
4730e5dd7070Spatrick       OS << "  This attribute has been superseded by ``" << Replacement
4731e5dd7070Spatrick          << "``.";
4732e5dd7070Spatrick     OS << "\n\n";
4733e5dd7070Spatrick   }
4734e5dd7070Spatrick 
4735e5dd7070Spatrick   const StringRef ContentStr = Doc.Documentation->getValueAsString("Content");
4736e5dd7070Spatrick   // Trim leading and trailing newlines and spaces.
4737e5dd7070Spatrick   OS << ContentStr.trim();
4738e5dd7070Spatrick 
4739e5dd7070Spatrick   OS << "\n\n\n";
4740e5dd7070Spatrick }
4741e5dd7070Spatrick 
EmitClangAttrDocs(RecordKeeper & Records,raw_ostream & OS)4742e5dd7070Spatrick void EmitClangAttrDocs(RecordKeeper &Records, raw_ostream &OS) {
4743e5dd7070Spatrick   // Get the documentation introduction paragraph.
4744e5dd7070Spatrick   const Record *Documentation = Records.getDef("GlobalDocumentation");
4745e5dd7070Spatrick   if (!Documentation) {
4746e5dd7070Spatrick     PrintFatalError("The Documentation top-level definition is missing, "
4747e5dd7070Spatrick                     "no documentation will be generated.");
4748e5dd7070Spatrick     return;
4749e5dd7070Spatrick   }
4750e5dd7070Spatrick 
4751e5dd7070Spatrick   OS << Documentation->getValueAsString("Intro") << "\n";
4752e5dd7070Spatrick 
4753e5dd7070Spatrick   // Gather the Documentation lists from each of the attributes, based on the
4754e5dd7070Spatrick   // category provided.
4755e5dd7070Spatrick   std::vector<Record *> Attrs = Records.getAllDerivedDefinitions("Attr");
4756*12c85518Srobert   struct CategoryLess {
4757*12c85518Srobert     bool operator()(const Record *L, const Record *R) const {
4758*12c85518Srobert       return L->getValueAsString("Name") < R->getValueAsString("Name");
4759*12c85518Srobert     }
4760*12c85518Srobert   };
4761*12c85518Srobert   std::map<const Record *, std::vector<DocumentationData>, CategoryLess>
4762*12c85518Srobert       SplitDocs;
4763e5dd7070Spatrick   for (const auto *A : Attrs) {
4764e5dd7070Spatrick     const Record &Attr = *A;
4765e5dd7070Spatrick     std::vector<Record *> Docs = Attr.getValueAsListOfDefs("Documentation");
4766e5dd7070Spatrick     for (const auto *D : Docs) {
4767e5dd7070Spatrick       const Record &Doc = *D;
4768e5dd7070Spatrick       const Record *Category = Doc.getValueAsDef("Category");
4769*12c85518Srobert       // If the category is "InternalOnly", then there cannot be any other
4770*12c85518Srobert       // documentation categories (otherwise, the attribute would be
4771*12c85518Srobert       // emitted into the docs).
4772e5dd7070Spatrick       const StringRef Cat = Category->getValueAsString("Name");
4773*12c85518Srobert       bool InternalOnly = Cat == "InternalOnly";
4774*12c85518Srobert       if (InternalOnly && Docs.size() > 1)
4775e5dd7070Spatrick         PrintFatalError(Doc.getLoc(),
4776*12c85518Srobert                         "Attribute is \"InternalOnly\", but has multiple "
4777e5dd7070Spatrick                         "documentation categories");
4778e5dd7070Spatrick 
4779*12c85518Srobert       if (!InternalOnly)
4780e5dd7070Spatrick         SplitDocs[Category].push_back(DocumentationData(
4781*12c85518Srobert             Doc, Attr, GetAttributeHeadingAndSpellings(Doc, Attr, Cat)));
4782e5dd7070Spatrick     }
4783e5dd7070Spatrick   }
4784e5dd7070Spatrick 
4785e5dd7070Spatrick   // Having split the attributes out based on what documentation goes where,
4786e5dd7070Spatrick   // we can begin to generate sections of documentation.
4787e5dd7070Spatrick   for (auto &I : SplitDocs) {
4788e5dd7070Spatrick     WriteCategoryHeader(I.first, OS);
4789e5dd7070Spatrick 
4790e5dd7070Spatrick     llvm::sort(I.second,
4791e5dd7070Spatrick                [](const DocumentationData &D1, const DocumentationData &D2) {
4792e5dd7070Spatrick                  return D1.Heading < D2.Heading;
4793e5dd7070Spatrick                });
4794e5dd7070Spatrick 
4795e5dd7070Spatrick     // Walk over each of the attributes in the category and write out their
4796e5dd7070Spatrick     // documentation.
4797e5dd7070Spatrick     for (const auto &Doc : I.second)
4798e5dd7070Spatrick       WriteDocumentation(Records, Doc, OS);
4799e5dd7070Spatrick   }
4800e5dd7070Spatrick }
4801e5dd7070Spatrick 
EmitTestPragmaAttributeSupportedAttributes(RecordKeeper & Records,raw_ostream & OS)4802e5dd7070Spatrick void EmitTestPragmaAttributeSupportedAttributes(RecordKeeper &Records,
4803e5dd7070Spatrick                                                 raw_ostream &OS) {
4804e5dd7070Spatrick   PragmaClangAttributeSupport Support = getPragmaAttributeSupport(Records);
4805e5dd7070Spatrick   ParsedAttrMap Attrs = getParsedAttrList(Records);
4806e5dd7070Spatrick   OS << "#pragma clang attribute supports the following attributes:\n";
4807e5dd7070Spatrick   for (const auto &I : Attrs) {
4808e5dd7070Spatrick     if (!Support.isAttributedSupported(*I.second))
4809e5dd7070Spatrick       continue;
4810e5dd7070Spatrick     OS << I.first;
4811e5dd7070Spatrick     if (I.second->isValueUnset("Subjects")) {
4812e5dd7070Spatrick       OS << " ()\n";
4813e5dd7070Spatrick       continue;
4814e5dd7070Spatrick     }
4815e5dd7070Spatrick     const Record *SubjectObj = I.second->getValueAsDef("Subjects");
4816e5dd7070Spatrick     std::vector<Record *> Subjects =
4817e5dd7070Spatrick         SubjectObj->getValueAsListOfDefs("Subjects");
4818e5dd7070Spatrick     OS << " (";
4819a9ac8606Spatrick     bool PrintComma = false;
4820e5dd7070Spatrick     for (const auto &Subject : llvm::enumerate(Subjects)) {
4821a9ac8606Spatrick       if (!isSupportedPragmaClangAttributeSubject(*Subject.value()))
4822a9ac8606Spatrick         continue;
4823a9ac8606Spatrick       if (PrintComma)
4824e5dd7070Spatrick         OS << ", ";
4825a9ac8606Spatrick       PrintComma = true;
4826e5dd7070Spatrick       PragmaClangAttributeSupport::RuleOrAggregateRuleSet &RuleSet =
4827e5dd7070Spatrick           Support.SubjectsToRules.find(Subject.value())->getSecond();
4828e5dd7070Spatrick       if (RuleSet.isRule()) {
4829e5dd7070Spatrick         OS << RuleSet.getRule().getEnumValueName();
4830e5dd7070Spatrick         continue;
4831e5dd7070Spatrick       }
4832e5dd7070Spatrick       OS << "(";
4833e5dd7070Spatrick       for (const auto &Rule : llvm::enumerate(RuleSet.getAggregateRuleSet())) {
4834e5dd7070Spatrick         if (Rule.index())
4835e5dd7070Spatrick           OS << ", ";
4836e5dd7070Spatrick         OS << Rule.value().getEnumValueName();
4837e5dd7070Spatrick       }
4838e5dd7070Spatrick       OS << ")";
4839e5dd7070Spatrick     }
4840e5dd7070Spatrick     OS << ")\n";
4841e5dd7070Spatrick   }
4842e5dd7070Spatrick   OS << "End of supported attributes.\n";
4843e5dd7070Spatrick }
4844e5dd7070Spatrick 
4845e5dd7070Spatrick } // end namespace clang
4846