Statistics
| Branch: | Revision:

ffmpeg / libavcodec / i386 / h264dsp_mmx.c @ 9fa35729

History | View | Annotate | Download (58.7 KB)

1
/*
2
 * Copyright (c) 2004-2005 Michael Niedermayer, Loren Merritt
3
 *
4
 * This file is part of FFmpeg.
5
 *
6
 * FFmpeg 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.1 of the License, or (at your option) any later version.
10
 *
11
 * FFmpeg 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 FFmpeg; if not, write to the Free Software
18
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
19
 */
20

    
21
#include "dsputil_mmx.h"
22

    
23
/***********************************/
24
/* IDCT */
25

    
26
/* in/out: mma=mma+mmb, mmb=mmb-mma */
27
#define SUMSUB_BA( a, b ) \
28
    "paddw "#b", "#a" \n\t"\
29
    "paddw "#b", "#b" \n\t"\
30
    "psubw "#a", "#b" \n\t"
31

    
32
#define SUMSUB_BADC( a, b, c, d ) \
33
    "paddw "#b", "#a" \n\t"\
34
    "paddw "#d", "#c" \n\t"\
35
    "paddw "#b", "#b" \n\t"\
36
    "paddw "#d", "#d" \n\t"\
37
    "psubw "#a", "#b" \n\t"\
38
    "psubw "#c", "#d" \n\t"
39

    
40
#define SUMSUBD2_AB( a, b, t ) \
41
    "movq  "#b", "#t" \n\t"\
42
    "psraw  $1 , "#b" \n\t"\
43
    "paddw "#a", "#b" \n\t"\
44
    "psraw  $1 , "#a" \n\t"\
45
    "psubw "#t", "#a" \n\t"
46

    
47
#define IDCT4_1D( s02, s13, d02, d13, t ) \
48
    SUMSUB_BA  ( s02, d02 )\
49
    SUMSUBD2_AB( s13, d13, t )\
50
    SUMSUB_BADC( d13, s02, s13, d02 )
51

    
52
#define STORE_DIFF_4P( p, t, z ) \
53
    "psraw      $6,     "#p" \n\t"\
54
    "movd       (%0),   "#t" \n\t"\
55
    "punpcklbw "#z",    "#t" \n\t"\
56
    "paddsw    "#t",    "#p" \n\t"\
57
    "packuswb  "#z",    "#p" \n\t"\
58
    "movd      "#p",    (%0) \n\t"
59

    
60
static void ff_h264_idct_add_mmx(uint8_t *dst, int16_t *block, int stride)
61
{
62
    /* Load dct coeffs */
63
    asm volatile(
64
        "movq   (%0), %%mm0 \n\t"
65
        "movq  8(%0), %%mm1 \n\t"
66
        "movq 16(%0), %%mm2 \n\t"
67
        "movq 24(%0), %%mm3 \n\t"
68
    :: "r"(block) );
69

    
70
    asm volatile(
71
        /* mm1=s02+s13  mm2=s02-s13  mm4=d02+d13  mm0=d02-d13 */
72
        IDCT4_1D( %%mm2, %%mm1, %%mm0, %%mm3, %%mm4 )
73

    
74
        "movq      %0,    %%mm6 \n\t"
75
        /* in: 1,4,0,2  out: 1,2,3,0 */
76
        TRANSPOSE4( %%mm3, %%mm1, %%mm0, %%mm2, %%mm4 )
77

    
78
        "paddw     %%mm6, %%mm3 \n\t"
79

    
80
        /* mm2=s02+s13  mm3=s02-s13  mm4=d02+d13  mm1=d02-d13 */
81
        IDCT4_1D( %%mm4, %%mm2, %%mm3, %%mm0, %%mm1 )
82

    
83
        "pxor %%mm7, %%mm7    \n\t"
84
    :: "m"(ff_pw_32));
85

    
86
    asm volatile(
87
    STORE_DIFF_4P( %%mm0, %%mm1, %%mm7)
88
        "add %1, %0             \n\t"
89
    STORE_DIFF_4P( %%mm2, %%mm1, %%mm7)
90
        "add %1, %0             \n\t"
91
    STORE_DIFF_4P( %%mm3, %%mm1, %%mm7)
92
        "add %1, %0             \n\t"
93
    STORE_DIFF_4P( %%mm4, %%mm1, %%mm7)
94
        : "+r"(dst)
95
        : "r" ((long)stride)
96
    );
97
}
98

    
99
static inline void h264_idct8_1d(int16_t *block)
100
{
101
    asm volatile(
102
        "movq 112(%0), %%mm7  \n\t"
103
        "movq  80(%0), %%mm5  \n\t"
104
        "movq  48(%0), %%mm3  \n\t"
105
        "movq  16(%0), %%mm1  \n\t"
106

    
107
        "movq   %%mm7, %%mm4  \n\t"
108
        "movq   %%mm3, %%mm6  \n\t"
109
        "movq   %%mm5, %%mm0  \n\t"
110
        "movq   %%mm7, %%mm2  \n\t"
111
        "psraw  $1,    %%mm4  \n\t"
112
        "psraw  $1,    %%mm6  \n\t"
113
        "psubw  %%mm7, %%mm0  \n\t"
114
        "psubw  %%mm6, %%mm2  \n\t"
115
        "psubw  %%mm4, %%mm0  \n\t"
116
        "psubw  %%mm3, %%mm2  \n\t"
117
        "psubw  %%mm3, %%mm0  \n\t"
118
        "paddw  %%mm1, %%mm2  \n\t"
119

    
120
        "movq   %%mm5, %%mm4  \n\t"
121
        "movq   %%mm1, %%mm6  \n\t"
122
        "psraw  $1,    %%mm4  \n\t"
123
        "psraw  $1,    %%mm6  \n\t"
124
        "paddw  %%mm5, %%mm4  \n\t"
125
        "paddw  %%mm1, %%mm6  \n\t"
126
        "paddw  %%mm7, %%mm4  \n\t"
127
        "paddw  %%mm5, %%mm6  \n\t"
128
        "psubw  %%mm1, %%mm4  \n\t"
129
        "paddw  %%mm3, %%mm6  \n\t"
130

    
131
        "movq   %%mm0, %%mm1  \n\t"
132
        "movq   %%mm4, %%mm3  \n\t"
133
        "movq   %%mm2, %%mm5  \n\t"
134
        "movq   %%mm6, %%mm7  \n\t"
135
        "psraw  $2,    %%mm6  \n\t"
136
        "psraw  $2,    %%mm3  \n\t"
137
        "psraw  $2,    %%mm5  \n\t"
138
        "psraw  $2,    %%mm0  \n\t"
139
        "paddw  %%mm6, %%mm1  \n\t"
140
        "paddw  %%mm2, %%mm3  \n\t"
141
        "psubw  %%mm4, %%mm5  \n\t"
142
        "psubw  %%mm0, %%mm7  \n\t"
143

    
144
        "movq  32(%0), %%mm2  \n\t"
145
        "movq  96(%0), %%mm6  \n\t"
146
        "movq   %%mm2, %%mm4  \n\t"
147
        "movq   %%mm6, %%mm0  \n\t"
148
        "psraw  $1,    %%mm4  \n\t"
149
        "psraw  $1,    %%mm6  \n\t"
150
        "psubw  %%mm0, %%mm4  \n\t"
151
        "paddw  %%mm2, %%mm6  \n\t"
152

    
153
        "movq    (%0), %%mm2  \n\t"
154
        "movq  64(%0), %%mm0  \n\t"
155
        SUMSUB_BA( %%mm0, %%mm2 )
156
        SUMSUB_BA( %%mm6, %%mm0 )
157
        SUMSUB_BA( %%mm4, %%mm2 )
158
        SUMSUB_BA( %%mm7, %%mm6 )
159
        SUMSUB_BA( %%mm5, %%mm4 )
160
        SUMSUB_BA( %%mm3, %%mm2 )
161
        SUMSUB_BA( %%mm1, %%mm0 )
162
        :: "r"(block)
163
    );
164
}
165

    
166
static void ff_h264_idct8_add_mmx(uint8_t *dst, int16_t *block, int stride)
167
{
168
    int i;
169
    int16_t __attribute__ ((aligned(8))) b2[64];
170

    
171
    block[0] += 32;
172

    
173
    for(i=0; i<2; i++){
174
        DECLARE_ALIGNED_8(uint64_t, tmp);
175

    
176
        h264_idct8_1d(block+4*i);
177

    
178
        asm volatile(
179
            "movq   %%mm7,    %0   \n\t"
180
            TRANSPOSE4( %%mm0, %%mm2, %%mm4, %%mm6, %%mm7 )
181
            "movq   %%mm0,  8(%1)  \n\t"
182
            "movq   %%mm6, 24(%1)  \n\t"
183
            "movq   %%mm7, 40(%1)  \n\t"
184
            "movq   %%mm4, 56(%1)  \n\t"
185
            "movq    %0,    %%mm7  \n\t"
186
            TRANSPOSE4( %%mm7, %%mm5, %%mm3, %%mm1, %%mm0 )
187
            "movq   %%mm7,   (%1)  \n\t"
188
            "movq   %%mm1, 16(%1)  \n\t"
189
            "movq   %%mm0, 32(%1)  \n\t"
190
            "movq   %%mm3, 48(%1)  \n\t"
191
            : "=m"(tmp)
192
            : "r"(b2+32*i)
193
            : "memory"
194
        );
195
    }
196

    
197
    for(i=0; i<2; i++){
198
        h264_idct8_1d(b2+4*i);
199

    
200
        asm volatile(
201
            "psraw     $6, %%mm7  \n\t"
202
            "psraw     $6, %%mm6  \n\t"
203
            "psraw     $6, %%mm5  \n\t"
204
            "psraw     $6, %%mm4  \n\t"
205
            "psraw     $6, %%mm3  \n\t"
206
            "psraw     $6, %%mm2  \n\t"
207
            "psraw     $6, %%mm1  \n\t"
208
            "psraw     $6, %%mm0  \n\t"
209

    
210
            "movq   %%mm7,    (%0)  \n\t"
211
            "movq   %%mm5,  16(%0)  \n\t"
212
            "movq   %%mm3,  32(%0)  \n\t"
213
            "movq   %%mm1,  48(%0)  \n\t"
214
            "movq   %%mm0,  64(%0)  \n\t"
215
            "movq   %%mm2,  80(%0)  \n\t"
216
            "movq   %%mm4,  96(%0)  \n\t"
217
            "movq   %%mm6, 112(%0)  \n\t"
218
            :: "r"(b2+4*i)
219
            : "memory"
220
        );
221
    }
222

    
223
    add_pixels_clamped_mmx(b2, dst, stride);
224
}
225

    
226
static void ff_h264_idct_dc_add_mmx2(uint8_t *dst, int16_t *block, int stride)
227
{
228
    int dc = (block[0] + 32) >> 6;
229
    asm volatile(
230
        "movd          %0, %%mm0 \n\t"
231
        "pshufw $0, %%mm0, %%mm0 \n\t"
232
        "pxor       %%mm1, %%mm1 \n\t"
233
        "psubw      %%mm0, %%mm1 \n\t"
234
        "packuswb   %%mm0, %%mm0 \n\t"
235
        "packuswb   %%mm1, %%mm1 \n\t"
236
        ::"r"(dc)
237
    );
238
    asm volatile(
239
        "movd          %0, %%mm2 \n\t"
240
        "movd          %1, %%mm3 \n\t"
241
        "movd          %2, %%mm4 \n\t"
242
        "movd          %3, %%mm5 \n\t"
243
        "paddusb    %%mm0, %%mm2 \n\t"
244
        "paddusb    %%mm0, %%mm3 \n\t"
245
        "paddusb    %%mm0, %%mm4 \n\t"
246
        "paddusb    %%mm0, %%mm5 \n\t"
247
        "psubusb    %%mm1, %%mm2 \n\t"
248
        "psubusb    %%mm1, %%mm3 \n\t"
249
        "psubusb    %%mm1, %%mm4 \n\t"
250
        "psubusb    %%mm1, %%mm5 \n\t"
251
        "movd       %%mm2, %0    \n\t"
252
        "movd       %%mm3, %1    \n\t"
253
        "movd       %%mm4, %2    \n\t"
254
        "movd       %%mm5, %3    \n\t"
255
        :"+m"(*(uint32_t*)(dst+0*stride)),
256
         "+m"(*(uint32_t*)(dst+1*stride)),
257
         "+m"(*(uint32_t*)(dst+2*stride)),
258
         "+m"(*(uint32_t*)(dst+3*stride))
259
    );
260
}
261

    
262
static void ff_h264_idct8_dc_add_mmx2(uint8_t *dst, int16_t *block, int stride)
263
{
264
    int dc = (block[0] + 32) >> 6;
265
    int y;
266
    asm volatile(
267
        "movd          %0, %%mm0 \n\t"
268
        "pshufw $0, %%mm0, %%mm0 \n\t"
269
        "pxor       %%mm1, %%mm1 \n\t"
270
        "psubw      %%mm0, %%mm1 \n\t"
271
        "packuswb   %%mm0, %%mm0 \n\t"
272
        "packuswb   %%mm1, %%mm1 \n\t"
273
        ::"r"(dc)
274
    );
275
    for(y=2; y--; dst += 4*stride){
276
    asm volatile(
277
        "movq          %0, %%mm2 \n\t"
278
        "movq          %1, %%mm3 \n\t"
279
        "movq          %2, %%mm4 \n\t"
280
        "movq          %3, %%mm5 \n\t"
281
        "paddusb    %%mm0, %%mm2 \n\t"
282
        "paddusb    %%mm0, %%mm3 \n\t"
283
        "paddusb    %%mm0, %%mm4 \n\t"
284
        "paddusb    %%mm0, %%mm5 \n\t"
285
        "psubusb    %%mm1, %%mm2 \n\t"
286
        "psubusb    %%mm1, %%mm3 \n\t"
287
        "psubusb    %%mm1, %%mm4 \n\t"
288
        "psubusb    %%mm1, %%mm5 \n\t"
289
        "movq       %%mm2, %0    \n\t"
290
        "movq       %%mm3, %1    \n\t"
291
        "movq       %%mm4, %2    \n\t"
292
        "movq       %%mm5, %3    \n\t"
293
        :"+m"(*(uint64_t*)(dst+0*stride)),
294
         "+m"(*(uint64_t*)(dst+1*stride)),
295
         "+m"(*(uint64_t*)(dst+2*stride)),
296
         "+m"(*(uint64_t*)(dst+3*stride))
297
    );
298
    }
299
}
300

    
301

    
302
/***********************************/
303
/* deblocking */
304

    
305
// out: o = |x-y|>a
306
// clobbers: t
307
#define DIFF_GT_MMX(x,y,a,o,t)\
308
    "movq     "#y", "#t"  \n\t"\
