1 /*
2 * Copyright © 2023, VideoLAN and dav1d authors
3 * Copyright © 2023, Loongson Technology Corporation Limited
4 * All rights reserved.
5 *
6 * Redistribution and use in source and binary forms, with or without
7 * modification, are permitted provided that the following conditions are met:
8 *
9 * 1. Redistributions of source code must retain the above copyright notice, this
10 * list of conditions and the following disclaimer.
11 *
12 * 2. Redistributions in binary form must reproduce the above copyright notice,
13 * this list of conditions and the following disclaimer in the documentation
14 * and/or other materials provided with the distribution.
15 *
16 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
17 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
18 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
19 * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR
20 * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
21 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
22 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
23 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
24 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
25 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
26 */
27
28 #ifndef DAV1D_SRC_LOONGARCH_MC_H
29 #define DAV1D_SRC_LOONGARCH_MC_H
30
31 #include "config.h"
32 #include "src/mc.h"
33 #include "src/cpu.h"
34
35 #define init_mc_fn(type, name, suffix) \
36 c->mc[type] = BF(dav1d_put_##name, suffix)
37 #define init_mct_fn(type, name, suffix) \
38 c->mct[type] = BF(dav1d_prep_##name, suffix)
39
40 decl_avg_fn(BF(dav1d_avg, lsx));
41 decl_w_avg_fn(BF(dav1d_w_avg, lsx));
42 decl_mask_fn(BF(dav1d_mask, lsx));
43 decl_warp8x8_fn(BF(dav1d_warp_affine_8x8, lsx));
44 decl_warp8x8t_fn(BF(dav1d_warp_affine_8x8t, lsx));
45 decl_w_mask_fn(BF(dav1d_w_mask_420, lsx));
46
47 decl_mc_fn(BF(dav1d_put_8tap_regular, lsx));
48 decl_mc_fn(BF(dav1d_put_8tap_regular_smooth, lsx));
49 decl_mc_fn(BF(dav1d_put_8tap_regular_sharp, lsx));
50 decl_mc_fn(BF(dav1d_put_8tap_smooth, lsx));
51 decl_mc_fn(BF(dav1d_put_8tap_smooth_regular, lsx));
52 decl_mc_fn(BF(dav1d_put_8tap_smooth_sharp, lsx));
53 decl_mc_fn(BF(dav1d_put_8tap_sharp, lsx));
54 decl_mc_fn(BF(dav1d_put_8tap_sharp_regular, lsx));
55 decl_mc_fn(BF(dav1d_put_8tap_sharp_smooth, lsx));
56
57 decl_avg_fn(BF(dav1d_avg, lasx));
58 decl_w_avg_fn(BF(dav1d_w_avg, lasx));
59 decl_mask_fn(BF(dav1d_mask, lasx));
60 decl_warp8x8_fn(BF(dav1d_warp_affine_8x8, lasx));
61 decl_warp8x8t_fn(BF(dav1d_warp_affine_8x8t, lasx));
62 decl_w_mask_fn(BF(dav1d_w_mask_420, lasx));
63
64 decl_mct_fn(BF(dav1d_prep_8tap_regular, lasx));
65 decl_mct_fn(BF(dav1d_prep_8tap_regular_smooth, lasx));
66 decl_mct_fn(BF(dav1d_prep_8tap_regular_sharp, lasx));
67 decl_mct_fn(BF(dav1d_prep_8tap_smooth, lasx));
68 decl_mct_fn(BF(dav1d_prep_8tap_smooth_regular, lasx));
69 decl_mct_fn(BF(dav1d_prep_8tap_smooth_sharp, lasx));
70 decl_mct_fn(BF(dav1d_prep_8tap_sharp, lasx));
71 decl_mct_fn(BF(dav1d_prep_8tap_sharp_regular, lasx));
72 decl_mct_fn(BF(dav1d_prep_8tap_sharp_smooth, lasx));
73
mc_dsp_init_loongarch(Dav1dMCDSPContext * const c)74 static ALWAYS_INLINE void mc_dsp_init_loongarch(Dav1dMCDSPContext *const c) {
75 #if BITDEPTH == 8
76 const unsigned flags = dav1d_get_cpu_flags();
77
78 if (!(flags & DAV1D_LOONGARCH_CPU_FLAG_LSX)) return;
79
80 c->avg = BF(dav1d_avg, lsx);
81 c->w_avg = BF(dav1d_w_avg, lsx);
82 c->mask = BF(dav1d_mask, lsx);
83 c->warp8x8 = BF(dav1d_warp_affine_8x8, lsx);
84 c->warp8x8t = BF(dav1d_warp_affine_8x8t, lsx);
85 c->w_mask[2] = BF(dav1d_w_mask_420, lsx);
86
87 init_mc_fn(FILTER_2D_8TAP_REGULAR, 8tap_regular, lsx);
88 init_mc_fn(FILTER_2D_8TAP_REGULAR_SMOOTH, 8tap_regular_smooth, lsx);
89 init_mc_fn(FILTER_2D_8TAP_REGULAR_SHARP, 8tap_regular_sharp, lsx);
90 init_mc_fn(FILTER_2D_8TAP_SMOOTH_REGULAR, 8tap_smooth_regular, lsx);
91 init_mc_fn(FILTER_2D_8TAP_SMOOTH, 8tap_smooth, lsx);
92 init_mc_fn(FILTER_2D_8TAP_SMOOTH_SHARP, 8tap_smooth_sharp, lsx);
93 init_mc_fn(FILTER_2D_8TAP_SHARP_REGULAR, 8tap_sharp_regular, lsx);
94 init_mc_fn(FILTER_2D_8TAP_SHARP_SMOOTH, 8tap_sharp_smooth, lsx);
95 init_mc_fn(FILTER_2D_8TAP_SHARP, 8tap_sharp, lsx);
96
97 if (!(flags & DAV1D_LOONGARCH_CPU_FLAG_LASX)) return;
98
99 c->avg = BF(dav1d_avg, lasx);
100 c->w_avg = BF(dav1d_w_avg, lasx);
101 c->mask = BF(dav1d_mask, lasx);
102 c->warp8x8 = BF(dav1d_warp_affine_8x8, lasx);
103 c->warp8x8t = BF(dav1d_warp_affine_8x8t, lasx);
104 c->w_mask[2] = BF(dav1d_w_mask_420, lasx);
105
106 init_mct_fn(FILTER_2D_8TAP_REGULAR, 8tap_regular, lasx);
107 init_mct_fn(FILTER_2D_8TAP_REGULAR_SMOOTH, 8tap_regular_smooth, lasx);
108 init_mct_fn(FILTER_2D_8TAP_REGULAR_SHARP, 8tap_regular_sharp, lasx);
109 init_mct_fn(FILTER_2D_8TAP_SMOOTH_REGULAR, 8tap_smooth_regular, lasx);
110 init_mct_fn(FILTER_2D_8TAP_SMOOTH, 8tap_smooth, lasx);
111 init_mct_fn(FILTER_2D_8TAP_SMOOTH_SHARP, 8tap_smooth_sharp, lasx);
112 init_mct_fn(FILTER_2D_8TAP_SHARP_REGULAR, 8tap_sharp_regular, lasx);
113 init_mct_fn(FILTER_2D_8TAP_SHARP_SMOOTH, 8tap_sharp_smooth, lasx);
114 init_mct_fn(FILTER_2D_8TAP_SHARP, 8tap_sharp, lasx);
115 #endif
116 }
117
118 #endif /* DAV1D_SRC_LOONGARCH_MC_H */
119