LAV
/
FFmpeg
1
0
Fork 0
Code

lavc/vvc: Use pps->{width, height} over sps->{width, height} 

The PPS should be used instead of the SPS to get the current picture's
dimensions.  Using the SPS can cause issues if the resolution changes
mid-sequence.  In particular, it was leading to invalid memory accesses
if the resolution decreased.

Patch replaces sps->{width,height} with pps->{width,height}.  It also
removes sps->{width,height}, as these are no longer used anywhere.

Fixes crash when decoding DVB V&V test sequence
VVC_HDR_UHDTV1_ClosedGOP_Max3840x2160_50fps_HLG10_res_change_without_RPR

Signed-off-by: Frank Plowman <post@frankplowman.com>
This commit is contained in:
Frank Plowman 2024-02-15 11:54:24 +00:00 committed by Nuo Mi
parent 86367be5ef
commit acacf8a313
7 changed files with 32 additions and 36 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
libavcodec/vvc/vvc_ctu.c
View File

@ -2415,8 +2415,8 @@ void ff_vvc_decode_neighbour(VVCLocalContext *lc, const int x_ctb, const int y_c
VVCFrameContext *fc = lc->fc;
const int ctb_size = fc->ps.sps->ctb_size_y;
lc->end_of_tiles_x = fc->ps.sps->width;
lc->end_of_tiles_y = fc->ps.sps->height;
lc->end_of_tiles_x = fc->ps.pps->width;
lc->end_of_tiles_y = fc->ps.pps->height;
if (fc->ps.pps->ctb_to_col_bd[rx] != fc->ps.pps->ctb_to_col_bd[rx + 1])
lc->end_of_tiles_x = FFMIN(x_ctb + ctb_size, lc->end_of_tiles_x);
if (fc->ps.pps->ctb_to_row_bd[ry] != fc->ps.pps->ctb_to_row_bd[ry + 1])

2
libavcodec/vvc/vvc_ctu.h
View File

@ -85,7 +85,7 @@
/**
* Value of the luma sample at position (x, y) in the 2D array tab.
*/
#define SAMPLE(tab, x, y) ((tab)[(y) * s->sps->width + (x)])
#define SAMPLE(tab, x, y) ((tab)[(y) * s->pps->width + (x)])
#define SAMPLE_CTB(tab, x, y) ((tab)[(y) * min_cb_width + (x)])
#define CTB(tab, x, y) ((tab)[(y) * fc->ps.pps->ctb_width + (x)])

48
libavcodec/vvc/vvc_filter.c
View File

@ -102,8 +102,8 @@ static void copy_ctb_to_hv(VVCFrameContext *fc, const uint8_t *src,
const int c_idx, const int x_ctb, const int y_ctb, const int top)
{
const int ps = fc->ps.sps->pixel_shift;
const int w = fc->ps.sps->width >> fc->ps.sps->hshift[c_idx];
const int h = fc->ps.sps->height >> fc->ps.sps->vshift[c_idx];
const int w = fc->ps.pps->width >> fc->ps.sps->hshift[c_idx];
const int h = fc->ps.pps->height >> fc->ps.sps->vshift[c_idx];
if (top) {
/* top */
@ -133,8 +133,8 @@ static void sao_copy_ctb_to_hv(VVCLocalContext *lc, const int rx, const int ry,
const ptrdiff_t src_stride = fc->frame->linesize[c_idx];
const int ctb_size_h = ctb_size_y >> fc->ps.sps->hshift[c_idx];
const int ctb_size_v = ctb_size_y >> fc->ps.sps->vshift[c_idx];
const int width = FFMIN(ctb_size_h, (fc->ps.sps->width >> fc->ps.sps->hshift[c_idx]) - x);
const int height = FFMIN(ctb_size_v, (fc->ps.sps->height >> fc->ps.sps->vshift[c_idx]) - y);
const int width = FFMIN(ctb_size_h, (fc->ps.pps->width >> fc->ps.sps->hshift[c_idx]) - x);
const int height = FFMIN(ctb_size_v, (fc->ps.pps->height >> fc->ps.sps->vshift[c_idx]) - y);
const uint8_t *src = &fc->frame->data[c_idx][y * src_stride + (x << fc->ps.sps->pixel_shift)];
copy_ctb_to_hv(fc, src, src_stride, x, y, width, height, c_idx, rx, ry, top);
}
@ -216,8 +216,8 @@ void ff_vvc_sao_filter(VVCLocalContext *lc, int x, int y)
ptrdiff_t src_stride = fc->frame->linesize[c_idx];
int ctb_size_h = ctb_size_y >> fc->ps.sps->hshift[c_idx];
int ctb_size_v = ctb_size_y >> fc->ps.sps->vshift[c_idx];
int width = FFMIN(ctb_size_h, (fc->ps.sps->width >> fc->ps.sps->hshift[c_idx]) - x0);
int height = FFMIN(ctb_size_v, (fc->ps.sps->height >> fc->ps.sps->vshift[c_idx]) - y0);
int width = FFMIN(ctb_size_h, (fc->ps.pps->width >> fc->ps.sps->hshift[c_idx]) - x0);
int height = FFMIN(ctb_size_v, (fc->ps.pps->height >> fc->ps.sps->vshift[c_idx]) - y0);
int tab = sao_tab[(FFALIGN(width, 8) >> 3) - 1];
uint8_t *src = &fc->frame->data[c_idx][y0 * src_stride + (x0 << fc->ps.sps->pixel_shift)];
ptrdiff_t dst_stride;
@ -230,8 +230,8 @@ void ff_vvc_sao_filter(VVCLocalContext *lc, int x, int y)
break;
case SAO_EDGE:
{
const int w = fc->ps.sps->width >> fc->ps.sps->hshift[c_idx];
const int h = fc->ps.sps->height >> fc->ps.sps->vshift[c_idx];
const int w = fc->ps.pps->width >> fc->ps.sps->hshift[c_idx];
const int h = fc->ps.pps->height >> fc->ps.sps->vshift[c_idx];
const int sh = fc->ps.sps->pixel_shift;
dst_stride = 2*MAX_PB_SIZE + AV_INPUT_BUFFER_PADDING_SIZE;
@ -880,11 +880,11 @@ void ff_vvc_deblock_vertical(const VVCLocalContext *lc, int x0, int y0)
vvc_deblock_bs(lc, x0, y0, 1);
x_end = x0 + ctb_size;
if (x_end > fc->ps.sps->width)
x_end = fc->ps.sps->width;
if (x_end > fc->ps.pps->width)
x_end = fc->ps.pps->width;
y_end = y0 + ctb_size;
if (y_end > fc->ps.sps->height)
y_end = fc->ps.sps->height;
if (y_end > fc->ps.pps->height)
y_end = fc->ps.pps->height;
for (int c_idx = 0; c_idx < c_end; c_idx++) {
const int hs = sps->hshift[c_idx];
@ -950,11 +950,11 @@ void ff_vvc_deblock_horizontal(const VVCLocalContext *lc, int x0, int y0)
vvc_deblock_bs(lc, x0, y0, 0);
x_end = x0 + ctb_size;
if (x_end > fc->ps.sps->width)
x_end = fc->ps.sps->width;
if (x_end > fc->ps.pps->width)
x_end = fc->ps.pps->width;
y_end = y0 + ctb_size;
if (y_end > fc->ps.sps->height)
y_end = fc->ps.sps->height;
if (y_end > fc->ps.pps->height)
y_end = fc->ps.pps->height;
for (int c_idx = 0; c_idx < c_end; c_idx++) {
const int hs = sps->hshift[c_idx];
@ -1050,8 +1050,8 @@ static void alf_copy_ctb_to_hv(VVCFrameContext *fc, const uint8_t *src, const pt
const int x, const int y, const int width, const int height, const int x_ctb, const int y_ctb, const int c_idx)
{
const int ps = fc->ps.sps->pixel_shift;
const int w = fc->ps.sps->width >> fc->ps.sps->hshift[c_idx];
const int h = fc->ps.sps->height >> fc->ps.sps->vshift[c_idx];
const int w = fc->ps.pps->width >> fc->ps.sps->hshift[c_idx];
const int h = fc->ps.pps->height >> fc->ps.sps->vshift[c_idx];
const int border_pixels = (c_idx == 0) ? ALF_BORDER_LUMA : ALF_BORDER_CHROMA;
const int offset_h[] = { 0, height - border_pixels };
const int offset_v[] = { 0, width - border_pixels };
@ -1107,8 +1107,8 @@ static void alf_prepare_buffer(VVCFrameContext *fc, uint8_t *_dst, const uint8_t
const int c_idx, const int *edges)
{
const int ps = fc->ps.sps->pixel_shift;
const int w = fc->ps.sps->width >> fc->ps.sps->hshift[c_idx];
const int h = fc->ps.sps->height >> fc->ps.sps->vshift[c_idx];
const int w = fc->ps.pps->width >> fc->ps.sps->hshift[c_idx];
const int h = fc->ps.pps->height >> fc->ps.sps->vshift[c_idx];
const int border_pixels = c_idx == 0 ? ALF_BORDER_LUMA : ALF_BORDER_CHROMA;
uint8_t *dst, *src;
@ -1241,8 +1241,8 @@ void ff_vvc_alf_copy_ctu_to_hv(VVCLocalContext* lc, const int x0, const int y0)
const int vs = fc->ps.sps->vshift[c_idx];
const int x = x0 >> hs;
const int y = y0 >> vs;
const int width = FFMIN(fc->ps.sps->width - x0, ctb_size_y) >> hs;
const int height = FFMIN(fc->ps.sps->height - y0, ctb_size_y) >> vs;
const int width = FFMIN(fc->ps.pps->width - x0, ctb_size_y) >> hs;
const int height = FFMIN(fc->ps.pps->height - y0, ctb_size_y) >> vs;
const int src_stride = fc->frame->linesize[c_idx];
uint8_t* src = &fc->frame->data[c_idx][y * src_stride + (x << ps)];
@ -1286,8 +1286,8 @@ void ff_vvc_alf_filter(VVCLocalContext *lc, const int x0, const int y0)
const int ctb_size_v = ctb_size_y >> vs;
const int x = x0 >> hs;
const int y = y0 >> vs;
const int pic_width = fc->ps.sps->width >> hs;
const int pic_height = fc->ps.sps->height >> vs;
const int pic_width = fc->ps.pps->width >> hs;
const int pic_height = fc->ps.pps->height >> vs;
const int width = FFMIN(pic_width - x, ctb_size_h);
const int height = FFMIN(pic_height - y, ctb_size_v);
const int src_stride = fc->frame->linesize[c_idx];

4
libavcodec/vvc/vvc_mvs.c
View File

@ -226,8 +226,8 @@ static int temporal_luma_motion_vector(const VVCLocalContext *lc,
if (tab_mvf &&
(cu->y0 >> sps->ctb_log2_size_y) == (y >> sps->ctb_log2_size_y) &&
y < fc->ps.sps->height &&
x < fc->ps.sps->width) {
y < fc->ps.pps->height &&
x < fc->ps.pps->width) {
x &= ~7;
y &= ~7;
temp_col = TAB_MVF(x, y);

3
libavcodec/vvc/vvc_ps.c
View File

@ -177,9 +177,6 @@ static int sps_derive(VVCSPS *sps, void *log_ctx)
int ret;
const H266RawSPS *r = sps->r;
sps->width = r->sps_pic_width_max_in_luma_samples;
sps->height = r->sps_pic_height_max_in_luma_samples;
ret = sps_bit_depth(sps, log_ctx);
if (ret < 0)
return ret;

2
libavcodec/vvc/vvc_ps.h
View File

@ -59,8 +59,6 @@ typedef struct VVCSPS {
const H266RawSPS *r; ///< RefStruct reference
//derived values
uint16_t width;
uint16_t height;
uint8_t hshift[VVC_MAX_SAMPLE_ARRAYS];
uint8_t vshift[VVC_MAX_SAMPLE_ARRAYS];
uint32_t max_pic_order_cnt_lsb; ///< MaxPicOrderCntLsb

5
libavcodec/vvc/vvc_refs.c
View File

@ -316,6 +316,7 @@ static void mark_ref(VVCFrame *frame, int flag)
static VVCFrame *generate_missing_ref(VVCContext *s, VVCFrameContext *fc, int poc)
{
const VVCSPS *sps = fc->ps.sps;
const VVCPPS *pps = fc->ps.pps;
VVCFrame *frame;
frame = alloc_frame(s, fc);
@ -329,10 +330,10 @@ static VVCFrame *generate_missing_ref(VVCContext *s, VVCFrameContext *fc, int po
frame->frame->buf[i]->size);
} else {
for (int i = 0; frame->frame->data[i]; i++)
for (int y = 0; y < (sps->height >> sps->vshift[i]); y++) {
for (int y = 0; y < (pps->height >> sps->vshift[i]); y++) {
uint8_t *dst = frame->frame->data[i] + y * frame->frame->linesize[i];
AV_WN16(dst, 1 << (sps->bit_depth - 1));
av_memcpy_backptr(dst + 2, 2, 2*(sps->width >> sps->hshift[i]) - 2);
av_memcpy_backptr(dst + 2, 2, 2*(pps->width >> sps->hshift[i]) - 2);
}
}
}