swift4如何使用滤镜更改黑白图像的颜色

Question

我想改变作为书页的黑白图像的颜色。 我使用for循环来完成此操作，该循环逐个像素地更改颜色，这花费了太多时间。

var p2b: RGBAPixel = RGBAPixel(r: 0, g: 153, b: 0)
var p2t: RGBAPixel = RGBAPixel(r: 255, g: 255, b: 255)
for y in 0 ..< height {
    for x in 0 ..< width {

         let p1 = img.getPixel(x, y: y)

        if ( p1.red > 200 ) { img2.setPixel(p2b, x: x, y: y) }

        else { img.setPixel(p2t, x: x, y: y) }
    }
}

有更快的方法吗？

Answer 1

这仅适用于黑白图像，例如书页图像

将图像获取为CIImage并制作两个像素，例如newPix1和newPix2

var ciimage: CIImage
ciimage = CIImage(image: myImg)!     

var newPix1 = EFUIntPixel()
newPix1.red = UInt8(25); newPix1.green = UInt8(25); newPix1.blue = 
UInt8(25); newPix1.alpha = UInt8(200)

var newPix2 = EFUIntPixel()
newPix2.red = UInt8(0); newPix2.green = UInt8(255); newPix2.blue = 
UInt8(255); newPix2.alpha = UInt8(200)

现在调用函数替换颜色

ciimage = ciimage.replace(colorNew: newPix1, colorNew2: newPix2, 
isFirst: true)!

图像的白色将被newPix1替换，黑色将被newPix2替换

不要忘记添加以下代码，您可以将其放在所需的位置

struct EFUIntPixel {
    var red: UInt8 = 0
    var green: UInt8 = 0
    var blue: UInt8 = 0
    var alpha: UInt8 = 0
}

extension CIImage {

    // Replace color with another one
    func replace(colorNew: EFUIntPixel,colorNew2: EFUIntPixel, isFirst: Bool) -> CIImage? {
        let cubeSize = 64
        let cubeData = { () -> [Float] in
            let selectColor = (Float(0.862745106), Float(0.862745106), Float(0.862745106), Float(0.784313738))
            let raplaceColor = (Float(colorNew.red) / 255.0, Float(colorNew.green) / 255.0, Float(colorNew.blue) / 255.0, Float(colorNew.alpha) / 255.0)

            let raplaceColor2 = (Float(colorNew2.red) / 255.0, Float(colorNew2.green) / 255.0, Float(colorNew2.blue) / 255.0, Float(colorNew2.alpha) / 255.0)

            var data = [Float](repeating: 0, count: cubeSize * cubeSize * cubeSize * 4)
            var tempRGB: [Float] = [0, 0, 0]
            var newRGB: (r : Float, g : Float, b : Float, a: Float)
            var offset = 0
            for z in 0 ..< cubeSize {
                tempRGB[2] = Float(z) / Float(cubeSize) // blue value
                for y in 0 ..< cubeSize {
                    tempRGB[1] = Float(y) / Float(cubeSize) // green value
                    for x in 0 ..< cubeSize {
                        tempRGB[0] = Float(x) / Float(cubeSize) // red value

                        // Select colorOld
                        if tempRGB[0] <= selectColor.0 && tempRGB[1] <= selectColor.1 && tempRGB[2] <= selectColor.2 {
                            newRGB = (raplaceColor2.0, raplaceColor2.1, raplaceColor2.2, raplaceColor2.3)
                        }
                        // Select colorOld
                        else if tempRGB[0] >= selectColor.0 && tempRGB[1] >= selectColor.1 && tempRGB[2] >= selectColor.2 {
                            newRGB = (raplaceColor.0, raplaceColor.1, raplaceColor.2, raplaceColor.3)
                        } else {
                            newRGB = (tempRGB[0], tempRGB[1], tempRGB[2], 1)
                        }

                        data[offset] = newRGB.r
                        data[offset + 1] = newRGB.g
                        data[offset + 2] = newRGB.b
                        data[offset + 3] = 1.0
                        offset += 4
                    }
                }
            }
            return data
        }()

        let data = cubeData.withUnsafeBufferPointer { Data(buffer: $0) } as NSData
        let colorCube = CIFilter(name: "CIColorCube")!
        colorCube.setValue(cubeSize, forKey: "inputCubeDimension")
        colorCube.setValue(data, forKey: "inputCubeData")
        colorCube.setValue(self, forKey: kCIInputImageKey)
        return colorCube.outputImage
    }
}