遍历 UnsafeMutableRawPointer 图像数据
Traversing UnsafeMutableRawPointer image data
我似乎无法理解 Swift 中手动访问图像像素数据背后的方法。我正在尝试从 CGImage 创建图像蒙版,稍后可以在单独的图像上使用。我想识别特定值的所有像素并将图像中的所有其他内容转换为 black/white 或 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 {
for col in 0..<contextWidth {
let offset = (col * contextHeight) + row
好像添加 row
实际上会让您到达那一行。但是 row
只是所需行的 number,而不是该行开始的字节;在我看来,一行跳转的大小需要是一行中所有字节的大小。
我似乎无法理解 Swift 中手动访问图像像素数据背后的方法。我正在尝试从 CGImage 创建图像蒙版,稍后可以在单独的图像上使用。我想识别特定值的所有像素并将图像中的所有其他内容转换为 black/white 或 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 {
for col in 0..<contextWidth {
let offset = (col * contextHeight) + row
好像添加 row
实际上会让您到达那一行。但是 row
只是所需行的 number,而不是该行开始的字节;在我看来,一行跳转的大小需要是一行中所有字节的大小。