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