乐趣区

关于ios:IOS-Metal使用METAL自定义CIFilter导致vertex-function找不到的问题

jiezi

通常应用 CIFilter 能够对图像做一些解决，如果有些成果咱们不称心，咱们须要本人去实现的话，是能够通过写 metal 来自定义 CIFilter，和 fragment shader 相似，解决对象都是一个像素点。

如何应用 metal shader 自定义 CIFilter 网上的教程有很多，我这里还是赘述一下。

第一步：创立一个.metal 文件，定义 filter。记住本人的办法名，前面须要用到。
第二步：继承 CIFilter 定义出一个子类，通过加载 default.metallib 找到对应的办法即可。
第三步：在 Build Settings 外面退出两个 flag。

上面是第一步的代码，轻易定义一个文件比方叫 kernel.metal，外面放上这些代码。留神到这里的办法名是 myColor。

myColor 办法就是简略的返回以后点的色彩。

#include <metal_stdlib>
#include <CoreImage/CoreImage.h>

using namespace metal;

extern "C" {
    namespace coreimage {float4 myColor(sample_t s, float value) {return s.rgba;}
    }
}

第二步，建设一个本人的 CIFilter 子类，而后做上面这些事件。

class CustomFilter : CIFilter {
    
    var value: Double = 0
    
    private var kernel: CIKernel!
    
    override init() {super.init()
        commonInit()}
    
    required init?(coder: NSCoder) {super.init(coder: coder)
        commonInit()}
    
    private func commonInit() {
        // 找到默认的 default.metallib
        guard let url = Bundle.main.url(forResource: "default", withExtension: "metallib"),
            let data = try? Data(contentsOf: url) else {fatalError("Unable to get metallib")
        }
        
        // 从 lib 中加载到 myColor 办法
        guard let myKernel = try? CIKernel.init(functionName: "myColor", fromMetalLibraryData: data) else {fatalError("Unable to create CIKernel from myKernel")
        }
        
        kernel = myKernel
    }
    
    var inputImage:CIImage?
    override var outputImage: CIImage? {let src = CISampler(image: self.inputImage!)
        // apply 这个 filter
        return kernel.apply(extent: inputImage!.extent, roiCallback: { _, rect in return rect}, arguments: [src, value])
    }
}

第三步，在 Build Setting 外面退出 flag

须要加两个参数一个 -fcikernel，另一个是 -cikernel。这篇文档第 10 页有讲 https://developer.apple.com/metal/MetalCIKLReference6.pdf 照着加一下即可。

好耶，自定义的 CIFilter 能够应用了，BUT，此时此刻你之前写的 pipeline 却无奈失常工作了，会提醒找不到 vertex function。

为什么会这样呢？因为加了 flag 之后，编译的时候会将所有的 .metal 文件都通过某种模式进行非凡的编译使之成为适宜 CoreImage 框架应用的代码，失常的 pipeline 外面须要的 vertex shader 和 fragment shader 都生效了。

为了解决这个问题，须要将辨别哪些是须要通过 -cikernel, -fcikernel flag 打包的，哪些是不须要的。

这里有解决方案：https://stackoverflow.com/questions/57391441/metal-vertexfunction-defined-in-metal-file-becomes-nil-once-setting-compiler-a

我踩了不少坑才实现这项性能。

其实在 https://developer.apple.com/metal/MetalCIKLReference6.pdf 文档中曾经写了如何编译 metal 使之成为 kernel 实用的代码，见下图

原理就是如此，所以咱们第一步须要辨别 失常的 metal 和 cikernel 的 metal，为了辨别这一点，间接将刚刚写的 kernel.metal 的后缀改为 .kernel。这样一般的 metal 文件依然是 *.metal，而为 CIFilter 用的 metal 文件后缀则是 .kernel。

第二步，编译前须要改回 .metal 后缀，通过实际发现，metal 编译器间接疏忽了后缀名不正确的文件以至于无奈编译胜利，所以咱们须要先执行一个 cp 步骤。

第三步，执行如上图所示的编译，变成一个咱们自定义的 xxxx.metallib

第四步，将 xxxx.metallib 复制到打包门路中，以便打包的时候能够打进 ipa 文件。

第五步，在自定义的 CIFilter 中加载 metallib 时，应用刚刚自定义的 xxxx.metallib，而不是 default。

当初来讲讲具体操作。

在 Build Rules 中增加一个步骤

在外面的输入框中放入 

# 复制一下，不然 metal 编译器不意识
cp "${SRCROOT}/Varlens/Shader/kernel.kernel" "${DERIVED_FILES_DIR}/kernel.metal"

# 前面这两句就是编译了，留神输入输出门路即可
xcrun metal -fcikernel "${DERIVED_FILES_DIR}/kernel.metal" -c -o "${DERIVED_FILES_DIR}/MyKernels.air"
xcrun metallib -cikernel "${DERIVED_FILES_DIR}/MyKernels.air" -o ${DERIVED_FILE_DIR}/kernel.metallib

而后在 Build Phases 中增加一个步骤，做复制 metallib 的操作。

完事。

krosshj @ 2021-04-22 14:57

退出移动版