2 * Transform Filter (Base for decoders, etc...)
4 * Copyright 2005 Christian Costa
5 * Copyright 2010 Aric Stewart, CodeWeavers
7 * This library is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Lesser General Public
9 * License as published by the Free Software Foundation; either
10 * version 2.1 of the License, or (at your option) any later version.
12 * This library is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * Lesser General Public License for more details.
17 * You should have received a copy of the GNU Lesser General Public
18 * License along with this library; if not, write to the Free Software
19 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
35 #include "wine/unicode.h"
36 #include "wine/debug.h"
37 #include "wine/strmbase.h"
38 #include "strmbase_private.h"
40 WINE_DEFAULT_DEBUG_CHANNEL(strmbase
);
42 static const WCHAR wcsInputPinName
[] = {'i','n','p','u','t',' ','p','i','n',0};
43 static const WCHAR wcsOutputPinName
[] = {'o','u','t','p','u','t',' ','p','i','n',0};
45 static const IBaseFilterVtbl TransformFilter_Vtbl
;
46 static const IPinVtbl TransformFilter_InputPin_Vtbl
;
47 static const IPinVtbl TransformFilter_OutputPin_Vtbl
;
48 static const IQualityControlVtbl TransformFilter_QualityControl_Vtbl
;
50 static inline BaseInputPin
*impl_BaseInputPin_from_BasePin( BasePin
*iface
)
52 return CONTAINING_RECORD(iface
, BaseInputPin
, pin
);
55 static inline BasePin
*impl_BasePin_from_IPin( IPin
*iface
)
57 return CONTAINING_RECORD(iface
, BasePin
, IPin_iface
);
60 static inline BaseInputPin
*impl_BaseInputPin_from_IPin( IPin
*iface
)
62 return CONTAINING_RECORD(iface
, BaseInputPin
, pin
.IPin_iface
);
65 static inline BaseOutputPin
*impl_BaseOutputPin_from_IPin( IPin
*iface
)
67 return CONTAINING_RECORD(iface
, BaseOutputPin
, pin
.IPin_iface
);
70 static inline TransformFilter
*impl_from_IBaseFilter( IBaseFilter
*iface
)
72 return CONTAINING_RECORD(iface
, TransformFilter
, filter
.IBaseFilter_iface
);
75 static inline TransformFilter
*impl_from_BaseFilter( BaseFilter
*iface
)
77 return CONTAINING_RECORD(iface
, TransformFilter
, filter
);
80 static HRESULT WINAPI
TransformFilter_Input_CheckMediaType(BasePin
*iface
, const AM_MEDIA_TYPE
* pmt
)
82 BaseInputPin
* This
= impl_BaseInputPin_from_BasePin(iface
);
83 TransformFilter
* pTransform
;
86 pTransform
= impl_from_IBaseFilter(This
->pin
.pinInfo
.pFilter
);
88 if (pTransform
->pFuncsTable
->pfnCheckInputType
)
89 return pTransform
->pFuncsTable
->pfnCheckInputType(pTransform
, pmt
);
90 /* Assume OK if there's no query method (the connection will fail if
95 static HRESULT WINAPI
TransformFilter_Input_Receive(BaseInputPin
*This
, IMediaSample
*pInSample
)
98 TransformFilter
* pTransform
;
100 pTransform
= impl_from_IBaseFilter(This
->pin
.pinInfo
.pFilter
);
102 EnterCriticalSection(&pTransform
->csReceive
);
103 if (pTransform
->filter
.state
== State_Stopped
)
105 LeaveCriticalSection(&pTransform
->csReceive
);
106 return VFW_E_WRONG_STATE
;
109 if (This
->end_of_stream
|| This
->flushing
)
111 LeaveCriticalSection(&pTransform
->csReceive
);
115 LeaveCriticalSection(&pTransform
->csReceive
);
116 if (pTransform
->pFuncsTable
->pfnReceive
)
117 hr
= pTransform
->pFuncsTable
->pfnReceive(pTransform
, pInSample
);
124 static HRESULT WINAPI
TransformFilter_Output_QueryAccept(IPin
*iface
, const AM_MEDIA_TYPE
* pmt
)
126 BasePin
*This
= impl_BasePin_from_IPin(iface
);
127 TransformFilter
*pTransformFilter
= impl_from_IBaseFilter(This
->pinInfo
.pFilter
);
128 AM_MEDIA_TYPE
* outpmt
= &pTransformFilter
->pmt
;
129 TRACE("%p\n", iface
);
131 if (IsEqualIID(&pmt
->majortype
, &outpmt
->majortype
)
132 && (IsEqualIID(&pmt
->subtype
, &outpmt
->subtype
) || IsEqualIID(&outpmt
->subtype
, &GUID_NULL
)))
137 static HRESULT WINAPI
TransformFilter_Output_DecideBufferSize(BaseOutputPin
*This
, IMemAllocator
*pAlloc
, ALLOCATOR_PROPERTIES
*ppropInputRequest
)
139 TransformFilter
*pTransformFilter
= impl_from_IBaseFilter(This
->pin
.pinInfo
.pFilter
);
140 return pTransformFilter
->pFuncsTable
->pfnDecideBufferSize(pTransformFilter
, pAlloc
, ppropInputRequest
);
143 static HRESULT WINAPI
TransformFilter_Output_GetMediaType(BasePin
*This
, int iPosition
, AM_MEDIA_TYPE
*pmt
)
145 TransformFilter
*pTransform
= impl_from_IBaseFilter(This
->pinInfo
.pFilter
);
150 return VFW_S_NO_MORE_ITEMS
;
151 CopyMediaType(pmt
, &pTransform
->pmt
);
155 static IPin
* WINAPI
TransformFilter_GetPin(BaseFilter
*iface
, int pos
)
157 TransformFilter
*This
= impl_from_BaseFilter(iface
);
159 if (pos
>= This
->npins
|| pos
< 0)
162 IPin_AddRef(This
->ppPins
[pos
]);
163 return This
->ppPins
[pos
];
166 static LONG WINAPI
TransformFilter_GetPinCount(BaseFilter
*iface
)
168 TransformFilter
*This
= impl_from_BaseFilter(iface
);
170 return (This
->npins
+1);
173 static const BaseFilterFuncTable tfBaseFuncTable
= {
174 TransformFilter_GetPin
,
175 TransformFilter_GetPinCount
178 static const BasePinFuncTable tf_input_BaseFuncTable
= {
179 TransformFilter_Input_CheckMediaType
,
181 BasePinImpl_GetMediaTypeVersion
,
182 BasePinImpl_GetMediaType
185 static const BaseInputPinFuncTable tf_input_BaseInputFuncTable
= {
186 TransformFilter_Input_Receive
189 static const BasePinFuncTable tf_output_BaseFuncTable
= {
191 BaseOutputPinImpl_AttemptConnection
,
192 BasePinImpl_GetMediaTypeVersion
,
193 TransformFilter_Output_GetMediaType
196 static const BaseOutputPinFuncTable tf_output_BaseOutputFuncTable
= {
197 TransformFilter_Output_DecideBufferSize
,
198 BaseOutputPinImpl_DecideAllocator
,
199 BaseOutputPinImpl_BreakConnect
202 static HRESULT
TransformFilter_Init(const IBaseFilterVtbl
*pVtbl
, const CLSID
* pClsid
, const TransformFilterFuncTable
* pFuncsTable
, TransformFilter
* pTransformFilter
)
208 BaseFilter_Init(&pTransformFilter
->filter
, pVtbl
, pClsid
, (DWORD_PTR
)(__FILE__
": TransformFilter.csFilter"), &tfBaseFuncTable
);
210 InitializeCriticalSection(&pTransformFilter
->csReceive
);
211 pTransformFilter
->csReceive
.DebugInfo
->Spare
[0] = (DWORD_PTR
)(__FILE__
": TransformFilter.csReceive");
213 /* pTransformFilter is already allocated */
214 pTransformFilter
->pFuncsTable
= pFuncsTable
;
215 ZeroMemory(&pTransformFilter
->pmt
, sizeof(pTransformFilter
->pmt
));
216 pTransformFilter
->npins
= 2;
218 pTransformFilter
->ppPins
= CoTaskMemAlloc(2 * sizeof(IPin
*));
220 /* construct input pin */
221 piInput
.dir
= PINDIR_INPUT
;
222 piInput
.pFilter
= &pTransformFilter
->filter
.IBaseFilter_iface
;
223 lstrcpynW(piInput
.achName
, wcsInputPinName
, sizeof(piInput
.achName
) / sizeof(piInput
.achName
[0]));
224 piOutput
.dir
= PINDIR_OUTPUT
;
225 piOutput
.pFilter
= &pTransformFilter
->filter
.IBaseFilter_iface
;
226 lstrcpynW(piOutput
.achName
, wcsOutputPinName
, sizeof(piOutput
.achName
) / sizeof(piOutput
.achName
[0]));
228 hr
= BaseInputPin_Construct(&TransformFilter_InputPin_Vtbl
, &piInput
, &tf_input_BaseFuncTable
, &tf_input_BaseInputFuncTable
, &pTransformFilter
->filter
.csFilter
, NULL
, &pTransformFilter
->ppPins
[0]);
232 hr
= BaseOutputPin_Construct(&TransformFilter_OutputPin_Vtbl
, sizeof(BaseOutputPin
), &piOutput
, &tf_output_BaseFuncTable
, &tf_output_BaseOutputFuncTable
, &pTransformFilter
->filter
.csFilter
, &pTransformFilter
->ppPins
[1]);
235 ERR("Cannot create output pin (%x)\n", hr
);
237 QualityControlImpl_Create( pTransformFilter
->ppPins
[0], &pTransformFilter
->filter
.IBaseFilter_iface
, &pTransformFilter
->qcimpl
);
238 pTransformFilter
->qcimpl
->IQualityControl_iface
.lpVtbl
= &TransformFilter_QualityControl_Vtbl
;
243 CoTaskMemFree(pTransformFilter
->ppPins
);
244 BaseFilterImpl_Release(&pTransformFilter
->filter
.IBaseFilter_iface
);
250 HRESULT
TransformFilter_Construct(const IBaseFilterVtbl
*pVtbl
, LONG filter_size
, const CLSID
* pClsid
, const TransformFilterFuncTable
* pFuncsTable
, IBaseFilter
** ppTransformFilter
)
252 TransformFilter
* pTf
;
254 *ppTransformFilter
= NULL
;
256 assert(filter_size
>= sizeof(TransformFilter
));
258 pTf
= CoTaskMemAlloc(filter_size
);
261 return E_OUTOFMEMORY
;
263 ZeroMemory(pTf
, filter_size
);
265 if (SUCCEEDED(TransformFilter_Init(pVtbl
, pClsid
, pFuncsTable
, pTf
)))
267 *ppTransformFilter
= &pTf
->filter
.IBaseFilter_iface
;
275 HRESULT WINAPI
TransformFilterImpl_QueryInterface(IBaseFilter
* iface
, REFIID riid
, LPVOID
* ppv
)
278 TransformFilter
*This
= impl_from_IBaseFilter(iface
);
279 TRACE("(%p/%p)->(%s, %p)\n", This
, iface
, debugstr_guid(riid
), ppv
);
281 if (IsEqualIID(riid
, &IID_IQualityControl
)) {
282 *ppv
= (IQualityControl
*)This
->qcimpl
;
283 IUnknown_AddRef((IUnknown
*)*ppv
);
286 hr
= BaseFilterImpl_QueryInterface(iface
, riid
, ppv
);
288 if (FAILED(hr
) && !IsEqualIID(riid
, &IID_IPin
) && !IsEqualIID(riid
, &IID_IVideoWindow
) &&
289 !IsEqualIID(riid
, &IID_IAMFilterMiscFlags
))
290 FIXME("No interface for %s!\n", debugstr_guid(riid
));
295 ULONG WINAPI
TransformFilterImpl_Release(IBaseFilter
* iface
)
297 TransformFilter
*This
= impl_from_IBaseFilter(iface
);
298 ULONG refCount
= BaseFilterImpl_Release(iface
);
300 TRACE("(%p/%p)->() Release from %d\n", This
, iface
, refCount
+ 1);
306 for (i
= 0; i
< This
->npins
; i
++)
310 if (SUCCEEDED(IPin_ConnectedTo(This
->ppPins
[i
], &pConnectedTo
)))
312 IPin_Disconnect(pConnectedTo
);
313 IPin_Release(pConnectedTo
);
315 IPin_Disconnect(This
->ppPins
[i
]);
317 IPin_Release(This
->ppPins
[i
]);
320 CoTaskMemFree(This
->ppPins
);
322 TRACE("Destroying transform filter\n");
323 This
->csReceive
.DebugInfo
->Spare
[0] = 0;
324 DeleteCriticalSection(&This
->csReceive
);
325 FreeMediaType(&This
->pmt
);
326 QualityControlImpl_Destroy(This
->qcimpl
);
335 /** IMediaFilter methods **/
337 HRESULT WINAPI
TransformFilterImpl_Stop(IBaseFilter
* iface
)
339 TransformFilter
*This
= impl_from_IBaseFilter(iface
);
342 TRACE("(%p/%p)\n", This
, iface
);
344 EnterCriticalSection(&This
->csReceive
);
346 This
->filter
.state
= State_Stopped
;
347 if (This
->pFuncsTable
->pfnStopStreaming
)
348 hr
= This
->pFuncsTable
->pfnStopStreaming(This
);
350 LeaveCriticalSection(&This
->csReceive
);
355 HRESULT WINAPI
TransformFilterImpl_Pause(IBaseFilter
* iface
)
357 TransformFilter
*This
= impl_from_IBaseFilter(iface
);
360 TRACE("(%p/%p)->()\n", This
, iface
);
362 EnterCriticalSection(&This
->csReceive
);
364 if (This
->filter
.state
== State_Stopped
)
365 hr
= IBaseFilter_Run(iface
, -1);
370 This
->filter
.state
= State_Paused
;
372 LeaveCriticalSection(&This
->csReceive
);
377 HRESULT WINAPI
TransformFilterImpl_Run(IBaseFilter
* iface
, REFERENCE_TIME tStart
)
380 TransformFilter
*This
= impl_from_IBaseFilter(iface
);
382 TRACE("(%p/%p)->(%s)\n", This
, iface
, wine_dbgstr_longlong(tStart
));
384 EnterCriticalSection(&This
->csReceive
);
386 if (This
->filter
.state
== State_Stopped
)
388 impl_BaseInputPin_from_IPin(This
->ppPins
[0])->end_of_stream
= 0;
389 if (This
->pFuncsTable
->pfnStartStreaming
)
390 hr
= This
->pFuncsTable
->pfnStartStreaming(This
);
392 hr
= BaseOutputPinImpl_Active(impl_BaseOutputPin_from_IPin(This
->ppPins
[1]));
397 This
->filter
.rtStreamStart
= tStart
;
398 This
->filter
.state
= State_Running
;
401 LeaveCriticalSection(&This
->csReceive
);
406 HRESULT WINAPI
TransformFilterImpl_Notify(TransformFilter
*iface
, IBaseFilter
*sender
, Quality qm
)
408 return QualityControlImpl_Notify((IQualityControl
*)iface
->qcimpl
, sender
, qm
);
411 /** IBaseFilter implementation **/
413 HRESULT WINAPI
TransformFilterImpl_FindPin(IBaseFilter
* iface
, LPCWSTR Id
, IPin
**ppPin
)
415 TransformFilter
*This
= impl_from_IBaseFilter(iface
);
417 TRACE("(%p/%p)->(%s,%p)\n", This
, iface
, debugstr_w(Id
), ppPin
);
422 static const IBaseFilterVtbl TransformFilter_Vtbl
=
424 TransformFilterImpl_QueryInterface
,
425 BaseFilterImpl_AddRef
,
426 TransformFilterImpl_Release
,
427 BaseFilterImpl_GetClassID
,
428 TransformFilterImpl_Stop
,
429 TransformFilterImpl_Pause
,
430 TransformFilterImpl_Run
,
431 BaseFilterImpl_GetState
,
432 BaseFilterImpl_SetSyncSource
,
433 BaseFilterImpl_GetSyncSource
,
434 BaseFilterImpl_EnumPins
,
435 TransformFilterImpl_FindPin
,
436 BaseFilterImpl_QueryFilterInfo
,
437 BaseFilterImpl_JoinFilterGraph
,
438 BaseFilterImpl_QueryVendorInfo
441 static HRESULT WINAPI
TransformFilter_InputPin_EndOfStream(IPin
* iface
)
443 BaseInputPin
* This
= impl_BaseInputPin_from_IPin(iface
);
444 TransformFilter
* pTransform
;
448 TRACE("(%p)->()\n", iface
);
450 /* Since we process samples synchronously, just forward notification downstream */
451 pTransform
= impl_from_IBaseFilter(This
->pin
.pinInfo
.pFilter
);
455 hr
= IPin_ConnectedTo(pTransform
->ppPins
[1], &ppin
);
458 hr
= IPin_EndOfStream(ppin
);
467 static HRESULT WINAPI
TransformFilter_InputPin_ReceiveConnection(IPin
* iface
, IPin
* pReceivePin
, const AM_MEDIA_TYPE
* pmt
)
469 BaseInputPin
* This
= impl_BaseInputPin_from_IPin(iface
);
470 TransformFilter
* pTransform
;
473 TRACE("(%p)->(%p, %p)\n", iface
, pReceivePin
, pmt
);
475 pTransform
= impl_from_IBaseFilter(This
->pin
.pinInfo
.pFilter
);
477 if (pTransform
->pFuncsTable
->pfnSetMediaType
)
478 hr
= pTransform
->pFuncsTable
->pfnSetMediaType(pTransform
, PINDIR_INPUT
, pmt
);
480 if (SUCCEEDED(hr
) && pTransform
->pFuncsTable
->pfnCompleteConnect
)
481 hr
= pTransform
->pFuncsTable
->pfnCompleteConnect(pTransform
, PINDIR_INPUT
, pReceivePin
);
485 hr
= BaseInputPinImpl_ReceiveConnection(iface
, pReceivePin
, pmt
);
486 if (FAILED(hr
) && pTransform
->pFuncsTable
->pfnBreakConnect
)
487 pTransform
->pFuncsTable
->pfnBreakConnect(pTransform
, PINDIR_INPUT
);
493 static HRESULT WINAPI
TransformFilter_InputPin_Disconnect(IPin
* iface
)
495 BaseInputPin
* This
= impl_BaseInputPin_from_IPin(iface
);
496 TransformFilter
* pTransform
;
498 TRACE("(%p)->()\n", iface
);
500 pTransform
= impl_from_IBaseFilter(This
->pin
.pinInfo
.pFilter
);
501 if (pTransform
->pFuncsTable
->pfnBreakConnect
)
502 pTransform
->pFuncsTable
->pfnBreakConnect(pTransform
, PINDIR_INPUT
);
504 return BasePinImpl_Disconnect(iface
);
507 static HRESULT WINAPI
TransformFilter_InputPin_BeginFlush(IPin
* iface
)
509 BaseInputPin
* This
= impl_BaseInputPin_from_IPin(iface
);
510 TransformFilter
* pTransform
;
513 TRACE("(%p)->()\n", iface
);
515 pTransform
= impl_from_IBaseFilter(This
->pin
.pinInfo
.pFilter
);
516 EnterCriticalSection(&pTransform
->filter
.csFilter
);
517 if (pTransform
->pFuncsTable
->pfnBeginFlush
)
518 hr
= pTransform
->pFuncsTable
->pfnBeginFlush(pTransform
);
520 hr
= BaseInputPinImpl_BeginFlush(iface
);
521 LeaveCriticalSection(&pTransform
->filter
.csFilter
);
525 static HRESULT WINAPI
TransformFilter_InputPin_EndFlush(IPin
* iface
)
527 BaseInputPin
* This
= impl_BaseInputPin_from_IPin(iface
);
528 TransformFilter
* pTransform
;
531 TRACE("(%p)->()\n", iface
);
533 pTransform
= impl_from_IBaseFilter(This
->pin
.pinInfo
.pFilter
);
534 EnterCriticalSection(&pTransform
->filter
.csFilter
);
535 if (pTransform
->pFuncsTable
->pfnEndFlush
)
536 hr
= pTransform
->pFuncsTable
->pfnEndFlush(pTransform
);
538 hr
= BaseInputPinImpl_EndFlush(iface
);
539 LeaveCriticalSection(&pTransform
->filter
.csFilter
);
543 static HRESULT WINAPI
TransformFilter_InputPin_NewSegment(IPin
* iface
, REFERENCE_TIME tStart
, REFERENCE_TIME tStop
, double dRate
)
545 BaseInputPin
* This
= impl_BaseInputPin_from_IPin(iface
);
546 TransformFilter
* pTransform
;
549 TRACE("(%p)->()\n", iface
);
551 pTransform
= impl_from_IBaseFilter(This
->pin
.pinInfo
.pFilter
);
552 EnterCriticalSection(&pTransform
->filter
.csFilter
);
553 if (pTransform
->pFuncsTable
->pfnNewSegment
)
554 hr
= pTransform
->pFuncsTable
->pfnNewSegment(pTransform
, tStart
, tStop
, dRate
);
556 hr
= BaseInputPinImpl_NewSegment(iface
, tStart
, tStop
, dRate
);
557 LeaveCriticalSection(&pTransform
->filter
.csFilter
);
561 static const IPinVtbl TransformFilter_InputPin_Vtbl
=
563 BaseInputPinImpl_QueryInterface
,
565 BaseInputPinImpl_Release
,
566 BaseInputPinImpl_Connect
,
567 TransformFilter_InputPin_ReceiveConnection
,
568 TransformFilter_InputPin_Disconnect
,
569 BasePinImpl_ConnectedTo
,
570 BasePinImpl_ConnectionMediaType
,
571 BasePinImpl_QueryPinInfo
,
572 BasePinImpl_QueryDirection
,
574 BaseInputPinImpl_QueryAccept
,
575 BasePinImpl_EnumMediaTypes
,
576 BasePinImpl_QueryInternalConnections
,
577 TransformFilter_InputPin_EndOfStream
,
578 TransformFilter_InputPin_BeginFlush
,
579 TransformFilter_InputPin_EndFlush
,
580 TransformFilter_InputPin_NewSegment
583 static const IPinVtbl TransformFilter_OutputPin_Vtbl
=
585 BaseOutputPinImpl_QueryInterface
,
587 BaseOutputPinImpl_Release
,
588 BaseOutputPinImpl_Connect
,
589 BaseOutputPinImpl_ReceiveConnection
,
590 BaseOutputPinImpl_Disconnect
,
591 BasePinImpl_ConnectedTo
,
592 BasePinImpl_ConnectionMediaType
,
593 BasePinImpl_QueryPinInfo
,
594 BasePinImpl_QueryDirection
,
596 TransformFilter_Output_QueryAccept
,
597 BasePinImpl_EnumMediaTypes
,
598 BasePinImpl_QueryInternalConnections
,
599 BaseOutputPinImpl_EndOfStream
,
600 BaseOutputPinImpl_BeginFlush
,
601 BaseOutputPinImpl_EndFlush
,
602 BasePinImpl_NewSegment
605 static HRESULT WINAPI
TransformFilter_QualityControlImpl_Notify(IQualityControl
*iface
, IBaseFilter
*sender
, Quality qm
) {
606 QualityControlImpl
*qc
= (QualityControlImpl
*)iface
;
607 TransformFilter
*This
= impl_from_IBaseFilter(qc
->self
);
609 if (This
->pFuncsTable
->pfnNotify
)
610 return This
->pFuncsTable
->pfnNotify(This
, sender
, qm
);
612 return TransformFilterImpl_Notify(This
, sender
, qm
);
615 static const IQualityControlVtbl TransformFilter_QualityControl_Vtbl
= {
616 QualityControlImpl_QueryInterface
,
617 QualityControlImpl_AddRef
,
618 QualityControlImpl_Release
,
619 TransformFilter_QualityControlImpl_Notify
,
620 QualityControlImpl_SetSink