/***************************************************************************
                          ADM_xvid4.cpp  -  description
                             -------------------

	Xvid4 api

    begin                : Fri Jul 12 2002
    copyright            : (C) 2002 by mean
    email                : fixounet@free.fr
 ***************************************************************************/

/***************************************************************************
 *                                                                         *
 *   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.                                   *
 *                                                                         *
 ***************************************************************************/
#include "config.h"

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <math.h>

#include "ADM_library/default.h"
#include "ADM_codecs/ADM_codec.h"
#ifdef USE_XVID_4

#include "ADM_codecs/ADM_xvid4param.h"
#include "ADM_codecs/ADM_xvid4.h"
#include "xvid.h"
#include <ADM_assert.h>


static const int motion_presets[] = {	
			0,
			0,
			0,
			0,
			XVID_ME_HALFPELREFINE16,
			XVID_ME_HALFPELREFINE16 | XVID_ME_ADVANCEDDIAMOND16,
			XVID_ME_HALFPELREFINE16 | XVID_ME_EXTSEARCH16 |
			XVID_ME_HALFPELREFINE8  | XVID_ME_USESQUARES16
};

#define ME_ELEMENTS (sizeof(motion_presets)/sizeof(motion_presets[0]))


#define VOP_ELEMENTS (sizeof(vop_presets)/sizeof(vop_presets[0]))

   

static myENC_RESULT xvid_res;
static 	xvid_gbl_init_t xvid_gbl_init2;
static 	xvid_enc_create_t xvid_enc_create;
static  xvid_plugin_single_t single;
static  xvid_plugin_2pass1_t pass1;
static  xvid_plugin_2pass2_t pass2;
static  xvid_enc_plugin_t plugins[7];
static  xvid_enc_frame_t xvid_enc_frame;
static 	xvid_enc_stats_t xvid_enc_stats;
static  xvid_gbl_info_t xvid_gbl_info;
#define MMSET(x) memset(&x,0,sizeof(x))

void xvid4_init(void);
/*
	System wide init, do it once for all 
*/
void xvid4_init(void)
{
	MMSET(xvid_gbl_init2);
	MMSET(xvid_gbl_info);
	
	printf("Initializing global xvid 4\n");
	xvid_gbl_init2.version = XVID_VERSION;
	xvid_global(NULL, XVID_GBL_INIT, &xvid_gbl_init2, NULL);
	xvid_gbl_info.version = XVID_VERSION;
	xvid_global(NULL, XVID_GBL_INFO, &xvid_gbl_info, NULL);
	if(xvid_gbl_info.build)
		{
			printf("\txvid build:%s\n",xvid_gbl_info.build);
		}
	printf("\txvid thread:%d\n",xvid_gbl_info.num_threads);
	printf("\txvid SIMD supported:(%x)\n",xvid_gbl_info.cpu_flags);
	#define CPUF(x) if(xvid_gbl_info.cpu_flags  & XVID_CPU_##x) printf("\t\t"#x"\n");	
#if defined( ARCH_X86)  || defined(ARCH_X86_64)
	CPUF(MMX);
	CPUF(MMXEXT);
	CPUF(SSE);
	CPUF(SSE2);
	CPUF(3DNOW);
	CPUF(3DNOWEXT);
#endif	
#ifdef USE_ALTIVEC	
	CPUF(ALTIVEC);
#endif	
	
}
xvid4Encoder::~xvid4Encoder()
{
	stopEncoder();
}
//--------------------------------------------------

uint8_t xvid4Encoder::init(uint32_t q,uint32_t fps1000,xvid4EncParam *param)
{
	return 0;
}


