specialize shader

include/renderer_mtl/mtl_blit_pipeline_cache.hpp

@@ -0,0 +1,72 @@
+#pragma once
+
+#include "pica_to_mtl.hpp"
+
+using namespace PICA;
+
+namespace Metal {
+
+struct BlitPipelineHash {
+    // Formats
+    ColorFmt colorFmt;
+    DepthFmt depthFmt;
+};
+
+// This pipeline only caches the pipeline with all of its color and depth attachment variations
+class BlitPipelineCache {
+public:
+    BlitPipelineCache() = default;
+
+    ~BlitPipelineCache() {
+        clear();
+        vertexFunction->release();
+        fragmentFunction->release();
+    }
+
+    void set(MTL::Device* dev, MTL::Function* vert, MTL::Function* frag) {
+        device = dev;
+        vertexFunction = vert;
+        fragmentFunction = frag;
+    }
+
+    MTL::RenderPipelineState* get(BlitPipelineHash hash) {
+        u8 intHash = ((u8)hash.colorFmt << 3) | (u8)hash.depthFmt;
+        auto& pipeline = pipelineCache[intHash];
+        if (!pipeline) {
+            MTL::RenderPipelineDescriptor* desc = MTL::RenderPipelineDescriptor::alloc()->init();
+            desc->setVertexFunction(vertexFunction);
+            desc->setFragmentFunction(fragmentFunction);
+
+            auto colorAttachment = desc->colorAttachments()->object(0);
+            colorAttachment->setPixelFormat(toMTLPixelFormatColor(hash.colorFmt));
+
+            desc->setDepthAttachmentPixelFormat(toMTLPixelFormatDepth(hash.depthFmt));
+
+           	NS::Error* error = nullptr;
+           	pipeline = device->newRenderPipelineState(desc, &error);
+           	if (error) {
+          		Helpers::panic("Error creating blit pipeline state: %s", error->description()->cString(NS::ASCIIStringEncoding));
+           	}
+
+            desc->release();
+        }
+
+        return pipeline;
+    }
+
+    void clear() {
+        for (auto& pair : pipelineCache) {
+            pair.second->release();
+        }
+        pipelineCache.clear();
+    }
+
+private:
+    std::unordered_map<u8, MTL::RenderPipelineState*> pipelineCache;
+
+    MTL::Device* device;
+    MTL::Function* vertexFunction;
+    MTL::Function* fragmentFunction;
+};
+
+} // namespace Metal