如何捕获与SceneView屏幕亮度一致的快照

我太懂这个痛点了！做AR项目的时候几乎人人都会碰到默认快照偏暗的问题，毕竟sceneView.snapshot()默认的渲染逻辑和屏幕显示的流程不一样，没考虑到屏幕的颜色空间和实际显示增益。下面给你几个亲测有效的解决办法：

方案一：用drawHierarchy替代默认快照方法 #

这个方法会直接复刻屏幕上的渲染结果，能完美匹配你看到的亮度，是最省心的方案。修改你的代码如下：

@IBAction func ARSnapTapped(_ sender: Any) {
    if !draw {
        // 创建和屏幕同比例的绘图上下文
        UIGraphicsBeginImageContextWithOptions(sceneView.bounds.size, sceneView.isOpaque, UIScreen.main.scale)
        // 绘制SceneView的层级内容，afterScreenUpdates设为true确保捕获最新画面
        sceneView.drawHierarchy(in: sceneView.bounds, afterScreenUpdates: true)
        // 从上下文获取图片
        let newImg = UIGraphicsGetImageFromCurrentImageContext()
        // 结束上下文
        UIGraphicsEndImageContext()
        
        DispatchQueue.main.async {
            // 这里替换成你原本处理图片的逻辑，比如显示到imageView
            self.imageView.image = newImg
        }
    }
}

方案二：修正默认快照的颜色空间 #

如果你一定要用原生的snapshot()方法，可以手动把快照的线性颜色空间转换成屏幕使用的sRGB，就能修正亮度差：

@IBAction func ARSnapTapped(_ sender: Any) {
    if !draw {
        let rawSnapshot = sceneView.snapshot()
        
        // 获取sRGB颜色空间
        guard let srgbColorSpace = CGColorSpace(name: CGColorSpace.sRGB) else {
            print("无法获取sRGB颜色空间")
            return
        }
        
        // 基于sRGB创建绘图上下文，重新绘制快照
        if let cgRawImage = rawSnapshot.cgImage,
           let context = CGContext(data: nil,
                                   width: cgRawImage.width,
                                   height: cgRawImage.height,
                                   bitsPerComponent: 8,
                                   bytesPerRow: cgRawImage.bytesPerRow,
                                   space: srgbColorSpace,
                                   bitmapInfo: CGImageAlphaInfo.premultipliedLast.rawValue) {
            
            context.draw(cgRawImage, in: CGRect(x: 0, y: 0, width: cgRawImage.width, height: cgRawImage.height))
            
            if let correctedCgImage = context.makeImage() {
                let correctedImage = UIImage(cgImage: correctedCgImage, scale: rawSnapshot.scale, orientation: rawSnapshot.imageOrientation)
                
                DispatchQueue.main.async {
                    self.imageView.image = correctedImage
                }
            }
        }
    }
}

额外小提示：检查AR光照估计设置 #

如果你的AR场景启用了光照估计，确保配置里的isLightEstimationEnabled是开启的，这样快照时能正确使用当前环境光数据，避免因光照计算偏差导致的画面偏暗：

let configuration = ARWorldTrackingConfiguration()
// 开启光照估计
configuration.isLightEstimationEnabled = true
sceneView.session.run(configuration)

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