avfilter/vf_uspp: Allocate qp storage after qp_stride is known

Signed-off-by: Michael Niedermayer <michaelni@gmx.at>
This commit is contained in:
Michael Niedermayer 2014-12-12 01:40:26 +01:00
parent 327c5292f2
commit d2d8ac24b8
1 changed files with 10 additions and 9 deletions

View File

@ -306,15 +306,6 @@ static int config_input(AVFilterLink *inlink)
return AVERROR(EINVAL);
}
if (!uspp->use_bframe_qp) {
/* we are assuming here the qp blocks will not be smaller that 16x16 */
uspp->non_b_qp_alloc_size = FF_CEIL_RSHIFT(width, 4) * FF_CEIL_RSHIFT(height, 4);
uspp->non_b_qp_table = av_calloc(uspp->non_b_qp_alloc_size, sizeof(*uspp->non_b_qp_table));
if (!uspp->non_b_qp_table)
return AVERROR(ENOMEM);
}
for (i = 0; i < 3; i++) {
int is_chroma = !!i;
int w = ((width + 4 * BLOCK-1) & (~(2 * BLOCK-1))) >> is_chroma;
@ -391,6 +382,16 @@ static int filter_frame(AVFilterLink *inlink, AVFrame *in)
w = FF_CEIL_RSHIFT(qp_stride, 4);
h = FF_CEIL_RSHIFT(inlink->h, 4);
}
if (w * h > uspp->non_b_qp_alloc_size) {
int ret = av_reallocp_array(&uspp->non_b_qp_table, w, h);
if (ret < 0) {
uspp->non_b_qp_alloc_size = 0;
return ret;
}
uspp->non_b_qp_alloc_size = w * h;
}
av_assert0(w * h <= uspp->non_b_qp_alloc_size);
memcpy(uspp->non_b_qp_table, qp_table, w * h);
}