C语言判断h264中I帧位置代码实现

判断H.264码流中I帧的位置可以通过解析NAL单元实现，其中I帧的NAL单元类型为5。以下是一个简单的C语言代码实现：

#include <stdio.h>
#include <stdlib.h>

int main()
{
    FILE* fp = fopen("test.h264", "rb"); // 打开H.264文件
    if (fp == NULL) {
        printf("Failed to open the H.264 file.\n");
        return -1;
    }

    unsigned char* buf = (unsigned char*)malloc(1024 * 1024); // 申请一块足够大的缓存区
    if (buf == NULL) {
        printf("Failed to allocate memory.\n");
        return -1;
    }

    int pos = 0; // 记录I帧位置
    int nal_type; // NAL单元类型
    int nal_len; // NAL单元长度

    while (!feof(fp)) {
        int len = fread(buf, 1, 1024 * 1024, fp); // 读取数据到缓存区
        unsigned char* p = buf;
        while (p < buf + len) {
            // 查找NAL起始码
            if (p[0] == 0x00 && p[1] == 0x00 && p[2] == 0x01) {
                p += 3;
                // 获取NAL单元类型和长度
                nal_type = *p & 0x1f;
                nal_len = ((p[0] << 8) | p[1]) + 2;
                // 判断是否为I帧
                if (nal_type == 5) {
                    pos = p - buf - 3; // 记录I帧位置
                    printf("I frame found at position %d.\n", pos);
                }
                p += nal_len;
            } else {
                // 如果不是NAL起始码，则继续往后查找
                p++;
            }
        }
    }

    free(buf); // 释放缓存区
    fclose(fp); // 关闭文件

    return 0;
}

上述代码中，我们首先打开H.264文件并申请一块足够大的缓存区，然后循环读取数据到缓存区中，每次查找NAL起始码并获取NAL单元类型和长度，如果是I帧则记录位置并输出。最后释放缓存区并关闭文件。