FFmpeg
dsp_init.c
Go to the documentation of this file.
1 /*
2  * VVC filters DSP
3  *
4  * Copyright (C) 2024 Zhao Zhili
5  *
6  * This file is part of FFmpeg.
7  *
8  * FFmpeg is free software; you can redistribute it and/or
9  * modify it under the terms of the GNU Lesser General Public
10  * License as published by the Free Software Foundation; either
11  * version 2.1 of the License, or (at your option) any later version.
12  *
13  * FFmpeg is distributed in the hope that it will be useful,
14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
16  * Lesser General Public License for more details.
17  *
18  * You should have received a copy of the GNU Lesser General Public
19  * License along with FFmpeg; if not, write to the Free Software
20  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
21  */
22 
23 #include "libavutil/cpu.h"
24 #include "libavutil/aarch64/cpu.h"
25 #include "libavcodec/aarch64/h26x/dsp.h"
26 #include "libavcodec/vvc/dsp.h"
27 #include "libavcodec/vvc/dec.h"
28 #include "libavcodec/vvc/ctu.h"
29 
30 #define BIT_DEPTH 8
31 #include "alf_template.c"
32 #undef BIT_DEPTH
33 
34 #define BIT_DEPTH 10
35 #include "alf_template.c"
36 #undef BIT_DEPTH
37 
38 #define BIT_DEPTH 12
39 #include "alf_template.c"
40 #undef BIT_DEPTH
41 
42 int ff_vvc_sad_neon(const int16_t *src0, const int16_t *src1, int dx, int dy,
43  const int block_w, const int block_h);
44 
45 void ff_vvc_avg_8_neon(uint8_t *dst, ptrdiff_t dst_stride,
46  const int16_t *src0, const int16_t *src1, int width,
47  int height);
48 void ff_vvc_avg_10_neon(uint8_t *dst, ptrdiff_t dst_stride,
49  const int16_t *src0, const int16_t *src1, int width,
50  int height);
51 void ff_vvc_avg_12_neon(uint8_t *dst, ptrdiff_t dst_stride,
52  const int16_t *src0, const int16_t *src1, int width,
53  int height);
54 
55 void ff_vvc_w_avg_8_neon(uint8_t *_dst, ptrdiff_t _dst_stride,
56  const int16_t *src0, const int16_t *src1,
57  int width, int height,
58  uintptr_t w0_w1, uintptr_t offset_shift);
59 void ff_vvc_w_avg_10_neon(uint8_t *_dst, ptrdiff_t _dst_stride,
60  const int16_t *src0, const int16_t *src1,
61  int width, int height,
62  uintptr_t w0_w1, uintptr_t offset_shift);
63 void ff_vvc_w_avg_12_neon(uint8_t *_dst, ptrdiff_t _dst_stride,
64  const int16_t *src0, const int16_t *src1,
65  int width, int height,
66  uintptr_t w0_w1, uintptr_t offset_shift);
67 /* When passing arguments to functions, Apple platforms diverge from the ARM64
68  * standard ABI for functions that require passing arguments on the stack. To
69  * simplify portability in the assembly function interface, use a different
70  * function signature that doesn't require passing arguments on the stack.
71  */
72 #define W_AVG_FUN(bit_depth) \
73 static void vvc_w_avg_ ## bit_depth(uint8_t *dst, ptrdiff_t dst_stride, \
74  const int16_t *src0, const int16_t *src1, int width, int height, \
75  int denom, int w0, int w1, int o0, int o1) \
76 { \
77  int shift = denom + FFMAX(3, 15 - bit_depth); \
78  int offset = ((o0 + o1) * (1 << (bit_depth - 8)) + 1) * (1 << (shift - 1)); \
79  uintptr_t w0_w1 = ((uintptr_t)w0 << 32) | (uint32_t)w1; \
80  uintptr_t offset_shift = ((uintptr_t)offset << 32) | (uint32_t)shift; \
81  ff_vvc_w_avg_ ## bit_depth ## _neon(dst, dst_stride, src0, src1, width, height, w0_w1, offset_shift); \
82 }
83 
84 W_AVG_FUN(8)
85 W_AVG_FUN(10)
86 W_AVG_FUN(12)
87 
88 #define DMVR_FUN(fn, bd) \
89  void ff_vvc_dmvr_ ## fn ## bd ## _neon(int16_t *dst, \
90  const uint8_t *_src, ptrdiff_t _src_stride, int height, \
91  intptr_t mx, intptr_t my, int width);
92 
93 DMVR_FUN(, 8)
94 DMVR_FUN(, 12)
95 DMVR_FUN(hv_, 8)
96 DMVR_FUN(hv_, 10)
97 DMVR_FUN(hv_, 12)
98 
99 void ff_vvc_dsp_init_aarch64(VVCDSPContext *const c, const int bd)
100 {
101  int cpu_flags = av_get_cpu_flags();
102  if (!have_neon(cpu_flags))
103  return;
104 
105  if (bd == 8) {
106  c->inter.put[0][1][0][0] = ff_vvc_put_pel_pixels4_8_neon;
107  c->inter.put[0][2][0][0] = ff_vvc_put_pel_pixels8_8_neon;
108  c->inter.put[0][3][0][0] = ff_vvc_put_pel_pixels16_8_neon;
109  c->inter.put[0][4][0][0] = ff_vvc_put_pel_pixels32_8_neon;
110  c->inter.put[0][5][0][0] = ff_vvc_put_pel_pixels64_8_neon;
111  c->inter.put[0][6][0][0] = ff_vvc_put_pel_pixels128_8_neon;
112 
113  c->inter.put[0][1][0][1] = ff_vvc_put_qpel_h4_8_neon;
114  c->inter.put[0][2][0][1] = ff_vvc_put_qpel_h8_8_neon;
115  c->inter.put[0][3][0][1] = ff_vvc_put_qpel_h16_8_neon;
116  c->inter.put[0][4][0][1] =
117  c->inter.put[0][5][0][1] =
118  c->inter.put[0][6][0][1] = ff_vvc_put_qpel_h32_8_neon;
119 
120  c->inter.put[0][1][1][0] = ff_vvc_put_qpel_v4_8_neon;
121  c->inter.put[0][2][1][0] =
122  c->inter.put[0][3][1][0] =
123  c->inter.put[0][4][1][0] =
124  c->inter.put[0][5][1][0] =
125  c->inter.put[0][6][1][0] = ff_vvc_put_qpel_v8_8_neon;
126 
127  c->inter.put[0][1][1][1] = ff_vvc_put_qpel_hv4_8_neon;
128  c->inter.put[0][2][1][1] = ff_vvc_put_qpel_hv8_8_neon;
129  c->inter.put[0][3][1][1] = ff_vvc_put_qpel_hv16_8_neon;
130  c->inter.put[0][4][1][1] = ff_vvc_put_qpel_hv32_8_neon;
131  c->inter.put[0][5][1][1] = ff_vvc_put_qpel_hv64_8_neon;
132  c->inter.put[0][6][1][1] = ff_vvc_put_qpel_hv128_8_neon;
133 
134  c->inter.put[1][1][0][1] = ff_vvc_put_epel_h4_8_neon;
135  c->inter.put[1][2][0][1] = ff_vvc_put_epel_h8_8_neon;
136  c->inter.put[1][3][0][1] = ff_vvc_put_epel_h16_8_neon;
137  c->inter.put[1][4][0][1] =
138  c->inter.put[1][5][0][1] =
139  c->inter.put[1][6][0][1] = ff_vvc_put_epel_h32_8_neon;
140 
141  c->inter.put[1][1][1][1] = ff_vvc_put_epel_hv4_8_neon;
142  c->inter.put[1][2][1][1] = ff_vvc_put_epel_hv8_8_neon;
143  c->inter.put[1][3][1][1] = ff_vvc_put_epel_hv16_8_neon;
144  c->inter.put[1][4][1][1] = ff_vvc_put_epel_hv32_8_neon;
145  c->inter.put[1][5][1][1] = ff_vvc_put_epel_hv64_8_neon;
146  c->inter.put[1][6][1][1] = ff_vvc_put_epel_hv128_8_neon;
147 
148  c->inter.put_uni[0][1][0][0] = ff_vvc_put_pel_uni_pixels4_8_neon;
149  c->inter.put_uni[0][2][0][0] = ff_vvc_put_pel_uni_pixels8_8_neon;
150  c->inter.put_uni[0][3][0][0] = ff_vvc_put_pel_uni_pixels16_8_neon;
151  c->inter.put_uni[0][4][0][0] = ff_vvc_put_pel_uni_pixels32_8_neon;
152  c->inter.put_uni[0][5][0][0] = ff_vvc_put_pel_uni_pixels64_8_neon;
153  c->inter.put_uni[0][6][0][0] = ff_vvc_put_pel_uni_pixels128_8_neon;
154 
155  c->inter.put_uni[0][1][0][1] = ff_vvc_put_qpel_uni_h4_8_neon;
156  c->inter.put_uni[0][2][0][1] = ff_vvc_put_qpel_uni_h8_8_neon;
157  c->inter.put_uni[0][3][0][1] = ff_vvc_put_qpel_uni_h16_8_neon;
158  c->inter.put_uni[0][4][0][1] =
159  c->inter.put_uni[0][5][0][1] =
160  c->inter.put_uni[0][6][0][1] = ff_vvc_put_qpel_uni_h32_8_neon;
161 
162  c->inter.put_uni_w[0][1][0][0] = ff_vvc_put_pel_uni_w_pixels4_8_neon;
163  c->inter.put_uni_w[0][2][0][0] = ff_vvc_put_pel_uni_w_pixels8_8_neon;
164  c->inter.put_uni_w[0][3][0][0] = ff_vvc_put_pel_uni_w_pixels16_8_neon;
165  c->inter.put_uni_w[0][4][0][0] = ff_vvc_put_pel_uni_w_pixels32_8_neon;
166  c->inter.put_uni_w[0][5][0][0] = ff_vvc_put_pel_uni_w_pixels64_8_neon;
167  c->inter.put_uni_w[0][6][0][0] = ff_vvc_put_pel_uni_w_pixels128_8_neon;
168 
169  c->inter.avg = ff_vvc_avg_8_neon;
170  c->inter.w_avg = vvc_w_avg_8;
171  c->inter.dmvr[0][0] = ff_vvc_dmvr_8_neon;
172  c->inter.dmvr[1][1] = ff_vvc_dmvr_hv_8_neon;
173 
174  for (int i = 0; i < FF_ARRAY_ELEMS(c->sao.band_filter); i++)
175  c->sao.band_filter[i] = ff_h26x_sao_band_filter_8x8_8_neon;
176  c->sao.edge_filter[0] = ff_vvc_sao_edge_filter_8x8_8_neon;
177  for (int i = 1; i < FF_ARRAY_ELEMS(c->sao.edge_filter); i++)
178  c->sao.edge_filter[i] = ff_vvc_sao_edge_filter_16x16_8_neon;
179  c->alf.filter[LUMA] = alf_filter_luma_8_neon;
180  c->alf.filter[CHROMA] = alf_filter_chroma_8_neon;
181 
182  if (have_i8mm(cpu_flags)) {
183  c->inter.put[0][1][0][1] = ff_vvc_put_qpel_h4_8_neon_i8mm;
184  c->inter.put[0][2][0][1] = ff_vvc_put_qpel_h8_8_neon_i8mm;
185  c->inter.put[0][3][0][1] = ff_vvc_put_qpel_h16_8_neon_i8mm;
186  c->inter.put[0][4][0][1] = ff_vvc_put_qpel_h32_8_neon_i8mm;
187  c->inter.put[0][5][0][1] = ff_vvc_put_qpel_h64_8_neon_i8mm;
188  c->inter.put[0][6][0][1] = ff_vvc_put_qpel_h128_8_neon_i8mm;
189 
190  c->inter.put[0][1][1][1] = ff_vvc_put_qpel_hv4_8_neon_i8mm;
191  c->inter.put[0][2][1][1] = ff_vvc_put_qpel_hv8_8_neon_i8mm;
192  c->inter.put[0][3][1][1] = ff_vvc_put_qpel_hv16_8_neon_i8mm;
193  c->inter.put[0][4][1][1] = ff_vvc_put_qpel_hv32_8_neon_i8mm;
194  c->inter.put[0][5][1][1] = ff_vvc_put_qpel_hv64_8_neon_i8mm;
195  c->inter.put[0][6][1][1] = ff_vvc_put_qpel_hv128_8_neon_i8mm;
196 
197  c->inter.put[1][1][0][1] = ff_vvc_put_epel_h4_8_neon_i8mm;
198  c->inter.put[1][2][0][1] = ff_vvc_put_epel_h8_8_neon_i8mm;
199  c->inter.put[1][3][0][1] = ff_vvc_put_epel_h16_8_neon_i8mm;
200  c->inter.put[1][4][0][1] = ff_vvc_put_epel_h32_8_neon_i8mm;
201  c->inter.put[1][5][0][1] = ff_vvc_put_epel_h64_8_neon_i8mm;
202  c->inter.put[1][6][0][1] = ff_vvc_put_epel_h128_8_neon_i8mm;
203 
204  c->inter.put[1][1][1][1] = ff_vvc_put_epel_hv4_8_neon_i8mm;
205  c->inter.put[1][2][1][1] = ff_vvc_put_epel_hv8_8_neon_i8mm;
206  c->inter.put[1][3][1][1] = ff_vvc_put_epel_hv16_8_neon_i8mm;
207  c->inter.put[1][4][1][1] = ff_vvc_put_epel_hv32_8_neon_i8mm;
208  c->inter.put[1][5][1][1] = ff_vvc_put_epel_hv64_8_neon_i8mm;
209  c->inter.put[1][6][1][1] = ff_vvc_put_epel_hv128_8_neon_i8mm;
210  }
211  } else if (bd == 10) {
212  c->inter.avg = ff_vvc_avg_10_neon;
213  c->inter.w_avg = vvc_w_avg_10;
214  c->inter.dmvr[1][1] = ff_vvc_dmvr_hv_10_neon;
215 
216  c->alf.filter[LUMA] = alf_filter_luma_10_neon;
217  c->alf.filter[CHROMA] = alf_filter_chroma_10_neon;
218  } else if (bd == 12) {
219  c->inter.avg = ff_vvc_avg_12_neon;
220  c->inter.w_avg = vvc_w_avg_12;
221  c->inter.dmvr[0][0] = ff_vvc_dmvr_12_neon;
222  c->inter.dmvr[1][1] = ff_vvc_dmvr_hv_12_neon;
223 
224  c->alf.filter[LUMA] = alf_filter_luma_12_neon;
225  c->alf.filter[CHROMA] = alf_filter_chroma_12_neon;
226  }
227 
228  c->inter.sad = ff_vvc_sad_neon;
229 }
_dst
uint8_t * _dst
Definition: dsp.h:52
LUMA
#define LUMA
Definition: filter.c:31
src1
const pixel * src1
Definition: h264pred_template.c:421
ff_vvc_put_qpel_v8_8_neon
void ff_vvc_put_qpel_v8_8_neon(int16_t *dst, const uint8_t *_src, ptrdiff_t _srcstride, int height, const int8_t *hf, const int8_t *vf, int width)
av_get_cpu_flags
int av_get_cpu_flags(void)
Return the flags which specify extensions supported by the CPU.
Definition: cpu.c:107
cpu_flags
static atomic_int cpu_flags
Definition: cpu.c:56
ff_vvc_sad_neon
int ff_vvc_sad_neon(const int16_t *src0, const int16_t *src1, int dx, int dy, const int block_w, const int block_h)
dsp.h
ff_vvc_put_qpel_v4_8_neon
void ff_vvc_put_qpel_v4_8_neon(int16_t *dst, const uint8_t *_src, ptrdiff_t _srcstride, int height, const int8_t *hf, const int8_t *vf, int width)
FF_ARRAY_ELEMS
#define FF_ARRAY_ELEMS(a)
Definition: sinewin_tablegen.c:29
dsp.h
ff_vvc_avg_12_neon
void ff_vvc_avg_12_neon(uint8_t *dst, ptrdiff_t dst_stride, const int16_t *src0, const int16_t *src1, int width, int height)
ff_vvc_avg_8_neon
void ff_vvc_avg_8_neon(uint8_t *dst, ptrdiff_t dst_stride, const int16_t *src0, const int16_t *src1, int width, int height)
if
if(ret)
Definition: filter_design.txt:179
W_AVG_FUN
#define W_AVG_FUN(bit_depth)
Definition: dsp_init.c:72
ff_vvc_sao_edge_filter_8x8_8_neon
void ff_vvc_sao_edge_filter_8x8_8_neon(uint8_t *dst, const uint8_t *src, ptrdiff_t stride_dst, const int16_t *sao_offset_val, int eo, int width, int height)
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
have_i8mm
#define have_i8mm(flags)
Definition: cpu.h:29
height
#define height
Definition: dsp.h:85
dst
uint8_t ptrdiff_t const uint8_t ptrdiff_t int intptr_t intptr_t int int16_t * dst
Definition: dsp.h:83
cpu.h
for
for(k=2;k<=8;++k)
Definition: h264pred_template.c:425
ff_vvc_w_avg_12_neon
void ff_vvc_w_avg_12_neon(uint8_t *_dst, ptrdiff_t _dst_stride, const int16_t *src0, const int16_t *src1, int width, int height, uintptr_t w0_w1, uintptr_t offset_shift)
DMVR_FUN
#define DMVR_FUN(fn, bd)
Definition: dsp_init.c:88
have_neon
#define have_neon(flags)
Definition: cpu.h:26
ff_h26x_sao_band_filter_8x8_8_neon
void ff_h26x_sao_band_filter_8x8_8_neon(uint8_t *_dst, const uint8_t *_src, ptrdiff_t stride_dst, ptrdiff_t stride_src, const int16_t *sao_offset_val, int sao_left_class, int width, int height)
i
#define i(width, name, range_min, range_max)
Definition: cbs_h2645.c:256
ff_vvc_avg_10_neon
void ff_vvc_avg_10_neon(uint8_t *dst, ptrdiff_t dst_stride, const int16_t *src0, const int16_t *src1, int width, int height)
CHROMA
@ CHROMA
Definition: vf_waveform.c:49
ff_vvc_w_avg_10_neon
void ff_vvc_w_avg_10_neon(uint8_t *_dst, ptrdiff_t _dst_stride, const int16_t *src0, const int16_t *src1, int width, int height, uintptr_t w0_w1, uintptr_t offset_shift)
ff_vvc_dsp_init_aarch64
void ff_vvc_dsp_init_aarch64(VVCDSPContext *const c, const int bd)
Definition: dsp_init.c:99
src0
const pixel *const src0
Definition: h264pred_template.c:420
ff_vvc_sao_edge_filter_16x16_8_neon
void ff_vvc_sao_edge_filter_16x16_8_neon(uint8_t *dst, const uint8_t *src, ptrdiff_t stride_dst, const int16_t *sao_offset_val, int eo, int width, int height)
alf_template.c
ff_vvc_w_avg_8_neon
void ff_vvc_w_avg_8_neon(uint8_t *_dst, ptrdiff_t _dst_stride, const int16_t *src0, const int16_t *src1, int width, int height, uintptr_t w0_w1, uintptr_t offset_shift)
ctu.h
width
#define width
Definition: dsp.h:85
cpu.h
dec.h
VVCDSPContext
Definition: dsp.h:169
Generated on Fri Nov 15 2024 19:21:28 for FFmpeg by   doxygen 1.8.17