[cvs] / xvidcore / src / divx4.c Repository:
ViewVC logotype

View of /xvidcore/src/divx4.c

Parent Directory Parent Directory | Revision Log Revision Log


Revision 1.11 - (download) (annotate)
Sat May 4 12:26:06 2002 UTC (21 years, 10 months ago) by edgomez
Branch: MAIN
Changes since 1.10: +124 -15 lines
Added "per encoder" quality setting
/**************************************************************************
 *
 *	XVID MPEG-4 VIDEO CODEC
 *	OpenDivx API wrapper
 *
 *	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., 675 Mass Ave, Cambridge, MA 02139, USA.
 *
 *************************************************************************/

/**************************************************************************
 *
 *	History:
 *
 *	26.02.2001  fixed dec_csp bugs
 *	26.12.2001  xvid_init() support
 *	22.12.2001  removed some compiler warnings
 *	16.12.2001  inital version; (c)2001 peter ross <pross@cs.rmit.edu.au>
 *
 * $Id: divx4.c,v 1.11 2002/05/04 12:26:06 edgomez Exp $
 *
 *************************************************************************/

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

#include "xvid.h"
#include "divx4.h"
#include "decoder.h"
#include "encoder.h"

#define EMULATED_DIVX_VERSION 20011001

/**************************************************************************
 * Divx Instance Structure
 *
 * This chain list datatype allows XviD do instanciate multiples divx4
 * sessions.
 *
 * ToDo : The way this chain list is used does not guarantee reentrance
 *        because they are not protected by any kind of mutex to allow
 *        only one modifier. We should add a mutex for each element in
 *        the chainlist.
 *************************************************************************/


typedef struct DINST
{
	unsigned long key;
	struct DINST * next;

	void * handle;
	XVID_DEC_FRAME xframe;

} DINST;

typedef struct EINST
{
	struct EINST * next;

	void * handle;
	int quality;

} EINST;

/**************************************************************************
 * Global data (needed to emulate correctly exported symbols from divx4)
 *************************************************************************/

/* This is not used in this module but is required by some divx4 encoders*/
int quiet_encore = 1;

/**************************************************************************
 * Local data
 *************************************************************************/

/* The Divx4 instance chainlist */
static DINST * dhead = NULL;
static EINST * ehead = NULL;

/* Divx4 quality to XviD encoder motion flag presets */
static int const divx4_motion_presets[7] = {
	0, 

	PMV_QUICKSTOP16, 

	PMV_EARLYSTOP16, 

	PMV_EARLYSTOP16 | PMV_HALFPELREFINE16,

	PMV_EARLYSTOP16 | PMV_HALFPELREFINE16 |
	PMV_EARLYSTOP8  | PMV_HALFPELDIAMOND8,

	PMV_EARLYSTOP16 | PMV_HALFPELREFINE16 |
	PMV_EARLYSTOP8  | PMV_HALFPELDIAMOND8,

	PMV_EARLYSTOP16    | PMV_HALFPELREFINE16 |
	PMV_EXTSEARCH16    | PMV_EARLYSTOP8 |
	PMV_HALFPELREFINE8 | PMV_HALFPELDIAMOND8
};


/* Divx4 quality to general encoder flag presets */
static int const divx4_general_presets[7] = {
	0, 
	XVID_H263QUANT, 
	XVID_H263QUANT,
	XVID_H263QUANT | XVID_HALFPEL,
	XVID_H263QUANT | XVID_INTER4V | XVID_HALFPEL,
   	XVID_H263QUANT | XVID_INTER4V | XVID_HALFPEL, 
	XVID_H263QUANT | XVID_INTER4V | XVID_HALFPEL
};

/**************************************************************************
 * Local Prototypes
 *************************************************************************/

/* Chain list helper functions */
static DINST * dinst_find(unsigned long key);
static DINST * dinst_add(unsigned long key);
static void    dinst_remove(unsigned long key);

static EINST * einst_find(void *handle);
static EINST * einst_add(void *handle);
static void    einst_remove(void *handle);

/* Converts divx4 colorspaces codes to xvid codes */
static int xvid_to_opendivx_dec_csp(int csp);
static int xvid_to_opendivx_enc_csp(int csp);

/**************************************************************************
 * decore part
 *
 * decore is the divx4 entry point used to decompress the mpeg4 bitstream
 * into a user defined image format.
 *************************************************************************/