309
    "movq     "#x", "#o"  \n\t"\
310
    "psubusb  "#x", "#t"  \n\t"\
311
    "psubusb  "#y", "#o"  \n\t"\
312
    "por      "#t", "#o"  \n\t"\
313
    "psubusb  "#a", "#o"  \n\t"
314

    
315
// out: o = |x-y|>a
316
// clobbers: t
317
#define DIFF_GT2_MMX(x,y,a,o,t)\
318
    "movq     "#y", "#t"  \n\t"\
319
    "movq     "#x", "#o"  \n\t"\
320
    "psubusb  "#x", "#t"  \n\t"\
321
    "psubusb  "#y", "#o"  \n\t"\
322
    "psubusb  "#a", "#t"  \n\t"\
323
    "psubusb  "#a", "#o"  \n\t"\
324
    "pcmpeqb  "#t", "#o"  \n\t"\
325

    
326
// in: mm0=p1 mm1=p0 mm2=q0 mm3=q1
327
// out: mm5=beta-1, mm7=mask
328
// clobbers: mm4,mm6
329
#define H264_DEBLOCK_MASK(alpha1, beta1) \
330
    "pshufw $0, "#alpha1", %%mm4 \n\t"\
331
    "pshufw $0, "#beta1 ", %%mm5 \n\t"\
332
    "packuswb  %%mm4, %%mm4      \n\t"\
333
    "packuswb  %%mm5, %%mm5      \n\t"\
334
    DIFF_GT_MMX(%%mm1, %%mm2, %%mm4, %%mm7, %%mm6) /* |p0-q0| > alpha-1 */\
335
    DIFF_GT_MMX(%%mm0, %%mm1, %%mm5, %%mm4, %%mm6) /* |p1-p0| > beta-1 */\
336
    "por       %%mm4, %%mm7      \n\t"\
337
    DIFF_GT_MMX(%%mm3, %%mm2, %%mm5, %%mm4, %%mm6) /* |q1-q0| > beta-1 */\
338
    "por       %%mm4, %%mm7      \n\t"\
339
    "pxor      %%mm6, %%mm6      \n\t"\
340
    "pcmpeqb   %%mm6, %%mm7      \n\t"
341

    
342
// in: mm0=p1 mm1=p0 mm2=q0 mm3=q1 mm7=(tc&mask)
343
// out: mm1=p0' mm2=q0'
344
// clobbers: mm0,3-6
345
#define H264_DEBLOCK_P0_Q0(pb_01, pb_3f)\
346
        "movq    %%mm1              , %%mm5 \n\t"\
347
        "pxor    %%mm2              , %%mm5 \n\t" /* p0^q0*/\
348
        "pand    "#pb_01"           , %%mm5 \n\t" /* (p0^q0)&1*/\
349
        "pcmpeqb %%mm4              , %%mm4 \n\t"\
350
        "pxor    %%mm4              , %%mm3 \n\t"\
351
        "pavgb   %%mm0              , %%mm3 \n\t" /* (p1 - q1 + 256)>>1*/\
352
        "pavgb   "MANGLE(ff_pb_3)"  , %%mm3 \n\t" /*(((p1 - q1 + 256)>>1)+4)>>1 = 64+2+(p1-q1)>>2*/\
353
        "pxor    %%mm1              , %%mm4 \n\t"\
354
        "pavgb   %%mm2              , %%mm4 \n\t" /* (q0 - p0 + 256)>>1*/\
355
        "pavgb   %%mm5              , %%mm3 \n\t"\
356
        "paddusb %%mm4              , %%mm3 \n\t" /* d+128+33*/\
357
        "movq    "MANGLE(ff_pb_A1)" , %%mm6 \n\t"\
358
        "psubusb %%mm3              , %%mm6 \n\t"\
359
        "psubusb "MANGLE(ff_pb_A1)" , %%mm3 \n\t"\
360
        "pminub  %%mm7              , %%mm6 \n\t"\
361
        "pminub  %%mm7              , %%mm3 \n\t"\
362
        "psubusb %%mm6              , %%mm1 \n\t"\
363
        "psubusb %%mm3              , %%mm2 \n\t"\
364
        "paddusb %%mm3              , %%mm1 \n\t"\
365
        "paddusb %%mm6              , %%mm2 \n\t"
366

    
367
// in: mm0=p1 mm1=p0 mm2=q0 mm3=q1 mm7=(tc&mask) %8=ff_bone
368
// out: (q1addr) = av_clip( (q2+((p0+q0+1)>>1))>>1, q1-tc0, q1+tc0 )
369
// clobbers: q2, tmp, tc0
370
#define H264_DEBLOCK_Q1(p1, q2, q2addr, q1addr, tc0, tmp)\
371
        "movq     %%mm1,  "#tmp"   \n\t"\
372
        "pavgb    %%mm2,  "#tmp"   \n\t"\
373
        "pavgb    "#tmp", "#q2"    \n\t" /* avg(p2,avg(p0,q0)) */\
374
        "pxor   "q2addr", "#tmp"   \n\t"\
375
        "pand     %8,     "#tmp"   \n\t" /* (p2^avg(p0,q0))&1 */\
376
        "psubusb  "#tmp", "#q2"    \n\t" /* (p2+((p0+q0+1)>>1))>>1 */\
377
        "movq     "#p1",  "#tmp"   \n\t"\
378
        "psubusb  "#tc0", "#tmp"   \n\t"\
379
        "paddusb  "#p1",  "#tc0"   \n\t"\
380
        "pmaxub   "#tmp", "#q2"    \n\t"\
381
        "pminub   "#tc0", "#q2"    \n\t"\
382
        "movq     "#q2",  "q1addr" \n\t"
