lavfi/nlmeans: add AArch64 SIMD for compute_safe_ssd_integral_image
[ffmpeg.git] / libavfilter / aarch64 / vf_nlmeans_init.c
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 "libavutil/aarch64/cpu.h"
20 #include "libavfilter/vf_nlmeans.h"
21
22 void ff_compute_safe_ssd_integral_image_neon(uint32_t *dst, ptrdiff_t dst_linesize_32,
23                                              const uint8_t *s1, ptrdiff_t linesize1,
24                                              const uint8_t *s2, ptrdiff_t linesize2,
25                                              int w, int h);
26
27 av_cold void ff_nlmeans_init_aarch64(NLMeansDSPContext *dsp)
28 {
29     int cpu_flags = av_get_cpu_flags();
30
31     if (have_neon(cpu_flags))
32         dsp->compute_safe_ssd_integral_image = ff_compute_safe_ssd_integral_image_neon;
33 }