x86: h264qpel: int --> ptrdiff_t for some line_size parameters
[ffmpeg.git] / libavcodec / x86 / h264_qpel.c
1 /*
2  * Copyright (c) 2004-2005 Michael Niedermayer, Loren Merritt
3  * Copyright (c) 2011 Daniel Kang
4  *
5  * This file is part of Libav.
6  *
7  * Libav is free software; you can redistribute it and/or
8  * modify it under the terms of the GNU Lesser General Public
9  * License as published by the Free Software Foundation; either
10  * version 2.1 of the License, or (at your option) any later version.
11  *
12  * Libav is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
15  * Lesser General Public License for more details.
16  *
17  * You should have received a copy of the GNU Lesser General Public
18  * License along with Libav; if not, write to the Free Software
19  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20  */
21
22 #include "libavutil/attributes.h"
23 #include "libavutil/cpu.h"
24 #include "libavutil/x86/asm.h"
25 #include "libavutil/x86/cpu.h"
26 #include "libavcodec/h264qpel.h"
27 #include "libavcodec/mpegvideo.h"
28 #include "dsputil_mmx.h"
29
30 #if HAVE_YASM
31 void ff_put_pixels4_mmxext(uint8_t *block, const uint8_t *pixels,
32                            ptrdiff_t line_size, int h);
33 void ff_avg_pixels4_mmxext(uint8_t *block, const uint8_t *pixels,
34                            ptrdiff_t line_size, int h);
35 void ff_put_pixels8_mmxext(uint8_t *block, const uint8_t *pixels,
36                            ptrdiff_t line_size, int h);
37 static void ff_put_pixels16_mmxext(uint8_t *block, const uint8_t *pixels,
38                                    ptrdiff_t line_size, int h)
39 {
40     ff_put_pixels8_mmxext(block,     pixels,     line_size, h);
41     ff_put_pixels8_mmxext(block + 8, pixels + 8, line_size, h);
42 }
43 static void ff_avg_pixels16_mmxext(uint8_t *block, const uint8_t *pixels,
44                                    ptrdiff_t line_size, int h)
45 {
46     ff_avg_pixels8_mmxext(block,     pixels,     line_size, h);
47     ff_avg_pixels8_mmxext(block + 8, pixels + 8, line_size, h);
48 }
49 void ff_put_pixels4_l2_mmxext(uint8_t *dst, uint8_t *src1, uint8_t *src2,
50                               int dstStride, int src1Stride, int h);
51 void ff_avg_pixels4_l2_mmxext(uint8_t *dst, uint8_t *src1, uint8_t *src2,
52                               int dstStride, int src1Stride, int h);
53 void ff_put_pixels8_l2_mmxext(uint8_t *dst, uint8_t *src1, uint8_t *src2,
54                               int dstStride, int src1Stride, int h);
55 void ff_avg_pixels8_l2_mmxext(uint8_t *dst, uint8_t *src1, uint8_t *src2,
56                               int dstStride, int src1Stride, int h);
57 void ff_put_pixels16_l2_mmxext(uint8_t *dst, uint8_t *src1, uint8_t *src2,
58                                int dstStride, int src1Stride, int h);
59 void ff_avg_pixels16_l2_mmxext(uint8_t *dst, uint8_t *src1, uint8_t *src2,
60                                int dstStride, int src1Stride, int h);
61 void ff_put_pixels16_sse2(uint8_t *block, const uint8_t *pixels,
62                           int line_size, int h);
63 void ff_avg_pixels16_sse2(uint8_t *block, const uint8_t *pixels,
64                           int line_size, int h);
65 #define ff_put_pixels8_l2_sse2  ff_put_pixels8_l2_mmxext
66 #define ff_avg_pixels8_l2_sse2  ff_avg_pixels8_l2_mmxext
67 #define ff_put_pixels16_l2_sse2 ff_put_pixels16_l2_mmxext
68 #define ff_avg_pixels16_l2_sse2 ff_avg_pixels16_l2_mmxext
69
70 #define DEF_QPEL(OPNAME)\
71 void ff_ ## OPNAME ## _h264_qpel4_h_lowpass_mmxext(uint8_t *dst, uint8_t *src, int dstStride, int srcStride);\
72 void ff_ ## OPNAME ## _h264_qpel8_h_lowpass_mmxext(uint8_t *dst, uint8_t *src, int dstStride, int srcStride);\
73 void ff_ ## OPNAME ## _h264_qpel8_h_lowpass_ssse3(uint8_t *dst, uint8_t *src, int dstStride, int srcStride);\
74 void ff_ ## OPNAME ## _h264_qpel4_h_lowpass_l2_mmxext(uint8_t *dst, uint8_t *src, uint8_t *src2, int dstStride, int src2Stride);\
75 void ff_ ## OPNAME ## _h264_qpel8_h_lowpass_l2_mmxext(uint8_t *dst, uint8_t *src, uint8_t *src2, int dstStride, int src2Stride);\
76 void ff_ ## OPNAME ## _h264_qpel8_h_lowpass_l2_ssse3(uint8_t *dst, uint8_t *src, uint8_t *src2, int dstStride, int src2Stride);\
77 void ff_ ## OPNAME ## _h264_qpel4_v_lowpass_mmxext(uint8_t *dst, uint8_t *src, int dstStride, int srcStride);\
78 void ff_ ## OPNAME ## _h264_qpel8or16_v_lowpass_op_mmxext(uint8_t *dst, uint8_t *src, int dstStride, int srcStride, int h);\
79 void ff_ ## OPNAME ## _h264_qpel8or16_v_lowpass_sse2(uint8_t *dst, uint8_t *src, int dstStride, int srcStride, int h);\
80 void ff_ ## OPNAME ## _h264_qpel4_hv_lowpass_v_mmxext(uint8_t *src, int16_t *tmp, int srcStride);\
81 void ff_ ## OPNAME ## _h264_qpel4_hv_lowpass_h_mmxext(int16_t *tmp, uint8_t *dst, int dstStride);\
82 void ff_ ## OPNAME ## _h264_qpel8or16_hv1_lowpass_op_mmxext(uint8_t *src, int16_t *tmp, int srcStride, int size);\
83 void ff_ ## OPNAME ## _h264_qpel8or16_hv1_lowpass_op_sse2(uint8_t *src, int16_t *tmp, int srcStride, int size);\
84 void ff_ ## OPNAME ## _h264_qpel8or16_hv2_lowpass_op_mmxext(uint8_t *dst, int16_t *tmp, int dstStride, int unused, int h);\
85 void ff_ ## OPNAME ## _h264_qpel8or16_hv2_lowpass_ssse3(uint8_t *dst, int16_t *tmp, int dstStride, int tmpStride, int size);\
86 void ff_ ## OPNAME ## _pixels4_l2_shift5_mmxext(uint8_t *dst, int16_t *src16, uint8_t *src8, int dstStride, int src8Stride, int h);\
87 void ff_ ## OPNAME ## _pixels8_l2_shift5_mmxext(uint8_t *dst, int16_t *src16, uint8_t *src8, int dstStride, int src8Stride, int h);
88
89 DEF_QPEL(avg)
90 DEF_QPEL(put)
91
92 #define QPEL_H264(OPNAME, OP, MMX)\
93 static av_always_inline void ff_ ## OPNAME ## h264_qpel4_hv_lowpass_ ## MMX(uint8_t *dst, int16_t *tmp, uint8_t *src, int dstStride, int tmpStride, int srcStride){\
94     int w=3;\
95     src -= 2*srcStride+2;\
96     while(w--){\
97         ff_ ## OPNAME ## h264_qpel4_hv_lowpass_v_mmxext(src, tmp, srcStride);\
98         tmp += 4;\
99         src += 4;\
100     }\
101     tmp -= 3*4;\
102     ff_ ## OPNAME ## h264_qpel4_hv_lowpass_h_mmxext(tmp, dst, dstStride);\
103 }\
104 \
105 static av_always_inline void ff_ ## OPNAME ## h264_qpel8or16_v_lowpass_ ## MMX(uint8_t *dst, uint8_t *src, int dstStride, int srcStride, int h){\
106     src -= 2*srcStride;\
107     ff_ ## OPNAME ## h264_qpel8or16_v_lowpass_op_mmxext(dst, src, dstStride, srcStride, h);\
108     src += 4;\
109     dst += 4;\
110     ff_ ## OPNAME ## h264_qpel8or16_v_lowpass_op_mmxext(dst, src, dstStride, srcStride, h);\
111 }\
112 static av_always_inline void ff_ ## OPNAME ## h264_qpel8or16_hv1_lowpass_ ## MMX(int16_t *tmp, uint8_t *src, int tmpStride, int srcStride, int size){\
113     int w = (size+8)>>2;\
114     src -= 2*srcStride+2;\
115     while(w--){\
116         ff_ ## OPNAME ## h264_qpel8or16_hv1_lowpass_op_mmxext(src, tmp, srcStride, size);\
117         tmp += 4;\
118         src += 4;\
119     }\
120 }\
121 static av_always_inline void ff_ ## OPNAME ## h264_qpel8or16_hv2_lowpass_ ## MMX(uint8_t *dst, int16_t *tmp, int dstStride, int tmpStride, int size){\
122     int w = size>>4;\
123     do{\
124     ff_ ## OPNAME ## h264_qpel8or16_hv2_lowpass_op_mmxext(dst, tmp, dstStride, 0, size);\
125     tmp += 8;\
126     dst += 8;\
127     }while(w--);\
128 }\
129 \
130 static av_always_inline void ff_ ## OPNAME ## h264_qpel8_v_lowpass_ ## MMX(uint8_t *dst, uint8_t *src, int dstStride, int srcStride){\
131     ff_ ## OPNAME ## h264_qpel8or16_v_lowpass_ ## MMX(dst  , src  , dstStride, srcStride, 8);\
132 }\
133 static av_always_inline void ff_ ## OPNAME ## h264_qpel16_v_lowpass_ ## MMX(uint8_t *dst, uint8_t *src, int dstStride, int srcStride){\
134     ff_ ## OPNAME ## h264_qpel8or16_v_lowpass_ ## MMX(dst  , src  , dstStride, srcStride, 16);\
135     ff_ ## OPNAME ## h264_qpel8or16_v_lowpass_ ## MMX(dst+8, src+8, dstStride, srcStride, 16);\
136 }\
137 \
138 static av_always_inline void ff_ ## OPNAME ## h264_qpel16_h_lowpass_ ## MMX(uint8_t *dst, uint8_t *src, int dstStride, int srcStride){\
139     ff_ ## OPNAME ## h264_qpel8_h_lowpass_ ## MMX(dst  , src  , dstStride, srcStride);\
140     ff_ ## OPNAME ## h264_qpel8_h_lowpass_ ## MMX(dst+8, src+8, dstStride, srcStride);\
141     src += 8*srcStride;\
142     dst += 8*dstStride;\
143     ff_ ## OPNAME ## h264_qpel8_h_lowpass_ ## MMX(dst  , src  , dstStride, srcStride);\
144     ff_ ## OPNAME ## h264_qpel8_h_lowpass_ ## MMX(dst+8, src+8, dstStride, srcStride);\
145 }\
146 \
147 static av_always_inline void ff_ ## OPNAME ## h264_qpel16_h_lowpass_l2_ ## MMX(uint8_t *dst, uint8_t *src, uint8_t *src2, int dstStride, int src2Stride){\
148     ff_ ## OPNAME ## h264_qpel8_h_lowpass_l2_ ## MMX(dst  , src  , src2  , dstStride, src2Stride);\
149     ff_ ## OPNAME ## h264_qpel8_h_lowpass_l2_ ## MMX(dst+8, src+8, src2+8, dstStride, src2Stride);\
150     src += 8*dstStride;\
151     dst += 8*dstStride;\
152     src2 += 8*src2Stride;\
153     ff_ ## OPNAME ## h264_qpel8_h_lowpass_l2_ ## MMX(dst  , src  , src2  , dstStride, src2Stride);\
154     ff_ ## OPNAME ## h264_qpel8_h_lowpass_l2_ ## MMX(dst+8, src+8, src2+8, dstStride, src2Stride);\
155 }\
156 \
157 static av_always_inline void ff_ ## OPNAME ## h264_qpel8or16_hv_lowpass_ ## MMX(uint8_t *dst, int16_t *tmp, uint8_t *src, int dstStride, int tmpStride, int srcStride, int size){\
158     ff_put_h264_qpel8or16_hv1_lowpass_ ## MMX(tmp, src, tmpStride, srcStride, size);\
159     ff_ ## OPNAME ## h264_qpel8or16_hv2_lowpass_ ## MMX(dst, tmp, dstStride, tmpStride, size);\
160 }\
161 static av_always_inline void ff_ ## OPNAME ## h264_qpel8_hv_lowpass_ ## MMX(uint8_t *dst, int16_t *tmp, uint8_t *src, int dstStride, int tmpStride, int srcStride){\
162     ff_ ## OPNAME ## h264_qpel8or16_hv_lowpass_ ## MMX(dst  , tmp  , src  , dstStride, tmpStride, srcStride, 8);\
163 }\
164 \
165 static av_always_inline void ff_ ## OPNAME ## h264_qpel16_hv_lowpass_ ## MMX(uint8_t *dst, int16_t *tmp, uint8_t *src, int dstStride, int tmpStride, int srcStride){\
166     ff_ ## OPNAME ## h264_qpel8or16_hv_lowpass_ ## MMX(dst  , tmp  , src  , dstStride, tmpStride, srcStride, 16);\
167 }\
168 \
169 static av_always_inline void ff_ ## OPNAME ## pixels16_l2_shift5_ ## MMX(uint8_t *dst, int16_t *src16, uint8_t *src8, int dstStride, int src8Stride, int h)\
170 {\
171     ff_ ## OPNAME ## pixels8_l2_shift5_ ## MMX(dst  , src16  , src8  , dstStride, src8Stride, h);\
172     ff_ ## OPNAME ## pixels8_l2_shift5_ ## MMX(dst+8, src16+8, src8+8, dstStride, src8Stride, h);\
173 }\
174
175
176 #if ARCH_X86_64
177 #define QPEL_H264_H16_XMM(OPNAME, OP, MMX)\
178
179 void ff_avg_h264_qpel16_h_lowpass_l2_ssse3(uint8_t *dst, uint8_t *src, uint8_t *src2, int dstStride, int src2Stride);
180 void ff_put_h264_qpel16_h_lowpass_l2_ssse3(uint8_t *dst, uint8_t *src, uint8_t *src2, int dstStride, int src2Stride);
181
182 #else // ARCH_X86_64
183 #define QPEL_H264_H16_XMM(OPNAME, OP, MMX)\
184 static av_always_inline void ff_ ## OPNAME ## h264_qpel16_h_lowpass_l2_ ## MMX(uint8_t *dst, uint8_t *src, uint8_t *src2, int dstStride, int src2Stride){\
185     ff_ ## OPNAME ## h264_qpel8_h_lowpass_l2_ ## MMX(dst  , src  , src2  , dstStride, src2Stride);\
186     ff_ ## OPNAME ## h264_qpel8_h_lowpass_l2_ ## MMX(dst+8, src+8, src2+8, dstStride, src2Stride);\
187     src += 8*dstStride;\
188     dst += 8*dstStride;\
189     src2 += 8*src2Stride;\
190     ff_ ## OPNAME ## h264_qpel8_h_lowpass_l2_ ## MMX(dst  , src  , src2  , dstStride, src2Stride);\
191     ff_ ## OPNAME ## h264_qpel8_h_lowpass_l2_ ## MMX(dst+8, src+8, src2+8, dstStride, src2Stride);\
192 }
193 #endif // ARCH_X86_64
194
195 #define QPEL_H264_H_XMM(OPNAME, OP, MMX)\
196 QPEL_H264_H16_XMM(OPNAME, OP, MMX)\
197 static av_always_inline void ff_ ## OPNAME ## h264_qpel16_h_lowpass_ ## MMX(uint8_t *dst, uint8_t *src, int dstStride, int srcStride){\
198     ff_ ## OPNAME ## h264_qpel8_h_lowpass_ ## MMX(dst  , src  , dstStride, srcStride);\
199     ff_ ## OPNAME ## h264_qpel8_h_lowpass_ ## MMX(dst+8, src+8, dstStride, srcStride);\
200     src += 8*srcStride;\
201     dst += 8*dstStride;\
202     ff_ ## OPNAME ## h264_qpel8_h_lowpass_ ## MMX(dst  , src  , dstStride, srcStride);\
203     ff_ ## OPNAME ## h264_qpel8_h_lowpass_ ## MMX(dst+8, src+8, dstStride, srcStride);\
204 }\
205
206 #define QPEL_H264_V_XMM(OPNAME, OP, MMX)\
207 static av_always_inline void ff_ ## OPNAME ## h264_qpel8_v_lowpass_ ## MMX(uint8_t *dst, uint8_t *src, int dstStride, int srcStride){\
208     ff_ ## OPNAME ## h264_qpel8or16_v_lowpass_ ## MMX(dst  , src  , dstStride, srcStride, 8);\
209 }\
210 static av_always_inline void ff_ ## OPNAME ## h264_qpel16_v_lowpass_ ## MMX(uint8_t *dst, uint8_t *src, int dstStride, int srcStride){\
211     ff_ ## OPNAME ## h264_qpel8or16_v_lowpass_ ## MMX(dst  , src  , dstStride, srcStride, 16);\
212     ff_ ## OPNAME ## h264_qpel8or16_v_lowpass_ ## MMX(dst+8, src+8, dstStride, srcStride, 16);\
213 }
214
215 static av_always_inline void ff_put_h264_qpel8or16_hv1_lowpass_sse2(int16_t *tmp, uint8_t *src, int tmpStride, int srcStride, int size){
216     int w = (size+8)>>3;
217     src -= 2*srcStride+2;
218     while(w--){
219         ff_put_h264_qpel8or16_hv1_lowpass_op_sse2(src, tmp, srcStride, size);
220         tmp += 8;
221         src += 8;
222     }
223 }
224
225 #define QPEL_H264_HV_XMM(OPNAME, OP, MMX)\
226 static av_always_inline void ff_ ## OPNAME ## h264_qpel8or16_hv_lowpass_ ## MMX(uint8_t *dst, int16_t *tmp, uint8_t *src, int dstStride, int tmpStride, int srcStride, int size){\
227     ff_put_h264_qpel8or16_hv1_lowpass_sse2(tmp, src, tmpStride, srcStride, size);\
228     ff_ ## OPNAME ## h264_qpel8or16_hv2_lowpass_ ## MMX(dst, tmp, dstStride, tmpStride, size);\
229 }\
230 static av_always_inline void ff_ ## OPNAME ## h264_qpel8_hv_lowpass_ ## MMX(uint8_t *dst, int16_t *tmp, uint8_t *src, int dstStride, int tmpStride, int srcStride){\
231     ff_ ## OPNAME ## h264_qpel8or16_hv_lowpass_ ## MMX(dst, tmp, src, dstStride, tmpStride, srcStride, 8);\
232 }\
233 static av_always_inline void ff_ ## OPNAME ## h264_qpel16_hv_lowpass_ ## MMX(uint8_t *dst, int16_t *tmp, uint8_t *src, int dstStride, int tmpStride, int srcStride){\
234     ff_ ## OPNAME ## h264_qpel8or16_hv_lowpass_ ## MMX(dst, tmp, src, dstStride, tmpStride, srcStride, 16);\
235 }\
236
237 #define ff_put_h264_qpel8_h_lowpass_l2_sse2  ff_put_h264_qpel8_h_lowpass_l2_mmxext
238 #define ff_avg_h264_qpel8_h_lowpass_l2_sse2  ff_avg_h264_qpel8_h_lowpass_l2_mmxext
239 #define ff_put_h264_qpel16_h_lowpass_l2_sse2 ff_put_h264_qpel16_h_lowpass_l2_mmxext
240 #define ff_avg_h264_qpel16_h_lowpass_l2_sse2 ff_avg_h264_qpel16_h_lowpass_l2_mmxext
241
242 #define ff_put_h264_qpel8_v_lowpass_ssse3  ff_put_h264_qpel8_v_lowpass_sse2
243 #define ff_avg_h264_qpel8_v_lowpass_ssse3  ff_avg_h264_qpel8_v_lowpass_sse2
244 #define ff_put_h264_qpel16_v_lowpass_ssse3 ff_put_h264_qpel16_v_lowpass_sse2
245 #define ff_avg_h264_qpel16_v_lowpass_ssse3 ff_avg_h264_qpel16_v_lowpass_sse2
246
247 #define ff_put_h264_qpel8or16_hv2_lowpass_sse2 ff_put_h264_qpel8or16_hv2_lowpass_mmxext
248 #define ff_avg_h264_qpel8or16_hv2_lowpass_sse2 ff_avg_h264_qpel8or16_hv2_lowpass_mmxext
249
250 #define H264_MC(OPNAME, SIZE, MMX, ALIGN) \
251 H264_MC_C(OPNAME, SIZE, MMX, ALIGN)\
252 H264_MC_V(OPNAME, SIZE, MMX, ALIGN)\
253 H264_MC_H(OPNAME, SIZE, MMX, ALIGN)\
254 H264_MC_HV(OPNAME, SIZE, MMX, ALIGN)\
255
256 static void put_h264_qpel16_mc00_sse2 (uint8_t *dst, uint8_t *src,
257                                        ptrdiff_t stride)
258 {
259     ff_put_pixels16_sse2(dst, src, stride, 16);
260 }
261 static void avg_h264_qpel16_mc00_sse2 (uint8_t *dst, uint8_t *src,
262                                        ptrdiff_t stride)
263 {
264     ff_avg_pixels16_sse2(dst, src, stride, 16);
265 }
266 #define put_h264_qpel8_mc00_sse2 put_h264_qpel8_mc00_mmxext
267 #define avg_h264_qpel8_mc00_sse2 avg_h264_qpel8_mc00_mmxext
268
269 #define H264_MC_C(OPNAME, SIZE, MMX, ALIGN) \
270 static void OPNAME ## h264_qpel ## SIZE ## _mc00_ ## MMX (uint8_t *dst, uint8_t *src, ptrdiff_t stride)\
271 {\
272     ff_ ## OPNAME ## pixels ## SIZE ## _ ## MMX(dst, src, stride, SIZE);\
273 }\
274
275 #define H264_MC_H(OPNAME, SIZE, MMX, ALIGN) \
276 static void OPNAME ## h264_qpel ## SIZE ## _mc10_ ## MMX(uint8_t *dst, uint8_t *src, ptrdiff_t stride)\
277 {\
278     ff_ ## OPNAME ## h264_qpel ## SIZE ## _h_lowpass_l2_ ## MMX(dst, src, src, stride, stride);\
279 }\
280 \
281 static void OPNAME ## h264_qpel ## SIZE ## _mc20_ ## MMX(uint8_t *dst, uint8_t *src, ptrdiff_t stride)\
282 {\
283     ff_ ## OPNAME ## h264_qpel ## SIZE ## _h_lowpass_ ## MMX(dst, src, stride, stride);\
284 }\
285 \
286 static void OPNAME ## h264_qpel ## SIZE ## _mc30_ ## MMX(uint8_t *dst, uint8_t *src, ptrdiff_t stride)\
287 {\
288     ff_ ## OPNAME ## h264_qpel ## SIZE ## _h_lowpass_l2_ ## MMX(dst, src, src+1, stride, stride);\
289 }\
290
291 #define H264_MC_V(OPNAME, SIZE, MMX, ALIGN) \
292 static void OPNAME ## h264_qpel ## SIZE ## _mc01_ ## MMX(uint8_t *dst, uint8_t *src, ptrdiff_t stride)\
293 {\
294     DECLARE_ALIGNED(ALIGN, uint8_t, temp)[SIZE*SIZE];\
295     ff_put_h264_qpel ## SIZE ## _v_lowpass_ ## MMX(temp, src, SIZE, stride);\
296     ff_ ## OPNAME ## pixels ## SIZE ## _l2_ ## MMX(dst, src, temp, stride, stride, SIZE);\
297 }\
298 \
299 static void OPNAME ## h264_qpel ## SIZE ## _mc02_ ## MMX(uint8_t *dst, uint8_t *src, ptrdiff_t stride)\
300 {\
301     ff_ ## OPNAME ## h264_qpel ## SIZE ## _v_lowpass_ ## MMX(dst, src, stride, stride);\
302 }\
303 \
304 static void OPNAME ## h264_qpel ## SIZE ## _mc03_ ## MMX(uint8_t *dst, uint8_t *src, ptrdiff_t stride)\
305 {\
306     DECLARE_ALIGNED(ALIGN, uint8_t, temp)[SIZE*SIZE];\
307     ff_put_h264_qpel ## SIZE ## _v_lowpass_ ## MMX(temp, src, SIZE, stride);\
308     ff_ ## OPNAME ## pixels ## SIZE ## _l2_ ## MMX(dst, src+stride, temp, stride, stride, SIZE);\
309 }\
310
311 #define H264_MC_HV(OPNAME, SIZE, MMX, ALIGN) \
312 static void OPNAME ## h264_qpel ## SIZE ## _mc11_ ## MMX(uint8_t *dst, uint8_t *src, ptrdiff_t stride)\
313 {\
314     DECLARE_ALIGNED(ALIGN, uint8_t, temp)[SIZE*SIZE];\
315     ff_put_h264_qpel ## SIZE ## _v_lowpass_ ## MMX(temp, src, SIZE, stride);\
316     ff_ ## OPNAME ## h264_qpel ## SIZE ## _h_lowpass_l2_ ## MMX(dst, src, temp, stride, SIZE);\
317 }\
318 \
319 static void OPNAME ## h264_qpel ## SIZE ## _mc31_ ## MMX(uint8_t *dst, uint8_t *src, ptrdiff_t stride)\
320 {\
321     DECLARE_ALIGNED(ALIGN, uint8_t, temp)[SIZE*SIZE];\
322     ff_put_h264_qpel ## SIZE ## _v_lowpass_ ## MMX(temp, src+1, SIZE, stride);\
323     ff_ ## OPNAME ## h264_qpel ## SIZE ## _h_lowpass_l2_ ## MMX(dst, src, temp, stride, SIZE);\
324 }\
325 \
326 static void OPNAME ## h264_qpel ## SIZE ## _mc13_ ## MMX(uint8_t *dst, uint8_t *src, ptrdiff_t stride)\
327 {\
328     DECLARE_ALIGNED(ALIGN, uint8_t, temp)[SIZE*SIZE];\
329     ff_put_h264_qpel ## SIZE ## _v_lowpass_ ## MMX(temp, src, SIZE, stride);\
330     ff_ ## OPNAME ## h264_qpel ## SIZE ## _h_lowpass_l2_ ## MMX(dst, src+stride, temp, stride, SIZE);\
331 }\
332 \
333 static void OPNAME ## h264_qpel ## SIZE ## _mc33_ ## MMX(uint8_t *dst, uint8_t *src, ptrdiff_t stride)\
334 {\
335     DECLARE_ALIGNED(ALIGN, uint8_t, temp)[SIZE*SIZE];\
336     ff_put_h264_qpel ## SIZE ## _v_lowpass_ ## MMX(temp, src+1, SIZE, stride);\
337     ff_ ## OPNAME ## h264_qpel ## SIZE ## _h_lowpass_l2_ ## MMX(dst, src+stride, temp, stride, SIZE);\
338 }\
339 \
340 static void OPNAME ## h264_qpel ## SIZE ## _mc22_ ## MMX(uint8_t *dst, uint8_t *src, ptrdiff_t stride)\
341 {\
342     DECLARE_ALIGNED(ALIGN, uint16_t, temp)[SIZE*(SIZE<8?12:24)];\
343     ff_ ## OPNAME ## h264_qpel ## SIZE ## _hv_lowpass_ ## MMX(dst, temp, src, stride, SIZE, stride);\
344 }\
345 \
346 static void OPNAME ## h264_qpel ## SIZE ## _mc21_ ## MMX(uint8_t *dst, uint8_t *src, ptrdiff_t stride)\
347 {\
348     DECLARE_ALIGNED(ALIGN, uint8_t, temp)[SIZE*(SIZE<8?12:24)*2 + SIZE*SIZE];\
349     uint8_t * const halfHV= temp;\
350     int16_t * const halfV= (int16_t*)(temp + SIZE*SIZE);\
351     assert(((int)temp & 7) == 0);\
352     ff_put_h264_qpel ## SIZE ## _hv_lowpass_ ## MMX(halfHV, halfV, src, SIZE, SIZE, stride);\
353     ff_ ## OPNAME ## h264_qpel ## SIZE ## _h_lowpass_l2_ ## MMX(dst, src, halfHV, stride, SIZE);\
354 }\
355 \
356 static void OPNAME ## h264_qpel ## SIZE ## _mc23_ ## MMX(uint8_t *dst, uint8_t *src, ptrdiff_t stride)\
357 {\
358     DECLARE_ALIGNED(ALIGN, uint8_t, temp)[SIZE*(SIZE<8?12:24)*2 + SIZE*SIZE];\
359     uint8_t * const halfHV= temp;\
360     int16_t * const halfV= (int16_t*)(temp + SIZE*SIZE);\
361     assert(((int)temp & 7) == 0);\
362     ff_put_h264_qpel ## SIZE ## _hv_lowpass_ ## MMX(halfHV, halfV, src, SIZE, SIZE, stride);\
363     ff_ ## OPNAME ## h264_qpel ## SIZE ## _h_lowpass_l2_ ## MMX(dst, src+stride, halfHV, stride, SIZE);\
364 }\
365 \
366 static void OPNAME ## h264_qpel ## SIZE ## _mc12_ ## MMX(uint8_t *dst, uint8_t *src, ptrdiff_t stride)\
367 {\
368     DECLARE_ALIGNED(ALIGN, uint8_t, temp)[SIZE*(SIZE<8?12:24)*2 + SIZE*SIZE];\
369     uint8_t * const halfHV= temp;\
370     int16_t * const halfV= (int16_t*)(temp + SIZE*SIZE);\
371     assert(((int)temp & 7) == 0);\
372     ff_put_h264_qpel ## SIZE ## _hv_lowpass_ ## MMX(halfHV, halfV, src, SIZE, SIZE, stride);\
373     ff_ ## OPNAME ## pixels ## SIZE ## _l2_shift5_mmxext(dst, halfV+2, halfHV, stride, SIZE, SIZE);\
374 }\
375 \
376 static void OPNAME ## h264_qpel ## SIZE ## _mc32_ ## MMX(uint8_t *dst, uint8_t *src, ptrdiff_t stride)\
377 {\
378     DECLARE_ALIGNED(ALIGN, uint8_t, temp)[SIZE*(SIZE<8?12:24)*2 + SIZE*SIZE];\
379     uint8_t * const halfHV= temp;\
380     int16_t * const halfV= (int16_t*)(temp + SIZE*SIZE);\
381     assert(((int)temp & 7) == 0);\
382     ff_put_h264_qpel ## SIZE ## _hv_lowpass_ ## MMX(halfHV, halfV, src, SIZE, SIZE, stride);\
383     ff_ ## OPNAME ## pixels ## SIZE ## _l2_shift5_mmxext(dst, halfV+3, halfHV, stride, SIZE, SIZE);\
384 }\
385
386 #define H264_MC_4816(MMX)\
387 H264_MC(put_, 4, MMX, 8)\
388 H264_MC(put_, 8, MMX, 8)\
389 H264_MC(put_, 16,MMX, 8)\
390 H264_MC(avg_, 4, MMX, 8)\
391 H264_MC(avg_, 8, MMX, 8)\
392 H264_MC(avg_, 16,MMX, 8)\
393
394 #define H264_MC_816(QPEL, XMM)\
395 QPEL(put_, 8, XMM, 16)\
396 QPEL(put_, 16,XMM, 16)\
397 QPEL(avg_, 8, XMM, 16)\
398 QPEL(avg_, 16,XMM, 16)\
399
400 #undef PAVGB
401 #define PAVGB "pavgb"
402 QPEL_H264(put_,        PUT_OP, mmxext)
403 QPEL_H264(avg_, AVG_MMXEXT_OP, mmxext)
404 QPEL_H264_V_XMM(put_,       PUT_OP, sse2)
405 QPEL_H264_V_XMM(avg_,AVG_MMXEXT_OP, sse2)
406 QPEL_H264_HV_XMM(put_,       PUT_OP, sse2)
407 QPEL_H264_HV_XMM(avg_,AVG_MMXEXT_OP, sse2)
408 QPEL_H264_H_XMM(put_,       PUT_OP, ssse3)
409 QPEL_H264_H_XMM(avg_,AVG_MMXEXT_OP, ssse3)
410 QPEL_H264_HV_XMM(put_,       PUT_OP, ssse3)
411 QPEL_H264_HV_XMM(avg_,AVG_MMXEXT_OP, ssse3)
412 #undef PAVGB
413
414 H264_MC_4816(mmxext)
415 H264_MC_816(H264_MC_V, sse2)
416 H264_MC_816(H264_MC_HV, sse2)
417 H264_MC_816(H264_MC_H, ssse3)
418 H264_MC_816(H264_MC_HV, ssse3)
419
420
421 //10bit
422 #define LUMA_MC_OP(OP, NUM, DEPTH, TYPE, OPT) \
423 void ff_ ## OP ## _h264_qpel ## NUM ## _ ## TYPE ## _ ## DEPTH ## _ ## OPT \
424     (uint8_t *dst, uint8_t *src, ptrdiff_t stride);
425
426 #define LUMA_MC_ALL(DEPTH, TYPE, OPT) \
427     LUMA_MC_OP(put,  4, DEPTH, TYPE, OPT) \
428     LUMA_MC_OP(avg,  4, DEPTH, TYPE, OPT) \
429     LUMA_MC_OP(put,  8, DEPTH, TYPE, OPT) \
430     LUMA_MC_OP(avg,  8, DEPTH, TYPE, OPT) \
431     LUMA_MC_OP(put, 16, DEPTH, TYPE, OPT) \
432     LUMA_MC_OP(avg, 16, DEPTH, TYPE, OPT)
433
434 #define LUMA_MC_816(DEPTH, TYPE, OPT) \
435     LUMA_MC_OP(put,  8, DEPTH, TYPE, OPT) \
436     LUMA_MC_OP(avg,  8, DEPTH, TYPE, OPT) \
437     LUMA_MC_OP(put, 16, DEPTH, TYPE, OPT) \
438     LUMA_MC_OP(avg, 16, DEPTH, TYPE, OPT)
439
440 LUMA_MC_ALL(10, mc00, mmxext)
441 LUMA_MC_ALL(10, mc10, mmxext)
442 LUMA_MC_ALL(10, mc20, mmxext)
443 LUMA_MC_ALL(10, mc30, mmxext)
444 LUMA_MC_ALL(10, mc01, mmxext)
445 LUMA_MC_ALL(10, mc11, mmxext)
446 LUMA_MC_ALL(10, mc21, mmxext)
447 LUMA_MC_ALL(10, mc31, mmxext)
448 LUMA_MC_ALL(10, mc02, mmxext)
449 LUMA_MC_ALL(10, mc12, mmxext)
450 LUMA_MC_ALL(10, mc22, mmxext)
451 LUMA_MC_ALL(10, mc32, mmxext)
452 LUMA_MC_ALL(10, mc03, mmxext)
453 LUMA_MC_ALL(10, mc13, mmxext)
454 LUMA_MC_ALL(10, mc23, mmxext)
455 LUMA_MC_ALL(10, mc33, mmxext)
456
457 LUMA_MC_816(10, mc00, sse2)
458 LUMA_MC_816(10, mc10, sse2)
459 LUMA_MC_816(10, mc10, sse2_cache64)
460 LUMA_MC_816(10, mc10, ssse3_cache64)
461 LUMA_MC_816(10, mc20, sse2)
462 LUMA_MC_816(10, mc20, sse2_cache64)
463 LUMA_MC_816(10, mc20, ssse3_cache64)
464 LUMA_MC_816(10, mc30, sse2)
465 LUMA_MC_816(10, mc30, sse2_cache64)
466 LUMA_MC_816(10, mc30, ssse3_cache64)
467 LUMA_MC_816(10, mc01, sse2)
468 LUMA_MC_816(10, mc11, sse2)
469 LUMA_MC_816(10, mc21, sse2)
470 LUMA_MC_816(10, mc31, sse2)
471 LUMA_MC_816(10, mc02, sse2)
472 LUMA_MC_816(10, mc12, sse2)
473 LUMA_MC_816(10, mc22, sse2)
474 LUMA_MC_816(10, mc32, sse2)
475 LUMA_MC_816(10, mc03, sse2)
476 LUMA_MC_816(10, mc13, sse2)
477 LUMA_MC_816(10, mc23, sse2)
478 LUMA_MC_816(10, mc33, sse2)
479
480 #define QPEL16_OPMC(OP, MC, MMX)\
481 void ff_ ## OP ## _h264_qpel16_ ## MC ## _10_ ## MMX(uint8_t *dst, uint8_t *src, ptrdiff_t stride){\
482     ff_ ## OP ## _h264_qpel8_ ## MC ## _10_ ## MMX(dst   , src   , stride);\
483     ff_ ## OP ## _h264_qpel8_ ## MC ## _10_ ## MMX(dst+16, src+16, stride);\
484     src += 8*stride;\
485     dst += 8*stride;\
486     ff_ ## OP ## _h264_qpel8_ ## MC ## _10_ ## MMX(dst   , src   , stride);\
487     ff_ ## OP ## _h264_qpel8_ ## MC ## _10_ ## MMX(dst+16, src+16, stride);\
488 }
489
490 #define QPEL16_OP(MC, MMX)\
491 QPEL16_OPMC(put, MC, MMX)\
492 QPEL16_OPMC(avg, MC, MMX)
493
494 #define QPEL16(MMX)\
495 QPEL16_OP(mc00, MMX)\
496 QPEL16_OP(mc01, MMX)\
497 QPEL16_OP(mc02, MMX)\
498 QPEL16_OP(mc03, MMX)\
499 QPEL16_OP(mc10, MMX)\
500 QPEL16_OP(mc11, MMX)\
501 QPEL16_OP(mc12, MMX)\
502 QPEL16_OP(mc13, MMX)\
503 QPEL16_OP(mc20, MMX)\
504 QPEL16_OP(mc21, MMX)\
505 QPEL16_OP(mc22, MMX)\
506 QPEL16_OP(mc23, MMX)\
507 QPEL16_OP(mc30, MMX)\
508 QPEL16_OP(mc31, MMX)\
509 QPEL16_OP(mc32, MMX)\
510 QPEL16_OP(mc33, MMX)
511
512 #if ARCH_X86_32 && HAVE_YASM && CONFIG_H264QPEL // ARCH_X86_64 implies SSE2+
513 QPEL16(mmxext)
514 #endif
515
516 #endif /* HAVE_YASM */
517
518 #define SET_QPEL_FUNCS(PFX, IDX, SIZE, CPU, PREFIX)                          \
519     do {                                                                     \
520     c->PFX ## _pixels_tab[IDX][ 0] = PREFIX ## PFX ## SIZE ## _mc00_ ## CPU; \
521     c->PFX ## _pixels_tab[IDX][ 1] = PREFIX ## PFX ## SIZE ## _mc10_ ## CPU; \
522     c->PFX ## _pixels_tab[IDX][ 2] = PREFIX ## PFX ## SIZE ## _mc20_ ## CPU; \
523     c->PFX ## _pixels_tab[IDX][ 3] = PREFIX ## PFX ## SIZE ## _mc30_ ## CPU; \
524     c->PFX ## _pixels_tab[IDX][ 4] = PREFIX ## PFX ## SIZE ## _mc01_ ## CPU; \
525     c->PFX ## _pixels_tab[IDX][ 5] = PREFIX ## PFX ## SIZE ## _mc11_ ## CPU; \
526     c->PFX ## _pixels_tab[IDX][ 6] = PREFIX ## PFX ## SIZE ## _mc21_ ## CPU; \
527     c->PFX ## _pixels_tab[IDX][ 7] = PREFIX ## PFX ## SIZE ## _mc31_ ## CPU; \
528     c->PFX ## _pixels_tab[IDX][ 8] = PREFIX ## PFX ## SIZE ## _mc02_ ## CPU; \
529     c->PFX ## _pixels_tab[IDX][ 9] = PREFIX ## PFX ## SIZE ## _mc12_ ## CPU; \
530     c->PFX ## _pixels_tab[IDX][10] = PREFIX ## PFX ## SIZE ## _mc22_ ## CPU; \
531     c->PFX ## _pixels_tab[IDX][11] = PREFIX ## PFX ## SIZE ## _mc32_ ## CPU; \
532     c->PFX ## _pixels_tab[IDX][12] = PREFIX ## PFX ## SIZE ## _mc03_ ## CPU; \
533     c->PFX ## _pixels_tab[IDX][13] = PREFIX ## PFX ## SIZE ## _mc13_ ## CPU; \
534     c->PFX ## _pixels_tab[IDX][14] = PREFIX ## PFX ## SIZE ## _mc23_ ## CPU; \
535     c->PFX ## _pixels_tab[IDX][15] = PREFIX ## PFX ## SIZE ## _mc33_ ## CPU; \
536     } while (0)
537
538 #define H264_QPEL_FUNCS(x, y, CPU)                                                            \
539     do {                                                                                      \
540         c->put_h264_qpel_pixels_tab[0][x + y * 4] = put_h264_qpel16_mc ## x ## y ## _ ## CPU; \
541         c->put_h264_qpel_pixels_tab[1][x + y * 4] = put_h264_qpel8_mc  ## x ## y ## _ ## CPU; \
542         c->avg_h264_qpel_pixels_tab[0][x + y * 4] = avg_h264_qpel16_mc ## x ## y ## _ ## CPU; \
543         c->avg_h264_qpel_pixels_tab[1][x + y * 4] = avg_h264_qpel8_mc  ## x ## y ## _ ## CPU; \
544     } while (0)
545
546 #define H264_QPEL_FUNCS_10(x, y, CPU)                                                               \
547     do {                                                                                            \
548         c->put_h264_qpel_pixels_tab[0][x + y * 4] = ff_put_h264_qpel16_mc ## x ## y ## _10_ ## CPU; \
549         c->put_h264_qpel_pixels_tab[1][x + y * 4] = ff_put_h264_qpel8_mc  ## x ## y ## _10_ ## CPU; \
550         c->avg_h264_qpel_pixels_tab[0][x + y * 4] = ff_avg_h264_qpel16_mc ## x ## y ## _10_ ## CPU; \
551         c->avg_h264_qpel_pixels_tab[1][x + y * 4] = ff_avg_h264_qpel8_mc  ## x ## y ## _10_ ## CPU; \
552     } while (0)
553
554 av_cold void ff_h264qpel_init_x86(H264QpelContext *c, int bit_depth)
555 {
556 #if HAVE_YASM
557     int high_bit_depth = bit_depth > 8;
558     int mm_flags = av_get_cpu_flags();
559
560     if (EXTERNAL_MMXEXT(mm_flags)) {
561         if (!high_bit_depth) {
562             SET_QPEL_FUNCS(put_h264_qpel, 0, 16, mmxext, );
563             SET_QPEL_FUNCS(put_h264_qpel, 1,  8, mmxext, );
564             SET_QPEL_FUNCS(put_h264_qpel, 2,  4, mmxext, );
565             SET_QPEL_FUNCS(avg_h264_qpel, 0, 16, mmxext, );
566             SET_QPEL_FUNCS(avg_h264_qpel, 1,  8, mmxext, );
567             SET_QPEL_FUNCS(avg_h264_qpel, 2,  4, mmxext, );
568         } else if (bit_depth == 10) {
569 #if ARCH_X86_32
570             SET_QPEL_FUNCS(avg_h264_qpel, 0, 16, 10_mmxext, ff_);
571             SET_QPEL_FUNCS(put_h264_qpel, 0, 16, 10_mmxext, ff_);
572             SET_QPEL_FUNCS(put_h264_qpel, 1,  8, 10_mmxext, ff_);
573             SET_QPEL_FUNCS(avg_h264_qpel, 1,  8, 10_mmxext, ff_);
574 #endif
575             SET_QPEL_FUNCS(put_h264_qpel, 2, 4,  10_mmxext, ff_);
576             SET_QPEL_FUNCS(avg_h264_qpel, 2, 4,  10_mmxext, ff_);
577         }
578     }
579
580     if (EXTERNAL_SSE2(mm_flags)) {
581         if (!(mm_flags & AV_CPU_FLAG_SSE2SLOW) && !high_bit_depth) {
582             // these functions are slower than mmx on AMD, but faster on Intel
583             H264_QPEL_FUNCS(0, 0, sse2);
584         }
585
586         if (!high_bit_depth) {
587             H264_QPEL_FUNCS(0, 1, sse2);
588             H264_QPEL_FUNCS(0, 2, sse2);
589             H264_QPEL_FUNCS(0, 3, sse2);
590             H264_QPEL_FUNCS(1, 1, sse2);
591             H264_QPEL_FUNCS(1, 2, sse2);
592             H264_QPEL_FUNCS(1, 3, sse2);
593             H264_QPEL_FUNCS(2, 1, sse2);
594             H264_QPEL_FUNCS(2, 2, sse2);
595             H264_QPEL_FUNCS(2, 3, sse2);
596             H264_QPEL_FUNCS(3, 1, sse2);
597             H264_QPEL_FUNCS(3, 2, sse2);
598             H264_QPEL_FUNCS(3, 3, sse2);
599         }
600
601         if (bit_depth == 10) {
602             SET_QPEL_FUNCS(put_h264_qpel, 0, 16, 10_sse2, ff_);
603             SET_QPEL_FUNCS(put_h264_qpel, 1,  8, 10_sse2, ff_);
604             SET_QPEL_FUNCS(avg_h264_qpel, 0, 16, 10_sse2, ff_);
605             SET_QPEL_FUNCS(avg_h264_qpel, 1,  8, 10_sse2, ff_);
606             H264_QPEL_FUNCS_10(1, 0, sse2_cache64);
607             H264_QPEL_FUNCS_10(2, 0, sse2_cache64);
608             H264_QPEL_FUNCS_10(3, 0, sse2_cache64);
609         }
610     }
611
612     if (EXTERNAL_SSSE3(mm_flags)) {
613         if (!high_bit_depth) {
614             H264_QPEL_FUNCS(1, 0, ssse3);
615             H264_QPEL_FUNCS(1, 1, ssse3);
616             H264_QPEL_FUNCS(1, 2, ssse3);
617             H264_QPEL_FUNCS(1, 3, ssse3);
618             H264_QPEL_FUNCS(2, 0, ssse3);
619             H264_QPEL_FUNCS(2, 1, ssse3);
620             H264_QPEL_FUNCS(2, 2, ssse3);
621             H264_QPEL_FUNCS(2, 3, ssse3);
622             H264_QPEL_FUNCS(3, 0, ssse3);
623             H264_QPEL_FUNCS(3, 1, ssse3);
624             H264_QPEL_FUNCS(3, 2, ssse3);
625             H264_QPEL_FUNCS(3, 3, ssse3);
626         }
627
628         if (bit_depth == 10) {
629             H264_QPEL_FUNCS_10(1, 0, ssse3_cache64);
630             H264_QPEL_FUNCS_10(2, 0, ssse3_cache64);
631             H264_QPEL_FUNCS_10(3, 0, ssse3_cache64);
632         }
633     }
634
635     if (EXTERNAL_AVX(mm_flags)) {
636         /* AVX implies 64 byte cache lines without the need to avoid unaligned
637          * memory accesses that cross the boundary between two cache lines.
638          * TODO: Port X264_CPU_CACHELINE_32/64 detection from x264 to avoid
639          * having to treat SSE2 functions with such properties as AVX. */
640         if (bit_depth == 10) {
641             H264_QPEL_FUNCS_10(1, 0, sse2);
642             H264_QPEL_FUNCS_10(2, 0, sse2);
643             H264_QPEL_FUNCS_10(3, 0, sse2);
644         }
645     }
646 #endif
647 }