383

    
384
static inline void h264_loop_filter_luma_mmx2(uint8_t *pix, int stride, int alpha1, int beta1, int8_t *tc0)
385
{
386
    DECLARE_ALIGNED_8(uint64_t, tmp0[2]);
387

    
388
    asm volatile(
389
        "movq    (%1,%3), %%mm0    \n\t" //p1
390
        "movq    (%1,%3,2), %%mm1  \n\t" //p0
391
        "movq    (%2),    %%mm2    \n\t" //q0
392
        "movq    (%2,%3), %%mm3    \n\t" //q1
393
        H264_DEBLOCK_MASK(%6, %7)
394

    
395
        "movd      %5,    %%mm4    \n\t"
396
        "punpcklbw %%mm4, %%mm4    \n\t"
397
        "punpcklwd %%mm4, %%mm4    \n\t"
398
        "pcmpeqb   %%mm3, %%mm3    \n\t"
399
        "movq      %%mm4, %%mm6    \n\t"
400
        "pcmpgtb   %%mm3, %%mm4    \n\t"
401
        "movq      %%mm6, 8+%0     \n\t"
402
        "pand      %%mm4, %%mm7    \n\t"
403
        "movq      %%mm7, %0       \n\t"
404

    
405
        /* filter p1 */
406
        "movq     (%1),   %%mm3    \n\t" //p2
407
        DIFF_GT2_MMX(%%mm1, %%mm3, %%mm5, %%mm6, %%mm4) // |p2-p0|>beta-1
408
        "pand     %%mm7,  %%mm6    \n\t" // mask & |p2-p0|<beta
409
        "pand     8+%0,   %%mm7    \n\t" // mask & tc0
410
        "movq     %%mm7,  %%mm4    \n\t"
411
        "psubb    %%mm6,  %%mm7    \n\t"
412
        "pand     %%mm4,  %%mm6    \n\t" // mask & |p2-p0|<beta & tc0
413
        H264_DEBLOCK_Q1(%%mm0, %%mm3, "(%1)", "(%1,%3)", %%mm6, %%mm4)
414

    
415
        /* filter q1 */
416
        "movq    (%2,%3,2), %%mm4  \n\t" //q2
417
        DIFF_GT2_MMX(%%mm2, %%mm4, %%mm5, %%mm6, %%mm3) // |q2-q0|>beta-1
418
        "pand     %0,     %%mm6    \n\t"
419
        "movq     8+%0,   %%mm5    \n\t" // can be merged with the and below but is slower then
420
        "pand     %%mm6,  %%mm5    \n\t"
421
        "psubb    %%mm6,  %%mm7    \n\t"
422
        "movq    (%2,%3), %%mm3    \n\t"
423
        H264_DEBLOCK_Q1(%%mm3, %%mm4, "(%2,%3,2)", "(%2,%3)", %%mm5, %%mm6)
424

    
425
        /* filter p0, q0 */
426
        H264_DEBLOCK_P0_Q0(%8, unused)
427
        "movq      %%mm1, (%1,%3,2) \n\t"
428
        "movq      %%mm2, (%2)      \n\t"
429

    
430
        : "=m"(*tmp0)
431
        : "r"(pix-3*stride), "r"(pix), "r"((long)stride),
432
          "m"(*tmp0/*unused*/), "m"(*(uint32_t*)tc0), "m"(alpha1), "m"(beta1),
433
          "m"(ff_bone)
434
    );
435
}
436

    
437
static void h264_v_loop_filter_luma_mmx2(uint8_t *pix, int stride, int alpha, int beta, int8_t *tc0)
438
{
439
    if((tc0[0] & tc0[1]) >= 0)
440
        h264_loop_filter_luma_mmx2(pix, stride, alpha-1, beta-1, tc0);
441
    if((tc0[2] & tc0[3]) >= 0)
442
        h264_loop_filter_luma_mmx2(pix+8, stride, alpha-1, beta-1, tc0+2);
443
}
444
static void h264_h_loop_filter_luma_mmx2(uint8_t *pix, int stride, int alpha, int beta, int8_t *tc0)
445
{
446
    //FIXME: could cut some load/stores by merging transpose with filter
447
    // also, it only needs to transpose 6x8
448
    DECLARE_ALIGNED_8(uint8_t, trans[8*8]);
449
    int i;
450
    for(i=0; i<2; i++, pix+=8*stride, tc0+=2) {
451
        if((tc0[0] & tc0[1]) < 0)
452
            continue;
453
        transpose4x4(trans,       pix-4,          8, stride);
454
        transpose4x4(trans  +4*8, pix,            8, stride);
455
        transpose4x4(trans+4,     pix-4+4*stride, 8, stride);
456
        transpose4x4(trans+4+4*8, pix  +4*stride, 8, stride);
457
        h264_loop_filter_luma_mmx2(trans+4*8, 8, alpha-1, beta-1, tc0);
458
        transpose4x4(pix-2,          trans  +2*8, stride, 8);
459
        transpose4x4(pix-2+4*stride, trans+4+2*8, stride, 8);
460
    }
461
}
462

    
463
static inline void h264_loop_filter_chroma_mmx2(uint8_t *pix, int stride, int alpha1, int beta1, int8_t *tc0)
464
{
465
    asm volatile(
466
        "movq    (%0),    %%mm0     \n\t" //p1
467
        "movq    (%0,%2), %%mm1     \n\t" //p0
468
        "movq    (%1),    %%mm2     \n\t" //q0
469
        "movq    (%1,%2), %%mm3     \n\t" //q1
470
        H264_DEBLOCK_MASK(%4, %5)
471
        "movd      %3,    %%mm6     \n\t"
472
        "punpcklbw %%mm6, %%mm6     \n\t"
473
        "pand      %%mm6, %%mm7     \n\t" // mm7 = tc&mask
474
        H264_DEBLOCK_P0_Q0(%6, %7)
475
        "movq      %%mm1, (%0,%2)   \n\t"
476
        "movq      %%mm2, (%1)      \n\t"
477

    
478
        :: "r"(pix-2*stride), "r"(pix), "r"((long)stride),
479
           "r"(*(uint32_t*)tc0),
480
           "m"(alpha1), "m"(beta1), "m"(ff_bone), "m"(ff_pb_3F)
481
    );
482
}
483

    
484
static void h264_v_loop_filter_chroma_mmx2(uint8_t *pix, int stride, int alpha, int beta, int8_t *tc0)
485
{
486
    h264_loop_filter_chroma_mmx2(pix, stride, alpha-1, beta-1, tc0);
487
}
488

    
489
static void h264_h_loop_filter_chroma_mmx2(uint8_t *pix, int stride, int alpha, int beta, int8_t *tc0)
490
{
491
    //FIXME: could cut some load/stores by merging transpose with filter
492
    DECLARE_ALIGNED_8(uint8_t, trans[8*4]);
493
    transpose4x4(trans, pix-2, 8, stride);
494
    transpose4x4(trans+4, pix-2+4*stride, 8, stride);
495
    h264_loop_filter_chroma_mmx2(trans+2*8, 8, alpha-1, beta-1, tc0);
496
    transpose4x4(pix-2, trans, stride, 8);
497
    transpose4x4(pix-2+4*stride, trans+4, stride, 8);
498
}
499

    
500
// p0 = (p0 + q1 + 2*p1 + 2) >> 2
501
#define H264_FILTER_CHROMA4(p0, p1, q1, one) \
502
    "movq    "#p0", %%mm4  \n\t"\
503
    "pxor    "#q1", %%mm4  \n\t"\
504
    "pand   "#one", %%mm4  \n\t" /* mm4 = (p0^q1)&1 */\
505
    "pavgb   "#q1", "#p0"  \n\t"\
506
    "psubusb %%mm4, "#p0"  \n\t"\
507
    "pavgb   "#p1", "#p0"  \n\t" /* dst = avg(p1, avg(p0,q1) - ((p0^q1)&1)) */\
508

    
509
static inline void h264_loop_filter_chroma_intra_mmx2(uint8_t *pix, int stride, int alpha1, int beta1)
510
{
511
    asm volatile(
512
        "movq    (%0),    %%mm0     \n\t"
513
        "movq    (%0,%2), %%mm1     \n\t"
514
        "movq    (%1),    %%mm2     \n\t"
515
        "movq    (%1,%2), %%mm3     \n\t"
516
        H264_DEBLOCK_MASK(%3, %4)
517
        "movq    %%mm1,   %%mm5     \n\t"
518
        "movq    %%mm2,   %%mm6     \n\t"
519
        H264_FILTER_CHROMA4(%%mm1, %%mm0, %%mm3, %5) //p0'
520
        H264_FILTER_CHROMA4(%%mm2, %%mm3, %%mm0, %5) //q0'
521
        "psubb   %%mm5,   %%mm1     \n\t"
522
        "psubb   %%mm6,   %%mm2     \n\t"
523
        "pand    %%mm7,   %%mm1     \n\t"
524
        "pand    %%mm7,   %%mm2     \n\t"
525
        "paddb   %%mm5,   %%mm1     \n\t"
526
        "paddb   %%mm6,   %%mm2     \n\t"
527
        "movq    %%mm1,   (%0,%2)   \n\t"
528
        "movq    %%mm2,   (%1)      \n\t"
529
        :: "r"(pix-2*stride), "r"(pix), "r"((long)stride),
530
           "m"(alpha1), "m"(beta1), "m"(ff_bone)
531
    );
532
}
533

    
534
static void h264_v_loop_filter_chroma_intra_mmx2(uint8_t *pix, int stride, int alpha, int beta)
535
{
536
    h264_loop_filter_chroma_intra_mmx2(pix, stride, alpha-1, beta-1);
537
}
538

    
539
static void h264_h_loop_filter_chroma_intra_mmx2(uint8_t *pix, int stride, int alpha, int beta)
540
{
541
    //FIXME: could cut some load/stores by merging transpose with filter
542
    DECLARE_ALIGNED_8(uint8_t, trans[8*4]);
543
    transpose4x4(trans, pix-2, 8, stride);
544
    transpose4x4(trans+4, pix-2+4*stride, 8, stride);
545
    h264_loop_filter_chroma_intra_mmx2(trans+2*8, 8, alpha-1, beta-1);
546
    transpose4x4(pix-2, trans, stride, 8);
547
    transpose4x4(pix-2+4*stride, trans+4, stride, 8);
548
}
549

    
550
static void h264_loop_filter_strength_mmx2( int16_t bS[2][4][4], uint8_t nnz[40], int8_t ref[2][40], int16_t mv[2][40][2],
551
                                            int bidir, int edges, int step, int mask_mv0, int mask_mv1 ) {
552
    int dir;
553
    asm volatile(
554
        "pxor %%mm7, %%mm7 \n\t"
555
        "movq %0, %%mm6 \n\t"
556
        "movq %1, %%mm5 \n\t"
557
        "movq %2, %%mm4 \n\t"
558
        ::"m"(ff_pb_1), "m"(ff_pb_3), "m"(ff_pb_7)
559
    );
560
    // could do a special case for dir==0 && edges==1, but it only reduces the
561
    // average filter time by 1.2%
562
    for( dir=1; dir>=0; dir-- ) {
563
        const int d_idx = dir ? -8 : -1;
564
        const int mask_mv = dir ? mask_mv1 : mask_mv0;
565
        DECLARE_ALIGNED_8(const uint64_t, mask_dir) = dir ? 0 : 0xffffffffffffffffULL;
566
        int b_idx, edge, l;
567
        for( b_idx=12, edge=0; edge<edges; edge+=step, b_idx+=8*step ) {
568
            asm volatile(
569
                "pand %0, %%mm0 \n\t"
570
                ::"m"(mask_dir)
571
            );
572
            if(!(mask_mv & edge)) {
573
                asm volatile("pxor %%mm0, %%mm0 \n\t":);
574
                for( l = bidir; l >= 0; l-- ) {
575
                    asm volatile(
576
                        "movd %0, %%mm1 \n\t"
577
                        "punpckldq %1, %%mm1 \n\t"
578
                        "movq %%mm1, %%mm2 \n\t"
579
                        "psrlw $7, %%mm2 \n\t"
580
                        "pand %%mm6, %%mm2 \n\t"
581
                        "por %%mm2, %%mm1 \n\t" // ref_cache with -2 mapped to -1
582
                        "punpckldq %%mm1, %%mm2 \n\t"
583
                        "pcmpeqb %%mm2, %%mm1 \n\t"
584
                        "paddb %%mm6, %%mm1 \n\t"
585
                        "punpckhbw %%mm7, %%mm1 \n\t" // ref[b] != ref[bn]
586
                        "por %%mm1, %%mm0 \n\t"
587

    
588
                        "movq %2, %%mm1 \n\t"
589
                        "movq %3, %%mm2 \n\t"
590
                        "psubw %4, %%mm1 \n\t"
591
                        "psubw %5, %%mm2 \n\t"
592
                        "packsswb %%mm2, %%mm1 \n\t"
593
                        "paddb %%mm5, %%mm1 \n\t"
594
                        "pminub %%mm4, %%mm1 \n\t"
595
                        "pcmpeqb %%mm4, %%mm1 \n\t" // abs(mv[b] - mv[bn]) >= limit
596
                        "por %%mm1, %%mm0 \n\t"
597
                        ::"m"(ref[l][b_idx]),
598
                          "m"(ref[l][b_idx+d_idx]),
599
                          "m"(mv[l][b_idx][0]),
600
                          "m"(mv[l][b_idx+2][0]),
601
                          "m"(mv[l][b_idx+d_idx][0]),
602
                          "m"(mv[l][b_idx+d_idx+2][0])
603
                    );
604
                }
605
            }
606
            asm volatile(
607
                "movd %0, %%mm1 \n\t"
608
                "por  %1, %%mm1 \n\t"
609
                "punpcklbw %%mm7, %%mm1 \n\t"
610
                "pcmpgtw %%mm7, %%mm1 \n\t" // nnz[b] || nnz[bn]
611
                ::"m"(nnz[b_idx]),
612
                  "m"(nnz[b_idx+d_idx])
613
            );
614
            asm volatile(
615
                "pcmpeqw %%mm7, %%mm0 \n\t"
616
                "pcmpeqw %%mm7, %%mm0 \n\t"
617
                "psrlw $15, %%mm0 \n\t" // nonzero -> 1
618
                "psrlw $14, %%mm1 \n\t"
619
                "movq %%mm0, %%mm2 \n\t"
620
                "por %%mm1, %%mm2 \n\t"
621
                "psrlw $1, %%mm1 \n\t"
622
                "pandn %%mm2, %%mm1 \n\t"
623
                "movq %%mm1, %0 \n\t"
624
                :"=m"(*bS[dir][edge])
625
                ::"memory"
626
            );
627
        }
628
        edges = 4;
629
        step = 1;
630
    }
631
    asm volatile(
632
        "movq   (%0), %%mm0 \n\t"
633
        "movq  8(%0), %%mm1 \n\t"
634
        "movq 16(%0), %%mm2 \n\t"
635
        "movq 24(%0), %%mm3 \n\t"
636
        TRANSPOSE4(%%mm0, %%mm1, %%mm2, %%mm3, %%mm4)
637
        "movq %%mm0,   (%0) \n\t"
638
        "movq %%mm3,  8(%0) \n\t"
639
        "movq %%mm4, 16(%0) \n\t"
640
        "movq %%mm2, 24(%0) \n\t"
641
        ::"r"(bS[0])
642
        :"memory"
643
    );
644
}
645

    
646
/***********************************/
647
/* motion compensation */
648

    
649
#define QPEL_H264V(A,B,C,D,E,F,OP)\
650
        "movd (%0), "#F"            \n\t"\
