libmpcodecs/vd_lzo.c

   1 #include <stdio.h>
   2 #include <stdlib.h>
   3
   4 #include "config.h"
   5 #include "mp_msg.h"
   6
   7 #include "vd_internal.h"
   8
   9 #ifdef USE_LIBLZO
  10 #include <lzo1x.h>
  11 #else
  12 #include "native/minilzo.h"
  13 #define lzo_malloc malloc
  14 #define lzo_free free
  15 #endif
  16
  17 #define MOD_NAME "DecLZO"
  18
  19 static vd_info_t info = {
  20         "LZO compressed Video",
  21         "lzo",
  22         "Tilmann Bitterberg",
  23         "Transcode development team <http://www.theorie.physik.uni-goettingen.de/~ostreich/transcode/>",
  24         "based on liblzo: http://www.oberhumer.com/opensource/lzo/"
  25 };
  26
  27 LIBVD_EXTERN(lzo)
  28
  29 typedef struct {
  30     lzo_byte *wrkmem;
  31     int codec;
  32 } lzo_context_t;
  33
  34 // to set/get/query special features/parameters
  35 static int control (sh_video_t *sh, int cmd, void* arg, ...)
  36 {
  37     lzo_context_t *priv = sh->context;
  38     //printf("[%s] Query!! (%s)\n", MOD_NAME, (codec==IMGFMT_BGR24)?"BGR":"none");
  39     //printf("[%s] Query!! (%s)\n", MOD_NAME, (codec==IMGFMT_YV12)?"YV12":"none");
  40     switch(cmd){
  41     case VDCTRL_QUERY_FORMAT:
  42         if( (*((int*)arg)) == IMGFMT_BGR24 && priv->codec == IMGFMT_BGR24) return CONTROL_TRUE;
  43         if( (*((int*)arg)) == IMGFMT_YV12 && priv->codec == IMGFMT_YV12) return CONTROL_TRUE;
  44         return CONTROL_FALSE;
  45     }
  46     return CONTROL_UNKNOWN;
  47 }
  48
  49
  50 // init driver
  51 static int init(sh_video_t *sh)
  52 {
  53     lzo_context_t *priv;
  54
  55     if (lzo_init() != LZO_E_OK) {
  56         mp_msg (MSGT_DECVIDEO, MSGL_ERR, "[%s] lzo_init() failed\n", MOD_NAME);
  57         return 0;
  58     }
  59
  60     priv = malloc(sizeof(lzo_context_t));
  61     if (!priv)
  62     {
  63         mp_msg (MSGT_DECVIDEO, MSGL_ERR, "[%s] memory allocation failed\n", MOD_NAME);
  64         return 0;
  65     }
  66     priv->codec = -1;
  67     sh->context = priv;
  68
  69     priv->wrkmem = (lzo_bytep) lzo_malloc(LZO1X_1_MEM_COMPRESS);
  70
  71     if (priv->wrkmem == NULL) {
  72         mp_msg (MSGT_DECVIDEO, MSGL_ERR, "[%s] Cannot alloc work memory\n", MOD_NAME);
  73         return 0;
  74     }
  75
  76     return 1;
  77 }
  78
  79 // uninit driver
  80 static void uninit(sh_video_t *sh)
  81 {
  82     lzo_context_t *priv = sh->context;
  83
  84     if (priv)
  85     {
  86         if (priv->wrkmem)
  87             lzo_free(priv->wrkmem);
  88         free(priv);
  89     }
  90
  91     sh->context = NULL;
  92 }
  93
  94 //mp_image_t* mpcodecs_get_image(sh_video_t *sh, int mp_imgtype, int mp_imgflag, int w, int h);
  95
  96 // decode a frame
  97 static mp_image_t* decode(sh_video_t *sh,void* data,int len,int flags)
  98 {
  99     static int init_done = 0;
 100     int r;
 101     mp_image_t* mpi;
 102     int w;
 103     lzo_context_t *priv = sh->context;
 104
 105     if (len <= 0) {
 106             return NULL; // skipped frame
 107     }
 108
 109
 110     if (!init_done) {
 111         lzo_byte *tmp=NULL;
 112
 113         // decompress one frame to see if its
 114         // either YV12 or RGB24
 115         if (!tmp) tmp = lzo_malloc(sh->bih->biSizeImage);
 116
 117         mp_msg (MSGT_DECVIDEO, MSGL_V, "[%s] 2 depth %d, format %d data %p len (%d) (%d)\n",
 118             MOD_NAME, sh->bih->biBitCount, sh->format, data, len, sh->bih->biSizeImage
 119             );
 120
 121         /* decompress the frame */
 122         w = sh->bih->biSizeImage;
 123         r = lzo1x_decompress_safe (data, len, tmp, &w, priv->wrkmem);
 124         free(tmp);
 125
 126         if (r != LZO_E_OK) {
 127             /* this should NEVER happen */
 128             mp_msg (MSGT_DECVIDEO, MSGL_ERR,
 129                     "[%s] internal error - decompression failed: %d\n", MOD_NAME, r);
 130             return NULL;
 131         }
 132
 133         if        (w == (sh->bih->biSizeImage))   {
 134             priv->codec = IMGFMT_BGR24;
 135             mp_msg (MSGT_DECVIDEO, MSGL_V, "[%s] codec choosen is BGR24\n", MOD_NAME);
 136         } else if (w == (sh->bih->biSizeImage)/2) {
 137             priv->codec = IMGFMT_YV12;
 138             mp_msg (MSGT_DECVIDEO, MSGL_V, "[%s] codec choosen is YV12\n", MOD_NAME);
 139         } else {
 140             priv->codec = -1;
 141             mp_msg(MSGT_DECVIDEO,MSGL_ERR,"[%s] Unsupported out_fmt\n", MOD_NAME);
 142             return NULL;
 143         }
 144
 145         if(!mpcodecs_config_vo(sh,sh->disp_w,sh->disp_h,priv->codec)) return NULL;
 146         init_done++;
 147     }
 148
 149     mpi = mpcodecs_get_image(sh, MP_IMGTYPE_TEMP, 0,
 150         sh->disp_w, sh->disp_h);
 151
 152
 153     if (!mpi) {
 154             mp_msg (MSGT_DECVIDEO, MSGL_ERR, "[%s] mpcodecs_get_image failed\n", MOD_NAME);
 155             return NULL;
 156     }
 157
 158     w = mpi->w * mpi->h;
 159     r = lzo1x_decompress_safe (data, len, mpi->planes[0], &w, priv->wrkmem);
 160     if (r != LZO_E_OK) {
 161         /* this should NEVER happen */
 162         mp_msg (MSGT_DECVIDEO, MSGL_ERR,
 163                 "[%s] internal error - decompression failed: %d\n", MOD_NAME, r);
 164       return NULL;
 165     }
 166
 167     mp_msg (MSGT_DECVIDEO, MSGL_DBG2,
 168                 "[%s] decompressed %lu bytes into %lu bytes\n", MOD_NAME,
 169                 (long) len, (long)w);
 170
 171     return mpi;
 172 }
 173
 174 /* vim: sw=4
 175    */