无法仅获取在Swift中显示相机(AVCapturePhotoOutput)的UIView的屏幕截图

Question

我想保存显示相机视图的屏幕截图（avCapturePhotoOutput）。

我设置了cameraview（显示相机视图），并添加其他具有Uibutton等部分的Uiview。

但是，如果我保存它，则将图片保存到相机卷只是一个白色的视图。（当我试图保存自我时。

我该如何解决？

class ViewController: UIViewController, AVCapturePhotoCaptureDelegate{
    //for camera
    var captureDevice : AVCaptureDevice!
    var captureSesssion: AVCaptureSession!
    var stillImageOutput: AVCapturePhotoOutput?
    var previewLayer: AVCaptureVideoPreviewLayer?
    var videoConnection:AVCaptureConnection!
    var topView: TopView! //on cameraView. It has some buttons and imageViews.
    var cameraView:UIView! //that shows camera's view
    override func viewDidLoad() {
        super.viewDidLoad()
        self.view.backgroundColor = (UIColor.black)
        cameraView = UIView(frame: CGRect(x: 0.0, y: 0.0, width: self.view.frame.size.width, height: self.view.frame.size.height))
        self.view.addSubview(cameraView)
        captureSesssion = AVCaptureSession()
        stillImageOutput = AVCapturePhotoOutput()
        captureSesssion.sessionPreset = AVCaptureSessionPreset1280x720
        captureDevice = AVCaptureDevice.defaultDevice(withDeviceType: .builtInWideAngleCamera, mediaType: AVMediaTypeVideo, position: .front)
        do {
            let input = try AVCaptureDeviceInput(device: captureDevice)
            if (captureSesssion.canAddInput(input)) {
                captureSesssion.addInput(input)
                if (captureSesssion.canAddOutput(stillImageOutput)) {
                    captureSesssion.addOutput(stillImageOutput)
                    captureSesssion.startRunning()
                    previewLayer = AVCaptureVideoPreviewLayer(session: captureSesssion)
                    previewLayer?.videoGravity = AVLayerVideoGravityResizeAspect 
                    previewLayer?.connection.videoOrientation = AVCaptureVideoOrientation.portrait 
                    previewLayer?.connection.automaticallyAdjustsVideoMirroring = false
                    previewLayer?.connection.isVideoMirrored = true
                    cameraView.layer.addSublayer(previewLayer!)
                    previewLayer?.position = CGPoint(x: self.view.frame.width / 2, y: self.view.frame.height / 2)
                    previewLayer?.bounds = self.view.frame
                }
            }
        }
        catch {
            print(error)
        }
        topView = TopView.init(frame: CGRect(x: 0, y: 0, width: 320, height: 568))
        topView.viewController = self
        self.view.addSubview(topView)
    }
    override func touchesBegan(_ touches: Set<UITouch>, with event: UIEvent?) {
        let touch = touches.first!
        captureSesssion.stopRunning() //for taking still picture
        let layer = UIApplication.shared.keyWindow?.layer
        let scale = UIScreen.main.scale
        UIGraphicsBeginImageContextWithOptions((layer?.frame.size)!, false, scale);
        cameraView.layer.render(in: UIGraphicsGetCurrentContext()!)
        let screenshot = UIGraphicsGetImageFromCurrentImageContext()
        UIGraphicsEndImageContext()
        //save the screenshot to camera roll        
        UIImageWriteToSavedPhotosAlbum(screenshot!, nil, nil, nil);
      }
}

更新：

添加了制作CashApelayer的代码，但它无法显示相机视图，并且完成保存的图像后，图像仅为白色图片....

cameraView = UIView(frame: CGRect(x: 0.0, y: 0.0, width: self.view.frame.size.width, height: self.view.frame.size.height))
self.view.addSubview(cameraView)
let cropRectOverlay = CAShapeLayer() //added
cameraView.layer.mask = cropRectOverlay //added
cameraView.layer.addSublayer(cropRectOverlay) //added
...
previewLayer?.connection.automaticallyAdjustsVideoMirroring = false
previewLayer?.connection.isVideoMirrored = true
cameraView.layer.addSublayer(cropRectOverlay) //added
...
UIGraphicsBeginImageContextWithOptions(cameraView.bounds.size, false, 0); //added 
cameraView.layer.render(in: UIGraphicsGetCurrentContext()!) //added
let image = UIGraphicsGetImageFromCurrentImageContext(); //added
UIGraphicsEndImageContext(); //added
//When added resultImageView, it shows black view.          
//let resultImageView = UIImageView(frame: self.view.frame)
//resultImageView.image = image
//self.view.addSubview(resultImageView)
UIImageWriteToSavedPhotosAlbum(image!, nil, nil, nil);

Answer 1

捕获会话的图像

let stillImageOutput = AVCaptureStillImageOutput()
stillImageOutput.outputSettings = [AVVideoCodecKey:AVVideoCodecJPEG]
if captureSession.canAddOutput(stillImageOutput) {
      captureSession.addOutput(stillImageOutput)
}

func captureImage() {
     let videoConnection = stillImageOutput.connection(withMediaType: AVMediaTypeVideo)
     stillImageOutput.captureStillImageAsynchronously(from: videoConnection, completionHandler: {(ingDataBuffer, error) in
         let imageData = AVCaptureStillImageOutput.jpegStillImageNSDataRepresentation(ingDataBuffer)
         self.imgView.image = UIImage(data: imageData!)
    })
}

uiview的屏幕截图

UIGraphicsBeginImageContextWithOptions(cameraView.bounds.size, false, 0);
cameraView.layer.render(in: UIGraphicsGetCurrentContext()!)
let image = UIGraphicsGetImageFromCurrentImageContext();
UIGraphicsEndImageContext();

avcapturephotocapturedelegate

将委托给自我

func capture(_ captureOutput: AVCapturePhotoOutput, didFinishProcessingPhotoSampleBuffer photoSampleBuffer: CMSampleBuffer?, previewPhotoSampleBuffer: CMSampleBuffer?, resolvedSettings: AVCaptureResolvedPhotoSettings, bracketSettings: AVCaptureBracketedStillImageSettings?, error: NSError?) {
    if let sampleBuffer = photoSampleBuffer, let previewBuffer = previewPhotoSampleBuffer, let dataImage = AVCapturePhotoOutput.jpegPhotoDataRepresentation(forJPEGSampleBuffer: sampleBuffer, previewPhotoSampleBuffer: previewBuffer) {
      print(image: UIImage(data: dataImage).size)
    }
}