651
        "movq "#C", %%mm6           \n\t"\
652
        "paddw "#D", %%mm6          \n\t"\
653
        "psllw $2, %%mm6            \n\t"\
654
        "psubw "#B", %%mm6          \n\t"\
655
        "psubw "#E", %%mm6          \n\t"\
656
        "pmullw %4, %%mm6           \n\t"\
657
        "add %2, %0                 \n\t"\
658
        "punpcklbw %%mm7, "#F"      \n\t"\
659
        "paddw %5, "#A"             \n\t"\
660
        "paddw "#F", "#A"           \n\t"\
661
        "paddw "#A", %%mm6          \n\t"\
662
        "psraw $5, %%mm6            \n\t"\
663
        "packuswb %%mm6, %%mm6      \n\t"\
664
        OP(%%mm6, (%1), A, d)\
665
        "add %3, %1                 \n\t"
666

    
667
#define QPEL_H264HV(A,B,C,D,E,F,OF)\
668
        "movd (%0), "#F"            \n\t"\
669
        "movq "#C", %%mm6           \n\t"\
670
        "paddw "#D", %%mm6          \n\t"\
671
        "psllw $2, %%mm6            \n\t"\
672
        "psubw "#B", %%mm6          \n\t"\
673
        "psubw "#E", %%mm6          \n\t"\
674
        "pmullw %3, %%mm6           \n\t"\
675
        "add %2, %0                 \n\t"\
676
        "punpcklbw %%mm7, "#F"      \n\t"\
677
        "paddw "#F", "#A"           \n\t"\
678
        "paddw "#A", %%mm6          \n\t"\
679
        "movq %%mm6, "#OF"(%1)      \n\t"
680

    
681
#define QPEL_H264(OPNAME, OP, MMX)\
682
static av_noinline void OPNAME ## h264_qpel4_h_lowpass_ ## MMX(uint8_t *dst, uint8_t *src, int dstStride, int srcStride){\
683
    int h=4;\
684
\
685
    asm volatile(\
686
        "pxor %%mm7, %%mm7          \n\t"\
687
        "movq %5, %%mm4             \n\t"\
688
        "movq %6, %%mm5             \n\t"\
689
        "1:                         \n\t"\
690
        "movd  -1(%0), %%mm1        \n\t"\
691
        "movd    (%0), %%mm2        \n\t"\
692
        "movd   1(%0), %%mm3        \n\t"\
693
        "movd   2(%0), %%mm0        \n\t"\
694
        "punpcklbw %%mm7, %%mm1     \n\t"\
695
        "punpcklbw %%mm7, %%mm2     \n\t"\
696
        "punpcklbw %%mm7, %%mm3     \n\t"\
697
        "punpcklbw %%mm7, %%mm0     \n\t"\
698
        "paddw %%mm0, %%mm1         \n\t"\
699
        "paddw %%mm3, %%mm2         \n\t"\
700
        "movd  -2(%0), %%mm0        \n\t"\
701
        "movd   3(%0), %%mm3        \n\t"\
702
        "punpcklbw %%mm7, %%mm0     \n\t"\
703
        "punpcklbw %%mm7, %%mm3     \n\t"\
704
        "paddw %%mm3, %%mm0         \n\t"\
705
        "psllw $2, %%mm2            \n\t"\
706
        "psubw %%mm1, %%mm2         \n\t"\
707
        "pmullw %%mm4, %%mm2        \n\t"\
708
        "paddw %%mm5, %%mm0         \n\t"\
709
        "paddw %%mm2, %%mm0         \n\t"\
710
        "psraw $5, %%mm0            \n\t"\
711
        "packuswb %%mm0, %%mm0      \n\t"\
712
        OP(%%mm0, (%1),%%mm6, d)\
713
        "add %3, %0                 \n\t"\
714
        "add %4, %1                 \n\t"\
715
        "decl %2                    \n\t"\
716
        " jnz 1b                    \n\t"\
717
        : "+a"(src), "+c"(dst), "+m"(h)\
718
        : "d"((long)srcStride), "S"((long)dstStride), "m"(ff_pw_5), "m"(ff_pw_16)\
719
        : "memory"\
720
    );\
721
}\
722
static av_noinline void OPNAME ## h264_qpel4_h_lowpass_l2_ ## MMX(uint8_t *dst, uint8_t *src, uint8_t *src2, int dstStride, int src2Stride){\
723
    int h=4;\
724
    asm volatile(\
725
        "pxor %%mm7, %%mm7          \n\t"\
726
        "movq %0, %%mm4             \n\t"\
727
        "movq %1, %%mm5             \n\t"\
728
        :: "m"(ff_pw_5), "m"(ff_pw_16)\
729
    );\
730
    do{\
731
    asm volatile(\
732
        "movd  -1(%0), %%mm1        \n\t"\
733
        "movd    (%0), %%mm2        \n\t"\
734
        "movd   1(%0), %%mm3        \n\t"\
735
        "movd   2(%0), %%mm0        \n\t"\
736
        "punpcklbw %%mm7, %%mm1     \n\t"\
737
        "punpcklbw %%mm7, %%mm2     \n\t"\
738
        "punpcklbw %%mm7, %%mm3     \n\t"\
739
        "punpcklbw %%mm7, %%mm0     \n\t"\
740
        "paddw %%mm0, %%mm1         \n\t"\
741
        "paddw %%mm3, %%mm2         \n\t"\
742
        "movd  -2(%0), %%mm0        \n\t"\
743
        "movd   3(%0), %%mm3        \n\t"\
744
        "punpcklbw %%mm7, %%mm0     \n\t"\
745
        "punpcklbw %%mm7, %%mm3     \n\t"\
746
        "paddw %%mm3, %%mm0         \n\t"\
747
        "psllw $2, %%mm2            \n\t"\
748
        "psubw %%mm1, %%mm2         \n\t"\
749
        "pmullw %%mm4, %%mm2        \n\t"\
750
        "paddw %%mm5, %%mm0         \n\t"\
751
        "paddw %%mm2, %%mm0         \n\t"\
752
        "movd   (%2), %%mm3         \n\t"\
753
        "psraw $5, %%mm0            \n\t"\
754
        "packuswb %%mm0, %%mm0      \n\t"\
755
        PAVGB" %%mm3, %%mm0         \n\t"\
756
        OP(%%mm0, (%1),%%mm6, d)\
757
        "add %4, %0                 \n\t"\
758
        "add %4, %1                 \n\t"\
759
        "add %3, %2                 \n\t"\
760
        : "+a"(src), "+c"(dst), "+d"(src2)\
761
        : "D"((long)src2Stride), "S"((long)dstStride)\
762
        : "memory"\
763
    );\
764
    }while(--h);\
765
}\
766
static av_noinline void OPNAME ## h264_qpel4_v_lowpass_ ## MMX(uint8_t *dst, uint8_t *src, int dstStride, int srcStride){\
767
    src -= 2*srcStride;\
768
    asm volatile(\
769
        "pxor %%mm7, %%mm7          \n\t"\
770
        "movd (%0), %%mm0           \n\t"\
771
        "add %2, %0                 \n\t"\
772
        "movd (%0), %%mm1           \n\t"\
773
        "add %2, %0                 \n\t"\
774
        "movd (%0), %%mm2           \n\t"\
775
        "add %2, %0                 \n\t"\
776
        "movd (%0), %%mm3           \n\t"\
777
        "add %2, %0                 \n\t"\
778
        "movd (%0), %%mm4           \n\t"\
779
        "add %2, %0                 \n\t"\
780
        "punpcklbw %%mm7, %%mm0     \n\t"\
781
        "punpcklbw %%mm7, %%mm1     \n\t"\
782
        "punpcklbw %%mm7, %%mm2     \n\t"\
783
        "punpcklbw %%mm7, %%mm3     \n\t"\
784
        "punpcklbw %%mm7, %%mm4     \n\t"\
785
        QPEL_H264V(%%mm0, %%mm1, %%mm2, %%mm3, %%mm4, %%mm5, OP)\
786
        QPEL_H264V(%%mm1, %%mm2, %%mm3, %%mm4, %%mm5, %%mm0, OP)\
787
        QPEL_H264V(%%mm2, %%mm3, %%mm4, %%mm5, %%mm0, %%mm1, OP)\
788
        QPEL_H264V(%%mm3, %%mm4, %%mm5, %%mm0, %%mm1, %%mm2, OP)\
789
         \
790
        : "+a"(src), "+c"(dst)\
791
        : "S"((long)srcStride), "D"((long)dstStride), "m"(ff_pw_5), "m"(ff_pw_16)\
792
        : "memory"\
793
    );\
794
}\
795
static av_noinline void OPNAME ## h264_qpel4_hv_lowpass_ ## MMX(uint8_t *dst, int16_t *tmp, uint8_t *src, int dstStride, int tmpStride, int srcStride){\
796
    int h=4;\
797
    int w=3;\
798
    src -= 2*srcStride+2;\
799
    while(w--){\
800
        asm volatile(\
801
            "pxor %%mm7, %%mm7      \n\t"\
802
            "movd (%0), %%mm0       \n\t"\
803
            "add %2, %0             \n\t"\
804
            "movd (%0), %%mm1       \n\t"\
805
            "add %2, %0             \n\t"\
806
            "movd (%0), %%mm2       \n\t"\
807
            "add %2, %0             \n\t"\
808
            "movd (%0), %%mm3       \n\t"\
809
            "add %2, %0             \n\t"\
810
            "movd (%0), %%mm4       \n\t"\
811
            "add %2, %0             \n\t"\
812
            "punpcklbw %%mm7, %%mm0 \n\t"\
813
            "punpcklbw %%mm7, %%mm1 \n\t"\
814
            "punpcklbw %%mm7, %%mm2 \n\t"\
815
            "punpcklbw %%mm7, %%mm3 \n\t"\
816
            "punpcklbw %%mm7, %%mm4 \n\t"\
817
            QPEL_H264HV(%%mm0, %%mm1, %%mm2, %%mm3, %%mm4, %%mm5, 0*8*3)\
818
            QPEL_H264HV(%%mm1, %%mm2, %%mm3, %%mm4, %%mm5, %%mm0, 1*8*3)\
819
            QPEL_H264HV(%%mm2, %%mm3, %%mm4, %%mm5, %%mm0, %%mm1, 2*8*3)\
820
            QPEL_H264HV(%%mm3, %%mm4, %%mm5, %%mm0, %%mm1, %%mm2, 3*8*3)\
821
             \
822
            : "+a"(src)\
823
            : "c"(tmp), "S"((long)srcStride), "m"(ff_pw_5)\
824
            : "memory"\
825
        );\
826
        tmp += 4;\
827
        src += 4 - 9*srcStride;\
828
    }\
