libmpcodecs/vd_qtvideo.c

   1 /*
   2  * This file is part of MPlayer.
   3  *
   4  * MPlayer is free software; you can redistribute it and/or modify
   5  * it under the terms of the GNU General Public License as published by
   6  * the Free Software Foundation; either version 2 of the License, or
   7  * (at your option) any later version.
   8  *
   9  * MPlayer is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12  * GNU General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU General Public License along
  15  * with MPlayer; if not, write to the Free Software Foundation, Inc.,
  16  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  17  */
  18
  19 #include <stdio.h>
  20 #include <stdlib.h>
  21 #include <inttypes.h>
  22
  23 #include "config.h"
  24 #include "mp_msg.h"
  25 #include "mpbswap.h"
  26 #include "vd_internal.h"
  27
  28 #ifdef CONFIG_QUICKTIME
  29 #include <QuickTime/ImageCodec.h>
  30 #define dump_ImageDescription(x)
  31 #else
  32 #include "loader/ldt_keeper.h"
  33 #include "loader/qtx/qtxsdk/components.h"
  34 #include "loader/wine/windef.h"
  35 #endif
  36
  37 static const vd_info_t info = {
  38         "Quicktime Video decoder",
  39         "qtvideo",
  40         "A'rpi",
  41         "Sascha Sommer",
  42         "win32"
  43 };
  44
  45 LIBVD_EXTERN(qtvideo)
  46
  47 //static ComponentDescription desc; // for FindNextComponent()
  48 static ComponentInstance ci=NULL; // codec handle
  49 //static CodecInfo cinfo;       // for ImageCodecGetCodecInfo()
  50 //Component prev=NULL;
  51 //ComponentResult cres; //
  52 static CodecCapabilities codeccap; // for decpar
  53 static CodecDecompressParams decpar; // for ImageCodecPreDecompress()
  54 //static ImageSubCodecDecompressCapabilities icap; // for ImageCodecInitialize()
  55 static Rect OutBufferRect;              //the dimensions of our GWorld
  56
  57 static GWorldPtr OutBufferGWorld = NULL;//a GWorld is some kind of description for a drawing environment
  58 static ImageDescriptionHandle framedescHandle;
  59
  60 #ifndef CONFIG_QUICKTIME
  61 HMODULE   WINAPI LoadLibraryA(LPCSTR);
  62 FARPROC   WINAPI GetProcAddress(HMODULE,LPCSTR);
  63 int       WINAPI FreeLibrary(HMODULE);
  64 static    HINSTANCE qtime_qts; // handle to the preloaded quicktime.qts
  65 static    HMODULE handler;
  66 static    Component (*FindNextComponent)(Component prev,ComponentDescription* desc);
  67 static    OSErr (*GetComponentInfo)(Component prev,ComponentDescription* desc,Handle h1,Handle h2,Handle h3);
  68 static    long (*CountComponents)(ComponentDescription* desc);
  69 static    OSErr (*InitializeQTML)(long flags);
  70 static    OSErr (*EnterMovies)(void);
  71 static    ComponentInstance (*OpenComponent)(Component c);
  72 static    ComponentResult (*ImageCodecInitialize)(ComponentInstance ci,
  73                                  ImageSubCodecDecompressCapabilities * cap);
  74 static    ComponentResult (*ImageCodecBeginBand)(ComponentInstance      ci,
  75                                  CodecDecompressParams * params,
  76                                  ImageSubCodecDecompressRecord * drp,
  77                                  long                   flags);
  78 static    ComponentResult (*ImageCodecGetCodecInfo)(ComponentInstance      ci,
  79                                  CodecInfo *            info);
  80 static    ComponentResult (*ImageCodecPreDecompress)(ComponentInstance      ci,
  81                                  CodecDecompressParams * params);
  82 static    ComponentResult (*ImageCodecBandDecompress)(ComponentInstance      ci,
  83                                  CodecDecompressParams * params);
  84 static    PixMapHandle    (*GetGWorldPixMap)(GWorldPtr offscreenGWorld);
  85 static    OSErr           (*QTNewGWorldFromPtr)(GWorldPtr *gw,
  86                                OSType pixelFormat,
  87                                const Rect *boundsRect,
  88                                CTabHandle cTable,
  89                                /*GDHandle*/void* aGDevice, //unused anyway
  90                                GWorldFlags flags,
  91                                void *baseAddr,
  92                                long rowBytes);
  93 static    OSErr           (*NewHandleClear)(Size byteCount);
  94 #endif /* #ifndef CONFIG_QUICKTIME */
  95
  96 // to set/get/query special features/parameters
  97 static int control(sh_video_t *sh,int cmd,void* arg,...){
  98     return CONTROL_UNKNOWN;
  99 }
 100
 101 static int codec_initialized=0;
 102
 103 // init driver
 104 static int init(sh_video_t *sh){
 105 #ifndef CONFIG_QUICKTIME
 106     long result = 1;
 107 #endif
 108     ComponentResult cres;
 109     ComponentDescription desc;
 110     Component prev=NULL;
 111     CodecInfo cinfo;    // for ImageCodecGetCodecInfo()
 112     ImageSubCodecDecompressCapabilities icap; // for ImageCodecInitialize()
 113
 114     codec_initialized = 0;
 115 #ifdef CONFIG_QUICKTIME
 116     EnterMovies();
 117 #else
 118
 119 #ifdef WIN32_LOADER
 120     Setup_LDT_Keeper();
 121 #endif
 122
 123     //preload quicktime.qts to avoid the problems caused by the hardcoded path inside the dll
 124     qtime_qts = LoadLibraryA("QuickTime.qts");
 125     if(!qtime_qts){
 126     mp_msg(MSGT_DECVIDEO,MSGL_ERR,"unable to load QuickTime.qts\n" );
 127     return 0;
 128     }
 129
 130     handler = LoadLibraryA("qtmlClient.dll");
 131     if(!handler){
 132     mp_msg(MSGT_DECVIDEO,MSGL_ERR,"unable to load qtmlClient.dll\n");
 133     return 0;
 134     }
 135
 136     InitializeQTML = (OSErr (*)(long))GetProcAddress(handler, "InitializeQTML");
 137     EnterMovies = (OSErr (*)(void))GetProcAddress(handler, "EnterMovies");
 138     FindNextComponent = (Component (*)(Component,ComponentDescription*))GetProcAddress(handler, "FindNextComponent");
 139     CountComponents = (long (*)(ComponentDescription*))GetProcAddress(handler, "CountComponents");
 140     GetComponentInfo = (OSErr (*)(Component,ComponentDescription*,Handle,Handle,Handle))GetProcAddress(handler, "GetComponentInfo");
 141     OpenComponent = (ComponentInstance (*)(Component))GetProcAddress(handler, "OpenComponent");
 142     ImageCodecInitialize = (ComponentResult (*)(ComponentInstance,ImageSubCodecDecompressCapabilities *))GetProcAddress(handler, "ImageCodecInitialize");
 143     ImageCodecGetCodecInfo = (ComponentResult (*)(ComponentInstance,CodecInfo *))GetProcAddress(handler, "ImageCodecGetCodecInfo");
 144     ImageCodecBeginBand = (ComponentResult (*)(ComponentInstance,CodecDecompressParams *,ImageSubCodecDecompressRecord *,long))GetProcAddress(handler, "ImageCodecBeginBand");
 145     ImageCodecPreDecompress = (ComponentResult (*)(ComponentInstance,CodecDecompressParams *))GetProcAddress(handler, "ImageCodecPreDecompress");
 146     ImageCodecBandDecompress = (ComponentResult (*)(ComponentInstance,CodecDecompressParams *))GetProcAddress(handler, "ImageCodecBandDecompress");
 147     GetGWorldPixMap = (PixMapHandle (*)(GWorldPtr))GetProcAddress(handler, "GetGWorldPixMap");
 148     QTNewGWorldFromPtr = (OSErr(*)(GWorldPtr *,OSType,const Rect *,CTabHandle,void*,GWorldFlags,void *,long))GetProcAddress(handler, "QTNewGWorldFromPtr");
 149     NewHandleClear = (OSErr(*)(Size))GetProcAddress(handler, "NewHandleClear");
 150     //     = GetProcAddress(handler, "");
 151
 152     if(!InitializeQTML || !EnterMovies || !FindNextComponent || !ImageCodecBandDecompress){
 153         mp_msg(MSGT_DECVIDEO,MSGL_ERR,"invalid qtmlClient.dll!\n");
 154         return 0;
 155     }
 156
 157     result=InitializeQTML(6+16);
 158 //    result=InitializeQTML(0);
 159     mp_msg(MSGT_DECVIDEO,MSGL_DBG2,"InitializeQTML returned %li\n",result);
 160 //    result=EnterMovies();
 161 //    printf("EnterMovies->%d\n",result);
 162 #endif /* CONFIG_QUICKTIME */
 163
 164 #if 0
 165     memset(&desc,0,sizeof(desc));
 166     while((prev=FindNextComponent(prev,&desc))){
 167         ComponentDescription desc2;
 168         unsigned char* c1=&desc2.componentType;
 169         unsigned char* c2=&desc2.componentSubType;
 170         memset(&desc2,0,sizeof(desc2));
 171 //      printf("juhee %p (%p)\n",prev,&desc);
 172         GetComponentInfo(prev,&desc2,NULL,NULL,NULL);
 173         mp_msg(MSGT_DECVIDEO,MSGL_DGB2,"DESC: %c%c%c%c/%c%c%c%c [0x%X/0x%X] 0x%X\n",
 174             c1[3],c1[2],c1[1],c1[0],
 175             c2[3],c2[2],c2[1],c2[0],
 176             desc2.componentType,desc2.componentSubType,
 177             desc2.componentFlags);
 178     }
 179 #endif
 180
 181
 182     memset(&desc,0,sizeof(desc));
 183     desc.componentType= (((unsigned char)'i')<<24)|
 184                         (((unsigned char)'m')<<16)|
 185                         (((unsigned char)'d')<<8)|
 186                         (((unsigned char)'c'));
 187 #if 0
 188     desc.componentSubType=
 189                     (((unsigned char)'S'<<24))|
 190                         (((unsigned char)'V')<<16)|
 191                         (((unsigned char)'Q')<<8)|
 192                         (((unsigned char)'3'));
 193 #else
 194     desc.componentSubType = bswap_32(sh->format);
 195 #endif
 196     desc.componentManufacturer=0;
 197     desc.componentFlags=0;
 198     desc.componentFlagsMask=0;
 199
 200     mp_msg(MSGT_DECVIDEO,MSGL_DBG2,"Count = %ld\n",CountComponents(&desc));
 201     prev=FindNextComponent(NULL,&desc);
 202     if(!prev){
 203         mp_msg(MSGT_DECVIDEO,MSGL_ERR,"Cannot find requested component\n");
 204         return 0;
 205     }
 206     mp_msg(MSGT_DECVIDEO,MSGL_DBG2,"Found it! ID = %p\n",prev);
 207
 208     ci=OpenComponent(prev);
 209     mp_msg(MSGT_DECVIDEO,MSGL_DBG2,"ci=%p\n",ci);
 210
 211     memset(&icap,0,sizeof(icap));
 212     cres=ImageCodecInitialize(ci,&icap);
 213     mp_msg(MSGT_DECVIDEO,MSGL_DBG2,"ImageCodecInitialize->%#x  size=%d (%d)\n",cres,icap.recordSize,icap.decompressRecordSize);
 214
 215     memset(&cinfo,0,sizeof(cinfo));
 216     cres=ImageCodecGetCodecInfo(ci,&cinfo);
 217     mp_msg(MSGT_DECVIDEO,MSGL_DBG2,"Flags: compr: 0x%X  decomp: 0x%X format: 0x%X\n",
 218         cinfo.compressFlags, cinfo.decompressFlags, cinfo.formatFlags);
 219     mp_msg(MSGT_DECVIDEO,MSGL_DBG2,"Codec name: %.*s\n",((unsigned char*)&cinfo.typeName)[0],
 220         ((unsigned char*)&cinfo.typeName)+1);
 221
 222     //make a yuy2 gworld
 223     OutBufferRect.top=0;
 224     OutBufferRect.left=0;
 225     OutBufferRect.right=sh->disp_w;
 226     OutBufferRect.bottom=sh->disp_h;
 227
 228     //Fill the imagedescription for our SVQ3 frame
 229     //we can probably get this from Demuxer
 230 #if 0
 231     framedescHandle=(ImageDescriptionHandle)NewHandleClear(sizeof(ImageDescription)+200);
 232     printf("framedescHandle=%p  *p=%p\n",framedescHandle,*framedescHandle);
 233 { FILE* f=fopen("/root/.wine/fake_windows/IDesc","r");
 234   if(!f) printf("filenot found: IDesc\n");
 235   fread(*framedescHandle,sizeof(ImageDescription)+200,1,f);
 236   fclose(f);
 237 }
 238 #else
 239     if(!sh->ImageDesc) sh->ImageDesc=(sh->bih+1); // hack for SVQ3-in-AVI
 240     mp_msg(MSGT_DECVIDEO,MSGL_DBG2,"ImageDescription size: %d\n",((ImageDescription*)(sh->ImageDesc))->idSize);
 241     framedescHandle=(ImageDescriptionHandle)NewHandleClear(((ImageDescription*)(sh->ImageDesc))->idSize);
 242     memcpy(*framedescHandle,sh->ImageDesc,((ImageDescription*)(sh->ImageDesc))->idSize);
 243     dump_ImageDescription(*framedescHandle);
 244 #endif
 245 //Find codecscomponent for video decompression
 246 //    result = FindCodec ('SVQ1',anyCodec,&compressor,&decompressor );
 247 //    printf("FindCodec SVQ1 returned:%i compressor: 0x%X decompressor: 0x%X\n",result,compressor,decompressor);
 248
 249     sh->context = (void *)kYUVSPixelFormat;
 250 #if 1
 251     {
 252         int imgfmt = sh->codec->outfmt[sh->outfmtidx];
 253         int qt_imgfmt;
 254     switch(imgfmt)
 255     {
 256         case IMGFMT_YUY2:
 257             qt_imgfmt = kYUVSPixelFormat;
 258             break;
 259         case IMGFMT_YVU9:
 260             qt_imgfmt = 0x73797639; //kYVU9PixelFormat;
 261             break;
 262         case IMGFMT_YV12:
 263             qt_imgfmt = 0x79343230;
 264             break;
 265         case IMGFMT_UYVY:
 266             qt_imgfmt = kUYVY422PixelFormat;
 267             break;
 268         case IMGFMT_YVYU:
 269             qt_imgfmt = kYVYU422PixelFormat;
 270             imgfmt = IMGFMT_YUY2;
 271             break;
 272         case IMGFMT_RGB16:
 273             qt_imgfmt = k16LE555PixelFormat;
 274             break;
 275         case IMGFMT_BGR24:
 276             qt_imgfmt = k24BGRPixelFormat;
 277             break;
 278         case IMGFMT_BGR32:
 279             qt_imgfmt = k32BGRAPixelFormat;
 280             break;
 281         case IMGFMT_RGB32:
 282             qt_imgfmt = k32RGBAPixelFormat;
 283             break;
 284         default:
 285             mp_msg(MSGT_DECVIDEO,MSGL_ERR,"Unknown requested csp\n");
 286             return 0;
 287     }
 288     mp_msg(MSGT_DECVIDEO,MSGL_DBG2,"imgfmt: %s qt_imgfmt: %.4s\n", vo_format_name(imgfmt), (char *)&qt_imgfmt);
 289     sh->context = (void *)qt_imgfmt;
 290     if(!mpcodecs_config_vo(sh,sh->disp_w,sh->disp_h,imgfmt)) return 0;
 291     }
 292 #else
 293     if(!mpcodecs_config_vo(sh,sh->disp_w,sh->disp_h,IMGFMT_YUY2)) return 0;
 294 #endif
 295
 296     return 1;
 297 }
 298
 299 // uninit driver
 300 static void uninit(sh_video_t *sh){
 301 #ifdef CONFIG_QUICKTIME
 302     ExitMovies();
 303 #endif
 304 }
 305
 306 // decode a frame
 307 static mp_image_t* decode(sh_video_t *sh,void* data,int len,int flags){
 308     long result = 1;
 309     int i;
 310     mp_image_t* mpi;
 311     ComponentResult cres;
 312
 313     if(len<=0) return NULL; // skipped frame
 314
 315     mpi=mpcodecs_get_image(sh, MP_IMGTYPE_STATIC, MP_IMGFLAG_PRESERVE,
 316         sh->disp_w, sh->disp_h);
 317     if(!mpi) return NULL;
 318
 319     decpar.data = (char*)data;
 320     decpar.bufferSize = len;
 321     (**framedescHandle).dataSize=len;
 322
 323 if(!codec_initialized){
 324     result = QTNewGWorldFromPtr(
 325         &OutBufferGWorld,
 326 //        kYUVSPixelFormat, //pixel format of new GWorld == YUY2
 327         (OSType)sh->context,
 328         &OutBufferRect,   //we should benchmark if yvu9 is faster for svq3, too
 329         0,
 330         0,
 331         0,
 332         mpi->planes[0],
 333         mpi->stride[0]);
 334     mp_msg(MSGT_DECVIDEO,MSGL_DBG2,"NewGWorldFromPtr returned:%ld\n",65536-(result&0xffff));
 335 //    if (65536-(result&0xFFFF) != 10000)
 336 //      return NULL;
 337
 338 //    printf("IDesc=%d\n",sizeof(ImageDescription));
 339
 340     decpar.imageDescription = framedescHandle;
 341     decpar.startLine=0;
 342     decpar.stopLine=(**framedescHandle).height;
 343     decpar.frameNumber = 1; //1
 344 //    decpar.conditionFlags=0xFFD; // first
 345 //    decpar.callerFlags=0x2001; // first
 346     decpar.matrixFlags = 0;
 347     decpar.matrixType = 0;
 348     decpar.matrix = 0;
 349     decpar.capabilities=&codeccap;
 350 //    decpar.accuracy = 0x1680000; //codecNormalQuality;
 351     decpar.accuracy = codecNormalQuality;
 352 //    decpar.port = OutBufferGWorld;
 353 //    decpar.preferredOffscreenPixelSize=17207;
 354
 355 //    decpar.sequenceID=malloc(1000);
 356 //    memset(decpar.sequenceID,0,1000);
 357
 358 //    SrcRect.top=17207;
 359 //    SrcRect.left=0;
 360 //    SrcRect.right=0;//image_width;
 361 //    SrcRect.bottom=0;//image_height;
 362
 363 //    decpar.srcRect = SrcRect;
 364     decpar.srcRect = OutBufferRect;
 365
 366     decpar.transferMode = srcCopy;
 367     decpar.dstPixMap = **GetGWorldPixMap( OutBufferGWorld);//destPixmap;
 368
 369     cres=ImageCodecPreDecompress(ci,&decpar);
 370     mp_msg(MSGT_DECVIDEO,MSGL_DBG2,"ImageCodecPreDecompress cres=0x%X\n",cres);
 371
 372     if(decpar.wantedDestinationPixelTypes)
 373     { OSType *p=*(decpar.wantedDestinationPixelTypes);
 374       if(p) while(*p){
 375           mp_msg(MSGT_DECVIDEO,MSGL_DBG2,"supported csp: 0x%08X %.4s\n",*p,(char *)p);
 376           ++p;
 377       }
 378     }
 379
 380
 381 //    decpar.conditionFlags=0x10FFF; // first
 382 //    decpar.preferredOffscreenPixelSize=17207;
 383
 384 //    decpar.conditionFlags=0x10FFD; // first
 385
 386 //      cres=ImageCodecPreDecompress(ci,&decpar);
 387 //    printf("ImageCodecPreDecompress cres=0x%X\n",cres);
 388
 389
 390     codec_initialized=1;
 391 }
 392
 393 #if 0
 394     if(decpar.frameNumber==124){
 395         decpar.frameNumber=1;
 396         cres=ImageCodecPreDecompress(ci,&decpar);
 397         mp_msg(MSGT_DECVIDEO,MSGL_DBG2,"ImageCodecPreDecompress cres=0x%lX\n",cres);
 398     }
 399 #endif
 400
 401     cres=ImageCodecBandDecompress(ci,&decpar);
 402
 403     ++decpar.frameNumber;
 404
 405     if(cres&0xFFFF){
 406         mp_msg(MSGT_DECVIDEO,MSGL_DBG2,"ImageCodecBandDecompress cres=0x%X (-0x%X) %d\n",cres,-cres,cres);
 407         return NULL;
 408     }
 409
 410 //    for(i=0;i<8;i++)
 411 //      printf("img_base[%d]=%p\n",i,((int*)decpar.dstPixMap.baseAddr)[i]);
 412
 413 if((int)sh->context==0x73797639){       // Sorenson 16-bit YUV -> std YVU9
 414
 415     short *src0=(short *)((char*)decpar.dstPixMap.baseAddr+0x20);
 416
 417     for(i=0;i<mpi->h;i++){
 418         int x;
 419         unsigned char* dst=mpi->planes[0]+i*mpi->stride[0];
 420         unsigned short* src=src0+i*((mpi->w+15)&(~15));
 421         for(x=0;x<mpi->w;x++) dst[x]=src[x];
 422     }
 423     src0+=((mpi->w+15)&(~15))*((mpi->h+15)&(~15));
 424     for(i=0;i<mpi->h/4;i++){
 425         int x;
 426         unsigned char* dst=mpi->planes[1]+i*mpi->stride[1];
 427         unsigned short* src=src0+i*(((mpi->w+63)&(~63))/4);
 428         for(x=0;x<mpi->w/4;x++) dst[x]=src[x];
 429         src+=((mpi->w+63)&(~63))/4;
 430     }
 431     src0+=(((mpi->w+63)&(~63))/4)*(((mpi->h+63)&(~63))/4);
 432     for(i=0;i<mpi->h/4;i++){
 433         int x;
 434         unsigned char* dst=mpi->planes[2]+i*mpi->stride[2];
 435         unsigned short* src=src0+i*(((mpi->w+63)&(~63))/4);
 436         for(x=0;x<mpi->w/4;x++) dst[x]=src[x];
 437         src+=((mpi->w+63)&(~63))/4;
 438     }
 439
 440 }
 441
 442
 443     return mpi;
 444 }