概述

• 音視頻采集是直播架構的第一環，是視頻的來源
  • 其實視頻的采集有多個應用場景：比如二維碼開發
• 音視頻采集包括兩部分：
  • 視頻采集
  • 音頻采集
• 在iOS開發中，是可以同步采集視頻&音頻的，使用方式也非常簡單
• 相關的采集API都封裝在AVFoundation框架中，導入對應框架，實現功能即可

采集步驟文字描述

• 導入框架

  • AVFoundation框架

• 創建捕捉會話（AVCaptureSession）

  • 該會話用于連接之后的輸入源&輸出源
  • 輸入源：攝像頭&話筒
  • 輸出源：拿到對應的音頻&視頻數據的出口
  • 會話：用于將輸入源&輸出源連接起來

• 設置視頻輸入源&輸出源

  • 輸入源（AVCaptureDeviceInput）：從攝像頭輸入
  • 輸出源（AVCaptureVideoDataOutput）：可以設置代理，在代理方法中拿到數據
  • 將輸入&輸出添加到會話中

• 設置音頻輸入源&輸出源

  • 輸入源（AVCaptureDeviceInput）：從話筒輸入
  • 輸出源（AVCaptureAudioDataOutput）：可以設置代理，在代理方法中拿到數據
  • 將輸入&輸出添加到會話中

• 添加預覽圖層（可選）

  • 如果希望用戶看到采集的畫面，可以添加預覽圖層
  • 該預覽圖層不是必須的，及時沒有添加也可以正常采集數據

• 采集

  • 調用會話（AVCaptureSession）的startRunning方法即可開始采集
  • 調用會話（AVCaptureSession）的stopRunning方法即可結束采集

代碼實現

• 整體步驟

  //創建捕捉會話 - 懶加載
  fileprivate lazy var session :AVCaptureSession = AVCaptureSession()
  override func viewDidLoad() {
        super.viewDidLoad()
        
        //1.初始化視頻輸入輸出
        setupVideoInputOutput()
        //2.初始化音頻輸入輸出
        setupAudioInputOutput()
        //3.初始化預覽圖層
        setupPreviewLayer()
    }

• 設置視頻輸入源&輸出源

 //1.添加視頻的輸入
guard let devices = AVCaptureDevice.devices() as?[AVCaptureDevice] else{return}
guard let device = devices.filter({$0.position == .front}).first else {return}
guard let input = try? AVCaptureDeviceInput(device: device) else {return}
//2.添加視頻的輸出
let output = AVCaptureVideoDataOutput()
output.setSampleBufferDelegate(self, queue: DispatchQueue.global())
videoconnection = output.connection(withMediaType: AVMediaTypeVideo)
 //3.添加輸入輸出
if session.canAddInput(input) {
    session.addInput(input)
}
if session.canAddOutput(output){
    session.addOutput(output)
}

//完成配置
session.commitConfiguration()

• 設置音頻輸入源&輸出源

//1.創建輸入
guard let device = AVCaptureDevice.defaultDevice(withMediaType: AVMediaTypeAudio) else { return }
guard let input = try?AVCaptureDeviceInput(device:device) else { return }

//2.創建輸出
let output = AVCaptureAudioDataOutput()
output.setSampleBufferDelegate(self, queue: DispatchQueue.global())
//3.添加輸入輸出
if session.canAddInput(input) {
    session.addInput(input)
}
if session.canAddOutput(output){
    session.addOutput(output)
}

//完成配置
session.commitConfiguration()

• 添加預覽圖層

//創建預覽圖層
guard let previewLayer = AVCaptureVideoPreviewLayer(session: session) else { return }

//設置previewLayer的屬性
previewLayer.frame = view.bounds

//將圖層添加到控制器的view的layer中
view.layer.insertSublayer(previewLayer, at: 0)

• 采集

@IBAction func startCapture(_ sender: Any) {
    session.startRunning()
    
}

@IBAction func endCapture(_ sender: Any) {
    session.stopRunning()
}

• 鏡頭切換

//存儲設備的input
fileprivate var videoInput: AVCaptureDeviceInput?
//切換攝像頭
@IBAction func chageCamera(_ sender: Any) {
    //1.取出之前鏡頭的方向
    guard let videoInput = videoInput else { return}
    let oritation: AVCaptureDevicePosition = videoInput.device.position == .front ? .back : .front
    guard let devices = AVCaptureDevice.devices() as?[AVCaptureDevice] else{return}
    guard let device = devices.filter({$0.position == oritation}).first else {return}
    guard let newInput = try? AVCaptureDeviceInput(device: device) else {return}
    //移除之前的input 添加新的input
    session.beginConfiguration()
    session.removeInput(videoInput)
    if session.canAddInput(newInput) {
        session.addInput(newInput)
        self.videoInput = newInput
    }
    session.commitConfiguration()
}

寫入文件

• 寫入文件步驟

  • 創建AVCaptureMovieFileOutput對象
  • 用于將音頻視頻寫入文件
  • 將movieFileOutput對象，添加到session的輸出中
    • 寫入文件也是一種輸出
  • 設置視頻的穩定模式
    • 不設置可能會出現視頻跳幀等問題
    • 通常設置為自動即可
  • 開始寫入
  • 錄制完成，停止寫入即可

• 寫入文件代碼實現

func 開始采集(){
  session?.startRunning()      
  //在session運行后開始錄制視頻 存儲
  setupMovieFileOutput()
}
func 結束采集(){
  fileOutput?.stopRecording()
}
//1.創建寫入文件的輸出
fileOutput = AVCaptureMovieFileOutput()
// 獲取視頻的connection
let connection = fileOutput!.connection(withMediaType: AVMediaTypeVideo)
// 設置視頻的穩定模式
connection?.automaticallyAdjustsVideoMirroring = true
//        fileOutput
//2.開始寫入文件
let filePath = NSSearchPathForDirectoriesInDomains(.documentDirectory, .userDomainMask, true).first! + "/abc.mp4"
let fileUrl = URL(fileURLWithPath: filePath)
if session!.canAddOutput(fileOutput){
    session?.addOutput(fileOutput)
}

fileOutput?.startRecording(toOutputFileURL: fileUrl, recordingDelegate: self)

• 代理方法中監聽開始結束事件

extension ViewController : AVCaptureFileOutputRecordingDelegate {
    func capture(_ captureOutput: AVCaptureFileOutput!, didStartRecordingToOutputFileAt fileURL: URL!, fromConnections connections: [Any]!) {
        print("開始錄制")
    }

    func capture(_ captureOutput: AVCaptureFileOutput!, didFinishRecordingToOutputFileAt outputFileURL: URL!, fromConnections connections: [Any]!, error: Error!) {
        print("停止錄制")
    }
}

demo地址