Browse code

Add unsharp video filter. Contributed by Daniel G. Taylor (dan/at/programmer-art/dot/org)

Originally committed as revision 22813 to svn://svn.ffmpeg.org/ffmpeg/trunk

Bobby Bingham authored on 2010/04/07 10:05:24
Showing 4 changed files
... ...
@@ -217,6 +217,59 @@ not specified it will use the default value of 16.
217 217
 Adding this in the beginning of filter chains should make filtering
218 218
 faster due to better use of the memory cache.
219 219
 
220
+@section unsharp
221
+
222
+Sharpen or blur the input video. It accepts the following parameters:
223
+
224
+@multitable @columnfractions .2 .5 .1 .1 .1
225
+@headitem Name @tab Description @tab Min @tab Max @tab Default
226
+@item @var{luma_msize_x}
227
+@tab Luma matrix horizontal size
228
+@tab 3
229
+@tab 13
230
+@tab 5
231
+@item @var{luma_msize_y}
232
+@tab Luma matrix vertical size
233
+@tab 3
234
+@tab 13
235
+@tab 5
236
+@item @var{luma_amount}
237
+@tab Luma effect strength
238
+@tab -2.0
239
+@tab 5.0
240
+@tab 1.0
241
+@item @var{chroma_msize_x}
242
+@tab Chroma matrix horizontal size
243
+@tab 3
244
+@tab 13
245
+@tab 0
246
+@item @var{chroma_msize_y}
247
+@tab Chroma matrix vertical size
248
+@tab 3
249
+@tab 13
250
+@tab 0
251
+@item @var{chroma_amount}
252
+@tab Chroma effect strength
253
+@tab -2.0
254
+@tab 5.0
255
+@tab 0.0
256
+@end multitable
257
+
258
+Negative values for the amount will blur the input video, while positive
259
+values will sharpen. All parameters are optional and default to the
260
+equivalent of the string '5:5:1.0:0:0:0.0'.
261
+
262
+@example
263
+# Strong luma sharpen effect parameters
264
+unsharp=7:7:2.5
265
+
266
+# Strong blur of both luma and chroma parameters
267
+unsharp=7:7:-2:7:7:-2
268
+
269
+# Use the default values with @command{ffmpeg}
270
+./ffmpeg -i in.avi -vfilters "unsharp" out.mp4
271
+@end example
272
+
220 273
 @section vflip
221 274
 
222 275
 Flip the input video vertically.
... ...
@@ -22,6 +22,7 @@ OBJS-$(CONFIG_NULL_FILTER)                   += vf_null.o
22 22
 OBJS-$(CONFIG_PIXELASPECT_FILTER)            += vf_aspect.o
23 23
 OBJS-$(CONFIG_SCALE_FILTER)                  += vf_scale.o
24 24
 OBJS-$(CONFIG_SLICIFY_FILTER)                += vf_slicify.o
25
+OBJS-$(CONFIG_UNSHARP_FILTER)                += vf_unsharp.o
25 26
 OBJS-$(CONFIG_VFLIP_FILTER)                  += vf_vflip.o
26 27
 
27 28
 OBJS-$(CONFIG_NULLSRC_FILTER)                += vsrc_nullsrc.o
... ...
@@ -42,6 +42,7 @@ void avfilter_register_all(void)
42 42
     REGISTER_FILTER (PIXELASPECT, pixelaspect, vf);
43 43
     REGISTER_FILTER (SCALE,       scale,       vf);
44 44
     REGISTER_FILTER (SLICIFY,     slicify,     vf);
45
+    REGISTER_FILTER (UNSHARP,     unsharp,     vf);
45 46
     REGISTER_FILTER (VFLIP,       vflip,       vf);
46 47
 
47 48
     REGISTER_FILTER (NULLSRC,     nullsrc,     vsrc);
