ios - 遍历 UIImage 的所有像素

Question

我需要一些帮助来了解如何遍历 UIImage 中的每个像素。我在每个像素中找到红色、绿色、蓝色和 alpha 颜色值，然后比较它们以查看照片中是否存在某种颜色。

我有一个扩展可以返回像素的这些颜色值:

extension UIImage { func getPixelColor(pos: CGPoint) -> UIColor {

let pixelData = self.cgImage!.dataProvider!.data
let data: UnsafePointer<UInt8> = CFDataGetBytePtr(pixelData)

let pixelInfo: Int = ((Int(self.size.width) * Int(pos.y)) + Int(pos.x)) * 4

let r = CGFloat(data[pixelInfo]) / CGFloat(255.0)
let g = CGFloat(data[pixelInfo+1]) / CGFloat(255.0)
let b = CGFloat(data[pixelInfo+2]) / CGFloat(255.0)
let a = CGFloat(data[pixelInfo+3]) / CGFloat(255.0)

return UIColor(red: r, green: g, blue: b, alpha: a)
}

我能够找到图像中特定像素的颜色值并检查它是否与特定颜色值匹配:

let specificPoint = CGPoint(x: 25, y: 25)
    let findColorValue = displayImage.image?.getPixelColor(pos: specificPoint)

    var redval: CGFloat = 0
    var greenval: CGFloat = 0
    var blueval: CGFloat = 0
    var alphaval: CGFloat = 0

    resultLabel.text = "r: \(redval) g: \(greenval) b: \(blueval) a: \(alphaval)"
    findColorValue?.getRed(&redval, green: &greenval, blue: &blueval, alpha: &alphaval)
    if resultLabel.text == "r: 1.0 g: 0.792156862745098 b: 0.474509803921569 a: 0.0156862745098039" {
        resultLabel.text = "YES"
    } else {
        resultLabel.text = "NO"
    }

我需要帮助创建一个 for 循环来遍历图像中的所有像素并能够测试它们的特定颜色值。

我也不确定我获取像素颜色值的方式是否 100% 正确，所以如果我做错了什么，请随时告诉我。

谢谢

Answer 1

使用嵌套的 for 循环:

for yCo in 0 ..< Int(image.size.height) { 
    for xCo in 0 ..< Int(image.size.width) {

        if image.getPixelColor(pos: CGPoint(x: xCo, y: yCo)) == UIColor(red: 1, green: 1, blue: 1, alpha: 1) {
            //Next step
        }
    }
}