0

我写了一个函数来计算图像特定部分的平均颜色。例如，使用自上而下的黑白渐变PNG示例，我可以使用myImage.averageColor(.bottom)

extension UIImage {
    enum Region {
        case top, bottom
    }
    
    func averageColor(_ region: Region) -> Color? {
        guard let inputImage = CIImage(image: self) else { return nil }
    
        let extentVector: CIVector = {
            switch region {
            case .top:
                return CIVector(x: 0, y: inputImage.extent.size.height - 1, z: inputImage.extent.size.width, w: inputImage.extent.size.height)
            case .bottom:
                return CIVector(x: 0, y: 0, z: inputImage.extent.size.width, w: 1)
            }
        }()
        
        guard let filter = CIFilter(name: "CIAreaAverage", parameters: [kCIInputImageKey: inputImage, kCIInputExtentKey: extentVector]) else { return nil }
        guard let outputImage = filter.outputImage else { return nil }
        
        var bitmap = [UInt8](repeating: 0, count: 4)
        let context = CIContext(options: [.workingColorSpace: kCFNull])
        context.render(outputImage, toBitmap: &bitmap, rowBytes: 4, bounds: CGRect(x: 0, y: 0, width: 1, height: 1), format: .RGBA8, colorSpace: nil)
        
        return Color(red: Double(bitmap[0]) / 255, green: Double(bitmap[1]) / 255, blue: Double(bitmap[2]) / 255)
    }
}

但是当我将源图像旋转为左右时，函数仍然返回白色（原始底部）而不是预期的灰色。

我该如何解决？

这在我的情况下不起作用。 UIImage 在转换为 CIImage 时失去方向，然后转换为 CGImage 并返回 UIImage

4

0 回答 0