vImageFloodFill_ARGB8888替换全部像素而非仅相邻像素问题求助

环境与修改说明 #

• 开发环境：Xcode Version 14.1 beta 3 (14B5033e)
• 基于Apple官方vImage泛洪填充示例做了两处修改：
  1. 替换编译失败的imageLiteral代码为：

     guard let inputCGImage = img.cgImage else { print("Can't create input CGImage") return }
     

  2. 原示例未定义tmp变量，根据vImageFloodFill_ARGB8888文档，传入nil让系统自动创建临时缓冲区

问题：点击图像触发泛洪填充后，整图变为单色块，而非仅替换点击位置的相邻同色像素。

问题排查与解决方案 #

1. 检查颜色匹配阈值（threshold参数） #

泛洪填充的核心是通过阈值判断相邻像素是否属于同一区域。如果阈值设置过大（比如设为255），会判定所有像素与目标颜色匹配，直接填充整图。
解决：
设置合理的阈值，针对ARGB8888格式，建议用10-30的小数值（数值越小，颜色匹配越严格）：

let threshold: UInt32 = 15 // 根据需求调整

2. 修正点击坐标的转换逻辑 #

如果UIKit点击坐标未正确转换为CGImage的像素坐标（忽略了图像缩放、contentMode或坐标系差异），会导致泛洪填充的起始位置错误，甚至误触发全图填充。
解决：
将点击点转换为图像像素坐标：

// 适配UIImageView的contentMode，转换点击点为图像像素坐标
func convertTapPointToPixel(_ point: CGPoint, in imageView: UIImageView) -> CGPoint? {
    guard let image = imageView.image else { return nil }
    let scale = image.scale
    let imgSize = image.size
    let viewSize = imageView.bounds.size
    
    // 计算图像在视图中的实际显示区域
    let aspectRatio = imgSize.width / imgSize.height
    var drawRect = CGRect.zero
    if viewSize.width / viewSize.height > aspectRatio {
        drawRect.size.height = viewSize.height
        drawRect.size.width = viewSize.height * aspectRatio
        drawRect.origin.x = (viewSize.width - drawRect.width) / 2
    } else {
        drawRect.size.width = viewSize.width
        drawRect.size.height = viewSize.width / aspectRatio
        drawRect.origin.y = (viewSize.height - drawRect.height) / 2
    }
    
    // 归一化后转换为像素坐标
    let normX = (point.x - drawRect.origin.x) / drawRect.width
    let normY = (point.y - drawRect.origin.y) / drawRect.height
    let pixelX = normX * imgSize.width * scale
    let pixelY = normY * imgSize.height * scale
    return CGPoint(x: pixelX, y: pixelY)
}

调用该方法得到正确像素坐标后，再传入泛洪填充函数。

3. 验证vImageBuffer的格式与初始化 #

如果输入图像格式不是ARGB8888，或者vImageBuffer的行字节数计算错误，会导致颜色判断逻辑异常，引发全图填充。
解决：
确保缓冲区初始化时格式匹配：

guard let inputCGImage = img.cgImage else { 
    print("Can't create input CGImage") 
    return 
}
var format = vImage_CGImageFormat(bitsPerComponent: 8,
                                  bitsPerPixel: 32,
                                  colorSpace: nil,
                                  bitmapInfo: CGBitmapInfo(rawValue: CGImageAlphaInfo.premultipliedFirst.rawValue),
                                  version: 0,
                                  decode: nil,
                                  renderingIntent: .defaultIntent)
var sourceBuffer = vImage_Buffer()
let status = vImageBuffer_InitWithCGImage(&sourceBuffer,
                                          &format,
                                          nil,
                                          inputCGImage,
                                          vImage_Flags(kvImageNoFlags))
guard status == kvImageNoError else {
    print("Buffer init failed with status: \(status)")
    return
}

注意bitmapInfo要和图像的alpha通道类型对应（比如premultipliedFirst对应ARGB格式）。

4. 检查填充颜色的格式 #

如果填充颜色的通道顺序与图像格式不匹配（比如传入RGBA而非ARGB），会导致颜色判断混乱，引发异常填充。
解决：
针对ARGB8888格式，按A-R-G-B顺序设置填充颜色：

// 示例：不透明红色（A=255, R=255, G=0, B=0）
let fillColor: UInt32 = 0xFF0000FF

5. 手动创建临时缓冲区排查 #

虽然文档说明传入nil会自动创建临时缓冲区，但部分场景下可能存在初始化异常，建议手动创建缓冲区验证：

var tmpBuffer = vImage_Buffer()
let tmpStatus = vImageBuffer_Init(&tmpBuffer,
                                  sourceBuffer.height,
                                  sourceBuffer.width,
                                  format.bitsPerPixel,
                                  vImage_Flags(kvImageNoFlags))
guard tmpStatus == kvImageNoError else {
    print("Tmp buffer init failed")
    vImageBuffer_Free(&sourceBuffer)
    return
}
// 调用泛洪填充时传入tmpBuffer
let floodStatus = vImageFloodFill_ARGB8888(&sourceBuffer,
                                           &tmpBuffer,
                                           Int32(pixelX),
                                           Int32(pixelY),
                                           fillColor,
                                           threshold,
                                           kvImageNoFlags)
// 使用后释放临时缓冲区
vImageBuffer_Free(&tmpBuffer)

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