FFmpeg
vf_nlmeans.c
Go to the documentation of this file.
1 /*
2  * Copyright (c) 2018 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 #include "checkasm.h"
22 #include "libavfilter/vf_nlmeans.h"
23 #include "libavutil/avassert.h"
24 
25 #define randomize_buffer(buf, size) do { \
26  int i; \
27  for (i = 0; i < size / 4; i++) \
28  ((uint32_t *)buf)[i] = rnd(); \
29 } while (0)
30 
32 {
33  NLMeansDSPContext dsp = {0};
34 
35  const int w = 123; // source width
36  const int h = 45; // source height
37  const int p = 3; // patch half size
38  const int r = 2; // research window half size
39 
40  ff_nlmeans_init(&dsp);
41 
42  /* See the filter's code for the explanations on the variables */
43  if (check_func(dsp.compute_safe_ssd_integral_image, "ssd_integral_image")) {
44  int offx, offy;
45  const int e = p + r;
46  const int ii_w = w + e*2;
47  const int ii_h = h + e*2;
48  const int ii_lz_32 = FFALIGN(ii_w + 1, 4);
49  uint32_t *ii_orig_ref = av_mallocz_array(ii_h + 1, ii_lz_32 * sizeof(*ii_orig_ref));
50  uint32_t *ii_ref = ii_orig_ref + ii_lz_32 + 1;
51  uint32_t *ii_orig_new = av_mallocz_array(ii_h + 1, ii_lz_32 * sizeof(*ii_orig_new));
52  uint32_t *ii_new = ii_orig_new + ii_lz_32 + 1;
53  const int src_lz = FFALIGN(w, 16);
54  uint8_t *src = av_mallocz_array(h, src_lz);
55 
56  declare_func(void, uint32_t *dst, ptrdiff_t dst_linesize_32,
57  const uint8_t *s1, ptrdiff_t linesize1,
58  const uint8_t *s2, ptrdiff_t linesize2,
59  int w, int h);
60 
61  randomize_buffer(src, h * src_lz);
62 
63  for (offy = -r; offy <= r; offy++) {
64  for (offx = -r; offx <= r; offx++) {
65  if (offx || offy) {
66  const int s1x = e;
67  const int s1y = e;
68  const int s2x = e + offx;
69  const int s2y = e + offy;
70  const int startx_safe = FFMAX(s1x, s2x);
71  const int starty_safe = FFMAX(s1y, s2y);
72  const int u_endx_safe = FFMIN(s1x + w, s2x + w);
73  const int endy_safe = FFMIN(s1y + h, s2y + h);
74  const int safe_pw = (u_endx_safe - startx_safe) & ~0xf;
75  const int safe_ph = endy_safe - starty_safe;
76 
77  av_assert0(safe_pw && safe_ph);
78  av_assert0(startx_safe - s1x >= 0); av_assert0(startx_safe - s1x < w);
79  av_assert0(starty_safe - s1y >= 0); av_assert0(starty_safe - s1y < h);
80  av_assert0(startx_safe - s2x >= 0); av_assert0(startx_safe - s2x < w);
81  av_assert0(starty_safe - s2y >= 0); av_assert0(starty_safe - s2y < h);
82 
83  memset(ii_ref, 0, (ii_lz_32 * ii_h - 1) * sizeof(*ii_ref));
84  memset(ii_new, 0, (ii_lz_32 * ii_h - 1) * sizeof(*ii_new));
85 
86  call_ref(ii_ref + starty_safe*ii_lz_32 + startx_safe, ii_lz_32,
87  src + (starty_safe - s1y) * src_lz + (startx_safe - s1x), src_lz,
88  src + (starty_safe - s2y) * src_lz + (startx_safe - s2x), src_lz,
89  safe_pw, safe_ph);
90  call_new(ii_new + starty_safe*ii_lz_32 + startx_safe, ii_lz_32,
91  src + (starty_safe - s1y) * src_lz + (startx_safe - s1x), src_lz,
92  src + (starty_safe - s2y) * src_lz + (startx_safe - s2x), src_lz,
93  safe_pw, safe_ph);
94 
95  if (memcmp(ii_ref, ii_new, (ii_lz_32 * ii_h - 1) * sizeof(*ii_ref)))
96  fail();
97 
98  memset(ii_new, 0, (ii_lz_32 * ii_h - 1) * sizeof(*ii_new));
99  bench_new(ii_new + starty_safe*ii_lz_32 + startx_safe, ii_lz_32,
100  src + (starty_safe - s1y) * src_lz + (startx_safe - s1x), src_lz,
101  src + (starty_safe - s2y) * src_lz + (startx_safe - s2x), src_lz,
102  safe_pw, safe_ph);
103  }
104  }
105  }
106 
107  av_freep(&ii_orig_ref);
108  av_freep(&ii_orig_new);
109  av_freep(&src);
110  }
111 
112  report("dsp");
113 }
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
#define randomize_buffer(buf, size)
Definition: vf_nlmeans.c:25
#define src
Definition: vp8dsp.c:254
#define report
Definition: checkasm.h:125
#define av_assert0(cond)
assert() equivalent, that is always enabled.
Definition: avassert.h:37
void checkasm_check_nlmeans(void)
Definition: vf_nlmeans.c:31
uint8_t
#define FFALIGN(x, a)
Definition: macros.h:48
#define s2
Definition: regdef.h:39
#define declare_func(ret,...)
Definition: checkasm.h:117
const char * r
Definition: vf_curves.c:114
simple assert() macros that are a bit more flexible than ISO C assert().
#define FFMAX(a, b)
Definition: common.h:94
#define fail()
Definition: checkasm.h:122
#define FFMIN(a, b)
Definition: common.h:96
uint8_t w
Definition: llviddspenc.c:38
#define call_ref(...)
Definition: checkasm.h:128
#define check_func(func,...)
Definition: checkasm.h:113
#define s1
Definition: regdef.h:38
#define xf(width, name, var, range_min, range_max, subs,...)
Definition: cbs_av1.c:664
#define bench_new(...)
Definition: checkasm.h:255
#define av_freep(p)
#define call_new(...)
Definition: checkasm.h:195
void ff_nlmeans_init(NLMeansDSPContext *dsp)
Definition: vf_nlmeans.c:509
void * av_mallocz_array(size_t nmemb, size_t size)
Definition: mem.c:191