一尘不染

如何在iOS的Swift中以像素级别加载和编辑位图文件?

swift

我想在Swift中以像素级别处理图像。针对目标c回答了此问题:如何访问和处理JPEG图像像素?,但我希望看到Swift的等效资源。


阅读 262

收藏
2020-07-07

共1个答案

一尘不染

这就是我从触摸位置的图像中获取颜色的方式。

(此示例没有错误检查是否为零)

func createARGBBitmapContext(inImage: CGImage) -> CGContext {
  var bitmapByteCount = 0
  var bitmapBytesPerRow = 0

  //Get image width, height
  let pixelsWide = CGImageGetWidth(inImage)
  let pixelsHigh = CGImageGetHeight(inImage)

  // Declare the number of bytes per row. Each pixel in the bitmap in this
  // example is represented by 4 bytes; 8 bits each of red, green, blue, and
  // alpha.
  bitmapBytesPerRow = Int(pixelsWide) * 4
  bitmapByteCount = bitmapBytesPerRow * Int(pixelsHigh)

  // Use the generic RGB color space.
  let colorSpace = CGColorSpaceCreateDeviceRGB()

  // Allocate memory for image data. This is the destination in memory
  // where any drawing to the bitmap context will be rendered.
  let bitmapData = malloc(bitmapByteCount)

  // Create the bitmap context. We want pre-multiplied ARGB, 8-bits
  // per component. Regardless of what the source image format is
  // (CMYK, Grayscale, and so on) it will be converted over to the format
  // specified here by CGBitmapContextCreate.
  let context = CGBitmapContextCreate(bitmapData, pixelsWide, pixelsHigh, 8, bitmapBytesPerRow, colorSpace, CGImageAlphaInfo.PremultipliedFirst.rawValue)

  // Make sure and release colorspace before returning
  return context!
}

func getPixelColorAtLocation(point:CGPoint, inImage:CGImageRef) -> NSColor {
  // Create off screen bitmap context to draw the image into. Format ARGB is 4 bytes for each pixel: Alpa, Red, Green, Blue
  let context = self.createARGBBitmapContext(inImage)

  let pixelsWide = CGImageGetWidth(inImage)
  let pixelsHigh = CGImageGetHeight(inImage)
  let rect = CGRect(x:0, y:0, width:Int(pixelsWide), height:Int(pixelsHigh))

  //Clear the context
  CGContextClearRect(context, rect)

  // Draw the image to the bitmap context. Once we draw, the memory
  // allocated for the context for rendering will then contain the
  // raw image data in the specified color space.
  CGContextDrawImage(context, rect, inImage)

  // Now we can get a pointer to the image data associated with the bitmap
  // context.
  let data = CGBitmapContextGetData(context)
  let dataType = UnsafePointer<UInt8>(data)

  let offset = 4*((Int(pixelsWide) * Int(point.y)) + Int(point.x))
  let alpha = dataType[offset]
  let red = dataType[offset+1]
  let green = dataType[offset+2]
  let blue = dataType[offset+3]
  let color = NSColor(red: CGFloat(red)/255.0, green: CGFloat(green)/255.0, blue: CGFloat(blue)/255.0, alpha: CGFloat(alpha)/255.0)

  // Free image data memory for the context
  free(data)
  return color;
}
2020-07-07