/* * Modified for use with MPlayer, detailed changelog at * http://svn.mplayerhq.hu/mplayer/trunk/ * $Id$ */#include "config.h"#if (C_HAS_DIRECTSHOW)#include "DS_Filter.h"#include "driver.h"#include "com.h"#include "win32.h"typedef long STDCALL (*GETCLASS) (const GUID*, const GUID*, void**);#ifndef WIN32_LOADER#if (!C_HAVE_WXGUI)const GUID IID_IUnknown ={    0x00000000, 0x0000, 0x0000,    {0xc0, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x46}};const GUID IID_IClassFactory ={    0x00000001, 0x0000, 0x0000,    {0xc0, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x46}};#endifHRESULT STDCALL CoInitialize(LPVOID pvReserved);void STDCALL CoUninitialize(void);#endifstatic void DS_Filter_Start(DS_Filter* This){    HRESULT hr;    //LOG_MSG("DS_Filter_Start(%p)", This);    hr = This->m_pFilter->vt->Run(This->m_pFilter, (REFERENCE_TIME)0);    if (hr != 0)    {	LOG_MSG("WARNING: m_Filter->Run() failed, error code %x", (int)hr);    }}static void DS_Filter_Stop(DS_Filter* This){    if (This->m_pAll)    {	//LOG_MSG("DS_Filter_Stop(%p)", This);	This->m_pFilter->vt->Stop(This->m_pFilter); // causes weird crash ??? FIXME	This->m_pAll->vt->Release((IUnknown*)This->m_pAll);	This->m_pAll = 0;    }}void DS_Filter_Destroy(DS_Filter* This){    This->Stop(This);    if (This->m_pOurInput)	This->m_pOurInput->vt->Release((IUnknown*)This->m_pOurInput);    if (This->m_pInputPin)	This->m_pInputPin->vt->Disconnect(This->m_pInputPin);    if (This->m_pOutputPin)	This->m_pOutputPin->vt->Disconnect(This->m_pOutputPin);    if (This->m_pFilter)	This->m_pFilter->vt->Release((IUnknown*)This->m_pFilter);    if (This->m_pOutputPin)	This->m_pOutputPin->vt->Release((IUnknown*)This->m_pOutputPin);    if (This->m_pInputPin)	This->m_pInputPin->vt->Release((IUnknown*)This->m_pInputPin);    if (This->m_pImp)	This->m_pImp->vt->Release((IUnknown*)This->m_pImp);    if (This->m_pOurOutput)	This->m_pOurOutput->vt->Release((IUnknown*)This->m_pOurOutput);    if (This->m_pParentFilter)	This->m_pParentFilter->vt->Release((IUnknown*)This->m_pParentFilter);    if (This->m_pSrcFilter)	This->m_pSrcFilter->vt->Release((IUnknown*)This->m_pSrcFilter);    // FIXME - we are still leaving few things allocated!    if (This->m_iHandle)	FreeLibrary((unsigned)This->m_iHandle);    free(This);#ifdef WIN32_LOADER    CodecRelease();#else    CoUninitialize();#endif}static HRESULT STDCALL DS_Filter_CopySample(void* pUserData, IMediaSample* pSample) {    int len;    LONGLONG time_start;    SampleProcUserData* pData=(SampleProcUserData*)pUserData;    len = pSample->vt->GetActualDataLength(pSample);    if (len == 0)	len = pSample->vt->GetSize(pSample); //for iv50    // Retrieve PTS stamp for this sample    pSample->vt->GetTime(pSample, &time_start, NULL);    pData->size = len;    pData->frame_pts = time_start;    LOG_MSG("CopySample returned size %d, pts %u", pData->size, pData->frame_pts);    return 0;}DS_Filter* DS_FilterCreate(const char* dllname, const GUID* id,			   AM_MEDIA_TYPE* in_fmt,			   AM_MEDIA_TYPE* out_fmt, SampleProcUserData* pUserData){    int init = 0;    const char* em = NULL;    MemAllocator* tempAll;    ALLOCATOR_PROPERTIES props,props1;    HRESULT result;    DS_Filter* This = (DS_Filter*) malloc(sizeof(DS_Filter));    if (!This)	return NULL;#ifdef WIN32_LOADER    CodecAlloc();#else    CoInitialize(0L);#endif    /*	tempAll is not used anywhere.	MemAllocatorCreate() is called to ensure that RegisterComObject for IMemoryAllocator	will be	called before possible call	to CoCreateInstance(...,&IID_IMemoryAllocator,...) from binary codec.    */    tempAll=MemAllocatorCreate();    This->m_pFilter = NULL;    This->m_pInputPin = NULL;    This->m_pOutputPin = NULL;    This->m_pSrcFilter = NULL;    This->m_pParentFilter = NULL;    This->m_pOurInput = NULL;    This->m_pOurOutput = NULL;    This->m_pAll = NULL;    This->m_pImp = NULL;    This->Start = DS_Filter_Start;    This->Stop = DS_Filter_Stop;    for (;;)    {	GETCLASS func;	struct IClassFactory* factory = NULL;	struct IUnknown* object = NULL;	IEnumPins* enum_pins = 0;	IPin* array[256];	ULONG fetched;        unsigned int i;	This->m_iHandle = LoadLibraryA(dllname);	if (!This->m_iHandle)	{	    em = "could not open DirectShow DLL";	    break;	}	func = (GETCLASS)GetProcAddress((unsigned)This->m_iHandle, "DllGetClassObject");	if (!func)	{	    em = "illegal or corrupt DirectShow DLL";	    break;	}	result = func(id, &IID_IClassFactory, (void**)&factory);	if (result || !factory)	{	    em = "no such class object";	    break;	}	result = factory->vt->CreateInstance(factory, 0, &IID_IUnknown, (void**)&object);	factory->vt->Release((IUnknown*)factory);	if (result || !object)	{	    em = "class factory failure";	    break;	}	result = object->vt->QueryInterface(object, &IID_IBaseFilter, (void**)&This->m_pFilter);	object->vt->Release((IUnknown*)object);	if (result || !This->m_pFilter)	{	    em = "object does not provide IBaseFilter interface";            break;	}	// enumerate pins	result = This->m_pFilter->vt->EnumPins(This->m_pFilter, &enum_pins);	if (result || !enum_pins)	{	    em = "could not enumerate pins";            break;	}	enum_pins->vt->Reset(enum_pins);	result = enum_pins->vt->Next(enum_pins, (ULONG)256, (IPin**)array, &fetched);	LOG_MSG("Pins enumeration returned %ld pins, error is %x", fetched, (int)result);	for (i = 0; i < fetched; i++)	{	    int direction = -1;	    array[i]->vt->QueryDirection(array[i], (PIN_DIRECTION*)&direction);	    if (!This->m_pInputPin && direction == 0)	    {		This->m_pInputPin = array[i];		This->m_pInputPin->vt->AddRef((IUnknown*)This->m_pInputPin);	    }	    if (!This->m_pOutputPin && direction == 1)	    {		This->m_pOutputPin = array[i];		This->m_pOutputPin->vt->AddRef((IUnknown*)This->m_pOutputPin);	    }	    array[i]->vt->Release((IUnknown*)(array[i]));	}	if (!This->m_pInputPin)	{	    em = "could not find input pin";            break;	}	if (!This->m_pOutputPin)	{	    em = "could not find output pin";            break;	}	result = This->m_pInputPin->vt->QueryInterface((IUnknown*)This->m_pInputPin,						       &IID_IMemInputPin,						       (void**)&This->m_pImp);	if (result)	{	    em = "could not get IMemInputPin interface";	    break;	}	This->m_pOurType = in_fmt;	This->m_pDestType = out_fmt;        result = This->m_pInputPin->vt->QueryAccept(This->m_pInputPin, This->m_pOurType);	if (result)	{	    em = "source format is not accepted";            break;	}	This->m_pParentFilter = CBaseFilter2Create();	This->m_pSrcFilter = CBaseFilterCreate(This->m_pOurType, This->m_pParentFilter);	This->m_pOurInput = This->m_pSrcFilter->GetPin(This->m_pSrcFilter);	This->m_pOurInput->vt->AddRef((IUnknown*)This->m_pOurInput);	result = This->m_pInputPin->vt->ReceiveConnection(This->m_pInputPin,							  This->m_pOurInput,							  This->m_pOurType);	if (result)	{	    em = "could not connect to input pin";            break;	}	result = This->m_pImp->vt->GetAllocator(This->m_pImp, &This->m_pAll);	if (result || !This->m_pAll)	{	    em="error getting IMemAllocator interface";            break;	}	//Seting allocator property according to our media type	props.cBuffers=1;	props.cbBuffer=This->m_pOurType->lSampleSize;	props.cbAlign=1;	props.cbPrefix=0;	This->m_pAll->vt->SetProperties(This->m_pAll, &props, &props1);	//Notify remote pin about choosed allocator	This->m_pImp->vt->NotifyAllocator(This->m_pImp, This->m_pAll, 0);	This->m_pOurOutput = COutputPinCreate(This->m_pDestType,DS_Filter_CopySample,pUserData);	result = This->m_pOutputPin->vt->QueryAccept(This->m_pOutputPin,							This->m_pDestType);	// Only connect if we can.  Otherwise delay the connection until	// DS_VideoDecoder_SetDestFmt is called	if (!result) {	    result = This->m_pOutputPin->vt->ReceiveConnection(This->m_pOutputPin,							   (IPin*) This->m_pOurOutput,							   This->m_pDestType);	    if(result)	    {		em = "could not connect to output pin";        	break;	    }	}	init++;        break;    }    tempAll->vt->Release((IUnknown*)tempAll);    if (!init)    {	DS_Filter_Destroy(This);	ERROR_MSG("Warning: DS_Filter() %s.  (DLL=%.200s, r=0x%x)", em, dllname, result);        This = 0;    }    return This;}#endif // (C_HAS_DIRECTSHOW)