2 * Transform Filter (Base for decoders, etc...)
4 * Copyright 2005 Christian Costa
6 * This library is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
11 * This library is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with this library; if not, write to the Free Software
18 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
23 #include "quartz_private.h"
24 #include "control_private.h"
42 #include "wine/unicode.h"
43 #include "wine/debug.h"
45 #include "transform.h"
47 WINE_DEFAULT_DEBUG_CHANNEL(quartz
);
49 static const WCHAR wcsInputPinName
[] = {'i','n','p','u','t',' ','p','i','n',0};
50 static const WCHAR wcsOutputPinName
[] = {'o','u','t','p','u','t',' ','p','i','n',0};
52 static const IBaseFilterVtbl TransformFilter_Vtbl
;
53 static const IPinVtbl TransformFilter_InputPin_Vtbl
;
54 static const IMemInputPinVtbl MemInputPin_Vtbl
;
55 static const IPinVtbl TransformFilter_OutputPin_Vtbl
;
57 static HRESULT
TransformFilter_Sample(LPVOID iface
, IMediaSample
* pSample
)
59 TransformFilterImpl
*This
= (TransformFilterImpl
*)iface
;
60 LPBYTE pbSrcStream
= NULL
;
62 REFERENCE_TIME tStart
, tStop
;
65 TRACE("%p %p\n", iface
, pSample
);
67 hr
= IMediaSample_GetPointer(pSample
, &pbSrcStream
);
70 ERR("Cannot get pointer to sample data (%lx)\n", hr
);
74 hr
= IMediaSample_GetTime(pSample
, &tStart
, &tStop
);
76 ERR("Cannot get sample time (%lx)\n", hr
);
78 cbSrcStream
= IMediaSample_GetActualDataLength(pSample
);
80 TRACE("Sample data ptr = %p, size = %ld\n", pbSrcStream
, cbSrcStream
);
82 #if 0 /* For debugging purpose */
85 for(i
= 0; i
< cbSrcStream
; i
++)
87 if ((i
!=0) && !(i
%16))
89 DPRINTF("%02x ", pbSrcStream
[i
]);
95 This
->pFuncsTable
->pfnProcessSampleData(This
, pbSrcStream
, cbSrcStream
);
100 static HRESULT
TransformFilter_Input_QueryAccept(LPVOID iface
, const AM_MEDIA_TYPE
* pmt
)
102 TransformFilterImpl
* This
= (TransformFilterImpl
*)iface
;
103 TRACE("%p\n", iface
);
104 dump_AM_MEDIA_TYPE(pmt
);
106 return This
->pFuncsTable
->pfnConnectInput(This
, pmt
);
110 static HRESULT
TransformFilter_Output_QueryAccept(LPVOID iface
, const AM_MEDIA_TYPE
* pmt
)
112 TransformFilterImpl
* pTransformFilter
= (TransformFilterImpl
*)iface
;
113 AM_MEDIA_TYPE
* outpmt
= &((OutputPin
*)pTransformFilter
->ppPins
[1])->pin
.mtCurrent
;
114 TRACE("%p\n", iface
);
116 if (IsEqualIID(&pmt
->majortype
, &outpmt
->majortype
) && IsEqualIID(&pmt
->subtype
, &outpmt
->subtype
))
121 static HRESULT
TransformFilter_InputPin_Construct(const PIN_INFO
* pPinInfo
, SAMPLEPROC pSampleProc
, LPVOID pUserData
, QUERYACCEPTPROC pQueryAccept
, LPCRITICAL_SECTION pCritSec
, IPin
** ppPin
)
127 if (pPinInfo
->dir
!= PINDIR_INPUT
)
129 ERR("Pin direction(%x) != PINDIR_INPUT\n", pPinInfo
->dir
);
133 pPinImpl
= CoTaskMemAlloc(sizeof(*pPinImpl
));
136 return E_OUTOFMEMORY
;
138 if (SUCCEEDED(InputPin_Init(pPinInfo
, pSampleProc
, pUserData
, pQueryAccept
, pCritSec
, pPinImpl
)))
140 pPinImpl
->pin
.lpVtbl
= &TransformFilter_InputPin_Vtbl
;
141 pPinImpl
->lpVtblMemInput
= &MemInputPin_Vtbl
;
143 *ppPin
= (IPin
*)(&pPinImpl
->pin
.lpVtbl
);
149 HRESULT
TransformFilter_OutputPin_Construct(const PIN_INFO
* pPinInfo
, ALLOCATOR_PROPERTIES
*props
, LPVOID pUserData
, QUERYACCEPTPROC pQueryAccept
, LPCRITICAL_SECTION pCritSec
, IPin
** ppPin
)
151 OutputPin
* pPinImpl
;
155 if (pPinInfo
->dir
!= PINDIR_OUTPUT
)
157 ERR("Pin direction(%x) != PINDIR_OUTPUT\n", pPinInfo
->dir
);
161 pPinImpl
= CoTaskMemAlloc(sizeof(*pPinImpl
));
164 return E_OUTOFMEMORY
;
166 if (SUCCEEDED(OutputPin_Init(pPinInfo
, props
, pUserData
, pQueryAccept
, pCritSec
, pPinImpl
)))
168 pPinImpl
->pin
.lpVtbl
= &TransformFilter_OutputPin_Vtbl
;
170 *ppPin
= (IPin
*)(&pPinImpl
->pin
.lpVtbl
);
176 HRESULT
TransformFilter_Create(TransformFilterImpl
* pTransformFilter
, const CLSID
* pClsid
, TransformFuncsTable
* pFuncsTable
)
182 /* pTransformFilter is already allocated */
183 pTransformFilter
->clsid
= *pClsid
;
184 pTransformFilter
->pFuncsTable
= pFuncsTable
;
186 pTransformFilter
->lpVtbl
= &TransformFilter_Vtbl
;
188 pTransformFilter
->refCount
= 1;
189 InitializeCriticalSection(&pTransformFilter
->csFilter
);
190 pTransformFilter
->state
= State_Stopped
;
191 pTransformFilter
->pClock
= NULL
;
192 ZeroMemory(&pTransformFilter
->filterInfo
, sizeof(FILTER_INFO
));
194 pTransformFilter
->ppPins
= CoTaskMemAlloc(2 * sizeof(IPin
*));
196 /* construct input pin */
197 piInput
.dir
= PINDIR_INPUT
;
198 piInput
.pFilter
= (IBaseFilter
*)pTransformFilter
;
199 lstrcpynW(piInput
.achName
, wcsInputPinName
, sizeof(piInput
.achName
) / sizeof(piInput
.achName
[0]));
200 piOutput
.dir
= PINDIR_OUTPUT
;
201 piOutput
.pFilter
= (IBaseFilter
*)pTransformFilter
;
202 lstrcpynW(piOutput
.achName
, wcsOutputPinName
, sizeof(piOutput
.achName
) / sizeof(piOutput
.achName
[0]));
204 hr
= TransformFilter_InputPin_Construct(&piInput
, TransformFilter_Sample
, (LPVOID
)pTransformFilter
, TransformFilter_Input_QueryAccept
, &pTransformFilter
->csFilter
, &pTransformFilter
->ppPins
[0]);
208 ALLOCATOR_PROPERTIES props
;
211 props
.cbBuffer
= 0; /* Will be updated at connection time */
214 hr
= TransformFilter_OutputPin_Construct(&piOutput
, &props
, NULL
, TransformFilter_Output_QueryAccept
, &pTransformFilter
->csFilter
, &pTransformFilter
->ppPins
[1]);
217 ERR("Cannot create output pin (%lx)\n", hr
);
221 CoTaskMemFree(pTransformFilter
->ppPins
);
222 DeleteCriticalSection(&pTransformFilter
->csFilter
);
223 CoTaskMemFree(pTransformFilter
);
229 static HRESULT WINAPI
TransformFilter_QueryInterface(IBaseFilter
* iface
, REFIID riid
, LPVOID
* ppv
)
231 TransformFilterImpl
*This
= (TransformFilterImpl
*)iface
;
232 TRACE("(%p/%p)->(%s, %p)\n", This
, iface
, qzdebugstr_guid(riid
), ppv
);
236 if (IsEqualIID(riid
, &IID_IUnknown
))
238 else if (IsEqualIID(riid
, &IID_IPersist
))
240 else if (IsEqualIID(riid
, &IID_IMediaFilter
))
242 else if (IsEqualIID(riid
, &IID_IBaseFilter
))
247 IUnknown_AddRef((IUnknown
*)(*ppv
));
251 FIXME("No interface for %s!\n", qzdebugstr_guid(riid
));
253 return E_NOINTERFACE
;
256 static ULONG WINAPI
TransformFilter_AddRef(IBaseFilter
* iface
)
258 TransformFilterImpl
*This
= (TransformFilterImpl
*)iface
;
259 ULONG refCount
= InterlockedIncrement(&This
->refCount
);
261 TRACE("(%p/%p)->() AddRef from %ld\n", This
, iface
, refCount
- 1);
266 static ULONG WINAPI
TransformFilter_Release(IBaseFilter
* iface
)
268 TransformFilterImpl
*This
= (TransformFilterImpl
*)iface
;
269 ULONG refCount
= InterlockedDecrement(&This
->refCount
);
271 TRACE("(%p/%p)->() Release from %ld\n", This
, iface
, refCount
+ 1);
277 DeleteCriticalSection(&This
->csFilter
);
278 IReferenceClock_Release(This
->pClock
);
280 for (i
= 0; i
< 2; i
++)
281 IPin_Release(This
->ppPins
[i
]);
283 HeapFree(GetProcessHeap(), 0, This
->ppPins
);
286 This
->pFuncsTable
->pfnCleanup(This
);
288 TRACE("Destroying transform filter\n");
297 /** IPersist methods **/
299 static HRESULT WINAPI
TransformFilter_GetClassID(IBaseFilter
* iface
, CLSID
* pClsid
)
301 TransformFilterImpl
*This
= (TransformFilterImpl
*)iface
;
303 TRACE("(%p/%p)->(%p)\n", This
, iface
, pClsid
);
305 *pClsid
= This
->clsid
;
310 /** IMediaFilter methods **/
312 static HRESULT WINAPI
TransformFilter_Stop(IBaseFilter
* iface
)
314 TransformFilterImpl
*This
= (TransformFilterImpl
*)iface
;
316 TRACE("(%p/%p)\n", This
, iface
);
318 EnterCriticalSection(&This
->csFilter
);
320 This
->state
= State_Stopped
;
321 if (This
->pFuncsTable
->pfnProcessEnd
)
322 This
->pFuncsTable
->pfnProcessEnd(This
);
324 LeaveCriticalSection(&This
->csFilter
);
329 static HRESULT WINAPI
TransformFilter_Pause(IBaseFilter
* iface
)
331 TransformFilterImpl
*This
= (TransformFilterImpl
*)iface
;
333 TRACE("(%p/%p)->()\n", This
, iface
);
335 EnterCriticalSection(&This
->csFilter
);
337 This
->state
= State_Paused
;
339 LeaveCriticalSection(&This
->csFilter
);
344 static HRESULT WINAPI
TransformFilter_Run(IBaseFilter
* iface
, REFERENCE_TIME tStart
)
347 TransformFilterImpl
*This
= (TransformFilterImpl
*)iface
;
349 TRACE("(%p/%p)->(%s)\n", This
, iface
, wine_dbgstr_longlong(tStart
));
351 EnterCriticalSection(&This
->csFilter
);
353 This
->rtStreamStart
= tStart
;
354 This
->state
= State_Running
;
355 OutputPin_CommitAllocator((OutputPin
*)This
->ppPins
[1]);
356 if (This
->pFuncsTable
->pfnProcessBegin
)
357 This
->pFuncsTable
->pfnProcessBegin(This
);
359 LeaveCriticalSection(&This
->csFilter
);
364 static HRESULT WINAPI
TransformFilter_GetState(IBaseFilter
* iface
, DWORD dwMilliSecsTimeout
, FILTER_STATE
*pState
)
366 TransformFilterImpl
*This
= (TransformFilterImpl
*)iface
;
368 TRACE("(%p/%p)->(%ld, %p)\n", This
, iface
, dwMilliSecsTimeout
, pState
);
370 EnterCriticalSection(&This
->csFilter
);
372 *pState
= This
->state
;
374 LeaveCriticalSection(&This
->csFilter
);
379 static HRESULT WINAPI
TransformFilter_SetSyncSource(IBaseFilter
* iface
, IReferenceClock
*pClock
)
381 TransformFilterImpl
*This
= (TransformFilterImpl
*)iface
;
383 TRACE("(%p/%p)->(%p)\n", This
, iface
, pClock
);
385 EnterCriticalSection(&This
->csFilter
);
388 IReferenceClock_Release(This
->pClock
);
389 This
->pClock
= pClock
;
391 IReferenceClock_AddRef(This
->pClock
);
393 LeaveCriticalSection(&This
->csFilter
);
398 static HRESULT WINAPI
TransformFilter_GetSyncSource(IBaseFilter
* iface
, IReferenceClock
**ppClock
)
400 TransformFilterImpl
*This
= (TransformFilterImpl
*)iface
;
402 TRACE("(%p/%p)->(%p)\n", This
, iface
, ppClock
);
404 EnterCriticalSection(&This
->csFilter
);
406 *ppClock
= This
->pClock
;
407 IReferenceClock_AddRef(This
->pClock
);
409 LeaveCriticalSection(&This
->csFilter
);
414 /** IBaseFilter implementation **/
416 static HRESULT WINAPI
TransformFilter_EnumPins(IBaseFilter
* iface
, IEnumPins
**ppEnum
)
419 TransformFilterImpl
*This
= (TransformFilterImpl
*)iface
;
421 TRACE("(%p/%p)->(%p)\n", This
, iface
, ppEnum
);
423 epd
.cPins
= 2; /* input and output pins */
424 epd
.ppPins
= This
->ppPins
;
425 return IEnumPinsImpl_Construct(&epd
, ppEnum
);
428 static HRESULT WINAPI
TransformFilter_FindPin(IBaseFilter
* iface
, LPCWSTR Id
, IPin
**ppPin
)
430 TransformFilterImpl
*This
= (TransformFilterImpl
*)iface
;
432 TRACE("(%p/%p)->(%p,%p)\n", This
, iface
, debugstr_w(Id
), ppPin
);
437 static HRESULT WINAPI
TransformFilter_QueryFilterInfo(IBaseFilter
* iface
, FILTER_INFO
*pInfo
)
439 TransformFilterImpl
*This
= (TransformFilterImpl
*)iface
;
441 TRACE("(%p/%p)->(%p)\n", This
, iface
, pInfo
);
443 strcpyW(pInfo
->achName
, This
->filterInfo
.achName
);
444 pInfo
->pGraph
= This
->filterInfo
.pGraph
;
447 IFilterGraph_AddRef(pInfo
->pGraph
);
452 static HRESULT WINAPI
TransformFilter_JoinFilterGraph(IBaseFilter
* iface
, IFilterGraph
*pGraph
, LPCWSTR pName
)
455 TransformFilterImpl
*This
= (TransformFilterImpl
*)iface
;
457 TRACE("(%p/%p)->(%p, %s)\n", This
, iface
, pGraph
, debugstr_w(pName
));
459 EnterCriticalSection(&This
->csFilter
);
462 strcpyW(This
->filterInfo
.achName
, pName
);
464 *This
->filterInfo
.achName
= '\0';
465 This
->filterInfo
.pGraph
= pGraph
; /* NOTE: do NOT increase ref. count */
467 LeaveCriticalSection(&This
->csFilter
);
472 static HRESULT WINAPI
TransformFilter_QueryVendorInfo(IBaseFilter
* iface
, LPWSTR
*pVendorInfo
)
474 TransformFilterImpl
*This
= (TransformFilterImpl
*)iface
;
475 TRACE("(%p/%p)->(%p)\n", This
, iface
, pVendorInfo
);
479 static const IBaseFilterVtbl TransformFilter_Vtbl
=
481 TransformFilter_QueryInterface
,
482 TransformFilter_AddRef
,
483 TransformFilter_Release
,
484 TransformFilter_GetClassID
,
485 TransformFilter_Stop
,
486 TransformFilter_Pause
,
488 TransformFilter_GetState
,
489 TransformFilter_SetSyncSource
,
490 TransformFilter_GetSyncSource
,
491 TransformFilter_EnumPins
,
492 TransformFilter_FindPin
,
493 TransformFilter_QueryFilterInfo
,
494 TransformFilter_JoinFilterGraph
,
495 TransformFilter_QueryVendorInfo
498 HRESULT WINAPI
TransformFilter_InputPin_EndOfStream(IPin
* iface
)
500 InputPin
* This
= (InputPin
*) iface
;
501 TransformFilterImpl
* pTransform
;
505 TRACE("(%p)->()\n", iface
);
507 /* Since we process samples synchronously, just forward notification downstream */
508 pTransform
= (TransformFilterImpl
*)This
->pin
.pinInfo
.pFilter
;
512 hr
= IPin_ConnectedTo(pTransform
->ppPins
[1], &ppin
);
515 hr
= IPin_EndOfStream(ppin
);
524 static const IPinVtbl TransformFilter_InputPin_Vtbl
=
526 InputPin_QueryInterface
,
530 InputPin_ReceiveConnection
,
532 IPinImpl_ConnectedTo
,
533 IPinImpl_ConnectionMediaType
,
534 IPinImpl_QueryPinInfo
,
535 IPinImpl_QueryDirection
,
537 IPinImpl_QueryAccept
,
538 IPinImpl_EnumMediaTypes
,
539 IPinImpl_QueryInternalConnections
,
540 TransformFilter_InputPin_EndOfStream
,
546 HRESULT WINAPI
TransformFilter_Output_EnumMediaTypes(IPin
* iface
, IEnumMediaTypes
** ppEnum
)
548 IPinImpl
*This
= (IPinImpl
*)iface
;
549 ENUMMEDIADETAILS emd
;
551 TRACE("(%p/%p)->(%p)\n", This
, iface
, ppEnum
);
554 emd
.pMediaTypes
= &This
->mtCurrent
;
556 return IEnumMediaTypesImpl_Construct(&emd
, ppEnum
);
559 HRESULT WINAPI
TransformFilter_Output_Disconnect(IPin
* iface
)
561 OutputPin
* This
= (OutputPin
*)iface
;
563 TransformFilterImpl
* pTransformFilter
= (TransformFilterImpl
*)This
->pin
.pinInfo
.pFilter
;
565 TRACE("(%p/%p)->()\n", This
, iface
);
567 hr
= OutputPin_Disconnect(iface
);
571 pTransformFilter
->pFuncsTable
->pfnCleanup(pTransformFilter
);
577 static const IPinVtbl TransformFilter_OutputPin_Vtbl
=
579 OutputPin_QueryInterface
,
583 OutputPin_ReceiveConnection
,
584 TransformFilter_Output_Disconnect
,
585 IPinImpl_ConnectedTo
,
586 IPinImpl_ConnectionMediaType
,
587 IPinImpl_QueryPinInfo
,
588 IPinImpl_QueryDirection
,
590 IPinImpl_QueryAccept
,
591 TransformFilter_Output_EnumMediaTypes
,
592 IPinImpl_QueryInternalConnections
,
593 OutputPin_EndOfStream
,
594 OutputPin_BeginFlush
,
599 static const IMemInputPinVtbl MemInputPin_Vtbl
=
601 MemInputPin_QueryInterface
,
604 MemInputPin_GetAllocator
,
605 MemInputPin_NotifyAllocator
,
606 MemInputPin_GetAllocatorRequirements
,
608 MemInputPin_ReceiveMultiple
,
609 MemInputPin_ReceiveCanBlock