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_LIBAVUTIL_SO
  10 #include <ffmpeg/lzo.h>
  11 #else
  12 #include "libavutil/lzo.h"
  13 #endif
  14
  15 #define MOD_NAME "DecLZO"
  16
  17 static vd_info_t info = {
  18         "LZO compressed Video",
  19         "lzo",
  20         "Tilmann Bitterberg",
  21         "Transcode development team <http://www.theorie.physik.uni-goettingen.de/~ostreich/transcode/>",
  22         "based on liblzo: http://www.oberhumer.com/opensource/lzo/"
  23 };
  24
  25 LIBVD_EXTERN(lzo)
  26
  27 typedef struct {
  28     uint8_t *buffer;
  29     int bufsz;
  30     int codec;
  31 } lzo_context_t;
  32
  33 // to set/get/query special features/parameters
  34 static int control (sh_video_t *sh, int cmd, void* arg, ...)
  35 {
  36     lzo_context_t *priv = sh->context;
  37     switch(cmd){
  38     case VDCTRL_QUERY_FORMAT:
  39         if (*(int *)arg == priv->codec) return CONTROL_TRUE;
  40         return CONTROL_FALSE;
  41     }
  42     return CONTROL_UNKNOWN;
  43 }
  44
  45
  46 // init driver
  47 static int init(sh_video_t *sh)
  48 {
  49     lzo_context_t *priv;
  50
  51     if (sh->bih->biSizeImage <= 0) {
  52         mp_msg (MSGT_DECVIDEO, MSGL_ERR, "[%s] Invalid frame size\n", MOD_NAME);
  53         return 0;
  54     }
  55
  56     priv = malloc(sizeof(lzo_context_t));
  57     if (!priv)
  58     {
  59         mp_msg (MSGT_DECVIDEO, MSGL_ERR, "[%s] memory allocation failed\n", MOD_NAME);
  60         return 0;
  61     }
  62     priv->bufsz = sh->bih->biSizeImage;
  63     priv->buffer = malloc(priv->bufsz + LZO_OUTPUT_PADDING);
  64     priv->codec = -1;
  65     sh->context = priv;
  66
  67     return 1;
  68 }
  69
  70 // uninit driver
  71 static void uninit(sh_video_t *sh)
  72 {
  73     lzo_context_t *priv = sh->context;
  74
  75     if (priv)
  76     {
  77         free(priv->buffer);
  78         free(priv);
  79     }
  80
  81     sh->context = NULL;
  82 }
  83
  84 // decode a frame
  85 static mp_image_t* decode(sh_video_t *sh,void* data,int len,int flags)
  86 {
  87     int r;
  88     mp_image_t* mpi;
  89     lzo_context_t *priv = sh->context;
  90     int w = priv->bufsz;
  91
  92     if (len <= 0) {
  93             return NULL; // skipped frame
  94     }
  95
  96     r = lzo1x_decode(priv->buffer, &w, data, &len);
  97     if (r) {
  98         /* this should NEVER happen */
  99         mp_msg (MSGT_DECVIDEO, MSGL_ERR,
 100                 "[%s] internal error - decompression failed: %d\n", MOD_NAME, r);
 101       return NULL;
 102     }
 103
 104     if (priv->codec == -1) {
 105         // detect RGB24 vs. YV12 via decoded size
 106         mp_msg (MSGT_DECVIDEO, MSGL_V, "[%s] 2 depth %d, format %d data %p len (%d) (%d)\n",
 107             MOD_NAME, sh->bih->biBitCount, sh->format, data, len, sh->bih->biSizeImage
 108             );
 109
 110         if (w == 0) {
 111             priv->codec = IMGFMT_BGR24;
 112             mp_msg (MSGT_DECVIDEO, MSGL_V, "[%s] codec choosen is BGR24\n", MOD_NAME);
 113         } else if (w == (sh->bih->biSizeImage)/2) {
 114             priv->codec = IMGFMT_YV12;
 115             mp_msg (MSGT_DECVIDEO, MSGL_V, "[%s] codec choosen is YV12\n", MOD_NAME);
 116         } else {
 117             priv->codec = -1;
 118             mp_msg(MSGT_DECVIDEO,MSGL_ERR,"[%s] Unsupported out_fmt\n", MOD_NAME);
 119             return NULL;
 120         }
 121
 122         if(!mpcodecs_config_vo(sh,sh->disp_w,sh->disp_h,priv->codec)) {
 123             priv->codec = -1;
 124             return NULL;
 125         }
 126     }
 127
 128     mpi = mpcodecs_get_image(sh, MP_IMGTYPE_EXPORT, 0,
 129         sh->disp_w, sh->disp_h);
 130
 131
 132     if (!mpi) {
 133             mp_msg (MSGT_DECVIDEO, MSGL_ERR, "[%s] mpcodecs_get_image failed\n", MOD_NAME);
 134             return NULL;
 135     }
 136
 137     mpi->planes[0] = priv->buffer;
 138     if (priv->codec == IMGFMT_BGR24)
 139         mpi->stride[0] = 3 * sh->disp_w;
 140     else {
 141         mpi->stride[0] = sh->disp_w;
 142         mpi->planes[2] = priv->buffer + sh->disp_w*sh->disp_h;
 143         mpi->stride[2] = sh->disp_w / 2;
 144         mpi->planes[1] = priv->buffer + sh->disp_w*sh->disp_h*5/4;
 145         mpi->stride[1] = sh->disp_w / 2;
 146     }
 147
 148     mp_msg (MSGT_DECVIDEO, MSGL_DBG2,
 149                 "[%s] decompressed %lu bytes into %lu bytes\n", MOD_NAME,
 150                 (long) len, (long)w);
 151
 152     return mpi;
 153 }
 154
 155 /* vim: sw=4
 156    */