829
    tmp -= 3*4;\
830
    asm volatile(\
831
        "movq %4, %%mm6             \n\t"\
832
        "1:                         \n\t"\
833
        "movq     (%0), %%mm0       \n\t"\
834
        "paddw  10(%0), %%mm0       \n\t"\
835
        "movq    2(%0), %%mm1       \n\t"\
836
        "paddw   8(%0), %%mm1       \n\t"\
837
        "movq    4(%0), %%mm2       \n\t"\
838
        "paddw   6(%0), %%mm2       \n\t"\
839
        "psubw %%mm1, %%mm0         \n\t"/*a-b   (abccba)*/\
840
        "psraw $2, %%mm0            \n\t"/*(a-b)/4 */\
841
        "psubw %%mm1, %%mm0         \n\t"/*(a-b)/4-b */\
842
        "paddsw %%mm2, %%mm0        \n\t"\
843
        "psraw $2, %%mm0            \n\t"/*((a-b)/4-b+c)/4 */\
844
        "paddw %%mm6, %%mm2         \n\t"\
845
        "paddw %%mm2, %%mm0         \n\t"/*(a-5*b+20*c)/16 +32 */\
846
        "psraw $6, %%mm0            \n\t"\
847
        "packuswb %%mm0, %%mm0      \n\t"\
848
        OP(%%mm0, (%1),%%mm7, d)\
849
        "add $24, %0                \n\t"\
850
        "add %3, %1                 \n\t"\
851
        "decl %2                    \n\t"\
852
        " jnz 1b                    \n\t"\
853
        : "+a"(tmp), "+c"(dst), "+m"(h)\
854
        : "S"((long)dstStride), "m"(ff_pw_32)\
855
        : "memory"\
856
    );\
857
}\
858
\
859
static av_noinline void OPNAME ## h264_qpel8_h_lowpass_ ## MMX(uint8_t *dst, uint8_t *src, int dstStride, int srcStride){\
860
    int h=8;\
861
    asm volatile(\
862
        "pxor %%mm7, %%mm7          \n\t"\
863
        "movq %5, %%mm6             \n\t"\
864
        "1:                         \n\t"\
865
        "movq    (%0), %%mm0        \n\t"\
866
        "movq   1(%0), %%mm2        \n\t"\
867
        "movq %%mm0, %%mm1          \n\t"\
868
        "movq %%mm2, %%mm3          \n\t"\
869
        "punpcklbw %%mm7, %%mm0     \n\t"\
870
        "punpckhbw %%mm7, %%mm1     \n\t"\
871
        "punpcklbw %%mm7, %%mm2     \n\t"\
872
        "punpckhbw %%mm7, %%mm3     \n\t"\
873
        "paddw %%mm2, %%mm0         \n\t"\
874
        "paddw %%mm3, %%mm1         \n\t"\
875
        "psllw $2, %%mm0            \n\t"\
876
        "psllw $2, %%mm1            \n\t"\
877
        "movq   -1(%0), %%mm2       \n\t"\
878
        "movq    2(%0), %%mm4       \n\t"\
879
        "movq %%mm2, %%mm3          \n\t"\
880
        "movq %%mm4, %%mm5          \n\t"\
881
        "punpcklbw %%mm7, %%mm2     \n\t"\
882
        "punpckhbw %%mm7, %%mm3     \n\t"\
883
        "punpcklbw %%mm7, %%mm4     \n\t"\
884
        "punpckhbw %%mm7, %%mm5     \n\t"\
885
        "paddw %%mm4, %%mm2         \n\t"\
886
        "paddw %%mm3, %%mm5         \n\t"\
887
        "psubw %%mm2, %%mm0         \n\t"\
888
        "psubw %%mm5, %%mm1         \n\t"\
889
        "pmullw %%mm6, %%mm0        \n\t"\
890
        "pmullw %%mm6, %%mm1        \n\t"\
891
        "movd   -2(%0), %%mm2       \n\t"\
892
        "movd    7(%0), %%mm5       \n\t"\
893
        "punpcklbw %%mm7, %%mm2     \n\t"\
894
        "punpcklbw %%mm7, %%mm5     \n\t"\
895
        "paddw %%mm3, %%mm2         \n\t"\
896
        "paddw %%mm5, %%mm4         \n\t"\
897
        "movq %6, %%mm5             \n\t"\
898
        "paddw %%mm5, %%mm2         \n\t"\
899
        "paddw %%mm5, %%mm4         \n\t"\
900
        "paddw %%mm2, %%mm0         \n\t"\
901
        "paddw %%mm4, %%mm1         \n\t"\
902
        "psraw $5, %%mm0            \n\t"\
903
        "psraw $5, %%mm1            \n\t"\
904
        "packuswb %%mm1, %%mm0      \n\t"\
905
        OP(%%mm0, (%1),%%mm5, q)\
906
        "add %3, %0                 \n\t"\
907
        "add %4, %1                 \n\t"\
908
        "decl %2                    \n\t"\
909
        " jnz 1b                    \n\t"\
910
        : "+a"(src), "+c"(dst), "+m"(h)\
911
        : "d"((long)srcStride), "S"((long)dstStride), "m"(ff_pw_5), "m"(ff_pw_16)\
912
        : "memory"\
913
    );\
914
}\
915
\
916
static av_noinline void OPNAME ## h264_qpel8_h_lowpass_l2_ ## MMX(uint8_t *dst, uint8_t *src, uint8_t *src2, int dstStride, int src2Stride){\
917
    int h=8;\
918
    asm volatile(\
919
        "pxor %%mm7, %%mm7          \n\t"\
920
        "movq %0, %%mm6             \n\t"\
921
        :: "m"(ff_pw_5)\
922
    );\
923
    do{\
924
    asm volatile(\
925
        "movq    (%0), %%mm0        \n\t"\
926
        "movq   1(%0), %%mm2        \n\t"\
927
        "movq %%mm0, %%mm1          \n\t"\
928
        "movq %%mm2, %%mm3          \n\t"\
929
        "punpcklbw %%mm7, %%mm0     \n\t"\
930
        "punpckhbw %%mm7, %%mm1     \n\t"\
931
        "punpcklbw %%mm7, %%mm2     \n\t"\
932
        "punpckhbw %%mm7, %%mm3     \n\t"\
933
        "paddw %%mm2, %%mm0         \n\t"\
934
        "paddw %%mm3, %%mm1         \n\t"\
935
        "psllw $2, %%mm0            \n\t"\
936
        "psllw $2, %%mm1            \n\t"\
937
        "movq   -1(%0), %%mm2       \n\t"\
938
        "movq    2(%0), %%mm4       \n\t"\
939
        "movq %%mm2, %%mm3          \n\t"\
940
        "movq %%mm4, %%mm5          \n\t"\
941
        "punpcklbw %%mm7, %%mm2     \n\t"\
942
        "punpckhbw %%mm7, %%mm3     \n\t"\
943
        "punpcklbw %%mm7, %%mm4     \n\t"\
944
        "punpckhbw %%mm7, %%mm5     \n\t"\
945
        "paddw %%mm4, %%mm2         \n\t"\
946
        "paddw %%mm3, %%mm5         \n\t"\
947
        "psubw %%mm2, %%mm0         \n\t"\
948
        "psubw %%mm5, %%mm1         \n\t"\
949
        "pmullw %%mm6, %%mm0        \n\t"\
950
        "pmullw %%mm6, %%mm1        \n\t"\
951
        "movd   -2(%0), %%mm2       \n\t"\
952
        "movd    7(%0), %%mm5       \n\t"\
953
        "punpcklbw %%mm7, %%mm2     \n\t"\
954
        "punpcklbw %%mm7, %%mm5     \n\t"\
955
        "paddw %%mm3, %%mm2         \n\t"\
956
        "paddw %%mm5, %%mm4         \n\t"\
957
        "movq %5, %%mm5             \n\t"\
958
        "paddw %%mm5, %%mm2         \n\t"\
959
        "paddw %%mm5, %%mm4         \n\t"\
960
        "paddw %%mm2, %%mm0         \n\t"\
961
        "paddw %%mm4, %%mm1         \n\t"\
962
        "psraw $5, %%mm0            \n\t"\
963
        "psraw $5, %%mm1            \n\t"\
964
        "movq (%2), %%mm4           \n\t"\
965
        "packuswb %%mm1, %%mm0      \n\t"\
966
        PAVGB" %%mm4, %%mm0         \n\t"\
967
        OP(%%mm0, (%1),%%mm5, q)\
968
        "add %4, %0                 \n\t"\
969
        "add %4, %1                 \n\t"\
970
        "add %3, %2                 \n\t"\
971
        : "+a"(src), "+c"(dst), "+d"(src2)\
972
        : "D"((long)src2Stride), "S"((long)dstStride),\
973
          "m"(ff_pw_16)\
974
        : "memory"\
975
    );\
976
    }while(--h);\
977
}\
978
\
979
static av_noinline void OPNAME ## h264_qpel8or16_v_lowpass_ ## MMX(uint8_t *dst, uint8_t *src, int dstStride, int srcStride, int h){\
980
    int w= 2;\
981
    src -= 2*srcStride;\
982
    \
983
    while(w--){\
984
      asm volatile(\
985
        "pxor %%mm7, %%mm7          \n\t"\
986
        "movd (%0), %%mm0           \n\t"\
987
        "add %2, %0                 \n\t"\
988
        "movd (%0), %%mm1           \n\t"\
989
        "add %2, %0                 \n\t"\
990
        "movd (%0), %%mm2           \n\t"\
991
        "add %2, %0                 \n\t"\
992
        "movd (%0), %%mm3           \n\t"\
993
        "add %2, %0                 \n\t"\
994
        "movd (%0), %%mm4           \n\t"\
995
        "add %2, %0                 \n\t"\
996
        "punpcklbw %%mm7, %%mm0     \n\t"\
997
        "punpcklbw %%mm7, %%mm1     \n\t"\
998
        "punpcklbw %%mm7, %%mm2     \n\t"\
999
        "punpcklbw %%mm7, %%mm3     \n\t"\
1000
        "punpcklbw %%mm7, %%mm4     \n\t"\
1001
        QPEL_H264V(%%mm0, %%mm1, %%mm2, %%mm3, %%mm4, %%mm5, OP)\
1002
        QPEL_H264V(%%mm1, %%mm2, %%mm3, %%mm4, %%mm5, %%mm0, OP)\
1003
        QPEL_H264V(%%mm2, %%mm3, %%mm4, %%mm5, %%mm0, %%mm1, OP)\
1004
        QPEL_H264V(%%mm3, %%mm4, %%mm5, %%mm0, %%mm1, %%mm2, OP)\
1005
        QPEL_H264V(%%mm4, %%mm5, %%mm0, %%mm1, %%mm2, %%mm3, OP)\
1006
        QPEL_H264V(%%mm5, %%mm0, %%mm1, %%mm2, %%mm3, %%mm4, OP)\
1007
        QPEL_H264V(%%mm0, %%mm1, %%mm2, %%mm3, %%mm4, %%mm5, OP)\
1008
        QPEL_H264V(%%mm1, %%mm2, %%mm3, %%mm4, %%mm5, %%mm0, OP)\
1009
         \
1010
        : "+a"(src), "+c"(dst)\
1011
        : "S"((long)srcStride), "D"((long)dstStride), "m"(ff_pw_5), "m"(ff_pw_16)\
1012
        : "memory"\
1013
     );\
1014
     if(h==16){\
1015
        asm volatile(\
1016
            QPEL_H264V(%%mm2, %%mm3, %%mm4, %%mm5, %%mm0, %%mm1, OP)\
1017
            QPEL_H264V(%%mm3, %%mm4, %%mm5, %%mm0, %%mm1, %%mm2, OP)\
1018
            QPEL_H264V(%%mm4, %%mm5, %%mm0, %%mm1, %%mm2, %%mm3, OP)\
1019
            QPEL_H264V(%%mm5, %%mm0, %%mm1, %%mm2, %%mm3, %%mm4, OP)\
1020
            QPEL_H264V(%%mm0, %%mm1, %%mm2, %%mm3, %%mm4, %%mm5, OP)\
1021
            QPEL_H264V(%%mm1, %%mm2, %%mm3, %%mm4, %%mm5, %%mm0, OP)\
1022
            QPEL_H264V(%%mm2, %%mm3, %%mm4, %%mm5, %%mm0, %%mm1, OP)\
1023
            QPEL_H264V(%%mm3, %%mm4, %%mm5, %%mm0, %%mm1, %%mm2, OP)\
1024
            \
1025
           : "+a"(src), "+c"(dst)\
1026
           : "S"((long)srcStride), "D"((long)dstStride), "m"(ff_pw_5), "m"(ff_pw_16)\
1027
           : "memory"\
1028
        );\
1029
     }\
1030
     src += 4-(h+5)*srcStride;\
1031
     dst += 4-h*dstStride;\
1032
   }\
