mirror of
https://github.com/FFmpeg/FFmpeg.git
synced 2024-12-23 12:43:46 +02:00
MMX for H.264 iDCT (adapted from x264)
Originally committed as revision 4329 to svn://svn.ffmpeg.org/ffmpeg/trunk
This commit is contained in:
parent
ef7b5aee5d
commit
1d62fc8560
@ -3205,6 +3205,8 @@ void ff_vp3_idct_sse2(int16_t *input_data);
|
||||
void ff_vp3_idct_mmx(int16_t *data);
|
||||
void ff_vp3_dsp_init_mmx(void);
|
||||
|
||||
void ff_h264_idct_add_mmx2(uint8_t *dst, int16_t *block, int stride);
|
||||
|
||||
/* XXX: those functions should be suppressed ASAP when all IDCTs are
|
||||
converted */
|
||||
static void ff_libmpeg2mmx_idct_put(uint8_t *dest, int line_size, DCTELEM *block)
|
||||
@ -3424,6 +3426,8 @@ void dsputil_init_mmx(DSPContext* c, AVCodecContext *avctx)
|
||||
c->vsad[4]= vsad_intra16_mmx2;
|
||||
#endif //CONFIG_ENCODERS
|
||||
|
||||
c->h264_idct_add= ff_h264_idct_add_mmx2;
|
||||
|
||||
if(!(avctx->flags & CODEC_FLAG_BITEXACT)){
|
||||
c->put_no_rnd_pixels_tab[0][1] = put_no_rnd_pixels16_x2_mmx2;
|
||||
c->put_no_rnd_pixels_tab[0][2] = put_no_rnd_pixels16_y2_mmx2;
|
||||
|
@ -597,3 +597,89 @@ declare_idct (ff_mmxext_idct, mmxext_table,
|
||||
|
||||
declare_idct (ff_mmx_idct, mmx_table,
|
||||
mmx_row_head, mmx_row, mmx_row_tail, mmx_row_mid)
|
||||
|
||||
|
||||
|
||||
/* in/out: mma=mma+mmb, mmb=mmb-mma */
|
||||
#define SUMSUB_BA( a, b ) \
|
||||
"paddw "#b", "#a" \n\t"\
|
||||
"paddw "#b", "#b" \n\t"\
|
||||
"psubw "#a", "#b" \n\t"
|
||||
|
||||
#define SUMSUB_BADC( a, b, c, d ) \
|
||||
"paddw "#b", "#a" \n\t"\
|
||||
"paddw "#d", "#c" \n\t"\
|
||||
"paddw "#b", "#b" \n\t"\
|
||||
"paddw "#d", "#d" \n\t"\
|
||||
"psubw "#a", "#b" \n\t"\
|
||||
"psubw "#c", "#d" \n\t"
|
||||
|
||||
/* in: a,b out: a,s */
|
||||
#define SUMSUBD2_AB( a, b, t, s ) \
|
||||
"movq "#a", "#s" \n\t"\
|
||||
"movq "#b", "#t" \n\t"\
|
||||
"psraw $1 , "#b" \n\t"\
|
||||
"psraw $1 , "#s" \n\t"\
|
||||
"paddw "#b", "#a" \n\t"\
|
||||
"psubw "#t", "#s" \n\t"
|
||||
|
||||
#define IDCT4_1D( s02, s13, d02, d13, t, u ) \
|
||||
SUMSUB_BA ( s02, d02 )\
|
||||
SUMSUBD2_AB( s13, d13, u, t )\
|
||||
SUMSUB_BADC( s13, s02, t, d02 )
|
||||
|
||||
#define SBUTTERFLY( a, b, t, n ) \
|
||||
"movq "#a", "#t" \n\t" /* abcd */\
|
||||
"punpckl"#n" "#b", "#a" \n\t" /* aebf */\
|
||||
"punpckh"#n" "#b", "#t" \n\t" /* cgdh */
|
||||
|
||||
#define TRANSPOSE4( a, b, c, d, t ) \
|
||||
SBUTTERFLY( a, b, t, wd ) /* a=aebf t=cgdh */\
|
||||
SBUTTERFLY( c, d, b, wd ) /* c=imjn b=kolp */\
|
||||
SBUTTERFLY( a, c, d, dq ) /* a=aeim d=bfjn */\
|
||||
SBUTTERFLY( t, b, c, dq ) /* t=cgko c=dhlp */
|
||||
|
||||
#define STORE_DIFF_4P( p, t, pw32, z, dst ) \
|
||||
asm volatile(\
|
||||
"paddw "#pw32", "#p" \n\t"\
|
||||
"psraw $6, "#p" \n\t"\
|
||||
"movd (%0), "#t" \n\t"\
|
||||
"punpcklbw "#z", "#t" \n\t"\
|
||||
"paddsw "#t", "#p" \n\t"\
|
||||
"packuswb "#z", "#p" \n\t"\
|
||||
"movd "#p", (%0) \n\t" :: "r"(dst) )
|
||||
|
||||
static const uint64_t ff_pw_32 attribute_used __attribute__ ((aligned(8))) = 0x0020002000200020ULL;
|
||||
|
||||
void ff_h264_idct_add_mmx2(uint8_t *dst, int16_t *block, int stride)
|
||||
{
|
||||
/* Load dct coeffs */
|
||||
asm volatile(
|
||||
"movq (%0), %%mm0 \n\t"
|
||||
"movq 8(%0), %%mm1 \n\t"
|
||||
"movq 16(%0), %%mm2 \n\t"
|
||||
"movq 24(%0), %%mm3 \n\t"
|
||||
:: "r"(block) );
|
||||
|
||||
asm volatile(
|
||||
/* mm1=s02+s13 mm2=s02-s13 mm4=d02+d13 mm0=d02-d13 */
|
||||
IDCT4_1D( %%mm2, %%mm1, %%mm0, %%mm3, %%mm4, %%mm5 )
|
||||
|
||||
/* in: 1,4,0,2 out: 1,2,3,0 */
|
||||
TRANSPOSE4( %%mm1, %%mm4, %%mm0, %%mm2, %%mm3 )
|
||||
|
||||
/* mm2=s02+s13 mm3=s02-s13 mm4=d02+d13 mm1=d02-d13 */
|
||||
IDCT4_1D( %%mm3, %%mm2, %%mm1, %%mm0, %%mm4, %%mm5 )
|
||||
|
||||
/* in: 2,4,1,3 out: 2,3,0,1 */
|
||||
TRANSPOSE4( %%mm2, %%mm4, %%mm1, %%mm3, %%mm0 )
|
||||
|
||||
"pxor %%mm7, %%mm7 \n\t"
|
||||
"movq ff_pw_32, %%mm6 \n\t"
|
||||
:: );
|
||||
|
||||
STORE_DIFF_4P( %%mm2, %%mm4, %%mm6, %%mm7, &dst[0*stride] );
|
||||
STORE_DIFF_4P( %%mm3, %%mm4, %%mm6, %%mm7, &dst[1*stride] );
|
||||
STORE_DIFF_4P( %%mm0, %%mm4, %%mm6, %%mm7, &dst[2*stride] );
|
||||
STORE_DIFF_4P( %%mm1, %%mm4, %%mm6, %%mm7, &dst[3*stride] );
|
||||
}
|
||||
|
Loading…
Reference in New Issue
Block a user