Functors (#88)

* add functors * fix formatting * update mps+functors * replace inout with pointer
s1ddok · Jun 11, 2020 · c29f4eb · c29f4eb
1 parent bae5270
commit c29f4eb
Show file tree

Hide file tree

Showing 72 changed files with 519 additions and 524 deletions.
diff --git a/Alloy/CVPixelBuffer+MTLTexture.swift b/Alloy/CVPixelBuffer+MTLTexture.swift
@@ -1,11 +1,3 @@
-//
-//  CVPixelFormat+MTLTexture.swift
-//  Alloy
-//
-//  Created by Andrey Volodin on 20.10.17.
-//  Copyright © 2017 Andrey Volodin. All rights reserved.
-//
-
 import Metal
 import CoreVideo.CVPixelBuffer
 

diff --git a/Alloy/CodableConformance.swift b/Alloy/CodableConformance.swift
@@ -1,10 +1,3 @@
-//
-//  CodableConformance.swift
-//  Beautifier
-//
-//  Created by Andrey Volodin on 28/08/2019.
-//
-
 import Metal
 
 extension MTLCPUCacheMode: Codable {}

diff --git a/Alloy/ComputeCommand.swift b/Alloy/ComputeCommand.swift
@@ -1,11 +1,3 @@
-//
-//  ComputeFunction.swift
-//  AlloyDemo
-//
-//  Created by Andrey Volodin on 28.08.2018.
-//  Copyright © 2018 Andrey Volodin. All rights reserved.
-//
-
 import Metal
 
 @dynamicMemberLookup

diff --git a/Alloy/Encoders/BlockSize.swift b/Alloy/Encoders/BlockSize.swift
@@ -1,10 +1,3 @@
-//
-//  BlockSize.swift
-//  Alloy
-//
-//  Created by Eugene Bokhan on 14/02/2019.
-//
-
 import Foundation
 
 public struct BlockSize {

diff --git a/Alloy/Encoders/EuclideanDistance.swift b/Alloy/Encoders/EuclideanDistance.swift
@@ -1,10 +1,3 @@
-//
-//  EuclideanDistance.swift
-//  Alloy
-//
-//  Created by Eugene Bokhan on 30/08/2019.
-//
-
 import Metal
 
 final public class EuclideanDistance {
@@ -29,6 +22,26 @@ final public class EuclideanDistance {
 
     // MARK: - Encode
 
+    public func callAsFunction(textureOne: MTLTexture,
+                               textureTwo: MTLTexture,
+                               resultBuffer: MTLBuffer,
+                               in commandBuffer: MTLCommandBuffer) {
+        self.encode(textureOne: textureOne,
+                    textureTwo: textureTwo,
+                    resultBuffer: resultBuffer,
+                    in: commandBuffer)
+    }
+
+    public func callAsFunction(textureOne: MTLTexture,
+                               textureTwo: MTLTexture,
+                               resultBuffer: MTLBuffer,
+                               using encoder: MTLComputeCommandEncoder) {
+        self.encode(textureOne: textureOne,
+                    textureTwo: textureTwo,
+                    resultBuffer: resultBuffer,
+                    using: encoder)
+    }
+
     public func encode(textureOne: MTLTexture,
                        textureTwo: MTLTexture,
                        resultBuffer: MTLBuffer,

diff --git a/Alloy/Encoders/LookUpTable.swift b/Alloy/Encoders/LookUpTable.swift
@@ -1,10 +1,3 @@
-//
-//  LookUpTable.swift
-//  Alloy
-//
-//  Created by Andrey Volodin on 29.10.2019.
-//
-
 import Metal
 
 final public class LookUpTable {
@@ -33,6 +26,30 @@ final public class LookUpTable {
 
     // MARK: - Encode
 
+    public func callAsFunction(sourceTexture: MTLTexture,
+                               outputTexture: MTLTexture,
+                               lut: MTLTexture,
+                               intensity: Float,
+                               in commandBuffer: MTLCommandBuffer) {
+        self.encode(sourceTexture: sourceTexture,
+                    outputTexture: outputTexture,
+                    lut: lut,
+                    intensity: intensity,
+                    in: commandBuffer)
+    }
+
+    public func callAsFunction(sourceTexture: MTLTexture,
+                               outputTexture: MTLTexture,
+                               lut: MTLTexture,
+                               intensity: Float,
+                               using encoder: MTLComputeCommandEncoder) {
+        self.encode(sourceTexture: sourceTexture,
+                    outputTexture: outputTexture,
+                    lut: lut,
+                    intensity: intensity,
+                    using: encoder)
+    }
+
     public func encode(sourceTexture: MTLTexture,
                        outputTexture: MTLTexture,
                        lut: MTLTexture,

diff --git a/Alloy/Encoders/MPSUnaryImageKernels.swift b/Alloy/Encoders/MPSUnaryImageKernels.swift
@@ -1,10 +1,3 @@
-//
-//  MPSUnaryImageKernels.swift
-//  Alloy
-//
-//  Created by Eugene Bokhan on 27.09.2019.
-//
-
 import MetalPerformanceShaders
 
 final public class MPSUnaryImageKernels {
@@ -21,6 +14,14 @@ final public class MPSUnaryImageKernels {
 
     // MARK: - Encode
 
+    public func callAsFunction(sourceTexture: MTLTexture,
+                               destinationTexture: MTLTexture,
+                               in commandBuffer: MTLCommandBuffer) {
+        self.encode(sourceTexture: sourceTexture,
+                    destinationTexture: destinationTexture,
+                    in: commandBuffer)
+    }
+
     public func encode(sourceTexture: MTLTexture,
                        destinationTexture: MTLTexture,
                        in commandBuffer: MTLCommandBuffer) {

diff --git a/Alloy/Encoders/MaskGuidedBlur.swift b/Alloy/Encoders/MaskGuidedBlur.swift
@@ -1,10 +1,3 @@
-//
-//  MaskGuidedBlur.swift
-//  Alloy
-//
-//  Created by Eugene Bokhan on 29/08/2019.
-//
-
 import Metal
 import MetalPerformanceShaders
 
@@ -36,6 +29,18 @@ final public class MaskGuidedBlur {
 
     // MARK: - Encode
 
+    public func callAsFunction(sourceTexture: MTLTexture,
+                               maskTexture: MTLTexture,
+                               destinationTexture: MTLTexture,
+                               sigma: Float,
+                               in commandBuffer: MTLCommandBuffer) {
+        self.encode(sourceTexture: sourceTexture,
+                    maskTexture: maskTexture,
+                    destinationTexture: destinationTexture,
+                    sigma: sigma,
+                    in: commandBuffer)
+    }
+
     public func encode(sourceTexture: MTLTexture,
                        maskTexture: MTLTexture,
                        destinationTexture: MTLTexture,

diff --git a/Alloy/Encoders/NormalizeKernel.swift b/Alloy/Encoders/NormalizeKernel.swift
@@ -1,10 +1,3 @@
-//
-//  NormalizeKernel.swift
-//  Alloy
-//
-//  Created by Eugene Bokhan on 08/05/2019.
-//
-
 import Metal
 import simd
 
@@ -33,6 +26,30 @@ final public class NormalizeKernel {
 
     // MARK: - Encode
 
+    public func callAsFunction(sourceTexture: MTLTexture,
+                               destinationTexture: MTLTexture,
+                               mean: SIMD3<Float>,
+                               std: SIMD3<Float>,
+                               in commandBuffer: MTLCommandBuffer) {
+        self.encode(sourceTexture: sourceTexture,
+                    destinationTexture: destinationTexture,
+                    mean: mean,
+                    std: std,
+                    in: commandBuffer)
+    }
+
+    public func callAsFunction(sourceTexture: MTLTexture,
+                               destinationTexture: MTLTexture,
+                               mean: SIMD3<Float>,
+                               std: SIMD3<Float>,
+                               using encoder: MTLComputeCommandEncoder) {
+        self.encode(sourceTexture: sourceTexture,
+                    destinationTexture: destinationTexture,
+                    mean: mean,
+                    std: std,
+                    using: encoder)
+    }
+
     public func encode(sourceTexture: MTLTexture,
                        destinationTexture: MTLTexture,
                        mean: SIMD3<Float>,

diff --git a/Alloy/Encoders/TextureAddConstant.swift b/Alloy/Encoders/TextureAddConstant.swift
@@ -1,10 +1,3 @@
-//
-//  TextureAddConstant.swift
-//  Alloy
-//
-//  Created by Eugene Bokhan on 03/09/2019.
-//
-
 import Metal
 
 final public class TextureAddConstant {
@@ -37,6 +30,26 @@ final public class TextureAddConstant {
 
     // MARK: - Encode
 
+    public func callAsFunction(sourceTexture: MTLTexture,
+                               destinationTexture: MTLTexture,
+                               constant: SIMD4<Float>,
+                               in commandBuffer: MTLCommandBuffer) {
+        self.encode(sourceTexture: sourceTexture,
+                    destinationTexture: destinationTexture,
+                    constant: constant,
+                    in: commandBuffer)
+    }
+
+    public func callAsFunction(sourceTexture: MTLTexture,
+                               destinationTexture: MTLTexture,
+                               constant: SIMD4<Float>,
+                               using encoder: MTLComputeCommandEncoder) {
+        self.encode(sourceTexture: sourceTexture,
+                    destinationTexture: destinationTexture,
+                    constant: constant,
+                    using: encoder)
+    }
+
     public func encode(sourceTexture: MTLTexture,
                        destinationTexture: MTLTexture,
                        constant: SIMD4<Float>,

diff --git a/Alloy/Encoders/TextureAffineCrop.swift b/Alloy/Encoders/TextureAffineCrop.swift
@@ -1,10 +1,3 @@
-//
-//  TextureAffineCrop.swift
-//  Pods
-//
-//  Created by Andrey Volodin on 18.11.2019.
-//
-
 import Metal
 
 final public class TextureAffineCrop {
@@ -33,6 +26,26 @@ final public class TextureAffineCrop {
 
     // MARK: - Encode
 
+    public func callAsFunction(sourceTexture: MTLTexture,
+                               destinationTexture: MTLTexture,
+                               affineTransform: simd_float3x3,
+                               in commandBuffer: MTLCommandBuffer) {
+        self.encode(sourceTexture: sourceTexture,
+                    destinationTexture: destinationTexture,
+                    affineTransform: affineTransform,
+                    in: commandBuffer)
+    }
+
+    public func callAsFunction(sourceTexture: MTLTexture,
+                               destinationTexture: MTLTexture,
+                               affineTransform: simd_float3x3,
+                               using encoder: MTLComputeCommandEncoder) {
+        self.encode(sourceTexture: sourceTexture,
+                    destinationTexture: destinationTexture,
+                    affineTransform: affineTransform,
+                    using: encoder)
+    }
+
     public func encode(sourceTexture: MTLTexture,
                        destinationTexture: MTLTexture,
                        affineTransform: simd_float3x3,

diff --git a/Alloy/Encoders/TextureCopy.swift b/Alloy/Encoders/TextureCopy.swift
@@ -1,10 +1,3 @@
-//
-//  TextureCopy.swift
-//  Alloy
-//
-//  Created by Andrey Volodin on 31/01/2019.
-//
-
 import Metal
 
 final public class TextureCopy {
@@ -36,6 +29,46 @@ final public class TextureCopy {
 
     // MARK: - Encode
 
+    public func callAsFunction(sourceTexture: MTLTexture,
+                               destinationTexture: MTLTexture,
+                               in commandBuffer: MTLCommandBuffer) {
+        self.encode(sourceTexture: sourceTexture,
+                    destinationTexture: destinationTexture,
+                    in: commandBuffer)
+    }
+
+    public func callAsFunction(sourceTexture: MTLTexture,
+                               destinationTexture: MTLTexture,
+                               using encoder: MTLComputeCommandEncoder) {
+        self.encode(sourceTexture: sourceTexture,
+                    destinationTexture: destinationTexture,
+                    using: encoder)
+    }
+
+    public func callAsFunction(region sourceTexureRegion: MTLRegion,
+                               from sourceTexture: MTLTexture,
+                               to destinationTextureOrigin: MTLOrigin,
+                               of destinationTexture: MTLTexture,
+                               in commandBuffer: MTLCommandBuffer) {
+        self.copy(region: sourceTexureRegion,
+                  from: sourceTexture,
+                  to: destinationTextureOrigin,
+                  of: destinationTexture,
+                  in: commandBuffer)
+    }
+
+    public func callAsFunction(region sourceTexureRegion: MTLRegion,
+                               from sourceTexture: MTLTexture,
+                               to destinationTextureOrigin: MTLOrigin,
+                               of destinationTexture: MTLTexture,
+                               using encoder: MTLComputeCommandEncoder) {
+        self.copy(region: sourceTexureRegion,
+                  from: sourceTexture,
+                  to: destinationTextureOrigin,
+                  of: destinationTexture,
+                  using: encoder)
+    }
+
     public func encode(sourceTexture: MTLTexture,
                        destinationTexture: MTLTexture,
                        in commandBuffer: MTLCommandBuffer) {