Merge pull request #3 from yukiny0811/feat/sample-usage

add sample texture usage

Author: yukiny0811

Examples/ExampleMacOS/ExampleMacOS/Logic/MyRenderer.swift

@@ -26,7 +26,7 @@ class MyRenderer: ShaderRenderer {
     override func draw(view: MTKView, drawable: CAMetalDrawable) {
         let dispatch = EMMetalDispatch()
         dispatch.compute { [self] encoder in
-            compute.tex = EMMetalTexture(texture: drawable.texture)
+            compute.tex = drawable.texture.emTexture
             compute.intensity = abs(sin(Float(Date().timeIntervalSince(date)))) * 100
             compute.dispatch(encoder, textureSizeReference: drawable.texture)
         }

Sources/EasyMetalShader/Models/EMMetalTexture.swift

@@ -14,12 +14,12 @@ public class EMMetalTexture: NSObject {
     public var texture: MTLTexture?
     public var usage: EMMetalTextureUsage
 
-    public init(texture: MTLTexture?, usage: EMMetalTextureUsage = .read_write) {
+    public init(texture: MTLTexture?, usage: EMMetalTextureUsage) {
         self.texture = texture
         self.usage = usage
     }
 
-    public static func create(width: Int, height: Int, pixelFormat: MTLPixelFormat, label: String?, usage: EMMetalTextureUsage) -> EMMetalTexture {
+    public static func create(width: Int, height: Int, pixelFormat: MTLPixelFormat, label: String?) -> MTLTexture {
         let descriptor = MTLTextureDescriptor()
         descriptor.pixelFormat = pixelFormat
         descriptor.textureType = .type2D
@@ -29,10 +29,10 @@ public class EMMetalTexture: NSObject {
         descriptor.resourceOptions = .storageModePrivate
         let texture = ShaderCore.device.makeTexture(descriptor: descriptor)!
         texture.label = label
-        return EMMetalTexture(texture: texture, usage: usage)
+        return texture
     }
 
-    public static func createManaged(width: Int, height: Int, pixelFormat: MTLPixelFormat, label: String?, usage: EMMetalTextureUsage) -> EMMetalTexture {
+    public static func createManaged(width: Int, height: Int, pixelFormat: MTLPixelFormat, label: String?) -> MTLTexture {
         let descriptor = MTLTextureDescriptor()
         descriptor.pixelFormat = pixelFormat
         descriptor.textureType = .type2D
@@ -46,6 +46,14 @@ public class EMMetalTexture: NSObject {
         #endif
         let texture = ShaderCore.device.makeTexture(descriptor: descriptor)!
         texture.label = label
-        return EMMetalTexture(texture: texture, usage: usage)
+        return texture
+    }
+}
+
+public extension MTLTexture {
+    
+    // dont use this before shader compilation!
+    var emTexture: EMMetalTexture {
+        .init(texture: self, usage: .read)
     }
 }

Sources/EasyMetalShader/Models/EMMetalTextureUsage.swift

@@ -9,4 +9,5 @@ public enum EMMetalTextureUsage {
     case read
     case write
     case read_write
+    case sample
 }

Sources/EasyMetalShader/Pipelines/EMMetalComputeFunction.swift

@@ -67,6 +67,8 @@ open class EMMetalComputeFunction: NSObject, EMMetalFunction {
                     functionImpl += "texture2d<float, access::write> \(key) [[texture(\(i+1))]]"
                 case .read_write:
                     functionImpl += "texture2d<float, access::read_write> \(key) [[texture(\(i+1))]]"
+                case .sample:
+                    functionImpl += "texture2d<float, access::sample> \(key) [[texture(\(i+1))]]"
                 }
             case .none:
                 break

Sources/EasyMetalShader/Pipelines/EMMetalRenderFunction.swift

@@ -70,6 +70,8 @@ open class EMMetalRenderFunction: NSObject, EMMetalFunction {
                     functionImpl += "texture2d<float, access::write> \(key) [[texture(\(i+1))]]"
                 case .read_write:
                     functionImpl += "texture2d<float, access::read_write> \(key) [[texture(\(i+1))]]"
+                case .sample:
+                    functionImpl += "texture2d<float, access::sample> \(key) [[texture(\(i+1))]]"
                 }
             case .none:
                 break
@@ -142,6 +144,8 @@ open class EMMetalRenderFunction: NSObject, EMMetalFunction {
                     functionImpl += "texture2d<float, access::write> \(key) [[texture(\(i+1))]]"
                 case .read_write:
                     functionImpl += "texture2d<float, access::read_write> \(key) [[texture(\(i+1))]]"
+                case .sample:
+                    functionImpl += "texture2d<float, access::sample> \(key) [[texture(\(i+1))]]"
                 }
             case .none:
                 break