git.sesse.net Git - ffmpeg/blob - libavfilter/boxblur.c

   1 /*
   2  * Copyright (c) 2002 Michael Niedermayer <michaelni@gmx.at>
   3  * Copyright (c) 2011 Stefano Sabatini
   4  * Copyright (c) 2018 Danil Iashchenko
   5  *
   6  * This file is part of FFmpeg.
   7  *
   8  * FFmpeg is free software; you can redistribute it and/or
   9  * modify it under the terms of the GNU Lesser General Public
  10  * License as published by the Free Software Foundation; either
  11  * version 2.1 of the License, or (at your option) any later version.
  12  *
  13  * FFmpeg is distributed in the hope that it will be useful,
  14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  16  * Lesser General Public License for more details.
  17  *
  18  * You should have received a copy of the GNU Lesser General Public
  19  * License along with FFmpeg; if not, write to the Free Software
  20  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  21  */
  22
  23 #include "boxblur.h"
  24
  25 static const char *const var_names[] = {
  26     "w",
  27     "h",
  28     "cw",
  29     "ch",
  30     "hsub",
  31     "vsub",
  32     NULL
  33 };
  34
  35 enum var_name {
  36     VAR_W,
  37     VAR_H,
  38     VAR_CW,
  39     VAR_CH,
  40     VAR_HSUB,
  41     VAR_VSUB,
  42     VARS_NB
  43 };
  44
  45
  46 int ff_boxblur_eval_filter_params(AVFilterLink *inlink,
  47                                   FilterParam *luma_param,
  48                                   FilterParam *chroma_param,
  49                                   FilterParam *alpha_param)
  50 {
  51     const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(inlink->format);
  52     AVFilterContext *ctx = inlink->dst;
  53     int w = inlink->w, h = inlink->h;
  54     int cw, ch;
  55     double var_values[VARS_NB], res;
  56     char *expr;
  57     int ret;
  58
  59     if (!luma_param->radius_expr) {
  60         av_log(ctx, AV_LOG_ERROR, "Luma radius expression is not set.\n");
  61         return AVERROR(EINVAL);
  62     }
  63
  64     /* fill missing params */
  65     if (!chroma_param->radius_expr) {
  66         chroma_param->radius_expr = av_strdup(luma_param->radius_expr);
  67         if (!chroma_param->radius_expr)
  68             return AVERROR(ENOMEM);
  69     }
  70     if (chroma_param->power < 0)
  71         chroma_param->power = luma_param->power;
  72
  73     if (!alpha_param->radius_expr) {
  74         alpha_param->radius_expr = av_strdup(luma_param->radius_expr);
  75         if (!alpha_param->radius_expr)
  76             return AVERROR(ENOMEM);
  77     }
  78     if (alpha_param->power < 0)
  79         alpha_param->power = luma_param->power;
  80
  81     var_values[VAR_W]       = inlink->w;
  82     var_values[VAR_H]       = inlink->h;
  83     var_values[VAR_CW] = cw = w>>(desc->log2_chroma_w);
  84     var_values[VAR_CH] = ch = h>>(desc->log2_chroma_h);
  85     var_values[VAR_HSUB]    = 1<<(desc->log2_chroma_w);
  86     var_values[VAR_VSUB]    = 1<<(desc->log2_chroma_h);
  87
  88 #define EVAL_RADIUS_EXPR(comp)                                          \
  89     expr = comp->radius_expr;                                           \
  90     ret = av_expr_parse_and_eval(&res, expr, var_names, var_values,     \
  91                                  NULL, NULL, NULL, NULL, NULL, 0, ctx); \
  92     comp->radius = res;                                                 \
  93     if (ret < 0) {                                                      \
  94         av_log(ctx, AV_LOG_ERROR,                                      \
  95                "Error when evaluating " #comp " radius expression '%s'\n", expr); \
  96         return ret;                                                     \
  97     }
  98
  99     EVAL_RADIUS_EXPR(luma_param);
 100     EVAL_RADIUS_EXPR(chroma_param);
 101     EVAL_RADIUS_EXPR(alpha_param);
 102
 103     av_log(ctx, AV_LOG_VERBOSE,
 104            "luma_radius:%d luma_power:%d "
 105            "chroma_radius:%d chroma_power:%d "
 106            "alpha_radius:%d alpha_power:%d "
 107            "w:%d chroma_w:%d h:%d chroma_h:%d\n",
 108            luma_param  ->radius, luma_param  ->power,
 109            chroma_param->radius, chroma_param->power,
 110            alpha_param ->radius, alpha_param ->power,
 111            w, cw, h, ch);
 112
 113
 114 #define CHECK_RADIUS_VAL(w_, h_, comp)                                  \
 115     if (comp->radius < 0 ||                                   \
 116         2*comp->radius > FFMIN(w_, h_)) {                     \
 117         av_log(ctx, AV_LOG_ERROR,                                       \
 118                "Invalid " #comp " radius value %d, must be >= 0 and <= %d\n", \
 119                comp->radius, FFMIN(w_, h_)/2);                \
 120         return AVERROR(EINVAL);                                         \
 121     }
 122     CHECK_RADIUS_VAL(w,  h,  luma_param);
 123     CHECK_RADIUS_VAL(cw, ch, chroma_param);
 124     CHECK_RADIUS_VAL(w,  h,  alpha_param);
 125
 126     return 0;
 127 }