FFmpeg
vf_fftdnoiz.c
Go to the documentation of this file.
1 /*
2  * This file is part of FFmpeg.
3  *
4  * FFmpeg is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU Lesser General Public
6  * License as published by the Free Software Foundation; either
7  * version 2.1 of the License, or (at your option) any later version.
8  *
9  * FFmpeg is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12  * Lesser General Public License for more details.
13  *
14  * You should have received a copy of the GNU Lesser General Public
15  * License along with FFmpeg; if not, write to the Free Software
16  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
17  */
18 
19 #include <float.h>
20 
21 #include "libavutil/common.h"
22 #include "libavutil/imgutils.h"
23 #include "libavutil/opt.h"
24 #include "libavutil/pixdesc.h"
25 #include "libavutil/tx.h"
26 #include "internal.h"
27 #include "video.h"
28 #include "window_func.h"
29 
30 #define MAX_BLOCK 256
31 #define MAX_THREADS 32
32 
33 enum BufferTypes {
34  CURRENT,
35  PREV,
36  NEXT,
37  BSIZE
38 };
39 
40 typedef struct PlaneContext {
41  int planewidth, planeheight;
42  int nox, noy;
43  int b;
44  int o;
45  float n;
46 
47  float *buffer[MAX_THREADS][BSIZE];
48  AVComplexFloat *hdata[MAX_THREADS], *vdata[MAX_THREADS];
49  AVComplexFloat *hdata_out[MAX_THREADS], *vdata_out[MAX_THREADS];
50  int data_linesize;
51  int buffer_linesize;
52 } PlaneContext;
53 
54 typedef struct FFTdnoizContext {
55  const AVClass *class;
56 
57  float sigma;
58  float amount;
59  int block_size;
60  float overlap;
61  int method;
62  int window;
63  int nb_prev;
64  int nb_next;
65  int planesf;
66 
67  AVFrame *prev, *cur, *next;
68 
69  int depth;
70  int nb_planes;
71  int nb_threads;
72  PlaneContext planes[4];
73  float win[MAX_BLOCK][MAX_BLOCK];
74 
75  AVTXContext *fft[MAX_THREADS], *ifft[MAX_THREADS];
76  AVTXContext *fft_r[MAX_THREADS], *ifft_r[MAX_THREADS];
77 
78  av_tx_fn tx_fn, itx_fn;
79  av_tx_fn tx_r_fn, itx_r_fn;
80 
81  void (*import_row)(AVComplexFloat *dst, uint8_t *src, int rw, float scale, float *win, int off);
82  void (*export_row)(AVComplexFloat *src, uint8_t *dst, int rw, int depth, float *win);
83 } FFTdnoizContext;
84 
85 #define OFFSET(x) offsetof(FFTdnoizContext, x)
86 #define FLAGS AV_OPT_FLAG_FILTERING_PARAM|AV_OPT_FLAG_VIDEO_PARAM
87 #define TFLAGS AV_OPT_FLAG_FILTERING_PARAM|AV_OPT_FLAG_VIDEO_PARAM|AV_OPT_FLAG_RUNTIME_PARAM
88 static const AVOption fftdnoiz_options[] = {
89  { "sigma", "set denoise strength",
90  OFFSET(sigma), AV_OPT_TYPE_FLOAT, {.dbl=1}, 0, 100, .flags = TFLAGS },
91  { "amount", "set amount of denoising",
92  OFFSET(amount), AV_OPT_TYPE_FLOAT, {.dbl=1}, 0.01, 1, .flags = TFLAGS },
93  { "block", "set block size",
94  OFFSET(block_size), AV_OPT_TYPE_INT, {.i64=32}, 8, MAX_BLOCK, .flags = FLAGS },
95  { "overlap", "set block overlap",
96  OFFSET(overlap), AV_OPT_TYPE_FLOAT, {.dbl=0.5}, 0.2, 0.8, .flags = FLAGS },
97  { "method", "set method of denoising",
98  OFFSET(method), AV_OPT_TYPE_INT, {.i64=0}, 0, 1, .flags = TFLAGS, "method" },
99  { "wiener", "wiener method",
100  0, AV_OPT_TYPE_CONST, {.i64=0}, 0, 0, .flags = TFLAGS, "method" },
101  { "hard", "hard thresholding",
102  0, AV_OPT_TYPE_CONST, {.i64=1}, 0, 0, .flags = TFLAGS, "method" },
103  { "prev", "set number of previous frames for temporal denoising",
104  OFFSET(nb_prev), AV_OPT_TYPE_INT, {.i64=0}, 0, 1, .flags = FLAGS },
105  { "next", "set number of next frames for temporal denoising",
106  OFFSET(nb_next), AV_OPT_TYPE_INT, {.i64=0}, 0, 1, .flags = FLAGS },
107  { "planes", "set planes to filter",
108  OFFSET(planesf), AV_OPT_TYPE_INT, {.i64=7}, 0, 15, .flags = TFLAGS },
109  WIN_FUNC_OPTION("window", OFFSET(window), FLAGS, WFUNC_HANNING),
110  { NULL }
111 };
112 
113 AVFILTER_DEFINE_CLASS(fftdnoiz);
114 
115 static const enum AVPixelFormat pix_fmts[] = {
116  AV_PIX_FMT_GRAY8, AV_PIX_FMT_GRAY9,
117  AV_PIX_FMT_GRAY10, AV_PIX_FMT_GRAY12,
118  AV_PIX_FMT_GRAY14, AV_PIX_FMT_GRAY16,
119  AV_PIX_FMT_YUV410P, AV_PIX_FMT_YUV411P,
120  AV_PIX_FMT_YUV420P, AV_PIX_FMT_YUV422P,
121  AV_PIX_FMT_YUV440P, AV_PIX_FMT_YUV444P,
122  AV_PIX_FMT_YUVJ420P, AV_PIX_FMT_YUVJ422P,
123  AV_PIX_FMT_YUVJ440P, AV_PIX_FMT_YUVJ444P,
124  AV_PIX_FMT_YUVJ411P,
125  AV_PIX_FMT_YUV420P9, AV_PIX_FMT_YUV422P9, AV_PIX_FMT_YUV444P9,
126  AV_PIX_FMT_YUV420P10, AV_PIX_FMT_YUV422P10, AV_PIX_FMT_YUV444P10,
127  AV_PIX_FMT_YUV440P10,
128  AV_PIX_FMT_YUV444P12, AV_PIX_FMT_YUV422P12, AV_PIX_FMT_YUV420P12,
129  AV_PIX_FMT_YUV440P12,
130  AV_PIX_FMT_YUV444P14, AV_PIX_FMT_YUV422P14, AV_PIX_FMT_YUV420P14,
131  AV_PIX_FMT_YUV420P16, AV_PIX_FMT_YUV422P16, AV_PIX_FMT_YUV444P16,
132  AV_PIX_FMT_GBRP, AV_PIX_FMT_GBRP9, AV_PIX_FMT_GBRP10,
133  AV_PIX_FMT_GBRP12, AV_PIX_FMT_GBRP14, AV_PIX_FMT_GBRP16,
134  AV_PIX_FMT_YUVA420P, AV_PIX_FMT_YUVA422P, AV_PIX_FMT_YUVA444P,
135  AV_PIX_FMT_YUVA444P9, AV_PIX_FMT_YUVA444P10, AV_PIX_FMT_YUVA444P12, AV_PIX_FMT_YUVA444P16,
136  AV_PIX_FMT_YUVA422P9, AV_PIX_FMT_YUVA422P10, AV_PIX_FMT_YUVA422P12, AV_PIX_FMT_YUVA422P16,
137  AV_PIX_FMT_YUVA420P9, AV_PIX_FMT_YUVA420P10, AV_PIX_FMT_YUVA420P16,
138  AV_PIX_FMT_GBRAP, AV_PIX_FMT_GBRAP10, AV_PIX_FMT_GBRAP12, AV_PIX_FMT_GBRAP16,
139  AV_PIX_FMT_NONE
140 };
141 
142 typedef struct ThreadData {
143  float *src, *dst;
144 } ThreadData;
145 
146 static void import_row8(AVComplexFloat *dst, uint8_t *src, int rw,
147  float scale, float *win, int off)
148 {
149  for (int j = 0; j < rw; j++) {
150  const int i = abs(j + off);
151  dst[j].re = src[i] * scale * win[j];
152  dst[j].im = 0.f;
153  }
154 }
155 
156 static void export_row8(AVComplexFloat *src, uint8_t *dst, int rw, int depth, float *win)
157 {
158  for (int j = 0; j < rw; j++)
159  dst[j] = av_clip_uint8(lrintf(src[j].re / win[j]));
160 }
161 
162 static void import_row16(AVComplexFloat *dst, uint8_t *srcp, int rw,
163  float scale, float *win, int off)
164 {
165  uint16_t *src = (uint16_t *)srcp;
166 
167  for (int j = 0; j < rw; j++) {
168  const int i = abs(j + off);
169  dst[j].re = src[i] * scale * win[j];
170  dst[j].im = 0;
171  }
172 }
173 
174 static void export_row16(AVComplexFloat *src, uint8_t *dstp, int rw, int depth, float *win)
175 {
176  uint16_t *dst = (uint16_t *)dstp;
177 
178  for (int j = 0; j < rw; j++)
179  dst[j] = av_clip_uintp2_c(lrintf(src[j].re / win[j]), depth);
180 }
181 
182 static int config_input(AVFilterLink *inlink)
183 {
184  AVFilterContext *ctx = inlink->dst;
185  const AVPixFmtDescriptor *desc;
186  FFTdnoizContext *s = ctx->priv;
187  float lut[MAX_BLOCK + 1];
188  float overlap;
189  int i;
190 
191  desc = av_pix_fmt_desc_get(inlink->format);
192  s->depth = desc->comp[0].depth;
193 
194  if (s->depth <= 8) {
195  s->import_row = import_row8;
196  s->export_row = export_row8;
197  } else {
198  s->import_row = import_row16;
199  s->export_row = export_row16;
200  }
201 
202  s->planes[1].planewidth = s->planes[2].planewidth = AV_CEIL_RSHIFT(inlink->w, desc->log2_chroma_w);
203  s->planes[0].planewidth = s->planes[3].planewidth = inlink->w;
204  s->planes[1].planeheight = s->planes[2].planeheight = AV_CEIL_RSHIFT(inlink->h, desc->log2_chroma_h);
205  s->planes[0].planeheight = s->planes[3].planeheight = inlink->h;
206 
207  s->nb_planes = av_pix_fmt_count_planes(inlink->format);
208  s->nb_threads = FFMIN(ff_filter_get_nb_threads(ctx), MAX_THREADS);
209 
210  for (int i = 0; i < s->nb_threads; i++) {
211  float scale = 1.f, iscale = 1.f;
212  int ret;
213 
214  if ((ret = av_tx_init(&s->fft[i], &s->tx_fn, AV_TX_FLOAT_FFT,
215  0, s->block_size, &scale, 0)) < 0 ||
216  (ret = av_tx_init(&s->ifft[i], &s->itx_fn, AV_TX_FLOAT_FFT,
217  1, s->block_size, &iscale, 0)) < 0 ||
218  (ret = av_tx_init(&s->fft_r[i], &s->tx_r_fn, AV_TX_FLOAT_FFT,
219  0, 1 + s->nb_prev + s->nb_next, &scale, 0)) < 0 ||
220  (ret = av_tx_init(&s->ifft_r[i], &s->itx_r_fn, AV_TX_FLOAT_FFT,
221  1, 1 + s->nb_prev + s->nb_next, &iscale, 0)) < 0)
222  return ret;
223  }
224 
225  for (i = 0; i < s->nb_planes; i++) {
226  PlaneContext *p = &s->planes[i];
227  int size;
228 
229  p->b = s->block_size;
230  p->n = 1.f / (p->b * p->b);
231  p->o = lrintf(p->b * s->overlap);
232  size = p->b - p->o;
233  p->nox = (p->planewidth + (size - 1)) / size;
234  p->noy = (p->planeheight + (size - 1)) / size;
235 
236  av_log(ctx, AV_LOG_DEBUG, "nox:%d noy:%d size:%d\n", p->nox, p->noy, size);
237 
238  p->buffer_linesize = p->b * sizeof(AVComplexFloat);
239  p->data_linesize = 2 * p->b * sizeof(float);
240  for (int j = 0; j < s->nb_threads; j++) {
241  p->hdata[j] = av_calloc(p->b, p->data_linesize);
242  p->hdata_out[j] = av_calloc(p->b, p->data_linesize);
243  p->vdata[j] = av_calloc(p->b, p->data_linesize);
244  p->vdata_out[j] = av_calloc(p->b, p->data_linesize);
245  p->buffer[j][CURRENT] = av_calloc(p->b, p->buffer_linesize);
246  if (!p->buffer[j][CURRENT])
247  return AVERROR(ENOMEM);
248  if (s->nb_prev > 0) {
249  p->buffer[j][PREV] = av_calloc(p->b, p->buffer_linesize);
250  if (!p->buffer[j][PREV])
251  return AVERROR(ENOMEM);
252  }
253  if (s->nb_next > 0) {
254  p->buffer[j][NEXT] = av_calloc(p->b, p->buffer_linesize);
255  if (!p->buffer[j][NEXT])
256  return AVERROR(ENOMEM);
257  }
258  if (!p->hdata[j] || !p->vdata[j] ||
259  !p->hdata_out[j] || !p->vdata_out[j])
260  return AVERROR(ENOMEM);
261  }
262  }
263 
264  generate_window_func(lut, s->block_size + 1, s->window, &overlap);
265 
266  for (int y = 0; y < s->block_size; y++) {
267  for (int x = 0; x < s->block_size; x++)
268  s->win[y][x] = lut[y] * lut[x];
269  }
270 
271  return 0;
272 }
273 
274 static void import_block(FFTdnoizContext *s,
275  uint8_t *srcp, int src_linesize,
276  float *buffer, int buffer_linesize, int plane,
277  int jobnr, int y, int x)
278 {
279  PlaneContext *p = &s->planes[plane];
280  const int width = p->planewidth;
281  const int height = p->planeheight;
282  const int block = p->b;
283  const int overlap = p->o;
284  const int hoverlap = overlap / 2;
285  const int size = block - overlap;
286  const int bpp = (s->depth + 7) / 8;
287  const int data_linesize = p->data_linesize / sizeof(AVComplexFloat);
288  const float scale = 1.f / ((1.f + s->nb_prev + s->nb_next) * s->block_size * s->block_size);
289  AVComplexFloat *hdata = p->hdata[jobnr];
290  AVComplexFloat *hdata_out = p->hdata_out[jobnr];
291  AVComplexFloat *vdata_out = p->vdata_out[jobnr];
292  const int woff = -hoverlap;
293  const int hoff = -hoverlap;
294  const int rh = FFMIN(block, height - y * size + hoverlap);
295  const int rw = FFMIN(block, width - x * size + hoverlap);
296  AVComplexFloat *ssrc, *ddst, *dst = hdata, *dst_out = hdata_out;
297  float *bdst = buffer;
298 
299  buffer_linesize /= sizeof(float);
300 
301  for (int i = 0; i < rh; i++) {
302  uint8_t *src = srcp + src_linesize * abs(y * size + i + hoff) + x * size * bpp;
303 
304  s->import_row(dst, src, rw, scale, s->win[i], woff);
305  for (int j = rw; j < block; j++) {
306  dst[j].re = dst[rw - 1].re;
307  dst[j].im = 0.f;
308  }
309  s->tx_fn(s->fft[jobnr], dst_out, dst, sizeof(AVComplexFloat));
310 
311  ddst = dst_out;
312  dst += data_linesize;
313  dst_out += data_linesize;
314  }
315 
316  dst = dst_out;
317  for (int i = rh; i < block; i++) {
318  for (int j = 0; j < block; j++) {
319  dst[j].re = ddst[j].re;
320  dst[j].im = ddst[j].im;
321  }
322 
323  dst += data_linesize;
324  }
325 
326  ssrc = hdata_out;
327  dst = vdata_out;
328  for (int i = 0; i < block; i++) {
329  for (int j = 0; j < block; j++)
330  dst[j] = ssrc[j * data_linesize + i];
331  s->tx_fn(s->fft[jobnr], bdst, dst, sizeof(AVComplexFloat));
332 
333  dst += data_linesize;
334  bdst += buffer_linesize;
335  }
336 }
337 
338 static void export_block(FFTdnoizContext *s,
339  uint8_t *dstp, int dst_linesize,
340  float *buffer, int buffer_linesize, int plane,
341  int jobnr, int y, int x)
342 {
343  PlaneContext *p = &s->planes[plane];
344  const int depth = s->depth;
345  const int bpp = (depth + 7) / 8;
346  const int width = p->planewidth;
347  const int height = p->planeheight;
348  const int block = p->b;
349  const int overlap = p->o;
350  const int hoverlap = overlap / 2;
351  const int size = block - overlap;
352  const int data_linesize = p->data_linesize / sizeof(AVComplexFloat);
353  AVComplexFloat *hdata = p->hdata[jobnr];
354  AVComplexFloat *hdata_out = p->hdata_out[jobnr];
355  AVComplexFloat *vdata_out = p->vdata_out[jobnr];
356  const int rw = FFMIN(size, width - x * size);
357  const int rh = FFMIN(size, height - y * size);
358  AVComplexFloat *hdst, *vdst = vdata_out, *hdst_out = hdata_out;
359  float *bsrc = buffer;
360 
361  hdst = hdata;
362  buffer_linesize /= sizeof(float);
363 
364  for (int i = 0; i < block; i++) {
365  s->itx_fn(s->ifft[jobnr], vdst, bsrc, sizeof(AVComplexFloat));
366  for (int j = 0; j < block; j++)
367  hdst[j * data_linesize + i] = vdst[j];
368 
369  vdst += data_linesize;
370  bsrc += buffer_linesize;
371  }
372 
373  hdst = hdata + hoverlap * data_linesize;
374  for (int i = 0; i < rh && (y * size + i) < height; i++) {
375  uint8_t *dst = dstp + dst_linesize * (y * size + i) + x * size * bpp;
376 
377  s->itx_fn(s->ifft[jobnr], hdst_out, hdst, sizeof(AVComplexFloat));
378  s->export_row(hdst_out + hoverlap, dst, rw, depth, s->win[i + hoverlap] + hoverlap);
379 
380  hdst += data_linesize;
381  hdst_out += data_linesize;
382  }
383 }
384 
385 static void filter_block3d2(FFTdnoizContext *s, int plane, float *pbuffer, float *nbuffer,
386  int jobnr)
387 {
388  PlaneContext *p = &s->planes[plane];
389  const int block = p->b;
390  const int buffer_linesize = p->buffer_linesize / sizeof(float);
391  const float depthx = (1 << (s->depth - 8)) * (1 << (s->depth - 8));
392  const float sigma = s->sigma * depthx / (3.f * s->block_size * s->block_size);
393  const float limit = 1.f - s->amount;
394  float *cbuffer = p->buffer[jobnr][CURRENT];
395  const int method = s->method;
396  float *cbuff = cbuffer;
397  float *pbuff = pbuffer;
398  float *nbuff = nbuffer;
399 
400  for (int i = 0; i < block; i++) {
401  for (int j = 0; j < block; j++) {
402  AVComplexFloat buffer[BSIZE];
403  AVComplexFloat outbuffer[BSIZE];
404 
405  buffer[0].re = pbuff[2 * j ];
406  buffer[0].im = pbuff[2 * j + 1];
407 
408  buffer[1].re = cbuff[2 * j ];
409  buffer[1].im = cbuff[2 * j + 1];
410 
411  buffer[2].re = nbuff[2 * j ];
412  buffer[2].im = nbuff[2 * j + 1];
413 
414  s->tx_r_fn(s->fft_r[jobnr], outbuffer, buffer, sizeof(AVComplexFloat));
415 
416  for (int z = 0; z < 3; z++) {
417  const float re = outbuffer[z].re;
418  const float im = outbuffer[z].im;
419  const float power = re * re + im * im;
420  float factor;
421 
422  switch (method) {
423  case 0:
424  factor = fmaxf(limit, (power - sigma) / (power + 1e-15f));
425  break;
426  case 1:
427  factor = power < sigma ? limit : 1.f;
428  break;
429  }
430 
431  outbuffer[z].re *= factor;
432  outbuffer[z].im *= factor;
433  }
434 
435  s->itx_r_fn(s->ifft_r[jobnr], buffer, outbuffer, sizeof(AVComplexFloat));
436 
437  cbuff[2 * j + 0] = buffer[1].re;
438  cbuff[2 * j + 1] = buffer[1].im;
439  }
440 
441  cbuff += buffer_linesize;
442  pbuff += buffer_linesize;
443  nbuff += buffer_linesize;
444  }
445 }
446 
447 static void filter_block3d1(FFTdnoizContext *s, int plane, float *pbuffer,
448  int jobnr)
449 {
450  PlaneContext *p = &s->planes[plane];
451  const int block = p->b;
452  const int buffer_linesize = p->buffer_linesize / sizeof(float);
453  const float depthx = (1 << (s->depth - 8)) * (1 << (s->depth - 8));
454  const float sigma = s->sigma * depthx / (2.f * s->block_size * s->block_size);
455  const float limit = 1.f - s->amount;
456  float *cbuffer = p->buffer[jobnr][CURRENT];
457  const int method = s->method;
458  float *cbuff = cbuffer;
459  float *pbuff = pbuffer;
460 
461  for (int i = 0; i < block; i++) {
462  for (int j = 0; j < block; j++) {
463  AVComplexFloat buffer[BSIZE];
464  AVComplexFloat outbuffer[BSIZE];
465 
466  buffer[0].re = pbuff[2 * j ];
467  buffer[0].im = pbuff[2 * j + 1];
468 
469  buffer[1].re = cbuff[2 * j ];
470  buffer[1].im = cbuff[2 * j + 1];
471 
472  s->tx_r_fn(s->fft_r[jobnr], outbuffer, buffer, sizeof(AVComplexFloat));
473 
474  for (int z = 0; z < 2; z++) {
475  const float re = outbuffer[z].re;
476  const float im = outbuffer[z].im;
477  const float power = re * re + im * im;
478  float factor;
479 
480  switch (method) {
481  case 0:
482  factor = fmaxf(limit, (power - sigma) / (power + 1e-15f));
483  break;
484  case 1:
485  factor = power < sigma ? limit : 1.f;
486  break;
487  }
488 
489  outbuffer[z].re *= factor;
490  outbuffer[z].im *= factor;
491  }
492 
493  s->itx_r_fn(s->ifft_r[jobnr], buffer, outbuffer, sizeof(AVComplexFloat));
494 
495  cbuff[2 * j + 0] = buffer[1].re;
496  cbuff[2 * j + 1] = buffer[1].im;
497  }
498 
499  cbuff += buffer_linesize;
500  pbuff += buffer_linesize;
501  }
502 }
503 
504 static void filter_block2d(FFTdnoizContext *s, int plane,
505  int jobnr)
506 {
507  PlaneContext *p = &s->planes[plane];
508  const int block = p->b;
509  const int method = s->method;
510  const int buffer_linesize = p->buffer_linesize / sizeof(float);
511  const float depthx = (1 << (s->depth - 8)) * (1 << (s->depth - 8));
512  const float sigma = s->sigma * depthx / (s->block_size * s->block_size);
513  const float limit = 1.f - s->amount;
514  float *buff = p->buffer[jobnr][CURRENT];
515 
516  for (int i = 0; i < block; i++) {
517  for (int j = 0; j < block; j++) {
518  float factor, power, re, im;
519 
520  re = buff[j * 2 ];
521  im = buff[j * 2 + 1];
522  power = re * re + im * im;
523  switch (method) {
524  case 0:
525  factor = fmaxf(limit, (power - sigma) / (power + 1e-15f));
526  break;
527  case 1:
528  factor = power < sigma ? limit : 1.f;
529  break;
530  }
531 
532  buff[j * 2 ] *= factor;
533  buff[j * 2 + 1] *= factor;
534  }
535 
536  buff += buffer_linesize;
537  }
538 }
539 
540 static int denoise(AVFilterContext *ctx, void *arg,
541  int jobnr, int nb_jobs)
542 {
543  FFTdnoizContext *s = ctx->priv;
544  AVFrame *out = arg;
545 
546  for (int plane = 0; plane < s->nb_planes; plane++) {
547  PlaneContext *p = &s->planes[plane];
548  const int nox = p->nox;
549  const int noy = p->noy;
550  const int slice_start = (noy * jobnr) / nb_jobs;
551  const int slice_end = (noy * (jobnr+1)) / nb_jobs;
552 
553  if (!((1 << plane) & s->planesf) || ctx->is_disabled)
554  continue;
555 
556  for (int y = slice_start; y < slice_end; y++) {
557  for (int x = 0; x < nox; x++) {
558  if (s->next) {
559  import_block(s, s->next->data[plane], s->next->linesize[plane],
560  p->buffer[jobnr][NEXT], p->buffer_linesize, plane,
561  jobnr, y, x);
562  }
563 
564  if (s->prev) {
565  import_block(s, s->prev->data[plane], s->prev->linesize[plane],
566  p->buffer[jobnr][PREV], p->buffer_linesize, plane,
567  jobnr, y, x);
568  }
569 
570  import_block(s, s->cur->data[plane], s->cur->linesize[plane],
571  p->buffer[jobnr][CURRENT], p->buffer_linesize, plane,
572  jobnr, y, x);
573 
574  if (s->next && s->prev) {
575  filter_block3d2(s, plane, p->buffer[jobnr][PREV], p->buffer[jobnr][NEXT], jobnr);
576  } else if (s->next) {
577  filter_block3d1(s, plane, p->buffer[jobnr][NEXT], jobnr);
578  } else if (s->prev) {
579  filter_block3d1(s, plane, p->buffer[jobnr][PREV], jobnr);
580  } else {
581  filter_block2d(s, plane, jobnr);
582  }
583 
584  export_block(s, out->data[plane], out->linesize[plane],
585  p->buffer[jobnr][CURRENT], p->buffer_linesize, plane,
586  jobnr, y, x);
587  }
588  }
589  }
590 
591  return 0;
592 }
593 
594 static int filter_frame(AVFilterLink *inlink, AVFrame *in)
595 {
596  AVFilterContext *ctx = inlink->dst;
597  FFTdnoizContext *s = ctx->priv;
598  AVFilterLink *outlink = ctx->outputs[0];
599  int direct, plane;
600  AVFrame *out;
601 
602  if (s->nb_next > 0 && s->nb_prev > 0) {
603  av_frame_free(&s->prev);
604  s->prev = s->cur;
605  s->cur = s->next;
606  s->next = in;
607 
608  if (!s->prev && s->cur) {
609  s->prev = av_frame_clone(s->cur);
610  if (!s->prev)
611  return AVERROR(ENOMEM);
612  }
613  if (!s->cur)
614  return 0;
615  } else if (s->nb_next > 0) {
616  av_frame_free(&s->cur);
617  s->cur = s->next;
618  s->next = in;
619 
620  if (!s->cur)
621  return 0;
622  } else if (s->nb_prev > 0) {
623  av_frame_free(&s->prev);
624  s->prev = s->cur;
625  s->cur = in;
626 
627  if (!s->prev)
628  s->prev = av_frame_clone(s->cur);
629  if (!s->prev)
630  return AVERROR(ENOMEM);
631  } else {
632  s->cur = in;
633  }
634 
635  if (av_frame_is_writable(in) && s->nb_next == 0 && s->nb_prev == 0) {
636  direct = 1;
637  out = in;
638  } else {
639  direct = 0;
640  out = ff_get_video_buffer(outlink, outlink->w, outlink->h);
641  if (!out)
642  return AVERROR(ENOMEM);
643  av_frame_copy_props(out, s->cur);
644  }
645 
646  ff_filter_execute(ctx, denoise, out, NULL,
647  FFMIN(s->planes[0].noy, s->nb_threads));
648 
649  for (plane = 0; plane < s->nb_planes; plane++) {
650  PlaneContext *p = &s->planes[plane];
651 
652  if (!((1 << plane) & s->planesf) || ctx->is_disabled) {
653  if (!direct)
654  av_image_copy_plane(out->data[plane], out->linesize[plane],
655  s->cur->data[plane], s->cur->linesize[plane],
656  p->planewidth * (1 + (s->depth > 8)), p->planeheight);
657  continue;
658  }
659  }
660 
661  if (s->nb_next == 0 && s->nb_prev == 0) {
662  if (direct) {
663  s->cur = NULL;
664  } else {
665  av_frame_free(&s->cur);
666  }
667  }
668  return ff_filter_frame(outlink, out);
669 }
670 
671 static int request_frame(AVFilterLink *outlink)
672 {
673  AVFilterContext *ctx = outlink->src;
674  FFTdnoizContext *s = ctx->priv;
675  int ret = 0;
676 
677  ret = ff_request_frame(ctx->inputs[0]);
678 
679  if (ret == AVERROR_EOF && (s->nb_next > 0)) {
680  AVFrame *buf;
681 
682  if (s->next && s->nb_next > 0)
683  buf = av_frame_clone(s->next);
684  else if (s->cur)
685  buf = av_frame_clone(s->cur);
686  else
687  buf = av_frame_clone(s->prev);
688  if (!buf)
689  return AVERROR(ENOMEM);
690 
691  ret = filter_frame(ctx->inputs[0], buf);
692  if (ret < 0)
693  return ret;
694  ret = AVERROR_EOF;
695  }
696 
697  return ret;
698 }
699 
700 static av_cold void uninit(AVFilterContext *ctx)
701 {
702  FFTdnoizContext *s = ctx->priv;
703  int i;
704 
705  for (i = 0; i < 4; i++) {
706  PlaneContext *p = &s->planes[i];
707 
708  for (int j = 0; j < s->nb_threads; j++) {
709  av_freep(&p->hdata[j]);
710  av_freep(&p->vdata[j]);
711  av_freep(&p->hdata_out[j]);
712  av_freep(&p->vdata_out[j]);
713  av_freep(&p->buffer[j][PREV]);
714  av_freep(&p->buffer[j][CURRENT]);
715  av_freep(&p->buffer[j][NEXT]);
716  }
717  }
718 
719  for (i = 0; i < s->nb_threads; i++) {
720  av_tx_uninit(&s->fft[i]);
721  av_tx_uninit(&s->ifft[i]);
722  av_tx_uninit(&s->fft_r[i]);
723  av_tx_uninit(&s->ifft_r[i]);
724  }
725 
726  av_frame_free(&s->prev);
727  av_frame_free(&s->cur);
728  av_frame_free(&s->next);
729 }
730 
731 static const AVFilterPad fftdnoiz_inputs[] = {
732  {
733  .name = "default",
734  .type = AVMEDIA_TYPE_VIDEO,
735  .filter_frame = filter_frame,
736  .config_props = config_input,
737  },
738 };
739 
740 static const AVFilterPad fftdnoiz_outputs[] = {
741  {
742  .name = "default",
743  .type = AVMEDIA_TYPE_VIDEO,
744  .request_frame = request_frame,
745  },
746 };
747 
748 const AVFilter ff_vf_fftdnoiz = {
749  .name = "fftdnoiz",
750  .description = NULL_IF_CONFIG_SMALL("Denoise frames using 3D FFT."),
751  .priv_size = sizeof(FFTdnoizContext),
752  .uninit = uninit,
753  FILTER_INPUTS(fftdnoiz_inputs),
754  FILTER_OUTPUTS(fftdnoiz_outputs),
755  FILTER_PIXFMTS_ARRAY(pix_fmts),
756  .priv_class = &fftdnoiz_class,
757  .flags = AVFILTER_FLAG_SUPPORT_TIMELINE_INTERNAL |
758  AVFILTER_FLAG_SLICE_THREADS,
759  .process_command = ff_filter_process_command,
760 };
filter_block2d
static void filter_block2d(FFTdnoizContext *s, int plane, int jobnr)
Definition: vf_fftdnoiz.c:504
ff_get_video_buffer
AVFrame * ff_get_video_buffer(AVFilterLink *link, int w, int h)
Request a picture buffer with a specific set of permissions.
Definition: video.c:108
AV_PIX_FMT_YUVA422P16
#define AV_PIX_FMT_YUVA422P16
Definition: pixfmt.h:512
CURRENT
@ CURRENT
Definition: vf_fftdnoiz.c:34
denoise
static int denoise(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs)
Definition: vf_fftdnoiz.c:540
AV_PIX_FMT_GBRAP16
#define AV_PIX_FMT_GBRAP16
Definition: pixfmt.h:491
FFTdnoizContext::import_row
void(* import_row)(AVComplexFloat *dst, uint8_t *src, int rw, float scale, float *win, int off)
Definition: vf_fftdnoiz.c:81
AVPixelFormat
AVPixelFormat
Pixel format.
Definition: pixfmt.h:64
PlaneContext::buffer_linesize
int buffer_linesize
Definition: vf_fftdnoiz.c:51
AVERROR
Filter the word “frame” indicates either a video frame or a group of audio as stored in an AVFrame structure Format for each input and each output the list of supported formats For video that means pixel format For audio that means channel sample they are references to shared objects When the negotiation mechanism computes the intersection of the formats supported at each end of a all references to both lists are replaced with a reference to the intersection And when a single format is eventually chosen for a link amongst the remaining all references to the list are updated That means that if a filter requires that its input and output have the same format amongst a supported all it has to do is use a reference to the same list of formats query_formats can leave some formats unset and return AVERROR(EAGAIN) to cause the negotiation mechanism toagain later. That can be used by filters with complex requirements to use the format negotiated on one link to set the formats supported on another. Frame references ownership and permissions
opt.h
request_frame
static int request_frame(AVFilterLink *outlink)
Definition: vf_fftdnoiz.c:671
MAX_THREADS
#define MAX_THREADS
Definition: vf_fftdnoiz.c:31
out
FILE * out
Definition: movenc.c:54
FFTdnoizContext::window
int window
Definition: vf_fftdnoiz.c:62
FFTdnoizContext::export_row
void(* export_row)(AVComplexFloat *src, uint8_t *dst, int rw, int depth, float *win)
Definition: vf_fftdnoiz.c:82
ff_filter_frame
int ff_filter_frame(AVFilterLink *link, AVFrame *frame)
Send a frame of data to the next filter.
Definition: avfilter.c:978
FFTdnoizContext::nb_next
int nb_next
Definition: vf_fftdnoiz.c:64
av_pix_fmt_desc_get
const AVPixFmtDescriptor * av_pix_fmt_desc_get(enum AVPixelFormat pix_fmt)
Definition: pixdesc.c:2964
AVERROR_EOF
#define AVERROR_EOF
End of file.
Definition: error.h:57
PlaneContext::buffer
float * buffer[MAX_THREADS][BSIZE]
Definition: vf_fftdnoiz.c:47
FILTER_PIXFMTS_ARRAY
#define FILTER_PIXFMTS_ARRAY(array)
Definition: internal.h:172
AVTXContext
Definition: tx_priv.h:235
inlink
The exact code depends on how similar the blocks are and how related they are to the and needs to apply these operations to the correct inlink or outlink if there are several Macros are available to factor that when no extra processing is inlink
Definition: filter_design.txt:212
import_row16
static void import_row16(AVComplexFloat *dst, uint8_t *srcp, int rw, float scale, float *win, int off)
Definition: vf_fftdnoiz.c:162
av_frame_free
void av_frame_free(AVFrame **frame)
Free the frame and any dynamically allocated objects in it, e.g.
Definition: frame.c:100
AV_PIX_FMT_YUVA422P9
#define AV_PIX_FMT_YUVA422P9
Definition: pixfmt.h:504
AVFrame
This structure describes decoded (raw) audio or video data.
Definition: frame.h:340
pixdesc.h
av_clip_uintp2_c
static av_always_inline av_const unsigned av_clip_uintp2_c(int a, int p)
Clip a signed integer to an unsigned power of two range.
Definition: common.h:276
AV_PIX_FMT_YUVA420P16
#define AV_PIX_FMT_YUVA420P16
Definition: pixfmt.h:511
AV_PIX_FMT_YUVA420P10
#define AV_PIX_FMT_YUVA420P10
Definition: pixfmt.h:506
AVOption
AVOption.
Definition: opt.h:251
PlaneContext::b
int b
Definition: vf_fftdnoiz.c:43
fftdnoiz_outputs
static const AVFilterPad fftdnoiz_outputs[]
Definition: vf_fftdnoiz.c:740
FFTdnoizContext::ifft
AVTXContext * ifft[MAX_THREADS]
Definition: vf_fftdnoiz.c:75
AV_PIX_FMT_YUV420P10
#define AV_PIX_FMT_YUV420P10
Definition: pixfmt.h:468
FFTdnoizContext::sigma
float sigma
Definition: vf_fftdnoiz.c:57
ff_request_frame
int ff_request_frame(AVFilterLink *link)
Request an input frame from the filter at the other end of the link.
Definition: avfilter.c:431
float.h
AVComplexFloat
Definition: tx.h:27
WIN_FUNC_OPTION
#define WIN_FUNC_OPTION(win_func_opt_name, win_func_offset, flag, default_window_func)
Definition: window_func.h:37
AV_PIX_FMT_YUV440P
@ AV_PIX_FMT_YUV440P
planar YUV 4:4:0 (1 Cr & Cb sample per 1x2 Y samples)
Definition: pixfmt.h:99
AVFilter::name
const char * name
Filter name.
Definition: avfilter.h:170
video.h
AV_PIX_FMT_YUVA422P10
#define AV_PIX_FMT_YUVA422P10
Definition: pixfmt.h:507
av_tx_init
av_cold int av_tx_init(AVTXContext **ctx, av_tx_fn *tx, enum AVTXType type, int inv, int len, const void *scale, uint64_t flags)
Initialize a transform context with the given configuration (i)MDCTs with an odd length are currently...
Definition: tx.c:901
AV_PIX_FMT_GRAY9
#define AV_PIX_FMT_GRAY9
Definition: pixfmt.h:448
av_image_copy_plane
void av_image_copy_plane(uint8_t *dst, int dst_linesize, const uint8_t *src, int src_linesize, int bytewidth, int height)
Copy image plane from src to dst.
Definition: imgutils.c:374
win
static float win(SuperEqualizerContext *s, float n, int N)
Definition: af_superequalizer.c:119
FFTdnoizContext::cur
AVFrame * cur
Definition: vf_fftdnoiz.c:67
av_pix_fmt_count_planes
int av_pix_fmt_count_planes(enum AVPixelFormat pix_fmt)
Definition: pixdesc.c:3004
AV_PIX_FMT_YUVA420P9
#define AV_PIX_FMT_YUVA420P9
Definition: pixfmt.h:503
AVComplexFloat::im
float im
Definition: tx.h:28
window
static SDL_Window * window
Definition: ffplay.c:364
AV_PIX_FMT_GBRP14
#define AV_PIX_FMT_GBRP14
Definition: pixfmt.h:486
AV_PIX_FMT_GBRAP
@ AV_PIX_FMT_GBRAP
planar GBRA 4:4:4:4 32bpp
Definition: pixfmt.h:205
PlaneContext::planeheight
int planeheight
Definition: vf_fftdnoiz.c:41
AV_PIX_FMT_GBRP10
#define AV_PIX_FMT_GBRP10
Definition: pixfmt.h:484
AV_PIX_FMT_YUVA444P16
#define AV_PIX_FMT_YUVA444P16
Definition: pixfmt.h:513
AV_PIX_FMT_YUV422P9
#define AV_PIX_FMT_YUV422P9
Definition: pixfmt.h:466
import_block
static void import_block(FFTdnoizContext *s, uint8_t *srcp, int src_linesize, float *buffer, int buffer_linesize, int plane, int jobnr, int y, int x)
Definition: vf_fftdnoiz.c:274
scale
static av_always_inline float scale(float x, float s)
Definition: vf_v360.c:1389
AV_PIX_FMT_GRAY16
#define AV_PIX_FMT_GRAY16
Definition: pixfmt.h:452
TFLAGS
#define TFLAGS
Definition: vf_fftdnoiz.c:87
AVFilterPad
A filter pad used for either input or output.
Definition: internal.h:47
MAX_BLOCK
#define MAX_BLOCK
Definition: vf_fftdnoiz.c:30
AV_PIX_FMT_YUV444P10
#define AV_PIX_FMT_YUV444P10
Definition: pixfmt.h:471
AV_PIX_FMT_YUVJ411P
@ AV_PIX_FMT_YUVJ411P
planar YUV 4:1:1, 12bpp, (1 Cr & Cb sample per 4x1 Y samples) full scale (JPEG), deprecated in favor ...
Definition: pixfmt.h:276
PlaneContext::vdata
AVComplexFloat * vdata[MAX_THREADS]
Definition: vf_fftdnoiz.c:48
av_cold
#define av_cold
Definition: attributes.h:90
AV_PIX_FMT_YUV422P16
#define AV_PIX_FMT_YUV422P16
Definition: pixfmt.h:480
av_tx_fn
void(* av_tx_fn)(AVTXContext *s, void *out, void *in, ptrdiff_t stride)
Function pointer to a function to perform the transform.
Definition: tx.h:151
FFTdnoizContext::overlap
float overlap
Definition: vf_fftdnoiz.c:60
FFTdnoizContext::itx_r_fn
av_tx_fn itx_r_fn
Definition: vf_fftdnoiz.c:79
AV_PIX_FMT_YUVJ422P
@ AV_PIX_FMT_YUVJ422P
planar YUV 4:2:2, 16bpp, full scale (JPEG), deprecated in favor of AV_PIX_FMT_YUV422P and setting col...
Definition: pixfmt.h:79
AV_PIX_FMT_GBRAP10
#define AV_PIX_FMT_GBRAP10
Definition: pixfmt.h:488
float
float
Definition: af_crystalizer.c:121
width
#define width
s
#define s(width, name)
Definition: cbs_vp9.c:198
AV_PIX_FMT_GBRAP12
#define AV_PIX_FMT_GBRAP12
Definition: pixfmt.h:489
AV_PIX_FMT_YUVA420P
@ AV_PIX_FMT_YUVA420P
planar YUV 4:2:0, 20bpp, (1 Cr & Cb sample per 2x2 Y & A samples)
Definition: pixfmt.h:101
AV_PIX_FMT_YUV444P16
#define AV_PIX_FMT_YUV444P16
Definition: pixfmt.h:481
AV_CEIL_RSHIFT
#define AV_CEIL_RSHIFT(a, b)
Definition: common.h:51
slice_end
static int slice_end(AVCodecContext *avctx, AVFrame *pict)
Handle slice ends.
Definition: mpeg12dec.c:1714
AV_TX_FLOAT_FFT
@ AV_TX_FLOAT_FFT
Standard complex to complex FFT with sample data type of AVComplexFloat, AVComplexDouble or AVComplex...
Definition: tx.h:47
AV_PIX_FMT_YUVA444P12
#define AV_PIX_FMT_YUVA444P12
Definition: pixfmt.h:510
AV_PIX_FMT_YUV420P9
#define AV_PIX_FMT_YUV420P9
Definition: pixfmt.h:465
AV_LOG_DEBUG
#define AV_LOG_DEBUG
Stuff which is only useful for libav* developers.
Definition: log.h:201
FFTdnoizContext::win
float win[MAX_BLOCK][MAX_BLOCK]
Definition: vf_fftdnoiz.c:73
AV_PIX_FMT_YUV420P16
#define AV_PIX_FMT_YUV420P16
Definition: pixfmt.h:479
ctx
AVFormatContext * ctx
Definition: movenc.c:48
AV_PIX_FMT_GRAY14
#define AV_PIX_FMT_GRAY14
Definition: pixfmt.h:451
av_frame_clone
AVFrame * av_frame_clone(const AVFrame *src)
Create a new frame that references the same data as src.
Definition: frame.c:609
pix_fmts
static enum AVPixelFormat pix_fmts[]
Definition: vf_fftdnoiz.c:115
AV_PIX_FMT_YUV420P
@ AV_PIX_FMT_YUV420P
planar YUV 4:2:0, 12bpp, (1 Cr & Cb sample per 2x2 Y samples)
Definition: pixfmt.h:66
FILTER_INPUTS
#define FILTER_INPUTS(array)
Definition: internal.h:192
AV_PIX_FMT_YUVJ444P
@ AV_PIX_FMT_YUVJ444P
planar YUV 4:4:4, 24bpp, full scale (JPEG), deprecated in favor of AV_PIX_FMT_YUV444P and setting col...
Definition: pixfmt.h:80
arg
const char * arg
Definition: jacosubdec.c:67
AV_PIX_FMT_GRAY10
#define AV_PIX_FMT_GRAY10
Definition: pixfmt.h:449
ThreadData::dst
AVFrame * dst
Definition: vf_blend.c:56
PlaneContext::hdata
AVComplexFloat * hdata[MAX_THREADS]
Definition: vf_fftdnoiz.c:48
AV_PIX_FMT_GBRP16
#define AV_PIX_FMT_GBRP16
Definition: pixfmt.h:487
FFTdnoizContext::next
AVFrame * next
Definition: vf_fftdnoiz.c:67
FFTdnoizContext::tx_fn
av_tx_fn tx_fn
Definition: vf_fftdnoiz.c:78
AVClass
Describe the class of an AVClass context structure.
Definition: log.h:66
NULL
#define NULL
Definition: coverity.c:32
av_frame_copy_props
int av_frame_copy_props(AVFrame *dst, const AVFrame *src)
Copy only "metadata" fields from src to dst.
Definition: frame.c:736
PlaneContext::planewidth
int planewidth
Definition: vf_fftdnoiz.c:41
ThreadData::src
const uint8_t * src
Definition: vf_bm3d.c:54
FFTdnoizContext::fft_r
AVTXContext * fft_r[MAX_THREADS]
Definition: vf_fftdnoiz.c:76
FFTdnoizContext
Definition: vf_fftdnoiz.c:54
AV_PIX_FMT_YUVJ420P
@ AV_PIX_FMT_YUVJ420P
planar YUV 4:2:0, 12bpp, full scale (JPEG), deprecated in favor of AV_PIX_FMT_YUV420P and setting col...
Definition: pixfmt.h:78
FFTdnoizContext::prev
AVFrame * prev
Definition: vf_fftdnoiz.c:67
AV_PIX_FMT_YUV440P10
#define AV_PIX_FMT_YUV440P10
Definition: pixfmt.h:470
PlaneContext
Definition: ffv1.h:62
generate_window_func
static void generate_window_func(float *lut, int N, int win_func, float *overlap)
Definition: window_func.h:63
PlaneContext::data_linesize
int data_linesize
Definition: vf_fftdnoiz.c:50
abs
#define abs(x)
Definition: cuda_runtime.h:35
WFUNC_HANNING
@ WFUNC_HANNING
Definition: window_func.h:29
AV_PIX_FMT_YUV422P10
#define AV_PIX_FMT_YUV422P10
Definition: pixfmt.h:469
AV_PIX_FMT_GRAY8
@ AV_PIX_FMT_GRAY8
Y , 8bpp.
Definition: pixfmt.h:74
AV_PIX_FMT_GBRP9
#define AV_PIX_FMT_GBRP9
Definition: pixfmt.h:483
FFTdnoizContext::nb_threads
int nb_threads
Definition: vf_fftdnoiz.c:71
FFTdnoizContext::depth
int depth
Definition: vf_fftdnoiz.c:69
config_input
static int config_input(AVFilterLink *inlink)
Definition: vf_fftdnoiz.c:182
filter_block3d2
static void filter_block3d2(FFTdnoizContext *s, int plane, float *pbuffer, float *nbuffer, int jobnr)
Definition: vf_fftdnoiz.c:385
f
f
Definition: af_crystalizer.c:121
BSIZE
@ BSIZE
Definition: vf_fftdnoiz.c:37
NULL_IF_CONFIG_SMALL
#define NULL_IF_CONFIG_SMALL(x)
Return NULL if CONFIG_SMALL is true, otherwise the argument without modification.
Definition: internal.h:106
fmaxf
float fmaxf(float, float)
AV_PIX_FMT_YUV422P12
#define AV_PIX_FMT_YUV422P12
Definition: pixfmt.h:473
size
int size
Definition: twinvq_data.h:10344
AVComplexFloat::re
float re
Definition: tx.h:28
PlaneContext::o
int o
Definition: vf_fftdnoiz.c:44
AV_PIX_FMT_YUV444P12
#define AV_PIX_FMT_YUV444P12
Definition: pixfmt.h:475
av_frame_is_writable
int av_frame_is_writable(AVFrame *frame)
Check if the frame data is writable.
Definition: frame.c:666
FLAGS
#define FLAGS
Definition: vf_fftdnoiz.c:86
FFTdnoizContext::tx_r_fn
av_tx_fn tx_r_fn
Definition: vf_fftdnoiz.c:79
PREV
@ PREV
Definition: vf_fftdnoiz.c:35
ff_filter_process_command
int ff_filter_process_command(AVFilterContext *ctx, const char *cmd, const char *arg, char *res, int res_len, int flags)
Generic processing of user supplied commands that are set in the same way as the filter options.
Definition: avfilter.c:851
height
#define height
AV_PIX_FMT_YUVA444P
@ AV_PIX_FMT_YUVA444P
planar YUV 4:4:4 32bpp, (1 Cr & Cb sample per 1x1 Y & A samples)
Definition: pixfmt.h:167
AV_PIX_FMT_YUVA444P10
#define AV_PIX_FMT_YUVA444P10
Definition: pixfmt.h:508
av_tx_uninit
av_cold void av_tx_uninit(AVTXContext **ctx)
Frees a context and sets *ctx to NULL, does nothing when *ctx == NULL.
Definition: tx.c:294
filter_block3d1
static void filter_block3d1(FFTdnoizContext *s, int plane, float *pbuffer, int jobnr)
Definition: vf_fftdnoiz.c:447
internal.h
AV_OPT_TYPE_FLOAT
@ AV_OPT_TYPE_FLOAT
Definition: opt.h:228
FFTdnoizContext::amount
float amount
Definition: vf_fftdnoiz.c:58
FFTdnoizContext::nb_planes
int nb_planes
Definition: vf_fftdnoiz.c:70
FFTdnoizContext::planes
PlaneContext planes[4]
Definition: vf_fftdnoiz.c:72
PlaneContext::vdata_out
AVComplexFloat * vdata_out[MAX_THREADS]
Definition: vf_fftdnoiz.c:49
lrintf
#define lrintf(x)
Definition: libm_mips.h:72
i
#define i(width, name, range_min, range_max)
Definition: cbs_h2645.c:245
FFTdnoizContext::planesf
int planesf
Definition: vf_fftdnoiz.c:65
AV_PIX_FMT_GBRP12
#define AV_PIX_FMT_GBRP12
Definition: pixfmt.h:485
common.h
PlaneContext::nox
int nox
Definition: vf_fftdnoiz.c:42
ff_filter_get_nb_threads
int ff_filter_get_nb_threads(AVFilterContext *ctx)
Get number of threads for current filter instance.
Definition: avfilter.c:786
ThreadData
Used for passing data between threads.
Definition: dsddec.c:69
FFMIN
#define FFMIN(a, b)
Definition: macros.h:49
AV_PIX_FMT_YUVJ440P
@ AV_PIX_FMT_YUVJ440P
planar YUV 4:4:0 full scale (JPEG), deprecated in favor of AV_PIX_FMT_YUV440P and setting color_range
Definition: pixfmt.h:100
AVFilterPad::name
const char * name
Pad name.
Definition: internal.h:53
OFFSET
#define OFFSET(x)
Definition: vf_fftdnoiz.c:85
av_calloc
void * av_calloc(size_t nmemb, size_t size)
Definition: mem.c:262
AV_PIX_FMT_YUV444P9
#define AV_PIX_FMT_YUV444P9
Definition: pixfmt.h:467
limit
static double limit(double x)
Definition: vf_pseudocolor.c:142
NEXT
@ NEXT
Definition: vf_fftdnoiz.c:36
FFTdnoizContext::itx_fn
av_tx_fn itx_fn
Definition: vf_fftdnoiz.c:78
AVFilter
Filter definition.
Definition: avfilter.h:166
FFTdnoizContext::nb_prev
int nb_prev
Definition: vf_fftdnoiz.c:63
ret
ret
Definition: filter_design.txt:187
uninit
static av_cold void uninit(AVFilterContext *ctx)
Definition: vf_fftdnoiz.c:700
export_block
static void export_block(FFTdnoizContext *s, uint8_t *dstp, int dst_linesize, float *buffer, int buffer_linesize, int plane, int jobnr, int y, int x)
Definition: vf_fftdnoiz.c:338
FFTdnoizContext::ifft_r
AVTXContext * ifft_r[MAX_THREADS]
Definition: vf_fftdnoiz.c:76
AV_PIX_FMT_YUVA444P9
#define AV_PIX_FMT_YUVA444P9
Definition: pixfmt.h:505
fftdnoiz_inputs
static const AVFilterPad fftdnoiz_inputs[]
Definition: vf_fftdnoiz.c:731
AV_PIX_FMT_YUV420P12
#define AV_PIX_FMT_YUV420P12
Definition: pixfmt.h:472
AV_PIX_FMT_YUV422P14
#define AV_PIX_FMT_YUV422P14
Definition: pixfmt.h:477
window_func.h
PlaneContext::hdata_out
AVComplexFloat * hdata_out[MAX_THREADS]
Definition: vf_fftdnoiz.c:49
power
static float power(float r, float g, float b, float max)
Definition: preserve_color.h:45
PlaneContext::noy
int noy
Definition: vf_fftdnoiz.c:42
ff_vf_fftdnoiz
const AVFilter ff_vf_fftdnoiz
Definition: vf_fftdnoiz.c:748
buffer
the frame and frame reference mechanism is intended to as much as expensive copies of that data while still allowing the filters to produce correct results The data is stored in buffers represented by AVFrame structures Several references can point to the same frame buffer
Definition: filter_design.txt:49
AV_PIX_FMT_NONE
@ AV_PIX_FMT_NONE
Definition: pixfmt.h:65
AV_PIX_FMT_YUVA422P12
#define AV_PIX_FMT_YUVA422P12
Definition: pixfmt.h:509
AV_OPT_TYPE_INT
@ AV_OPT_TYPE_INT
Definition: opt.h:225
export_row16
static void export_row16(AVComplexFloat *src, uint8_t *dstp, int rw, int depth, float *win)
Definition: vf_fftdnoiz.c:174
av_clip_uint8
#define av_clip_uint8
Definition: common.h:102
AV_PIX_FMT_YUV444P
@ AV_PIX_FMT_YUV444P
planar YUV 4:4:4, 24bpp, (1 Cr & Cb sample per 1x1 Y samples)
Definition: pixfmt.h:71
AVFilterContext
An instance of a filter.
Definition: avfilter.h:397
factor
static const int factor[16]
Definition: vf_pp7.c:78
AV_PIX_FMT_GBRP
@ AV_PIX_FMT_GBRP
planar GBR 4:4:4 24bpp
Definition: pixfmt.h:158
AVFILTER_FLAG_SLICE_THREADS
#define AVFILTER_FLAG_SLICE_THREADS
The filter supports multithreading by splitting frames into multiple parts and processing them concur...
Definition: avfilter.h:117
desc
const char * desc
Definition: libsvtav1.c:83
AVMEDIA_TYPE_VIDEO
@ AVMEDIA_TYPE_VIDEO
Definition: avutil.h:201
AV_PIX_FMT_YUV422P
@ AV_PIX_FMT_YUV422P
planar YUV 4:2:2, 16bpp, (1 Cr & Cb sample per 2x1 Y samples)
Definition: pixfmt.h:70
import_row8
static void import_row8(AVComplexFloat *dst, uint8_t *src, int rw, float scale, float *win, int off)
Definition: vf_fftdnoiz.c:146
FFTdnoizContext::block_size
int block_size
Definition: vf_fftdnoiz.c:59
AVPixFmtDescriptor
Descriptor that unambiguously describes how the bits of a pixel are stored in the up to 4 data planes...
Definition: pixdesc.h:69
FFTdnoizContext::fft
AVTXContext * fft[MAX_THREADS]
Definition: vf_fftdnoiz.c:75
filter_frame
static int filter_frame(AVFilterLink *inlink, AVFrame *in)
Definition: vf_fftdnoiz.c:594
PlaneContext::n
float n
Definition: vf_fftdnoiz.c:45
FILTER_OUTPUTS
#define FILTER_OUTPUTS(array)
Definition: internal.h:193
av_freep
#define av_freep(p)
Definition: tableprint_vlc.h:34
src
INIT_CLIP pixel * src
Definition: h264pred_template.c:418
AV_PIX_FMT_YUV411P
@ AV_PIX_FMT_YUV411P
planar YUV 4:1:1, 12bpp, (1 Cr & Cb sample per 4x1 Y samples)
Definition: pixfmt.h:73
AVFILTER_FLAG_SUPPORT_TIMELINE_INTERNAL
#define AVFILTER_FLAG_SUPPORT_TIMELINE_INTERNAL
Same as AVFILTER_FLAG_SUPPORT_TIMELINE_GENERIC, except that the filter will have its filter_frame() c...
Definition: avfilter.h:155
imgutils.h
AV_PIX_FMT_YUV410P
@ AV_PIX_FMT_YUV410P
planar YUV 4:1:0, 9bpp, (1 Cr & Cb sample per 4x4 Y samples)
Definition: pixfmt.h:72
block
The exact code depends on how similar the blocks are and how related they are to the block
Definition: filter_design.txt:207
av_log
#define av_log(a,...)
Definition: tableprint_vlc.h:27
export_row8
static void export_row8(AVComplexFloat *src, uint8_t *dst, int rw, int depth, float *win)
Definition: vf_fftdnoiz.c:156
AV_PIX_FMT_YUV440P12
#define AV_PIX_FMT_YUV440P12
Definition: pixfmt.h:474
AV_PIX_FMT_YUV444P14
#define AV_PIX_FMT_YUV444P14
Definition: pixfmt.h:478
AVFILTER_DEFINE_CLASS
AVFILTER_DEFINE_CLASS(fftdnoiz)
fftdnoiz_options
static const AVOption fftdnoiz_options[]
Definition: vf_fftdnoiz.c:88
AV_PIX_FMT_GRAY12
#define AV_PIX_FMT_GRAY12
Definition: pixfmt.h:450
ff_filter_execute
static av_always_inline int ff_filter_execute(AVFilterContext *ctx, avfilter_action_func *func, void *arg, int *ret, int nb_jobs)
Definition: internal.h:144
AV_OPT_TYPE_CONST
@ AV_OPT_TYPE_CONST
Definition: opt.h:234
BufferTypes
BufferTypes
Definition: vf_fftdnoiz.c:33
AV_PIX_FMT_YUVA422P
@ AV_PIX_FMT_YUVA422P
planar YUV 4:2:2 24bpp, (1 Cr & Cb sample per 2x1 Y & A samples)
Definition: pixfmt.h:166
AV_PIX_FMT_YUV420P14
#define AV_PIX_FMT_YUV420P14
Definition: pixfmt.h:476
tx.h
FFTdnoizContext::method
int method
Definition: vf_fftdnoiz.c:61
Generated on Sat Dec 2 2023 19:22:06 for FFmpeg by   doxygen 1.8.17