Details | Last modification | View Log | RSS feed
Rev | Author | Line No. | Line |
---|---|---|---|
6147 | serge | 1 | /* |
2 | * Copyright (c) 2012 Laurent de Soras |
||
3 | * Copyright (c) 2013 Fredrik Mellbin |
||
4 | * Copyright (c) 2015 Paul B Mahol |
||
5 | * Copyright (c) 2015 James Darnley |
||
6 | * |
||
7 | * This file is part of FFmpeg. |
||
8 | * |
||
9 | * FFmpeg is free software; you can redistribute it and/or |
||
10 | * modify it under the terms of the GNU Lesser General Public |
||
11 | * License as published by the Free Software Foundation; either |
||
12 | * version 2.1 of the License, or (at your option) any later version. |
||
13 | * |
||
14 | * FFmpeg is distributed in the hope that it will be useful, |
||
15 | * but WITHOUT ANY WARRANTY; without even the implied warranty of |
||
16 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
||
17 | * Lesser General Public License for more details. |
||
18 | * |
||
19 | * You should have received a copy of the GNU Lesser General Public |
||
20 | * License along with FFmpeg; if not, write to the Free Software |
||
21 | * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
||
22 | */ |
||
23 | |||
24 | #include "libavutil/imgutils.h" |
||
25 | #include "libavutil/opt.h" |
||
26 | #include "libavutil/pixdesc.h" |
||
27 | #include "avfilter.h" |
||
28 | #include "formats.h" |
||
29 | #include "internal.h" |
||
30 | #include "removegrain.h" |
||
31 | #include "video.h" |
||
32 | |||
33 | #define OFFSET(x) offsetof(RemoveGrainContext, x) |
||
34 | #define FLAGS AV_OPT_FLAG_FILTERING_PARAM|AV_OPT_FLAG_VIDEO_PARAM |
||
35 | |||
36 | static const AVOption removegrain_options[] = { |
||
37 | { "m0", "set mode for 1st plane", OFFSET(mode[0]), AV_OPT_TYPE_INT, {.i64=0}, 0, 24, FLAGS }, |
||
38 | { "m1", "set mode for 2nd plane", OFFSET(mode[1]), AV_OPT_TYPE_INT, {.i64=0}, 0, 24, FLAGS }, |
||
39 | { "m2", "set mode for 3rd plane", OFFSET(mode[2]), AV_OPT_TYPE_INT, {.i64=0}, 0, 24, FLAGS }, |
||
40 | { "m3", "set mode for 4th plane", OFFSET(mode[3]), AV_OPT_TYPE_INT, {.i64=0}, 0, 24, FLAGS }, |
||
41 | {NULL} |
||
42 | }; |
||
43 | |||
44 | AVFILTER_DEFINE_CLASS(removegrain); |
||
45 | |||
46 | static int query_formats(AVFilterContext *ctx) |
||
47 | { |
||
48 | static const enum AVPixelFormat pix_fmts[] = { |
||
49 | AV_PIX_FMT_GRAY8, |
||
50 | AV_PIX_FMT_YUV420P, AV_PIX_FMT_YUV422P, AV_PIX_FMT_YUV444P, |
||
51 | AV_PIX_FMT_YUVA420P, AV_PIX_FMT_YUVA422P, AV_PIX_FMT_YUVA444P, |
||
52 | AV_PIX_FMT_YUV440P, AV_PIX_FMT_YUV411P, AV_PIX_FMT_YUV410P, |
||
53 | AV_PIX_FMT_YUVJ411P, AV_PIX_FMT_YUVJ420P, AV_PIX_FMT_YUVJ422P, |
||
54 | AV_PIX_FMT_YUVJ440P, AV_PIX_FMT_YUVJ444P, |
||
55 | AV_PIX_FMT_GBRP, AV_PIX_FMT_GBRAP, |
||
56 | AV_PIX_FMT_NONE |
||
57 | }; |
||
58 | |||
59 | AVFilterFormats *fmts_list = ff_make_format_list(pix_fmts); |
||
60 | if (!fmts_list) |
||
61 | return AVERROR(ENOMEM); |
||
62 | return ff_set_common_formats(ctx, fmts_list); |
||
63 | } |
||
64 | |||
65 | #define REMOVE_GRAIN_SORT_AXIS \ |
||
66 | const int ma1 = FFMAX(a1, a8); \ |
||
67 | const int mi1 = FFMIN(a1, a8); \ |
||
68 | const int ma2 = FFMAX(a2, a7); \ |
||
69 | const int mi2 = FFMIN(a2, a7); \ |
||
70 | const int ma3 = FFMAX(a3, a6); \ |
||
71 | const int mi3 = FFMIN(a3, a6); \ |
||
72 | const int ma4 = FFMAX(a4, a5); \ |
||
73 | const int mi4 = FFMIN(a4, a5); |
||
74 | |||
75 | static int mode01(int c, int a1, int a2, int a3, int a4, int a5, int a6, int a7, int a8) |
||
76 | { |
||
77 | const int mi = FFMIN(FFMIN(FFMIN(a1, a2), FFMIN(a3, a4)), FFMIN(FFMIN(a5, a6), FFMIN(a7, a8))); |
||
78 | const int ma = FFMAX(FFMAX(FFMAX(a1, a2), FFMAX(a3, a4)), FFMAX(FFMAX(a5, a6), FFMAX(a7, a8))); |
||
79 | |||
80 | return av_clip(c, mi, ma); |
||
81 | } |
||
82 | |||
83 | static int cmp_int(const void *p1, const void *p2) |
||
84 | { |
||
85 | int left = *(const int *)p1; |
||
86 | int right = *(const int *)p2; |
||
87 | |||
88 | return ((left > right) - (left < right)); |
||
89 | } |
||
90 | |||
91 | static int mode02(int c, int a1, int a2, int a3, int a4, int a5, int a6, int a7, int a8) |
||
92 | { |
||
93 | int a[8] = { a1, a2, a3, a4, a5, a6, a7, a8 }; |
||
94 | |||
95 | qsort(&a, 8, sizeof(a[0]), cmp_int); |
||
96 | |||
97 | return av_clip(c, a[2 - 1 ], a[7 - 1]); |
||
98 | } |
||
99 | |||
100 | static int mode03(int c, int a1, int a2, int a3, int a4, int a5, int a6, int a7, int a8) |
||
101 | { |
||
102 | int a[8] = { a1, a2, a3, a4, a5, a6, a7, a8 }; |
||
103 | |||
104 | qsort(&a, 8, sizeof(a[0]), cmp_int); |
||
105 | |||
106 | return av_clip(c, a[3 - 1 ], a[6 - 1]); |
||
107 | } |
||
108 | |||
109 | static int mode04(int c, int a1, int a2, int a3, int a4, int a5, int a6, int a7, int a8) |
||
110 | { |
||
111 | int a[8] = { a1, a2, a3, a4, a5, a6, a7, a8 }; |
||
112 | |||
113 | qsort(&a, 8, sizeof(a[0]), cmp_int); |
||
114 | |||
115 | return av_clip(c, a[4 - 1 ], a[5 - 1]); |
||
116 | } |
||
117 | |||
118 | static int mode05(int c, int a1, int a2, int a3, int a4, int a5, int a6, int a7, int a8) |
||
119 | { |
||
120 | REMOVE_GRAIN_SORT_AXIS |
||
121 | |||
122 | const int c1 = FFABS(c - av_clip(c, mi1, ma1)); |
||
123 | const int c2 = FFABS(c - av_clip(c, mi2, ma2)); |
||
124 | const int c3 = FFABS(c - av_clip(c, mi3, ma3)); |
||
125 | const int c4 = FFABS(c - av_clip(c, mi4, ma4)); |
||
126 | |||
127 | const int mindiff = FFMIN(FFMIN(c1, c2), FFMIN(c3, c4)); |
||
128 | |||
129 | /* When adding SIMD notice the return order here: 4, 2, 3, 1. */ |
||
130 | if (mindiff == c4) { |
||
131 | return av_clip(c, mi4, ma4); |
||
132 | } else if (mindiff == c2) { |
||
133 | return av_clip(c, mi2, ma2); |
||
134 | } else if (mindiff == c3) { |
||
135 | return av_clip(c, mi3, ma3); |
||
136 | } |
||
137 | |||
138 | return av_clip(c, mi1, ma1); |
||
139 | } |
||
140 | |||
141 | static int mode06(int c, int a1, int a2, int a3, int a4, int a5, int a6, int a7, int a8) |
||
142 | { |
||
143 | REMOVE_GRAIN_SORT_AXIS |
||
144 | |||
145 | const int d1 = ma1 - mi1; |
||
146 | const int d2 = ma2 - mi2; |
||
147 | const int d3 = ma3 - mi3; |
||
148 | const int d4 = ma4 - mi4; |
||
149 | |||
150 | const int cli1 = av_clip(c, mi1, ma1); |
||
151 | const int cli2 = av_clip(c, mi2, ma2); |
||
152 | const int cli3 = av_clip(c, mi3, ma3); |
||
153 | const int cli4 = av_clip(c, mi4, ma4); |
||
154 | |||
155 | const int c1 = av_clip_uint16((FFABS(c - cli1) << 1) + d1); |
||
156 | const int c2 = av_clip_uint16((FFABS(c - cli2) << 1) + d2); |
||
157 | const int c3 = av_clip_uint16((FFABS(c - cli3) << 1) + d3); |
||
158 | const int c4 = av_clip_uint16((FFABS(c - cli4) << 1) + d4); |
||
159 | |||
160 | const int mindiff = FFMIN(FFMIN(c1, c2), FFMIN(c3, c4)); |
||
161 | |||
162 | if (mindiff == c4) { |
||
163 | return cli4; |
||
164 | } else if (mindiff == c2) { |
||
165 | return cli2; |
||
166 | } else if (mindiff == c3) { |
||
167 | return cli3; |
||
168 | } |
||
169 | |||
170 | return cli1; |
||
171 | } |
||
172 | |||
173 | static int mode07(int c, int a1, int a2, int a3, int a4, int a5, int a6, int a7, int a8) |
||
174 | { |
||
175 | REMOVE_GRAIN_SORT_AXIS |
||
176 | |||
177 | const int d1 = ma1 - mi1; |
||
178 | const int d2 = ma2 - mi2; |
||
179 | const int d3 = ma3 - mi3; |
||
180 | const int d4 = ma4 - mi4; |
||
181 | |||
182 | const int cli1 = av_clip(c, mi1, ma1); |
||
183 | const int cli2 = av_clip(c, mi2, ma2); |
||
184 | const int cli3 = av_clip(c, mi3, ma3); |
||
185 | const int cli4 = av_clip(c, mi4, ma4); |
||
186 | |||
187 | const int c1 = FFABS(c - cli1) + d1; |
||
188 | const int c2 = FFABS(c - cli2) + d2; |
||
189 | const int c3 = FFABS(c - cli3) + d3; |
||
190 | const int c4 = FFABS(c - cli4) + d4; |
||
191 | |||
192 | const int mindiff = FFMIN(FFMIN(c1, c2), FFMIN(c3, c4)); |
||
193 | |||
194 | if (mindiff == c4) { |
||
195 | return cli4; |
||
196 | } else if (mindiff == c2) { |
||
197 | return cli2; |
||
198 | } else if (mindiff == c3) { |
||
199 | return cli3; |
||
200 | } |
||
201 | |||
202 | return cli1; |
||
203 | } |
||
204 | |||
205 | static int mode08(int c, int a1, int a2, int a3, int a4, int a5, int a6, int a7, int a8) |
||
206 | { |
||
207 | REMOVE_GRAIN_SORT_AXIS |
||
208 | |||
209 | const int d1 = ma1 - mi1; |
||
210 | const int d2 = ma2 - mi2; |
||
211 | const int d3 = ma3 - mi3; |
||
212 | const int d4 = ma4 - mi4; |
||
213 | |||
214 | const int cli1 = av_clip(c, mi1, ma1); |
||
215 | const int cli2 = av_clip(c, mi2, ma2); |
||
216 | const int cli3 = av_clip(c, mi3, ma3); |
||
217 | const int cli4 = av_clip(c, mi4, ma4); |
||
218 | |||
219 | const int c1 = av_clip_uint16(FFABS(c - cli1) + (d1 << 1)); |
||
220 | const int c2 = av_clip_uint16(FFABS(c - cli2) + (d2 << 1)); |
||
221 | const int c3 = av_clip_uint16(FFABS(c - cli3) + (d3 << 1)); |
||
222 | const int c4 = av_clip_uint16(FFABS(c - cli4) + (d4 << 1)); |
||
223 | |||
224 | const int mindiff = FFMIN(FFMIN(c1, c2), FFMIN(c3, c4)); |
||
225 | |||
226 | if (mindiff == c4) { |
||
227 | return cli4; |
||
228 | } else if (mindiff == c2) { |
||
229 | return cli2; |
||
230 | } else if (mindiff == c3) { |
||
231 | return cli3; |
||
232 | } |
||
233 | |||
234 | return cli1; |
||
235 | } |
||
236 | |||
237 | static int mode09(int c, int a1, int a2, int a3, int a4, int a5, int a6, int a7, int a8) |
||
238 | { |
||
239 | REMOVE_GRAIN_SORT_AXIS |
||
240 | |||
241 | const int d1 = ma1 - mi1; |
||
242 | const int d2 = ma2 - mi2; |
||
243 | const int d3 = ma3 - mi3; |
||
244 | const int d4 = ma4 - mi4; |
||
245 | |||
246 | const int mindiff = FFMIN(FFMIN(d1, d2), FFMIN(d3, d4)); |
||
247 | |||
248 | if (mindiff == d4) { |
||
249 | return av_clip(c, mi4, ma4); |
||
250 | } else if (mindiff == d2) { |
||
251 | return av_clip(c, mi2, ma2); |
||
252 | } else if (mindiff == d3) { |
||
253 | return av_clip(c, mi3, ma3); |
||
254 | } |
||
255 | |||
256 | return av_clip(c, mi1, ma1); |
||
257 | } |
||
258 | |||
259 | static int mode10(int c, int a1, int a2, int a3, int a4, int a5, int a6, int a7, int a8) |
||
260 | { |
||
261 | const int d1 = FFABS(c - a1); |
||
262 | const int d2 = FFABS(c - a2); |
||
263 | const int d3 = FFABS(c - a3); |
||
264 | const int d4 = FFABS(c - a4); |
||
265 | const int d5 = FFABS(c - a5); |
||
266 | const int d6 = FFABS(c - a6); |
||
267 | const int d7 = FFABS(c - a7); |
||
268 | const int d8 = FFABS(c - a8); |
||
269 | |||
270 | const int mindiff = FFMIN(FFMIN(FFMIN(d1, d2), FFMIN(d3, d4)), |
||
271 | FFMIN(FFMIN(d5, d6), FFMIN(d7, d8))); |
||
272 | |||
273 | if (mindiff == d7) return a7; |
||
274 | if (mindiff == d8) return a8; |
||
275 | if (mindiff == d6) return a6; |
||
276 | if (mindiff == d2) return a2; |
||
277 | if (mindiff == d3) return a3; |
||
278 | if (mindiff == d1) return a1; |
||
279 | if (mindiff == d5) return a5; |
||
280 | |||
281 | return a4; |
||
282 | } |
||
283 | |||
284 | static int mode1112(int c, int a1, int a2, int a3, int a4, int a5, int a6, int a7, int a8) |
||
285 | { |
||
286 | const int sum = 4 * c + 2 * (a2 + a4 + a5 + a7) + a1 + a3 + a6 + a8; |
||
287 | const int val = (sum + 8) >> 4; |
||
288 | |||
289 | return val; |
||
290 | } |
||
291 | |||
292 | static int mode1314(int c, int a1, int a2, int a3, int a4, int a5, int a6, int a7, int a8) |
||
293 | { |
||
294 | const int d1 = FFABS(a1 - a8); |
||
295 | const int d2 = FFABS(a2 - a7); |
||
296 | const int d3 = FFABS(a3 - a6); |
||
297 | |||
298 | const int mindiff = FFMIN(FFMIN(d1, d2), d3); |
||
299 | |||
300 | if (mindiff == d2) { |
||
301 | return (a2 + a7 + 1) >> 1; |
||
302 | } |
||
303 | if (mindiff == d3) { |
||
304 | return (a3 + a6 + 1) >> 1; |
||
305 | } |
||
306 | |||
307 | return (a1 + a8 + 1) >> 1; |
||
308 | } |
||
309 | |||
310 | static int mode1516(int c, int a1, int a2, int a3, int a4, int a5, int a6, int a7, int a8) |
||
311 | { |
||
312 | const int d1 = FFABS(a1 - a8); |
||
313 | const int d2 = FFABS(a2 - a7); |
||
314 | const int d3 = FFABS(a3 - a6); |
||
315 | |||
316 | const int mindiff = FFMIN(FFMIN(d1, d2), d3); |
||
317 | const int average = (2 * (a2 + a7) + a1 + a3 + a6 + a8 + 4) >> 3; |
||
318 | |||
319 | if (mindiff == d2) { |
||
320 | return av_clip(average, FFMIN(a2, a7), FFMAX(a2, a7)); |
||
321 | } |
||
322 | if (mindiff == d3) { |
||
323 | return av_clip(average, FFMIN(a3, a6), FFMAX(a3, a6)); |
||
324 | } |
||
325 | |||
326 | return av_clip(average, FFMIN(a1, a8), FFMAX(a1, a8)); |
||
327 | } |
||
328 | |||
329 | static int mode17(int c, int a1, int a2, int a3, int a4, int a5, int a6, int a7, int a8) |
||
330 | { |
||
331 | REMOVE_GRAIN_SORT_AXIS |
||
332 | |||
333 | const int l = FFMAX(FFMAX(mi1, mi2), FFMAX(mi3, mi4)); |
||
334 | const int u = FFMIN(FFMIN(ma1, ma2), FFMIN(ma3, ma4)); |
||
335 | |||
336 | return av_clip(c, FFMIN(l, u), FFMAX(l, u)); |
||
337 | } |
||
338 | |||
339 | static int mode18(int c, int a1, int a2, int a3, int a4, int a5, int a6, int a7, int a8) |
||
340 | { |
||
341 | const int d1 = FFMAX(FFABS(c - a1), FFABS(c - a8)); |
||
342 | const int d2 = FFMAX(FFABS(c - a2), FFABS(c - a7)); |
||
343 | const int d3 = FFMAX(FFABS(c - a3), FFABS(c - a6)); |
||
344 | const int d4 = FFMAX(FFABS(c - a4), FFABS(c - a5)); |
||
345 | |||
346 | const int mindiff = FFMIN(FFMIN(d1, d2), FFMIN(d3, d4)); |
||
347 | |||
348 | if (mindiff == d4) { |
||
349 | return av_clip(c, FFMIN(a4, a5), FFMAX(a4, a5)); |
||
350 | } |
||
351 | if (mindiff == d2) { |
||
352 | return av_clip(c, FFMIN(a2, a7), FFMAX(a2, a7)); |
||
353 | } |
||
354 | if (mindiff == d3) { |
||
355 | return av_clip(c, FFMIN(a3, a6), FFMAX(a3, a6)); |
||
356 | } |
||
357 | |||
358 | return av_clip(c, FFMIN(a1, a8), FFMAX(a1, a8)); |
||
359 | } |
||
360 | |||
361 | static int mode19(int c, int a1, int a2, int a3, int a4, int a5, int a6, int a7, int a8) |
||
362 | { |
||
363 | const int sum = a1 + a2 + a3 + a4 + a5 + a6 + a7 + a8; |
||
364 | const int val = (sum + 4) >> 3; |
||
365 | |||
366 | return val; |
||
367 | } |
||
368 | |||
369 | static int mode20(int c, int a1, int a2, int a3, int a4, int a5, int a6, int a7, int a8) |
||
370 | { |
||
371 | const int sum = a1 + a2 + a3 + a4 + c + a5 + a6 + a7 + a8; |
||
372 | const int val = (sum + 4) / 9; |
||
373 | |||
374 | return val; |
||
375 | } |
||
376 | |||
377 | static int mode21(int c, int a1, int a2, int a3, int a4, int a5, int a6, int a7, int a8) |
||
378 | { |
||
379 | const int l1l = (a1 + a8) >> 1; |
||
380 | const int l2l = (a2 + a7) >> 1; |
||
381 | const int l3l = (a3 + a6) >> 1; |
||
382 | const int l4l = (a4 + a5) >> 1; |
||
383 | |||
384 | const int l1h = (a1 + a8 + 1) >> 1; |
||
385 | const int l2h = (a2 + a7 + 1) >> 1; |
||
386 | const int l3h = (a3 + a6 + 1) >> 1; |
||
387 | const int l4h = (a4 + a5 + 1) >> 1; |
||
388 | |||
389 | const int mi = FFMIN(FFMIN(l1l, l2l), FFMIN(l3l, l4l)); |
||
390 | const int ma = FFMAX(FFMAX(l1h, l2h), FFMAX(l3h, l4h)); |
||
391 | |||
392 | return av_clip(c, mi, ma); |
||
393 | } |
||
394 | |||
395 | static int mode22(int c, int a1, int a2, int a3, int a4, int a5, int a6, int a7, int a8) |
||
396 | { |
||
397 | const int l1 = (a1 + a8 + 1) >> 1; |
||
398 | const int l2 = (a2 + a7 + 1) >> 1; |
||
399 | const int l3 = (a3 + a6 + 1) >> 1; |
||
400 | const int l4 = (a4 + a5 + 1) >> 1; |
||
401 | |||
402 | const int mi = FFMIN(FFMIN(l1, l2), FFMIN(l3, l4)); |
||
403 | const int ma = FFMAX(FFMAX(l1, l2), FFMAX(l3, l4)); |
||
404 | |||
405 | return av_clip(c, mi, ma); |
||
406 | } |
||
407 | |||
408 | static int mode23(int c, int a1, int a2, int a3, int a4, int a5, int a6, int a7, int a8) |
||
409 | { |
||
410 | REMOVE_GRAIN_SORT_AXIS |
||
411 | |||
412 | const int linediff1 = ma1 - mi1; |
||
413 | const int linediff2 = ma2 - mi2; |
||
414 | const int linediff3 = ma3 - mi3; |
||
415 | const int linediff4 = ma4 - mi4; |
||
416 | |||
417 | const int u1 = FFMIN(c - ma1, linediff1); |
||
418 | const int u2 = FFMIN(c - ma2, linediff2); |
||
419 | const int u3 = FFMIN(c - ma3, linediff3); |
||
420 | const int u4 = FFMIN(c - ma4, linediff4); |
||
421 | const int u = FFMAX(FFMAX(FFMAX(u1, u2), FFMAX(u3, u4)), 0); |
||
422 | |||
423 | const int d1 = FFMIN(mi1 - c, linediff1); |
||
424 | const int d2 = FFMIN(mi2 - c, linediff2); |
||
425 | const int d3 = FFMIN(mi3 - c, linediff3); |
||
426 | const int d4 = FFMIN(mi4 - c, linediff4); |
||
427 | const int d = FFMAX(FFMAX(FFMAX(d1, d2), FFMAX(d3, d4)), 0); |
||
428 | |||
429 | return c - u + d; // This probably will never overflow. |
||
430 | } |
||
431 | |||
432 | static int mode24(int c, int a1, int a2, int a3, int a4, int a5, int a6, int a7, int a8) |
||
433 | { |
||
434 | REMOVE_GRAIN_SORT_AXIS |
||
435 | |||
436 | const int linediff1 = ma1 - mi1; |
||
437 | const int linediff2 = ma2 - mi2; |
||
438 | const int linediff3 = ma3 - mi3; |
||
439 | const int linediff4 = ma4 - mi4; |
||
440 | |||
441 | const int tu1 = c - ma1; |
||
442 | const int tu2 = c - ma2; |
||
443 | const int tu3 = c - ma3; |
||
444 | const int tu4 = c - ma4; |
||
445 | |||
446 | const int u1 = FFMIN(tu1, linediff1 - tu1); |
||
447 | const int u2 = FFMIN(tu2, linediff2 - tu2); |
||
448 | const int u3 = FFMIN(tu3, linediff3 - tu3); |
||
449 | const int u4 = FFMIN(tu4, linediff4 - tu4); |
||
450 | const int u = FFMAX(FFMAX(FFMAX(u1, u2), FFMAX(u3, u4)), 0); |
||
451 | |||
452 | const int td1 = mi1 - c; |
||
453 | const int td2 = mi2 - c; |
||
454 | const int td3 = mi3 - c; |
||
455 | const int td4 = mi4 - c; |
||
456 | |||
457 | const int d1 = FFMIN(td1, linediff1 - td1); |
||
458 | const int d2 = FFMIN(td2, linediff2 - td2); |
||
459 | const int d3 = FFMIN(td3, linediff3 - td3); |
||
460 | const int d4 = FFMIN(td4, linediff4 - td4); |
||
461 | const int d = FFMAX(FFMAX(FFMAX(d1, d2), FFMAX(d3, d4)), 0); |
||
462 | |||
463 | return c - u + d; // This probably will never overflow. |
||
464 | } |
||
465 | |||
466 | static int config_input(AVFilterLink *inlink) |
||
467 | { |
||
468 | RemoveGrainContext *s = inlink->dst->priv; |
||
469 | const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(inlink->format); |
||
470 | int i; |
||
471 | |||
472 | s->nb_planes = av_pix_fmt_count_planes(inlink->format); |
||
473 | |||
474 | s->planeheight[1] = s->planeheight[2] = FF_CEIL_RSHIFT(inlink->h, desc->log2_chroma_h); |
||
475 | s->planeheight[0] = s->planeheight[3] = inlink->h; |
||
476 | s->planewidth[1] = s->planewidth[2] = FF_CEIL_RSHIFT(inlink->w, desc->log2_chroma_w); |
||
477 | s->planewidth[0] = s->planewidth[3] = inlink->w; |
||
478 | |||
479 | for (i = 0; i < s->nb_planes; i++) { |
||
480 | switch (s->mode[i]) { |
||
481 | case 1: s->rg[i] = mode01; break; |
||
482 | case 2: s->rg[i] = mode02; break; |
||
483 | case 3: s->rg[i] = mode03; break; |
||
484 | case 4: s->rg[i] = mode04; break; |
||
485 | case 5: s->rg[i] = mode05; break; |
||
486 | case 6: s->rg[i] = mode06; break; |
||
487 | case 7: s->rg[i] = mode07; break; |
||
488 | case 8: s->rg[i] = mode08; break; |
||
489 | case 9: s->rg[i] = mode09; break; |
||
490 | case 10: s->rg[i] = mode10; break; |
||
491 | case 11: s->rg[i] = mode1112; break; |
||
492 | case 12: s->rg[i] = mode1112; break; |
||
493 | case 13: s->skip_odd = 1; |
||
494 | s->rg[i] = mode1314; break; |
||
495 | case 14: s->skip_even = 1; |
||
496 | s->rg[i] = mode1314; break; |
||
497 | case 15: s->skip_odd = 1; |
||
498 | s->rg[i] = mode1516; break; |
||
499 | case 16: s->skip_even = 1; |
||
500 | s->rg[i] = mode1516; break; |
||
501 | case 17: s->rg[i] = mode17; break; |
||
502 | case 18: s->rg[i] = mode18; break; |
||
503 | case 19: s->rg[i] = mode19; break; |
||
504 | case 20: s->rg[i] = mode20; break; |
||
505 | case 21: s->rg[i] = mode21; break; |
||
506 | case 22: s->rg[i] = mode22; break; |
||
507 | case 23: s->rg[i] = mode23; break; |
||
508 | case 24: s->rg[i] = mode24; break; |
||
509 | } |
||
510 | } |
||
511 | |||
512 | if (ARCH_X86) |
||
513 | ff_removegrain_init_x86(s); |
||
514 | |||
515 | return 0; |
||
516 | } |
||
517 | |||
518 | typedef struct ThreadData { |
||
519 | AVFrame *in, *out; |
||
520 | int plane; |
||
521 | } ThreadData; |
||
522 | |||
523 | static int filter_slice(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs) |
||
524 | { |
||
525 | RemoveGrainContext *s = ctx->priv; |
||
526 | ThreadData *td = arg; |
||
527 | AVFrame *in = td->in; |
||
528 | AVFrame *out = td->out; |
||
529 | const int i = td->plane; |
||
530 | const int height = s->planeheight[i]; |
||
531 | const int om = in->linesize[i] - 1; |
||
532 | const int o0 = in->linesize[i] ; |
||
533 | const int op = in->linesize[i] + 1; |
||
534 | int start = (height * jobnr ) / nb_jobs; |
||
535 | int end = (height * (jobnr+1)) / nb_jobs; |
||
536 | int x, y; |
||
537 | |||
538 | start = FFMAX(1, start); |
||
539 | end = FFMIN(height-1, end); |
||
540 | for (y = start; y < end; y++) { |
||
541 | uint8_t *dst = out->data[i]; |
||
542 | uint8_t *src = in->data[i]; |
||
543 | |||
544 | src = in->data[i] + y * in->linesize[i]; |
||
545 | dst = out->data[i] + y * out->linesize[i]; |
||
546 | |||
547 | if (s->skip_even && !(y & 1)) { |
||
548 | memcpy(dst, src, s->planewidth[i]); |
||
549 | continue; |
||
550 | } |
||
551 | if (s->skip_odd && y & 1) { |
||
552 | memcpy(dst, src, s->planewidth[i]); |
||
553 | continue; |
||
554 | } |
||
555 | |||
556 | *dst++ = *src++; |
||
557 | |||
558 | if (s->fl[i]) { |
||
559 | int w_asm = (s->planewidth[i] - 2) & ~15; |
||
560 | |||
561 | s->fl[i](dst, src, in->linesize[i], w_asm); |
||
562 | |||
563 | x = 1 + w_asm; |
||
564 | dst += w_asm; |
||
565 | src += w_asm; |
||
566 | } else |
||
567 | x = 1; |
||
568 | |||
569 | for (; x < s->planewidth[i] - 1; x++) { |
||
570 | const int a1 = src[-op]; |
||
571 | const int a2 = src[-o0]; |
||
572 | const int a3 = src[-om]; |
||
573 | const int a4 = src[-1 ]; |
||
574 | const int c = src[ 0 ]; |
||
575 | const int a5 = src[ 1 ]; |
||
576 | const int a6 = src[ om]; |
||
577 | const int a7 = src[ o0]; |
||
578 | const int a8 = src[ op]; |
||
579 | |||
580 | const int res = s->rg[i](c, a1, a2, a3, a4, a5, a6, a7, a8); |
||
581 | |||
582 | *dst = res; |
||
583 | dst++, src++; |
||
584 | } |
||
585 | dst[0] = src[0]; |
||
586 | } |
||
587 | |||
588 | return 0; |
||
589 | } |
||
590 | |||
591 | static int filter_frame(AVFilterLink *inlink, AVFrame *in) |
||
592 | { |
||
593 | AVFilterContext *ctx = inlink->dst; |
||
594 | AVFilterLink *outlink = ctx->outputs[0]; |
||
595 | RemoveGrainContext *s = ctx->priv; |
||
596 | ThreadData td; |
||
597 | AVFrame *out; |
||
598 | int i; |
||
599 | |||
600 | out = ff_get_video_buffer(outlink, outlink->w, outlink->h); |
||
601 | if (!out) { |
||
602 | av_frame_free(&in); |
||
603 | return AVERROR(ENOMEM); |
||
604 | } |
||
605 | av_frame_copy_props(out, in); |
||
606 | |||
607 | for (i = 0; i < s->nb_planes; i++) { |
||
608 | uint8_t *dst = out->data[i]; |
||
609 | uint8_t *src = in->data[i]; |
||
610 | |||
611 | if (s->mode[i] == 0) { |
||
612 | av_image_copy_plane(dst, out->linesize[i], |
||
613 | src, in->linesize[i], |
||
614 | s->planewidth[i], s->planeheight[i]); |
||
615 | continue; |
||
616 | } |
||
617 | |||
618 | memcpy(dst, src, s->planewidth[i]); |
||
619 | |||
620 | td.in = in; td.out = out; td.plane = i; |
||
621 | ctx->internal->execute(ctx, filter_slice, &td, NULL, |
||
622 | FFMIN(s->planeheight[i], ctx->graph->nb_threads)); |
||
623 | |||
624 | src = in->data[i] + (s->planeheight[i] - 1) * in->linesize[i]; |
||
625 | dst = out->data[i] + (s->planeheight[i] - 1) * out->linesize[i]; |
||
626 | memcpy(dst, src, s->planewidth[i]); |
||
627 | } |
||
628 | |||
629 | av_frame_free(&in); |
||
630 | return ff_filter_frame(outlink, out); |
||
631 | } |
||
632 | |||
633 | static const AVFilterPad removegrain_inputs[] = { |
||
634 | { |
||
635 | .name = "default", |
||
636 | .type = AVMEDIA_TYPE_VIDEO, |
||
637 | .filter_frame = filter_frame, |
||
638 | .config_props = config_input, |
||
639 | }, |
||
640 | { NULL } |
||
641 | }; |
||
642 | |||
643 | static const AVFilterPad removegrain_outputs[] = { |
||
644 | { |
||
645 | .name = "default", |
||
646 | .type = AVMEDIA_TYPE_VIDEO, |
||
647 | }, |
||
648 | { NULL } |
||
649 | }; |
||
650 | |||
651 | AVFilter ff_vf_removegrain = { |
||
652 | .name = "removegrain", |
||
653 | .description = NULL_IF_CONFIG_SMALL("Remove grain."), |
||
654 | .priv_size = sizeof(RemoveGrainContext), |
||
655 | .query_formats = query_formats, |
||
656 | .inputs = removegrain_inputs, |
||
657 | .outputs = removegrain_outputs, |
||
658 | .priv_class = &removegrain_class, |
||
659 | .flags = AVFILTER_FLAG_SUPPORT_TIMELINE_GENERIC | AVFILTER_FLAG_SLICE_THREADS, |
||
660 | };>>>>><>><>><>><>><>><>><>><>> |