xref: /aosp_15_r20/external/libdav1d/src/riscv/mc.h (revision c09093415860a1c2373dacd84c4fde00c507cdfd)
1 /*
2  * Copyright © 2024, VideoLAN and dav1d authors
3  * Copyright © 2024, Nathan Egge
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 #include "src/cpu.h"
29 #include "src/mc.h"
30 
31 decl_blend_fn(BF(dav1d_blend, rvv));
32 decl_blend_dir_fn(BF(dav1d_blend_h, rvv));
33 decl_blend_dir_fn(BF(dav1d_blend_v, rvv));
34 
35 decl_blend_fn(BF(dav1d_blend_vl256, rvv));
36 decl_blend_dir_fn(BF(dav1d_blend_h_vl256, rvv));
37 decl_blend_dir_fn(BF(dav1d_blend_v_vl256, rvv));
38 
39 decl_avg_fn(BF(dav1d_avg, rvv));
40 decl_w_avg_fn(BF(dav1d_w_avg, rvv));
41 decl_mask_fn(BF(dav1d_mask, rvv));
42 
43 decl_warp8x8_fn(BF(dav1d_warp_8x8, rvv));
44 decl_warp8x8t_fn(BF(dav1d_warp_8x8t, rvv));
45 
mc_dsp_init_riscv(Dav1dMCDSPContext * const c)46 static ALWAYS_INLINE void mc_dsp_init_riscv(Dav1dMCDSPContext *const c) {
47   const unsigned flags = dav1d_get_cpu_flags();
48 
49   if (!(flags & DAV1D_RISCV_CPU_FLAG_V)) return;
50 
51 #if BITDEPTH == 8
52   c->blend = BF(dav1d_blend, rvv);
53   c->blend_h = BF(dav1d_blend_h, rvv);
54   c->blend_v = BF(dav1d_blend_v, rvv);
55 
56   if (dav1d_get_vlen() >= 256) {
57     c->blend = BF(dav1d_blend_vl256, rvv);
58     c->blend_h = BF(dav1d_blend_h_vl256, rvv);
59     c->blend_v = BF(dav1d_blend_v_vl256, rvv);
60   }
61 
62   c->avg     = BF(dav1d_avg, rvv);
63   c->w_avg   = BF(dav1d_w_avg, rvv);
64   c->mask    = BF(dav1d_mask, rvv);
65 
66   c->warp8x8 = BF(dav1d_warp_8x8, rvv);
67   c->warp8x8t = BF(dav1d_warp_8x8t, rvv);
68 #endif
69 }
70