2 * Parser (Base for parsers and splitters)
4 * Copyright 2003 Robert Shearman
5 * Copyright 2004-2005 Christian Costa
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., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
22 #include "quartz_private.h"
23 #include "control_private.h"
34 #include "wine/unicode.h"
35 #include "wine/debug.h"
42 WINE_DEFAULT_DEBUG_CHANNEL(quartz);
44 static const WCHAR wcsInputPinName[] = {'i','n','p','u','t',' ','p','i','n',0};
45 static const IBaseFilterVtbl Parser_Vtbl;
46 static const IMediaSeekingVtbl Parser_Seeking_Vtbl;
47 static const IPinVtbl Parser_OutputPin_Vtbl;
48 static const IPinVtbl Parser_InputPin_Vtbl;
50 static HRESULT Parser_OutputPin_QueryAccept(LPVOID iface, const AM_MEDIA_TYPE * pmt);
51 static HRESULT Parser_ChangeStart(LPVOID iface);
52 static HRESULT Parser_ChangeStop(LPVOID iface);
53 static HRESULT Parser_ChangeRate(LPVOID iface);
55 static HRESULT Parser_InputPin_Construct(const PIN_INFO * pPinInfo, SAMPLEPROC pSampleProc, LPVOID pUserData, QUERYACCEPTPROC pQueryAccept, LPCRITICAL_SECTION pCritSec, IPin ** ppPin);
57 #define _IMediaSeeking_Offset ((int)(&(((Parser_OutputPin*)0)->mediaSeeking)))
58 #define ICOM_THIS_From_IMediaSeeking(impl, iface) impl* This = (impl*)(((char*)iface)-_IMediaSeeking_Offset);
60 HRESULT Parser_Create(ParserImpl* pParser, const CLSID* pClsid, PFN_PROCESS_SAMPLE fnProcessSample, PFN_QUERY_ACCEPT fnQueryAccept, PFN_PRE_CONNECT fnPreConnect)
65 /* pTransformFilter is already allocated */
66 pParser->clsid = *pClsid;
68 pParser->lpVtbl = &Parser_Vtbl;
69 pParser->refCount = 1;
70 InitializeCriticalSection(&pParser->csFilter);
71 pParser->state = State_Stopped;
72 pParser->pClock = NULL;
73 ZeroMemory(&pParser->filterInfo, sizeof(FILTER_INFO));
75 pParser->cStreams = 0;
76 pParser->ppPins = CoTaskMemAlloc(1 * sizeof(IPin *));
78 /* construct input pin */
79 piInput.dir = PINDIR_INPUT;
80 piInput.pFilter = (IBaseFilter *)pParser;
81 lstrcpynW(piInput.achName, wcsInputPinName, sizeof(piInput.achName) / sizeof(piInput.achName[0]));
83 hr = Parser_InputPin_Construct(&piInput, fnProcessSample, (LPVOID)pParser, fnQueryAccept, &pParser->csFilter, (IPin **)&pParser->pInputPin);
87 pParser->ppPins[0] = (IPin *)pParser->pInputPin;
88 pParser->pInputPin->fnPreConnect = fnPreConnect;
92 CoTaskMemFree(pParser->ppPins);
93 DeleteCriticalSection(&pParser->csFilter);
94 CoTaskMemFree(pParser);
100 static HRESULT Parser_OutputPin_Init(const PIN_INFO * pPinInfo, ALLOCATOR_PROPERTIES * props, LPVOID pUserData, QUERYACCEPTPROC pQueryAccept, const AM_MEDIA_TYPE * pmt, float fSamplesPerSec, LPCRITICAL_SECTION pCritSec, Parser_OutputPin * pPinImpl)
102 pPinImpl->pmt = CoTaskMemAlloc(sizeof(AM_MEDIA_TYPE));
103 CopyMediaType(pPinImpl->pmt, pmt);
104 pPinImpl->dwSamplesProcessed = 0;
105 pPinImpl->dwSampleSize = 0;
106 pPinImpl->fSamplesPerSec = fSamplesPerSec;
108 MediaSeekingImpl_Init((LPVOID)pPinInfo->pFilter, Parser_ChangeStop, Parser_ChangeStart, Parser_ChangeRate, &pPinImpl->mediaSeeking);
109 pPinImpl->mediaSeeking.lpVtbl = &Parser_Seeking_Vtbl;
111 return OutputPin_Init(pPinInfo, props, pUserData, pQueryAccept, pCritSec, &pPinImpl->pin);
114 static HRESULT Parser_OutputPin_Construct(const PIN_INFO * pPinInfo, ALLOCATOR_PROPERTIES * props, LPVOID pUserData, QUERYACCEPTPROC pQueryAccept, const AM_MEDIA_TYPE * pmt, float fSamplesPerSec, LPCRITICAL_SECTION pCritSec, IPin ** ppPin)
116 Parser_OutputPin * pPinImpl;
120 assert(pPinInfo->dir == PINDIR_OUTPUT);
122 pPinImpl = CoTaskMemAlloc(sizeof(Parser_OutputPin));
125 return E_OUTOFMEMORY;
127 if (SUCCEEDED(Parser_OutputPin_Init(pPinInfo, props, pUserData, pQueryAccept, pmt, fSamplesPerSec, pCritSec, pPinImpl)))
129 pPinImpl->pin.pin.lpVtbl = &Parser_OutputPin_Vtbl;
131 *ppPin = (IPin *)pPinImpl;
137 static HRESULT WINAPI Parser_QueryInterface(IBaseFilter * iface, REFIID riid, LPVOID * ppv)
139 ParserImpl *This = (ParserImpl *)iface;
140 TRACE("(%s, %p)\n", qzdebugstr_guid(riid), ppv);
144 if (IsEqualIID(riid, &IID_IUnknown))
146 else if (IsEqualIID(riid, &IID_IPersist))
148 else if (IsEqualIID(riid, &IID_IMediaFilter))
150 else if (IsEqualIID(riid, &IID_IBaseFilter))
155 IUnknown_AddRef((IUnknown *)(*ppv));
159 FIXME("No interface for %s!\n", qzdebugstr_guid(riid));
161 return E_NOINTERFACE;
164 static ULONG WINAPI Parser_AddRef(IBaseFilter * iface)
166 ParserImpl *This = (ParserImpl *)iface;
167 ULONG refCount = InterlockedIncrement(&This->refCount);
169 TRACE("(%p/%p)->() AddRef from %ld\n", This, iface, refCount - 1);
174 static ULONG WINAPI Parser_Release(IBaseFilter * iface)
176 ParserImpl *This = (ParserImpl *)iface;
177 ULONG refCount = InterlockedDecrement(&This->refCount);
179 TRACE("(%p/%p)->() Release from %ld\n", This, iface, refCount + 1);
185 DeleteCriticalSection(&This->csFilter);
187 IReferenceClock_Release(This->pClock);
189 for (i = 0; i < This->cStreams + 1; i++)
190 IPin_Release(This->ppPins[i]);
192 HeapFree(GetProcessHeap(), 0, This->ppPins);
195 TRACE("Destroying AVI splitter\n");
204 /** IPersist methods **/
206 static HRESULT WINAPI Parser_GetClassID(IBaseFilter * iface, CLSID * pClsid)
208 TRACE("(%p)\n", pClsid);
210 *pClsid = CLSID_AviSplitter;
215 /** IMediaFilter methods **/
217 static HRESULT WINAPI Parser_Stop(IBaseFilter * iface)
220 ParserImpl *This = (ParserImpl *)iface;
224 EnterCriticalSection(&This->csFilter);
226 if (This->state == State_Stopped)
228 LeaveCriticalSection(&This->csFilter);
231 hr = PullPin_StopProcessing(This->pInputPin);
232 This->state = State_Stopped;
234 LeaveCriticalSection(&This->csFilter);
239 static HRESULT WINAPI Parser_Pause(IBaseFilter * iface)
243 ParserImpl *This = (ParserImpl *)iface;
247 EnterCriticalSection(&This->csFilter);
249 if (This->state == State_Paused)
251 LeaveCriticalSection(&This->csFilter);
254 bInit = (This->state == State_Stopped);
255 This->state = State_Paused;
257 LeaveCriticalSection(&This->csFilter);
263 hr = PullPin_Seek(This->pInputPin, 0, ((LONGLONG)0x7fffffff << 32) | 0xffffffff);
266 hr = PullPin_InitProcessing(This->pInputPin);
270 for (i = 1; i < This->cStreams + 1; i++)
272 Parser_OutputPin* StreamPin = (Parser_OutputPin *)This->ppPins[i];
273 OutputPin_DeliverNewSegment((OutputPin *)This->ppPins[i], 0, (LONGLONG)ceil(10000000.0 * (float)StreamPin->dwLength / StreamPin->fSamplesPerSec), 1.0);
274 StreamPin->mediaSeeking.llDuration = (LONGLONG)ceil(10000000.0 * (float)StreamPin->dwLength / StreamPin->fSamplesPerSec);
275 StreamPin->mediaSeeking.llStop = (LONGLONG)ceil(10000000.0 * (float)StreamPin->dwLength / StreamPin->fSamplesPerSec);
276 OutputPin_CommitAllocator((OutputPin *)This->ppPins[i]);
279 /* FIXME: this is a little hacky: we have to deliver (at least?) one sample
280 * to each renderer before they will complete their transitions. We should probably
281 * seek through the stream for the first of each, rather than do it this way which is
282 * probably a bit prone to deadlocking */
283 hr = PullPin_StartProcessing(This->pInputPin);
286 /* FIXME: else pause thread */
289 hr = PullPin_PauseProcessing(This->pInputPin);
294 static HRESULT WINAPI Parser_Run(IBaseFilter * iface, REFERENCE_TIME tStart)
297 ParserImpl *This = (ParserImpl *)iface;
300 TRACE("(%s)\n", wine_dbgstr_longlong(tStart));
302 EnterCriticalSection(&This->csFilter);
304 if (This->state == State_Running)
306 LeaveCriticalSection(&This->csFilter);
310 This->rtStreamStart = tStart;
312 hr = PullPin_Seek(This->pInputPin, tStart, ((LONGLONG)0x7fffffff << 32) | 0xffffffff);
314 if (SUCCEEDED(hr) && (This->state == State_Stopped))
316 hr = PullPin_InitProcessing(This->pInputPin);
320 for (i = 1; i < (This->cStreams + 1); i++)
322 OutputPin_CommitAllocator((OutputPin *)This->ppPins[i]);
328 hr = PullPin_StartProcessing(This->pInputPin);
331 This->state = State_Running;
333 LeaveCriticalSection(&This->csFilter);
338 static HRESULT WINAPI Parser_GetState(IBaseFilter * iface, DWORD dwMilliSecsTimeout, FILTER_STATE *pState)
340 ParserImpl *This = (ParserImpl *)iface;
342 TRACE("(%ld, %p)\n", dwMilliSecsTimeout, pState);
344 EnterCriticalSection(&This->csFilter);
346 *pState = This->state;
348 LeaveCriticalSection(&This->csFilter);
350 /* FIXME: this is a little bit unsafe, but I don't see that we can do this
351 * while in the critical section. Maybe we could copy the pointer and addref in the
352 * critical section and then release after this.
354 if (This->pInputPin && (PullPin_WaitForStateChange(This->pInputPin, dwMilliSecsTimeout) == S_FALSE))
355 return VFW_S_STATE_INTERMEDIATE;
360 static HRESULT WINAPI Parser_SetSyncSource(IBaseFilter * iface, IReferenceClock *pClock)
362 ParserImpl *This = (ParserImpl *)iface;
364 TRACE("(%p)\n", pClock);
366 EnterCriticalSection(&This->csFilter);
369 IReferenceClock_Release(This->pClock);
370 This->pClock = pClock;
372 IReferenceClock_AddRef(This->pClock);
374 LeaveCriticalSection(&This->csFilter);
379 static HRESULT WINAPI Parser_GetSyncSource(IBaseFilter * iface, IReferenceClock **ppClock)
381 ParserImpl *This = (ParserImpl *)iface;
383 TRACE("(%p)\n", ppClock);
385 EnterCriticalSection(&This->csFilter);
387 *ppClock = This->pClock;
389 IReferenceClock_AddRef(This->pClock);
391 LeaveCriticalSection(&This->csFilter);
396 /** IBaseFilter implementation **/
398 static HRESULT WINAPI Parser_EnumPins(IBaseFilter * iface, IEnumPins **ppEnum)
401 ParserImpl *This = (ParserImpl *)iface;
403 TRACE("(%p)\n", ppEnum);
405 epd.cPins = This->cStreams + 1; /* +1 for input pin */
406 epd.ppPins = This->ppPins;
407 return IEnumPinsImpl_Construct(&epd, ppEnum);
410 static HRESULT WINAPI Parser_FindPin(IBaseFilter * iface, LPCWSTR Id, IPin **ppPin)
412 FIXME("(%p)->(%s,%p)\n", iface, debugstr_w(Id), ppPin);
414 /* FIXME: critical section */
419 static HRESULT WINAPI Parser_QueryFilterInfo(IBaseFilter * iface, FILTER_INFO *pInfo)
421 ParserImpl *This = (ParserImpl *)iface;
423 TRACE("(%p)\n", pInfo);
425 strcpyW(pInfo->achName, This->filterInfo.achName);
426 pInfo->pGraph = This->filterInfo.pGraph;
429 IFilterGraph_AddRef(pInfo->pGraph);
434 static HRESULT WINAPI Parser_JoinFilterGraph(IBaseFilter * iface, IFilterGraph *pGraph, LPCWSTR pName)
437 ParserImpl *This = (ParserImpl *)iface;
439 TRACE("(%p, %s)\n", pGraph, debugstr_w(pName));
441 EnterCriticalSection(&This->csFilter);
444 strcpyW(This->filterInfo.achName, pName);
446 *This->filterInfo.achName = '\0';
447 This->filterInfo.pGraph = pGraph; /* NOTE: do NOT increase ref. count */
449 LeaveCriticalSection(&This->csFilter);
454 static HRESULT WINAPI Parser_QueryVendorInfo(IBaseFilter * iface, LPWSTR *pVendorInfo)
456 TRACE("(%p)\n", pVendorInfo);
460 static const IBaseFilterVtbl Parser_Vtbl =
462 Parser_QueryInterface,
470 Parser_SetSyncSource,
471 Parser_GetSyncSource,
474 Parser_QueryFilterInfo,
475 Parser_JoinFilterGraph,
476 Parser_QueryVendorInfo
479 HRESULT Parser_AddPin(ParserImpl * This, PIN_INFO * piOutput, ALLOCATOR_PROPERTIES * props, AM_MEDIA_TYPE * amt, float fSamplesPerSec, DWORD dwSampleSize, DWORD dwLength)
484 ppOldPins = This->ppPins;
486 This->ppPins = HeapAlloc(GetProcessHeap(), 0, (This->cStreams + 2) * sizeof(IPin *));
487 memcpy(This->ppPins, ppOldPins, (This->cStreams + 1) * sizeof(IPin *));
489 hr = Parser_OutputPin_Construct(piOutput, props, NULL, Parser_OutputPin_QueryAccept, amt, fSamplesPerSec, &This->csFilter, This->ppPins + This->cStreams + 1);
493 ((Parser_OutputPin *)(This->ppPins[This->cStreams + 1]))->dwSampleSize = dwSampleSize;
494 ((Parser_OutputPin *)(This->ppPins[This->cStreams + 1]))->dwLength = dwLength;
495 ((Parser_OutputPin *)(This->ppPins[This->cStreams + 1]))->pin.pin.pUserData = (LPVOID)This->ppPins[This->cStreams + 1];
497 HeapFree(GetProcessHeap(), 0, ppOldPins);
501 HeapFree(GetProcessHeap(), 0, This->ppPins);
502 This->ppPins = ppOldPins;
503 ERR("Failed with error %lx\n", hr);
509 static HRESULT Parser_RemoveOutputPins(ParserImpl * This)
511 /* NOTE: should be in critical section when calling this function */
514 IPin ** ppOldPins = This->ppPins;
516 /* reduce the pin array down to 1 (just our input pin) */
517 This->ppPins = HeapAlloc(GetProcessHeap(), 0, sizeof(IPin *) * 1);
518 memcpy(This->ppPins, ppOldPins, sizeof(IPin *) * 1);
520 for (i = 0; i < This->cStreams; i++)
522 OutputPin_DeliverDisconnect((OutputPin *)ppOldPins[i + 1]);
523 IPin_Release(ppOldPins[i + 1]);
527 HeapFree(GetProcessHeap(), 0, ppOldPins);
532 static HRESULT Parser_ChangeStart(LPVOID iface)
534 FIXME("(%p)\n", iface);
538 static HRESULT Parser_ChangeStop(LPVOID iface)
540 FIXME("(%p)\n", iface);
544 static HRESULT Parser_ChangeRate(LPVOID iface)
546 FIXME("(%p)\n", iface);
551 static HRESULT WINAPI Parser_Seeking_QueryInterface(IMediaSeeking * iface, REFIID riid, LPVOID * ppv)
553 ICOM_THIS_From_IMediaSeeking(Parser_OutputPin, iface);
555 return IUnknown_QueryInterface((IUnknown *)This, riid, ppv);
558 static ULONG WINAPI Parser_Seeking_AddRef(IMediaSeeking * iface)
560 ICOM_THIS_From_IMediaSeeking(Parser_OutputPin, iface);
562 return IUnknown_AddRef((IUnknown *)This);
565 static ULONG WINAPI Parser_Seeking_Release(IMediaSeeking * iface)
567 ICOM_THIS_From_IMediaSeeking(Parser_OutputPin, iface);
569 return IUnknown_Release((IUnknown *)This);
572 static const IMediaSeekingVtbl Parser_Seeking_Vtbl =
574 Parser_Seeking_QueryInterface,
575 Parser_Seeking_AddRef,
576 Parser_Seeking_Release,
577 MediaSeekingImpl_GetCapabilities,
578 MediaSeekingImpl_CheckCapabilities,
579 MediaSeekingImpl_IsFormatSupported,
580 MediaSeekingImpl_QueryPreferredFormat,
581 MediaSeekingImpl_GetTimeFormat,
582 MediaSeekingImpl_IsUsingTimeFormat,
583 MediaSeekingImpl_SetTimeFormat,
584 MediaSeekingImpl_GetDuration,
585 MediaSeekingImpl_GetStopPosition,
586 MediaSeekingImpl_GetCurrentPosition,
587 MediaSeekingImpl_ConvertTimeFormat,
588 MediaSeekingImpl_SetPositions,
589 MediaSeekingImpl_GetPositions,
590 MediaSeekingImpl_GetAvailable,
591 MediaSeekingImpl_SetRate,
592 MediaSeekingImpl_GetRate,
593 MediaSeekingImpl_GetPreroll
596 HRESULT WINAPI Parser_OutputPin_QueryInterface(IPin * iface, REFIID riid, LPVOID * ppv)
598 Parser_OutputPin *This = (Parser_OutputPin *)iface;
600 TRACE("(%s, %p)\n", qzdebugstr_guid(riid), ppv);
604 if (IsEqualIID(riid, &IID_IUnknown))
605 *ppv = (LPVOID)iface;
606 else if (IsEqualIID(riid, &IID_IPin))
607 *ppv = (LPVOID)iface;
608 else if (IsEqualIID(riid, &IID_IMediaSeeking))
609 *ppv = (LPVOID)&This->mediaSeeking;
613 IUnknown_AddRef((IUnknown *)(*ppv));
617 FIXME("No interface for %s!\n", qzdebugstr_guid(riid));
619 return E_NOINTERFACE;
622 static ULONG WINAPI Parser_OutputPin_Release(IPin * iface)
624 Parser_OutputPin *This = (Parser_OutputPin *)iface;
625 ULONG refCount = InterlockedDecrement(&This->pin.pin.refCount);
627 TRACE("(%p)->() Release from %ld\n", iface, refCount + 1);
631 FreeMediaType(This->pmt);
632 CoTaskMemFree(This->pmt);
633 FreeMediaType(&This->pin.pin.mtCurrent);
640 static HRESULT WINAPI Parser_OutputPin_EnumMediaTypes(IPin * iface, IEnumMediaTypes ** ppEnum)
642 ENUMMEDIADETAILS emd;
643 Parser_OutputPin *This = (Parser_OutputPin *)iface;
645 TRACE("(%p)\n", ppEnum);
647 /* override this method to allow enumeration of your types */
649 emd.pMediaTypes = This->pmt;
651 return IEnumMediaTypesImpl_Construct(&emd, ppEnum);
654 static HRESULT Parser_OutputPin_QueryAccept(LPVOID iface, const AM_MEDIA_TYPE * pmt)
656 Parser_OutputPin *This = (Parser_OutputPin *)iface;
659 dump_AM_MEDIA_TYPE(pmt);
661 return (memcmp(This->pmt, pmt, sizeof(AM_MEDIA_TYPE)) == 0);
664 static const IPinVtbl Parser_OutputPin_Vtbl =
666 Parser_OutputPin_QueryInterface,
668 Parser_OutputPin_Release,
670 OutputPin_ReceiveConnection,
671 OutputPin_Disconnect,
672 IPinImpl_ConnectedTo,
673 IPinImpl_ConnectionMediaType,
674 IPinImpl_QueryPinInfo,
675 IPinImpl_QueryDirection,
677 IPinImpl_QueryAccept,
678 Parser_OutputPin_EnumMediaTypes,
679 IPinImpl_QueryInternalConnections,
680 OutputPin_EndOfStream,
681 OutputPin_BeginFlush,
686 static HRESULT Parser_InputPin_Construct(const PIN_INFO * pPinInfo, SAMPLEPROC pSampleProc, LPVOID pUserData, QUERYACCEPTPROC pQueryAccept, LPCRITICAL_SECTION pCritSec, IPin ** ppPin)
692 if (pPinInfo->dir != PINDIR_INPUT)
694 ERR("Pin direction(%x) != PINDIR_INPUT\n", pPinInfo->dir);
698 pPinImpl = CoTaskMemAlloc(sizeof(*pPinImpl));
701 return E_OUTOFMEMORY;
703 if (SUCCEEDED(PullPin_Init(pPinInfo, pSampleProc, pUserData, pQueryAccept, pCritSec, pPinImpl)))
705 pPinImpl->pin.lpVtbl = &Parser_InputPin_Vtbl;
707 *ppPin = (IPin *)(&pPinImpl->pin.lpVtbl);
713 static HRESULT WINAPI Parser_InputPin_Disconnect(IPin * iface)
716 IPinImpl *This = (IPinImpl *)iface;
720 EnterCriticalSection(This->pCritSec);
722 if (This->pConnectedTo)
726 hr = IBaseFilter_GetState(This->pinInfo.pFilter, 0, &state);
728 if (SUCCEEDED(hr) && (state == State_Stopped))
730 IPin_Release(This->pConnectedTo);
731 This->pConnectedTo = NULL;
732 hr = Parser_RemoveOutputPins((ParserImpl *)This->pinInfo.pFilter);
735 hr = VFW_E_NOT_STOPPED;
740 LeaveCriticalSection(This->pCritSec);
745 static const IPinVtbl Parser_InputPin_Vtbl =
747 PullPin_QueryInterface,
751 PullPin_ReceiveConnection,
752 Parser_InputPin_Disconnect,
753 IPinImpl_ConnectedTo,
754 IPinImpl_ConnectionMediaType,
755 IPinImpl_QueryPinInfo,
756 IPinImpl_QueryDirection,
758 IPinImpl_QueryAccept,
759 IPinImpl_EnumMediaTypes,
760 IPinImpl_QueryInternalConnections,