如何在xCode中访问iphone的锁屏和主屏幕图像？

func captureScreen() -> UIImage? {
    guard let window = UIApplication.shared.windows.first else {
        return nil
    }
    
    UIGraphicsBeginImageContextWithOptions(window.bounds.size, false, 0)
    window.drawHierarchy(in: window.bounds, afterScreenUpdates: false)
    let screenshot = UIGraphicsGetImageFromCurrentImageContext()
    UIGraphicsEndImageContext()
    
    return screenshot
}

if let screenshot = captureScreen() {
    // 在这里可以对截图进行进一步的处理或保存
}