Mercurial > mplayer.hg
diff libmpcodecs/ve_qtvideo.c @ 8471:0b7839c27be9
Sorenson 1/3 encoding just for fun :)
ve_qtvideo code by Sascha Sommer
SVQ3 support hack by /me
author | arpi |
---|---|
date | Mon, 16 Dec 2002 01:49:39 +0000 |
parents | |
children | 2b955a07fdcf |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/libmpcodecs/ve_qtvideo.c Mon Dec 16 01:49:39 2002 +0000 @@ -0,0 +1,325 @@ +/*qt video encoder using win32 libs + released under gnu gpl + (C)Sascha Sommer */ + +#define MAX_IDSIZE 0x6F + +#include <stdio.h> +#include <stdlib.h> +#include <string.h> + +#include "../config.h" +#include "../mp_msg.h" +#include "../bswap.h" + +#ifdef USE_QTX_CODECS +#include "../loader/qtx/qtxsdk/components.h" +#include "wine/windef.h" + +#include "codec-cfg.h" +#include "stream.h" +#include "demuxer.h" +#include "stheader.h" + +#include "aviwrite.h" + +#include "img_format.h" +#include "mp_image.h" +#include "vf.h" + +HMODULE WINAPI LoadLibraryA(LPCSTR); +FARPROC WINAPI GetProcAddress(HMODULE,LPCSTR); +int WINAPI FreeLibrary(HMODULE); +static HMODULE handler; + +static OSErr (*FindCodec)(CodecType cType, + CodecComponent specCodec, + CompressorComponent * compressor, + DecompressorComponent * decompressor); +static OSErr (*InitializeQTML)(long flags); +static PixMapHandle (*GetGWorldPixMap)(GWorldPtr offscreenGWorld); +static OSErr (*QTNewGWorldFromPtr)(GWorldPtr *gw, + OSType pixelFormat, + const Rect *boundsRect, + CTabHandle cTable, + /*GDHandle*/void* aGDevice, /*unused anyway*/ + GWorldFlags flags, + void *baseAddr, + long rowBytes); +static OSErr (*NewHandleClear)(Size byteCount); +static OSErr (*CompressSequenceBegin) ( + ImageSequence *seqID, + PixMapHandle src, + PixMapHandle prev, + const Rect *srcRect, + const Rect *prevRect, + short colorDepth, + CodecType cType, + CompressorComponent codec, + CodecQ spatialQuality, + CodecQ temporalQuality, + long keyFrameRate, + CTabHandle ctable, + CodecFlags flags, + ImageDescriptionHandle desc ); + +static OSErr (*CompressSequenceFrame) ( + ImageSequence seqID, + PixMapHandle src, + const Rect *srcRect, + CodecFlags flags, + Ptr data, + long *dataSize, + UInt8 *similarity, + ICMCompletionProcRecordPtr asyncCompletionProc ); + +static OSErr (*GetMaxCompressionSize)(PixMapHandle src, + const Rect *srcRect, + short colorDepth, + CodecQ quality, + CodecType cType, + CompressorComponent codec, + long *size ); +static OSErr (*CDSequenceEnd)( ImageSequence seqID ); +static Component (*FindNextComponent)(Component prev,ComponentDescription* desc); +static long (*CountComponents)(ComponentDescription* desc); +static OSErr (*GetComponentInfo)(Component prev,ComponentDescription* desc,Handle h1,Handle h2,Handle h3); + + +extern void mencoder_write_chunk(aviwrite_stream_t *s,int len,unsigned int flags); + + +//static int format=mmioFOURCC('S','V','Q','1'); +static int format=mmioFOURCC('S','V','Q','3'); + + + +//static void *frame_in; //input frame +static void *frame_prev; //previous frame +static void *frame_comp; //compressed frame +static GWorldPtr frame_GWorld_in = NULL;//a GWorld is some kind of description for a drawing environment +static GWorldPtr frame_GWorld_prev = NULL; +static Rect FrameRect; + +static CompressorComponent compressor; +static DecompressorComponent decompressor; +static ImageDescriptionHandle desc; +static ImageSequence seq; + + + + + +struct vf_priv_s { + aviwrite_stream_t* mux; + //dv_encoder_t* enc; + +}; +#define mux_v (vf->priv->mux) + +//===========================================================================// + +static int config(struct vf_instance_s* vf, + int width, int height, int d_width, int d_height, + unsigned int flags, unsigned int outfmt){ + OSErr cres; + ComponentDescription cdesc; + mux_v->bih->biWidth=width; + mux_v->bih->biHeight=height; + mux_v->bih->biSizeImage=width*height*2; + + + + memset(&desc,0,sizeof(cdesc)); + cdesc.componentType= (((unsigned char)'i')<<24)| + (((unsigned char)'m')<<16)| + (((unsigned char)'c')<<8)| + (((unsigned char)'o')); + + cdesc.componentSubType=bswap_32(format); + cdesc.componentManufacturer=0; + cdesc.componentFlags=0; + cdesc.componentFlagsMask=0; + + + printf("Count = %d\n",CountComponents(&cdesc)); + compressor=FindNextComponent(NULL,&cdesc); + if(!compressor){ + printf("Cannot find requested component\n"); + return(0); + } + printf("Found it! ID = 0x%X\n",compressor); + +// cres= FindCodec (fourcc,anyCodec,&compressor,&decompressor ); +// printf("FindCodec returned:%i compressor: 0x%X decompressor: 0x%X\n",cres&0xFFFF,compressor,decompressor); + + return 1; +} + +static int control(struct vf_instance_s* vf, int request, void* data){ + + return CONTROL_UNKNOWN; +} + +static int query_format(struct vf_instance_s* vf, unsigned int fmt){ + if(fmt==IMGFMT_YUY2) return 3; + return 0; +} + +static int codec_inited = 0; + +static int put_image(struct vf_instance_s* vf, mp_image_t *mpi){ + + OSErr cres; + long framesizemax; + UInt8 similarity; + long compressedsize; + int in_format=kYUVSPixelFormat; + int width = mpi->width; + int height = mpi->height; + int stride = width*2; +if(!codec_inited){ + FrameRect.top=0; + FrameRect.left=0; + FrameRect.right=width; + FrameRect.bottom=height; + cres = QTNewGWorldFromPtr( + &frame_GWorld_in, + in_format, + &FrameRect, + 0, + 0, + 0, + mpi->planes[0], + stride); + printf("NewGWorldFromPtr returned:%i\n",cres&0xFFFF); + //dunno what todo about this + frame_prev = malloc(stride * height); + cres = QTNewGWorldFromPtr( + &frame_GWorld_prev, + in_format, + &FrameRect, + 0, + 0, + 0, + frame_prev, + stride); + printf("height:%i width:%i stride:%i\n",height,width,stride); + printf("NewGWorldFromPtr returned:%i\n",cres&0xFFFF); + cres= GetMaxCompressionSize ( + GetGWorldPixMap(frame_GWorld_in), + &FrameRect, + 24, + codecNormalQuality, + bswap_32(format), + compressor, + &framesizemax ); + printf("GetMaxCompressionSize returned:%i : MaxSize:%i\n",cres&0xFFFF,framesizemax); + frame_comp=malloc(framesizemax); + desc = (ImageDescriptionHandle)NewHandleClear(MAX_IDSIZE); //memory where the desc will be stored + (*desc)->idSize=MAX_IDSIZE; + + cres= CompressSequenceBegin ( + &seq, + GetGWorldPixMap( frame_GWorld_in), + GetGWorldPixMap( frame_GWorld_prev), + &FrameRect, + &FrameRect, + 24, // color depth + bswap_32(format), // fourcc + compressor, // codec component + codecNormalQuality, //codecNormalQuality, + codecMaxQuality, //codecNormalQuality, + 10*25, // keyframe rate + 0, + 0, + desc); + printf("CompressSequenceBegin returned:%i\n",cres&0xFFFF); + printf("Sequence ID:%i\n",seq); + + dump_ImageDescription(*desc); + codec_inited++; +} + cres = CompressSequenceFrame ( + seq, + GetGWorldPixMap(frame_GWorld_in), + &FrameRect, + 0, + (char*)mux_v->buffer, + &compressedsize, + &similarity, + 0); + + if(cres&0xFFFF)printf("CompressSequenceFrame returned:%i\n",cres&0xFFFF); + printf("Size %i->%i \n",stride*height,compressedsize); +#if 0 + printf("Ratio: %i:1\n",(stride*height)/compressedsize); +#endif + mencoder_write_chunk(mux_v, compressedsize , 0x10); + + if(((*desc)->idSize)>MAX_IDSIZE){ + printf("FATAL! idSize=%d too big, increase MAX_IDSIZE in ve_qtvideo.c!\n",((*desc)->idSize)); + } else { + // according to QT docs, imagedescription may be changed while encoding + // a frame (even its size may (and does!) change!) + memcpy(mux_v->bih+1,*desc,(*desc)->idSize); + } + + return 1; +} + +//===========================================================================// + +static int vf_open(vf_instance_t *vf, char* args){ + OSErr cres = 1; + vf->config=config; + vf->control=control; + vf->query_format=query_format; + vf->put_image=put_image; + vf->priv=malloc(sizeof(struct vf_priv_s)); + memset(vf->priv,0,sizeof(struct vf_priv_s)); + vf->priv->mux=(aviwrite_stream_t*)args; + + mux_v->bih=malloc(sizeof(BITMAPINFOHEADER)+MAX_IDSIZE); + mux_v->bih->biSize=sizeof(BITMAPINFOHEADER)+MAX_IDSIZE; + mux_v->bih->biWidth=0; + mux_v->bih->biHeight=0; + mux_v->bih->biCompression=format; + mux_v->bih->biPlanes=1; + mux_v->bih->biBitCount=24; + + + Setup_LDT_Keeper(); + handler = LoadLibraryA("qtmlClient.dll"); + InitializeQTML = GetProcAddress(handler, "InitializeQTML"); + GetGWorldPixMap = GetProcAddress(handler, "GetGWorldPixMap"); + QTNewGWorldFromPtr = GetProcAddress(handler, "QTNewGWorldFromPtr"); + NewHandleClear = GetProcAddress(handler, "NewHandleClear"); + FindCodec = GetProcAddress(handler,"FindCodec"); + CompressSequenceBegin = GetProcAddress(handler,"CompressSequenceBegin"); + CompressSequenceFrame = GetProcAddress(handler,"CompressSequenceFrame"); + GetMaxCompressionSize = GetProcAddress(handler,"GetMaxCompressionSize"); + CDSequenceEnd = GetProcAddress(handler,"CDSequenceEnd"); + FindNextComponent = GetProcAddress(handler, "FindNextComponent"); + CountComponents = GetProcAddress(handler, "CountComponents"); + GetComponentInfo = GetProcAddress(handler, "GetComponentInfo"); + if(!InitializeQTML ||!CompressSequenceBegin){ + printf("invalid qt DLL!\n"); + return 0; + } + //printf("%i,%i,%i\n",mmioFOURCC('S','V','Q','1'),'SVQ1',bswap_32(mmioFOURCC('S','V','Q','1'))); + cres=InitializeQTML(6+16); + printf("InitializeQTML returned %i\n",cres); + return 1; +} + +vf_info_t ve_info_qtvideo = { + "Quicktime video encoder using win32 DLLs", + "qtvideo", + "Sascha Sommer", + "for internal use by mencoder", + vf_open +}; + +//===========================================================================// +#endif