Swift-AVCapture視頻采集

import UIKit
import AVFoundation

protocol CaptureManagerDelegate: AnyObject {
    func processSampleBuffer(sampleBuffer: CMSampleBuffer, type: AVMediaType)
}

class ZGCapture: NSObject {
    
    weak var delegate: CaptureManagerDelegate?
    
    /// 最小感光度ISO
    var minISO: Float {
        return currentDevice?.activeFormat.minISO ?? 0
    }

    /// 最大感光度ISO
    var maxISO: Float {
        return currentDevice?.activeFormat.maxISO ?? 0
    }

    /// 最小快門速度S
    var minExposureDuration: CMTime {
        return currentDevice?.activeFormat.minExposureDuration ?? .zero
    }

    /// 最大快門速度S
    var maxExposureDuration: CMTime {
        return currentDevice?.activeFormat.maxExposureDuration ?? .zero
    }
    
    var minExposureTargetBias: Float {
        return currentDevice?.minExposureTargetBias ?? .zero
    }

    var maxExposureTargetBias: Float {
        return currentDevice?.maxExposureTargetBias ?? .zero
    }
    
    /// 當前縮放
    var zoom: CGFloat {
        return currentDevice?.videoZoomFactor ?? 1
    }
    
    /// 最大縮放
    var maxZoom: CGFloat {
        return currentDevice?.maxAvailableVideoZoomFactor ?? 1
    }

    /// 最小縮放
    var minZoom: CGFloat {
        return currentDevice?.minAvailableVideoZoomFactor ?? 1
    }
    
    // MARK: - 曝光補償
    var exposureTargetBias: Float {
        return currentDevice?.exposureTargetBias ?? .zero
    }
    
    private var autoExposureDuration: CMTime = .zero
    
    private(set) var videoDevices = [AVCaptureDevice]()
    private(set) var currentDevice: AVCaptureDevice?
    private var captureSession = AVCaptureSession()
    private var captureConnection: AVCaptureConnection?
    private var currentVideoInput: AVCaptureDeviceInput?
    private var videoQueue = DispatchQueue(label: "videoQueue")
    private var previewLayer: AVCaptureVideoPreviewLayer?
    private var currentBackDeviceIndex = 0
    
    override init() {
        super.init()
        getVideoDevices()
        initCapture(deviceIndex: 0)
    }
    
    func setPreview(preview: UIView) {
        previewLayer?.removeFromSuperlayer()
        previewLayer = AVCaptureVideoPreviewLayer(session: captureSession)
        previewLayer?.frame = preview.bounds
        previewLayer?.videoGravity = .resizeAspectFill
        preview.layer.insertSublayer(previewLayer!, at: 0)
    }
    
    func startRecordVideo() {
        startCapture()
    }
    
    func stopRecordVideo(){
        stopCapture()
    }
    
    private func startCapture() {
        if captureSession.isRunning {
            captureSession.stopRunning()
        }
        videoQueue.async {
            self.captureSession.startRunning()
        }
    }

    // 結束采集
    private func stopCapture() {
        if captureSession.isRunning {
            captureSession.stopRunning()
        }
    }

