Swift iOS 中的屏幕截图

在Swift iOS开发中，屏幕截图是一个非常常见的需求。屏幕截图允许我们捕捉当前屏幕上的内容并保存为图像，这对于创建教程、记录应用程序状态或者与其他应用程序共享内容非常有用。在本文中，我们将探讨如何使用Swift在iOS应用程序中实现屏幕截图功能，并提供一些实用的案例代码。

swift
import UIKit
func takeScreenshot() {
    let window = UIApplication.shared.windows.first // 获取应用程序的主窗口
    let screenshot = window?.snapshotView(afterScreenUpdates: true) // 获取屏幕截图视图
    
    if let image = screenshot?.toImage() { // 将视图转换为图像
        UIImageWriteToSavedPhotosAlbum(image, nil, nil, nil) // 保存图像到相册
    }
}
extension UIView {
    func toImage() -> UIImage? {
        UIGraphicsBeginImageContextWithOptions(bounds.size, true, UIScreen.main.scale)
        drawHierarchy(in: bounds, afterScreenUpdates: true)
        let image = UIGraphicsGetImageFromCurrentImageContext()
        UIGraphicsEndImageContext()
        return image
    }
}

swift
import UIKit
func takeCustomScreenshot() {
    let window = UIApplication.shared.windows.first
    let customRect = CGRect(x: 100, y: 100, width: 200, height: 200) // 自定义截图区域
    
    UIGraphicsBeginImageContextWithOptions(customRect.size, true, UIScreen.main.scale)
    window?.drawHierarchy(in: customRect, afterScreenUpdates: true)
    let image = UIGraphicsGetImageFromCurrentImageContext()
    UIGraphicsEndImageContext()
    
    if let image = image {
        UIImageWriteToSavedPhotosAlbum(image, nil, nil, nil)
    }
}

swift
import UIKit
class ViewController: UIViewController {
    
    @IBAction func takeScreenshotButtonTapped(_ sender: UIButton) {
        UIGraphicsBeginImageContextWithOptions(view.bounds.size, true, UIScreen.main.scale)
        view.drawHierarchy(in: view.bounds, afterScreenUpdates: true)
        let image = UIGraphicsGetImageFromCurrentImageContext()
        UIGraphicsEndImageContext()
        
        if let image = image {
            UIImageWriteToSavedPhotosAlbum(image, nil, nil, nil)
        }
    }
}