在UIImage中查找特定颜色值

import UIKit

func findColorInImage(image: UIImage, targetColor: UIColor) -> [CGPoint] {
    guard let cgImage = image.cgImage else {
        return []
    }
    
    let width = cgImage.width
    let height = cgImage.height
    
    let colorSpace = CGColorSpaceCreateDeviceRGB()
    let bytesPerPixel = 4
    let bytesPerRow = bytesPerPixel * width
    let bitsPerComponent = 8
    
    let bitmapInfo = CGImageAlphaInfo.premultipliedLast.rawValue | CGBitmapInfo.byteOrder32Big.rawValue
    
    guard let context = CGContext(data: nil, width: width, height: height, bitsPerComponent: bitsPerComponent, bytesPerRow: bytesPerRow, space: colorSpace, bitmapInfo: bitmapInfo) else {
        return []
    }
    
    context.draw(cgImage, in: CGRect(x: 0, y: 0, width: width, height: height))
    
    guard let pixelData = context.data else {
        return []
    }
    
    let data = pixelData.bindMemory(to: UInt32.self, capacity: width * height)
    
    var matchingPoints: [CGPoint] = []
    
    for y in 0..<height {
        for x in 0..<width {
            let offset = y * width + x
            let color = data[offset]
            
            let red = CGFloat((color >> 16) & 0xff) / 255.0
            let green = CGFloat((color >> 8) & 0xff) / 255.0
            let blue = CGFloat(color & 0xff) / 255.0
            
            let pixelColor = UIColor(red: red, green: green, blue: blue, alpha: 1.0)
            
            if pixelColor == targetColor {
                let point = CGPoint(x: x, y: y)
                matchingPoints.append(point)
            }
        }
    }
    
    return matchingPoints
}

// 使用示例
let image = UIImage(named: "example_image")
let targetColor = UIColor.red

let matchingPoints = findColorInImage(image: image, targetColor: targetColor)
print("匹配的像素点：\(matchingPoints)")