    private func getVideoDevices(){
        //廣角、長焦
        var deviceTypes:[AVCaptureDevice.DeviceType] = [.builtInWideAngleCamera, .builtInTelephotoCamera]
        if #available(iOS 10.2, *) {
            //雙攝廣角
            deviceTypes.append(.builtInDualCamera)
        }
        if #available(iOS 11.1, *) {
            //組合
            deviceTypes.append(.builtInTrueDepthCamera)
        }
        if #available(iOS 13.0, *) {
            //超廣角、超廣角+廣角、超廣角+廣角+長焦
            deviceTypes += [.builtInUltraWideCamera, .builtInDualWideCamera, .builtInTripleCamera]
        }
        let deviceSession = AVCaptureDevice.DiscoverySession(deviceTypes: deviceTypes, mediaType: .video, position: .back)
        videoDevices = deviceSession.devices
    }
    
    ///選擇或切換后置鏡頭
    func initCapture(deviceIndex: Int){
        guard deviceIndex < videoDevices.count else {
            return
        }
        let videoDevice = videoDevices[deviceIndex]
        currentBackDeviceIndex = deviceIndex
        DispatchQueue.global().async {
            self.initDevice(device: videoDevice)
        }
    }
    
    ///切換前后置攝像頭
    func switchPosition(position: AVCaptureDevice.Position){
        if position == .front,
           let device = AVCaptureDevice.DiscoverySession(deviceTypes: [.builtInWideAngleCamera], mediaType: .video, position: .front).devices.first  {
            setTorchMode(mode: .off)
            initDevice(device: device)
        }
        else{
            initCapture(deviceIndex: currentBackDeviceIndex)
        }
    }
    

    private func initDevice(device: AVCaptureDevice){
        let isRestoration = captureSession.isRunning
        stopCapture()
        // 設置輸入源
        guard let videoInput = try? AVCaptureDeviceInput(device: device) else {
            return
        }
        // 記錄當前采集設備
        currentDevice = device
        autoExposureDuration = device.exposureDuration
        
        // 設置輸出源
        let videoOutput = AVCaptureVideoDataOutput()
        videoOutput.setSampleBufferDelegate(self, queue: videoQueue)
        // 拋棄過期幀
        videoOutput.alwaysDiscardsLateVideoFrames = true
        // 設置輸出格式
        videoOutput.videoSettings = [
            kCVPixelBufferPixelFormatTypeKey as String: Int(kCVPixelFormatType_420YpCbCr8BiPlanarVideoRange),
        ]
        
        captureSession.beginConfiguration()
        //移除舊源
        if let currentVideoDeviceInput = captureSession.inputs.first as? AVCaptureDeviceInput {
            captureSession.removeInput(currentVideoDeviceInput)
        }
        if let currentVideoOutput = captureSession.outputs.first {
            captureSession.removeOutput(currentVideoOutput)
        }
        //添加源
        if captureSession.canAddInput(videoInput) {
            captureSession.addInput(videoInput)
            currentVideoInput = videoInput
        }
        if captureSession.canAddOutput(videoOutput) {
            captureSession.addOutput(videoOutput)
        }
        captureSession.commitConfiguration()

        captureConnection = videoOutput.connection(with: AVMediaType.video)
        // 視頻錄制方向
        setupVideoOrientation(orientation: .portrait)
        
        if isRestoration {
            startCapture()
        }
    }
    
    private func setupVideoOrientation(deviceOrientation: UIDeviceOrientation) {
        // 設置視頻方向
        var videoOrientation: AVCaptureVideoOrientation = .portrait
        switch deviceOrientation {
            case .portrait:
                videoOrientation = .portrait
            case .portraitUpsideDown:
                videoOrientation = .portraitUpsideDown
            case .landscapeLeft:
                videoOrientation = .landscapeLeft
            case .landscapeRight:
                videoOrientation = .landscapeRight
            default:
                break
        }
        self.setupVideoOrientation(orientation: videoOrientation)
    }

    /// 設置視頻方向
    private func setupVideoOrientation(orientation: AVCaptureVideoOrientation) {
        if (self.captureConnection?.isVideoOrientationSupported ?? false) {
            captureConnection?.videoOrientation = orientation
        }
    }
    
