realtime yv12->mpeg1 with libavcodec qscale=1

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

1 files changed, 147 insertions, 0 deletions

diff --git a/libmpcodecs/vf_lavc.c b/libmpcodecs/vf_lavc.c
new file mode 100644
index 0000000000..f6ec6ccd15
--- /dev/null
+++ b/libmpcodecs/vf_lavc.c
@@ -0,0 +1,147 @@
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <inttypes.h>
+
+#include "../config.h"
+#include "../mp_msg.h"
+#include "../help_mp.h"
+
+#ifdef USE_LIBAVCODEC
+
+#include "img_format.h"
+#include "mp_image.h"
+#include "vf.h"
+
+//#include "../libvo/fastmemcpy.h"
+
+#ifdef USE_LIBAVCODEC_SO
+#include <libffmpeg/avcodec.h>
+#else
+#include "libavcodec/avcodec.h"
+#endif
+
+extern int avcodec_inited;
+
+struct vf_priv_s {
+    unsigned char* outbuf;
+    int outbuf_size;
+    AVCodecContext context;
+    AVCodec *codec;
+    vo_mpegpes_t pes;
+};
+
+#define lavc_venc_context (vf->priv->context)
+
+//===========================================================================//
+
+static int config(struct vf_instance_s* vf,
+        int width, int height, int d_width, int d_height,
+	unsigned int flags, unsigned int outfmt){
+    if(vf_next_query_format(vf,IMGFMT_MPEGPES)<=0) return 0;
+
+    lavc_venc_context.width = width;
+    lavc_venc_context.height = height;
+
+    if(vf->priv->outbuf) free(vf->priv->outbuf);
+
+    vf->priv->outbuf_size=10000+width*height;  // must be enough!
+    if(vf->priv->outbuf) free(vf->priv->outbuf);
+    vf->priv->outbuf = malloc(vf->priv->outbuf_size);
+
+    if (avcodec_open(&lavc_venc_context, vf->priv->codec) != 0) {
+	mp_msg(MSGT_MENCODER,MSGL_ERR,MSGTR_CantOpenCodec);
+	return 0;
+    }
+
+    if (lavc_venc_context.codec->encode == NULL) {
+	mp_msg(MSGT_MENCODER,MSGL_ERR,"avcodec init failed (ctx->codec->encode == NULL)!\n");
+	return 0;
+    }
+
+    return vf_next_config(vf,width,height,d_width,d_height,flags,IMGFMT_MPEGPES);
+}
+
+static void put_image(struct vf_instance_s* vf, mp_image_t *mpi){
+    mp_image_t* dmpi;
+    int out_size;
+    AVPicture lavc_venc_picture;
+
+    lavc_venc_picture.data[0]=mpi->planes[0];
+    lavc_venc_picture.data[1]=mpi->planes[1];
+    lavc_venc_picture.data[2]=mpi->planes[2];
+    lavc_venc_picture.linesize[0]=mpi->stride[0];
+    lavc_venc_picture.linesize[1]=mpi->stride[1];
+    lavc_venc_picture.linesize[2]=mpi->stride[2];
+
+    out_size = avcodec_encode_video(&lavc_venc_context, 
+	vf->priv->outbuf, vf->priv->outbuf_size, &lavc_venc_picture);
+
+    if(out_size<=0) return;
+
+    dmpi=vf_get_image(vf->next,IMGFMT_MPEGPES,
+	MP_IMGTYPE_EXPORT, 0,
+	mpi->w, mpi->h);
+    
+    vf->priv->pes.data=vf->priv->outbuf;
+    vf->priv->pes.size=out_size;
+    vf->priv->pes.id=0x1E0;
+    vf->priv->pes.timestamp=-1; // dunno
+    
+    dmpi->planes[0]=&vf->priv->pes;
+    
+    vf_next_put_image(vf,dmpi);
+}
+
+//===========================================================================//
+
+static int query_format(struct vf_instance_s* vf, unsigned int fmt){
+    switch(fmt){
+    case IMGFMT_YV12:
+    case IMGFMT_I420:
+    case IMGFMT_IYUV:
+	return (vf_next_query_format(vf,fmt) & (~(VFCAP_CSP_SUPPORTED_BY_HW|VFCAP_ACCEPT_STRIDE)));
+    }
+    return 0;
+}
+
+static int open(vf_instance_t *vf, char* args){
+    vf->config=config;
+    vf->put_image=put_image;
+    vf->query_format=query_format;
+    vf->priv=malloc(sizeof(struct vf_priv_s));
+    memset(vf->priv,0,sizeof(struct vf_priv_s));
+
+    if (!avcodec_inited){
+	avcodec_init();
+	avcodec_register_all();
+	avcodec_inited=1;
+    }
+
+    vf->priv->codec = (AVCodec *)avcodec_find_encoder_by_name("mpeg1video");
+    if (!vf->priv->codec) {
+	mp_msg(MSGT_MENCODER,MSGL_ERR,MSGTR_MissingLAVCcodec, "mpeg1video");
+	return 0;
+    }
+
+    // TODO: parse args ->
+    lavc_venc_context.bit_rate = 8000000;
+    lavc_venc_context.frame_rate = 25 * FRAME_RATE_BASE;
+    lavc_venc_context.qmin= 1;
+    lavc_venc_context.gop_size = 0; // I-only
+    lavc_venc_context.flags = CODEC_FLAG_QSCALE;
+    lavc_venc_context.quality = 1;
+
+    return 1;
+}
+
+vf_info_t vf_info_lavc = {
+    "realtime mpeg1 encoding with libavcodec",
+    "lavc",
+    "A'rpi",
+    "",
+    open
+};
+
+//===========================================================================//
+#endif