x86: dsputil: Move cavs and vc1-specific functions where they belong
[ffmpeg.git] / libavcodec / x86 / cavsdsp.c
1 /*
2  * Chinese AVS video (AVS1-P2, JiZhun profile) decoder.
3  * Copyright (c) 2006  Stefan Gehrer <stefan.gehrer@gmx.de>
4  *
5  * MMX-optimized DSP functions, based on H.264 optimizations by
6  * Michael Niedermayer and Loren Merritt
7  *
8  * This file is part of Libav.
9  *
10  * Libav is free software; you can redistribute it and/or
11  * modify it under the terms of the GNU Lesser General Public
12  * License as published by the Free Software Foundation; either
13  * version 2.1 of the License, or (at your option) any later version.
14  *
15  * Libav is distributed in the hope that it will be useful,
16  * but WITHOUT ANY WARRANTY; without even the implied warranty of
17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
18  * Lesser General Public License for more details.
19  *
20  * You should have received a copy of the GNU Lesser General Public
21  * License along with Libav; if not, write to the Free Software
22  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
23  */
24
25 #include "libavutil/attributes.h"
26 #include "libavutil/common.h"
27 #include "libavutil/cpu.h"
28 #include "libavutil/x86/asm.h"
29 #include "libavutil/x86/cpu.h"
30 #include "libavcodec/cavsdsp.h"
31 #include "dsputil_mmx.h"
32 #include "config.h"
33
34 #if HAVE_MMX_INLINE
35
36 /* in/out: mma=mma+mmb, mmb=mmb-mma */
37 #define SUMSUB_BA( a, b ) \
38     "paddw "#b", "#a" \n\t"\
39     "paddw "#b", "#b" \n\t"\
40     "psubw "#a", "#b" \n\t"
41
42 /*****************************************************************************
43  *
44  * inverse transform
45  *
46  ****************************************************************************/
47
48 static inline void cavs_idct8_1d(int16_t *block, uint64_t bias)
49 {
50     __asm__ volatile(
51         "movq 112(%0), %%mm4  \n\t" /* mm4 = src7 */
52         "movq  16(%0), %%mm5  \n\t" /* mm5 = src1 */
53         "movq  80(%0), %%mm2  \n\t" /* mm2 = src5 */
54         "movq  48(%0), %%mm7  \n\t" /* mm7 = src3 */
55         "movq   %%mm4, %%mm0  \n\t"
56         "movq   %%mm5, %%mm3  \n\t"
57         "movq   %%mm2, %%mm6  \n\t"
58         "movq   %%mm7, %%mm1  \n\t"
59
60         "paddw  %%mm4, %%mm4  \n\t" /* mm4 = 2*src7 */
61         "paddw  %%mm3, %%mm3  \n\t" /* mm3 = 2*src1 */
62         "paddw  %%mm6, %%mm6  \n\t" /* mm6 = 2*src5 */
63         "paddw  %%mm1, %%mm1  \n\t" /* mm1 = 2*src3 */
64         "paddw  %%mm4, %%mm0  \n\t" /* mm0 = 3*src7 */
65         "paddw  %%mm3, %%mm5  \n\t" /* mm5 = 3*src1 */
66         "paddw  %%mm6, %%mm2  \n\t" /* mm2 = 3*src5 */
67         "paddw  %%mm1, %%mm7  \n\t" /* mm7 = 3*src3 */
68         "psubw  %%mm4, %%mm5  \n\t" /* mm5 = 3*src1 - 2*src7 = a0 */
69         "paddw  %%mm6, %%mm7  \n\t" /* mm7 = 3*src3 + 2*src5 = a1 */
70         "psubw  %%mm2, %%mm1  \n\t" /* mm1 = 2*src3 - 3*src5 = a2 */
71         "paddw  %%mm0, %%mm3  \n\t" /* mm3 = 2*src1 + 3*src7 = a3 */
72
73         "movq   %%mm5, %%mm4  \n\t"
74         "movq   %%mm7, %%mm6  \n\t"
75         "movq   %%mm3, %%mm0  \n\t"
76         "movq   %%mm1, %%mm2  \n\t"
77         SUMSUB_BA( %%mm7, %%mm5 )   /* mm7 = a0 + a1  mm5 = a0 - a1 */
78         "paddw  %%mm3, %%mm7  \n\t" /* mm7 = a0 + a1 + a3 */
79         "paddw  %%mm1, %%mm5  \n\t" /* mm5 = a0 - a1 + a2 */
80         "paddw  %%mm7, %%mm7  \n\t"
81         "paddw  %%mm5, %%mm5  \n\t"
82         "paddw  %%mm6, %%mm7  \n\t" /* mm7 = b4 */
83         "paddw  %%mm4, %%mm5  \n\t" /* mm5 = b5 */
84
85         SUMSUB_BA( %%mm1, %%mm3 )   /* mm1 = a3 + a2  mm3 = a3 - a2 */
86         "psubw  %%mm1, %%mm4  \n\t" /* mm4 = a0 - a2 - a3 */
87         "movq   %%mm4, %%mm1  \n\t" /* mm1 = a0 - a2 - a3 */
88         "psubw  %%mm6, %%mm3  \n\t" /* mm3 = a3 - a2 - a1 */
89         "paddw  %%mm1, %%mm1  \n\t"
90         "paddw  %%mm3, %%mm3  \n\t"
91         "psubw  %%mm2, %%mm1  \n\t" /* mm1 = b7 */
92         "paddw  %%mm0, %%mm3  \n\t" /* mm3 = b6 */
93
94         "movq  32(%0), %%mm2  \n\t" /* mm2 = src2 */
95         "movq  96(%0), %%mm6  \n\t" /* mm6 = src6 */
96         "movq   %%mm2, %%mm4  \n\t"
97         "movq   %%mm6, %%mm0  \n\t"
98         "psllw  $2,    %%mm4  \n\t" /* mm4 = 4*src2 */
99         "psllw  $2,    %%mm6  \n\t" /* mm6 = 4*src6 */
100         "paddw  %%mm4, %%mm2  \n\t" /* mm2 = 5*src2 */
101         "paddw  %%mm6, %%mm0  \n\t" /* mm0 = 5*src6 */
102         "paddw  %%mm2, %%mm2  \n\t"
103         "paddw  %%mm0, %%mm0  \n\t"
104         "psubw  %%mm0, %%mm4  \n\t" /* mm4 = 4*src2 - 10*src6 = a7 */
105         "paddw  %%mm2, %%mm6  \n\t" /* mm6 = 4*src6 + 10*src2 = a6 */
106
107         "movq    (%0), %%mm2  \n\t" /* mm2 = src0 */
108         "movq  64(%0), %%mm0  \n\t" /* mm0 = src4 */
109         SUMSUB_BA( %%mm0, %%mm2 )   /* mm0 = src0+src4  mm2 = src0-src4 */
110         "psllw  $3,    %%mm0  \n\t"
111         "psllw  $3,    %%mm2  \n\t"
112         "paddw  %1,    %%mm0  \n\t" /* add rounding bias */
113         "paddw  %1,    %%mm2  \n\t" /* add rounding bias */
114
115         SUMSUB_BA( %%mm6, %%mm0 )   /* mm6 = a4 + a6  mm0 = a4 - a6 */
116         SUMSUB_BA( %%mm4, %%mm2 )   /* mm4 = a5 + a7  mm2 = a5 - a7 */
117         SUMSUB_BA( %%mm7, %%mm6 )   /* mm7 = dst0  mm6 = dst7 */
118         SUMSUB_BA( %%mm5, %%mm4 )   /* mm5 = dst1  mm4 = dst6 */
119         SUMSUB_BA( %%mm3, %%mm2 )   /* mm3 = dst2  mm2 = dst5 */
120         SUMSUB_BA( %%mm1, %%mm0 )   /* mm1 = dst3  mm0 = dst4 */
121         :: "r"(block), "m"(bias)
122     );
123 }
124
125 static void cavs_idct8_add_mmx(uint8_t *dst, int16_t *block, int stride)
126 {
127     int i;
128     DECLARE_ALIGNED(8, int16_t, b2)[64];
129
130     for(i=0; i<2; i++){
131         DECLARE_ALIGNED(8, uint64_t, tmp);
132
133         cavs_idct8_1d(block+4*i, ff_pw_4.a);
134
135         __asm__ volatile(
136             "psraw     $3, %%mm7  \n\t"
137             "psraw     $3, %%mm6  \n\t"
138             "psraw     $3, %%mm5  \n\t"
139             "psraw     $3, %%mm4  \n\t"
140             "psraw     $3, %%mm3  \n\t"
141             "psraw     $3, %%mm2  \n\t"
142             "psraw     $3, %%mm1  \n\t"
143             "psraw     $3, %%mm0  \n\t"
144             "movq   %%mm7,    %0   \n\t"
145             TRANSPOSE4( %%mm0, %%mm2, %%mm4, %%mm6, %%mm7 )
146             "movq   %%mm0,  8(%1)  \n\t"
147             "movq   %%mm6, 24(%1)  \n\t"
148             "movq   %%mm7, 40(%1)  \n\t"
149             "movq   %%mm4, 56(%1)  \n\t"
150             "movq    %0,    %%mm7  \n\t"
151             TRANSPOSE4( %%mm7, %%mm5, %%mm3, %%mm1, %%mm0 )
152             "movq   %%mm7,   (%1)  \n\t"
153             "movq   %%mm1, 16(%1)  \n\t"
154             "movq   %%mm0, 32(%1)  \n\t"
155             "movq   %%mm3, 48(%1)  \n\t"
156             : "=m"(tmp)
157             : "r"(b2+32*i)
158             : "memory"
159         );
160     }
161
162     for(i=0; i<2; i++){
163         cavs_idct8_1d(b2+4*i, ff_pw_64.a);
164
165         __asm__ volatile(
166             "psraw     $7, %%mm7  \n\t"
167             "psraw     $7, %%mm6  \n\t"
168             "psraw     $7, %%mm5  \n\t"
169             "psraw     $7, %%mm4  \n\t"
170             "psraw     $7, %%mm3  \n\t"
171             "psraw     $7, %%mm2  \n\t"
172             "psraw     $7, %%mm1  \n\t"
173             "psraw     $7, %%mm0  \n\t"
174             "movq   %%mm7,    (%0)  \n\t"
175             "movq   %%mm5,  16(%0)  \n\t"
176             "movq   %%mm3,  32(%0)  \n\t"
177             "movq   %%mm1,  48(%0)  \n\t"
178             "movq   %%mm0,  64(%0)  \n\t"
179             "movq   %%mm2,  80(%0)  \n\t"
180             "movq   %%mm4,  96(%0)  \n\t"
181             "movq   %%mm6, 112(%0)  \n\t"
182             :: "r"(b2+4*i)
183             : "memory"
184         );
185     }
186
187     ff_add_pixels_clamped_mmx(b2, dst, stride);
188 }
189
190 #endif /* HAVE_MMX_INLINE */
191
192 #if (HAVE_MMXEXT_INLINE || HAVE_AMD3DNOW_INLINE)
193
194 /*****************************************************************************
195  *
196  * motion compensation
197  *
198  ****************************************************************************/
199
200 /* vertical filter [-1 -2 96 42 -7  0]  */
201 #define QPEL_CAVSV1(A,B,C,D,E,F,OP,MUL2) \
202         "movd (%0), "#F"            \n\t"\
203         "movq "#C", %%mm6           \n\t"\
204         "pmullw %5, %%mm6           \n\t"\
205         "movq "#D", %%mm7           \n\t"\
206         "pmullw "MANGLE(MUL2)", %%mm7\n\t"\
207         "psllw $3, "#E"             \n\t"\
208         "psubw "#E", %%mm6          \n\t"\
209         "psraw $3, "#E"             \n\t"\
210         "paddw %%mm7, %%mm6         \n\t"\
211         "paddw "#E", %%mm6          \n\t"\
212         "paddw "#B", "#B"           \n\t"\
213         "pxor %%mm7, %%mm7          \n\t"\
214         "add %2, %0                 \n\t"\
215         "punpcklbw %%mm7, "#F"      \n\t"\
216         "psubw "#B", %%mm6          \n\t"\
217         "psraw $1, "#B"             \n\t"\
218         "psubw "#A", %%mm6          \n\t"\
219         "paddw %4, %%mm6            \n\t"\
220         "psraw $7, %%mm6            \n\t"\
221         "packuswb %%mm6, %%mm6      \n\t"\
222         OP(%%mm6, (%1), A, d)            \
223         "add %3, %1                 \n\t"
224
225 /* vertical filter [ 0 -1  5  5 -1  0]  */
226 #define QPEL_CAVSV2(A,B,C,D,E,F,OP,MUL2) \
227         "movd (%0), "#F"            \n\t"\
228         "movq "#C", %%mm6           \n\t"\
229         "paddw "#D", %%mm6          \n\t"\
230         "pmullw %5, %%mm6           \n\t"\
231         "add %2, %0                 \n\t"\
232         "punpcklbw %%mm7, "#F"      \n\t"\
233         "psubw "#B", %%mm6          \n\t"\
234         "psubw "#E", %%mm6          \n\t"\
235         "paddw %4, %%mm6            \n\t"\
236         "psraw $3, %%mm6            \n\t"\
237         "packuswb %%mm6, %%mm6      \n\t"\
238         OP(%%mm6, (%1), A, d)            \
239         "add %3, %1                 \n\t"
240
241 /* vertical filter [ 0 -7 42 96 -2 -1]  */
242 #define QPEL_CAVSV3(A,B,C,D,E,F,OP,MUL2) \
243         "movd (%0), "#F"            \n\t"\
244         "movq "#C", %%mm6           \n\t"\
245         "pmullw "MANGLE(MUL2)", %%mm6\n\t"\
246         "movq "#D", %%mm7           \n\t"\
247         "pmullw %5, %%mm7           \n\t"\
248         "psllw $3, "#B"             \n\t"\
249         "psubw "#B", %%mm6          \n\t"\
250         "psraw $3, "#B"             \n\t"\
251         "paddw %%mm7, %%mm6         \n\t"\
252         "paddw "#B", %%mm6          \n\t"\
253         "paddw "#E", "#E"           \n\t"\
254         "pxor %%mm7, %%mm7          \n\t"\
255         "add %2, %0                 \n\t"\
256         "punpcklbw %%mm7, "#F"      \n\t"\
257         "psubw "#E", %%mm6          \n\t"\
258         "psraw $1, "#E"             \n\t"\
259         "psubw "#F", %%mm6          \n\t"\
260         "paddw %4, %%mm6            \n\t"\
261         "psraw $7, %%mm6            \n\t"\
262         "packuswb %%mm6, %%mm6      \n\t"\
263         OP(%%mm6, (%1), A, d)            \
264         "add %3, %1                 \n\t"
265
266
267 #define QPEL_CAVSVNUM(VOP,OP,ADD,MUL1,MUL2)\
268     int w= 2;\
269     src -= 2*srcStride;\
270     \
271     while(w--){\
272       __asm__ volatile(\
273         "pxor %%mm7, %%mm7          \n\t"\
274         "movd (%0), %%mm0           \n\t"\
275         "add %2, %0                 \n\t"\
276         "movd (%0), %%mm1           \n\t"\
277         "add %2, %0                 \n\t"\
278         "movd (%0), %%mm2           \n\t"\
279         "add %2, %0                 \n\t"\
280         "movd (%0), %%mm3           \n\t"\
281         "add %2, %0                 \n\t"\
282         "movd (%0), %%mm4           \n\t"\
283         "add %2, %0                 \n\t"\
284         "punpcklbw %%mm7, %%mm0     \n\t"\
285         "punpcklbw %%mm7, %%mm1     \n\t"\
286         "punpcklbw %%mm7, %%mm2     \n\t"\
287         "punpcklbw %%mm7, %%mm3     \n\t"\
288         "punpcklbw %%mm7, %%mm4     \n\t"\
289         VOP(%%mm0, %%mm1, %%mm2, %%mm3, %%mm4, %%mm5, OP, MUL2)\
290         VOP(%%mm1, %%mm2, %%mm3, %%mm4, %%mm5, %%mm0, OP, MUL2)\
291         VOP(%%mm2, %%mm3, %%mm4, %%mm5, %%mm0, %%mm1, OP, MUL2)\
292         VOP(%%mm3, %%mm4, %%mm5, %%mm0, %%mm1, %%mm2, OP, MUL2)\
293         VOP(%%mm4, %%mm5, %%mm0, %%mm1, %%mm2, %%mm3, OP, MUL2)\
294         VOP(%%mm5, %%mm0, %%mm1, %%mm2, %%mm3, %%mm4, OP, MUL2)\
295         VOP(%%mm0, %%mm1, %%mm2, %%mm3, %%mm4, %%mm5, OP, MUL2)\
296         VOP(%%mm1, %%mm2, %%mm3, %%mm4, %%mm5, %%mm0, OP, MUL2)\
297         \
298         : "+a"(src), "+c"(dst)\
299         : "S"((x86_reg)srcStride), "r"((x86_reg)dstStride), "m"(ADD), "m"(MUL1)\
300         : "memory"\
301      );\
302      if(h==16){\
303         __asm__ volatile(\
304             VOP(%%mm2, %%mm3, %%mm4, %%mm5, %%mm0, %%mm1, OP, MUL2)\
305             VOP(%%mm3, %%mm4, %%mm5, %%mm0, %%mm1, %%mm2, OP, MUL2)\
306             VOP(%%mm4, %%mm5, %%mm0, %%mm1, %%mm2, %%mm3, OP, MUL2)\
307             VOP(%%mm5, %%mm0, %%mm1, %%mm2, %%mm3, %%mm4, OP, MUL2)\
308             VOP(%%mm0, %%mm1, %%mm2, %%mm3, %%mm4, %%mm5, OP, MUL2)\
309             VOP(%%mm1, %%mm2, %%mm3, %%mm4, %%mm5, %%mm0, OP, MUL2)\
310             VOP(%%mm2, %%mm3, %%mm4, %%mm5, %%mm0, %%mm1, OP, MUL2)\
311             VOP(%%mm3, %%mm4, %%mm5, %%mm0, %%mm1, %%mm2, OP, MUL2)\
312             \
313            : "+a"(src), "+c"(dst)\
314            : "S"((x86_reg)srcStride), "r"((x86_reg)dstStride), "m"(ADD),  "m"(MUL1)\
315            : "memory"\
316         );\
317      }\
318      src += 4-(h+5)*srcStride;\
319      dst += 4-h*dstStride;\
320    }
321
322 #define QPEL_CAVS(OPNAME, OP, MMX)\
323 static void OPNAME ## cavs_qpel8_h_ ## MMX(uint8_t *dst, uint8_t *src, int dstStride, int srcStride){\
324     int h=8;\
325     __asm__ volatile(\
326         "pxor %%mm7, %%mm7          \n\t"\
327         "movq %5, %%mm6             \n\t"\
328         "1:                         \n\t"\
329         "movq    (%0), %%mm0        \n\t"\
330         "movq   1(%0), %%mm2        \n\t"\
331         "movq %%mm0, %%mm1          \n\t"\
332         "movq %%mm2, %%mm3          \n\t"\
333         "punpcklbw %%mm7, %%mm0     \n\t"\
334         "punpckhbw %%mm7, %%mm1     \n\t"\
335         "punpcklbw %%mm7, %%mm2     \n\t"\
336         "punpckhbw %%mm7, %%mm3     \n\t"\
337         "paddw %%mm2, %%mm0         \n\t"\
338         "paddw %%mm3, %%mm1         \n\t"\
339         "pmullw %%mm6, %%mm0        \n\t"\
340         "pmullw %%mm6, %%mm1        \n\t"\
341         "movq   -1(%0), %%mm2       \n\t"\
342         "movq    2(%0), %%mm4       \n\t"\
343         "movq %%mm2, %%mm3          \n\t"\
344         "movq %%mm4, %%mm5          \n\t"\
345         "punpcklbw %%mm7, %%mm2     \n\t"\
346         "punpckhbw %%mm7, %%mm3     \n\t"\
347         "punpcklbw %%mm7, %%mm4     \n\t"\
348         "punpckhbw %%mm7, %%mm5     \n\t"\
349         "paddw %%mm4, %%mm2         \n\t"\
350         "paddw %%mm3, %%mm5         \n\t"\
351         "psubw %%mm2, %%mm0         \n\t"\
352         "psubw %%mm5, %%mm1         \n\t"\
353         "movq %6, %%mm5             \n\t"\
354         "paddw %%mm5, %%mm0         \n\t"\
355         "paddw %%mm5, %%mm1         \n\t"\
356         "psraw $3, %%mm0            \n\t"\
357         "psraw $3, %%mm1            \n\t"\
358         "packuswb %%mm1, %%mm0      \n\t"\
359         OP(%%mm0, (%1),%%mm5, q)         \
360         "add %3, %0                 \n\t"\
361         "add %4, %1                 \n\t"\
362         "decl %2                    \n\t"\
363         " jnz 1b                    \n\t"\
364         : "+a"(src), "+c"(dst), "+m"(h)\
365         : "d"((x86_reg)srcStride), "S"((x86_reg)dstStride), "m"(ff_pw_5), "m"(ff_pw_4)\
366         : "memory"\
367     );\
368 }\
369 \
370 static inline void OPNAME ## cavs_qpel8or16_v1_ ## MMX(uint8_t *dst, uint8_t *src, int dstStride, int srcStride, int h){\
371   QPEL_CAVSVNUM(QPEL_CAVSV1,OP,ff_pw_64,ff_pw_96,ff_pw_42)      \
372 }\
373 \
374 static inline void OPNAME ## cavs_qpel8or16_v2_ ## MMX(uint8_t *dst, uint8_t *src, int dstStride, int srcStride, int h){\
375   QPEL_CAVSVNUM(QPEL_CAVSV2,OP,ff_pw_4,ff_pw_5,ff_pw_5)         \
376 }\
377 \
378 static inline void OPNAME ## cavs_qpel8or16_v3_ ## MMX(uint8_t *dst, uint8_t *src, int dstStride, int srcStride, int h){\
379   QPEL_CAVSVNUM(QPEL_CAVSV3,OP,ff_pw_64,ff_pw_96,ff_pw_42)      \
380 }\
381 \
382 static void OPNAME ## cavs_qpel8_v1_ ## MMX(uint8_t *dst, uint8_t *src, int dstStride, int srcStride){\
383     OPNAME ## cavs_qpel8or16_v1_ ## MMX(dst  , src  , dstStride, srcStride, 8);\
384 }\
385 static void OPNAME ## cavs_qpel16_v1_ ## MMX(uint8_t *dst, uint8_t *src, int dstStride, int srcStride){\
386     OPNAME ## cavs_qpel8or16_v1_ ## MMX(dst  , src  , dstStride, srcStride, 16);\
387     OPNAME ## cavs_qpel8or16_v1_ ## MMX(dst+8, src+8, dstStride, srcStride, 16);\
388 }\
389 \
390 static void OPNAME ## cavs_qpel8_v2_ ## MMX(uint8_t *dst, uint8_t *src, int dstStride, int srcStride){\
391     OPNAME ## cavs_qpel8or16_v2_ ## MMX(dst  , src  , dstStride, srcStride, 8);\
392 }\
393 static void OPNAME ## cavs_qpel16_v2_ ## MMX(uint8_t *dst, uint8_t *src, int dstStride, int srcStride){\
394     OPNAME ## cavs_qpel8or16_v2_ ## MMX(dst  , src  , dstStride, srcStride, 16);\
395     OPNAME ## cavs_qpel8or16_v2_ ## MMX(dst+8, src+8, dstStride, srcStride, 16);\
396 }\
397 \
398 static void OPNAME ## cavs_qpel8_v3_ ## MMX(uint8_t *dst, uint8_t *src, int dstStride, int srcStride){\
399     OPNAME ## cavs_qpel8or16_v3_ ## MMX(dst  , src  , dstStride, srcStride, 8);\
400 }\
401 static void OPNAME ## cavs_qpel16_v3_ ## MMX(uint8_t *dst, uint8_t *src, int dstStride, int srcStride){\
402     OPNAME ## cavs_qpel8or16_v3_ ## MMX(dst  , src  , dstStride, srcStride, 16);\
403     OPNAME ## cavs_qpel8or16_v3_ ## MMX(dst+8, src+8, dstStride, srcStride, 16);\
404 }\
405 \
406 static void OPNAME ## cavs_qpel16_h_ ## MMX(uint8_t *dst, uint8_t *src, int dstStride, int srcStride){\
407     OPNAME ## cavs_qpel8_h_ ## MMX(dst  , src  , dstStride, srcStride);\
408     OPNAME ## cavs_qpel8_h_ ## MMX(dst+8, src+8, dstStride, srcStride);\
409     src += 8*srcStride;\
410     dst += 8*dstStride;\
411     OPNAME ## cavs_qpel8_h_ ## MMX(dst  , src  , dstStride, srcStride);\
412     OPNAME ## cavs_qpel8_h_ ## MMX(dst+8, src+8, dstStride, srcStride);\
413 }\
414
415 #define CAVS_MC(OPNAME, SIZE, MMX) \
416 static void OPNAME ## cavs_qpel ## SIZE ## _mc20_ ## MMX(uint8_t *dst, uint8_t *src, ptrdiff_t stride)\
417 {\
418     OPNAME ## cavs_qpel ## SIZE ## _h_ ## MMX(dst, src, stride, stride);\
419 }\
420 \
421 static void OPNAME ## cavs_qpel ## SIZE ## _mc01_ ## MMX(uint8_t *dst, uint8_t *src, ptrdiff_t stride)\
422 {\
423     OPNAME ## cavs_qpel ## SIZE ## _v1_ ## MMX(dst, src, stride, stride);\
424 }\
425 \
426 static void OPNAME ## cavs_qpel ## SIZE ## _mc02_ ## MMX(uint8_t *dst, uint8_t *src, ptrdiff_t stride)\
427 {\
428     OPNAME ## cavs_qpel ## SIZE ## _v2_ ## MMX(dst, src, stride, stride);\
429 }\
430 \
431 static void OPNAME ## cavs_qpel ## SIZE ## _mc03_ ## MMX(uint8_t *dst, uint8_t *src, ptrdiff_t stride)\
432 {\
433     OPNAME ## cavs_qpel ## SIZE ## _v3_ ## MMX(dst, src, stride, stride);\
434 }\
435
436 #define PUT_OP(a,b,temp, size) "mov" #size " " #a ", " #b "    \n\t"
437 #define AVG_3DNOW_OP(a,b,temp, size) \
438 "mov" #size " " #b ", " #temp "   \n\t"\
439 "pavgusb " #temp ", " #a "        \n\t"\
440 "mov" #size " " #a ", " #b "      \n\t"
441 #define AVG_MMXEXT_OP(a, b, temp, size) \
442 "mov" #size " " #b ", " #temp "   \n\t"\
443 "pavgb " #temp ", " #a "          \n\t"\
444 "mov" #size " " #a ", " #b "      \n\t"
445
446 #endif /* (HAVE_MMXEXT_INLINE || HAVE_AMD3DNOW_INLINE) */
447
448 #if HAVE_MMX_INLINE
449 static void put_cavs_qpel8_mc00_mmx(uint8_t *dst, uint8_t *src,
450                                     ptrdiff_t stride)
451 {
452     ff_put_pixels8_mmx(dst, src, stride, 8);
453 }
454
455 static void avg_cavs_qpel8_mc00_mmx(uint8_t *dst, uint8_t *src,
456                                     ptrdiff_t stride)
457 {
458     ff_avg_pixels8_mmx(dst, src, stride, 8);
459 }
460
461 static void put_cavs_qpel16_mc00_mmx(uint8_t *dst, uint8_t *src,
462                                      ptrdiff_t stride)
463 {
464     ff_put_pixels16_mmx(dst, src, stride, 16);
465 }
466
467 static void avg_cavs_qpel16_mc00_mmx(uint8_t *dst, uint8_t *src,
468                                      ptrdiff_t stride)
469 {
470     ff_avg_pixels16_mmx(dst, src, stride, 16);
471 }
472
473 static av_cold void cavsdsp_init_mmx(CAVSDSPContext *c,
474                                      AVCodecContext *avctx)
475 {
476     c->put_cavs_qpel_pixels_tab[0][0] = put_cavs_qpel16_mc00_mmx;
477     c->put_cavs_qpel_pixels_tab[1][0] = put_cavs_qpel8_mc00_mmx;
478     c->avg_cavs_qpel_pixels_tab[0][0] = avg_cavs_qpel16_mc00_mmx;
479     c->avg_cavs_qpel_pixels_tab[1][0] = avg_cavs_qpel8_mc00_mmx;
480
481     c->cavs_idct8_add = cavs_idct8_add_mmx;
482     c->idct_perm      = FF_TRANSPOSE_IDCT_PERM;
483 }
484 #endif /* HAVE_MMX_INLINE */
485
486 #define DSPFUNC(PFX, IDX, NUM, EXT)                                                       \
487     c->PFX ## _cavs_qpel_pixels_tab[IDX][ 2] = PFX ## _cavs_qpel ## NUM ## _mc20_ ## EXT; \
488     c->PFX ## _cavs_qpel_pixels_tab[IDX][ 4] = PFX ## _cavs_qpel ## NUM ## _mc01_ ## EXT; \
489     c->PFX ## _cavs_qpel_pixels_tab[IDX][ 8] = PFX ## _cavs_qpel ## NUM ## _mc02_ ## EXT; \
490     c->PFX ## _cavs_qpel_pixels_tab[IDX][12] = PFX ## _cavs_qpel ## NUM ## _mc03_ ## EXT; \
491
492 #if HAVE_MMXEXT_INLINE
493 QPEL_CAVS(put_,        PUT_OP, mmxext)
494 QPEL_CAVS(avg_, AVG_MMXEXT_OP, mmxext)
495
496 CAVS_MC(put_,  8, mmxext)
497 CAVS_MC(put_, 16, mmxext)
498 CAVS_MC(avg_,  8, mmxext)
499 CAVS_MC(avg_, 16, mmxext)
500
501 static av_cold void cavsdsp_init_mmxext(CAVSDSPContext *c,
502                                         AVCodecContext *avctx)
503 {
504     DSPFUNC(put, 0, 16, mmxext);
505     DSPFUNC(put, 1,  8, mmxext);
506     DSPFUNC(avg, 0, 16, mmxext);
507     DSPFUNC(avg, 1,  8, mmxext);
508 }
509 #endif /* HAVE_MMXEXT_INLINE */
510
511 #if HAVE_AMD3DNOW_INLINE
512 QPEL_CAVS(put_,       PUT_OP, 3dnow)
513 QPEL_CAVS(avg_, AVG_3DNOW_OP, 3dnow)
514
515 CAVS_MC(put_, 8, 3dnow)
516 CAVS_MC(put_, 16,3dnow)
517 CAVS_MC(avg_, 8, 3dnow)
518 CAVS_MC(avg_, 16,3dnow)
519
520 static av_cold void cavsdsp_init_3dnow(CAVSDSPContext *c,
521                                        AVCodecContext *avctx)
522 {
523     DSPFUNC(put, 0, 16, 3dnow);
524     DSPFUNC(put, 1,  8, 3dnow);
525     DSPFUNC(avg, 0, 16, 3dnow);
526     DSPFUNC(avg, 1,  8, 3dnow);
527 }
528 #endif /* HAVE_AMD3DNOW_INLINE */
529
530 av_cold void ff_cavsdsp_init_x86(CAVSDSPContext *c, AVCodecContext *avctx)
531 {
532 #if HAVE_MMX_INLINE
533     int mm_flags = av_get_cpu_flags();
534
535     if (mm_flags & AV_CPU_FLAG_MMX)
536         cavsdsp_init_mmx(c, avctx);
537 #endif /* HAVE_MMX_INLINE */
538 #if HAVE_MMXEXT_INLINE
539     if (mm_flags & AV_CPU_FLAG_MMXEXT)
540         cavsdsp_init_mmxext(c, avctx);
541 #endif /* HAVE_MMXEXT_INLINE */
542 #if HAVE_AMD3DNOW_INLINE
543     if (mm_flags & AV_CPU_FLAG_3DNOW)
544         cavsdsp_init_3dnow(c, avctx);
545 #endif /* HAVE_AMD3DNOW_INLINE */
546 }