MTLCommandBuffer extension

Author: yukiny0811

EasyMetalShader.xcworkspace/xcshareddata/swiftpm/Package.resolved

@@ -53,3 +53,27 @@ public class EMMetalDispatch {
         commandBuffer.waitUntilCompleted()
     }
 }
+
+extension MTLCommandBuffer {
+    
+    public func compute(_ process: @escaping (MTLComputeCommandEncoder) -> ()) {
+        let encoder = self.makeComputeCommandEncoder()!
+        process(encoder)
+        encoder.endEncoding()
+    }
+    
+    public func render(renderTargetTexture: MTLTexture, needsClear: Bool, process: @escaping (MTLRenderCommandEncoder) -> ()) {
+        let descriptor = MTLRenderPassDescriptor()
+        descriptor.colorAttachments[0].texture = renderTargetTexture
+        if needsClear {
+            descriptor.colorAttachments[0].loadAction = .clear
+        } else {
+            descriptor.colorAttachments[0].loadAction = .load
+        }
+        descriptor.colorAttachments[0].storeAction = .store
+        descriptor.colorAttachments[0].clearColor = .init(red: 0, green: 0, blue: 0, alpha: 0)
+        let encoder = self.makeRenderCommandEncoder(descriptor: descriptor)!
+        process(encoder)
+        encoder.endEncoding()
+    }
+}