adding aditional data to attributes

Author: joaosaffran-zz

clang/include/clang/Basic/Attr.td

@@ -4314,6 +4314,10 @@ def HLSLRootSignature: Attr {
                               ErrorDiag, "'function'">;
   let LangOpts = [HLSL];
   let Documentation = [HLSLLoopHintDocs];
+  let AdditionalMembers = [{
+    public:
+      llvm::hlsl::HLSLRootElement Elements;
+    }];
 }
 
 def CapturedRecord : InheritableAttr {

clang/include/clang/Sema/HLSLRootSignature.h

@@ -42,12 +42,14 @@ struct RSTknInfo {
 class RootSignaturParser {
 
 public:
-  RootSignaturParser(StringRef Signature) : Signature(Signature) {}
+  RootSignaturParser(HLSLRootSignatureAttr *Attr, StringRef Signature)
+      : Signature(Signature), Attr(Attr) {}
 
   void ParseRootDefinition();
 
 private:
   StringRef Signature;
+  HLSLRootSignatureAttr *Attr;
 
   RSTknInfo CurTok;
   std::string IdentifierStr;

clang/lib/Sema/SemaHLSL.cpp

@@ -712,9 +712,14 @@ void SemaHLSL::handleHLSLRootSignature(Decl *D, const ParsedAttr &AL) {
   if (!SemaRef.checkStringLiteralArgumentAttr(AL, 0, Signature))
     return;
 
-  RootSignaturParser Parser(Signature);
+  HLSLRootSignatureAttr *NewAttr = ::new (getASTContext())
+      HLSLRootSignatureAttr(getASTContext(), AL, Signature);
 
+  RootSignaturParser Parser(NewAttr, Signature);
   Parser.ParseRootDefinition();
+
+  if (NewAttr)
+    D->addAttr(NewAttr);
 }
 
 void SemaHLSL::handleWaveSizeAttr(Decl *D, const ParsedAttr &AL) {

llvm/include/llvm/Frontend/HLSL/HLSLResource.h

@@ -44,6 +44,19 @@ class FrontendResource {
   uint32_t getSpace();
   MDNode *getMetadata() { return Entry; }
 };
+
+class HLSLRootElement {
+public:
+  HLSLRootElement() {}
+  StringRef getName();
+
+  ~HLSLRootElement() {}
+};
+
+class HLSLRootFlag : public HLSLRootElement {
+public:
+  std::size_t Flag = 0x111;
+};
 } // namespace hlsl
 } // namespace llvm
 

llvm/lib/Frontend/HLSL/HLSLResource.cpp

@@ -67,3 +67,5 @@ FrontendResource::FrontendResource(GlobalVariable *GV, ResourceKind RK,
             ConstantAsMetadata::get(B.getInt32(ResIndex)),
             ConstantAsMetadata::get(B.getInt32(Space))});
 }
+
+llvm::StringRef llvm::hlsl::HLSLRootElement::getName() { return StringRef(""); }