如何优化iOS中基于ImageView特定区域颜色设置对比色的代码？

核心问题分析 #

你的代码当前直接使用子视图在UIImageView中的frame去裁剪原图，但忽略了contentMode的影响——当图片没有填满UIImageView时，图片在View内的实际显示区域和View的bounds并不重合，导致裁剪出来的区域和Label实际覆盖的图片区域完全错位。

优化方案与代码实现 #

步骤1：计算图片在ImageView中的实际显示区域 #

首先需要根据contentMode算出图片在UIImageView内的真实展示frame，这一步是解决错位的关键。

步骤2：坐标系转换修正 #

将子视图的区域从UIImageView的坐标系，转换到原图的像素坐标系，这样裁剪出来的区域才是Label真正覆盖的图片部分。

完整优化代码 #

// 扩展UIImageView，添加计算图片实际显示区域的方法
extension UIImageView {
    func imageDisplayFrame() -> CGRect {
        guard let image = image else { return bounds }
        let imageSize = image.size
        let viewSize = bounds.size
        
        let widthRatio = viewSize.width / imageSize.width
        let heightRatio = viewSize.height / imageSize.height
        
        var displaySize: CGSize
        switch contentMode {
        case .scaleToFill:
            displaySize = viewSize
        case .scaleAspectFit, .scaleAspectFill, .center, .top, .bottom, .left, .right, .topLeft, .topRight, .bottomLeft, .bottomRight:
            let scale = contentMode == .scaleAspectFill ? max(widthRatio, heightRatio) : min(widthRatio, heightRatio)
            displaySize = CGSize(width: imageSize.width * scale, height: imageSize.height * scale)
        @unknown default:
            displaySize = viewSize
        }
        
        // 计算居中或对齐后的frame
        let x = (viewSize.width - displaySize.width) / 2
        let y = (viewSize.height - displaySize.height) / 2
        return CGRect(x: x, y: y, width: displaySize.width, height: displaySize.height)
    }
}

// 修正后的获取主色调方法
func getDominantColor(from subview: UIView, in backgroundImageView: UIImageView) -> UIColor? {
    guard let image = backgroundImageView.image, let cgImage = image.cgImage else {
        return nil
    }
    
    // 确保布局完成
    backgroundImageView.layoutIfNeeded()
    subview.layoutIfNeeded()
    
    // 1. 获取图片在ImageView中的实际显示区域
    let imageDisplayFrame = backgroundImageView.imageDisplayFrame()
    let imageSize = image.size
    let imageViewSize = backgroundImageView.bounds.size
    
    // 2. 将子视图的frame转换到ImageView坐标系
    let subviewFrameInImageView = subview.convert(subview.bounds, to: backgroundImageView)
    
    // 3. 将ImageView坐标系下的子视图区域，转换为原图的像素坐标系
    // 计算比例：原图像素尺寸 / ImageView中图片的显示尺寸
    let scaleX = imageSize.width / imageDisplayFrame.width
    let scaleY = imageSize.height / imageDisplayFrame.height
    
    // 计算子视图区域相对于图片显示区域的偏移
    let subviewXInImage = (subviewFrameInImageView.minX - imageDisplayFrame.minX) * scaleX
    let subviewYInImage = (subviewFrameInImageView.minY - imageDisplayFrame.minY) * scaleY
    let subviewWidthInImage = subviewFrameInImageView.width * scaleX
    let subviewHeightInImage = subviewFrameInImageView.height * scaleY
    
    // 注意：CGImage的坐标系是原点在左下角，需要翻转Y轴
    let cgImageY = imageSize.height - (subviewYInImage + subviewHeightInImage)
    let cropRect = CGRect(x: subviewXInImage, y: cgImageY, width: subviewWidthInImage, height: subviewHeightInImage)
    
    // 裁剪图片
    guard let croppedCGImage = cgImage.cropping(to: cropRect) else {
        return nil
    }
    
    let croppedImage = UIImage(cgImage: croppedCGImage)
    return croppedImage.getDominantColor()
}

// 优化setContrastingColor方法：减少重复逻辑，避免强制解包
extension UIView {
    func setContrastingColor(basedOn backgroundColor: UIColor?) {
        guard let bgColor = backgroundColor else {
            // 默认 fallback 颜色
            if let label = self as? UILabel {
                label.textColor = .black
            } else if let imageView = self as? UIImageView {
                imageView.tintColor = .black
            }
            return
        }
        
        let isDark = bgColor.isDarkColor()
        let contrastColor: UIColor = isDark ? .white : .black
        
        if let label = self as? UILabel {
            label.textColor = contrastColor
        } else if let imageView = self as? UIImageView {
            imageView.tintColor = contrastColor
        }
    }
}

调用方式优化 #

// 去掉强制解包，增加安全判断
if let dominantColor = getDominantColor(from: self.participantImageView, in: self.eventImageView) {
    self.participantImageView.setContrastingColor(basedOn: dominantColor)
} else {
    // 处理获取颜色失败的情况，设置默认值
    self.participantImageView.setContrastingColor(basedOn: nil)
}

其他优化点说明 #

• 避免强制解包：原代码中直接使用dominantColor!，一旦获取失败会崩溃，优化后增加了安全判断和默认 fallback。
• 减少重复逻辑：setContrastingColor方法通过提取公共逻辑，避免了两次判断isDarkColor()的重复代码。
• 明确方法命名：将getDominantColorFromSubview改为getDominantColor(from:in:)，更符合Swift命名规范。
• 坐标系适配：处理了CGImage坐标系（左下角原点）和UIKit坐标系（左上角原点）的差异，确保裁剪区域正确。

内容的提问来源于stack exchange，提问作者Melodias