swift 制作一个获取当前帧的函数

在Swift中，获取当前帧可以使用AVFoundation库中的AVCaptureVideoDataOutput类来实现。以下是一个获取当前帧的函数示例：

import AVFoundation

func captureCurrentFrame() -> UIImage? {
    let captureSession = AVCaptureSession()
    captureSession.sessionPreset = AVCaptureSession.Preset.photo

    // 获取设备
    guard let device = AVCaptureDevice.default(for: .video) else { return nil }

    // 添加输入
    do {
        let input = try AVCaptureDeviceInput(device: device)
        if captureSession.canAddInput(input) {
            captureSession.addInput(input)
        }
    } catch {
        print(error.localizedDescription)
        return nil
    }

    // 添加输出
    let output = AVCaptureVideoDataOutput()
    output.setSampleBufferDelegate(self, queue: DispatchQueue(label: "videoQueue"))
    if captureSession.canAddOutput(output) {
        captureSession.addOutput(output)
    }

    // 启动会话
    captureSession.startRunning()

    // 获取当前帧
    if let connection = output.connection(with: .video),
       let sampleBuffer = output.copyNextSampleBuffer(),
       let imageBuffer = CMSampleBufferGetImageBuffer(sampleBuffer) {
        let ciImage = CIImage(cvPixelBuffer: imageBuffer)
        let context = CIContext(options: nil)
        let cgImage = context.createCGImage(ciImage, from: ciImage.extent)
        let uiImage = UIImage(cgImage: cgImage!)
        return uiImage
    }
    return nil
}

这个函数使用AVCaptureSession创建一个视频捕获会话，并配置了一个AVCaptureVideoDataOutput来获取视频数据。在获取到视频数据后，它将其转换为CIImage对象，并将其转换为UIImage对象以供使用。