Move REG_* macros from libavcodec/i386/mmx.h to libavutil/x86_cpu.h
[ffmpeg.git] / libavcodec / i386 / dsputil_mmx.c
1 /*
2  * MMX optimized DSP utils
3  * Copyright (c) 2000, 2001 Fabrice Bellard.
4  * Copyright (c) 2002-2004 Michael Niedermayer <michaelni@gmx.at>
5  *
6  * This library is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU Lesser General Public
8  * License as published by the Free Software Foundation; either
9  * version 2 of the License, or (at your option) any later version.
10  *
11  * This library is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14  * Lesser General Public License for more details.
15  *
16  * You should have received a copy of the GNU Lesser General Public
17  * License along with this library; if not, write to the Free Software
18  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
19  *
20  * MMX optimization by Nick Kurshev <nickols_k@mail.ru>
21  */
22
23 #include "../dsputil.h"
24 #include "../simple_idct.h"
25 #include "../mpegvideo.h"
26 #include "x86_cpu.h"
27 #include "mmx.h"
28
29 //#undef NDEBUG
30 //#include <assert.h>
31
32 extern const uint8_t ff_h263_loop_filter_strength[32];
33 extern void ff_idct_xvid_mmx(short *block);
34 extern void ff_idct_xvid_mmx2(short *block);
35
36 int mm_flags; /* multimedia extension flags */
37
38 /* pixel operations */
39 static const uint64_t mm_bone attribute_used __attribute__ ((aligned(8))) = 0x0101010101010101ULL;
40 static const uint64_t mm_wone attribute_used __attribute__ ((aligned(8))) = 0x0001000100010001ULL;
41 static const uint64_t mm_wtwo attribute_used __attribute__ ((aligned(8))) = 0x0002000200020002ULL;
42
43 static const uint64_t ff_pw_20 attribute_used __attribute__ ((aligned(8))) = 0x0014001400140014ULL;
44 static const uint64_t ff_pw_3  attribute_used __attribute__ ((aligned(8))) = 0x0003000300030003ULL;
45 static const uint64_t ff_pw_4  attribute_used __attribute__ ((aligned(8))) = 0x0004000400040004ULL;
46 static const uint64_t ff_pw_5  attribute_used __attribute__ ((aligned(8))) = 0x0005000500050005ULL;
47 static const uint64_t ff_pw_8  attribute_used __attribute__ ((aligned(8))) = 0x0008000800080008ULL;
48 static const uint64_t ff_pw_16 attribute_used __attribute__ ((aligned(8))) = 0x0010001000100010ULL;
49 static const uint64_t ff_pw_32 attribute_used __attribute__ ((aligned(8))) = 0x0020002000200020ULL;
50 static const uint64_t ff_pw_64 attribute_used __attribute__ ((aligned(8))) = 0x0040004000400040ULL;
51 static const uint64_t ff_pw_15 attribute_used __attribute__ ((aligned(8))) = 0x000F000F000F000FULL;
52
53 static const uint64_t ff_pb_3F attribute_used __attribute__ ((aligned(8))) = 0x3F3F3F3F3F3F3F3FULL;
54 static const uint64_t ff_pb_FC attribute_used __attribute__ ((aligned(8))) = 0xFCFCFCFCFCFCFCFCULL;
55
56 #define JUMPALIGN() __asm __volatile (".balign 8"::)
57 #define MOVQ_ZERO(regd)  __asm __volatile ("pxor %%" #regd ", %%" #regd ::)
58
59 #define MOVQ_WONE(regd) \
60     __asm __volatile ( \
61     "pcmpeqd %%" #regd ", %%" #regd " \n\t" \
62     "psrlw $15, %%" #regd ::)
63
64 #define MOVQ_BFE(regd) \
65     __asm __volatile ( \
66     "pcmpeqd %%" #regd ", %%" #regd " \n\t"\
67     "paddb %%" #regd ", %%" #regd " \n\t" ::)
68
69 #ifndef PIC
70 #define MOVQ_BONE(regd)  __asm __volatile ("movq %0, %%" #regd " \n\t" ::"m"(mm_bone))
71 #define MOVQ_WTWO(regd)  __asm __volatile ("movq %0, %%" #regd " \n\t" ::"m"(mm_wtwo))
72 #else
73 // for shared library it's better to use this way for accessing constants
74 // pcmpeqd -> -1
75 #define MOVQ_BONE(regd) \
76     __asm __volatile ( \
77     "pcmpeqd %%" #regd ", %%" #regd " \n\t" \
78     "psrlw $15, %%" #regd " \n\t" \
79     "packuswb %%" #regd ", %%" #regd " \n\t" ::)
80
81 #define MOVQ_WTWO(regd) \
82     __asm __volatile ( \
83     "pcmpeqd %%" #regd ", %%" #regd " \n\t" \
84     "psrlw $15, %%" #regd " \n\t" \
85     "psllw $1, %%" #regd " \n\t"::)
86
87 #endif
88
89 // using regr as temporary and for the output result
90 // first argument is unmodifed and second is trashed
91 // regfe is supposed to contain 0xfefefefefefefefe
92 #define PAVGB_MMX_NO_RND(rega, regb, regr, regfe) \
93     "movq " #rega ", " #regr "  \n\t"\
94     "pand " #regb ", " #regr "  \n\t"\
95     "pxor " #rega ", " #regb "  \n\t"\
96     "pand " #regfe "," #regb "  \n\t"\
97     "psrlq $1, " #regb "        \n\t"\
98     "paddb " #regb ", " #regr " \n\t"
99
100 #define PAVGB_MMX(rega, regb, regr, regfe) \
101     "movq " #rega ", " #regr "  \n\t"\
102     "por  " #regb ", " #regr "  \n\t"\
103     "pxor " #rega ", " #regb "  \n\t"\
104     "pand " #regfe "," #regb "  \n\t"\
105     "psrlq $1, " #regb "        \n\t"\
106     "psubb " #regb ", " #regr " \n\t"
107
108 // mm6 is supposed to contain 0xfefefefefefefefe
109 #define PAVGBP_MMX_NO_RND(rega, regb, regr,  regc, regd, regp) \
110     "movq " #rega ", " #regr "  \n\t"\
111     "movq " #regc ", " #regp "  \n\t"\
112     "pand " #regb ", " #regr "  \n\t"\
113     "pand " #regd ", " #regp "  \n\t"\
114     "pxor " #rega ", " #regb "  \n\t"\
115     "pxor " #regc ", " #regd "  \n\t"\
116     "pand %%mm6, " #regb "      \n\t"\
117     "pand %%mm6, " #regd "      \n\t"\
118     "psrlq $1, " #regb "        \n\t"\
119     "psrlq $1, " #regd "        \n\t"\
120     "paddb " #regb ", " #regr " \n\t"\
121     "paddb " #regd ", " #regp " \n\t"
122
123 #define PAVGBP_MMX(rega, regb, regr, regc, regd, regp) \
124     "movq " #rega ", " #regr "  \n\t"\
125     "movq " #regc ", " #regp "  \n\t"\
126     "por  " #regb ", " #regr "  \n\t"\
127     "por  " #regd ", " #regp "  \n\t"\
128     "pxor " #rega ", " #regb "  \n\t"\
129     "pxor " #regc ", " #regd "  \n\t"\
130     "pand %%mm6, " #regb "      \n\t"\
131     "pand %%mm6, " #regd "      \n\t"\
132     "psrlq $1, " #regd "        \n\t"\
133     "psrlq $1, " #regb "        \n\t"\
134     "psubb " #regb ", " #regr " \n\t"\
135     "psubb " #regd ", " #regp " \n\t"
136
137 /***********************************/
138 /* MMX no rounding */
139 #define DEF(x, y) x ## _no_rnd_ ## y ##_mmx
140 #define SET_RND  MOVQ_WONE
141 #define PAVGBP(a, b, c, d, e, f)        PAVGBP_MMX_NO_RND(a, b, c, d, e, f)
142 #define PAVGB(a, b, c, e)               PAVGB_MMX_NO_RND(a, b, c, e)
143
144 #include "dsputil_mmx_rnd.h"
145
146 #undef DEF
147 #undef SET_RND
148 #undef PAVGBP
149 #undef PAVGB
150 /***********************************/
151 /* MMX rounding */
152
153 #define DEF(x, y) x ## _ ## y ##_mmx
154 #define SET_RND  MOVQ_WTWO
155 #define PAVGBP(a, b, c, d, e, f)        PAVGBP_MMX(a, b, c, d, e, f)
156 #define PAVGB(a, b, c, e)               PAVGB_MMX(a, b, c, e)
157
158 #include "dsputil_mmx_rnd.h"
159
160 #undef DEF
161 #undef SET_RND
162 #undef PAVGBP
163 #undef PAVGB
164
165 /***********************************/
166 /* 3Dnow specific */
167
168 #define DEF(x) x ## _3dnow
169 /* for Athlons PAVGUSB is prefered */
170 #define PAVGB "pavgusb"
171
172 #include "dsputil_mmx_avg.h"
173
174 #undef DEF
175 #undef PAVGB
176
177 /***********************************/
178 /* MMX2 specific */
179
180 #define DEF(x) x ## _mmx2
181
182 /* Introduced only in MMX2 set */
183 #define PAVGB "pavgb"
184
185 #include "dsputil_mmx_avg.h"
186
187 #undef DEF
188 #undef PAVGB
189
190 /***********************************/
191 /* standard MMX */
192
193 #ifdef CONFIG_ENCODERS
194 static void get_pixels_mmx(DCTELEM *block, const uint8_t *pixels, int line_size)
195 {
196     asm volatile(
197         "mov $-128, %%"REG_a"           \n\t"
198         "pxor %%mm7, %%mm7              \n\t"
199         ".balign 16                     \n\t"
200         "1:                             \n\t"
201         "movq (%0), %%mm0               \n\t"
202         "movq (%0, %2), %%mm2           \n\t"
203         "movq %%mm0, %%mm1              \n\t"
204         "movq %%mm2, %%mm3              \n\t"
205         "punpcklbw %%mm7, %%mm0         \n\t"
206         "punpckhbw %%mm7, %%mm1         \n\t"
207         "punpcklbw %%mm7, %%mm2         \n\t"
208         "punpckhbw %%mm7, %%mm3         \n\t"
209         "movq %%mm0, (%1, %%"REG_a")    \n\t"
210         "movq %%mm1, 8(%1, %%"REG_a")   \n\t"
211         "movq %%mm2, 16(%1, %%"REG_a")  \n\t"
212         "movq %%mm3, 24(%1, %%"REG_a")  \n\t"
213         "add %3, %0                     \n\t"
214         "add $32, %%"REG_a"             \n\t"
215         "js 1b                          \n\t"
216         : "+r" (pixels)
217         : "r" (block+64), "r" ((long)line_size), "r" ((long)line_size*2)
218         : "%"REG_a
219     );
220 }
221
222 static inline void diff_pixels_mmx(DCTELEM *block, const uint8_t *s1, const uint8_t *s2, int stride)
223 {
224     asm volatile(
225         "pxor %%mm7, %%mm7              \n\t"
226         "mov $-128, %%"REG_a"           \n\t"
227         ".balign 16                     \n\t"
228         "1:                             \n\t"
229         "movq (%0), %%mm0               \n\t"
230         "movq (%1), %%mm2               \n\t"
231         "movq %%mm0, %%mm1              \n\t"
232         "movq %%mm2, %%mm3              \n\t"
233         "punpcklbw %%mm7, %%mm0         \n\t"
234         "punpckhbw %%mm7, %%mm1         \n\t"
235         "punpcklbw %%mm7, %%mm2         \n\t"
236         "punpckhbw %%mm7, %%mm3         \n\t"
237         "psubw %%mm2, %%mm0             \n\t"
238         "psubw %%mm3, %%mm1             \n\t"
239         "movq %%mm0, (%2, %%"REG_a")    \n\t"
240         "movq %%mm1, 8(%2, %%"REG_a")   \n\t"
241         "add %3, %0                     \n\t"
242         "add %3, %1                     \n\t"
243         "add $16, %%"REG_a"             \n\t"
244         "jnz 1b                         \n\t"
245         : "+r" (s1), "+r" (s2)
246         : "r" (block+64), "r" ((long)stride)
247         : "%"REG_a
248     );
249 }
250 #endif //CONFIG_ENCODERS
251
252 void put_pixels_clamped_mmx(const DCTELEM *block, uint8_t *pixels, int line_size)
253 {
254     const DCTELEM *p;
255     uint8_t *pix;
256
257     /* read the pixels */
258     p = block;
259     pix = pixels;
260     /* unrolled loop */
261         __asm __volatile(
262                 "movq   %3, %%mm0               \n\t"
263                 "movq   8%3, %%mm1              \n\t"
264                 "movq   16%3, %%mm2             \n\t"
265                 "movq   24%3, %%mm3             \n\t"
266                 "movq   32%3, %%mm4             \n\t"
267                 "movq   40%3, %%mm5             \n\t"
268                 "movq   48%3, %%mm6             \n\t"
269                 "movq   56%3, %%mm7             \n\t"
270                 "packuswb %%mm1, %%mm0          \n\t"
271                 "packuswb %%mm3, %%mm2          \n\t"
272                 "packuswb %%mm5, %%mm4          \n\t"
273                 "packuswb %%mm7, %%mm6          \n\t"
274                 "movq   %%mm0, (%0)             \n\t"
275                 "movq   %%mm2, (%0, %1)         \n\t"
276                 "movq   %%mm4, (%0, %1, 2)      \n\t"
277                 "movq   %%mm6, (%0, %2)         \n\t"
278                 ::"r" (pix), "r" ((long)line_size), "r" ((long)line_size*3), "m"(*p)
279                 :"memory");
280         pix += line_size*4;
281         p += 32;
282
283     // if here would be an exact copy of the code above
284     // compiler would generate some very strange code
285     // thus using "r"
286     __asm __volatile(
287             "movq       (%3), %%mm0             \n\t"
288             "movq       8(%3), %%mm1            \n\t"
289             "movq       16(%3), %%mm2           \n\t"
290             "movq       24(%3), %%mm3           \n\t"
291             "movq       32(%3), %%mm4           \n\t"
292             "movq       40(%3), %%mm5           \n\t"
293             "movq       48(%3), %%mm6           \n\t"
294             "movq       56(%3), %%mm7           \n\t"
295             "packuswb %%mm1, %%mm0              \n\t"
296             "packuswb %%mm3, %%mm2              \n\t"
297             "packuswb %%mm5, %%mm4              \n\t"
298             "packuswb %%mm7, %%mm6              \n\t"
299             "movq       %%mm0, (%0)             \n\t"
300             "movq       %%mm2, (%0, %1)         \n\t"
301             "movq       %%mm4, (%0, %1, 2)      \n\t"
302             "movq       %%mm6, (%0, %2)         \n\t"
303             ::"r" (pix), "r" ((long)line_size), "r" ((long)line_size*3), "r"(p)
304             :"memory");
305 }
306
307 static DECLARE_ALIGNED_8(const unsigned char, vector128[8]) =
308   { 0x80, 0x80, 0x80, 0x80, 0x80, 0x80, 0x80, 0x80 };
309
310 void put_signed_pixels_clamped_mmx(const DCTELEM *block, uint8_t *pixels, int line_size)
311 {
312     int i;
313
314     movq_m2r(*vector128, mm1);
315     for (i = 0; i < 8; i++) {
316         movq_m2r(*(block), mm0);
317         packsswb_m2r(*(block + 4), mm0);
318         block += 8;
319         paddb_r2r(mm1, mm0);
320         movq_r2m(mm0, *pixels);
321         pixels += line_size;
322     }
323 }
324
325 void add_pixels_clamped_mmx(const DCTELEM *block, uint8_t *pixels, int line_size)
326 {
327     const DCTELEM *p;
328     uint8_t *pix;
329     int i;
330
331     /* read the pixels */
332     p = block;
333     pix = pixels;
334     MOVQ_ZERO(mm7);
335     i = 4;
336     do {
337         __asm __volatile(
338                 "movq   (%2), %%mm0     \n\t"
339                 "movq   8(%2), %%mm1    \n\t"
340                 "movq   16(%2), %%mm2   \n\t"
341                 "movq   24(%2), %%mm3   \n\t"
342                 "movq   %0, %%mm4       \n\t"
343                 "movq   %1, %%mm6       \n\t"
344                 "movq   %%mm4, %%mm5    \n\t"
345                 "punpcklbw %%mm7, %%mm4 \n\t"
346                 "punpckhbw %%mm7, %%mm5 \n\t"
347                 "paddsw %%mm4, %%mm0    \n\t"
348                 "paddsw %%mm5, %%mm1    \n\t"
349                 "movq   %%mm6, %%mm5    \n\t"
350                 "punpcklbw %%mm7, %%mm6 \n\t"
351                 "punpckhbw %%mm7, %%mm5 \n\t"
352                 "paddsw %%mm6, %%mm2    \n\t"
353                 "paddsw %%mm5, %%mm3    \n\t"
354                 "packuswb %%mm1, %%mm0  \n\t"
355                 "packuswb %%mm3, %%mm2  \n\t"
356                 "movq   %%mm0, %0       \n\t"
357                 "movq   %%mm2, %1       \n\t"
358                 :"+m"(*pix), "+m"(*(pix+line_size))
359                 :"r"(p)
360                 :"memory");
361         pix += line_size*2;
362         p += 16;
363     } while (--i);
364 }
365
366 static void put_pixels4_mmx(uint8_t *block, const uint8_t *pixels, int line_size, int h)
367 {
368     __asm __volatile(
369          "lea (%3, %3), %%"REG_a"       \n\t"
370          ".balign 8                     \n\t"
371          "1:                            \n\t"
372          "movd (%1), %%mm0              \n\t"
373          "movd (%1, %3), %%mm1          \n\t"
374          "movd %%mm0, (%2)              \n\t"
375          "movd %%mm1, (%2, %3)          \n\t"
376          "add %%"REG_a", %1             \n\t"
377          "add %%"REG_a", %2             \n\t"
378          "movd (%1), %%mm0              \n\t"
379          "movd (%1, %3), %%mm1          \n\t"
380          "movd %%mm0, (%2)              \n\t"
381          "movd %%mm1, (%2, %3)          \n\t"
382          "add %%"REG_a", %1             \n\t"
383          "add %%"REG_a", %2             \n\t"
384          "subl $4, %0                   \n\t"
385          "jnz 1b                        \n\t"
386          : "+g"(h), "+r" (pixels),  "+r" (block)
387          : "r"((long)line_size)
388          : "%"REG_a, "memory"
389         );
390 }
391
392 static void put_pixels8_mmx(uint8_t *block, const uint8_t *pixels, int line_size, int h)
393 {
394     __asm __volatile(
395          "lea (%3, %3), %%"REG_a"       \n\t"
396          ".balign 8                     \n\t"
397          "1:                            \n\t"
398          "movq (%1), %%mm0              \n\t"
399          "movq (%1, %3), %%mm1          \n\t"
400          "movq %%mm0, (%2)              \n\t"
401          "movq %%mm1, (%2, %3)          \n\t"
402          "add %%"REG_a", %1             \n\t"
403          "add %%"REG_a", %2             \n\t"
404          "movq (%1), %%mm0              \n\t"
405          "movq (%1, %3), %%mm1          \n\t"
406          "movq %%mm0, (%2)              \n\t"
407          "movq %%mm1, (%2, %3)          \n\t"
408          "add %%"REG_a", %1             \n\t"
409          "add %%"REG_a", %2             \n\t"
410          "subl $4, %0                   \n\t"
411          "jnz 1b                        \n\t"
412          : "+g"(h), "+r" (pixels),  "+r" (block)
413          : "r"((long)line_size)
414          : "%"REG_a, "memory"
415         );
416 }
417
418 static void put_pixels16_mmx(uint8_t *block, const uint8_t *pixels, int line_size, int h)
419 {
420     __asm __volatile(
421          "lea (%3, %3), %%"REG_a"       \n\t"
422          ".balign 8                     \n\t"
423          "1:                            \n\t"
424          "movq (%1), %%mm0              \n\t"
425          "movq 8(%1), %%mm4             \n\t"
426          "movq (%1, %3), %%mm1          \n\t"
427          "movq 8(%1, %3), %%mm5         \n\t"
428          "movq %%mm0, (%2)              \n\t"
429          "movq %%mm4, 8(%2)             \n\t"
430          "movq %%mm1, (%2, %3)          \n\t"
431          "movq %%mm5, 8(%2, %3)         \n\t"
432          "add %%"REG_a", %1             \n\t"
433          "add %%"REG_a", %2             \n\t"
434          "movq (%1), %%mm0              \n\t"
435          "movq 8(%1), %%mm4             \n\t"
436          "movq (%1, %3), %%mm1          \n\t"
437          "movq 8(%1, %3), %%mm5         \n\t"
438          "movq %%mm0, (%2)              \n\t"
439          "movq %%mm4, 8(%2)             \n\t"
440          "movq %%mm1, (%2, %3)          \n\t"
441          "movq %%mm5, 8(%2, %3)         \n\t"
442          "add %%"REG_a", %1             \n\t"
443          "add %%"REG_a", %2             \n\t"
444          "subl $4, %0                   \n\t"
445          "jnz 1b                        \n\t"
446          : "+g"(h), "+r" (pixels),  "+r" (block)
447          : "r"((long)line_size)
448          : "%"REG_a, "memory"
449         );
450 }
451
452 static void clear_blocks_mmx(DCTELEM *blocks)
453 {
454     __asm __volatile(
455                 "pxor %%mm7, %%mm7              \n\t"
456                 "mov $-128*6, %%"REG_a"         \n\t"
457                 "1:                             \n\t"
458                 "movq %%mm7, (%0, %%"REG_a")    \n\t"
459                 "movq %%mm7, 8(%0, %%"REG_a")   \n\t"
460                 "movq %%mm7, 16(%0, %%"REG_a")  \n\t"
461                 "movq %%mm7, 24(%0, %%"REG_a")  \n\t"
462                 "add $32, %%"REG_a"             \n\t"
463                 " js 1b                         \n\t"
464                 : : "r" (((uint8_t *)blocks)+128*6)
465                 : "%"REG_a
466         );
467 }
468
469 #ifdef CONFIG_ENCODERS
470 static int pix_sum16_mmx(uint8_t * pix, int line_size){
471     const int h=16;
472     int sum;
473     long index= -line_size*h;
474
475     __asm __volatile(
476                 "pxor %%mm7, %%mm7              \n\t"
477                 "pxor %%mm6, %%mm6              \n\t"
478                 "1:                             \n\t"
479                 "movq (%2, %1), %%mm0           \n\t"
480                 "movq (%2, %1), %%mm1           \n\t"
481                 "movq 8(%2, %1), %%mm2          \n\t"
482                 "movq 8(%2, %1), %%mm3          \n\t"
483                 "punpcklbw %%mm7, %%mm0         \n\t"
484                 "punpckhbw %%mm7, %%mm1         \n\t"
485                 "punpcklbw %%mm7, %%mm2         \n\t"
486                 "punpckhbw %%mm7, %%mm3         \n\t"
487                 "paddw %%mm0, %%mm1             \n\t"
488                 "paddw %%mm2, %%mm3             \n\t"
489                 "paddw %%mm1, %%mm3             \n\t"
490                 "paddw %%mm3, %%mm6             \n\t"
491                 "add %3, %1                     \n\t"
492                 " js 1b                         \n\t"
493                 "movq %%mm6, %%mm5              \n\t"
494                 "psrlq $32, %%mm6               \n\t"
495                 "paddw %%mm5, %%mm6             \n\t"
496                 "movq %%mm6, %%mm5              \n\t"
497                 "psrlq $16, %%mm6               \n\t"
498                 "paddw %%mm5, %%mm6             \n\t"
499                 "movd %%mm6, %0                 \n\t"
500                 "andl $0xFFFF, %0               \n\t"
501                 : "=&r" (sum), "+r" (index)
502                 : "r" (pix - index), "r" ((long)line_size)
503         );
504
505         return sum;
506 }
507 #endif //CONFIG_ENCODERS
508
509 static void add_bytes_mmx(uint8_t *dst, uint8_t *src, int w){
510     long i=0;
511     asm volatile(
512         "1:                             \n\t"
513         "movq  (%1, %0), %%mm0          \n\t"
514         "movq  (%2, %0), %%mm1          \n\t"
515         "paddb %%mm0, %%mm1             \n\t"
516         "movq %%mm1, (%2, %0)           \n\t"
517         "movq 8(%1, %0), %%mm0          \n\t"
518         "movq 8(%2, %0), %%mm1          \n\t"
519         "paddb %%mm0, %%mm1             \n\t"
520         "movq %%mm1, 8(%2, %0)          \n\t"
521         "add $16, %0                    \n\t"
522         "cmp %3, %0                     \n\t"
523         " jb 1b                         \n\t"
524         : "+r" (i)
525         : "r"(src), "r"(dst), "r"((long)w-15)
526     );
527     for(; i<w; i++)
528         dst[i+0] += src[i+0];
529 }
530
531 #define H263_LOOP_FILTER \
532         "pxor %%mm7, %%mm7              \n\t"\
533         "movq  %0, %%mm0                \n\t"\
534         "movq  %0, %%mm1                \n\t"\
535         "movq  %3, %%mm2                \n\t"\
536         "movq  %3, %%mm3                \n\t"\
537         "punpcklbw %%mm7, %%mm0         \n\t"\
538         "punpckhbw %%mm7, %%mm1         \n\t"\
539         "punpcklbw %%mm7, %%mm2         \n\t"\
540         "punpckhbw %%mm7, %%mm3         \n\t"\
541         "psubw %%mm2, %%mm0             \n\t"\
542         "psubw %%mm3, %%mm1             \n\t"\
543         "movq  %1, %%mm2                \n\t"\
544         "movq  %1, %%mm3                \n\t"\
545         "movq  %2, %%mm4                \n\t"\
546         "movq  %2, %%mm5                \n\t"\
547         "punpcklbw %%mm7, %%mm2         \n\t"\
548         "punpckhbw %%mm7, %%mm3         \n\t"\
549         "punpcklbw %%mm7, %%mm4         \n\t"\
550         "punpckhbw %%mm7, %%mm5         \n\t"\
551         "psubw %%mm2, %%mm4             \n\t"\
552         "psubw %%mm3, %%mm5             \n\t"\
553         "psllw $2, %%mm4                \n\t"\
554         "psllw $2, %%mm5                \n\t"\
555         "paddw %%mm0, %%mm4             \n\t"\
556         "paddw %%mm1, %%mm5             \n\t"\
557         "pxor %%mm6, %%mm6              \n\t"\
558         "pcmpgtw %%mm4, %%mm6           \n\t"\
559         "pcmpgtw %%mm5, %%mm7           \n\t"\
560         "pxor %%mm6, %%mm4              \n\t"\
561         "pxor %%mm7, %%mm5              \n\t"\
562         "psubw %%mm6, %%mm4             \n\t"\
563         "psubw %%mm7, %%mm5             \n\t"\
564         "psrlw $3, %%mm4                \n\t"\
565         "psrlw $3, %%mm5                \n\t"\
566         "packuswb %%mm5, %%mm4          \n\t"\
567         "packsswb %%mm7, %%mm6          \n\t"\
568         "pxor %%mm7, %%mm7              \n\t"\
569         "movd %4, %%mm2                 \n\t"\
570         "punpcklbw %%mm2, %%mm2         \n\t"\
571         "punpcklbw %%mm2, %%mm2         \n\t"\
572         "punpcklbw %%mm2, %%mm2         \n\t"\
573         "psubusb %%mm4, %%mm2           \n\t"\
574         "movq %%mm2, %%mm3              \n\t"\
575         "psubusb %%mm4, %%mm3           \n\t"\
576         "psubb %%mm3, %%mm2             \n\t"\
577         "movq %1, %%mm3                 \n\t"\
578         "movq %2, %%mm4                 \n\t"\
579         "pxor %%mm6, %%mm3              \n\t"\
580         "pxor %%mm6, %%mm4              \n\t"\
581         "paddusb %%mm2, %%mm3           \n\t"\
582         "psubusb %%mm2, %%mm4           \n\t"\
583         "pxor %%mm6, %%mm3              \n\t"\
584         "pxor %%mm6, %%mm4              \n\t"\
585         "paddusb %%mm2, %%mm2           \n\t"\
586         "packsswb %%mm1, %%mm0          \n\t"\
587         "pcmpgtb %%mm0, %%mm7           \n\t"\
588         "pxor %%mm7, %%mm0              \n\t"\
589         "psubb %%mm7, %%mm0             \n\t"\
590         "movq %%mm0, %%mm1              \n\t"\
591         "psubusb %%mm2, %%mm0           \n\t"\
592         "psubb %%mm0, %%mm1             \n\t"\
593         "pand %5, %%mm1                 \n\t"\
594         "psrlw $2, %%mm1                \n\t"\
595         "pxor %%mm7, %%mm1              \n\t"\
596         "psubb %%mm7, %%mm1             \n\t"\
597         "movq %0, %%mm5                 \n\t"\
598         "movq %3, %%mm6                 \n\t"\
599         "psubb %%mm1, %%mm5             \n\t"\
600         "paddb %%mm1, %%mm6             \n\t"
601
602 static void h263_v_loop_filter_mmx(uint8_t *src, int stride, int qscale){
603     const int strength= ff_h263_loop_filter_strength[qscale];
604
605     asm volatile(
606
607         H263_LOOP_FILTER
608
609         "movq %%mm3, %1                 \n\t"
610         "movq %%mm4, %2                 \n\t"
611         "movq %%mm5, %0                 \n\t"
612         "movq %%mm6, %3                 \n\t"
613         : "+m" (*(uint64_t*)(src - 2*stride)),
614           "+m" (*(uint64_t*)(src - 1*stride)),
615           "+m" (*(uint64_t*)(src + 0*stride)),
616           "+m" (*(uint64_t*)(src + 1*stride))
617         : "g" (2*strength), "m"(ff_pb_FC)
618     );
619 }
620
621 static inline void transpose4x4(uint8_t *dst, uint8_t *src, int dst_stride, int src_stride){
622     asm volatile( //FIXME could save 1 instruction if done as 8x4 ...
623         "movd  %4, %%mm0                \n\t"
624         "movd  %5, %%mm1                \n\t"
625         "movd  %6, %%mm2                \n\t"
626         "movd  %7, %%mm3                \n\t"
627         "punpcklbw %%mm1, %%mm0         \n\t"
628         "punpcklbw %%mm3, %%mm2         \n\t"
629         "movq %%mm0, %%mm1              \n\t"
630         "punpcklwd %%mm2, %%mm0         \n\t"
631         "punpckhwd %%mm2, %%mm1         \n\t"
632         "movd  %%mm0, %0                \n\t"
633         "punpckhdq %%mm0, %%mm0         \n\t"
634         "movd  %%mm0, %1                \n\t"
635         "movd  %%mm1, %2                \n\t"
636         "punpckhdq %%mm1, %%mm1         \n\t"
637         "movd  %%mm1, %3                \n\t"
638
639         : "=m" (*(uint32_t*)(dst + 0*dst_stride)),
640           "=m" (*(uint32_t*)(dst + 1*dst_stride)),
641           "=m" (*(uint32_t*)(dst + 2*dst_stride)),
642           "=m" (*(uint32_t*)(dst + 3*dst_stride))
643         :  "m" (*(uint32_t*)(src + 0*src_stride)),
644            "m" (*(uint32_t*)(src + 1*src_stride)),
645            "m" (*(uint32_t*)(src + 2*src_stride)),
646            "m" (*(uint32_t*)(src + 3*src_stride))
647     );
648 }
649
650 static void h263_h_loop_filter_mmx(uint8_t *src, int stride, int qscale){
651     const int strength= ff_h263_loop_filter_strength[qscale];
652     uint64_t temp[4] __attribute__ ((aligned(8)));
653     uint8_t *btemp= (uint8_t*)temp;
654
655     src -= 2;
656
657     transpose4x4(btemp  , src           , 8, stride);
658     transpose4x4(btemp+4, src + 4*stride, 8, stride);
659     asm volatile(
660         H263_LOOP_FILTER // 5 3 4 6
661
662         : "+m" (temp[0]),
663           "+m" (temp[1]),
664           "+m" (temp[2]),
665           "+m" (temp[3])
666         : "g" (2*strength), "m"(ff_pb_FC)
667     );
668
669     asm volatile(
670         "movq %%mm5, %%mm1              \n\t"
671         "movq %%mm4, %%mm0              \n\t"
672         "punpcklbw %%mm3, %%mm5         \n\t"
673         "punpcklbw %%mm6, %%mm4         \n\t"
674         "punpckhbw %%mm3, %%mm1         \n\t"
675         "punpckhbw %%mm6, %%mm0         \n\t"
676         "movq %%mm5, %%mm3              \n\t"
677         "movq %%mm1, %%mm6              \n\t"
678         "punpcklwd %%mm4, %%mm5         \n\t"
679         "punpcklwd %%mm0, %%mm1         \n\t"
680         "punpckhwd %%mm4, %%mm3         \n\t"
681         "punpckhwd %%mm0, %%mm6         \n\t"
682         "movd %%mm5, (%0)               \n\t"
683         "punpckhdq %%mm5, %%mm5         \n\t"
684         "movd %%mm5, (%0,%2)            \n\t"
685         "movd %%mm3, (%0,%2,2)          \n\t"
686         "punpckhdq %%mm3, %%mm3         \n\t"
687         "movd %%mm3, (%0,%3)            \n\t"
688         "movd %%mm1, (%1)               \n\t"
689         "punpckhdq %%mm1, %%mm1         \n\t"
690         "movd %%mm1, (%1,%2)            \n\t"
691         "movd %%mm6, (%1,%2,2)          \n\t"
692         "punpckhdq %%mm6, %%mm6         \n\t"
693         "movd %%mm6, (%1,%3)            \n\t"
694         :: "r" (src),
695            "r" (src + 4*stride),
696            "r" ((long)   stride ),
697            "r" ((long)(3*stride))
698     );
699 }
700
701 #ifdef CONFIG_ENCODERS
702 static int pix_norm1_mmx(uint8_t *pix, int line_size) {
703     int tmp;
704   asm volatile (
705       "movl $16,%%ecx\n"
706       "pxor %%mm0,%%mm0\n"
707       "pxor %%mm7,%%mm7\n"
708       "1:\n"
709       "movq (%0),%%mm2\n"       /* mm2 = pix[0-7] */
710       "movq 8(%0),%%mm3\n"      /* mm3 = pix[8-15] */
711
712       "movq %%mm2,%%mm1\n"      /* mm1 = mm2 = pix[0-7] */
713
714       "punpckhbw %%mm0,%%mm1\n" /* mm1 = [pix4-7] */
715       "punpcklbw %%mm0,%%mm2\n" /* mm2 = [pix0-3] */
716
717       "movq %%mm3,%%mm4\n"      /* mm4 = mm3 = pix[8-15] */
718       "punpckhbw %%mm0,%%mm3\n" /* mm3 = [pix12-15] */
719       "punpcklbw %%mm0,%%mm4\n" /* mm4 = [pix8-11] */
720
721       "pmaddwd %%mm1,%%mm1\n"   /* mm1 = (pix0^2+pix1^2,pix2^2+pix3^2) */
722       "pmaddwd %%mm2,%%mm2\n"   /* mm2 = (pix4^2+pix5^2,pix6^2+pix7^2) */
723
724       "pmaddwd %%mm3,%%mm3\n"
725       "pmaddwd %%mm4,%%mm4\n"
726
727       "paddd %%mm1,%%mm2\n"     /* mm2 = (pix0^2+pix1^2+pix4^2+pix5^2,
728                                           pix2^2+pix3^2+pix6^2+pix7^2) */
729       "paddd %%mm3,%%mm4\n"
730       "paddd %%mm2,%%mm7\n"
731
732       "add %2, %0\n"
733       "paddd %%mm4,%%mm7\n"
734       "dec %%ecx\n"
735       "jnz 1b\n"
736
737       "movq %%mm7,%%mm1\n"
738       "psrlq $32, %%mm7\n"      /* shift hi dword to lo */
739       "paddd %%mm7,%%mm1\n"
740       "movd %%mm1,%1\n"
741       : "+r" (pix), "=r"(tmp) : "r" ((long)line_size) : "%ecx" );
742     return tmp;
743 }
744
745 static int sse8_mmx(void *v, uint8_t * pix1, uint8_t * pix2, int line_size, int h) {
746     int tmp;
747   asm volatile (
748       "movl %4,%%ecx\n"
749       "shr $1,%%ecx\n"
750       "pxor %%mm0,%%mm0\n"      /* mm0 = 0 */
751       "pxor %%mm7,%%mm7\n"      /* mm7 holds the sum */
752       "1:\n"
753       "movq (%0),%%mm1\n"       /* mm1 = pix1[0][0-7] */
754       "movq (%1),%%mm2\n"       /* mm2 = pix2[0][0-7] */
755       "movq (%0,%3),%%mm3\n"    /* mm3 = pix1[1][0-7] */
756       "movq (%1,%3),%%mm4\n"    /* mm4 = pix2[1][0-7] */
757
758       /* todo: mm1-mm2, mm3-mm4 */
759       /* algo: substract mm1 from mm2 with saturation and vice versa */
760       /*       OR the results to get absolute difference */
761       "movq %%mm1,%%mm5\n"
762       "movq %%mm3,%%mm6\n"
763       "psubusb %%mm2,%%mm1\n"
764       "psubusb %%mm4,%%mm3\n"
765       "psubusb %%mm5,%%mm2\n"
766       "psubusb %%mm6,%%mm4\n"
767
768       "por %%mm1,%%mm2\n"
769       "por %%mm3,%%mm4\n"
770
771       /* now convert to 16-bit vectors so we can square them */
772       "movq %%mm2,%%mm1\n"
773       "movq %%mm4,%%mm3\n"
774
775       "punpckhbw %%mm0,%%mm2\n"
776       "punpckhbw %%mm0,%%mm4\n"
777       "punpcklbw %%mm0,%%mm1\n" /* mm1 now spread over (mm1,mm2) */
778       "punpcklbw %%mm0,%%mm3\n" /* mm4 now spread over (mm3,mm4) */
779
780       "pmaddwd %%mm2,%%mm2\n"
781       "pmaddwd %%mm4,%%mm4\n"
782       "pmaddwd %%mm1,%%mm1\n"
783       "pmaddwd %%mm3,%%mm3\n"
784
785       "lea (%0,%3,2), %0\n"     /* pix1 += 2*line_size */
786       "lea (%1,%3,2), %1\n"     /* pix2 += 2*line_size */
787
788       "paddd %%mm2,%%mm1\n"
789       "paddd %%mm4,%%mm3\n"
790       "paddd %%mm1,%%mm7\n"
791       "paddd %%mm3,%%mm7\n"
792
793       "decl %%ecx\n"
794       "jnz 1b\n"
795
796       "movq %%mm7,%%mm1\n"
797       "psrlq $32, %%mm7\n"      /* shift hi dword to lo */
798       "paddd %%mm7,%%mm1\n"
799       "movd %%mm1,%2\n"
800       : "+r" (pix1), "+r" (pix2), "=r"(tmp)
801       : "r" ((long)line_size) , "m" (h)
802       : "%ecx");
803     return tmp;
804 }
805
806 static int sse16_mmx(void *v, uint8_t * pix1, uint8_t * pix2, int line_size, int h) {
807     int tmp;
808   asm volatile (
809       "movl %4,%%ecx\n"
810       "pxor %%mm0,%%mm0\n"      /* mm0 = 0 */
811       "pxor %%mm7,%%mm7\n"      /* mm7 holds the sum */
812       "1:\n"
813       "movq (%0),%%mm1\n"       /* mm1 = pix1[0-7] */
814       "movq (%1),%%mm2\n"       /* mm2 = pix2[0-7] */
815       "movq 8(%0),%%mm3\n"      /* mm3 = pix1[8-15] */
816       "movq 8(%1),%%mm4\n"      /* mm4 = pix2[8-15] */
817
818       /* todo: mm1-mm2, mm3-mm4 */
819       /* algo: substract mm1 from mm2 with saturation and vice versa */
820       /*       OR the results to get absolute difference */
821       "movq %%mm1,%%mm5\n"
822       "movq %%mm3,%%mm6\n"
823       "psubusb %%mm2,%%mm1\n"
824       "psubusb %%mm4,%%mm3\n"
825       "psubusb %%mm5,%%mm2\n"
826       "psubusb %%mm6,%%mm4\n"
827
828       "por %%mm1,%%mm2\n"
829       "por %%mm3,%%mm4\n"
830
831       /* now convert to 16-bit vectors so we can square them */
832       "movq %%mm2,%%mm1\n"
833       "movq %%mm4,%%mm3\n"
834
835       "punpckhbw %%mm0,%%mm2\n"
836       "punpckhbw %%mm0,%%mm4\n"
837       "punpcklbw %%mm0,%%mm1\n" /* mm1 now spread over (mm1,mm2) */
838       "punpcklbw %%mm0,%%mm3\n" /* mm4 now spread over (mm3,mm4) */
839
840       "pmaddwd %%mm2,%%mm2\n"
841       "pmaddwd %%mm4,%%mm4\n"
842       "pmaddwd %%mm1,%%mm1\n"
843       "pmaddwd %%mm3,%%mm3\n"
844
845       "add %3,%0\n"
846       "add %3,%1\n"
847
848       "paddd %%mm2,%%mm1\n"
849       "paddd %%mm4,%%mm3\n"
850       "paddd %%mm1,%%mm7\n"
851       "paddd %%mm3,%%mm7\n"
852
853       "decl %%ecx\n"
854       "jnz 1b\n"
855
856       "movq %%mm7,%%mm1\n"
857       "psrlq $32, %%mm7\n"      /* shift hi dword to lo */
858       "paddd %%mm7,%%mm1\n"
859       "movd %%mm1,%2\n"
860       : "+r" (pix1), "+r" (pix2), "=r"(tmp)
861       : "r" ((long)line_size) , "m" (h)
862       : "%ecx");
863     return tmp;
864 }
865
866 static int sse16_sse2(void *v, uint8_t * pix1, uint8_t * pix2, int line_size, int h) {
867     int tmp;
868   asm volatile (
869       "shr $1,%2\n"
870       "pxor %%xmm0,%%xmm0\n"    /* mm0 = 0 */
871       "pxor %%xmm7,%%xmm7\n"    /* mm7 holds the sum */
872       "1:\n"
873       "movdqu (%0),%%xmm1\n"    /* mm1 = pix1[0][0-15] */
874       "movdqu (%1),%%xmm2\n"    /* mm2 = pix2[0][0-15] */
875       "movdqu (%0,%4),%%xmm3\n" /* mm3 = pix1[1][0-15] */
876       "movdqu (%1,%4),%%xmm4\n" /* mm4 = pix2[1][0-15] */
877
878       /* todo: mm1-mm2, mm3-mm4 */
879       /* algo: substract mm1 from mm2 with saturation and vice versa */
880       /*       OR the results to get absolute difference */
881       "movdqa %%xmm1,%%xmm5\n"
882       "movdqa %%xmm3,%%xmm6\n"
883       "psubusb %%xmm2,%%xmm1\n"
884       "psubusb %%xmm4,%%xmm3\n"
885       "psubusb %%xmm5,%%xmm2\n"
886       "psubusb %%xmm6,%%xmm4\n"
887
888       "por %%xmm1,%%xmm2\n"
889       "por %%xmm3,%%xmm4\n"
890
891       /* now convert to 16-bit vectors so we can square them */
892       "movdqa %%xmm2,%%xmm1\n"
893       "movdqa %%xmm4,%%xmm3\n"
894
895       "punpckhbw %%xmm0,%%xmm2\n"
896       "punpckhbw %%xmm0,%%xmm4\n"
897       "punpcklbw %%xmm0,%%xmm1\n"  /* mm1 now spread over (mm1,mm2) */
898       "punpcklbw %%xmm0,%%xmm3\n"  /* mm4 now spread over (mm3,mm4) */
899
900       "pmaddwd %%xmm2,%%xmm2\n"
901       "pmaddwd %%xmm4,%%xmm4\n"
902       "pmaddwd %%xmm1,%%xmm1\n"
903       "pmaddwd %%xmm3,%%xmm3\n"
904
905       "lea (%0,%4,2), %0\n"        /* pix1 += 2*line_size */
906       "lea (%1,%4,2), %1\n"        /* pix2 += 2*line_size */
907
908       "paddd %%xmm2,%%xmm1\n"
909       "paddd %%xmm4,%%xmm3\n"
910       "paddd %%xmm1,%%xmm7\n"
911       "paddd %%xmm3,%%xmm7\n"
912
913       "decl %2\n"
914       "jnz 1b\n"
915
916       "movdqa %%xmm7,%%xmm1\n"
917       "psrldq $8, %%xmm7\n"        /* shift hi qword to lo */
918       "paddd %%xmm1,%%xmm7\n"
919       "movdqa %%xmm7,%%xmm1\n"
920       "psrldq $4, %%xmm7\n"        /* shift hi dword to lo */
921       "paddd %%xmm1,%%xmm7\n"
922       "movd %%xmm7,%3\n"
923       : "+r" (pix1), "+r" (pix2), "+r"(h), "=r"(tmp)
924       : "r" ((long)line_size));
925     return tmp;
926 }
927
928 static int hf_noise8_mmx(uint8_t * pix1, int line_size, int h) {
929     int tmp;
930   asm volatile (
931       "movl %3,%%ecx\n"
932       "pxor %%mm7,%%mm7\n"
933       "pxor %%mm6,%%mm6\n"
934
935       "movq (%0),%%mm0\n"
936       "movq %%mm0, %%mm1\n"
937       "psllq $8, %%mm0\n"
938       "psrlq $8, %%mm1\n"
939       "psrlq $8, %%mm0\n"
940       "movq %%mm0, %%mm2\n"
941       "movq %%mm1, %%mm3\n"
942       "punpcklbw %%mm7,%%mm0\n"
943       "punpcklbw %%mm7,%%mm1\n"
944       "punpckhbw %%mm7,%%mm2\n"
945       "punpckhbw %%mm7,%%mm3\n"
946       "psubw %%mm1, %%mm0\n"
947       "psubw %%mm3, %%mm2\n"
948
949       "add %2,%0\n"
950
951       "movq (%0),%%mm4\n"
952       "movq %%mm4, %%mm1\n"
953       "psllq $8, %%mm4\n"
954       "psrlq $8, %%mm1\n"
955       "psrlq $8, %%mm4\n"
956       "movq %%mm4, %%mm5\n"
957       "movq %%mm1, %%mm3\n"
958       "punpcklbw %%mm7,%%mm4\n"
959       "punpcklbw %%mm7,%%mm1\n"
960       "punpckhbw %%mm7,%%mm5\n"
961       "punpckhbw %%mm7,%%mm3\n"
962       "psubw %%mm1, %%mm4\n"
963       "psubw %%mm3, %%mm5\n"
964       "psubw %%mm4, %%mm0\n"
965       "psubw %%mm5, %%mm2\n"
966       "pxor %%mm3, %%mm3\n"
967       "pxor %%mm1, %%mm1\n"
968       "pcmpgtw %%mm0, %%mm3\n\t"
969       "pcmpgtw %%mm2, %%mm1\n\t"
970       "pxor %%mm3, %%mm0\n"
971       "pxor %%mm1, %%mm2\n"
972       "psubw %%mm3, %%mm0\n"
973       "psubw %%mm1, %%mm2\n"
974       "paddw %%mm0, %%mm2\n"
975       "paddw %%mm2, %%mm6\n"
976
977       "add %2,%0\n"
978       "1:\n"
979
980       "movq (%0),%%mm0\n"
981       "movq %%mm0, %%mm1\n"
982       "psllq $8, %%mm0\n"
983       "psrlq $8, %%mm1\n"
984       "psrlq $8, %%mm0\n"
985       "movq %%mm0, %%mm2\n"
986       "movq %%mm1, %%mm3\n"
987       "punpcklbw %%mm7,%%mm0\n"
988       "punpcklbw %%mm7,%%mm1\n"
989       "punpckhbw %%mm7,%%mm2\n"
990       "punpckhbw %%mm7,%%mm3\n"
991       "psubw %%mm1, %%mm0\n"
992       "psubw %%mm3, %%mm2\n"
993       "psubw %%mm0, %%mm4\n"
994       "psubw %%mm2, %%mm5\n"
995       "pxor %%mm3, %%mm3\n"
996       "pxor %%mm1, %%mm1\n"
997       "pcmpgtw %%mm4, %%mm3\n\t"
998       "pcmpgtw %%mm5, %%mm1\n\t"
999       "pxor %%mm3, %%mm4\n"
1000       "pxor %%mm1, %%mm5\n"
1001       "psubw %%mm3, %%mm4\n"
1002       "psubw %%mm1, %%mm5\n"
1003       "paddw %%mm4, %%mm5\n"
1004       "paddw %%mm5, %%mm6\n"
1005
1006       "add %2,%0\n"
1007
1008       "movq (%0),%%mm4\n"
1009       "movq %%mm4, %%mm1\n"
1010       "psllq $8, %%mm4\n"
1011       "psrlq $8, %%mm1\n"
1012       "psrlq $8, %%mm4\n"
1013       "movq %%mm4, %%mm5\n"
1014       "movq %%mm1, %%mm3\n"
1015       "punpcklbw %%mm7,%%mm4\n"
1016       "punpcklbw %%mm7,%%mm1\n"
1017       "punpckhbw %%mm7,%%mm5\n"
1018       "punpckhbw %%mm7,%%mm3\n"
1019       "psubw %%mm1, %%mm4\n"
1020       "psubw %%mm3, %%mm5\n"
1021       "psubw %%mm4, %%mm0\n"
1022       "psubw %%mm5, %%mm2\n"
1023       "pxor %%mm3, %%mm3\n"
1024       "pxor %%mm1, %%mm1\n"
1025       "pcmpgtw %%mm0, %%mm3\n\t"
1026       "pcmpgtw %%mm2, %%mm1\n\t"
1027       "pxor %%mm3, %%mm0\n"
1028       "pxor %%mm1, %%mm2\n"
1029       "psubw %%mm3, %%mm0\n"
1030       "psubw %%mm1, %%mm2\n"
1031       "paddw %%mm0, %%mm2\n"
1032       "paddw %%mm2, %%mm6\n"
1033
1034       "add %2,%0\n"
1035       "subl $2, %%ecx\n"
1036       " jnz 1b\n"
1037
1038       "movq %%mm6, %%mm0\n"
1039       "punpcklwd %%mm7,%%mm0\n"
1040       "punpckhwd %%mm7,%%mm6\n"
1041       "paddd %%mm0, %%mm6\n"
1042
1043       "movq %%mm6,%%mm0\n"
1044       "psrlq $32, %%mm6\n"
1045       "paddd %%mm6,%%mm0\n"
1046       "movd %%mm0,%1\n"
1047       : "+r" (pix1), "=r"(tmp)
1048       : "r" ((long)line_size) , "g" (h-2)
1049       : "%ecx");
1050       return tmp;
1051 }
1052
1053 static int hf_noise16_mmx(uint8_t * pix1, int line_size, int h) {
1054     int tmp;
1055     uint8_t * pix= pix1;
1056   asm volatile (
1057       "movl %3,%%ecx\n"
1058       "pxor %%mm7,%%mm7\n"
1059       "pxor %%mm6,%%mm6\n"
1060
1061       "movq (%0),%%mm0\n"
1062       "movq 1(%0),%%mm1\n"
1063       "movq %%mm0, %%mm2\n"
1064       "movq %%mm1, %%mm3\n"
1065       "punpcklbw %%mm7,%%mm0\n"
1066       "punpcklbw %%mm7,%%mm1\n"
1067       "punpckhbw %%mm7,%%mm2\n"
1068       "punpckhbw %%mm7,%%mm3\n"
1069       "psubw %%mm1, %%mm0\n"
1070       "psubw %%mm3, %%mm2\n"
1071
1072       "add %2,%0\n"
1073
1074       "movq (%0),%%mm4\n"
1075       "movq 1(%0),%%mm1\n"
1076       "movq %%mm4, %%mm5\n"
1077       "movq %%mm1, %%mm3\n"
1078       "punpcklbw %%mm7,%%mm4\n"
1079       "punpcklbw %%mm7,%%mm1\n"
1080       "punpckhbw %%mm7,%%mm5\n"
1081       "punpckhbw %%mm7,%%mm3\n"
1082       "psubw %%mm1, %%mm4\n"
1083       "psubw %%mm3, %%mm5\n"
1084       "psubw %%mm4, %%mm0\n"
1085       "psubw %%mm5, %%mm2\n"
1086       "pxor %%mm3, %%mm3\n"
1087       "pxor %%mm1, %%mm1\n"
1088       "pcmpgtw %%mm0, %%mm3\n\t"
1089       "pcmpgtw %%mm2, %%mm1\n\t"
1090       "pxor %%mm3, %%mm0\n"
1091       "pxor %%mm1, %%mm2\n"
1092       "psubw %%mm3, %%mm0\n"
1093       "psubw %%mm1, %%mm2\n"
1094       "paddw %%mm0, %%mm2\n"
1095       "paddw %%mm2, %%mm6\n"
1096
1097       "add %2,%0\n"
1098       "1:\n"
1099
1100       "movq (%0),%%mm0\n"
1101       "movq 1(%0),%%mm1\n"
1102       "movq %%mm0, %%mm2\n"
1103       "movq %%mm1, %%mm3\n"
1104       "punpcklbw %%mm7,%%mm0\n"
1105       "punpcklbw %%mm7,%%mm1\n"
1106       "punpckhbw %%mm7,%%mm2\n"
1107       "punpckhbw %%mm7,%%mm3\n"
1108       "psubw %%mm1, %%mm0\n"
1109       "psubw %%mm3, %%mm2\n"
1110       "psubw %%mm0, %%mm4\n"
1111       "psubw %%mm2, %%mm5\n"
1112       "pxor %%mm3, %%mm3\n"
1113       "pxor %%mm1, %%mm1\n"
1114       "pcmpgtw %%mm4, %%mm3\n\t"
1115       "pcmpgtw %%mm5, %%mm1\n\t"
1116       "pxor %%mm3, %%mm4\n"
1117       "pxor %%mm1, %%mm5\n"
1118       "psubw %%mm3, %%mm4\n"
1119       "psubw %%mm1, %%mm5\n"
1120       "paddw %%mm4, %%mm5\n"
1121       "paddw %%mm5, %%mm6\n"
1122
1123       "add %2,%0\n"
1124
1125       "movq (%0),%%mm4\n"
1126       "movq 1(%0),%%mm1\n"
1127       "movq %%mm4, %%mm5\n"
1128       "movq %%mm1, %%mm3\n"
1129       "punpcklbw %%mm7,%%mm4\n"
1130       "punpcklbw %%mm7,%%mm1\n"
1131       "punpckhbw %%mm7,%%mm5\n"
1132       "punpckhbw %%mm7,%%mm3\n"
1133       "psubw %%mm1, %%mm4\n"
1134       "psubw %%mm3, %%mm5\n"
1135       "psubw %%mm4, %%mm0\n"
1136       "psubw %%mm5, %%mm2\n"
1137       "pxor %%mm3, %%mm3\n"
1138       "pxor %%mm1, %%mm1\n"
1139       "pcmpgtw %%mm0, %%mm3\n\t"
1140       "pcmpgtw %%mm2, %%mm1\n\t"
1141       "pxor %%mm3, %%mm0\n"
1142       "pxor %%mm1, %%mm2\n"
1143       "psubw %%mm3, %%mm0\n"
1144       "psubw %%mm1, %%mm2\n"
1145       "paddw %%mm0, %%mm2\n"
1146       "paddw %%mm2, %%mm6\n"
1147
1148       "add %2,%0\n"
1149       "subl $2, %%ecx\n"
1150       " jnz 1b\n"
1151
1152       "movq %%mm6, %%mm0\n"
1153       "punpcklwd %%mm7,%%mm0\n"
1154       "punpckhwd %%mm7,%%mm6\n"
1155       "paddd %%mm0, %%mm6\n"
1156
1157       "movq %%mm6,%%mm0\n"
1158       "psrlq $32, %%mm6\n"
1159       "paddd %%mm6,%%mm0\n"
1160       "movd %%mm0,%1\n"
1161       : "+r" (pix1), "=r"(tmp)
1162       : "r" ((long)line_size) , "g" (h-2)
1163       : "%ecx");
1164       return tmp + hf_noise8_mmx(pix+8, line_size, h);
1165 }
1166
1167 static int nsse16_mmx(void *p, uint8_t * pix1, uint8_t * pix2, int line_size, int h) {
1168     MpegEncContext *c = p;
1169     int score1, score2;
1170
1171     if(c) score1 = c->dsp.sse[0](c, pix1, pix2, line_size, h);
1172     else  score1 = sse16_mmx(c, pix1, pix2, line_size, h);
1173     score2= hf_noise16_mmx(pix1, line_size, h) - hf_noise16_mmx(pix2, line_size, h);
1174
1175     if(c) return score1 + ABS(score2)*c->avctx->nsse_weight;
1176     else  return score1 + ABS(score2)*8;
1177 }
1178
1179 static int nsse8_mmx(void *p, uint8_t * pix1, uint8_t * pix2, int line_size, int h) {
1180     MpegEncContext *c = p;
1181     int score1= sse8_mmx(c, pix1, pix2, line_size, h);
1182     int score2= hf_noise8_mmx(pix1, line_size, h) - hf_noise8_mmx(pix2, line_size, h);
1183
1184     if(c) return score1 + ABS(score2)*c->avctx->nsse_weight;
1185     else  return score1 + ABS(score2)*8;
1186 }
1187
1188 static int vsad_intra16_mmx(void *v, uint8_t * pix, uint8_t * dummy, int line_size, int h) {
1189     int tmp;
1190
1191     assert( (((int)pix) & 7) == 0);
1192     assert((line_size &7) ==0);
1193
1194 #define SUM(in0, in1, out0, out1) \
1195       "movq (%0), %%mm2\n"\
1196       "movq 8(%0), %%mm3\n"\
1197       "add %2,%0\n"\
1198       "movq %%mm2, " #out0 "\n"\
1199       "movq %%mm3, " #out1 "\n"\
1200       "psubusb " #in0 ", %%mm2\n"\
1201       "psubusb " #in1 ", %%mm3\n"\
1202       "psubusb " #out0 ", " #in0 "\n"\
1203       "psubusb " #out1 ", " #in1 "\n"\
1204       "por %%mm2, " #in0 "\n"\
1205       "por %%mm3, " #in1 "\n"\
1206       "movq " #in0 ", %%mm2\n"\
1207       "movq " #in1 ", %%mm3\n"\
1208       "punpcklbw %%mm7, " #in0 "\n"\
1209       "punpcklbw %%mm7, " #in1 "\n"\
1210       "punpckhbw %%mm7, %%mm2\n"\
1211       "punpckhbw %%mm7, %%mm3\n"\
1212       "paddw " #in1 ", " #in0 "\n"\
1213       "paddw %%mm3, %%mm2\n"\
1214       "paddw %%mm2, " #in0 "\n"\
1215       "paddw " #in0 ", %%mm6\n"
1216
1217
1218   asm volatile (
1219       "movl %3,%%ecx\n"
1220       "pxor %%mm6,%%mm6\n"
1221       "pxor %%mm7,%%mm7\n"
1222       "movq (%0),%%mm0\n"
1223       "movq 8(%0),%%mm1\n"
1224       "add %2,%0\n"
1225       "subl $2, %%ecx\n"
1226       SUM(%%mm0, %%mm1, %%mm4, %%mm5)
1227       "1:\n"
1228
1229       SUM(%%mm4, %%mm5, %%mm0, %%mm1)
1230
1231       SUM(%%mm0, %%mm1, %%mm4, %%mm5)
1232
1233       "subl $2, %%ecx\n"
1234       "jnz 1b\n"
1235
1236       "movq %%mm6,%%mm0\n"
1237       "psrlq $32, %%mm6\n"
1238       "paddw %%mm6,%%mm0\n"
1239       "movq %%mm0,%%mm6\n"
1240       "psrlq $16, %%mm0\n"
1241       "paddw %%mm6,%%mm0\n"
1242       "movd %%mm0,%1\n"
1243       : "+r" (pix), "=r"(tmp)
1244       : "r" ((long)line_size) , "m" (h)
1245       : "%ecx");
1246     return tmp & 0xFFFF;
1247 }
1248 #undef SUM
1249
1250 static int vsad_intra16_mmx2(void *v, uint8_t * pix, uint8_t * dummy, int line_size, int h) {
1251     int tmp;
1252
1253     assert( (((int)pix) & 7) == 0);
1254     assert((line_size &7) ==0);
1255
1256 #define SUM(in0, in1, out0, out1) \
1257       "movq (%0), " #out0 "\n"\
1258       "movq 8(%0), " #out1 "\n"\
1259       "add %2,%0\n"\
1260       "psadbw " #out0 ", " #in0 "\n"\
1261       "psadbw " #out1 ", " #in1 "\n"\
1262       "paddw " #in1 ", " #in0 "\n"\
1263       "paddw " #in0 ", %%mm6\n"
1264
1265   asm volatile (
1266       "movl %3,%%ecx\n"
1267       "pxor %%mm6,%%mm6\n"
1268       "pxor %%mm7,%%mm7\n"
1269       "movq (%0),%%mm0\n"
1270       "movq 8(%0),%%mm1\n"
1271       "add %2,%0\n"
1272       "subl $2, %%ecx\n"
1273       SUM(%%mm0, %%mm1, %%mm4, %%mm5)
1274       "1:\n"
1275
1276       SUM(%%mm4, %%mm5, %%mm0, %%mm1)
1277
1278       SUM(%%mm0, %%mm1, %%mm4, %%mm5)
1279
1280       "subl $2, %%ecx\n"
1281       "jnz 1b\n"
1282
1283       "movd %%mm6,%1\n"
1284       : "+r" (pix), "=r"(tmp)
1285       : "r" ((long)line_size) , "m" (h)
1286       : "%ecx");
1287     return tmp;
1288 }
1289 #undef SUM
1290
1291 static int vsad16_mmx(void *v, uint8_t * pix1, uint8_t * pix2, int line_size, int h) {
1292     int tmp;
1293
1294     assert( (((int)pix1) & 7) == 0);
1295     assert( (((int)pix2) & 7) == 0);
1296     assert((line_size &7) ==0);
1297
1298 #define SUM(in0, in1, out0, out1) \
1299       "movq (%0),%%mm2\n"\
1300       "movq (%1)," #out0 "\n"\
1301       "movq 8(%0),%%mm3\n"\
1302       "movq 8(%1)," #out1 "\n"\
1303       "add %3,%0\n"\
1304       "add %3,%1\n"\
1305       "psubb " #out0 ", %%mm2\n"\
1306       "psubb " #out1 ", %%mm3\n"\
1307       "pxor %%mm7, %%mm2\n"\
1308       "pxor %%mm7, %%mm3\n"\
1309       "movq %%mm2, " #out0 "\n"\
1310       "movq %%mm3, " #out1 "\n"\
1311       "psubusb " #in0 ", %%mm2\n"\
1312       "psubusb " #in1 ", %%mm3\n"\
1313       "psubusb " #out0 ", " #in0 "\n"\
1314       "psubusb " #out1 ", " #in1 "\n"\
1315       "por %%mm2, " #in0 "\n"\
1316       "por %%mm3, " #in1 "\n"\
1317       "movq " #in0 ", %%mm2\n"\
1318       "movq " #in1 ", %%mm3\n"\
1319       "punpcklbw %%mm7, " #in0 "\n"\
1320       "punpcklbw %%mm7, " #in1 "\n"\
1321       "punpckhbw %%mm7, %%mm2\n"\
1322       "punpckhbw %%mm7, %%mm3\n"\
1323       "paddw " #in1 ", " #in0 "\n"\
1324       "paddw %%mm3, %%mm2\n"\
1325       "paddw %%mm2, " #in0 "\n"\
1326       "paddw " #in0 ", %%mm6\n"
1327
1328
1329   asm volatile (
1330       "movl %4,%%ecx\n"
1331       "pxor %%mm6,%%mm6\n"
1332       "pcmpeqw %%mm7,%%mm7\n"
1333       "psllw $15, %%mm7\n"
1334       "packsswb %%mm7, %%mm7\n"
1335       "movq (%0),%%mm0\n"
1336       "movq (%1),%%mm2\n"
1337       "movq 8(%0),%%mm1\n"
1338       "movq 8(%1),%%mm3\n"
1339       "add %3,%0\n"
1340       "add %3,%1\n"
1341       "subl $2, %%ecx\n"
1342       "psubb %%mm2, %%mm0\n"
1343       "psubb %%mm3, %%mm1\n"
1344       "pxor %%mm7, %%mm0\n"
1345       "pxor %%mm7, %%mm1\n"
1346       SUM(%%mm0, %%mm1, %%mm4, %%mm5)
1347       "1:\n"
1348
1349       SUM(%%mm4, %%mm5, %%mm0, %%mm1)
1350
1351       SUM(%%mm0, %%mm1, %%mm4, %%mm5)
1352
1353       "subl $2, %%ecx\n"
1354       "jnz 1b\n"
1355
1356       "movq %%mm6,%%mm0\n"
1357       "psrlq $32, %%mm6\n"
1358       "paddw %%mm6,%%mm0\n"
1359       "movq %%mm0,%%mm6\n"
1360       "psrlq $16, %%mm0\n"
1361       "paddw %%mm6,%%mm0\n"
1362       "movd %%mm0,%2\n"
1363       : "+r" (pix1), "+r" (pix2), "=r"(tmp)
1364       : "r" ((long)line_size) , "m" (h)
1365       : "%ecx");
1366     return tmp & 0x7FFF;
1367 }
1368 #undef SUM
1369
1370 static int vsad16_mmx2(void *v, uint8_t * pix1, uint8_t * pix2, int line_size, int h) {
1371     int tmp;
1372
1373     assert( (((int)pix1) & 7) == 0);
1374     assert( (((int)pix2) & 7) == 0);
1375     assert((line_size &7) ==0);
1376
1377 #define SUM(in0, in1, out0, out1) \
1378       "movq (%0)," #out0 "\n"\
1379       "movq (%1),%%mm2\n"\
1380       "movq 8(%0)," #out1 "\n"\
1381       "movq 8(%1),%%mm3\n"\
1382       "add %3,%0\n"\
1383       "add %3,%1\n"\
1384       "psubb %%mm2, " #out0 "\n"\
1385       "psubb %%mm3, " #out1 "\n"\
1386       "pxor %%mm7, " #out0 "\n"\
1387       "pxor %%mm7, " #out1 "\n"\
1388       "psadbw " #out0 ", " #in0 "\n"\
1389       "psadbw " #out1 ", " #in1 "\n"\
1390       "paddw " #in1 ", " #in0 "\n"\
1391       "paddw " #in0 ", %%mm6\n"
1392
1393   asm volatile (
1394       "movl %4,%%ecx\n"
1395       "pxor %%mm6,%%mm6\n"
1396       "pcmpeqw %%mm7,%%mm7\n"
1397       "psllw $15, %%mm7\n"
1398       "packsswb %%mm7, %%mm7\n"
1399       "movq (%0),%%mm0\n"
1400       "movq (%1),%%mm2\n"
1401       "movq 8(%0),%%mm1\n"
1402       "movq 8(%1),%%mm3\n"
1403       "add %3,%0\n"
1404       "add %3,%1\n"
1405       "subl $2, %%ecx\n"
1406       "psubb %%mm2, %%mm0\n"
1407       "psubb %%mm3, %%mm1\n"
1408       "pxor %%mm7, %%mm0\n"
1409       "pxor %%mm7, %%mm1\n"
1410       SUM(%%mm0, %%mm1, %%mm4, %%mm5)
1411       "1:\n"
1412
1413       SUM(%%mm4, %%mm5, %%mm0, %%mm1)
1414
1415       SUM(%%mm0, %%mm1, %%mm4, %%mm5)
1416
1417       "subl $2, %%ecx\n"
1418       "jnz 1b\n"
1419
1420       "movd %%mm6,%2\n"
1421       : "+r" (pix1), "+r" (pix2), "=r"(tmp)
1422       : "r" ((long)line_size) , "m" (h)
1423       : "%ecx");
1424     return tmp;
1425 }
1426 #undef SUM
1427
1428 static void diff_bytes_mmx(uint8_t *dst, uint8_t *src1, uint8_t *src2, int w){
1429     long i=0;
1430     asm volatile(
1431         "1:                             \n\t"
1432         "movq  (%2, %0), %%mm0          \n\t"
1433         "movq  (%1, %0), %%mm1          \n\t"
1434         "psubb %%mm0, %%mm1             \n\t"
1435         "movq %%mm1, (%3, %0)           \n\t"
1436         "movq 8(%2, %0), %%mm0          \n\t"
1437         "movq 8(%1, %0), %%mm1          \n\t"
1438         "psubb %%mm0, %%mm1             \n\t"
1439         "movq %%mm1, 8(%3, %0)          \n\t"
1440         "add $16, %0                    \n\t"
1441         "cmp %4, %0                     \n\t"
1442         " jb 1b                         \n\t"
1443         : "+r" (i)
1444         : "r"(src1), "r"(src2), "r"(dst), "r"((long)w-15)
1445     );
1446     for(; i<w; i++)
1447         dst[i+0] = src1[i+0]-src2[i+0];
1448 }
1449
1450 static void sub_hfyu_median_prediction_mmx2(uint8_t *dst, uint8_t *src1, uint8_t *src2, int w, int *left, int *left_top){
1451     long i=0;
1452     uint8_t l, lt;
1453
1454     asm volatile(
1455         "1:                             \n\t"
1456         "movq  -1(%1, %0), %%mm0        \n\t" // LT
1457         "movq  (%1, %0), %%mm1          \n\t" // T
1458         "movq  -1(%2, %0), %%mm2        \n\t" // L
1459         "movq  (%2, %0), %%mm3          \n\t" // X
1460         "movq %%mm2, %%mm4              \n\t" // L
1461         "psubb %%mm0, %%mm2             \n\t"
1462         "paddb %%mm1, %%mm2             \n\t" // L + T - LT
1463         "movq %%mm4, %%mm5              \n\t" // L
1464         "pmaxub %%mm1, %%mm4            \n\t" // max(T, L)
1465         "pminub %%mm5, %%mm1            \n\t" // min(T, L)
1466         "pminub %%mm2, %%mm4            \n\t"
1467         "pmaxub %%mm1, %%mm4            \n\t"
1468         "psubb %%mm4, %%mm3             \n\t" // dst - pred
1469         "movq %%mm3, (%3, %0)           \n\t"
1470         "add $8, %0                     \n\t"
1471         "cmp %4, %0                     \n\t"
1472         " jb 1b                         \n\t"
1473         : "+r" (i)
1474         : "r"(src1), "r"(src2), "r"(dst), "r"((long)w)
1475     );
1476
1477     l= *left;
1478     lt= *left_top;
1479
1480     dst[0]= src2[0] - mid_pred(l, src1[0], (l + src1[0] - lt)&0xFF);
1481
1482     *left_top= src1[w-1];
1483     *left    = src2[w-1];
1484 }
1485
1486 #define LBUTTERFLY2(a1,b1,a2,b2)\
1487     "paddw " #b1 ", " #a1 "           \n\t"\
1488     "paddw " #b2 ", " #a2 "           \n\t"\
1489     "paddw " #b1 ", " #b1 "           \n\t"\
1490     "paddw " #b2 ", " #b2 "           \n\t"\
1491     "psubw " #a1 ", " #b1 "           \n\t"\
1492     "psubw " #a2 ", " #b2 "           \n\t"
1493
1494 #define HADAMARD48\
1495         LBUTTERFLY2(%%mm0, %%mm1, %%mm2, %%mm3)\
1496         LBUTTERFLY2(%%mm4, %%mm5, %%mm6, %%mm7)\
1497         LBUTTERFLY2(%%mm0, %%mm2, %%mm1, %%mm3)\
1498         LBUTTERFLY2(%%mm4, %%mm6, %%mm5, %%mm7)\
1499         LBUTTERFLY2(%%mm0, %%mm4, %%mm1, %%mm5)\
1500         LBUTTERFLY2(%%mm2, %%mm6, %%mm3, %%mm7)\
1501
1502 #define MMABS(a,z)\
1503     "pxor " #z ", " #z "              \n\t"\
1504     "pcmpgtw " #a ", " #z "           \n\t"\
1505     "pxor " #z ", " #a "              \n\t"\
1506     "psubw " #z ", " #a "             \n\t"
1507
1508 #define MMABS_SUM(a,z, sum)\
1509     "pxor " #z ", " #z "              \n\t"\
1510     "pcmpgtw " #a ", " #z "           \n\t"\
1511     "pxor " #z ", " #a "              \n\t"\
1512     "psubw " #z ", " #a "             \n\t"\
1513     "paddusw " #a ", " #sum "         \n\t"
1514
1515 #define MMABS_MMX2(a,z)\
1516     "pxor " #z ", " #z "              \n\t"\
1517     "psubw " #a ", " #z "             \n\t"\
1518     "pmaxsw " #z ", " #a "            \n\t"
1519
1520 #define MMABS_SUM_MMX2(a,z, sum)\
1521     "pxor " #z ", " #z "              \n\t"\
1522     "psubw " #a ", " #z "             \n\t"\
1523     "pmaxsw " #z ", " #a "            \n\t"\
1524     "paddusw " #a ", " #sum "         \n\t"
1525
1526 #define SBUTTERFLY(a,b,t,n)\
1527     "movq " #a ", " #t "              \n\t" /* abcd */\
1528     "punpckl" #n " " #b ", " #a "     \n\t" /* aebf */\
1529     "punpckh" #n " " #b ", " #t "     \n\t" /* cgdh */\
1530
1531 #define TRANSPOSE4(a,b,c,d,t)\
1532     SBUTTERFLY(a,b,t,wd) /* a=aebf t=cgdh */\
1533     SBUTTERFLY(c,d,b,wd) /* c=imjn b=kolp */\
1534     SBUTTERFLY(a,c,d,dq) /* a=aeim d=bfjn */\
1535     SBUTTERFLY(t,b,c,dq) /* t=cgko c=dhlp */
1536
1537 #define LOAD4(o, a, b, c, d)\
1538         "movq "#o"(%1), " #a "        \n\t"\
1539         "movq "#o"+16(%1), " #b "     \n\t"\
1540         "movq "#o"+32(%1), " #c "     \n\t"\
1541         "movq "#o"+48(%1), " #d "     \n\t"
1542
1543 #define STORE4(o, a, b, c, d)\
1544         "movq "#a", "#o"(%1)          \n\t"\
1545         "movq "#b", "#o"+16(%1)       \n\t"\
1546         "movq "#c", "#o"+32(%1)       \n\t"\
1547         "movq "#d", "#o"+48(%1)       \n\t"\
1548
1549 static int hadamard8_diff_mmx(void *s, uint8_t *src1, uint8_t *src2, int stride, int h){
1550     DECLARE_ALIGNED_8(uint64_t, temp[16]);
1551     int sum=0;
1552
1553     assert(h==8);
1554
1555     diff_pixels_mmx((DCTELEM*)temp, src1, src2, stride);
1556
1557     asm volatile(
1558         LOAD4(0 , %%mm0, %%mm1, %%mm2, %%mm3)
1559         LOAD4(64, %%mm4, %%mm5, %%mm6, %%mm7)
1560
1561         HADAMARD48
1562
1563         "movq %%mm7, 112(%1)            \n\t"
1564
1565         TRANSPOSE4(%%mm0, %%mm1, %%mm2, %%mm3, %%mm7)
1566         STORE4(0 , %%mm0, %%mm3, %%mm7, %%mm2)
1567
1568         "movq 112(%1), %%mm7            \n\t"
1569         TRANSPOSE4(%%mm4, %%mm5, %%mm6, %%mm7, %%mm0)
1570         STORE4(64, %%mm4, %%mm7, %%mm0, %%mm6)
1571
1572         LOAD4(8 , %%mm0, %%mm1, %%mm2, %%mm3)
1573         LOAD4(72, %%mm4, %%mm5, %%mm6, %%mm7)
1574
1575         HADAMARD48
1576
1577         "movq %%mm7, 120(%1)            \n\t"
1578
1579         TRANSPOSE4(%%mm0, %%mm1, %%mm2, %%mm3, %%mm7)
1580         STORE4(8 , %%mm0, %%mm3, %%mm7, %%mm2)
1581
1582         "movq 120(%1), %%mm7            \n\t"
1583         TRANSPOSE4(%%mm4, %%mm5, %%mm6, %%mm7, %%mm0)
1584         "movq %%mm7, %%mm5              \n\t"//FIXME remove
1585         "movq %%mm6, %%mm7              \n\t"
1586         "movq %%mm0, %%mm6              \n\t"
1587 //        STORE4(72, %%mm4, %%mm7, %%mm0, %%mm6) //FIXME remove
1588
1589         LOAD4(64, %%mm0, %%mm1, %%mm2, %%mm3)
1590 //        LOAD4(72, %%mm4, %%mm5, %%mm6, %%mm7)
1591
1592         HADAMARD48
1593         "movq %%mm7, 64(%1)             \n\t"
1594         MMABS(%%mm0, %%mm7)
1595         MMABS_SUM(%%mm1, %%mm7, %%mm0)
1596         MMABS_SUM(%%mm2, %%mm7, %%mm0)
1597         MMABS_SUM(%%mm3, %%mm7, %%mm0)
1598         MMABS_SUM(%%mm4, %%mm7, %%mm0)
1599         MMABS_SUM(%%mm5, %%mm7, %%mm0)
1600         MMABS_SUM(%%mm6, %%mm7, %%mm0)
1601         "movq 64(%1), %%mm1             \n\t"
1602         MMABS_SUM(%%mm1, %%mm7, %%mm0)
1603         "movq %%mm0, 64(%1)             \n\t"
1604
1605         LOAD4(0 , %%mm0, %%mm1, %%mm2, %%mm3)
1606         LOAD4(8 , %%mm4, %%mm5, %%mm6, %%mm7)
1607
1608         HADAMARD48
1609         "movq %%mm7, (%1)               \n\t"
1610         MMABS(%%mm0, %%mm7)
1611         MMABS_SUM(%%mm1, %%mm7, %%mm0)
1612         MMABS_SUM(%%mm2, %%mm7, %%mm0)
1613         MMABS_SUM(%%mm3, %%mm7, %%mm0)
1614         MMABS_SUM(%%mm4, %%mm7, %%mm0)
1615         MMABS_SUM(%%mm5, %%mm7, %%mm0)
1616         MMABS_SUM(%%mm6, %%mm7, %%mm0)
1617         "movq (%1), %%mm1               \n\t"
1618         MMABS_SUM(%%mm1, %%mm7, %%mm0)
1619         "movq 64(%1), %%mm1             \n\t"
1620         MMABS_SUM(%%mm1, %%mm7, %%mm0)
1621
1622         "movq %%mm0, %%mm1              \n\t"
1623         "psrlq $32, %%mm0               \n\t"
1624         "paddusw %%mm1, %%mm0           \n\t"
1625         "movq %%mm0, %%mm1              \n\t"
1626         "psrlq $16, %%mm0               \n\t"
1627         "paddusw %%mm1, %%mm0           \n\t"
1628         "movd %%mm0, %0                 \n\t"
1629
1630         : "=r" (sum)
1631         : "r"(temp)
1632     );
1633     return sum&0xFFFF;
1634 }
1635
1636 static int hadamard8_diff_mmx2(void *s, uint8_t *src1, uint8_t *src2, int stride, int h){
1637     DECLARE_ALIGNED_8(uint64_t, temp[16]);
1638     int sum=0;
1639
1640     assert(h==8);
1641
1642     diff_pixels_mmx((DCTELEM*)temp, src1, src2, stride);
1643
1644     asm volatile(
1645         LOAD4(0 , %%mm0, %%mm1, %%mm2, %%mm3)
1646         LOAD4(64, %%mm4, %%mm5, %%mm6, %%mm7)
1647
1648         HADAMARD48
1649
1650         "movq %%mm7, 112(%1)            \n\t"
1651
1652         TRANSPOSE4(%%mm0, %%mm1, %%mm2, %%mm3, %%mm7)
1653         STORE4(0 , %%mm0, %%mm3, %%mm7, %%mm2)
1654
1655         "movq 112(%1), %%mm7            \n\t"
1656         TRANSPOSE4(%%mm4, %%mm5, %%mm6, %%mm7, %%mm0)
1657         STORE4(64, %%mm4, %%mm7, %%mm0, %%mm6)
1658
1659         LOAD4(8 , %%mm0, %%mm1, %%mm2, %%mm3)
1660         LOAD4(72, %%mm4, %%mm5, %%mm6, %%mm7)
1661
1662         HADAMARD48
1663
1664         "movq %%mm7, 120(%1)            \n\t"
1665
1666         TRANSPOSE4(%%mm0, %%mm1, %%mm2, %%mm3, %%mm7)
1667         STORE4(8 , %%mm0, %%mm3, %%mm7, %%mm2)
1668
1669         "movq 120(%1), %%mm7            \n\t"
1670         TRANSPOSE4(%%mm4, %%mm5, %%mm6, %%mm7, %%mm0)
1671         "movq %%mm7, %%mm5              \n\t"//FIXME remove
1672         "movq %%mm6, %%mm7              \n\t"
1673         "movq %%mm0, %%mm6              \n\t"
1674 //        STORE4(72, %%mm4, %%mm7, %%mm0, %%mm6) //FIXME remove
1675
1676         LOAD4(64, %%mm0, %%mm1, %%mm2, %%mm3)
1677 //        LOAD4(72, %%mm4, %%mm5, %%mm6, %%mm7)
1678
1679         HADAMARD48
1680         "movq %%mm7, 64(%1)             \n\t"
1681         MMABS_MMX2(%%mm0, %%mm7)
1682         MMABS_SUM_MMX2(%%mm1, %%mm7, %%mm0)
1683         MMABS_SUM_MMX2(%%mm2, %%mm7, %%mm0)
1684         MMABS_SUM_MMX2(%%mm3, %%mm7, %%mm0)
1685         MMABS_SUM_MMX2(%%mm4, %%mm7, %%mm0)
1686         MMABS_SUM_MMX2(%%mm5, %%mm7, %%mm0)
1687         MMABS_SUM_MMX2(%%mm6, %%mm7, %%mm0)
1688         "movq 64(%1), %%mm1             \n\t"
1689         MMABS_SUM_MMX2(%%mm1, %%mm7, %%mm0)
1690         "movq %%mm0, 64(%1)             \n\t"
1691
1692         LOAD4(0 , %%mm0, %%mm1, %%mm2, %%mm3)
1693         LOAD4(8 , %%mm4, %%mm5, %%mm6, %%mm7)
1694
1695         HADAMARD48
1696         "movq %%mm7, (%1)               \n\t"
1697         MMABS_MMX2(%%mm0, %%mm7)
1698         MMABS_SUM_MMX2(%%mm1, %%mm7, %%mm0)
1699         MMABS_SUM_MMX2(%%mm2, %%mm7, %%mm0)
1700         MMABS_SUM_MMX2(%%mm3, %%mm7, %%mm0)
1701         MMABS_SUM_MMX2(%%mm4, %%mm7, %%mm0)
1702         MMABS_SUM_MMX2(%%mm5, %%mm7, %%mm0)
1703         MMABS_SUM_MMX2(%%mm6, %%mm7, %%mm0)
1704         "movq (%1), %%mm1               \n\t"
1705         MMABS_SUM_MMX2(%%mm1, %%mm7, %%mm0)
1706         "movq 64(%1), %%mm1             \n\t"
1707         MMABS_SUM_MMX2(%%mm1, %%mm7, %%mm0)
1708
1709         "pshufw $0x0E, %%mm0, %%mm1     \n\t"
1710         "paddusw %%mm1, %%mm0           \n\t"
1711         "pshufw $0x01, %%mm0, %%mm1     \n\t"
1712         "paddusw %%mm1, %%mm0           \n\t"
1713         "movd %%mm0, %0                 \n\t"
1714
1715         : "=r" (sum)
1716         : "r"(temp)
1717     );
1718     return sum&0xFFFF;
1719 }
1720
1721
1722 WARPER8_16_SQ(hadamard8_diff_mmx, hadamard8_diff16_mmx)
1723 WARPER8_16_SQ(hadamard8_diff_mmx2, hadamard8_diff16_mmx2)
1724 #endif //CONFIG_ENCODERS
1725
1726 #define put_no_rnd_pixels8_mmx(a,b,c,d) put_pixels8_mmx(a,b,c,d)
1727 #define put_no_rnd_pixels16_mmx(a,b,c,d) put_pixels16_mmx(a,b,c,d)
1728
1729 #define QPEL_V_LOW(m3,m4,m5,m6, pw_20, pw_3, rnd, in0, in1, in2, in7, out, OP)\
1730         "paddw " #m4 ", " #m3 "           \n\t" /* x1 */\
1731         "movq "MANGLE(ff_pw_20)", %%mm4   \n\t" /* 20 */\
1732         "pmullw " #m3 ", %%mm4            \n\t" /* 20x1 */\
1733         "movq "#in7", " #m3 "             \n\t" /* d */\
1734         "movq "#in0", %%mm5               \n\t" /* D */\
1735         "paddw " #m3 ", %%mm5             \n\t" /* x4 */\
1736         "psubw %%mm5, %%mm4               \n\t" /* 20x1 - x4 */\
1737         "movq "#in1", %%mm5               \n\t" /* C */\
1738         "movq "#in2", %%mm6               \n\t" /* B */\
1739         "paddw " #m6 ", %%mm5             \n\t" /* x3 */\
1740         "paddw " #m5 ", %%mm6             \n\t" /* x2 */\
1741         "paddw %%mm6, %%mm6               \n\t" /* 2x2 */\
1742         "psubw %%mm6, %%mm5               \n\t" /* -2x2 + x3 */\
1743         "pmullw "MANGLE(ff_pw_3)", %%mm5  \n\t" /* -6x2 + 3x3 */\
1744         "paddw " #rnd ", %%mm4            \n\t" /* x2 */\
1745         "paddw %%mm4, %%mm5               \n\t" /* 20x1 - 6x2 + 3x3 - x4 */\
1746         "psraw $5, %%mm5                  \n\t"\
1747         "packuswb %%mm5, %%mm5            \n\t"\
1748         OP(%%mm5, out, %%mm7, d)
1749
1750 #define QPEL_BASE(OPNAME, ROUNDER, RND, OP_MMX2, OP_3DNOW)\
1751 static void OPNAME ## mpeg4_qpel16_h_lowpass_mmx2(uint8_t *dst, uint8_t *src, int dstStride, int srcStride, int h){\
1752     uint64_t temp;\
1753 \
1754     asm volatile(\
1755         "pxor %%mm7, %%mm7                \n\t"\
1756         "1:                               \n\t"\
1757         "movq  (%0), %%mm0                \n\t" /* ABCDEFGH */\
1758         "movq %%mm0, %%mm1                \n\t" /* ABCDEFGH */\
1759         "movq %%mm0, %%mm2                \n\t" /* ABCDEFGH */\
1760         "punpcklbw %%mm7, %%mm0           \n\t" /* 0A0B0C0D */\
1761         "punpckhbw %%mm7, %%mm1           \n\t" /* 0E0F0G0H */\
1762         "pshufw $0x90, %%mm0, %%mm5       \n\t" /* 0A0A0B0C */\
1763         "pshufw $0x41, %%mm0, %%mm6       \n\t" /* 0B0A0A0B */\
1764         "movq %%mm2, %%mm3                \n\t" /* ABCDEFGH */\
1765         "movq %%mm2, %%mm4                \n\t" /* ABCDEFGH */\
1766         "psllq $8, %%mm2                  \n\t" /* 0ABCDEFG */\
1767         "psllq $16, %%mm3                 \n\t" /* 00ABCDEF */\
1768         "psllq $24, %%mm4                 \n\t" /* 000ABCDE */\
1769         "punpckhbw %%mm7, %%mm2           \n\t" /* 0D0E0F0G */\
1770         "punpckhbw %%mm7, %%mm3           \n\t" /* 0C0D0E0F */\
1771         "punpckhbw %%mm7, %%mm4           \n\t" /* 0B0C0D0E */\
1772         "paddw %%mm3, %%mm5               \n\t" /* b */\
1773         "paddw %%mm2, %%mm6               \n\t" /* c */\
1774         "paddw %%mm5, %%mm5               \n\t" /* 2b */\
1775         "psubw %%mm5, %%mm6               \n\t" /* c - 2b */\
1776         "pshufw $0x06, %%mm0, %%mm5       \n\t" /* 0C0B0A0A */\
1777         "pmullw "MANGLE(ff_pw_3)", %%mm6  \n\t" /* 3c - 6b */\
1778         "paddw %%mm4, %%mm0               \n\t" /* a */\
1779         "paddw %%mm1, %%mm5               \n\t" /* d */\
1780         "pmullw "MANGLE(ff_pw_20)", %%mm0 \n\t" /* 20a */\
1781         "psubw %%mm5, %%mm0               \n\t" /* 20a - d */\
1782         "paddw %6, %%mm6                  \n\t"\
1783         "paddw %%mm6, %%mm0               \n\t" /* 20a - 6b + 3c - d */\
1784         "psraw $5, %%mm0                  \n\t"\
1785         "movq %%mm0, %5                   \n\t"\
1786         /* mm1=EFGH, mm2=DEFG, mm3=CDEF, mm4=BCDE, mm7=0 */\
1787         \
1788         "movq 5(%0), %%mm0                \n\t" /* FGHIJKLM */\
1789         "movq %%mm0, %%mm5                \n\t" /* FGHIJKLM */\
1790         "movq %%mm0, %%mm6                \n\t" /* FGHIJKLM */\
1791         "psrlq $8, %%mm0                  \n\t" /* GHIJKLM0 */\
1792         "psrlq $16, %%mm5                 \n\t" /* HIJKLM00 */\
1793         "punpcklbw %%mm7, %%mm0           \n\t" /* 0G0H0I0J */\
1794         "punpcklbw %%mm7, %%mm5           \n\t" /* 0H0I0J0K */\
1795         "paddw %%mm0, %%mm2               \n\t" /* b */\
1796         "paddw %%mm5, %%mm3               \n\t" /* c */\
1797         "paddw %%mm2, %%mm2               \n\t" /* 2b */\
1798         "psubw %%mm2, %%mm3               \n\t" /* c - 2b */\
1799         "movq %%mm6, %%mm2                \n\t" /* FGHIJKLM */\
1800         "psrlq $24, %%mm6                 \n\t" /* IJKLM000 */\
1801         "punpcklbw %%mm7, %%mm2           \n\t" /* 0F0G0H0I */\
1802         "punpcklbw %%mm7, %%mm6           \n\t" /* 0I0J0K0L */\
1803         "pmullw "MANGLE(ff_pw_3)", %%mm3  \n\t" /* 3c - 6b */\
1804         "paddw %%mm2, %%mm1               \n\t" /* a */\
1805         "paddw %%mm6, %%mm4               \n\t" /* d */\
1806         "pmullw "MANGLE(ff_pw_20)", %%mm1 \n\t" /* 20a */\
1807         "psubw %%mm4, %%mm3               \n\t" /* - 6b +3c - d */\
1808         "paddw %6, %%mm1                  \n\t"\
1809         "paddw %%mm1, %%mm3               \n\t" /* 20a - 6b +3c - d */\
1810         "psraw $5, %%mm3                  \n\t"\
1811         "movq %5, %%mm1                   \n\t"\
1812         "packuswb %%mm3, %%mm1            \n\t"\
1813         OP_MMX2(%%mm1, (%1),%%mm4, q)\
1814         /* mm0= GHIJ, mm2=FGHI, mm5=HIJK, mm6=IJKL, mm7=0 */\
1815         \
1816         "movq 9(%0), %%mm1                \n\t" /* JKLMNOPQ */\
1817         "movq %%mm1, %%mm4                \n\t" /* JKLMNOPQ */\
1818         "movq %%mm1, %%mm3                \n\t" /* JKLMNOPQ */\
1819         "psrlq $8, %%mm1                  \n\t" /* KLMNOPQ0 */\
1820         "psrlq $16, %%mm4                 \n\t" /* LMNOPQ00 */\
1821         "punpcklbw %%mm7, %%mm1           \n\t" /* 0K0L0M0N */\
1822         "punpcklbw %%mm7, %%mm4           \n\t" /* 0L0M0N0O */\
1823         "paddw %%mm1, %%mm5               \n\t" /* b */\
1824         "paddw %%mm4, %%mm0               \n\t" /* c */\
1825         "paddw %%mm5, %%mm5               \n\t" /* 2b */\
1826         "psubw %%mm5, %%mm0               \n\t" /* c - 2b */\
1827         "movq %%mm3, %%mm5                \n\t" /* JKLMNOPQ */\
1828         "psrlq $24, %%mm3                 \n\t" /* MNOPQ000 */\
1829         "pmullw "MANGLE(ff_pw_3)", %%mm0  \n\t" /* 3c - 6b */\
1830         "punpcklbw %%mm7, %%mm3           \n\t" /* 0M0N0O0P */\
1831         "paddw %%mm3, %%mm2               \n\t" /* d */\
1832         "psubw %%mm2, %%mm0               \n\t" /* -6b + 3c - d */\
1833         "movq %%mm5, %%mm2                \n\t" /* JKLMNOPQ */\
1834         "punpcklbw %%mm7, %%mm2           \n\t" /* 0J0K0L0M */\
1835         "punpckhbw %%mm7, %%mm5           \n\t" /* 0N0O0P0Q */\
1836         "paddw %%mm2, %%mm6               \n\t" /* a */\
1837         "pmullw "MANGLE(ff_pw_20)", %%mm6 \n\t" /* 20a */\
1838         "paddw %6, %%mm0                  \n\t"\
1839         "paddw %%mm6, %%mm0               \n\t" /* 20a - 6b + 3c - d */\
1840         "psraw $5, %%mm0                  \n\t"\
1841         /* mm1=KLMN, mm2=JKLM, mm3=MNOP, mm4=LMNO, mm5=NOPQ mm7=0 */\
1842         \
1843         "paddw %%mm5, %%mm3               \n\t" /* a */\
1844         "pshufw $0xF9, %%mm5, %%mm6       \n\t" /* 0O0P0Q0Q */\
1845         "paddw %%mm4, %%mm6               \n\t" /* b */\
1846         "pshufw $0xBE, %%mm5, %%mm4       \n\t" /* 0P0Q0Q0P */\
1847         "pshufw $0x6F, %%mm5, %%mm5       \n\t" /* 0Q0Q0P0O */\
1848         "paddw %%mm1, %%mm4               \n\t" /* c */\
1849         "paddw %%mm2, %%mm5               \n\t" /* d */\
1850         "paddw %%mm6, %%mm6               \n\t" /* 2b */\
1851         "psubw %%mm6, %%mm4               \n\t" /* c - 2b */\
1852         "pmullw "MANGLE(ff_pw_20)", %%mm3 \n\t" /* 20a */\
1853         "pmullw "MANGLE(ff_pw_3)", %%mm4  \n\t" /* 3c - 6b */\
1854         "psubw %%mm5, %%mm3               \n\t" /* -6b + 3c - d */\
1855         "paddw %6, %%mm4                  \n\t"\
1856         "paddw %%mm3, %%mm4               \n\t" /* 20a - 6b + 3c - d */\
1857         "psraw $5, %%mm4                  \n\t"\
1858         "packuswb %%mm4, %%mm0            \n\t"\
1859         OP_MMX2(%%mm0, 8(%1), %%mm4, q)\
1860         \
1861         "add %3, %0                       \n\t"\
1862         "add %4, %1                       \n\t"\
1863         "decl %2                          \n\t"\
1864         " jnz 1b                          \n\t"\
1865         : "+a"(src), "+c"(dst), "+m"(h)\
1866         : "d"((long)srcStride), "S"((long)dstStride), /*"m"(ff_pw_20), "m"(ff_pw_3),*/ "m"(temp), "m"(ROUNDER)\
1867         : "memory"\
1868     );\
1869 }\
1870 \
1871 static void OPNAME ## mpeg4_qpel16_h_lowpass_3dnow(uint8_t *dst, uint8_t *src, int dstStride, int srcStride, int h){\
1872     int i;\
1873     int16_t temp[16];\
1874     /* quick HACK, XXX FIXME MUST be optimized */\
1875     for(i=0; i<h; i++)\
1876     {\
1877         temp[ 0]= (src[ 0]+src[ 1])*20 - (src[ 0]+src[ 2])*6 + (src[ 1]+src[ 3])*3 - (src[ 2]+src[ 4]);\
1878         temp[ 1]= (src[ 1]+src[ 2])*20 - (src[ 0]+src[ 3])*6 + (src[ 0]+src[ 4])*3 - (src[ 1]+src[ 5]);\
1879         temp[ 2]= (src[ 2]+src[ 3])*20 - (src[ 1]+src[ 4])*6 + (src[ 0]+src[ 5])*3 - (src[ 0]+src[ 6]);\
1880         temp[ 3]= (src[ 3]+src[ 4])*20 - (src[ 2]+src[ 5])*6 + (src[ 1]+src[ 6])*3 - (src[ 0]+src[ 7]);\
1881         temp[ 4]= (src[ 4]+src[ 5])*20 - (src[ 3]+src[ 6])*6 + (src[ 2]+src[ 7])*3 - (src[ 1]+src[ 8]);\
1882         temp[ 5]= (src[ 5]+src[ 6])*20 - (src[ 4]+src[ 7])*6 + (src[ 3]+src[ 8])*3 - (src[ 2]+src[ 9]);\
1883         temp[ 6]= (src[ 6]+src[ 7])*20 - (src[ 5]+src[ 8])*6 + (src[ 4]+src[ 9])*3 - (src[ 3]+src[10]);\
1884         temp[ 7]= (src[ 7]+src[ 8])*20 - (src[ 6]+src[ 9])*6 + (src[ 5]+src[10])*3 - (src[ 4]+src[11]);\
1885         temp[ 8]= (src[ 8]+src[ 9])*20 - (src[ 7]+src[10])*6 + (src[ 6]+src[11])*3 - (src[ 5]+src[12]);\
1886         temp[ 9]= (src[ 9]+src[10])*20 - (src[ 8]+src[11])*6 + (src[ 7]+src[12])*3 - (src[ 6]+src[13]);\
1887         temp[10]= (src[10]+src[11])*20 - (src[ 9]+src[12])*6 + (src[ 8]+src[13])*3 - (src[ 7]+src[14]);\
1888         temp[11]= (src[11]+src[12])*20 - (src[10]+src[13])*6 + (src[ 9]+src[14])*3 - (src[ 8]+src[15]);\
1889         temp[12]= (src[12]+src[13])*20 - (src[11]+src[14])*6 + (src[10]+src[15])*3 - (src[ 9]+src[16]);\
1890         temp[13]= (src[13]+src[14])*20 - (src[12]+src[15])*6 + (src[11]+src[16])*3 - (src[10]+src[16]);\
1891         temp[14]= (src[14]+src[15])*20 - (src[13]+src[16])*6 + (src[12]+src[16])*3 - (src[11]+src[15]);\
1892         temp[15]= (src[15]+src[16])*20 - (src[14]+src[16])*6 + (src[13]+src[15])*3 - (src[12]+src[14]);\
1893         asm volatile(\
1894             "movq (%0), %%mm0               \n\t"\
1895             "movq 8(%0), %%mm1              \n\t"\
1896             "paddw %2, %%mm0                \n\t"\
1897             "paddw %2, %%mm1                \n\t"\
1898             "psraw $5, %%mm0                \n\t"\
1899             "psraw $5, %%mm1                \n\t"\
1900             "packuswb %%mm1, %%mm0          \n\t"\
1901             OP_3DNOW(%%mm0, (%1), %%mm1, q)\
1902             "movq 16(%0), %%mm0             \n\t"\
1903             "movq 24(%0), %%mm1             \n\t"\
1904             "paddw %2, %%mm0                \n\t"\
1905             "paddw %2, %%mm1                \n\t"\
1906             "psraw $5, %%mm0                \n\t"\
1907             "psraw $5, %%mm1                \n\t"\
1908             "packuswb %%mm1, %%mm0          \n\t"\
1909             OP_3DNOW(%%mm0, 8(%1), %%mm1, q)\
1910             :: "r"(temp), "r"(dst), "m"(ROUNDER)\
1911             : "memory"\
1912         );\
1913         dst+=dstStride;\
1914         src+=srcStride;\
1915     }\
1916 }\
1917 \
1918 static void OPNAME ## mpeg4_qpel8_h_lowpass_mmx2(uint8_t *dst, uint8_t *src, int dstStride, int srcStride, int h){\
1919     uint64_t temp;\
1920 \
1921     asm volatile(\
1922         "pxor %%mm7, %%mm7                \n\t"\
1923         "1:                               \n\t"\
1924         "movq  (%0), %%mm0                \n\t" /* ABCDEFGH */\
1925         "movq %%mm0, %%mm1                \n\t" /* ABCDEFGH */\
1926         "movq %%mm0, %%mm2                \n\t" /* ABCDEFGH */\
1927         "punpcklbw %%mm7, %%mm0           \n\t" /* 0A0B0C0D */\
1928         "punpckhbw %%mm7, %%mm1           \n\t" /* 0E0F0G0H */\
1929         "pshufw $0x90, %%mm0, %%mm5       \n\t" /* 0A0A0B0C */\
1930         "pshufw $0x41, %%mm0, %%mm6       \n\t" /* 0B0A0A0B */\
1931         "movq %%mm2, %%mm3                \n\t" /* ABCDEFGH */\
1932         "movq %%mm2, %%mm4                \n\t" /* ABCDEFGH */\
1933         "psllq $8, %%mm2                  \n\t" /* 0ABCDEFG */\
1934         "psllq $16, %%mm3                 \n\t" /* 00ABCDEF */\
1935         "psllq $24, %%mm4                 \n\t" /* 000ABCDE */\
1936         "punpckhbw %%mm7, %%mm2           \n\t" /* 0D0E0F0G */\
1937         "punpckhbw %%mm7, %%mm3           \n\t" /* 0C0D0E0F */\
1938         "punpckhbw %%mm7, %%mm4           \n\t" /* 0B0C0D0E */\
1939         "paddw %%mm3, %%mm5               \n\t" /* b */\
1940         "paddw %%mm2, %%mm6               \n\t" /* c */\
1941         "paddw %%mm5, %%mm5               \n\t" /* 2b */\
1942         "psubw %%mm5, %%mm6               \n\t" /* c - 2b */\
1943         "pshufw $0x06, %%mm0, %%mm5       \n\t" /* 0C0B0A0A */\
1944         "pmullw "MANGLE(ff_pw_3)", %%mm6  \n\t" /* 3c - 6b */\
1945         "paddw %%mm4, %%mm0               \n\t" /* a */\
1946         "paddw %%mm1, %%mm5               \n\t" /* d */\
1947         "pmullw "MANGLE(ff_pw_20)", %%mm0 \n\t" /* 20a */\
1948         "psubw %%mm5, %%mm0               \n\t" /* 20a - d */\
1949         "paddw %6, %%mm6                  \n\t"\
1950         "paddw %%mm6, %%mm0               \n\t" /* 20a - 6b + 3c - d */\
1951         "psraw $5, %%mm0                  \n\t"\
1952         /* mm1=EFGH, mm2=DEFG, mm3=CDEF, mm4=BCDE, mm7=0 */\
1953         \
1954         "movd 5(%0), %%mm5                \n\t" /* FGHI */\
1955         "punpcklbw %%mm7, %%mm5           \n\t" /* 0F0G0H0I */\
1956         "pshufw $0xF9, %%mm5, %%mm6       \n\t" /* 0G0H0I0I */\
1957         "paddw %%mm5, %%mm1               \n\t" /* a */\
1958         "paddw %%mm6, %%mm2               \n\t" /* b */\
1959         "pshufw $0xBE, %%mm5, %%mm6       \n\t" /* 0H0I0I0H */\
1960         "pshufw $0x6F, %%mm5, %%mm5       \n\t" /* 0I0I0H0G */\
1961         "paddw %%mm6, %%mm3               \n\t" /* c */\
1962         "paddw %%mm5, %%mm4               \n\t" /* d */\
1963         "paddw %%mm2, %%mm2               \n\t" /* 2b */\
1964         "psubw %%mm2, %%mm3               \n\t" /* c - 2b */\
1965         "pmullw "MANGLE(ff_pw_20)", %%mm1 \n\t" /* 20a */\
1966         "pmullw "MANGLE(ff_pw_3)", %%mm3  \n\t" /* 3c - 6b */\
1967         "psubw %%mm4, %%mm3               \n\t" /* -6b + 3c - d */\
1968         "paddw %6, %%mm1                  \n\t"\
1969         "paddw %%mm1, %%mm3               \n\t" /* 20a - 6b + 3c - d */\
1970         "psraw $5, %%mm3                  \n\t"\
1971         "packuswb %%mm3, %%mm0            \n\t"\
1972         OP_MMX2(%%mm0, (%1), %%mm4, q)\
1973         \
1974         "add %3, %0                       \n\t"\
1975         "add %4, %1                       \n\t"\
1976         "decl %2                          \n\t"\
1977         " jnz 1b                          \n\t"\
1978         : "+a"(src), "+c"(dst), "+m"(h)\
1979         : "S"((long)srcStride), "D"((long)dstStride), /*"m"(ff_pw_20), "m"(ff_pw_3),*/ "m"(temp), "m"(ROUNDER)\
1980         : "memory"\
1981     );\
1982 }\
1983 \
1984 static void OPNAME ## mpeg4_qpel8_h_lowpass_3dnow(uint8_t *dst, uint8_t *src, int dstStride, int srcStride, int h){\
1985     int i;\
1986     int16_t temp[8];\
1987     /* quick HACK, XXX FIXME MUST be optimized */\
1988     for(i=0; i<h; i++)\
1989     {\
1990         temp[ 0]= (src[ 0]+src[ 1])*20 - (src[ 0]+src[ 2])*6 + (src[ 1]+src[ 3])*3 - (src[ 2]+src[ 4]);\
1991         temp[ 1]= (src[ 1]+src[ 2])*20 - (src[ 0]+src[ 3])*6 + (src[ 0]+src[ 4])*3 - (src[ 1]+src[ 5]);\
1992         temp[ 2]= (src[ 2]+src[ 3])*20 - (src[ 1]+src[ 4])*6 + (src[ 0]+src[ 5])*3 - (src[ 0]+src[ 6]);\
1993         temp[ 3]= (src[ 3]+src[ 4])*20 - (src[ 2]+src[ 5])*6 + (src[ 1]+src[ 6])*3 - (src[ 0]+src[ 7]);\
1994         temp[ 4]= (src[ 4]+src[ 5])*20 - (src[ 3]+src[ 6])*6 + (src[ 2]+src[ 7])*3 - (src[ 1]+src[ 8]);\
1995         temp[ 5]= (src[ 5]+src[ 6])*20 - (src[ 4]+src[ 7])*6 + (src[ 3]+src[ 8])*3 - (src[ 2]+src[ 8]);\
1996         temp[ 6]= (src[ 6]+src[ 7])*20 - (src[ 5]+src[ 8])*6 + (src[ 4]+src[ 8])*3 - (src[ 3]+src[ 7]);\
1997         temp[ 7]= (src[ 7]+src[ 8])*20 - (src[ 6]+src[ 8])*6 + (src[ 5]+src[ 7])*3 - (src[ 4]+src[ 6]);\
1998         asm volatile(\
1999             "movq (%0), %%mm0           \n\t"\
2000             "movq 8(%0), %%mm1          \n\t"\
2001             "paddw %2, %%mm0            \n\t"\
2002             "paddw %2, %%mm1            \n\t"\
2003             "psraw $5, %%mm0            \n\t"\
2004             "psraw $5, %%mm1            \n\t"\
2005             "packuswb %%mm1, %%mm0      \n\t"\
2006             OP_3DNOW(%%mm0, (%1), %%mm1, q)\
2007             :: "r"(temp), "r"(dst), "m"(ROUNDER)\
2008             :"memory"\
2009         );\
2010         dst+=dstStride;\
2011         src+=srcStride;\
2012     }\
2013 }
2014
2015 #define QPEL_OP(OPNAME, ROUNDER, RND, OP, MMX)\
2016 \
2017 static void OPNAME ## mpeg4_qpel16_v_lowpass_ ## MMX(uint8_t *dst, uint8_t *src, int dstStride, int srcStride){\
2018     uint64_t temp[17*4];\
2019     uint64_t *temp_ptr= temp;\
2020     int count= 17;\
2021 \
2022     /*FIXME unroll */\
2023     asm volatile(\
2024         "pxor %%mm7, %%mm7              \n\t"\
2025         "1:                             \n\t"\
2026         "movq (%0), %%mm0               \n\t"\
2027         "movq (%0), %%mm1               \n\t"\
2028         "movq 8(%0), %%mm2              \n\t"\
2029         "movq 8(%0), %%mm3              \n\t"\
2030         "punpcklbw %%mm7, %%mm0         \n\t"\
2031         "punpckhbw %%mm7, %%mm1         \n\t"\
2032         "punpcklbw %%mm7, %%mm2         \n\t"\
2033         "punpckhbw %%mm7, %%mm3         \n\t"\
2034         "movq %%mm0, (%1)               \n\t"\
2035         "movq %%mm1, 17*8(%1)           \n\t"\
2036         "movq %%mm2, 2*17*8(%1)         \n\t"\
2037         "movq %%mm3, 3*17*8(%1)         \n\t"\
2038         "add $8, %1                     \n\t"\
2039         "add %3, %0                     \n\t"\
2040         "decl %2                        \n\t"\
2041         " jnz 1b                        \n\t"\
2042         : "+r" (src), "+r" (temp_ptr), "+r"(count)\
2043         : "r" ((long)srcStride)\
2044         : "memory"\
2045     );\
2046     \
2047     temp_ptr= temp;\
2048     count=4;\
2049     \
2050 /*FIXME reorder for speed */\
2051     asm volatile(\
2052         /*"pxor %%mm7, %%mm7              \n\t"*/\
2053         "1:                             \n\t"\
2054         "movq (%0), %%mm0               \n\t"\
2055         "movq 8(%0), %%mm1              \n\t"\
2056         "movq 16(%0), %%mm2             \n\t"\
2057         "movq 24(%0), %%mm3             \n\t"\
2058         QPEL_V_LOW(%%mm0, %%mm1, %%mm2, %%mm3, %5, %6, %5, 16(%0),  8(%0),   (%0), 32(%0), (%1), OP)\
2059         QPEL_V_LOW(%%mm1, %%mm2, %%mm3, %%mm0, %5, %6, %5,  8(%0),   (%0),   (%0), 40(%0), (%1, %3), OP)\
2060         "add %4, %1                     \n\t"\
2061         QPEL_V_LOW(%%mm2, %%mm3, %%mm0, %%mm1, %5, %6, %5,   (%0),   (%0),  8(%0), 48(%0), (%1), OP)\
2062         \
2063         QPEL_V_LOW(%%mm3, %%mm0, %%mm1, %%mm2, %5, %6, %5,   (%0),  8(%0), 16(%0), 56(%0), (%1, %3), OP)\
2064         "add %4, %1                     \n\t"\
2065         QPEL_V_LOW(%%mm0, %%mm1, %%mm2, %%mm3, %5, %6, %5,  8(%0), 16(%0), 24(%0), 64(%0), (%1), OP)\
2066         QPEL_V_LOW(%%mm1, %%mm2, %%mm3, %%mm0, %5, %6, %5, 16(%0), 24(%0), 32(%0), 72(%0), (%1, %3), OP)\
2067         "add %4, %1                     \n\t"\
2068         QPEL_V_LOW(%%mm2, %%mm3, %%mm0, %%mm1, %5, %6, %5, 24(%0), 32(%0), 40(%0), 80(%0), (%1), OP)\
2069         QPEL_V_LOW(%%mm3, %%mm0, %%mm1, %%mm2, %5, %6, %5, 32(%0), 40(%0), 48(%0), 88(%0), (%1, %3), OP)\
2070         "add %4, %1                     \n\t"\
2071         QPEL_V_LOW(%%mm0, %%mm1, %%mm2, %%mm3, %5, %6, %5, 40(%0), 48(%0), 56(%0), 96(%0), (%1), OP)\
2072         QPEL_V_LOW(%%mm1, %%mm2, %%mm3, %%mm0, %5, %6, %5, 48(%0), 56(%0), 64(%0),104(%0), (%1, %3), OP)\
2073         "add %4, %1                     \n\t"\
2074         QPEL_V_LOW(%%mm2, %%mm3, %%mm0, %%mm1, %5, %6, %5, 56(%0), 64(%0), 72(%0),112(%0), (%1), OP)\
2075         QPEL_V_LOW(%%mm3, %%mm0, %%mm1, %%mm2, %5, %6, %5, 64(%0), 72(%0), 80(%0),120(%0), (%1, %3), OP)\
2076         "add %4, %1                     \n\t"\
2077         QPEL_V_LOW(%%mm0, %%mm1, %%mm2, %%mm3, %5, %6, %5, 72(%0), 80(%0), 88(%0),128(%0), (%1), OP)\
2078         \
2079         QPEL_V_LOW(%%mm1, %%mm2, %%mm3, %%mm0, %5, %6, %5, 80(%0), 88(%0), 96(%0),128(%0), (%1, %3), OP)\
2080         "add %4, %1                     \n\t"  \
2081         QPEL_V_LOW(%%mm2, %%mm3, %%mm0, %%mm1, %5, %6, %5, 88(%0), 96(%0),104(%0),120(%0), (%1), OP)\
2082         QPEL_V_LOW(%%mm3, %%mm0, %%mm1, %%mm2, %5, %6, %5, 96(%0),104(%0),112(%0),112(%0), (%1, %3), OP)\
2083         \
2084         "add $136, %0                   \n\t"\
2085         "add %6, %1                     \n\t"\
2086         "decl %2                        \n\t"\
2087         " jnz 1b                        \n\t"\
2088         \
2089         : "+r"(temp_ptr), "+r"(dst), "+g"(count)\
2090         : "r"((long)dstStride), "r"(2*(long)dstStride), /*"m"(ff_pw_20), "m"(ff_pw_3),*/ "m"(ROUNDER), "g"(4-14*(long)dstStride)\
2091         :"memory"\
2092     );\
2093 }\
2094 \
2095 static void OPNAME ## mpeg4_qpel8_v_lowpass_ ## MMX(uint8_t *dst, uint8_t *src, int dstStride, int srcStride){\
2096     uint64_t temp[9*2];\
2097     uint64_t *temp_ptr= temp;\
2098     int count= 9;\
2099 \
2100     /*FIXME unroll */\
2101     asm volatile(\
2102         "pxor %%mm7, %%mm7              \n\t"\
2103         "1:                             \n\t"\
2104         "movq (%0), %%mm0               \n\t"\
2105         "movq (%0), %%mm1               \n\t"\
2106         "punpcklbw %%mm7, %%mm0         \n\t"\
2107         "punpckhbw %%mm7, %%mm1         \n\t"\
2108         "movq %%mm0, (%1)               \n\t"\
2109         "movq %%mm1, 9*8(%1)            \n\t"\
2110         "add $8, %1                     \n\t"\
2111         "add %3, %0                     \n\t"\
2112         "decl %2                        \n\t"\
2113         " jnz 1b                        \n\t"\
2114         : "+r" (src), "+r" (temp_ptr), "+r"(count)\
2115         : "r" ((long)srcStride)\
2116         : "memory"\
2117     );\
2118     \
2119     temp_ptr= temp;\
2120     count=2;\
2121     \
2122 /*FIXME reorder for speed */\
2123     asm volatile(\
2124         /*"pxor %%mm7, %%mm7              \n\t"*/\
2125         "1:                             \n\t"\
2126         "movq (%0), %%mm0               \n\t"\
2127         "movq 8(%0), %%mm1              \n\t"\
2128         "movq 16(%0), %%mm2             \n\t"\
2129         "movq 24(%0), %%mm3             \n\t"\
2130         QPEL_V_LOW(%%mm0, %%mm1, %%mm2, %%mm3, %5, %6, %5, 16(%0),  8(%0),   (%0), 32(%0), (%1), OP)\
2131         QPEL_V_LOW(%%mm1, %%mm2, %%mm3, %%mm0, %5, %6, %5,  8(%0),   (%0),   (%0), 40(%0), (%1, %3), OP)\
2132         "add %4, %1                     \n\t"\
2133         QPEL_V_LOW(%%mm2, %%mm3, %%mm0, %%mm1, %5, %6, %5,   (%0),   (%0),  8(%0), 48(%0), (%1), OP)\
2134         \
2135         QPEL_V_LOW(%%mm3, %%mm0, %%mm1, %%mm2, %5, %6, %5,   (%0),  8(%0), 16(%0), 56(%0), (%1, %3), OP)\
2136         "add %4, %1                     \n\t"\
2137         QPEL_V_LOW(%%mm0, %%mm1, %%mm2, %%mm3, %5, %6, %5,  8(%0), 16(%0), 24(%0), 64(%0), (%1), OP)\
2138         \
2139         QPEL_V_LOW(%%mm1, %%mm2, %%mm3, %%mm0, %5, %6, %5, 16(%0), 24(%0), 32(%0), 64(%0), (%1, %3), OP)\
2140         "add %4, %1                     \n\t"\
2141         QPEL_V_LOW(%%mm2, %%mm3, %%mm0, %%mm1, %5, %6, %5, 24(%0), 32(%0), 40(%0), 56(%0), (%1), OP)\
2142         QPEL_V_LOW(%%mm3, %%mm0, %%mm1, %%mm2, %5, %6, %5, 32(%0), 40(%0), 48(%0), 48(%0), (%1, %3), OP)\
2143                 \
2144         "add $72, %0                    \n\t"\
2145         "add %6, %1                     \n\t"\
2146         "decl %2                        \n\t"\
2147         " jnz 1b                        \n\t"\
2148          \
2149         : "+r"(temp_ptr), "+r"(dst), "+g"(count)\
2150         : "r"((long)dstStride), "r"(2*(long)dstStride), /*"m"(ff_pw_20), "m"(ff_pw_3),*/ "m"(ROUNDER), "g"(4-6*(long)dstStride)\
2151         : "memory"\
2152    );\
2153 }\
2154 \
2155 static void OPNAME ## qpel8_mc00_ ## MMX (uint8_t *dst, uint8_t *src, int stride){\
2156     OPNAME ## pixels8_mmx(dst, src, stride, 8);\
2157 }\
2158 \
2159 static void OPNAME ## qpel8_mc10_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
2160     uint64_t temp[8];\
2161     uint8_t * const half= (uint8_t*)temp;\
2162     put ## RND ## mpeg4_qpel8_h_lowpass_ ## MMX(half, src, 8, stride, 8);\
2163     OPNAME ## pixels8_l2_ ## MMX(dst, src, half, stride, stride, 8);\
2164 }\
2165 \
2166 static void OPNAME ## qpel8_mc20_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
2167     OPNAME ## mpeg4_qpel8_h_lowpass_ ## MMX(dst, src, stride, stride, 8);\
2168 }\
2169 \
2170 static void OPNAME ## qpel8_mc30_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
2171     uint64_t temp[8];\
2172     uint8_t * const half= (uint8_t*)temp;\
2173     put ## RND ## mpeg4_qpel8_h_lowpass_ ## MMX(half, src, 8, stride, 8);\
2174     OPNAME ## pixels8_l2_ ## MMX(dst, src+1, half, stride, stride, 8);\
2175 }\
2176 \
2177 static void OPNAME ## qpel8_mc01_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
2178     uint64_t temp[8];\
2179     uint8_t * const half= (uint8_t*)temp;\
2180     put ## RND ## mpeg4_qpel8_v_lowpass_ ## MMX(half, src, 8, stride);\
2181     OPNAME ## pixels8_l2_ ## MMX(dst, src, half, stride, stride, 8);\
2182 }\
2183 \
2184 static void OPNAME ## qpel8_mc02_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
2185     OPNAME ## mpeg4_qpel8_v_lowpass_ ## MMX(dst, src, stride, stride);\
2186 }\
2187 \
2188 static void OPNAME ## qpel8_mc03_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
2189     uint64_t temp[8];\
2190     uint8_t * const half= (uint8_t*)temp;\
2191     put ## RND ## mpeg4_qpel8_v_lowpass_ ## MMX(half, src, 8, stride);\
2192     OPNAME ## pixels8_l2_ ## MMX(dst, src+stride, half, stride, stride, 8);\
2193 }\
2194 static void OPNAME ## qpel8_mc11_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
2195     uint64_t half[8 + 9];\
2196     uint8_t * const halfH= ((uint8_t*)half) + 64;\
2197     uint8_t * const halfHV= ((uint8_t*)half);\
2198     put ## RND ## mpeg4_qpel8_h_lowpass_ ## MMX(halfH, src, 8, stride, 9);\
2199     put ## RND ## pixels8_l2_ ## MMX(halfH, src, halfH, 8, stride, 9);\
2200     put ## RND ## mpeg4_qpel8_v_lowpass_ ## MMX(halfHV, halfH, 8, 8);\
2201     OPNAME ## pixels8_l2_ ## MMX(dst, halfH, halfHV, stride, 8, 8);\
2202 }\
2203 static void OPNAME ## qpel8_mc31_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
2204     uint64_t half[8 + 9];\
2205     uint8_t * const halfH= ((uint8_t*)half) + 64;\
2206     uint8_t * const halfHV= ((uint8_t*)half);\
2207     put ## RND ## mpeg4_qpel8_h_lowpass_ ## MMX(halfH, src, 8, stride, 9);\
2208     put ## RND ## pixels8_l2_ ## MMX(halfH, src+1, halfH, 8, stride, 9);\
2209     put ## RND ## mpeg4_qpel8_v_lowpass_ ## MMX(halfHV, halfH, 8, 8);\
2210     OPNAME ## pixels8_l2_ ## MMX(dst, halfH, halfHV, stride, 8, 8);\
2211 }\
2212 static void OPNAME ## qpel8_mc13_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
2213     uint64_t half[8 + 9];\
2214     uint8_t * const halfH= ((uint8_t*)half) + 64;\
2215     uint8_t * const halfHV= ((uint8_t*)half);\
2216     put ## RND ## mpeg4_qpel8_h_lowpass_ ## MMX(halfH, src, 8, stride, 9);\
2217     put ## RND ## pixels8_l2_ ## MMX(halfH, src, halfH, 8, stride, 9);\
2218     put ## RND ## mpeg4_qpel8_v_lowpass_ ## MMX(halfHV, halfH, 8, 8);\
2219     OPNAME ## pixels8_l2_ ## MMX(dst, halfH+8, halfHV, stride, 8, 8);\
2220 }\
2221 static void OPNAME ## qpel8_mc33_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
2222     uint64_t half[8 + 9];\
2223     uint8_t * const halfH= ((uint8_t*)half) + 64;\
2224     uint8_t * const halfHV= ((uint8_t*)half);\
2225     put ## RND ## mpeg4_qpel8_h_lowpass_ ## MMX(halfH, src, 8, stride, 9);\
2226     put ## RND ## pixels8_l2_ ## MMX(halfH, src+1, halfH, 8, stride, 9);\
2227     put ## RND ## mpeg4_qpel8_v_lowpass_ ## MMX(halfHV, halfH, 8, 8);\
2228     OPNAME ## pixels8_l2_ ## MMX(dst, halfH+8, halfHV, stride, 8, 8);\
2229 }\
2230 static void OPNAME ## qpel8_mc21_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
2231     uint64_t half[8 + 9];\
2232     uint8_t * const halfH= ((uint8_t*)half) + 64;\
2233     uint8_t * const halfHV= ((uint8_t*)half);\
2234     put ## RND ## mpeg4_qpel8_h_lowpass_ ## MMX(halfH, src, 8, stride, 9);\
2235     put ## RND ## mpeg4_qpel8_v_lowpass_ ## MMX(halfHV, halfH, 8, 8);\
2236     OPNAME ## pixels8_l2_ ## MMX(dst, halfH, halfHV, stride, 8, 8);\
2237 }\
2238 static void OPNAME ## qpel8_mc23_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
2239     uint64_t half[8 + 9];\
2240     uint8_t * const halfH= ((uint8_t*)half) + 64;\
2241     uint8_t * const halfHV= ((uint8_t*)half);\
2242     put ## RND ## mpeg4_qpel8_h_lowpass_ ## MMX(halfH, src, 8, stride, 9);\
2243     put ## RND ## mpeg4_qpel8_v_lowpass_ ## MMX(halfHV, halfH, 8, 8);\
2244     OPNAME ## pixels8_l2_ ## MMX(dst, halfH+8, halfHV, stride, 8, 8);\
2245 }\
2246 static void OPNAME ## qpel8_mc12_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
2247     uint64_t half[8 + 9];\
2248     uint8_t * const halfH= ((uint8_t*)half);\
2249     put ## RND ## mpeg4_qpel8_h_lowpass_ ## MMX(halfH, src, 8, stride, 9);\
2250     put ## RND ## pixels8_l2_ ## MMX(halfH, src, halfH, 8, stride, 9);\
2251     OPNAME ## mpeg4_qpel8_v_lowpass_ ## MMX(dst, halfH, stride, 8);\
2252 }\
2253 static void OPNAME ## qpel8_mc32_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
2254     uint64_t half[8 + 9];\
2255     uint8_t * const halfH= ((uint8_t*)half);\
2256     put ## RND ## mpeg4_qpel8_h_lowpass_ ## MMX(halfH, src, 8, stride, 9);\
2257     put ## RND ## pixels8_l2_ ## MMX(halfH, src+1, halfH, 8, stride, 9);\
2258     OPNAME ## mpeg4_qpel8_v_lowpass_ ## MMX(dst, halfH, stride, 8);\
2259 }\
2260 static void OPNAME ## qpel8_mc22_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
2261     uint64_t half[9];\
2262     uint8_t * const halfH= ((uint8_t*)half);\
2263     put ## RND ## mpeg4_qpel8_h_lowpass_ ## MMX(halfH, src, 8, stride, 9);\
2264     OPNAME ## mpeg4_qpel8_v_lowpass_ ## MMX(dst, halfH, stride, 8);\
2265 }\
2266 static void OPNAME ## qpel16_mc00_ ## MMX (uint8_t *dst, uint8_t *src, int stride){\
2267     OPNAME ## pixels16_mmx(dst, src, stride, 16);\
2268 }\
2269 \
2270 static void OPNAME ## qpel16_mc10_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
2271     uint64_t temp[32];\
2272     uint8_t * const half= (uint8_t*)temp;\
2273     put ## RND ## mpeg4_qpel16_h_lowpass_ ## MMX(half, src, 16, stride, 16);\
2274     OPNAME ## pixels16_l2_ ## MMX(dst, src, half, stride, stride, 16);\
2275 }\
2276 \
2277 static void OPNAME ## qpel16_mc20_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
2278     OPNAME ## mpeg4_qpel16_h_lowpass_ ## MMX(dst, src, stride, stride, 16);\
2279 }\
2280 \
2281 static void OPNAME ## qpel16_mc30_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
2282     uint64_t temp[32];\
2283     uint8_t * const half= (uint8_t*)temp;\
2284     put ## RND ## mpeg4_qpel16_h_lowpass_ ## MMX(half, src, 16, stride, 16);\
2285     OPNAME ## pixels16_l2_ ## MMX(dst, src+1, half, stride, stride, 16);\
2286 }\
2287 \
2288 static void OPNAME ## qpel16_mc01_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
2289     uint64_t temp[32];\
2290     uint8_t * const half= (uint8_t*)temp;\
2291     put ## RND ## mpeg4_qpel16_v_lowpass_ ## MMX(half, src, 16, stride);\
2292     OPNAME ## pixels16_l2_ ## MMX(dst, src, half, stride, stride, 16);\
2293 }\
2294 \
2295 static void OPNAME ## qpel16_mc02_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
2296     OPNAME ## mpeg4_qpel16_v_lowpass_ ## MMX(dst, src, stride, stride);\
2297 }\
2298 \
2299 static void OPNAME ## qpel16_mc03_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
2300     uint64_t temp[32];\
2301     uint8_t * const half= (uint8_t*)temp;\
2302     put ## RND ## mpeg4_qpel16_v_lowpass_ ## MMX(half, src, 16, stride);\
2303     OPNAME ## pixels16_l2_ ## MMX(dst, src+stride, half, stride, stride, 16);\
2304 }\
2305 static void OPNAME ## qpel16_mc11_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
2306     uint64_t half[16*2 + 17*2];\
2307     uint8_t * const halfH= ((uint8_t*)half) + 256;\
2308     uint8_t * const halfHV= ((uint8_t*)half);\
2309     put ## RND ## mpeg4_qpel16_h_lowpass_ ## MMX(halfH, src, 16, stride, 17);\
2310     put ## RND ## pixels16_l2_ ## MMX(halfH, src, halfH, 16, stride, 17);\
2311     put ## RND ## mpeg4_qpel16_v_lowpass_ ## MMX(halfHV, halfH, 16, 16);\
2312     OPNAME ## pixels16_l2_ ## MMX(dst, halfH, halfHV, stride, 16, 16);\
2313 }\
2314 static void OPNAME ## qpel16_mc31_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
2315     uint64_t half[16*2 + 17*2];\
2316     uint8_t * const halfH= ((uint8_t*)half) + 256;\
2317     uint8_t * const halfHV= ((uint8_t*)half);\
2318     put ## RND ## mpeg4_qpel16_h_lowpass_ ## MMX(halfH, src, 16, stride, 17);\
2319     put ## RND ## pixels16_l2_ ## MMX(halfH, src+1, halfH, 16, stride, 17);\
2320     put ## RND ## mpeg4_qpel16_v_lowpass_ ## MMX(halfHV, halfH, 16, 16);\
2321     OPNAME ## pixels16_l2_ ## MMX(dst, halfH, halfHV, stride, 16, 16);\
2322 }\
2323 static void OPNAME ## qpel16_mc13_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
2324     uint64_t half[16*2 + 17*2];\
2325     uint8_t * const halfH= ((uint8_t*)half) + 256;\
2326     uint8_t * const halfHV= ((uint8_t*)half);\
2327     put ## RND ## mpeg4_qpel16_h_lowpass_ ## MMX(halfH, src, 16, stride, 17);\
2328     put ## RND ## pixels16_l2_ ## MMX(halfH, src, halfH, 16, stride, 17);\
2329     put ## RND ## mpeg4_qpel16_v_lowpass_ ## MMX(halfHV, halfH, 16, 16);\
2330     OPNAME ## pixels16_l2_ ## MMX(dst, halfH+16, halfHV, stride, 16, 16);\
2331 }\
2332 static void OPNAME ## qpel16_mc33_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
2333     uint64_t half[16*2 + 17*2];\
2334     uint8_t * const halfH= ((uint8_t*)half) + 256;\
2335     uint8_t * const halfHV= ((uint8_t*)half);\
2336     put ## RND ## mpeg4_qpel16_h_lowpass_ ## MMX(halfH, src, 16, stride, 17);\
2337     put ## RND ## pixels16_l2_ ## MMX(halfH, src+1, halfH, 16, stride, 17);\
2338     put ## RND ## mpeg4_qpel16_v_lowpass_ ## MMX(halfHV, halfH, 16, 16);\
2339     OPNAME ## pixels16_l2_ ## MMX(dst, halfH+16, halfHV, stride, 16, 16);\
2340 }\
2341 static void OPNAME ## qpel16_mc21_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
2342     uint64_t half[16*2 + 17*2];\
2343     uint8_t * const halfH= ((uint8_t*)half) + 256;\
2344     uint8_t * const halfHV= ((uint8_t*)half);\
2345     put ## RND ## mpeg4_qpel16_h_lowpass_ ## MMX(halfH, src, 16, stride, 17);\
2346     put ## RND ## mpeg4_qpel16_v_lowpass_ ## MMX(halfHV, halfH, 16, 16);\
2347     OPNAME ## pixels16_l2_ ## MMX(dst, halfH, halfHV, stride, 16, 16);\
2348 }\
2349 static void OPNAME ## qpel16_mc23_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
2350     uint64_t half[16*2 + 17*2];\
2351     uint8_t * const halfH= ((uint8_t*)half) + 256;\
2352     uint8_t * const halfHV= ((uint8_t*)half);\
2353     put ## RND ## mpeg4_qpel16_h_lowpass_ ## MMX(halfH, src, 16, stride, 17);\
2354     put ## RND ## mpeg4_qpel16_v_lowpass_ ## MMX(halfHV, halfH, 16, 16);\
2355     OPNAME ## pixels16_l2_ ## MMX(dst, halfH+16, halfHV, stride, 16, 16);\
2356 }\
2357 static void OPNAME ## qpel16_mc12_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
2358     uint64_t half[17*2];\
2359     uint8_t * const halfH= ((uint8_t*)half);\
2360     put ## RND ## mpeg4_qpel16_h_lowpass_ ## MMX(halfH, src, 16, stride, 17);\
2361     put ## RND ## pixels16_l2_ ## MMX(halfH, src, halfH, 16, stride, 17);\
2362     OPNAME ## mpeg4_qpel16_v_lowpass_ ## MMX(dst, halfH, stride, 16);\
2363 }\
2364 static void OPNAME ## qpel16_mc32_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
2365     uint64_t half[17*2];\
2366     uint8_t * const halfH= ((uint8_t*)half);\
2367     put ## RND ## mpeg4_qpel16_h_lowpass_ ## MMX(halfH, src, 16, stride, 17);\
2368     put ## RND ## pixels16_l2_ ## MMX(halfH, src+1, halfH, 16, stride, 17);\
2369     OPNAME ## mpeg4_qpel16_v_lowpass_ ## MMX(dst, halfH, stride, 16);\
2370 }\
2371 static void OPNAME ## qpel16_mc22_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
2372     uint64_t half[17*2];\
2373     uint8_t * const halfH= ((uint8_t*)half);\
2374     put ## RND ## mpeg4_qpel16_h_lowpass_ ## MMX(halfH, src, 16, stride, 17);\
2375     OPNAME ## mpeg4_qpel16_v_lowpass_ ## MMX(dst, halfH, stride, 16);\
2376 }
2377
2378 #define PUT_OP(a,b,temp, size) "mov" #size " " #a ", " #b "        \n\t"
2379 #define AVG_3DNOW_OP(a,b,temp, size) \
2380 "mov" #size " " #b ", " #temp "   \n\t"\
2381 "pavgusb " #temp ", " #a "        \n\t"\
2382 "mov" #size " " #a ", " #b "      \n\t"
2383 #define AVG_MMX2_OP(a,b,temp, size) \
2384 "mov" #size " " #b ", " #temp "   \n\t"\
2385 "pavgb " #temp ", " #a "          \n\t"\
2386 "mov" #size " " #a ", " #b "      \n\t"
2387
2388 QPEL_BASE(put_       , ff_pw_16, _       , PUT_OP, PUT_OP)
2389 QPEL_BASE(avg_       , ff_pw_16, _       , AVG_MMX2_OP, AVG_3DNOW_OP)
2390 QPEL_BASE(put_no_rnd_, ff_pw_15, _no_rnd_, PUT_OP, PUT_OP)
2391 QPEL_OP(put_       , ff_pw_16, _       , PUT_OP, 3dnow)
2392 QPEL_OP(avg_       , ff_pw_16, _       , AVG_3DNOW_OP, 3dnow)
2393 QPEL_OP(put_no_rnd_, ff_pw_15, _no_rnd_, PUT_OP, 3dnow)
2394 QPEL_OP(put_       , ff_pw_16, _       , PUT_OP, mmx2)
2395 QPEL_OP(avg_       , ff_pw_16, _       , AVG_MMX2_OP, mmx2)
2396 QPEL_OP(put_no_rnd_, ff_pw_15, _no_rnd_, PUT_OP, mmx2)
2397
2398 #if 0
2399 static void just_return() { return; }
2400 #endif
2401
2402 #define SET_QPEL_FUNC(postfix1, postfix2) \
2403     c->put_ ## postfix1 = put_ ## postfix2;\
2404     c->put_no_rnd_ ## postfix1 = put_no_rnd_ ## postfix2;\
2405     c->avg_ ## postfix1 = avg_ ## postfix2;
2406
2407 static void gmc_mmx(uint8_t *dst, uint8_t *src, int stride, int h, int ox, int oy,
2408                     int dxx, int dxy, int dyx, int dyy, int shift, int r, int width, int height){
2409     const int w = 8;
2410     const int ix = ox>>(16+shift);
2411     const int iy = oy>>(16+shift);
2412     const int oxs = ox>>4;
2413     const int oys = oy>>4;
2414     const int dxxs = dxx>>4;
2415     const int dxys = dxy>>4;
2416     const int dyxs = dyx>>4;
2417     const int dyys = dyy>>4;
2418     const uint16_t r4[4] = {r,r,r,r};
2419     const uint16_t dxy4[4] = {dxys,dxys,dxys,dxys};
2420     const uint16_t dyy4[4] = {dyys,dyys,dyys,dyys};
2421     const uint64_t shift2 = 2*shift;
2422     uint8_t edge_buf[(h+1)*stride];
2423     int x, y;
2424
2425     const int dxw = (dxx-(1<<(16+shift)))*(w-1);
2426     const int dyh = (dyy-(1<<(16+shift)))*(h-1);
2427     const int dxh = dxy*(h-1);
2428     const int dyw = dyx*(w-1);
2429     if( // non-constant fullpel offset (3% of blocks)
2430         (ox^(ox+dxw) | ox^(ox+dxh) | ox^(ox+dxw+dxh) |
2431          oy^(oy+dyw) | oy^(oy+dyh) | oy^(oy+dyw+dyh)) >> (16+shift)
2432         // uses more than 16 bits of subpel mv (only at huge resolution)
2433         || (dxx|dxy|dyx|dyy)&15 )
2434     {
2435         //FIXME could still use mmx for some of the rows
2436         ff_gmc_c(dst, src, stride, h, ox, oy, dxx, dxy, dyx, dyy, shift, r, width, height);
2437         return;
2438     }
2439
2440     src += ix + iy*stride;
2441     if( (unsigned)ix >= width-w ||
2442         (unsigned)iy >= height-h )
2443     {
2444         ff_emulated_edge_mc(edge_buf, src, stride, w+1, h+1, ix, iy, width, height);
2445         src = edge_buf;
2446     }
2447
2448     asm volatile(
2449         "movd         %0, %%mm6 \n\t"
2450         "pxor      %%mm7, %%mm7 \n\t"
2451         "punpcklwd %%mm6, %%mm6 \n\t"
2452         "punpcklwd %%mm6, %%mm6 \n\t"
2453         :: "r"(1<<shift)
2454     );
2455
2456     for(x=0; x<w; x+=4){
2457         uint16_t dx4[4] = { oxs - dxys + dxxs*(x+0),
2458                             oxs - dxys + dxxs*(x+1),
2459                             oxs - dxys + dxxs*(x+2),
2460                             oxs - dxys + dxxs*(x+3) };
2461         uint16_t dy4[4] = { oys - dyys + dyxs*(x+0),
2462                             oys - dyys + dyxs*(x+1),
2463                             oys - dyys + dyxs*(x+2),
2464                             oys - dyys + dyxs*(x+3) };
2465
2466         for(y=0; y<h; y++){
2467             asm volatile(
2468                 "movq   %0,  %%mm4 \n\t"
2469                 "movq   %1,  %%mm5 \n\t"
2470                 "paddw  %2,  %%mm4 \n\t"
2471                 "paddw  %3,  %%mm5 \n\t"
2472                 "movq   %%mm4, %0  \n\t"
2473                 "movq   %%mm5, %1  \n\t"
2474                 "psrlw  $12, %%mm4 \n\t"
2475                 "psrlw  $12, %%mm5 \n\t"
2476                 : "+m"(*dx4), "+m"(*dy4)
2477                 : "m"(*dxy4), "m"(*dyy4)
2478             );
2479
2480             asm volatile(
2481                 "movq   %%mm6, %%mm2 \n\t"
2482                 "movq   %%mm6, %%mm1 \n\t"
2483                 "psubw  %%mm4, %%mm2 \n\t"
2484                 "psubw  %%mm5, %%mm1 \n\t"
2485                 "movq   %%mm2, %%mm0 \n\t"
2486                 "movq   %%mm4, %%mm3 \n\t"
2487                 "pmullw %%mm1, %%mm0 \n\t" // (s-dx)*(s-dy)
2488                 "pmullw %%mm5, %%mm3 \n\t" // dx*dy
2489                 "pmullw %%mm5, %%mm2 \n\t" // (s-dx)*dy
2490                 "pmullw %%mm4, %%mm1 \n\t" // dx*(s-dy)
2491
2492                 "movd   %4,    %%mm5 \n\t"
2493                 "movd   %3,    %%mm4 \n\t"
2494                 "punpcklbw %%mm7, %%mm5 \n\t"
2495                 "punpcklbw %%mm7, %%mm4 \n\t"
2496                 "pmullw %%mm5, %%mm3 \n\t" // src[1,1] * dx*dy
2497                 "pmullw %%mm4, %%mm2 \n\t" // src[0,1] * (s-dx)*dy
2498
2499                 "movd   %2,    %%mm5 \n\t"
2500                 "movd   %1,    %%mm4 \n\t"
2501                 "punpcklbw %%mm7, %%mm5 \n\t"
2502                 "punpcklbw %%mm7, %%mm4 \n\t"
2503                 "pmullw %%mm5, %%mm1 \n\t" // src[1,0] * dx*(s-dy)
2504                 "pmullw %%mm4, %%mm0 \n\t" // src[0,0] * (s-dx)*(s-dy)
2505                 "paddw  %5,    %%mm1 \n\t"
2506                 "paddw  %%mm3, %%mm2 \n\t"
2507                 "paddw  %%mm1, %%mm0 \n\t"
2508                 "paddw  %%mm2, %%mm0 \n\t"
2509
2510                 "psrlw    %6,    %%mm0 \n\t"
2511                 "packuswb %%mm0, %%mm0 \n\t"
2512                 "movd     %%mm0, %0    \n\t"
2513
2514                 : "=m"(dst[x+y*stride])
2515                 : "m"(src[0]), "m"(src[1]),
2516                   "m"(src[stride]), "m"(src[stride+1]),
2517                   "m"(*r4), "m"(shift2)
2518             );
2519             src += stride;
2520         }
2521         src += 4-h*stride;
2522     }
2523 }
2524
2525 static int try_8x8basis_mmx(int16_t rem[64], int16_t weight[64], int16_t basis[64], int scale){
2526     long i=0;
2527
2528     assert(ABS(scale) < 256);
2529     scale<<= 16 + 1 - BASIS_SHIFT + RECON_SHIFT;
2530
2531     asm volatile(
2532         "pcmpeqw %%mm6, %%mm6           \n\t" // -1w
2533         "psrlw $15, %%mm6               \n\t" //  1w
2534         "pxor %%mm7, %%mm7              \n\t"
2535         "movd  %4, %%mm5                \n\t"
2536         "punpcklwd %%mm5, %%mm5         \n\t"
2537         "punpcklwd %%mm5, %%mm5         \n\t"
2538         "1:                             \n\t"
2539         "movq  (%1, %0), %%mm0          \n\t"
2540         "movq  8(%1, %0), %%mm1         \n\t"
2541         "pmulhw %%mm5, %%mm0            \n\t"
2542         "pmulhw %%mm5, %%mm1            \n\t"
2543         "paddw %%mm6, %%mm0             \n\t"
2544         "paddw %%mm6, %%mm1             \n\t"
2545         "psraw $1, %%mm0                \n\t"
2546         "psraw $1, %%mm1                \n\t"
2547         "paddw (%2, %0), %%mm0          \n\t"
2548         "paddw 8(%2, %0), %%mm1         \n\t"
2549         "psraw $6, %%mm0                \n\t"
2550         "psraw $6, %%mm1                \n\t"
2551         "pmullw (%3, %0), %%mm0         \n\t"
2552         "pmullw 8(%3, %0), %%mm1        \n\t"
2553         "pmaddwd %%mm0, %%mm0           \n\t"
2554         "pmaddwd %%mm1, %%mm1           \n\t"
2555         "paddd %%mm1, %%mm0             \n\t"
2556         "psrld $4, %%mm0                \n\t"
2557         "paddd %%mm0, %%mm7             \n\t"
2558         "add $16, %0                    \n\t"
2559         "cmp $128, %0                   \n\t" //FIXME optimize & bench
2560         " jb 1b                         \n\t"
2561         "movq %%mm7, %%mm6              \n\t"
2562         "psrlq $32, %%mm7               \n\t"
2563         "paddd %%mm6, %%mm7             \n\t"
2564         "psrld $2, %%mm7                \n\t"
2565         "movd %%mm7, %0                 \n\t"
2566
2567         : "+r" (i)
2568         : "r"(basis), "r"(rem), "r"(weight), "g"(scale)
2569     );
2570     return i;
2571 }
2572
2573 static void add_8x8basis_mmx(int16_t rem[64], int16_t basis[64], int scale){
2574     long i=0;
2575
2576     if(ABS(scale) < 256){
2577         scale<<= 16 + 1 - BASIS_SHIFT + RECON_SHIFT;
2578         asm volatile(
2579                 "pcmpeqw %%mm6, %%mm6   \n\t" // -1w
2580                 "psrlw $15, %%mm6       \n\t" //  1w
2581                 "movd  %3, %%mm5        \n\t"
2582                 "punpcklwd %%mm5, %%mm5 \n\t"
2583                 "punpcklwd %%mm5, %%mm5 \n\t"
2584                 "1:                     \n\t"
2585                 "movq  (%1, %0), %%mm0  \n\t"
2586                 "movq  8(%1, %0), %%mm1 \n\t"
2587                 "pmulhw %%mm5, %%mm0    \n\t"
2588                 "pmulhw %%mm5, %%mm1    \n\t"
2589                 "paddw %%mm6, %%mm0     \n\t"
2590                 "paddw %%mm6, %%mm1     \n\t"
2591                 "psraw $1, %%mm0        \n\t"
2592                 "psraw $1, %%mm1        \n\t"
2593                 "paddw (%2, %0), %%mm0  \n\t"
2594                 "paddw 8(%2, %0), %%mm1 \n\t"
2595                 "movq %%mm0, (%2, %0)   \n\t"
2596                 "movq %%mm1, 8(%2, %0)  \n\t"
2597                 "add $16, %0            \n\t"
2598                 "cmp $128, %0           \n\t" //FIXME optimize & bench
2599                 " jb 1b                 \n\t"
2600
2601                 : "+r" (i)
2602                 : "r"(basis), "r"(rem), "g"(scale)
2603         );
2604     }else{
2605         for(i=0; i<8*8; i++){
2606             rem[i] += (basis[i]*scale + (1<<(BASIS_SHIFT - RECON_SHIFT-1)))>>(BASIS_SHIFT - RECON_SHIFT);
2607         }
2608     }
2609 }
2610
2611 #define PREFETCH(name, op) \
2612 void name(void *mem, int stride, int h){\
2613     const uint8_t *p= mem;\
2614     do{\
2615         asm volatile(#op" %0" :: "m"(*p));\
2616         p+= stride;\
2617     }while(--h);\
2618 }
2619 PREFETCH(prefetch_mmx2,  prefetcht0)
2620 PREFETCH(prefetch_3dnow, prefetch)
2621 #undef PREFETCH
2622
2623 #include "h264dsp_mmx.c"
2624
2625 /* external functions, from idct_mmx.c */
2626 void ff_mmx_idct(DCTELEM *block);
2627 void ff_mmxext_idct(DCTELEM *block);
2628
2629 void ff_vp3_idct_sse2(int16_t *input_data);
2630 void ff_vp3_idct_mmx(int16_t *data);
2631 void ff_vp3_dsp_init_mmx(void);
2632
2633 /* XXX: those functions should be suppressed ASAP when all IDCTs are
2634    converted */
2635 static void ff_libmpeg2mmx_idct_put(uint8_t *dest, int line_size, DCTELEM *block)
2636 {
2637     ff_mmx_idct (block);
2638     put_pixels_clamped_mmx(block, dest, line_size);
2639 }
2640 static void ff_libmpeg2mmx_idct_add(uint8_t *dest, int line_size, DCTELEM *block)
2641 {
2642     ff_mmx_idct (block);
2643     add_pixels_clamped_mmx(block, dest, line_size);
2644 }
2645 static void ff_libmpeg2mmx2_idct_put(uint8_t *dest, int line_size, DCTELEM *block)
2646 {
2647     ff_mmxext_idct (block);
2648     put_pixels_clamped_mmx(block, dest, line_size);
2649 }
2650 static void ff_libmpeg2mmx2_idct_add(uint8_t *dest, int line_size, DCTELEM *block)
2651 {
2652     ff_mmxext_idct (block);
2653     add_pixels_clamped_mmx(block, dest, line_size);
2654 }
2655 static void ff_vp3_idct_put_sse2(uint8_t *dest, int line_size, DCTELEM *block)
2656 {
2657     ff_vp3_idct_sse2(block);
2658     put_signed_pixels_clamped_mmx(block, dest, line_size);
2659 }
2660 static void ff_vp3_idct_add_sse2(uint8_t *dest, int line_size, DCTELEM *block)
2661 {
2662     ff_vp3_idct_sse2(block);
2663     add_pixels_clamped_mmx(block, dest, line_size);
2664 }
2665 static void ff_vp3_idct_put_mmx(uint8_t *dest, int line_size, DCTELEM *block)
2666 {
2667     ff_vp3_idct_mmx(block);
2668     put_signed_pixels_clamped_mmx(block, dest, line_size);
2669 }
2670 static void ff_vp3_idct_add_mmx(uint8_t *dest, int line_size, DCTELEM *block)
2671 {
2672     ff_vp3_idct_mmx(block);
2673     add_pixels_clamped_mmx(block, dest, line_size);
2674 }
2675 #ifdef CONFIG_GPL
2676 static void ff_idct_xvid_mmx_put(uint8_t *dest, int line_size, DCTELEM *block)
2677 {
2678     ff_idct_xvid_mmx (block);
2679     put_pixels_clamped_mmx(block, dest, line_size);
2680 }
2681 static void ff_idct_xvid_mmx_add(uint8_t *dest, int line_size, DCTELEM *block)
2682 {
2683     ff_idct_xvid_mmx (block);
2684     add_pixels_clamped_mmx(block, dest, line_size);
2685 }
2686 static void ff_idct_xvid_mmx2_put(uint8_t *dest, int line_size, DCTELEM *block)
2687 {
2688     ff_idct_xvid_mmx2 (block);
2689     put_pixels_clamped_mmx(block, dest, line_size);
2690 }
2691 static void ff_idct_xvid_mmx2_add(uint8_t *dest, int line_size, DCTELEM *block)
2692 {
2693     ff_idct_xvid_mmx2 (block);
2694     add_pixels_clamped_mmx(block, dest, line_size);
2695 }
2696 #endif
2697
2698 #ifdef CONFIG_SNOW_ENCODER
2699 extern void ff_snow_horizontal_compose97i_sse2(DWTELEM *b, int width);
2700 extern void ff_snow_horizontal_compose97i_mmx(DWTELEM *b, int width);
2701 extern void ff_snow_vertical_compose97i_sse2(DWTELEM *b0, DWTELEM *b1, DWTELEM *b2, DWTELEM *b3, DWTELEM *b4, DWTELEM *b5, int width);
2702 extern void ff_snow_vertical_compose97i_mmx(DWTELEM *b0, DWTELEM *b1, DWTELEM *b2, DWTELEM *b3, DWTELEM *b4, DWTELEM *b5, int width);
2703 extern void ff_snow_inner_add_yblock_sse2(uint8_t *obmc, const int obmc_stride, uint8_t * * block, int b_w, int b_h,
2704                            int src_x, int src_y, int src_stride, slice_buffer * sb, int add, uint8_t * dst8);
2705 extern void ff_snow_inner_add_yblock_mmx(uint8_t *obmc, const int obmc_stride, uint8_t * * block, int b_w, int b_h,
2706                           int src_x, int src_y, int src_stride, slice_buffer * sb, int add, uint8_t * dst8);
2707 #endif
2708
2709 void dsputil_init_mmx(DSPContext* c, AVCodecContext *avctx)
2710 {
2711     mm_flags = mm_support();
2712
2713     if (avctx->dsp_mask) {
2714         if (avctx->dsp_mask & FF_MM_FORCE)
2715             mm_flags |= (avctx->dsp_mask & 0xffff);
2716         else
2717             mm_flags &= ~(avctx->dsp_mask & 0xffff);
2718     }
2719
2720 #if 0
2721     av_log(avctx, AV_LOG_INFO, "libavcodec: CPU flags:");
2722     if (mm_flags & MM_MMX)
2723         av_log(avctx, AV_LOG_INFO, " mmx");
2724     if (mm_flags & MM_MMXEXT)
2725         av_log(avctx, AV_LOG_INFO, " mmxext");
2726     if (mm_flags & MM_3DNOW)
2727         av_log(avctx, AV_LOG_INFO, " 3dnow");
2728     if (mm_flags & MM_SSE)
2729         av_log(avctx, AV_LOG_INFO, " sse");
2730     if (mm_flags & MM_SSE2)
2731         av_log(avctx, AV_LOG_INFO, " sse2");
2732     av_log(avctx, AV_LOG_INFO, "\n");
2733 #endif
2734
2735     if (mm_flags & MM_MMX) {
2736         const int idct_algo= avctx->idct_algo;
2737
2738 #ifdef CONFIG_ENCODERS
2739         const int dct_algo = avctx->dct_algo;
2740         if(dct_algo==FF_DCT_AUTO || dct_algo==FF_DCT_MMX){
2741             if(mm_flags & MM_SSE2){
2742                 c->fdct = ff_fdct_sse2;
2743             }else if(mm_flags & MM_MMXEXT){
2744                 c->fdct = ff_fdct_mmx2;
2745             }else{
2746                 c->fdct = ff_fdct_mmx;
2747             }
2748         }
2749 #endif //CONFIG_ENCODERS
2750         if(avctx->lowres==0){
2751             if(idct_algo==FF_IDCT_AUTO || idct_algo==FF_IDCT_SIMPLEMMX){
2752                 c->idct_put= ff_simple_idct_put_mmx;
2753                 c->idct_add= ff_simple_idct_add_mmx;
2754                 c->idct    = ff_simple_idct_mmx;
2755                 c->idct_permutation_type= FF_SIMPLE_IDCT_PERM;
2756             }else if(idct_algo==FF_IDCT_LIBMPEG2MMX){
2757                 if(mm_flags & MM_MMXEXT){
2758                     c->idct_put= ff_libmpeg2mmx2_idct_put;
2759                     c->idct_add= ff_libmpeg2mmx2_idct_add;
2760                     c->idct    = ff_mmxext_idct;
2761                 }else{
2762                     c->idct_put= ff_libmpeg2mmx_idct_put;
2763                     c->idct_add= ff_libmpeg2mmx_idct_add;
2764                     c->idct    = ff_mmx_idct;
2765                 }
2766                 c->idct_permutation_type= FF_LIBMPEG2_IDCT_PERM;
2767             }else if(idct_algo==FF_IDCT_VP3){
2768                 if(mm_flags & MM_SSE2){
2769                     c->idct_put= ff_vp3_idct_put_sse2;
2770                     c->idct_add= ff_vp3_idct_add_sse2;
2771                     c->idct    = ff_vp3_idct_sse2;
2772                     c->idct_permutation_type= FF_TRANSPOSE_IDCT_PERM;
2773                 }else{
2774                     ff_vp3_dsp_init_mmx();
2775                     c->idct_put= ff_vp3_idct_put_mmx;
2776                     c->idct_add= ff_vp3_idct_add_mmx;
2777                     c->idct    = ff_vp3_idct_mmx;
2778                     c->idct_permutation_type= FF_PARTTRANS_IDCT_PERM;
2779                 }
2780 #ifdef CONFIG_GPL
2781             }else if(idct_algo==FF_IDCT_XVIDMMX){
2782                 if(mm_flags & MM_MMXEXT){
2783                     c->idct_put= ff_idct_xvid_mmx2_put;
2784                     c->idct_add= ff_idct_xvid_mmx2_add;
2785                     c->idct    = ff_idct_xvid_mmx2;
2786                 }else{
2787                     c->idct_put= ff_idct_xvid_mmx_put;
2788                     c->idct_add= ff_idct_xvid_mmx_add;
2789                     c->idct    = ff_idct_xvid_mmx;
2790                 }
2791 #endif
2792             }
2793         }
2794
2795 #ifdef CONFIG_ENCODERS
2796         c->get_pixels = get_pixels_mmx;
2797         c->diff_pixels = diff_pixels_mmx;
2798 #endif //CONFIG_ENCODERS
2799         c->put_pixels_clamped = put_pixels_clamped_mmx;
2800         c->put_signed_pixels_clamped = put_signed_pixels_clamped_mmx;
2801         c->add_pixels_clamped = add_pixels_clamped_mmx;
2802         c->clear_blocks = clear_blocks_mmx;
2803 #ifdef CONFIG_ENCODERS
2804         c->pix_sum = pix_sum16_mmx;
2805 #endif //CONFIG_ENCODERS
2806
2807         c->put_pixels_tab[0][0] = put_pixels16_mmx;
2808         c->put_pixels_tab[0][1] = put_pixels16_x2_mmx;
2809         c->put_pixels_tab[0][2] = put_pixels16_y2_mmx;
2810         c->put_pixels_tab[0][3] = put_pixels16_xy2_mmx;
2811
2812         c->put_no_rnd_pixels_tab[0][0] = put_pixels16_mmx;
2813         c->put_no_rnd_pixels_tab[0][1] = put_no_rnd_pixels16_x2_mmx;
2814         c->put_no_rnd_pixels_tab[0][2] = put_no_rnd_pixels16_y2_mmx;
2815         c->put_no_rnd_pixels_tab[0][3] = put_no_rnd_pixels16_xy2_mmx;
2816
2817         c->avg_pixels_tab[0][0] = avg_pixels16_mmx;
2818         c->avg_pixels_tab[0][1] = avg_pixels16_x2_mmx;
2819         c->avg_pixels_tab[0][2] = avg_pixels16_y2_mmx;
2820         c->avg_pixels_tab[0][3] = avg_pixels16_xy2_mmx;
2821
2822         c->avg_no_rnd_pixels_tab[0][0] = avg_no_rnd_pixels16_mmx;
2823         c->avg_no_rnd_pixels_tab[0][1] = avg_no_rnd_pixels16_x2_mmx;
2824         c->avg_no_rnd_pixels_tab[0][2] = avg_no_rnd_pixels16_y2_mmx;
2825         c->avg_no_rnd_pixels_tab[0][3] = avg_no_rnd_pixels16_xy2_mmx;
2826
2827         c->put_pixels_tab[1][0] = put_pixels8_mmx;
2828         c->put_pixels_tab[1][1] = put_pixels8_x2_mmx;
2829         c->put_pixels_tab[1][2] = put_pixels8_y2_mmx;
2830         c->put_pixels_tab[1][3] = put_pixels8_xy2_mmx;
2831
2832         c->put_no_rnd_pixels_tab[1][0] = put_pixels8_mmx;
2833         c->put_no_rnd_pixels_tab[1][1] = put_no_rnd_pixels8_x2_mmx;
2834         c->put_no_rnd_pixels_tab[1][2] = put_no_rnd_pixels8_y2_mmx;
2835         c->put_no_rnd_pixels_tab[1][3] = put_no_rnd_pixels8_xy2_mmx;
2836
2837         c->avg_pixels_tab[1][0] = avg_pixels8_mmx;
2838         c->avg_pixels_tab[1][1] = avg_pixels8_x2_mmx;
2839         c->avg_pixels_tab[1][2] = avg_pixels8_y2_mmx;
2840         c->avg_pixels_tab[1][3] = avg_pixels8_xy2_mmx;
2841
2842         c->avg_no_rnd_pixels_tab[1][0] = avg_no_rnd_pixels8_mmx;
2843         c->avg_no_rnd_pixels_tab[1][1] = avg_no_rnd_pixels8_x2_mmx;
2844         c->avg_no_rnd_pixels_tab[1][2] = avg_no_rnd_pixels8_y2_mmx;
2845         c->avg_no_rnd_pixels_tab[1][3] = avg_no_rnd_pixels8_xy2_mmx;
2846
2847         c->gmc= gmc_mmx;
2848
2849         c->add_bytes= add_bytes_mmx;
2850 #ifdef CONFIG_ENCODERS
2851         c->diff_bytes= diff_bytes_mmx;
2852
2853         c->hadamard8_diff[0]= hadamard8_diff16_mmx;
2854         c->hadamard8_diff[1]= hadamard8_diff_mmx;
2855
2856         c->pix_norm1 = pix_norm1_mmx;
2857         c->sse[0] = (mm_flags & MM_SSE2) ? sse16_sse2 : sse16_mmx;
2858           c->sse[1] = sse8_mmx;
2859         c->vsad[4]= vsad_intra16_mmx;
2860
2861         c->nsse[0] = nsse16_mmx;
2862         c->nsse[1] = nsse8_mmx;
2863         if(!(avctx->flags & CODEC_FLAG_BITEXACT)){
2864             c->vsad[0] = vsad16_mmx;
2865         }
2866
2867         if(!(avctx->flags & CODEC_FLAG_BITEXACT)){
2868             c->try_8x8basis= try_8x8basis_mmx;
2869         }
2870         c->add_8x8basis= add_8x8basis_mmx;
2871
2872 #endif //CONFIG_ENCODERS
2873
2874         c->h263_v_loop_filter= h263_v_loop_filter_mmx;
2875         c->h263_h_loop_filter= h263_h_loop_filter_mmx;
2876         c->put_h264_chroma_pixels_tab[0]= put_h264_chroma_mc8_mmx;
2877         c->put_h264_chroma_pixels_tab[1]= put_h264_chroma_mc4_mmx;
2878
2879         c->h264_idct_dc_add=
2880         c->h264_idct_add= ff_h264_idct_add_mmx;
2881         c->h264_idct8_dc_add=
2882         c->h264_idct8_add= ff_h264_idct8_add_mmx;
2883
2884         if (mm_flags & MM_MMXEXT) {
2885             c->prefetch = prefetch_mmx2;
2886
2887             c->put_pixels_tab[0][1] = put_pixels16_x2_mmx2;
2888             c->put_pixels_tab[0][2] = put_pixels16_y2_mmx2;
2889
2890             c->avg_pixels_tab[0][0] = avg_pixels16_mmx2;
2891             c->avg_pixels_tab[0][1] = avg_pixels16_x2_mmx2;
2892             c->avg_pixels_tab[0][2] = avg_pixels16_y2_mmx2;
2893
2894             c->put_pixels_tab[1][1] = put_pixels8_x2_mmx2;
2895             c->put_pixels_tab[1][2] = put_pixels8_y2_mmx2;
2896
2897             c->avg_pixels_tab[1][0] = avg_pixels8_mmx2;
2898             c->avg_pixels_tab[1][1] = avg_pixels8_x2_mmx2;
2899             c->avg_pixels_tab[1][2] = avg_pixels8_y2_mmx2;
2900
2901 #ifdef CONFIG_ENCODERS
2902             c->hadamard8_diff[0]= hadamard8_diff16_mmx2;
2903             c->hadamard8_diff[1]= hadamard8_diff_mmx2;
2904             c->vsad[4]= vsad_intra16_mmx2;
2905 #endif //CONFIG_ENCODERS
2906
2907             c->h264_idct_dc_add= ff_h264_idct_dc_add_mmx2;
2908             c->h264_idct8_dc_add= ff_h264_idct8_dc_add_mmx2;
2909
2910             if(!(avctx->flags & CODEC_FLAG_BITEXACT)){
2911                 c->put_no_rnd_pixels_tab[0][1] = put_no_rnd_pixels16_x2_mmx2;
2912                 c->put_no_rnd_pixels_tab[0][2] = put_no_rnd_pixels16_y2_mmx2;
2913                 c->put_no_rnd_pixels_tab[1][1] = put_no_rnd_pixels8_x2_mmx2;
2914                 c->put_no_rnd_pixels_tab[1][2] = put_no_rnd_pixels8_y2_mmx2;
2915                 c->avg_pixels_tab[0][3] = avg_pixels16_xy2_mmx2;
2916                 c->avg_pixels_tab[1][3] = avg_pixels8_xy2_mmx2;
2917 #ifdef CONFIG_ENCODERS
2918                 c->vsad[0] = vsad16_mmx2;
2919 #endif //CONFIG_ENCODERS
2920             }
2921
2922 #if 1
2923             SET_QPEL_FUNC(qpel_pixels_tab[0][ 0], qpel16_mc00_mmx2)
2924             SET_QPEL_FUNC(qpel_pixels_tab[0][ 1], qpel16_mc10_mmx2)
2925             SET_QPEL_FUNC(qpel_pixels_tab[0][ 2], qpel16_mc20_mmx2)
2926             SET_QPEL_FUNC(qpel_pixels_tab[0][ 3], qpel16_mc30_mmx2)
2927             SET_QPEL_FUNC(qpel_pixels_tab[0][ 4], qpel16_mc01_mmx2)
2928             SET_QPEL_FUNC(qpel_pixels_tab[0][ 5], qpel16_mc11_mmx2)
2929             SET_QPEL_FUNC(qpel_pixels_tab[0][ 6], qpel16_mc21_mmx2)
2930             SET_QPEL_FUNC(qpel_pixels_tab[0][ 7], qpel16_mc31_mmx2)
2931             SET_QPEL_FUNC(qpel_pixels_tab[0][ 8], qpel16_mc02_mmx2)
2932             SET_QPEL_FUNC(qpel_pixels_tab[0][ 9], qpel16_mc12_mmx2)
2933             SET_QPEL_FUNC(qpel_pixels_tab[0][10], qpel16_mc22_mmx2)
2934             SET_QPEL_FUNC(qpel_pixels_tab[0][11], qpel16_mc32_mmx2)
2935             SET_QPEL_FUNC(qpel_pixels_tab[0][12], qpel16_mc03_mmx2)
2936             SET_QPEL_FUNC(qpel_pixels_tab[0][13], qpel16_mc13_mmx2)
2937             SET_QPEL_FUNC(qpel_pixels_tab[0][14], qpel16_mc23_mmx2)
2938             SET_QPEL_FUNC(qpel_pixels_tab[0][15], qpel16_mc33_mmx2)
2939             SET_QPEL_FUNC(qpel_pixels_tab[1][ 0], qpel8_mc00_mmx2)
2940             SET_QPEL_FUNC(qpel_pixels_tab[1][ 1], qpel8_mc10_mmx2)
2941             SET_QPEL_FUNC(qpel_pixels_tab[1][ 2], qpel8_mc20_mmx2)
2942             SET_QPEL_FUNC(qpel_pixels_tab[1][ 3], qpel8_mc30_mmx2)
2943             SET_QPEL_FUNC(qpel_pixels_tab[1][ 4], qpel8_mc01_mmx2)
2944             SET_QPEL_FUNC(qpel_pixels_tab[1][ 5], qpel8_mc11_mmx2)
2945             SET_QPEL_FUNC(qpel_pixels_tab[1][ 6], qpel8_mc21_mmx2)
2946             SET_QPEL_FUNC(qpel_pixels_tab[1][ 7], qpel8_mc31_mmx2)
2947             SET_QPEL_FUNC(qpel_pixels_tab[1][ 8], qpel8_mc02_mmx2)
2948             SET_QPEL_FUNC(qpel_pixels_tab[1][ 9], qpel8_mc12_mmx2)
2949             SET_QPEL_FUNC(qpel_pixels_tab[1][10], qpel8_mc22_mmx2)
2950             SET_QPEL_FUNC(qpel_pixels_tab[1][11], qpel8_mc32_mmx2)
2951             SET_QPEL_FUNC(qpel_pixels_tab[1][12], qpel8_mc03_mmx2)
2952             SET_QPEL_FUNC(qpel_pixels_tab[1][13], qpel8_mc13_mmx2)
2953             SET_QPEL_FUNC(qpel_pixels_tab[1][14], qpel8_mc23_mmx2)
2954             SET_QPEL_FUNC(qpel_pixels_tab[1][15], qpel8_mc33_mmx2)
2955 #endif
2956
2957 //FIXME 3dnow too
2958 #define dspfunc(PFX, IDX, NUM) \
2959     c->PFX ## _pixels_tab[IDX][ 0] = PFX ## NUM ## _mc00_mmx2; \
2960     c->PFX ## _pixels_tab[IDX][ 1] = PFX ## NUM ## _mc10_mmx2; \
2961     c->PFX ## _pixels_tab[IDX][ 2] = PFX ## NUM ## _mc20_mmx2; \
2962     c->PFX ## _pixels_tab[IDX][ 3] = PFX ## NUM ## _mc30_mmx2; \
2963     c->PFX ## _pixels_tab[IDX][ 4] = PFX ## NUM ## _mc01_mmx2; \
2964     c->PFX ## _pixels_tab[IDX][ 5] = PFX ## NUM ## _mc11_mmx2; \
2965     c->PFX ## _pixels_tab[IDX][ 6] = PFX ## NUM ## _mc21_mmx2; \
2966     c->PFX ## _pixels_tab[IDX][ 7] = PFX ## NUM ## _mc31_mmx2; \
2967     c->PFX ## _pixels_tab[IDX][ 8] = PFX ## NUM ## _mc02_mmx2; \
2968     c->PFX ## _pixels_tab[IDX][ 9] = PFX ## NUM ## _mc12_mmx2; \
2969     c->PFX ## _pixels_tab[IDX][10] = PFX ## NUM ## _mc22_mmx2; \
2970     c->PFX ## _pixels_tab[IDX][11] = PFX ## NUM ## _mc32_mmx2; \
2971     c->PFX ## _pixels_tab[IDX][12] = PFX ## NUM ## _mc03_mmx2; \
2972     c->PFX ## _pixels_tab[IDX][13] = PFX ## NUM ## _mc13_mmx2; \
2973     c->PFX ## _pixels_tab[IDX][14] = PFX ## NUM ## _mc23_mmx2; \
2974     c->PFX ## _pixels_tab[IDX][15] = PFX ## NUM ## _mc33_mmx2
2975
2976             dspfunc(put_h264_qpel, 0, 16);
2977             dspfunc(put_h264_qpel, 1, 8);
2978             dspfunc(put_h264_qpel, 2, 4);
2979             dspfunc(avg_h264_qpel, 0, 16);
2980             dspfunc(avg_h264_qpel, 1, 8);
2981             dspfunc(avg_h264_qpel, 2, 4);
2982 #undef dspfunc
2983
2984             c->avg_h264_chroma_pixels_tab[0]= avg_h264_chroma_mc8_mmx2;
2985             c->avg_h264_chroma_pixels_tab[1]= avg_h264_chroma_mc4_mmx2;
2986             c->avg_h264_chroma_pixels_tab[2]= avg_h264_chroma_mc2_mmx2;
2987             c->put_h264_chroma_pixels_tab[2]= put_h264_chroma_mc2_mmx2;
2988             c->h264_v_loop_filter_luma= h264_v_loop_filter_luma_mmx2;
2989             c->h264_h_loop_filter_luma= h264_h_loop_filter_luma_mmx2;
2990             c->h264_v_loop_filter_chroma= h264_v_loop_filter_chroma_mmx2;
2991             c->h264_h_loop_filter_chroma= h264_h_loop_filter_chroma_mmx2;
2992             c->h264_v_loop_filter_chroma_intra= h264_v_loop_filter_chroma_intra_mmx2;
2993             c->h264_h_loop_filter_chroma_intra= h264_h_loop_filter_chroma_intra_mmx2;
2994
2995             c->weight_h264_pixels_tab[0]= ff_h264_weight_16x16_mmx2;
2996             c->weight_h264_pixels_tab[1]= ff_h264_weight_16x8_mmx2;
2997             c->weight_h264_pixels_tab[2]= ff_h264_weight_8x16_mmx2;
2998             c->weight_h264_pixels_tab[3]= ff_h264_weight_8x8_mmx2;
2999             c->weight_h264_pixels_tab[4]= ff_h264_weight_8x4_mmx2;
3000             c->weight_h264_pixels_tab[5]= ff_h264_weight_4x8_mmx2;
3001             c->weight_h264_pixels_tab[6]= ff_h264_weight_4x4_mmx2;
3002             c->weight_h264_pixels_tab[7]= ff_h264_weight_4x2_mmx2;
3003
3004             c->biweight_h264_pixels_tab[0]= ff_h264_biweight_16x16_mmx2;
3005             c->biweight_h264_pixels_tab[1]= ff_h264_biweight_16x8_mmx2;
3006             c->biweight_h264_pixels_tab[2]= ff_h264_biweight_8x16_mmx2;
3007             c->biweight_h264_pixels_tab[3]= ff_h264_biweight_8x8_mmx2;
3008             c->biweight_h264_pixels_tab[4]= ff_h264_biweight_8x4_mmx2;
3009             c->biweight_h264_pixels_tab[5]= ff_h264_biweight_4x8_mmx2;
3010             c->biweight_h264_pixels_tab[6]= ff_h264_biweight_4x4_mmx2;
3011             c->biweight_h264_pixels_tab[7]= ff_h264_biweight_4x2_mmx2;
3012
3013 #ifdef CONFIG_ENCODERS
3014             c->sub_hfyu_median_prediction= sub_hfyu_median_prediction_mmx2;
3015 #endif //CONFIG_ENCODERS
3016         } else if (mm_flags & MM_3DNOW) {
3017             c->prefetch = prefetch_3dnow;
3018
3019             c->put_pixels_tab[0][1] = put_pixels16_x2_3dnow;
3020             c->put_pixels_tab[0][2] = put_pixels16_y2_3dnow;
3021
3022             c->avg_pixels_tab[0][0] = avg_pixels16_3dnow;
3023             c->avg_pixels_tab[0][1] = avg_pixels16_x2_3dnow;
3024             c->avg_pixels_tab[0][2] = avg_pixels16_y2_3dnow;
3025
3026             c->put_pixels_tab[1][1] = put_pixels8_x2_3dnow;
3027             c->put_pixels_tab[1][2] = put_pixels8_y2_3dnow;
3028
3029             c->avg_pixels_tab[1][0] = avg_pixels8_3dnow;
3030             c->avg_pixels_tab[1][1] = avg_pixels8_x2_3dnow;
3031             c->avg_pixels_tab[1][2] = avg_pixels8_y2_3dnow;
3032
3033             if(!(avctx->flags & CODEC_FLAG_BITEXACT)){
3034                 c->put_no_rnd_pixels_tab[0][1] = put_no_rnd_pixels16_x2_3dnow;
3035                 c->put_no_rnd_pixels_tab[0][2] = put_no_rnd_pixels16_y2_3dnow;
3036                 c->put_no_rnd_pixels_tab[1][1] = put_no_rnd_pixels8_x2_3dnow;
3037                 c->put_no_rnd_pixels_tab[1][2] = put_no_rnd_pixels8_y2_3dnow;
3038                 c->avg_pixels_tab[0][3] = avg_pixels16_xy2_3dnow;
3039                 c->avg_pixels_tab[1][3] = avg_pixels8_xy2_3dnow;
3040             }
3041
3042             SET_QPEL_FUNC(qpel_pixels_tab[0][ 0], qpel16_mc00_3dnow)
3043             SET_QPEL_FUNC(qpel_pixels_tab[0][ 1], qpel16_mc10_3dnow)
3044             SET_QPEL_FUNC(qpel_pixels_tab[0][ 2], qpel16_mc20_3dnow)
3045             SET_QPEL_FUNC(qpel_pixels_tab[0][ 3], qpel16_mc30_3dnow)
3046             SET_QPEL_FUNC(qpel_pixels_tab[0][ 4], qpel16_mc01_3dnow)
3047             SET_QPEL_FUNC(qpel_pixels_tab[0][ 5], qpel16_mc11_3dnow)
3048             SET_QPEL_FUNC(qpel_pixels_tab[0][ 6], qpel16_mc21_3dnow)
3049             SET_QPEL_FUNC(qpel_pixels_tab[0][ 7], qpel16_mc31_3dnow)
3050             SET_QPEL_FUNC(qpel_pixels_tab[0][ 8], qpel16_mc02_3dnow)
3051             SET_QPEL_FUNC(qpel_pixels_tab[0][ 9], qpel16_mc12_3dnow)
3052             SET_QPEL_FUNC(qpel_pixels_tab[0][10], qpel16_mc22_3dnow)
3053             SET_QPEL_FUNC(qpel_pixels_tab[0][11], qpel16_mc32_3dnow)
3054             SET_QPEL_FUNC(qpel_pixels_tab[0][12], qpel16_mc03_3dnow)
3055             SET_QPEL_FUNC(qpel_pixels_tab[0][13], qpel16_mc13_3dnow)
3056             SET_QPEL_FUNC(qpel_pixels_tab[0][14], qpel16_mc23_3dnow)
3057             SET_QPEL_FUNC(qpel_pixels_tab[0][15], qpel16_mc33_3dnow)
3058             SET_QPEL_FUNC(qpel_pixels_tab[1][ 0], qpel8_mc00_3dnow)
3059             SET_QPEL_FUNC(qpel_pixels_tab[1][ 1], qpel8_mc10_3dnow)
3060             SET_QPEL_FUNC(qpel_pixels_tab[1][ 2], qpel8_mc20_3dnow)
3061             SET_QPEL_FUNC(qpel_pixels_tab[1][ 3], qpel8_mc30_3dnow)
3062             SET_QPEL_FUNC(qpel_pixels_tab[1][ 4], qpel8_mc01_3dnow)
3063             SET_QPEL_FUNC(qpel_pixels_tab[1][ 5], qpel8_mc11_3dnow)
3064             SET_QPEL_FUNC(qpel_pixels_tab[1][ 6], qpel8_mc21_3dnow)
3065             SET_QPEL_FUNC(qpel_pixels_tab[1][ 7], qpel8_mc31_3dnow)
3066             SET_QPEL_FUNC(qpel_pixels_tab[1][ 8], qpel8_mc02_3dnow)
3067             SET_QPEL_FUNC(qpel_pixels_tab[1][ 9], qpel8_mc12_3dnow)
3068             SET_QPEL_FUNC(qpel_pixels_tab[1][10], qpel8_mc22_3dnow)
3069             SET_QPEL_FUNC(qpel_pixels_tab[1][11], qpel8_mc32_3dnow)
3070             SET_QPEL_FUNC(qpel_pixels_tab[1][12], qpel8_mc03_3dnow)
3071             SET_QPEL_FUNC(qpel_pixels_tab[1][13], qpel8_mc13_3dnow)
3072             SET_QPEL_FUNC(qpel_pixels_tab[1][14], qpel8_mc23_3dnow)
3073             SET_QPEL_FUNC(qpel_pixels_tab[1][15], qpel8_mc33_3dnow)
3074
3075 #define dspfunc(PFX, IDX, NUM) \
3076     c->PFX ## _pixels_tab[IDX][ 0] = PFX ## NUM ## _mc00_3dnow; \
3077     c->PFX ## _pixels_tab[IDX][ 1] = PFX ## NUM ## _mc10_3dnow; \
3078     c->PFX ## _pixels_tab[IDX][ 2] = PFX ## NUM ## _mc20_3dnow; \
3079     c->PFX ## _pixels_tab[IDX][ 3] = PFX ## NUM ## _mc30_3dnow; \
3080     c->PFX ## _pixels_tab[IDX][ 4] = PFX ## NUM ## _mc01_3dnow; \
3081     c->PFX ## _pixels_tab[IDX][ 5] = PFX ## NUM ## _mc11_3dnow; \
3082     c->PFX ## _pixels_tab[IDX][ 6] = PFX ## NUM ## _mc21_3dnow; \
3083     c->PFX ## _pixels_tab[IDX][ 7] = PFX ## NUM ## _mc31_3dnow; \
3084     c->PFX ## _pixels_tab[IDX][ 8] = PFX ## NUM ## _mc02_3dnow; \
3085     c->PFX ## _pixels_tab[IDX][ 9] = PFX ## NUM ## _mc12_3dnow; \
3086     c->PFX ## _pixels_tab[IDX][10] = PFX ## NUM ## _mc22_3dnow; \
3087     c->PFX ## _pixels_tab[IDX][11] = PFX ## NUM ## _mc32_3dnow; \
3088     c->PFX ## _pixels_tab[IDX][12] = PFX ## NUM ## _mc03_3dnow; \
3089     c->PFX ## _pixels_tab[IDX][13] = PFX ## NUM ## _mc13_3dnow; \
3090     c->PFX ## _pixels_tab[IDX][14] = PFX ## NUM ## _mc23_3dnow; \
3091     c->PFX ## _pixels_tab[IDX][15] = PFX ## NUM ## _mc33_3dnow
3092
3093             dspfunc(put_h264_qpel, 0, 16);
3094             dspfunc(put_h264_qpel, 1, 8);
3095             dspfunc(put_h264_qpel, 2, 4);
3096             dspfunc(avg_h264_qpel, 0, 16);
3097             dspfunc(avg_h264_qpel, 1, 8);
3098             dspfunc(avg_h264_qpel, 2, 4);
3099
3100             c->avg_h264_chroma_pixels_tab[0]= avg_h264_chroma_mc8_3dnow;
3101             c->avg_h264_chroma_pixels_tab[1]= avg_h264_chroma_mc4_3dnow;
3102         }
3103
3104 #ifdef CONFIG_SNOW_ENCODER
3105         if(mm_flags & MM_SSE2){
3106             c->horizontal_compose97i = ff_snow_horizontal_compose97i_sse2;
3107             c->vertical_compose97i = ff_snow_vertical_compose97i_sse2;
3108             c->inner_add_yblock = ff_snow_inner_add_yblock_sse2;
3109         }
3110         else{
3111             c->horizontal_compose97i = ff_snow_horizontal_compose97i_mmx;
3112             c->vertical_compose97i = ff_snow_vertical_compose97i_mmx;
3113             c->inner_add_yblock = ff_snow_inner_add_yblock_mmx;
3114         }
3115 #endif
3116     }
3117
3118 #ifdef CONFIG_ENCODERS
3119     dsputil_init_pix_mmx(c, avctx);
3120 #endif //CONFIG_ENCODERS
3121 #if 0
3122     // for speed testing
3123     get_pixels = just_return;
3124     put_pixels_clamped = just_return;
3125     add_pixels_clamped = just_return;
3126
3127     pix_abs16x16 = just_return;
3128     pix_abs16x16_x2 = just_return;
3129     pix_abs16x16_y2 = just_return;
3130     pix_abs16x16_xy2 = just_return;
3131
3132     put_pixels_tab[0] = just_return;
3133     put_pixels_tab[1] = just_return;
3134     put_pixels_tab[2] = just_return;
3135     put_pixels_tab[3] = just_return;
3136
3137     put_no_rnd_pixels_tab[0] = just_return;
3138     put_no_rnd_pixels_tab[1] = just_return;
3139     put_no_rnd_pixels_tab[2] = just_return;
3140     put_no_rnd_pixels_tab[3] = just_return;
3141
3142     avg_pixels_tab[0] = just_return;
3143     avg_pixels_tab[1] = just_return;
3144     avg_pixels_tab[2] = just_return;
3145     avg_pixels_tab[3] = just_return;
3146
3147     avg_no_rnd_pixels_tab[0] = just_return;
3148     avg_no_rnd_pixels_tab[1] = just_return;
3149     avg_no_rnd_pixels_tab[2] = just_return;
3150     avg_no_rnd_pixels_tab[3] = just_return;
3151
3152     //av_fdct = just_return;
3153     //ff_idct = just_return;
3154 #endif
3155 }