C++ 解析 H264 数据包并识别帧类型

C++ 函数解析 H264 数据包并识别帧类型

H264 编码中，每个 NAL 单元（网络抽象层单元）的类型和作用不同，因此可以通过解析 NAL 单元的类型来确定每一包数据是什么帧。

以下是解析 H264 数据包并识别帧类型的 C++ 函数：

void parseH264FrameType(unsigned char* data, int length)
{
    // 检查起始码前缀
    if (length < 4 || data[0] != 0 || data[1] != 0 || data[2] != 1)
        return;

    int pos = 0;
    int nalType = -1;

    // 解析每个 NAL 单元
    while (pos < length)
    {
        // 寻找下一个 NAL 单元的起始位置
        int nextStart = pos + 4;
        while (nextStart < length && !(data[nextStart] == 0 && data[nextStart+1] == 0 && data[nextStart+2] == 1))
            nextStart++;

        // 计算当前 NAL 单元的长度
        int nalLength = nextStart - pos;

        // 获取 NAL 单元类型
        if (nalLength > 1)
            nalType = (data[pos+4] & 0x1F);

        // 根据 NAL 单元类型确定帧类型
        switch (nalType)
        {
            case 1: // 非 IDR 片段
            case 2: // 片段数据分区 A
            case 3: // 片段数据分区 B
            case 4: // 片段数据分区 C
            case 5: // IDR 片段
                // 视频帧
                // ...
                break;
            case 6: // 补充增强信息 (SEI)
                // SEI 消息
                // ...
                break;
            case 7: // 序列参数集 (SPS)
                // SPS 消息
                // ...
                break;
            case 8: // 图像参数集 (PPS)
                // PPS 消息
                // ...
                break;
            case 9: // 存取单元分隔符
                // AUD 消息
                // ...
                break;
            case 10: // 编码序列结束
                // 编码序列结束消息
                // ...
                break;
            case 11: // 流结束
                // 流结束消息
                // ...
                break;
            case 12: // 填充数据
                // 填充数据
                // ...
                break;
            default:
                // 未知 NAL 单元类型
                // ...
                break;
        }

        // 移动到下一个 NAL 单元
        pos = nextStart;
    }
}

该函数首先检查 H264 数据包是否包含起始码前缀 (0x000001)，然后解析每个 NAL 单元的类型，根据 NAL 单元类型确定每一包数据是什么帧。常见的帧类型包括：

• 非 IDR (参考帧) 片段
• IDR (关键帧) 片段
• SEI 消息
• SPS 消息 (序列参数集)
• PPS 消息 (图像参数集)
• AUD 消息 (存取单元分隔符)
• 编码序列结束
• 流结束

通过解析每个 NAL 单元的类型，可以准确地确定每一包 H264 数据是什么帧。