Mercurial > libavcodec.hg

view i386/dsputil_mmx_avg.h @ 1746:67949dc3117b libavcodec

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
dont calculate stuff if its not used ...
author michael
date Wed, 14 Jan 2004 02:35:04 +0000
parents 07a484280a82
children f65d87bfdd5a
line wrap: on
line source

/*
 * DSP utils : average functions are compiled twice for 3dnow/mmx2
 * Copyright (c) 2000, 2001 Fabrice Bellard.
 * Copyright (c) 2002-2004 Michael Niedermayer
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2 of the License, or (at your option) any later version.
 *
 * This library 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
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 *
 * MMX optimization by Nick Kurshev <nickols_k@mail.ru>
 * mostly rewritten by Michael Niedermayer <michaelni@gmx.at>
 * and improved by Zdenek Kabelac <kabi@users.sf.net>
 */
 
/* XXX: we use explicit registers to avoid a gcc 2.95.2 register asm
   clobber bug - now it will work with 2.95.2 and also with -fPIC
 */
static void DEF(put_pixels8_x2)(uint8_t *block, const uint8_t *pixels, int line_size, int h)
{
    __asm __volatile(
	"lea (%3, %3), %%eax		\n\t"
	"1:				\n\t"
	"movq (%1), %%mm0		\n\t"
	"movq (%1, %3), %%mm1		\n\t"
	PAVGB" 1(%1), %%mm0		\n\t"
	PAVGB" 1(%1, %3), %%mm1		\n\t"
	"movq %%mm0, (%2)		\n\t"
	"movq %%mm1, (%2, %3)		\n\t"
	"addl %%eax, %1			\n\t"
	"addl %%eax, %2			\n\t"
	"movq (%1), %%mm0		\n\t"
	"movq (%1, %3), %%mm1		\n\t"
	PAVGB" 1(%1), %%mm0		\n\t"
	PAVGB" 1(%1, %3), %%mm1		\n\t"
	"addl %%eax, %1			\n\t"
	"movq %%mm0, (%2)		\n\t"
	"movq %%mm1, (%2, %3)		\n\t"
	"addl %%eax, %2			\n\t"
	"subl $4, %0			\n\t"
	"jnz 1b				\n\t"
	:"+g"(h), "+S"(pixels), "+D"(block)
	:"r" (line_size)
	:"%eax", "memory");
}

static void DEF(put_pixels8_l2)(uint8_t *dst, uint8_t *src1, uint8_t *src2, int dstStride, int src1Stride, int h)
{
    __asm __volatile(
	"1:				\n\t"
	"movq	(%1), %%mm0		\n\t"
	"addl	%4, %1			\n\t"
	"movq	(%1), %%mm1		\n\t"
	"addl	%4, %1			\n\t"
	PAVGB" (%2), %%mm0		\n\t"
	PAVGB" 8(%2), %%mm1		\n\t"
	"movq	%%mm0, (%3)		\n\t"
	"addl	%5, %3			\n\t"
	"movq	%%mm1, (%3)		\n\t"
	"addl	%5, %3			\n\t"
	"movq	(%1), %%mm0		\n\t"
	"addl	%4, %1			\n\t"
	"movq	(%1), %%mm1		\n\t"
	"addl	%4, %1			\n\t"
	PAVGB" 16(%2), %%mm0		\n\t"
	PAVGB" 24(%2), %%mm1		\n\t"
	"movq	%%mm0, (%3)		\n\t"
	"addl	%5, %3			\n\t"
	"movq	%%mm1, (%3)		\n\t"
	"addl	%5, %3			\n\t"
        "addl	$32, %2			\n\t"
	"subl	$4, %0			\n\t"
	"jnz	1b			\n\t"
	:"+g"(h), "+r"(src1), "+r"(src2), "+r"(dst)
	:"r"(src1Stride), "r"(dstStride)
	:"memory");
}

