]> git.sesse.net Git - ffmpeg/blob - libavcodec/x86/cavsdsp.c
Merge commit '218aefce4472dc02ee3f12830a9a894bf7916da9'
[ffmpeg] / 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 FFmpeg.
9  *
10  * FFmpeg 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  * FFmpeg 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 FFmpeg; 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/dsputil.h"
31 #include "libavcodec/cavsdsp.h"
32 #include "dsputil_mmx.h"
33 #include "config.h"
34
35 #if (HAVE_MMXEXT_INLINE || HAVE_AMD3DNOW_INLINE)
36
37 /* in/out: mma=mma+mmb, mmb=mmb-mma */
38 #define SUMSUB_BA( a, b ) \
39     "paddw "#b", "#a" \n\t"\
40     "paddw "#b", "#b" \n\t"\
41     "psubw "#a", "#b" \n\t"
42
43 /*****************************************************************************
44  *
45  * inverse transform
46  *
47  ****************************************************************************/
48
49 static inline void cavs_idct8_1d(int16_t *block, uint64_t bias)
50 {
51     __asm__ volatile(
52         "movq 112(%0), %%mm4  \n\t" /* mm4 = src7 */
53         "movq  16(%0), %%mm5  \n\t" /* mm5 = src1 */
54         "movq  80(%0), %%mm2  \n\t" /* mm2 = src5 */
55         "movq  48(%0), %%mm7  \n\t" /* mm7 = src3 */
56         "movq   %%mm4, %%mm0  \n\t"
57         "movq   %%mm5, %%mm3  \n\t"
58         "movq   %%mm2, %%mm6  \n\t"
59         "movq   %%mm7, %%mm1  \n\t"
60
61         "paddw  %%mm4, %%mm4  \n\t" /* mm4 = 2*src7 */
62         "paddw  %%mm3, %%mm3  \n\t" /* mm3 = 2*src1 */
63         "paddw  %%mm6, %%mm6  \n\t" /* mm6 = 2*src5 */
64         "paddw  %%mm1, %%mm1  \n\t" /* mm1 = 2*src3 */
65         "paddw  %%mm4, %%mm0  \n\t" /* mm0 = 3*src7 */
66         "paddw  %%mm3, %%mm5  \n\t" /* mm5 = 3*src1 */
67         "paddw  %%mm6, %%mm2  \n\t" /* mm2 = 3*src5 */
68         "paddw  %%mm1, %%mm7  \n\t" /* mm7 = 3*src3 */
69         "psubw  %%mm4, %%mm5  \n\t" /* mm5 = 3*src1 - 2*src7 = a0 */
70         "paddw  %%mm6, %%mm7  \n\t" /* mm7 = 3*src3 + 2*src5 = a1 */
71         "psubw  %%mm2, %%mm1  \n\t" /* mm1 = 2*src3 - 3*src5 = a2 */
72         "paddw  %%mm0, %%mm3  \n\t" /* mm3 = 2*src1 + 3*src7 = a3 */
73
74         "movq   %%mm5, %%mm4  \n\t"
75         "movq   %%mm7, %%mm6  \n\t"
76         "movq   %%mm3, %%mm0  \n\t"
77         "movq   %%mm1, %%mm2  \n\t"
78         SUMSUB_BA( %%mm7, %%mm5 )   /* mm7 = a0 + a1  mm5 = a0 - a1 */
79         "paddw  %%mm3, %%mm7  \n\t" /* mm7 = a0 + a1 + a3 */
80         "paddw  %%mm1, %%mm5  \n\t" /* mm5 = a0 - a1 + a2 */
81         "paddw  %%mm7, %%mm7  \n\t"
82         "paddw  %%mm5, %%mm5  \n\t"
83         "paddw  %%mm6, %%mm7  \n\t" /* mm7 = b4 */
84         "paddw  %%mm4, %%mm5  \n\t" /* mm5 = b5 */
85
86         SUMSUB_BA( %%mm1, %%mm3 )   /* mm1 = a3 + a2  mm3 = a3 - a2 */
87         "psubw  %%mm1, %%mm4  \n\t" /* mm4 = a0 - a2 - a3 */
88         "movq   %%mm4, %%mm1  \n\t" /* mm1 = a0 - a2 - a3 */
89         "psubw  %%mm6, %%mm3  \n\t" /* mm3 = a3 - a2 - a1 */
90         "paddw  %%mm1, %%mm1  \n\t"
91         "paddw  %%mm3, %%mm3  \n\t"
92         "psubw  %%mm2, %%mm1  \n\t" /* mm1 = b7 */
93         "paddw  %%mm0, %%mm3  \n\t" /* mm3 = b6 */
94
95         "movq  32(%0), %%mm2  \n\t" /* mm2 = src2 */
96         "movq  96(%0), %%mm6  \n\t" /* mm6 = src6 */
97         "movq   %%mm2, %%mm4  \n\t"
98         "movq   %%mm6, %%mm0  \n\t"
99         "psllw  $2,    %%mm4  \n\t" /* mm4 = 4*src2 */
100         "psllw  $2,    %%mm6  \n\t" /* mm6 = 4*src6 */
101         "paddw  %%mm4, %%mm2  \n\t" /* mm2 = 5*src2 */
102         "paddw  %%mm6, %%mm0  \n\t" /* mm0 = 5*src6 */
103         "paddw  %%mm2, %%mm2  \n\t"
104         "paddw  %%mm0, %%mm0  \n\t"
105         "psubw  %%mm0, %%mm4  \n\t" /* mm4 = 4*src2 - 10*src6 = a7 */
106         "paddw  %%mm2, %%mm6  \n\t" /* mm6 = 4*src6 + 10*src2 = a6 */
107
108         "movq    (%0), %%mm2  \n\t" /* mm2 = src0 */
109         "movq  64(%0), %%mm0  \n\t" /* mm0 = src4 */
110         SUMSUB_BA( %%mm0, %%mm2 )   /* mm0 = src0+src4  mm2 = src0-src4 */
111         "psllw  $3,    %%mm0  \n\t"
112         "psllw  $3,    %%mm2  \n\t"
113         "paddw  %1,    %%mm0  \n\t" /* add rounding bias */
114         "paddw  %1,    %%mm2  \n\t" /* add rounding bias */
115
116         SUMSUB_BA( %%mm6, %%mm0 )   /* mm6 = a4 + a6  mm0 = a4 - a6 */
117         SUMSUB_BA( %%mm4, %%mm2 )   /* mm4 = a5 + a7  mm2 = a5 - a7 */
118         SUMSUB_BA( %%mm7, %%mm6 )   /* mm7 = dst0  mm6 = dst7 */
119         SUMSUB_BA( %%mm5, %%mm4 )   /* mm5 = dst1  mm4 = dst6 */
120         SUMSUB_BA( %%mm3, %%mm2 )   /* mm3 = dst2  mm2 = dst5 */
121         SUMSUB_BA( %%mm1, %%mm0 )   /* mm1 = dst3  mm0 = dst4 */
122         :: "r"(block), "m"(bias)
123     );
124 }
125
126 static void cavs_idct8_add_mmx(uint8_t *dst, int16_t *block, int stride)
127 {
128     int i;
129     DECLARE_ALIGNED(8, int16_t, b2)[64];
130
131     for(i=0; i<2; i++){
132         DECLARE_ALIGNED(8, uint64_t, tmp);
133
134         cavs_idct8_1d(block+4*i, ff_pw_4.a);
135
136         __asm__ volatile(
137             "psraw     $3, %%mm7  \n\t"
138             "psraw     $3, %%mm6  \n\t"
139             "psraw     $3, %%mm5  \n\t"
140             "psraw     $3, %%mm4  \n\t"
141             "psraw     $3, %%mm3  \n\t"
142             "psraw     $3, %%mm2  \n\t"
143             "psraw     $3, %%mm1  \n\t"
144             "psraw     $3, %%mm0  \n\t"
145             "movq   %%mm7,    %0   \n\t"
146             TRANSPOSE4( %%mm0, %%mm2, %%mm4, %%mm6, %%mm7 )
147             "movq   %%mm0,  8(%1)  \n\t"
148             "movq   %%mm6, 24(%1)  \n\t"
149             "movq   %%mm7, 40(%1)  \n\t"
150             "movq   %%mm4, 56(%1)  \n\t"
151             "movq    %0,    %%mm7  \n\t"
152             TRANSPOSE4( %%mm7, %%mm5, %%mm3, %%mm1, %%mm0 )
153             "movq   %%mm7,   (%1)  \n\t"
154             "movq   %%mm1, 16(%1)  \n\t"
155             "movq   %%mm0, 32(%1)  \n\t"
156             "movq   %%mm3, 48(%1)  \n\t"
157             : "=m"(tmp)
158             : "r"(b2+32*i)
159             : "memory"
160         );
161     }
162
163     for(i=0; i<2; i++){
164         cavs_idct8_1d(b2+4*i, ff_pw_64.a);
165
166         __asm__ volatile(
167             "psraw     $7, %%mm7  \n\t"
168             "psraw     $7, %%mm6  \n\t"
169             "psraw     $7, %%mm5  \n\t"
170             "psraw     $7, %%mm4  \n\t"
171             "psraw     $7, %%mm3  \n\t"
172             "psraw     $7, %%mm2  \n\t"
173             "psraw     $7, %%mm1  \n\t"
174             "psraw     $7, %%mm0  \n\t"
175             "movq   %%mm7,    (%0)  \n\t"
176             "movq   %%mm5,  16(%0)  \n\t"
177             "movq   %%mm3,  32(%0)  \n\t"
178             "movq   %%mm1,  48(%0)  \n\t"
179             "movq   %%mm0,  64(%0)  \n\t"
180             "movq   %%mm2,  80(%0)  \n\t"
181             "movq   %%mm4,  96(%0)  \n\t"
182             "movq   %%mm6, 112(%0)  \n\t"
183             :: "r"(b2+4*i)
184             : "memory"
185         );
186     }
187
188     ff_add_pixels_clamped_mmx(b2, dst, stride);
189 }
190
191 /*****************************************************************************
192  *
193  * motion compensation
194  *
195  ****************************************************************************/
196
197 /* vertical filter [-1 -2 96 42 -7  0]  */
198 #define QPEL_CAVSV1(A,B,C,D,E,F,OP,MUL2) \
199         "movd (%0), "#F"            \n\t"\
200         "movq "#C", %%mm6           \n\t"\
201         "pmullw %5, %%mm6           \n\t"\
202         "movq "#D", %%mm7           \n\t"\
203         "pmullw "MANGLE(MUL2)", %%mm7\n\t"\
204         "psllw $3, "#E"             \n\t"\
205         "psubw "#E", %%mm6          \n\t"\
206         "psraw $3, "#E"             \n\t"\
207         "paddw %%mm7, %%mm6         \n\t"\
208         "paddw "#E", %%mm6          \n\t"\
209         "paddw "#B", "#B"           \n\t"\
210         "pxor %%mm7, %%mm7          \n\t"\
211         "add %2, %0                 \n\t"\
212         "punpcklbw %%mm7, "#F"      \n\t"\
213         "psubw "#B", %%mm6          \n\t"\
214         "psraw $1, "#B"             \n\t"\
215         "psubw "#A", %%mm6          \n\t"\
216         "paddw %4, %%mm6            \n\t"\
217         "psraw $7, %%mm6            \n\t"\
218         "packuswb %%mm6, %%mm6      \n\t"\
219         OP(%%mm6, (%1), A, d)            \
220         "add %3, %1                 \n\t"
221
222 /* vertical filter [ 0 -1  5  5 -1  0]  */
223 #define QPEL_CAVSV2(A,B,C,D,E,F,OP,MUL2) \
224         "movd (%0), "#F"            \n\t"\
225         "movq "#C", %%mm6           \n\t"\
226         "paddw "#D", %%mm6          \n\t"\
227         "pmullw %5, %%mm6           \n\t"\
228         "add %2, %0                 \n\t"\
229         "punpcklbw %%mm7, "#F"      \n\t"\
230         "psubw "#B", %%mm6          \n\t"\
231         "psubw "#E", %%mm6          \n\t"\
232         "paddw %4, %%mm6            \n\t"\
233         "psraw $3, %%mm6            \n\t"\
234         "packuswb %%mm6, %%mm6      \n\t"\
235         OP(%%mm6, (%1), A, d)            \
236         "add %3, %1                 \n\t"
237
238 /* vertical filter [ 0 -7 42 96 -2 -1]  */
239 #define QPEL_CAVSV3(A,B,C,D,E,F,OP,MUL2) \
240         "movd (%0), "#F"            \n\t"\
241         "movq "#C", %%mm6           \n\t"\
242         "pmullw "MANGLE(MUL2)", %%mm6\n\t"\
243         "movq "#D", %%mm7           \n\t"\
244         "pmullw %5, %%mm7           \n\t"\
245         "psllw $3, "#B"             \n\t"\
246         "psubw "#B", %%mm6          \n\t"\
247         "psraw $3, "#B"             \n\t"\
248         "paddw %%mm7, %%mm6         \n\t"\
249         "paddw "#B", %%mm6          \n\t"\
250         "paddw "#E", "#E"           \n\t"\
251         "pxor %%mm7, %%mm7          \n\t"\
252         "add %2, %0                 \n\t"\
253         "punpcklbw %%mm7, "#F"      \n\t"\
254         "psubw "#E", %%mm6          \n\t"\
255         "psraw $1, "#E"             \n\t"\
256         "psubw "#F", %%mm6          \n\t"\
257         "paddw %4, %%mm6            \n\t"\
258         "psraw $7, %%mm6            \n\t"\
259         "packuswb %%mm6, %%mm6      \n\t"\
260         OP(%%mm6, (%1), A, d)            \
261         "add %3, %1                 \n\t"
262
263
264 #define QPEL_CAVSVNUM(VOP,OP,ADD,MUL1,MUL2)\
265     int w= 2;\
266     src -= 2*srcStride;\
267     \
268     while(w--){\
269       __asm__ volatile(\
270         "pxor %%mm7, %%mm7          \n\t"\
271         "movd (%0), %%mm0           \n\t"\
272         "add %2, %0                 \n\t"\
273         "movd (%0), %%mm1           \n\t"\
274         "add %2, %0                 \n\t"\
275         "movd (%0), %%mm2           \n\t"\
276         "add %2, %0                 \n\t"\
277         "movd (%0), %%mm3           \n\t"\
278         "add %2, %0                 \n\t"\
279         "movd (%0), %%mm4           \n\t"\
280         "add %2, %0                 \n\t"\
281         "punpcklbw %%mm7, %%mm0     \n\t"\
282         "punpcklbw %%mm7, %%mm1     \n\t"\
283         "punpcklbw %%mm7, %%mm2     \n\t"\
284         "punpcklbw %%mm7, %%mm3     \n\t"\
285         "punpcklbw %%mm7, %%mm4     \n\t"\
286         VOP(%%mm0, %%mm1, %%mm2, %%mm3, %%mm4, %%mm5, OP, MUL2)\
287         VOP(%%mm1, %%mm2, %%mm3, %%mm4, %%mm5, %%mm0, OP, MUL2)\
288         VOP(%%mm2, %%mm3, %%mm4, %%mm5, %%mm0, %%mm1, OP, MUL2)\
289         VOP(%%mm3, %%mm4, %%mm5, %%mm0, %%mm1, %%mm2, OP, MUL2)\
290         VOP(%%mm4, %%mm5, %%mm0, %%mm1, %%mm2, %%mm3, OP, MUL2)\
291         VOP(%%mm5, %%mm0, %%mm1, %%mm2, %%mm3, %%mm4, OP, MUL2)\
292         VOP(%%mm0, %%mm1, %%mm2, %%mm3, %%mm4, %%mm5, OP, MUL2)\
293         VOP(%%mm1, %%mm2, %%mm3, %%mm4, %%mm5, %%mm0, OP, MUL2)\
294         \
295         : "+a"(src), "+c"(dst)\
296         : "S"((x86_reg)srcStride), "r"((x86_reg)dstStride), "m"(ADD), "m"(MUL1)\
297         : "memory"\
298      );\
299      if(h==16){\
300         __asm__ volatile(\
301             VOP(%%mm2, %%mm3, %%mm4, %%mm5, %%mm0, %%mm1, OP, MUL2)\
302             VOP(%%mm3, %%mm4, %%mm5, %%mm0, %%mm1, %%mm2, OP, MUL2)\
303             VOP(%%mm4, %%mm5, %%mm0, %%mm1, %%mm2, %%mm3, OP, MUL2)\
304             VOP(%%mm5, %%mm0, %%mm1, %%mm2, %%mm3, %%mm4, OP, MUL2)\
305             VOP(%%mm0, %%mm1, %%mm2, %%mm3, %%mm4, %%mm5, OP, MUL2)\
306             VOP(%%mm1, %%mm2, %%mm3, %%mm4, %%mm5, %%mm0, OP, MUL2)\
307             VOP(%%mm2, %%mm3, %%mm4, %%mm5, %%mm0, %%mm1, OP, MUL2)\
308             VOP(%%mm3, %%mm4, %%mm5, %%mm0, %%mm1, %%mm2, OP, MUL2)\
309             \
310            : "+a"(src), "+c"(dst)\
311            : "S"((x86_reg)srcStride), "r"((x86_reg)dstStride), "m"(ADD),  "m"(MUL1)\
312            : "memory"\
313         );\
314      }\
315      src += 4-(h+5)*srcStride;\
316      dst += 4-h*dstStride;\
317    }
318
319 #define QPEL_CAVS(OPNAME, OP, MMX)\
320 static void OPNAME ## cavs_qpel8_h_ ## MMX(uint8_t *dst, uint8_t *src, int dstStride, int srcStride){\
321     int h=8;\
322     __asm__ volatile(\
323         "pxor %%mm7, %%mm7          \n\t"\
324         "movq %5, %%mm6             \n\t"\
325         "1:                         \n\t"\
326         "movq    (%0), %%mm0        \n\t"\
327         "movq   1(%0), %%mm2        \n\t"\
328         "movq %%mm0, %%mm1          \n\t"\
329         "movq %%mm2, %%mm3          \n\t"\
330         "punpcklbw %%mm7, %%mm0     \n\t"\
331         "punpckhbw %%mm7, %%mm1     \n\t"\
332         "punpcklbw %%mm7, %%mm2     \n\t"\
333         "punpckhbw %%mm7, %%mm3     \n\t"\
334         "paddw %%mm2, %%mm0         \n\t"\
335         "paddw %%mm3, %%mm1         \n\t"\
336         "pmullw %%mm6, %%mm0        \n\t"\
337         "pmullw %%mm6, %%mm1        \n\t"\
338         "movq   -1(%0), %%mm2       \n\t"\
339         "movq    2(%0), %%mm4       \n\t"\
340         "movq %%mm2, %%mm3          \n\t"\
341         "movq %%mm4, %%mm5          \n\t"\
342         "punpcklbw %%mm7, %%mm2     \n\t"\
343         "punpckhbw %%mm7, %%mm3     \n\t"\
344         "punpcklbw %%mm7, %%mm4     \n\t"\
345         "punpckhbw %%mm7, %%mm5     \n\t"\
346         "paddw %%mm4, %%mm2         \n\t"\
347         "paddw %%mm3, %%mm5         \n\t"\
348         "psubw %%mm2, %%mm0         \n\t"\
349         "psubw %%mm5, %%mm1         \n\t"\
350         "movq %6, %%mm5             \n\t"\
351         "paddw %%mm5, %%mm0         \n\t"\
352         "paddw %%mm5, %%mm1         \n\t"\
353         "psraw $3, %%mm0            \n\t"\
354         "psraw $3, %%mm1            \n\t"\
355         "packuswb %%mm1, %%mm0      \n\t"\
356         OP(%%mm0, (%1),%%mm5, q)         \
357         "add %3, %0                 \n\t"\
358         "add %4, %1                 \n\t"\
359         "decl %2                    \n\t"\
360         " jnz 1b                    \n\t"\
361         : "+a"(src), "+c"(dst), "+m"(h)\
362         : "d"((x86_reg)srcStride), "S"((x86_reg)dstStride), "m"(ff_pw_5), "m"(ff_pw_4)\
363         : "memory"\
364     );\
365 }\
366 \
367 static inline void OPNAME ## cavs_qpel8or16_v1_ ## MMX(uint8_t *dst, uint8_t *src, int dstStride, int srcStride, int h){\
368   QPEL_CAVSVNUM(QPEL_CAVSV1,OP,ff_pw_64,ff_pw_96,ff_pw_42)      \
369 }\
370 \
371 static inline void OPNAME ## cavs_qpel8or16_v2_ ## MMX(uint8_t *dst, uint8_t *src, int dstStride, int srcStride, int h){\
372   QPEL_CAVSVNUM(QPEL_CAVSV2,OP,ff_pw_4,ff_pw_5,ff_pw_5)         \
373 }\
374 \
375 static inline void OPNAME ## cavs_qpel8or16_v3_ ## MMX(uint8_t *dst, uint8_t *src, int dstStride, int srcStride, int h){\
376   QPEL_CAVSVNUM(QPEL_CAVSV3,OP,ff_pw_64,ff_pw_96,ff_pw_42)      \
377 }\
378 \
379 static void OPNAME ## cavs_qpel8_v1_ ## MMX(uint8_t *dst, uint8_t *src, int dstStride, int srcStride){\
380     OPNAME ## cavs_qpel8or16_v1_ ## MMX(dst  , src  , dstStride, srcStride, 8);\
381 }\
382 static void OPNAME ## cavs_qpel16_v1_ ## MMX(uint8_t *dst, uint8_t *src, int dstStride, int srcStride){\
383     OPNAME ## cavs_qpel8or16_v1_ ## MMX(dst  , src  , dstStride, srcStride, 16);\
384     OPNAME ## cavs_qpel8or16_v1_ ## MMX(dst+8, src+8, dstStride, srcStride, 16);\
385 }\
386 \
387 static void OPNAME ## cavs_qpel8_v2_ ## MMX(uint8_t *dst, uint8_t *src, int dstStride, int srcStride){\
388     OPNAME ## cavs_qpel8or16_v2_ ## MMX(dst  , src  , dstStride, srcStride, 8);\
389 }\
390 static void OPNAME ## cavs_qpel16_v2_ ## MMX(uint8_t *dst, uint8_t *src, int dstStride, int srcStride){\
391     OPNAME ## cavs_qpel8or16_v2_ ## MMX(dst  , src  , dstStride, srcStride, 16);\
392     OPNAME ## cavs_qpel8or16_v2_ ## MMX(dst+8, src+8, dstStride, srcStride, 16);\
393 }\
394 \
395 static void OPNAME ## cavs_qpel8_v3_ ## MMX(uint8_t *dst, uint8_t *src, int dstStride, int srcStride){\
396     OPNAME ## cavs_qpel8or16_v3_ ## MMX(dst  , src  , dstStride, srcStride, 8);\
397 }\
398 static void OPNAME ## cavs_qpel16_v3_ ## MMX(uint8_t *dst, uint8_t *src, int dstStride, int srcStride){\
399     OPNAME ## cavs_qpel8or16_v3_ ## MMX(dst  , src  , dstStride, srcStride, 16);\
400     OPNAME ## cavs_qpel8or16_v3_ ## MMX(dst+8, src+8, dstStride, srcStride, 16);\
401 }\
402 \
403 static void OPNAME ## cavs_qpel16_h_ ## MMX(uint8_t *dst, uint8_t *src, int dstStride, int srcStride){\
404     OPNAME ## cavs_qpel8_h_ ## MMX(dst  , src  , dstStride, srcStride);\
405     OPNAME ## cavs_qpel8_h_ ## MMX(dst+8, src+8, dstStride, srcStride);\
406     src += 8*srcStride;\
407     dst += 8*dstStride;\
408     OPNAME ## cavs_qpel8_h_ ## MMX(dst  , src  , dstStride, srcStride);\
409     OPNAME ## cavs_qpel8_h_ ## MMX(dst+8, src+8, dstStride, srcStride);\
410 }\
411
412 #define CAVS_MC(OPNAME, SIZE, MMX) \
413 static void ff_ ## OPNAME ## cavs_qpel ## SIZE ## _mc20_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
414     OPNAME ## cavs_qpel ## SIZE ## _h_ ## MMX(dst, src, stride, stride);\
415 }\
416 \
417 static void ff_ ## OPNAME ## cavs_qpel ## SIZE ## _mc01_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
418     OPNAME ## cavs_qpel ## SIZE ## _v1_ ## MMX(dst, src, stride, stride);\
419 }\
420 \
421 static void ff_ ## OPNAME ## cavs_qpel ## SIZE ## _mc02_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
422     OPNAME ## cavs_qpel ## SIZE ## _v2_ ## MMX(dst, src, stride, stride);\
423 }\
424 \
425 static void ff_ ## OPNAME ## cavs_qpel ## SIZE ## _mc03_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
426     OPNAME ## cavs_qpel ## SIZE ## _v3_ ## MMX(dst, src, stride, stride);\
427 }\
428
429 #define PUT_OP(a,b,temp, size) "mov" #size " " #a ", " #b "    \n\t"
430 #define AVG_3DNOW_OP(a,b,temp, size) \
431 "mov" #size " " #b ", " #temp "   \n\t"\
432 "pavgusb " #temp ", " #a "        \n\t"\
433 "mov" #size " " #a ", " #b "      \n\t"
434 #define AVG_MMXEXT_OP(a, b, temp, size) \
435 "mov" #size " " #b ", " #temp "   \n\t"\
436 "pavgb " #temp ", " #a "          \n\t"\
437 "mov" #size " " #a ", " #b "      \n\t"
438
439 #endif /* (HAVE_MMXEXT_INLINE || HAVE_AMD3DNOW_INLINE) */
440
441 #if HAVE_MMXEXT_INLINE
442 QPEL_CAVS(put_,        PUT_OP, mmxext)
443 QPEL_CAVS(avg_, AVG_MMXEXT_OP, mmxext)
444
445 CAVS_MC(put_,  8, mmxext)
446 CAVS_MC(put_, 16, mmxext)
447 CAVS_MC(avg_,  8, mmxext)
448 CAVS_MC(avg_, 16, mmxext)
449
450 static av_cold void ff_cavsdsp_init_mmxext(CAVSDSPContext *c,
451                                            AVCodecContext *avctx)
452 {
453 #define dspfunc(PFX, IDX, NUM) \
454     c->PFX ## _pixels_tab[IDX][ 0] = ff_ ## PFX ## NUM ## _mc00_mmxext; \
455     c->PFX ## _pixels_tab[IDX][ 2] = ff_ ## PFX ## NUM ## _mc20_mmxext; \
456     c->PFX ## _pixels_tab[IDX][ 4] = ff_ ## PFX ## NUM ## _mc01_mmxext; \
457     c->PFX ## _pixels_tab[IDX][ 8] = ff_ ## PFX ## NUM ## _mc02_mmxext; \
458     c->PFX ## _pixels_tab[IDX][12] = ff_ ## PFX ## NUM ## _mc03_mmxext; \
459
460     dspfunc(put_cavs_qpel, 0, 16);
461     dspfunc(put_cavs_qpel, 1, 8);
462     dspfunc(avg_cavs_qpel, 0, 16);
463     dspfunc(avg_cavs_qpel, 1, 8);
464 #undef dspfunc
465     c->cavs_idct8_add = cavs_idct8_add_mmx;
466     c->idct_perm = FF_TRANSPOSE_IDCT_PERM;
467 }
468 #endif /* HAVE_MMXEXT_INLINE */
469
470 #if HAVE_AMD3DNOW_INLINE
471 QPEL_CAVS(put_,       PUT_OP, 3dnow)
472 QPEL_CAVS(avg_, AVG_3DNOW_OP, 3dnow)
473
474 CAVS_MC(put_, 8, 3dnow)
475 CAVS_MC(put_, 16,3dnow)
476 CAVS_MC(avg_, 8, 3dnow)
477 CAVS_MC(avg_, 16,3dnow)
478
479 static av_cold void ff_cavsdsp_init_3dnow(CAVSDSPContext *c,
480                                           AVCodecContext *avctx)
481 {
482 #define dspfunc(PFX, IDX, NUM) \
483     c->PFX ## _pixels_tab[IDX][ 0] = ff_ ## PFX ## NUM ## _mc00_mmxext; \
484     c->PFX ## _pixels_tab[IDX][ 2] = ff_ ## PFX ## NUM ## _mc20_3dnow; \
485     c->PFX ## _pixels_tab[IDX][ 4] = ff_ ## PFX ## NUM ## _mc01_3dnow; \
486     c->PFX ## _pixels_tab[IDX][ 8] = ff_ ## PFX ## NUM ## _mc02_3dnow; \
487     c->PFX ## _pixels_tab[IDX][12] = ff_ ## PFX ## NUM ## _mc03_3dnow; \
488
489     dspfunc(put_cavs_qpel, 0, 16);
490     dspfunc(put_cavs_qpel, 1, 8);
491     dspfunc(avg_cavs_qpel, 0, 16);
492     dspfunc(avg_cavs_qpel, 1, 8);
493 #undef dspfunc
494     c->cavs_idct8_add = cavs_idct8_add_mmx;
495     c->idct_perm = FF_TRANSPOSE_IDCT_PERM;
496 }
497 #endif /* HAVE_AMD3DNOW_INLINE */
498
499 av_cold void ff_cavsdsp_init_x86(CAVSDSPContext *c, AVCodecContext *avctx)
500 {
501     int mm_flags = av_get_cpu_flags();
502
503 #if HAVE_MMXEXT_INLINE
504     if (mm_flags & AV_CPU_FLAG_MMXEXT) ff_cavsdsp_init_mmxext(c, avctx);
505 #endif /* HAVE_MMXEXT_INLINE */
506 #if HAVE_AMD3DNOW_INLINE
507     if (mm_flags & AV_CPU_FLAG_3DNOW) ff_cavsdsp_init_3dnow(c, avctx);
508 #endif /* HAVE_AMD3DNOW_INLINE */
509 }