diff options
author | albeu <albeu@b3059339-0415-0410-9bf9-f77b7e298cf2> | 2003-03-03 11:03:19 +0000 |
---|---|---|
committer | albeu <albeu@b3059339-0415-0410-9bf9-f77b7e298cf2> | 2003-03-03 11:03:19 +0000 |
commit | b9fab7021c782292e12ec85cb257ddc2a3ea6f7d (patch) | |
tree | 81f8cc39be56eaa75b60bb64458a90c2b3588bdf /libmpcodecs/ve_nuv.c | |
parent | 64839d94aeb71b39a3b261f4d808f93be7316e96 (diff) | |
download | mpv-b9fab7021c782292e12ec85cb257ddc2a3ea6f7d.tar.bz2 mpv-b9fab7021c782292e12ec85cb257ddc2a3ea6f7d.tar.xz |
A new nuppel video encoder. Mainly for RT encoding on slow box.
git-svn-id: svn://svn.mplayerhq.hu/mplayer/trunk@9521 b3059339-0415-0410-9bf9-f77b7e298cf2
Diffstat (limited to 'libmpcodecs/ve_nuv.c')
-rw-r--r-- | libmpcodecs/ve_nuv.c | 219 |
1 files changed, 219 insertions, 0 deletions
diff --git a/libmpcodecs/ve_nuv.c b/libmpcodecs/ve_nuv.c new file mode 100644 index 0000000000..f88e558cb9 --- /dev/null +++ b/libmpcodecs/ve_nuv.c @@ -0,0 +1,219 @@ +#include <stdio.h> +#include <stdlib.h> +#include <string.h> + +#include "../config.h" +#include "../mp_msg.h" + +#include "m_option.h" + +#include "codec-cfg.h" +#include "stream.h" +#include "demuxer.h" +#include "stheader.h" + +#include "muxer.h" + +#include "img_format.h" +#include "mp_image.h" +#include "vf.h" + +#include "libmpdemux/nuppelvideo.h" +#include "native/minilzo.h" +#include "native/RTjpegN.h" + +#define LZO_AL(size) (((size) + (sizeof(long) - 1)) / sizeof(long)) +#define LZO_IN_LEN (1024*1024L) +#define LZO_OUT_LEN (LZO_IN_LEN + LZO_IN_LEN / 64 + 16 + 3) + +//===========================================================================// + +struct vf_priv_s { + int raw; // Do not use RTjpeg + int lzo; // Use lzo + unsigned int l,c,q; // Mjpeg param + muxer_stream_t* mux; + uint8_t* buffer; + + int buf_size; + int tbl_wrote; + lzo_byte *zbuffer; + long __LZO_MMODEL *zmem; +}; +#define mux_v (vf->priv->mux) + +struct vf_priv_s nuv_priv_dflt = { + 0, // raw + 1, // lzo + 1,1, // l,c + 255, // q + NULL, + NULL, + 0,0, + NULL,NULL +}; + +m_option_t nuvopts_conf[]={ + {"raw", &nuv_priv_dflt.raw, CONF_TYPE_FLAG, 0, 0, 1, NULL}, + {"rtjpeg", &nuv_priv_dflt.raw, CONF_TYPE_FLAG, 0, 1, 0, NULL}, + {"lzo", &nuv_priv_dflt.lzo, CONF_TYPE_FLAG, 0, 0, 1, NULL}, + {"nolzo", &nuv_priv_dflt.lzo, CONF_TYPE_FLAG, 0, 1, 0, NULL}, + {"q", &nuv_priv_dflt.q, CONF_TYPE_INT, M_OPT_RANGE,3,255, NULL}, + {"l", &nuv_priv_dflt.l, CONF_TYPE_INT, M_OPT_RANGE,0,20, NULL}, + {"c", &nuv_priv_dflt.c, CONF_TYPE_INT, M_OPT_RANGE,0,20, NULL}, + {NULL, NULL, 0, 0, 0, 0, NULL} +}; + +//===========================================================================// + + +static int config(struct vf_instance_s* vf, + int width, int height, int d_width, int d_height, + unsigned int flags, unsigned int outfmt){ + + // We need a buffer wich can holda header and a whole YV12 picture + // or a RTJpeg table + vf->priv->buf_size = width*height*3/2+FRAMEHEADERSIZE; + if(vf->priv->buf_size < (int)(128*sizeof(long int) + FRAMEHEADERSIZE)) + vf->priv->buf_size = 128*sizeof(long int) + FRAMEHEADERSIZE; + + mux_v->bih->biWidth=width; + mux_v->bih->biHeight=height; + mux_v->bih->biSizeImage=mux_v->bih->biWidth*mux_v->bih->biHeight*(mux_v->bih->biBitCount/8); + vf->priv->buffer = realloc(vf->priv->buffer,vf->priv->buf_size); + vf->priv->tbl_wrote = 0; + + 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){ + if(fmt==IMGFMT_I420) return 3; + return 0; +} + +static int put_image(struct vf_instance_s* vf, mp_image_t *mpi){ + struct rtframeheader* ench = (struct rtframeheader*)vf->priv->buffer; + uint8_t* data = vf->priv->buffer + FRAMEHEADERSIZE; + uint8_t* zdata = vf->priv->zbuffer + FRAMEHEADERSIZE; + int len = 0, zlen = 0,r; + + memset(vf->priv->buffer,0,FRAMEHEADERSIZE); // Reset the header + if(vf->priv->lzo) + memset(vf->priv->zbuffer,0,FRAMEHEADERSIZE); + + // This has to be don here otherwise tv with sound doesn't work + if(!vf->priv->tbl_wrote) { + RTjpeg_init_compress((long int*)data,mpi->width,mpi->height,vf->priv->q); + RTjpeg_init_mcompress(); + + ench->frametype = 'D'; // compressor data + ench->comptype = 'R'; // compressor data for RTjpeg + ench->packetlength = 128*sizeof(long int); + + mux_v->buffer=vf->priv->buffer; + muxer_write_chunk(mux_v,FRAMEHEADERSIZE + 128*sizeof(long int), 0x10); + vf->priv->tbl_wrote = 1; + memset(ench,0,FRAMEHEADERSIZE); // Reset the header + } + + // Raw picture + if(vf->priv->raw) { + len = mpi->width*mpi->height*3/2; + // Try lzo ??? + if(vf->priv->lzo) { + r = lzo1x_1_compress(mpi->planes[0],mpi->width*mpi->height*3/2, + zdata,&zlen,vf->priv->zmem); + if(r != LZO_E_OK) { + mp_msg(MSGT_VFILTER,MSGL_ERR,"LZO compress error\n"); + zlen = 0; + } + } + + if(zlen <= 0 || zlen > len) { + memcpy(data,mpi->planes[0],len); + ench->comptype = '0'; + } else { // Use lzo only if it's littler + ench = (struct rtframeheader*)vf->priv->zbuffer; + ench->comptype = '3'; + len = zlen; + } + + } else { // RTjpeg compression + len = RTjpeg_mcompressYUV420(data,mpi->planes[0],vf->priv->l, + vf->priv->c); + if(len <= 0) { + mp_msg(MSGT_VFILTER,MSGL_ERR,"RTjpeg_mcompressYUV420 error (%d)\n",len); + return 0; + } + + if(vf->priv->lzo) { + r = lzo1x_1_compress(data,len,zdata,&zlen,vf->priv->zmem); + if(r != LZO_E_OK) { + mp_msg(MSGT_VFILTER,MSGL_ERR,"LZO compress error\n"); + zlen = 0; + } + } + + if(zlen <= 0 || zlen > len) + ench->comptype = '1'; + else { + ench = (struct rtframeheader*)vf->priv->zbuffer; + ench->comptype = '2'; + len = zlen; + } + + } + + ench->frametype = 'V'; // video frame + ench->packetlength = len; + mux_v->buffer=(void*)ench; + muxer_write_chunk(mux_v, len + FRAMEHEADERSIZE, 0x10); + return 1; +} + +//===========================================================================// + +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)); + memcpy(vf->priv, &nuv_priv_dflt,sizeof(struct vf_priv_s)); + //memset(vf->priv,0,sizeof(struct vf_priv_s)); + vf->priv->mux=(muxer_stream_t*)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=12; + mux_v->bih->biCompression = mmioFOURCC('N','U','V','1'); + + if(vf->priv->lzo) { + if(lzo_init() != LZO_E_OK) { + mp_msg(MSGT_VFILTER,MSGL_WARN,"LZO init failed: no lzo compression\n"); + vf->priv->lzo = 0; + } + vf->priv->zbuffer = (lzo_bytep)malloc(FRAMEHEADERSIZE + LZO_OUT_LEN); + vf->priv->zmem = (long*)malloc(sizeof(long)*LZO_AL(LZO1X_1_MEM_COMPRESS)); + } + + return 1; +} + +vf_info_t ve_info_nuv = { + "nuv encoder", + "nuv", + "Albeu", + "for internal use by mencoder", + vf_open +}; + +//===========================================================================// |