strmbase: Implement BaseControlWindow.
[wine/multimedia.git] / dlls / strmbase / transform.c
blob493324d2446ad4b62b7232cdbf50420bb284e128
1 /*
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
21 #include "config.h"
22 #include <stdarg.h>
24 #define COBJMACROS
26 #include "windef.h"
27 #include "winbase.h"
28 #include "dshow.h"
29 #include "amvideo.h"
30 #include "strmif.h"
31 #include "vfw.h"
33 #include <assert.h>
35 #include "wine/unicode.h"
36 #include "wine/debug.h"
37 #include "wine/strmbase.h"
39 WINE_DEFAULT_DEBUG_CHANNEL(strmbase);
41 static const WCHAR wcsInputPinName[] = {'i','n','p','u','t',' ','p','i','n',0};
42 static const WCHAR wcsOutputPinName[] = {'o','u','t','p','u','t',' ','p','i','n',0};
44 static const IBaseFilterVtbl TransformFilter_Vtbl;
45 static const IPinVtbl TransformFilter_InputPin_Vtbl;
46 static const IPinVtbl TransformFilter_OutputPin_Vtbl;
47 static const IQualityControlVtbl TransformFilter_QualityControl_Vtbl;
49 static inline BaseInputPin *impl_BaseInputPin_from_BasePin( BasePin *iface )
51 return CONTAINING_RECORD(iface, BaseInputPin, pin);
54 static inline BasePin *impl_BasePin_from_IPin( IPin *iface )
56 return CONTAINING_RECORD(iface, BasePin, IPin_iface);
59 static inline BaseInputPin *impl_BaseInputPin_from_IPin( IPin *iface )
61 return CONTAINING_RECORD(iface, BaseInputPin, pin.IPin_iface);
64 static inline BaseOutputPin *impl_BaseOutputPin_from_IPin( IPin *iface )
66 return CONTAINING_RECORD(iface, BaseOutputPin, pin.IPin_iface);
69 static inline TransformFilter *impl_from_IBaseFilter( IBaseFilter *iface )
71 return CONTAINING_RECORD(iface, TransformFilter, filter.IBaseFilter_iface);
74 static inline TransformFilter *impl_from_BaseFilter( BaseFilter *iface )
76 return CONTAINING_RECORD(iface, TransformFilter, filter);
79 static HRESULT WINAPI TransformFilter_Input_CheckMediaType(BasePin *iface, const AM_MEDIA_TYPE * pmt)
81 BaseInputPin* This = impl_BaseInputPin_from_BasePin(iface);
82 TransformFilter * pTransform;
84 TRACE("%p\n", iface);
85 pTransform = impl_from_IBaseFilter(This->pin.pinInfo.pFilter);
87 if (pTransform->pFuncsTable->pfnCheckInputType)
88 return pTransform->pFuncsTable->pfnCheckInputType(pTransform, pmt);
89 /* Assume OK if there's no query method (the connection will fail if
90 needed) */
91 return S_OK;
94 static HRESULT WINAPI TransformFilter_Input_Receive(BaseInputPin *This, IMediaSample *pInSample)
96 HRESULT hr = S_FALSE;
97 TransformFilter * pTransform;
98 TRACE("%p\n", This);
99 pTransform = impl_from_IBaseFilter(This->pin.pinInfo.pFilter);
101 EnterCriticalSection(&pTransform->csReceive);
102 if (pTransform->filter.state == State_Stopped)
104 LeaveCriticalSection(&pTransform->csReceive);
105 return VFW_E_WRONG_STATE;
108 if (This->end_of_stream || This->flushing)
110 LeaveCriticalSection(&pTransform->csReceive);
111 return S_FALSE;
114 LeaveCriticalSection(&pTransform->csReceive);
115 if (pTransform->pFuncsTable->pfnReceive)
116 hr = pTransform->pFuncsTable->pfnReceive(pTransform, pInSample);
117 else
118 hr = S_FALSE;
120 return hr;
123 static HRESULT WINAPI TransformFilter_Output_QueryAccept(IPin *iface, const AM_MEDIA_TYPE * pmt)
125 BasePin *This = impl_BasePin_from_IPin(iface);
126 TransformFilter *pTransformFilter = impl_from_IBaseFilter(This->pinInfo.pFilter);
127 AM_MEDIA_TYPE* outpmt = &pTransformFilter->pmt;
128 TRACE("%p\n", iface);
130 if (IsEqualIID(&pmt->majortype, &outpmt->majortype)
131 && (IsEqualIID(&pmt->subtype, &outpmt->subtype) || IsEqualIID(&outpmt->subtype, &GUID_NULL)))
132 return S_OK;
133 return S_FALSE;
136 static HRESULT WINAPI TransformFilter_Output_DecideBufferSize(BaseOutputPin *This, IMemAllocator *pAlloc, ALLOCATOR_PROPERTIES *ppropInputRequest)
138 TransformFilter *pTransformFilter = impl_from_IBaseFilter(This->pin.pinInfo.pFilter);
139 return pTransformFilter->pFuncsTable->pfnDecideBufferSize(pTransformFilter, pAlloc, ppropInputRequest);
142 static HRESULT WINAPI TransformFilter_Output_GetMediaType(BasePin *This, int iPosition, AM_MEDIA_TYPE *pmt)
144 TransformFilter *pTransform = impl_from_IBaseFilter(This->pinInfo.pFilter);
146 if (iPosition < 0)
147 return E_INVALIDARG;
148 if (iPosition > 0)
149 return VFW_S_NO_MORE_ITEMS;
150 CopyMediaType(pmt, &pTransform->pmt);
151 return S_OK;
154 static IPin* WINAPI TransformFilter_GetPin(BaseFilter *iface, int pos)
156 TransformFilter *This = impl_from_BaseFilter(iface);
158 if (pos >= This->npins || pos < 0)
159 return NULL;
161 IPin_AddRef(This->ppPins[pos]);
162 return This->ppPins[pos];
165 static LONG WINAPI TransformFilter_GetPinCount(BaseFilter *iface)
167 TransformFilter *This = impl_from_BaseFilter(iface);
169 return (This->npins+1);
172 static const BaseFilterFuncTable tfBaseFuncTable = {
173 TransformFilter_GetPin,
174 TransformFilter_GetPinCount
177 static const BasePinFuncTable tf_input_BaseFuncTable = {
178 TransformFilter_Input_CheckMediaType,
179 NULL,
180 BasePinImpl_GetMediaTypeVersion,
181 BasePinImpl_GetMediaType
184 static const BaseInputPinFuncTable tf_input_BaseInputFuncTable = {
185 TransformFilter_Input_Receive
188 static const BasePinFuncTable tf_output_BaseFuncTable = {
189 NULL,
190 BaseOutputPinImpl_AttemptConnection,
191 BasePinImpl_GetMediaTypeVersion,
192 TransformFilter_Output_GetMediaType
195 static const BaseOutputPinFuncTable tf_output_BaseOutputFuncTable = {
196 TransformFilter_Output_DecideBufferSize,
197 BaseOutputPinImpl_DecideAllocator,
198 BaseOutputPinImpl_BreakConnect
201 static HRESULT TransformFilter_Init(const IBaseFilterVtbl *pVtbl, const CLSID* pClsid, const TransformFilterFuncTable* pFuncsTable, TransformFilter* pTransformFilter)
203 HRESULT hr;
204 PIN_INFO piInput;
205 PIN_INFO piOutput;
207 BaseFilter_Init(&pTransformFilter->filter, pVtbl, pClsid, (DWORD_PTR)(__FILE__ ": TransformFilter.csFilter"), &tfBaseFuncTable);
209 InitializeCriticalSection(&pTransformFilter->csReceive);
210 pTransformFilter->csReceive.DebugInfo->Spare[0] = (DWORD_PTR)(__FILE__": TransformFilter.csReceive");
212 /* pTransformFilter is already allocated */
213 pTransformFilter->pFuncsTable = pFuncsTable;
214 ZeroMemory(&pTransformFilter->pmt, sizeof(pTransformFilter->pmt));
215 pTransformFilter->npins = 2;
217 pTransformFilter->ppPins = CoTaskMemAlloc(2 * sizeof(IPin *));
219 /* construct input pin */
220 piInput.dir = PINDIR_INPUT;
221 piInput.pFilter = &pTransformFilter->filter.IBaseFilter_iface;
222 lstrcpynW(piInput.achName, wcsInputPinName, sizeof(piInput.achName) / sizeof(piInput.achName[0]));
223 piOutput.dir = PINDIR_OUTPUT;
224 piOutput.pFilter = &pTransformFilter->filter.IBaseFilter_iface;
225 lstrcpynW(piOutput.achName, wcsOutputPinName, sizeof(piOutput.achName) / sizeof(piOutput.achName[0]));
227 hr = BaseInputPin_Construct(&TransformFilter_InputPin_Vtbl, &piInput, &tf_input_BaseFuncTable, &tf_input_BaseInputFuncTable, &pTransformFilter->filter.csFilter, NULL, &pTransformFilter->ppPins[0]);
229 if (SUCCEEDED(hr))
231 hr = BaseOutputPin_Construct(&TransformFilter_OutputPin_Vtbl, sizeof(BaseOutputPin), &piOutput, &tf_output_BaseFuncTable, &tf_output_BaseOutputFuncTable, &pTransformFilter->filter.csFilter, &pTransformFilter->ppPins[1]);
233 if (FAILED(hr))
234 ERR("Cannot create output pin (%x)\n", hr);
235 else {
236 QualityControlImpl_init(&pTransformFilter->qcimpl, pTransformFilter->ppPins[0], &pTransformFilter->filter.IBaseFilter_iface);
237 pTransformFilter->qcimpl.lpVtbl = &TransformFilter_QualityControl_Vtbl;
240 if (FAILED(hr))
242 CoTaskMemFree(pTransformFilter->ppPins);
243 BaseFilterImpl_Release(&pTransformFilter->filter.IBaseFilter_iface);
246 return hr;
249 HRESULT TransformFilter_Construct(const IBaseFilterVtbl *pVtbl, LONG filter_size, const CLSID* pClsid, const TransformFilterFuncTable* pFuncsTable, IBaseFilter ** ppTransformFilter)
251 TransformFilter* pTf;
253 *ppTransformFilter = NULL;
255 assert(filter_size >= sizeof(TransformFilter));
257 pTf = CoTaskMemAlloc(filter_size);
259 if (!pTf)
260 return E_OUTOFMEMORY;
262 ZeroMemory(pTf, filter_size);
264 if (SUCCEEDED(TransformFilter_Init(pVtbl, pClsid, pFuncsTable, pTf)))
266 *ppTransformFilter = &pTf->filter.IBaseFilter_iface;
267 return S_OK;
270 CoTaskMemFree(pTf);
271 return E_FAIL;
274 HRESULT WINAPI TransformFilterImpl_QueryInterface(IBaseFilter * iface, REFIID riid, LPVOID * ppv)
276 HRESULT hr;
277 TransformFilter *This = impl_from_IBaseFilter(iface);
278 TRACE("(%p/%p)->(%s, %p)\n", This, iface, debugstr_guid(riid), ppv);
280 if (IsEqualIID(riid, &IID_IQualityControl)) {
281 *ppv = (IQualityControl*)&This->qcimpl;
282 IUnknown_AddRef((IUnknown*)*ppv);
283 return S_OK;
285 hr = BaseFilterImpl_QueryInterface(iface, riid, ppv);
287 if (FAILED(hr) && !IsEqualIID(riid, &IID_IPin) && !IsEqualIID(riid, &IID_IVideoWindow) &&
288 !IsEqualIID(riid, &IID_IAMFilterMiscFlags))
289 FIXME("No interface for %s!\n", debugstr_guid(riid));
291 return hr;
294 ULONG WINAPI TransformFilterImpl_Release(IBaseFilter * iface)
296 TransformFilter *This = impl_from_IBaseFilter(iface);
297 ULONG refCount = BaseFilterImpl_Release(iface);
299 TRACE("(%p/%p)->() Release from %d\n", This, iface, refCount + 1);
301 if (!refCount)
303 ULONG i;
305 for (i = 0; i < This->npins; i++)
307 IPin *pConnectedTo;
309 if (SUCCEEDED(IPin_ConnectedTo(This->ppPins[i], &pConnectedTo)))
311 IPin_Disconnect(pConnectedTo);
312 IPin_Release(pConnectedTo);
314 IPin_Disconnect(This->ppPins[i]);
316 IPin_Release(This->ppPins[i]);
319 CoTaskMemFree(This->ppPins);
321 TRACE("Destroying transform filter\n");
322 This->csReceive.DebugInfo->Spare[0] = 0;
323 DeleteCriticalSection(&This->csReceive);
324 FreeMediaType(&This->pmt);
325 CoTaskMemFree(This);
327 return 0;
329 else
330 return refCount;
333 /** IMediaFilter methods **/
335 HRESULT WINAPI TransformFilterImpl_Stop(IBaseFilter * iface)
337 TransformFilter *This = impl_from_IBaseFilter(iface);
338 HRESULT hr = S_OK;
340 TRACE("(%p/%p)\n", This, iface);
342 EnterCriticalSection(&This->csReceive);
344 This->filter.state = State_Stopped;
345 if (This->pFuncsTable->pfnStopStreaming)
346 hr = This->pFuncsTable->pfnStopStreaming(This);
348 LeaveCriticalSection(&This->csReceive);
350 return hr;
353 HRESULT WINAPI TransformFilterImpl_Pause(IBaseFilter * iface)
355 TransformFilter *This = impl_from_IBaseFilter(iface);
356 HRESULT hr;
358 TRACE("(%p/%p)->()\n", This, iface);
360 EnterCriticalSection(&This->csReceive);
362 if (This->filter.state == State_Stopped)
363 hr = IBaseFilter_Run(iface, -1);
364 else
365 hr = S_OK;
367 if (SUCCEEDED(hr))
368 This->filter.state = State_Paused;
370 LeaveCriticalSection(&This->csReceive);
372 return hr;
375 HRESULT WINAPI TransformFilterImpl_Run(IBaseFilter * iface, REFERENCE_TIME tStart)
377 HRESULT hr = S_OK;
378 TransformFilter *This = impl_from_IBaseFilter(iface);
380 TRACE("(%p/%p)->(%s)\n", This, iface, wine_dbgstr_longlong(tStart));
382 EnterCriticalSection(&This->csReceive);
384 if (This->filter.state == State_Stopped)
386 impl_BaseInputPin_from_IPin(This->ppPins[0])->end_of_stream = 0;
387 if (This->pFuncsTable->pfnStartStreaming)
388 hr = This->pFuncsTable->pfnStartStreaming(This);
389 if (SUCCEEDED(hr))
390 hr = BaseOutputPinImpl_Active(impl_BaseOutputPin_from_IPin(This->ppPins[1]));
393 if (SUCCEEDED(hr))
395 This->filter.rtStreamStart = tStart;
396 This->filter.state = State_Running;
399 LeaveCriticalSection(&This->csReceive);
401 return hr;
404 /** IBaseFilter implementation **/
406 HRESULT WINAPI TransformFilterImpl_FindPin(IBaseFilter * iface, LPCWSTR Id, IPin **ppPin)
408 TransformFilter *This = impl_from_IBaseFilter(iface);
410 TRACE("(%p/%p)->(%s,%p)\n", This, iface, debugstr_w(Id), ppPin);
412 return E_NOTIMPL;
415 static const IBaseFilterVtbl TransformFilter_Vtbl =
417 TransformFilterImpl_QueryInterface,
418 BaseFilterImpl_AddRef,
419 TransformFilterImpl_Release,
420 BaseFilterImpl_GetClassID,
421 TransformFilterImpl_Stop,
422 TransformFilterImpl_Pause,
423 TransformFilterImpl_Run,
424 BaseFilterImpl_GetState,
425 BaseFilterImpl_SetSyncSource,
426 BaseFilterImpl_GetSyncSource,
427 BaseFilterImpl_EnumPins,
428 TransformFilterImpl_FindPin,
429 BaseFilterImpl_QueryFilterInfo,
430 BaseFilterImpl_JoinFilterGraph,
431 BaseFilterImpl_QueryVendorInfo
434 static HRESULT WINAPI TransformFilter_InputPin_EndOfStream(IPin * iface)
436 BaseInputPin* This = impl_BaseInputPin_from_IPin(iface);
437 TransformFilter* pTransform;
438 IPin* ppin;
439 HRESULT hr;
441 TRACE("(%p)->()\n", iface);
443 /* Since we process samples synchronously, just forward notification downstream */
444 pTransform = impl_from_IBaseFilter(This->pin.pinInfo.pFilter);
445 if (!pTransform)
446 hr = E_FAIL;
447 else
448 hr = IPin_ConnectedTo(pTransform->ppPins[1], &ppin);
449 if (SUCCEEDED(hr))
451 hr = IPin_EndOfStream(ppin);
452 IPin_Release(ppin);
455 if (FAILED(hr))
456 ERR("%x\n", hr);
457 return hr;
460 static HRESULT WINAPI TransformFilter_InputPin_ReceiveConnection(IPin * iface, IPin * pReceivePin, const AM_MEDIA_TYPE * pmt)
462 BaseInputPin* This = impl_BaseInputPin_from_IPin(iface);
463 TransformFilter* pTransform;
464 HRESULT hr = S_OK;
466 TRACE("(%p)->(%p, %p)\n", iface, pReceivePin, pmt);
468 pTransform = impl_from_IBaseFilter(This->pin.pinInfo.pFilter);
470 if (pTransform->pFuncsTable->pfnSetMediaType)
471 hr = pTransform->pFuncsTable->pfnSetMediaType(pTransform, PINDIR_INPUT, pmt);
473 if (SUCCEEDED(hr) && pTransform->pFuncsTable->pfnCompleteConnect)
474 hr = pTransform->pFuncsTable->pfnCompleteConnect(pTransform, PINDIR_INPUT, pReceivePin);
476 if (SUCCEEDED(hr))
478 hr = BaseInputPinImpl_ReceiveConnection(iface, pReceivePin, pmt);
479 if (FAILED(hr) && pTransform->pFuncsTable->pfnBreakConnect)
480 pTransform->pFuncsTable->pfnBreakConnect(pTransform, PINDIR_INPUT);
483 return hr;
486 static HRESULT WINAPI TransformFilter_InputPin_Disconnect(IPin * iface)
488 BaseInputPin* This = impl_BaseInputPin_from_IPin(iface);
489 TransformFilter* pTransform;
491 TRACE("(%p)->()\n", iface);
493 pTransform = impl_from_IBaseFilter(This->pin.pinInfo.pFilter);
494 if (pTransform->pFuncsTable->pfnBreakConnect)
495 pTransform->pFuncsTable->pfnBreakConnect(pTransform, PINDIR_INPUT);
497 return BasePinImpl_Disconnect(iface);
500 static HRESULT WINAPI TransformFilter_InputPin_BeginFlush(IPin * iface)
502 BaseInputPin* This = impl_BaseInputPin_from_IPin(iface);
503 TransformFilter* pTransform;
504 HRESULT hr = S_OK;
506 TRACE("(%p)->()\n", iface);
508 pTransform = impl_from_IBaseFilter(This->pin.pinInfo.pFilter);
509 EnterCriticalSection(&pTransform->filter.csFilter);
510 if (pTransform->pFuncsTable->pfnBeginFlush)
511 hr = pTransform->pFuncsTable->pfnBeginFlush(pTransform);
512 if (SUCCEEDED(hr))
513 hr = BaseInputPinImpl_BeginFlush(iface);
514 LeaveCriticalSection(&pTransform->filter.csFilter);
515 return hr;
518 static HRESULT WINAPI TransformFilter_InputPin_EndFlush(IPin * iface)
520 BaseInputPin* This = impl_BaseInputPin_from_IPin(iface);
521 TransformFilter* pTransform;
522 HRESULT hr = S_OK;
524 TRACE("(%p)->()\n", iface);
526 pTransform = impl_from_IBaseFilter(This->pin.pinInfo.pFilter);
527 EnterCriticalSection(&pTransform->filter.csFilter);
528 if (pTransform->pFuncsTable->pfnEndFlush)
529 hr = pTransform->pFuncsTable->pfnEndFlush(pTransform);
530 if (SUCCEEDED(hr))
531 hr = BaseInputPinImpl_EndFlush(iface);
532 LeaveCriticalSection(&pTransform->filter.csFilter);
533 return hr;
536 static HRESULT WINAPI TransformFilter_InputPin_NewSegment(IPin * iface, REFERENCE_TIME tStart, REFERENCE_TIME tStop, double dRate)
538 BaseInputPin* This = impl_BaseInputPin_from_IPin(iface);
539 TransformFilter* pTransform;
540 HRESULT hr = S_OK;
542 TRACE("(%p)->()\n", iface);
544 pTransform = impl_from_IBaseFilter(This->pin.pinInfo.pFilter);
545 EnterCriticalSection(&pTransform->filter.csFilter);
546 if (pTransform->pFuncsTable->pfnNewSegment)
547 hr = pTransform->pFuncsTable->pfnNewSegment(pTransform, tStart, tStop, dRate);
548 if (SUCCEEDED(hr))
549 hr = BaseInputPinImpl_NewSegment(iface, tStart, tStop, dRate);
550 LeaveCriticalSection(&pTransform->filter.csFilter);
551 return hr;
554 static const IPinVtbl TransformFilter_InputPin_Vtbl =
556 BaseInputPinImpl_QueryInterface,
557 BasePinImpl_AddRef,
558 BaseInputPinImpl_Release,
559 BaseInputPinImpl_Connect,
560 TransformFilter_InputPin_ReceiveConnection,
561 TransformFilter_InputPin_Disconnect,
562 BasePinImpl_ConnectedTo,
563 BasePinImpl_ConnectionMediaType,
564 BasePinImpl_QueryPinInfo,
565 BasePinImpl_QueryDirection,
566 BasePinImpl_QueryId,
567 BaseInputPinImpl_QueryAccept,
568 BasePinImpl_EnumMediaTypes,
569 BasePinImpl_QueryInternalConnections,
570 TransformFilter_InputPin_EndOfStream,
571 TransformFilter_InputPin_BeginFlush,
572 TransformFilter_InputPin_EndFlush,
573 TransformFilter_InputPin_NewSegment
576 static const IPinVtbl TransformFilter_OutputPin_Vtbl =
578 BaseOutputPinImpl_QueryInterface,
579 BasePinImpl_AddRef,
580 BaseOutputPinImpl_Release,
581 BaseOutputPinImpl_Connect,
582 BaseOutputPinImpl_ReceiveConnection,
583 BaseOutputPinImpl_Disconnect,
584 BasePinImpl_ConnectedTo,
585 BasePinImpl_ConnectionMediaType,
586 BasePinImpl_QueryPinInfo,
587 BasePinImpl_QueryDirection,
588 BasePinImpl_QueryId,
589 TransformFilter_Output_QueryAccept,
590 BasePinImpl_EnumMediaTypes,
591 BasePinImpl_QueryInternalConnections,
592 BaseOutputPinImpl_EndOfStream,
593 BaseOutputPinImpl_BeginFlush,
594 BaseOutputPinImpl_EndFlush,
595 BasePinImpl_NewSegment
598 static HRESULT WINAPI TransformFilter_QualityControlImpl_Notify(IQualityControl *iface, IBaseFilter *sender, Quality qm) {
599 QualityControlImpl *qc = (QualityControlImpl*)iface;
600 TransformFilter *This = impl_from_IBaseFilter(qc->self);
602 if (This->pFuncsTable->pfnNotify)
603 return This->pFuncsTable->pfnNotify(This, sender, qm);
604 else
605 return QualityControlImpl_Notify(iface, sender, qm);
608 static const IQualityControlVtbl TransformFilter_QualityControl_Vtbl = {
609 QualityControlImpl_QueryInterface,
610 QualityControlImpl_AddRef,
611 QualityControlImpl_Release,
612 TransformFilter_QualityControlImpl_Notify,
613 QualityControlImpl_SetSink