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