Swift iOS：Core Image搭配自定义Metal Kernel运行异常如何排查

问题根源与修复方案 #

你遇到的异常由三个典型的Metal/Core Image Kernel开发问题导致：

1. 整数除法逻辑错误
   Metal是C系语言，整数常量相除默认执行整数截断，你代码中所有1/3、1/6、2/3的运算结果都是0，完全不符合预期的浮点数值，这是输出黑白图像的核心原因。Swift中因为类型推断会自动将整数常量转为参与浮点运算，所以Playground中运行正常。

2. 浮点数相等判断精度问题
   直接用==判断maxComp == minComp、adjustedSat == 0会因为浮点数精度误差出现分支判断错误，导致部分像素的HSL转换逻辑异常。

3. 颜色分量溢出
   调整后的饱和度、亮度以及最终RGB分量没有做范围钳制，超出[0,1]合法范围的数值会导致像素溢出，出现细碎的黑白噪点块。

修复后的Metal Kernel代码 #

#include <metal_stdlib>
using namespace metal;
#include <CoreImage/CoreImage.h>

// 浮点数精度判断阈值
constant float EPS = 1e-6;

extern "C" {
    namespace coreimage {
        float4 hslFilterKernel(sample_t s, float center, float hueOffset, float satOffset, float lumOffset) {
            float maxComp = (s.r > s.g && s.r > s.b) ? s.r : (s.g > s.b) ? s.g : s.b ;
            float minComp = (s.r < s.g && s.r < s.b) ? s.r : (s.g < s.b) ? s.g : s.b ;
            
            float inputHue = (maxComp + minComp)/2.0 ;
            float inputSat = (maxComp + minComp)/2.0 ;
            float inputLum = (maxComp + minComp)/2.0 ;
            
            if (fabs(maxComp - minComp) < EPS) {
                inputHue = 0.0 ;
                inputSat = 0.0 ;
            } else {
                float delta = maxComp - minComp ;
                inputSat = inputLum > 0.5 ? delta/(2.0 - maxComp - minComp) : delta/(maxComp + minComp);
                if (s.r > s.g && s.r > s.b) {
                    inputHue = (s.g - s.b)/delta + (s.g < s.b ? 6.0 : 0.0);
                } else if (s.g > s.b) {
                    inputHue = (s.b - s.r)/delta + 2.0;
                } else {
                    inputHue = (s.r - s.g)/delta + 4.0;
                }
                inputHue = inputHue/6.0 ;
            }
            
            float minHue = center - 22.5/(360.0) ;
            float maxHue = center + 22.5/(360.0) ;
            
            float adjustedHue = inputHue + ((inputHue > minHue && inputHue < maxHue) ? hueOffset : 0.0 );
            // 饱和度钳制在0-1范围
            float adjustedSat = clamp(inputSat + ((inputHue > minHue && inputHue < maxHue) ? satOffset : 0.0 ), 0.0, 1.0);
            // 亮度钳制在0-1范围
            float adjustedLum = clamp(inputLum + ((inputHue > minHue && inputHue < maxHue) ? lumOffset : 0.0 ), 0.0, 1.0);
            
            float red = 0.0 ;
            float green = 0.0 ;
            float blue = 0.0 ;
            
            if (adjustedSat < EPS) {
                red = adjustedLum;
                green = adjustedLum;
                blue = adjustedLum;
            } else {
                float q = adjustedLum < 0.5 ? adjustedLum*(1.0+adjustedSat) : adjustedLum + adjustedSat - (adjustedLum*adjustedSat);
                float p = 2.0*adjustedLum - q;
                
                // 所有除法改为浮点运算
                float t = adjustedHue + 1.0/3.0;
                if (t < 0.0) { t += 1.0; }
                if (t > 1.0) { t -= 1.0; }
                if (t < 1.0/6.0) { red = p + (q - p)*6.0*t; }
                else if (t < 1.0/2.0) { red = q; }
                else if (t < 2.0/3.0) { red = p + (q - p)*(2.0/3.0 - t)*6.0; }
                else { red = p; }
                
                t = adjustedHue;
                if (t < 0.0) { t += 1.0; }
                if (t > 1.0) { t -= 1.0; }
                if (t < 1.0/6.0) { green = p + (q - p)*6.0*t; }
                else if (t < 1.0/2.0) { green = q ;}
                else if (t < 2.0/3.0) { green = p + (q - p)*(2.0/3.0 - t)*6.0; }
                else { green = p; }
                
                t = adjustedHue - 1.0/3.0;
                if (t < 0.0) { t += 1.0; }
                if (t > 1.0) { t -= 1.0; }
                if (t < 1.0/6.0) { blue = p + (q - p)*6.0*t; }
                else if (t < 1.0/2.0) { blue = q; }
                else if (t < 2.0/3.0) { blue = p + (q - p)*(2.0/3.0 - t)*6.0;}
                else { blue = p; }
            }
            
            float4 outColor;
            // 最终RGB分量钳制在0-1范围
            outColor.r = clamp(red, 0.0, 1.0);
            outColor.g = clamp(green, 0.0, 1.0);
            outColor.b = clamp(blue, 0.0, 1.0);
            outColor.a = s.a;
            
            return outColor;
        }
    }
}

附加排查技巧 #

Metal Kernel无法直接输出日志时，可以将中间计算结果（比如inputHue、inputSat）赋值给输出颜色的RGB通道，通过观察输出图像的颜色分布就能定位哪一步计算出现问题。

内容的提问来源于stack exchange，提问作者diclonius9