encoders

git-svn-id: svn://svn.mplayerhq.hu/mplayer/trunk@5551 b3059339-0415-0410-9bf9-f77b7e298cf2

1 files changed, 194 insertions, 0 deletions

diff --git a/libmpcodecs/ve_divx4.c b/libmpcodecs/ve_divx4.c
new file mode 100644
index 0000000000..c272efbe1f
--- /dev/null
+++ b/libmpcodecs/ve_divx4.c
@@ -0,0 +1,194 @@
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+
+#include "../config.h"
+#include "../mp_msg.h"
+
+#ifdef HAVE_DIVX4ENCORE
+
+#include "codec-cfg.h"
+#include "stream.h"
+#include "demuxer.h"
+#include "stheader.h"
+
+#include "aviwrite.h"
+
+#include "../libvo/img_format.h"
+#include "../mp_image.h"
+#include "vf.h"
+
+//===========================================================================//
+
+#include "divx4_vbr.h"
+
+extern int pass;
+extern char* passtmpfile;
+
+#include <encore2.h>
+
+ENC_PARAM divx4_param;
+int divx4_crispness;
+
+#include "cfgparser.h"
+
+struct config divx4opts_conf[]={
+	{"br", &divx4_param.bitrate, CONF_TYPE_INT, CONF_RANGE, 4, 24000000, NULL},
+	{"rc_period", &divx4_param.rc_period, CONF_TYPE_INT, 0,0,0, NULL},
+	{"rc_reaction_period", &divx4_param.rc_reaction_period, CONF_TYPE_INT, 0,0,0, NULL},
+	{"rc_reaction_ratio", &divx4_param.rc_reaction_ratio, CONF_TYPE_INT, 0,0,0, NULL},
+	{"min_quant", &divx4_param.min_quantizer, CONF_TYPE_INT, CONF_RANGE,0,32, NULL},
+	{"max_quant", &divx4_param.max_quantizer, CONF_TYPE_INT, CONF_RANGE,0,32, NULL},
+	{"key", &divx4_param.max_key_interval, CONF_TYPE_INT, CONF_MIN,0,0, NULL},
+	{"deinterlace", &divx4_param.deinterlace, CONF_TYPE_FLAG, 0,0,1, NULL},
+	{"q", &divx4_param.quality, CONF_TYPE_INT, CONF_RANGE, 1, 5, NULL},
+	{"crispness", &divx4_crispness, CONF_TYPE_INT, CONF_RANGE,0,100, NULL},
+	{"help", "TODO: divx4opts help!\n", CONF_TYPE_PRINT, CONF_NOCFG, 0, 0, NULL},
+	{NULL, NULL, 0, 0, 0, 0, NULL}
+};
+
+struct vf_priv_s {
+    aviwrite_stream_t* mux;
+    ENC_RESULT enc_result;
+    ENC_FRAME enc_frame;
+    void* enc_handle;
+};
+
+#define mux_v (vf->priv->mux)
+
+static int config(struct vf_instance_s* vf,
+        int width, int height, int d_width, int d_height,
+	unsigned int flags, unsigned int outfmt){
+
+    mux_v->bih->biWidth=width;
+    mux_v->bih->biHeight=height;
+
+    divx4_param.x_dim=width;
+    divx4_param.y_dim=height;
+    divx4_param.framerate=(float)mux_v->h.dwRate/mux_v->h.dwScale;
+    mux_v->bih->biSizeImage=mux_v->bih->biWidth*mux_v->bih->biHeight*3;
+
+    if(!divx4_param.bitrate) divx4_param.bitrate=800000;
+    else if(divx4_param.bitrate<=16000) divx4_param.bitrate*=1000;
+    if(!divx4_param.quality) divx4_param.quality=5; // the quality of compression ( 1 - fastest, 5 - best )
+
+    divx4_param.handle=NULL;
+    encore(NULL,ENC_OPT_INIT,&divx4_param,NULL);
+    vf->priv->enc_handle=divx4_param.handle;
+    switch(outfmt){
+    case IMGFMT_YV12:	vf->priv->enc_frame.colorspace=ENC_CSP_YV12; break;
+    case IMGFMT_IYUV:
+    case IMGFMT_I420:	vf->priv->enc_frame.colorspace=ENC_CSP_I420; break;
+    case IMGFMT_YUY2:	vf->priv->enc_frame.colorspace=ENC_CSP_YUY2; break;
+    case IMGFMT_UYVY:	vf->priv->enc_frame.colorspace=ENC_CSP_UYVY; break;
+    case IMGFMT_RGB24:
+    case IMGFMT_BGR24:
+    	vf->priv->enc_frame.colorspace=ENC_CSP_RGB24; break;
+    default:
+	mp_msg(MSGT_MENCODER,MSGL_ERR,"divx4: unsupported picture format (%s)!\n",
+	    vo_format_name(outfmt));
+	return 0;
+    }
+
+    switch(pass){
+    case 1:
+	if (VbrControl_init_2pass_vbr_analysis(passtmpfile, divx4_param.quality) == -1){
+	    mp_msg(MSGT_MENCODER,MSGL_ERR,"2pass failed: filename=%s\n", passtmpfile);
+	    pass=0;
+	}
+	break;
+    case 2:
+        if (VbrControl_init_2pass_vbr_encoding(passtmpfile,
+					 divx4_param.bitrate,
+					 divx4_param.framerate,
+					 divx4_crispness,
+					 divx4_param.quality) == -1){
+	    mp_msg(MSGT_MENCODER,MSGL_ERR,"2pass failed: filename=%s\n", passtmpfile);
+	    pass=0;
+	}
+	break;
+    }
+
+    return 1;
+}
+
+static int control(struct vf_instance_s* vf, int request, void* data){
+
+    return CONTROL_UNKNOWN;
+}
+
+static int query_format(struct vf_instance_s* vf, unsigned int fmt){
+    switch(fmt){
+    case IMGFMT_YV12:
+    case IMGFMT_IYUV:
+    case IMGFMT_I420:
+	return 3; // no conversion
+    case IMGFMT_YUY2:
+    case IMGFMT_UYVY:
+    case IMGFMT_RGB24:
+    case IMGFMT_BGR24:
+	return 1; // conversion
+    }
+    return 0;
+}
+
+static void put_image(struct vf_instance_s* vf, mp_image_t *mpi){
+    ENC_RESULT enc_result;
+    vf->priv->enc_frame.image=mpi->planes[0];
+    vf->priv->enc_frame.bitstream=mux_v->buffer;
+    vf->priv->enc_frame.length=mux_v->buffer_size;
+    vf->priv->enc_frame.mvs=NULL;
+    if(pass==2){	// handle 2-pass:
+    	vf->priv->enc_frame.quant = VbrControl_get_quant();
+	vf->priv->enc_frame.intra = VbrControl_get_intra();
+	encore(vf->priv->enc_handle,ENC_OPT_ENCODE_VBR,&vf->priv->enc_frame,&enc_result);
+        VbrControl_update_2pass_vbr_encoding(enc_result.motion_bits,
+					    enc_result.texture_bits,
+					    enc_result.total_bits);
+    } else {
+	vf->priv->enc_frame.quant=0;
+	vf->priv->enc_frame.intra=0;
+	encore(vf->priv->enc_handle,ENC_OPT_ENCODE,&vf->priv->enc_frame,&enc_result);
+	if(pass==1){
+	  VbrControl_update_2pass_vbr_analysis(enc_result.is_key_frame, 
+					       enc_result.motion_bits, 
+					       enc_result.texture_bits, 
+					       enc_result.total_bits, 
+					       enc_result.quantizer);
+	}
+    }
+    mencoder_write_frame(mux_v,vf->priv->enc_frame.length,enc_result.is_key_frame?0x10:0);
+}
+
+//===========================================================================//
+
+static int vf_open(vf_instance_t *vf, char* args){
+    vf->config=config;
+    vf->control=control;
+    vf->query_format=query_format;
+    vf->put_image=put_image;
+    vf->priv=malloc(sizeof(struct vf_priv_s));
+    memset(vf->priv,0,sizeof(struct vf_priv_s));
+    vf->priv->mux=args;
+
+    mux_v->bih=malloc(sizeof(BITMAPINFOHEADER));
+    mux_v->bih->biSize=sizeof(BITMAPINFOHEADER);
+    mux_v->bih->biWidth=0;
+    mux_v->bih->biHeight=0;
+    mux_v->bih->biPlanes=1;
+    mux_v->bih->biBitCount=24;
+    mux_v->bih->biCompression=mmioFOURCC('d','i','v','x');
+
+    return 1;
+}
+
+vf_info_t ve_info_divx4 = {
+    "divx4 encoder",
+    "divx4",
+    "A'rpi",
+    "for internal use by mencoder",
+    vf_open
+};
+
+//===========================================================================//
+#endif