uint8_t 	xvid4Encoder::getResult( void *ress)
{
 		myENC_RESULT *res;

		res=(myENC_RESULT *)ress;
		memcpy(res,&xvid_res,sizeof(myENC_RESULT));
		return 1;

}
void     xvid4Encoder::createUpdate(void )
{
		xvid_enc_create.max_bframes=_param.bframes;
		xvid_enc_create.max_key_interval=_param.max_key_interval;		
		xvid_enc_create.bquant_ratio = 	_param.bquant_ratio;
		xvid_enc_create.bquant_offset = 	_param.bquant_offset;
		
		if (_param.packed)
			xvid_enc_create.global |= XVID_GLOBAL_PACKED;
		
		if (_param.closed_gop)
			xvid_enc_create.global |= XVID_GLOBAL_CLOSED_GOP;
		
		for(uint32_t i=0;i<3;i++)
		{
			xvid_enc_create.min_quant[i]=_param.qmin[i];
			xvid_enc_create.max_quant[i]=_param.qmax[i];
		}

		preAmble(NULL);
#define C_VOP(x) if(xvid_enc_frame.vop_flags& XVID_VOP_##x) printf("Vop :"#x" is set\n");		
#define C_VOL(x) if(xvid_enc_frame.vol_flags& XVID_VOL_##x) printf("Vol :"#x" is set\n");
#define C_ME(x) if(xvid_enc_frame.motion& XVID_ME_##x) printf("ME :"#x" is set\n");

	C_VOP(DEBUG);
	C_VOP(CHROMAOPT);
	C_VOP(HALFPEL);
	C_VOP(INTER4V);
	C_VOP(TRELLISQUANT);
	C_VOP(CHROMAOPT);
	C_VOP(CARTOON);
	C_VOP(GREYSCALE);
	C_VOP(HQACPRED);
	C_VOP(MODEDECISION_RD);
	
	C_VOL(MPEGQUANT);
	C_VOL(EXTRASTATS);
	C_VOL(GMC);
	C_VOL(QUARTERPEL);
	C_VOL(INTERLACING);
	
	C_ME(ADVANCEDDIAMOND16);
	C_ME(ADVANCEDDIAMOND8);
	
	C_ME(USESQUARES16);
	C_ME(USESQUARES8);
	
	C_ME(HALFPELREFINE16);
	C_ME(HALFPELREFINE8);
	
	// Turbo Mode
	C_ME(FASTREFINE16);
	C_ME(FASTREFINE8);
	C_ME(SKIP_DELTASEARCH);
	C_ME(FAST_MODEINTERPOLATE);
	C_ME(BFRAME_EARLYSTOP);
	
	C_ME(GME_REFINE);
	C_ME(QUARTERPELREFINE8);
	C_ME(QUARTERPELREFINE16);
	C_ME(EXTSEARCH16);
	C_ME(EXTSEARCH8);
	C_ME(CHROMA_BVOP);
	C_ME(CHROMA_PVOP);
	
	
		
}
void     xvid4Encoder::encUpdate(void )
{
		

}