static void DEF(put_pixels16_x2)(uint8_t *block, const uint8_t *pixels, int line_size, int h)
{
    __asm __volatile(
	"lea (%3, %3), %%eax		\n\t"
	"1:				\n\t"
	"movq (%1), %%mm0		\n\t"
	"movq (%1, %3), %%mm1		\n\t"
	"movq 8(%1), %%mm2		\n\t"
	"movq 8(%1, %3), %%mm3		\n\t"
	PAVGB" 1(%1), %%mm0		\n\t"
	PAVGB" 1(%1, %3), %%mm1		\n\t"
	PAVGB" 9(%1), %%mm2		\n\t"
	PAVGB" 9(%1, %3), %%mm3		\n\t"
	"movq %%mm0, (%2)		\n\t"
	"movq %%mm1, (%2, %3)		\n\t"
	"movq %%mm2, 8(%2)		\n\t"
	"movq %%mm3, 8(%2, %3)		\n\t"
	"addl %%eax, %1			\n\t"
	"addl %%eax, %2			\n\t"
	"movq (%1), %%mm0		\n\t"
	"movq (%1, %3), %%mm1		\n\t"
	"movq 8(%1), %%mm2		\n\t"
	"movq 8(%1, %3), %%mm3		\n\t"
	PAVGB" 1(%1), %%mm0		\n\t"
	PAVGB" 1(%1, %3), %%mm1		\n\t"
	PAVGB" 9(%1), %%mm2		\n\t"
	PAVGB" 9(%1, %3), %%mm3		\n\t"
	"addl %%eax, %1			\n\t"
	"movq %%mm0, (%2)		\n\t"
	"movq %%mm1, (%2, %3)		\n\t"
	"movq %%mm2, 8(%2)		\n\t"
	"movq %%mm3, 8(%2, %3)		\n\t"
	"addl %%eax, %2			\n\t"
	"subl $4, %0			\n\t"
	"jnz 1b				\n\t"
	:"+g"(h), "+S"(pixels), "+D"(block)
	:"r" (line_size)
	:"%eax", "memory");
}

static void DEF(put_pixels16_l2)(uint8_t *dst, uint8_t *src1, uint8_t *src2, int dstStride, int src1Stride, int h)
{
    __asm __volatile(
	"1:				\n\t"
	"movq	(%1), %%mm0		\n\t"
	"movq	8(%1), %%mm1		\n\t"
	"addl	%4, %1			\n\t"
	PAVGB" (%2), %%mm0		\n\t"
	PAVGB" 8(%2), %%mm1		\n\t"
	"movq	%%mm0, (%3)		\n\t"
	"movq	%%mm1, 8(%3)		\n\t"
	"addl	%5, %3			\n\t"
	"movq	(%1), %%mm0		\n\t"
	"movq	8(%1), %%mm1		\n\t"
	"addl	%4, %1			\n\t"
	PAVGB" 16(%2), %%mm0		\n\t"
	PAVGB" 24(%2), %%mm1		\n\t"
	"movq	%%mm0, (%3)		\n\t"
	"movq	%%mm1, 8(%3)		\n\t"
	"addl	%5, %3			\n\t"
        "addl	$32, %2			\n\t"
	"subl	$2, %0			\n\t"
	"jnz	1b			\n\t"
	:"+g"(h), "+r"(src1), "+r"(src2), "+r"(dst)
	:"r"(src1Stride), "r"(dstStride)
	:"memory");
}
 
/* GL: this function does incorrect rounding if overflow */
static void DEF(put_no_rnd_pixels8_x2)(uint8_t *block, const uint8_t *pixels, int line_size, int h)
{
    MOVQ_BONE(mm6);
    __asm __volatile(
	"lea (%3, %3), %%eax		\n\t"
	"1:				\n\t"
	"movq (%1), %%mm0		\n\t"
	"movq (%1, %3), %%mm2		\n\t"
	"movq 1(%1), %%mm1		\n\t"
	"movq 1(%1, %3), %%mm3		\n\t"
	"addl %%eax, %1			\n\t"
	"psubusb %%mm6, %%mm0		\n\t"
	"psubusb %%mm6, %%mm2		\n\t"
	PAVGB" %%mm1, %%mm0		\n\t"
	PAVGB" %%mm3, %%mm2		\n\t"
	"movq %%mm0, (%2)		\n\t"
	"movq %%mm2, (%2, %3)		\n\t"
	"movq (%1), %%mm0		\n\t"
	"movq 1(%1), %%mm1		\n\t"
	"movq (%1, %3), %%mm2		\n\t"
	"movq 1(%1, %3), %%mm3		\n\t"
	"addl %%eax, %2			\n\t"
	"addl %%eax, %1			\n\t"
	"psubusb %%mm6, %%mm0		\n\t"
	"psubusb %%mm6, %%mm2		\n\t"
	PAVGB" %%mm1, %%mm0		\n\t"
	PAVGB" %%mm3, %%mm2		\n\t"
	"movq %%mm0, (%2)		\n\t"
	"movq %%mm2, (%2, %3)		\n\t"
	"addl %%eax, %2			\n\t"
	"subl $4, %0			\n\t"
	"jnz 1b				\n\t"
	:"+g"(h), "+S"(pixels), "+D"(block)
	:"r" (line_size)
	:"%eax", "memory");
}

