h265/hevc码流中获取slice type类型的代码

获取H.265/HEVC 码流中 Slice Type 类型的代码示例如下：

#include <stdio.h>

#define MAX_SLICE_NUM 1024

// NALU的类型
typedef enum {
    NALU_TYPE_SLICE = 1,
    NALU_TYPE_IDR = 5,
    NALU_TYPE_SPS = 7,
    NALU_TYPE_PPS = 8,
} NaluType;

// slice类型
typedef enum {
    SLICE_TYPE_B = 0,
    SLICE_TYPE_P = 1,
    SLICE_TYPE_I = 2,
} SliceType;

int main(int argc, char *argv[]) {
    // 打开码流文件
    FILE *fp = fopen("test.h265", "rb");
    if (!fp) {
        printf("Open file failed!\n");
        return -1;
    }

    // 初始化变量
    int nalu_len = 0;
    unsigned char nalu_header[4] = {0x00, 0x00, 0x00, 0x01}; // NALU头
    unsigned char slice_header[4] = {0x00, 0x00, 0x00, 0x01}; // Slice头
    unsigned char buffer[MAX_SLICE_NUM] = {0};
    int buffer_len = 0;
    int slice_type = -1;

    // 循环读取NALU
    while (!feof(fp)) {
        // 读取NALU长度
        if (fread(&nalu_len, 4, 1, fp) != 1) {
            break;
        }

        // 读取NALU
        fread(buffer, 1, nalu_len, fp);

        // 获取NALU类型
        int nalu_type = buffer[0] & 0x1f;

        // 如果是Slice类型，则获取Slice Type
        if (nalu_type == NALU_TYPE_SLICE || nalu_type == NALU_TYPE_IDR) {
            // 获取Slice头
            unsigned char *slice_start = buffer;
            while (slice_start < buffer + nalu_len) {
                if (slice_start[0] == 0x00 && slice_start[1] == 0x00 &&
                    slice_start[2] == 0x00 && slice_start[3] == 0x01) {
                    break;
                }
                slice_start++;
            }
            if (slice_start >= buffer + nalu_len - 4) {
                continue;
            }

            // 获取Slice Type
            slice_type = slice_start[4] % 5;

            // 输出Slice Type
            printf("Slice Type: %d\n", slice_type);
        }
    }

    // 关闭文件
    fclose(fp);

    return 0;
}

上述代码通过循环读取NALU，判断NALU类型是否为 Slice 类型，如果是，则从NALU中获取 Slice Type 类型。其中，Slice Type 类型定义如下：

- 0：B slice
- 1：P slice
- 2：I slice

需要注意的是，在 H.265/HEVC 码流中，Slice Type 类型可能存在于 Slice Header 中的第一个字节的后 5 位。因此，需要先找到 Slice Header 的起始位置，再从中获取 Slice Type。