Browse Source

zscale: Factor out graph building

tags/n3.4
Vittorio Giovara 8 years ago
parent
commit
2e91a96d7b
1 changed files with 31 additions and 21 deletions
  1. +31
    -21
      libavfilter/vf_zscale.c

+ 31
- 21
libavfilter/vf_zscale.c View File

@@ -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;


Loading…
Cancel
Save