diff options
| -rw-r--r-- | libmpcodecs/Makefile | 2 | ||||
| -rw-r--r-- | libmpcodecs/ve.c | 4 | ||||
| -rw-r--r-- | libmpcodecs/ve_xvid.c | 374 |
3 files changed, 379 insertions, 1 deletions
diff --git a/libmpcodecs/Makefile b/libmpcodecs/Makefile index 2390605a7b..30b962d517 100644 --- a/libmpcodecs/Makefile +++ b/libmpcodecs/Makefile @@ -7,7 +7,7 @@ LIBNAME2 = libmpencoders.a AUDIO_SRCS=dec_audio.c ad.c ad_liba52.c ad_acm.c ad_alaw.c ad_dk3adpcm.c ad_dshow.c ad_dvdpcm.c ad_ffmpeg.c ad_hwac3.c ad_imaadpcm.c ad_mp3lib.c ad_msadpcm.c ad_pcm.c ad_roqaudio.c ad_msgsm.c ad_faad.c ad_libvorbis.c ad_libmad.c ad_realaud.c ad_libdv.c VIDEO_SRCS=dec_video.c vd.c vd_null.c vd_realvid.c vd_cinepak.c vd_qtrpza.c vd_ffmpeg.c vd_dshow.c vd_vfw.c vd_vfwex.c vd_odivx.c vd_divx4.c vd_raw.c vd_xanim.c vd_msvidc.c vd_fli.c vd_qtrle.c vd_qtsmc.c vd_roqvideo.c vd_cyuv.c vd_nuv.c vd_libmpeg2.c vd_msrle.c vd_huffyuv.c vd_mpegpes.c vd_svq1.c vd_xvid.c vd_libdv.c vd_lcl.c vd_mtga.c VFILTER_SRCS=vf.c vf_vo.c vf_crop.c vf_expand.c vf_pp.c vf_scale.c vf_format.c vf_yuy2.c vf_flip.c vf_rgb2bgr.c vf_rotate.c vf_mirror.c vf_palette.c vf_lavc.c vf_dvbscale.c vf_cropdetect.c vf_test.c vf_noise.c vf_yvu9.c vf_rectangle.c vf_lavcdeint.c vf_eq.c vf_halfpack.c vf_dint.c -ENCODER_SRCS=ve.c ve_divx4.c ve_lavc.c ve_vfw.c ve_rawrgb.c ve_libdv.c +ENCODER_SRCS=ve.c ve_divx4.c ve_lavc.c ve_vfw.c ve_rawrgb.c ve_libdv.c ve_xvid.c NATIVE_SRCS=native/RTjpegN.c native/cinepak.c native/cyuv.c native/fli.c native/minilzo.c native/msvidc.c native/nuppelvideo.c native/qtrle.c native/qtrpza.c native/qtsmc.c native/roqav.c native/xa_gsm.c native/svq1.c ifeq ($(FAME),yes) diff --git a/libmpcodecs/ve.c b/libmpcodecs/ve.c index 00b559de70..1dee6ac7c5 100644 --- a/libmpcodecs/ve.c +++ b/libmpcodecs/ve.c @@ -14,6 +14,7 @@ extern vf_info_t ve_info_lavc; extern vf_info_t ve_info_vfw; extern vf_info_t ve_info_rawrgb; extern vf_info_t ve_info_libdv; +extern vf_info_t ve_info_xvid; static vf_info_t* encoder_list[]={ #ifdef HAVE_DIVX4ENCORE @@ -29,6 +30,9 @@ static vf_info_t* encoder_list[]={ &ve_info_libdv, #endif &ve_info_rawrgb, +#if defined(HAVE_XVID) + &ve_info_xvid, +#endif NULL }; diff --git a/libmpcodecs/ve_xvid.c b/libmpcodecs/ve_xvid.c new file mode 100644 index 0000000000..1875528e18 --- /dev/null +++ b/libmpcodecs/ve_xvid.c @@ -0,0 +1,374 @@ +#include <stdio.h> +#include <stdlib.h> +#include <string.h> + +#include "../config.h" +#include "../mp_msg.h" + +#ifdef HAVE_XVID + +#include "codec-cfg.h" +#include "stream.h" +#include "demuxer.h" +#include "stheader.h" + +#include "aviwrite.h" + +#include "img_format.h" +#include "mp_image.h" +#include "vf.h" + +#include <xvid.h> +#include "xvid_vbr.h" + +#include "cfgparser.h" + +/**********************************************************************/ +/* Divx4 quality to XviD encoder motion flag presets */ +static int const divx4_motion_presets[7] = { + 0, + PMV_EARLYSTOP16, + PMV_EARLYSTOP16 | PMV_ADVANCEDDIAMOND16, + PMV_EARLYSTOP16 | PMV_HALFPELREFINE16, + PMV_EARLYSTOP16 | PMV_HALFPELREFINE16 | PMV_EARLYSTOP8 | PMV_HALFPELREFINE8, + PMV_EARLYSTOP16 | PMV_HALFPELREFINE16 | PMV_EARLYSTOP8 | PMV_HALFPELREFINE8, + PMV_EARLYSTOP16 | PMV_HALFPELREFINE16 | PMV_EXTSEARCH16 | PMV_EARLYSTOP8 | PMV_HALFPELREFINE8 +}; + +/* 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 +}; + +struct { + int quality; + int bitrate; + int rc_reaction_delay_factor; + int rc_averaging_period; + int rc_buffer; + int max_quantizer; + int min_quantizer; + int max_key_interval; + enum { + XVID_MODE_CBR = 0, XVID_MODE_2PASS_1, XVID_MODE_2PASS_2, XVID_MODE_FIXED_QUANT, + XVID_MODE_UNSPEC = -1 + } mode; + int debug; + char *stats_file;; + int keyframe_boost; + int kfthreshold; + int kfreduction; + int min_key_interval; + int fixed_quant; +} xvidenc_param = { + sizeof(divx4_motion_presets) / sizeof(divx4_motion_presets[0]) - 1, /* quality */ + 0, 0, 0, 0, 0, 0, 0, + XVID_MODE_CBR, + 1, /* debug */ + NULL, /* stats_file */ + -1, -1, -1, /* keyframe_boost, kfthreshold, kfreduction */ + -1, /* min_key_interval */ + -1, /* fixed_quant */ +}; + +struct config xvidencopts_conf[] = { + { "quality", &xvidenc_param.quality, CONF_TYPE_INT, CONF_RANGE, 0, + sizeof(divx4_motion_presets) / sizeof(divx4_motion_presets[0]) - 1, NULL}, + { "br", &xvidenc_param.bitrate, CONF_TYPE_INT, 0, 0, 0, NULL}, + { "rc_reaction_delay_factor", &xvidenc_param.rc_reaction_delay_factor, CONF_TYPE_INT, 0, 0, NULL}, + { "rc_averaging_period", &xvidenc_param.rc_averaging_period, CONF_TYPE_INT, 0, 0, NULL}, + { "rc_buffer", &xvidenc_param.rc_buffer, CONF_TYPE_INT, 0, 0, NULL}, + { "max_quantizer", &xvidenc_param.max_quantizer, CONF_TYPE_INT, 0, 0, NULL}, + { "min_quantizer", &xvidenc_param.max_quantizer, CONF_TYPE_INT, 0, 0, NULL}, + { "max_key_interval", &xvidenc_param.max_key_interval, CONF_TYPE_INT, 0, 0, NULL}, + /* cbr, vbrqual, vbrquant, 2pass-1, 2pass-2-int, 2pass-2-ext */ + { "cbr", &xvidenc_param.mode, CONF_TYPE_FLAG, 0, 0, XVID_MODE_CBR, NULL}, + { "2pass-1", &xvidenc_param.mode, CONF_TYPE_FLAG, 0, 0, XVID_MODE_2PASS_1, NULL}, + { "2pass-2", &xvidenc_param.mode, CONF_TYPE_FLAG, 0, 0, XVID_MODE_2PASS_2, NULL}, + { "fixedquant", &xvidenc_param.mode, CONF_TYPE_FLAG, 0, 0, XVID_MODE_FIXED_QUANT, NULL}, + { "nodebug", &xvidenc_param.debug, CONF_TYPE_FLAG, 0, 0, 0, NULL}, + { "debug", &xvidenc_param.debug, CONF_TYPE_FLAG, 0, 0, 1, NULL}, + { "statsfile", &xvidenc_param.stats_file, CONF_TYPE_STRING, 0, 0, 0, NULL}, /* for XVID_MODE_2PASS_1/22 */ + { "keyframe_boost", &xvidenc_param.keyframe_boost, CONF_TYPE_INT, 0, 0, 0, NULL}, /* for XVID_MODE_2PASS_2 */ + { "kfthreshold", &xvidenc_param.kfthreshold, CONF_TYPE_INT, 0, 0, 0, NULL}, /* for XVID_MODE_2PASS_2 */ + { "kfreduction", &xvidenc_param.kfreduction, CONF_TYPE_INT, 0, 0, 0, NULL}, /* for XVID_MODE_2PASS_2 */ + { "min_key_interval", &xvidenc_param.max_key_interval, CONF_TYPE_INT, 0, 0, 0, NULL}, /* for XVID_MODE_2PASS_2 */ + { "fixed_quant", &xvidenc_param.fixed_quant, CONF_TYPE_INT, CONF_RANGE, 1, 31, NULL}, /* for XVID_MODE_FIXED_QUANT */ + { NULL, NULL, 0, 0, 0, 0, NULL} +}; + +struct vf_priv_s { + aviwrite_stream_t* mux; + XVID_ENC_FRAME enc_frame; + void* enc_handle; + vbr_control_t vbr_state; + FILE *hintfile; +}; + +static FILE * +get_hint_file(struct vf_instance_s* vf, unsigned char *mode) +{ + if (vf->priv->hintfile == NULL) { + vf->priv->hintfile = fopen("xvid_hint_me.dat", mode); + if (vf->priv->hintfile == NULL) + perror("xvid: could not open xvid_hint_me.dat"); + } + return vf->priv->hintfile; +} + +static int +config(struct vf_instance_s* vf, + int width, int height, int d_width, int d_height, + unsigned int flags, unsigned int outfmt) +{ + XVID_ENC_PARAM enc_param; + + vf->priv->mux->bih->biWidth = width; + vf->priv->mux->bih->biHeight = height; + vf->priv->mux->bih->biSizeImage = vf->priv->mux->bih->biWidth * vf->priv->mux->bih->biHeight * 3; + + memset(&enc_param, 0, sizeof(enc_param)); + enc_param.width = width; + enc_param.height = height; + enc_param.fincr = vf->priv->mux->h.dwScale; + enc_param.fbase = vf->priv->mux->h.dwRate; + enc_param.rc_bitrate = xvidenc_param.bitrate; + enc_param.rc_reaction_delay_factor = xvidenc_param.rc_reaction_delay_factor; + enc_param.rc_averaging_period = xvidenc_param.rc_averaging_period; + enc_param.rc_buffer = xvidenc_param.rc_buffer; + enc_param.max_quantizer = xvidenc_param.max_quantizer; + enc_param.min_quantizer = xvidenc_param.min_quantizer; + enc_param.max_key_interval = xvidenc_param.max_key_interval; + switch (xvid_encore(NULL, XVID_ENC_CREATE, &enc_param, NULL)) { + case XVID_ERR_FAIL: + fprintf(stderr, "xvid: encoder creation failed\n"); + return 0; + case XVID_ERR_MEMORY: + fprintf(stderr, "xvid: encoder creation failed, out of memory\n"); + return 0; + case XVID_ERR_FORMAT: + fprintf(stderr, "xvid: encoder creation failed, bad format\n"); + return 0; + } + vf->priv->enc_handle = enc_param.handle; + + vf->priv->enc_frame.general = divx4_general_presets[xvidenc_param.quality]; + vf->priv->enc_frame.motion = divx4_motion_presets[xvidenc_param.quality]; + switch (outfmt) { + case IMGFMT_YV12: + vf->priv->enc_frame.colorspace = XVID_CSP_YV12; + break; + case IMGFMT_IYUV: case IMGFMT_I420: + vf->priv->enc_frame.colorspace = XVID_CSP_I420; + break; + case IMGFMT_YUY2: + vf->priv->enc_frame.colorspace = XVID_CSP_YUY2; + break; + case IMGFMT_UYVY: + vf->priv->enc_frame.colorspace = XVID_CSP_UYVY; + break; + case IMGFMT_RGB24: case IMGFMT_BGR24: + vf->priv->enc_frame.colorspace = XVID_CSP_RGB24; + break; + default: + mp_msg(MSGT_MENCODER,MSGL_ERR,"xvid: unsupported picture format (%s)!\n", + vo_format_name(outfmt)); + return 0; + } + vf->priv->enc_frame.quant_intra_matrix = 0; + vf->priv->enc_frame.quant_inter_matrix = 0; + + vf->priv->vbr_state.debug = xvidenc_param.debug; + vbrSetDefaults(&vf->priv->vbr_state); + vf->priv->vbr_state.fps = (double)enc_param.fbase / enc_param.fincr; + if (xvidenc_param.stats_file) + vf->priv->vbr_state.filename = xvidenc_param.stats_file; + if (xvidenc_param.bitrate) + vf->priv->vbr_state.desired_bitrate = xvidenc_param.bitrate; + if (xvidenc_param.keyframe_boost) + vf->priv->vbr_state.keyframe_boost = xvidenc_param.keyframe_boost; + if (xvidenc_param.kfthreshold) + vf->priv->vbr_state.kftreshold = xvidenc_param.kfthreshold; + if (xvidenc_param.kfreduction) + vf->priv->vbr_state.kfreduction = xvidenc_param.kfreduction; + if (xvidenc_param.min_key_interval) + vf->priv->vbr_state.min_key_interval = xvidenc_param.min_key_interval; + if (xvidenc_param.max_key_interval) + vf->priv->vbr_state.max_key_interval = xvidenc_param.max_key_interval; + if (xvidenc_param.fixed_quant) + vf->priv->vbr_state.fixed_quant = xvidenc_param.fixed_quant; + switch (xvidenc_param.mode) { + case XVID_MODE_CBR: + vf->priv->vbr_state.mode = VBR_MODE_1PASS; + break; + case XVID_MODE_FIXED_QUANT: + vf->priv->vbr_state.mode = VBR_MODE_FIXED_QUANT; + break; + case XVID_MODE_2PASS_1: + vf->priv->vbr_state.mode = VBR_MODE_2PASS_1; + break; + case XVID_MODE_2PASS_2: + vf->priv->vbr_state.mode = VBR_MODE_2PASS_2; + break; + default: + abort(); + } + vbrInit(&vf->priv->vbr_state); + return 1; +} + +static void +uninit(struct vf_instance_s* vf) +{ + if (vf->priv->hintfile) + fclose(vf->priv->hintfile); + vbrFinish(&vf->priv->vbr_state); +} + +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 VFCAP_CSP_SUPPORTED | VFCAP_CSP_SUPPORTED_BY_HW; + case IMGFMT_YUY2: case IMGFMT_UYVY: + return VFCAP_CSP_SUPPORTED; + case IMGFMT_RGB24: case IMGFMT_BGR24: + return VFCAP_CSP_SUPPORTED | VFCAP_FLIPPED; + } + return 0; +} + +static int +put_image(struct vf_instance_s* vf, mp_image_t *mpi) +{ + XVID_ENC_STATS enc_stats; + + vf->priv->enc_frame.bitstream = vf->priv->mux->buffer; + vf->priv->enc_frame.length = -1 /* vf->priv->mux->buffer_size */; + vf->priv->enc_frame.image = mpi->planes[0]; + vf->priv->enc_frame.quant = vbrGetQuant(&vf->priv->vbr_state); + vf->priv->enc_frame.intra = vbrGetIntra(&vf->priv->vbr_state); +#if 0 + if (xvidenc_param.mode == XVID_MODE_2PASS_1) { + vf->priv->enc_frame.hint.hintstream = hintstream; + vf->priv->enc_frame.hint.rawhints = 0; + vf->priv->enc_frame.general |= XVID_HINTEDME_GET; + } + else if (xvidenc_param.mode == XVID_MODE_2PASS_2) { + FILE *f = get_hint_file(vf, "r"); + vf->priv->enc_frame.general &= ~XVID_HINTEDME_SET; + if (f) { + int blocksize; + int read; + read = fread(&blocksize, sizeof(blocksize), 1, f); + if (read == 1) { + read = fread(hintstream, blocksize, 1, f); + if (read == blocksize) { + vf->priv->enc_frame.hint.hintstream = hintstream; + vf->priv->enc_frame.hint.rawhints = 0; + vf->priv->enc_frame.general |= XVID_HINTEDME_SET; + } + else + perror("xvid: hint file read block failure"); + } + else + perror("xvid: hint file read failure"); + } + } +#endif + switch (xvid_encore(vf->priv->enc_handle, XVID_ENC_ENCODE, &vf->priv->enc_frame, &enc_stats)) { + case XVID_ERR_OK: + break; + case XVID_ERR_MEMORY: + mp_msg(MSGT_MENCODER, MSGL_ERR, "xvid: out of memory\n"); + break; + case XVID_ERR_FORMAT: + mp_msg(MSGT_MENCODER, MSGL_ERR, "xvid: bad format\n"); + break; + default: + mp_msg(MSGT_MENCODER, MSGL_ERR, "xvid: failure\n"); + break; + } + mencoder_write_chunk(vf->priv->mux, vf->priv->enc_frame.length, vf->priv->enc_frame.intra ? 0x10 : 0); + vbrUpdate(&vf->priv->vbr_state, enc_stats.quant, vf->priv->enc_frame.intra, + enc_stats.hlength, vf->priv->enc_frame.length, enc_stats.kblks, enc_stats.mblks, enc_stats.ublks); +#if 1 + if (vf->priv->enc_frame.general & XVID_HINTEDME_GET) { + FILE *f = get_hint_file(vf, "w"); + if (f) { + unsigned int wrote; + wrote = fwrite(&vf->priv->enc_frame.hint.hintlength, sizeof(vf->priv->enc_frame.hint.hintlength), 1, f); + if (wrote == 1) { + wrote = fwrite(&vf->priv->enc_frame.hint.hintstream, vf->priv->enc_frame.hint.hintlength, 1, f); + if (wrote != 1) + perror("xvid: hint write block failure"); + } + else + perror("xvid: hint write failure"); + } + } +#endif + return 1; +} + +//===========================================================================// + +static int +vf_open(vf_instance_t *vf, char* args) +{ + XVID_INIT_PARAM params = { 0, 0, 0}; + vf->config = config; + vf->control = control; + vf->uninit = uninit; + 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 = (aviwrite_stream_t*)args; + + vf->priv->mux->bih = malloc(sizeof(BITMAPINFOHEADER)); + vf->priv->mux->bih->biSize = sizeof(BITMAPINFOHEADER); + vf->priv->mux->bih->biWidth = 0; + vf->priv->mux->bih->biHeight = 0; + vf->priv->mux->bih->biPlanes = 1; + vf->priv->mux->bih->biBitCount = 24; + vf->priv->mux->bih->biCompression = mmioFOURCC('X','V','I','D'); + + if (xvid_init(NULL, 0, ¶ms, NULL) != XVID_ERR_OK) { + fprintf(stderr, "xvid: initialisation failure\n"); + abort(); + } + if (params.api_version != API_VERSION) { + fprintf(stderr, "xvid: XviD library API version mismatch\n" + "\texpected %d.%d, got %d.%d, you should recompile MPlayer.\n", + API_VERSION >> 16, API_VERSION & 0xff, + params.api_version >> 16, params.api_version & 0xff); + abort(); + } + + return 1; +} + +vf_info_t ve_info_xvid = { + "XviD encoder", + "xvid", + "Kim Minh Kaplan", + "for internal use by mencoder", + vf_open +}; + +//===========================================================================// +#endif |