More clang reflection data.

Author: MaxwellGengYF

include/luisa/clangcxx/build_arguments.h

@@ -2,13 +2,17 @@
 #include <luisa/core/stl/string.h>
 #include <luisa/core/stl/vector.h>
 #include <luisa/ast/type.h>
+#include <luisa/ast/usage.h>
 namespace luisa::clangcxx {
 struct BuildArgument {
-    compute::Type const *type;
+    compute::Type const *type{};
     string var_name;
+    uint resource_var_id{~0u};
+    compute::Usage var_usage{compute::Usage::NONE};
 };
 struct ShaderReflection {
     uint dimension;
+    uint3 block_size;
     luisa::vector<BuildArgument> kernel_args;
 };
 }// namespace luisa::clangcxx

src/clangcxx/src/llvm/ASTConsumer.cpp

@@ -1453,8 +1453,10 @@ auto FunctionBuilderBuilder::build(const clang::FunctionDecl *S, bool allowKerne
                     //     LUISA_WARNING("Set kernel arg. {}", luisa::string_view{});
                     // }
                     if (auto lcType = db->FindOrAddType(Ty, param->getBeginLoc())) {
+                        BuildArgument *build_arg = nullptr;
                         if (is_kernel) {
-                            db->refl.kernel_args.emplace_back(lcType, luisa::string{param->getName().data(), param->getName().size()});
+                            db->refl.kernel_args.emplace_back(lcType, luisa::string{param->getName().data(), param->getName().size()}, ~0u);
+                            build_arg = &db->refl.kernel_args.back();
                         }
                         const luisa::compute::RefExpr *local = nullptr;
                         switch (lcType->tag()) {
@@ -1485,6 +1487,9 @@ auto FunctionBuilderBuilder::build(const clang::FunctionDecl *S, bool allowKerne
                                 local = LC_ArgOrRef(Ty, builder, lcType);
                             } break;
                         }
+                        if (build_arg && local) {
+                            build_arg->resource_var_id = local->variable().uid();
+                        }
                         stack.SetLocal(param, local);
                         return local;
                     } else {
@@ -1692,6 +1697,10 @@ ASTConsumer::~ASTConsumer() {
     } else {
         if (kernel_arg_reflect) {
             *kernel_arg_reflect = std::move(db.refl);
+            kernel_arg_reflect->block_size = db.kernel_builder->block_size();
+            for (auto &i : kernel_arg_reflect->kernel_args) {
+                i.var_usage = db.kernel_builder->variable_usage(i.resource_var_id);
+            }
         }
         device->impl()->create_shader(option, luisa::compute::Function{db.kernel_builder.get()});
     }