FFmpeg
vp9dsp_init.h
Go to the documentation of this file.
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 #ifndef AVCODEC_X86_VP9DSP_INIT_H
24 #define AVCODEC_X86_VP9DSP_INIT_H
25 
26 #include "libavutil/attributes.h"
27 #include "libavutil/mem_internal.h"
28 
29 #include "libavcodec/vp9dsp.h"
30 
31 // hack to force-expand BPC
32 #define cat(a, bpp, b) a##bpp##b
33 
34 #define decl_fpel_func(avg, sz, bpp, opt) \
35 void ff_vp9_##avg##sz##bpp##_##opt(uint8_t *dst, ptrdiff_t dst_stride, \
36  const uint8_t *src, ptrdiff_t src_stride, \
37  int h, int mx, int my)
38 
39 #define decl_mc_func(avg, sz, dir, opt, type, f_sz, bpp) \
40 void ff_vp9_##avg##_8tap_1d_##dir##_##sz##_##bpp##_##opt(uint8_t *dst, ptrdiff_t dst_stride, \
41  const uint8_t *src, ptrdiff_t src_stride, \
42  int h, const type (*filter)[f_sz])
43 
44 #define decl_mc_funcs(sz, opt, type, fsz, bpp) \
45 decl_mc_func(put, sz, h, opt, type, fsz, bpp); \
46 decl_mc_func(avg, sz, h, opt, type, fsz, bpp); \
47 decl_mc_func(put, sz, v, opt, type, fsz, bpp); \
48 decl_mc_func(avg, sz, v, opt, type, fsz, bpp)
49 
50 #define decl_ipred_fn(type, sz, bpp, opt) \
51 void ff_vp9_ipred_##type##_##sz##x##sz##_##bpp##_##opt(uint8_t *dst, \
52  ptrdiff_t stride, \
53  const uint8_t *l, \
54  const uint8_t *a)
55 
56 #define decl_ipred_fns(type, bpp, opt4, opt8_16_32) \
57 decl_ipred_fn(type, 4, bpp, opt4); \
58 decl_ipred_fn(type, 8, bpp, opt8_16_32); \
59 decl_ipred_fn(type, 16, bpp, opt8_16_32); \
60 decl_ipred_fn(type, 32, bpp, opt8_16_32)
61 
62 #define decl_itxfm_func(typea, typeb, size, bpp, opt) \
63 void cat(ff_vp9_##typea##_##typeb##_##size##x##size##_add_, bpp, _##opt)(uint8_t *dst, \
64  ptrdiff_t stride, \
65  int16_t *block, \
66  int eob)
67 
68 #define decl_itxfm_funcs(size, bpp, opt) \
69 decl_itxfm_func(idct, idct, size, bpp, opt); \
70 decl_itxfm_func(iadst, idct, size, bpp, opt); \
71 decl_itxfm_func(idct, iadst, size, bpp, opt); \
72 decl_itxfm_func(iadst, iadst, size, bpp, opt)
73 
74 #define mc_rep_func(avg, sz, hsz, hszb, dir, opt, type, f_sz, bpp) \
75 static av_always_inline void \
76 ff_vp9_##avg##_8tap_1d_##dir##_##sz##_##bpp##_##opt(uint8_t *dst, ptrdiff_t dst_stride, \
77  const uint8_t *src, ptrdiff_t src_stride, \
78  int h, const type (*filter)[f_sz]) \
79 { \
80  ff_vp9_##avg##_8tap_1d_##dir##_##hsz##_##bpp##_##opt(dst, dst_stride, src, \
81  src_stride, h, filter); \
82  ff_vp9_##avg##_8tap_1d_##dir##_##hsz##_##bpp##_##opt(dst + hszb, dst_stride, src + hszb, \
83  src_stride, h, filter); \
84 }
85 
86 #define mc_rep_funcs(sz, hsz, hszb, opt, type, fsz, bpp) \
87 mc_rep_func(put, sz, hsz, hszb, h, opt, type, fsz, bpp) \
88 mc_rep_func(avg, sz, hsz, hszb, h, opt, type, fsz, bpp) \
89 mc_rep_func(put, sz, hsz, hszb, v, opt, type, fsz, bpp) \
90 mc_rep_func(avg, sz, hsz, hszb, v, opt, type, fsz, bpp)
91 
92 #define filter_8tap_1d_fn(op, sz, f, f_opt, fname, dir, dvar, bpp, opt) \
93 static void op##_8tap_##fname##_##sz##dir##_##bpp##_##opt(uint8_t *dst, ptrdiff_t dst_stride, \
94  const uint8_t *src, ptrdiff_t src_stride, \
95  int h, int mx, int my) \
96 { \
97  ff_vp9_##op##_8tap_1d_##dir##_##sz##_##bpp##_##opt(dst, dst_stride, src, src_stride, \
98  h, ff_filters_##f_opt[f][dvar - 1]); \
99 }
100 
101 #define filters_8tap_1d_fn(op, sz, dir, dvar, bpp, opt, f_opt) \
102 filter_8tap_1d_fn(op, sz, FILTER_8TAP_REGULAR, f_opt, regular, dir, dvar, bpp, opt) \
103 filter_8tap_1d_fn(op, sz, FILTER_8TAP_SHARP, f_opt, sharp, dir, dvar, bpp, opt) \
104 filter_8tap_1d_fn(op, sz, FILTER_8TAP_SMOOTH, f_opt, smooth, dir, dvar, bpp, opt)
105 
106 #define filters_8tap_1d_fn2(op, sz, bpp, opt, f_opt) \
107 filters_8tap_1d_fn(op, sz, h, mx, bpp, opt, f_opt) \
108 filters_8tap_1d_fn(op, sz, v, my, bpp, opt, f_opt)
109 
110 #define filters_8tap_1d_fn3(op, bpp, opt4, opt8, f_opt) \
111 filters_8tap_1d_fn2(op, 64, bpp, opt8, f_opt) \
112 filters_8tap_1d_fn2(op, 32, bpp, opt8, f_opt) \
113 filters_8tap_1d_fn2(op, 16, bpp, opt8, f_opt) \
114 filters_8tap_1d_fn2(op, 8, bpp, opt8, f_opt) \
115 filters_8tap_1d_fn2(op, 4, bpp, opt4, f_opt)
116 
117 #define filter_8tap_2d_fn(op, sz, f, f_opt, fname, align, bpp, bytes, opt) \
118 static void op##_8tap_##fname##_##sz##hv_##bpp##_##opt(uint8_t *dst, ptrdiff_t dst_stride, \
119  const uint8_t *src, ptrdiff_t src_stride, \
120  int h, int mx, int my) \
121 { \
122  LOCAL_ALIGNED_##align(uint8_t, temp, [71 * 64 * bytes]); \
123  ff_vp9_put_8tap_1d_h_##sz##_##bpp##_##opt(temp, 64 * bytes, src - 3 * src_stride, \
124  src_stride, h + 7, \
125  ff_filters_##f_opt[f][mx - 1]); \
126  ff_vp9_##op##_8tap_1d_v_##sz##_##bpp##_##opt(dst, dst_stride, temp + 3 * bytes * 64, \
127  64 * bytes, h, \
128  ff_filters_##f_opt[f][my - 1]); \
129 }
130 
131 #define filters_8tap_2d_fn(op, sz, align, bpp, bytes, opt, f_opt) \
132 filter_8tap_2d_fn(op, sz, FILTER_8TAP_REGULAR, f_opt, regular, align, bpp, bytes, opt) \
133 filter_8tap_2d_fn(op, sz, FILTER_8TAP_SHARP, f_opt, sharp, align, bpp, bytes, opt) \
134 filter_8tap_2d_fn(op, sz, FILTER_8TAP_SMOOTH, f_opt, smooth, align, bpp, bytes, opt)
135 
136 #define filters_8tap_2d_fn2(op, align, bpp, bytes, opt4, opt8, f_opt) \
137 filters_8tap_2d_fn(op, 64, align, bpp, bytes, opt8, f_opt) \
138 filters_8tap_2d_fn(op, 32, align, bpp, bytes, opt8, f_opt) \
139 filters_8tap_2d_fn(op, 16, align, bpp, bytes, opt8, f_opt) \
140 filters_8tap_2d_fn(op, 8, align, bpp, bytes, opt8, f_opt) \
141 filters_8tap_2d_fn(op, 4, align, bpp, bytes, opt4, f_opt)
142 
143 #define init_fpel_func(idx1, idx2, sz, type, bpp, opt) \
144  dsp->mc[idx1][FILTER_8TAP_SMOOTH ][idx2][0][0] = \
145  dsp->mc[idx1][FILTER_8TAP_REGULAR][idx2][0][0] = \
146  dsp->mc[idx1][FILTER_8TAP_SHARP ][idx2][0][0] = \
147  dsp->mc[idx1][FILTER_BILINEAR ][idx2][0][0] = ff_vp9_##type##sz##bpp##_##opt
148 
149 #define init_subpel1(idx1, idx2, idxh, idxv, sz, dir, type, bpp, opt) \
150  dsp->mc[idx1][FILTER_8TAP_SMOOTH ][idx2][idxh][idxv] = \
151  type##_8tap_smooth_##sz##dir##_##bpp##_##opt; \
152  dsp->mc[idx1][FILTER_8TAP_REGULAR][idx2][idxh][idxv] = \
153  type##_8tap_regular_##sz##dir##_##bpp##_##opt; \
154  dsp->mc[idx1][FILTER_8TAP_SHARP ][idx2][idxh][idxv] = \
155  type##_8tap_sharp_##sz##dir##_##bpp##_##opt
156 
157 #define init_subpel2(idx1, idx2, sz, type, bpp, opt) \
158  init_subpel1(idx1, idx2, 1, 1, sz, hv, type, bpp, opt); \
159  init_subpel1(idx1, idx2, 0, 1, sz, v, type, bpp, opt); \
160  init_subpel1(idx1, idx2, 1, 0, sz, h, type, bpp, opt)
161 
162 #define init_subpel3_32_64(idx, type, bpp, opt) \
163  init_subpel2(0, idx, 64, type, bpp, opt); \
164  init_subpel2(1, idx, 32, type, bpp, opt)
165 
166 #define init_subpel3_8to64(idx, type, bpp, opt) \
167  init_subpel3_32_64(idx, type, bpp, opt); \
168  init_subpel2(2, idx, 16, type, bpp, opt); \
169  init_subpel2(3, idx, 8, type, bpp, opt)
170 
171 #define init_subpel3(idx, type, bpp, opt) \
172  init_subpel3_8to64(idx, type, bpp, opt); \
173  init_subpel2(4, idx, 4, type, bpp, opt)
174 
175 #define init_ipred_func(type, enum, sz, bpp, opt) \
176  dsp->intra_pred[TX_##sz##X##sz][enum##_PRED] = \
177  cat(ff_vp9_ipred_##type##_##sz##x##sz##_, bpp, _##opt)
178 
179 #define init_8_16_32_ipred_funcs(type, enum, bpp, opt) \
180  init_ipred_func(type, enum, 8, bpp, opt); \
181  init_ipred_func(type, enum, 16, bpp, opt); \
182  init_ipred_func(type, enum, 32, bpp, opt)
183 
184 #define init_ipred_funcs(type, enum, bpp, opt) \
185  init_ipred_func(type, enum, 4, bpp, opt); \
186  init_8_16_32_ipred_funcs(type, enum, bpp, opt)
187 
188 void ff_vp9dsp_init_10bpp_x86(VP9DSPContext *dsp, int bitexact);
189 void ff_vp9dsp_init_12bpp_x86(VP9DSPContext *dsp, int bitexact);
191 
192 #endif /* AVCODEC_X86_VP9DSP_INIT_H */
mem_internal.h
VP9DSPContext
Definition: vp9dsp.h:40
ff_vp9dsp_init_10bpp_x86
void ff_vp9dsp_init_10bpp_x86(VP9DSPContext *dsp, int bitexact)
ff_vp9dsp_init_12bpp_x86
void ff_vp9dsp_init_12bpp_x86(VP9DSPContext *dsp, int bitexact)
vp9dsp.h
ff_vp9dsp_init_16bpp_x86
void ff_vp9dsp_init_16bpp_x86(VP9DSPContext *dsp)
Definition: vp9dsp_init_16bpp.c:73
attributes.h