static void DEF(put_pixels8_y2)(uint8_t *block, const uint8_t *pixels, int line_size, int h)
{
    __asm __volatile(
	"lea (%3, %3), %%eax		\n\t"
	"movq (%1), %%mm0		\n\t"
	"subl %3, %2			\n\t"
	"1:				\n\t"
	"movq (%1, %3), %%mm1		\n\t"
	"movq (%1, %%eax), %%mm2	\n\t"
	"addl %%eax, %1			\n\t"
	PAVGB" %%mm1, %%mm0		\n\t"
	PAVGB" %%mm2, %%mm1		\n\t"
	"movq %%mm0, (%2, %3)		\n\t"
	"movq %%mm1, (%2, %%eax)	\n\t"
	"movq (%1, %3), %%mm1		\n\t"
	"movq (%1, %%eax), %%mm0	\n\t"
	"addl %%eax, %2			\n\t"
	"addl %%eax, %1			\n\t"
	PAVGB" %%mm1, %%mm2		\n\t"
	PAVGB" %%mm0, %%mm1		\n\t"
	"movq %%mm2, (%2, %3)		\n\t"
	"movq %%mm1, (%2, %%eax)	\n\t"
	"addl %%eax, %2			\n\t"
	"subl $4, %0			\n\t"
	"jnz 1b				\n\t"
	:"+g"(h), "+S"(pixels), "+D" (block)
	:"r" (line_size)
	:"%eax", "memory");
}

/* GL: this function does incorrect rounding if overflow */
static void DEF(put_no_rnd_pixels8_y2)(uint8_t *block, const uint8_t *pixels, int line_size, int h)
{
    MOVQ_BONE(mm6);
    __asm __volatile(
	"lea (%3, %3), %%eax		\n\t"
	"movq (%1), %%mm0		\n\t"
	"subl %3, %2			\n\t"
	"1:				\n\t"
	"movq (%1, %3), %%mm1		\n\t"
	"movq (%1, %%eax), %%mm2	\n\t"
	"addl %%eax, %1			\n\t"
	"psubusb %%mm6, %%mm1		\n\t"
	PAVGB" %%mm1, %%mm0		\n\t"
	PAVGB" %%mm2, %%mm1		\n\t"
	"movq %%mm0, (%2, %3)		\n\t"
	"movq %%mm1, (%2, %%eax)	\n\t"
	"movq (%1, %3), %%mm1		\n\t"
	"movq (%1, %%eax), %%mm0	\n\t"
	"addl %%eax, %2			\n\t"
	"addl %%eax, %1			\n\t"
	"psubusb %%mm6, %%mm1		\n\t"
	PAVGB" %%mm1, %%mm2		\n\t"
	PAVGB" %%mm0, %%mm1		\n\t"
	"movq %%mm2, (%2, %3)		\n\t"
	"movq %%mm1, (%2, %%eax)	\n\t"
	"addl %%eax, %2			\n\t"
	"subl $4, %0			\n\t"
	"jnz 1b				\n\t"
	:"+g"(h), "+S"(pixels), "+D" (block)
	:"r" (line_size)
	:"%eax", "memory");
}

