/*
 * DSP utils : average functions are compiled twice for 3dnow/mmx2
 * Copyright (c) 2000, 2001 Fabrice Bellard.
 * Copyright (c) 2002 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);
}