int
decore(unsigned long key, unsigned long opt, void * param1, void * param2)
{

	int xerr;

	switch (opt) {

	case DEC_OPT_MEMORY_REQS :
	{
		memset(param2, 0, sizeof(DEC_MEM_REQS));
		return DEC_OK;
	}
	
	case DEC_OPT_INIT :
	{
		XVID_INIT_PARAM xinit;
		XVID_DEC_PARAM xparam;
		DINST * dcur;
		DEC_PARAM * dparam = (DEC_PARAM *)param1;

		/* Find the divx4 instance */
		if ((dcur = dinst_find(key)) == NULL)
		{
			dcur = dinst_add(key);
		}

		/*
		 * XviD initialization
		 * XviD will detect the host cpu type and activate optimized
		 * functions according to the host cpu features.
		 */
		xinit.cpu_flags = 0;
		xvid_init(NULL, 0, &xinit, NULL);

		/* XviD decoder initialization for this instance */
		xparam.width = dparam->x_dim;
		xparam.height = dparam->y_dim;
		dcur->xframe.colorspace =
			xvid_to_opendivx_dec_csp(dparam->output_format);
				
		xerr = decoder_create(&xparam);

		/* Store the xvid handle into the divx4 instance chainlist */
		dcur->handle = xparam.handle;
			
		break;
	}

	case DEC_OPT_RELEASE :
	{
		DINST * dcur;

		/* Find the divx4 instance into the chain list */
		if ((dcur = dinst_find(key)) == NULL)
		{
			return DEC_EXIT;
		}
			
		/* Destroy the XviD decoder attached to this divx4 instance */
		xerr = decoder_destroy(dcur->handle);

		/* Remove the divx4 instance from the chainlist */
		dinst_remove(key);

		break;
	}
	
	case DEC_OPT_SETPP :
	{
		DINST * dcur;

		/* Find the divx4 instance into the chain list */
		if ((dcur = dinst_find(key)) == NULL)
		{
			return DEC_EXIT;
		}

		/*
		 * We return DEC_OK but XviD has no postprocessing implemented
		 * in core.
		 */
		return DEC_OK;
	}

	case DEC_OPT_SETOUT :
	{
		DINST * dcur;
		DEC_PARAM * dparam = (DEC_PARAM *)param1;

		if ((dcur = dinst_find(key)) == NULL)
		{
			return DEC_EXIT;
		}

		/* Change the output colorspace */
		dcur->xframe.colorspace =
			xvid_to_opendivx_dec_csp(dparam->output_format);

		return DEC_OK;
	}
	
	case DEC_OPT_FRAME:
	{
		int csp_tmp = 0;
		DINST * dcur;
		DEC_FRAME * dframe = (DEC_FRAME *)param1;

		if ((dcur = dinst_find(key)) == NULL)
		{
			return DEC_EXIT;
		}

		/* Copy the divx4 fields to the XviD decoder structure */
		dcur->xframe.bitstream = dframe->bitstream;
		dcur->xframe.length = dframe->length;
		dcur->xframe.image = dframe->bmp;
		dcur->xframe.stride = dframe->stride;
			
		/* Does the frame need to be skipped ? */
		if (!dframe->render_flag)
		{
			/*
			 * Then we use the null colorspace to force XviD to
			 * skip the frame. The original colorspace will be
			 * restored after the decoder call
			 */
			csp_tmp = dcur->xframe.colorspace;
			dcur->xframe.colorspace = XVID_CSP_NULL;
		}
			
		/* Decode the bitstream */
		xerr = decoder_decode(dcur->handle, &dcur->xframe);

		/* Restore the real colorspace for this instance */
		if (!dframe->render_flag)
		{
			dcur->xframe.colorspace = csp_tmp;
		}

		break;
	}

	case DEC_OPT_FRAME_311 :
		/* XviD does not handle Divx ;-) 3.11 yet */
		return DEC_EXIT;

	case DEC_OPT_VERSION:
		return EMULATED_DIVX_VERSION;

	default :
		return DEC_EXIT;
	}


	/* XviD error code -> Divx4 */
	switch(xerr)
	{
	case XVID_ERR_OK :
		return DEC_OK;
	case XVID_ERR_MEMORY :
		return DEC_MEMORY;
	case XVID_ERR_FORMAT :
		return DEC_BAD_FORMAT;
	default : 
		return DEC_EXIT;
	}
}