static void DEF(avg_pixels8)(uint8_t *block, const uint8_t *pixels, int line_size, int h)
{
    __asm __volatile(
	"lea (%3, %3), %%eax		\n\t"
	"1:				\n\t"
	"movq (%2), %%mm0		\n\t"
	"movq (%2, %3), %%mm1		\n\t"
	PAVGB" (%1), %%mm0		\n\t"
	PAVGB" (%1, %3), %%mm1		\n\t"
	"movq %%mm0, (%2)		\n\t"
	"movq %%mm1, (%2, %3)		\n\t"
	"addl %%eax, %1			\n\t"
	"addl %%eax, %2			\n\t"
	"movq (%2), %%mm0		\n\t"
	"movq (%2, %3), %%mm1		\n\t"
	PAVGB" (%1), %%mm0		\n\t"
	PAVGB" (%1, %3), %%mm1		\n\t"
	"addl %%eax, %1			\n\t"
	"movq %%mm0, (%2)		\n\t"
	"movq %%mm1, (%2, %3)		\n\t"
	"addl %%eax, %2			\n\t"
	"subl $4, %0			\n\t"
	"jnz 1b				\n\t"
	:"+g"(h), "+S"(pixels), "+D"(block)
	:"r" (line_size)
	:"%eax", "memory");
}

static void DEF(avg_pixels8_x2)(uint8_t *block, const uint8_t *pixels, int line_size, int h)
{
    __asm __volatile(
	"lea (%3, %3), %%eax		\n\t"
	"1:				\n\t"
	"movq (%1), %%mm0		\n\t"
	"movq (%1, %3), %%mm2		\n\t"
	PAVGB" 1(%1), %%mm0		\n\t"
	PAVGB" 1(%1, %3), %%mm2		\n\t"
	PAVGB" (%2), %%mm0		\n\t"
	PAVGB" (%2, %3), %%mm2		\n\t"
	"addl %%eax, %1			\n\t"
	"movq %%mm0, (%2)		\n\t"
	"movq %%mm2, (%2, %3)		\n\t"
	"movq (%1), %%mm0		\n\t"
	"movq (%1, %3), %%mm2		\n\t"
	PAVGB" 1(%1), %%mm0		\n\t"
	PAVGB" 1(%1, %3), %%mm2		\n\t"
	"addl %%eax, %2			\n\t"
	"addl %%eax, %1			\n\t"
	PAVGB" (%2), %%mm0		\n\t"
	PAVGB" (%2, %3), %%mm2		\n\t"
	"movq %%mm0, (%2)		\n\t"
	"movq %%mm2, (%2, %3)		\n\t"
	"addl %%eax, %2			\n\t"
	"subl $4, %0			\n\t"
	"jnz 1b				\n\t"
	:"+g"(h), "+S"(pixels), "+D"(block)
	:"r" (line_size)
	:"%eax", "memory");
}

static void DEF(avg_pixels8_y2)(uint8_t *block, const uint8_t *pixels, int line_size, int h)
{
    __asm __volatile(
	"lea (%3, %3), %%eax		\n\t"
	"movq (%1), %%mm0		\n\t"
	"subl %3, %2			\n\t"
	"1:				\n\t"
	"movq (%1, %3), %%mm1		\n\t"
	"movq (%1, %%eax), %%mm2	\n\t"
	"addl %%eax, %1			\n\t"
	PAVGB" %%mm1, %%mm0		\n\t"
	PAVGB" %%mm2, %%mm1		\n\t"
	"movq (%2, %3), %%mm3		\n\t"
	"movq (%2, %%eax), %%mm4	\n\t"
	PAVGB" %%mm3, %%mm0		\n\t"
	PAVGB" %%mm4, %%mm1		\n\t"
	"movq %%mm0, (%2, %3)		\n\t"
	"movq %%mm1, (%2, %%eax)	\n\t"
	"movq (%1, %3), %%mm1		\n\t"
	"movq (%1, %%eax), %%mm0	\n\t"
	PAVGB" %%mm1, %%mm2		\n\t"
	PAVGB" %%mm0, %%mm1		\n\t"
	"addl %%eax, %2			\n\t"
	"addl %%eax, %1			\n\t"
	"movq (%2, %3), %%mm3		\n\t"
	"movq (%2, %%eax), %%mm4	\n\t"
	PAVGB" %%mm3, %%mm2		\n\t"
	PAVGB" %%mm4, %%mm1		\n\t"
	"movq %%mm2, (%2, %3)		\n\t"
	"movq %%mm1, (%2, %%eax)	\n\t"
	"addl %%eax, %2			\n\t"
	"subl $4, %0			\n\t"
	"jnz 1b				\n\t"
	:"+g"(h), "+S"(pixels), "+D"(block)
	:"r" (line_size)
	:"%eax", "memory");
}

