63 const float *
src,
int src_linesize,
64 float *dst,
int dst_linesize,
70 float **
src,
int src_linesize,
76 #define DEFAULT_NBITS 3
78 #define OFFSET(x) offsetof(DCTdnoizContext, x)
79 #define FLAGS AV_OPT_FLAG_FILTERING_PARAM|AV_OPT_FLAG_VIDEO_PARAM
93 int dst_stridea,
int dst_strideb,
94 int src_stridea,
int src_strideb)
98 for (
i = 0;
i < 8;
i++) {
99 const float x00 =
src[0*src_stridea] +
src[7*src_stridea];
100 const float x01 =
src[1*src_stridea] +
src[6*src_stridea];
101 const float x02 =
src[2*src_stridea] +
src[5*src_stridea];
102 const float x03 =
src[3*src_stridea] +
src[4*src_stridea];
103 const float x04 =
src[0*src_stridea] -
src[7*src_stridea];
104 const float x05 =
src[1*src_stridea] -
src[6*src_stridea];
105 const float x06 =
src[2*src_stridea] -
src[5*src_stridea];
106 const float x07 =
src[3*src_stridea] -
src[4*src_stridea];
107 const float x08 = x00 + x03;
108 const float x09 = x01 + x02;
109 const float x0a = x00 - x03;
110 const float x0b = x01 - x02;
111 const float x0c = 1.38703984532215f*x04 + 0.275899379282943f*x07;
112 const float x0d = 1.17587560241936f*x05 + 0.785694958387102f*x06;
113 const float x0e = -0.785694958387102f*x05 + 1.17587560241936f*x06;
114 const float x0f = 0.275899379282943f*x04 - 1.38703984532215f*x07;
115 const float x10 = 0.353553390593274f * (x0c - x0d);
116 const float x11 = 0.353553390593274f * (x0e - x0f);
117 dst[0*dst_stridea] = 0.353553390593274f * (x08 + x09);
118 dst[1*dst_stridea] = 0.353553390593274f * (x0c + x0d);
119 dst[2*dst_stridea] = 0.461939766255643f*x0a + 0.191341716182545f*x0b;
120 dst[3*dst_stridea] = 0.707106781186547f * (x10 - x11);
121 dst[4*dst_stridea] = 0.353553390593274f * (x08 - x09);
122 dst[5*dst_stridea] = 0.707106781186547f * (x10 + x11);
123 dst[6*dst_stridea] = 0.191341716182545f*x0a - 0.461939766255643f*x0b;
124 dst[7*dst_stridea] = 0.353553390593274f * (x0e + x0f);
131 int dst_stridea,
int dst_strideb,
132 int src_stridea,
int src_strideb,
137 for (
i = 0;
i < 8;
i++) {
138 const float x00 = 1.4142135623731f *
src[0*src_stridea];
139 const float x01 = 1.38703984532215f *
src[1*src_stridea] + 0.275899379282943f*
src[7*src_stridea];
140 const float x02 = 1.30656296487638f *
src[2*src_stridea] + 0.541196100146197f*
src[6*src_stridea];
141 const float x03 = 1.17587560241936f *
src[3*src_stridea] + 0.785694958387102f*
src[5*src_stridea];
142 const float x04 = 1.4142135623731f *
src[4*src_stridea];
143 const float x05 = -0.785694958387102f*
src[3*src_stridea] + 1.17587560241936f*
src[5*src_stridea];
144 const float x06 = 0.541196100146197f*
src[2*src_stridea] - 1.30656296487638f*
src[6*src_stridea];
145 const float x07 = -0.275899379282943f*
src[1*src_stridea] + 1.38703984532215f*
src[7*src_stridea];
146 const float x09 = x00 + x04;
147 const float x0a = x01 + x03;
148 const float x0b = 1.4142135623731f*x02;
149 const float x0c = x00 - x04;
150 const float x0d = x01 - x03;
151 const float x0e = 0.353553390593274f * (x09 - x0b);
152 const float x0f = 0.353553390593274f * (x0c + x0d);
153 const float x10 = 0.353553390593274f * (x0c - x0d);
154 const float x11 = 1.4142135623731f*x06;
155 const float x12 = x05 + x07;
156 const float x13 = x05 - x07;
157 const float x14 = 0.353553390593274f * (x11 + x12);
158 const float x15 = 0.353553390593274f * (x11 - x12);
159 const float x16 = 0.5f*x13;
160 dst[0*dst_stridea] = (
add ? dst[ 0*dst_stridea] : 0) + 0.25f * (x09 + x0b) + 0.353553390593274f*x0a;
161 dst[1*dst_stridea] = (
add ? dst[ 1*dst_stridea] : 0) + 0.707106781186547f * (x0f + x15);
162 dst[2*dst_stridea] = (
add ? dst[ 2*dst_stridea] : 0) + 0.707106781186547f * (x0f - x15);
163 dst[3*dst_stridea] = (
add ? dst[ 3*dst_stridea] : 0) + 0.707106781186547f * (x0e + x16);
164 dst[4*dst_stridea] = (
add ? dst[ 4*dst_stridea] : 0) + 0.707106781186547f * (x0e - x16);
165 dst[5*dst_stridea] = (
add ? dst[ 5*dst_stridea] : 0) + 0.707106781186547f * (x10 - x14);
166 dst[6*dst_stridea] = (
add ? dst[ 6*dst_stridea] : 0) + 0.707106781186547f * (x10 + x14);
167 dst[7*dst_stridea] = (
add ? dst[ 7*dst_stridea] : 0) + 0.25f * (x09 + x0b) - 0.353553390593274f*x0a;
175 int dst_stridea,
int dst_strideb,
176 int src_stridea,
int src_strideb)
180 for (
i = 0;
i < 16;
i++) {
181 const float x00 =
src[ 0*src_stridea] +
src[15*src_stridea];
182 const float x01 =
src[ 1*src_stridea] +
src[14*src_stridea];
183 const float x02 =
src[ 2*src_stridea] +
src[13*src_stridea];
184 const float x03 =
src[ 3*src_stridea] +
src[12*src_stridea];
185 const float x04 =
src[ 4*src_stridea] +
src[11*src_stridea];
186 const float x05 =
src[ 5*src_stridea] +
src[10*src_stridea];
187 const float x06 =
src[ 6*src_stridea] +
src[ 9*src_stridea];
188 const float x07 =
src[ 7*src_stridea] +
src[ 8*src_stridea];
189 const float x08 =
src[ 0*src_stridea] -
src[15*src_stridea];
190 const float x09 =
src[ 1*src_stridea] -
src[14*src_stridea];
191 const float x0a =
src[ 2*src_stridea] -
src[13*src_stridea];
192 const float x0b =
src[ 3*src_stridea] -
src[12*src_stridea];
193 const float x0c =
src[ 4*src_stridea] -
src[11*src_stridea];
194 const float x0d =
src[ 5*src_stridea] -
src[10*src_stridea];
195 const float x0e =
src[ 6*src_stridea] -
src[ 9*src_stridea];
196 const float x0f =
src[ 7*src_stridea] -
src[ 8*src_stridea];
197 const float x10 = x00 + x07;
198 const float x11 = x01 + x06;
199 const float x12 = x02 + x05;
200 const float x13 = x03 + x04;
201 const float x14 = x00 - x07;
202 const float x15 = x01 - x06;
203 const float x16 = x02 - x05;
204 const float x17 = x03 - x04;
205 const float x18 = x10 + x13;
206 const float x19 = x11 + x12;
207 const float x1a = x10 - x13;
208 const float x1b = x11 - x12;
209 const float x1c = 1.38703984532215f*x14 + 0.275899379282943f*x17;
210 const float x1d = 1.17587560241936f*x15 + 0.785694958387102f*x16;
211 const float x1e = -0.785694958387102f*x15 + 1.17587560241936f *x16;
212 const float x1f = 0.275899379282943f*x14 - 1.38703984532215f *x17;
213 const float x20 = 0.25f * (x1c - x1d);
214 const float x21 = 0.25f * (x1e - x1f);
215 const float x22 = 1.40740373752638f *x08 + 0.138617169199091f*x0f;
216 const float x23 = 1.35331800117435f *x09 + 0.410524527522357f*x0e;
217 const float x24 = 1.24722501298667f *x0a + 0.666655658477747f*x0d;
218 const float x25 = 1.09320186700176f *x0b + 0.897167586342636f*x0c;
219 const float x26 = -0.897167586342636f*x0b + 1.09320186700176f *x0c;
220 const float x27 = 0.666655658477747f*x0a - 1.24722501298667f *x0d;
221 const float x28 = -0.410524527522357f*x09 + 1.35331800117435f *x0e;
222 const float x29 = 0.138617169199091f*x08 - 1.40740373752638f *x0f;
223 const float x2a = x22 + x25;
224 const float x2b = x23 + x24;
225 const float x2c = x22 - x25;
226 const float x2d = x23 - x24;
227 const float x2e = 0.25f * (x2a - x2b);
228 const float x2f = 0.326640741219094f*x2c + 0.135299025036549f*x2d;
229 const float x30 = 0.135299025036549f*x2c - 0.326640741219094f*x2d;
230 const float x31 = x26 + x29;
231 const float x32 = x27 + x28;
232 const float x33 = x26 - x29;
233 const float x34 = x27 - x28;
234 const float x35 = 0.25f * (x31 - x32);
235 const float x36 = 0.326640741219094f*x33 + 0.135299025036549f*x34;
236 const float x37 = 0.135299025036549f*x33 - 0.326640741219094f*x34;
237 dst[ 0*dst_stridea] = 0.25f * (x18 + x19);
238 dst[ 1*dst_stridea] = 0.25f * (x2a + x2b);
239 dst[ 2*dst_stridea] = 0.25f * (x1c + x1d);
240 dst[ 3*dst_stridea] = 0.707106781186547f * (x2f - x37);
241 dst[ 4*dst_stridea] = 0.326640741219094f*x1a + 0.135299025036549f*x1b;
242 dst[ 5*dst_stridea] = 0.707106781186547f * (x2f + x37);
243 dst[ 6*dst_stridea] = 0.707106781186547f * (x20 - x21);
244 dst[ 7*dst_stridea] = 0.707106781186547f * (x2e + x35);
245 dst[ 8*dst_stridea] = 0.25f * (x18 - x19);
246 dst[ 9*dst_stridea] = 0.707106781186547f * (x2e - x35);
247 dst[10*dst_stridea] = 0.707106781186547f * (x20 + x21);
248 dst[11*dst_stridea] = 0.707106781186547f * (x30 - x36);
249 dst[12*dst_stridea] = 0.135299025036549f*x1a - 0.326640741219094f*x1b;
250 dst[13*dst_stridea] = 0.707106781186547f * (x30 + x36);
251 dst[14*dst_stridea] = 0.25f * (x1e + x1f);
252 dst[15*dst_stridea] = 0.25f * (x31 + x32);
259 int dst_stridea,
int dst_strideb,
260 int src_stridea,
int src_strideb,
265 for (
i = 0;
i < 16;
i++) {
266 const float x00 = 1.4142135623731f *
src[ 0*src_stridea];
267 const float x01 = 1.40740373752638f *
src[ 1*src_stridea] + 0.138617169199091f*
src[15*src_stridea];
268 const float x02 = 1.38703984532215f *
src[ 2*src_stridea] + 0.275899379282943f*
src[14*src_stridea];
269 const float x03 = 1.35331800117435f *
src[ 3*src_stridea] + 0.410524527522357f*
src[13*src_stridea];
270 const float x04 = 1.30656296487638f *
src[ 4*src_stridea] + 0.541196100146197f*
src[12*src_stridea];
271 const float x05 = 1.24722501298667f *
src[ 5*src_stridea] + 0.666655658477747f*
src[11*src_stridea];
272 const float x06 = 1.17587560241936f *
src[ 6*src_stridea] + 0.785694958387102f*
src[10*src_stridea];
273 const float x07 = 1.09320186700176f *
src[ 7*src_stridea] + 0.897167586342636f*
src[ 9*src_stridea];
274 const float x08 = 1.4142135623731f *
src[ 8*src_stridea];
275 const float x09 = -0.897167586342636f*
src[ 7*src_stridea] + 1.09320186700176f*
src[ 9*src_stridea];
276 const float x0a = 0.785694958387102f*
src[ 6*src_stridea] - 1.17587560241936f*
src[10*src_stridea];
277 const float x0b = -0.666655658477747f*
src[ 5*src_stridea] + 1.24722501298667f*
src[11*src_stridea];
278 const float x0c = 0.541196100146197f*
src[ 4*src_stridea] - 1.30656296487638f*
src[12*src_stridea];
279 const float x0d = -0.410524527522357f*
src[ 3*src_stridea] + 1.35331800117435f*
src[13*src_stridea];
280 const float x0e = 0.275899379282943f*
src[ 2*src_stridea] - 1.38703984532215f*
src[14*src_stridea];
281 const float x0f = -0.138617169199091f*
src[ 1*src_stridea] + 1.40740373752638f*
src[15*src_stridea];
282 const float x12 = x00 + x08;
283 const float x13 = x01 + x07;
284 const float x14 = x02 + x06;
285 const float x15 = x03 + x05;
286 const float x16 = 1.4142135623731f*x04;
287 const float x17 = x00 - x08;
288 const float x18 = x01 - x07;
289 const float x19 = x02 - x06;
290 const float x1a = x03 - x05;
291 const float x1d = x12 + x16;
292 const float x1e = x13 + x15;
293 const float x1f = 1.4142135623731f*x14;
294 const float x20 = x12 - x16;
295 const float x21 = x13 - x15;
296 const float x22 = 0.25f * (x1d - x1f);
297 const float x23 = 0.25f * (x20 + x21);
298 const float x24 = 0.25f * (x20 - x21);
299 const float x25 = 1.4142135623731f*x17;
300 const float x26 = 1.30656296487638f*x18 + 0.541196100146197f*x1a;
301 const float x27 = 1.4142135623731f*x19;
302 const float x28 = -0.541196100146197f*x18 + 1.30656296487638f*x1a;
303 const float x29 = 0.176776695296637f * (x25 + x27) + 0.25f*x26;
304 const float x2a = 0.25f * (x25 - x27);
305 const float x2b = 0.176776695296637f * (x25 + x27) - 0.25f*x26;
306 const float x2c = 0.353553390593274f*x28;
307 const float x1b = 0.707106781186547f * (x2a - x2c);
308 const float x1c = 0.707106781186547f * (x2a + x2c);
309 const float x2d = 1.4142135623731f*x0c;
310 const float x2e = x0b + x0d;
311 const float x2f = x0a + x0e;
312 const float x30 = x09 + x0f;
313 const float x31 = x09 - x0f;
314 const float x32 = x0a - x0e;
315 const float x33 = x0b - x0d;
316 const float x37 = 1.4142135623731f*x2d;
317 const float x38 = 1.30656296487638f*x2e + 0.541196100146197f*x30;
318 const float x39 = 1.4142135623731f*x2f;
319 const float x3a = -0.541196100146197f*x2e + 1.30656296487638f*x30;
320 const float x3b = 0.176776695296637f * (x37 + x39) + 0.25f*x38;
321 const float x3c = 0.25f * (x37 - x39);
322 const float x3d = 0.176776695296637f * (x37 + x39) - 0.25f*x38;
323 const float x3e = 0.353553390593274f*x3a;
324 const float x34 = 0.707106781186547f * (x3c - x3e);
325 const float x35 = 0.707106781186547f * (x3c + x3e);
326 const float x3f = 1.4142135623731f*x32;
327 const float x40 = x31 + x33;
328 const float x41 = x31 - x33;
329 const float x42 = 0.25f * (x3f + x40);
330 const float x43 = 0.25f * (x3f - x40);
331 const float x44 = 0.353553390593274f*x41;
332 dst[ 0*dst_stridea] = (
add ? dst[ 0*dst_stridea] : 0) + 0.176776695296637f * (x1d + x1f) + 0.25f*x1e;
333 dst[ 1*dst_stridea] = (
add ? dst[ 1*dst_stridea] : 0) + 0.707106781186547f * (x29 + x3d);
334 dst[ 2*dst_stridea] = (
add ? dst[ 2*dst_stridea] : 0) + 0.707106781186547f * (x29 - x3d);
335 dst[ 3*dst_stridea] = (
add ? dst[ 3*dst_stridea] : 0) + 0.707106781186547f * (x23 - x43);
336 dst[ 4*dst_stridea] = (
add ? dst[ 4*dst_stridea] : 0) + 0.707106781186547f * (x23 + x43);
337 dst[ 5*dst_stridea] = (
add ? dst[ 5*dst_stridea] : 0) + 0.707106781186547f * (x1b - x35);
338 dst[ 6*dst_stridea] = (
add ? dst[ 6*dst_stridea] : 0) + 0.707106781186547f * (x1b + x35);
339 dst[ 7*dst_stridea] = (
add ? dst[ 7*dst_stridea] : 0) + 0.707106781186547f * (x22 + x44);
340 dst[ 8*dst_stridea] = (
add ? dst[ 8*dst_stridea] : 0) + 0.707106781186547f * (x22 - x44);
341 dst[ 9*dst_stridea] = (
add ? dst[ 9*dst_stridea] : 0) + 0.707106781186547f * (x1c + x34);
342 dst[10*dst_stridea] = (
add ? dst[10*dst_stridea] : 0) + 0.707106781186547f * (x1c - x34);
343 dst[11*dst_stridea] = (
add ? dst[11*dst_stridea] : 0) + 0.707106781186547f * (x24 + x42);
344 dst[12*dst_stridea] = (
add ? dst[12*dst_stridea] : 0) + 0.707106781186547f * (x24 - x42);
345 dst[13*dst_stridea] = (
add ? dst[13*dst_stridea] : 0) + 0.707106781186547f * (x2b - x3b);
346 dst[14*dst_stridea] = (
add ? dst[14*dst_stridea] : 0) + 0.707106781186547f * (x2b + x3b);
347 dst[15*dst_stridea] = (
add ? dst[15*dst_stridea] : 0) + 0.176776695296637f * (x1d + x1f) - 0.25f*x1e;
353 #define DEF_FILTER_FREQ_FUNCS(bsize) \
354 static av_always_inline void filter_freq_##bsize(const float *src, int src_linesize, \
355 float *dst, int dst_linesize, \
356 AVExpr *expr, double *var_values, \
360 DECLARE_ALIGNED(32, float, tmp_block1)[bsize * bsize]; \
361 DECLARE_ALIGNED(32, float, tmp_block2)[bsize * bsize]; \
364 fdct##bsize##_1d(tmp_block1, src, 1, bsize, 1, src_linesize); \
365 fdct##bsize##_1d(tmp_block2, tmp_block1, bsize, 1, bsize, 1); \
367 for (i = 0; i < bsize*bsize; i++) { \
368 float *b = &tmp_block2[i]; \
371 var_values[VAR_C] = fabsf(*b); \
372 *b *= av_expr_eval(expr, var_values, NULL); \
374 if (fabsf(*b) < sigma_th) \
380 idct##bsize##_1d(tmp_block1, tmp_block2, 1, bsize, 1, bsize, 0); \
381 idct##bsize##_1d(dst, tmp_block1, dst_linesize, 1, bsize, 1, 1); \
384 static void filter_freq_sigma_##bsize(DCTdnoizContext *s, \
385 const float *src, int src_linesize, \
386 float *dst, int dst_linesize, int thread_id) \
388 filter_freq_##bsize(src, src_linesize, dst, dst_linesize, NULL, NULL, s->th); \
391 static void filter_freq_expr_##bsize(DCTdnoizContext *s, \
392 const float *src, int src_linesize, \
393 float *dst, int dst_linesize, int thread_id) \
395 filter_freq_##bsize(src, src_linesize, dst, dst_linesize, \
396 s->expr[thread_id], s->var_values[thread_id], 0); \
402 #define DCT3X3_0_0 0.5773502691896258f
403 #define DCT3X3_0_1 0.5773502691896258f
404 #define DCT3X3_0_2 0.5773502691896258f
405 #define DCT3X3_1_0 0.7071067811865475f
406 #define DCT3X3_1_2 -0.7071067811865475f
407 #define DCT3X3_2_0 0.4082482904638631f
408 #define DCT3X3_2_1 -0.8164965809277261f
409 #define DCT3X3_2_2 0.4082482904638631f
417 float *dstp_r = dst[0];
418 float *dstp_g = dst[1];
419 float *dstp_b = dst[2];
422 for (y = 0; y <
h; y++) {
423 for (x = 0; x <
w; x++) {
429 srcp += src_linesize -
w * 3;
430 dstp_r += dst_linesize;
431 dstp_g += dst_linesize;
432 dstp_b += dst_linesize;
437 float **
src,
int src_linesize,
442 const float *src_r =
src[0];
443 const float *src_g =
src[1];
444 const float *src_b =
src[2];
447 for (y = 0; y <
h; y++) {
448 for (x = 0; x <
w; x++) {
454 dstp += dst_linesize -
w * 3;
455 src_r += src_linesize;
456 src_g += src_linesize;
457 src_b += src_linesize;
461 #define DECLARE_COLOR_FUNCS(name, r, g, b) \
462 static void color_decorrelation_##name(float **dst, int dst_linesize, \
463 const uint8_t **src, int src_linesize, \
466 color_decorrelation(dst, dst_linesize, src, src_linesize, w, h, r, g, b); \
469 static void color_correlation_##name(uint8_t **dst, int dst_linesize, \
470 float **src, int src_linesize, \
473 color_correlation(dst, dst_linesize, src, src_linesize, w, h, r, g, b); \
484 float *dstp_r = dst[0];
485 float *dstp_g = dst[1];
486 float *dstp_b = dst[2];
491 for (y = 0; y <
h; y++) {
492 for (x = 0; x <
w; x++) {
497 srcp_r += src_linesize;
498 srcp_g += src_linesize;
499 srcp_b += src_linesize;
500 dstp_r += dst_linesize;
501 dstp_g += dst_linesize;
502 dstp_b += dst_linesize;
507 float **
src,
int src_linesize,
511 const float *src_r =
src[0];
512 const float *src_g =
src[1];
513 const float *src_b =
src[2];
518 for (y = 0; y <
h; y++) {
519 for (x = 0; x <
w; x++) {
524 dstp_r += dst_linesize;
525 dstp_g += dst_linesize;
526 dstp_b += dst_linesize;
527 src_r += src_linesize;
528 src_g += src_linesize;
529 src_b += src_linesize;
537 int i, x, y, bx, by, linesize, *iweights, max_slice_h, slice_h;
538 const int bsize = 1 <<
s->n;
542 s->color_decorrelation = color_decorrelation_bgr;
543 s->color_correlation = color_correlation_bgr;
546 s->color_decorrelation = color_decorrelation_rgb;
547 s->color_correlation = color_correlation_rgb;
557 s->pr_width = inlink->
w - (inlink->
w -
bsize) %
s->step;
558 s->pr_height = inlink->
h - (inlink->
h -
bsize) %
s->step;
559 if (
s->pr_width != inlink->
w)
561 inlink->
w -
s->pr_width);
562 if (
s->pr_height != inlink->
h)
564 inlink->
h -
s->pr_height);
566 max_slice_h =
s->pr_height / ((
s->bsize - 1) * 2);
571 s->p_linesize = linesize =
FFALIGN(
s->pr_width, 32);
572 for (
i = 0;
i < 2;
i++) {
576 if (!
s->cbuf[
i][0] || !
s->cbuf[
i][1] || !
s->cbuf[
i][2])
583 for (
i = 0;
i <
s->nb_threads;
i++) {
594 slice_h = (
int)
ceilf(
s->pr_height / (
float)
s->nb_threads) + (
s->bsize - 1) * 2;
595 for (
i = 0;
i <
s->nb_threads;
i++) {
601 s->weights =
av_malloc(
s->pr_height * linesize *
sizeof(*
s->weights));
604 iweights =
av_calloc(
s->pr_height, linesize *
sizeof(*iweights));
607 for (y = 0; y <
s->pr_height -
bsize + 1; y +=
s->step)
608 for (x = 0; x <
s->pr_width -
bsize + 1; x +=
s->step)
609 for (by = 0; by <
bsize; by++)
610 for (bx = 0; bx <
bsize; bx++)
611 iweights[(y + by)*linesize + x + bx]++;
612 for (y = 0; y <
s->pr_height; y++)
613 for (x = 0; x <
s->pr_width; x++)
614 s->weights[y*linesize + x] = 1. / iweights[y*linesize + x];
624 s->bsize = 1 <<
s->n;
625 if (
s->overlap == -1)
626 s->overlap =
s->bsize - 1;
628 if (
s->overlap >
s->bsize - 1) {
630 "with a block size of %dx%d\n",
631 s->bsize - 1,
s->bsize,
s->bsize);
637 case 3:
s->filter_freq_func = filter_freq_expr_8;
break;
638 case 4:
s->filter_freq_func = filter_freq_expr_16;
break;
643 case 3:
s->filter_freq_func = filter_freq_sigma_8;
break;
644 case 4:
s->filter_freq_func = filter_freq_sigma_16;
break;
649 s->th =
s->sigma * 3.;
650 s->step =
s->bsize -
s->overlap;
672 void *
arg,
int jobnr,
int nb_jobs)
677 const int w =
s->pr_width;
678 const int h =
s->pr_height;
679 const int slice_start = (
h * jobnr ) / nb_jobs;
680 const int slice_end = (
h * (jobnr+1)) / nb_jobs;
681 const int slice_start_ctx =
FFMAX(slice_start -
s->bsize + 1, 0);
683 const int slice_h = slice_end_ctx - slice_start_ctx;
684 const int src_linesize =
s->p_linesize;
685 const int dst_linesize =
s->p_linesize;
686 const int slice_linesize =
s->p_linesize;
688 const float *
src =
td->src + slice_start_ctx * src_linesize;
689 const float *
weights =
s->weights + slice_start * dst_linesize;
690 float *slice =
s->slices[jobnr];
693 memset(slice, 0, (slice_h +
s->bsize - 1) * dst_linesize *
sizeof(*slice));
696 for (y = 0; y < slice_h; y +=
s->step) {
697 for (x = 0; x <
w -
s->bsize + 1; x +=
s->step)
698 s->filter_freq_func(
s,
src + x, src_linesize,
699 slice + x, slice_linesize,
701 src +=
s->step * src_linesize;
702 slice +=
s->step * slice_linesize;
706 slice =
s->slices[jobnr] + (slice_start - slice_start_ctx) * slice_linesize;
707 dst =
td->dst + slice_start * dst_linesize;
708 for (y = slice_start; y <
slice_end; y++) {
709 for (x = 0; x <
w; x++)
710 dst[x] = slice[x] *
weights[x];
711 slice += slice_linesize;
740 s->color_decorrelation(
s->cbuf[0],
s->p_linesize,
742 s->pr_width,
s->pr_height);
743 for (plane = 0; plane < 3; plane++) {
745 .src =
s->cbuf[0][plane],
746 .dst =
s->cbuf[1][plane],
750 s->color_correlation(
out->data,
out->linesize[0],
751 s->cbuf[1],
s->p_linesize,
752 s->pr_width,
s->pr_height);
758 const int dst_linesize =
out->linesize[0];
759 const int src_linesize =
in->linesize[0];
760 const int hpad = (inlink->
w -
s->pr_width) * 3;
761 const int vpad = (inlink->
h -
s->pr_height);
764 uint8_t *dstp = dst +
s->pr_width * 3;
767 for (y = 0; y <
s->pr_height; y++) {
768 memcpy(dstp, srcp, hpad);
769 dstp += dst_linesize;
770 srcp += src_linesize;
774 uint8_t *dstp = dst +
s->pr_height * dst_linesize;
775 const uint8_t *srcp =
src +
s->pr_height * src_linesize;
777 for (y = 0; y < vpad; y++) {
778 memcpy(dstp, srcp, inlink->
w * 3);
779 dstp += dst_linesize;
780 srcp += src_linesize;
796 for (
i = 0;
i < 2;
i++) {
801 for (
i = 0;
i <
s->nb_threads;
i++) {
834 .priv_class = &dctdnoiz_class,
static const AVFilterPad inputs[]
static const AVFilterPad outputs[]
uint8_t pi<< 24) CONV_FUNC_GROUP(AV_SAMPLE_FMT_FLT, float, AV_SAMPLE_FMT_U8, uint8_t,(*(const uint8_t *) pi - 0x80) *(1.0f/(1<< 7))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_DBL, double, AV_SAMPLE_FMT_U8, uint8_t,(*(const uint8_t *) pi - 0x80) *(1.0/(1<< 7))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_S16, int16_t,(*(const int16_t *) pi >> 8)+0x80) CONV_FUNC_GROUP(AV_SAMPLE_FMT_FLT, float, AV_SAMPLE_FMT_S16, int16_t, *(const int16_t *) pi *(1.0f/(1<< 15))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_DBL, double, AV_SAMPLE_FMT_S16, int16_t, *(const int16_t *) pi *(1.0/(1<< 15))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_S32, int32_t,(*(const int32_t *) pi >> 24)+0x80) CONV_FUNC_GROUP(AV_SAMPLE_FMT_FLT, float, AV_SAMPLE_FMT_S32, int32_t, *(const int32_t *) pi *(1.0f/(1U<< 31))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_DBL, double, AV_SAMPLE_FMT_S32, int32_t, *(const int32_t *) pi *(1.0/(1U<< 31))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_FLT, float, av_clip_uint8(lrintf(*(const float *) pi *(1<< 7))+0x80)) CONV_FUNC_GROUP(AV_SAMPLE_FMT_S16, int16_t, AV_SAMPLE_FMT_FLT, float, av_clip_int16(lrintf(*(const float *) pi *(1<< 15)))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_S32, int32_t, AV_SAMPLE_FMT_FLT, float, av_clipl_int32(llrintf(*(const float *) pi *(1U<< 31)))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_DBL, double, av_clip_uint8(lrint(*(const double *) pi *(1<< 7))+0x80)) CONV_FUNC_GROUP(AV_SAMPLE_FMT_S16, int16_t, AV_SAMPLE_FMT_DBL, double, av_clip_int16(lrint(*(const double *) pi *(1<< 15)))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_S32, int32_t, AV_SAMPLE_FMT_DBL, double, av_clipl_int32(llrint(*(const double *) pi *(1U<< 31)))) #define SET_CONV_FUNC_GROUP(ofmt, ifmt) static void set_generic_function(AudioConvert *ac) { } void ff_audio_convert_free(AudioConvert **ac) { if(! *ac) return;ff_dither_free(&(*ac) ->dc);av_freep(ac);} AudioConvert *ff_audio_convert_alloc(AVAudioResampleContext *avr, enum AVSampleFormat out_fmt, enum AVSampleFormat in_fmt, int channels, int sample_rate, int apply_map) { AudioConvert *ac;int in_planar, out_planar;ac=av_mallocz(sizeof(*ac));if(!ac) return NULL;ac->avr=avr;ac->out_fmt=out_fmt;ac->in_fmt=in_fmt;ac->channels=channels;ac->apply_map=apply_map;if(avr->dither_method !=AV_RESAMPLE_DITHER_NONE &&av_get_packed_sample_fmt(out_fmt)==AV_SAMPLE_FMT_S16 &&av_get_bytes_per_sample(in_fmt) > 2) { ac->dc=ff_dither_alloc(avr, out_fmt, in_fmt, channels, sample_rate, apply_map);if(!ac->dc) { av_free(ac);return NULL;} return ac;} in_planar=ff_sample_fmt_is_planar(in_fmt, channels);out_planar=ff_sample_fmt_is_planar(out_fmt, channels);if(in_planar==out_planar) { ac->func_type=CONV_FUNC_TYPE_FLAT;ac->planes=in_planar ? ac->channels :1;} else if(in_planar) ac->func_type=CONV_FUNC_TYPE_INTERLEAVE;else ac->func_type=CONV_FUNC_TYPE_DEINTERLEAVE;set_generic_function(ac);if(ARCH_AARCH64) ff_audio_convert_init_aarch64(ac);if(ARCH_ARM) ff_audio_convert_init_arm(ac);if(ARCH_X86) ff_audio_convert_init_x86(ac);return ac;} int ff_audio_convert(AudioConvert *ac, AudioData *out, AudioData *in) { int use_generic=1;int len=in->nb_samples;int p;if(ac->dc) { av_log(ac->avr, AV_LOG_TRACE, "%d samples - audio_convert: %s to %s (dithered)\n", len, av_get_sample_fmt_name(ac->in_fmt), av_get_sample_fmt_name(ac->out_fmt));return ff_convert_dither(ac-> in
simple assert() macros that are a bit more flexible than ISO C assert().
#define av_assert0(cond)
assert() equivalent, that is always enabled.
int ff_filter_frame(AVFilterLink *link, AVFrame *frame)
Send a frame of data to the next filter.
int ff_filter_get_nb_threads(AVFilterContext *ctx)
Get number of threads for current filter instance.
#define flags(name, subs,...)
static __device__ float ceilf(float a)
static float add(float src0, float src1)
void av_expr_free(AVExpr *e)
Free a parsed expression previously created with av_expr_parse().
int av_expr_parse(AVExpr **expr, const char *s, const char *const *const_names, const char *const *func1_names, double(*const *funcs1)(void *, double), const char *const *func2_names, double(*const *funcs2)(void *, double, double), int log_offset, void *log_ctx)
Parse an expression.
simple arithmetic expression evaluator
#define AVFILTER_FLAG_SUPPORT_TIMELINE_GENERIC
Some filters support a generic "enable" expression option that can be used to enable or disable a fil...
#define AVFILTER_FLAG_SLICE_THREADS
The filter supports multithreading by splitting frames into multiple parts and processing them concur...
int av_frame_is_writable(AVFrame *frame)
Check if the frame data is writable.
void av_frame_free(AVFrame **frame)
Free the frame and any dynamically allocated objects in it, e.g.
int av_frame_copy_props(AVFrame *dst, const AVFrame *src)
Copy only "metadata" fields from src to dst.
#define AV_LOG_DEBUG
Stuff which is only useful for libav* developers.
#define AV_LOG_WARNING
Something somehow does not look correct.
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
void * av_calloc(size_t nmemb, size_t size)
Non-inlined equivalent of av_mallocz_array().
static const int weights[]
static void direct(const float *in, const FFTComplex *ir, int len, float *out)
common internal API header
#define NULL_IF_CONFIG_SMALL(x)
Return NULL if CONFIG_SMALL is true, otherwise the argument without modification.
static enum AVPixelFormat pix_fmts[]
static int slice_end(AVCodecContext *avctx, AVFrame *pict)
Handle slice ends.
AVPixelFormat
Pixel format.
@ AV_PIX_FMT_RGB24
packed RGB 8:8:8, 24bpp, RGBRGB...
@ AV_PIX_FMT_BGR24
packed RGB 8:8:8, 24bpp, BGRBGR...
@ AV_PIX_FMT_GBRP
planar GBR 4:4:4 24bpp
typedef void(RENAME(mix_any_func_type))
Describe the class of an AVClass context structure.
AVFilterLink ** outputs
array of pointers to output links
A link between two filters.
int w
agreed upon image width
int h
agreed upon image height
AVFilterContext * dst
dest filter
int format
agreed upon media format
A filter pad used for either input or output.
const char * name
Pad name.
const char * name
Filter name.
AVFormatInternal * internal
An opaque field for libavformat internal usage.
This structure describes decoded (raw) audio or video data.
void(* color_correlation)(uint8_t **dst, int dst_linesize, float **src, int src_linesize, int w, int h)
double var_values[MAX_THREADS][VAR_VARS_NB]
AVExpr * expr[MAX_THREADS]
void(* filter_freq_func)(struct DCTdnoizContext *s, const float *src, int src_linesize, float *dst, int dst_linesize, int thread_id)
void(* color_decorrelation)(float **dst, int dst_linesize, const uint8_t **src, int src_linesize, int w, int h)
float * slices[MAX_THREADS]
Used for passing data between threads.
#define av_malloc_array(a, b)
static av_always_inline void color_correlation(uint8_t **dst, int dst_linesize, float **src, int src_linesize, int w, int h, int r, int g, int b)
static int filter_slice(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs)
static void av_always_inline idct16_1d(float *dst, const float *src, int dst_stridea, int dst_strideb, int src_stridea, int src_strideb, int add)
static int query_formats(AVFilterContext *ctx)
static const AVFilterPad dctdnoiz_outputs[]
static int config_input(AVFilterLink *inlink)
static void av_always_inline fdct8_1d(float *dst, const float *src, int dst_stridea, int dst_strideb, int src_stridea, int src_strideb)
static av_always_inline void color_decorrelation(float **dst, int dst_linesize, const uint8_t **src, int src_linesize, int w, int h, int r, int g, int b)
static int filter_frame(AVFilterLink *inlink, AVFrame *in)
static av_always_inline void color_decorrelation_gbrp(float **dst, int dst_linesize, const uint8_t **src, int src_linesize, int w, int h)
static const AVOption dctdnoiz_options[]
static void av_always_inline fdct16_1d(float *dst, const float *src, int dst_stridea, int dst_strideb, int src_stridea, int src_strideb)
static void av_always_inline idct8_1d(float *dst, const float *src, int dst_stridea, int dst_strideb, int src_stridea, int src_strideb, int add)
static const char *const var_names[]
A simple, relatively efficient and slow DCT image denoiser.
AVFILTER_DEFINE_CLASS(dctdnoiz)
static av_always_inline void color_correlation_gbrp(uint8_t **dst, int dst_linesize, float **src, int src_linesize, int w, int h)
static av_cold int init(AVFilterContext *ctx)
static av_cold void uninit(AVFilterContext *ctx)
static const AVFilterPad dctdnoiz_inputs[]
#define DECLARE_COLOR_FUNCS(name, r, g, b)
#define DEF_FILTER_FREQ_FUNCS(bsize)
AVFrame * ff_get_video_buffer(AVFilterLink *link, int w, int h)
Request a picture buffer with a specific set of permissions.