void xvid4Encoder::dump(void)
{
	printf("Xvid4 codec configuration:\n");
	if(_param.mpegQuantizer)
		printf("\tMPEGQUANT\n");
	if(_param.interlaced)
		printf("\tINTERLACED\n");
		

#define ST(x,y) \
	if(_param.x) printf("\t"#y"\n");				
	
	
	ST(  inter4mv,  INTER4V);
	ST(  trellis,	TRELLISQUANT);
	ST(  cartoon,	CARTOON);
	ST(  greyscale, GREYSCALE);
#undef ST

}
//____________________________________________
//
//  Initialize the compressor
//
uint8_t     xvid4Encoder::stopEncoder(void )
{
	if(_handle)
	{
		xvid_encore(_handle, XVID_ENC_DESTROY, NULL, NULL);
		printf("Destroying xvid codec\n");
		_handle=NULL;
	}
    return 1;
}
uint8_t     xvid4Encoder::preAmble(uint8_t *in )
{
/* Version for the frame and the stats */
	MMSET(xvid_enc_frame);
	MMSET(xvid_enc_stats);
	
	xvid_enc_frame.version = XVID_VERSION;	
	xvid_enc_stats.version = XVID_VERSION;

	/* Bind output buffer */

	xvid_enc_frame.length = 0;
	if(_param.mpegQuantizer)
		xvid_enc_frame.vol_flags|=XVID_VOL_MPEGQUANT;
	if(_param.interlaced)
		xvid_enc_frame.vol_flags|=XVID_VOL_INTERLACING;
		
	
#define SVOP(x,y) if(_param.x) xvid_enc_frame.vop_flags|=XVID_VOP_##y
					
	
	xvid_enc_frame.motion = motion_presets[_param.guiLevel];
	
	SVOP(  inter4mv	,INTER4V);
	SVOP(  trellis	,TRELLISQUANT);
	SVOP(  hqac	,HQACPRED);
	SVOP(  greyscale,GREYSCALE);
	SVOP(  cartoon	,CARTOON);
	SVOP(  chroma_opt,CHROMAOPT);
	//SVOP(  1	,HALFPEL);
	xvid_enc_frame.vop_flags|=XVID_VOP_HALFPEL;	
	// ME 
	if(_param.chroma_me)
		{
			xvid_enc_frame.motion|=XVID_ME_CHROMA_BVOP;	
			xvid_enc_frame.motion|=XVID_ME_CHROMA_PVOP;
		}
	if(_param.cartoon) 
	{
			xvid_enc_frame.motion|=XVID_ME_DETECT_STATIC_MOTION;
	}
	//printf("Vhq mode : %d\n",_param.vhqmode);
#define VHME(x) xvid_enc_frame.motion|=	XVID_ME_##x
	switch(_param.vhqmode)
	{
		case 4: VHME(EXTSEARCH_RD);
		case 3: VHME(HALFPELREFINE8_RD);
			VHME(QUARTERPELREFINE8_RD);
			VHME(CHECKPREDICTION_RD);
		
		case 2:
			VHME(HALFPELREFINE16_RD);
			VHME(QUARTERPELREFINE16_RD);
		case 1:
			SVOP(vhqmode,MODEDECISION_RD);			
		case 0:
			break;
		default:
			ADM_assert(0);
	}	
	if(_param.qpel)
		{
			xvid_enc_frame.vol_flags|=XVID_VOL_QUARTERPEL;			
			
			xvid_enc_frame.motion    |= XVID_ME_QUARTERPELREFINE16;
			xvid_enc_frame.motion    |= XVID_ME_QUARTERPELREFINE8;	
		}
			
	if (_param.gmc)
		{
			xvid_enc_frame.vol_flags |= XVID_VOL_GMC;
			xvid_enc_frame.motion    |= XVID_ME_GME_REFINE;
		}
									
	if (_param.turbo)
		{
			xvid_enc_frame.motion    |= XVID_ME_FASTREFINE16;
			xvid_enc_frame.motion    |= XVID_ME_FASTREFINE8;
			xvid_enc_frame.motion    |= XVID_ME_SKIP_DELTASEARCH;
			xvid_enc_frame.motion	 |= XVID_ME_FAST_MODEINTERPOLATE;
			xvid_enc_frame.motion	 |= XVID_ME_BFRAME_EARLYSTOP;
		}
		
	xvid_enc_frame.bframe_threshold = 	_param.bframe_threshold;
		
	xvid_enc_frame.input.csp = XVID_CSP_YV12;
	xvid_enc_frame.input.stride[0] = _w;
	xvid_enc_frame.input.stride[1] = _w>>1;
	xvid_enc_frame.input.stride[2] = _w>>1;
	xvid_enc_frame.type = XVID_TYPE_AUTO;
	
	
	/* Set up motion estimation flags */
	xvid_enc_frame.input.plane[0] = in;
	xvid_enc_frame.input.plane[1] = in+(_w*_h);
	xvid_enc_frame.input.plane[2] = in+((_w*_h*5)>>2);
	return 1;
}	

uint8_t     xvid4Encoder::postAmble(uint32_t *flags )
{
	xvid_res.is_key_frame=*flags=0;
	
	if(xvid_enc_frame.out_flags & XVID_KEYFRAME) 
	{
		*flags=AVI_KEY_FRAME;
		xvid_res.is_key_frame=1;
	}
	else
	if(xvid_enc_stats.type == XVID_TYPE_BVOP)
	{
		*flags=AVI_B_FRAME;
	
	}
	xvid_res.out_quantizer=xvid_enc_stats.quant;
	return 1;
}

// *************************************************
// *************************************************
//									CQ
// *************************************************
// *************************************************
uint8_t xvid4EncoderCQ::init(uint32_t q,uint32_t fps1000, xvid4EncParam *param)
{
    int xerr;
    	printf("Using Xvid 4 codec with CQ mode = %lu (%dx%d)\n",q,_w,_h);
    	_q=q;
	_fps1000=fps1000;	
	memcpy(&_param,param,sizeof(_param));
	dump();
	// Here we go...
	MMSET(xvid_enc_create);
	xvid_enc_create.version = XVID_VERSION;
	xvid_enc_create.width = _w;
	xvid_enc_create.height = _h;
	MMSET(single);
	

	plugins[0].func = xvid_plugin_single;
	plugins[0].param = &single;
	
	single.version = XVID_VERSION;
	single.bitrate = 1500;

	xvid_enc_create.plugins = plugins;
	xvid_enc_create.num_plugins = 1;
	
	//Framerate
	xvid_enc_create.fincr = 1000;
	xvid_enc_create.fbase = _fps1000;
	//
	createUpdate();
	xerr = xvid_encore(NULL, XVID_ENC_CREATE, &xvid_enc_create, NULL);
	if(xerr<0)
	{
		printf("Xvid-4 init error :%d\n",xerr);
		return 0;
	
	}
	
	_handle = xvid_enc_create.handle;

	printf("Xvid-4 CQ init Ok\n");

 	return 1;
}


uint8_t xvid4EncoderCQ::encode(
			ADMImage * in,
			uint8_t * out,
			uint32_t * len,
       			uint32_t * flags)
{
int ret;

	
	
	preAmble(in->data);
	
	xvid_enc_frame.quant = _q;
	xvid_enc_frame.bitstream = out;	
	
	/* Encode the frame */
	ret = xvid_encore(_handle, XVID_ENC_ENCODE, &xvid_enc_frame,
					  &xvid_enc_stats);
	if(ret<0)
		{
			printf("Error calling xvid4 %d\n",ret);
			return 0;
			
		}
	postAmble(flags);
	
	*len=xvid_res.total_bits=ret;
	xvid_res.total_bits*=8;
	
	return 1;

}
// *************************************************
// *************************************************
//		CQ but Q is controlled externally
// *************************************************
// *************************************************

uint8_t xvid4EncoderVBRExternal::init(uint32_t q,uint32_t fps1000, xvid4EncParam *param)
{
    int xerr;
    	printf("Using Xvid 4 codec with CQ mode = %lu (%dx%d)\n",q,_w,_h);
    	_q=q;
	_fps1000=fps1000;	
	memcpy(&_param,param,sizeof(_param));
	dump();
	// Here we go...
	MMSET(xvid_enc_create);
	xvid_enc_create.version = XVID_VERSION;
	xvid_enc_create.width = _w;
	xvid_enc_create.height = _h;
	MMSET(single);
	

	plugins[0].func = xvid_plugin_single;
	plugins[0].param = &single;
	
	single.version = XVID_VERSION;
	single.bitrate = 1500;

	xvid_enc_create.plugins = plugins;
	xvid_enc_create.num_plugins = 1;
	
	//Framerate
	xvid_enc_create.fincr = 1000;
	xvid_enc_create.fbase = _fps1000;
	//
	createUpdate();
	xerr = xvid_encore(NULL, XVID_ENC_CREATE, &xvid_enc_create, NULL);
	if(xerr<0)
	{
		printf("Xvid-4 init error :%d\n",xerr);
		return 0;
	
	}
	
	_handle = xvid_enc_create.handle;

	printf("Xvid-4 CQ init Ok\n");

 	return 1;
}


uint8_t xvid4EncoderVBRExternal::encode(
			ADMImage * in,
			uint8_t * out,
			uint32_t * len,
       			uint32_t * flags)
{
int q;
	q=(*flags)>>16;
	*flags&=0xFFFF;
	return encodeVBR(in,out,len,q,flags);
}

uint8_t xvid4EncoderVBRExternal::encodeVBR(
			ADMImage * in,
			uint8_t * out,
			uint32_t * len,
			uint32_t q,
       			uint32_t * flags)
{	
int  ret;
	preAmble(in->data);
	
	xvid_enc_frame.quant = q;
	xvid_enc_frame.bitstream = out;	
	if(*flags & AVI_KEY_FRAME)
	{
		xvid_enc_frame.type=XVID_TYPE_IVOP;
	}
	else if(*flags & AVI_B_FRAME)
	{
		xvid_enc_frame.type=XVID_TYPE_BVOP;
	}
	else
	{	//P
			xvid_enc_frame.type=XVID_TYPE_PVOP;
	}
	
	/* Encode the frame */
	ret = xvid_encore(_handle, XVID_ENC_ENCODE, &xvid_enc_frame,
					  &xvid_enc_stats);
	if(ret<0)
		{
			printf("Error calling xvid4 %d\n",ret);
			return 0;
			
		}
	postAmble(flags);
	
	*len=xvid_res.total_bits=ret;
	xvid_res.total_bits*=8;
	
	return 1;

}
// *************************************************
// *************************************************
//		CBR
// *************************************************
// *************************************************
uint8_t xvid4EncoderCBR::init(uint32_t br,uint32_t fps1000, xvid4EncParam *param)
{
    int xerr;
    	printf("Using Xvid 4 codec with CBR mode = %lu\n",br);
    	_bitrate=br;
	_fps1000=fps1000;
	memcpy(&_param,param,sizeof(_param));
	dump();
	// Here we go...
	MMSET(single);
	MMSET(xvid_enc_create);
	
	xvid_enc_create.version = XVID_VERSION;
	xvid_enc_create.width = _w;
	xvid_enc_create.height = _h;
	

	plugins[0].func = xvid_plugin_single;
	plugins[0].param = &single;
	
	single.version = XVID_VERSION;
	single.bitrate = _bitrate;

	xvid_enc_create.plugins = plugins;
	xvid_enc_create.num_plugins = 1;
	
	//Framerate
	xvid_enc_create.fincr = 1000;
	xvid_enc_create.fbase = _fps1000;
	//
	createUpdate();
	xerr = xvid_encore(NULL, XVID_ENC_CREATE, &xvid_enc_create, NULL);
	if(xerr<0)
	{
		printf("Xvid-4 init error :%d\n",xerr);
		return 0;
	
	}
	
	_handle = xvid_enc_create.handle;

	printf("Xvid-4 CBR init Ok\n");

 	return 1;
}


uint8_t xvid4EncoderCBR::encode(
			ADMImage * in,
			uint8_t * out,
			uint32_t * len,
       			uint32_t * flags)
{
int ret;

	
	
	preAmble(in->data);
	
	xvid_enc_frame.quant = 0;
	xvid_enc_frame.bitstream = out;
	
	//--
	
	
	//--

	
	/* Encode the frame */
	ret = xvid_encore(_handle, XVID_ENC_ENCODE, &xvid_enc_frame,
					  &xvid_enc_stats);
	if(ret<0)
		{
			printf("Error calling xvid4 %d\n",ret);
			return 0;
			
		}
	postAmble(flags);
	
	*len=xvid_res.total_bits=ret;
	xvid_res.total_bits*=8;
	
	return 1;

}
// *************************************************
// *************************************************
//		Pass1
// *************************************************
// *************************************************
uint8_t xvid4EncoderPass1::init(uint32_t br,uint32_t fps1000, xvid4EncParam *param)
{
    int xerr;
    	printf("Using Xvid 4 codec pass 1 of 2 (log: %s)\n",(char *)param->logName);
    	
	_fps1000=fps1000;
	memcpy(&_param,param,sizeof(_param));
	dump();
	// Here we go...
	MMSET(xvid_enc_create);
	MMSET(pass1);
	
	xvid_enc_create.version = XVID_VERSION;
	xvid_enc_create.width = _w;
	xvid_enc_create.height = _h;
	
	

	plugins[0].func = xvid_plugin_2pass1;
	plugins[0].param = &pass1;
	
	pass1.version = XVID_VERSION;
	pass1.filename = (char *)_param.logName;

	xvid_enc_create.plugins = plugins;
	xvid_enc_create.num_plugins = 1;
	
	
	//Framerate
	xvid_enc_create.fincr = 1000;
	xvid_enc_create.fbase = _fps1000;
	//
	createUpdate();
	// Force Qz=2 !
	xerr = xvid_encore(NULL, XVID_ENC_CREATE, &xvid_enc_create, NULL);
	if(xerr<0)
	{
		printf("Xvid-4 init error :%d\n",xerr);
		return 0;
	
	}
	
	_handle = xvid_enc_create.handle;

	printf("Xvid-4 Pass1 init Ok\n");

 	return 1;
}


uint8_t xvid4EncoderPass1::encode(
			ADMImage * in,
			uint8_t * out,
			uint32_t * len,
       			uint32_t * flags)
{
int ret;

	
	
	preAmble(in->data);
	
	//xvid_enc_frame.quant = 2;
	xvid_enc_frame.bitstream = out;
	xvid_enc_frame.input.plane[0] = YPLANE(in);
	

	
	/* Encode the frame */
	ret = xvid_encore(_handle, XVID_ENC_ENCODE, &xvid_enc_frame,
					  &xvid_enc_stats);
	if(ret<0)
		{
			printf("Error calling xvid4 %d\n",ret);
			return 0;
			
		}
	postAmble(flags);
	
	*len=xvid_res.total_bits=ret;
	xvid_res.total_bits*=8;
	
	return 1;

}
// *************************************************
// *************************************************
//		Pass2
// *************************************************
// *************************************************
uint8_t xvid4EncoderPass2::init(uint32_t br,uint32_t fps1000, xvid4EncParam *param)
{
    int xerr;
    	printf("Using Xvid 4 codec with 2pass  mode pass 2 of 2, average bitrate = %lu\n",br);
    	_bitrate=br;
	_fps1000=fps1000;
	memcpy(&_param,param,sizeof(_param));
	dump();
	// Here we go...
	MMSET(xvid_enc_create);
	MMSET(pass2);
	
	xvid_enc_create.version = XVID_VERSION;
	xvid_enc_create.width = _w;
	xvid_enc_create.height = _h;
	
	

	plugins[0].func = xvid_plugin_2pass2;
	plugins[0].param = &pass2;
	
	pass2.version = XVID_VERSION;
	pass2.filename = (char *)_param.logName;

	pass2.bitrate=br;	// Average bitrate
	#define CPY(x) pass2.x=_param.x;printf(#x"=%d\n",pass2.x);
	
	CPY( keyframe_boost);
	CPY( curve_compression_high);
	CPY( curve_compression_low);
	CPY( overflow_control_strength);
	CPY( max_overflow_improvement);
	CPY( max_overflow_degradation);
	CPY( kfreduction);
	CPY( kfthreshold);
	
	CPY( container_frame_overhead);
	
	
	
	xvid_enc_create.plugins = plugins;
	xvid_enc_create.num_plugins = 1;
	
	//Framerate
	xvid_enc_create.fincr = 1000;
	xvid_enc_create.fbase = _fps1000;
	//
	createUpdate();
	xerr = xvid_encore(NULL, XVID_ENC_CREATE, &xvid_enc_create, NULL);
	if(xerr<0)
	{
		printf("Xvid-4 init error :%d\n",xerr);
		return 0;
	
	}
	
	_handle = xvid_enc_create.handle;

	printf("Xvid-4 init Ok\n");

 	return 1;
}


uint8_t xvid4EncoderPass2::encode(
			ADMImage * in,
			uint8_t * out,
			uint32_t * len,
       			uint32_t * flags)
{
int ret;

	
	
	preAmble(in->data);
		
	xvid_enc_frame.bitstream = out;
	xvid_enc_frame.input.plane[0] = YPLANE(in);
	

	
	/* Encode the frame */
	ret = xvid_encore(_handle, XVID_ENC_ENCODE, &xvid_enc_frame,
					  &xvid_enc_stats);
	if(ret<0)
		{
			printf("Error calling xvid4 %d\n",ret);
			return 0;
			
		}
	postAmble(flags);
	
	*len=xvid_res.total_bits=ret;
	xvid_res.total_bits*=8;
	
	return 1;

}
#endif
