如何修改代码截取UIImageView而非主视图的快照？

嘿，我来帮你搞定这个截图问题！你原来的代码是截取整个应用窗口的，但现在要只截指定的UIImageView，还遇到了黑屏的问题，咱们一步步调整：

问题根源分析 #

你之前的代码用了UIApplication.shared.keyWindow!.layer，这个在iOS 13+已经被废弃了，而且它截取的是整个窗口（包括那些按钮）。黑屏大概率是因为获取的窗口不对，或者没有正确渲染UIImageView的内容。

修改后的代码 #

直接把目标换成你的UIImageView的layer就行，同时修复窗口兼容问题，还要确保在主线程执行（UI操作必须在主线程）：

// 改成传入要截取的UIImageView，默认保存到相册
open func takeScreenshot(of imageView: UIImageView, shouldSave: Bool = true) -> UIImage? {
    // 确保在主线程执行，避免渲染异常
    guard Thread.isMainThread else {
        DispatchQueue.main.sync {
            return takeScreenshot(of: imageView, shouldSave: shouldSave)
        }
    }
    
    let scale = UIScreen.main.scale
    // 用UIImageView的bounds创建上下文，精准匹配目标视图尺寸
    UIGraphicsBeginImageContextWithOptions(imageView.bounds.size, false, scale)
    
    defer {
        UIGraphicsEndImageContext() // 确保上下文一定会被释放，避免内存泄漏
    }
    
    guard let context = UIGraphicsGetCurrentContext() else {
        print("获取图形上下文失败")
        return nil
    }
    
    // 渲染UIImageView的layer到上下文里
    imageView.layer.render(in: context)
    
    guard let screenshotImage = UIGraphicsGetImageFromCurrentImageContext() else {
        print("从上下文捕获图片失败")
        return nil
    }
    
    // 保存到相册（如果需要）
    if shouldSave {
        UIImageWriteToSavedPhotosAlbum(screenshotImage, nil, nil, nil)
    }
    
    return screenshotImage
}

使用方式 #

调用的时候直接传你的UIImageView实例就行：

// 假设你的UIImageView叫myImageView
let capturedImage = takeScreenshot(of: myImageView)

额外注意事项 #

• 如果你的UIImageView里有视频播放（比如AVPlayerLayer），单纯渲染layer可能拿不到实时帧，这时候需要用AVAssetImageGenerator来获取当前播放帧。
• 确保调用这个方法时，UIImageView已经完成布局（比如在viewDidAppear之后调用，而不是viewDidLoad），否则bounds可能不正确。
• 如果还是遇到黑屏，检查UIImageView的isHidden属性、alpha值，或者它是否被其他视图完全遮挡。

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