feat: begin shader load logic

Signed-off-by: Michael Pollind <mpollind@gmail.com>

Author: pollend

Source/Metal/CommandBufferMTL.mm

@@ -39,20 +39,59 @@
     PipelineMTL& pipelineImpl = (PipelineMTL&)pipeline;
     m_CurrentPipeline = &pipelineImpl;
 
-    switch(m_CurrentPipeline->m_pipelineType) {
-        case nri::PipelineMTL::Compute:
-            m_ComputeEncoder = [m_Handle computeCommandEncoderWithDescriptor: NULL];
-            break;
-        default:
-            break;
-    }
+//    if(m_CurrentPipeline->m_pipelineType == nri::PipelineMTL::Compute) {
+ //       m_ComputeEncoder = [m_Handle computeCommandEncoderWithDescriptor: NULL];
+  //  }
+
+}
+void CommandBufferMTL::SetPipelineLayout(const PipelineLayout& pipelineLayout) {
+
+}
+void CommandBufferMTL::SetDescriptorSet(uint32_t setIndexInPipelineLayout, const DescriptorSet& descriptorSet, const uint32_t* dynamicConstantBufferOffsets) {
+    
+}
+void CommandBufferMTL::SetConstants(uint32_t pushConstantIndex, const void* data, uint32_t size) {
+    //if (pDesc->mUsedStages & SHADER_STAGE_VERT)
+    //{
+    //    [m_RendererEncoder setVertexBytes:data length:size atIndex:pushConstantIndex];
+    //}
+
+    //if (pDesc->mUsedStages & SHADER_STAGE_FRAG)
+    //{
+    //    [m_RendererEncoder  setFragmentBytes:data length:size atIndex:pushConstantIndex];
+    //}
+
+    //if (pDesc->mUsedStages & SHADER_STAGE_COMP)
+    //{
+    //    [m_RendererEncoder setBytes:data length:size atIndex:pushConstantIndex];
+    //}
 
 }
-void CommandBufferMTL::SetPipelineLayout(const PipelineLayout& pipelineLayout) {}
-void CommandBufferMTL::SetDescriptorSet(uint32_t setIndexInPipelineLayout, const DescriptorSet& descriptorSet, const uint32_t* dynamicConstantBufferOffsets) {}
-void CommandBufferMTL::SetConstants(uint32_t pushConstantIndex, const void* data, uint32_t size) {}
 void CommandBufferMTL::SetDescriptorPool(const DescriptorPool& descriptorPool) {}
-void CommandBufferMTL::Barrier(const BarrierGroupDesc& barrierGroupDesc) {}
+void CommandBufferMTL::Barrier(const BarrierGroupDesc& barrierGroupDesc) {
+    //if (pCmd->pQueue->mBarrierFlags & BARRIER_FLAG_BUFFERS)
+    {
+        [m_RendererEncoder memoryBarrierWithScope:MTLBarrierScopeBuffers
+                                         afterStages:MTLRenderStageFragment
+                                        beforeStages:MTLRenderStageVertex];
+    }
+
+    //if (pCmd->pQueue->mBarrierFlags & BARRIER_FLAG_TEXTURES)
+    {
+        [m_RendererEncoder memoryBarrierWithScope:MTLBarrierScopeTextures
+                                         afterStages:MTLRenderStageFragment
+                                        beforeStages:MTLRenderStageVertex];
+    }
+
+    //if (pCmd->pQueue->mBarrierFlags & BARRIER_FLAG_RENDERTARGETS)
+    {
+        [m_RendererEncoder memoryBarrierWithScope:MTLBarrierScopeRenderTargets
+                                         afterStages:MTLRenderStageFragment
+                                        beforeStages:MTLRenderStageVertex];
+    }
+
+    
+}
 void CommandBufferMTL::BeginRendering(const AttachmentsDesc& attachmentsDesc) {
     m_RendererEncoder = [m_Handle renderCommandEncoderWithDescriptor: NULL];
 }
@@ -61,7 +100,11 @@
     m_RendererEncoder = nil;
     m_ComputeEncoder = nil;
 }
-void CommandBufferMTL::SetViewports(const Viewport* viewports, uint32_t viewportNum) {}
+void CommandBufferMTL::SetViewports(const Viewport* viewports, uint32_t viewportNum) {
+    MTLViewport* mtlViewports = StackAlloc(MTLViewport, viewportNum);
+
+  //  [m_RendererEncoder setViewports:<#(const MTLViewport * _Nonnull)#> count:<#(NSUInteger)#>
+}
 void CommandBufferMTL::SetScissors(const Rect* rects, uint32_t rectNum) {
     NSCAssert(m_RendererEncoder, @"encoder set");
     MTLScissorRect rect;

Source/Metal/CommandQueueMTL.h

@@ -7,6 +7,13 @@ namespace nri {
 
 struct DeviceMTL;
 
+NriBits(QueueBarrierBits, uint8_t,
+    NONE = 0,
+    BARRIER_FLAG_BUFFERS = NriBit(0),
+    BARRIER_FLAG_TEXTURES = NriBit(1),
+    BARRIER_FLAG_RENDERTARGETS = NriBit(2),
+    BARRIER_FLAG_FENCE = NriBit(3));
+
 struct CommandQueueMTL {
 
     inline CommandQueueMTL(DeviceMTL& device)
@@ -35,11 +42,14 @@ struct CommandQueueMTL {
 
     void SetDebugName(const char* name);
     Result Create(CommandQueueType type);
+    QueueBarrierBits m_BarrierBits = QueueBarrierBits::NONE;
+
 private:
     DeviceMTL& m_Device;
     CommandQueueType m_Type = CommandQueueType(-1);
     id<MTLCommandQueue> m_Handle;
     Lock m_Lock;
+
 };
 
 }

Source/Metal/PipelineMTL.h

@@ -7,12 +7,13 @@ namespace nri {
 struct DeviceMTL;
 struct PipelineLayoutMTL;
 
+NriEnum(PipelineType, uint8_t,
+    Compute,
+    Graphics,
+    Raytracing
+);
+
 struct PipelineMTL {
-    enum PipelineType {
-        Compute,
-        Graphics,
-        Raytracing
-    };
 
     inline PipelineMTL(DeviceMTL& device)
         : m_Device(device) {
@@ -26,6 +27,7 @@ struct PipelineMTL {
     PipelineType m_pipelineType;
     MTLPrimitiveTopologyClass m_topologyClass;
     MTLPrimitiveType m_primitiveType;
+    StageBits m_usedBits;
 private:
     DeviceMTL& m_Device;
 };

Source/Metal/PipelineMTL.mm

@@ -43,6 +43,20 @@
     }
 
     for (uint32_t i = 0; i < graphicsPipelineDesc.shaderNum; i++) {
+        const ShaderDesc& shader = graphicsPipelineDesc.shaders[i];
+        m_usedBits |= shader.stage;
+
+        dispatch_data_t byteCode =
+            dispatch_data_create(shader.bytecode, shader.size, nil,
+                                 DISPATCH_DATA_DESTRUCTOR_DEFAULT);
+        id<MTLLibrary> lib = [m_Device newLibraryWithData:byteCode error:nil];
+
+        // Create a MTLFunction from the loaded MTLLibrary.
+        NSString *entryPointNStr = [lib functionNames][0];
+        if (shader.entryPointName) {
+          entryPointNStr =
+              [[NSString alloc] initWithUTF8String:shader.entryPointName];
+        }
     }
     // Depth-stencil
     const DepthAttachmentDesc& da = graphicsPipelineDesc.outputMerger.depth;