Add userAttribs to entryPoints in reflection JSON (#6366)

Author: davidar

source/slang/slang-reflection-json.cpp

@@ -328,6 +328,22 @@ static void emitUserAttributes(PrettyWriter& writer, slang::VariableReflection*
         writer << "]";
     }
 }
+static void emitUserAttributes(PrettyWriter& writer, slang::FunctionReflection* func)
+{
+    auto attribCount = func->getUserAttributeCount();
+    if (attribCount)
+    {
+        writer << ",\n\"userAttribs\": [";
+        for (unsigned int i = 0; i < attribCount; i++)
+        {
+            if (i > 0)
+                writer << ",\n";
+            auto attrib = func->getUserAttributeByIndex(i);
+            emitUserAttributeJSON(writer, attrib);
+        }
+        writer << "]";
+    }
+}
 
 static void emitReflectionVarLayoutJSON(PrettyWriter& writer, slang::VariableLayoutReflection* var)
 {
@@ -1091,6 +1107,8 @@ static void emitReflectionEntryPointJSON(
         writer << "\n]";
     }
 
+    emitUserAttributes(writer, entryPoint->getFunction());
+
     writer.dedent();
     writer << "\n}";
 }

tests/reflection/attribute.slang

@@ -5,6 +5,7 @@
 //TEST:REFLECTION:-stage compute -entry main -target hlsl -no-codegen
 
 [__AttributeUsage(_AttributeTargets.Struct)]
+[__AttributeUsage(_AttributeTargets.Function)]
 struct MyStructAttribute
 {
     int iParam;
@@ -58,6 +59,7 @@ D param3;
 [StructVarParam(1)] int globalInt2;
 
 
+[MyStruct(2, 3.0)]
 [numthreads(1, 1, 1)]
 void main(
     uint3 dispatchThreadID : SV_DispatchThreadID,

tests/reflection/attribute.slang.expected

@@ -310,7 +310,15 @@ standard output = {
                     }
                 }
             ],
-            "threadGroupSize": [1, 1, 1]
+            "threadGroupSize": [1, 1, 1],
+            "userAttribs": [{
+                "name": "MyStruct",
+                "arguments": [
+                    2,
+                    3.000000
+                ]
+            }
+            ]
         }
     ]
 }