遍历UnsafeMutableRawPointer图像数据

时间:2018-04-05 19:48:08

标签: ios swift core-graphics cgcontext

我似乎无法围绕手动访问Swift中的图像像素数据的方法。我试图从CGImage创建一个图像蒙版,以后可以在单独的图像上使用。我想识别特定值的所有像素,并将图像中的所有其他像素转换为黑/白或可能是alpha(但此时并不重要)。我玩的代码看起来像这样:

    let colorSpace: CGColorSpace = CGColorSpaceCreateDeviceRGB()
    let contextWidth: Int = Int(snapshot.size.width)
    let contextHeight: Int = Int(snapshot.size.height)
    let bytesPerPixel: Int = 24
    let bitsPerComponent: Int = 8
    let bytesPerRow: Int = bytesPerPixel * contextWidth
    let bitmapInfo: CGBitmapInfo = CGBitmapInfo(rawValue: CGImageAlphaInfo.noneSkipLast.rawValue)

    guard let context: CGContext = CGContext(data: nil, width: contextWidth, height: contextHeight, bitsPerComponent: bitsPerComponent, bytesPerRow: bytesPerRow, space: colorSpace, bitmapInfo: bitmapInfo.rawValue) else {
        print("Could not create CGContext")
        return
    }

    context.draw(maskCGImage, in: CGRect(x: 0, y: 0, width: contextWidth, height: contextHeight))
    guard let contextDataRaw: UnsafeMutableRawPointer = context.data else {
        print("Could not get UnsafeMutableRawPointer from CGContext")
        return
    }

    let contextData: UnsafeMutablePointer<UInt8> = contextDataRaw.bindMemory(to: UInt8.self, capacity: contextWidth * contextHeight)

    for row in 0..<contextHeight {
        for col in 0..<contextWidth {
            let offset = (col * contextHeight) + row
            let pixelArray = [contextData[offset], contextData[offset + 1], contextData[offset + 2]]
            if pixelArray == [120, 120, 120] {
                contextData[offset] = 0
                contextData[offset + 1] = 0
                contextData[offset + 2] = 0
            }

        }
    }

我尝试了各种行和列的排列,试图找出正确的顺序,即let offset = (row * contextWidth) + collet offset = (col * contextHeight) + rowlet offset = ((row * contextWidth) + col) * 3let offset = ((row * contextWidth) + col) * 4

我得到的输出看起来像这样(请记住,这个图像应该看起来像一团随机颜色):

enter image description here

正如我喜欢的小箭头所示,顶部的黑色样本是我编辑过的像素,然而这些像素确实应该变成黑色,所有其他灰色像素(例如箭头下的像素)也是如此。绝对是相同的RGB值120,120,120。

我知道这个问题是我在数组中移动的顺序,我似乎无法弄清楚模式是什么。另外,作为一个注释,使用copy(maskingColorComponents :)不会做,因为我想删除一些特定的颜色,而不是它们的范围。

任何帮助都会一如既往地受到高度赞赏。提前谢谢!

1 个答案:

答案 0 :(得分:1)

您显然是在正确的轨道上,因为您已经正确地点击了左上角的所有像素。但你不能继续完成 down 图像;显然你没有调查足够的行。所以问题可能仅仅在于您对 行的想法略有不同。

你在说

for row in 0..<contextHeight {
    for col in 0..<contextWidth {
        let offset = (col * contextHeight) + row

好像添加row实际上会让你到那一行。但row只是所需行的数字,而不是开始该行的字节;在我看来,一行跳转的大小需要是一行中所有字节的大小。