libmpcodecs/vd_vfw.c

   1 #include <stdio.h>
   2 #include <stdlib.h>
   3
   4 #include "config.h"
   5 #include "mp_msg.h"
   6 #include "help_mp.h"
   7
   8 #ifdef USE_WIN32DLL
   9
  10 #include "vd_internal.h"
  11
  12 #include "wine/driver.h"
  13 #include "wine/vfw.h"
  14
  15 static vd_info_t info = {
  16 #ifdef BUILD_VFWEX
  17         "Win32/VfWex video codecs",
  18         "vfwex",
  19 #else
  20         "Win32/VfW video codecs",
  21         "vfw",
  22 #endif
  23         "A'rpi & Alex",
  24         "avifile.sf.net",
  25         "win32 codecs"
  26 };
  27
  28 #ifdef BUILD_VFWEX
  29 LIBVD_EXTERN(vfwex)
  30 #else
  31 LIBVD_EXTERN(vfw)
  32 #endif
  33
  34 typedef struct {
  35     BITMAPINFOHEADER *o_bih;
  36     HIC handle;
  37     unsigned char *palette;
  38 } vd_vfw_ctx;
  39
  40 extern int divx_quality;
  41
  42 static int vfw_set_postproc(sh_video_t* sh, int quality)
  43 {
  44     vd_vfw_ctx *priv = sh->context;
  45     // Works only with opendivx/divx4 based DLL
  46     return ICSendMessage(priv->handle, ICM_USER+80, (long)(&quality), 0);
  47 }
  48
  49 static void set_csp(BITMAPINFOHEADER *o_bih,unsigned int outfmt){
  50     int yuv = 0;
  51
  52         switch (outfmt)
  53         {
  54         /* planar format */
  55         case IMGFMT_YV12:
  56         case IMGFMT_I420:
  57         case IMGFMT_IYUV:
  58             o_bih->biBitCount=12;
  59             yuv=1;
  60             break;
  61         case IMGFMT_YVU9:
  62         case IMGFMT_IF09:
  63             o_bih->biBitCount=9;
  64             yuv=1;
  65             break;
  66         /* packed format */
  67         case IMGFMT_YUY2:
  68         case IMGFMT_UYVY:
  69         case IMGFMT_YVYU:
  70             o_bih->biBitCount=16;
  71             yuv=1;
  72             break;
  73         /* rgb/bgr format */
  74         case IMGFMT_RGB8:
  75         case IMGFMT_BGR8:
  76             o_bih->biBitCount=8;
  77             break;
  78         case IMGFMT_RGB15:
  79         case IMGFMT_RGB16:
  80         case IMGFMT_BGR15:
  81         case IMGFMT_BGR16:
  82             o_bih->biBitCount=16;
  83             break;
  84         case IMGFMT_RGB24:
  85         case IMGFMT_BGR24:
  86             o_bih->biBitCount=24;
  87             break;
  88         case IMGFMT_RGB32:
  89         case IMGFMT_BGR32:
  90             o_bih->biBitCount=32;
  91             break;
  92         default:
  93             mp_msg(MSGT_WIN32,MSGL_ERR,"Unsupported image format: %s\n", vo_format_name(outfmt));
  94             return;
  95         }
  96
  97         o_bih->biSizeImage = abs(o_bih->biWidth * o_bih->biHeight * (o_bih->biBitCount/8));
  98
  99 // Note: we cannot rely on sh->outfmtidx here, it's undefined at this stage!!!
 100 //      if (yuv && !(sh->codec->outflags[sh->outfmtidx] & CODECS_FLAG_YUVHACK))
 101         if (yuv)
 102             o_bih->biCompression = outfmt;
 103         else
 104             o_bih->biCompression = 0;
 105 }
 106
 107 // to set/get/query special features/parameters
 108 static int control(sh_video_t *sh,int cmd,void* arg,...){
 109     vd_vfw_ctx *priv = sh->context;
 110     switch(cmd){
 111     case VDCTRL_QUERY_MAX_PP_LEVEL:
 112         return 9;
 113     case VDCTRL_SET_PP_LEVEL:
 114         vfw_set_postproc(sh,10*(*((int*)arg)));
 115         return CONTROL_OK;
 116 #if 1
 117     // FIXME: make this optional...
 118     case VDCTRL_QUERY_FORMAT:
 119       {
 120         HRESULT ret;
 121         if(!(sh->codec->outflags[sh->outfmtidx]&CODECS_FLAG_QUERY))
 122             return CONTROL_UNKNOWN;     // do not query!
 123         set_csp(priv->o_bih,*((int*)arg));
 124 #ifdef BUILD_VFWEX
 125         ret = ICDecompressQueryEx(priv->handle, sh->bih, priv->o_bih);
 126 #else
 127         ret = ICDecompressQuery(priv->handle, sh->bih, priv->o_bih);
 128 #endif
 129         if (ret)
 130         {
 131             mp_msg(MSGT_WIN32, MSGL_DBG2, "ICDecompressQuery failed:: Error %d\n", (int)ret);
 132             return CONTROL_FALSE;
 133         }
 134         return CONTROL_TRUE;
 135       }
 136 #endif
 137     }
 138     return CONTROL_UNKNOWN;
 139 }
 140
 141 extern void print_video_header(BITMAPINFOHEADER *h);
 142
 143 // init driver
 144 static int init(sh_video_t *sh){
 145     HRESULT ret;
 146 //    unsigned int outfmt=sh->codec->outfmt[sh->outfmtidx];
 147     int i, o_bih_len;
 148     vd_vfw_ctx *priv;
 149
 150     /* Hack for VSSH codec: new dll can't decode old files
 151      * In my samples old files have no extradata, so use that info
 152      * to decide what dll should be used (here and in vd_dshow).
 153      */
 154     if (!strcmp(sh->codec->dll, "vssh264.dll") && (sh->bih->biSize > 40))
 155       return 0;
 156
 157     priv = malloc(sizeof(vd_vfw_ctx));
 158     if (!priv)
 159         return 0;
 160     memset(priv, 0, sizeof(vd_vfw_ctx));
 161     sh->context = priv;
 162
 163     mp_msg(MSGT_WIN32,MSGL_V,"======= Win32 (VFW) VIDEO Codec init =======\n");
 164
 165
 166 //    win32_codec_name = sh->codec->dll;
 167 //    sh->hic = ICOpen( 0x63646976, sh->bih->biCompression, ICMODE_FASTDECOMPRESS);
 168 //    priv->handle = ICOpen( 0x63646976, sh->bih->biCompression, ICMODE_DECOMPRESS);
 169     priv->handle = ICOpen( (long)(sh->codec->dll), sh->bih->biCompression, ICMODE_DECOMPRESS);
 170     if(!priv->handle){
 171         mp_msg(MSGT_WIN32,MSGL_ERR,"ICOpen failed! unknown codec / wrong parameters?\n");
 172         return 0;
 173     }
 174
 175 //    sh->bih->biBitCount=32;
 176
 177     o_bih_len = ICDecompressGetFormatSize(priv->handle, sh->bih);
 178
 179     priv->o_bih = malloc(o_bih_len);
 180     memset(priv->o_bih, 0, o_bih_len);
 181
 182     mp_msg(MSGT_WIN32,MSGL_V,"ICDecompressGetFormatSize ret: %d\n", o_bih_len);
 183
 184     ret = ICDecompressGetFormat(priv->handle, sh->bih, priv->o_bih);
 185     if(ret < 0){
 186         mp_msg(MSGT_WIN32,MSGL_ERR,"ICDecompressGetFormat failed: Error %d\n", (int)ret);
 187         for (i=0; i < o_bih_len; i++) mp_msg(MSGT_WIN32, MSGL_DBG2, "%02x ", priv->o_bih[i]);
 188         return 0;
 189     }
 190     mp_msg(MSGT_WIN32,MSGL_V,"ICDecompressGetFormat OK\n");
 191
 192 #if 0
 193     // workaround for pegasus MJPEG:
 194     if(!sh_video->o_bih.biWidth) sh_video->o_bih.biWidth=sh_video->bih->biWidth;
 195     if(!sh_video->o_bih.biHeight) sh_video->o_bih.biHeight=sh_video->bih->biHeight;
 196     if(!sh_video->o_bih.biPlanes) sh_video->o_bih.biPlanes=sh_video->bih->biPlanes;
 197 #endif
 198
 199     // ok let libvo and vd core to handshake and decide the optimal csp:
 200     if(!mpcodecs_config_vo(sh,sh->disp_w,sh->disp_h,IMGFMT_YUY2)) return 0;
 201
 202     if (!(sh->codec->outflags[sh->outfmtidx]&CODECS_FLAG_FLIP)) {
 203         priv->o_bih->biHeight=-sh->bih->biHeight; // flip image!
 204     }
 205
 206     // ok, let's set the choosen colorspace:
 207     set_csp(priv->o_bih,sh->codec->outfmt[sh->outfmtidx]);
 208
 209     // fake it to RGB for broken DLLs (divx3)
 210     if(sh->codec->outflags[sh->outfmtidx] & CODECS_FLAG_YUVHACK)
 211         priv->o_bih->biCompression = 0;
 212
 213     // sanity check:
 214 #if 1
 215 #ifdef BUILD_VFWEX
 216     ret = ICDecompressQueryEx(priv->handle, sh->bih, priv->o_bih);
 217 #else
 218     ret = ICDecompressQuery(priv->handle, sh->bih, priv->o_bih);
 219 #endif
 220     if (ret)
 221     {
 222         mp_msg(MSGT_WIN32,MSGL_WARN,"ICDecompressQuery failed: Error %d\n", (int)ret);
 223 //      return 0;
 224     } else
 225         mp_msg(MSGT_WIN32,MSGL_V,"ICDecompressQuery OK\n");
 226 #endif
 227
 228 #ifdef BUILD_VFWEX
 229     ret = ICDecompressBeginEx(priv->handle, sh->bih, priv->o_bih);
 230 #else
 231     ret = ICDecompressBegin(priv->handle, sh->bih, priv->o_bih);
 232 #endif
 233     if (ret)
 234     {
 235         mp_msg(MSGT_WIN32,MSGL_WARN,"ICDecompressBegin failed: Error %d\n", (int)ret);
 236 //      return 0;
 237     }
 238
 239     // for broken codecs set it again:
 240     if(sh->codec->outflags[sh->outfmtidx] & CODECS_FLAG_YUVHACK)
 241         set_csp(priv->o_bih,sh->codec->outfmt[sh->outfmtidx]);
 242
 243     mp_msg(MSGT_WIN32, MSGL_V, "Input format:\n");
 244     if(verbose>0) print_video_header(sh->bih);
 245     mp_msg(MSGT_WIN32, MSGL_V, "Output format:\n");
 246     if(verbose>0) print_video_header(priv->o_bih);
 247
 248     // set postprocessing level in xvid/divx4 .dll
 249     ICSendMessage(priv->handle, ICM_USER+80, (long)(&divx_quality), 0);
 250
 251     // don't do this palette mess always, it makes div3 dll crashing...
 252     if(sh->codec->outfmt[sh->outfmtidx]==IMGFMT_BGR8){
 253         if(ICDecompressGetPalette(priv->handle, sh->bih, priv->o_bih)){
 254             priv->palette = (unsigned char*)(priv->o_bih+1);
 255             mp_msg(MSGT_WIN32,MSGL_V,"ICDecompressGetPalette OK\n");
 256         } else {
 257             if(sh->bih->biSize>=40+4*4)
 258                 priv->palette = (unsigned char*)(sh->bih+1);
 259         }
 260     }
 261
 262     mp_msg(MSGT_DECVIDEO,MSGL_V,"INFO: Win32 video codec init OK!\n");
 263     return 1;
 264 }
 265
 266 // uninit driver
 267 static void uninit(sh_video_t *sh){
 268     HRESULT ret;
 269     vd_vfw_ctx *priv = sh->context;
 270
 271 #ifdef BUILD_VFWEX
 272     ret = ICDecompressEndEx(priv->handle);
 273 #else
 274     ret = ICDecompressEnd(priv->handle);
 275 #endif
 276     if (ret)
 277     {
 278         mp_msg(MSGT_WIN32, MSGL_WARN, "ICDecompressEnd failed: %d\n", ret);
 279         return;
 280     }
 281
 282     ret = ICClose(priv->handle);
 283     if (ret)
 284     {
 285         mp_msg(MSGT_WIN32, MSGL_WARN, "ICClose failed: %d\n", ret);
 286         return;
 287     }
 288
 289     free(priv->o_bih);
 290     free(priv);
 291 }
 292
 293 // decode a frame
 294 static mp_image_t* decode(sh_video_t *sh,void* data,int len,int flags){
 295     vd_vfw_ctx *priv = sh->context;
 296     mp_image_t* mpi;
 297     HRESULT ret;
 298
 299     if(len<=0) return NULL; // skipped frame
 300
 301     mpi=mpcodecs_get_image(sh,
 302         (sh->codec->outflags[sh->outfmtidx] & CODECS_FLAG_STATIC) ?
 303         MP_IMGTYPE_STATIC : MP_IMGTYPE_TEMP, MP_IMGFLAG_ACCEPT_WIDTH,
 304         sh->disp_w, sh->disp_h);
 305     if(!mpi){   // temporary!
 306         printf("couldn't allocate image for cinepak codec\n");
 307         return NULL;
 308     }
 309
 310     // set stride:  (trick discovered by Andreas Ackermann - thanx!)
 311     sh->bih->biWidth=mpi->width; //mpi->stride[0]/(mpi->bpp/8);
 312     priv->o_bih->biWidth=mpi->width; //mpi->stride[0]/(mpi->bpp/8);
 313
 314     sh->bih->biSizeImage = len;
 315
 316 #ifdef BUILD_VFWEX
 317     ret = ICDecompressEx(priv->handle,
 318 #else
 319     ret = ICDecompress(priv->handle,
 320 #endif
 321           ( (sh->ds->flags&1) ? 0 : ICDECOMPRESS_NOTKEYFRAME ) |
 322           ( ((flags&3)==2 && !(sh->ds->flags&1))?(ICDECOMPRESS_HURRYUP|ICDECOMPRESS_PREROL):0 ),
 323            sh->bih, data, priv->o_bih, (flags&3) ? 0 : mpi->planes[0]);
 324
 325     if ((int)ret){
 326       mp_msg(MSGT_DECVIDEO,MSGL_WARN,"Error decompressing frame, err=%d\n",ret);
 327       return NULL;
 328     }
 329
 330     // export palette:
 331     if(mpi->imgfmt==IMGFMT_RGB8 || mpi->imgfmt==IMGFMT_BGR8){
 332         if (priv->palette)
 333         {
 334             mpi->planes[1] = priv->palette;
 335             mpi->flags |= MP_IMGFLAG_RGB_PALETTE;
 336             mp_dbg(MSGT_DECVIDEO, MSGL_DBG2, "Found and copied palette\n");
 337         }
 338         else
 339             mpi->planes[1]=NULL;
 340     }
 341
 342     return mpi;
 343 }
 344 #endif