在Swift中获取UIImage的主颜色

import UIKit
import CoreImage

func getMainColor(from image: UIImage) -> UIColor? {
    guard let ciImage = image.ciImage ?? CIImage(image: image) else {
        return nil
    }
    
    let context = CIContext()
    guard let cgImage = context.createCGImage(ciImage, from: ciImage.extent) else {
        return nil
    }
    
    let colorCube = CIFilter(name: "CIColorCube")
    colorCube?.setValue(ciImage, forKey: kCIInputImageKey)
    
    guard let outputImage = colorCube?.outputImage else {
        return nil
    }
    
    guard let cgOutputImage = context.createCGImage(outputImage, from: outputImage.extent) else {
        return nil
    }
    
    let dataProvider = cgOutputImage.dataProvider
    let data = dataProvider?.data
    let bytes = CFDataGetBytePtr(data)
    
    var colorCount = [String: Int]()
    
    for i in stride(from: 0, to: data!.count, by: 4) {
        let red = Int(bytes![i])
        let green = Int(bytes![i+1])
        let blue = Int(bytes![i+2])
        
        let colorKey = "\(red)-\(green)-\(blue)"
        if let count = colorCount[colorKey] {
            colorCount[colorKey] = count + 1
        } else {
            colorCount[colorKey] = 1
        }
    }
    
    let sortedColors = colorCount.sorted { $0.value > $1.value }
    if let mostFrequentColor = sortedColors.first?.key {
        let components = mostFrequentColor.components(separatedBy: "-")
        if let red = Int(components[0]), let green = Int(components[1]), let blue = Int(components[2]) {
            let color = UIColor(red: CGFloat(red)/255, green: CGFloat(green)/255, blue: CGFloat(blue)/255, alpha: 1.0)
            return color
        }
    }
    
    return nil
}

// 使用示例
let image = UIImage(named: "example_image")
if let mainColor = getMainColor(from: image) {
    print(mainColor)
} else {
    print("无法获取主颜色")
}