/**************************************************************************
 *
 *  XVID MPEG-4 VIDEO CODEC
 *  -  Motion sad header  -
 *
 *  This program is an implementation of a part of one or more MPEG-4
 *  Video tools as specified in ISO/IEC 14496-2 standard.  Those intending
 *  to use this software module in hardware or software products are
 *  advised that its use may infringe existing patents or copyrights, and
 *  any such use would be at such party's own risk.  The original
 *  developer of this software module and his/her company, and subsequent
 *  editors and their companies, will have no liability for use of this
 *  software or modifications or derivatives thereof.
 *
 *  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., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
 *
 *  $Id: motion.h,v 1.13.2.5 2002/11/02 15:52:31 chl Exp $
 *
 ***************************************************************************/

#ifndef _MOTION_H_
#define _MOTION_H_

#include "../portab.h"
#include "../global.h"

// fast ((A)/2)*2
#define EVEN(A)		(((A)<0?(A)+1:(A)) & ~1)

#define MVzero(A) ( ((A).x)==(0) && ((A).y)==(0) )
#define MVequal(A,B) ( ((A).x)==((B).x) && ((A).y)==((B).y) )

static const uint32_t roundtab[16] =
		{ 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 2, 2 };

/*
 * modified rounding tables
 * original tables see ISO spec tables 7-6 -> 7-9
 */

/* K = 4 */
static const uint32_t roundtab_76[16] = { 0, 0, 0, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0, 0, 1, 1 };
/* K = 2 */
static const uint32_t roundtab_78[8] = { 0, 0, 1, 1, 0, 0, 0, 1  };
/* K = 1 */
static const uint32_t roundtab_79[4] = { 0, 1, 0, 0 };


/*
 * getref: calculate reference image pointer
 * the decision to use interpolation h/v/hv or the normal image is
 * based on dx & dy.
 */

static __inline const uint8_t *
get_ref(const uint8_t * const refn,
		const uint8_t * const refh,
		const uint8_t * const refv,
		const uint8_t * const refhv,
		const uint32_t x,
		const uint32_t y,
		const uint32_t block,	/* block dimension, 8 or 16 */

		const int32_t dx,
		const int32_t dy,
		const uint32_t stride)
{


	switch (((dx & 1) << 1) + (dy & 1)) {	/* ((dx%2)?2:0)+((dy%2)?1:0) */
	case 0:
		return refn + (int) ((x * block + dx / 2) + (y * block + dy / 2) * stride);
	case 1:
		return refv + (int) ((x * block + dx / 2) + (y * block +
											  (dy - 1) / 2) * stride);
	case 2:
		return refh + (int) ((x * block + (dx - 1) / 2) + (y * block +
													dy / 2) * stride);
	default:
		return refhv + (int) ((x * block + (dx - 1) / 2) + (y * block +
													 (dy - 1) / 2) * stride);
	}

}


/* This is somehow a copy of get_ref, but with MV instead of X,Y */

static __inline const uint8_t *
get_ref_mv(const uint8_t * const refn,
		   const uint8_t * const refh,
		   const uint8_t * const refv,
		   const uint8_t * const refhv,
		   const uint32_t x,
		   const uint32_t y,
		   const uint32_t block,	/* block dimension, 8 or 16 */

		   const VECTOR * mv,	/* measured in half-pel! */

		   const uint32_t stride)
{

	switch ((((mv->x) & 1) << 1) + ((mv->y) & 1)) {
	case 0:
		return refn + (int) ((x * block + (mv->x) / 2) + (y * block +
												   (mv->y) / 2) * stride);
	case 1:
		return refv + (int) ((x * block + (mv->x) / 2) + (y * block +
												   ((mv->y) - 1) / 2) * stride);
	case 2:
		return refh + (int) ((x * block + ((mv->x) - 1) / 2) + (y * block +
														 (mv->y) / 2) * stride);
	default:
		return refhv + (int) ((x * block + ((mv->x) - 1) / 2) + (y * block +
														  ((mv->y) -
														   1) / 2) * stride);
	}

}

void MotionEstimationBVOP(MBParam * const pParam,
						  FRAMEINFO * const frame,
						  // forward (past) reference 
						  const int32_t time_bp,
						  const int32_t time_pp,
						  const MACROBLOCK * const f_mbs,
						  const IMAGE * const f_ref,
						  const IMAGE * const f_refH,
						  const IMAGE * const f_refV,
						  const IMAGE * const f_refHV,
						  // backward (future) reference
						  const MACROBLOCK * const b_mbs,
						  const IMAGE * const b_ref,
						  const IMAGE * const b_refH,
						  const IMAGE * const b_refV,
						  const IMAGE * const b_refHV);

void MBMotionCompensationBVOP(MBParam * pParam,
							  MACROBLOCK * const mb,
							  const uint32_t i,
							  const uint32_t j,
							  IMAGE * const cur,
							  const IMAGE * const f_ref,
							  const IMAGE * const f_refh,
							  const IMAGE * const f_refv,
							  const IMAGE * const f_refhv,
							  const IMAGE * const b_ref,
							  const IMAGE * const b_refh,
							  const IMAGE * const b_refv,
							  const IMAGE * const b_refhv,
							  int16_t * dct_codes);

void
MotionEstimationHinted(	MBParam * const pParam,
						FRAMEINFO * const current,
						FRAMEINFO * const reference,
						const IMAGE * const pRefH,
						const IMAGE * const pRefV,
						const IMAGE * const pRefHV);

int
MEanalysis(	const IMAGE * const pRef,	
			const IMAGE * const pCurrent,
			MBParam * const pParam,
			MACROBLOCK * const pMBs,
			const uint32_t iFcode);

int
FindFcode(	const MBParam * const pParam,
			const FRAMEINFO * const current);

/* used internally only */

static VECTOR
GlobalMotionEst(const MACROBLOCK * const pMBs, 
				const MBParam * const pParam, const uint32_t iFcode);
				

#endif							/* _MOTION_H_ */