|
|
|
@ -438,6 +438,34 @@ static void format_init(zimg_image_format *format, AVFrame *frame, const AVPixFm |
|
|
|
|
format->chroma_location = location == -1 ? convert_chroma_location(frame->chroma_location) : location; |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
static int graph_build(zimg_filter_graph **graph, zimg_graph_builder_params *params, |
|
|
|
|
zimg_image_format *src_format, zimg_image_format *dst_format, |
|
|
|
|
void **tmp, size_t *tmp_size) |
|
|
|
|
{ |
|
|
|
|
int ret; |
|
|
|
|
size_t size; |
|
|
|
|
|
|
|
|
|
zimg_filter_graph_free(*graph); |
|
|
|
|
*graph = zimg_filter_graph_build(src_format, dst_format, params); |
|
|
|
|
if (!*graph) |
|
|
|
|
return print_zimg_error(NULL); |
|
|
|
|
|
|
|
|
|
ret = zimg_filter_graph_get_tmp_size(*graph, &size); |
|
|
|
|
if (ret) |
|
|
|
|
return print_zimg_error(NULL); |
|
|
|
|
|
|
|
|
|
if (size > *tmp_size) { |
|
|
|
|
av_freep(tmp); |
|
|
|
|
*tmp = av_malloc(size); |
|
|
|
|
if (!*tmp) |
|
|
|
|
return AVERROR(ENOMEM); |
|
|
|
|
|
|
|
|
|
*tmp_size = size; |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
return 0; |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
static int filter_frame(AVFilterLink *link, AVFrame *in) |
|
|
|
|
{ |
|
|
|
|
ZScaleContext *s = link->dst->priv; |
|
|
|
@ -447,7 +475,6 @@ static int filter_frame(AVFilterLink *link, AVFrame *in) |
|
|
|
|
zimg_image_buffer_const src_buf = { ZIMG_API_VERSION }; |
|
|
|
|
zimg_image_buffer dst_buf = { ZIMG_API_VERSION }; |
|
|
|
|
char buf[32]; |
|
|
|
|
size_t tmp_size; |
|
|
|
|
int ret = 0, plane; |
|
|
|
|
AVFrame *out; |
|
|
|
|
|
|
|
|
@ -520,27 +547,10 @@ static int filter_frame(AVFilterLink *link, AVFrame *in) |
|
|
|
|
if (s->chromal != -1) |
|
|
|
|
out->chroma_location = (int)s->dst_format.chroma_location - 1; |
|
|
|
|
|
|
|
|
|
zimg_filter_graph_free(s->graph); |
|
|
|
|
s->graph = zimg_filter_graph_build(&s->src_format, &s->dst_format, &s->params); |
|
|
|
|
if (!s->graph) { |
|
|
|
|
ret = print_zimg_error(link->dst); |
|
|
|
|
goto fail; |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
if ((ret = zimg_filter_graph_get_tmp_size(s->graph, &tmp_size))) { |
|
|
|
|
ret = print_zimg_error(link->dst); |
|
|
|
|
ret = graph_build(&s->graph, &s->params, &s->src_format, &s->dst_format, |
|
|
|
|
&s->tmp, &s->tmp_size); |
|
|
|
|
if (ret < 0) |
|
|
|
|
goto fail; |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
if (tmp_size > s->tmp_size) { |
|
|
|
|
av_freep(&s->tmp); |
|
|
|
|
s->tmp = av_malloc(tmp_size); |
|
|
|
|
if (!s->tmp) { |
|
|
|
|
ret = AVERROR(ENOMEM); |
|
|
|
|
goto fail; |
|
|
|
|
} |
|
|
|
|
s->tmp_size = tmp_size; |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
s->in_colorspace = in->colorspace; |
|
|
|
|
s->in_trc = in->color_trc; |
|
|
|
|