1033
}\
1034
static av_noinline void OPNAME ## h264_qpel8or16_hv_lowpass_ ## MMX(uint8_t *dst, int16_t *tmp, uint8_t *src, int dstStride, int tmpStride, int srcStride, int size){\
1035
    int h = size;\
1036
    int w = (size+8)>>2;\
1037
    src -= 2*srcStride+2;\
1038
    while(w--){\
1039
        asm volatile(\
1040
            "pxor %%mm7, %%mm7      \n\t"\
1041
            "movd (%0), %%mm0       \n\t"\
1042
            "add %2, %0             \n\t"\
1043
            "movd (%0), %%mm1       \n\t"\
1044
            "add %2, %0             \n\t"\
1045
            "movd (%0), %%mm2       \n\t"\
1046
            "add %2, %0             \n\t"\
1047
            "movd (%0), %%mm3       \n\t"\
1048
            "add %2, %0             \n\t"\
1049
            "movd (%0), %%mm4       \n\t"\
1050
            "add %2, %0             \n\t"\
1051
            "punpcklbw %%mm7, %%mm0 \n\t"\
1052
            "punpcklbw %%mm7, %%mm1 \n\t"\
1053
            "punpcklbw %%mm7, %%mm2 \n\t"\
1054
            "punpcklbw %%mm7, %%mm3 \n\t"\
1055
            "punpcklbw %%mm7, %%mm4 \n\t"\
1056
            QPEL_H264HV(%%mm0, %%mm1, %%mm2, %%mm3, %%mm4, %%mm5, 0*48)\
1057
            QPEL_H264HV(%%mm1, %%mm2, %%mm3, %%mm4, %%mm5, %%mm0, 1*48)\
1058
            QPEL_H264HV(%%mm2, %%mm3, %%mm4, %%mm5, %%mm0, %%mm1, 2*48)\
1059
            QPEL_H264HV(%%mm3, %%mm4, %%mm5, %%mm0, %%mm1, %%mm2, 3*48)\
1060
            QPEL_H264HV(%%mm4, %%mm5, %%mm0, %%mm1, %%mm2, %%mm3, 4*48)\
1061
            QPEL_H264HV(%%mm5, %%mm0, %%mm1, %%mm2, %%mm3, %%mm4, 5*48)\
1062
            QPEL_H264HV(%%mm0, %%mm1, %%mm2, %%mm3, %%mm4, %%mm5, 6*48)\
1063
            QPEL_H264HV(%%mm1, %%mm2, %%mm3, %%mm4, %%mm5, %%mm0, 7*48)\
1064
            : "+a"(src)\
1065
            : "c"(tmp), "S"((long)srcStride), "m"(ff_pw_5)\
1066
            : "memory"\
1067
        );\
1068
        if(size==16){\
1069
            asm volatile(\
1070
                QPEL_H264HV(%%mm2, %%mm3, %%mm4, %%mm5, %%mm0, %%mm1,  8*48)\
1071
                QPEL_H264HV(%%mm3, %%mm4, %%mm5, %%mm0, %%mm1, %%mm2,  9*48)\
1072
                QPEL_H264HV(%%mm4, %%mm5, %%mm0, %%mm1, %%mm2, %%mm3, 10*48)\
1073
                QPEL_H264HV(%%mm5, %%mm0, %%mm1, %%mm2, %%mm3, %%mm4, 11*48)\
1074
                QPEL_H264HV(%%mm0, %%mm1, %%mm2, %%mm3, %%mm4, %%mm5, 12*48)\
1075
                QPEL_H264HV(%%mm1, %%mm2, %%mm3, %%mm4, %%mm5, %%mm0, 13*48)\
1076
                QPEL_H264HV(%%mm2, %%mm3, %%mm4, %%mm5, %%mm0, %%mm1, 14*48)\
1077
                QPEL_H264HV(%%mm3, %%mm4, %%mm5, %%mm0, %%mm1, %%mm2, 15*48)\
1078
                : "+a"(src)\
1079
                : "c"(tmp), "S"((long)srcStride), "m"(ff_pw_5)\
1080
                : "memory"\
1081
            );\
1082
        }\
1083
        tmp += 4;\
1084
        src += 4 - (size+5)*srcStride;\
1085
    }\
1086
    tmp -= size+8;\
1087
    w = size>>4;\
1088
    do{\
1089
    h = size;\
1090
    asm volatile(\
1091
        "movq %4, %%mm6             \n\t"\
1092
        "1:                         \n\t"\
1093
        "movq     (%0), %%mm0       \n\t"\
1094
        "movq    8(%0), %%mm3       \n\t"\
1095
        "movq    2(%0), %%mm1       \n\t"\
1096
        "movq   10(%0), %%mm4       \n\t"\
1097
        "paddw   %%mm4, %%mm0       \n\t"\
1098
        "paddw   %%mm3, %%mm1       \n\t"\
1099
        "paddw  18(%0), %%mm3       \n\t"\
1100
        "paddw  16(%0), %%mm4       \n\t"\
1101
        "movq    4(%0), %%mm2       \n\t"\
1102
        "movq   12(%0), %%mm5       \n\t"\
1103
        "paddw   6(%0), %%mm2       \n\t"\
1104
        "paddw  14(%0), %%mm5       \n\t"\
1105
        "psubw %%mm1, %%mm0         \n\t"\
1106
        "psubw %%mm4, %%mm3         \n\t"\
1107
        "psraw $2, %%mm0            \n\t"\
1108
        "psraw $2, %%mm3            \n\t"\
1109
        "psubw %%mm1, %%mm0         \n\t"\
1110
        "psubw %%mm4, %%mm3         \n\t"\
1111
        "paddsw %%mm2, %%mm0        \n\t"\
1112
        "paddsw %%mm5, %%mm3        \n\t"\
1113
        "psraw $2, %%mm0            \n\t"\
1114
        "psraw $2, %%mm3            \n\t"\
1115
        "paddw %%mm6, %%mm2         \n\t"\
1116
        "paddw %%mm6, %%mm5         \n\t"\
1117
        "paddw %%mm2, %%mm0         \n\t"\
1118
        "paddw %%mm5, %%mm3         \n\t"\
1119
        "psraw $6, %%mm0            \n\t"\
1120
        "psraw $6, %%mm3            \n\t"\
1121
        "packuswb %%mm3, %%mm0      \n\t"\
1122
        OP(%%mm0, (%1),%%mm7, q)\
1123
        "add $48, %0                \n\t"\
1124
        "add %3, %1                 \n\t"\
1125
        "decl %2                    \n\t"\
1126
        " jnz 1b                    \n\t"\
1127
        : "+a"(tmp), "+c"(dst), "+m"(h)\
1128
        : "S"((long)dstStride), "m"(ff_pw_32)\
1129
        : "memory"\
1130
    );\
1131
    tmp += 8 - size*24;\
1132
    dst += 8 - size*dstStride;\
1133
    }while(w--);\
1134
}\
1135
\
1136
static void OPNAME ## h264_qpel8_v_lowpass_ ## MMX(uint8_t *dst, uint8_t *src, int dstStride, int srcStride){\
1137
    OPNAME ## h264_qpel8or16_v_lowpass_ ## MMX(dst  , src  , dstStride, srcStride, 8);\
1138
}\
1139
static av_noinline void OPNAME ## h264_qpel16_v_lowpass_ ## MMX(uint8_t *dst, uint8_t *src, int dstStride, int srcStride){\
1140
    OPNAME ## h264_qpel8or16_v_lowpass_ ## MMX(dst  , src  , dstStride, srcStride, 16);\
1141
    OPNAME ## h264_qpel8or16_v_lowpass_ ## MMX(dst+8, src+8, dstStride, srcStride, 16);\
1142
}\
1143
\
1144
static av_noinline void OPNAME ## h264_qpel16_h_lowpass_ ## MMX(uint8_t *dst, uint8_t *src, int dstStride, int srcStride){\
1145
    OPNAME ## h264_qpel8_h_lowpass_ ## MMX(dst  , src  , dstStride, srcStride);\
1146
    OPNAME ## h264_qpel8_h_lowpass_ ## MMX(dst+8, src+8, dstStride, srcStride);\
1147
    src += 8*srcStride;\
1148
    dst += 8*dstStride;\
1149
    OPNAME ## h264_qpel8_h_lowpass_ ## MMX(dst  , src  , dstStride, srcStride);\
1150
    OPNAME ## h264_qpel8_h_lowpass_ ## MMX(dst+8, src+8, dstStride, srcStride);\
1151
}\
1152
\
1153
static av_noinline void OPNAME ## h264_qpel16_h_lowpass_l2_ ## MMX(uint8_t *dst, uint8_t *src, uint8_t *src2, int dstStride, int src2Stride){\
1154
    OPNAME ## h264_qpel8_h_lowpass_l2_ ## MMX(dst  , src  , src2  , dstStride, src2Stride);\
1155
    OPNAME ## h264_qpel8_h_lowpass_l2_ ## MMX(dst+8, src+8, src2+8, dstStride, src2Stride);\
1156
    src += 8*dstStride;\
1157
    dst += 8*dstStride;\
1158
    src2 += 8*src2Stride;\
1159
    OPNAME ## h264_qpel8_h_lowpass_l2_ ## MMX(dst  , src  , src2  , dstStride, src2Stride);\
1160
    OPNAME ## h264_qpel8_h_lowpass_l2_ ## MMX(dst+8, src+8, src2+8, dstStride, src2Stride);\
1161
}\
1162
\
1163
static void OPNAME ## h264_qpel8_hv_lowpass_ ## MMX(uint8_t *dst, int16_t *tmp, uint8_t *src, int dstStride, int tmpStride, int srcStride){\
1164
    OPNAME ## h264_qpel8or16_hv_lowpass_ ## MMX(dst  , tmp  , src  , dstStride, tmpStride, srcStride, 8);\
