从UIScreen中获取快照

Question

我想从视图中获取快照，包括状态栏。

下面的代码在模拟器中工作，但不适用于物理设备。

extension UIScreen {

    class func screenshot() -> UIImage {
        let view = main.snapshotView(afterScreenUpdates: false)

        UIGraphicsBeginImageContext(view.bounds.size)
        view.drawHierarchy(in: view, afterScreenUpdates: true)

        let screenshot = UIGraphicsGetImageFromCurrentImageContext()
        UIGraphicsEndImageContext()
        return screenshot!
    }
}

Answer 1

要解决问题，可以使用以下函数：

func takeScreenshot() -> UIImage? {
    var screenshotImage :UIImage?
    let layer = UIApplication.shared.keyWindow!.layer
    let scale = UIScreen.main.scale
    UIGraphicsBeginImageContextWithOptions(layer.frame.size, false, scale);
    guard let context = UIGraphicsGetCurrentContext() else {return nil}
    layer.render(in:context)
    screenshotImage = UIGraphicsGetImageFromCurrentImageContext()
    UIGraphicsEndImageContext()

    return screenshotImage
}

我刚刚测试了这个，它能在一个物理设备上工作。

Answer 2

创建一个扩展如下：

extension UIApplication {

    func getViewScreenShot(vc: UIViewController) -> UIImage? {

        let scale = UIScreen.main.scale
        let bounds = vc.view.bounds
        UIGraphicsBeginImageContextWithOptions(bounds.size, false, scale)
        if UIGraphicsGetCurrentContext() != nil {
            vc.view.drawHierarchy(in: bounds, afterScreenUpdates: true)
            let screenshot = UIGraphicsGetImageFromCurrentImageContext()
            UIGraphicsEndImageContext()
            return screenshot!
        }
        return nil
    }
}

并称之为：

weak var weakSelf = self
let screenShotImage: UIImage = UIApplication.shared.getViewScreenShot(vc: weakSelf!)