lavfi: Add boxblur_opencl filter
[ffmpeg.git] / 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(NULL, 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 }