1165
}\
1166
\
1167
static void OPNAME ## h264_qpel16_hv_lowpass_ ## MMX(uint8_t *dst, int16_t *tmp, uint8_t *src, int dstStride, int tmpStride, int srcStride){\
1168
    OPNAME ## h264_qpel8or16_hv_lowpass_ ## MMX(dst  , tmp  , src  , dstStride, tmpStride, srcStride, 16);\
1169
}\
1170
\
1171
static av_noinline void OPNAME ## pixels4_l2_shift5_ ## MMX(uint8_t *dst, int16_t *src16, uint8_t *src8, int dstStride, int src8Stride, int h)\
1172
{\
1173
    asm volatile(\
1174
        "movq       %5,  %%mm6          \n\t"\
1175
        "movq      (%1), %%mm0          \n\t"\
1176
        "movq    24(%1), %%mm1          \n\t"\
1177
        "paddw    %%mm6, %%mm0          \n\t"\
1178
        "paddw    %%mm6, %%mm1          \n\t"\
1179
        "psraw      $5,  %%mm0          \n\t"\
1180
        "psraw      $5,  %%mm1          \n\t"\
1181
        "packuswb %%mm0, %%mm0          \n\t"\
1182
        "packuswb %%mm1, %%mm1          \n\t"\
1183
        PAVGB"     (%0), %%mm0          \n\t"\
1184
        PAVGB"  (%0,%3), %%mm1          \n\t"\
1185
        OP(%%mm0, (%2),    %%mm4, d)\
1186
        OP(%%mm1, (%2,%4), %%mm5, d)\
1187
        "lea  (%0,%3,2), %0             \n\t"\
1188
        "lea  (%2,%4,2), %2             \n\t"\
1189
        "movq    48(%1), %%mm0          \n\t"\
1190
        "movq    72(%1), %%mm1          \n\t"\
1191
        "paddw    %%mm6, %%mm0          \n\t"\
1192
        "paddw    %%mm6, %%mm1          \n\t"\
1193
        "psraw      $5,  %%mm0          \n\t"\
1194
        "psraw      $5,  %%mm1          \n\t"\
1195
        "packuswb %%mm0, %%mm0          \n\t"\
1196
        "packuswb %%mm1, %%mm1          \n\t"\
1197
        PAVGB"     (%0), %%mm0          \n\t"\
1198
        PAVGB"  (%0,%3), %%mm1          \n\t"\
1199
        OP(%%mm0, (%2),    %%mm4, d)\
1200
        OP(%%mm1, (%2,%4), %%mm5, d)\
1201
        :"+a"(src8), "+c"(src16), "+d"(dst)\
1202
        :"S"((long)src8Stride), "D"((long)dstStride), "m"(ff_pw_16)\
1203
        :"memory");\
1204
}\
1205
static av_noinline void OPNAME ## pixels8_l2_shift5_ ## MMX(uint8_t *dst, int16_t *src16, uint8_t *src8, int dstStride, int src8Stride, int h)\
1206
{\
1207
    asm volatile(\
1208
        "movq       %0,  %%mm6          \n\t"\
1209
        ::"m"(ff_pw_16)\
1210
        );\
1211
    while(h--){\
1212
    asm volatile(\
1213
        "movq      (%1), %%mm0          \n\t"\
1214
        "movq     8(%1), %%mm1          \n\t"\
1215
        "paddw    %%mm6, %%mm0          \n\t"\
1216
        "paddw    %%mm6, %%mm1          \n\t"\
1217
        "psraw      $5,  %%mm0          \n\t"\
1218
        "psraw      $5,  %%mm1          \n\t"\
1219
        "packuswb %%mm1, %%mm0          \n\t"\
1220
        PAVGB"     (%0), %%mm0          \n\t"\
1221
        OP(%%mm0, (%2), %%mm5, q)\
1222
        ::"a"(src8), "c"(src16), "d"(dst)\
1223
        :"memory");\
1224
        src8 += src8Stride;\
1225
        src16 += 24;\
1226
        dst += dstStride;\
1227
    }\
1228
}\
1229
static void OPNAME ## pixels16_l2_shift5_ ## MMX(uint8_t *dst, int16_t *src16, uint8_t *src8, int dstStride, int src8Stride, int h)\
1230
{\
1231
    OPNAME ## pixels8_l2_shift5_ ## MMX(dst  , src16  , src8  , dstStride, src8Stride, h);\
1232
    OPNAME ## pixels8_l2_shift5_ ## MMX(dst+8, src16+8, src8+8, dstStride, src8Stride, h);\
1233
}\
1234

    
1235

    
1236
#define H264_MC(OPNAME, SIZE, MMX) \
1237
static void OPNAME ## h264_qpel ## SIZE ## _mc00_ ## MMX (uint8_t *dst, uint8_t *src, int stride){\
1238
    OPNAME ## pixels ## SIZE ## _mmx(dst, src, stride, SIZE);\
1239
}\
1240
\
1241
static void OPNAME ## h264_qpel ## SIZE ## _mc10_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
1242
    OPNAME ## h264_qpel ## SIZE ## _h_lowpass_l2_ ## MMX(dst, src, src, stride, stride);\
1243
}\
1244
\
1245
static void OPNAME ## h264_qpel ## SIZE ## _mc20_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
1246
    OPNAME ## h264_qpel ## SIZE ## _h_lowpass_ ## MMX(dst, src, stride, stride);\
1247
}\
1248
\
1249
static void OPNAME ## h264_qpel ## SIZE ## _mc30_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
1250
    OPNAME ## h264_qpel ## SIZE ## _h_lowpass_l2_ ## MMX(dst, src, src+1, stride, stride);\
1251
}\
1252
\
1253
static void OPNAME ## h264_qpel ## SIZE ## _mc01_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
1254
    uint64_t temp[SIZE*SIZE/8];\
1255
    uint8_t * const half= (uint8_t*)temp;\
1256
    put_h264_qpel ## SIZE ## _v_lowpass_ ## MMX(half, src, SIZE, stride);\
1257
    OPNAME ## pixels ## SIZE ## _l2_ ## MMX(dst, src, half, stride, stride, SIZE);\
1258
}\
1259
\
1260
static void OPNAME ## h264_qpel ## SIZE ## _mc02_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
1261
    OPNAME ## h264_qpel ## SIZE ## _v_lowpass_ ## MMX(dst, src, stride, stride);\
1262
}\
1263
\
1264
static void OPNAME ## h264_qpel ## SIZE ## _mc03_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
1265
    uint64_t temp[SIZE*SIZE/8];\
1266
    uint8_t * const half= (uint8_t*)temp;\
1267
    put_h264_qpel ## SIZE ## _v_lowpass_ ## MMX(half, src, SIZE, stride);\
1268
    OPNAME ## pixels ## SIZE ## _l2_ ## MMX(dst, src+stride, half, stride, stride, SIZE);\
1269
}\
1270
\
1271
static void OPNAME ## h264_qpel ## SIZE ## _mc11_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
1272
    uint64_t temp[SIZE*SIZE/8];\
1273
    uint8_t * const halfV= (uint8_t*)temp;\
1274
    put_h264_qpel ## SIZE ## _v_lowpass_ ## MMX(halfV, src, SIZE, stride);\
1275
    OPNAME ## h264_qpel ## SIZE ## _h_lowpass_l2_ ## MMX(dst, src, halfV, stride, SIZE);\
1276
}\
1277
\
1278
static void OPNAME ## h264_qpel ## SIZE ## _mc31_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
1279
    uint64_t temp[SIZE*SIZE/8];\
1280
    uint8_t * const halfV= (uint8_t*)temp;\
1281
    put_h264_qpel ## SIZE ## _v_lowpass_ ## MMX(halfV, src+1, SIZE, stride);\
1282
    OPNAME ## h264_qpel ## SIZE ## _h_lowpass_l2_ ## MMX(dst, src, halfV, stride, SIZE);\
1283
}\
1284
\
1285
static void OPNAME ## h264_qpel ## SIZE ## _mc13_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
1286
    uint64_t temp[SIZE*SIZE/8];\
1287
    uint8_t * const halfV= (uint8_t*)temp;\
1288
    put_h264_qpel ## SIZE ## _v_lowpass_ ## MMX(halfV, src, SIZE, stride);\
1289
    OPNAME ## h264_qpel ## SIZE ## _h_lowpass_l2_ ## MMX(dst, src+stride, halfV, stride, SIZE);\
1290
}\
1291
\
1292
static void OPNAME ## h264_qpel ## SIZE ## _mc33_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
1293
    uint64_t temp[SIZE*SIZE/8];\
1294
    uint8_t * const halfV= (uint8_t*)temp;\
1295
    put_h264_qpel ## SIZE ## _v_lowpass_ ## MMX(halfV, src+1, SIZE, stride);\
1296
    OPNAME ## h264_qpel ## SIZE ## _h_lowpass_l2_ ## MMX(dst, src+stride, halfV, stride, SIZE);\
1297
}\
1298
\
1299
static void OPNAME ## h264_qpel ## SIZE ## _mc22_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
1300
    uint64_t temp[SIZE*(SIZE<8?12:24)/4];\
1301
    int16_t * const tmp= (int16_t*)temp;\
1302
    OPNAME ## h264_qpel ## SIZE ## _hv_lowpass_ ## MMX(dst, tmp, src, stride, SIZE, stride);\
1303
}\
1304
\
1305
static void OPNAME ## h264_qpel ## SIZE ## _mc21_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
1306
    uint64_t temp[SIZE*(SIZE<8?12:24)/4 + SIZE*SIZE/8];\
1307
    uint8_t * const halfHV= (uint8_t*)temp;\
1308
    int16_t * const tmp= ((int16_t*)temp) + SIZE*SIZE/2;\
1309
    assert(((int)temp & 7) == 0);\
1310
    put_h264_qpel ## SIZE ## _hv_lowpass_ ## MMX(halfHV, tmp, src, SIZE, SIZE, stride);\
1311
    OPNAME ## h264_qpel ## SIZE ## _h_lowpass_l2_ ## MMX(dst, src, halfHV, stride, SIZE);\
1312
}\
1313
\
1314
static void OPNAME ## h264_qpel ## SIZE ## _mc23_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
1315
    uint64_t temp[SIZE*(SIZE<8?12:24)/4 + SIZE*SIZE/8];\
1316
    uint8_t * const halfHV= (uint8_t*)temp;\
1317
    int16_t * const tmp= ((int16_t*)temp) + SIZE*SIZE/2;\
1318
    assert(((int)temp & 7) == 0);\
1319
    put_h264_qpel ## SIZE ## _hv_lowpass_ ## MMX(halfHV, tmp, src, SIZE, SIZE, stride);\
1320
    OPNAME ## h264_qpel ## SIZE ## _h_lowpass_l2_ ## MMX(dst, src+stride, halfHV, stride, SIZE);\
1321
}\
1322
\
1323
static void OPNAME ## h264_qpel ## SIZE ## _mc12_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
1324
    uint64_t temp[SIZE*(SIZE<8?12:24)/4 + SIZE*SIZE/8];\
1325
    int16_t * const halfV= ((int16_t*)temp) + SIZE*SIZE/2;\
1326
    uint8_t * const halfHV= ((uint8_t*)temp);\
1327
    assert(((int)temp & 7) == 0);\
1328
    put_h264_qpel ## SIZE ## _hv_lowpass_ ## MMX(halfHV, halfV, src, SIZE, SIZE, stride);\
1329
    OPNAME ## pixels ## SIZE ## _l2_shift5_ ## MMX(dst, halfV+2, halfHV, stride, SIZE, SIZE);\
1330
}\
1331
\
1332
static void OPNAME ## h264_qpel ## SIZE ## _mc32_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
1333
    uint64_t temp[SIZE*(SIZE<8?12:24)/4 + SIZE*SIZE/8];\
