HM12 & NV12 "decoder" (specially interleaved YUV formats, used by Hauppauge's PVR cards)

based on patch by Alex <d18c7db@hotmail.com>


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

1 files changed, 138 insertions, 0 deletions

diff --git a/libmpcodecs/vd_hmblck.c b/libmpcodecs/vd_hmblck.c
new file mode 100644
index 0000000000..bbecf4a6f8
--- /dev/null
+++ b/libmpcodecs/vd_hmblck.c
@@ -0,0 +1,138 @@
+#include <stdio.h>
+#include <stdlib.h>
+
+#include "config.h"
+#include "mp_msg.h"
+
+#include "vd_internal.h"
+
+#define TEMP_BUF_SIZE (720*576)
+
+static vd_info_t info = {
+	"Hauppauge Macroblock/NV12 Decoder",
+	"hmblck",
+	"Alex <d18c7db@hotmail.com>, A'rpi",
+	"Alex <d18c7db@hotmail.com>",
+	"uncompressed"
+};
+
+LIBVD_EXTERN(hmblck)
+
+static void de_macro_y(unsigned char* dst,unsigned char* src,int dstride,int w,int h){
+    unsigned int y;
+    // descramble Y plane
+    for (y=0; y<h; y+=16) {
+	unsigned int x;
+        for (x=0; x<w; x+=16) {
+	    unsigned int i;
+            for (i=0; i<16; i++) {
+                memcpy(dst + x + (y+i)*dstride, src, 16);
+                src+=16;
+            }
+        }
+    }
+}
+
+static void de_macro_uv(unsigned char* dstu,unsigned char* dstv,unsigned char* src,int dstride,int w,int h){
+    unsigned int y;
+    // descramble U/V plane
+    for (y=0; y<h; y+=16) {
+	unsigned int x;
+        for (x=0; x<w; x+=8) {
+	    unsigned int i;
+            for (i=0; i<16; i++) {
+		int idx=x + (y+i)*dstride;
+		dstu[idx+0]=src[0]; dstv[idx+0]=src[1];
+		dstu[idx+1]=src[2]; dstv[idx+1]=src[3];
+		dstu[idx+2]=src[4]; dstv[idx+2]=src[5];
+		dstu[idx+3]=src[6]; dstv[idx+3]=src[7];
+		dstu[idx+4]=src[8]; dstv[idx+4]=src[9];
+		dstu[idx+5]=src[10]; dstv[idx+5]=src[11];
+		dstu[idx+6]=src[12]; dstv[idx+6]=src[13];
+		dstu[idx+7]=src[14]; dstv[idx+7]=src[15];
+                src+=16;
+            }
+        }
+    }
+}
+
+/*************************************************************************
+ * convert a nv12 buffer to yv12
+ */
+static int nv12_to_yv12(void *data, int len, mp_image_t* mpi) {
+    unsigned int Y_size  = mpi->width * mpi->height;
+    unsigned int UV_size = mpi->chroma_width * mpi->chroma_height;
+    unsigned int idx;
+    unsigned char *dst_Y = mpi->planes[0];
+    unsigned char *dst_U = mpi->planes[1];
+    unsigned char *dst_V = mpi->planes[2];
+    unsigned char *src   = data + Y_size;
+
+    // sanity check raw stream
+    if ( (len != (Y_size + (UV_size<<1))) ) {
+        mp_msg(MSGT_DECVIDEO, MSGL_ERR,
+               "hmblck: Image size inconsistent with data size.\n");
+        return 0;
+    }
+    if ( (mpi->width > 720) || (mpi->height > 576) ) {
+        mp_msg(MSGT_DECVIDEO,MSGL_ERR,
+               "hmblck: Image size is too big.\n");
+        return 0;
+    }
+    if (mpi->num_planes != 3) {
+        mp_msg(MSGT_DECVIDEO,MSGL_ERR,
+               "hmblck: Incorrect number of image planes.\n");
+        return 0;
+    }
+
+    // luma data is easy, just copy it
+    memcpy(dst_Y, data, Y_size);
+
+    // chroma data is interlaced UVUV... so deinterlace it
+    for(idx=0; idx<UV_size; idx++ ) {
+        *(dst_U + idx) = *(src + (idx<<1) + 0); 
+        *(dst_V + idx) = *(src + (idx<<1) + 1);
+    }
+    return 1;
+}
+
+/*************************************************************************
+ * set/get/query special features/parameters
+ */
+static int control(sh_video_t *sh,int cmd, void *arg,...){
+    return CONTROL_UNKNOWN;
+}
+/*************************************************************************
+ * init driver
+ */
+static int init(sh_video_t *sh){
+    return mpcodecs_config_vo(sh,sh->disp_w,sh->disp_h,sh->format);
+}
+/*************************************************************************
+ * uninit driver
+ */
+static void uninit(sh_video_t *sh){
+}
+/*************************************************************************
+ * decode a frame
+ */
+static mp_image_t* decode(sh_video_t *sh,void* data,int len,int flags){
+    mp_image_t* mpi;
+
+    if(len<=0) return NULL; // skipped frame
+
+    mpi=mpcodecs_get_image(sh, MP_IMGTYPE_TEMP, MP_IMGFLAG_ACCEPT_STRIDE,
+        sh->disp_w, sh->disp_h);
+    if(!mpi) return NULL;
+
+    if(sh->format == IMGFMT_HM12) {
+        //if(!de_macro(sh, data, len, flags, mpi)) return NULL;
+	de_macro_y(mpi->planes[0],data,mpi->stride[0],mpi->w,mpi->h);
+	de_macro_uv(mpi->planes[1],mpi->planes[2],data+mpi->w*mpi->h,mpi->stride[1],mpi->w/2,mpi->h/2);
+    } else {
+	if(!nv12_to_yv12(data, len, mpi)) return NULL;
+    }
+
+    return mpi;
+}
+/*************************************************************************/