realtime yv12->mpeg1 with libavcodec qscale=1

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

libmpcodecs/Makefile

@@ -6,7 +6,7 @@ LIBNAME2 = libmpencoders.a
 
 AUDIO_SRCS=dec_audio.c ad.c ad_a52.c ad_acm.c ad_alaw.c ad_dk3adpcm.c ad_dk4adpcm.c ad_dshow.c ad_dvdpcm.c ad_ffmpeg.c ad_hwac3.c ad_imaadpcm.c ad_mp3.c ad_msadpcm.c ad_pcm.c ad_roqaudio.c ad_msgsm.c ad_faad.c ad_vorbis.c ad_libmad.c
 VIDEO_SRCS=dec_video.c vd.c vd_null.c vd_cinepak.c vd_qtrpza.c vd_ffmpeg.c vd_dshow.c vd_vfw.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_zlib.c vd_mpegpes.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
+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
 ENCODER_SRCS=ve.c ve_divx4.c ve_lavc.c ve_vfw.c ve_rawrgb.c ve_libdv.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
 

libmpcodecs/vf.c

@@ -24,6 +24,7 @@ extern vf_info_t vf_info_rgb2bgr;
 extern vf_info_t vf_info_rotate;
 extern vf_info_t vf_info_mirror;
 extern vf_info_t vf_info_palette;
+extern vf_info_t vf_info_lavc;
 
 char** vo_plugin_args=(char**) NULL;
 
@@ -45,6 +46,7 @@ static vf_info_t* filter_list[]={
     &vf_info_rotate,
     &vf_info_mirror,
     &vf_info_palette,
+    &vf_info_lavc,
     NULL
 };
 

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