2 * Optimized for ia32 CPUs by Nick Kurshev <nickols_k@mail.ru>
3 * h263, mpeg1, mpeg2 dequantizer & draw_edges by Michael Niedermayer <michaelni@gmx.at>
5 * This file is part of FFmpeg.
7 * FFmpeg is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Lesser General Public
9 * License as published by the Free Software Foundation; either
10 * version 2.1 of the License, or (at your option) any later version.
12 * FFmpeg 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 GNU
15 * Lesser General Public License for more details.
17 * You should have received a copy of the GNU Lesser General Public
18 * License along with FFmpeg; if not, write to the Free Software
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
22 #include "libavutil/attributes.h"
23 #include "libavutil/cpu.h"
24 #include "libavutil/x86/asm.h"
25 #include "libavutil/x86/cpu.h"
26 #include "libavcodec/avcodec.h"
27 #include "libavcodec/mpegvideo.h"
28 #include "dsputil_x86.h"
32 static void dct_unquantize_h263_intra_mmx(MpegEncContext *s,
33 int16_t *block, int n, int qscale)
35 x86_reg level, qmul, qadd, nCoeffs;
39 av_assert2(s->block_last_index[n]>=0 || s->h263_aic);
43 level = block[0] * s->y_dc_scale;
45 level = block[0] * s->c_dc_scale;
46 qadd = (qscale - 1) | 1;
54 nCoeffs= s->inter_scantable.raster_end[ s->block_last_index[n] ];
57 "movd %1, %%mm6 \n\t" //qmul
58 "packssdw %%mm6, %%mm6 \n\t"
59 "packssdw %%mm6, %%mm6 \n\t"
60 "movd %2, %%mm5 \n\t" //qadd
61 "pxor %%mm7, %%mm7 \n\t"
62 "packssdw %%mm5, %%mm5 \n\t"
63 "packssdw %%mm5, %%mm5 \n\t"
64 "psubw %%mm5, %%mm7 \n\t"
65 "pxor %%mm4, %%mm4 \n\t"
68 "movq (%0, %3), %%mm0 \n\t"
69 "movq 8(%0, %3), %%mm1 \n\t"
71 "pmullw %%mm6, %%mm0 \n\t"
72 "pmullw %%mm6, %%mm1 \n\t"
74 "movq (%0, %3), %%mm2 \n\t"
75 "movq 8(%0, %3), %%mm3 \n\t"
77 "pcmpgtw %%mm4, %%mm2 \n\t" // block[i] < 0 ? -1 : 0
78 "pcmpgtw %%mm4, %%mm3 \n\t" // block[i] < 0 ? -1 : 0
80 "pxor %%mm2, %%mm0 \n\t"
81 "pxor %%mm3, %%mm1 \n\t"
83 "paddw %%mm7, %%mm0 \n\t"
84 "paddw %%mm7, %%mm1 \n\t"
86 "pxor %%mm0, %%mm2 \n\t"
87 "pxor %%mm1, %%mm3 \n\t"
89 "pcmpeqw %%mm7, %%mm0 \n\t" // block[i] == 0 ? -1 : 0
90 "pcmpeqw %%mm7, %%mm1 \n\t" // block[i] == 0 ? -1 : 0
92 "pandn %%mm2, %%mm0 \n\t"
93 "pandn %%mm3, %%mm1 \n\t"
95 "movq %%mm0, (%0, %3) \n\t"
96 "movq %%mm1, 8(%0, %3) \n\t"
100 ::"r" (block+nCoeffs), "rm"(qmul), "rm" (qadd), "r" (2*(-nCoeffs))
107 static void dct_unquantize_h263_inter_mmx(MpegEncContext *s,
108 int16_t *block, int n, int qscale)
110 x86_reg qmul, qadd, nCoeffs;
113 qadd = (qscale - 1) | 1;
115 av_assert2(s->block_last_index[n]>=0 || s->h263_aic);
117 nCoeffs= s->inter_scantable.raster_end[ s->block_last_index[n] ];
120 "movd %1, %%mm6 \n\t" //qmul
121 "packssdw %%mm6, %%mm6 \n\t"
122 "packssdw %%mm6, %%mm6 \n\t"
123 "movd %2, %%mm5 \n\t" //qadd
124 "pxor %%mm7, %%mm7 \n\t"
125 "packssdw %%mm5, %%mm5 \n\t"
126 "packssdw %%mm5, %%mm5 \n\t"
127 "psubw %%mm5, %%mm7 \n\t"
128 "pxor %%mm4, %%mm4 \n\t"
131 "movq (%0, %3), %%mm0 \n\t"
132 "movq 8(%0, %3), %%mm1 \n\t"
134 "pmullw %%mm6, %%mm0 \n\t"
135 "pmullw %%mm6, %%mm1 \n\t"
137 "movq (%0, %3), %%mm2 \n\t"
138 "movq 8(%0, %3), %%mm3 \n\t"
140 "pcmpgtw %%mm4, %%mm2 \n\t" // block[i] < 0 ? -1 : 0
141 "pcmpgtw %%mm4, %%mm3 \n\t" // block[i] < 0 ? -1 : 0
143 "pxor %%mm2, %%mm0 \n\t"
144 "pxor %%mm3, %%mm1 \n\t"
146 "paddw %%mm7, %%mm0 \n\t"
147 "paddw %%mm7, %%mm1 \n\t"
149 "pxor %%mm0, %%mm2 \n\t"
150 "pxor %%mm1, %%mm3 \n\t"
152 "pcmpeqw %%mm7, %%mm0 \n\t" // block[i] == 0 ? -1 : 0
153 "pcmpeqw %%mm7, %%mm1 \n\t" // block[i] == 0 ? -1 : 0
155 "pandn %%mm2, %%mm0 \n\t"
156 "pandn %%mm3, %%mm1 \n\t"
158 "movq %%mm0, (%0, %3) \n\t"
159 "movq %%mm1, 8(%0, %3) \n\t"
163 ::"r" (block+nCoeffs), "rm"(qmul), "rm" (qadd), "r" (2*(-nCoeffs))
168 static void dct_unquantize_mpeg1_intra_mmx(MpegEncContext *s,
169 int16_t *block, int n, int qscale)
172 const uint16_t *quant_matrix;
175 av_assert2(s->block_last_index[n]>=0);
177 nCoeffs= s->intra_scantable.raster_end[ s->block_last_index[n] ]+1;
180 block0 = block[0] * s->y_dc_scale;
182 block0 = block[0] * s->c_dc_scale;
183 /* XXX: only mpeg1 */
184 quant_matrix = s->intra_matrix;
186 "pcmpeqw %%mm7, %%mm7 \n\t"
187 "psrlw $15, %%mm7 \n\t"
188 "movd %2, %%mm6 \n\t"
189 "packssdw %%mm6, %%mm6 \n\t"
190 "packssdw %%mm6, %%mm6 \n\t"
191 "mov %3, %%"REG_a" \n\t"
194 "movq (%0, %%"REG_a"), %%mm0 \n\t"
195 "movq 8(%0, %%"REG_a"), %%mm1 \n\t"
196 "movq (%1, %%"REG_a"), %%mm4 \n\t"
197 "movq 8(%1, %%"REG_a"), %%mm5 \n\t"
198 "pmullw %%mm6, %%mm4 \n\t" // q=qscale*quant_matrix[i]
199 "pmullw %%mm6, %%mm5 \n\t" // q=qscale*quant_matrix[i]
200 "pxor %%mm2, %%mm2 \n\t"
201 "pxor %%mm3, %%mm3 \n\t"
202 "pcmpgtw %%mm0, %%mm2 \n\t" // block[i] < 0 ? -1 : 0
203 "pcmpgtw %%mm1, %%mm3 \n\t" // block[i] < 0 ? -1 : 0
204 "pxor %%mm2, %%mm0 \n\t"
205 "pxor %%mm3, %%mm1 \n\t"
206 "psubw %%mm2, %%mm0 \n\t" // abs(block[i])
207 "psubw %%mm3, %%mm1 \n\t" // abs(block[i])
208 "pmullw %%mm4, %%mm0 \n\t" // abs(block[i])*q
209 "pmullw %%mm5, %%mm1 \n\t" // abs(block[i])*q
210 "pxor %%mm4, %%mm4 \n\t"
211 "pxor %%mm5, %%mm5 \n\t" // FIXME slow
212 "pcmpeqw (%0, %%"REG_a"), %%mm4 \n\t" // block[i] == 0 ? -1 : 0
213 "pcmpeqw 8(%0, %%"REG_a"), %%mm5\n\t" // block[i] == 0 ? -1 : 0
214 "psraw $3, %%mm0 \n\t"
215 "psraw $3, %%mm1 \n\t"
216 "psubw %%mm7, %%mm0 \n\t"
217 "psubw %%mm7, %%mm1 \n\t"
218 "por %%mm7, %%mm0 \n\t"
219 "por %%mm7, %%mm1 \n\t"
220 "pxor %%mm2, %%mm0 \n\t"
221 "pxor %%mm3, %%mm1 \n\t"
222 "psubw %%mm2, %%mm0 \n\t"
223 "psubw %%mm3, %%mm1 \n\t"
224 "pandn %%mm0, %%mm4 \n\t"
225 "pandn %%mm1, %%mm5 \n\t"
226 "movq %%mm4, (%0, %%"REG_a") \n\t"
227 "movq %%mm5, 8(%0, %%"REG_a") \n\t"
229 "add $16, %%"REG_a" \n\t"
231 ::"r" (block+nCoeffs), "r"(quant_matrix+nCoeffs), "rm" (qscale), "g" (-2*nCoeffs)
237 static void dct_unquantize_mpeg1_inter_mmx(MpegEncContext *s,
238 int16_t *block, int n, int qscale)
241 const uint16_t *quant_matrix;
243 av_assert2(s->block_last_index[n]>=0);
245 nCoeffs= s->intra_scantable.raster_end[ s->block_last_index[n] ]+1;
247 quant_matrix = s->inter_matrix;
249 "pcmpeqw %%mm7, %%mm7 \n\t"
250 "psrlw $15, %%mm7 \n\t"
251 "movd %2, %%mm6 \n\t"
252 "packssdw %%mm6, %%mm6 \n\t"
253 "packssdw %%mm6, %%mm6 \n\t"
254 "mov %3, %%"REG_a" \n\t"
257 "movq (%0, %%"REG_a"), %%mm0 \n\t"
258 "movq 8(%0, %%"REG_a"), %%mm1 \n\t"
259 "movq (%1, %%"REG_a"), %%mm4 \n\t"
260 "movq 8(%1, %%"REG_a"), %%mm5 \n\t"
261 "pmullw %%mm6, %%mm4 \n\t" // q=qscale*quant_matrix[i]
262 "pmullw %%mm6, %%mm5 \n\t" // q=qscale*quant_matrix[i]
263 "pxor %%mm2, %%mm2 \n\t"
264 "pxor %%mm3, %%mm3 \n\t"
265 "pcmpgtw %%mm0, %%mm2 \n\t" // block[i] < 0 ? -1 : 0
266 "pcmpgtw %%mm1, %%mm3 \n\t" // block[i] < 0 ? -1 : 0
267 "pxor %%mm2, %%mm0 \n\t"
268 "pxor %%mm3, %%mm1 \n\t"
269 "psubw %%mm2, %%mm0 \n\t" // abs(block[i])
270 "psubw %%mm3, %%mm1 \n\t" // abs(block[i])
271 "paddw %%mm0, %%mm0 \n\t" // abs(block[i])*2
272 "paddw %%mm1, %%mm1 \n\t" // abs(block[i])*2
273 "paddw %%mm7, %%mm0 \n\t" // abs(block[i])*2 + 1
274 "paddw %%mm7, %%mm1 \n\t" // abs(block[i])*2 + 1
275 "pmullw %%mm4, %%mm0 \n\t" // (abs(block[i])*2 + 1)*q
276 "pmullw %%mm5, %%mm1 \n\t" // (abs(block[i])*2 + 1)*q
277 "pxor %%mm4, %%mm4 \n\t"
278 "pxor %%mm5, %%mm5 \n\t" // FIXME slow
279 "pcmpeqw (%0, %%"REG_a"), %%mm4 \n\t" // block[i] == 0 ? -1 : 0
280 "pcmpeqw 8(%0, %%"REG_a"), %%mm5\n\t" // block[i] == 0 ? -1 : 0
281 "psraw $4, %%mm0 \n\t"
282 "psraw $4, %%mm1 \n\t"
283 "psubw %%mm7, %%mm0 \n\t"
284 "psubw %%mm7, %%mm1 \n\t"
285 "por %%mm7, %%mm0 \n\t"
286 "por %%mm7, %%mm1 \n\t"
287 "pxor %%mm2, %%mm0 \n\t"
288 "pxor %%mm3, %%mm1 \n\t"
289 "psubw %%mm2, %%mm0 \n\t"
290 "psubw %%mm3, %%mm1 \n\t"
291 "pandn %%mm0, %%mm4 \n\t"
292 "pandn %%mm1, %%mm5 \n\t"
293 "movq %%mm4, (%0, %%"REG_a") \n\t"
294 "movq %%mm5, 8(%0, %%"REG_a") \n\t"
296 "add $16, %%"REG_a" \n\t"
298 ::"r" (block+nCoeffs), "r"(quant_matrix+nCoeffs), "rm" (qscale), "g" (-2*nCoeffs)
303 static void dct_unquantize_mpeg2_intra_mmx(MpegEncContext *s,
304 int16_t *block, int n, int qscale)
307 const uint16_t *quant_matrix;
310 av_assert2(s->block_last_index[n]>=0);
312 if(s->alternate_scan) nCoeffs= 63; //FIXME
313 else nCoeffs= s->intra_scantable.raster_end[ s->block_last_index[n] ];
316 block0 = block[0] * s->y_dc_scale;
318 block0 = block[0] * s->c_dc_scale;
319 quant_matrix = s->intra_matrix;
321 "pcmpeqw %%mm7, %%mm7 \n\t"
322 "psrlw $15, %%mm7 \n\t"
323 "movd %2, %%mm6 \n\t"
324 "packssdw %%mm6, %%mm6 \n\t"
325 "packssdw %%mm6, %%mm6 \n\t"
326 "mov %3, %%"REG_a" \n\t"
329 "movq (%0, %%"REG_a"), %%mm0 \n\t"
330 "movq 8(%0, %%"REG_a"), %%mm1 \n\t"
331 "movq (%1, %%"REG_a"), %%mm4 \n\t"
332 "movq 8(%1, %%"REG_a"), %%mm5 \n\t"
333 "pmullw %%mm6, %%mm4 \n\t" // q=qscale*quant_matrix[i]
334 "pmullw %%mm6, %%mm5 \n\t" // q=qscale*quant_matrix[i]
335 "pxor %%mm2, %%mm2 \n\t"
336 "pxor %%mm3, %%mm3 \n\t"
337 "pcmpgtw %%mm0, %%mm2 \n\t" // block[i] < 0 ? -1 : 0
338 "pcmpgtw %%mm1, %%mm3 \n\t" // block[i] < 0 ? -1 : 0
339 "pxor %%mm2, %%mm0 \n\t"
340 "pxor %%mm3, %%mm1 \n\t"
341 "psubw %%mm2, %%mm0 \n\t" // abs(block[i])
342 "psubw %%mm3, %%mm1 \n\t" // abs(block[i])
343 "pmullw %%mm4, %%mm0 \n\t" // abs(block[i])*q
344 "pmullw %%mm5, %%mm1 \n\t" // abs(block[i])*q
345 "pxor %%mm4, %%mm4 \n\t"
346 "pxor %%mm5, %%mm5 \n\t" // FIXME slow
347 "pcmpeqw (%0, %%"REG_a"), %%mm4 \n\t" // block[i] == 0 ? -1 : 0
348 "pcmpeqw 8(%0, %%"REG_a"), %%mm5\n\t" // block[i] == 0 ? -1 : 0
349 "psraw $3, %%mm0 \n\t"
350 "psraw $3, %%mm1 \n\t"
351 "pxor %%mm2, %%mm0 \n\t"
352 "pxor %%mm3, %%mm1 \n\t"
353 "psubw %%mm2, %%mm0 \n\t"
354 "psubw %%mm3, %%mm1 \n\t"
355 "pandn %%mm0, %%mm4 \n\t"
356 "pandn %%mm1, %%mm5 \n\t"
357 "movq %%mm4, (%0, %%"REG_a") \n\t"
358 "movq %%mm5, 8(%0, %%"REG_a") \n\t"
360 "add $16, %%"REG_a" \n\t"
362 ::"r" (block+nCoeffs), "r"(quant_matrix+nCoeffs), "rm" (qscale), "g" (-2*nCoeffs)
366 //Note, we do not do mismatch control for intra as errors cannot accumulate
369 static void dct_unquantize_mpeg2_inter_mmx(MpegEncContext *s,
370 int16_t *block, int n, int qscale)
373 const uint16_t *quant_matrix;
375 av_assert2(s->block_last_index[n]>=0);
377 if(s->alternate_scan) nCoeffs= 63; //FIXME
378 else nCoeffs= s->intra_scantable.raster_end[ s->block_last_index[n] ];
380 quant_matrix = s->inter_matrix;
382 "pcmpeqw %%mm7, %%mm7 \n\t"
383 "psrlq $48, %%mm7 \n\t"
384 "movd %2, %%mm6 \n\t"
385 "packssdw %%mm6, %%mm6 \n\t"
386 "packssdw %%mm6, %%mm6 \n\t"
387 "mov %3, %%"REG_a" \n\t"
390 "movq (%0, %%"REG_a"), %%mm0 \n\t"
391 "movq 8(%0, %%"REG_a"), %%mm1 \n\t"
392 "movq (%1, %%"REG_a"), %%mm4 \n\t"
393 "movq 8(%1, %%"REG_a"), %%mm5 \n\t"
394 "pmullw %%mm6, %%mm4 \n\t" // q=qscale*quant_matrix[i]
395 "pmullw %%mm6, %%mm5 \n\t" // q=qscale*quant_matrix[i]
396 "pxor %%mm2, %%mm2 \n\t"
397 "pxor %%mm3, %%mm3 \n\t"
398 "pcmpgtw %%mm0, %%mm2 \n\t" // block[i] < 0 ? -1 : 0
399 "pcmpgtw %%mm1, %%mm3 \n\t" // block[i] < 0 ? -1 : 0
400 "pxor %%mm2, %%mm0 \n\t"
401 "pxor %%mm3, %%mm1 \n\t"
402 "psubw %%mm2, %%mm0 \n\t" // abs(block[i])
403 "psubw %%mm3, %%mm1 \n\t" // abs(block[i])
404 "paddw %%mm0, %%mm0 \n\t" // abs(block[i])*2
405 "paddw %%mm1, %%mm1 \n\t" // abs(block[i])*2
406 "pmullw %%mm4, %%mm0 \n\t" // abs(block[i])*2*q
407 "pmullw %%mm5, %%mm1 \n\t" // abs(block[i])*2*q
408 "paddw %%mm4, %%mm0 \n\t" // (abs(block[i])*2 + 1)*q
409 "paddw %%mm5, %%mm1 \n\t" // (abs(block[i])*2 + 1)*q
410 "pxor %%mm4, %%mm4 \n\t"
411 "pxor %%mm5, %%mm5 \n\t" // FIXME slow
412 "pcmpeqw (%0, %%"REG_a"), %%mm4 \n\t" // block[i] == 0 ? -1 : 0
413 "pcmpeqw 8(%0, %%"REG_a"), %%mm5\n\t" // block[i] == 0 ? -1 : 0
414 "psrlw $4, %%mm0 \n\t"
415 "psrlw $4, %%mm1 \n\t"
416 "pxor %%mm2, %%mm0 \n\t"
417 "pxor %%mm3, %%mm1 \n\t"
418 "psubw %%mm2, %%mm0 \n\t"
419 "psubw %%mm3, %%mm1 \n\t"
420 "pandn %%mm0, %%mm4 \n\t"
421 "pandn %%mm1, %%mm5 \n\t"
422 "pxor %%mm4, %%mm7 \n\t"
423 "pxor %%mm5, %%mm7 \n\t"
424 "movq %%mm4, (%0, %%"REG_a") \n\t"
425 "movq %%mm5, 8(%0, %%"REG_a") \n\t"
427 "add $16, %%"REG_a" \n\t"
429 "movd 124(%0, %3), %%mm0 \n\t"
430 "movq %%mm7, %%mm6 \n\t"
431 "psrlq $32, %%mm7 \n\t"
432 "pxor %%mm6, %%mm7 \n\t"
433 "movq %%mm7, %%mm6 \n\t"
434 "psrlq $16, %%mm7 \n\t"
435 "pxor %%mm6, %%mm7 \n\t"
436 "pslld $31, %%mm7 \n\t"
437 "psrlq $15, %%mm7 \n\t"
438 "pxor %%mm7, %%mm0 \n\t"
439 "movd %%mm0, 124(%0, %3) \n\t"
441 ::"r" (block+nCoeffs), "r"(quant_matrix+nCoeffs), "rm" (qscale), "r" (-2*nCoeffs)
446 #endif /* HAVE_MMX_INLINE */
448 av_cold void ff_MPV_common_init_x86(MpegEncContext *s)
451 int cpu_flags = av_get_cpu_flags();
453 if (INLINE_MMX(cpu_flags)) {
454 s->dct_unquantize_h263_intra = dct_unquantize_h263_intra_mmx;
455 s->dct_unquantize_h263_inter = dct_unquantize_h263_inter_mmx;
456 s->dct_unquantize_mpeg1_intra = dct_unquantize_mpeg1_intra_mmx;
457 s->dct_unquantize_mpeg1_inter = dct_unquantize_mpeg1_inter_mmx;
458 if(!(s->flags & CODEC_FLAG_BITEXACT))
459 s->dct_unquantize_mpeg2_intra = dct_unquantize_mpeg2_intra_mmx;
460 s->dct_unquantize_mpeg2_inter = dct_unquantize_mpeg2_inter_mmx;
462 #endif /* HAVE_MMX_INLINE */