x86/vp9: add avx2 subpel MC SIMD for 10/12bpp
[ffmpeg.git] / libavcodec / x86 / vp9dsp_init_16bpp_template.c
1 /*
2  * VP9 SIMD optimizations
3  *
4  * Copyright (c) 2013 Ronald S. Bultje <rsbultje gmail com>
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/attributes.h"
24 #include "libavutil/cpu.h"
25 #include "libavutil/mem.h"
26 #include "libavutil/x86/cpu.h"
27 #include "libavcodec/vp9dsp.h"
28 #include "libavcodec/x86/vp9dsp_init.h"
29
30 #if HAVE_YASM
31
32 extern const int16_t ff_filters_16bpp[3][15][4][16];
33
34 decl_mc_funcs(4, sse2, int16_t, 16, BPC);
35 decl_mc_funcs(8, sse2, int16_t, 16, BPC);
36 decl_mc_funcs(16, avx2, int16_t, 16, BPC);
37
38 mc_rep_funcs(16,  8, 16, sse2, int16_t, 16, BPC);
39 mc_rep_funcs(32, 16, 32, sse2, int16_t, 16, BPC);
40 mc_rep_funcs(64, 32, 64, sse2, int16_t, 16, BPC);
41 mc_rep_funcs(32, 16, 32, avx2, int16_t, 16, BPC);
42 mc_rep_funcs(64, 32, 64, avx2, int16_t, 16, BPC);
43
44 filters_8tap_2d_fn2(put, 16, BPC, 2, sse2, sse2, 16bpp)
45 filters_8tap_2d_fn2(avg, 16, BPC, 2, sse2, sse2, 16bpp)
46 filters_8tap_2d_fn(put, 64, 32, BPC, 2, avx2, 16bpp)
47 filters_8tap_2d_fn(avg, 64, 32, BPC, 2, avx2, 16bpp)
48 filters_8tap_2d_fn(put, 32, 32, BPC, 2, avx2, 16bpp)
49 filters_8tap_2d_fn(avg, 32, 32, BPC, 2, avx2, 16bpp)
50 filters_8tap_2d_fn(put, 16, 32, BPC, 2, avx2, 16bpp)
51 filters_8tap_2d_fn(avg, 16, 32, BPC, 2, avx2, 16bpp)
52
53 filters_8tap_1d_fn3(put, BPC, sse2, sse2, 16bpp)
54 filters_8tap_1d_fn3(avg, BPC, sse2, sse2, 16bpp)
55 filters_8tap_1d_fn2(put, 64, BPC, avx2, 16bpp)
56 filters_8tap_1d_fn2(avg, 64, BPC, avx2, 16bpp)
57 filters_8tap_1d_fn2(put, 32, BPC, avx2, 16bpp)
58 filters_8tap_1d_fn2(avg, 32, BPC, avx2, 16bpp)
59 filters_8tap_1d_fn2(put, 16, BPC, avx2, 16bpp)
60 filters_8tap_1d_fn2(avg, 16, BPC, avx2, 16bpp)
61
62 #endif /* HAVE_YASM */
63
64 av_cold void INIT_FUNC(VP9DSPContext *dsp)
65 {
66 #if HAVE_YASM
67     int cpu_flags = av_get_cpu_flags();
68
69     if (EXTERNAL_SSE2(cpu_flags)) {
70         init_subpel3(0, put, BPC, sse2);
71         init_subpel3(1, avg, BPC, sse2);
72     }
73
74     if (EXTERNAL_AVX2(cpu_flags)) {
75         init_subpel3_32_64(0,  put, BPC, avx2);
76         init_subpel3_32_64(1,  avg, BPC, avx2);
77         init_subpel2(2, 0, 16, put, BPC, avx2);
78         init_subpel2(2, 1, 16, avg, BPC, avx2);
79     }
80
81 #endif /* HAVE_YASM */
82
83     ff_vp9dsp_init_16bpp_x86(dsp);
84 }