FFmpeg
vf_nlmeans.c
Go to the documentation of this file.
1 /*
2  * Copyright (c) 2016 Clément Bœsch <u pkh me>
3  *
4  * This file is part of FFmpeg.
5  *
6  * FFmpeg is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU Lesser General Public
8  * License as published by the Free Software Foundation; either
9  * version 2.1 of the License, or (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 GNU
14  * Lesser General Public License for more details.
15  *
16  * You should have received a copy of the GNU Lesser General Public
17  * License along with FFmpeg; if not, write to the Free Software
18  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
19  */
20 
21 /**
22  * @todo
23  * - better automatic defaults? see "Parameters" @ http://www.ipol.im/pub/art/2011/bcm_nlm/
24  * - temporal support (probably doesn't need any displacement according to
25  * "Denoising image sequences does not require motion estimation")
26  * - Bayer pixel format support for at least raw photos? (DNG support would be
27  * handy here)
28  * - FATE test (probably needs visual threshold test mechanism due to the use
29  * of floats)
30  */
31 
32 #include "libavutil/avassert.h"
33 #include "libavutil/opt.h"
34 #include "libavutil/pixdesc.h"
35 #include "avfilter.h"
36 #include "internal.h"
37 #include "vf_nlmeans.h"
38 #include "vf_nlmeans_init.h"
39 #include "video.h"
40 
41 typedef struct NLMeansContext {
42  const AVClass *class;
43  int nb_planes;
45  double pdiff_scale; // invert of the filtering parameter (sigma*10) squared
46  double sigma; // denoising strength
47  int patch_size, patch_hsize; // patch size and half size
48  int patch_size_uv, patch_hsize_uv; // patch size and half size for chroma planes
49  int research_size, research_hsize; // research size and half size
50  int research_size_uv, research_hsize_uv; // research size and half size for chroma planes
51  uint32_t *ii_orig; // integral image
52  uint32_t *ii; // integral image starting after the 0-line and 0-column
53  int ii_w, ii_h; // width and height of the integral image
54  ptrdiff_t ii_lz_32; // linesize in 32-bit units of the integral image
55  float *total_weight; // total weight for every pixel
56  float *sum; // weighted sum for every pixel
57  int linesize; // sum and total_weight linesize
58  float *weight_lut; // lookup table mapping (scaled) patch differences to their associated weights
59  uint32_t max_meaningful_diff; // maximum difference considered (if the patch difference is too high we ignore the pixel)
62 
63 #define OFFSET(x) offsetof(NLMeansContext, x)
64 #define FLAGS AV_OPT_FLAG_FILTERING_PARAM|AV_OPT_FLAG_VIDEO_PARAM
65 static const AVOption nlmeans_options[] = {
66  { "s", "denoising strength", OFFSET(sigma), AV_OPT_TYPE_DOUBLE, { .dbl = 1.0 }, 1.0, 30.0, FLAGS },
67  { "p", "patch size", OFFSET(patch_size), AV_OPT_TYPE_INT, { .i64 = 3*2+1 }, 0, 99, FLAGS },
68  { "pc", "patch size for chroma planes", OFFSET(patch_size_uv), AV_OPT_TYPE_INT, { .i64 = 0 }, 0, 99, FLAGS },
69  { "r", "research window", OFFSET(research_size), AV_OPT_TYPE_INT, { .i64 = 7*2+1 }, 0, 99, FLAGS },
70  { "rc", "research window for chroma planes", OFFSET(research_size_uv), AV_OPT_TYPE_INT, { .i64 = 0 }, 0, 99, FLAGS },
71  { NULL }
72 };
73 
74 AVFILTER_DEFINE_CLASS(nlmeans);
75 
76 static const enum AVPixelFormat pix_fmts[] = {
85 };
86 
87 /**
88  * Compute squared difference of an unsafe area (the zone nor s1 nor s2 could
89  * be readable).
90  *
91  * On the other hand, the line above dst and the column to its left are always
92  * readable.
93  *
94  * There is little point in having this function SIMDified as it is likely too
95  * complex and only handle small portions of the image.
96  *
97  * @param dst integral image
98  * @param dst_linesize_32 integral image linesize (in 32-bit integers unit)
99  * @param startx integral starting x position
100  * @param starty integral starting y position
101  * @param src source plane buffer
102  * @param linesize source plane linesize
103  * @param offx source offsetting in x
104  * @param offy source offsetting in y
105  * @paran r absolute maximum source offsetting
106  * @param sw source width
107  * @param sh source height
108  * @param w width to compute
109  * @param h height to compute
110  */
111 static inline void compute_unsafe_ssd_integral_image(uint32_t *dst, ptrdiff_t dst_linesize_32,
112  int startx, int starty,
113  const uint8_t *src, ptrdiff_t linesize,
114  int offx, int offy, int r, int sw, int sh,
115  int w, int h)
116 {
117  for (int y = starty; y < starty + h; y++) {
118  uint32_t acc = dst[y*dst_linesize_32 + startx - 1] - dst[(y-1)*dst_linesize_32 + startx - 1];
119  const int s1y = av_clip(y - r, 0, sh - 1);
120  const int s2y = av_clip(y - (r + offy), 0, sh - 1);
121 
122  for (int x = startx; x < startx + w; x++) {
123  const int s1x = av_clip(x - r, 0, sw - 1);
124  const int s2x = av_clip(x - (r + offx), 0, sw - 1);
125  const uint8_t v1 = src[s1y*linesize + s1x];
126  const uint8_t v2 = src[s2y*linesize + s2x];
127  const int d = v1 - v2;
128  acc += d * d;
129  dst[y*dst_linesize_32 + x] = dst[(y-1)*dst_linesize_32 + x] + acc;
130  }
131  }
132 }
133 
134 /*
135  * Compute the sum of squared difference integral image
136  * http://www.ipol.im/pub/art/2014/57/
137  * Integral Images for Block Matching - Gabriele Facciolo, Nicolas Limare, Enric Meinhardt-Llopis
138  *
139  * @param ii integral image of dimension (w+e*2) x (h+e*2) with
140  * an additional zeroed top line and column already
141  * "applied" to the pointer value
142  * @param ii_linesize_32 integral image linesize (in 32-bit integers unit)
143  * @param src source plane buffer
144  * @param linesize source plane linesize
145  * @param offx x-offsetting ranging in [-e;e]
146  * @param offy y-offsetting ranging in [-e;e]
147  * @param w source width
148  * @param h source height
149  * @param e research padding edge
150  */
152  uint32_t *ii, ptrdiff_t ii_linesize_32,
153  const uint8_t *src, ptrdiff_t linesize, int offx, int offy,
154  int e, int w, int h)
155 {
156  // ii has a surrounding padding of thickness "e"
157  const int ii_w = w + e*2;
158  const int ii_h = h + e*2;
159 
160  // we center the first source
161  const int s1x = e;
162  const int s1y = e;
163 
164  // 2nd source is the frame with offsetting
165  const int s2x = e + offx;
166  const int s2y = e + offy;
167 
168  // get the dimension of the overlapping rectangle where it is always safe
169  // to compare the 2 sources pixels
170  const int startx_safe = FFMAX(s1x, s2x);
171  const int starty_safe = FFMAX(s1y, s2y);
172  const int u_endx_safe = FFMIN(s1x + w, s2x + w); // unaligned
173  const int endy_safe = FFMIN(s1y + h, s2y + h);
174 
175  // deduce the safe area width and height
176  const int safe_pw = (u_endx_safe - startx_safe) & ~0xf;
177  const int safe_ph = endy_safe - starty_safe;
178 
179  // adjusted end x position of the safe area after width of the safe area gets aligned
180  const int endx_safe = startx_safe + safe_pw;
181 
182  // top part where only one of s1 and s2 is still readable, or none at all
183  compute_unsafe_ssd_integral_image(ii, ii_linesize_32,
184  0, 0,
185  src, linesize,
186  offx, offy, e, w, h,
187  ii_w, starty_safe);
188 
189  // fill the left column integral required to compute the central
190  // overlapping one
191  compute_unsafe_ssd_integral_image(ii, ii_linesize_32,
192  0, starty_safe,
193  src, linesize,
194  offx, offy, e, w, h,
195  startx_safe, safe_ph);
196 
197  // main and safe part of the integral
198  av_assert1(startx_safe - s1x >= 0); av_assert1(startx_safe - s1x < w);
199  av_assert1(starty_safe - s1y >= 0); av_assert1(starty_safe - s1y < h);
200  av_assert1(startx_safe - s2x >= 0); av_assert1(startx_safe - s2x < w);
201  av_assert1(starty_safe - s2y >= 0); av_assert1(starty_safe - s2y < h);
202  if (safe_pw && safe_ph)
203  dsp->compute_safe_ssd_integral_image(ii + starty_safe*ii_linesize_32 + startx_safe, ii_linesize_32,
204  src + (starty_safe - s1y) * linesize + (startx_safe - s1x), linesize,
205  src + (starty_safe - s2y) * linesize + (startx_safe - s2x), linesize,
206  safe_pw, safe_ph);
207 
208  // right part of the integral
209  compute_unsafe_ssd_integral_image(ii, ii_linesize_32,
210  endx_safe, starty_safe,
211  src, linesize,
212  offx, offy, e, w, h,
213  ii_w - endx_safe, safe_ph);
214 
215  // bottom part where only one of s1 and s2 is still readable, or none at all
216  compute_unsafe_ssd_integral_image(ii, ii_linesize_32,
217  0, endy_safe,
218  src, linesize,
219  offx, offy, e, w, h,
220  ii_w, ii_h - endy_safe);
221 }
222 
224 {
225  AVFilterContext *ctx = inlink->dst;
226  NLMeansContext *s = ctx->priv;
228  const int e = FFMAX(s->research_hsize, s->research_hsize_uv)
229  + FFMAX(s->patch_hsize, s->patch_hsize_uv);
230 
231  s->chroma_w = AV_CEIL_RSHIFT(inlink->w, desc->log2_chroma_w);
232  s->chroma_h = AV_CEIL_RSHIFT(inlink->h, desc->log2_chroma_h);
233  s->nb_planes = av_pix_fmt_count_planes(inlink->format);
234 
235  /* Allocate the integral image with extra edges of thickness "e"
236  *
237  * +_+-------------------------------+
238  * |0|0000000000000000000000000000000|
239  * +-x-------------------------------+
240  * |0|\ ^ |
241  * |0| ii | e |
242  * |0| v |
243  * |0| +-----------------------+ |
244  * |0| | | |
245  * |0|<->| | |
246  * |0| e | | |
247  * |0| | | |
248  * |0| +-----------------------+ |
249  * |0| |
250  * |0| |
251  * |0| |
252  * +-+-------------------------------+
253  */
254  s->ii_w = inlink->w + e*2;
255  s->ii_h = inlink->h + e*2;
256 
257  // align to 4 the linesize, "+1" is for the space of the left 0-column
258  s->ii_lz_32 = FFALIGN(s->ii_w + 1, 4);
259 
260  // "+1" is for the space of the top 0-line
261  s->ii_orig = av_calloc(s->ii_h + 1, s->ii_lz_32 * sizeof(*s->ii_orig));
262  if (!s->ii_orig)
263  return AVERROR(ENOMEM);
264 
265  // skip top 0-line and left 0-column
266  s->ii = s->ii_orig + s->ii_lz_32 + 1;
267 
268  // allocate weighted average for every pixel
269  s->linesize = inlink->w + 100;
270  s->total_weight = av_malloc_array(s->linesize, inlink->h * sizeof(*s->total_weight));
271  s->sum = av_malloc_array(s->linesize, inlink->h * sizeof(*s->sum));
272  if (!s->total_weight || !s->sum)
273  return AVERROR(ENOMEM);
274 
275  return 0;
276 }
277 
278 struct thread_data {
279  const uint8_t *src;
280  ptrdiff_t src_linesize;
282  int endx, endy;
283  const uint32_t *ii_start;
284  int p;
285 };
286 
287 static int nlmeans_slice(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs)
288 {
289  NLMeansContext *s = ctx->priv;
290  const uint32_t max_meaningful_diff = s->max_meaningful_diff;
291  const struct thread_data *td = arg;
292  const ptrdiff_t src_linesize = td->src_linesize;
293  const int process_h = td->endy - td->starty;
294  const int slice_start = (process_h * jobnr ) / nb_jobs;
295  const int slice_end = (process_h * (jobnr+1)) / nb_jobs;
296  const int starty = td->starty + slice_start;
297  const int endy = td->starty + slice_end;
298  const int p = td->p;
299  const uint32_t *ii = td->ii_start + (starty - p - 1) * s->ii_lz_32 - p - 1;
300  const int dist_b = 2*p + 1;
301  const int dist_d = dist_b * s->ii_lz_32;
302  const int dist_e = dist_d + dist_b;
303  const float *const weight_lut = s->weight_lut;
304  NLMeansDSPContext *dsp = &s->dsp;
305 
306  for (int y = starty; y < endy; y++) {
307  const uint8_t *const src = td->src + y*src_linesize;
308  float *total_weight = s->total_weight + y*s->linesize;
309  float *sum = s->sum + y*s->linesize;
310  const uint32_t *const iia = ii;
311  const uint32_t *const iib = ii + dist_b;
312  const uint32_t *const iid = ii + dist_d;
313  const uint32_t *const iie = ii + dist_e;
314 
315  dsp->compute_weights_line(iia, iib, iid, iie, src, total_weight, sum,
316  weight_lut, max_meaningful_diff,
317  td->startx, td->endx);
318  ii += s->ii_lz_32;
319  }
320  return 0;
321 }
322 
323 static void weight_averages(uint8_t *dst, ptrdiff_t dst_linesize,
324  const uint8_t *src, ptrdiff_t src_linesize,
325  float *total_weight, float *sum, ptrdiff_t linesize,
326  int w, int h)
327 {
328  for (int y = 0; y < h; y++) {
329  for (int x = 0; x < w; x++) {
330  // Also weight the centered pixel
331  total_weight[x] += 1.f;
332  sum[x] += 1.f * src[x];
333  dst[x] = av_clip_uint8(sum[x] / total_weight[x] + 0.5f);
334  }
335  dst += dst_linesize;
336  src += src_linesize;
337  total_weight += linesize;
338  sum += linesize;
339  }
340 }
341 
342 static int nlmeans_plane(AVFilterContext *ctx, int w, int h, int p, int r,
343  uint8_t *dst, ptrdiff_t dst_linesize,
344  const uint8_t *src, ptrdiff_t src_linesize)
345 {
346  NLMeansContext *s = ctx->priv;
347  /* patches center points cover the whole research window so the patches
348  * themselves overflow the research window */
349  const int e = r + p;
350  /* focus an integral pointer on the centered image (s1) */
351  const uint32_t *centered_ii = s->ii + e*s->ii_lz_32 + e;
352 
353  memset(s->total_weight, 0, s->linesize * h * sizeof(*s->total_weight));
354  memset(s->sum, 0, s->linesize * h * sizeof(*s->sum));
355 
356  for (int offy = -r; offy <= r; offy++) {
357  for (int offx = -r; offx <= r; offx++) {
358  if (offx || offy) {
359  struct thread_data td = {
360  .src = src + offy*src_linesize + offx,
361  .src_linesize = src_linesize,
362  .startx = FFMAX(0, -offx),
363  .starty = FFMAX(0, -offy),
364  .endx = FFMIN(w, w - offx),
365  .endy = FFMIN(h, h - offy),
366  .ii_start = centered_ii + offy*s->ii_lz_32 + offx,
367  .p = p,
368  };
369 
370  compute_ssd_integral_image(&s->dsp, s->ii, s->ii_lz_32,
371  src, src_linesize,
372  offx, offy, e, w, h);
374  FFMIN(td.endy - td.starty, ff_filter_get_nb_threads(ctx)));
375  }
376  }
377  }
378 
379  weight_averages(dst, dst_linesize, src, src_linesize,
380  s->total_weight, s->sum, s->linesize, w, h);
381 
382  return 0;
383 }
384 
386 {
387  AVFilterContext *ctx = inlink->dst;
388  NLMeansContext *s = ctx->priv;
389  AVFilterLink *outlink = ctx->outputs[0];
390 
391  AVFrame *out = ff_get_video_buffer(outlink, outlink->w, outlink->h);
392  if (!out) {
393  av_frame_free(&in);
394  return AVERROR(ENOMEM);
395  }
397 
398  for (int i = 0; i < s->nb_planes; i++) {
399  const int w = i ? s->chroma_w : inlink->w;
400  const int h = i ? s->chroma_h : inlink->h;
401  const int p = i ? s->patch_hsize_uv : s->patch_hsize;
402  const int r = i ? s->research_hsize_uv : s->research_hsize;
403  nlmeans_plane(ctx, w, h, p, r,
404  out->data[i], out->linesize[i],
405  in->data[i], in->linesize[i]);
406  }
407 
408  av_frame_free(&in);
409  return ff_filter_frame(outlink, out);
410 }
411 
412 #define CHECK_ODD_FIELD(field, name) do { \
413  if (!(s->field & 1)) { \
414  s->field |= 1; \
415  av_log(ctx, AV_LOG_WARNING, name " size must be odd, " \
416  "setting it to %d\n", s->field); \
417  } \
418 } while (0)
419 
421 {
422  NLMeansContext *s = ctx->priv;
423  const double h = s->sigma * 10.;
424 
425  s->pdiff_scale = 1. / (h * h);
426  s->max_meaningful_diff = log(255.) / s->pdiff_scale;
427  s->weight_lut = av_calloc(s->max_meaningful_diff + 1, sizeof(*s->weight_lut));
428  if (!s->weight_lut)
429  return AVERROR(ENOMEM);
430  for (int i = 0; i < s->max_meaningful_diff; i++)
431  s->weight_lut[i] = exp(-i * s->pdiff_scale);
432 
433  CHECK_ODD_FIELD(research_size, "Luma research window");
434  CHECK_ODD_FIELD(patch_size, "Luma patch");
435 
436  if (!s->research_size_uv) s->research_size_uv = s->research_size;
437  if (!s->patch_size_uv) s->patch_size_uv = s->patch_size;
438 
439  CHECK_ODD_FIELD(research_size_uv, "Chroma research window");
440  CHECK_ODD_FIELD(patch_size_uv, "Chroma patch");
441 
442  s->research_hsize = s->research_size / 2;
443  s->research_hsize_uv = s->research_size_uv / 2;
444  s->patch_hsize = s->patch_size / 2;
445  s->patch_hsize_uv = s->patch_size_uv / 2;
446 
447  av_log(ctx, AV_LOG_DEBUG, "Research window: %dx%d / %dx%d, patch size: %dx%d / %dx%d\n",
448  s->research_size, s->research_size, s->research_size_uv, s->research_size_uv,
449  s->patch_size, s->patch_size, s->patch_size_uv, s->patch_size_uv);
450 
451  ff_nlmeans_init(&s->dsp);
452 
453  return 0;
454 }
455 
457 {
458  NLMeansContext *s = ctx->priv;
459  av_freep(&s->weight_lut);
460  av_freep(&s->ii_orig);
461  av_freep(&s->total_weight);
462  av_freep(&s->sum);
463 }
464 
465 static const AVFilterPad nlmeans_inputs[] = {
466  {
467  .name = "default",
468  .type = AVMEDIA_TYPE_VIDEO,
469  .config_props = config_input,
470  .filter_frame = filter_frame,
471  },
472 };
473 
475  .name = "nlmeans",
476  .description = NULL_IF_CONFIG_SMALL("Non-local means denoiser."),
477  .priv_size = sizeof(NLMeansContext),
478  .init = init,
479  .uninit = uninit,
483  .priv_class = &nlmeans_class,
485 };
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:112
td
#define td
Definition: regdef.h:70
AVPixelFormat
AVPixelFormat
Pixel format.
Definition: pixfmt.h:71
pix_fmts
static enum AVPixelFormat pix_fmts[]
Definition: vf_nlmeans.c:76
av_clip
#define av_clip
Definition: common.h:98
NLMeansContext::weight_lut
float * weight_lut
Definition: vf_nlmeans.c:58
r
const char * r
Definition: vf_curves.c:126
acc
int acc
Definition: yuv2rgb.c:554
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
thread_data::src_linesize
ptrdiff_t src_linesize
Definition: vf_nlmeans.c:280
ff_filter_frame
int ff_filter_frame(AVFilterLink *link, AVFrame *frame)
Send a frame of data to the next filter.
Definition: avfilter.c:1018
av_pix_fmt_desc_get
const AVPixFmtDescriptor * av_pix_fmt_desc_get(enum AVPixelFormat pix_fmt)
Definition: pixdesc.c:2962
FILTER_PIXFMTS_ARRAY
#define FILTER_PIXFMTS_ARRAY(array)
Definition: internal.h:162
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
normalize.log
log
Definition: normalize.py:21
av_frame_free
void av_frame_free(AVFrame **frame)
Free the frame and any dynamically allocated objects in it, e.g.
Definition: frame.c:88
AVFrame
This structure describes decoded (raw) audio or video data.
Definition: frame.h:340
pixdesc.h
w
uint8_t w
Definition: llviddspenc.c:38
NLMeansContext::max_meaningful_diff
uint32_t max_meaningful_diff
Definition: vf_nlmeans.c:59
AVOption
AVOption.
Definition: opt.h:346
AV_PIX_FMT_YUV440P
@ AV_PIX_FMT_YUV440P
planar YUV 4:4:0 (1 Cr & Cb sample per 1x2 Y samples)
Definition: pixfmt.h:106
FFMAX
#define FFMAX(a, b)
Definition: macros.h:47
AVFilter::name
const char * name
Filter name.
Definition: avfilter.h:170
video.h
CHECK_ODD_FIELD
#define CHECK_ODD_FIELD(field, name)
Definition: vf_nlmeans.c:412
NLMeansContext::sum
float * sum
Definition: vf_nlmeans.c:56
AVFrame::data
uint8_t * data[AV_NUM_DATA_POINTERS]
pointer to the picture/channel planes.
Definition: frame.h:361
NLMeansContext::ii_w
int ii_w
Definition: vf_nlmeans.c:53
av_pix_fmt_count_planes
int av_pix_fmt_count_planes(enum AVPixelFormat pix_fmt)
Definition: pixdesc.c:3002
NLMeansDSPContext
Definition: vf_nlmeans.h:25
NLMeansContext::nb_planes
int nb_planes
Definition: vf_nlmeans.c:43
compute_ssd_integral_image
static void compute_ssd_integral_image(const NLMeansDSPContext *dsp, uint32_t *ii, ptrdiff_t ii_linesize_32, const uint8_t *src, ptrdiff_t linesize, int offx, int offy, int e, int w, int h)
Definition: vf_nlmeans.c:151
nlmeans_inputs
static const AVFilterPad nlmeans_inputs[]
Definition: vf_nlmeans.c:465
NLMeansDSPContext::compute_safe_ssd_integral_image
void(* compute_safe_ssd_integral_image)(uint32_t *dst, ptrdiff_t dst_linesize_32, const uint8_t *s1, ptrdiff_t linesize1, const uint8_t *s2, ptrdiff_t linesize2, int w, int h)
Definition: vf_nlmeans.h:26
AVFilterPad
A filter pad used for either input or output.
Definition: internal.h:33
thread_data::starty
int starty
Definition: vf_nlmeans.c:281
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:283
NLMeansContext::linesize
int linesize
Definition: vf_nlmeans.c:57
slice_start
static int slice_start(SliceContext *sc, VVCContext *s, VVCFrameContext *fc, const CodedBitstreamUnit *unit, const int is_first_slice)
Definition: vvcdec.c:685
avassert.h
compute_unsafe_ssd_integral_image
static void compute_unsafe_ssd_integral_image(uint32_t *dst, ptrdiff_t dst_linesize_32, int startx, int starty, const uint8_t *src, ptrdiff_t linesize, int offx, int offy, int r, int sw, int sh, int w, int h)
Compute squared difference of an unsafe area (the zone nor s1 nor s2 could be readable).
Definition: vf_nlmeans.c:111
av_cold
#define av_cold
Definition: attributes.h:90
NLMeansContext::dsp
NLMeansDSPContext dsp
Definition: vf_nlmeans.c:60
ff_video_default_filterpad
const AVFilterPad ff_video_default_filterpad[1]
An AVFilterPad array whose only entry has name "default" and is of type AVMEDIA_TYPE_VIDEO.
Definition: video.c:37
thread_data
Definition: vf_lut.c:338
NLMeansContext::research_hsize_uv
int research_hsize_uv
Definition: vf_nlmeans.c:50
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:86
NLMeansContext::chroma_h
int chroma_h
Definition: vf_nlmeans.c:44
s
#define s(width, name)
Definition: cbs_vp9.c:198
NLMeansContext::research_hsize
int research_hsize
Definition: vf_nlmeans.c:49
AV_CEIL_RSHIFT
#define AV_CEIL_RSHIFT(a, b)
Definition: common.h:58
AV_OPT_TYPE_DOUBLE
@ AV_OPT_TYPE_DOUBLE
Definition: opt.h:237
slice_end
static int slice_end(AVCodecContext *avctx, AVFrame *pict)
Handle slice ends.
Definition: mpeg12dec.c:1717
AV_LOG_DEBUG
#define AV_LOG_DEBUG
Stuff which is only useful for libav* developers.
Definition: log.h:201
ctx
AVFormatContext * ctx
Definition: movenc.c:48
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:73
thread_data::src
const uint8_t * src
Definition: vf_nlmeans.c:279
FILTER_INPUTS
#define FILTER_INPUTS(array)
Definition: internal.h:182
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:87
arg
const char * arg
Definition: jacosubdec.c:67
thread_data::endx
int endx
Definition: vf_nlmeans.c:282
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:637
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:85
ff_vf_nlmeans
const AVFilter ff_vf_nlmeans
Definition: vf_nlmeans.c:474
weight_averages
static void weight_averages(uint8_t *dst, ptrdiff_t dst_linesize, const uint8_t *src, ptrdiff_t src_linesize, float *total_weight, float *sum, ptrdiff_t linesize, int w, int h)
Definition: vf_nlmeans.c:323
thread_data::startx
int startx
Definition: vf_nlmeans.c:281
NLMeansContext
Definition: vf_nlmeans.c:41
NLMeansContext::ii_lz_32
ptrdiff_t ii_lz_32
Definition: vf_nlmeans.c:54
AV_PIX_FMT_GRAY8
@ AV_PIX_FMT_GRAY8
Y , 8bpp.
Definition: pixfmt.h:81
FLAGS
#define FLAGS
Definition: vf_nlmeans.c:64
exp
int8_t exp
Definition: eval.c:74
NLMeansContext::total_weight
float * total_weight
Definition: vf_nlmeans.c:55
f
f
Definition: af_crystalizer.c:121
uninit
static av_cold void uninit(AVFilterContext *ctx)
Definition: vf_nlmeans.c:456
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
vf_nlmeans_init.h
vf_nlmeans.h
for
for(k=2;k<=8;++k)
Definition: h264pred_template.c:425
nlmeans_slice
static int nlmeans_slice(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs)
Definition: vf_nlmeans.c:287
NLMeansContext::pdiff_scale
double pdiff_scale
Definition: vf_nlmeans.c:45
NLMeansContext::ii_orig
uint32_t * ii_orig
Definition: vf_nlmeans.c:51
config_input
static int config_input(AVFilterLink *inlink)
Definition: vf_nlmeans.c:223
thread_data::endy
int endy
Definition: vf_nlmeans.c:282
internal.h
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:147
NLMeansContext::ii_h
int ii_h
Definition: vf_nlmeans.c:53
NLMeansContext::patch_size
int patch_size
Definition: vf_nlmeans.c:47
NLMeansContext::ii
uint32_t * ii
Definition: vf_nlmeans.c:52
i
#define i(width, name, range_min, range_max)
Definition: cbs_h2645.c:255
av_malloc_array
#define av_malloc_array(a, b)
Definition: tableprint_vlc.h:31
ff_filter_get_nb_threads
int ff_filter_get_nb_threads(AVFilterContext *ctx)
Get number of threads for current filter instance.
Definition: avfilter.c:825
av_assert1
#define av_assert1(cond)
assert() equivalent, that does not lie in speed critical code.
Definition: avassert.h:56
xf
#define xf(width, name, var, range_min, range_max, subs,...)
Definition: cbs_av1.c:590
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:107
AVFilterPad::name
const char * name
Pad name.
Definition: internal.h:39
av_calloc
void * av_calloc(size_t nmemb, size_t size)
Definition: mem.c:262
NLMeansContext::chroma_w
int chroma_w
Definition: vf_nlmeans.c:44
ff_nlmeans_init
static av_unused void ff_nlmeans_init(NLMeansDSPContext *dsp)
Definition: vf_nlmeans_init.h:127
AVFilter
Filter definition.
Definition: avfilter.h:166
NLMeansContext::sigma
double sigma
Definition: vf_nlmeans.c:46
init
static av_cold int init(AVFilterContext *ctx)
Definition: vf_nlmeans.c:420
AV_PIX_FMT_NONE
@ AV_PIX_FMT_NONE
Definition: pixfmt.h:72
AV_OPT_TYPE_INT
@ AV_OPT_TYPE_INT
Definition: opt.h:235
avfilter.h
NLMeansContext::patch_hsize
int patch_hsize
Definition: vf_nlmeans.c:47
NLMeansContext::research_size
int research_size
Definition: vf_nlmeans.c:49
NLMeansContext::patch_hsize_uv
int patch_hsize_uv
Definition: vf_nlmeans.c:48
av_clip_uint8
#define av_clip_uint8
Definition: common.h:104
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:78
AVFilterContext
An instance of a filter.
Definition: avfilter.h:407
AV_PIX_FMT_GBRP
@ AV_PIX_FMT_GBRP
planar GBR 4:4:4 24bpp
Definition: pixfmt.h:165
AVFILTER_DEFINE_CLASS
AVFILTER_DEFINE_CLASS(nlmeans)
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:73
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:77
NLMeansContext::research_size_uv
int research_size_uv
Definition: vf_nlmeans.c:50
filter_frame
static int filter_frame(AVFilterLink *inlink, AVFrame *in)
Definition: vf_nlmeans.c:385
AVPixFmtDescriptor
Descriptor that unambiguously describes how the bits of a pixel are stored in the up to 4 data planes...
Definition: pixdesc.h:69
nlmeans_plane
static int nlmeans_plane(AVFilterContext *ctx, int w, int h, int p, int r, uint8_t *dst, ptrdiff_t dst_linesize, const uint8_t *src, ptrdiff_t src_linesize)
Definition: vf_nlmeans.c:342
FFALIGN
#define FFALIGN(x, a)
Definition: macros.h:78
FILTER_OUTPUTS
#define FILTER_OUTPUTS(array)
Definition: internal.h:183
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:80
thread_data::in
AVFrame * in
Definition: vf_lut.c:339
d
d
Definition: ffmpeg_filter.c:425
nlmeans_options
static const AVOption nlmeans_options[]
Definition: vf_nlmeans.c:65
AVFrame::linesize
int linesize[AV_NUM_DATA_POINTERS]
For video, a positive or negative value, which is typically indicating the size in bytes of each pict...
Definition: frame.h:385
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:79
av_log
#define av_log(a,...)
Definition: tableprint_vlc.h:27
h
h
Definition: vp9dsp_template.c:2038
OFFSET
#define OFFSET(x)
Definition: vf_nlmeans.c:63
thread_data::p
int p
Definition: vf_nlmeans.c:284
thread_data::ii_start
const uint32_t * ii_start
Definition: vf_nlmeans.c:283
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:134
NLMeansContext::patch_size_uv
int patch_size_uv
Definition: vf_nlmeans.c:48