libmpcodecs/ve_qtvideo.c

   1 /*qt video encoder using win32 libs
   2   released under gnu gpl
   3   (C)Sascha Sommer                 */
   4
   5 #define MAX_IDSIZE 0x6F
   6
   7 #include <stdio.h>
   8 #include <stdlib.h>
   9 #include <string.h>
  10
  11 #include "config.h"
  12 #include "mp_msg.h"
  13 #include "libavutil/common.h"
  14 #include "mpbswap.h"
  15
  16 #ifdef WIN32_LOADER
  17 #include "loader/ldt_keeper.h"
  18 #endif
  19
  20 #include "loader/qtx/qtxsdk/components.h"
  21 #include "loader/wine/windef.h"
  22
  23 #include "codec-cfg.h"
  24 #include "stream/stream.h"
  25 #include "libmpdemux/demuxer.h"
  26 #include "libmpdemux/stheader.h"
  27
  28 #include "stream/stream.h"
  29 #include "libmpdemux/muxer.h"
  30
  31 #include "img_format.h"
  32 #include "mp_image.h"
  33 #include "vf.h"
  34
  35 HMODULE   WINAPI LoadLibraryA(LPCSTR);
  36 FARPROC   WINAPI GetProcAddress(HMODULE,LPCSTR);
  37 int       WINAPI FreeLibrary(HMODULE);
  38 static HINSTANCE qtime_qts; //handle to preloaded quicktime.qts
  39 static HMODULE handler;
  40
  41 static OSErr        (*FindCodec)(CodecType              cType,
  42                         CodecComponent         specCodec,
  43                         CompressorComponent *  compressor,
  44                         DecompressorComponent * decompressor);
  45 static OSErr        (*InitializeQTML)(long flags);
  46 static PixMapHandle (*GetGWorldPixMap)(GWorldPtr offscreenGWorld);
  47 static OSErr        (*QTNewGWorldFromPtr)(GWorldPtr *gw,
  48                                     OSType pixelFormat,
  49                                     const Rect *boundsRect,
  50                                     CTabHandle cTable,
  51                         /*GDHandle*/void* aGDevice, /*unused anyway*/
  52                         GWorldFlags flags,
  53                         void *baseAddr,
  54                         long rowBytes);
  55 static OSErr        (*NewHandleClear)(Size byteCount);
  56 static OSErr        (*CompressSequenceBegin) (
  57      ImageSequence             *seqID,
  58      PixMapHandle              src,
  59      PixMapHandle              prev,
  60      const Rect                *srcRect,
  61      const Rect                *prevRect,
  62      short                     colorDepth,
  63      CodecType                 cType,
  64      CompressorComponent       codec,
  65      CodecQ                    spatialQuality,
  66      CodecQ                    temporalQuality,
  67      long                      keyFrameRate,
  68      CTabHandle                ctable,
  69      CodecFlags                flags,
  70      ImageDescriptionHandle    desc );
  71
  72 static OSErr (*CompressSequenceFrame) (
  73      ImageSequence                 seqID,
  74      PixMapHandle                  src,
  75      const Rect                    *srcRect,
  76      CodecFlags                    flags,
  77      Ptr                           data,
  78      long                          *dataSize,
  79      UInt8                         *similarity,
  80      ICMCompletionProcRecordPtr    asyncCompletionProc );
  81
  82 static    OSErr (*GetMaxCompressionSize)(PixMapHandle src,
  83      const Rect *srcRect,
  84      short colorDepth,
  85      CodecQ quality,
  86      CodecType cType,
  87      CompressorComponent codec,
  88      long *size );
  89 static    OSErr (*CDSequenceEnd)( ImageSequence seqID );
  90 static    Component (*FindNextComponent)(Component prev,ComponentDescription* desc);
  91 static    long (*CountComponents)(ComponentDescription* desc);
  92 static    OSErr (*GetComponentInfo)(Component prev,ComponentDescription* desc,Handle h1,Handle h2,Handle h3);
  93
  94
  95
  96 //static int format=mmioFOURCC('S','V','Q','1');
  97 static int format=mmioFOURCC('S','V','Q','3');
  98
  99
 100
 101 //static void *frame_in;    //input frame
 102 static void *frame_prev;  //previous frame
 103 static void *frame_comp;  //compressed frame
 104 static GWorldPtr frame_GWorld_in = NULL;//a GWorld is some kind of description for a drawing environment
 105 static GWorldPtr frame_GWorld_prev = NULL;
 106 static Rect FrameRect;
 107
 108 static CompressorComponent compressor;
 109 static ImageDescriptionHandle desc;
 110 static ImageSequence seq;
 111
 112
 113
 114
 115
 116 struct vf_priv_s {
 117     muxer_stream_t* mux;
 118     //dv_encoder_t* enc;
 119
 120 };
 121 #define mux_v (vf->priv->mux)
 122
 123 //===========================================================================//
 124
 125 static int config(struct vf_instance_s* vf,
 126         int width, int height, int d_width, int d_height,
 127         unsigned int flags, unsigned int outfmt){
 128 //    OSErr cres;
 129     ComponentDescription cdesc;
 130     mux_v->bih->biWidth=width;
 131     mux_v->bih->biHeight=height;
 132     mux_v->bih->biSizeImage=width*height*2;
 133     mux_v->aspect = (float)d_width/d_height;
 134
 135
 136
 137     memset(&cdesc,0,sizeof(cdesc));
 138     cdesc.componentType= (((unsigned char)'i')<<24)|
 139                         (((unsigned char)'m')<<16)|
 140                         (((unsigned char)'c')<<8)|
 141                         (((unsigned char)'o'));
 142
 143     cdesc.componentSubType=bswap_32(format);
 144     cdesc.componentManufacturer=0;
 145     cdesc.componentFlags=0;
 146     cdesc.componentFlagsMask=0;
 147
 148
 149     mp_msg(MSGT_MENCODER,MSGL_DBG2,"Count = %ld\n",CountComponents(&cdesc));
 150     compressor=FindNextComponent(NULL,&cdesc);
 151     if(!compressor){
 152         mp_msg(MSGT_MENCODER,MSGL_ERR,"Cannot find requested component\n");
 153         return(0);
 154     }
 155     mp_msg(MSGT_MENCODER,MSGL_DBG2,"Found it! ID = %p\n",compressor);
 156
 157 //      cres= FindCodec (fourcc,anyCodec,&compressor,&decompressor );
 158 //      printf("FindCodec returned:%i compressor: 0x%X decompressor: 0x%X\n",cres&0xFFFF,compressor,decompressor);
 159
 160     return 1;
 161 }
 162
 163 static int control(struct vf_instance_s* vf, int request, void* data){
 164
 165     return CONTROL_UNKNOWN;
 166 }
 167
 168 static int query_format(struct vf_instance_s* vf, unsigned int fmt){
 169     if(fmt==IMGFMT_YUY2) return VFCAP_CSP_SUPPORTED | VFCAP_CSP_SUPPORTED_BY_HW;
 170     return 0;
 171 }
 172
 173 static int codec_inited = 0;
 174
 175 static int put_image(struct vf_instance_s* vf, mp_image_t *mpi, double pts){
 176
 177     OSErr cres;
 178     long framesizemax;
 179     UInt8 similarity=0;
 180     long compressedsize;
 181     OSType in_format=kYUVSPixelFormat;
 182     int width = mpi->width;
 183     int height = mpi->height;
 184     int stride = width*2;
 185 if(!codec_inited){
 186     FrameRect.top=0;
 187     FrameRect.left=0;
 188     FrameRect.right=width;
 189     FrameRect.bottom=height;
 190     cres = QTNewGWorldFromPtr(
 191         &frame_GWorld_in,
 192         in_format,
 193         &FrameRect,
 194         0,
 195         0,
 196         0,
 197         mpi->planes[0],
 198         stride);
 199     mp_msg(MSGT_MENCODER,MSGL_DBG2,"NewGWorldFromPtr returned:%i\n",cres&0xFFFF);
 200     //dunno what todo about this
 201     frame_prev = malloc(stride * height);
 202     cres = QTNewGWorldFromPtr(
 203         &frame_GWorld_prev,
 204         in_format,
 205         &FrameRect,
 206         0,
 207         0,
 208         0,
 209         frame_prev,
 210         stride);
 211     mp_msg(MSGT_MENCODER,MSGL_DBG2,"height:%i width:%i stride:%i\n",height,width,stride);
 212     mp_msg(MSGT_MENCODER,MSGL_DBG2,"NewGWorldFromPtr returned:%i\n",cres&0xFFFF);
 213     cres=  GetMaxCompressionSize (
 214        GetGWorldPixMap(frame_GWorld_in),
 215        &FrameRect,
 216        24,
 217        codecNormalQuality,
 218        bswap_32(format),
 219        compressor,
 220        &framesizemax );
 221     mp_msg(MSGT_MENCODER,MSGL_DBG2,"GetMaxCompressionSize returned:%i : MaxSize:%li\n",cres&0xFFFF,framesizemax);
 222     frame_comp=malloc(framesizemax);
 223
 224     desc = (ImageDescriptionHandle)NewHandleClear(MAX_IDSIZE); //memory where the desc will be stored
 225     (*desc)->idSize=MAX_IDSIZE;
 226
 227     cres= CompressSequenceBegin (
 228        &seq,
 229        GetGWorldPixMap( frame_GWorld_in),
 230        GetGWorldPixMap( frame_GWorld_prev),
 231        &FrameRect,
 232        &FrameRect,
 233        24, // color depth
 234        bswap_32(format), // fourcc
 235        compressor,  // codec component
 236        codecNormalQuality, //codecNormalQuality,
 237        codecMaxQuality, //codecNormalQuality,
 238        10*30, // keyframe rate
 239        0,
 240        0,
 241        desc);
 242     mp_msg(MSGT_MENCODER,MSGL_DBG2,"CompressSequenceBegin returned:%i\n",cres&0xFFFF);
 243     mp_msg(MSGT_MENCODER,MSGL_DBG2,"Sequence ID:%i\n",seq);
 244
 245     dump_ImageDescription(*desc);
 246     codec_inited++;
 247 }
 248     cres = CompressSequenceFrame (
 249         seq,
 250         GetGWorldPixMap(frame_GWorld_in),
 251         &FrameRect,
 252         0,
 253         (char*)mux_v->buffer,
 254         &compressedsize,
 255         &similarity,
 256         0);
 257
 258     if(cres&0xFFFF)mp_msg(MSGT_MENCODER,MSGL_DBG2,"CompressSequenceFrame returned:%i\n",cres&0xFFFF);
 259 #if 0
 260     printf("Size %i->%i   \n",stride*height,compressedsize);
 261     printf("Ratio: %i:1\n",(stride*height)/compressedsize);
 262 #endif
 263     muxer_write_chunk(mux_v, compressedsize , similarity?0:0x10, MP_NOPTS_VALUE, MP_NOPTS_VALUE);
 264
 265     if(((*desc)->idSize)>MAX_IDSIZE){
 266         mp_msg(MSGT_MENCODER,MSGL_ERR,"FATAL! idSize=%d too big, increase MAX_IDSIZE in ve_qtvideo.c!\n",((*desc)->idSize));
 267     } else {
 268         // according to QT docs, imagedescription may be changed while encoding
 269         // a frame (even its size may (and does!) change!)
 270         memcpy(mux_v->bih+1,*desc,(*desc)->idSize);
 271     }
 272
 273     return 1;
 274 }
 275
 276 //===========================================================================//
 277
 278 static int vf_open(vf_instance_t *vf, char* args){
 279     OSErr cres = 1;
 280     vf->config=config;
 281     vf->default_caps=VFCAP_CONSTANT;
 282     vf->control=control;
 283     vf->query_format=query_format;
 284     vf->put_image=put_image;
 285     vf->priv=malloc(sizeof(struct vf_priv_s));
 286     memset(vf->priv,0,sizeof(struct vf_priv_s));
 287     vf->priv->mux=(muxer_stream_t*)args;
 288
 289     mux_v->bih=calloc(1, sizeof(BITMAPINFOHEADER)+MAX_IDSIZE);
 290     mux_v->bih->biSize=sizeof(BITMAPINFOHEADER)+MAX_IDSIZE;
 291     mux_v->bih->biWidth=0;
 292     mux_v->bih->biHeight=0;
 293     mux_v->bih->biCompression=format;
 294     mux_v->bih->biPlanes=1;
 295     mux_v->bih->biBitCount=24;
 296
 297
 298 #ifdef WIN32_LOADER
 299     Setup_LDT_Keeper();
 300 #endif
 301     //preload quicktime.qts to avoid the problems caused by the hardcoded path inside the dll
 302     qtime_qts = LoadLibraryA("QuickTime.qts");
 303     if(!qtime_qts){
 304         mp_msg(MSGT_MENCODER,MSGL_ERR,"unable to load QuickTime.qts\n" );
 305         return 0;
 306     }
 307
 308     handler = LoadLibraryA("qtmlClient.dll");
 309     if(!handler){
 310         mp_msg(MSGT_MENCODER,MSGL_ERR,"unable to load qtmlClient.dll\n");
 311         return 0;
 312     }
 313     InitializeQTML = (OSErr (*)(long))GetProcAddress(handler, "InitializeQTML");
 314     GetGWorldPixMap = (PixMapHandle (*)(GWorldPtr))GetProcAddress(handler, "GetGWorldPixMap");
 315     QTNewGWorldFromPtr = (OSErr(*)(GWorldPtr *,OSType,const Rect *,CTabHandle,void*,GWorldFlags,void *,long))GetProcAddress(handler, "QTNewGWorldFromPtr");
 316     NewHandleClear = (OSErr(*)(Size))GetProcAddress(handler, "NewHandleClear");
 317     FindCodec = (OSErr (*)(CodecType,CodecComponent,CompressorComponent *,DecompressorComponent *))GetProcAddress(handler,"FindCodec");
 318     CompressSequenceBegin = (OSErr(*)(ImageSequence *,PixMapHandle,PixMapHandle,const Rect *,const Rect *,short,CodecType,CompressorComponent,CodecQ,CodecQ,long,CTabHandle,CodecFlags,ImageDescriptionHandle))GetProcAddress(handler,"CompressSequenceBegin");
 319     CompressSequenceFrame = (OSErr(*)(ImageSequence,PixMapHandle,const Rect *,CodecFlags,Ptr,long *,UInt8 *,ICMCompletionProcRecordPtr))GetProcAddress(handler,"CompressSequenceFrame");
 320     GetMaxCompressionSize = (OSErr(*)(PixMapHandle,const Rect *,short,CodecQ,CodecType,CompressorComponent,long *))GetProcAddress(handler,"GetMaxCompressionSize");
 321     CDSequenceEnd = (OSErr (*)(ImageSequence))GetProcAddress(handler,"CDSequenceEnd");
 322     FindNextComponent = (Component (*)(Component,ComponentDescription*))GetProcAddress(handler, "FindNextComponent");
 323     CountComponents = (long (*)(ComponentDescription*))GetProcAddress(handler, "CountComponents");
 324     GetComponentInfo = (OSErr (*)(Component,ComponentDescription*,Handle,Handle,Handle))GetProcAddress(handler, "GetComponentInfo");
 325     if(!InitializeQTML  ||!CompressSequenceBegin){
 326         mp_msg(MSGT_MENCODER,MSGL_ERR,"invalid qt DLL!\n");
 327         return 0;
 328     }
 329     //printf("%i,%i,%i\n",mmioFOURCC('S','V','Q','1'),'SVQ1',bswap_32(mmioFOURCC('S','V','Q','1')));
 330     cres=InitializeQTML(6+16);
 331     mp_msg(MSGT_MENCODER,MSGL_DBG2,"InitializeQTML returned %i\n",cres);
 332     return 1;
 333 }
 334
 335 vf_info_t ve_info_qtvideo = {
 336     "Quicktime video encoder using win32 DLLs",
 337     "qtvideo",
 338     "Sascha Sommer",
 339     "for internal use by mencoder",
 340     vf_open
 341 };
 342
 343 //===========================================================================//