//設置分辨率和幀率,因為耗時,在未設置完又beginConfiguration會出錯,所以加個Queue用于取消
    private let setResolutionAndFpsQueue = DispatchQueue(label: "serialQueue")
    private var workItem: DispatchWorkItem?
    func setResolutionAndFps(width: Int, height: Int, frameRate: Float64) {
        workItem?.cancel()
        workItem = DispatchWorkItem { [self] in
            guard let captureDevice = currentDevice else { return }
            let size = CGSize(width: width, height: height)
            for vFormat in captureDevice.formats {
                let maxRate = vFormat.videoSupportedFrameRateRanges.first?.maxFrameRate ?? 30
                guard maxRate >= frameRate else { continue }
                let description = vFormat.formatDescription
                let dims = CMVideoFormatDescriptionGetDimensions(description)
                //分辨率
                if dims.width == Int32(size.width) && dims.height == Int32(size.height) {
                    //幀率
                    captureSession.beginConfiguration()
                    do {
                        try captureDevice.lockForConfiguration()
                        captureDevice.activeFormat = vFormat
                        captureDevice.activeVideoMinFrameDuration = CMTimeMake(value: 1, timescale: Int32(frameRate))
                        captureDevice.activeVideoMaxFrameDuration = CMTimeMake(value: 1, timescale: Int32(frameRate))
                        captureDevice.unlockForConfiguration()
                        captureSession.commitConfiguration()
                        break
                    } catch {
                        captureSession.commitConfiguration()
                    }
                }
            }
            print("設置分辨率:\(size) 幀率:\(frameRate)")
        }
        setResolutionAndFpsQueue.async(execute: workItem!)
    }
    
    /// 設置ISO/快門自動
    func setAutoMode() {
        guard let currentDevice = currentDevice,
              currentDevice.isExposureModeSupported(.autoExpose)
        else { return }

        try? currentDevice.lockForConfiguration()
        currentDevice.setExposureModeCustom(duration: autoExposureDuration, iso: currentDevice.iso) { [weak currentDevice] _ in
            currentDevice?.exposureMode = .autoExpose
            currentDevice?.unlockForConfiguration()
        }
    }
    /// 感光度?ISO
    func setISO(value: Float) {
        guard let currentDevice = currentDevice else {
            return
        }
        try? currentDevice.lockForConfiguration()
        let clampedISO = min(max(value, minISO), maxISO) // 確保ISO值在范圍內
        currentDevice.setExposureModeCustom(duration: AVCaptureDevice.currentExposureDuration, iso: clampedISO, completionHandler: nil)
        currentDevice.unlockForConfiguration()
    }
    /// 快門速度
    func setShutterSpeed(value: Float) {
        guard let currentDevice = currentDevice else {
            return
        }
        let maxValue = maxExposureDuration.seconds
        let minValue = minExposureDuration.seconds
        let value = Double(value)

        var duration: CMTime
        if value < minValue {
            duration = minExposureDuration
        } else if value > maxValue {
            duration = maxExposureDuration
        } else {
            let scale: Int32 = 10000
            let v = max(Int64(value * Double(scale)), 1)
            duration = CMTimeMake(value: v, timescale: scale)
        }

        try? currentDevice.lockForConfiguration()
        currentDevice.setExposureModeCustom(duration: duration, iso: currentDevice.iso) { [weak currentDevice] _ in
            currentDevice?.exposureMode = .custom
            currentDevice?.unlockForConfiguration()
        }
    }
    
    // 對焦模式
    func setFocus(mode: AVCaptureDevice.FocusMode, point: CGPoint? = nil) {
        guard let currentDevice = currentDevice,
              currentDevice.isFocusModeSupported(mode)
        else { return }
        try? currentDevice.lockForConfiguration()
        if currentDevice.isFocusPointOfInterestSupported,
           let point = point,
           let castPoint = previewLayer?.captureDevicePointConverted(fromLayerPoint: point) {
            currentDevice.focusPointOfInterest = castPoint
        }
        currentDevice.focusMode = mode
        currentDevice.unlockForConfiguration()
    }
    
    func setFocus(lensPosition: Float) {
        guard let currentDevice = currentDevice,
              currentDevice.isLockingFocusWithCustomLensPositionSupported
        else { return }
        try? currentDevice.lockForConfiguration()
        currentDevice.setFocusModeLocked(lensPosition: lensPosition)
        currentDevice.unlockForConfiguration()
    }
    
    // 設置曝光量 EV (-8 -- 8)
    func setExposure(value: Float) {
        guard let currentDevice = currentDevice,
              currentDevice.isExposureModeSupported(.locked)
        else { return }
        try? currentDevice.lockForConfiguration()
        currentDevice.setExposureTargetBias(value)
        currentDevice.unlockForConfiguration()
    }
    
    /// 設置白平衡模式 WB
    func setWhiteBalance(mode: AVCaptureDevice.WhiteBalanceMode) {
        guard let currentDevice = currentDevice,
              currentDevice.isWhiteBalanceModeSupported(mode)
        else { return }
        try? currentDevice.lockForConfiguration()
        currentDevice.whiteBalanceMode = mode
        currentDevice.unlockForConfiguration()
    }

    /// 設置白平衡量 WB
    func setWhiteBalance(temperature: Float) {
        guard let currentDevice = currentDevice,
              currentDevice.isWhiteBalanceModeSupported(.locked)
        else { return }
        
        let temperatureAndTintValues = AVCaptureDevice.WhiteBalanceTemperatureAndTintValues(temperature: temperature, tint: 0)
        let whiteBalanceGains = currentDevice.deviceWhiteBalanceGains(for: temperatureAndTintValues)

        let maxWhiteBalanceGain = currentDevice.maxWhiteBalanceGain
        var fixWhiteBalanceGains = whiteBalanceGains
        fixWhiteBalanceGains.redGain = max(1.0, min(maxWhiteBalanceGain, whiteBalanceGains.redGain))
        fixWhiteBalanceGains.greenGain = max(1.0, min(maxWhiteBalanceGain, whiteBalanceGains.greenGain))
        fixWhiteBalanceGains.blueGain = max(1.0, min(maxWhiteBalanceGain, whiteBalanceGains.blueGain))
        
        try? currentDevice.lockForConfiguration()
        currentDevice.setWhiteBalanceModeLocked(with: fixWhiteBalanceGains)
        currentDevice.unlockForConfiguration()
    }
    ///縮放
    func setZoom(factor: CGFloat){
        guard let currentDevice = currentDevice else { return }

        let minZoom = currentDevice.minAvailableVideoZoomFactor
        let maxZoom = currentDevice.maxAvailableVideoZoomFactor
        let zoom = min(maxZoom, max(factor, minZoom))
        try? currentDevice.lockForConfiguration()
        currentDevice.ramp(toVideoZoomFactor: zoom, withRate: 4.0)
        currentDevice.unlockForConfiguration()
    }
    
    ///閃光燈
    func setTorchMode(mode: AVCaptureDevice.TorchMode){
        guard let currentDevice = currentDevice else { return }
        try? currentDevice.lockForConfiguration()
        currentDevice.torchMode = mode
        currentDevice.unlockForConfiguration()
    }
    
}

extension ZGCapture: AVCaptureVideoDataOutputSampleBufferDelegate, AVCaptureAudioDataOutputSampleBufferDelegate {
    func captureOutput(_ output: AVCaptureOutput, didOutput sampleBuffer: CMSampleBuffer, from connection: AVCaptureConnection) {
        
        if connection == captureConnection {
            // sampleBuffer 就是我們拿到的畫面,美顏等操作都是對 sampleBuffer 進行的
//            print("已經采集視頻—-video")
            
            delegate?.processSampleBuffer(sampleBuffer: sampleBuffer, type: .video)
            
        } else {
            print("已經采集音頻--audio")
        }
        
    }
}

最后編輯于
?著作權歸作者所有,轉載或內容合作請聯系作者
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發布,文章內容僅代表作者本人觀點,簡書系信息發布平臺,僅提供信息存儲服務。

推薦閱讀更多精彩內容