512 lines
16 KiB
C
512 lines
16 KiB
C
/*
|
|
* Transform Filter (Base for decoders, etc...)
|
|
*
|
|
* Copyright 2005 Christian Costa
|
|
*
|
|
* This library is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Lesser General Public
|
|
* License as published by the Free Software Foundation; either
|
|
* version 2.1 of the License, or (at your option) any later version.
|
|
*
|
|
* This library is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
* Lesser General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Lesser General Public
|
|
* License along with this library; if not, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
|
|
*/
|
|
|
|
#include "config.h"
|
|
|
|
#include "quartz_private.h"
|
|
#include "control_private.h"
|
|
#include "pin.h"
|
|
|
|
#include "amvideo.h"
|
|
#include "windef.h"
|
|
#include "winbase.h"
|
|
#include "dshow.h"
|
|
#include "strmif.h"
|
|
#include "vfw.h"
|
|
|
|
#include <assert.h>
|
|
|
|
#include "wine/unicode.h"
|
|
#include "wine/debug.h"
|
|
|
|
#include "transform.h"
|
|
|
|
WINE_DEFAULT_DEBUG_CHANNEL(quartz);
|
|
|
|
static const WCHAR wcsInputPinName[] = {'i','n','p','u','t',' ','p','i','n',0};
|
|
static const WCHAR wcsOutputPinName[] = {'o','u','t','p','u','t',' ','p','i','n',0};
|
|
|
|
static const IBaseFilterVtbl TransformFilter_Vtbl;
|
|
static const IPinVtbl TransformFilter_InputPin_Vtbl;
|
|
static const IPinVtbl TransformFilter_OutputPin_Vtbl;
|
|
|
|
static HRESULT WINAPI TransformFilter_Input_CheckMediaType(IPin *iface, const AM_MEDIA_TYPE * pmt)
|
|
{
|
|
TransformFilterImpl* This = (TransformFilterImpl *)((BasePin *)iface)->pinInfo.pFilter;
|
|
TRACE("%p\n", iface);
|
|
dump_AM_MEDIA_TYPE(pmt);
|
|
|
|
if (This->pFuncsTable->pfnQueryConnect)
|
|
return This->pFuncsTable->pfnQueryConnect(This, pmt);
|
|
/* Assume OK if there's no query method (the connection will fail if
|
|
needed) */
|
|
return S_OK;
|
|
}
|
|
|
|
|
|
static HRESULT WINAPI TransformFilter_Output_QueryAccept(IPin *iface, const AM_MEDIA_TYPE * pmt)
|
|
{
|
|
BasePin *This = (BasePin *)iface;
|
|
TransformFilterImpl *pTransformFilter = (TransformFilterImpl *)This->pinInfo.pFilter;
|
|
AM_MEDIA_TYPE* outpmt = &pTransformFilter->pmt;
|
|
TRACE("%p\n", iface);
|
|
|
|
if (IsEqualIID(&pmt->majortype, &outpmt->majortype)
|
|
&& (IsEqualIID(&pmt->subtype, &outpmt->subtype) || IsEqualIID(&outpmt->subtype, &GUID_NULL)))
|
|
return S_OK;
|
|
return S_FALSE;
|
|
}
|
|
|
|
HRESULT TransformFilter_Create(TransformFilterImpl* pTransformFilter, const CLSID* pClsid, const TransformFuncsTable* pFuncsTable)
|
|
{
|
|
HRESULT hr;
|
|
PIN_INFO piInput;
|
|
PIN_INFO piOutput;
|
|
|
|
BaseFilter_Init(&pTransformFilter->filter, &TransformFilter_Vtbl, pClsid, (DWORD_PTR)(__FILE__ ": TransformFilterImpl.csFilter"));
|
|
|
|
/* pTransformFilter is already allocated */
|
|
pTransformFilter->pFuncsTable = pFuncsTable;
|
|
ZeroMemory(&pTransformFilter->pmt, sizeof(pTransformFilter->pmt));
|
|
pTransformFilter->npins = 2;
|
|
|
|
pTransformFilter->ppPins = CoTaskMemAlloc(2 * sizeof(IPin *));
|
|
|
|
/* construct input pin */
|
|
piInput.dir = PINDIR_INPUT;
|
|
piInput.pFilter = (IBaseFilter *)pTransformFilter;
|
|
lstrcpynW(piInput.achName, wcsInputPinName, sizeof(piInput.achName) / sizeof(piInput.achName[0]));
|
|
piOutput.dir = PINDIR_OUTPUT;
|
|
piOutput.pFilter = (IBaseFilter *)pTransformFilter;
|
|
lstrcpynW(piOutput.achName, wcsOutputPinName, sizeof(piOutput.achName) / sizeof(piOutput.achName[0]));
|
|
|
|
hr = BaseInputPin_Construct(&TransformFilter_InputPin_Vtbl, &piInput, TransformFilter_Input_CheckMediaType, pFuncsTable->pfnProcessSampleData, &pTransformFilter->filter.csFilter, NULL, &pTransformFilter->ppPins[0]);
|
|
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
ALLOCATOR_PROPERTIES props;
|
|
props.cbAlign = 1;
|
|
props.cbPrefix = 0;
|
|
props.cbBuffer = 0; /* Will be updated at connection time */
|
|
props.cBuffers = 1;
|
|
|
|
hr = BaseOutputPin_Construct(&TransformFilter_OutputPin_Vtbl, sizeof(BaseOutputPin), &piOutput, &props, NULL, &pTransformFilter->filter.csFilter, &pTransformFilter->ppPins[1]);
|
|
|
|
if (FAILED(hr))
|
|
ERR("Cannot create output pin (%x)\n", hr);
|
|
else
|
|
{
|
|
ISeekingPassThru *passthru;
|
|
hr = CoCreateInstance(&CLSID_SeekingPassThru, (IUnknown*)pTransformFilter, CLSCTX_INPROC_SERVER, &IID_IUnknown, (void**)&pTransformFilter->seekthru_unk);
|
|
IUnknown_QueryInterface(pTransformFilter->seekthru_unk, &IID_ISeekingPassThru, (void**)&passthru);
|
|
ISeekingPassThru_Init(passthru, FALSE, (IPin*)pTransformFilter->ppPins[0]);
|
|
ISeekingPassThru_Release(passthru);
|
|
}
|
|
}
|
|
if (FAILED(hr))
|
|
{
|
|
CoTaskMemFree(pTransformFilter->ppPins);
|
|
BaseFilterImpl_Release((IBaseFilter*)pTransformFilter);
|
|
CoTaskMemFree(pTransformFilter);
|
|
}
|
|
|
|
return hr;
|
|
}
|
|
|
|
static HRESULT WINAPI TransformFilter_QueryInterface(IBaseFilter * iface, REFIID riid, LPVOID * ppv)
|
|
{
|
|
TransformFilterImpl *This = (TransformFilterImpl *)iface;
|
|
TRACE("(%p/%p)->(%s, %p)\n", This, iface, qzdebugstr_guid(riid), ppv);
|
|
|
|
*ppv = NULL;
|
|
|
|
if (IsEqualIID(riid, &IID_IUnknown))
|
|
*ppv = This;
|
|
else if (IsEqualIID(riid, &IID_IPersist))
|
|
*ppv = This;
|
|
else if (IsEqualIID(riid, &IID_IMediaFilter))
|
|
*ppv = This;
|
|
else if (IsEqualIID(riid, &IID_IBaseFilter))
|
|
*ppv = This;
|
|
else if (IsEqualIID(riid, &IID_IMediaSeeking))
|
|
return IUnknown_QueryInterface(This->seekthru_unk, riid, ppv);
|
|
|
|
if (*ppv)
|
|
{
|
|
IUnknown_AddRef((IUnknown *)(*ppv));
|
|
return S_OK;
|
|
}
|
|
|
|
if (!IsEqualIID(riid, &IID_IPin) && !IsEqualIID(riid, &IID_IVideoWindow))
|
|
FIXME("No interface for %s!\n", qzdebugstr_guid(riid));
|
|
|
|
return E_NOINTERFACE;
|
|
}
|
|
|
|
static ULONG WINAPI TransformFilter_Release(IBaseFilter * iface)
|
|
{
|
|
TransformFilterImpl *This = (TransformFilterImpl *)iface;
|
|
ULONG refCount = BaseFilterImpl_Release(iface);
|
|
|
|
TRACE("(%p/%p)->() Release from %d\n", This, iface, refCount + 1);
|
|
|
|
if (!refCount)
|
|
{
|
|
ULONG i;
|
|
|
|
for (i = 0; i < This->npins; i++)
|
|
{
|
|
IPin *pConnectedTo;
|
|
|
|
if (SUCCEEDED(IPin_ConnectedTo(This->ppPins[i], &pConnectedTo)))
|
|
{
|
|
IPin_Disconnect(pConnectedTo);
|
|
IPin_Release(pConnectedTo);
|
|
}
|
|
IPin_Disconnect(This->ppPins[i]);
|
|
|
|
IPin_Release(This->ppPins[i]);
|
|
}
|
|
|
|
CoTaskMemFree(This->ppPins);
|
|
|
|
TRACE("Destroying transform filter\n");
|
|
FreeMediaType(&This->pmt);
|
|
CoTaskMemFree(This);
|
|
|
|
return 0;
|
|
}
|
|
else
|
|
return refCount;
|
|
}
|
|
|
|
/** IMediaFilter methods **/
|
|
|
|
static HRESULT WINAPI TransformFilter_Stop(IBaseFilter * iface)
|
|
{
|
|
TransformFilterImpl *This = (TransformFilterImpl *)iface;
|
|
HRESULT hr = S_OK;
|
|
|
|
TRACE("(%p/%p)\n", This, iface);
|
|
|
|
EnterCriticalSection(&This->filter.csFilter);
|
|
{
|
|
This->filter.state = State_Stopped;
|
|
if (This->pFuncsTable->pfnProcessEnd)
|
|
hr = This->pFuncsTable->pfnProcessEnd(This);
|
|
}
|
|
LeaveCriticalSection(&This->filter.csFilter);
|
|
|
|
return hr;
|
|
}
|
|
|
|
static HRESULT WINAPI TransformFilter_Pause(IBaseFilter * iface)
|
|
{
|
|
TransformFilterImpl *This = (TransformFilterImpl *)iface;
|
|
HRESULT hr;
|
|
|
|
TRACE("(%p/%p)->()\n", This, iface);
|
|
|
|
EnterCriticalSection(&This->filter.csFilter);
|
|
{
|
|
if (This->filter.state == State_Stopped)
|
|
hr = IBaseFilter_Run(iface, -1);
|
|
else
|
|
hr = S_OK;
|
|
|
|
if (SUCCEEDED(hr))
|
|
This->filter.state = State_Paused;
|
|
}
|
|
LeaveCriticalSection(&This->filter.csFilter);
|
|
|
|
return hr;
|
|
}
|
|
|
|
static HRESULT WINAPI TransformFilter_Run(IBaseFilter * iface, REFERENCE_TIME tStart)
|
|
{
|
|
HRESULT hr = S_OK;
|
|
TransformFilterImpl *This = (TransformFilterImpl *)iface;
|
|
|
|
TRACE("(%p/%p)->(%s)\n", This, iface, wine_dbgstr_longlong(tStart));
|
|
|
|
EnterCriticalSection(&This->filter.csFilter);
|
|
{
|
|
if (This->filter.state == State_Stopped)
|
|
{
|
|
((BaseInputPin *)This->ppPins[0])->end_of_stream = 0;
|
|
if (This->pFuncsTable->pfnProcessBegin)
|
|
hr = This->pFuncsTable->pfnProcessBegin(This);
|
|
if (SUCCEEDED(hr))
|
|
hr = BaseOutputPinImpl_Active((BaseOutputPin *)This->ppPins[1]);
|
|
}
|
|
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
This->filter.rtStreamStart = tStart;
|
|
This->filter.state = State_Running;
|
|
}
|
|
}
|
|
LeaveCriticalSection(&This->filter.csFilter);
|
|
|
|
return hr;
|
|
}
|
|
|
|
/** IBaseFilter implementation **/
|
|
|
|
static IPin* WINAPI TransformFilter_GetPin(IBaseFilter *iface, int pos)
|
|
{
|
|
TransformFilterImpl *This = (TransformFilterImpl *)iface;
|
|
|
|
if (pos >= This->npins || pos < 0)
|
|
return NULL;
|
|
|
|
IPin_AddRef(This->ppPins[pos]);
|
|
return This->ppPins[pos];
|
|
}
|
|
|
|
static LONG WINAPI TransformFilter_GetPinCount(IBaseFilter *iface)
|
|
{
|
|
TransformFilterImpl *This = (TransformFilterImpl *)iface;
|
|
|
|
return (This->npins+1);
|
|
}
|
|
|
|
static LONG WINAPI TransformFilter_GetPinVersion(IBaseFilter *iface)
|
|
{
|
|
/* Our pins are static, not changing so setting static tick count is ok */
|
|
return 0;
|
|
}
|
|
|
|
static HRESULT WINAPI TransformFilter_EnumPins(IBaseFilter * iface, IEnumPins **ppEnum)
|
|
{
|
|
TransformFilterImpl *This = (TransformFilterImpl *)iface;
|
|
|
|
TRACE("(%p/%p)->(%p)\n", This, iface, ppEnum);
|
|
|
|
return EnumPins_Construct(iface, TransformFilter_GetPin, TransformFilter_GetPinCount, TransformFilter_GetPinVersion, ppEnum);
|
|
}
|
|
|
|
static HRESULT WINAPI TransformFilter_FindPin(IBaseFilter * iface, LPCWSTR Id, IPin **ppPin)
|
|
{
|
|
TransformFilterImpl *This = (TransformFilterImpl *)iface;
|
|
|
|
TRACE("(%p/%p)->(%p,%p)\n", This, iface, debugstr_w(Id), ppPin);
|
|
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static const IBaseFilterVtbl TransformFilter_Vtbl =
|
|
{
|
|
TransformFilter_QueryInterface,
|
|
BaseFilterImpl_AddRef,
|
|
TransformFilter_Release,
|
|
BaseFilterImpl_GetClassID,
|
|
TransformFilter_Stop,
|
|
TransformFilter_Pause,
|
|
TransformFilter_Run,
|
|
BaseFilterImpl_GetState,
|
|
BaseFilterImpl_SetSyncSource,
|
|
BaseFilterImpl_GetSyncSource,
|
|
TransformFilter_EnumPins,
|
|
TransformFilter_FindPin,
|
|
BaseFilterImpl_QueryFilterInfo,
|
|
BaseFilterImpl_JoinFilterGraph,
|
|
BaseFilterImpl_QueryVendorInfo
|
|
};
|
|
|
|
static HRESULT WINAPI TransformFilter_InputPin_EndOfStream(IPin * iface)
|
|
{
|
|
BaseInputPin* This = (BaseInputPin*) iface;
|
|
TransformFilterImpl* pTransform;
|
|
IPin* ppin;
|
|
HRESULT hr;
|
|
|
|
TRACE("(%p)->()\n", iface);
|
|
|
|
/* Since we process samples synchronously, just forward notification downstream */
|
|
pTransform = (TransformFilterImpl*)This->pin.pinInfo.pFilter;
|
|
if (!pTransform)
|
|
hr = E_FAIL;
|
|
else
|
|
hr = IPin_ConnectedTo(pTransform->ppPins[1], &ppin);
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
hr = IPin_EndOfStream(ppin);
|
|
IPin_Release(ppin);
|
|
}
|
|
|
|
if (FAILED(hr))
|
|
ERR("%x\n", hr);
|
|
return hr;
|
|
}
|
|
|
|
static HRESULT WINAPI TransformFilter_InputPin_ReceiveConnection(IPin * iface, IPin * pReceivePin, const AM_MEDIA_TYPE * pmt)
|
|
{
|
|
BaseInputPin* This = (BaseInputPin*) iface;
|
|
TransformFilterImpl* pTransform;
|
|
HRESULT hr;
|
|
|
|
TRACE("(%p)->(%p, %p)\n", iface, pReceivePin, pmt);
|
|
|
|
pTransform = (TransformFilterImpl*)This->pin.pinInfo.pFilter;
|
|
|
|
hr = pTransform->pFuncsTable->pfnConnectInput(This, pmt);
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
hr = BaseInputPinImpl_ReceiveConnection(iface, pReceivePin, pmt);
|
|
if (FAILED(hr))
|
|
pTransform->pFuncsTable->pfnCleanup(This);
|
|
}
|
|
|
|
return hr;
|
|
}
|
|
|
|
static HRESULT WINAPI TransformFilter_InputPin_Disconnect(IPin * iface)
|
|
{
|
|
BaseInputPin* This = (BaseInputPin*) iface;
|
|
TransformFilterImpl* pTransform;
|
|
|
|
TRACE("(%p)->()\n", iface);
|
|
|
|
pTransform = (TransformFilterImpl*)This->pin.pinInfo.pFilter;
|
|
pTransform->pFuncsTable->pfnCleanup(This);
|
|
|
|
return BasePinImpl_Disconnect(iface);
|
|
}
|
|
|
|
static HRESULT WINAPI TransformFilter_InputPin_BeginFlush(IPin * iface)
|
|
{
|
|
BaseInputPin* This = (BaseInputPin*) iface;
|
|
TransformFilterImpl* pTransform;
|
|
HRESULT hr = S_OK;
|
|
|
|
TRACE("(%p)->()\n", iface);
|
|
|
|
pTransform = (TransformFilterImpl*)This->pin.pinInfo.pFilter;
|
|
EnterCriticalSection(&pTransform->filter.csFilter);
|
|
if (pTransform->pFuncsTable->pfnBeginFlush)
|
|
hr = pTransform->pFuncsTable->pfnBeginFlush(This);
|
|
if (SUCCEEDED(hr))
|
|
hr = BaseInputPinImpl_BeginFlush(iface);
|
|
LeaveCriticalSection(&pTransform->filter.csFilter);
|
|
return hr;
|
|
}
|
|
|
|
static HRESULT WINAPI TransformFilter_InputPin_EndFlush(IPin * iface)
|
|
{
|
|
BaseInputPin* This = (BaseInputPin*) iface;
|
|
TransformFilterImpl* pTransform;
|
|
HRESULT hr = S_OK;
|
|
|
|
TRACE("(%p)->()\n", iface);
|
|
|
|
pTransform = (TransformFilterImpl*)This->pin.pinInfo.pFilter;
|
|
EnterCriticalSection(&pTransform->filter.csFilter);
|
|
if (pTransform->pFuncsTable->pfnEndFlush)
|
|
hr = pTransform->pFuncsTable->pfnEndFlush(This);
|
|
if (SUCCEEDED(hr))
|
|
hr = BaseInputPinImpl_EndFlush(iface);
|
|
LeaveCriticalSection(&pTransform->filter.csFilter);
|
|
return hr;
|
|
}
|
|
|
|
static HRESULT WINAPI TransformFilter_InputPin_NewSegment(IPin * iface, REFERENCE_TIME tStart, REFERENCE_TIME tStop, double dRate)
|
|
{
|
|
BaseInputPin* This = (BaseInputPin*) iface;
|
|
TransformFilterImpl* pTransform;
|
|
HRESULT hr = S_OK;
|
|
|
|
TRACE("(%p)->()\n", iface);
|
|
|
|
pTransform = (TransformFilterImpl*)This->pin.pinInfo.pFilter;
|
|
EnterCriticalSection(&pTransform->filter.csFilter);
|
|
if (pTransform->pFuncsTable->pfnNewSegment)
|
|
hr = pTransform->pFuncsTable->pfnNewSegment(This, tStart, tStop, dRate);
|
|
if (SUCCEEDED(hr))
|
|
hr = BaseInputPinImpl_NewSegment(iface, tStart, tStop, dRate);
|
|
LeaveCriticalSection(&pTransform->filter.csFilter);
|
|
return hr;
|
|
}
|
|
|
|
static const IPinVtbl TransformFilter_InputPin_Vtbl =
|
|
{
|
|
BaseInputPinImpl_QueryInterface,
|
|
BasePinImpl_AddRef,
|
|
BaseInputPinImpl_Release,
|
|
BaseInputPinImpl_Connect,
|
|
TransformFilter_InputPin_ReceiveConnection,
|
|
TransformFilter_InputPin_Disconnect,
|
|
BasePinImpl_ConnectedTo,
|
|
BasePinImpl_ConnectionMediaType,
|
|
BasePinImpl_QueryPinInfo,
|
|
BasePinImpl_QueryDirection,
|
|
BasePinImpl_QueryId,
|
|
BaseInputPinImpl_QueryAccept,
|
|
BasePinImpl_EnumMediaTypes,
|
|
BasePinImpl_QueryInternalConnections,
|
|
TransformFilter_InputPin_EndOfStream,
|
|
TransformFilter_InputPin_BeginFlush,
|
|
TransformFilter_InputPin_EndFlush,
|
|
TransformFilter_InputPin_NewSegment
|
|
};
|
|
|
|
static HRESULT WINAPI TransformFilter_Output_GetMediaType(IPin *iface, int iPosition, AM_MEDIA_TYPE *pmt)
|
|
{
|
|
BasePin *This = (BasePin *)iface;
|
|
TransformFilterImpl *pTransform = (TransformFilterImpl *)This->pinInfo.pFilter;
|
|
|
|
if (iPosition < 0)
|
|
return E_INVALIDARG;
|
|
if (iPosition > 0)
|
|
return VFW_S_NO_MORE_ITEMS;
|
|
CopyMediaType(pmt, &pTransform->pmt);
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI TransformFilter_Output_EnumMediaTypes(IPin * iface, IEnumMediaTypes ** ppEnum)
|
|
{
|
|
BasePin *This = (BasePin *)iface;
|
|
TRACE("(%p/%p)->(%p)\n", This, iface, ppEnum);
|
|
|
|
return EnumMediaTypes_Construct(iface, TransformFilter_Output_GetMediaType, BasePinImpl_GetMediaTypeVersion, ppEnum);
|
|
}
|
|
|
|
static const IPinVtbl TransformFilter_OutputPin_Vtbl =
|
|
{
|
|
BaseOutputPinImpl_QueryInterface,
|
|
BasePinImpl_AddRef,
|
|
BaseOutputPinImpl_Release,
|
|
BaseOutputPinImpl_Connect,
|
|
BaseOutputPinImpl_ReceiveConnection,
|
|
BaseOutputPinImpl_Disconnect,
|
|
BasePinImpl_ConnectedTo,
|
|
BasePinImpl_ConnectionMediaType,
|
|
BasePinImpl_QueryPinInfo,
|
|
BasePinImpl_QueryDirection,
|
|
BasePinImpl_QueryId,
|
|
TransformFilter_Output_QueryAccept,
|
|
TransformFilter_Output_EnumMediaTypes,
|
|
BasePinImpl_QueryInternalConnections,
|
|
BaseOutputPinImpl_EndOfStream,
|
|
BaseOutputPinImpl_BeginFlush,
|
|
BaseOutputPinImpl_EndFlush,
|
|
BaseOutputPinImpl_NewSegment
|
|
};
|