1 /*
2 * Copyright © 2019, VideoLAN and dav1d authors
3 * Copyright © 2019, Two Orioles, LLC
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_X86_MSAC_H
29 #define DAV1D_SRC_X86_MSAC_H
30
31 #include "src/cpu.h"
32
33 unsigned dav1d_msac_decode_symbol_adapt4_sse2(MsacContext *s, uint16_t *cdf,
34 size_t n_symbols);
35 unsigned dav1d_msac_decode_symbol_adapt8_sse2(MsacContext *s, uint16_t *cdf,
36 size_t n_symbols);
37 unsigned dav1d_msac_decode_symbol_adapt16_sse2(MsacContext *s, uint16_t *cdf,
38 size_t n_symbols);
39 unsigned dav1d_msac_decode_symbol_adapt16_avx2(MsacContext *s, uint16_t *cdf,
40 size_t n_symbols);
41 unsigned dav1d_msac_decode_bool_adapt_sse2(MsacContext *s, uint16_t *cdf);
42 unsigned dav1d_msac_decode_bool_equi_sse2(MsacContext *s);
43 unsigned dav1d_msac_decode_bool_sse2(MsacContext *s, unsigned f);
44 unsigned dav1d_msac_decode_hi_tok_sse2(MsacContext *s, uint16_t *cdf);
45
46 #if ARCH_X86_64 || defined(__SSE2__) || (defined(_M_IX86_FP) && _M_IX86_FP >= 2)
47 #define dav1d_msac_decode_symbol_adapt4 dav1d_msac_decode_symbol_adapt4_sse2
48 #define dav1d_msac_decode_symbol_adapt8 dav1d_msac_decode_symbol_adapt8_sse2
49 #define dav1d_msac_decode_hi_tok dav1d_msac_decode_hi_tok_sse2
50 #endif
51
52 #define dav1d_msac_decode_bool_adapt dav1d_msac_decode_bool_adapt_sse2
53 #define dav1d_msac_decode_bool_equi dav1d_msac_decode_bool_equi_sse2
54 #define dav1d_msac_decode_bool dav1d_msac_decode_bool_sse2
55
56 #if ARCH_X86_64
57 #define dav1d_msac_decode_symbol_adapt16(ctx, cdf, symb) ((ctx)->symbol_adapt16(ctx, cdf, symb))
58
msac_init_x86(MsacContext * const s)59 static ALWAYS_INLINE void msac_init_x86(MsacContext *const s) {
60 const unsigned flags = dav1d_get_cpu_flags();
61
62 if (flags & DAV1D_X86_CPU_FLAG_SSE2) {
63 s->symbol_adapt16 = dav1d_msac_decode_symbol_adapt16_sse2;
64 }
65
66 if (flags & DAV1D_X86_CPU_FLAG_AVX2) {
67 s->symbol_adapt16 = dav1d_msac_decode_symbol_adapt16_avx2;
68 }
69 }
70
71 #elif defined(__SSE2__) || (defined(_M_IX86_FP) && _M_IX86_FP >= 2)
72 #define dav1d_msac_decode_symbol_adapt16 dav1d_msac_decode_symbol_adapt16_sse2
73 #endif
74
75 #endif /* DAV1D_SRC_X86_MSAC_H */
76