iOS h264解码前数据处理

972 阅读2分钟

背景

解码器使用VideoToolBox硬解码。

H264是annex-b格式,以0x00000001作为起始码,h264数据结构

CocoaAsyncSocket建立TCP连接,接收H264数据。

待解决的问题

目前TCP每次传输过来的数据,有可能是0.5条,虽然以0x00000001开头,但是数据不完整。 有可能是1.5条、2条等。

那么就需要将每次传输过来的数据,拆分成1条1条完整的数据给解码器。

解决过程

刚开始的时候,我就单纯的用一个大数组,将所有的数据存起来,然后遍历分割保存到数组,最后遍历取出分割好的数据给解码器。虽然逻辑上说得通,但是效率很低,数据量太大特别耗费CPU、内存,最终因为太卡顿放弃了。

然后我就找有没有人也用tcp获取数据解析,终于找到一个答案

提供了CocoaAsyncSocket一个很关键的方法:

- (void)readDataToData:(NSData *)data withTimeout:(NSTimeInterval)timeout tag:(long)tag

读取到某个位置截止,那么我只需要读取到起始码0x00000001就可以实现数据分割了。

let startCode: [UInt8] = [0,0,0,1]

func socket(_ sock: GCDAsyncSocket, didRead data: Data, withTag tag: Int) {

    socket.readData(to: Data(startCode), withTimeout: -1, tag: 0)
}

func socket(_ sock: GCDAsyncSocket, didConnectToHost host: String, port: UInt16) {   
    socket.readData(withTimeout: -1, tag: 0)
}

第一次获取到的数据

有1.5条数据,第二条不完整

// 用10进制表示获取到的byte数组
0,0,0,1,2,3,4,5,6,0,0,0,1,8,4,5,6,7,7.....

第二条获取到的数据

有0.5条数据,需要和第一次获取的第二条数据合并成完整数据。

9,9,4,3,2,0,0,0,1

第三条及以后获取到的数据

都是以起始码结尾的数据,只需要将起始码拼接到前面就是完整的数据。

9,3,4,5,6,3,2,3,5,2,3,4,5,2,3,4,5,5,4,3,2,4,2,2,.....0,0,0,1

最后将切分好的数据,给解码器就可以了。

总结

  1. h264原始数据,需要切割才能给解码器
  2. 使用CocoaAsyncSocketreadData可以简化切割