Merge commit 'f81be06cf614919d71ded29b8f595bef40123ad8'
[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 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/cavsdsp.h"
31 #include "libavcodec/idctdsp.h"
32 #include "constants.h"
33 #include "fpel.h"
34 #include "idctdsp.h"
35 #include "config.h"
36
37 #if HAVE_MMX_INLINE
38
39 /* in/out: mma=mma+mmb, mmb=mmb-mma */
40 #define SUMSUB_BA( a, b ) \
41     "paddw "#b", "#a" \n\t"\
42     "paddw "#b", "#b" \n\t"\
43     "psubw "#a", "#b" \n\t"
44
45 /*****************************************************************************
46  *
47  * inverse transform
48  *
49  ****************************************************************************/
50
51 static inline void cavs_idct8_1d(int16_t *block, uint64_t bias)
52 {
53     __asm__ volatile(
54         "movq 112(%0), %%mm4  \n\t" /* mm4 = src7 */
55         "movq  16(%0), %%mm5  \n\t" /* mm5 = src1 */
56         "movq  80(%0), %%mm2  \n\t" /* mm2 = src5 */
57         "movq  48(%0), %%mm7  \n\t" /* mm7 = src3 */
58         "movq   %%mm4, %%mm0  \n\t"
59         "movq   %%mm5, %%mm3  \n\t"
60         "movq   %%mm2, %%mm6  \n\t"
61         "movq   %%mm7, %%mm1  \n\t"
62
63         "paddw  %%mm4, %%mm4  \n\t" /* mm4 = 2*src7 */
64         "paddw  %%mm3, %%mm3  \n\t" /* mm3 = 2*src1 */
65         "paddw  %%mm6, %%mm6  \n\t" /* mm6 = 2*src5 */
66         "paddw  %%mm1, %%mm1  \n\t" /* mm1 = 2*src3 */
67         "paddw  %%mm4, %%mm0  \n\t" /* mm0 = 3*src7 */
68         "paddw  %%mm3, %%mm5  \n\t" /* mm5 = 3*src1 */
69         "paddw  %%mm6, %%mm2  \n\t" /* mm2 = 3*src5 */
70         "paddw  %%mm1, %%mm7  \n\t" /* mm7 = 3*src3 */
71         "psubw  %%mm4, %%mm5  \n\t" /* mm5 = 3*src1 - 2*src7 = a0 */
72         "paddw  %%mm6, %%mm7  \n\t" /* mm7 = 3*src3 + 2*src5 = a1 */
73         "psubw  %%mm2, %%mm1  \n\t" /* mm1 = 2*src3 - 3*src5 = a2 */
74         "paddw  %%mm0, %%mm3  \n\t" /* mm3 = 2*src1 + 3*src7 = a3 */
75
76         "movq   %%mm5, %%mm4  \n\t"
77         "movq   %%mm7, %%mm6  \n\t"
78         "movq   %%mm3, %%mm0  \n\t"
79         "movq   %%mm1, %%mm2  \n\t"
80         SUMSUB_BA( %%mm7, %%mm5 )   /* mm7 = a0 + a1  mm5 = a0 - a1 */
81         "paddw  %%mm3, %%mm7  \n\t" /* mm7 = a0 + a1 + a3 */
82         "paddw  %%mm1, %%mm5  \n\t" /* mm5 = a0 - a1 + a2 */
83         "paddw  %%mm7, %%mm7  \n\t"
84         "paddw  %%mm5, %%mm5  \n\t"
85         "paddw  %%mm6, %%mm7  \n\t" /* mm7 = b4 */
86         "paddw  %%mm4, %%mm5  \n\t" /* mm5 = b5 */
87
88         SUMSUB_BA( %%mm1, %%mm3 )   /* mm1 = a3 + a2  mm3 = a3 - a2 */
89         "psubw  %%mm1, %%mm4  \n\t" /* mm4 = a0 - a2 - a3 */
90         "movq   %%mm4, %%mm1  \n\t" /* mm1 = a0 - a2 - a3 */
91         "psubw  %%mm6, %%mm3  \n\t" /* mm3 = a3 - a2 - a1 */
92         "paddw  %%mm1, %%mm1  \n\t"
93         "paddw  %%mm3, %%mm3  \n\t"
94         "psubw  %%mm2, %%mm1  \n\t" /* mm1 = b7 */
95         "paddw  %%mm0, %%mm3  \n\t" /* mm3 = b6 */
96
97         "movq  32(%0), %%mm2  \n\t" /* mm2 = src2 */
98         "movq  96(%0), %%mm6  \n\t" /* mm6 = src6 */
99         "movq   %%mm2, %%mm4  \n\t"
100         "movq   %%mm6, %%mm0  \n\t"
101         "psllw  $2,    %%mm4  \n\t" /* mm4 = 4*src2 */
102         "psllw  $2,    %%mm6  \n\t" /* mm6 = 4*src6 */
103         "paddw  %%mm4, %%mm2  \n\t" /* mm2 = 5*src2 */
104         "paddw  %%mm6, %%mm0  \n\t" /* mm0 = 5*src6 */
105         "paddw  %%mm2, %%mm2  \n\t"
106         "paddw  %%mm0, %%mm0  \n\t"
107         "psubw  %%mm0, %%mm4  \n\t" /* mm4 = 4*src2 - 10*src6 = a7 */
108         "paddw  %%mm2, %%mm6  \n\t" /* mm6 = 4*src6 + 10*src2 = a6 */
109
110         "movq    (%0), %%mm2  \n\t" /* mm2 = src0 */
111         "movq  64(%0), %%mm0  \n\t" /* mm0 = src4 */
112         SUMSUB_BA( %%mm0, %%mm2 )   /* mm0 = src0+src4  mm2 = src0-src4 */
113         "psllw  $3,    %%mm0  \n\t"
114         "psllw  $3,    %%mm2  \n\t"
115         "paddw  %1,    %%mm0  \n\t" /* add rounding bias */
116         "paddw  %1,    %%mm2  \n\t" /* add rounding bias */
117
118         SUMSUB_BA( %%mm6, %%mm0 )   /* mm6 = a4 + a6  mm0 = a4 - a6 */
119         SUMSUB_BA( %%mm4, %%mm2 )   /* mm4 = a5 + a7  mm2 = a5 - a7 */
120         SUMSUB_BA( %%mm7, %%mm6 )   /* mm7 = dst0  mm6 = dst7 */
121         SUMSUB_BA( %%mm5, %%mm4 )   /* mm5 = dst1  mm4 = dst6 */
122         SUMSUB_BA( %%mm3, %%mm2 )   /* mm3 = dst2  mm2 = dst5 */
123         SUMSUB_BA( %%mm1, %%mm0 )   /* mm1 = dst3  mm0 = dst4 */
124         :: "r"(block), "m"(bias)
125     );
126 }
127
128 #define SBUTTERFLY(a,b,t,n,m)\
129     "mov" #m " " #a ", " #t "         \n\t" /* abcd */\
130     "punpckl" #n " " #b ", " #a "     \n\t" /* aebf */\
131     "punpckh" #n " " #b ", " #t "     \n\t" /* cgdh */\
132
133 #define TRANSPOSE4(a,b,c,d,t)\
134     SBUTTERFLY(a,b,t,wd,q) /* a=aebf t=cgdh */\
135     SBUTTERFLY(c,d,b,wd,q) /* c=imjn b=kolp */\
136     SBUTTERFLY(a,c,d,dq,q) /* a=aeim d=bfjn */\
137     SBUTTERFLY(t,b,c,dq,q) /* t=cgko c=dhlp */
138
139 static void cavs_idct8_add_mmx(uint8_t *dst, int16_t *block, ptrdiff_t stride)
140 {
141     int i;
142     LOCAL_ALIGNED(16, int16_t, b2, [64]);
143
144     for(i=0; i<2; i++){
145         cavs_idct8_1d(block + 4 * i, ff_pw_4.a);
146
147         __asm__ volatile(
148             "psraw     $3, %%mm7  \n\t"
149             "psraw     $3, %%mm6  \n\t"
150             "psraw     $3, %%mm5  \n\t"
151             "psraw     $3, %%mm4  \n\t"
152             "psraw     $3, %%mm3  \n\t"
153             "psraw     $3, %%mm2  \n\t"
154             "psraw     $3, %%mm1  \n\t"
155             "psraw     $3, %%mm0  \n\t"
156             "movq   %%mm7,  (%0)  \n\t"
157             TRANSPOSE4( %%mm0, %%mm2, %%mm4, %%mm6, %%mm7 )
158             "movq   %%mm0,  8(%0)  \n\t"
159             "movq   %%mm6, 24(%0)  \n\t"
160             "movq   %%mm7, 40(%0)  \n\t"
161             "movq   %%mm4, 56(%0)  \n\t"
162             "movq    (%0),  %%mm7  \n\t"
163             TRANSPOSE4( %%mm7, %%mm5, %%mm3, %%mm1, %%mm0 )
164             "movq   %%mm7,   (%0)  \n\t"
165             "movq   %%mm1, 16(%0)  \n\t"
166             "movq   %%mm0, 32(%0)  \n\t"
167             "movq   %%mm3, 48(%0)  \n\t"
168             :
169             : "r"(b2 + 32 * i)
170             : "memory"
171         );
172     }
173
174     for(i=0; i<2; i++){
175         cavs_idct8_1d(b2+4*i, ff_pw_64.a);
176
177         __asm__ volatile(
178             "psraw     $7, %%mm7  \n\t"
179             "psraw     $7, %%mm6  \n\t"
180             "psraw     $7, %%mm5  \n\t"
181             "psraw     $7, %%mm4  \n\t"
182             "psraw     $7, %%mm3  \n\t"
183             "psraw     $7, %%mm2  \n\t"
184             "psraw     $7, %%mm1  \n\t"
185             "psraw     $7, %%mm0  \n\t"
186             "movq   %%mm7,    (%0)  \n\t"
187             "movq   %%mm5,  16(%0)  \n\t"
188             "movq   %%mm3,  32(%0)  \n\t"
189             "movq   %%mm1,  48(%0)  \n\t"
190             "movq   %%mm0,  64(%0)  \n\t"
191             "movq   %%mm2,  80(%0)  \n\t"
192             "movq   %%mm4,  96(%0)  \n\t"
193             "movq   %%mm6, 112(%0)  \n\t"
194             :: "r"(b2+4*i)
195             : "memory"
196         );
197     }
198
199     ff_add_pixels_clamped(b2, dst, stride);
200 }
201
202 #endif /* HAVE_MMX_INLINE */
203
204 #if (HAVE_MMXEXT_INLINE || HAVE_AMD3DNOW_INLINE)
205
206 /*****************************************************************************
207  *
208  * motion compensation
209  *
210  ****************************************************************************/
211
212 /* vertical filter [-1 -2 96 42 -7  0]  */
213 #define QPEL_CAVSV1(A,B,C,D,E,F,OP,ADD, MUL1, MUL2) \
214         "movd (%0), "#F"            \n\t"\
215         "movq "#C", %%mm6           \n\t"\
216         "pmullw "MANGLE(MUL1)", %%mm6\n\t"\
217         "movq "#D", %%mm7           \n\t"\
218         "pmullw "MANGLE(MUL2)", %%mm7\n\t"\
219         "psllw $3, "#E"             \n\t"\
220         "psubw "#E", %%mm6          \n\t"\
221         "psraw $3, "#E"             \n\t"\
222         "paddw %%mm7, %%mm6         \n\t"\
223         "paddw "#E", %%mm6          \n\t"\
224         "paddw "#B", "#B"           \n\t"\
225         "pxor %%mm7, %%mm7          \n\t"\
226         "add %2, %0                 \n\t"\
227         "punpcklbw %%mm7, "#F"      \n\t"\
228         "psubw "#B", %%mm6          \n\t"\
229         "psraw $1, "#B"             \n\t"\
230         "psubw "#A", %%mm6          \n\t"\
231         "paddw "MANGLE(ADD)", %%mm6 \n\t"\
232         "psraw $7, %%mm6            \n\t"\
233         "packuswb %%mm6, %%mm6      \n\t"\
234         OP(%%mm6, (%1), A, d)            \
235         "add %3, %1                 \n\t"
236
237 /* vertical filter [ 0 -1  5  5 -1  0]  */
238 #define QPEL_CAVSV2(A,B,C,D,E,F,OP,ADD, MUL1, MUL2) \
239         "movd (%0), "#F"            \n\t"\
240         "movq "#C", %%mm6           \n\t"\
241         "paddw "#D", %%mm6          \n\t"\
242         "pmullw "MANGLE(MUL1)", %%mm6\n\t"\
243         "add %2, %0                 \n\t"\
244         "punpcklbw %%mm7, "#F"      \n\t"\
245         "psubw "#B", %%mm6          \n\t"\
246         "psubw "#E", %%mm6          \n\t"\
247         "paddw "MANGLE(ADD)", %%mm6 \n\t"\
248         "psraw $3, %%mm6            \n\t"\
249         "packuswb %%mm6, %%mm6      \n\t"\
250         OP(%%mm6, (%1), A, d)            \
251         "add %3, %1                 \n\t"
252
253 /* vertical filter [ 0 -7 42 96 -2 -1]  */
254 #define QPEL_CAVSV3(A,B,C,D,E,F,OP,ADD, MUL1, MUL2) \
255         "movd (%0), "#F"            \n\t"\
256         "movq "#C", %%mm6           \n\t"\
257         "pmullw "MANGLE(MUL2)", %%mm6\n\t"\
258         "movq "#D", %%mm7           \n\t"\
259         "pmullw "MANGLE(MUL1)", %%mm7\n\t"\
260         "psllw $3, "#B"             \n\t"\
261         "psubw "#B", %%mm6          \n\t"\
262         "psraw $3, "#B"             \n\t"\
263         "paddw %%mm7, %%mm6         \n\t"\
264         "paddw "#B", %%mm6          \n\t"\
265         "paddw "#E", "#E"           \n\t"\
266         "pxor %%mm7, %%mm7          \n\t"\
267         "add %2, %0                 \n\t"\
268         "punpcklbw %%mm7, "#F"      \n\t"\
269         "psubw "#E", %%mm6          \n\t"\
270         "psraw $1, "#E"             \n\t"\
271         "psubw "#F", %%mm6          \n\t"\
272         "paddw "MANGLE(ADD)", %%mm6 \n\t"\
273         "psraw $7, %%mm6            \n\t"\
274         "packuswb %%mm6, %%mm6      \n\t"\
275         OP(%%mm6, (%1), A, d)            \
276         "add %3, %1                 \n\t"
277
278
279 #define QPEL_CAVSVNUM(VOP,OP,ADD,MUL1,MUL2)\
280     int w= 2;\
281     src -= 2*srcStride;\
282     \
283     while(w--){\
284       __asm__ volatile(\
285         "pxor %%mm7, %%mm7          \n\t"\
286         "movd (%0), %%mm0           \n\t"\
287         "add %2, %0                 \n\t"\
288         "movd (%0), %%mm1           \n\t"\
289         "add %2, %0                 \n\t"\
290         "movd (%0), %%mm2           \n\t"\
291         "add %2, %0                 \n\t"\
292         "movd (%0), %%mm3           \n\t"\
293         "add %2, %0                 \n\t"\
294         "movd (%0), %%mm4           \n\t"\
295         "add %2, %0                 \n\t"\
296         "punpcklbw %%mm7, %%mm0     \n\t"\
297         "punpcklbw %%mm7, %%mm1     \n\t"\
298         "punpcklbw %%mm7, %%mm2     \n\t"\
299         "punpcklbw %%mm7, %%mm3     \n\t"\
300         "punpcklbw %%mm7, %%mm4     \n\t"\
301         VOP(%%mm0, %%mm1, %%mm2, %%mm3, %%mm4, %%mm5, OP, ADD, MUL1, MUL2)\
302         VOP(%%mm1, %%mm2, %%mm3, %%mm4, %%mm5, %%mm0, OP, ADD, MUL1, MUL2)\
303         VOP(%%mm2, %%mm3, %%mm4, %%mm5, %%mm0, %%mm1, OP, ADD, MUL1, MUL2)\
304         VOP(%%mm3, %%mm4, %%mm5, %%mm0, %%mm1, %%mm2, OP, ADD, MUL1, MUL2)\
305         VOP(%%mm4, %%mm5, %%mm0, %%mm1, %%mm2, %%mm3, OP, ADD, MUL1, MUL2)\
306         VOP(%%mm5, %%mm0, %%mm1, %%mm2, %%mm3, %%mm4, OP, ADD, MUL1, MUL2)\
307         VOP(%%mm0, %%mm1, %%mm2, %%mm3, %%mm4, %%mm5, OP, ADD, MUL1, MUL2)\
308         VOP(%%mm1, %%mm2, %%mm3, %%mm4, %%mm5, %%mm0, OP, ADD, MUL1, MUL2)\
309         \
310         : "+a"(src), "+c"(dst)\
311         : "S"((x86_reg)srcStride), "r"((x86_reg)dstStride)\
312           NAMED_CONSTRAINTS_ADD(ADD,MUL1,MUL2)\
313         : "memory"\
314      );\
315      if(h==16){\
316         __asm__ volatile(\
317             VOP(%%mm2, %%mm3, %%mm4, %%mm5, %%mm0, %%mm1, OP, ADD, MUL1, MUL2)\
318             VOP(%%mm3, %%mm4, %%mm5, %%mm0, %%mm1, %%mm2, OP, ADD, MUL1, MUL2)\
319             VOP(%%mm4, %%mm5, %%mm0, %%mm1, %%mm2, %%mm3, OP, ADD, MUL1, MUL2)\
320             VOP(%%mm5, %%mm0, %%mm1, %%mm2, %%mm3, %%mm4, OP, ADD, MUL1, MUL2)\
321             VOP(%%mm0, %%mm1, %%mm2, %%mm3, %%mm4, %%mm5, OP, ADD, MUL1, MUL2)\
322             VOP(%%mm1, %%mm2, %%mm3, %%mm4, %%mm5, %%mm0, OP, ADD, MUL1, MUL2)\
323             VOP(%%mm2, %%mm3, %%mm4, %%mm5, %%mm0, %%mm1, OP, ADD, MUL1, MUL2)\
324             VOP(%%mm3, %%mm4, %%mm5, %%mm0, %%mm1, %%mm2, OP, ADD, MUL1, MUL2)\
325             \
326            : "+a"(src), "+c"(dst)\
327            : "S"((x86_reg)srcStride), "r"((x86_reg)dstStride)\
328              NAMED_CONSTRAINTS_ADD(ADD,MUL1,MUL2)\
329            : "memory"\
330         );\
331      }\
332      src += 4-(h+5)*srcStride;\
333      dst += 4-h*dstStride;\
334    }
335
336 #define QPEL_CAVS(OPNAME, OP, MMX)\
337 static void OPNAME ## cavs_qpel8_h_ ## MMX(uint8_t *dst, const uint8_t *src, ptrdiff_t dstStride, ptrdiff_t srcStride)\
338 {\
339     int h=8;\
340     __asm__ volatile(\
341         "pxor %%mm7, %%mm7          \n\t"\
342         "movq "MANGLE(ff_pw_5)", %%mm6\n\t"\
343         "1:                         \n\t"\
344         "movq    (%0), %%mm0        \n\t"\
345         "movq   1(%0), %%mm2        \n\t"\
346         "movq %%mm0, %%mm1          \n\t"\
347         "movq %%mm2, %%mm3          \n\t"\
348         "punpcklbw %%mm7, %%mm0     \n\t"\
349         "punpckhbw %%mm7, %%mm1     \n\t"\
350         "punpcklbw %%mm7, %%mm2     \n\t"\
351         "punpckhbw %%mm7, %%mm3     \n\t"\
352         "paddw %%mm2, %%mm0         \n\t"\
353         "paddw %%mm3, %%mm1         \n\t"\
354         "pmullw %%mm6, %%mm0        \n\t"\
355         "pmullw %%mm6, %%mm1        \n\t"\
356         "movq   -1(%0), %%mm2       \n\t"\
357         "movq    2(%0), %%mm4       \n\t"\
358         "movq %%mm2, %%mm3          \n\t"\
359         "movq %%mm4, %%mm5          \n\t"\
360         "punpcklbw %%mm7, %%mm2     \n\t"\
361         "punpckhbw %%mm7, %%mm3     \n\t"\
362         "punpcklbw %%mm7, %%mm4     \n\t"\
363         "punpckhbw %%mm7, %%mm5     \n\t"\
364         "paddw %%mm4, %%mm2         \n\t"\
365         "paddw %%mm3, %%mm5         \n\t"\
366         "psubw %%mm2, %%mm0         \n\t"\
367         "psubw %%mm5, %%mm1         \n\t"\
368         "movq "MANGLE(ff_pw_4)", %%mm5\n\t"\
369         "paddw %%mm5, %%mm0         \n\t"\
370         "paddw %%mm5, %%mm1         \n\t"\
371         "psraw $3, %%mm0            \n\t"\
372         "psraw $3, %%mm1            \n\t"\
373         "packuswb %%mm1, %%mm0      \n\t"\
374         OP(%%mm0, (%1),%%mm5, q)         \
375         "add %3, %0                 \n\t"\
376         "add %4, %1                 \n\t"\
377         "decl %2                    \n\t"\
378         " jnz 1b                    \n\t"\
379         : "+a"(src), "+c"(dst), "+m"(h)\
380         : "d"((x86_reg)srcStride), "S"((x86_reg)dstStride)\
381           NAMED_CONSTRAINTS_ADD(ff_pw_4,ff_pw_5)\
382         : "memory"\
383     );\
384 }\
385 \
386 static inline void OPNAME ## cavs_qpel8or16_v1_ ## MMX(uint8_t *dst, const uint8_t *src, ptrdiff_t dstStride, ptrdiff_t srcStride, int h)\
387 {                                                                       \
388   QPEL_CAVSVNUM(QPEL_CAVSV1,OP,ff_pw_64,ff_pw_96,ff_pw_42)      \
389 }\
390 \
391 static inline void OPNAME ## cavs_qpel8or16_v2_ ## MMX(uint8_t *dst, const uint8_t *src, ptrdiff_t dstStride, ptrdiff_t srcStride, int h)\
392 {                                                                       \
393   QPEL_CAVSVNUM(QPEL_CAVSV2,OP,ff_pw_4,ff_pw_5,ff_pw_42)        \
394 }\
395 \
396 static inline void OPNAME ## cavs_qpel8or16_v3_ ## MMX(uint8_t *dst, const uint8_t *src, ptrdiff_t dstStride, ptrdiff_t srcStride, int h)\
397 {                                                                       \
398   QPEL_CAVSVNUM(QPEL_CAVSV3,OP,ff_pw_64,ff_pw_96,ff_pw_42)      \
399 }\
400 \
401 static void OPNAME ## cavs_qpel8_v1_ ## MMX(uint8_t *dst, const uint8_t *src, ptrdiff_t dstStride, ptrdiff_t srcStride)\
402 {                                                                       \
403     OPNAME ## cavs_qpel8or16_v1_ ## MMX(dst  , src  , dstStride, srcStride, 8);\
404 }\
405 static void OPNAME ## cavs_qpel16_v1_ ## MMX(uint8_t *dst, const uint8_t *src, ptrdiff_t dstStride, ptrdiff_t srcStride)\
406 {                                                                       \
407     OPNAME ## cavs_qpel8or16_v1_ ## MMX(dst  , src  , dstStride, srcStride, 16);\
408     OPNAME ## cavs_qpel8or16_v1_ ## MMX(dst+8, src+8, dstStride, srcStride, 16);\
409 }\
410 \
411 static void OPNAME ## cavs_qpel8_v2_ ## MMX(uint8_t *dst, const uint8_t *src, ptrdiff_t dstStride, ptrdiff_t srcStride)\
412 {                                                                       \
413     OPNAME ## cavs_qpel8or16_v2_ ## MMX(dst  , src  , dstStride, srcStride, 8);\
414 }\
415 static void OPNAME ## cavs_qpel16_v2_ ## MMX(uint8_t *dst, const uint8_t *src, ptrdiff_t dstStride, ptrdiff_t srcStride)\
416 {                                                                       \
417     OPNAME ## cavs_qpel8or16_v2_ ## MMX(dst  , src  , dstStride, srcStride, 16);\
418     OPNAME ## cavs_qpel8or16_v2_ ## MMX(dst+8, src+8, dstStride, srcStride, 16);\
419 }\
420 \
421 static void OPNAME ## cavs_qpel8_v3_ ## MMX(uint8_t *dst, const uint8_t *src, ptrdiff_t dstStride, ptrdiff_t srcStride)\
422 {                                                                       \
423     OPNAME ## cavs_qpel8or16_v3_ ## MMX(dst  , src  , dstStride, srcStride, 8);\
424 }\
425 static void OPNAME ## cavs_qpel16_v3_ ## MMX(uint8_t *dst, const uint8_t *src, ptrdiff_t dstStride, ptrdiff_t srcStride)\
426 {                                                                       \
427     OPNAME ## cavs_qpel8or16_v3_ ## MMX(dst  , src  , dstStride, srcStride, 16);\
428     OPNAME ## cavs_qpel8or16_v3_ ## MMX(dst+8, src+8, dstStride, srcStride, 16);\
429 }\
430 \
431 static void OPNAME ## cavs_qpel16_h_ ## MMX(uint8_t *dst, const uint8_t *src, ptrdiff_t dstStride, ptrdiff_t srcStride)\
432 {                                                                       \
433     OPNAME ## cavs_qpel8_h_ ## MMX(dst  , src  , dstStride, srcStride);\
434     OPNAME ## cavs_qpel8_h_ ## MMX(dst+8, src+8, dstStride, srcStride);\
435     src += 8*srcStride;\
436     dst += 8*dstStride;\
437     OPNAME ## cavs_qpel8_h_ ## MMX(dst  , src  , dstStride, srcStride);\
438     OPNAME ## cavs_qpel8_h_ ## MMX(dst+8, src+8, dstStride, srcStride);\
439 }\
440
441 #define CAVS_MC(OPNAME, SIZE, MMX) \
442 static void OPNAME ## cavs_qpel ## SIZE ## _mc20_ ## MMX(uint8_t *dst, const uint8_t *src, ptrdiff_t stride)\
443 {\
444     OPNAME ## cavs_qpel ## SIZE ## _h_ ## MMX(dst, src, stride, stride);\
445 }\
446 \
447 static void OPNAME ## cavs_qpel ## SIZE ## _mc01_ ## MMX(uint8_t *dst, const uint8_t *src, ptrdiff_t stride)\
448 {\
449     OPNAME ## cavs_qpel ## SIZE ## _v1_ ## MMX(dst, src, stride, stride);\
450 }\
451 \
452 static void OPNAME ## cavs_qpel ## SIZE ## _mc02_ ## MMX(uint8_t *dst, const uint8_t *src, ptrdiff_t stride)\
453 {\
454     OPNAME ## cavs_qpel ## SIZE ## _v2_ ## MMX(dst, src, stride, stride);\
455 }\
456 \
457 static void OPNAME ## cavs_qpel ## SIZE ## _mc03_ ## MMX(uint8_t *dst, const uint8_t *src, ptrdiff_t stride)\
458 {\
459     OPNAME ## cavs_qpel ## SIZE ## _v3_ ## MMX(dst, src, stride, stride);\
460 }\
461
462 #define PUT_OP(a,b,temp, size) "mov" #size " " #a ", " #b "    \n\t"
463 #define AVG_3DNOW_OP(a,b,temp, size) \
464 "mov" #size " " #b ", " #temp "   \n\t"\
465 "pavgusb " #temp ", " #a "        \n\t"\
466 "mov" #size " " #a ", " #b "      \n\t"
467 #define AVG_MMXEXT_OP(a, b, temp, size) \
468 "mov" #size " " #b ", " #temp "   \n\t"\
469 "pavgb " #temp ", " #a "          \n\t"\
470 "mov" #size " " #a ", " #b "      \n\t"
471
472 #endif /* (HAVE_MMXEXT_INLINE || HAVE_AMD3DNOW_INLINE) */
473
474 #if HAVE_MMX_EXTERNAL
475 static void put_cavs_qpel8_mc00_mmx(uint8_t *dst, const uint8_t *src,
476                                     ptrdiff_t stride)
477 {
478     ff_put_pixels8_mmx(dst, src, stride, 8);
479 }
480
481 static void avg_cavs_qpel8_mc00_mmx(uint8_t *dst, const uint8_t *src,
482                                     ptrdiff_t stride)
483 {
484     ff_avg_pixels8_mmx(dst, src, stride, 8);
485 }
486
487 static void avg_cavs_qpel8_mc00_mmxext(uint8_t *dst, const uint8_t *src,
488                                        ptrdiff_t stride)
489 {
490     ff_avg_pixels8_mmxext(dst, src, stride, 8);
491 }
492
493 static void put_cavs_qpel16_mc00_mmx(uint8_t *dst, const uint8_t *src,
494                                      ptrdiff_t stride)
495 {
496     ff_put_pixels16_mmx(dst, src, stride, 16);
497 }
498
499 static void avg_cavs_qpel16_mc00_mmx(uint8_t *dst, const uint8_t *src,
500                                      ptrdiff_t stride)
501 {
502     ff_avg_pixels16_mmx(dst, src, stride, 16);
503 }
504
505 static void avg_cavs_qpel16_mc00_mmxext(uint8_t *dst, const uint8_t *src,
506                                         ptrdiff_t stride)
507 {
508     ff_avg_pixels16_mmxext(dst, src, stride, 16);
509 }
510
511 static void put_cavs_qpel16_mc00_sse2(uint8_t *dst, const uint8_t *src,
512                                       ptrdiff_t stride)
513 {
514     ff_put_pixels16_sse2(dst, src, stride, 16);
515 }
516
517 static void avg_cavs_qpel16_mc00_sse2(uint8_t *dst, const uint8_t *src,
518                                       ptrdiff_t stride)
519 {
520     ff_avg_pixels16_sse2(dst, src, stride, 16);
521 }
522 #endif
523
524 static av_cold void cavsdsp_init_mmx(CAVSDSPContext *c,
525                                      AVCodecContext *avctx)
526 {
527 #if HAVE_MMX_EXTERNAL
528     c->put_cavs_qpel_pixels_tab[0][0] = put_cavs_qpel16_mc00_mmx;
529     c->put_cavs_qpel_pixels_tab[1][0] = put_cavs_qpel8_mc00_mmx;
530     c->avg_cavs_qpel_pixels_tab[0][0] = avg_cavs_qpel16_mc00_mmx;
531     c->avg_cavs_qpel_pixels_tab[1][0] = avg_cavs_qpel8_mc00_mmx;
532 #endif
533
534 #if HAVE_MMX_INLINE
535     c->cavs_idct8_add = cavs_idct8_add_mmx;
536     c->idct_perm      = FF_IDCT_PERM_TRANSPOSE;
537 #endif /* HAVE_MMX_INLINE */
538 }
539
540 #define DSPFUNC(PFX, IDX, NUM, EXT)                                                       \
541     c->PFX ## _cavs_qpel_pixels_tab[IDX][ 2] = PFX ## _cavs_qpel ## NUM ## _mc20_ ## EXT; \
542     c->PFX ## _cavs_qpel_pixels_tab[IDX][ 4] = PFX ## _cavs_qpel ## NUM ## _mc01_ ## EXT; \
543     c->PFX ## _cavs_qpel_pixels_tab[IDX][ 8] = PFX ## _cavs_qpel ## NUM ## _mc02_ ## EXT; \
544     c->PFX ## _cavs_qpel_pixels_tab[IDX][12] = PFX ## _cavs_qpel ## NUM ## _mc03_ ## EXT; \
545
546 #if HAVE_MMXEXT_INLINE
547 QPEL_CAVS(put_,        PUT_OP, mmxext)
548 QPEL_CAVS(avg_, AVG_MMXEXT_OP, mmxext)
549
550 CAVS_MC(put_,  8, mmxext)
551 CAVS_MC(put_, 16, mmxext)
552 CAVS_MC(avg_,  8, mmxext)
553 CAVS_MC(avg_, 16, mmxext)
554 #endif /* HAVE_MMXEXT_INLINE */
555
556 #if HAVE_AMD3DNOW_INLINE
557 QPEL_CAVS(put_,       PUT_OP, 3dnow)
558 QPEL_CAVS(avg_, AVG_3DNOW_OP, 3dnow)
559
560 CAVS_MC(put_, 8, 3dnow)
561 CAVS_MC(put_, 16,3dnow)
562 CAVS_MC(avg_, 8, 3dnow)
563 CAVS_MC(avg_, 16,3dnow)
564
565 static av_cold void cavsdsp_init_3dnow(CAVSDSPContext *c,
566                                        AVCodecContext *avctx)
567 {
568     DSPFUNC(put, 0, 16, 3dnow);
569     DSPFUNC(put, 1,  8, 3dnow);
570     DSPFUNC(avg, 0, 16, 3dnow);
571     DSPFUNC(avg, 1,  8, 3dnow);
572 }
573 #endif /* HAVE_AMD3DNOW_INLINE */
574
575 av_cold void ff_cavsdsp_init_x86(CAVSDSPContext *c, AVCodecContext *avctx)
576 {
577     av_unused int cpu_flags = av_get_cpu_flags();
578
579     if (X86_MMX(cpu_flags))
580         cavsdsp_init_mmx(c, avctx);
581
582 #if HAVE_AMD3DNOW_INLINE
583     if (INLINE_AMD3DNOW(cpu_flags))
584         cavsdsp_init_3dnow(c, avctx);
585 #endif /* HAVE_AMD3DNOW_INLINE */
586 #if HAVE_MMXEXT_INLINE
587     if (INLINE_MMXEXT(cpu_flags)) {
588         DSPFUNC(put, 0, 16, mmxext);
589         DSPFUNC(put, 1,  8, mmxext);
590         DSPFUNC(avg, 0, 16, mmxext);
591         DSPFUNC(avg, 1,  8, mmxext);
592     }
593 #endif
594 #if HAVE_MMX_EXTERNAL
595     if (EXTERNAL_MMXEXT(cpu_flags)) {
596         c->avg_cavs_qpel_pixels_tab[0][0] = avg_cavs_qpel16_mc00_mmxext;
597         c->avg_cavs_qpel_pixels_tab[1][0] = avg_cavs_qpel8_mc00_mmxext;
598     }
599 #endif
600 #if HAVE_SSE2_EXTERNAL
601     if (EXTERNAL_SSE2(cpu_flags)) {
602         c->put_cavs_qpel_pixels_tab[0][0] = put_cavs_qpel16_mc00_sse2;
603         c->avg_cavs_qpel_pixels_tab[0][0] = avg_cavs_qpel16_mc00_sse2;
604     }
605 #endif
606 }