virtualenv/FFmpeg
1
0
Fork 0
mirror of https://github.com/FFmpeg/FFmpeg.git synced 2024-11-21 10:55:51 +02:00
Code Issues Releases Activity

avcodec/vvcdec: ps, derive virtual boundaries

Browse Source
This commit is contained in:
Nuo Mi 2024-06-09 14:35:08 +08:00
parent 8b3d1f1fb3
commit 80ded9cfe7
2 changed files with 51 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

45
libavcodec/vvc/ps.c
View File

@ -796,8 +796,49 @@ static int ph_max_num_subblock_merge_cand(const H266RawSPS *sps, const H266RawPi
return sps->sps_sbtmvp_enabled_flag && ph->ph_temporal_mvp_enabled_flag;
}
static int ph_vb_pos(uint16_t *vbs, uint8_t *num_vbs, const uint16_t *pos_minus_1, const uint8_t num_pos, uint16_t max, const int ctb_size_y)
{
max = FF_CEIL_RSHIFT(max, 3) - 2;
for (int i = 0; i < num_pos; i++) {
if (pos_minus_1[i] > max)
return AVERROR_INVALIDDATA;
vbs[i] = (pos_minus_1[i] + 1) << 3;
// The distance between any two vertical virtual boundaries shall be greater than or equal to CtbSizeY luma samples
if (i && vbs[i] < vbs[i - 1] + ctb_size_y)
return AVERROR_INVALIDDATA;
}
*num_vbs = num_pos;
return 0;
}
#define VBF(f) (sps->sps_virtual_boundaries_present_flag ? sps->sps_##f : ph->r->ph_##f)
#define VBFS(c, d) VBF(virtual_boundary_pos_##c##_minus1), VBF(num_##d##_virtual_boundaries)
static int ph_vb(VVCPH *ph, const H266RawSPS *sps, const H266RawPPS *pps)
{
const int ctb_size_y = 1 << (sps->sps_log2_ctu_size_minus5 + 5);
int ret;
if (!sps->sps_virtual_boundaries_enabled_flag)
return 0;
ret = ph_vb_pos(ph->vb_pos_x, &ph->num_ver_vbs, VBFS(x, ver), pps->pps_pic_width_in_luma_samples, ctb_size_y);
if (ret < 0)
return ret;
ret = ph_vb_pos(ph->vb_pos_y, &ph->num_hor_vbs, VBFS(y, hor), pps->pps_pic_height_in_luma_samples, ctb_size_y);
if (ret < 0)
return ret;
return 0;
}
static int ph_derive(VVCPH *ph, const H266RawSPS *sps, const H266RawPPS *pps, const int poc_tid0, const int is_clvss)
{
int ret;
ph->max_num_subblock_merge_cand = ph_max_num_subblock_merge_cand(sps, ph->r);
ph->poc = ph_compute_poc(ph->r, sps, poc_tid0, is_clvss);
@ -805,6 +846,10 @@ static int ph_derive(VVCPH *ph, const H266RawSPS *sps, const H266RawPPS *pps, co
if (pps->pps_wp_info_in_ph_flag)
pred_weight_table(&ph->pwt, &ph->r->ph_pred_weight_table);
ret = ph_vb(ph, sps, pps);
if (ret < 0)
return ret;
return 0;
}

6
libavcodec/vvc/ps.h
View File

@ -151,6 +151,12 @@ typedef struct VVCPH {
//derived values
uint32_t max_num_subblock_merge_cand; ///< MaxNumSubblockMergeCand
int32_t poc; ///< PicOrderCntVal
uint8_t num_ver_vbs; ///< NumVerVirtualBoundaries
uint16_t vb_pos_x[VVC_MAX_VBS]; ///< VirtualBoundaryPosX
uint8_t num_hor_vbs; ///< NumHorVirtualBoundaries
uint16_t vb_pos_y[VVC_MAX_VBS]; ///< VirtualBoundaryPosY
PredWeightTable pwt;
} VVCPH;