quartz: Fix bugs that occur during connection.
[wine/wine64.git] / dlls / quartz / pin.c
blobe2b868e397c241c7e10781f86eb9baf4100ed961
1 /*
2 * Generic Implementation of IPin Interface
4 * Copyright 2003 Robert Shearman
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., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
21 #include "quartz_private.h"
22 #include "pin.h"
24 #include "wine/debug.h"
25 #include "wine/unicode.h"
26 #include "uuids.h"
27 #include "vfwmsgs.h"
28 #include <assert.h>
30 WINE_DEFAULT_DEBUG_CHANNEL(quartz);
32 static const IPinVtbl InputPin_Vtbl;
33 static const IPinVtbl OutputPin_Vtbl;
34 static const IMemInputPinVtbl MemInputPin_Vtbl;
35 static const IPinVtbl PullPin_Vtbl;
37 #define ALIGNDOWN(value,boundary) ((value)/(boundary)*(boundary))
38 #define ALIGNUP(value,boundary) (ALIGNDOWN((value)+(boundary)-1, (boundary)))
40 typedef HRESULT (*SendPinFunc)( IPin *to, LPVOID arg );
42 /** Helper function, there are a lot of places where the error code is inherited
43 * The following rules apply:
45 * Return the first received error code (E_NOTIMPL is ignored)
46 * If no errors occur: return the first received non-error-code that isn't S_OK
48 HRESULT updatehres( HRESULT original, HRESULT new )
50 if (FAILED( original ) || new == E_NOTIMPL)
51 return original;
53 if (FAILED( new ) || original == S_OK)
54 return new;
56 return original;
59 /** Sends a message from a pin further to other, similar pins
60 * fnMiddle is called on each pin found further on the stream.
61 * fnEnd (can be NULL) is called when the message can't be sent any further (this is a renderer or source)
63 * If the pin given is an input pin, the message will be sent downstream to other input pins
64 * If the pin given is an output pin, the message will be sent upstream to other output pins
66 static HRESULT SendFurther( IPin *from, SendPinFunc fnMiddle, LPVOID arg, SendPinFunc fnEnd )
68 PIN_INFO pin_info;
69 ULONG amount = 0;
70 HRESULT hr = S_OK;
71 HRESULT hr_return = S_OK;
72 IEnumPins *enumpins = NULL;
73 BOOL foundend = TRUE;
74 PIN_DIRECTION from_dir;
76 IPin_QueryDirection( from, &from_dir );
78 hr = IPin_QueryInternalConnections( from, NULL, &amount );
79 if (hr != E_NOTIMPL && amount)
80 FIXME("Use QueryInternalConnections!\n");
81 hr = S_OK;
83 pin_info.pFilter = NULL;
84 hr = IPin_QueryPinInfo( from, &pin_info );
85 if (FAILED(hr))
86 goto out;
88 hr = IBaseFilter_EnumPins( pin_info.pFilter, &enumpins );
89 if (FAILED(hr))
90 goto out;
92 hr = IEnumPins_Reset( enumpins );
93 while (hr == S_OK) {
94 IPin *pin = NULL;
95 hr = IEnumPins_Next( enumpins, 1, &pin, NULL );
96 if (hr == VFW_E_ENUM_OUT_OF_SYNC)
98 hr = IEnumPins_Reset( enumpins );
99 continue;
101 if (pin)
103 PIN_DIRECTION dir;
105 IPin_QueryDirection( pin, &dir );
106 if (dir != from_dir)
108 IPin *connected = NULL;
110 foundend = FALSE;
111 IPin_ConnectedTo( pin, &connected );
112 if (connected)
114 HRESULT hr_local;
116 hr_local = fnMiddle( connected, arg );
117 hr_return = updatehres( hr_return, hr_local );
118 IPin_Release(connected);
121 IPin_Release( pin );
123 else
125 hr = S_OK;
126 break;
130 if (!foundend)
131 hr = hr_return;
132 else if (fnEnd) {
133 HRESULT hr_local;
135 hr_local = fnEnd( from, arg );
136 hr_return = updatehres( hr_return, hr_local );
139 out:
140 if (pin_info.pFilter)
141 IBaseFilter_Release( pin_info.pFilter );
142 return hr;
145 static inline InputPin *impl_from_IMemInputPin( IMemInputPin *iface )
147 return (InputPin *)((char*)iface - FIELD_OFFSET(InputPin, lpVtblMemInput));
151 static void Copy_PinInfo(PIN_INFO * pDest, const PIN_INFO * pSrc)
153 /* Tempting to just do a memcpy, but the name field is
154 128 characters long! We will probably never exceed 10
155 most of the time, so we are better off copying
156 each field manually */
157 strcpyW(pDest->achName, pSrc->achName);
158 pDest->dir = pSrc->dir;
159 pDest->pFilter = pSrc->pFilter;
162 /* Function called as a helper to IPin_Connect */
163 /* specific AM_MEDIA_TYPE - it cannot be NULL */
164 /* NOTE: not part of standard interface */
165 static HRESULT OutputPin_ConnectSpecific(IPin * iface, IPin * pReceivePin, const AM_MEDIA_TYPE * pmt)
167 OutputPin *This = (OutputPin *)iface;
168 HRESULT hr;
169 IMemAllocator * pMemAlloc = NULL;
170 ALLOCATOR_PROPERTIES actual; /* FIXME: should we put the actual props back in to This? */
172 TRACE("(%p, %p)\n", pReceivePin, pmt);
173 dump_AM_MEDIA_TYPE(pmt);
175 /* FIXME: call queryacceptproc */
177 This->pin.pConnectedTo = pReceivePin;
178 IPin_AddRef(pReceivePin);
179 CopyMediaType(&This->pin.mtCurrent, pmt);
181 hr = IPin_ReceiveConnection(pReceivePin, iface, pmt);
183 /* get the IMemInputPin interface we will use to deliver samples to the
184 * connected pin */
185 if (SUCCEEDED(hr))
187 This->pMemInputPin = NULL;
188 hr = IPin_QueryInterface(pReceivePin, &IID_IMemInputPin, (LPVOID)&This->pMemInputPin);
190 if (SUCCEEDED(hr) && !This->custom_allocator)
192 hr = IMemInputPin_GetAllocator(This->pMemInputPin, &pMemAlloc);
194 if (hr == VFW_E_NO_ALLOCATOR)
195 /* Input pin provides no allocator, use standard memory allocator */
196 hr = CoCreateInstance(&CLSID_MemoryAllocator, NULL, CLSCTX_INPROC_SERVER, &IID_IMemAllocator, (LPVOID*)&pMemAlloc);
198 if (SUCCEEDED(hr))
199 hr = IMemAllocator_SetProperties(pMemAlloc, &This->allocProps, &actual);
201 if (SUCCEEDED(hr))
202 hr = IMemInputPin_NotifyAllocator(This->pMemInputPin, pMemAlloc, This->readonly);
204 if (pMemAlloc)
205 IMemAllocator_Release(pMemAlloc);
207 else if (SUCCEEDED(hr))
209 if (This->alloc)
211 hr = IMemInputPin_NotifyAllocator(This->pMemInputPin, This->alloc, This->readonly);
213 else
214 hr = VFW_E_NO_ALLOCATOR;
217 /* break connection if we couldn't get the allocator */
218 if (FAILED(hr))
220 if (This->pMemInputPin)
221 IMemInputPin_Release(This->pMemInputPin);
222 This->pMemInputPin = NULL;
224 IPin_Disconnect(pReceivePin);
228 if (FAILED(hr))
230 IPin_Release(This->pin.pConnectedTo);
231 This->pin.pConnectedTo = NULL;
232 FreeMediaType(&This->pin.mtCurrent);
235 TRACE(" -- %x\n", hr);
236 return hr;
239 static HRESULT InputPin_Init(const IPinVtbl *InputPin_Vtbl, const PIN_INFO * pPinInfo, SAMPLEPROC_PUSH pSampleProc, LPVOID pUserData,
240 QUERYACCEPTPROC pQueryAccept, CLEANUPPROC pCleanUp, LPCRITICAL_SECTION pCritSec, IMemAllocator *allocator, InputPin * pPinImpl)
242 TRACE("\n");
244 /* Common attributes */
245 pPinImpl->pin.refCount = 1;
246 pPinImpl->pin.pConnectedTo = NULL;
247 pPinImpl->pin.fnQueryAccept = pQueryAccept;
248 pPinImpl->pin.pUserData = pUserData;
249 pPinImpl->pin.pCritSec = pCritSec;
250 Copy_PinInfo(&pPinImpl->pin.pinInfo, pPinInfo);
251 ZeroMemory(&pPinImpl->pin.mtCurrent, sizeof(AM_MEDIA_TYPE));
253 /* Input pin attributes */
254 pPinImpl->fnSampleProc = pSampleProc;
255 pPinImpl->fnCleanProc = pCleanUp;
256 pPinImpl->pAllocator = pPinImpl->preferred_allocator = allocator;
257 if (pPinImpl->preferred_allocator)
258 IMemAllocator_AddRef(pPinImpl->preferred_allocator);
259 pPinImpl->tStart = 0;
260 pPinImpl->tStop = 0;
261 pPinImpl->dRate = 1.0;
262 pPinImpl->pin.lpVtbl = InputPin_Vtbl;
263 pPinImpl->lpVtblMemInput = &MemInputPin_Vtbl;
264 pPinImpl->flushing = pPinImpl->end_of_stream = 0;
266 return S_OK;
269 static HRESULT OutputPin_Init(const IPinVtbl *OutputPin_Vtbl, const PIN_INFO * pPinInfo, const ALLOCATOR_PROPERTIES * props, LPVOID pUserData,
270 QUERYACCEPTPROC pQueryAccept, LPCRITICAL_SECTION pCritSec, OutputPin * pPinImpl)
272 TRACE("\n");
274 /* Common attributes */
275 pPinImpl->pin.lpVtbl = OutputPin_Vtbl;
276 pPinImpl->pin.refCount = 1;
277 pPinImpl->pin.pConnectedTo = NULL;
278 pPinImpl->pin.fnQueryAccept = pQueryAccept;
279 pPinImpl->pin.pUserData = pUserData;
280 pPinImpl->pin.pCritSec = pCritSec;
281 Copy_PinInfo(&pPinImpl->pin.pinInfo, pPinInfo);
282 ZeroMemory(&pPinImpl->pin.mtCurrent, sizeof(AM_MEDIA_TYPE));
284 /* Output pin attributes */
285 pPinImpl->pMemInputPin = NULL;
286 pPinImpl->pConnectSpecific = OutputPin_ConnectSpecific;
287 /* If custom_allocator is set, you will need to specify an allocator
288 * in the alloc member of the struct before an output pin can connect
290 pPinImpl->custom_allocator = 0;
291 pPinImpl->alloc = NULL;
292 pPinImpl->readonly = FALSE;
293 if (props)
295 pPinImpl->allocProps = *props;
296 if (pPinImpl->allocProps.cbAlign == 0)
297 pPinImpl->allocProps.cbAlign = 1;
299 else
300 ZeroMemory(&pPinImpl->allocProps, sizeof(pPinImpl->allocProps));
302 return S_OK;
305 HRESULT InputPin_Construct(const IPinVtbl *InputPin_Vtbl, const PIN_INFO * pPinInfo, SAMPLEPROC_PUSH pSampleProc, LPVOID pUserData, QUERYACCEPTPROC pQueryAccept, CLEANUPPROC pCleanUp, LPCRITICAL_SECTION pCritSec, IMemAllocator *allocator, IPin ** ppPin)
307 InputPin * pPinImpl;
309 *ppPin = NULL;
311 if (pPinInfo->dir != PINDIR_INPUT)
313 ERR("Pin direction(%x) != PINDIR_INPUT\n", pPinInfo->dir);
314 return E_INVALIDARG;
317 pPinImpl = CoTaskMemAlloc(sizeof(*pPinImpl));
319 if (!pPinImpl)
320 return E_OUTOFMEMORY;
322 if (SUCCEEDED(InputPin_Init(InputPin_Vtbl, pPinInfo, pSampleProc, pUserData, pQueryAccept, pCleanUp, pCritSec, allocator, pPinImpl)))
324 *ppPin = (IPin *)pPinImpl;
325 return S_OK;
328 CoTaskMemFree(pPinImpl);
329 return E_FAIL;
332 HRESULT OutputPin_Construct(const IPinVtbl *OutputPin_Vtbl, long outputpin_size, const PIN_INFO * pPinInfo, ALLOCATOR_PROPERTIES *props, LPVOID pUserData, QUERYACCEPTPROC pQueryAccept, LPCRITICAL_SECTION pCritSec, IPin ** ppPin)
334 OutputPin * pPinImpl;
336 *ppPin = NULL;
338 if (pPinInfo->dir != PINDIR_OUTPUT)
340 ERR("Pin direction(%x) != PINDIR_OUTPUT\n", pPinInfo->dir);
341 return E_INVALIDARG;
344 assert(outputpin_size >= sizeof(OutputPin));
346 pPinImpl = CoTaskMemAlloc(outputpin_size);
348 if (!pPinImpl)
349 return E_OUTOFMEMORY;
351 if (SUCCEEDED(OutputPin_Init(OutputPin_Vtbl, pPinInfo, props, pUserData, pQueryAccept, pCritSec, pPinImpl)))
353 *ppPin = (IPin *)(&pPinImpl->pin.lpVtbl);
354 return S_OK;
357 CoTaskMemFree(pPinImpl);
358 return E_FAIL;
361 /*** Common pin functions ***/
363 ULONG WINAPI IPinImpl_AddRef(IPin * iface)
365 IPinImpl *This = (IPinImpl *)iface;
366 ULONG refCount = InterlockedIncrement(&This->refCount);
368 TRACE("(%p)->() AddRef from %d\n", iface, refCount - 1);
370 return refCount;
373 HRESULT WINAPI IPinImpl_Disconnect(IPin * iface)
375 HRESULT hr;
376 IPinImpl *This = (IPinImpl *)iface;
378 TRACE("()\n");
380 EnterCriticalSection(This->pCritSec);
382 if (This->pConnectedTo)
384 IPin_Release(This->pConnectedTo);
385 This->pConnectedTo = NULL;
386 FreeMediaType(&This->mtCurrent);
387 ZeroMemory(&This->mtCurrent, sizeof(This->mtCurrent));
388 hr = S_OK;
390 else
391 hr = S_FALSE;
393 LeaveCriticalSection(This->pCritSec);
395 return hr;
398 HRESULT WINAPI IPinImpl_ConnectedTo(IPin * iface, IPin ** ppPin)
400 HRESULT hr;
401 IPinImpl *This = (IPinImpl *)iface;
403 TRACE("(%p)\n", ppPin);
405 EnterCriticalSection(This->pCritSec);
407 if (This->pConnectedTo)
409 *ppPin = This->pConnectedTo;
410 IPin_AddRef(*ppPin);
411 hr = S_OK;
413 else
415 hr = VFW_E_NOT_CONNECTED;
416 *ppPin = NULL;
419 LeaveCriticalSection(This->pCritSec);
421 return hr;
424 HRESULT WINAPI IPinImpl_ConnectionMediaType(IPin * iface, AM_MEDIA_TYPE * pmt)
426 HRESULT hr;
427 IPinImpl *This = (IPinImpl *)iface;
429 TRACE("(%p/%p)->(%p)\n", This, iface, pmt);
431 EnterCriticalSection(This->pCritSec);
433 if (This->pConnectedTo)
435 CopyMediaType(pmt, &This->mtCurrent);
436 hr = S_OK;
438 else
440 ZeroMemory(pmt, sizeof(*pmt));
441 hr = VFW_E_NOT_CONNECTED;
444 LeaveCriticalSection(This->pCritSec);
446 return hr;
449 HRESULT WINAPI IPinImpl_QueryPinInfo(IPin * iface, PIN_INFO * pInfo)
451 IPinImpl *This = (IPinImpl *)iface;
453 TRACE("(%p/%p)->(%p)\n", This, iface, pInfo);
455 Copy_PinInfo(pInfo, &This->pinInfo);
456 IBaseFilter_AddRef(pInfo->pFilter);
458 return S_OK;
461 HRESULT WINAPI IPinImpl_QueryDirection(IPin * iface, PIN_DIRECTION * pPinDir)
463 IPinImpl *This = (IPinImpl *)iface;
465 TRACE("(%p/%p)->(%p)\n", This, iface, pPinDir);
467 *pPinDir = This->pinInfo.dir;
469 return S_OK;
472 HRESULT WINAPI IPinImpl_QueryId(IPin * iface, LPWSTR * Id)
474 IPinImpl *This = (IPinImpl *)iface;
476 TRACE("(%p/%p)->(%p)\n", This, iface, Id);
478 *Id = CoTaskMemAlloc((strlenW(This->pinInfo.achName) + 1) * sizeof(WCHAR));
479 if (!*Id)
480 return E_OUTOFMEMORY;
482 strcpyW(*Id, This->pinInfo.achName);
484 return S_OK;
487 HRESULT WINAPI IPinImpl_QueryAccept(IPin * iface, const AM_MEDIA_TYPE * pmt)
489 IPinImpl *This = (IPinImpl *)iface;
491 TRACE("(%p/%p)->(%p)\n", This, iface, pmt);
493 return (This->fnQueryAccept(This->pUserData, pmt) == S_OK ? S_OK : S_FALSE);
496 HRESULT WINAPI IPinImpl_EnumMediaTypes(IPin * iface, IEnumMediaTypes ** ppEnum)
498 IPinImpl *This = (IPinImpl *)iface;
499 ENUMMEDIADETAILS emd;
501 TRACE("(%p/%p)->(%p)\n", This, iface, ppEnum);
503 /* override this method to allow enumeration of your types */
504 emd.cMediaTypes = 0;
505 emd.pMediaTypes = NULL;
507 return IEnumMediaTypesImpl_Construct(&emd, ppEnum);
510 HRESULT WINAPI IPinImpl_QueryInternalConnections(IPin * iface, IPin ** apPin, ULONG * cPin)
512 IPinImpl *This = (IPinImpl *)iface;
514 TRACE("(%p/%p)->(%p, %p)\n", This, iface, apPin, cPin);
516 return E_NOTIMPL; /* to tell caller that all input pins connected to all output pins */
519 /*** IPin implementation for an input pin ***/
521 HRESULT WINAPI InputPin_QueryInterface(IPin * iface, REFIID riid, LPVOID * ppv)
523 InputPin *This = (InputPin *)iface;
525 TRACE("(%p)->(%s, %p)\n", iface, qzdebugstr_guid(riid), ppv);
527 *ppv = NULL;
529 if (IsEqualIID(riid, &IID_IUnknown))
530 *ppv = (LPVOID)iface;
531 else if (IsEqualIID(riid, &IID_IPin))
532 *ppv = (LPVOID)iface;
533 else if (IsEqualIID(riid, &IID_IMemInputPin))
534 *ppv = (LPVOID)&This->lpVtblMemInput;
535 else if (IsEqualIID(riid, &IID_IMediaSeeking))
537 return IBaseFilter_QueryInterface(This->pin.pinInfo.pFilter, &IID_IMediaSeeking, ppv);
540 if (*ppv)
542 IUnknown_AddRef((IUnknown *)(*ppv));
543 return S_OK;
546 FIXME("No interface for %s!\n", qzdebugstr_guid(riid));
548 return E_NOINTERFACE;
551 ULONG WINAPI InputPin_Release(IPin * iface)
553 InputPin *This = (InputPin *)iface;
554 ULONG refCount = InterlockedDecrement(&This->pin.refCount);
556 TRACE("(%p)->() Release from %d\n", iface, refCount + 1);
558 if (!refCount)
560 FreeMediaType(&This->pin.mtCurrent);
561 if (This->pAllocator)
562 IMemAllocator_Release(This->pAllocator);
563 This->pAllocator = NULL;
564 This->pin.lpVtbl = NULL;
565 CoTaskMemFree(This);
566 return 0;
568 else
569 return refCount;
572 HRESULT WINAPI InputPin_Connect(IPin * iface, IPin * pConnector, const AM_MEDIA_TYPE * pmt)
574 ERR("Outgoing connection on an input pin! (%p, %p)\n", pConnector, pmt);
576 return E_UNEXPECTED;
580 HRESULT WINAPI InputPin_ReceiveConnection(IPin * iface, IPin * pReceivePin, const AM_MEDIA_TYPE * pmt)
582 InputPin *This = (InputPin *)iface;
583 PIN_DIRECTION pindirReceive;
584 HRESULT hr = S_OK;
586 TRACE("(%p, %p)\n", pReceivePin, pmt);
587 dump_AM_MEDIA_TYPE(pmt);
589 EnterCriticalSection(This->pin.pCritSec);
591 if (This->pin.pConnectedTo)
592 hr = VFW_E_ALREADY_CONNECTED;
594 if (SUCCEEDED(hr) && This->pin.fnQueryAccept(This->pin.pUserData, pmt) != S_OK)
595 hr = VFW_E_TYPE_NOT_ACCEPTED; /* FIXME: shouldn't we just map common errors onto
596 * VFW_E_TYPE_NOT_ACCEPTED and pass the value on otherwise? */
598 if (SUCCEEDED(hr))
600 IPin_QueryDirection(pReceivePin, &pindirReceive);
602 if (pindirReceive != PINDIR_OUTPUT)
604 ERR("Can't connect from non-output pin\n");
605 hr = VFW_E_INVALID_DIRECTION;
609 if (SUCCEEDED(hr))
611 CopyMediaType(&This->pin.mtCurrent, pmt);
612 This->pin.pConnectedTo = pReceivePin;
613 IPin_AddRef(pReceivePin);
616 LeaveCriticalSection(This->pin.pCritSec);
618 return hr;
621 static HRESULT deliver_endofstream(IPin* pin, LPVOID unused)
623 return IPin_EndOfStream( pin );
626 HRESULT WINAPI InputPin_EndOfStream(IPin * iface)
628 HRESULT hr = S_OK;
629 InputPin *This = (InputPin *)iface;
631 TRACE("(%p)\n", This);
633 EnterCriticalSection(This->pin.pCritSec);
634 if (This->flushing)
635 hr = S_FALSE;
636 else
637 This->end_of_stream = 1;
638 LeaveCriticalSection(This->pin.pCritSec);
640 if (hr == S_OK)
641 hr = SendFurther( iface, deliver_endofstream, NULL, NULL );
642 return hr;
645 static HRESULT deliver_beginflush(IPin* pin, LPVOID unused)
647 return IPin_BeginFlush( pin );
650 HRESULT WINAPI InputPin_BeginFlush(IPin * iface)
652 InputPin *This = (InputPin *)iface;
653 HRESULT hr;
654 TRACE("() semi-stub\n");
656 EnterCriticalSection(This->pin.pCritSec);
657 This->flushing = 1;
659 if (This->fnCleanProc)
660 This->fnCleanProc(This->pin.pUserData);
662 hr = SendFurther( iface, deliver_beginflush, NULL, NULL );
663 LeaveCriticalSection(This->pin.pCritSec);
665 return hr;
668 static HRESULT deliver_endflush(IPin* pin, LPVOID unused)
670 return IPin_EndFlush( pin );
673 HRESULT WINAPI InputPin_EndFlush(IPin * iface)
675 InputPin *This = (InputPin *)iface;
676 HRESULT hr;
677 TRACE("(%p)\n", This);
679 EnterCriticalSection(This->pin.pCritSec);
680 This->flushing = This->end_of_stream = 0;
682 hr = SendFurther( iface, deliver_endflush, NULL, NULL );
683 LeaveCriticalSection(This->pin.pCritSec);
685 return hr;
688 typedef struct newsegmentargs
690 REFERENCE_TIME tStart, tStop;
691 double rate;
692 } newsegmentargs;
694 static HRESULT deliver_newsegment(IPin *pin, LPVOID data)
696 newsegmentargs *args = data;
697 return IPin_NewSegment(pin, args->tStart, args->tStop, args->rate);
700 HRESULT WINAPI InputPin_NewSegment(IPin * iface, REFERENCE_TIME tStart, REFERENCE_TIME tStop, double dRate)
702 InputPin *This = (InputPin *)iface;
703 newsegmentargs args;
705 TRACE("(%x%08x, %x%08x, %e)\n", (ULONG)(tStart >> 32), (ULONG)tStart, (ULONG)(tStop >> 32), (ULONG)tStop, dRate);
707 args.tStart = This->tStart = tStart;
708 args.tStop = This->tStop = tStop;
709 args.rate = This->dRate = dRate;
711 return SendFurther( iface, deliver_newsegment, &args, NULL );
714 static const IPinVtbl InputPin_Vtbl =
716 InputPin_QueryInterface,
717 IPinImpl_AddRef,
718 InputPin_Release,
719 InputPin_Connect,
720 InputPin_ReceiveConnection,
721 IPinImpl_Disconnect,
722 IPinImpl_ConnectedTo,
723 IPinImpl_ConnectionMediaType,
724 IPinImpl_QueryPinInfo,
725 IPinImpl_QueryDirection,
726 IPinImpl_QueryId,
727 IPinImpl_QueryAccept,
728 IPinImpl_EnumMediaTypes,
729 IPinImpl_QueryInternalConnections,
730 InputPin_EndOfStream,
731 InputPin_BeginFlush,
732 InputPin_EndFlush,
733 InputPin_NewSegment
736 /*** IMemInputPin implementation ***/
738 HRESULT WINAPI MemInputPin_QueryInterface(IMemInputPin * iface, REFIID riid, LPVOID * ppv)
740 InputPin *This = impl_from_IMemInputPin(iface);
742 return IPin_QueryInterface((IPin *)&This->pin, riid, ppv);
745 ULONG WINAPI MemInputPin_AddRef(IMemInputPin * iface)
747 InputPin *This = impl_from_IMemInputPin(iface);
749 return IPin_AddRef((IPin *)&This->pin);
752 ULONG WINAPI MemInputPin_Release(IMemInputPin * iface)
754 InputPin *This = impl_from_IMemInputPin(iface);
756 return IPin_Release((IPin *)&This->pin);
759 HRESULT WINAPI MemInputPin_GetAllocator(IMemInputPin * iface, IMemAllocator ** ppAllocator)
761 InputPin *This = impl_from_IMemInputPin(iface);
763 TRACE("(%p/%p)->(%p)\n", This, iface, ppAllocator);
765 *ppAllocator = This->pAllocator;
766 if (*ppAllocator)
767 IMemAllocator_AddRef(*ppAllocator);
769 return *ppAllocator ? S_OK : VFW_E_NO_ALLOCATOR;
772 HRESULT WINAPI MemInputPin_NotifyAllocator(IMemInputPin * iface, IMemAllocator * pAllocator, BOOL bReadOnly)
774 InputPin *This = impl_from_IMemInputPin(iface);
776 TRACE("(%p/%p)->(%p, %d)\n", This, iface, pAllocator, bReadOnly);
778 if (bReadOnly)
779 FIXME("Read only flag not handled yet!\n");
781 /* FIXME: Should we release the allocator on disconnection? */
782 if (!pAllocator)
784 WARN("Null allocator\n");
785 return E_POINTER;
788 if (This->preferred_allocator && pAllocator != This->preferred_allocator)
789 return E_FAIL;
791 if (This->pAllocator)
792 IMemAllocator_Release(This->pAllocator);
793 This->pAllocator = pAllocator;
794 if (This->pAllocator)
795 IMemAllocator_AddRef(This->pAllocator);
797 return S_OK;
800 HRESULT WINAPI MemInputPin_GetAllocatorRequirements(IMemInputPin * iface, ALLOCATOR_PROPERTIES * pProps)
802 InputPin *This = impl_from_IMemInputPin(iface);
804 TRACE("(%p/%p)->(%p)\n", This, iface, pProps);
806 /* override this method if you have any specific requirements */
808 return E_NOTIMPL;
811 HRESULT WINAPI MemInputPin_Receive(IMemInputPin * iface, IMediaSample * pSample)
813 InputPin *This = impl_from_IMemInputPin(iface);
814 HRESULT hr;
816 /* this trace commented out for performance reasons */
817 /*TRACE("(%p/%p)->(%p)\n", This, iface, pSample);*/
818 hr = This->fnSampleProc(This->pin.pUserData, pSample);
819 return hr;
822 HRESULT WINAPI MemInputPin_ReceiveMultiple(IMemInputPin * iface, IMediaSample ** pSamples, long nSamples, long *nSamplesProcessed)
824 HRESULT hr = S_OK;
825 InputPin *This = impl_from_IMemInputPin(iface);
827 TRACE("(%p/%p)->(%p, %ld, %p)\n", This, iface, pSamples, nSamples, nSamplesProcessed);
829 for (*nSamplesProcessed = 0; *nSamplesProcessed < nSamples; (*nSamplesProcessed)++)
831 hr = IMemInputPin_Receive(iface, pSamples[*nSamplesProcessed]);
832 if (hr != S_OK)
833 break;
836 return hr;
839 HRESULT WINAPI MemInputPin_ReceiveCanBlock(IMemInputPin * iface)
841 InputPin *This = impl_from_IMemInputPin(iface);
843 TRACE("(%p/%p)->()\n", This, iface);
845 return S_OK;
848 static const IMemInputPinVtbl MemInputPin_Vtbl =
850 MemInputPin_QueryInterface,
851 MemInputPin_AddRef,
852 MemInputPin_Release,
853 MemInputPin_GetAllocator,
854 MemInputPin_NotifyAllocator,
855 MemInputPin_GetAllocatorRequirements,
856 MemInputPin_Receive,
857 MemInputPin_ReceiveMultiple,
858 MemInputPin_ReceiveCanBlock
861 HRESULT WINAPI OutputPin_QueryInterface(IPin * iface, REFIID riid, LPVOID * ppv)
863 OutputPin *This = (OutputPin *)iface;
865 TRACE("(%p/%p)->(%s, %p)\n", This, iface, qzdebugstr_guid(riid), ppv);
867 *ppv = NULL;
869 if (IsEqualIID(riid, &IID_IUnknown))
870 *ppv = (LPVOID)iface;
871 else if (IsEqualIID(riid, &IID_IPin))
872 *ppv = (LPVOID)iface;
873 else if (IsEqualIID(riid, &IID_IMediaSeeking))
875 return IBaseFilter_QueryInterface(This->pin.pinInfo.pFilter, &IID_IMediaSeeking, ppv);
878 if (*ppv)
880 IUnknown_AddRef((IUnknown *)(*ppv));
881 return S_OK;
884 FIXME("No interface for %s!\n", qzdebugstr_guid(riid));
886 return E_NOINTERFACE;
889 ULONG WINAPI OutputPin_Release(IPin * iface)
891 OutputPin *This = (OutputPin *)iface;
892 ULONG refCount = InterlockedDecrement(&This->pin.refCount);
894 TRACE("(%p)->() Release from %d\n", iface, refCount + 1);
896 if (!refCount)
898 FreeMediaType(&This->pin.mtCurrent);
899 CoTaskMemFree(This);
900 return 0;
902 return refCount;
905 HRESULT WINAPI OutputPin_Connect(IPin * iface, IPin * pReceivePin, const AM_MEDIA_TYPE * pmt)
907 HRESULT hr;
908 OutputPin *This = (OutputPin *)iface;
910 TRACE("(%p/%p)->(%p, %p)\n", This, iface, pReceivePin, pmt);
911 dump_AM_MEDIA_TYPE(pmt);
913 /* If we try to connect to ourself, we will definitely deadlock.
914 * There are other cases where we could deadlock too, but this
915 * catches the obvious case */
916 assert(pReceivePin != iface);
918 EnterCriticalSection(This->pin.pCritSec);
920 /* if we have been a specific type to connect with, then we can either connect
921 * with that or fail. We cannot choose different AM_MEDIA_TYPE */
922 if (pmt && !IsEqualGUID(&pmt->majortype, &GUID_NULL) && !IsEqualGUID(&pmt->subtype, &GUID_NULL))
923 hr = This->pConnectSpecific(iface, pReceivePin, pmt);
924 else
926 /* negotiate media type */
928 IEnumMediaTypes * pEnumCandidates;
929 AM_MEDIA_TYPE * pmtCandidate = NULL; /* Candidate media type */
931 if (SUCCEEDED(hr = IPin_EnumMediaTypes(iface, &pEnumCandidates)))
933 hr = VFW_E_NO_ACCEPTABLE_TYPES; /* Assume the worst, but set to S_OK if connected successfully */
935 /* try this filter's media types first */
936 while (S_OK == IEnumMediaTypes_Next(pEnumCandidates, 1, &pmtCandidate, NULL))
938 assert(pmtCandidate);
939 dump_AM_MEDIA_TYPE(pmtCandidate);
940 if (!IsEqualGUID(&FORMAT_None, &pmtCandidate->formattype)
941 && !IsEqualGUID(&GUID_NULL, &pmtCandidate->formattype))
942 assert(pmtCandidate->pbFormat);
943 if (( !pmt || CompareMediaTypes(pmt, pmtCandidate, TRUE) ) &&
944 (This->pConnectSpecific(iface, pReceivePin, pmtCandidate) == S_OK))
946 hr = S_OK;
947 DeleteMediaType(pmtCandidate);
948 break;
950 DeleteMediaType(pmtCandidate);
951 pmtCandidate = NULL;
953 IEnumMediaTypes_Release(pEnumCandidates);
956 /* then try receiver filter's media types */
957 if (hr != S_OK && SUCCEEDED(hr = IPin_EnumMediaTypes(pReceivePin, &pEnumCandidates))) /* if we haven't already connected successfully */
959 hr = VFW_E_NO_ACCEPTABLE_TYPES; /* Assume the worst, but set to S_OK if connected successfully */
961 while (S_OK == IEnumMediaTypes_Next(pEnumCandidates, 1, &pmtCandidate, NULL))
963 assert(pmtCandidate);
964 dump_AM_MEDIA_TYPE(pmtCandidate);
965 if (!IsEqualGUID(&FORMAT_None, &pmtCandidate->formattype)
966 && !IsEqualGUID(&GUID_NULL, &pmtCandidate->formattype))
967 assert(pmtCandidate->pbFormat);
968 if (( !pmt || CompareMediaTypes(pmt, pmtCandidate, TRUE) ) &&
969 (This->pConnectSpecific(iface, pReceivePin, pmtCandidate) == S_OK))
971 hr = S_OK;
972 DeleteMediaType(pmtCandidate);
973 break;
975 DeleteMediaType(pmtCandidate);
976 pmtCandidate = NULL;
977 } /* while */
978 IEnumMediaTypes_Release(pEnumCandidates);
979 } /* if not found */
980 } /* if negotiate media type */
981 } /* if succeeded */
982 LeaveCriticalSection(This->pin.pCritSec);
984 TRACE(" -- %x\n", hr);
985 return hr;
988 HRESULT WINAPI OutputPin_ReceiveConnection(IPin * iface, IPin * pReceivePin, const AM_MEDIA_TYPE * pmt)
990 ERR("Incoming connection on an output pin! (%p, %p)\n", pReceivePin, pmt);
992 return E_UNEXPECTED;
995 HRESULT WINAPI OutputPin_Disconnect(IPin * iface)
997 HRESULT hr;
998 OutputPin *This = (OutputPin *)iface;
1000 TRACE("()\n");
1002 EnterCriticalSection(This->pin.pCritSec);
1004 if (This->pMemInputPin)
1006 IMemInputPin_Release(This->pMemInputPin);
1007 This->pMemInputPin = NULL;
1009 if (This->pin.pConnectedTo)
1011 IPin_Release(This->pin.pConnectedTo);
1012 This->pin.pConnectedTo = NULL;
1013 FreeMediaType(&This->pin.mtCurrent);
1014 ZeroMemory(&This->pin.mtCurrent, sizeof(This->pin.mtCurrent));
1015 hr = S_OK;
1017 else
1018 hr = S_FALSE;
1020 LeaveCriticalSection(This->pin.pCritSec);
1022 return hr;
1025 HRESULT WINAPI OutputPin_EndOfStream(IPin * iface)
1027 TRACE("()\n");
1029 /* not supposed to do anything in an output pin */
1031 return E_UNEXPECTED;
1034 HRESULT WINAPI OutputPin_BeginFlush(IPin * iface)
1036 TRACE("(%p)->()\n", iface);
1038 /* not supposed to do anything in an output pin */
1040 return E_UNEXPECTED;
1043 HRESULT WINAPI OutputPin_EndFlush(IPin * iface)
1045 TRACE("(%p)->()\n", iface);
1047 /* not supposed to do anything in an output pin */
1049 return E_UNEXPECTED;
1052 HRESULT WINAPI OutputPin_NewSegment(IPin * iface, REFERENCE_TIME tStart, REFERENCE_TIME tStop, double dRate)
1054 TRACE("(%p)->(%x%08x, %x%08x, %e)\n", iface, (ULONG)(tStart >> 32), (ULONG)tStart, (ULONG)(tStop >> 32), (ULONG)tStop, dRate);
1056 /* not supposed to do anything in an output pin */
1058 return E_UNEXPECTED;
1061 static const IPinVtbl OutputPin_Vtbl =
1063 OutputPin_QueryInterface,
1064 IPinImpl_AddRef,
1065 OutputPin_Release,
1066 OutputPin_Connect,
1067 OutputPin_ReceiveConnection,
1068 OutputPin_Disconnect,
1069 IPinImpl_ConnectedTo,
1070 IPinImpl_ConnectionMediaType,
1071 IPinImpl_QueryPinInfo,
1072 IPinImpl_QueryDirection,
1073 IPinImpl_QueryId,
1074 IPinImpl_QueryAccept,
1075 IPinImpl_EnumMediaTypes,
1076 IPinImpl_QueryInternalConnections,
1077 OutputPin_EndOfStream,
1078 OutputPin_BeginFlush,
1079 OutputPin_EndFlush,
1080 OutputPin_NewSegment
1083 HRESULT OutputPin_GetDeliveryBuffer(OutputPin * This, IMediaSample ** ppSample, REFERENCE_TIME * tStart, REFERENCE_TIME * tStop, DWORD dwFlags)
1085 HRESULT hr;
1087 TRACE("(%p, %p, %p, %x)\n", ppSample, tStart, tStop, dwFlags);
1089 EnterCriticalSection(This->pin.pCritSec);
1091 if (!This->pin.pConnectedTo)
1092 hr = VFW_E_NOT_CONNECTED;
1093 else
1095 IMemAllocator * pAlloc = NULL;
1097 hr = IMemInputPin_GetAllocator(This->pMemInputPin, &pAlloc);
1099 if (SUCCEEDED(hr))
1100 hr = IMemAllocator_GetBuffer(pAlloc, ppSample, tStart, tStop, dwFlags);
1102 if (SUCCEEDED(hr))
1103 hr = IMediaSample_SetTime(*ppSample, tStart, tStop);
1105 if (pAlloc)
1106 IMemAllocator_Release(pAlloc);
1109 LeaveCriticalSection(This->pin.pCritSec);
1111 return hr;
1114 HRESULT OutputPin_SendSample(OutputPin * This, IMediaSample * pSample)
1116 HRESULT hr = S_OK;
1117 IMemInputPin * pMemConnected = NULL;
1118 PIN_INFO pinInfo;
1120 EnterCriticalSection(This->pin.pCritSec);
1122 if (!This->pin.pConnectedTo || !This->pMemInputPin)
1123 hr = VFW_E_NOT_CONNECTED;
1124 else
1126 /* we don't have the lock held when using This->pMemInputPin,
1127 * so we need to AddRef it to stop it being deleted while we are
1128 * using it. Same with its filter. */
1129 pMemConnected = This->pMemInputPin;
1130 IMemInputPin_AddRef(pMemConnected);
1131 hr = IPin_QueryPinInfo(This->pin.pConnectedTo, &pinInfo);
1134 LeaveCriticalSection(This->pin.pCritSec);
1136 if (SUCCEEDED(hr))
1138 /* NOTE: if we are in a critical section when Receive is called
1139 * then it causes some problems (most notably with the native Video
1140 * Renderer) if we are re-entered for whatever reason */
1141 hr = IMemInputPin_Receive(pMemConnected, pSample);
1143 /* If the filter's destroyed, tell upstream to stop sending data */
1144 if(IBaseFilter_Release(pinInfo.pFilter) == 0 && SUCCEEDED(hr))
1145 hr = S_FALSE;
1147 if (pMemConnected)
1148 IMemInputPin_Release(pMemConnected);
1150 return hr;
1153 HRESULT OutputPin_DeliverNewSegment(OutputPin * This, REFERENCE_TIME tStart, REFERENCE_TIME tStop, double dRate)
1155 HRESULT hr;
1157 EnterCriticalSection(This->pin.pCritSec);
1159 if (!This->pin.pConnectedTo)
1160 hr = VFW_E_NOT_CONNECTED;
1161 else
1162 hr = IPin_NewSegment(This->pin.pConnectedTo, tStart, tStop, dRate);
1164 LeaveCriticalSection(This->pin.pCritSec);
1166 return hr;
1169 HRESULT OutputPin_CommitAllocator(OutputPin * This)
1171 HRESULT hr = S_OK;
1173 TRACE("(%p)->()\n", This);
1175 EnterCriticalSection(This->pin.pCritSec);
1177 if (!This->pin.pConnectedTo || !This->pMemInputPin)
1178 hr = VFW_E_NOT_CONNECTED;
1179 else
1181 IMemAllocator * pAlloc = NULL;
1183 hr = IMemInputPin_GetAllocator(This->pMemInputPin, &pAlloc);
1185 if (SUCCEEDED(hr))
1186 hr = IMemAllocator_Commit(pAlloc);
1188 if (pAlloc)
1189 IMemAllocator_Release(pAlloc);
1192 LeaveCriticalSection(This->pin.pCritSec);
1194 TRACE("--> %08x\n", hr);
1195 return hr;
1198 HRESULT OutputPin_DecommitAllocator(OutputPin * This)
1200 HRESULT hr = S_OK;
1202 TRACE("(%p)->()\n", This);
1204 EnterCriticalSection(This->pin.pCritSec);
1206 if (!This->pin.pConnectedTo || !This->pMemInputPin)
1207 hr = VFW_E_NOT_CONNECTED;
1208 else
1210 IMemAllocator * pAlloc = NULL;
1212 hr = IMemInputPin_GetAllocator(This->pMemInputPin, &pAlloc);
1214 if (SUCCEEDED(hr))
1215 hr = IMemAllocator_Decommit(pAlloc);
1217 if (pAlloc)
1218 IMemAllocator_Release(pAlloc);
1221 LeaveCriticalSection(This->pin.pCritSec);
1223 TRACE("--> %08x\n", hr);
1224 return hr;
1227 HRESULT OutputPin_DeliverDisconnect(OutputPin * This)
1229 HRESULT hr;
1231 TRACE("(%p)->()\n", This);
1233 EnterCriticalSection(This->pin.pCritSec);
1235 if (!This->pin.pConnectedTo || !This->pMemInputPin)
1236 hr = VFW_E_NOT_CONNECTED;
1237 else if (!This->custom_allocator)
1239 IMemAllocator * pAlloc = NULL;
1241 hr = IMemInputPin_GetAllocator(This->pMemInputPin, &pAlloc);
1243 if (SUCCEEDED(hr))
1244 hr = IMemAllocator_Decommit(pAlloc);
1246 if (pAlloc)
1247 IMemAllocator_Release(pAlloc);
1249 if (SUCCEEDED(hr))
1250 hr = IPin_Disconnect(This->pin.pConnectedTo);
1252 else /* Kill the allocator! */
1254 hr = IPin_Disconnect(This->pin.pConnectedTo);
1256 IPin_Disconnect((IPin *)This);
1258 LeaveCriticalSection(This->pin.pCritSec);
1260 return hr;
1264 static HRESULT PullPin_Init(const IPinVtbl *PullPin_Vtbl, const PIN_INFO * pPinInfo, SAMPLEPROC_PULL pSampleProc, LPVOID pUserData,
1265 QUERYACCEPTPROC pQueryAccept, CLEANUPPROC pCleanUp, REQUESTPROC pCustomRequest, STOPPROCESSPROC pDone, LPCRITICAL_SECTION pCritSec, PullPin * pPinImpl)
1267 /* Common attributes */
1268 pPinImpl->pin.lpVtbl = PullPin_Vtbl;
1269 pPinImpl->pin.refCount = 1;
1270 pPinImpl->pin.pConnectedTo = NULL;
1271 pPinImpl->pin.fnQueryAccept = pQueryAccept;
1272 pPinImpl->pin.pUserData = pUserData;
1273 pPinImpl->pin.pCritSec = pCritSec;
1274 Copy_PinInfo(&pPinImpl->pin.pinInfo, pPinInfo);
1275 ZeroMemory(&pPinImpl->pin.mtCurrent, sizeof(AM_MEDIA_TYPE));
1277 /* Input pin attributes */
1278 pPinImpl->fnSampleProc = pSampleProc;
1279 pPinImpl->fnCleanProc = pCleanUp;
1280 pPinImpl->fnDone = pDone;
1281 pPinImpl->fnPreConnect = NULL;
1282 pPinImpl->pAlloc = NULL;
1283 pPinImpl->pReader = NULL;
1284 pPinImpl->hThread = NULL;
1285 pPinImpl->hEventStateChanged = CreateEventW(NULL, TRUE, TRUE, NULL);
1286 pPinImpl->thread_sleepy = CreateEventW(NULL, FALSE, FALSE, NULL);
1288 pPinImpl->rtStart = 0;
1289 pPinImpl->rtCurrent = 0;
1290 pPinImpl->rtStop = ((LONGLONG)0x7fffffff << 32) | 0xffffffff;
1291 pPinImpl->dRate = 1.0;
1292 pPinImpl->state = Req_Die;
1293 pPinImpl->fnCustomRequest = pCustomRequest;
1294 pPinImpl->stop_playback = 1;
1296 InitializeCriticalSection(&pPinImpl->thread_lock);
1297 pPinImpl->thread_lock.DebugInfo->Spare[0] = (DWORD_PTR)( __FILE__ ": PullPin.thread_lock");
1299 return S_OK;
1302 HRESULT PullPin_Construct(const IPinVtbl *PullPin_Vtbl, const PIN_INFO * pPinInfo, SAMPLEPROC_PULL pSampleProc, LPVOID pUserData, QUERYACCEPTPROC pQueryAccept, CLEANUPPROC pCleanUp, REQUESTPROC pCustomRequest, STOPPROCESSPROC pDone, LPCRITICAL_SECTION pCritSec, IPin ** ppPin)
1304 PullPin * pPinImpl;
1306 *ppPin = NULL;
1308 if (pPinInfo->dir != PINDIR_INPUT)
1310 ERR("Pin direction(%x) != PINDIR_INPUT\n", pPinInfo->dir);
1311 return E_INVALIDARG;
1314 pPinImpl = CoTaskMemAlloc(sizeof(*pPinImpl));
1316 if (!pPinImpl)
1317 return E_OUTOFMEMORY;
1319 if (SUCCEEDED(PullPin_Init(PullPin_Vtbl, pPinInfo, pSampleProc, pUserData, pQueryAccept, pCleanUp, pCustomRequest, pDone, pCritSec, pPinImpl)))
1321 *ppPin = (IPin *)(&pPinImpl->pin.lpVtbl);
1322 return S_OK;
1325 CoTaskMemFree(pPinImpl);
1326 return E_FAIL;
1329 static HRESULT PullPin_InitProcessing(PullPin * This);
1331 HRESULT WINAPI PullPin_ReceiveConnection(IPin * iface, IPin * pReceivePin, const AM_MEDIA_TYPE * pmt)
1333 PIN_DIRECTION pindirReceive;
1334 HRESULT hr = S_OK;
1335 PullPin *This = (PullPin *)iface;
1337 TRACE("(%p/%p)->(%p, %p)\n", This, iface, pReceivePin, pmt);
1338 dump_AM_MEDIA_TYPE(pmt);
1340 EnterCriticalSection(This->pin.pCritSec);
1341 if (!This->pin.pConnectedTo)
1343 ALLOCATOR_PROPERTIES props;
1345 props.cBuffers = 3;
1346 props.cbBuffer = 64 * 1024; /* 64k bytes */
1347 props.cbAlign = 1;
1348 props.cbPrefix = 0;
1350 if (SUCCEEDED(hr) && (This->pin.fnQueryAccept(This->pin.pUserData, pmt) != S_OK))
1351 hr = VFW_E_TYPE_NOT_ACCEPTED; /* FIXME: shouldn't we just map common errors onto
1352 * VFW_E_TYPE_NOT_ACCEPTED and pass the value on otherwise? */
1354 if (SUCCEEDED(hr))
1356 IPin_QueryDirection(pReceivePin, &pindirReceive);
1358 if (pindirReceive != PINDIR_OUTPUT)
1360 ERR("Can't connect from non-output pin\n");
1361 hr = VFW_E_INVALID_DIRECTION;
1365 This->pReader = NULL;
1366 This->pAlloc = NULL;
1367 if (SUCCEEDED(hr))
1369 hr = IPin_QueryInterface(pReceivePin, &IID_IAsyncReader, (LPVOID *)&This->pReader);
1372 if (SUCCEEDED(hr) && This->fnPreConnect)
1374 hr = This->fnPreConnect(iface, pReceivePin, &props);
1377 if (SUCCEEDED(hr))
1379 hr = IAsyncReader_RequestAllocator(This->pReader, NULL, &props, &This->pAlloc);
1382 if (SUCCEEDED(hr))
1384 CopyMediaType(&This->pin.mtCurrent, pmt);
1385 This->pin.pConnectedTo = pReceivePin;
1386 IPin_AddRef(pReceivePin);
1387 hr = IMemAllocator_Commit(This->pAlloc);
1390 if (SUCCEEDED(hr))
1391 hr = PullPin_InitProcessing(This);
1393 if (FAILED(hr))
1395 if (This->pReader)
1396 IAsyncReader_Release(This->pReader);
1397 This->pReader = NULL;
1398 if (This->pAlloc)
1399 IMemAllocator_Release(This->pAlloc);
1400 This->pAlloc = NULL;
1403 else
1404 hr = VFW_E_ALREADY_CONNECTED;
1405 LeaveCriticalSection(This->pin.pCritSec);
1406 return hr;
1409 HRESULT WINAPI PullPin_QueryInterface(IPin * iface, REFIID riid, LPVOID * ppv)
1411 PullPin *This = (PullPin *)iface;
1413 TRACE("(%p/%p)->(%s, %p)\n", This, iface, qzdebugstr_guid(riid), ppv);
1415 *ppv = NULL;
1417 if (IsEqualIID(riid, &IID_IUnknown))
1418 *ppv = (LPVOID)iface;
1419 else if (IsEqualIID(riid, &IID_IPin))
1420 *ppv = (LPVOID)iface;
1421 else if (IsEqualIID(riid, &IID_IMediaSeeking))
1423 return IBaseFilter_QueryInterface(This->pin.pinInfo.pFilter, &IID_IMediaSeeking, ppv);
1426 if (*ppv)
1428 IUnknown_AddRef((IUnknown *)(*ppv));
1429 return S_OK;
1432 FIXME("No interface for %s!\n", qzdebugstr_guid(riid));
1434 return E_NOINTERFACE;
1437 ULONG WINAPI PullPin_Release(IPin *iface)
1439 PullPin *This = (PullPin *)iface;
1440 ULONG refCount = InterlockedDecrement(&This->pin.refCount);
1442 TRACE("(%p)->() Release from %d\n", This, refCount + 1);
1444 if (!refCount)
1446 WaitForSingleObject(This->hEventStateChanged, INFINITE);
1447 assert(!This->hThread);
1449 if(This->pAlloc)
1450 IMemAllocator_Release(This->pAlloc);
1451 if(This->pReader)
1452 IAsyncReader_Release(This->pReader);
1453 CloseHandle(This->thread_sleepy);
1454 CloseHandle(This->hEventStateChanged);
1455 This->thread_lock.DebugInfo->Spare[0] = 0;
1456 DeleteCriticalSection(&This->thread_lock);
1457 CoTaskMemFree(This);
1458 return 0;
1460 return refCount;
1463 static void CALLBACK PullPin_Flush(PullPin *This)
1465 IMediaSample *pSample;
1466 TRACE("Flushing!\n");
1468 if (This->pReader)
1470 /* Flush outstanding samples */
1471 IAsyncReader_BeginFlush(This->pReader);
1473 for (;;)
1475 DWORD_PTR dwUser;
1477 IAsyncReader_WaitForNext(This->pReader, 0, &pSample, &dwUser);
1479 if (!pSample)
1480 break;
1482 assert(!IMediaSample_GetActualDataLength(pSample));
1484 IMediaSample_Release(pSample);
1487 IAsyncReader_EndFlush(This->pReader);
1491 static void CALLBACK PullPin_Thread_Process(PullPin *This)
1493 HRESULT hr;
1494 IMediaSample * pSample = NULL;
1495 ALLOCATOR_PROPERTIES allocProps;
1497 hr = IMemAllocator_GetProperties(This->pAlloc, &allocProps);
1499 This->cbAlign = allocProps.cbAlign;
1501 if (This->rtCurrent < This->rtStart)
1502 This->rtCurrent = MEDIATIME_FROM_BYTES(ALIGNDOWN(BYTES_FROM_MEDIATIME(This->rtStart), This->cbAlign));
1504 TRACE("Start\n");
1506 if (This->rtCurrent >= This->rtStop)
1508 IPin_EndOfStream((IPin *)This);
1509 return;
1512 /* There is no sample in our buffer */
1513 hr = This->fnCustomRequest(This->pin.pUserData);
1515 if (FAILED(hr))
1516 ERR("Request error: %x\n", hr);
1518 EnterCriticalSection(This->pin.pCritSec);
1519 SetEvent(This->hEventStateChanged);
1520 LeaveCriticalSection(This->pin.pCritSec);
1522 if (SUCCEEDED(hr))
1525 DWORD_PTR dwUser;
1527 TRACE("Process sample\n");
1529 hr = IAsyncReader_WaitForNext(This->pReader, 10000, &pSample, &dwUser);
1531 /* Return an empty sample on error to the implementation in case it does custom parsing, so it knows it's gone */
1532 if (SUCCEEDED(hr))
1534 hr = This->fnSampleProc(This->pin.pUserData, pSample, dwUser);
1536 else
1538 /* FIXME: This is not well handled yet! */
1539 ERR("Processing error: %x\n", hr);
1542 if (pSample)
1544 IMediaSample_Release(pSample);
1545 pSample = NULL;
1547 } while (This->rtCurrent < This->rtStop && hr == S_OK && !This->stop_playback);
1549 /* Sample was rejected, and we are asked to terminate */
1550 if (pSample)
1552 IMediaSample_Release(pSample);
1555 /* Can't reset state to Sleepy here because that might race, instead PauseProcessing will do that for us
1556 * Flush remaining samples
1558 if (This->fnDone)
1559 This->fnDone(This->pin.pUserData);
1561 TRACE("End: %08x, %d\n", hr, This->stop_playback);
1564 static void CALLBACK PullPin_Thread_Pause(PullPin *This)
1566 PullPin_Flush(This);
1568 EnterCriticalSection(This->pin.pCritSec);
1569 This->state = Req_Sleepy;
1570 SetEvent(This->hEventStateChanged);
1571 LeaveCriticalSection(This->pin.pCritSec);
1574 static void CALLBACK PullPin_Thread_Stop(PullPin *This)
1576 TRACE("(%p)->()\n", This);
1578 EnterCriticalSection(This->pin.pCritSec);
1580 CloseHandle(This->hThread);
1581 This->hThread = NULL;
1582 SetEvent(This->hEventStateChanged);
1584 LeaveCriticalSection(This->pin.pCritSec);
1586 IBaseFilter_Release(This->pin.pinInfo.pFilter);
1588 CoUninitialize();
1589 ExitThread(0);
1592 static DWORD WINAPI PullPin_Thread_Main(LPVOID pv)
1594 PullPin *This = pv;
1595 CoInitializeEx(NULL, COINIT_MULTITHREADED);
1597 PullPin_Flush(This);
1599 for (;;)
1601 WaitForSingleObject(This->thread_sleepy, INFINITE);
1603 TRACE("State: %d\n", This->state);
1605 switch (This->state)
1607 case Req_Die: PullPin_Thread_Stop(This); break;
1608 case Req_Run: PullPin_Thread_Process(This); break;
1609 case Req_Pause: PullPin_Thread_Pause(This); break;
1610 case Req_Sleepy: ERR("Should not be signalled with SLEEPY!\n"); break;
1611 default: ERR("Unknown state request: %d\n", This->state); break;
1616 static HRESULT PullPin_InitProcessing(PullPin * This)
1618 HRESULT hr = S_OK;
1620 TRACE("(%p)->()\n", This);
1622 /* if we are connected */
1623 if (This->pAlloc)
1625 DWORD dwThreadId;
1627 WaitForSingleObject(This->hEventStateChanged, INFINITE);
1628 EnterCriticalSection(This->pin.pCritSec);
1630 assert(!This->hThread);
1631 assert(This->state == Req_Die);
1632 assert(This->stop_playback);
1633 assert(WaitForSingleObject(This->thread_sleepy, 0) == WAIT_TIMEOUT);
1634 This->state = Req_Sleepy;
1636 /* AddRef the filter to make sure it and it's pins will be around
1637 * as long as the thread */
1638 IBaseFilter_AddRef(This->pin.pinInfo.pFilter);
1641 This->hThread = CreateThread(NULL, 0, PullPin_Thread_Main, This, 0, &dwThreadId);
1642 if (!This->hThread)
1644 hr = HRESULT_FROM_WIN32(GetLastError());
1645 IBaseFilter_Release(This->pin.pinInfo.pFilter);
1648 if (SUCCEEDED(hr))
1650 SetEvent(This->hEventStateChanged);
1651 /* If assert fails, that means a command was not processed before the thread previously terminated */
1653 LeaveCriticalSection(This->pin.pCritSec);
1656 TRACE(" -- %x\n", hr);
1658 return hr;
1661 HRESULT PullPin_StartProcessing(PullPin * This)
1663 /* if we are connected */
1664 TRACE("(%p)->()\n", This);
1665 if(This->pAlloc)
1667 assert(This->hThread);
1669 PullPin_WaitForStateChange(This, INFINITE);
1671 assert(This->state == Req_Sleepy);
1673 /* Wake up! */
1674 assert(WaitForSingleObject(This->thread_sleepy, 0) == WAIT_TIMEOUT);
1675 This->state = Req_Run;
1676 This->stop_playback = 0;
1677 ResetEvent(This->hEventStateChanged);
1678 SetEvent(This->thread_sleepy);
1681 return S_OK;
1684 HRESULT PullPin_PauseProcessing(PullPin * This)
1686 /* if we are connected */
1687 TRACE("(%p)->()\n", This);
1688 if(This->pAlloc)
1690 assert(This->hThread);
1692 PullPin_WaitForStateChange(This, INFINITE);
1694 EnterCriticalSection(This->pin.pCritSec);
1696 assert(!This->stop_playback);
1697 assert(This->state == Req_Run|| This->state == Req_Sleepy);
1699 assert(WaitForSingleObject(This->thread_sleepy, 0) == WAIT_TIMEOUT);
1700 This->state = Req_Pause;
1701 This->stop_playback = 1;
1702 ResetEvent(This->hEventStateChanged);
1703 SetEvent(This->thread_sleepy);
1705 LeaveCriticalSection(This->pin.pCritSec);
1708 return S_OK;
1711 static HRESULT PullPin_StopProcessing(PullPin * This)
1713 TRACE("(%p)->()\n", This);
1715 /* if we are alive */
1716 assert(This->hThread);
1718 PullPin_WaitForStateChange(This, INFINITE);
1720 assert(This->state == Req_Pause || This->state == Req_Sleepy);
1722 This->stop_playback = 1;
1723 This->state = Req_Die;
1724 assert(WaitForSingleObject(This->thread_sleepy, 0) == WAIT_TIMEOUT);
1725 ResetEvent(This->hEventStateChanged);
1726 SetEvent(This->thread_sleepy);
1727 return S_OK;
1730 HRESULT PullPin_WaitForStateChange(PullPin * This, DWORD dwMilliseconds)
1732 if (WaitForSingleObject(This->hEventStateChanged, dwMilliseconds) == WAIT_TIMEOUT)
1733 return S_FALSE;
1734 return S_OK;
1737 HRESULT WINAPI PullPin_EndOfStream(IPin * iface)
1739 FIXME("(%p)->() stub\n", iface);
1741 return SendFurther( iface, deliver_endofstream, NULL, NULL );
1744 HRESULT WINAPI PullPin_BeginFlush(IPin * iface)
1746 PullPin *This = (PullPin *)iface;
1747 TRACE("(%p)->()\n", This);
1749 EnterCriticalSection(This->pin.pCritSec);
1751 SendFurther( iface, deliver_beginflush, NULL, NULL );
1753 LeaveCriticalSection(This->pin.pCritSec);
1755 EnterCriticalSection(&This->thread_lock);
1757 if (This->pReader)
1758 IAsyncReader_BeginFlush(This->pReader);
1759 PullPin_WaitForStateChange(This, INFINITE);
1761 if (This->hThread && This->state == Req_Run)
1763 PullPin_PauseProcessing(This);
1764 PullPin_WaitForStateChange(This, INFINITE);
1767 LeaveCriticalSection(&This->thread_lock);
1769 EnterCriticalSection(This->pin.pCritSec);
1771 This->fnCleanProc(This->pin.pUserData);
1773 LeaveCriticalSection(This->pin.pCritSec);
1775 return S_OK;
1778 HRESULT WINAPI PullPin_EndFlush(IPin * iface)
1780 PullPin *This = (PullPin *)iface;
1782 TRACE("(%p)->()\n", iface);
1784 /* Send further first: Else a race condition might terminate processing early */
1785 EnterCriticalSection(This->pin.pCritSec);
1786 SendFurther( iface, deliver_endflush, NULL, NULL );
1787 LeaveCriticalSection(This->pin.pCritSec);
1789 EnterCriticalSection(&This->thread_lock);
1791 FILTER_STATE state;
1792 IBaseFilter_GetState(This->pin.pinInfo.pFilter, INFINITE, &state);
1794 if (state != State_Stopped)
1795 PullPin_StartProcessing(This);
1797 PullPin_WaitForStateChange(This, INFINITE);
1799 LeaveCriticalSection(&This->thread_lock);
1801 return S_OK;
1804 HRESULT WINAPI PullPin_Disconnect(IPin *iface)
1806 HRESULT hr;
1807 PullPin *This = (PullPin *)iface;
1809 TRACE("()\n");
1811 EnterCriticalSection(This->pin.pCritSec);
1813 if (FAILED(hr = IMemAllocator_Decommit(This->pAlloc)))
1814 ERR("Allocator decommit failed with error %x. Possible memory leak\n", hr);
1816 if (This->pin.pConnectedTo)
1818 IPin_Release(This->pin.pConnectedTo);
1819 This->pin.pConnectedTo = NULL;
1820 PullPin_StopProcessing(This);
1822 FreeMediaType(&This->pin.mtCurrent);
1823 ZeroMemory(&This->pin.mtCurrent, sizeof(This->pin.mtCurrent));
1824 hr = S_OK;
1826 else
1827 hr = S_FALSE;
1829 LeaveCriticalSection(This->pin.pCritSec);
1831 return hr;
1834 HRESULT WINAPI PullPin_NewSegment(IPin * iface, REFERENCE_TIME tStart, REFERENCE_TIME tStop, double dRate)
1836 newsegmentargs args;
1837 FIXME("(%p)->(%s, %s, %g) stub\n", iface, wine_dbgstr_longlong(tStart), wine_dbgstr_longlong(tStop), dRate);
1839 args.tStart = tStart;
1840 args.tStop = tStop;
1841 args.rate = dRate;
1843 return SendFurther( iface, deliver_newsegment, &args, NULL );
1846 static const IPinVtbl PullPin_Vtbl =
1848 PullPin_QueryInterface,
1849 IPinImpl_AddRef,
1850 PullPin_Release,
1851 InputPin_Connect,
1852 PullPin_ReceiveConnection,
1853 PullPin_Disconnect,
1854 IPinImpl_ConnectedTo,
1855 IPinImpl_ConnectionMediaType,
1856 IPinImpl_QueryPinInfo,
1857 IPinImpl_QueryDirection,
1858 IPinImpl_QueryId,
1859 IPinImpl_QueryAccept,
1860 IPinImpl_EnumMediaTypes,
1861 IPinImpl_QueryInternalConnections,
1862 PullPin_EndOfStream,
1863 PullPin_BeginFlush,
1864 PullPin_EndFlush,
1865 PullPin_NewSegment