Commit 7f839595 authored by Anton Khirnov's avatar Anton Khirnov

vf_crop: use the name 's' for the pointer to the private context

This is shorter and consistent across filters.
parent 762df56e
...@@ -117,10 +117,12 @@ static int query_formats(AVFilterContext *ctx) ...@@ -117,10 +117,12 @@ static int query_formats(AVFilterContext *ctx)
static av_cold void uninit(AVFilterContext *ctx) static av_cold void uninit(AVFilterContext *ctx)
{ {
CropContext *crop = ctx->priv; CropContext *s = ctx->priv;
av_expr_free(crop->x_pexpr); crop->x_pexpr = NULL; av_expr_free(s->x_pexpr);
av_expr_free(crop->y_pexpr); crop->y_pexpr = NULL; s->x_pexpr = NULL;
av_expr_free(s->y_pexpr);
s->y_pexpr = NULL;
} }
static inline int normalize_double(int *n, double d) static inline int normalize_double(int *n, double d)
...@@ -141,74 +143,74 @@ static inline int normalize_double(int *n, double d) ...@@ -141,74 +143,74 @@ static inline int normalize_double(int *n, double d)
static int config_input(AVFilterLink *link) static int config_input(AVFilterLink *link)
{ {
AVFilterContext *ctx = link->dst; AVFilterContext *ctx = link->dst;
CropContext *crop = ctx->priv; CropContext *s = ctx->priv;
const AVPixFmtDescriptor *pix_desc = av_pix_fmt_desc_get(link->format); const AVPixFmtDescriptor *pix_desc = av_pix_fmt_desc_get(link->format);
int ret; int ret;
const char *expr; const char *expr;
double res; double res;
crop->var_values[VAR_E] = M_E; s->var_values[VAR_E] = M_E;
crop->var_values[VAR_PHI] = M_PHI; s->var_values[VAR_PHI] = M_PHI;
crop->var_values[VAR_PI] = M_PI; s->var_values[VAR_PI] = M_PI;
crop->var_values[VAR_IN_W] = crop->var_values[VAR_IW] = ctx->inputs[0]->w; s->var_values[VAR_IN_W] = s->var_values[VAR_IW] = ctx->inputs[0]->w;
crop->var_values[VAR_IN_H] = crop->var_values[VAR_IH] = ctx->inputs[0]->h; s->var_values[VAR_IN_H] = s->var_values[VAR_IH] = ctx->inputs[0]->h;
crop->var_values[VAR_X] = NAN; s->var_values[VAR_X] = NAN;
crop->var_values[VAR_Y] = NAN; s->var_values[VAR_Y] = NAN;
crop->var_values[VAR_OUT_W] = crop->var_values[VAR_OW] = NAN; s->var_values[VAR_OUT_W] = s->var_values[VAR_OW] = NAN;
crop->var_values[VAR_OUT_H] = crop->var_values[VAR_OH] = NAN; s->var_values[VAR_OUT_H] = s->var_values[VAR_OH] = NAN;
crop->var_values[VAR_N] = 0; s->var_values[VAR_N] = 0;
crop->var_values[VAR_T] = NAN; s->var_values[VAR_T] = NAN;
av_image_fill_max_pixsteps(crop->max_step, NULL, pix_desc); av_image_fill_max_pixsteps(s->max_step, NULL, pix_desc);
crop->hsub = pix_desc->log2_chroma_w; s->hsub = pix_desc->log2_chroma_w;
crop->vsub = pix_desc->log2_chroma_h; s->vsub = pix_desc->log2_chroma_h;
if ((ret = av_expr_parse_and_eval(&res, (expr = crop->ow_expr), if ((ret = av_expr_parse_and_eval(&res, (expr = s->ow_expr),
var_names, crop->var_values, var_names, s->var_values,
NULL, NULL, NULL, NULL, NULL, 0, ctx)) < 0) goto fail_expr; NULL, NULL, NULL, NULL, NULL, 0, ctx)) < 0) goto fail_expr;
crop->var_values[VAR_OUT_W] = crop->var_values[VAR_OW] = res; s->var_values[VAR_OUT_W] = s->var_values[VAR_OW] = res;
if ((ret = av_expr_parse_and_eval(&res, (expr = crop->oh_expr), if ((ret = av_expr_parse_and_eval(&res, (expr = s->oh_expr),
var_names, crop->var_values, var_names, s->var_values,
NULL, NULL, NULL, NULL, NULL, 0, ctx)) < 0) goto fail_expr; NULL, NULL, NULL, NULL, NULL, 0, ctx)) < 0) goto fail_expr;
crop->var_values[VAR_OUT_H] = crop->var_values[VAR_OH] = res; s->var_values[VAR_OUT_H] = s->var_values[VAR_OH] = res;
/* evaluate again ow as it may depend on oh */ /* evaluate again ow as it may depend on oh */
if ((ret = av_expr_parse_and_eval(&res, (expr = crop->ow_expr), if ((ret = av_expr_parse_and_eval(&res, (expr = s->ow_expr),
var_names, crop->var_values, var_names, s->var_values,
NULL, NULL, NULL, NULL, NULL, 0, ctx)) < 0) goto fail_expr; NULL, NULL, NULL, NULL, NULL, 0, ctx)) < 0) goto fail_expr;
crop->var_values[VAR_OUT_W] = crop->var_values[VAR_OW] = res; s->var_values[VAR_OUT_W] = s->var_values[VAR_OW] = res;
if (normalize_double(&crop->w, crop->var_values[VAR_OUT_W]) < 0 || if (normalize_double(&s->w, s->var_values[VAR_OUT_W]) < 0 ||
normalize_double(&crop->h, crop->var_values[VAR_OUT_H]) < 0) { normalize_double(&s->h, s->var_values[VAR_OUT_H]) < 0) {
av_log(ctx, AV_LOG_ERROR, av_log(ctx, AV_LOG_ERROR,
"Too big value or invalid expression for out_w/ow or out_h/oh. " "Too big value or invalid expression for out_w/ow or out_h/oh. "
"Maybe the expression for out_w:'%s' or for out_h:'%s' is self-referencing.\n", "Maybe the expression for out_w:'%s' or for out_h:'%s' is self-referencing.\n",
crop->ow_expr, crop->oh_expr); s->ow_expr, s->oh_expr);
return AVERROR(EINVAL); return AVERROR(EINVAL);
} }
crop->w &= ~((1 << crop->hsub) - 1); s->w &= ~((1 << s->hsub) - 1);
crop->h &= ~((1 << crop->vsub) - 1); s->h &= ~((1 << s->vsub) - 1);
if ((ret = av_expr_parse(&crop->x_pexpr, crop->x_expr, var_names, if ((ret = av_expr_parse(&s->x_pexpr, s->x_expr, var_names,
NULL, NULL, NULL, NULL, 0, ctx)) < 0 || NULL, NULL, NULL, NULL, 0, ctx)) < 0 ||
(ret = av_expr_parse(&crop->y_pexpr, crop->y_expr, var_names, (ret = av_expr_parse(&s->y_pexpr, s->y_expr, var_names,
NULL, NULL, NULL, NULL, 0, ctx)) < 0) NULL, NULL, NULL, NULL, 0, ctx)) < 0)
return AVERROR(EINVAL); return AVERROR(EINVAL);
av_log(ctx, AV_LOG_VERBOSE, "w:%d h:%d -> w:%d h:%d\n", av_log(ctx, AV_LOG_VERBOSE, "w:%d h:%d -> w:%d h:%d\n",
link->w, link->h, crop->w, crop->h); link->w, link->h, s->w, s->h);
if (crop->w <= 0 || crop->h <= 0 || if (s->w <= 0 || s->h <= 0 ||
crop->w > link->w || crop->h > link->h) { s->w > link->w || s->h > link->h) {
av_log(ctx, AV_LOG_ERROR, av_log(ctx, AV_LOG_ERROR,
"Invalid too big or non positive size for width '%d' or height '%d'\n", "Invalid too big or non positive size for width '%d' or height '%d'\n",
crop->w, crop->h); s->w, s->h);
return AVERROR(EINVAL); return AVERROR(EINVAL);
} }
/* set default, required in the case the first computed value for x/y is NAN */ /* set default, required in the case the first computed value for x/y is NAN */
crop->x = (link->w - crop->w) / 2; s->x = (link->w - s->w) / 2;
crop->y = (link->h - crop->h) / 2; s->y = (link->h - s->h) / 2;
crop->x &= ~((1 << crop->hsub) - 1); s->x &= ~((1 << s->hsub) - 1);
crop->y &= ~((1 << crop->vsub) - 1); s->y &= ~((1 << s->vsub) - 1);
return 0; return 0;
fail_expr: fail_expr:
...@@ -218,10 +220,10 @@ fail_expr: ...@@ -218,10 +220,10 @@ fail_expr:
static int config_output(AVFilterLink *link) static int config_output(AVFilterLink *link)
{ {
CropContext *crop = link->src->priv; CropContext *s = link->src->priv;
link->w = crop->w; link->w = s->w;
link->h = crop->h; link->h = s->h;
return 0; return 0;
} }
...@@ -229,52 +231,56 @@ static int config_output(AVFilterLink *link) ...@@ -229,52 +231,56 @@ static int config_output(AVFilterLink *link)
static int filter_frame(AVFilterLink *link, AVFrame *frame) static int filter_frame(AVFilterLink *link, AVFrame *frame)
{ {
AVFilterContext *ctx = link->dst; AVFilterContext *ctx = link->dst;
CropContext *crop = ctx->priv; CropContext *s = ctx->priv;
const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(link->format); const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(link->format);
int i; int i;
frame->width = crop->w; frame->width = s->w;
frame->height = crop->h; frame->height = s->h;
crop->var_values[VAR_T] = frame->pts == AV_NOPTS_VALUE ? s->var_values[VAR_T] = frame->pts == AV_NOPTS_VALUE ?
NAN : frame->pts * av_q2d(link->time_base); NAN : frame->pts * av_q2d(link->time_base);
crop->var_values[VAR_X] = av_expr_eval(crop->x_pexpr, crop->var_values, NULL); s->var_values[VAR_X] = av_expr_eval(s->x_pexpr, s->var_values, NULL);
crop->var_values[VAR_Y] = av_expr_eval(crop->y_pexpr, crop->var_values, NULL); s->var_values[VAR_Y] = av_expr_eval(s->y_pexpr, s->var_values, NULL);
crop->var_values[VAR_X] = av_expr_eval(crop->x_pexpr, crop->var_values, NULL); s->var_values[VAR_X] = av_expr_eval(s->x_pexpr, s->var_values, NULL);
normalize_double(&crop->x, crop->var_values[VAR_X]); normalize_double(&s->x, s->var_values[VAR_X]);
normalize_double(&crop->y, crop->var_values[VAR_Y]); normalize_double(&s->y, s->var_values[VAR_Y]);
if (crop->x < 0) crop->x = 0; if (s->x < 0)
if (crop->y < 0) crop->y = 0; s->x = 0;
if ((unsigned)crop->x + (unsigned)crop->w > link->w) crop->x = link->w - crop->w; if (s->y < 0)
if ((unsigned)crop->y + (unsigned)crop->h > link->h) crop->y = link->h - crop->h; s->y = 0;
crop->x &= ~((1 << crop->hsub) - 1); if ((unsigned)s->x + (unsigned)s->w > link->w)
crop->y &= ~((1 << crop->vsub) - 1); s->x = link->w - s->w;
if ((unsigned)s->y + (unsigned)s->h > link->h)
s->y = link->h - s->h;
s->x &= ~((1 << s->hsub) - 1);
s->y &= ~((1 << s->vsub) - 1);
av_dlog(ctx, "n:%d t:%f x:%d y:%d x+w:%d y+h:%d\n", av_dlog(ctx, "n:%d t:%f x:%d y:%d x+w:%d y+h:%d\n",
(int)crop->var_values[VAR_N], crop->var_values[VAR_T], crop->x, (int)s->var_values[VAR_N], s->var_values[VAR_T], s->x,
crop->y, crop->x+crop->w, crop->y+crop->h); s->y, s->x+s->w, s->y+s->h);
frame->data[0] += crop->y * frame->linesize[0]; frame->data[0] += s->y * frame->linesize[0];
frame->data[0] += crop->x * crop->max_step[0]; frame->data[0] += s->x * s->max_step[0];
if (!(desc->flags & AV_PIX_FMT_FLAG_PAL || desc->flags & AV_PIX_FMT_FLAG_PSEUDOPAL)) { if (!(desc->flags & AV_PIX_FMT_FLAG_PAL || desc->flags & AV_PIX_FMT_FLAG_PSEUDOPAL)) {
for (i = 1; i < 3; i ++) { for (i = 1; i < 3; i ++) {
if (frame->data[i]) { if (frame->data[i]) {
frame->data[i] += (crop->y >> crop->vsub) * frame->linesize[i]; frame->data[i] += (s->y >> s->vsub) * frame->linesize[i];
frame->data[i] += (crop->x * crop->max_step[i]) >> crop->hsub; frame->data[i] += (s->x * s->max_step[i]) >> s->hsub;
} }
} }
} }
/* alpha plane */ /* alpha plane */
if (frame->data[3]) { if (frame->data[3]) {
frame->data[3] += crop->y * frame->linesize[3]; frame->data[3] += s->y * frame->linesize[3];
frame->data[3] += crop->x * crop->max_step[3]; frame->data[3] += s->x * s->max_step[3];
} }
crop->var_values[VAR_N] += 1.0; s->var_values[VAR_N] += 1.0;
return ff_filter_frame(link->dst->outputs[0], frame); return ff_filter_frame(link->dst->outputs[0], frame);
} }
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment