vf_fspp: use LOCAL_MANGLE instead of MANGLE.
[ffmpeg.git] / libavfilter / libmpcodecs / vf_fspp.c
1 /*
2  * Copyright (C) 2003 Michael Niedermayer <michaelni@gmx.at>
3  * Copyright (C) 2005 Nikolaj Poroshin <porosh3@psu.ru>
4  *
5  * This file is part of MPlayer.
6  *
7  * MPlayer is free software; you can redistribute it and/or modify
8  * it under the terms of the GNU General Public License as published by
9  * the Free Software Foundation; either version 2 of the License, or
10  * (at your option) any later version.
11  *
12  * MPlayer is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15  * GNU General Public License for more details.
16  *
17  * You should have received a copy of the GNU General Public License along
18  * with MPlayer; if not, write to the Free Software Foundation, Inc.,
19  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
20  */
21
22 /*
23  * This implementation is based on an algorithm described in
24  * "Aria Nosratinia Embedded Post-Processing for
25  * Enhancement of Compressed Images (1999)"
26  * (http://citeseer.nj.nec.com/nosratinia99embedded.html)
27  * Futher, with splitting (i)dct into hor/ver passes, one of them can be
28  * performed once per block, not pixel. This allows for much better speed.
29  */
30
31 /*
32   Heavily optimized version of SPP filter by Nikolaj
33  */
34
35 #include <stdio.h>
36 #include <stdlib.h>
37 #include <string.h>
38 #include <inttypes.h>
39 #include <math.h>
40
41 #include "config.h"
42
43 #include "mp_msg.h"
44 #include "cpudetect.h"
45 #include "img_format.h"
46 #include "mp_image.h"
47 #include "vf.h"
48 #include "vd_ffmpeg.h"
49 #include "libvo/fastmemcpy.h"
50
51 #include "libavutil/internal.h"
52 #include "libavutil/intreadwrite.h"
53 #include "libavutil/mem.h"
54 #include "libavcodec/avcodec.h"
55 #include "libavcodec/dsputil.h"
56
57 #undef free
58 #undef malloc
59
60 //===========================================================================//
61 #define BLOCKSZ 12
62
63 static const short custom_threshold[64]=
64 // values (296) can't be too high
65 // -it causes too big quant dependence
66 // or maybe overflow(check), which results in some flashing
67 { 71, 296, 295, 237,  71,  40,  38,  19,
68   245, 193, 185, 121, 102,  73,  53,  27,
69   158, 129, 141, 107,  97,  73,  50,  26,
70   102, 116, 109,  98,  82,  66,  45,  23,
71   71,  94,  95,  81,  70,  56,  38,  20,
72   56,  77,  74,  66,  56,  44,  30,  15,
73   38,  53,  50,  45,  38,  30,  21,  11,
74   20,  27,  26,  23,  20,  15,  11,   5
75 };
76
77 static const uint8_t  __attribute__((aligned(32))) dither[8][8]={
78     {  0,  48,  12,  60,   3,  51,  15,  63, },
79     { 32,  16,  44,  28,  35,  19,  47,  31, },
80     {  8,  56,   4,  52,  11,  59,   7,  55, },
81     { 40,  24,  36,  20,  43,  27,  39,  23, },
82     {  2,  50,  14,  62,   1,  49,  13,  61, },
83     { 34,  18,  46,  30,  33,  17,  45,  29, },
84     { 10,  58,   6,  54,   9,  57,   5,  53, },
85     { 42,  26,  38,  22,  41,  25,  37,  21, },
86 };
87
88 struct vf_priv_s { //align 16 !
89     uint64_t threshold_mtx_noq[8*2];
90     uint64_t threshold_mtx[8*2];//used in both C & MMX (& later SSE2) versions
91
92     int log2_count;
93     int temp_stride;
94     int qp;
95     int mpeg2;
96     int prev_q;
97     uint8_t *src;
98     int16_t *temp;
99     int bframes;
100     char *non_b_qp;
101 };
102
103
104 #if !HAVE_MMX
105
106 //This func reads from 1 slice, 1 and clears 0 & 1
107 static void store_slice_c(uint8_t *dst, int16_t *src, int dst_stride, int src_stride, int width, int height, int log2_scale)
108 {int y, x;
109 #define STORE(pos)                                                        \
110     temp= (src[x + pos] + (d[pos]>>log2_scale))>>(6-log2_scale);        \
111     src[x + pos]=src[x + pos - 8*src_stride]=0;                                \
112     if(temp & 0x100) temp= ~(temp>>31);                                        \
113     dst[x + pos]= temp;
114
115     for(y=0; y<height; y++){
116         const uint8_t *d= dither[y];
117         for(x=0; x<width; x+=8){
118             int temp;
119             STORE(0);
120             STORE(1);
121             STORE(2);
122             STORE(3);
123             STORE(4);
124             STORE(5);
125             STORE(6);
126             STORE(7);
127         }
128         src+=src_stride;
129         dst+=dst_stride;
130     }
131 }
132
133 //This func reads from 2 slices, 0 & 2  and clears 2-nd
134 static void store_slice2_c(uint8_t *dst, int16_t *src, int dst_stride, int src_stride, int width, int height, int log2_scale)
135 {int y, x;
136 #define STORE2(pos)                                                        \
137     temp= (src[x + pos] + src[x + pos + 16*src_stride] + (d[pos]>>log2_scale))>>(6-log2_scale);        \
138     src[x + pos + 16*src_stride]=0;                                        \
139     if(temp & 0x100) temp= ~(temp>>31);                                        \
140     dst[x + pos]= temp;
141
142     for(y=0; y<height; y++){
143         const uint8_t *d= dither[y];
144         for(x=0; x<width; x+=8){
145             int temp;
146             STORE2(0);
147             STORE2(1);
148             STORE2(2);
149             STORE2(3);
150             STORE2(4);
151             STORE2(5);
152             STORE2(6);
153             STORE2(7);
154         }
155         src+=src_stride;
156         dst+=dst_stride;
157     }
158 }
159
160 static void mul_thrmat_c(struct vf_priv_s *p,int q)
161 {
162     int a;
163     for(a=0;a<64;a++)
164         ((short*)p->threshold_mtx)[a]=q * ((short*)p->threshold_mtx_noq)[a];//ints faster in C
165 }
166
167 static void column_fidct_c(int16_t* thr_adr, DCTELEM *data, DCTELEM *output, int cnt);
168 static void row_idct_c(DCTELEM* workspace,
169                        int16_t* output_adr, int output_stride, int cnt);
170 static void row_fdct_c(DCTELEM *data, const uint8_t *pixels, int line_size, int cnt);
171
172 //this is rather ugly, but there is no need for function pointers
173 #define store_slice_s store_slice_c
174 #define store_slice2_s store_slice2_c
175 #define mul_thrmat_s mul_thrmat_c
176 #define column_fidct_s column_fidct_c
177 #define row_idct_s row_idct_c
178 #define row_fdct_s row_fdct_c
179
180 #else /* HAVE_MMX */
181
182 //This func reads from 1 slice, 1 and clears 0 & 1
183 static void store_slice_mmx(uint8_t *dst, int16_t *src, long dst_stride, long src_stride, long width, long height, long log2_scale)
184 {
185     const uint8_t *od=&dither[0][0];
186     const uint8_t *end=&dither[height][0];
187     width = (width+7)&~7;
188     dst_stride-=width;
189     //src_stride=(src_stride-width)*2;
190     __asm__ volatile(
191         "mov %5, %%"REG_d"                \n\t"
192         "mov %6, %%"REG_S"                \n\t"
193         "mov %7, %%"REG_D"                \n\t"
194         "mov %1, %%"REG_a"                \n\t"
195         "movd %%"REG_d", %%mm5             \n\t"
196         "xor $-1, %%"REG_d"              \n\t"
197         "mov %%"REG_a", %%"REG_c"             \n\t"
198         "add $7, %%"REG_d"               \n\t"
199         "neg %%"REG_a"                   \n\t"
200         "sub %0, %%"REG_c"            \n\t"
201         "add %%"REG_c", %%"REG_c"             \n\t"
202         "movd %%"REG_d", %%mm2             \n\t"
203         "mov %%"REG_c", %1       \n\t"
204         "mov %2, %%"REG_d"               \n\t"
205         "shl $4, %%"REG_a"               \n\t"
206
207         "2:                        \n\t"
208         "movq (%%"REG_d"), %%mm3           \n\t"
209         "movq %%mm3, %%mm4             \n\t"
210         "pxor %%mm7, %%mm7             \n\t"
211         "punpcklbw %%mm7, %%mm3        \n\t"
212         "punpckhbw %%mm7, %%mm4        \n\t"
213         "mov %0, %%"REG_c"            \n\t"
214         "psraw %%mm5, %%mm3            \n\t"
215         "psraw %%mm5, %%mm4            \n\t"
216         "1:                        \n\t"
217         "movq %%mm7, (%%"REG_S",%%"REG_a",)     \n\t"
218         "movq (%%"REG_S"), %%mm0           \n\t"
219         "movq 8(%%"REG_S"), %%mm1          \n\t"
220
221         "movq %%mm7, 8(%%"REG_S",%%"REG_a",)    \n\t"
222         "paddw %%mm3, %%mm0            \n\t"
223         "paddw %%mm4, %%mm1            \n\t"
224
225         "movq %%mm7, (%%"REG_S")           \n\t"
226         "psraw %%mm2, %%mm0            \n\t"
227         "psraw %%mm2, %%mm1            \n\t"
228
229         "movq %%mm7, 8(%%"REG_S")          \n\t"
230         "packuswb %%mm1, %%mm0         \n\t"
231         "add $16, %%"REG_S"              \n\t"
232
233         "movq %%mm0, (%%"REG_D")           \n\t"
234         "add $8, %%"REG_D"               \n\t"
235         "sub $8, %%"REG_c"               \n\t"
236         "jg 1b                      \n\t"
237         "add %1, %%"REG_S"       \n\t"
238         "add $8, %%"REG_d"               \n\t"
239         "add %3, %%"REG_D"       \n\t"
240         "cmp %4, %%"REG_d"           \n\t"
241         "jl 2b                      \n\t"
242
243         :
244         : "m" (width), "m" (src_stride), "erm" (od), "m" (dst_stride), "erm" (end),
245           "m" (log2_scale), "m" (src), "m" (dst) //input
246         : "%"REG_a, "%"REG_c, "%"REG_d, "%"REG_S, "%"REG_D
247         );
248 }
249
250 //This func reads from 2 slices, 0 & 2  and clears 2-nd
251 static void store_slice2_mmx(uint8_t *dst, int16_t *src, long dst_stride, long src_stride, long width, long height, long log2_scale)
252 {
253     const uint8_t *od=&dither[0][0];
254     const uint8_t *end=&dither[height][0];
255     width = (width+7)&~7;
256     dst_stride-=width;
257     //src_stride=(src_stride-width)*2;
258     __asm__ volatile(
259         "mov %5, %%"REG_d"                \n\t"
260         "mov %6, %%"REG_S"                \n\t"
261         "mov %7, %%"REG_D"                \n\t"
262         "mov %1, %%"REG_a"            \n\t"
263         "movd %%"REG_d", %%mm5             \n\t"
264         "xor $-1, %%"REG_d"              \n\t"
265         "mov %%"REG_a", %%"REG_c"             \n\t"
266         "add $7, %%"REG_d"               \n\t"
267         "sub %0, %%"REG_c"            \n\t"
268         "add %%"REG_c", %%"REG_c"             \n\t"
269         "movd %%"REG_d", %%mm2             \n\t"
270         "mov %%"REG_c", %1       \n\t"
271         "mov %2, %%"REG_d"               \n\t"
272         "shl $5, %%"REG_a"               \n\t"
273
274         "2:                        \n\t"
275         "movq (%%"REG_d"), %%mm3           \n\t"
276         "movq %%mm3, %%mm4             \n\t"
277         "pxor %%mm7, %%mm7             \n\t"
278         "punpcklbw %%mm7, %%mm3        \n\t"
279         "punpckhbw %%mm7, %%mm4        \n\t"
280         "mov %0, %%"REG_c"            \n\t"
281         "psraw %%mm5, %%mm3            \n\t"
282         "psraw %%mm5, %%mm4            \n\t"
283         "1:                        \n\t"
284         "movq (%%"REG_S"), %%mm0           \n\t"
285         "movq 8(%%"REG_S"), %%mm1          \n\t"
286         "paddw %%mm3, %%mm0            \n\t"
287
288         "paddw (%%"REG_S",%%"REG_a",), %%mm0    \n\t"
289         "paddw %%mm4, %%mm1            \n\t"
290         "movq 8(%%"REG_S",%%"REG_a",), %%mm6    \n\t"
291
292         "movq %%mm7, (%%"REG_S",%%"REG_a",)     \n\t"
293         "psraw %%mm2, %%mm0            \n\t"
294         "paddw %%mm6, %%mm1            \n\t"
295
296         "movq %%mm7, 8(%%"REG_S",%%"REG_a",)    \n\t"
297         "psraw %%mm2, %%mm1            \n\t"
298         "packuswb %%mm1, %%mm0         \n\t"
299
300         "movq %%mm0, (%%"REG_D")           \n\t"
301         "add $16, %%"REG_S"              \n\t"
302         "add $8, %%"REG_D"               \n\t"
303         "sub $8, %%"REG_c"               \n\t"
304         "jg 1b                      \n\t"
305         "add %1, %%"REG_S"       \n\t"
306         "add $8, %%"REG_d"               \n\t"
307         "add %3, %%"REG_D"       \n\t"
308         "cmp %4, %%"REG_d"           \n\t"
309         "jl 2b                      \n\t"
310
311         :
312         : "m" (width), "m" (src_stride), "erm" (od), "m" (dst_stride), "erm" (end),
313           "m" (log2_scale), "m" (src), "m" (dst) //input
314         : "%"REG_a, "%"REG_c, "%"REG_d, "%"REG_D, "%"REG_S
315         );
316 }
317
318 static void mul_thrmat_mmx(struct vf_priv_s *p, int q)
319 {
320     uint64_t *adr=&p->threshold_mtx_noq[0];
321     __asm__ volatile(
322         "movd %0, %%mm7                \n\t"
323         "add $8*8*2, %%"REG_D"            \n\t"
324         "movq 0*8(%%"REG_S"), %%mm0        \n\t"
325         "punpcklwd %%mm7, %%mm7        \n\t"
326         "movq 1*8(%%"REG_S"), %%mm1        \n\t"
327         "punpckldq %%mm7, %%mm7        \n\t"
328         "pmullw %%mm7, %%mm0           \n\t"
329
330         "movq 2*8(%%"REG_S"), %%mm2        \n\t"
331         "pmullw %%mm7, %%mm1           \n\t"
332
333         "movq 3*8(%%"REG_S"), %%mm3        \n\t"
334         "pmullw %%mm7, %%mm2           \n\t"
335
336         "movq %%mm0, 0*8(%%"REG_D")        \n\t"
337         "movq 4*8(%%"REG_S"), %%mm4        \n\t"
338         "pmullw %%mm7, %%mm3           \n\t"
339
340         "movq %%mm1, 1*8(%%"REG_D")        \n\t"
341         "movq 5*8(%%"REG_S"), %%mm5        \n\t"
342         "pmullw %%mm7, %%mm4           \n\t"
343
344         "movq %%mm2, 2*8(%%"REG_D")        \n\t"
345         "movq 6*8(%%"REG_S"), %%mm6        \n\t"
346         "pmullw %%mm7, %%mm5           \n\t"
347
348         "movq %%mm3, 3*8(%%"REG_D")        \n\t"
349         "movq 7*8+0*8(%%"REG_S"), %%mm0    \n\t"
350         "pmullw %%mm7, %%mm6           \n\t"
351
352         "movq %%mm4, 4*8(%%"REG_D")        \n\t"
353         "movq 7*8+1*8(%%"REG_S"), %%mm1    \n\t"
354         "pmullw %%mm7, %%mm0           \n\t"
355
356         "movq %%mm5, 5*8(%%"REG_D")        \n\t"
357         "movq 7*8+2*8(%%"REG_S"), %%mm2    \n\t"
358         "pmullw %%mm7, %%mm1           \n\t"
359
360         "movq %%mm6, 6*8(%%"REG_D")        \n\t"
361         "movq 7*8+3*8(%%"REG_S"), %%mm3    \n\t"
362         "pmullw %%mm7, %%mm2           \n\t"
363
364         "movq %%mm0, 7*8+0*8(%%"REG_D")    \n\t"
365         "movq 7*8+4*8(%%"REG_S"), %%mm4    \n\t"
366         "pmullw %%mm7, %%mm3           \n\t"
367
368         "movq %%mm1, 7*8+1*8(%%"REG_D")    \n\t"
369         "movq 7*8+5*8(%%"REG_S"), %%mm5    \n\t"
370         "pmullw %%mm7, %%mm4           \n\t"
371
372         "movq %%mm2, 7*8+2*8(%%"REG_D")    \n\t"
373         "movq 7*8+6*8(%%"REG_S"), %%mm6    \n\t"
374         "pmullw %%mm7, %%mm5           \n\t"
375
376         "movq %%mm3, 7*8+3*8(%%"REG_D")    \n\t"
377         "movq 14*8+0*8(%%"REG_S"), %%mm0   \n\t"
378         "pmullw %%mm7, %%mm6           \n\t"
379
380         "movq %%mm4, 7*8+4*8(%%"REG_D")    \n\t"
381         "movq 14*8+1*8(%%"REG_S"), %%mm1   \n\t"
382         "pmullw %%mm7, %%mm0           \n\t"
383
384         "movq %%mm5, 7*8+5*8(%%"REG_D")    \n\t"
385         "pmullw %%mm7, %%mm1           \n\t"
386
387         "movq %%mm6, 7*8+6*8(%%"REG_D")    \n\t"
388         "movq %%mm0, 14*8+0*8(%%"REG_D")   \n\t"
389         "movq %%mm1, 14*8+1*8(%%"REG_D")   \n\t"
390
391         : "+g" (q), "+S" (adr), "+D" (adr)
392         :
393         );
394 }
395
396 static void column_fidct_mmx(int16_t* thr_adr,  DCTELEM *data,  DCTELEM *output,  int cnt);
397 static void row_idct_mmx(DCTELEM* workspace,
398                          int16_t* output_adr,  int output_stride,  int cnt);
399 static void row_fdct_mmx(DCTELEM *data,  const uint8_t *pixels,  int line_size,  int cnt);
400
401 #define store_slice_s store_slice_mmx
402 #define store_slice2_s store_slice2_mmx
403 #define mul_thrmat_s mul_thrmat_mmx
404 #define column_fidct_s column_fidct_mmx
405 #define row_idct_s row_idct_mmx
406 #define row_fdct_s row_fdct_mmx
407 #endif // HAVE_MMX
408
409 static void filter(struct vf_priv_s *p, uint8_t *dst, uint8_t *src,
410                    int dst_stride, int src_stride,
411                    int width, int height,
412                    uint8_t *qp_store, int qp_stride, int is_luma)
413 {
414     int x, x0, y, es, qy, t;
415     const int stride= is_luma ? p->temp_stride : (width+16);//((width+16+15)&(~15))
416     const int step=6-p->log2_count;
417     const int qps= 3 + is_luma;
418     int32_t __attribute__((aligned(32))) block_align[4*8*BLOCKSZ+ 4*8*BLOCKSZ];
419     DCTELEM *block= (DCTELEM *)block_align;
420     DCTELEM *block3=(DCTELEM *)(block_align+4*8*BLOCKSZ);
421
422     memset(block3, 0, 4*8*BLOCKSZ);
423
424     //p->src=src-src_stride*8-8;//!
425     if (!src || !dst) return; // HACK avoid crash for Y8 colourspace
426     for(y=0; y<height; y++){
427         int index= 8 + 8*stride + y*stride;
428         fast_memcpy(p->src + index, src + y*src_stride, width);//this line can be avoided by using DR & user fr.buffers
429         for(x=0; x<8; x++){
430             p->src[index         - x - 1]= p->src[index +         x    ];
431             p->src[index + width + x    ]= p->src[index + width - x - 1];
432         }
433     }
434     for(y=0; y<8; y++){
435         fast_memcpy(p->src + (      7-y)*stride, p->src + (      y+8)*stride, stride);
436         fast_memcpy(p->src + (height+8+y)*stride, p->src + (height-y+7)*stride, stride);
437     }
438     //FIXME (try edge emu)
439
440     for(y=8; y<24; y++)
441         memset(p->temp+ 8 +y*stride, 0,width*sizeof(int16_t));
442
443     for(y=step; y<height+8; y+=step){    //step= 1,2
444         qy=y-4;
445         if (qy>height-1) qy=height-1;
446         if (qy<0) qy=0;
447         qy=(qy>>qps)*qp_stride;
448         row_fdct_s(block, p->src + y*stride +2-(y&1), stride, 2);
449         for(x0=0; x0<width+8-8*(BLOCKSZ-1); x0+=8*(BLOCKSZ-1)){
450             row_fdct_s(block+8*8, p->src + y*stride+8+x0 +2-(y&1), stride, 2*(BLOCKSZ-1));
451             if(p->qp)
452                 column_fidct_s((int16_t*)(&p->threshold_mtx[0]), block+0*8, block3+0*8, 8*(BLOCKSZ-1)); //yes, this is a HOTSPOT
453             else
454                 for (x=0; x<8*(BLOCKSZ-1); x+=8) {
455                     t=x+x0-2; //correct t=x+x0-2-(y&1), but its the same
456                     if (t<0) t=0;//t always < width-2
457                     t=qp_store[qy+(t>>qps)];
458                     t=norm_qscale(t, p->mpeg2);
459                     if (t!=p->prev_q) p->prev_q=t, mul_thrmat_s(p, t);
460                     column_fidct_s((int16_t*)(&p->threshold_mtx[0]), block+x*8, block3+x*8, 8); //yes, this is a HOTSPOT
461                 }
462             row_idct_s(block3+0*8, p->temp + (y&15)*stride+x0+2-(y&1), stride, 2*(BLOCKSZ-1));
463             memmove(block, block+(BLOCKSZ-1)*64, 8*8*sizeof(DCTELEM)); //cycling
464             memmove(block3, block3+(BLOCKSZ-1)*64, 6*8*sizeof(DCTELEM));
465         }
466         //
467         es=width+8-x0; //  8, ...
468         if (es>8)
469             row_fdct_s(block+8*8, p->src + y*stride+8+x0 +2-(y&1), stride, (es-4)>>2);
470         column_fidct_s((int16_t*)(&p->threshold_mtx[0]), block, block3, es&(~1));
471         row_idct_s(block3+0*8, p->temp + (y&15)*stride+x0+2-(y&1), stride, es>>2);
472         {const int y1=y-8+step;//l5-7  l4-6
473             if (!(y1&7) && y1) {
474                 if (y1&8) store_slice_s(dst + (y1-8)*dst_stride, p->temp+ 8 +8*stride,
475                                         dst_stride, stride, width, 8, 5-p->log2_count);
476                 else store_slice2_s(dst + (y1-8)*dst_stride, p->temp+ 8 +0*stride,
477                                     dst_stride, stride, width, 8, 5-p->log2_count);
478             } }
479     }
480
481     if (y&7) {  // == height & 7
482         if (y&8) store_slice_s(dst + ((y-8)&~7)*dst_stride, p->temp+ 8 +8*stride,
483                                dst_stride, stride, width, y&7, 5-p->log2_count);
484         else store_slice2_s(dst + ((y-8)&~7)*dst_stride, p->temp+ 8 +0*stride,
485                             dst_stride, stride, width, y&7, 5-p->log2_count);
486     }
487 }
488
489 static int config(struct vf_instance *vf,
490                   int width, int height, int d_width, int d_height,
491                   unsigned int flags, unsigned int outfmt)
492 {
493     int h= (height+16+15)&(~15);
494
495     vf->priv->temp_stride= (width+16+15)&(~15);
496     vf->priv->temp= (int16_t*)av_mallocz(vf->priv->temp_stride*3*8*sizeof(int16_t));
497     //this can also be avoided, see above
498     vf->priv->src = (uint8_t*)av_malloc(vf->priv->temp_stride*h*sizeof(uint8_t));
499
500     return vf_next_config(vf,width,height,d_width,d_height,flags,outfmt);
501 }
502
503 static void get_image(struct vf_instance *vf, mp_image_t *mpi)
504 {
505     if(mpi->flags&MP_IMGFLAG_PRESERVE) return; // don't change
506     // ok, we can do pp in-place (or pp disabled):
507     vf->dmpi=vf_get_image(vf->next,mpi->imgfmt,
508                           mpi->type, mpi->flags, mpi->width, mpi->height);
509     mpi->planes[0]=vf->dmpi->planes[0];
510     mpi->stride[0]=vf->dmpi->stride[0];
511     mpi->width=vf->dmpi->width;
512     if(mpi->flags&MP_IMGFLAG_PLANAR){
513         mpi->planes[1]=vf->dmpi->planes[1];
514         mpi->planes[2]=vf->dmpi->planes[2];
515         mpi->stride[1]=vf->dmpi->stride[1];
516         mpi->stride[2]=vf->dmpi->stride[2];
517     }
518     mpi->flags|=MP_IMGFLAG_DIRECT;
519 }
520
521 static int put_image(struct vf_instance *vf, mp_image_t *mpi, double pts)
522 {
523     mp_image_t *dmpi;
524     if(!(mpi->flags&MP_IMGFLAG_DIRECT)){
525         // no DR, so get a new image! hope we'll get DR buffer:
526         dmpi=vf_get_image(vf->next,mpi->imgfmt,
527                           MP_IMGTYPE_TEMP,
528                           MP_IMGFLAG_ACCEPT_STRIDE|MP_IMGFLAG_PREFER_ALIGNED_STRIDE,
529                           mpi->width,mpi->height);
530         vf_clone_mpi_attributes(dmpi, mpi);
531     }else{
532         dmpi=vf->dmpi;
533     }
534
535     vf->priv->mpeg2= mpi->qscale_type;
536     if(mpi->pict_type != 3 && mpi->qscale && !vf->priv->qp){
537         int w = mpi->qstride;
538         int h = (mpi->h + 15) >> 4;
539         if (!w) {
540             w = (mpi->w + 15) >> 4;
541             h = 1;
542         }
543         if(!vf->priv->non_b_qp)
544             vf->priv->non_b_qp= malloc(w*h);
545         fast_memcpy(vf->priv->non_b_qp, mpi->qscale, w*h);
546     }
547     if(vf->priv->log2_count || !(mpi->flags&MP_IMGFLAG_DIRECT)){
548         char *qp_tab= vf->priv->non_b_qp;
549         if(vf->priv->bframes || !qp_tab)
550             qp_tab= mpi->qscale;
551
552         if(qp_tab || vf->priv->qp){
553             filter(vf->priv, dmpi->planes[0], mpi->planes[0], dmpi->stride[0], mpi->stride[0],
554                    mpi->w, mpi->h, qp_tab, mpi->qstride, 1);
555             filter(vf->priv, dmpi->planes[1], mpi->planes[1], dmpi->stride[1], mpi->stride[1],
556                    mpi->w>>mpi->chroma_x_shift, mpi->h>>mpi->chroma_y_shift, qp_tab, mpi->qstride, 0);
557             filter(vf->priv, dmpi->planes[2], mpi->planes[2], dmpi->stride[2], mpi->stride[2],
558                    mpi->w>>mpi->chroma_x_shift, mpi->h>>mpi->chroma_y_shift, qp_tab, mpi->qstride, 0);
559         }else{
560             memcpy_pic(dmpi->planes[0], mpi->planes[0], mpi->w, mpi->h, dmpi->stride[0], mpi->stride[0]);
561             memcpy_pic(dmpi->planes[1], mpi->planes[1], mpi->w>>mpi->chroma_x_shift, mpi->h>>mpi->chroma_y_shift, dmpi->stride[1], mpi->stride[1]);
562             memcpy_pic(dmpi->planes[2], mpi->planes[2], mpi->w>>mpi->chroma_x_shift, mpi->h>>mpi->chroma_y_shift, dmpi->stride[2], mpi->stride[2]);
563         }
564     }
565
566 #if HAVE_MMX
567     if(gCpuCaps.hasMMX) __asm__ volatile ("emms\n\t");
568 #endif
569 #if HAVE_MMX2
570     if(gCpuCaps.hasMMX2) __asm__ volatile ("sfence\n\t");
571 #endif
572     return vf_next_put_image(vf,dmpi, pts);
573 }
574
575 static void uninit(struct vf_instance *vf)
576 {
577     if(!vf->priv) return;
578
579     av_free(vf->priv->temp);
580     vf->priv->temp= NULL;
581     av_free(vf->priv->src);
582     vf->priv->src= NULL;
583     //free(vf->priv->avctx);
584     //vf->priv->avctx= NULL;
585     free(vf->priv->non_b_qp);
586     vf->priv->non_b_qp= NULL;
587
588     av_free(vf->priv);
589     vf->priv=NULL;
590 }
591
592 //===========================================================================//
593
594 static int query_format(struct vf_instance *vf, unsigned int fmt)
595 {
596     switch(fmt){
597     case IMGFMT_YVU9:
598     case IMGFMT_IF09:
599     case IMGFMT_YV12:
600     case IMGFMT_I420:
601     case IMGFMT_IYUV:
602     case IMGFMT_CLPL:
603     case IMGFMT_Y800:
604     case IMGFMT_Y8:
605     case IMGFMT_444P:
606     case IMGFMT_422P:
607     case IMGFMT_411P:
608         return vf_next_query_format(vf,fmt);
609     }
610     return 0;
611 }
612
613 static int control(struct vf_instance *vf, int request, void* data)
614 {
615     switch(request){
616     case VFCTRL_QUERY_MAX_PP_LEVEL:
617         return 5;
618     case VFCTRL_SET_PP_LEVEL:
619         vf->priv->log2_count= *((unsigned int*)data);
620         if (vf->priv->log2_count < 4) vf->priv->log2_count=4;
621         return CONTROL_TRUE;
622     }
623     return vf_next_control(vf,request,data);
624 }
625
626 static int vf_open(vf_instance_t *vf, char *args)
627 {
628     int i=0, bias;
629     int custom_threshold_m[64];
630     int log2c=-1;
631
632     vf->config=config;
633     vf->put_image=put_image;
634     vf->get_image=get_image;
635     vf->query_format=query_format;
636     vf->uninit=uninit;
637     vf->control= control;
638     vf->priv=av_mallocz(sizeof(struct vf_priv_s));//assumes align 16 !
639
640     init_avcodec();
641
642     //vf->priv->avctx= avcodec_alloc_context();
643     //dsputil_init(&vf->priv->dsp, vf->priv->avctx);
644
645     vf->priv->log2_count= 4;
646     vf->priv->bframes = 0;
647
648     if (args) sscanf(args, "%d:%d:%d:%d", &log2c, &vf->priv->qp, &i, &vf->priv->bframes);
649
650     if( log2c >=4 && log2c <=5 )
651         vf->priv->log2_count = log2c;
652     else if( log2c >= 6 )
653         vf->priv->log2_count = 5;
654
655     if(vf->priv->qp < 0)
656         vf->priv->qp = 0;
657
658     if (i < -15) i = -15;
659     if (i > 32) i = 32;
660
661     bias= (1<<4)+i; //regulable
662     vf->priv->prev_q=0;
663     //
664     for(i=0;i<64;i++) //FIXME: tune custom_threshold[] and remove this !
665         custom_threshold_m[i]=(int)(custom_threshold[i]*(bias/71.)+ 0.5);
666     for(i=0;i<8;i++){
667         vf->priv->threshold_mtx_noq[2*i]=(uint64_t)custom_threshold_m[i*8+2]
668             |(((uint64_t)custom_threshold_m[i*8+6])<<16)
669             |(((uint64_t)custom_threshold_m[i*8+0])<<32)
670             |(((uint64_t)custom_threshold_m[i*8+4])<<48);
671         vf->priv->threshold_mtx_noq[2*i+1]=(uint64_t)custom_threshold_m[i*8+5]
672             |(((uint64_t)custom_threshold_m[i*8+3])<<16)
673             |(((uint64_t)custom_threshold_m[i*8+1])<<32)
674             |(((uint64_t)custom_threshold_m[i*8+7])<<48);
675     }
676
677     if (vf->priv->qp) vf->priv->prev_q=vf->priv->qp, mul_thrmat_s(vf->priv, vf->priv->qp);
678
679     return 1;
680 }
681
682 const vf_info_t vf_info_fspp = {
683     "fast simple postprocess",
684     "fspp",
685     "Michael Niedermayer, Nikolaj Poroshin",
686     "",
687     vf_open,
688     NULL
689 };
690
691 //====================================================================
692 //Specific spp's dct, idct and threshold functions
693 //I'd prefer to have them in the separate file.
694
695 //typedef int16_t DCTELEM; //! only int16_t
696
697 #define DCTSIZE 8
698 #define DCTSIZE_S "8"
699
700 #define FIX(x,s)  ((int) ((x) * (1<<s) + 0.5)&0xffff)
701 #define C64(x)    ((uint64_t)((x)|(x)<<16))<<32 | (uint64_t)(x) | (uint64_t)(x)<<16
702 #define FIX64(x,s)  C64(FIX(x,s))
703
704 #define MULTIPLY16H(x,k)   (((x)*(k))>>16)
705 #define THRESHOLD(r,x,t) if(((unsigned)((x)+t))>t*2) r=(x);else r=0;
706 #define DESCALE(x,n)  (((x) + (1 << ((n)-1))) >> n)
707
708 #if HAVE_MMX
709
710 DECLARE_ASM_CONST(8, uint64_t, MM_FIX_0_382683433)=FIX64(0.382683433, 14);
711 DECLARE_ASM_CONST(8, uint64_t, MM_FIX_0_541196100)=FIX64(0.541196100, 14);
712 DECLARE_ASM_CONST(8, uint64_t, MM_FIX_0_707106781)=FIX64(0.707106781, 14);
713 DECLARE_ASM_CONST(8, uint64_t, MM_FIX_1_306562965)=FIX64(1.306562965, 14);
714
715 DECLARE_ASM_CONST(8, uint64_t, MM_FIX_1_414213562_A)=FIX64(1.414213562, 14);
716
717 DECLARE_ASM_CONST(8, uint64_t, MM_FIX_1_847759065)=FIX64(1.847759065, 13);
718 DECLARE_ASM_CONST(8, uint64_t, MM_FIX_2_613125930)=FIX64(-2.613125930, 13); //-
719 DECLARE_ASM_CONST(8, uint64_t, MM_FIX_1_414213562)=FIX64(1.414213562, 13);
720 DECLARE_ASM_CONST(8, uint64_t, MM_FIX_1_082392200)=FIX64(1.082392200, 13);
721 //for t3,t5,t7 == 0 shortcut
722 DECLARE_ASM_CONST(8, uint64_t, MM_FIX_0_847759065)=FIX64(0.847759065, 14);
723 DECLARE_ASM_CONST(8, uint64_t, MM_FIX_0_566454497)=FIX64(0.566454497, 14);
724 DECLARE_ASM_CONST(8, uint64_t, MM_FIX_0_198912367)=FIX64(0.198912367, 14);
725
726 DECLARE_ASM_CONST(8, uint64_t, MM_DESCALE_RND)=C64(4);
727 DECLARE_ASM_CONST(8, uint64_t, MM_2)=C64(2);
728
729 #else /* !HAVE_MMX */
730
731 typedef int32_t int_simd16_t;
732 static const int16_t FIX_0_382683433=FIX(0.382683433, 14);
733 static const int16_t FIX_0_541196100=FIX(0.541196100, 14);
734 static const int16_t FIX_0_707106781=FIX(0.707106781, 14);
735 static const int16_t FIX_1_306562965=FIX(1.306562965, 14);
736 static const int16_t FIX_1_414213562_A=FIX(1.414213562, 14);
737 static const int16_t FIX_1_847759065=FIX(1.847759065, 13);
738 static const int16_t FIX_2_613125930=FIX(-2.613125930, 13); //-
739 static const int16_t FIX_1_414213562=FIX(1.414213562, 13);
740 static const int16_t FIX_1_082392200=FIX(1.082392200, 13);
741
742 #endif
743
744 #if !HAVE_MMX
745
746 static void column_fidct_c(int16_t* thr_adr, DCTELEM *data, DCTELEM *output, int cnt)
747 {
748     int_simd16_t tmp0, tmp1, tmp2, tmp3, tmp4, tmp5, tmp6, tmp7;
749     int_simd16_t tmp10, tmp11, tmp12, tmp13;
750     int_simd16_t z1,z2,z3,z4,z5, z10, z11, z12, z13;
751     int_simd16_t d0, d1, d2, d3, d4, d5, d6, d7;
752
753     DCTELEM* dataptr;
754     DCTELEM* wsptr;
755     int16_t *threshold;
756     int ctr;
757
758     dataptr = data;
759     wsptr = output;
760
761     for (; cnt > 0; cnt-=2) { //start positions
762         threshold=(int16_t*)thr_adr;//threshold_mtx
763         for (ctr = DCTSIZE; ctr > 0; ctr--) {
764             // Process columns from input, add to output.
765             tmp0 = dataptr[DCTSIZE*0] + dataptr[DCTSIZE*7];
766             tmp7 = dataptr[DCTSIZE*0] - dataptr[DCTSIZE*7];
767
768             tmp1 = dataptr[DCTSIZE*1] + dataptr[DCTSIZE*6];
769             tmp6 = dataptr[DCTSIZE*1] - dataptr[DCTSIZE*6];
770
771             tmp2 = dataptr[DCTSIZE*2] + dataptr[DCTSIZE*5];
772             tmp5 = dataptr[DCTSIZE*2] - dataptr[DCTSIZE*5];
773
774             tmp3 = dataptr[DCTSIZE*3] + dataptr[DCTSIZE*4];
775             tmp4 = dataptr[DCTSIZE*3] - dataptr[DCTSIZE*4];
776
777             // Even part of FDCT
778
779             tmp10 = tmp0 + tmp3;
780             tmp13 = tmp0 - tmp3;
781             tmp11 = tmp1 + tmp2;
782             tmp12 = tmp1 - tmp2;
783
784             d0 = tmp10 + tmp11;
785             d4 = tmp10 - tmp11;
786
787             z1 = MULTIPLY16H((tmp12 + tmp13) <<2, FIX_0_707106781);
788             d2 = tmp13 + z1;
789             d6 = tmp13 - z1;
790
791             // Even part of IDCT
792
793             THRESHOLD(tmp0, d0, threshold[0*8]);
794             THRESHOLD(tmp1, d2, threshold[2*8]);
795             THRESHOLD(tmp2, d4, threshold[4*8]);
796             THRESHOLD(tmp3, d6, threshold[6*8]);
797             tmp0+=2;
798             tmp10 = (tmp0 + tmp2)>>2;
799             tmp11 = (tmp0 - tmp2)>>2;
800
801             tmp13 = (tmp1 + tmp3)>>2; //+2 !  (psnr decides)
802             tmp12 = MULTIPLY16H((tmp1 - tmp3), FIX_1_414213562_A) - tmp13; //<<2
803
804             tmp0 = tmp10 + tmp13; //->temps
805             tmp3 = tmp10 - tmp13; //->temps
806             tmp1 = tmp11 + tmp12; //->temps
807             tmp2 = tmp11 - tmp12; //->temps
808
809             // Odd part of FDCT
810
811             tmp10 = tmp4 + tmp5;
812             tmp11 = tmp5 + tmp6;
813             tmp12 = tmp6 + tmp7;
814
815             z5 = MULTIPLY16H((tmp10 - tmp12)<<2, FIX_0_382683433);
816             z2 = MULTIPLY16H(tmp10 <<2, FIX_0_541196100) + z5;
817             z4 = MULTIPLY16H(tmp12 <<2, FIX_1_306562965) + z5;
818             z3 = MULTIPLY16H(tmp11 <<2, FIX_0_707106781);
819
820             z11 = tmp7 + z3;
821             z13 = tmp7 - z3;
822
823             d5 = z13 + z2;
824             d3 = z13 - z2;
825             d1 = z11 + z4;
826             d7 = z11 - z4;
827
828             // Odd part of IDCT
829
830             THRESHOLD(tmp4, d1, threshold[1*8]);
831             THRESHOLD(tmp5, d3, threshold[3*8]);
832             THRESHOLD(tmp6, d5, threshold[5*8]);
833             THRESHOLD(tmp7, d7, threshold[7*8]);
834
835             //Simd version uses here a shortcut for the tmp5,tmp6,tmp7 == 0
836             z13 = tmp6 + tmp5;
837             z10 = (tmp6 - tmp5)<<1;
838             z11 = tmp4 + tmp7;
839             z12 = (tmp4 - tmp7)<<1;
840
841             tmp7 = (z11 + z13)>>2; //+2 !
842             tmp11 = MULTIPLY16H((z11 - z13)<<1, FIX_1_414213562);
843             z5 =    MULTIPLY16H(z10 + z12, FIX_1_847759065);
844             tmp10 = MULTIPLY16H(z12, FIX_1_082392200) - z5;
845             tmp12 = MULTIPLY16H(z10, FIX_2_613125930) + z5; // - !!
846
847             tmp6 = tmp12 - tmp7;
848             tmp5 = tmp11 - tmp6;
849             tmp4 = tmp10 + tmp5;
850
851             wsptr[DCTSIZE*0]+=  (tmp0 + tmp7);
852             wsptr[DCTSIZE*1]+=  (tmp1 + tmp6);
853             wsptr[DCTSIZE*2]+=  (tmp2 + tmp5);
854             wsptr[DCTSIZE*3]+=  (tmp3 - tmp4);
855             wsptr[DCTSIZE*4]+=  (tmp3 + tmp4);
856             wsptr[DCTSIZE*5]+=  (tmp2 - tmp5);
857             wsptr[DCTSIZE*6]=  (tmp1 - tmp6);
858             wsptr[DCTSIZE*7]=  (tmp0 - tmp7);
859             //
860             dataptr++; //next column
861             wsptr++;
862             threshold++;
863         }
864         dataptr+=8; //skip each second start pos
865         wsptr  +=8;
866     }
867 }
868
869 #else /* HAVE_MMX */
870
871 static void column_fidct_mmx(int16_t* thr_adr,  DCTELEM *data,  DCTELEM *output,  int cnt)
872 {
873     uint64_t __attribute__((aligned(8))) temps[4];
874     __asm__ volatile(
875         ASMALIGN(4)
876         "1:                   \n\t"
877         "movq "DCTSIZE_S"*0*2(%%"REG_S"), %%mm1 \n\t"
878         //
879         "movq "DCTSIZE_S"*3*2(%%"REG_S"), %%mm7 \n\t"
880         "movq %%mm1, %%mm0             \n\t"
881
882         "paddw "DCTSIZE_S"*7*2(%%"REG_S"), %%mm1 \n\t" //t0
883         "movq %%mm7, %%mm3             \n\t"
884
885         "paddw "DCTSIZE_S"*4*2(%%"REG_S"), %%mm7 \n\t" //t3
886         "movq %%mm1, %%mm5             \n\t"
887
888         "movq "DCTSIZE_S"*1*2(%%"REG_S"), %%mm6 \n\t"
889         "psubw %%mm7, %%mm1            \n\t" //t13
890
891         "movq "DCTSIZE_S"*2*2(%%"REG_S"), %%mm2 \n\t"
892         "movq %%mm6, %%mm4             \n\t"
893
894         "paddw "DCTSIZE_S"*6*2(%%"REG_S"), %%mm6 \n\t" //t1
895         "paddw %%mm7, %%mm5            \n\t" //t10
896
897         "paddw "DCTSIZE_S"*5*2(%%"REG_S"), %%mm2 \n\t" //t2
898         "movq %%mm6, %%mm7             \n\t"
899
900         "paddw %%mm2, %%mm6            \n\t" //t11
901         "psubw %%mm2, %%mm7            \n\t" //t12
902
903         "movq %%mm5, %%mm2             \n\t"
904         "paddw %%mm6, %%mm5            \n\t" //d0
905         // i0 t13 t12 i3 i1 d0 - d4
906         "psubw %%mm6, %%mm2            \n\t" //d4
907         "paddw %%mm1, %%mm7            \n\t"
908
909         "movq  4*16(%%"REG_d"), %%mm6      \n\t"
910         "psllw $2, %%mm7              \n\t"
911
912         "psubw 0*16(%%"REG_d"), %%mm5      \n\t"
913         "psubw %%mm6, %%mm2            \n\t"
914
915         "paddusw 0*16(%%"REG_d"), %%mm5    \n\t"
916         "paddusw %%mm6, %%mm2          \n\t"
917
918         "pmulhw "LOCAL_MANGLE(MM_FIX_0_707106781)", %%mm7 \n\t"
919         //
920         "paddw 0*16(%%"REG_d"), %%mm5      \n\t"
921         "paddw %%mm6, %%mm2            \n\t"
922
923         "psubusw 0*16(%%"REG_d"), %%mm5    \n\t"
924         "psubusw %%mm6, %%mm2          \n\t"
925
926 //This func is totally compute-bound,  operates at huge speed. So,  DC shortcut
927 // at this place isn't worthwhile due to BTB miss penalty (checked on Pent. 3).
928 //However,  typical numbers: nondc - 29%%,  dc - 46%%,  zero - 25%%. All <> 0 case is very rare.
929         "paddw "LOCAL_MANGLE(MM_2)", %%mm5            \n\t"
930         "movq %%mm2, %%mm6             \n\t"
931
932         "paddw %%mm5, %%mm2            \n\t"
933         "psubw %%mm6, %%mm5            \n\t"
934
935         "movq %%mm1, %%mm6             \n\t"
936         "paddw %%mm7, %%mm1            \n\t" //d2
937
938         "psubw 2*16(%%"REG_d"), %%mm1      \n\t"
939         "psubw %%mm7, %%mm6            \n\t" //d6
940
941         "movq 6*16(%%"REG_d"), %%mm7       \n\t"
942         "psraw $2, %%mm5              \n\t"
943
944         "paddusw 2*16(%%"REG_d"), %%mm1    \n\t"
945         "psubw %%mm7, %%mm6            \n\t"
946         // t7 d2 /t11 t4 t6 - d6 /t10
947
948         "paddw 2*16(%%"REG_d"), %%mm1      \n\t"
949         "paddusw %%mm7, %%mm6          \n\t"
950
951         "psubusw 2*16(%%"REG_d"), %%mm1    \n\t"
952         "paddw %%mm7, %%mm6            \n\t"
953
954         "psubw "DCTSIZE_S"*4*2(%%"REG_S"), %%mm3 \n\t"
955         "psubusw %%mm7, %%mm6          \n\t"
956
957         //movq [edi+"DCTSIZE_S"*2*2], mm1
958         //movq [edi+"DCTSIZE_S"*6*2], mm6
959         "movq %%mm1, %%mm7             \n\t"
960         "psraw $2, %%mm2              \n\t"
961
962         "psubw "DCTSIZE_S"*6*2(%%"REG_S"), %%mm4 \n\t"
963         "psubw %%mm6, %%mm1            \n\t"
964
965         "psubw "DCTSIZE_S"*7*2(%%"REG_S"), %%mm0 \n\t"
966         "paddw %%mm7, %%mm6            \n\t" //'t13
967
968         "psraw $2, %%mm6              \n\t" //paddw mm6, MM_2 !!    ---
969         "movq %%mm2, %%mm7             \n\t"
970
971         "pmulhw "LOCAL_MANGLE(MM_FIX_1_414213562_A)", %%mm1 \n\t"
972         "paddw %%mm6, %%mm2            \n\t" //'t0
973
974         "movq %%mm2, 0*8+%3            \n\t" //!
975         "psubw %%mm6, %%mm7            \n\t" //'t3
976
977         "movq "DCTSIZE_S"*2*2(%%"REG_S"), %%mm2 \n\t"
978         "psubw %%mm6, %%mm1            \n\t" //'t12
979
980         "psubw "DCTSIZE_S"*5*2(%%"REG_S"), %%mm2 \n\t" //t5
981         "movq %%mm5, %%mm6             \n\t"
982
983         "movq %%mm7, 3*8+%3            \n\t"
984         "paddw %%mm2, %%mm3            \n\t" //t10
985
986         "paddw %%mm4, %%mm2            \n\t" //t11
987         "paddw %%mm0, %%mm4            \n\t" //t12
988
989         "movq %%mm3, %%mm7             \n\t"
990         "psubw %%mm4, %%mm3            \n\t"
991
992         "psllw $2, %%mm3              \n\t"
993         "psllw $2, %%mm7              \n\t" //opt for P6
994
995         "pmulhw "LOCAL_MANGLE(MM_FIX_0_382683433)", %%mm3 \n\t"
996         "psllw $2, %%mm4              \n\t"
997
998         "pmulhw "LOCAL_MANGLE(MM_FIX_0_541196100)", %%mm7 \n\t"
999         "psllw $2, %%mm2              \n\t"
1000
1001         "pmulhw "LOCAL_MANGLE(MM_FIX_1_306562965)", %%mm4 \n\t"
1002         "paddw %%mm1, %%mm5            \n\t" //'t1
1003
1004         "pmulhw "LOCAL_MANGLE(MM_FIX_0_707106781)", %%mm2 \n\t"
1005         "psubw %%mm1, %%mm6            \n\t" //'t2
1006         // t7 't12 't11 t4 t6 - 't13 't10   ---
1007
1008         "paddw %%mm3, %%mm7            \n\t" //z2
1009
1010         "movq %%mm5, 1*8+%3            \n\t"
1011         "paddw %%mm3, %%mm4            \n\t" //z4
1012
1013         "movq 3*16(%%"REG_d"), %%mm3       \n\t"
1014         "movq %%mm0, %%mm1             \n\t"
1015
1016         "movq %%mm6, 2*8+%3            \n\t"
1017         "psubw %%mm2, %%mm1            \n\t" //z13
1018
1019 //===
1020         "paddw %%mm2, %%mm0            \n\t" //z11
1021         "movq %%mm1, %%mm5             \n\t"
1022
1023         "movq 5*16(%%"REG_d"), %%mm2       \n\t"
1024         "psubw %%mm7, %%mm1            \n\t" //d3
1025
1026         "paddw %%mm7, %%mm5            \n\t" //d5
1027         "psubw %%mm3, %%mm1            \n\t"
1028
1029         "movq 1*16(%%"REG_d"), %%mm7       \n\t"
1030         "psubw %%mm2, %%mm5            \n\t"
1031
1032         "movq %%mm0, %%mm6             \n\t"
1033         "paddw %%mm4, %%mm0            \n\t" //d1
1034
1035         "paddusw %%mm3, %%mm1          \n\t"
1036         "psubw %%mm4, %%mm6            \n\t" //d7
1037
1038         // d1 d3 - - - d5 d7 -
1039         "movq 7*16(%%"REG_d"), %%mm4       \n\t"
1040         "psubw %%mm7, %%mm0            \n\t"
1041
1042         "psubw %%mm4, %%mm6            \n\t"
1043         "paddusw %%mm2, %%mm5          \n\t"
1044
1045         "paddusw %%mm4, %%mm6          \n\t"
1046         "paddw %%mm3, %%mm1            \n\t"
1047
1048         "paddw %%mm2, %%mm5            \n\t"
1049         "paddw %%mm4, %%mm6            \n\t"
1050
1051         "psubusw %%mm3, %%mm1          \n\t"
1052         "psubusw %%mm2, %%mm5          \n\t"
1053
1054         "psubusw %%mm4, %%mm6          \n\t"
1055         "movq %%mm1, %%mm4             \n\t"
1056
1057         "por %%mm5, %%mm4              \n\t"
1058         "paddusw %%mm7, %%mm0          \n\t"
1059
1060         "por %%mm6, %%mm4              \n\t"
1061         "paddw %%mm7, %%mm0            \n\t"
1062
1063         "packssdw %%mm4, %%mm4         \n\t"
1064         "psubusw %%mm7, %%mm0          \n\t"
1065
1066         "movd %%mm4, %%"REG_a"             \n\t"
1067         "or %%"REG_a", %%"REG_a"              \n\t"
1068         "jnz 2f                 \n\t"
1069         //movq [edi+"DCTSIZE_S"*3*2], mm1
1070         //movq [edi+"DCTSIZE_S"*5*2], mm5
1071         //movq [edi+"DCTSIZE_S"*1*2], mm0
1072         //movq [edi+"DCTSIZE_S"*7*2], mm6
1073         // t4 t5 - - - t6 t7 -
1074         //--- t4 (mm0) may be <>0; mm1, mm5, mm6 == 0
1075 //Typical numbers: nondc - 19%%,  dc - 26%%,  zero - 55%%. zero case alone isn't worthwhile
1076         "movq 0*8+%3, %%mm4            \n\t"
1077         "movq %%mm0, %%mm1             \n\t"
1078
1079         "pmulhw "LOCAL_MANGLE(MM_FIX_0_847759065)", %%mm0 \n\t" //tmp6
1080         "movq %%mm1, %%mm2             \n\t"
1081
1082         "movq "DCTSIZE_S"*0*2(%%"REG_D"), %%mm5 \n\t"
1083         "movq %%mm2, %%mm3             \n\t"
1084
1085         "pmulhw "LOCAL_MANGLE(MM_FIX_0_566454497)", %%mm1 \n\t" //tmp5
1086         "paddw %%mm4, %%mm5            \n\t"
1087
1088         "movq 1*8+%3, %%mm6            \n\t"
1089         //paddw mm3, MM_2
1090         "psraw $2, %%mm3              \n\t" //tmp7
1091
1092         "pmulhw "LOCAL_MANGLE(MM_FIX_0_198912367)", %%mm2 \n\t" //-tmp4
1093         "psubw %%mm3, %%mm4            \n\t"
1094
1095         "movq "DCTSIZE_S"*1*2(%%"REG_D"), %%mm7 \n\t"
1096         "paddw %%mm3, %%mm5            \n\t"
1097
1098         "movq %%mm4, "DCTSIZE_S"*7*2(%%"REG_D") \n\t"
1099         "paddw %%mm6, %%mm7            \n\t"
1100
1101         "movq 2*8+%3, %%mm3            \n\t"
1102         "psubw %%mm0, %%mm6            \n\t"
1103
1104         "movq "DCTSIZE_S"*2*2(%%"REG_D"), %%mm4 \n\t"
1105         "paddw %%mm0, %%mm7            \n\t"
1106
1107         "movq %%mm5, "DCTSIZE_S"*0*2(%%"REG_D") \n\t"
1108         "paddw %%mm3, %%mm4            \n\t"
1109
1110         "movq %%mm6, "DCTSIZE_S"*6*2(%%"REG_D") \n\t"
1111         "psubw %%mm1, %%mm3            \n\t"
1112
1113         "movq "DCTSIZE_S"*5*2(%%"REG_D"), %%mm5 \n\t"
1114         "paddw %%mm1, %%mm4            \n\t"
1115
1116         "movq "DCTSIZE_S"*3*2(%%"REG_D"), %%mm6 \n\t"
1117         "paddw %%mm3, %%mm5            \n\t"
1118
1119         "movq 3*8+%3, %%mm0            \n\t"
1120         "add $8, %%"REG_S"               \n\t"
1121
1122         "movq %%mm7, "DCTSIZE_S"*1*2(%%"REG_D") \n\t"
1123         "paddw %%mm0, %%mm6            \n\t"
1124
1125         "movq %%mm4, "DCTSIZE_S"*2*2(%%"REG_D") \n\t"
1126         "psubw %%mm2, %%mm0            \n\t"
1127
1128         "movq "DCTSIZE_S"*4*2(%%"REG_D"), %%mm7 \n\t"
1129         "paddw %%mm2, %%mm6            \n\t"
1130
1131         "movq %%mm5, "DCTSIZE_S"*5*2(%%"REG_D") \n\t"
1132         "paddw %%mm0, %%mm7            \n\t"
1133
1134         "movq %%mm6, "DCTSIZE_S"*3*2(%%"REG_D") \n\t"
1135
1136         "movq %%mm7, "DCTSIZE_S"*4*2(%%"REG_D") \n\t"
1137         "add $8, %%"REG_D"               \n\t"
1138         "jmp 4f                  \n\t"
1139
1140         "2:                    \n\t"
1141         //--- non DC2
1142         //psraw mm1, 2 w/o it -> offset. thr1, thr1, thr1  (actually thr1, thr1, thr1-1)
1143         //psraw mm5, 2
1144         //psraw mm0, 2
1145         //psraw mm6, 2
1146         "movq %%mm5, %%mm3             \n\t"
1147         "psubw %%mm1, %%mm5            \n\t"
1148
1149         "psllw $1, %%mm5              \n\t" //'z10
1150         "paddw %%mm1, %%mm3            \n\t" //'z13
1151
1152         "movq %%mm0, %%mm2             \n\t"
1153         "psubw %%mm6, %%mm0            \n\t"
1154
1155         "movq %%mm5, %%mm1             \n\t"
1156         "psllw $1, %%mm0              \n\t" //'z12
1157
1158         "pmulhw "LOCAL_MANGLE(MM_FIX_2_613125930)", %%mm1 \n\t" //-
1159         "paddw %%mm0, %%mm5            \n\t"
1160
1161         "pmulhw "LOCAL_MANGLE(MM_FIX_1_847759065)", %%mm5 \n\t" //'z5
1162         "paddw %%mm6, %%mm2            \n\t" //'z11
1163
1164         "pmulhw "LOCAL_MANGLE(MM_FIX_1_082392200)", %%mm0 \n\t"
1165         "movq %%mm2, %%mm7             \n\t"
1166
1167         //---
1168         "movq 0*8+%3, %%mm4            \n\t"
1169         "psubw %%mm3, %%mm2            \n\t"
1170
1171         "psllw $1, %%mm2              \n\t"
1172         "paddw %%mm3, %%mm7            \n\t" //'t7
1173
1174         "pmulhw "LOCAL_MANGLE(MM_FIX_1_414213562)", %%mm2 \n\t" //'t11
1175         "movq %%mm4, %%mm6             \n\t"
1176         //paddw mm7, MM_2
1177         "psraw $2, %%mm7              \n\t"
1178
1179         "paddw "DCTSIZE_S"*0*2(%%"REG_D"), %%mm4 \n\t"
1180         "psubw %%mm7, %%mm6            \n\t"
1181
1182         "movq 1*8+%3, %%mm3            \n\t"
1183         "paddw %%mm7, %%mm4            \n\t"
1184
1185         "movq %%mm6, "DCTSIZE_S"*7*2(%%"REG_D") \n\t"
1186         "paddw %%mm5, %%mm1            \n\t" //'t12
1187
1188         "movq %%mm4, "DCTSIZE_S"*0*2(%%"REG_D") \n\t"
1189         "psubw %%mm7, %%mm1            \n\t" //'t6
1190
1191         "movq 2*8+%3, %%mm7            \n\t"
1192         "psubw %%mm5, %%mm0            \n\t" //'t10
1193
1194         "movq 3*8+%3, %%mm6            \n\t"
1195         "movq %%mm3, %%mm5             \n\t"
1196
1197         "paddw "DCTSIZE_S"*1*2(%%"REG_D"), %%mm3 \n\t"
1198         "psubw %%mm1, %%mm5            \n\t"
1199
1200         "psubw %%mm1, %%mm2            \n\t" //'t5
1201         "paddw %%mm1, %%mm3            \n\t"
1202
1203         "movq %%mm5, "DCTSIZE_S"*6*2(%%"REG_D") \n\t"
1204         "movq %%mm7, %%mm4             \n\t"
1205
1206         "paddw "DCTSIZE_S"*2*2(%%"REG_D"), %%mm7 \n\t"
1207         "psubw %%mm2, %%mm4            \n\t"
1208
1209         "paddw "DCTSIZE_S"*5*2(%%"REG_D"), %%mm4 \n\t"
1210         "paddw %%mm2, %%mm7            \n\t"
1211
1212         "movq %%mm3, "DCTSIZE_S"*1*2(%%"REG_D") \n\t"
1213         "paddw %%mm2, %%mm0            \n\t" //'t4
1214
1215         // 't4 't6 't5 - - - - 't7
1216         "movq %%mm7, "DCTSIZE_S"*2*2(%%"REG_D") \n\t"
1217         "movq %%mm6, %%mm1             \n\t"
1218
1219         "paddw "DCTSIZE_S"*4*2(%%"REG_D"), %%mm6 \n\t"
1220         "psubw %%mm0, %%mm1            \n\t"
1221
1222         "paddw "DCTSIZE_S"*3*2(%%"REG_D"), %%mm1 \n\t"
1223         "paddw %%mm0, %%mm6            \n\t"
1224
1225         "movq %%mm4, "DCTSIZE_S"*5*2(%%"REG_D") \n\t"
1226         "add $8, %%"REG_S"               \n\t"
1227
1228         "movq %%mm6, "DCTSIZE_S"*4*2(%%"REG_D") \n\t"
1229
1230         "movq %%mm1, "DCTSIZE_S"*3*2(%%"REG_D") \n\t"
1231         "add $8, %%"REG_D"               \n\t"
1232
1233         "4:                     \n\t"
1234 //=part 2 (the same)===========================================================
1235         "movq "DCTSIZE_S"*0*2(%%"REG_S"), %%mm1 \n\t"
1236         //
1237         "movq "DCTSIZE_S"*3*2(%%"REG_S"), %%mm7 \n\t"
1238         "movq %%mm1, %%mm0             \n\t"
1239
1240         "paddw "DCTSIZE_S"*7*2(%%"REG_S"), %%mm1 \n\t" //t0
1241         "movq %%mm7, %%mm3             \n\t"
1242
1243         "paddw "DCTSIZE_S"*4*2(%%"REG_S"), %%mm7 \n\t" //t3
1244         "movq %%mm1, %%mm5             \n\t"
1245
1246         "movq "DCTSIZE_S"*1*2(%%"REG_S"), %%mm6 \n\t"
1247         "psubw %%mm7, %%mm1            \n\t" //t13
1248
1249         "movq "DCTSIZE_S"*2*2(%%"REG_S"), %%mm2 \n\t"
1250         "movq %%mm6, %%mm4             \n\t"
1251
1252         "paddw "DCTSIZE_S"*6*2(%%"REG_S"), %%mm6 \n\t" //t1
1253         "paddw %%mm7, %%mm5            \n\t" //t10
1254
1255         "paddw "DCTSIZE_S"*5*2(%%"REG_S"), %%mm2 \n\t" //t2
1256         "movq %%mm6, %%mm7             \n\t"
1257
1258         "paddw %%mm2, %%mm6            \n\t" //t11
1259         "psubw %%mm2, %%mm7            \n\t" //t12
1260
1261         "movq %%mm5, %%mm2             \n\t"
1262         "paddw %%mm6, %%mm5            \n\t" //d0
1263         // i0 t13 t12 i3 i1 d0 - d4
1264         "psubw %%mm6, %%mm2            \n\t" //d4
1265         "paddw %%mm1, %%mm7            \n\t"
1266
1267         "movq  1*8+4*16(%%"REG_d"), %%mm6  \n\t"
1268         "psllw $2, %%mm7              \n\t"
1269
1270         "psubw 1*8+0*16(%%"REG_d"), %%mm5  \n\t"
1271         "psubw %%mm6, %%mm2            \n\t"
1272
1273         "paddusw 1*8+0*16(%%"REG_d"), %%mm5 \n\t"
1274         "paddusw %%mm6, %%mm2          \n\t"
1275
1276         "pmulhw "LOCAL_MANGLE(MM_FIX_0_707106781)", %%mm7 \n\t"
1277         //
1278         "paddw 1*8+0*16(%%"REG_d"), %%mm5  \n\t"
1279         "paddw %%mm6, %%mm2            \n\t"
1280
1281         "psubusw 1*8+0*16(%%"REG_d"), %%mm5 \n\t"
1282         "psubusw %%mm6, %%mm2          \n\t"
1283
1284 //This func is totally compute-bound,  operates at huge speed. So,  DC shortcut
1285 // at this place isn't worthwhile due to BTB miss penalty (checked on Pent. 3).
1286 //However,  typical numbers: nondc - 29%%,  dc - 46%%,  zero - 25%%. All <> 0 case is very rare.
1287         "paddw "LOCAL_MANGLE(MM_2)", %%mm5            \n\t"
1288         "movq %%mm2, %%mm6             \n\t"
1289
1290         "paddw %%mm5, %%mm2            \n\t"
1291         "psubw %%mm6, %%mm5            \n\t"
1292
1293         "movq %%mm1, %%mm6             \n\t"
1294         "paddw %%mm7, %%mm1            \n\t" //d2
1295
1296         "psubw 1*8+2*16(%%"REG_d"), %%mm1  \n\t"
1297         "psubw %%mm7, %%mm6            \n\t" //d6
1298
1299         "movq 1*8+6*16(%%"REG_d"), %%mm7   \n\t"
1300         "psraw $2, %%mm5              \n\t"
1301
1302         "paddusw 1*8+2*16(%%"REG_d"), %%mm1 \n\t"
1303         "psubw %%mm7, %%mm6            \n\t"
1304         // t7 d2 /t11 t4 t6 - d6 /t10
1305
1306         "paddw 1*8+2*16(%%"REG_d"), %%mm1  \n\t"
1307         "paddusw %%mm7, %%mm6          \n\t"
1308
1309         "psubusw 1*8+2*16(%%"REG_d"), %%mm1 \n\t"
1310         "paddw %%mm7, %%mm6            \n\t"
1311
1312         "psubw "DCTSIZE_S"*4*2(%%"REG_S"), %%mm3 \n\t"
1313         "psubusw %%mm7, %%mm6          \n\t"
1314
1315         //movq [edi+"DCTSIZE_S"*2*2], mm1
1316         //movq [edi+"DCTSIZE_S"*6*2], mm6
1317         "movq %%mm1, %%mm7             \n\t"
1318         "psraw $2, %%mm2              \n\t"
1319
1320         "psubw "DCTSIZE_S"*6*2(%%"REG_S"), %%mm4 \n\t"
1321         "psubw %%mm6, %%mm1            \n\t"
1322
1323         "psubw "DCTSIZE_S"*7*2(%%"REG_S"), %%mm0 \n\t"
1324         "paddw %%mm7, %%mm6            \n\t" //'t13
1325
1326         "psraw $2, %%mm6              \n\t" //paddw mm6, MM_2 !!    ---
1327         "movq %%mm2, %%mm7             \n\t"
1328
1329         "pmulhw "LOCAL_MANGLE(MM_FIX_1_414213562_A)", %%mm1 \n\t"
1330         "paddw %%mm6, %%mm2            \n\t" //'t0
1331
1332         "movq %%mm2, 0*8+%3            \n\t" //!
1333         "psubw %%mm6, %%mm7            \n\t" //'t3
1334
1335         "movq "DCTSIZE_S"*2*2(%%"REG_S"), %%mm2 \n\t"
1336         "psubw %%mm6, %%mm1            \n\t" //'t12
1337
1338         "psubw "DCTSIZE_S"*5*2(%%"REG_S"), %%mm2 \n\t" //t5
1339         "movq %%mm5, %%mm6             \n\t"
1340
1341         "movq %%mm7, 3*8+%3            \n\t"
1342         "paddw %%mm2, %%mm3            \n\t" //t10
1343
1344         "paddw %%mm4, %%mm2            \n\t" //t11
1345         "paddw %%mm0, %%mm4            \n\t" //t12
1346
1347         "movq %%mm3, %%mm7             \n\t"
1348         "psubw %%mm4, %%mm3            \n\t"
1349
1350         "psllw $2, %%mm3              \n\t"
1351         "psllw $2, %%mm7              \n\t" //opt for P6
1352
1353         "pmulhw "LOCAL_MANGLE(MM_FIX_0_382683433)", %%mm3 \n\t"
1354         "psllw $2, %%mm4              \n\t"
1355
1356         "pmulhw "LOCAL_MANGLE(MM_FIX_0_541196100)", %%mm7 \n\t"
1357         "psllw $2, %%mm2              \n\t"
1358
1359         "pmulhw "LOCAL_MANGLE(MM_FIX_1_306562965)", %%mm4 \n\t"
1360         "paddw %%mm1, %%mm5            \n\t" //'t1
1361
1362         "pmulhw "LOCAL_MANGLE(MM_FIX_0_707106781)", %%mm2 \n\t"
1363         "psubw %%mm1, %%mm6            \n\t" //'t2
1364         // t7 't12 't11 t4 t6 - 't13 't10   ---
1365
1366         "paddw %%mm3, %%mm7            \n\t" //z2
1367
1368         "movq %%mm5, 1*8+%3            \n\t"
1369         "paddw %%mm3, %%mm4            \n\t" //z4
1370
1371         "movq 1*8+3*16(%%"REG_d"), %%mm3   \n\t"
1372         "movq %%mm0, %%mm1             \n\t"
1373
1374         "movq %%mm6, 2*8+%3            \n\t"
1375         "psubw %%mm2, %%mm1            \n\t" //z13
1376
1377 //===
1378         "paddw %%mm2, %%mm0            \n\t" //z11
1379         "movq %%mm1, %%mm5             \n\t"
1380
1381         "movq 1*8+5*16(%%"REG_d"), %%mm2   \n\t"
1382         "psubw %%mm7, %%mm1            \n\t" //d3
1383
1384         "paddw %%mm7, %%mm5            \n\t" //d5
1385         "psubw %%mm3, %%mm1            \n\t"
1386
1387         "movq 1*8+1*16(%%"REG_d"), %%mm7   \n\t"
1388         "psubw %%mm2, %%mm5            \n\t"
1389
1390         "movq %%mm0, %%mm6             \n\t"
1391         "paddw %%mm4, %%mm0            \n\t" //d1
1392
1393         "paddusw %%mm3, %%mm1          \n\t"
1394         "psubw %%mm4, %%mm6            \n\t" //d7
1395
1396         // d1 d3 - - - d5 d7 -
1397         "movq 1*8+7*16(%%"REG_d"), %%mm4   \n\t"
1398         "psubw %%mm7, %%mm0            \n\t"
1399
1400         "psubw %%mm4, %%mm6            \n\t"
1401         "paddusw %%mm2, %%mm5          \n\t"
1402
1403         "paddusw %%mm4, %%mm6          \n\t"
1404         "paddw %%mm3, %%mm1            \n\t"
1405
1406         "paddw %%mm2, %%mm5            \n\t"
1407         "paddw %%mm4, %%mm6            \n\t"
1408
1409         "psubusw %%mm3, %%mm1          \n\t"
1410         "psubusw %%mm2, %%mm5          \n\t"
1411
1412         "psubusw %%mm4, %%mm6          \n\t"
1413         "movq %%mm1, %%mm4             \n\t"
1414
1415         "por %%mm5, %%mm4              \n\t"
1416         "paddusw %%mm7, %%mm0          \n\t"
1417
1418         "por %%mm6, %%mm4              \n\t"
1419         "paddw %%mm7, %%mm0            \n\t"
1420
1421         "packssdw %%mm4, %%mm4         \n\t"
1422         "psubusw %%mm7, %%mm0          \n\t"
1423
1424         "movd %%mm4, %%"REG_a"             \n\t"
1425         "or %%"REG_a", %%"REG_a"              \n\t"
1426         "jnz 3f                 \n\t"
1427         //movq [edi+"DCTSIZE_S"*3*2], mm1
1428         //movq [edi+"DCTSIZE_S"*5*2], mm5
1429         //movq [edi+"DCTSIZE_S"*1*2], mm0
1430         //movq [edi+"DCTSIZE_S"*7*2], mm6
1431         // t4 t5 - - - t6 t7 -
1432         //--- t4 (mm0) may be <>0; mm1, mm5, mm6 == 0
1433 //Typical numbers: nondc - 19%%,  dc - 26%%,  zero - 55%%. zero case alone isn't worthwhile
1434         "movq 0*8+%3, %%mm4            \n\t"
1435         "movq %%mm0, %%mm1             \n\t"
1436
1437         "pmulhw "LOCAL_MANGLE(MM_FIX_0_847759065)", %%mm0 \n\t" //tmp6
1438         "movq %%mm1, %%mm2             \n\t"
1439
1440         "movq "DCTSIZE_S"*0*2(%%"REG_D"), %%mm5 \n\t"
1441         "movq %%mm2, %%mm3             \n\t"
1442
1443         "pmulhw "LOCAL_MANGLE(MM_FIX_0_566454497)", %%mm1 \n\t" //tmp5
1444         "paddw %%mm4, %%mm5            \n\t"
1445
1446         "movq 1*8+%3, %%mm6            \n\t"
1447         //paddw mm3, MM_2
1448         "psraw $2, %%mm3              \n\t" //tmp7
1449
1450         "pmulhw "LOCAL_MANGLE(MM_FIX_0_198912367)", %%mm2 \n\t" //-tmp4
1451         "psubw %%mm3, %%mm4            \n\t"
1452
1453         "movq "DCTSIZE_S"*1*2(%%"REG_D"), %%mm7 \n\t"
1454         "paddw %%mm3, %%mm5            \n\t"
1455
1456         "movq %%mm4, "DCTSIZE_S"*7*2(%%"REG_D") \n\t"
1457         "paddw %%mm6, %%mm7            \n\t"
1458
1459         "movq 2*8+%3, %%mm3            \n\t"
1460         "psubw %%mm0, %%mm6            \n\t"
1461
1462         "movq "DCTSIZE_S"*2*2(%%"REG_D"), %%mm4 \n\t"
1463         "paddw %%mm0, %%mm7            \n\t"
1464
1465         "movq %%mm5, "DCTSIZE_S"*0*2(%%"REG_D") \n\t"
1466         "paddw %%mm3, %%mm4            \n\t"
1467
1468         "movq %%mm6, "DCTSIZE_S"*6*2(%%"REG_D") \n\t"
1469         "psubw %%mm1, %%mm3            \n\t"
1470
1471         "movq "DCTSIZE_S"*5*2(%%"REG_D"), %%mm5 \n\t"
1472         "paddw %%mm1, %%mm4            \n\t"
1473
1474         "movq "DCTSIZE_S"*3*2(%%"REG_D"), %%mm6 \n\t"
1475         "paddw %%mm3, %%mm5            \n\t"
1476
1477         "movq 3*8+%3, %%mm0            \n\t"
1478         "add $24, %%"REG_S"              \n\t"
1479
1480         "movq %%mm7, "DCTSIZE_S"*1*2(%%"REG_D") \n\t"
1481         "paddw %%mm0, %%mm6            \n\t"
1482
1483         "movq %%mm4, "DCTSIZE_S"*2*2(%%"REG_D") \n\t"
1484         "psubw %%mm2, %%mm0            \n\t"
1485
1486         "movq "DCTSIZE_S"*4*2(%%"REG_D"), %%mm7 \n\t"
1487         "paddw %%mm2, %%mm6            \n\t"
1488
1489         "movq %%mm5, "DCTSIZE_S"*5*2(%%"REG_D") \n\t"
1490         "paddw %%mm0, %%mm7            \n\t"
1491
1492         "movq %%mm6, "DCTSIZE_S"*3*2(%%"REG_D") \n\t"
1493
1494         "movq %%mm7, "DCTSIZE_S"*4*2(%%"REG_D") \n\t"
1495         "add $24, %%"REG_D"              \n\t"
1496         "sub $2, %%"REG_c"               \n\t"
1497         "jnz 1b                \n\t"
1498         "jmp 5f                   \n\t"
1499
1500         "3:                    \n\t"
1501         //--- non DC2
1502         //psraw mm1, 2 w/o it -> offset. thr1, thr1, thr1  (actually thr1, thr1, thr1-1)
1503         //psraw mm5, 2
1504         //psraw mm0, 2
1505         //psraw mm6, 2
1506         "movq %%mm5, %%mm3             \n\t"
1507         "psubw %%mm1, %%mm5            \n\t"
1508
1509         "psllw $1, %%mm5              \n\t" //'z10
1510         "paddw %%mm1, %%mm3            \n\t" //'z13
1511
1512         "movq %%mm0, %%mm2             \n\t"
1513         "psubw %%mm6, %%mm0            \n\t"
1514
1515         "movq %%mm5, %%mm1             \n\t"
1516         "psllw $1, %%mm0              \n\t" //'z12
1517
1518         "pmulhw "LOCAL_MANGLE(MM_FIX_2_613125930)", %%mm1 \n\t" //-
1519         "paddw %%mm0, %%mm5            \n\t"
1520
1521         "pmulhw "LOCAL_MANGLE(MM_FIX_1_847759065)", %%mm5 \n\t" //'z5
1522         "paddw %%mm6, %%mm2            \n\t" //'z11
1523
1524         "pmulhw "LOCAL_MANGLE(MM_FIX_1_082392200)", %%mm0 \n\t"
1525         "movq %%mm2, %%mm7             \n\t"
1526
1527         //---
1528         "movq 0*8+%3, %%mm4            \n\t"
1529         "psubw %%mm3, %%mm2            \n\t"
1530
1531         "psllw $1, %%mm2              \n\t"
1532         "paddw %%mm3, %%mm7            \n\t" //'t7
1533
1534         "pmulhw "LOCAL_MANGLE(MM_FIX_1_414213562)", %%mm2 \n\t" //'t11
1535         "movq %%mm4, %%mm6             \n\t"
1536         //paddw mm7, MM_2
1537         "psraw $2, %%mm7              \n\t"
1538
1539         "paddw "DCTSIZE_S"*0*2(%%"REG_D"), %%mm4 \n\t"
1540         "psubw %%mm7, %%mm6            \n\t"
1541
1542         "movq 1*8+%3, %%mm3            \n\t"
1543         "paddw %%mm7, %%mm4            \n\t"
1544
1545         "movq %%mm6, "DCTSIZE_S"*7*2(%%"REG_D") \n\t"
1546         "paddw %%mm5, %%mm1            \n\t" //'t12
1547
1548         "movq %%mm4, "DCTSIZE_S"*0*2(%%"REG_D") \n\t"
1549         "psubw %%mm7, %%mm1            \n\t" //'t6
1550
1551         "movq 2*8+%3, %%mm7            \n\t"
1552         "psubw %%mm5, %%mm0            \n\t" //'t10
1553
1554         "movq 3*8+%3, %%mm6            \n\t"
1555         "movq %%mm3, %%mm5             \n\t"
1556
1557         "paddw "DCTSIZE_S"*1*2(%%"REG_D"), %%mm3 \n\t"
1558         "psubw %%mm1, %%mm5            \n\t"
1559
1560         "psubw %%mm1, %%mm2            \n\t" //'t5
1561         "paddw %%mm1, %%mm3            \n\t"
1562
1563         "movq %%mm5, "DCTSIZE_S"*6*2(%%"REG_D") \n\t"
1564         "movq %%mm7, %%mm4             \n\t"
1565
1566         "paddw "DCTSIZE_S"*2*2(%%"REG_D"), %%mm7 \n\t"
1567         "psubw %%mm2, %%mm4            \n\t"
1568
1569         "paddw "DCTSIZE_S"*5*2(%%"REG_D"), %%mm4 \n\t"
1570         "paddw %%mm2, %%mm7            \n\t"
1571
1572         "movq %%mm3, "DCTSIZE_S"*1*2(%%"REG_D") \n\t"
1573         "paddw %%mm2, %%mm0            \n\t" //'t4
1574
1575         // 't4 't6 't5 - - - - 't7
1576         "movq %%mm7, "DCTSIZE_S"*2*2(%%"REG_D") \n\t"
1577         "movq %%mm6, %%mm1             \n\t"
1578
1579         "paddw "DCTSIZE_S"*4*2(%%"REG_D"), %%mm6 \n\t"
1580         "psubw %%mm0, %%mm1            \n\t"
1581
1582         "paddw "DCTSIZE_S"*3*2(%%"REG_D"), %%mm1 \n\t"
1583         "paddw %%mm0, %%mm6            \n\t"
1584
1585         "movq %%mm4, "DCTSIZE_S"*5*2(%%"REG_D") \n\t"
1586         "add $24, %%"REG_S"              \n\t"
1587
1588         "movq %%mm6, "DCTSIZE_S"*4*2(%%"REG_D") \n\t"
1589
1590         "movq %%mm1, "DCTSIZE_S"*3*2(%%"REG_D") \n\t"
1591         "add $24, %%"REG_D"              \n\t"
1592         "sub $2, %%"REG_c"               \n\t"
1593         "jnz 1b                \n\t"
1594         "5:                      \n\t"
1595
1596         : "+S"(data), "+D"(output), "+c"(cnt), "=o"(temps)
1597         : "d"(thr_adr)
1598         : "%"REG_a
1599         );
1600 }
1601
1602 #endif // HAVE_MMX
1603
1604 #if !HAVE_MMX
1605
1606 static void row_idct_c(DCTELEM* workspace,
1607                        int16_t* output_adr, int output_stride, int cnt)
1608 {
1609     int_simd16_t tmp0, tmp1, tmp2, tmp3, tmp4, tmp5, tmp6, tmp7;
1610     int_simd16_t tmp10, tmp11, tmp12, tmp13;
1611     int_simd16_t z5, z10, z11, z12, z13;
1612     int16_t* outptr;
1613     DCTELEM* wsptr;
1614
1615     cnt*=4;
1616     wsptr = workspace;
1617     outptr = output_adr;
1618     for (; cnt > 0; cnt--) {
1619         // Even part
1620         //Simd version reads 4x4 block and transposes it
1621         tmp10 = ( wsptr[2] +  wsptr[3]);
1622         tmp11 = ( wsptr[2] -  wsptr[3]);
1623
1624         tmp13 = ( wsptr[0] +  wsptr[1]);
1625         tmp12 = (MULTIPLY16H( wsptr[0] - wsptr[1], FIX_1_414213562_A)<<2) - tmp13;//this shift order to avoid overflow
1626
1627         tmp0 = tmp10 + tmp13; //->temps
1628         tmp3 = tmp10 - tmp13; //->temps
1629         tmp1 = tmp11 + tmp12;
1630         tmp2 = tmp11 - tmp12;
1631
1632         // Odd part
1633         //Also transpose, with previous:
1634         // ---- ----      ||||
1635         // ---- ---- idct ||||
1636         // ---- ---- ---> ||||
1637         // ---- ----      ||||
1638         z13 = wsptr[4] + wsptr[5];
1639         z10 = wsptr[4] - wsptr[5];
1640         z11 = wsptr[6] + wsptr[7];
1641         z12 = wsptr[6] - wsptr[7];
1642
1643         tmp7 = z11 + z13;
1644         tmp11 = MULTIPLY16H(z11 - z13, FIX_1_414213562);
1645
1646         z5 =    MULTIPLY16H(z10 + z12, FIX_1_847759065);
1647         tmp10 = MULTIPLY16H(z12, FIX_1_082392200) - z5;
1648         tmp12 = MULTIPLY16H(z10, FIX_2_613125930) + z5; // - FIX_
1649
1650         tmp6 = (tmp12<<3) - tmp7;
1651         tmp5 = (tmp11<<3) - tmp6;
1652         tmp4 = (tmp10<<3) + tmp5;
1653
1654         // Final output stage: descale and write column
1655         outptr[0*output_stride]+= DESCALE(tmp0 + tmp7, 3);
1656         outptr[1*output_stride]+= DESCALE(tmp1 + tmp6, 3);
1657         outptr[2*output_stride]+= DESCALE(tmp2 + tmp5, 3);
1658         outptr[3*output_stride]+= DESCALE(tmp3 - tmp4, 3);
1659         outptr[4*output_stride]+= DESCALE(tmp3 + tmp4, 3);
1660         outptr[5*output_stride]+= DESCALE(tmp2 - tmp5, 3);
1661         outptr[6*output_stride]+= DESCALE(tmp1 - tmp6, 3); //no += ?
1662         outptr[7*output_stride]+= DESCALE(tmp0 - tmp7, 3); //no += ?
1663         outptr++;
1664
1665         wsptr += DCTSIZE;       // advance pointer to next row
1666     }
1667 }
1668
1669 #else /* HAVE_MMX */
1670
1671 static void row_idct_mmx (DCTELEM* workspace,
1672                           int16_t* output_adr,  int output_stride,  int cnt)
1673 {
1674     uint64_t __attribute__((aligned(8))) temps[4];
1675     __asm__ volatile(
1676         "lea (%%"REG_a",%%"REG_a",2), %%"REG_d"    \n\t"
1677         "1:                     \n\t"
1678         "movq "DCTSIZE_S"*0*2(%%"REG_S"), %%mm0 \n\t"
1679         //
1680
1681         "movq "DCTSIZE_S"*1*2(%%"REG_S"), %%mm1 \n\t"
1682         "movq %%mm0, %%mm4             \n\t"
1683
1684         "movq "DCTSIZE_S"*2*2(%%"REG_S"), %%mm2 \n\t"
1685         "punpcklwd %%mm1, %%mm0        \n\t"
1686
1687         "movq "DCTSIZE_S"*3*2(%%"REG_S"), %%mm3 \n\t"
1688         "punpckhwd %%mm1, %%mm4        \n\t"
1689
1690         //transpose 4x4
1691         "movq %%mm2, %%mm7             \n\t"
1692         "punpcklwd %%mm3, %%mm2        \n\t"
1693
1694         "movq %%mm0, %%mm6             \n\t"
1695         "punpckldq %%mm2, %%mm0        \n\t" //0
1696
1697         "punpckhdq %%mm2, %%mm6        \n\t" //1
1698         "movq %%mm0, %%mm5             \n\t"
1699
1700         "punpckhwd %%mm3, %%mm7        \n\t"
1701         "psubw %%mm6, %%mm0            \n\t"
1702
1703         "pmulhw "LOCAL_MANGLE(MM_FIX_1_414213562_A)", %%mm0 \n\t"
1704         "movq %%mm4, %%mm2             \n\t"
1705
1706         "punpckldq %%mm7, %%mm4        \n\t" //2
1707         "paddw %%mm6, %%mm5            \n\t"
1708
1709         "punpckhdq %%mm7, %%mm2        \n\t" //3
1710         "movq %%mm4, %%mm1             \n\t"
1711
1712         "psllw $2, %%mm0              \n\t"
1713         "paddw %%mm2, %%mm4            \n\t" //t10
1714
1715         "movq "DCTSIZE_S"*0*2+"DCTSIZE_S"(%%"REG_S"), %%mm3 \n\t"
1716         "psubw %%mm2, %%mm1            \n\t" //t11
1717
1718         "movq "DCTSIZE_S"*1*2+"DCTSIZE_S"(%%"REG_S"), %%mm2 \n\t"
1719         "psubw %%mm5, %%mm0            \n\t"
1720
1721         "movq %%mm4, %%mm6             \n\t"
1722         "paddw %%mm5, %%mm4            \n\t" //t0
1723
1724         "psubw %%mm5, %%mm6            \n\t" //t3
1725         "movq %%mm1, %%mm7             \n\t"
1726
1727         "movq "DCTSIZE_S"*2*2+"DCTSIZE_S"(%%"REG_S"), %%mm5 \n\t"
1728         "paddw %%mm0, %%mm1            \n\t" //t1
1729
1730         "movq %%mm4, 0*8+%3            \n\t" //t0
1731         "movq %%mm3, %%mm4             \n\t"
1732
1733         "movq %%mm6, 1*8+%3            \n\t" //t3
1734         "punpcklwd %%mm2, %%mm3        \n\t"
1735
1736         //transpose 4x4
1737         "movq "DCTSIZE_S"*3*2+"DCTSIZE_S"(%%"REG_S"), %%mm6 \n\t"
1738         "punpckhwd %%mm2, %%mm4        \n\t"
1739
1740         "movq %%mm5, %%mm2             \n\t"
1741         "punpcklwd %%mm6, %%mm5        \n\t"
1742
1743         "psubw %%mm0, %%mm7            \n\t" //t2
1744         "punpckhwd %%mm6, %%mm2        \n\t"
1745
1746         "movq %%mm3, %%mm0             \n\t"
1747         "punpckldq %%mm5, %%mm3        \n\t" //4
1748
1749         "punpckhdq %%mm5, %%mm0        \n\t" //5
1750         "movq %%mm4, %%mm5             \n\t"
1751
1752         //
1753         "movq %%mm3, %%mm6             \n\t"
1754         "punpckldq %%mm2, %%mm4        \n\t" //6
1755
1756         "psubw %%mm0, %%mm3            \n\t" //z10
1757         "punpckhdq %%mm2, %%mm5        \n\t" //7
1758
1759         "paddw %%mm0, %%mm6            \n\t" //z13
1760         "movq %%mm4, %%mm2             \n\t"
1761
1762         "movq %%mm3, %%mm0             \n\t"
1763         "psubw %%mm5, %%mm4            \n\t" //z12
1764
1765         "pmulhw "LOCAL_MANGLE(MM_FIX_2_613125930)", %%mm0 \n\t" //-
1766         "paddw %%mm4, %%mm3            \n\t"
1767
1768         "pmulhw "LOCAL_MANGLE(MM_FIX_1_847759065)", %%mm3 \n\t" //z5
1769         "paddw %%mm5, %%mm2            \n\t" //z11  >
1770
1771         "pmulhw "LOCAL_MANGLE(MM_FIX_1_082392200)", %%mm4 \n\t"
1772         "movq %%mm2, %%mm5             \n\t"
1773
1774         "psubw %%mm6, %%mm2            \n\t"
1775         "paddw %%mm6, %%mm5            \n\t" //t7
1776
1777         "pmulhw "LOCAL_MANGLE(MM_FIX_1_414213562)", %%mm2 \n\t" //t11
1778         "paddw %%mm3, %%mm0            \n\t" //t12
1779
1780         "psllw $3, %%mm0              \n\t"
1781         "psubw %%mm3, %%mm4            \n\t" //t10
1782
1783         "movq 0*8+%3, %%mm6            \n\t"
1784         "movq %%mm1, %%mm3             \n\t"
1785
1786         "psllw $3, %%mm4              \n\t"
1787         "psubw %%mm5, %%mm0            \n\t" //t6
1788
1789         "psllw $3, %%mm2              \n\t"
1790         "paddw %%mm0, %%mm1            \n\t" //d1
1791
1792         "psubw %%mm0, %%mm2            \n\t" //t5
1793         "psubw %%mm0, %%mm3            \n\t" //d6
1794
1795         "paddw %%mm2, %%mm4            \n\t" //t4
1796         "movq %%mm7, %%mm0             \n\t"
1797
1798         "paddw %%mm2, %%mm7            \n\t" //d2
1799         "psubw %%mm2, %%mm0            \n\t" //d5
1800
1801         "movq "LOCAL_MANGLE(MM_DESCALE_RND)", %%mm2   \n\t" //4
1802         "psubw %%mm5, %%mm6            \n\t" //d7
1803
1804         "paddw 0*8+%3, %%mm5           \n\t" //d0
1805         "paddw %%mm2, %%mm1            \n\t"
1806
1807         "paddw %%mm2, %%mm5            \n\t"
1808         "psraw $3, %%mm1              \n\t"
1809
1810         "paddw %%mm2, %%mm7            \n\t"
1811         "psraw $3, %%mm5              \n\t"
1812
1813         "paddw (%%"REG_D"), %%mm5          \n\t"
1814         "psraw $3, %%mm7              \n\t"
1815
1816         "paddw (%%"REG_D",%%"REG_a",), %%mm1    \n\t"
1817         "paddw %%mm2, %%mm0            \n\t"
1818
1819         "paddw (%%"REG_D",%%"REG_a",2), %%mm7   \n\t"
1820         "paddw %%mm2, %%mm3            \n\t"
1821
1822         "movq %%mm5, (%%"REG_D")           \n\t"
1823         "paddw %%mm2, %%mm6            \n\t"
1824
1825         "movq %%mm1, (%%"REG_D",%%"REG_a",)     \n\t"
1826         "psraw $3, %%mm0              \n\t"
1827
1828         "movq %%mm7, (%%"REG_D",%%"REG_a",2)    \n\t"
1829         "add %%"REG_d", %%"REG_D"             \n\t" //3*ls
1830
1831         "movq 1*8+%3, %%mm5           \n\t" //t3
1832         "psraw $3, %%mm3              \n\t"
1833
1834         "paddw (%%"REG_D",%%"REG_a",2), %%mm0   \n\t"
1835         "psubw %%mm4, %%mm5            \n\t" //d3
1836
1837         "paddw (%%"REG_D",%%"REG_d",), %%mm3    \n\t"
1838         "psraw $3, %%mm6              \n\t"
1839
1840         "paddw 1*8+%3, %%mm4           \n\t" //d4
1841         "paddw %%mm2, %%mm5            \n\t"
1842
1843         "paddw (%%"REG_D",%%"REG_a",4), %%mm6   \n\t"
1844         "paddw %%mm2, %%mm4            \n\t"
1845
1846         "movq %%mm0, (%%"REG_D",%%"REG_a",2)    \n\t"
1847         "psraw $3, %%mm5              \n\t"
1848
1849         "paddw (%%"REG_D"), %%mm5          \n\t"
1850         "psraw $3, %%mm4              \n\t"
1851
1852         "paddw (%%"REG_D",%%"REG_a",), %%mm4    \n\t"
1853         "add $"DCTSIZE_S"*2*4, %%"REG_S"      \n\t" //4 rows
1854
1855         "movq %%mm3, (%%"REG_D",%%"REG_d",)     \n\t"
1856         "movq %%mm6, (%%"REG_D",%%"REG_a",4)    \n\t"
1857         "movq %%mm5, (%%"REG_D")           \n\t"
1858         "movq %%mm4, (%%"REG_D",%%"REG_a",)     \n\t"
1859
1860         "sub %%"REG_d", %%"REG_D"             \n\t"
1861         "add $8, %%"REG_D"               \n\t"
1862         "dec %%"REG_c"                   \n\t"
1863         "jnz 1b                  \n\t"
1864
1865         : "+S"(workspace), "+D"(output_adr), "+c"(cnt), "=o"(temps)
1866         : "a"(output_stride*sizeof(short))
1867         : "%"REG_d
1868         );
1869 }
1870
1871 #endif // HAVE_MMX
1872
1873 #if !HAVE_MMX
1874
1875 static void row_fdct_c(DCTELEM *data, const uint8_t *pixels, int line_size, int cnt)
1876 {
1877     int_simd16_t tmp0, tmp1, tmp2, tmp3, tmp4, tmp5, tmp6, tmp7;
1878     int_simd16_t tmp10, tmp11, tmp12, tmp13;
1879     int_simd16_t z1, z2, z3, z4, z5, z11, z13;
1880     DCTELEM *dataptr;
1881
1882     cnt*=4;
1883     // Pass 1: process rows.
1884
1885     dataptr = data;
1886     for (; cnt > 0; cnt--) {
1887         tmp0 = pixels[line_size*0] + pixels[line_size*7];
1888         tmp7 = pixels[line_size*0] - pixels[line_size*7];
1889         tmp1 = pixels[line_size*1] + pixels[line_size*6];
1890         tmp6 = pixels[line_size*1] - pixels[line_size*6];
1891         tmp2 = pixels[line_size*2] + pixels[line_size*5];
1892         tmp5 = pixels[line_size*2] - pixels[line_size*5];
1893         tmp3 = pixels[line_size*3] + pixels[line_size*4];
1894         tmp4 = pixels[line_size*3] - pixels[line_size*4];
1895
1896         // Even part
1897
1898         tmp10 = tmp0 + tmp3;
1899         tmp13 = tmp0 - tmp3;
1900         tmp11 = tmp1 + tmp2;
1901         tmp12 = tmp1 - tmp2;
1902         //Even columns are written first, this leads to different order of columns
1903         //in column_fidct(), but they are processed independently, so all ok.
1904         //Later in the row_idct() columns readed at the same order.
1905         dataptr[2] = tmp10 + tmp11;
1906         dataptr[3] = tmp10 - tmp11;
1907
1908         z1 = MULTIPLY16H((tmp12 + tmp13)<<2, FIX_0_707106781);
1909         dataptr[0] = tmp13 + z1;
1910         dataptr[1] = tmp13 - z1;
1911
1912         // Odd part
1913
1914         tmp10 = (tmp4 + tmp5) <<2;
1915         tmp11 = (tmp5 + tmp6) <<2;
1916         tmp12 = (tmp6 + tmp7) <<2;
1917
1918         z5 = MULTIPLY16H(tmp10 - tmp12, FIX_0_382683433);
1919         z2 = MULTIPLY16H(tmp10, FIX_0_541196100) + z5;
1920         z4 = MULTIPLY16H(tmp12, FIX_1_306562965) + z5;
1921         z3 = MULTIPLY16H(tmp11, FIX_0_707106781);
1922
1923         z11 = tmp7 + z3;
1924         z13 = tmp7 - z3;
1925
1926         dataptr[4] = z13 + z2;
1927         dataptr[5] = z13 - z2;
1928         dataptr[6] = z11 + z4;
1929         dataptr[7] = z11 - z4;
1930
1931         pixels++;               // advance pointer to next column
1932         dataptr += DCTSIZE;
1933     }
1934 }
1935
1936 #else /* HAVE_MMX */
1937
1938 static void row_fdct_mmx(DCTELEM *data,  const uint8_t *pixels,  int line_size,  int cnt)
1939 {
1940     uint64_t __attribute__((aligned(8))) temps[4];
1941     __asm__ volatile(
1942         "lea (%%"REG_a",%%"REG_a",2), %%"REG_d"    \n\t"
1943         "6:                     \n\t"
1944         "movd (%%"REG_S"), %%mm0           \n\t"
1945         "pxor %%mm7, %%mm7             \n\t"
1946
1947         "movd (%%"REG_S",%%"REG_a",), %%mm1     \n\t"
1948         "punpcklbw %%mm7, %%mm0        \n\t"
1949
1950         "movd (%%"REG_S",%%"REG_a",2), %%mm2    \n\t"
1951         "punpcklbw %%mm7, %%mm1        \n\t"
1952
1953         "punpcklbw %%mm7, %%mm2        \n\t"
1954         "add %%"REG_d", %%"REG_S"             \n\t"
1955
1956         "movq %%mm0, %%mm5             \n\t"
1957         //
1958
1959         "movd (%%"REG_S",%%"REG_a",4), %%mm3    \n\t" //7  ;prefetch!
1960         "movq %%mm1, %%mm6             \n\t"
1961
1962         "movd (%%"REG_S",%%"REG_d",), %%mm4     \n\t" //6
1963         "punpcklbw %%mm7, %%mm3        \n\t"
1964
1965         "psubw %%mm3, %%mm5            \n\t"
1966         "punpcklbw %%mm7, %%mm4        \n\t"
1967
1968         "paddw %%mm3, %%mm0            \n\t"
1969         "psubw %%mm4, %%mm6            \n\t"
1970
1971         "movd (%%"REG_S",%%"REG_a",2), %%mm3    \n\t" //5
1972         "paddw %%mm4, %%mm1            \n\t"
1973
1974         "movq %%mm5, 0*8+%3            \n\t" //t7
1975         "punpcklbw %%mm7, %%mm3        \n\t"
1976
1977         "movq %%mm6, 1*8+%3            \n\t" //t6
1978         "movq %%mm2, %%mm4             \n\t"
1979
1980         "movd (%%"REG_S"), %%mm5           \n\t" //3
1981         "paddw %%mm3, %%mm2            \n\t"
1982
1983         "movd (%%"REG_S",%%"REG_a",), %%mm6     \n\t" //4
1984         "punpcklbw %%mm7, %%mm5        \n\t"
1985
1986         "psubw %%mm3, %%mm4            \n\t"
1987         "punpcklbw %%mm7, %%mm6        \n\t"
1988
1989         "movq %%mm5, %%mm3             \n\t"
1990         "paddw %%mm6, %%mm5            \n\t" //t3
1991
1992         "psubw %%mm6, %%mm3            \n\t" //t4  ; t0 t1 t2 t4 t5 t3 - -
1993         "movq %%mm0, %%mm6             \n\t"
1994
1995         "movq %%mm1, %%mm7             \n\t"
1996         "psubw %%mm5, %%mm0            \n\t" //t13
1997
1998         "psubw %%mm2, %%mm1            \n\t"
1999         "paddw %%mm2, %%mm7            \n\t" //t11
2000
2001         "paddw %%mm0, %%mm1            \n\t"
2002         "movq %%mm7, %%mm2             \n\t"
2003
2004         "psllw $2, %%mm1              \n\t"
2005         "paddw %%mm5, %%mm6            \n\t" //t10
2006
2007         "pmulhw "LOCAL_MANGLE(MM_FIX_0_707106781)", %%mm1 \n\t"
2008         "paddw %%mm6, %%mm7            \n\t" //d2
2009
2010         "psubw %%mm2, %%mm6            \n\t" //d3
2011         "movq %%mm0, %%mm5             \n\t"
2012
2013         //transpose 4x4
2014         "movq %%mm7, %%mm2             \n\t"
2015         "punpcklwd %%mm6, %%mm7        \n\t"
2016
2017         "paddw %%mm1, %%mm0            \n\t" //d0
2018         "punpckhwd %%mm6, %%mm2        \n\t"
2019
2020         "psubw %%mm1, %%mm5            \n\t" //d1
2021         "movq %%mm0, %%mm6             \n\t"
2022
2023         "movq 1*8+%3, %%mm1            \n\t"
2024         "punpcklwd %%mm5, %%mm0        \n\t"
2025
2026         "punpckhwd %%mm5, %%mm6        \n\t"
2027         "movq %%mm0, %%mm5             \n\t"
2028
2029         "punpckldq %%mm7, %%mm0        \n\t" //0
2030         "paddw %%mm4, %%mm3            \n\t"
2031
2032         "punpckhdq %%mm7, %%mm5        \n\t" //1
2033         "movq %%mm6, %%mm7             \n\t"
2034
2035         "movq %%mm0, "DCTSIZE_S"*0*2(%%"REG_D") \n\t"
2036         "punpckldq %%mm2, %%mm6        \n\t" //2
2037
2038         "movq %%mm5, "DCTSIZE_S"*1*2(%%"REG_D") \n\t"
2039         "punpckhdq %%mm2, %%mm7        \n\t" //3
2040
2041         "movq %%mm6, "DCTSIZE_S"*2*2(%%"REG_D") \n\t"
2042         "paddw %%mm1, %%mm4            \n\t"
2043
2044         "movq %%mm7, "DCTSIZE_S"*3*2(%%"REG_D") \n\t"
2045         "psllw $2, %%mm3              \n\t" //t10
2046
2047         "movq 0*8+%3, %%mm2           \n\t"
2048         "psllw $2, %%mm4              \n\t" //t11
2049
2050         "pmulhw "LOCAL_MANGLE(MM_FIX_0_707106781)", %%mm4 \n\t" //z3
2051         "paddw %%mm2, %%mm1            \n\t"
2052
2053         "psllw $2, %%mm1              \n\t" //t12
2054         "movq %%mm3, %%mm0             \n\t"
2055
2056         "pmulhw "LOCAL_MANGLE(MM_FIX_0_541196100)", %%mm0 \n\t"
2057         "psubw %%mm1, %%mm3            \n\t"
2058
2059         "pmulhw "LOCAL_MANGLE(MM_FIX_0_382683433)", %%mm3 \n\t" //z5
2060         "movq %%mm2, %%mm5             \n\t"
2061
2062         "pmulhw "LOCAL_MANGLE(MM_FIX_1_306562965)", %%mm1 \n\t"
2063         "psubw %%mm4, %%mm2            \n\t" //z13
2064
2065         "paddw %%mm4, %%mm5            \n\t" //z11
2066         "movq %%mm2, %%mm6             \n\t"
2067
2068         "paddw %%mm3, %%mm0            \n\t" //z2
2069         "movq %%mm5, %%mm7             \n\t"
2070
2071         "paddw %%mm0, %%mm2            \n\t" //d4
2072         "psubw %%mm0, %%mm6            \n\t" //d5
2073
2074         "movq %%mm2, %%mm4             \n\t"
2075         "paddw %%mm3, %%mm1            \n\t" //z4
2076
2077         //transpose 4x4
2078         "punpcklwd %%mm6, %%mm2        \n\t"
2079         "paddw %%mm1, %%mm5            \n\t" //d6
2080
2081         "punpckhwd %%mm6, %%mm4        \n\t"
2082         "psubw %%mm1, %%mm7            \n\t" //d7
2083
2084         "movq %%mm5, %%mm6             \n\t"
2085         "punpcklwd %%mm7, %%mm5        \n\t"
2086
2087         "punpckhwd %%mm7, %%mm6        \n\t"
2088         "movq %%mm2, %%mm7             \n\t"
2089
2090         "punpckldq %%mm5, %%mm2        \n\t" //4
2091         "sub %%"REG_d", %%"REG_S"             \n\t"
2092
2093         "punpckhdq %%mm5, %%mm7        \n\t" //5
2094         "movq %%mm4, %%mm5             \n\t"
2095
2096         "movq %%mm2, "DCTSIZE_S"*0*2+"DCTSIZE_S"(%%"REG_D") \n\t"
2097         "punpckldq %%mm6, %%mm4        \n\t" //6
2098
2099         "movq %%mm7, "DCTSIZE_S"*1*2+"DCTSIZE_S"(%%"REG_D") \n\t"
2100         "punpckhdq %%mm6, %%mm5        \n\t" //7
2101
2102         "movq %%mm4, "DCTSIZE_S"*2*2+"DCTSIZE_S"(%%"REG_D") \n\t"
2103         "add $4, %%"REG_S"               \n\t"
2104
2105         "movq %%mm5, "DCTSIZE_S"*3*2+"DCTSIZE_S"(%%"REG_D") \n\t"
2106         "add $"DCTSIZE_S"*2*4, %%"REG_D"      \n\t" //4 rows
2107         "dec %%"REG_c"                   \n\t"
2108         "jnz 6b                  \n\t"
2109
2110         : "+S"(pixels), "+D"(data), "+c"(cnt), "=o"(temps)
2111         : "a"(line_size)
2112         : "%"REG_d);
2113 }
2114
2115 #endif // HAVE_MMX