mesh shader + push initial

Author: Try

Diff for: Engine/gapi/spirv/meshconverter.cpp

@@ -860,9 +860,14 @@ void MeshConverter::injectCountingPass(const uint32_t idMainFunc) {
 
   // Writeout meshlet descriptor
   {
+    const uint32_t ptrWorkGroupID = code.fetchAddBound();
+    fn.insert(spv::OpAccessChain, {_ptr_Input_uint, ptrWorkGroupID, idWorkGroupID, const1});
+    const uint32_t workIdX = code.fetchAddBound();
+    fn.insert(spv::OpLoad, {uint_t, workIdX, ptrWorkGroupID});
+
     const uint32_t ptrHeap0 = code.fetchAddBound();
     fn.insert(spv::OpAccessChain, {_ptr_Uniform_uint, ptrHeap0, vEngine1, const3, meshDest});
-    fn.insert(spv::OpStore, {ptrHeap0, const0}); // TODO: writeout self-id
+    fn.insert(spv::OpStore, {ptrHeap0, workIdX});
 
     const uint32_t dest1 = code.fetchAddBound();
     fn.insert(spv::OpIAdd, {uint_t, dest1, meshDest, const1});

Diff for: Engine/gapi/vulkan/vcommandbuffer.cpp

@@ -1045,6 +1045,15 @@ void VMeshCommandBuffer::setPipeline(AbstractGraphicsApi::Pipeline& p) {
   ms.bindVS(impl,    px.pipelineLayout);
   }
 
+void VMeshCommandBuffer::setBytes(AbstractGraphicsApi::Pipeline& p, const void* data, size_t size) {
+  VCommandBuffer::setBytes(p,data,size);
+
+  VPipeline& px = reinterpret_cast<VPipeline&>(p);
+  if(px.meshPipeline()==VK_NULL_HANDLE)
+    return;
+  vkCmdPushConstants(impl, px.meshPipelineLayout(), VK_PIPELINE_BIND_POINT_COMPUTE, 0, uint32_t(size), data);
+  }
+
 void VMeshCommandBuffer::setUniforms(AbstractGraphicsApi::Pipeline& p, AbstractGraphicsApi::Desc& u) {
   VCommandBuffer::setUniforms(p,u);
 

Diff for: Engine/gapi/vulkan/vcommandbuffer.h

@@ -137,6 +137,7 @@ class VMeshCommandBuffer:public VCommandBuffer {
     using VCommandBuffer::VCommandBuffer;
 
     void setPipeline(AbstractGraphicsApi::Pipeline& p) override;
+    void setBytes   (AbstractGraphicsApi::Pipeline& p, const void* data, size_t size) override;
     void setUniforms(AbstractGraphicsApi::Pipeline& p, AbstractGraphicsApi::Desc &u) override;
 
     void dispatchMesh(size_t firstInstance, size_t instanceCount) override;