/**************************************************************************
 * Encore Part
 *
 * encore is the divx4 entry point used to compress a frame to a mpeg4
 * bitstream.
 *************************************************************************/

#define FRAMERATE_INCR		1001

int
encore(void * handle, int opt, void * param1, void * param2)
{

	int xerr;

	switch(opt) {
	case ENC_OPT_INIT :
	{
		EINST *ecur;
		ENC_PARAM * eparam = (ENC_PARAM *)param1;
		XVID_INIT_PARAM xinit;
		XVID_ENC_PARAM xparam;

		/* Init XviD which will detect host cpu features */
		xinit.cpu_flags = 0;
		xvid_init(NULL, 0, &xinit, NULL);
						
		/* Settings are copied to the XviD encoder structure */
		xparam.width = eparam->x_dim;
		xparam.height = eparam->y_dim;
		if ((eparam->framerate - (int)eparam->framerate) == 0)
		{
			xparam.fincr = 1; 
			xparam.fbase = (int)eparam->framerate;
		}
		else
		{
			xparam.fincr = FRAMERATE_INCR;
			xparam.fbase = (int)(FRAMERATE_INCR * eparam->framerate);
		}
		xparam.rc_bitrate = eparam->bitrate;
		xparam.rc_reaction_delay_factor = 16;
		xparam.rc_averaging_period = 100;
		xparam.rc_buffer = 100;
		xparam.min_quantizer = eparam->min_quantizer;
		xparam.max_quantizer = eparam->max_quantizer;
		xparam.max_key_interval = eparam->max_key_interval;

		/* Create the encoder session */
		xerr = encoder_create(&xparam);

		eparam->handle = xparam.handle;

		/* Create an encoder instance in the chainlist */
		if ((ecur = einst_find(xparam.handle)) == NULL)
		{
			ecur = einst_add(xparam.handle);

			if(ecur == NULL) {
				encoder_destroy((Encoder*)xparam.handle);
				return ENC_MEMORY;
			}

		}

		ecur->quality = eparam->quality;
		if(ecur->quality < 0) ecur->quality = 0;
		if(ecur->quality > 6) ecur->quality = 6;

		break;
	}

	case ENC_OPT_RELEASE :
	{
		EINST *ecur;

		if ((ecur = einst_find(handle)) == NULL)
		{
			return ENC_FAIL;
		}

		xerr = encoder_destroy((Encoder *) handle);
		break;
	}

	case ENC_OPT_ENCODE :
	case ENC_OPT_ENCODE_VBR :
	{
		EINST *ecur;

		ENC_FRAME * eframe = (ENC_FRAME *)param1;
		ENC_RESULT * eresult = (ENC_RESULT *)param2;
		XVID_ENC_FRAME xframe;
		XVID_ENC_STATS xstats;

		if ((ecur = einst_find(handle)) == NULL)
		{
			return ENC_FAIL;
		}

		/* Copy the divx4 info into the xvid structure */
		xframe.bitstream = eframe->bitstream;
		xframe.length = eframe->length;
		xframe.motion = divx4_motion_presets[ecur->quality];
		xframe.general = divx4_general_presets[ecur->quality];

		xframe.image = eframe->image;
		xframe.colorspace =
			xvid_to_opendivx_enc_csp(eframe->colorspace);

		if (opt == ENC_OPT_ENCODE_VBR)	
		{
			xframe.intra = eframe->intra;
			xframe.quant = eframe->quant;
		}
		else
		{
			xframe.intra = -1;
			xframe.quant = 0;
		}
			
		/* Encode the frame */
		xerr = encoder_encode((Encoder *) handle, &xframe, (eresult ? &xstats : NULL) );

		/* Copy back the xvid structure to the divx4 one */
		if (eresult)
		{
			eresult->is_key_frame = xframe.intra;
			eresult->quantizer = xstats.quant;
			eresult->total_bits = xframe.length * 8;
			eresult->motion_bits = xstats.hlength * 8;
			eresult->texture_bits = eresult->total_bits - eresult->motion_bits;
		}

		eframe->length = xframe.length;

		break;
	}

	default:
		return ENC_FAIL;
	}

	/* XviD Error code  -> Divx4 error code */
	switch(xerr)
	{
	case XVID_ERR_OK :
		return ENC_OK;
	case XVID_ERR_MEMORY :
		return ENC_MEMORY;
	case XVID_ERR_FORMAT :
		return ENC_BAD_FORMAT;
	default : 
		return ENC_FAIL;
	}
}

