FFmpeg
vf_vaguedenoiser.c
Go to the documentation of this file.
1 /*
2  * Copyright (c) 2003 LeFunGus, lefungus@altern.org
3  *
4  * This file is part of FFmpeg
5  *
6  * FFmpeg is free software; you can redistribute it and/or modify
7  * it under the terms of the GNU General Public License as published by
8  * the Free Software Foundation; either version 2 of the License, or
9  * (at your option) any later version.
10  *
11  * FFmpeg is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14  * GNU General Public License for more details.
15  *
16  * You should have received a copy of the GNU General Public License along
17  * with FFmpeg; if not, write to the Free Software Foundation, Inc.,
18  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
19  */
20 
21 #include <float.h>
22 
23 #include "libavutil/imgutils.h"
24 #include "libavutil/attributes.h"
25 #include "libavutil/common.h"
26 #include "libavutil/pixdesc.h"
27 #include "libavutil/intreadwrite.h"
28 #include "libavutil/opt.h"
29 
30 #include "avfilter.h"
31 #include "formats.h"
32 #include "internal.h"
33 #include "video.h"
34 
35 typedef struct VagueDenoiserContext {
36  const AVClass *class;
37 
38  float threshold;
39  float percent;
40  int method;
41  int type;
42  int nsteps;
43  int planes;
44 
45  int depth;
46  int bpc;
47  int peak;
48  int nb_planes;
49  int planeheight[4];
50  int planewidth[4];
51 
52  float *block;
53  float *in;
54  float *out;
55  float *tmp;
56 
57  int hlowsize[4][32];
58  int hhighsize[4][32];
59  int vlowsize[4][32];
60  int vhighsize[4][32];
61 
62  void (*thresholding)(float *block, const int width, const int height,
63  const int stride, const float threshold,
64  const float percent);
66 
67 #define OFFSET(x) offsetof(VagueDenoiserContext, x)
68 #define FLAGS AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_FILTERING_PARAM
69 static const AVOption vaguedenoiser_options[] = {
70  { "threshold", "set filtering strength", OFFSET(threshold), AV_OPT_TYPE_FLOAT, {.dbl=2.}, 0,DBL_MAX, FLAGS },
71  { "method", "set filtering method", OFFSET(method), AV_OPT_TYPE_INT, {.i64=2 }, 0, 2, FLAGS, "method" },
72  { "hard", "hard thresholding", 0, AV_OPT_TYPE_CONST, {.i64=0}, 0, 0, FLAGS, "method" },
73  { "soft", "soft thresholding", 0, AV_OPT_TYPE_CONST, {.i64=1}, 0, 0, FLAGS, "method" },
74  { "garrote", "garrote thresholding", 0, AV_OPT_TYPE_CONST, {.i64=2}, 0, 0, FLAGS, "method" },
75  { "nsteps", "set number of steps", OFFSET(nsteps), AV_OPT_TYPE_INT, {.i64=6 }, 1, 32, FLAGS },
76  { "percent", "set percent of full denoising", OFFSET(percent),AV_OPT_TYPE_FLOAT, {.dbl=85}, 0,100, FLAGS },
77  { "planes", "set planes to filter", OFFSET(planes), AV_OPT_TYPE_INT, {.i64=15 }, 0, 15, FLAGS },
78  { "type", "set threshold type", OFFSET(type), AV_OPT_TYPE_INT, {.i64=0 }, 0, 1, FLAGS, "type" },
79  { "universal", "universal (VisuShrink)", 0, AV_OPT_TYPE_CONST, {.i64=0}, 0, 0, FLAGS, "type" },
80  { "bayes", "bayes (BayesShrink)", 0, AV_OPT_TYPE_CONST, {.i64=1}, 0, 0, FLAGS, "type" },
81  { NULL }
82 };
83 
84 AVFILTER_DEFINE_CLASS(vaguedenoiser);
85 
86 #define NPAD 10
87 
88 static const float analysis_low[9] = {
89  0.037828455506995f, -0.023849465019380f, -0.110624404418423f, 0.377402855612654f,
90  0.852698679009403f, 0.377402855612654f, -0.110624404418423f, -0.023849465019380f, 0.037828455506995f
91 };
92 
93 static const float analysis_high[7] = {
94  -0.064538882628938f, 0.040689417609558f, 0.418092273222212f, -0.788485616405664f,
95  0.418092273222212f, 0.040689417609558f, -0.064538882628938f
96 };
97 
98 static const float synthesis_low[7] = {
99  -0.064538882628938f, -0.040689417609558f, 0.418092273222212f, 0.788485616405664f,
100  0.418092273222212f, -0.040689417609558f, -0.064538882628938f
101 };
102 
103 static const float synthesis_high[9] = {
104  -0.037828455506995f, -0.023849465019380f, 0.110624404418423f, 0.377402855612654f,
105  -0.852698679009403f, 0.377402855612654f, 0.110624404418423f, -0.023849465019380f, -0.037828455506995f
106 };
107 
109 {
110  static const enum AVPixelFormat pix_fmts[] = {
134  };
136 }
137 
139 {
140  VagueDenoiserContext *s = inlink->dst->priv;
142  int p, i, nsteps_width, nsteps_height, nsteps_max;
143 
144  s->depth = desc->comp[0].depth;
145  s->bpc = (s->depth + 7) / 8;
146  s->nb_planes = desc->nb_components;
147 
148  s->planeheight[1] = s->planeheight[2] = AV_CEIL_RSHIFT(inlink->h, desc->log2_chroma_h);
149  s->planeheight[0] = s->planeheight[3] = inlink->h;
150  s->planewidth[1] = s->planewidth[2] = AV_CEIL_RSHIFT(inlink->w, desc->log2_chroma_w);
151  s->planewidth[0] = s->planewidth[3] = inlink->w;
152 
153  s->block = av_malloc_array(inlink->w * inlink->h, sizeof(*s->block));
154  s->in = av_malloc_array(32 + FFMAX(inlink->w, inlink->h), sizeof(*s->in));
155  s->out = av_malloc_array(32 + FFMAX(inlink->w, inlink->h), sizeof(*s->out));
156  s->tmp = av_malloc_array(32 + FFMAX(inlink->w, inlink->h), sizeof(*s->tmp));
157 
158  if (!s->block || !s->in || !s->out || !s->tmp)
159  return AVERROR(ENOMEM);
160 
161  s->threshold *= 1 << (s->depth - 8);
162  s->peak = (1 << s->depth) - 1;
163 
164  nsteps_width = ((s->planes & 2 || s->planes & 4) && s->nb_planes > 1) ? s->planewidth[1] : s->planewidth[0];
165  nsteps_height = ((s->planes & 2 || s->planes & 4) && s->nb_planes > 1) ? s->planeheight[1] : s->planeheight[0];
166 
167  for (nsteps_max = 1; nsteps_max < 15; nsteps_max++) {
168  if (pow(2, nsteps_max) >= nsteps_width || pow(2, nsteps_max) >= nsteps_height)
169  break;
170  }
171 
172  s->nsteps = FFMIN(s->nsteps, nsteps_max - 2);
173 
174  for (p = 0; p < 4; p++) {
175  s->hlowsize[p][0] = (s->planewidth[p] + 1) >> 1;
176  s->hhighsize[p][0] = s->planewidth[p] >> 1;
177  s->vlowsize[p][0] = (s->planeheight[p] + 1) >> 1;
178  s->vhighsize[p][0] = s->planeheight[p] >> 1;
179 
180  for (i = 1; i < s->nsteps; i++) {
181  s->hlowsize[p][i] = (s->hlowsize[p][i - 1] + 1) >> 1;
182  s->hhighsize[p][i] = s->hlowsize[p][i - 1] >> 1;
183  s->vlowsize[p][i] = (s->vlowsize[p][i - 1] + 1) >> 1;
184  s->vhighsize[p][i] = s->vlowsize[p][i - 1] >> 1;
185  }
186  }
187 
188  return 0;
189 }
190 
191 static inline void copy(const float *p1, float *p2, const int length)
192 {
193  memcpy(p2, p1, length * sizeof(float));
194 }
195 
196 static inline void copyv(const float *p1, const int stride1, float *p2, const int length)
197 {
198  int i;
199 
200  for (i = 0; i < length; i++) {
201  p2[i] = *p1;
202  p1 += stride1;
203  }
204 }
205 
206 static inline void copyh(const float *p1, float *p2, const int stride2, const int length)
207 {
208  int i;
209 
210  for (i = 0; i < length; i++) {
211  *p2 = p1[i];
212  p2 += stride2;
213  }
214 }
215 
216 // Do symmetric extension of data using prescribed symmetries
217 // Original values are in output[npad] through output[npad+size-1]
218 // New values will be placed in output[0] through output[npad] and in output[npad+size] through output[2*npad+size-1] (note: end values may not be filled in)
219 // extension at left bdry is ... 3 2 1 0 | 0 1 2 3 ...
220 // same for right boundary
221 // if right_ext=1 then ... 3 2 1 0 | 1 2 3
222 static void symmetric_extension(float *output, const int size, const int left_ext, const int right_ext)
223 {
224  int first = NPAD;
225  int last = NPAD - 1 + size;
226  const int originalLast = last;
227  int i, nextend, idx;
228 
229  if (left_ext == 2)
230  output[--first] = output[NPAD];
231  if (right_ext == 2)
232  output[++last] = output[originalLast];
233 
234  // extend left end
235  nextend = first;
236  for (i = 0; i < nextend; i++)
237  output[--first] = output[NPAD + 1 + i];
238 
239  idx = NPAD + NPAD - 1 + size;
240 
241  // extend right end
242  nextend = idx - last;
243  for (i = 0; i < nextend; i++)
244  output[++last] = output[originalLast - 1 - i];
245 }
246 
247 static void transform_step(float *input, float *output, const int size, const int low_size, VagueDenoiserContext *s)
248 {
249  int i;
250 
252 
253  for (i = NPAD; i < NPAD + low_size; i++) {
254  const float a = input[2 * i - 14] * analysis_low[0];
255  const float b = input[2 * i - 13] * analysis_low[1];
256  const float c = input[2 * i - 12] * analysis_low[2];
257  const float d = input[2 * i - 11] * analysis_low[3];
258  const float e = input[2 * i - 10] * analysis_low[4];
259  const float f = input[2 * i - 9] * analysis_low[3];
260  const float g = input[2 * i - 8] * analysis_low[2];
261  const float h = input[2 * i - 7] * analysis_low[1];
262  const float k = input[2 * i - 6] * analysis_low[0];
263 
264  output[i] = a + b + c + d + e + f + g + h + k;
265  }
266 
267  for (i = NPAD; i < NPAD + low_size; i++) {
268  const float a = input[2 * i - 12] * analysis_high[0];
269  const float b = input[2 * i - 11] * analysis_high[1];
270  const float c = input[2 * i - 10] * analysis_high[2];
271  const float d = input[2 * i - 9] * analysis_high[3];
272  const float e = input[2 * i - 8] * analysis_high[2];
273  const float f = input[2 * i - 7] * analysis_high[1];
274  const float g = input[2 * i - 6] * analysis_high[0];
275 
276  output[i + low_size] = a + b + c + d + e + f + g;
277  }
278 }
279 
280 static void invert_step(const float *input, float *output, float *temp, const int size, VagueDenoiserContext *s)
281 {
282  const int low_size = (size + 1) >> 1;
283  const int high_size = size >> 1;
284  int left_ext = 1, right_ext, i;
285  int findex;
286 
287  memcpy(temp + NPAD, input + NPAD, low_size * sizeof(float));
288 
289  right_ext = (size % 2 == 0) ? 2 : 1;
290  symmetric_extension(temp, low_size, left_ext, right_ext);
291 
292  memset(output, 0, (NPAD + NPAD + size) * sizeof(float));
293  findex = (size + 2) >> 1;
294 
295  for (i = 9; i < findex + 11; i++) {
296  const float a = temp[i] * synthesis_low[0];
297  const float b = temp[i] * synthesis_low[1];
298  const float c = temp[i] * synthesis_low[2];
299  const float d = temp[i] * synthesis_low[3];
300 
301  output[2 * i - 13] += a;
302  output[2 * i - 12] += b;
303  output[2 * i - 11] += c;
304  output[2 * i - 10] += d;
305  output[2 * i - 9] += c;
306  output[2 * i - 8] += b;
307  output[2 * i - 7] += a;
308  }
309 
310  memcpy(temp + NPAD, input + NPAD + low_size, high_size * sizeof(float));
311 
312  left_ext = 2;
313  right_ext = (size % 2 == 0) ? 1 : 2;
314  symmetric_extension(temp, high_size, left_ext, right_ext);
315 
316  for (i = 8; i < findex + 11; i++) {
317  const float a = temp[i] * synthesis_high[0];
318  const float b = temp[i] * synthesis_high[1];
319  const float c = temp[i] * synthesis_high[2];
320  const float d = temp[i] * synthesis_high[3];
321  const float e = temp[i] * synthesis_high[4];
322 
323  output[2 * i - 13] += a;
324  output[2 * i - 12] += b;
325  output[2 * i - 11] += c;
326  output[2 * i - 10] += d;
327  output[2 * i - 9] += e;
328  output[2 * i - 8] += d;
329  output[2 * i - 7] += c;
330  output[2 * i - 6] += b;
331  output[2 * i - 5] += a;
332  }
333 }
334 
335 static void hard_thresholding(float *block, const int width, const int height,
336  const int stride, const float threshold,
337  const float percent)
338 {
339  const float frac = 1.f - percent * 0.01f;
340  int y, x;
341 
342  for (y = 0; y < height; y++) {
343  for (x = 0; x < width; x++) {
344  if (FFABS(block[x]) <= threshold)
345  block[x] *= frac;
346  }
347  block += stride;
348  }
349 }
350 
351 static void soft_thresholding(float *block, const int width, const int height, const int stride,
352  const float threshold, const float percent)
353 {
354  const float frac = 1.f - percent * 0.01f;
355  const float shift = threshold * 0.01f * percent;
356  int y, x;
357 
358  for (y = 0; y < height; y++) {
359  for (x = 0; x < width; x++) {
360  const float temp = FFABS(block[x]);
361  if (temp <= threshold)
362  block[x] *= frac;
363  else
364  block[x] = (block[x] < 0.f ? -1.f : (block[x] > 0.f ? 1.f : 0.f)) * (temp - shift);
365  }
366  block += stride;
367  }
368 }
369 
370 static void qian_thresholding(float *block, const int width, const int height,
371  const int stride, const float threshold,
372  const float percent)
373 {
374  const float percent01 = percent * 0.01f;
375  const float tr2 = threshold * threshold * percent01;
376  const float frac = 1.f - percent01;
377  int y, x;
378 
379  for (y = 0; y < height; y++) {
380  for (x = 0; x < width; x++) {
381  const float temp = FFABS(block[x]);
382  if (temp <= threshold) {
383  block[x] *= frac;
384  } else {
385  const float tp2 = temp * temp;
386  block[x] *= (tp2 - tr2) / tp2;
387  }
388  }
389  block += stride;
390  }
391 }
392 
393 static float bayes_threshold(float *block, const int width, const int height,
394  const int stride, const float threshold)
395 {
396  float mean = 0.f;
397 
398  for (int y = 0; y < height; y++) {
399  for (int x = 0; x < width; x++) {
400  mean += block[x] * block[x];
401  }
402  block += stride;
403  }
404 
405  mean /= width * height;
406 
407  return threshold * threshold / (FFMAX(sqrtf(mean - threshold), FLT_EPSILON));
408 }
409 
411 {
412  int p, y, x, i, j;
413 
414  for (p = 0; p < s->nb_planes; p++) {
415  const int height = s->planeheight[p];
416  const int width = s->planewidth[p];
417  const uint8_t *srcp8 = in->data[p];
418  const uint16_t *srcp16 = (const uint16_t *)in->data[p];
419  uint8_t *dstp8 = out->data[p];
420  uint16_t *dstp16 = (uint16_t *)out->data[p];
421  float *output = s->block;
422  int h_low_size0 = width;
423  int v_low_size0 = height;
424  int nsteps_transform = s->nsteps;
425  int nsteps_invert = s->nsteps;
426  const float *input = s->block;
427 
428  if (!((1 << p) & s->planes)) {
429  av_image_copy_plane(out->data[p], out->linesize[p], in->data[p], in->linesize[p],
430  s->planewidth[p] * s->bpc, s->planeheight[p]);
431  continue;
432  }
433 
434  if (s->depth <= 8) {
435  for (y = 0; y < height; y++) {
436  for (x = 0; x < width; x++)
437  output[x] = srcp8[x];
438  srcp8 += in->linesize[p];
439  output += width;
440  }
441  } else {
442  for (y = 0; y < height; y++) {
443  for (x = 0; x < width; x++)
444  output[x] = srcp16[x];
445  srcp16 += in->linesize[p] / 2;
446  output += width;
447  }
448  }
449 
450  while (nsteps_transform--) {
451  int low_size = (h_low_size0 + 1) >> 1;
452  float *input = s->block;
453  for (j = 0; j < v_low_size0; j++) {
454  copy(input, s->in + NPAD, h_low_size0);
455  transform_step(s->in, s->out, h_low_size0, low_size, s);
456  copy(s->out + NPAD, input, h_low_size0);
457  input += width;
458  }
459 
460  low_size = (v_low_size0 + 1) >> 1;
461  input = s->block;
462  for (j = 0; j < h_low_size0; j++) {
463  copyv(input, width, s->in + NPAD, v_low_size0);
464  transform_step(s->in, s->out, v_low_size0, low_size, s);
465  copyh(s->out + NPAD, input, width, v_low_size0);
466  input++;
467  }
468 
469  h_low_size0 = (h_low_size0 + 1) >> 1;
470  v_low_size0 = (v_low_size0 + 1) >> 1;
471  }
472 
473  if (s->type == 0) {
474  s->thresholding(s->block, width, height, width, s->threshold, s->percent);
475  } else {
476  for (int n = 0; n < s->nsteps; n++) {
477  float threshold;
478  float *block;
479 
480  if (n == s->nsteps - 1) {
481  threshold = bayes_threshold(s->block, s->hlowsize[p][n], s->vlowsize[p][n], width, s->threshold);
482  s->thresholding(s->block, s->hlowsize[p][n], s->vlowsize[p][n], width, threshold, s->percent);
483  }
484  block = s->block + s->hlowsize[p][n];
485  threshold = bayes_threshold(block, s->hhighsize[p][n], s->vlowsize[p][n], width, s->threshold);
486  s->thresholding(block, s->hhighsize[p][n], s->vlowsize[p][n], width, threshold, s->percent);
487  block = s->block + s->vlowsize[p][n] * width;
488  threshold = bayes_threshold(block, s->hlowsize[p][n], s->vhighsize[p][n], width, s->threshold);
489  s->thresholding(block, s->hlowsize[p][n], s->vhighsize[p][n], width, threshold, s->percent);
490  block = s->block + s->hlowsize[p][n] + s->vlowsize[p][n] * width;
491  threshold = bayes_threshold(block, s->hhighsize[p][n], s->vhighsize[p][n], width, s->threshold);
492  s->thresholding(block, s->hhighsize[p][n], s->vhighsize[p][n], width, threshold, s->percent);
493  }
494  }
495 
496  while (nsteps_invert--) {
497  const int idx = s->vlowsize[p][nsteps_invert] + s->vhighsize[p][nsteps_invert];
498  const int idx2 = s->hlowsize[p][nsteps_invert] + s->hhighsize[p][nsteps_invert];
499  float * idx3 = s->block;
500  for (i = 0; i < idx2; i++) {
501  copyv(idx3, width, s->in + NPAD, idx);
502  invert_step(s->in, s->out, s->tmp, idx, s);
503  copyh(s->out + NPAD, idx3, width, idx);
504  idx3++;
505  }
506 
507  idx3 = s->block;
508  for (i = 0; i < idx; i++) {
509  copy(idx3, s->in + NPAD, idx2);
510  invert_step(s->in, s->out, s->tmp, idx2, s);
511  copy(s->out + NPAD, idx3, idx2);
512  idx3 += width;
513  }
514  }
515 
516  if (s->depth <= 8) {
517  for (y = 0; y < height; y++) {
518  for (x = 0; x < width; x++)
519  dstp8[x] = av_clip_uint8(input[x] + 0.5f);
520  input += width;
521  dstp8 += out->linesize[p];
522  }
523  } else {
524  for (y = 0; y < height; y++) {
525  for (x = 0; x < width; x++)
526  dstp16[x] = av_clip(input[x] + 0.5f, 0, s->peak);
527  input += width;
528  dstp16 += out->linesize[p] / 2;
529  }
530  }
531  }
532 }
533 
535 {
536  AVFilterContext *ctx = inlink->dst;
537  VagueDenoiserContext *s = ctx->priv;
538  AVFilterLink *outlink = ctx->outputs[0];
539  AVFrame *out;
540  int direct = av_frame_is_writable(in);
541 
542  if (direct) {
543  out = in;
544  } else {
545  out = ff_get_video_buffer(outlink, outlink->w, outlink->h);
546  if (!out) {
547  av_frame_free(&in);
548  return AVERROR(ENOMEM);
549  }
550 
552  }
553 
554  filter(s, in, out);
555 
556  if (!direct)
557  av_frame_free(&in);
558 
559  return ff_filter_frame(outlink, out);
560 }
561 
563 {
564  VagueDenoiserContext *s = ctx->priv;
565 
566  switch (s->method) {
567  case 0:
568  s->thresholding = hard_thresholding;
569  break;
570  case 1:
571  s->thresholding = soft_thresholding;
572  break;
573  case 2:
574  s->thresholding = qian_thresholding;
575  break;
576  }
577 
578  return 0;
579 }
580 
582 {
583  VagueDenoiserContext *s = ctx->priv;
584 
585  av_freep(&s->block);
586  av_freep(&s->in);
587  av_freep(&s->out);
588  av_freep(&s->tmp);
589 }
590 
592  {
593  .name = "default",
594  .type = AVMEDIA_TYPE_VIDEO,
595  .config_props = config_input,
596  .filter_frame = filter_frame,
597  },
598 };
599 
600 
602  {
603  .name = "default",
604  .type = AVMEDIA_TYPE_VIDEO
605  },
606 };
607 
609  .name = "vaguedenoiser",
610  .description = NULL_IF_CONFIG_SMALL("Apply a Wavelet based Denoiser."),
611  .priv_size = sizeof(VagueDenoiserContext),
612  .priv_class = &vaguedenoiser_class,
613  .init = init,
614  .uninit = uninit,
619 };
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:98
AV_PIX_FMT_YUVA422P16
#define AV_PIX_FMT_YUVA422P16
Definition: pixfmt.h:432
AV_PIX_FMT_GBRAP16
#define AV_PIX_FMT_GBRAP16
Definition: pixfmt.h:411
stride
int stride
Definition: mace.c:144
direct
static void direct(const float *in, const FFTComplex *ir, int len, float *out)
Definition: af_afir.c:61
AVPixelFormat
AVPixelFormat
Pixel format.
Definition: pixfmt.h:64
av_clip
#define av_clip
Definition: common.h:96
transform_step
static void transform_step(float *input, float *output, const int size, const int low_size, VagueDenoiserContext *s)
Definition: vf_vaguedenoiser.c:247
filter
static void filter(VagueDenoiserContext *s, AVFrame *in, AVFrame *out)
Definition: vf_vaguedenoiser.c:410
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
out
FILE * out
Definition: movenc.c:54
query_formats
static int query_formats(AVFilterContext *ctx)
Definition: vf_vaguedenoiser.c:108
ff_filter_frame
int ff_filter_frame(AVFilterLink *link, AVFrame *frame)
Send a frame of data to the next filter.
Definition: avfilter.c:1019
av_pix_fmt_desc_get
const AVPixFmtDescriptor * av_pix_fmt_desc_get(enum AVPixelFormat pix_fmt)
Definition: pixdesc.c:2540
copyv
static void copyv(const float *p1, const int stride1, float *p2, const int length)
Definition: vf_vaguedenoiser.c:196
OFFSET
#define OFFSET(x)
Definition: vf_vaguedenoiser.c:67
output
filter_frame For filters that do not use the this method is called when a frame is pushed to the filter s input It can be called at any time except in a reentrant way If the input frame is enough to produce output
Definition: filter_design.txt:225
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
av_frame_free
void av_frame_free(AVFrame **frame)
Free the frame and any dynamically allocated objects in it, e.g.
Definition: frame.c:112
AV_PIX_FMT_YUVA422P9
#define AV_PIX_FMT_YUVA422P9
Definition: pixfmt.h:424
VagueDenoiserContext::in
float * in
Definition: vf_vaguedenoiser.c:53
AVFrame
This structure describes decoded (raw) audio or video data.
Definition: frame.h:303
pixdesc.h
AV_PIX_FMT_YUVA420P16
#define AV_PIX_FMT_YUVA420P16
Definition: pixfmt.h:431
vaguedenoiser_outputs
static const AVFilterPad vaguedenoiser_outputs[]
Definition: vf_vaguedenoiser.c:601
AV_PIX_FMT_YUVA420P10
#define AV_PIX_FMT_YUVA420P10
Definition: pixfmt.h:426
AVOption
AVOption.
Definition: opt.h:247
b
#define b
Definition: input.c:40
AV_PIX_FMT_YUV420P10
#define AV_PIX_FMT_YUV420P10
Definition: pixfmt.h:389
float.h
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
FFMAX
#define FFMAX(a, b)
Definition: macros.h:47
AVFilter::name
const char * name
Filter name.
Definition: avfilter.h:153
video.h
copyh
static void copyh(const float *p1, float *p2, const int stride2, const int length)
Definition: vf_vaguedenoiser.c:206
AV_PIX_FMT_YUVA422P10
#define AV_PIX_FMT_YUVA422P10
Definition: pixfmt.h:427
AV_PIX_FMT_GRAY9
#define AV_PIX_FMT_GRAY9
Definition: pixfmt.h:369
symmetric_extension
static void symmetric_extension(float *output, const int size, const int left_ext, const int right_ext)
Definition: vf_vaguedenoiser.c:222
AVFrame::data
uint8_t * data[AV_NUM_DATA_POINTERS]
pointer to the picture/channel planes.
Definition: frame.h:317
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
VagueDenoiserContext::hhighsize
int hhighsize[4][32]
Definition: vf_vaguedenoiser.c:58
formats.h
AV_PIX_FMT_YUVA420P9
#define AV_PIX_FMT_YUVA420P9
Definition: pixfmt.h:423
AV_PIX_FMT_GBRP14
#define AV_PIX_FMT_GBRP14
Definition: pixfmt.h:407
AV_PIX_FMT_GBRAP
@ AV_PIX_FMT_GBRAP
planar GBRA 4:4:4:4 32bpp
Definition: pixfmt.h:205
analysis_high
static const float analysis_high[7]
Definition: vf_vaguedenoiser.c:93
AV_PIX_FMT_GBRP10
#define AV_PIX_FMT_GBRP10
Definition: pixfmt.h:405
AV_PIX_FMT_YUVA444P16
#define AV_PIX_FMT_YUVA444P16
Definition: pixfmt.h:433
VagueDenoiserContext::depth
int depth
Definition: vf_vaguedenoiser.c:45
VagueDenoiserContext::peak
int peak
Definition: vf_vaguedenoiser.c:47
AV_PIX_FMT_YUV422P9
#define AV_PIX_FMT_YUV422P9
Definition: pixfmt.h:387
invert_step
static void invert_step(const float *input, float *output, float *temp, const int size, VagueDenoiserContext *s)
Definition: vf_vaguedenoiser.c:280
type
it s the only field you need to keep assuming you have a context There is some magic you don t need to care about around this just let it vf type
Definition: writing_filters.txt:86
VagueDenoiserContext::nsteps
int nsteps
Definition: vf_vaguedenoiser.c:42
AV_PIX_FMT_GRAY16
#define AV_PIX_FMT_GRAY16
Definition: pixfmt.h:373
qian_thresholding
static void qian_thresholding(float *block, const int width, const int height, const int stride, const float threshold, const float percent)
Definition: vf_vaguedenoiser.c:370
AVFilterPad
A filter pad used for either input or output.
Definition: internal.h:50
VagueDenoiserContext::nb_planes
int nb_planes
Definition: vf_vaguedenoiser.c:48
AV_PIX_FMT_YUV444P10
#define AV_PIX_FMT_YUV444P10
Definition: pixfmt.h:392
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:248
first
trying all byte sequences megabyte in length and selecting the best looking sequence will yield cases to try But first
Definition: rate_distortion.txt:12
planes
static const struct @318 planes[]
av_cold
#define av_cold
Definition: attributes.h:90
AV_PIX_FMT_YUV422P16
#define AV_PIX_FMT_YUV422P16
Definition: pixfmt.h:401
VagueDenoiserContext::thresholding
void(* thresholding)(float *block, const int width, const int height, const int stride, const float threshold, const float percent)
Definition: vf_vaguedenoiser.c:62
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:409
width
#define width
synthesis_low
static const float synthesis_low[7]
Definition: vf_vaguedenoiser.c:98
intreadwrite.h
s
#define s(width, name)
Definition: cbs_vp9.c:257
AV_PIX_FMT_GBRAP12
#define AV_PIX_FMT_GBRAP12
Definition: pixfmt.h:410
VagueDenoiserContext::percent
float percent
Definition: vf_vaguedenoiser.c:39
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:402
AV_CEIL_RSHIFT
#define AV_CEIL_RSHIFT(a, b)
Definition: common.h:51
g
const char * g
Definition: vf_curves.c:117
ff_set_common_formats_from_list
int ff_set_common_formats_from_list(AVFilterContext *ctx, const int *fmts)
Equivalent to ff_set_common_formats(ctx, ff_make_format_list(fmts))
Definition: formats.c:703
pix_fmts
static enum AVPixelFormat pix_fmts[]
Definition: libkvazaar.c:290
AV_PIX_FMT_YUVA444P12
#define AV_PIX_FMT_YUVA444P12
Definition: pixfmt.h:430
AV_PIX_FMT_YUV420P9
#define AV_PIX_FMT_YUV420P9
Definition: pixfmt.h:386
AV_PIX_FMT_YUV420P16
#define AV_PIX_FMT_YUV420P16
Definition: pixfmt.h:400
ctx
AVFormatContext * ctx
Definition: movenc.c:48
AV_PIX_FMT_GRAY14
#define AV_PIX_FMT_GRAY14
Definition: pixfmt.h:372
synthesis_high
static const float synthesis_high[9]
Definition: vf_vaguedenoiser.c:103
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
VagueDenoiserContext::tmp
float * tmp
Definition: vf_vaguedenoiser.c:55
f
#define f(width, name)
Definition: cbs_vp9.c:255
FILTER_INPUTS
#define FILTER_INPUTS(array)
Definition: internal.h:152
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
bayes_threshold
static float bayes_threshold(float *block, const int width, const int height, const int stride, const float threshold)
Definition: vf_vaguedenoiser.c:393
FFABS
#define FFABS(a)
Absolute value, Note, INT_MIN / INT64_MIN result in undefined behavior as they are not representable ...
Definition: common.h:65
AV_PIX_FMT_GRAY10
#define AV_PIX_FMT_GRAY10
Definition: pixfmt.h:370
VagueDenoiserContext
Definition: vf_vaguedenoiser.c:35
AV_PIX_FMT_GBRP16
#define AV_PIX_FMT_GBRP16
Definition: pixfmt.h:408
AVClass
Describe the class of an AVClass context structure.
Definition: log.h:66
NULL
#define NULL
Definition: coverity.c:32
filter_frame
static int filter_frame(AVFilterLink *inlink, AVFrame *in)
Definition: vf_vaguedenoiser.c:534
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:537
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
hard_thresholding
static void hard_thresholding(float *block, const int width, const int height, const int stride, const float threshold, const float percent)
Definition: vf_vaguedenoiser.c:335
AV_PIX_FMT_YUV440P10
#define AV_PIX_FMT_YUV440P10
Definition: pixfmt.h:391
vaguedenoiser_inputs
static const AVFilterPad vaguedenoiser_inputs[]
Definition: vf_vaguedenoiser.c:591
AV_PIX_FMT_YUV422P10
#define AV_PIX_FMT_YUV422P10
Definition: pixfmt.h:390
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:404
VagueDenoiserContext::threshold
float threshold
Definition: vf_vaguedenoiser.c:38
c
Undefined Behavior In the C some operations are like signed integer dereferencing freed accessing outside allocated Undefined Behavior must not occur in a C it is not safe even if the output of undefined operations is unused The unsafety may seem nit picking but Optimizing compilers have in fact optimized code on the assumption that no undefined Behavior occurs Optimizing code based on wrong assumptions can and has in some cases lead to effects beyond the output of computations The signed integer overflow problem in speed critical code Code which is highly optimized and works with signed integers sometimes has the problem that often the output of the computation does not c
Definition: undefined.txt:32
FLAGS
#define FLAGS
Definition: vf_vaguedenoiser.c:68
VagueDenoiserContext::hlowsize
int hlowsize[4][32]
Definition: vf_vaguedenoiser.c:57
init
static av_cold int init(AVFilterContext *ctx)
Definition: vf_vaguedenoiser.c:562
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:117
copy
static void copy(const float *p1, float *p2, const int length)
Definition: vf_vaguedenoiser.c:191
AVFILTER_DEFINE_CLASS
AVFILTER_DEFINE_CLASS(vaguedenoiser)
ff_vf_vaguedenoiser
const AVFilter ff_vf_vaguedenoiser
Definition: vf_vaguedenoiser.c:608
AV_PIX_FMT_YUV422P12
#define AV_PIX_FMT_YUV422P12
Definition: pixfmt.h:394
size
int size
Definition: twinvq_data.h:10344
VagueDenoiserContext::vlowsize
int vlowsize[4][32]
Definition: vf_vaguedenoiser.c:59
AV_PIX_FMT_YUV444P12
#define AV_PIX_FMT_YUV444P12
Definition: pixfmt.h:396
VagueDenoiserContext::block
float * block
Definition: vf_vaguedenoiser.c:52
av_frame_is_writable
int av_frame_is_writable(AVFrame *frame)
Check if the frame data is writable.
Definition: frame.c:473
uninit
static av_cold void uninit(AVFilterContext *ctx)
Definition: vf_vaguedenoiser.c:581
height
#define height
a
The reader does not expect b to be semantically here and if the code is changed by maybe adding a a division or other the signedness will almost certainly be mistaken To avoid this confusion a new type was SUINT is the C unsigned type but it holds a signed int to use the same example SUINT a
Definition: undefined.txt:41
config_input
static int config_input(AVFilterLink *inlink)
Definition: vf_vaguedenoiser.c:138
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:428
attributes.h
VagueDenoiserContext::planewidth
int planewidth[4]
Definition: vf_vaguedenoiser.c:50
input
and forward the test the status of outputs and forward it to the corresponding return FFERROR_NOT_READY If the filters stores internally one or a few frame for some input
Definition: filter_design.txt:172
internal.h
VagueDenoiserContext::vhighsize
int vhighsize[4][32]
Definition: vf_vaguedenoiser.c:60
AVFILTER_FLAG_SUPPORT_TIMELINE_GENERIC
#define AVFILTER_FLAG_SUPPORT_TIMELINE_GENERIC
Some filters support a generic "enable" expression option that can be used to enable or disable a fil...
Definition: avfilter.h:130
AV_OPT_TYPE_FLOAT
@ AV_OPT_TYPE_FLOAT
Definition: opt.h:227
VagueDenoiserContext::method
int method
Definition: vf_vaguedenoiser.c:40
VagueDenoiserContext::type
int type
Definition: vf_vaguedenoiser.c:41
i
int i
Definition: input.c:406
av_malloc_array
#define av_malloc_array(a, b)
Definition: tableprint_vlc.h:32
AV_PIX_FMT_GBRP12
#define AV_PIX_FMT_GBRP12
Definition: pixfmt.h:406
common.h
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:56
AV_PIX_FMT_YUV444P9
#define AV_PIX_FMT_YUV444P9
Definition: pixfmt.h:388
vaguedenoiser_options
static const AVOption vaguedenoiser_options[]
Definition: vf_vaguedenoiser.c:69
AVFilter
Filter definition.
Definition: avfilter.h:149
VagueDenoiserContext::planeheight
int planeheight[4]
Definition: vf_vaguedenoiser.c:49
analysis_low
static const float analysis_low[9]
Definition: vf_vaguedenoiser.c:88
AV_PIX_FMT_YUVA444P9
#define AV_PIX_FMT_YUVA444P9
Definition: pixfmt.h:425
left_ext
static int left_ext(int wavelet_length, int levels, uint64_t sn)
Definition: af_afwtdn.c:525
AV_PIX_FMT_YUV420P12
#define AV_PIX_FMT_YUV420P12
Definition: pixfmt.h:393
AV_PIX_FMT_YUV422P14
#define AV_PIX_FMT_YUV422P14
Definition: pixfmt.h:398
soft_thresholding
static void soft_thresholding(float *block, const int width, const int height, const int stride, const float threshold, const float percent)
Definition: vf_vaguedenoiser.c:351
NPAD
#define NPAD
Definition: vf_vaguedenoiser.c:86
AV_PIX_FMT_NONE
@ AV_PIX_FMT_NONE
Definition: pixfmt.h:65
AV_PIX_FMT_YUVA422P12
#define AV_PIX_FMT_YUVA422P12
Definition: pixfmt.h:429
AV_OPT_TYPE_INT
@ AV_OPT_TYPE_INT
Definition: opt.h:224
avfilter.h
temp
else temp
Definition: vf_mcdeint.c:256
mean
static float mean(const float *input, int size)
Definition: vf_nnedi.c:860
VagueDenoiserContext::planes
int planes
Definition: vf_vaguedenoiser.c:43
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:346
shift
static int shift(int a, int b)
Definition: sonic.c:83
AV_PIX_FMT_GBRP
@ AV_PIX_FMT_GBRP
planar GBR 4:4:4 24bpp
Definition: pixfmt.h:158
desc
const char * desc
Definition: libsvtav1.c:79
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
VagueDenoiserContext::out
float * out
Definition: vf_vaguedenoiser.c:54
AVPixFmtDescriptor
Descriptor that unambiguously describes how the bits of a pixel are stored in the up to 4 data planes...
Definition: pixdesc.h:69
FILTER_OUTPUTS
#define FILTER_OUTPUTS(array)
Definition: internal.h:153
av_freep
#define av_freep(p)
Definition: tableprint_vlc.h:35
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
d
d
Definition: ffmpeg_filter.c:156
imgutils.h
AVFrame::linesize
int linesize[AV_NUM_DATA_POINTERS]
For video, size in bytes of each picture line.
Definition: frame.h:334
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_PIX_FMT_YUV440P12
#define AV_PIX_FMT_YUV440P12
Definition: pixfmt.h:395
h
h
Definition: vp9dsp_template.c:2038
AV_PIX_FMT_YUV444P14
#define AV_PIX_FMT_YUV444P14
Definition: pixfmt.h:399
AV_PIX_FMT_GRAY12
#define AV_PIX_FMT_GRAY12
Definition: pixfmt.h:371
AV_OPT_TYPE_CONST
@ AV_OPT_TYPE_CONST
Definition: opt.h:233
VagueDenoiserContext::bpc
int bpc
Definition: vf_vaguedenoiser.c:46
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:397