遍历 UnsafeMutableRawPointer 图像数据答案

【问题标题】：Traversing UnsafeMutableRawPointer image data遍历 UnsafeMutableRawPointer 图像数据
【发布时间】：2018-04-05 19:48:07
【问题描述】：

我似乎无法理解在 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) + col、let offset = (col * contextHeight) + row、let offset = ((row * contextWidth) + col) * 3、let offset = ((row * contextWidth) + col) * 4。

我得到的输出看起来像这样（请记住，这张图片应该看起来像一团随机颜色）：

正如我花哨的小箭头所示，顶部的黑色样本是我编辑的像素，这些像素确实应该变成黑色，但是，所有其他灰色像素也是如此（例如箭头下方的那些） .它们绝对是相同的 RGB 值 120、120、120。

我知道问题在于我在数组中移动的顺序，但我似乎无法弄清楚模式是什么。另外，请注意，使用 copy(maskingColorComponents:) 是行不通的，因为我想删除一些特定颜色，而不是它们的范围。

一如既往地非常感谢任何帮助。提前致谢！

【问题讨论】：

你试过调试你的代码吗？应该很容易看出您是否在图像网格中进行了尽可能多的传递。
我确实试过了。我试图访问一个特定的像素（也是它们的一个范围）并将它们涂成黑色，只是为了了解当我循环通过它时我在图像中的哪个位置。我已经尝试了我能想到的每一个行/列组合，但我仍然无法获得正确的排序。这令人难以置信的沮丧。这些数据没有标准顺序吗？没想到这么难！
您显然做得很好，因为您正确地击中了左上角的所有像素。所以看起来你对一行是什么的想法可能有点不对劲。您说的是for row in 0..<contextHeight，只是将row 添加到偏移量，一次一个row，但在我看来，一个row 跳转的大小需要是所有字节的大小在一行中.
@matt 你说得对。以let offset = (col * contextHeight) + (row * numComponents) 结束。所以跳到正确的行，然后以 4 个字节的步长沿着它移动。如果您想将其写成答案，我可以将其标记为正确，因为您有解决方案。
也许这与问题无关，但 bytesPerPixel 不应该是 3，而不是 24？

标签： ios swift core-graphics cgcontext

【解决方案1】：

您显然是在正确的轨道上，因为您正确地击中了左上角的所有像素。但是您不会继续向下图像的其余部分；显然你没有测量足够的行。所以问题可能仅仅是你对一行是的想法有点偏离。

你说的是

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

好像添加 row 实际上会让您进入该行。但是row 只是所需行的number，而不是开始该行的字节；在我看来，一行跳转的大小需要是一行中所有字节的大小。

【讨论】：