将捕获的图像存储到UIImage数组(AVFoundation)中

import AVFoundation

class ImageCaptureManager: NSObject, AVCaptureVideoDataOutputSampleBufferDelegate {
    var imageArray: [UIImage] = []
    var captureSession: AVCaptureSession?
    
    func startCapture() {
        captureSession = AVCaptureSession()
        
        guard let captureSession = captureSession else {
            return
        }
        
        guard let captureDevice = AVCaptureDevice.default(for: .video) else {
            return
        }
        
        do {
            let input = try AVCaptureDeviceInput(device: captureDevice)
            captureSession.addInput(input)
            
            let output = AVCaptureVideoDataOutput()
            output.setSampleBufferDelegate(self, queue: DispatchQueue.main)
            captureSession.addOutput(output)
            
            captureSession.startRunning()
        } catch {
            print("Error setting up capture session: \(error.localizedDescription)")
        }
    }
    
    func stopCapture() {
        captureSession?.stopRunning()
        captureSession = nil
    }
    
    func captureOutput(_ output: AVCaptureOutput, didOutput sampleBuffer: CMSampleBuffer, from connection: AVCaptureConnection) {
        guard let imageBuffer = CMSampleBufferGetImageBuffer(sampleBuffer) else {
            return
        }
        
        let ciImage = CIImage(cvPixelBuffer: imageBuffer)
        let context = CIContext()
        guard let cgImage = context.createCGImage(ciImage, from: ciImage.extent) else {
            return
        }
        
        let image = UIImage(cgImage: cgImage)
        imageArray.append(image)
    }
}

// 使用示例
let captureManager = ImageCaptureManager()
captureManager.startCapture()

// 在需要停止捕获时调用
captureManager.stopCapture()