eessppeelllloo/ffmpeg
1
0
Fork 0
mirror of https://git.ffmpeg.org/ffmpeg.git synced 2025-12-12 01:40:04 +01:00
Code Issues Packages Projects Releases Wiki Activity

vulkan_hevc: handle non-contiguous SPS/PPS/VPS ids

Browse Source 
Some clips (i.e. SLIST_B_Sony_9) will use PPS 0 and 8, before PPS 1-7.
vulkan_hevc expects {sps,pps,vps}_list to be filled in order, which
causes PPS 8 to not be added to the Vulkan session params when it is
being used by a picture.

This removes the expectation that these lists are filled in order. The
indicies into vps_list are saved since there are multiple usages of it.

This also fixes a bug with some clips (i.e. PPS_A_qualcomm_7) which use
all 64 available PPS slots, causing the old loop to think there are more
than 64 PPS-es.
This commit is contained in:
Benjamin Cheng
2023-09-30 12:57:29 -04:00
committed by Lynne
parent e8465aa5dd
commit 2a36e44af3
1 changed files with 27 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

40
libavcodec/vulkan_hevc.c
View File

@@ -70,6 +70,7 @@ typedef struct HEVCHeaderSet {
static int alloc_hevc_header_structs(FFVulkanDecodeContext *s, static int alloc_hevc_header_structs(FFVulkanDecodeContext *s,
int nb_vps, int nb_vps,
const int vps_list_idx[HEVC_MAX_VPS_COUNT],
AVBufferRef * const vps_list[HEVC_MAX_VPS_COUNT]) AVBufferRef * const vps_list[HEVC_MAX_VPS_COUNT])
{ {
uint8_t *data_ptr; uint8_t *data_ptr;
@@ -77,7 +78,7 @@ static int alloc_hevc_header_structs(FFVulkanDecodeContext *s,
size_t buf_size = sizeof(HEVCHeaderSet) + nb_vps*sizeof(HEVCHeaderVPS); size_t buf_size = sizeof(HEVCHeaderSet) + nb_vps*sizeof(HEVCHeaderVPS);
for (int i = 0; i < nb_vps; i++) { for (int i = 0; i < nb_vps; i++) {
const HEVCVPS *vps = (const HEVCVPS *)vps_list[i]->data; const HEVCVPS *vps = (const HEVCVPS *)vps_list[vps_list_idx[i]]->data;
buf_size += sizeof(HEVCHeaderVPSSet)*vps->vps_num_hrd_parameters; buf_size += sizeof(HEVCHeaderVPSSet)*vps->vps_num_hrd_parameters;
} }
@@ -96,7 +97,7 @@ static int alloc_hevc_header_structs(FFVulkanDecodeContext *s,
hdr->hvps = (HEVCHeaderVPS *)(data_ptr + sizeof(HEVCHeaderSet)); hdr->hvps = (HEVCHeaderVPS *)(data_ptr + sizeof(HEVCHeaderSet));
data_ptr += sizeof(HEVCHeaderSet) + nb_vps*sizeof(HEVCHeaderVPS); data_ptr += sizeof(HEVCHeaderSet) + nb_vps*sizeof(HEVCHeaderVPS);
for (int i = 0; i < nb_vps; i++) { for (int i = 0; i < nb_vps; i++) {
const HEVCVPS *vps = (const HEVCVPS *)vps_list[i]->data; const HEVCVPS *vps = (const HEVCVPS *)vps_list[vps_list_idx[i]]->data;
hdr->hvps[i].sls = (HEVCHeaderVPSSet *)data_ptr; hdr->hvps[i].sls = (HEVCHeaderVPSSet *)data_ptr;
data_ptr += sizeof(HEVCHeaderVPSSet)*vps->vps_num_hrd_parameters; data_ptr += sizeof(HEVCHeaderVPSSet)*vps->vps_num_hrd_parameters;
} }
@@ -655,13 +656,15 @@ static int vk_hevc_create_params(AVCodecContext *avctx, AVBufferRef **buf)
.videoSessionParametersTemplate = NULL, .videoSessionParametersTemplate = NULL,
}; };
int nb_vps = 0;
HEVCHeaderSet *hdr; HEVCHeaderSet *hdr;
int nb_vps = 0;
int vps_list_idx[HEVC_MAX_VPS_COUNT];
for (int i = 0; h->ps.vps_list[i]; i++) for (int i = 0; i < HEVC_MAX_VPS_COUNT; i++)
nb_vps++; if (h->ps.vps_list[i])
vps_list_idx[nb_vps++] = i;
err = alloc_hevc_header_structs(dec, nb_vps, h->ps.vps_list); err = alloc_hevc_header_structs(dec, nb_vps, vps_list_idx, h->ps.vps_list);
if (err < 0) if (err < 0)
return err; return err;
@@ -672,26 +675,31 @@ static int vk_hevc_create_params(AVCodecContext *avctx, AVBufferRef **buf)
h265_params_info.pStdVPSs = hdr->vps; h265_params_info.pStdVPSs = hdr->vps;
/* SPS list */ /* SPS list */
for (int i = 0; h->ps.sps_list[i]; i++) { for (int i = 0; i < HEVC_MAX_SPS_COUNT; i++) {
if (h->ps.sps_list[i]) {
const HEVCSPS *sps_l = (const HEVCSPS *)h->ps.sps_list[i]->data; const HEVCSPS *sps_l = (const HEVCSPS *)h->ps.sps_list[i]->data;
set_sps(sps_l, i, &hdr->hsps[i].scaling, &hdr->hsps[i].vui_header, int idx = h265_params_info.stdSPSCount++;
&hdr->hsps[i].vui, &hdr->sps[i], hdr->hsps[i].nal_hdr, set_sps(sps_l, i, &hdr->hsps[idx].scaling, &hdr->hsps[idx].vui_header,
hdr->hsps[i].vcl_hdr, &hdr->hsps[i].ptl, &hdr->hsps[i].dpbm, &hdr->hsps[idx].vui, &hdr->sps[idx], hdr->hsps[idx].nal_hdr,
&hdr->hsps[i].pal, hdr->hsps[i].str, &hdr->hsps[i].ltr); hdr->hsps[idx].vcl_hdr, &hdr->hsps[idx].ptl, &hdr->hsps[idx].dpbm,
h265_params_info.stdSPSCount++; &hdr->hsps[idx].pal, hdr->hsps[idx].str, &hdr->hsps[idx].ltr);
}
} }
/* PPS list */ /* PPS list */
for (int i = 0; h->ps.pps_list[i]; i++) { for (int i = 0; i < HEVC_MAX_PPS_COUNT; i++) {
if (h->ps.pps_list[i]) {
const HEVCPPS *pps_l = (const HEVCPPS *)h->ps.pps_list[i]->data; const HEVCPPS *pps_l = (const HEVCPPS *)h->ps.pps_list[i]->data;
const HEVCSPS *sps_l = (const HEVCSPS *)h->ps.sps_list[pps_l->sps_id]->data; const HEVCSPS *sps_l = (const HEVCSPS *)h->ps.sps_list[pps_l->sps_id]->data;
set_pps(pps_l, sps_l, &hdr->hpps[i].scaling, &hdr->pps[i], &hdr->hpps[i].pal); int idx = h265_params_info.stdPPSCount++;
h265_params_info.stdPPSCount++; set_pps(pps_l, sps_l, &hdr->hpps[idx].scaling,
&hdr->pps[idx], &hdr->hpps[idx].pal);
}
} }
/* VPS list */ /* VPS list */
for (int i = 0; i < nb_vps; i++) { for (int i = 0; i < nb_vps; i++) {
const HEVCVPS *vps_l = (const HEVCVPS *)h->ps.vps_list[i]->data; const HEVCVPS *vps_l = (const HEVCVPS *)h->ps.vps_list[vps_list_idx[i]]->data;
set_vps(vps_l, &hdr->vps[i], &hdr->hvps[i].ptl, &hdr->hvps[i].dpbm, set_vps(vps_l, &hdr->vps[i], &hdr->hvps[i].ptl, &hdr->hvps[i].dpbm,
hdr->hvps[i].hdr, hdr->hvps[i].sls); hdr->hvps[i].hdr, hdr->hvps[i].sls);
h265_params_info.stdVPSCount++; h265_params_info.stdVPSCount++;