如何为 Mac OS 截屏 NSView？

Question

我目前有一个功能可以在 Mac 应用程序中创建整个窗口的屏幕截图。但是，我想针对特定的视图。我知道在 iOS 中您可以执行以下操作：

extension WKWebView {

    var screenshot: UIImage {
        if #available(iOS 10.0, *) {
            return UIGraphicsImageRenderer(size: bounds.size).image { _ in
                drawHierarchy(in: CGRect(origin: .zero, size: bounds.size), afterScreenUpdates: true)
            }
        } else {
            UIGraphicsBeginImageContext(bounds.size)
            drawHierarchy(in: bounds, afterScreenUpdates: true)
            let image = UIGraphicsGetImageFromCurrentImageContext()!
            return image
        }
    }
}

如何使用 AppKit 而不是 UIKit 将其转换为 Mac OS。以下是我当前计算的整个窗口截图的属性。

extension WKWebView {

    var screenshot: NSImage {
        let displayID = CGWindowID() // cgmaindisplayid
        let imageRef = CGDisplayCreateImage(displayID)
        return NSImage(cgImage: imageRef!, size: (NSScreen.main?.frame.size)!)
    }
}

Answer 1

你可以使用NSView方法func dataWithPDF(inside rect: NSRect) -> Data

extension NSView {
    var image: NSImage? {
        return NSImage(data: dataWithPDF(inside: bounds))
    }
}