如何在Swift中获取UIImage中像素的颜色？

Question

我正在尝试使用Swift来获取UIImage中像素的颜色，但它似乎总是返回0。这是从@Minas
对此线程的回答中转换的代码：

func getPixelColor(pos: CGPoint) -> UIColor {
    var pixelData = CGDataProviderCopyData(CGImageGetDataProvider(self.CGImage))
    var data: UnsafePointer<UInt8> = CFDataGetBytePtr(pixelData)

    var pixelInfo: Int = ((Int(self.size.width) * Int(pos.y)) + Int(pos.x)) * 4

    var r = CGFloat(data[pixelInfo])
    var g = CGFloat(data[pixelInfo+1])
    var b = CGFloat(data[pixelInfo+2])
    var a = CGFloat(data[pixelInfo+3])

    return UIColor(red: r, green: g, blue: b, alpha: a)
}

提前致谢！

Answer 1

由于遇到了类似的问题，一些搜索将我引到了这里。您的代码工作正常。该问题可能是由您的图像引起的。

码：

  //On the top of your swift 
  extension UIImage {
      func getPixelColor(pos: CGPoint) -> UIColor {

          let pixelData = CGDataProviderCopyData(CGImageGetDataProvider(self.CGImage))
          let data: UnsafePointer<UInt8> = CFDataGetBytePtr(pixelData)

          let pixelInfo: Int = ((Int(self.size.width) * Int(pos.y)) + Int(pos.x)) * 4

          let r = CGFloat(data[pixelInfo]) / CGFloat(255.0)
          let g = CGFloat(data[pixelInfo+1]) / CGFloat(255.0)
          let b = CGFloat(data[pixelInfo+2]) / CGFloat(255.0)
          let a = CGFloat(data[pixelInfo+3]) / CGFloat(255.0)

          return UIColor(red: r, green: g, blue: b, alpha: a)
      }  
  }

发生的是此方法将从图像的CGImage中选择像素颜色。因此，请确保您选择的是正确的图像。例如，如果您的UIImage是200x200，但是来自Imgaes.xcassets或它来自何处的原始图像文件是400x400，并且您正在选取点（100,100），则实际上是在该图像的左上部分选取该点中。

两种解决方案： 1，使用来自Imgaes.xcassets的图像，并且在1x字段中仅放置一个@ 1x图像。将@ 2x，@
3x留空。确保您知道图像的大小，并选择一个在范围内的点。

//Make sure only 1x image is set
let image : UIImage = UIImage(named:"imageName") 
//Make sure point is within the image
let color : UIColor = image.getPixelColor(CGPointMake(xValue, yValue))

2，按比例缩放CGPoint以匹配UIImage。例如let point = CGPoint(100,100)在上面的例子中，

let xCoordinate : Float = Float(point.x) * (400.0/200.0)
let yCoordinate : Float = Float(point.y) * (400.0/200.0) 
let newCoordinate : CGPoint = CGPointMake(CGFloat(xCoordinate), CGFloat(yCoordinate))
let image : UIImage = largeImage
let color : UIColor = image.getPixelColor(CGPointMake(xValue, yValue))

我只测试了第一种方法，并且正在使用它从调色板中获得颜色。两者都应该起作用。快乐编码：）