// Note this is not correctly rounded, but this function is only used for b frames so it doesnt matter 
static void DEF(avg_pixels8_xy2)(uint8_t *block, const uint8_t *pixels, int line_size, int h)
{
    MOVQ_BONE(mm6);
    __asm __volatile(
	"lea (%3, %3), %%eax		\n\t"
	"movq (%1), %%mm0		\n\t"
	PAVGB" 1(%1), %%mm0		\n\t"
	".balign 8			\n\t"
	"1:				\n\t"
	"movq (%1, %%eax), %%mm2	\n\t"
	"movq (%1, %3), %%mm1		\n\t"
	"psubusb %%mm6, %%mm2		\n\t"
	PAVGB" 1(%1, %3), %%mm1		\n\t"
	PAVGB" 1(%1, %%eax), %%mm2	\n\t"
	"addl %%eax, %1			\n\t"
	PAVGB" %%mm1, %%mm0		\n\t"
	PAVGB" %%mm2, %%mm1		\n\t"
	PAVGB" (%2), %%mm0		\n\t"
	PAVGB" (%2, %3), %%mm1		\n\t"
	"movq %%mm0, (%2)		\n\t"
	"movq %%mm1, (%2, %3)		\n\t"
	"movq (%1, %3), %%mm1		\n\t"
	"movq (%1, %%eax), %%mm0	\n\t"
	PAVGB" 1(%1, %3), %%mm1		\n\t"
	PAVGB" 1(%1, %%eax), %%mm0	\n\t"
	"addl %%eax, %2			\n\t"
	"addl %%eax, %1			\n\t"
	PAVGB" %%mm1, %%mm2		\n\t"
	PAVGB" %%mm0, %%mm1		\n\t"
	PAVGB" (%2), %%mm2		\n\t"
	PAVGB" (%2, %3), %%mm1		\n\t"
	"movq %%mm2, (%2)		\n\t"
	"movq %%mm1, (%2, %3)		\n\t"
	"addl %%eax, %2			\n\t"
	"subl $4, %0			\n\t"
	"jnz 1b				\n\t"
	:"+g"(h), "+S"(pixels), "+D"(block)
	:"r" (line_size)
	:"%eax",  "memory");
}

//FIXME the following could be optimized too ...
static void DEF(put_no_rnd_pixels16_x2)(uint8_t *block, const uint8_t *pixels, int line_size, int h){
    DEF(put_no_rnd_pixels8_x2)(block  , pixels  , line_size, h);
    DEF(put_no_rnd_pixels8_x2)(block+8, pixels+8, line_size, h);
}
static void DEF(put_pixels16_y2)(uint8_t *block, const uint8_t *pixels, int line_size, int h){
    DEF(put_pixels8_y2)(block  , pixels  , line_size, h);
    DEF(put_pixels8_y2)(block+8, pixels+8, line_size, h);
}
static void DEF(put_no_rnd_pixels16_y2)(uint8_t *block, const uint8_t *pixels, int line_size, int h){
    DEF(put_no_rnd_pixels8_y2)(block  , pixels  , line_size, h);
    DEF(put_no_rnd_pixels8_y2)(block+8, pixels+8, line_size, h);
}
static void DEF(avg_pixels16)(uint8_t *block, const uint8_t *pixels, int line_size, int h){
    DEF(avg_pixels8)(block  , pixels  , line_size, h);
    DEF(avg_pixels8)(block+8, pixels+8, line_size, h);
}
static void DEF(avg_pixels16_x2)(uint8_t *block, const uint8_t *pixels, int line_size, int h){
    DEF(avg_pixels8_x2)(block  , pixels  , line_size, h);
    DEF(avg_pixels8_x2)(block+8, pixels+8, line_size, h);
}
static void DEF(avg_pixels16_y2)(uint8_t *block, const uint8_t *pixels, int line_size, int h){
    DEF(avg_pixels8_y2)(block  , pixels  , line_size, h);
    DEF(avg_pixels8_y2)(block+8, pixels+8, line_size, h);
}
static void DEF(avg_pixels16_xy2)(uint8_t *block, const uint8_t *pixels, int line_size, int h){
    DEF(avg_pixels8_xy2)(block  , pixels  , line_size, h);
    DEF(avg_pixels8_xy2)(block+8, pixels+8, line_size, h);
}