diff options
author | Fabrice Bellard <fabrice@bellard.org> | 2001-08-16 02:25:32 +0400 |
---|---|---|
committer | Fabrice Bellard <fabrice@bellard.org> | 2001-08-16 02:25:32 +0400 |
commit | 694ec0612b0a9d74d98a657b2425666c43996179 (patch) | |
tree | 5985143c9137e361cd21ebae2491c1cebbc12439 /libavcodec/i386/motion_est_mmx.c | |
parent | c72c6d2d12c00090304c4532c82ae2371041a31c (diff) |
suppressed nasm dependancy - rewrote forward DCT and motion estimation code
Originally committed as revision 104 to svn://svn.ffmpeg.org/ffmpeg/trunk
Diffstat (limited to 'libavcodec/i386/motion_est_mmx.c')
-rw-r--r-- | libavcodec/i386/motion_est_mmx.c | 244 |
1 files changed, 244 insertions, 0 deletions
diff --git a/libavcodec/i386/motion_est_mmx.c b/libavcodec/i386/motion_est_mmx.c new file mode 100644 index 0000000000..35b16b711c --- /dev/null +++ b/libavcodec/i386/motion_est_mmx.c @@ -0,0 +1,244 @@ +/* + * MMX optimized motion estimation + * Copyright (c) 2001 Gerard Lantau. + * + * This program is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation; either version 2 of the License, or + * (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program; if not, write to the Free Software + * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. + * + */ +#include "../dsputil.h" +#include "mmx.h" + +static const unsigned long long int mm_wone __attribute__ ((aligned(8))) = 0x0001000100010001; +static const unsigned long long int mm_wtwo __attribute__ ((aligned(8))) = 0x0002000200020002; + +/* mm7 is accumulator, mm6 is zero */ +static inline void sad_add(const UINT8 *p1, const UINT8 *p2) +{ + movq_m2r(*p1, mm0); + movq_m2r(*p2, mm1); + movq_r2r(mm0, mm2); + psubusb_r2r(mm1, mm0); + psubusb_r2r(mm2, mm1); + por_r2r(mm1, mm0); /* mm0 is absolute value */ + + movq_r2r(mm0, mm1); + punpcklbw_r2r(mm6, mm0); + punpckhbw_r2r(mm6, mm1); + paddusw_r2r(mm0, mm7); + paddusw_r2r(mm1, mm7); +} + +/* convert mm7 to value */ +static inline int sad_end(void) +{ + int res; + + movq_r2r(mm7, mm0); + psrlq_i2r(32, mm7); + paddusw_r2r(mm0, mm7); + + movq_r2r(mm7, mm0); + psrlq_i2r(16, mm7); + paddusw_r2r(mm0, mm7); + __asm __volatile ("movd %%mm7, %0" : "=a" (res)); + return res & 0xffff; +} + +int pix_abs16x16_mmx(UINT8 *blk1, UINT8 *blk2, int lx, int h) +{ + const UINT8 *p1, *p2; + + h >>= 1; + p1 = blk1; + p2 = blk2; + pxor_r2r(mm7, mm7); /* mm7 is accumulator */ + pxor_r2r(mm6, mm6); /* mm7 is zero constant */ + do { + sad_add(p1, p2); + sad_add(p1 + 8, p2 + 8); + p1 += lx; + p2 += lx; + sad_add(p1, p2); + sad_add(p1 + 8, p2 + 8); + p1 += lx; + p2 += lx; + } while (--h); + return sad_end(); +} + +/* please test it ! */ +static inline void sad_add_sse(const UINT8 *p1, const UINT8 *p2) +{ + movq_m2r(*(p1 + 0), mm0); + movq_m2r(*(p1 + 8), mm1); + psadbw_m2r(*(p2 + 0), mm0); + psadbw_m2r(*(p2 + 8), mm1); + paddusw_r2r(mm0, mm7); + paddusw_r2r(mm1, mm7); +} + +int pix_abs16x16_sse(UINT8 *blk1, UINT8 *blk2, int lx, int h) +{ + const UINT8 *p1, *p2; + + h >>= 1; + p1 = blk1; + p2 = blk2; + pxor_r2r(mm7, mm7); /* mm7 is accumulator */ + do { + sad_add_sse(p1, p2); + p1 += lx; + p2 += lx; + sad_add_sse(p1, p2); + p1 += lx; + p2 += lx; + } while (--h); + return sad_end(); +} + +#define DUMP(reg) { mmx_t tmp; movq_r2m(reg, tmp); printf(#reg "=%016Lx\n", tmp.uq); } + +/* mm7 is accumulator, mm6 is zero */ +static inline void sad_add_x2(const UINT8 *p1, const UINT8 *p2, const UINT8 *p3) +{ + movq_m2r(*(p2 + 0), mm0); + movq_m2r(*(p3 + 0), mm1); + movq_r2r(mm0, mm2); + movq_r2r(mm1, mm3); + punpcklbw_r2r(mm6, mm0); /* extract 4 bytes low */ + punpcklbw_r2r(mm6, mm1); + punpckhbw_r2r(mm6, mm2); /* high */ + punpckhbw_r2r(mm6, mm3); + paddusw_r2r(mm1, mm0); + paddusw_r2r(mm3, mm2); + movq_m2r(*(p1 + 0), mm1); /* mm1 : other value */ + paddusw_r2r(mm5, mm0); /* + 1 */ + paddusw_r2r(mm5, mm2); /* + 1 */ + psrlw_i2r(1, mm0); + psrlw_i2r(1, mm2); + packuswb_r2r(mm2, mm0); /* average is in mm0 */ + + movq_r2r(mm1, mm2); + psubusb_r2r(mm0, mm1); + psubusb_r2r(mm2, mm0); + por_r2r(mm1, mm0); /* mm0 is absolute value */ + + movq_r2r(mm0, mm1); + punpcklbw_r2r(mm6, mm0); + punpckhbw_r2r(mm6, mm1); + paddusw_r2r(mm0, mm7); + paddusw_r2r(mm1, mm7); +} + +int pix_abs16x16_x2_mmx(UINT8 *blk1, UINT8 *blk2, int lx, int h) +{ + const UINT8 *p1, *p2; + + p1 = blk1; + p2 = blk2; + pxor_r2r(mm7, mm7); /* mm7 is accumulator */ + pxor_r2r(mm6, mm6); /* mm7 is zero constant */ + movq_m2r(mm_wone, mm5); /* one constant */ + do { + sad_add_x2(p1, p2, p2 + 1); + sad_add_x2(p1 + 8, p2 + 8, p2 + 9); + p1 += lx; + p2 += lx; + } while (--h); + return sad_end(); +} + +int pix_abs16x16_y2_mmx(UINT8 *blk1, UINT8 *blk2, int lx, int h) +{ + const UINT8 *p1, *p2; + + p1 = blk1; + p2 = blk2; + pxor_r2r(mm7, mm7); /* mm7 is accumulator */ + pxor_r2r(mm6, mm6); /* mm7 is zero constant */ + movq_m2r(mm_wone, mm5); /* one constant */ + do { + sad_add_x2(p1, p2, p2 + lx); + sad_add_x2(p1 + 8, p2 + 8, p2 + 8 + lx); + p1 += lx; + p2 += lx; + } while (--h); + return sad_end(); +} + +/* mm7 is accumulator, mm6 is zero */ +static inline void sad_add_xy2(const UINT8 *p1, const UINT8 *p2, const UINT8 *p3) +{ + movq_m2r(*(p2 + 0), mm0); + movq_m2r(*(p3 + 0), mm1); + movq_r2r(mm0, mm2); + movq_r2r(mm1, mm3); + punpcklbw_r2r(mm6, mm0); /* extract 4 bytes low */ + punpcklbw_r2r(mm6, mm1); + punpckhbw_r2r(mm6, mm2); /* high */ + punpckhbw_r2r(mm6, mm3); + paddusw_r2r(mm1, mm0); + paddusw_r2r(mm3, mm2); + + movq_m2r(*(p2 + 1), mm1); + movq_m2r(*(p3 + 1), mm3); + movq_r2r(mm1, mm4); + punpcklbw_r2r(mm6, mm1); /* low */ + punpckhbw_r2r(mm6, mm4); /* high */ + paddusw_r2r(mm1, mm0); + paddusw_r2r(mm4, mm2); + movq_r2r(mm3, mm4); + punpcklbw_r2r(mm6, mm3); /* low */ + punpckhbw_r2r(mm6, mm4); /* high */ + paddusw_r2r(mm3, mm0); + paddusw_r2r(mm4, mm2); + + movq_m2r(*(p1 + 0), mm1); /* mm1 : other value */ + paddusw_r2r(mm5, mm0); /* + 2 */ + paddusw_r2r(mm5, mm2); /* + 2 */ + psrlw_i2r(2, mm0); + psrlw_i2r(2, mm2); + packuswb_r2r(mm2, mm0); /* average is in mm0 */ + + movq_r2r(mm1, mm2); + psubusb_r2r(mm0, mm1); + psubusb_r2r(mm2, mm0); + por_r2r(mm1, mm0); /* mm0 is absolute value */ + + movq_r2r(mm0, mm1); + punpcklbw_r2r(mm6, mm0); + punpckhbw_r2r(mm6, mm1); + paddusw_r2r(mm0, mm7); + paddusw_r2r(mm1, mm7); +} + +int pix_abs16x16_xy2_mmx(UINT8 *blk1, UINT8 *blk2, int lx, int h) +{ + const UINT8 *p1, *p2, *p3; + + p1 = blk1; + p2 = blk2; + p3 = blk2 + lx; + pxor_r2r(mm7, mm7); /* mm7 is accumulator */ + pxor_r2r(mm6, mm6); /* mm7 is zero constant */ + movq_m2r(mm_wtwo, mm5); /* one constant */ + do { + sad_add_xy2(p1, p2, p2 + lx); + sad_add_xy2(p1 + 8, p2 + 8, p2 + 8 + lx); + p1 += lx; + p2 += lx; + } while (--h); + return sad_end(); +} |