Генерация массива Float32 (данные Float32 PCM) с использованием CMSampleBuffer - PullRequest
Новая
       65

Генерация массива Float32 (данные Float32 PCM) с использованием CMSampleBuffer

0 голосов
/ 05 июня 2018

Я получаю обратные вызовы с камеры для аудио с данными в формате CMSampleBuffer, но я не могу преобразовать эти данные в данные PCM.

Я следовал документам, предоставленным Apple copyPCMData , UnsafeMutablePointer , AudioBufferList , но все, что я получил, это 0.0 в конце.

Вот мой код: 

private let pcmBufferPointer = UnsafeMutablePointer<AudioBufferList>.allocate(capacity: 1024)

init(....){
    //...
    let unsafeRawPointer = UnsafeMutableRawPointer.allocate(byteCount: 4, alignment: 0)
    let audioBuffer = AudioBuffer(mNumberChannels: 1, mDataByteSize: 4, mData: unsafeRawPointer)
    let audioBufferList = AudioBufferList(mNumberBuffers: 0, mBuffers: audioBuffer)
    self.pcmBufferPointer.initialize(repeating: audioBufferList, count: 1024)
}


//CMSampleBuffer obtained from AVCaptureAudioDataOutputSampleBufferDelegate
private func audioFrom(sampleBuffer: CMSampleBuffer) -> Void {
    let status = CMSampleBufferCopyPCMDataIntoAudioBufferList(sampleBuffer, 0, 1024, pcmBufferPointer)
    if status == 0 {
        Logger.log(key: "Audio Sample Buffer Status", message: "Buffer copied to pointer")
        let dataValue = pcmBufferPointer[0].mBuffers.mData!.load(as: Float32.self) //Tried with Int, Int16, Int32, Int64 and Float too
        Logger.log(key: "PCM Data Value", message: "Data value : \(dataValue)") //prints 0.0
    }else{
        Logger.log(key: "Audio Sample", message: "Buffer allocation failed with status \(status)")
    }
}

1 Ответ

0 голосов
/ 05 июня 2018

Наконец-то все заработало.

Пришлось добавить дополнительный шаг для преобразования указателя AudioBufferList в указатель AudioList 

if status == 0 {
    let inputDataPtr = UnsafeMutableAudioBufferListPointer(pcmBufferPointer)
    let mBuffers : AudioBuffer = inputDataPtr[0]
    if let bufferPointer = UnsafeMutableRawPointer(mBuffers.mData){
        let dataPointer = bufferPointer.assumingMemoryBound(to: Int16.self)
        let dataArray = Array(UnsafeBufferPointer.init(start: dataPointer, count: 1024))
        pcmArray.append(contentsOf: dataArray)
    }else{
        Logger.log(key: "Audio Sample", message: "Failed to generate audio sample")
    }
}else{
    Logger.log(key: "Audio Sample", message: "Buffer allocation failed with status \(status)")
}

Приведенный выше код работает только для одноканальных данных PCM,Для 2-канальных данных см. Следующую GIST - https://gist.github.com/hotpaw2/ba815fc23b5d642705f2b1dedfaf0107

...