/**************************************************************************
 * Local Functions
 *************************************************************************/

/***************************************
 * DINST chainlist helper functions    *
 ***************************************/

/* Find an element in the chainlist according to its key value */
static DINST * dinst_find(unsigned long key)
{
	DINST * dcur = dhead;

	while (dcur)
	{
		if (dcur->key == key)
		{
			return dcur;
		}
		dcur = dcur->next;
	}

	return NULL;
}


/* Add an element to the chainlist */
static DINST * dinst_add(unsigned long key)
{
	DINST * dnext = dhead;

	dhead = malloc(sizeof(DINST));
	if (dhead == NULL)
	{
		dhead = dnext;
		return NULL;
	}

	dhead->key = key;
	dhead->next = dnext;

	return dhead;
}


/* Remove an elmement from the chainlist */
static void dinst_remove(unsigned long key)
{
	DINST * dcur = dhead;

	if (dhead == NULL)
	{
		return;
	}

	if (dcur->key == key)
	{
		dhead = dcur->next;
		free(dcur);
		return;
	}

	while (dcur->next)
	{
		if (dcur->next->key == key)
		{
			DINST * tmp = dcur->next;
			dcur->next = tmp->next;
			free(tmp);
			return;
		}
		dcur = dcur->next;
	}
}


/***************************************
 * EINST chainlist helper functions    *
 ***************************************/

/* Find an element in the chainlist according to its handle */
static EINST * einst_find(void *handle)
{
	EINST * ecur = ehead;

	while (ecur)
	{
		if (ecur->handle == handle)
		{
			return ecur;
		}
		ecur = ecur->next;
	}

	return NULL;
}


/* Add an element to the chainlist */
static EINST * einst_add(void *handle)
{
	EINST * enext = ehead;

	ehead = malloc(sizeof(EINST));
	if (ehead == NULL)
	{
		ehead = enext;
		return NULL;
	}

	ehead->handle = handle;
	ehead->next = enext;

	return ehead;
}


/* Remove an elmement from the chainlist */
static void einst_remove(void *handle)
{
	EINST * ecur = ehead;

	if (ehead == NULL)
	{
		return;
	}

	if (ecur->handle == handle)
	{
		ehead = ecur->next;
		free(ecur);
		return;
	}

	while (ecur->next)
	{
		if (ecur->next->handle == handle)
		{
			EINST * tmp = ecur->next;
			ecur->next = tmp->next;
			free(tmp);
			return;
		}
		ecur = ecur->next;
	}
}
/***************************************
 * Colorspace code converter           *
 ***************************************/

static int xvid_to_opendivx_dec_csp(int csp)
{

	switch(csp)
	{
	case DEC_YV12 :
		return XVID_CSP_YV12;
	case DEC_420 :
		return XVID_CSP_I420;
	case DEC_YUY2 :
		return XVID_CSP_YUY2;
	case DEC_UYVY :
		return XVID_CSP_UYVY;
	case DEC_RGB32 :
		return XVID_CSP_VFLIP | XVID_CSP_RGB32;
	case DEC_RGB24 :
		return XVID_CSP_VFLIP | XVID_CSP_RGB24;
	case DEC_RGB565 :
		return XVID_CSP_VFLIP | XVID_CSP_RGB565;
	case DEC_RGB555 :
		return XVID_CSP_VFLIP | XVID_CSP_RGB555;
	case DEC_RGB32_INV :
		return XVID_CSP_RGB32;
	case DEC_RGB24_INV :
		return XVID_CSP_RGB24;
	case DEC_RGB565_INV :
		return XVID_CSP_RGB565;
	case DEC_RGB555_INV :
		return XVID_CSP_RGB555;
	case DEC_USER :
		return XVID_CSP_USER;    
	default :
		return -1;
	}
}

static int xvid_to_opendivx_enc_csp(int csp)
{

	switch (csp)
	{
	case ENC_CSP_RGB24 : 
		return XVID_CSP_VFLIP | XVID_CSP_RGB24;
	case ENC_CSP_YV12 :
		return XVID_CSP_YV12;
	case ENC_CSP_YUY2 :
		return XVID_CSP_YUY2;
	case ENC_CSP_UYVY :
		return XVID_CSP_UYVY;
	case ENC_CSP_I420 :
		return XVID_CSP_I420;
	default :
		return -1;
	}
}

No admin address has been configured
ViewVC Help
Powered by ViewVC 1.0.4