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

48次阅读

共计 2683 个字符,预计需要花费 7 分钟才能阅读完成。

通常应用 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 实用的代码,见下图

原理就是如此,所以咱们第一步须要辨别 失常的 metalcikernel 的 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

正文完
 0