H264 获取SPS与PPS

转载地址

H264 获取SPS与PPS

在用Android手机进行h264硬编码的时候如果要进行视频流的实时传输与播放，就需要知道视频流的Sequence Parameter Sets (SPS) 和Picture Parameter Set (PPS)。

今天算是看明白如何获取SPS和PPS，在这里记录下来，希望有需要的朋友可以在这里获取到一些些的帮助。

首先说一下大前提，我设置的视频录制参数为：

mMediaRecorder.setOutputFormat(MediaRecorder.OutputFormat.THREE_GPP);

mMediaRecorder.setVideoEncoder(MediaRecorder.VideoEncoder.H264);

为了让大家更加明白，我先贴出avcC的数据结构：

    aligned(8) class AVCDecoderConfigurationRecord {  
       unsigned int(8) configurationVersion = 1;  
       unsigned int(8) AVCProfileIndication;  
       unsigned int(8) profile_compatibility;  
       unsigned int(8) AVCLevelIndication;  

       bit(6) reserved = '111111'b;  
       unsigned int(2) lengthSizeMinusOne;  

       bit(3) reserved = '111'b;  
       unsigned int(5) numOfSequenceParameterSets;  
       for (i=0; i< numOfSequenceParameterSets; i++) {  
          unsigned int(16) sequenceParameterSetLength ;  
          bit(8*sequenceParameterSetLength) sequenceParameterSetNALUnit;  
       }  

       unsigned int(8) numOfPictureParameterSets;  
       for (i=0; i< numOfPictureParameterSets; i++) {  
          unsigned int(16) pictureParameterSetLength;  
          bit(8*pictureParameterSetLength) pictureParameterSetNALUnit;  
       }  
    }  

ok，数据结构贴出来了，我再贴出录制的3gp输出类型的h264码流片断。

阴影部分就是avcC的全部数据了。

其中： 0x61 0x76 0x63 0x43 就是字符avcC

0x01 是configurationVersion

0x42 是AVCProfileIndication

0x00 是profile_compatibility

0x1F是AVCLevelIndication

0xFF 是6bit的reserved 和2bit的lengthSizeMinusOne

0xE1 是3bit的reserved 和5bit的numOfSequenceParameterSets

0x00 0x09是sps的长度为9个字节。

故SPS的内容为接下来的9个字节：67 42 00 1f e9 02 c1 2c 80

接下来的：01为numOfPictureParameterSets

0x00和0x04是pps的长度为4个字节。

故PPS的内容为接下来的4个字节：68 ce 06 f2

通过这段数据片断，就可以获取到SPS和PPS了。

下面我将贴上用java代码获取输出格式为3gp的h264码流的SPS与PPS代码：

    package cn.edu.xmu.zgy;  

    import java.io.File;  
    import java.io.FileInputStream;  
    import java.io.IOException;  

    public class ObtainSPSAndPPS {  

        public void getSPSAndPPS(String fileName) throws IOException {  
            File file = new File(fileName);  
            FileInputStream fis = new FileInputStream(file);  

            int fileLength = (int) file.length();  
            byte[] fileData = new byte[fileLength];  
            fis.read(fileData);  

            // 'a'=0x61, 'v'=0x76, 'c'=0x63, 'C'=0x43  
            byte[] avcC = new byte[] { 0x61, 0x76, 0x63, 0x43 };  

            // avcC的起始位置  
            int avcRecord = 0;  
            for (int ix = 0; ix < fileLength; ++ix) {  
                if (fileData[ix] == avcC[0] && fileData[ix + 1] == avcC[1]  
                        && fileData[ix + 2] == avcC[2]  
                        && fileData[ix + 3] == avcC[3]) {  
                    // 找到avcC，则记录avcRecord起始位置，然后退出循环。  
                    avcRecord = ix + 4;  
                    break;  
                }  
            }  
            if (0 == avcRecord) {  
                System.out.println("没有找到avcC，请检查文件格式是否正确");  
                return;  
            }  

            // 加6的目的是为了跳过  
            // (1)8字节的 configurationVersion  
            // (2)8字节的 AVCProfileIndication  
            // (3)8字节的 profile_compatibility  
            // (4)8 字节的 AVCLevelIndication  
            // (5)6 bit 的 reserved  
            // (6)2 bit 的 lengthSizeMinusOne  
            // (7)3 bit 的 reserved  
            // (8)5 bit 的numOfSequenceParameterSets  
            // 共6个字节，然后到达sequenceParameterSetLength的位置  
            int spsStartPos = avcRecord + 6;  
            byte[] spsbt = new byte[] { fileData[spsStartPos],  
                    fileData[spsStartPos + 1] };  
            int spsLength = bytes2Int(spsbt);  
            byte[] SPS = new byte[spsLength];  
            // 跳过2个字节的 sequenceParameterSetLength  
            spsStartPos += 2;  
            System.arraycopy(fileData, spsStartPos, SPS, 0, spsLength);  
            printResult("SPS", SPS, spsLength);  

            // 底下部分为获取PPS  
            // spsStartPos + spsLength 可以跳到pps位置  
            // 再加1的目的是跳过1字节的 numOfPictureParameterSets  
            int ppsStartPos = spsStartPos + spsLength + 1;  
            byte[] ppsbt = new byte[] { fileData[ppsStartPos],  
                    fileData[ppsStartPos + 1] };  
            int ppsLength = bytes2Int(ppsbt);  
            byte[] PPS = new byte[ppsLength];  
            ppsStartPos += 2;  
            System.arraycopy(fileData, ppsStartPos, PPS, 0, ppsLength);  
            printResult("PPS", PPS, ppsLength);  
        }  

        private int bytes2Int(byte[] bt) {  
            int ret = bt[0];  
            ret <<= 8;  
            ret |= bt[1];  
            return ret;  
        }  

        private void printResult(String type, byte[] bt, int len) {  
            System.out.println(type + "长度为：" + len);  
            String cont = type + "的内容为：";  
            System.out.print(cont);  
            for (int ix = 0; ix < len; ++ix) {  
                System.out.printf("%02x ", bt[ix]);  
            }  
            System.out.println("\n----------");  
        }  

        public static void main(String[] args) throws IOException {  
            new ObtainSPSAndPPS().getSPSAndPPS("c:\\zgy.h264");  
        }  
    }  

运行结果如下：

    SPS长度为：9  
    SPS的内容为：67 42 00 1f e9 02 c1 2c 80   
    ----------  
    PPS长度为：4  
    PPS的内容为：68 ce 06 f2   
    ----------