1334
    int16_t * const halfV= ((int16_t*)temp) + SIZE*SIZE/2;\
1335
    uint8_t * const halfHV= ((uint8_t*)temp);\
1336
    assert(((int)temp & 7) == 0);\
1337
    put_h264_qpel ## SIZE ## _hv_lowpass_ ## MMX(halfHV, halfV, src, SIZE, SIZE, stride);\
1338
    OPNAME ## pixels ## SIZE ## _l2_shift5_ ## MMX(dst, halfV+3, halfHV, stride, SIZE, SIZE);\
1339
}\
1340

    
1341

    
1342
#define AVG_3DNOW_OP(a,b,temp, size) \
1343
"mov" #size " " #b ", " #temp "   \n\t"\
1344
"pavgusb " #temp ", " #a "        \n\t"\
1345
"mov" #size " " #a ", " #b "      \n\t"
1346
#define AVG_MMX2_OP(a,b,temp, size) \
1347
"mov" #size " " #b ", " #temp "   \n\t"\
1348
"pavgb " #temp ", " #a "          \n\t"\
1349
"mov" #size " " #a ", " #b "      \n\t"
1350

    
1351
#define PAVGB "pavgusb"
1352
QPEL_H264(put_,       PUT_OP, 3dnow)
1353
QPEL_H264(avg_, AVG_3DNOW_OP, 3dnow)
1354
#undef PAVGB
1355
#define PAVGB "pavgb"
1356
QPEL_H264(put_,       PUT_OP, mmx2)
1357
QPEL_H264(avg_,  AVG_MMX2_OP, mmx2)
1358
#undef PAVGB
1359

    
1360
H264_MC(put_, 4, 3dnow)
1361
H264_MC(put_, 8, 3dnow)
1362
H264_MC(put_, 16,3dnow)
1363
H264_MC(avg_, 4, 3dnow)
1364
H264_MC(avg_, 8, 3dnow)
1365
H264_MC(avg_, 16,3dnow)
1366
H264_MC(put_, 4, mmx2)
1367
H264_MC(put_, 8, mmx2)
1368
H264_MC(put_, 16,mmx2)
1369
H264_MC(avg_, 4, mmx2)
1370
H264_MC(avg_, 8, mmx2)
1371
H264_MC(avg_, 16,mmx2)
1372

    
1373

    
1374
#define H264_CHROMA_OP(S,D)
1375
#define H264_CHROMA_OP4(S,D,T)
1376
#define H264_CHROMA_MC8_TMPL put_h264_chroma_mc8_mmx
1377
#define H264_CHROMA_MC4_TMPL put_h264_chroma_mc4_mmx
1378
#define H264_CHROMA_MC2_TMPL put_h264_chroma_mc2_mmx2
1379
#define H264_CHROMA_MC8_MV0 put_pixels8_mmx
1380
#include "dsputil_h264_template_mmx.c"
1381

    
1382
static void put_h264_chroma_mc8_mmx_rnd(uint8_t *dst/*align 8*/, uint8_t *src/*align 1*/, int stride, int h, int x, int y)
1383
{
1384
    put_h264_chroma_mc8_mmx(dst, src, stride, h, x, y, 1);
1385
}
1386
static void put_h264_chroma_mc8_mmx_nornd(uint8_t *dst/*align 8*/, uint8_t *src/*align 1*/, int stride, int h, int x, int y)
1387
{
1388
    put_h264_chroma_mc8_mmx(dst, src, stride, h, x, y, 0);
1389
}
1390

    
1391
#undef H264_CHROMA_OP
1392
#undef H264_CHROMA_OP4
1393
#undef H264_CHROMA_MC8_TMPL
1394
#undef H264_CHROMA_MC4_TMPL
1395
#undef H264_CHROMA_MC2_TMPL
1396
#undef H264_CHROMA_MC8_MV0
1397

    
1398
#define H264_CHROMA_OP(S,D) "pavgb " #S ", " #D " \n\t"
1399
#define H264_CHROMA_OP4(S,D,T) "movd  " #S ", " #T " \n\t"\
1400
                               "pavgb " #T ", " #D " \n\t"
1401
#define H264_CHROMA_MC8_TMPL avg_h264_chroma_mc8_mmx2
1402
#define H264_CHROMA_MC4_TMPL avg_h264_chroma_mc4_mmx2
1403
#define H264_CHROMA_MC2_TMPL avg_h264_chroma_mc2_mmx2
1404
#define H264_CHROMA_MC8_MV0 avg_pixels8_mmx2
1405
#include "dsputil_h264_template_mmx.c"
1406
static void avg_h264_chroma_mc8_mmx2_rnd(uint8_t *dst/*align 8*/, uint8_t *src/*align 1*/, int stride, int h, int x, int y)
1407
{
1408
    avg_h264_chroma_mc8_mmx2(dst, src, stride, h, x, y, 1);
1409
}
1410
#undef H264_CHROMA_OP
1411
#undef H264_CHROMA_OP4
1412
#undef H264_CHROMA_MC8_TMPL
1413
#undef H264_CHROMA_MC4_TMPL
1414
#undef H264_CHROMA_MC2_TMPL
1415
#undef H264_CHROMA_MC8_MV0
1416

    
1417
#define H264_CHROMA_OP(S,D) "pavgusb " #S ", " #D " \n\t"
1418
#define H264_CHROMA_OP4(S,D,T) "movd " #S ", " #T " \n\t"\
1419
                               "pavgusb " #T ", " #D " \n\t"
1420
#define H264_CHROMA_MC8_TMPL avg_h264_chroma_mc8_3dnow
1421
#define H264_CHROMA_MC4_TMPL avg_h264_chroma_mc4_3dnow
1422
#define H264_CHROMA_MC8_MV0 avg_pixels8_3dnow
1423
#include "dsputil_h264_template_mmx.c"
1424
static void avg_h264_chroma_mc8_3dnow_rnd(uint8_t *dst/*align 8*/, uint8_t *src/*align 1*/, int stride, int h, int x, int y)
1425
{
1426
    avg_h264_chroma_mc8_3dnow(dst, src, stride, h, x, y, 1);
1427
}
1428
#undef H264_CHROMA_OP
1429
#undef H264_CHROMA_OP4
1430
#undef H264_CHROMA_MC8_TMPL
1431
#undef H264_CHROMA_MC4_TMPL
1432
#undef H264_CHROMA_MC8_MV0
1433

    
1434
/***********************************/
1435
/* weighted prediction */
1436

    
1437
static inline void ff_h264_weight_WxH_mmx2(uint8_t *dst, int stride, int log2_denom, int weight, int offset, int w, int h)
1438
{
1439
    int x, y;
1440
    offset <<= log2_denom;
1441
    offset += (1 << log2_denom) >> 1;
1442
    asm volatile(
1443
        "movd    %0, %%mm4        \n\t"
1444
        "movd    %1, %%mm5        \n\t"
1445
        "movd    %2, %%mm6        \n\t"
1446
        "pshufw  $0, %%mm4, %%mm4 \n\t"
1447
        "pshufw  $0, %%mm5, %%mm5 \n\t"
1448
        "pxor    %%mm7, %%mm7     \n\t"
1449
        :: "g"(weight), "g"(offset), "g"(log2_denom)
1450
    );
1451
    for(y=0; y<h; y+=2){
1452
        for(x=0; x<w; x+=4){
1453
            asm volatile(
1454
                "movd      %0,    %%mm0 \n\t"
1455
                "movd      %1,    %%mm1 \n\t"
1456
                "punpcklbw %%mm7, %%mm0 \n\t"
1457
                "punpcklbw %%mm7, %%mm1 \n\t"
1458
                "pmullw    %%mm4, %%mm0 \n\t"
1459
                "pmullw    %%mm4, %%mm1 \n\t"
1460
                "paddsw    %%mm5, %%mm0 \n\t"
1461
                "paddsw    %%mm5, %%mm1 \n\t"
1462
                "psraw     %%mm6, %%mm0 \n\t"
1463
                "psraw     %%mm6, %%mm1 \n\t"
1464
                "packuswb  %%mm7, %%mm0 \n\t"
1465
                "packuswb  %%mm7, %%mm1 \n\t"
1466
                "movd      %%mm0, %0    \n\t"
1467
                "movd      %%mm1, %1    \n\t"
1468
                : "+m"(*(uint32_t*)(dst+x)),
1469
                  "+m"(*(uint32_t*)(dst+x+stride))
1470
            );
1471
        }
1472
        dst += 2*stride;
1473
    }
1474
}
1475

    
1476
static inline void ff_h264_biweight_WxH_mmx2(uint8_t *dst, uint8_t *src, int stride, int log2_denom, int weightd, int weights, int offset, int w, int h)
1477
{
1478
    int x, y;
1479
    offset = ((offset + 1) | 1) << log2_denom;
1480
    asm volatile(
1481
        "movd    %0, %%mm3        \n\t"
1482
        "movd    %1, %%mm4        \n\t"
1483
        "movd    %2, %%mm5        \n\t"
1484
        "movd    %3, %%mm6        \n\t"
1485
        "pshufw  $0, %%mm3, %%mm3 \n\t"
1486
        "pshufw  $0, %%mm4, %%mm4 \n\t"
1487
        "pshufw  $0, %%mm5, %%mm5 \n\t"
1488
        "pxor    %%mm7, %%mm7     \n\t"
1489
        :: "g"(weightd), "g"(weights), "g"(offset), "g"(log2_denom+1)
1490
    );
1491
    for(y=0; y<h; y++){
1492
        for(x=0; x<w; x+=4){
1493
            asm volatile(
1494
                "movd      %0,    %%mm0 \n\t"
1495
                "movd      %1,    %%mm1 \n\t"
1496
                "punpcklbw %%mm7, %%mm0 \n\t"
1497
                "punpcklbw %%mm7, %%mm1 \n\t"
1498
                "pmullw    %%mm3, %%mm0 \n\t"
1499
                "pmullw    %%mm4, %%mm1 \n\t"
1500
                "paddsw    %%mm1, %%mm0 \n\t"
1501
                "paddsw    %%mm5, %%mm0 \n\t"
1502
                "psraw     %%mm6, %%mm0 \n\t"
1503
                "packuswb  %%mm0, %%mm0 \n\t"
1504
                "movd      %%mm0, %0    \n\t"
1505
                : "+m"(*(uint32_t*)(dst+x))
1506
                :  "m"(*(uint32_t*)(src+x))
1507
            );
1508
        }
1509
        src += stride;
1510
        dst += stride;
1511
    }
1512
}
1513

    
1514
#define H264_WEIGHT(W,H) \
1515
static void ff_h264_biweight_ ## W ## x ## H ## _mmx2(uint8_t *dst, uint8_t *src, int stride, int log2_denom, int weightd, int weights, int offset){ \
1516
    ff_h264_biweight_WxH_mmx2(dst, src, stride, log2_denom, weightd, weights, offset, W, H); \
1517
} \
1518
static void ff_h264_weight_ ## W ## x ## H ## _mmx2(uint8_t *dst, int stride, int log2_denom, int weight, int offset){ \
1519
    ff_h264_weight_WxH_mmx2(dst, stride, log2_denom, weight, offset, W, H); \
1520
}
1521

    
1522
H264_WEIGHT(16,16)
1523
H264_WEIGHT(16, 8)
1524
H264_WEIGHT( 8,16)
1525
H264_WEIGHT( 8, 8)
1526
H264_WEIGHT( 8, 4)
1527
H264_WEIGHT( 4, 8)
1528
H264_WEIGHT( 4, 4)
1529
H264_WEIGHT( 4, 2)
1530