48 49
new file mode 100644
... ...
@@ -0,0 +1,236 @@
0
+/*
1
+ * Ported to FFmpeg from MPlayer libmpcodecs/unsharp.c
2
+ * Original copyright (C) 2002 Remi Guyomarch <rguyom@pobox.com>
3
+ * Port copyright (C) 2010 Daniel G. Taylor <dan@programmer-art.org>
4
+ * Relicensed to the LGPL with permission from Remi Guyomarch.
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
+/**
24
+ * @file libavfilter/vf_unsharp.c
25
+ * blur / sharpen filter
26
+ *
27
+ * This code is based on:
28
+ *
29
+ * An Efficient algorithm for Gaussian blur using finite-state machines
30
+ * Frederick M. Waltz and John W. V. Miller
31
+ *
32
+ * SPIE Conf. on Machine Vision Systems for Inspection and Metrology VII
33
+ * Originally published Boston, Nov 98
34
+ *
35
+ * http://www.engin.umd.umich.edu/~jwvm/ece581/21_GBlur.pdf
36
+ */
37
+
38
+#include "avfilter.h"
39
+#include "libavutil/common.h"
40
+#include "libavutil/mem.h"
41
+#include "libavutil/pixdesc.h"
42
+
43
+#define MIN_SIZE 3
44
+#define MAX_SIZE 13
45
+
46
+#define CHROMA_WIDTH(link)  -((-link->w) >> av_pix_fmt_descriptors[link->format].log2_chroma_w)
47
+#define CHROMA_HEIGHT(link) -((-link->h) >> av_pix_fmt_descriptors[link->format].log2_chroma_h)
48
+
49
+typedef struct FilterParam {
50
+    int msize_x;                             ///< matrix width
51
+    int msize_y;                             ///< matrix height
52
+    int amount;                              ///< effect amount
53
+    int steps_x;                             ///< horizontal step count
54
+    int steps_y;                             ///< vertical step count
55
+    int scalebits;                           ///< bits to shift pixel
56
+    int32_t halfscale;                       ///< amount to add to pixel
57
+    uint32_t *sc[(MAX_SIZE * MAX_SIZE) - 1]; ///< finite state machine storage
58
+} FilterParam;
59
+
60
+typedef struct {
61
+    FilterParam luma;   ///< luma parameters (width, height, amount)
62
+    FilterParam chroma; ///< chroma parameters (width, height, amount)
63
+} UnsharpContext;
64
+
65
+static void unsharpen(uint8_t *dst, uint8_t *src, int dst_stride, int src_stride, int width, int height, FilterParam *fp)
66
+{
67
+    uint32_t **sc = fp->sc;
68
+    uint32_t sr[(MAX_SIZE * MAX_SIZE) - 1], tmp1, tmp2;
69
+
70
+    int32_t res;
71
+    int x, y, z;
72
+
73
+    if (!fp->amount) {
74
+        if (dst_stride == src_stride)
75
+            memcpy(dst, src, src_stride * height);
76
+        else
77
+            for (y = 0; y < height; y++, dst += dst_stride, src += src_stride)
78
+                memcpy(dst, src, width);
79
+        return;
80
+    }
81
+
82
+    for (y = 0; y < 2 * fp->steps_y; y++)
83
+        memset(sc[y], 0, sizeof(sc[y][0]) * (width + 2 * fp->steps_x));
84
+
85
+    for (y =- fp->steps_y; y < height + fp->steps_y; y++) {
86
+        memset(sr, 0, sizeof(sr[0]) * (2 * fp->steps_x - 1));
87
+        for (x =- fp->steps_x; x < width + fp->steps_x; x++) {
88
+            tmp1 = x <= 0 ? src[0] : x >= width ? src[width-1] : src[x];
89
+            for (z = 0; z < fp->steps_x * 2; z += 2) {
90
+                tmp2 = sr[z + 0] + tmp1; sr[z + 0] = tmp1;
91
+                tmp1 = sr[z + 1] + tmp2; sr[z + 1] = tmp2;
92
+            }
93
+            for (z = 0; z < fp->steps_y * 2; z += 2) {
94
+                tmp2 = sc[z + 0][x + fp->steps_x] + tmp1; sc[z + 0][x + fp->steps_x] = tmp1;
95
+                tmp1 = sc[z + 1][x + fp->steps_x] + tmp2; sc[z + 1][x + fp->steps_x] = tmp2;
96
+            }
97
+            if (x >= fp->steps_x && y >= fp->steps_y) {
98
+                uint8_t* srx = src - fp->steps_y * src_stride + x - fp->steps_x;
99
+                uint8_t* dsx = dst - fp->steps_y * dst_stride + x - fp->steps_x;
100
+
101
+                res = (int32_t)*srx + ((((int32_t) * srx - (int32_t)((tmp1 + fp->halfscale) >> fp->scalebits)) * fp->amount) >> 16);
102
+                *dsx = av_clip_uint8(res);
103
+            }
104
+        }
105
+        if (y >= 0) {
106
+            dst += dst_stride;
107
+            src += src_stride;
108
+        }
109
+    }
110
+}
111
+
112
+static void set_filter_param(FilterParam *fp, int msize_x, int msize_y, double amount)
113
+{
114
+    fp->msize_x = msize_x;
115
+    fp->msize_y = msize_y;
116
+    fp->amount = amount * 65536.0;
117
+
118
+    fp->steps_x = msize_x / 2;
119
+    fp->steps_y = msize_y / 2;
120
+    fp->scalebits = (fp->steps_x + fp->steps_y) * 2;
121
+    fp->halfscale = 1 << (fp->scalebits - 1);
122
+}
123
+
124
+static av_cold int init(AVFilterContext *ctx, const char *args, void *opaque)
125
+{
126
+    UnsharpContext *unsharp = ctx->priv;
127
+    int lmsize_x = 5, cmsize_x = 0;
128
+    int lmsize_y = 5, cmsize_y = 0;
129
+    double lamount = 1.0f, camount = 0.0f;
130
+
131
+    if (args)
132
+        sscanf(args, "%d:%d:%lf:%d:%d:%lf", &lmsize_x, &lmsize_y, &lamount,
133
+                                            &cmsize_x, &cmsize_y, &camount);
134
+
135
+    set_filter_param(&unsharp->luma,   lmsize_x, lmsize_y, lamount);
136
+    set_filter_param(&unsharp->chroma, cmsize_x, cmsize_y, camount);
137
+
138
+    return 0;
139
+}
140
+
141
+static int query_formats(AVFilterContext *ctx)
142
+{
143
+    enum PixelFormat pix_fmts[] = {
144
+        PIX_FMT_YUV420P,  PIX_FMT_YUV422P,  PIX_FMT_YUV444P,  PIX_FMT_YUV410P,
145
+        PIX_FMT_YUV411P,  PIX_FMT_YUV440P,  PIX_FMT_YUVJ420P, PIX_FMT_YUVJ422P,
146
+        PIX_FMT_YUVJ444P, PIX_FMT_YUVJ440P, PIX_FMT_NONE
147
+    };
148
+
149
+    avfilter_set_common_formats(ctx, avfilter_make_format_list(pix_fmts));
150
+
151
+    return 0;
152
+}
153
+
154
+static void init_filter_param(AVFilterContext *ctx, FilterParam *fp, const char *effect_type, int width)
155
+{
156
+    int z;
157
+    const char *effect;
158
+
159
+    effect = fp->amount == 0 ? "none" : fp->amount < 0 ? "blur" : "sharpen";
160
+
161
+    av_log(ctx, AV_LOG_INFO, "effect:%s type:%s msize_x:%d msize_y:%d amount:%0.2f\n",
162
+           effect, effect_type, fp->msize_x, fp->msize_y, fp->amount / 65535.0);
163
+
164
+    for (z = 0; z < 2 * fp->steps_y; z++)
165
+        fp->sc[z] = av_malloc(sizeof(*(fp->sc[z])) * (width + 2 * fp->steps_x));
166
+}
167
+
168
+static int config_props(AVFilterLink *link)
169
+{
170
+    UnsharpContext *unsharp = link->dst->priv;
171
+
172
+    init_filter_param(link->dst, &unsharp->luma,   "luma",   link->w);
173
+    init_filter_param(link->dst, &unsharp->chroma, "chroma", CHROMA_WIDTH(link));
174
+
175
+    return 0;
176
+}
177
+
178
+static void free_filter_param(FilterParam *fp)
179
+{
180
+    int z;
181
+
182
+    for (z = 0; z < 2 * fp->steps_y; z++)
183
+        av_free(fp->sc[z]);
184
+}
185
+
186
+static av_cold void uninit(AVFilterContext *ctx)
187
+{
188
+    UnsharpContext *unsharp = ctx->priv;
189
+
190
+    free_filter_param(&unsharp->luma);
191
+    free_filter_param(&unsharp->chroma);
192
+}
193
+
194
+static void end_frame(AVFilterLink *link)
195
+{
196
+    UnsharpContext *unsharp = link->dst->priv;
197
+    AVFilterPicRef *in  = link->cur_pic;
198
+    AVFilterPicRef *out = link->dst->outputs[0]->outpic;
199
+
200
+    unsharpen(out->data[0], in->data[0], out->linesize[0], in->linesize[0], link->w,            link->h,             &unsharp->luma);
201
+    unsharpen(out->data[1], in->data[1], out->linesize[1], in->linesize[1], CHROMA_WIDTH(link), CHROMA_HEIGHT(link), &unsharp->chroma);
202
+    unsharpen(out->data[2], in->data[2], out->linesize[2], in->linesize[2], CHROMA_WIDTH(link), CHROMA_HEIGHT(link), &unsharp->chroma);
203
+
204
+    avfilter_unref_pic(in);
205
+    avfilter_draw_slice(link->dst->outputs[0], 0, link->h, 1);
206
+    avfilter_end_frame(link->dst->outputs[0]);
207
+    avfilter_unref_pic(out);
208
+}
209
+
210
+static void draw_slice(AVFilterLink *link, int y, int h, int slice_dir)
211
+{
212
+}
213
+
214
+AVFilter avfilter_vf_unsharp = {
215
+    .name      = "unsharp",
216
+    .description = NULL_IF_CONFIG_SMALL("Sharpen or blur the input video."),
217
+
218
+    .priv_size = sizeof(UnsharpContext),
219
+
220
+    .init = init,
221
+    .uninit = uninit,
222
+    .query_formats = query_formats,
223
+
224
+    .inputs    = (AVFilterPad[]) {{ .name             = "default",
225
+                                    .type             = AVMEDIA_TYPE_VIDEO,
226
+                                    .draw_slice       = draw_slice,
227
+                                    .end_frame        = end_frame,
228
+                                    .config_props     = config_props,
229
+                                    .min_perms        = AV_PERM_READ, },
230
+                                  { .name = NULL}},
231
+
232
+    .outputs   = (AVFilterPad[]) {{ .name             = "default",
233
+                                    .type             = AVMEDIA_TYPE_VIDEO, },
234
+                                  { .name = NULL}},
235
+};