annotate ffmpeg/libavcodec/x86/mpegvideo.c @ 13:844d341cf643 tip

Back up before ISMIR
author Yading Song <yading.song@eecs.qmul.ac.uk>
date Thu, 31 Oct 2013 13:17:06 +0000
parents 6840f77b83aa
children
rev   line source
yading@10 1 /*
yading@10 2 * Optimized for ia32 CPUs by Nick Kurshev <nickols_k@mail.ru>
yading@10 3 * h263, mpeg1, mpeg2 dequantizer & draw_edges by Michael Niedermayer <michaelni@gmx.at>
yading@10 4 *
yading@10 5 * This file is part of FFmpeg.
yading@10 6 *
yading@10 7 * FFmpeg is free software; you can redistribute it and/or
yading@10 8 * modify it under the terms of the GNU Lesser General Public
yading@10 9 * License as published by the Free Software Foundation; either
yading@10 10 * version 2.1 of the License, or (at your option) any later version.
yading@10 11 *
yading@10 12 * FFmpeg is distributed in the hope that it will be useful,
yading@10 13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
yading@10 14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
yading@10 15 * Lesser General Public License for more details.
yading@10 16 *
yading@10 17 * You should have received a copy of the GNU Lesser General Public
yading@10 18 * License along with FFmpeg; if not, write to the Free Software
yading@10 19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
yading@10 20 */
yading@10 21
yading@10 22 #include "libavutil/attributes.h"
yading@10 23 #include "libavutil/cpu.h"
yading@10 24 #include "libavutil/x86/asm.h"
yading@10 25 #include "libavcodec/avcodec.h"
yading@10 26 #include "libavcodec/mpegvideo.h"
yading@10 27 #include "dsputil_mmx.h"
yading@10 28
yading@10 29 #if HAVE_INLINE_ASM
yading@10 30
yading@10 31 static void dct_unquantize_h263_intra_mmx(MpegEncContext *s,
yading@10 32 int16_t *block, int n, int qscale)
yading@10 33 {
yading@10 34 x86_reg level, qmul, qadd, nCoeffs;
yading@10 35
yading@10 36 qmul = qscale << 1;
yading@10 37
yading@10 38 av_assert2(s->block_last_index[n]>=0 || s->h263_aic);
yading@10 39
yading@10 40 if (!s->h263_aic) {
yading@10 41 if (n < 4)
yading@10 42 level = block[0] * s->y_dc_scale;
yading@10 43 else
yading@10 44 level = block[0] * s->c_dc_scale;
yading@10 45 qadd = (qscale - 1) | 1;
yading@10 46 }else{
yading@10 47 qadd = 0;
yading@10 48 level= block[0];
yading@10 49 }
yading@10 50 if(s->ac_pred)
yading@10 51 nCoeffs=63;
yading@10 52 else
yading@10 53 nCoeffs= s->inter_scantable.raster_end[ s->block_last_index[n] ];
yading@10 54
yading@10 55 __asm__ volatile(
yading@10 56 "movd %1, %%mm6 \n\t" //qmul
yading@10 57 "packssdw %%mm6, %%mm6 \n\t"
yading@10 58 "packssdw %%mm6, %%mm6 \n\t"
yading@10 59 "movd %2, %%mm5 \n\t" //qadd
yading@10 60 "pxor %%mm7, %%mm7 \n\t"
yading@10 61 "packssdw %%mm5, %%mm5 \n\t"
yading@10 62 "packssdw %%mm5, %%mm5 \n\t"
yading@10 63 "psubw %%mm5, %%mm7 \n\t"
yading@10 64 "pxor %%mm4, %%mm4 \n\t"
yading@10 65 ".p2align 4 \n\t"
yading@10 66 "1: \n\t"
yading@10 67 "movq (%0, %3), %%mm0 \n\t"
yading@10 68 "movq 8(%0, %3), %%mm1 \n\t"
yading@10 69
yading@10 70 "pmullw %%mm6, %%mm0 \n\t"
yading@10 71 "pmullw %%mm6, %%mm1 \n\t"
yading@10 72
yading@10 73 "movq (%0, %3), %%mm2 \n\t"
yading@10 74 "movq 8(%0, %3), %%mm3 \n\t"
yading@10 75
yading@10 76 "pcmpgtw %%mm4, %%mm2 \n\t" // block[i] < 0 ? -1 : 0
yading@10 77 "pcmpgtw %%mm4, %%mm3 \n\t" // block[i] < 0 ? -1 : 0
yading@10 78
yading@10 79 "pxor %%mm2, %%mm0 \n\t"
yading@10 80 "pxor %%mm3, %%mm1 \n\t"
yading@10 81
yading@10 82 "paddw %%mm7, %%mm0 \n\t"
yading@10 83 "paddw %%mm7, %%mm1 \n\t"
yading@10 84
yading@10 85 "pxor %%mm0, %%mm2 \n\t"
yading@10 86 "pxor %%mm1, %%mm3 \n\t"
yading@10 87
yading@10 88 "pcmpeqw %%mm7, %%mm0 \n\t" // block[i] == 0 ? -1 : 0
yading@10 89 "pcmpeqw %%mm7, %%mm1 \n\t" // block[i] == 0 ? -1 : 0
yading@10 90
yading@10 91 "pandn %%mm2, %%mm0 \n\t"
yading@10 92 "pandn %%mm3, %%mm1 \n\t"
yading@10 93
yading@10 94 "movq %%mm0, (%0, %3) \n\t"
yading@10 95 "movq %%mm1, 8(%0, %3) \n\t"
yading@10 96
yading@10 97 "add $16, %3 \n\t"
yading@10 98 "jng 1b \n\t"
yading@10 99 ::"r" (block+nCoeffs), "rm"(qmul), "rm" (qadd), "r" (2*(-nCoeffs))
yading@10 100 : "memory"
yading@10 101 );
yading@10 102 block[0]= level;
yading@10 103 }
yading@10 104
yading@10 105
yading@10 106 static void dct_unquantize_h263_inter_mmx(MpegEncContext *s,
yading@10 107 int16_t *block, int n, int qscale)
yading@10 108 {
yading@10 109 x86_reg qmul, qadd, nCoeffs;
yading@10 110
yading@10 111 qmul = qscale << 1;
yading@10 112 qadd = (qscale - 1) | 1;
yading@10 113
yading@10 114 av_assert2(s->block_last_index[n]>=0 || s->h263_aic);
yading@10 115
yading@10 116 nCoeffs= s->inter_scantable.raster_end[ s->block_last_index[n] ];
yading@10 117
yading@10 118 __asm__ volatile(
yading@10 119 "movd %1, %%mm6 \n\t" //qmul
yading@10 120 "packssdw %%mm6, %%mm6 \n\t"
yading@10 121 "packssdw %%mm6, %%mm6 \n\t"
yading@10 122 "movd %2, %%mm5 \n\t" //qadd
yading@10 123 "pxor %%mm7, %%mm7 \n\t"
yading@10 124 "packssdw %%mm5, %%mm5 \n\t"
yading@10 125 "packssdw %%mm5, %%mm5 \n\t"
yading@10 126 "psubw %%mm5, %%mm7 \n\t"
yading@10 127 "pxor %%mm4, %%mm4 \n\t"
yading@10 128 ".p2align 4 \n\t"
yading@10 129 "1: \n\t"
yading@10 130 "movq (%0, %3), %%mm0 \n\t"
yading@10 131 "movq 8(%0, %3), %%mm1 \n\t"
yading@10 132
yading@10 133 "pmullw %%mm6, %%mm0 \n\t"
yading@10 134 "pmullw %%mm6, %%mm1 \n\t"
yading@10 135
yading@10 136 "movq (%0, %3), %%mm2 \n\t"
yading@10 137 "movq 8(%0, %3), %%mm3 \n\t"
yading@10 138
yading@10 139 "pcmpgtw %%mm4, %%mm2 \n\t" // block[i] < 0 ? -1 : 0
yading@10 140 "pcmpgtw %%mm4, %%mm3 \n\t" // block[i] < 0 ? -1 : 0
yading@10 141
yading@10 142 "pxor %%mm2, %%mm0 \n\t"
yading@10 143 "pxor %%mm3, %%mm1 \n\t"
yading@10 144
yading@10 145 "paddw %%mm7, %%mm0 \n\t"
yading@10 146 "paddw %%mm7, %%mm1 \n\t"
yading@10 147
yading@10 148 "pxor %%mm0, %%mm2 \n\t"
yading@10 149 "pxor %%mm1, %%mm3 \n\t"
yading@10 150
yading@10 151 "pcmpeqw %%mm7, %%mm0 \n\t" // block[i] == 0 ? -1 : 0
yading@10 152 "pcmpeqw %%mm7, %%mm1 \n\t" // block[i] == 0 ? -1 : 0
yading@10 153
yading@10 154 "pandn %%mm2, %%mm0 \n\t"
yading@10 155 "pandn %%mm3, %%mm1 \n\t"
yading@10 156
yading@10 157 "movq %%mm0, (%0, %3) \n\t"
yading@10 158 "movq %%mm1, 8(%0, %3) \n\t"
yading@10 159
yading@10 160 "add $16, %3 \n\t"
yading@10 161 "jng 1b \n\t"
yading@10 162 ::"r" (block+nCoeffs), "rm"(qmul), "rm" (qadd), "r" (2*(-nCoeffs))
yading@10 163 : "memory"
yading@10 164 );
yading@10 165 }
yading@10 166
yading@10 167
yading@10 168 /*
yading@10 169 We can suppose that result of two multiplications can't be greater than 0xFFFF
yading@10 170 i.e. is 16-bit, so we use here only PMULLW instruction and can avoid
yading@10 171 a complex multiplication.
yading@10 172 =====================================================
yading@10 173 Full formula for multiplication of 2 integer numbers
yading@10 174 which are represent as high:low words:
yading@10 175 input: value1 = high1:low1
yading@10 176 value2 = high2:low2
yading@10 177 output: value3 = value1*value2
yading@10 178 value3=high3:low3 (on overflow: modulus 2^32 wrap-around)
yading@10 179 this mean that for 0x123456 * 0x123456 correct result is 0x766cb0ce4
yading@10 180 but this algorithm will compute only 0x66cb0ce4
yading@10 181 this limited by 16-bit size of operands
yading@10 182 ---------------------------------
yading@10 183 tlow1 = high1*low2
yading@10 184 tlow2 = high2*low1
yading@10 185 tlow1 = tlow1 + tlow2
yading@10 186 high3:low3 = low1*low2
yading@10 187 high3 += tlow1
yading@10 188 */
yading@10 189 static void dct_unquantize_mpeg1_intra_mmx(MpegEncContext *s,
yading@10 190 int16_t *block, int n, int qscale)
yading@10 191 {
yading@10 192 x86_reg nCoeffs;
yading@10 193 const uint16_t *quant_matrix;
yading@10 194 int block0;
yading@10 195
yading@10 196 av_assert2(s->block_last_index[n]>=0);
yading@10 197
yading@10 198 nCoeffs= s->intra_scantable.raster_end[ s->block_last_index[n] ]+1;
yading@10 199
yading@10 200 if (n < 4)
yading@10 201 block0 = block[0] * s->y_dc_scale;
yading@10 202 else
yading@10 203 block0 = block[0] * s->c_dc_scale;
yading@10 204 /* XXX: only mpeg1 */
yading@10 205 quant_matrix = s->intra_matrix;
yading@10 206 __asm__ volatile(
yading@10 207 "pcmpeqw %%mm7, %%mm7 \n\t"
yading@10 208 "psrlw $15, %%mm7 \n\t"
yading@10 209 "movd %2, %%mm6 \n\t"
yading@10 210 "packssdw %%mm6, %%mm6 \n\t"
yading@10 211 "packssdw %%mm6, %%mm6 \n\t"
yading@10 212 "mov %3, %%"REG_a" \n\t"
yading@10 213 ".p2align 4 \n\t"
yading@10 214 "1: \n\t"
yading@10 215 "movq (%0, %%"REG_a"), %%mm0 \n\t"
yading@10 216 "movq 8(%0, %%"REG_a"), %%mm1 \n\t"
yading@10 217 "movq (%1, %%"REG_a"), %%mm4 \n\t"
yading@10 218 "movq 8(%1, %%"REG_a"), %%mm5 \n\t"
yading@10 219 "pmullw %%mm6, %%mm4 \n\t" // q=qscale*quant_matrix[i]
yading@10 220 "pmullw %%mm6, %%mm5 \n\t" // q=qscale*quant_matrix[i]
yading@10 221 "pxor %%mm2, %%mm2 \n\t"
yading@10 222 "pxor %%mm3, %%mm3 \n\t"
yading@10 223 "pcmpgtw %%mm0, %%mm2 \n\t" // block[i] < 0 ? -1 : 0
yading@10 224 "pcmpgtw %%mm1, %%mm3 \n\t" // block[i] < 0 ? -1 : 0
yading@10 225 "pxor %%mm2, %%mm0 \n\t"
yading@10 226 "pxor %%mm3, %%mm1 \n\t"
yading@10 227 "psubw %%mm2, %%mm0 \n\t" // abs(block[i])
yading@10 228 "psubw %%mm3, %%mm1 \n\t" // abs(block[i])
yading@10 229 "pmullw %%mm4, %%mm0 \n\t" // abs(block[i])*q
yading@10 230 "pmullw %%mm5, %%mm1 \n\t" // abs(block[i])*q
yading@10 231 "pxor %%mm4, %%mm4 \n\t"
yading@10 232 "pxor %%mm5, %%mm5 \n\t" // FIXME slow
yading@10 233 "pcmpeqw (%0, %%"REG_a"), %%mm4 \n\t" // block[i] == 0 ? -1 : 0
yading@10 234 "pcmpeqw 8(%0, %%"REG_a"), %%mm5\n\t" // block[i] == 0 ? -1 : 0
yading@10 235 "psraw $3, %%mm0 \n\t"
yading@10 236 "psraw $3, %%mm1 \n\t"
yading@10 237 "psubw %%mm7, %%mm0 \n\t"
yading@10 238 "psubw %%mm7, %%mm1 \n\t"
yading@10 239 "por %%mm7, %%mm0 \n\t"
yading@10 240 "por %%mm7, %%mm1 \n\t"
yading@10 241 "pxor %%mm2, %%mm0 \n\t"
yading@10 242 "pxor %%mm3, %%mm1 \n\t"
yading@10 243 "psubw %%mm2, %%mm0 \n\t"
yading@10 244 "psubw %%mm3, %%mm1 \n\t"
yading@10 245 "pandn %%mm0, %%mm4 \n\t"
yading@10 246 "pandn %%mm1, %%mm5 \n\t"
yading@10 247 "movq %%mm4, (%0, %%"REG_a") \n\t"
yading@10 248 "movq %%mm5, 8(%0, %%"REG_a") \n\t"
yading@10 249
yading@10 250 "add $16, %%"REG_a" \n\t"
yading@10 251 "js 1b \n\t"
yading@10 252 ::"r" (block+nCoeffs), "r"(quant_matrix+nCoeffs), "rm" (qscale), "g" (-2*nCoeffs)
yading@10 253 : "%"REG_a, "memory"
yading@10 254 );
yading@10 255 block[0]= block0;
yading@10 256 }
yading@10 257
yading@10 258 static void dct_unquantize_mpeg1_inter_mmx(MpegEncContext *s,
yading@10 259 int16_t *block, int n, int qscale)
yading@10 260 {
yading@10 261 x86_reg nCoeffs;
yading@10 262 const uint16_t *quant_matrix;
yading@10 263
yading@10 264 av_assert2(s->block_last_index[n]>=0);
yading@10 265
yading@10 266 nCoeffs= s->intra_scantable.raster_end[ s->block_last_index[n] ]+1;
yading@10 267
yading@10 268 quant_matrix = s->inter_matrix;
yading@10 269 __asm__ volatile(
yading@10 270 "pcmpeqw %%mm7, %%mm7 \n\t"
yading@10 271 "psrlw $15, %%mm7 \n\t"
yading@10 272 "movd %2, %%mm6 \n\t"
yading@10 273 "packssdw %%mm6, %%mm6 \n\t"
yading@10 274 "packssdw %%mm6, %%mm6 \n\t"
yading@10 275 "mov %3, %%"REG_a" \n\t"
yading@10 276 ".p2align 4 \n\t"
yading@10 277 "1: \n\t"
yading@10 278 "movq (%0, %%"REG_a"), %%mm0 \n\t"
yading@10 279 "movq 8(%0, %%"REG_a"), %%mm1 \n\t"
yading@10 280 "movq (%1, %%"REG_a"), %%mm4 \n\t"
yading@10 281 "movq 8(%1, %%"REG_a"), %%mm5 \n\t"
yading@10 282 "pmullw %%mm6, %%mm4 \n\t" // q=qscale*quant_matrix[i]
yading@10 283 "pmullw %%mm6, %%mm5 \n\t" // q=qscale*quant_matrix[i]
yading@10 284 "pxor %%mm2, %%mm2 \n\t"
yading@10 285 "pxor %%mm3, %%mm3 \n\t"
yading@10 286 "pcmpgtw %%mm0, %%mm2 \n\t" // block[i] < 0 ? -1 : 0
yading@10 287 "pcmpgtw %%mm1, %%mm3 \n\t" // block[i] < 0 ? -1 : 0
yading@10 288 "pxor %%mm2, %%mm0 \n\t"
yading@10 289 "pxor %%mm3, %%mm1 \n\t"
yading@10 290 "psubw %%mm2, %%mm0 \n\t" // abs(block[i])
yading@10 291 "psubw %%mm3, %%mm1 \n\t" // abs(block[i])
yading@10 292 "paddw %%mm0, %%mm0 \n\t" // abs(block[i])*2
yading@10 293 "paddw %%mm1, %%mm1 \n\t" // abs(block[i])*2
yading@10 294 "paddw %%mm7, %%mm0 \n\t" // abs(block[i])*2 + 1
yading@10 295 "paddw %%mm7, %%mm1 \n\t" // abs(block[i])*2 + 1
yading@10 296 "pmullw %%mm4, %%mm0 \n\t" // (abs(block[i])*2 + 1)*q
yading@10 297 "pmullw %%mm5, %%mm1 \n\t" // (abs(block[i])*2 + 1)*q
yading@10 298 "pxor %%mm4, %%mm4 \n\t"
yading@10 299 "pxor %%mm5, %%mm5 \n\t" // FIXME slow
yading@10 300 "pcmpeqw (%0, %%"REG_a"), %%mm4 \n\t" // block[i] == 0 ? -1 : 0
yading@10 301 "pcmpeqw 8(%0, %%"REG_a"), %%mm5\n\t" // block[i] == 0 ? -1 : 0
yading@10 302 "psraw $4, %%mm0 \n\t"
yading@10 303 "psraw $4, %%mm1 \n\t"
yading@10 304 "psubw %%mm7, %%mm0 \n\t"
yading@10 305 "psubw %%mm7, %%mm1 \n\t"
yading@10 306 "por %%mm7, %%mm0 \n\t"
yading@10 307 "por %%mm7, %%mm1 \n\t"
yading@10 308 "pxor %%mm2, %%mm0 \n\t"
yading@10 309 "pxor %%mm3, %%mm1 \n\t"
yading@10 310 "psubw %%mm2, %%mm0 \n\t"
yading@10 311 "psubw %%mm3, %%mm1 \n\t"
yading@10 312 "pandn %%mm0, %%mm4 \n\t"
yading@10 313 "pandn %%mm1, %%mm5 \n\t"
yading@10 314 "movq %%mm4, (%0, %%"REG_a") \n\t"
yading@10 315 "movq %%mm5, 8(%0, %%"REG_a") \n\t"
yading@10 316
yading@10 317 "add $16, %%"REG_a" \n\t"
yading@10 318 "js 1b \n\t"
yading@10 319 ::"r" (block+nCoeffs), "r"(quant_matrix+nCoeffs), "rm" (qscale), "g" (-2*nCoeffs)
yading@10 320 : "%"REG_a, "memory"
yading@10 321 );
yading@10 322 }
yading@10 323
yading@10 324 static void dct_unquantize_mpeg2_intra_mmx(MpegEncContext *s,
yading@10 325 int16_t *block, int n, int qscale)
yading@10 326 {
yading@10 327 x86_reg nCoeffs;
yading@10 328 const uint16_t *quant_matrix;
yading@10 329 int block0;
yading@10 330
yading@10 331 av_assert2(s->block_last_index[n]>=0);
yading@10 332
yading@10 333 if(s->alternate_scan) nCoeffs= 63; //FIXME
yading@10 334 else nCoeffs= s->intra_scantable.raster_end[ s->block_last_index[n] ];
yading@10 335
yading@10 336 if (n < 4)
yading@10 337 block0 = block[0] * s->y_dc_scale;
yading@10 338 else
yading@10 339 block0 = block[0] * s->c_dc_scale;
yading@10 340 quant_matrix = s->intra_matrix;
yading@10 341 __asm__ volatile(
yading@10 342 "pcmpeqw %%mm7, %%mm7 \n\t"
yading@10 343 "psrlw $15, %%mm7 \n\t"
yading@10 344 "movd %2, %%mm6 \n\t"
yading@10 345 "packssdw %%mm6, %%mm6 \n\t"
yading@10 346 "packssdw %%mm6, %%mm6 \n\t"
yading@10 347 "mov %3, %%"REG_a" \n\t"
yading@10 348 ".p2align 4 \n\t"
yading@10 349 "1: \n\t"
yading@10 350 "movq (%0, %%"REG_a"), %%mm0 \n\t"
yading@10 351 "movq 8(%0, %%"REG_a"), %%mm1 \n\t"
yading@10 352 "movq (%1, %%"REG_a"), %%mm4 \n\t"
yading@10 353 "movq 8(%1, %%"REG_a"), %%mm5 \n\t"
yading@10 354 "pmullw %%mm6, %%mm4 \n\t" // q=qscale*quant_matrix[i]
yading@10 355 "pmullw %%mm6, %%mm5 \n\t" // q=qscale*quant_matrix[i]
yading@10 356 "pxor %%mm2, %%mm2 \n\t"
yading@10 357 "pxor %%mm3, %%mm3 \n\t"
yading@10 358 "pcmpgtw %%mm0, %%mm2 \n\t" // block[i] < 0 ? -1 : 0
yading@10 359 "pcmpgtw %%mm1, %%mm3 \n\t" // block[i] < 0 ? -1 : 0
yading@10 360 "pxor %%mm2, %%mm0 \n\t"
yading@10 361 "pxor %%mm3, %%mm1 \n\t"
yading@10 362 "psubw %%mm2, %%mm0 \n\t" // abs(block[i])
yading@10 363 "psubw %%mm3, %%mm1 \n\t" // abs(block[i])
yading@10 364 "pmullw %%mm4, %%mm0 \n\t" // abs(block[i])*q
yading@10 365 "pmullw %%mm5, %%mm1 \n\t" // abs(block[i])*q
yading@10 366 "pxor %%mm4, %%mm4 \n\t"
yading@10 367 "pxor %%mm5, %%mm5 \n\t" // FIXME slow
yading@10 368 "pcmpeqw (%0, %%"REG_a"), %%mm4 \n\t" // block[i] == 0 ? -1 : 0
yading@10 369 "pcmpeqw 8(%0, %%"REG_a"), %%mm5\n\t" // block[i] == 0 ? -1 : 0
yading@10 370 "psraw $3, %%mm0 \n\t"
yading@10 371 "psraw $3, %%mm1 \n\t"
yading@10 372 "pxor %%mm2, %%mm0 \n\t"
yading@10 373 "pxor %%mm3, %%mm1 \n\t"
yading@10 374 "psubw %%mm2, %%mm0 \n\t"
yading@10 375 "psubw %%mm3, %%mm1 \n\t"
yading@10 376 "pandn %%mm0, %%mm4 \n\t"
yading@10 377 "pandn %%mm1, %%mm5 \n\t"
yading@10 378 "movq %%mm4, (%0, %%"REG_a") \n\t"
yading@10 379 "movq %%mm5, 8(%0, %%"REG_a") \n\t"
yading@10 380
yading@10 381 "add $16, %%"REG_a" \n\t"
yading@10 382 "jng 1b \n\t"
yading@10 383 ::"r" (block+nCoeffs), "r"(quant_matrix+nCoeffs), "rm" (qscale), "g" (-2*nCoeffs)
yading@10 384 : "%"REG_a, "memory"
yading@10 385 );
yading@10 386 block[0]= block0;
yading@10 387 //Note, we do not do mismatch control for intra as errors cannot accumulate
yading@10 388 }
yading@10 389
yading@10 390 static void dct_unquantize_mpeg2_inter_mmx(MpegEncContext *s,
yading@10 391 int16_t *block, int n, int qscale)
yading@10 392 {
yading@10 393 x86_reg nCoeffs;
yading@10 394 const uint16_t *quant_matrix;
yading@10 395
yading@10 396 av_assert2(s->block_last_index[n]>=0);
yading@10 397
yading@10 398 if(s->alternate_scan) nCoeffs= 63; //FIXME
yading@10 399 else nCoeffs= s->intra_scantable.raster_end[ s->block_last_index[n] ];
yading@10 400
yading@10 401 quant_matrix = s->inter_matrix;
yading@10 402 __asm__ volatile(
yading@10 403 "pcmpeqw %%mm7, %%mm7 \n\t"
yading@10 404 "psrlq $48, %%mm7 \n\t"
yading@10 405 "movd %2, %%mm6 \n\t"
yading@10 406 "packssdw %%mm6, %%mm6 \n\t"
yading@10 407 "packssdw %%mm6, %%mm6 \n\t"
yading@10 408 "mov %3, %%"REG_a" \n\t"
yading@10 409 ".p2align 4 \n\t"
yading@10 410 "1: \n\t"
yading@10 411 "movq (%0, %%"REG_a"), %%mm0 \n\t"
yading@10 412 "movq 8(%0, %%"REG_a"), %%mm1 \n\t"
yading@10 413 "movq (%1, %%"REG_a"), %%mm4 \n\t"
yading@10 414 "movq 8(%1, %%"REG_a"), %%mm5 \n\t"
yading@10 415 "pmullw %%mm6, %%mm4 \n\t" // q=qscale*quant_matrix[i]
yading@10 416 "pmullw %%mm6, %%mm5 \n\t" // q=qscale*quant_matrix[i]
yading@10 417 "pxor %%mm2, %%mm2 \n\t"
yading@10 418 "pxor %%mm3, %%mm3 \n\t"
yading@10 419 "pcmpgtw %%mm0, %%mm2 \n\t" // block[i] < 0 ? -1 : 0
yading@10 420 "pcmpgtw %%mm1, %%mm3 \n\t" // block[i] < 0 ? -1 : 0
yading@10 421 "pxor %%mm2, %%mm0 \n\t"
yading@10 422 "pxor %%mm3, %%mm1 \n\t"
yading@10 423 "psubw %%mm2, %%mm0 \n\t" // abs(block[i])
yading@10 424 "psubw %%mm3, %%mm1 \n\t" // abs(block[i])
yading@10 425 "paddw %%mm0, %%mm0 \n\t" // abs(block[i])*2
yading@10 426 "paddw %%mm1, %%mm1 \n\t" // abs(block[i])*2
yading@10 427 "pmullw %%mm4, %%mm0 \n\t" // abs(block[i])*2*q
yading@10 428 "pmullw %%mm5, %%mm1 \n\t" // abs(block[i])*2*q
yading@10 429 "paddw %%mm4, %%mm0 \n\t" // (abs(block[i])*2 + 1)*q
yading@10 430 "paddw %%mm5, %%mm1 \n\t" // (abs(block[i])*2 + 1)*q
yading@10 431 "pxor %%mm4, %%mm4 \n\t"
yading@10 432 "pxor %%mm5, %%mm5 \n\t" // FIXME slow
yading@10 433 "pcmpeqw (%0, %%"REG_a"), %%mm4 \n\t" // block[i] == 0 ? -1 : 0
yading@10 434 "pcmpeqw 8(%0, %%"REG_a"), %%mm5\n\t" // block[i] == 0 ? -1 : 0
yading@10 435 "psrlw $4, %%mm0 \n\t"
yading@10 436 "psrlw $4, %%mm1 \n\t"
yading@10 437 "pxor %%mm2, %%mm0 \n\t"
yading@10 438 "pxor %%mm3, %%mm1 \n\t"
yading@10 439 "psubw %%mm2, %%mm0 \n\t"
yading@10 440 "psubw %%mm3, %%mm1 \n\t"
yading@10 441 "pandn %%mm0, %%mm4 \n\t"
yading@10 442 "pandn %%mm1, %%mm5 \n\t"
yading@10 443 "pxor %%mm4, %%mm7 \n\t"
yading@10 444 "pxor %%mm5, %%mm7 \n\t"
yading@10 445 "movq %%mm4, (%0, %%"REG_a") \n\t"
yading@10 446 "movq %%mm5, 8(%0, %%"REG_a") \n\t"
yading@10 447
yading@10 448 "add $16, %%"REG_a" \n\t"
yading@10 449 "jng 1b \n\t"
yading@10 450 "movd 124(%0, %3), %%mm0 \n\t"
yading@10 451 "movq %%mm7, %%mm6 \n\t"
yading@10 452 "psrlq $32, %%mm7 \n\t"
yading@10 453 "pxor %%mm6, %%mm7 \n\t"
yading@10 454 "movq %%mm7, %%mm6 \n\t"
yading@10 455 "psrlq $16, %%mm7 \n\t"
yading@10 456 "pxor %%mm6, %%mm7 \n\t"
yading@10 457 "pslld $31, %%mm7 \n\t"
yading@10 458 "psrlq $15, %%mm7 \n\t"
yading@10 459 "pxor %%mm7, %%mm0 \n\t"
yading@10 460 "movd %%mm0, 124(%0, %3) \n\t"
yading@10 461
yading@10 462 ::"r" (block+nCoeffs), "r"(quant_matrix+nCoeffs), "rm" (qscale), "r" (-2*nCoeffs)
yading@10 463 : "%"REG_a, "memory"
yading@10 464 );
yading@10 465 }
yading@10 466
yading@10 467 static void denoise_dct_mmx(MpegEncContext *s, int16_t *block){
yading@10 468 const int intra= s->mb_intra;
yading@10 469 int *sum= s->dct_error_sum[intra];
yading@10 470 uint16_t *offset= s->dct_offset[intra];
yading@10 471
yading@10 472 s->dct_count[intra]++;
yading@10 473
yading@10 474 __asm__ volatile(
yading@10 475 "pxor %%mm7, %%mm7 \n\t"
yading@10 476 "1: \n\t"
yading@10 477 "pxor %%mm0, %%mm0 \n\t"
yading@10 478 "pxor %%mm1, %%mm1 \n\t"
yading@10 479 "movq (%0), %%mm2 \n\t"
yading@10 480 "movq 8(%0), %%mm3 \n\t"
yading@10 481 "pcmpgtw %%mm2, %%mm0 \n\t"
yading@10 482 "pcmpgtw %%mm3, %%mm1 \n\t"
yading@10 483 "pxor %%mm0, %%mm2 \n\t"
yading@10 484 "pxor %%mm1, %%mm3 \n\t"
yading@10 485 "psubw %%mm0, %%mm2 \n\t"
yading@10 486 "psubw %%mm1, %%mm3 \n\t"
yading@10 487 "movq %%mm2, %%mm4 \n\t"
yading@10 488 "movq %%mm3, %%mm5 \n\t"
yading@10 489 "psubusw (%2), %%mm2 \n\t"
yading@10 490 "psubusw 8(%2), %%mm3 \n\t"
yading@10 491 "pxor %%mm0, %%mm2 \n\t"
yading@10 492 "pxor %%mm1, %%mm3 \n\t"
yading@10 493 "psubw %%mm0, %%mm2 \n\t"
yading@10 494 "psubw %%mm1, %%mm3 \n\t"
yading@10 495 "movq %%mm2, (%0) \n\t"
yading@10 496 "movq %%mm3, 8(%0) \n\t"
yading@10 497 "movq %%mm4, %%mm2 \n\t"
yading@10 498 "movq %%mm5, %%mm3 \n\t"
yading@10 499 "punpcklwd %%mm7, %%mm4 \n\t"
yading@10 500 "punpckhwd %%mm7, %%mm2 \n\t"
yading@10 501 "punpcklwd %%mm7, %%mm5 \n\t"
yading@10 502 "punpckhwd %%mm7, %%mm3 \n\t"
yading@10 503 "paddd (%1), %%mm4 \n\t"
yading@10 504 "paddd 8(%1), %%mm2 \n\t"
yading@10 505 "paddd 16(%1), %%mm5 \n\t"
yading@10 506 "paddd 24(%1), %%mm3 \n\t"
yading@10 507 "movq %%mm4, (%1) \n\t"
yading@10 508 "movq %%mm2, 8(%1) \n\t"
yading@10 509 "movq %%mm5, 16(%1) \n\t"
yading@10 510 "movq %%mm3, 24(%1) \n\t"
yading@10 511 "add $16, %0 \n\t"
yading@10 512 "add $32, %1 \n\t"
yading@10 513 "add $16, %2 \n\t"
yading@10 514 "cmp %3, %0 \n\t"
yading@10 515 " jb 1b \n\t"
yading@10 516 : "+r" (block), "+r" (sum), "+r" (offset)
yading@10 517 : "r"(block+64)
yading@10 518 );
yading@10 519 }
yading@10 520
yading@10 521 static void denoise_dct_sse2(MpegEncContext *s, int16_t *block){
yading@10 522 const int intra= s->mb_intra;
yading@10 523 int *sum= s->dct_error_sum[intra];
yading@10 524 uint16_t *offset= s->dct_offset[intra];
yading@10 525
yading@10 526 s->dct_count[intra]++;
yading@10 527
yading@10 528 __asm__ volatile(
yading@10 529 "pxor %%xmm7, %%xmm7 \n\t"
yading@10 530 "1: \n\t"
yading@10 531 "pxor %%xmm0, %%xmm0 \n\t"
yading@10 532 "pxor %%xmm1, %%xmm1 \n\t"
yading@10 533 "movdqa (%0), %%xmm2 \n\t"
yading@10 534 "movdqa 16(%0), %%xmm3 \n\t"
yading@10 535 "pcmpgtw %%xmm2, %%xmm0 \n\t"
yading@10 536 "pcmpgtw %%xmm3, %%xmm1 \n\t"
yading@10 537 "pxor %%xmm0, %%xmm2 \n\t"
yading@10 538 "pxor %%xmm1, %%xmm3 \n\t"
yading@10 539 "psubw %%xmm0, %%xmm2 \n\t"
yading@10 540 "psubw %%xmm1, %%xmm3 \n\t"
yading@10 541 "movdqa %%xmm2, %%xmm4 \n\t"
yading@10 542 "movdqa %%xmm3, %%xmm5 \n\t"
yading@10 543 "psubusw (%2), %%xmm2 \n\t"
yading@10 544 "psubusw 16(%2), %%xmm3 \n\t"
yading@10 545 "pxor %%xmm0, %%xmm2 \n\t"
yading@10 546 "pxor %%xmm1, %%xmm3 \n\t"
yading@10 547 "psubw %%xmm0, %%xmm2 \n\t"
yading@10 548 "psubw %%xmm1, %%xmm3 \n\t"
yading@10 549 "movdqa %%xmm2, (%0) \n\t"
yading@10 550 "movdqa %%xmm3, 16(%0) \n\t"
yading@10 551 "movdqa %%xmm4, %%xmm6 \n\t"
yading@10 552 "movdqa %%xmm5, %%xmm0 \n\t"
yading@10 553 "punpcklwd %%xmm7, %%xmm4 \n\t"
yading@10 554 "punpckhwd %%xmm7, %%xmm6 \n\t"
yading@10 555 "punpcklwd %%xmm7, %%xmm5 \n\t"
yading@10 556 "punpckhwd %%xmm7, %%xmm0 \n\t"
yading@10 557 "paddd (%1), %%xmm4 \n\t"
yading@10 558 "paddd 16(%1), %%xmm6 \n\t"
yading@10 559 "paddd 32(%1), %%xmm5 \n\t"
yading@10 560 "paddd 48(%1), %%xmm0 \n\t"
yading@10 561 "movdqa %%xmm4, (%1) \n\t"
yading@10 562 "movdqa %%xmm6, 16(%1) \n\t"
yading@10 563 "movdqa %%xmm5, 32(%1) \n\t"
yading@10 564 "movdqa %%xmm0, 48(%1) \n\t"
yading@10 565 "add $32, %0 \n\t"
yading@10 566 "add $64, %1 \n\t"
yading@10 567 "add $32, %2 \n\t"
yading@10 568 "cmp %3, %0 \n\t"
yading@10 569 " jb 1b \n\t"
yading@10 570 : "+r" (block), "+r" (sum), "+r" (offset)
yading@10 571 : "r"(block+64)
yading@10 572 XMM_CLOBBERS_ONLY("%xmm0", "%xmm1", "%xmm2", "%xmm3",
yading@10 573 "%xmm4", "%xmm5", "%xmm6", "%xmm7")
yading@10 574 );
yading@10 575 }
yading@10 576
yading@10 577 #endif /* HAVE_INLINE_ASM */
yading@10 578
yading@10 579 av_cold void ff_MPV_common_init_x86(MpegEncContext *s)
yading@10 580 {
yading@10 581 #if HAVE_INLINE_ASM
yading@10 582 int mm_flags = av_get_cpu_flags();
yading@10 583
yading@10 584 if (mm_flags & AV_CPU_FLAG_MMX) {
yading@10 585 s->dct_unquantize_h263_intra = dct_unquantize_h263_intra_mmx;
yading@10 586 s->dct_unquantize_h263_inter = dct_unquantize_h263_inter_mmx;
yading@10 587 s->dct_unquantize_mpeg1_intra = dct_unquantize_mpeg1_intra_mmx;
yading@10 588 s->dct_unquantize_mpeg1_inter = dct_unquantize_mpeg1_inter_mmx;
yading@10 589 if(!(s->flags & CODEC_FLAG_BITEXACT))
yading@10 590 s->dct_unquantize_mpeg2_intra = dct_unquantize_mpeg2_intra_mmx;
yading@10 591 s->dct_unquantize_mpeg2_inter = dct_unquantize_mpeg2_inter_mmx;
yading@10 592
yading@10 593 if (mm_flags & AV_CPU_FLAG_SSE2) {
yading@10 594 s->denoise_dct= denoise_dct_sse2;
yading@10 595 } else {
yading@10 596 s->denoise_dct= denoise_dct_mmx;
yading@10 597 }
yading@10 598 }
yading@10 599 #endif /* HAVE_INLINE_ASM */
yading@10 600 }