Correct the test for the ODS_SELECTED bit in the WM_DRAWITEM message
[wine] / dlls / quartz / avisplit.c
1 /*
2  * AVI Splitter Filter
3  *
4  * Copyright 2003 Robert Shearman
5  *
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.
10  *
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.
15  *
16  * You should have received a copy of the GNU Lesser General Public
17  * License along with this library; if not, write to the Free Software
18  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
19  */
20 /* FIXME:
21  * - we don't do anything with indices yet (we could use them when seeking)
22  * - we don't support multiple RIFF sections (i.e. large AVI files > 2Gb)
23  */
24
25 #include "quartz_private.h"
26 #include "control_private.h"
27 #include "pin.h"
28
29 #include "uuids.h"
30 #include "aviriff.h"
31 #include "mmreg.h"
32 #include "vfwmsgs.h"
33 #include "amvideo.h"
34
35 #include "fourcc.h"
36
37 #include "wine/unicode.h"
38 #include "wine/debug.h"
39
40 #include <math.h>
41 #include <assert.h>
42
43 WINE_DEFAULT_DEBUG_CHANNEL(quartz);
44
45 static const WCHAR wcsInputPinName[] = {'i','n','p','u','t',' ','p','i','n',0};
46 static const struct IBaseFilterVtbl AVISplitter_Vtbl;
47 static const struct IMediaSeekingVtbl AVISplitter_Seeking_Vtbl;
48 static const struct IPinVtbl AVISplitter_OutputPin_Vtbl;
49 static const struct IPinVtbl AVISplitter_InputPin_Vtbl;
50
51 static HRESULT AVISplitter_Sample(LPVOID iface, IMediaSample * pSample);
52 static HRESULT AVISplitter_OutputPin_QueryAccept(LPVOID iface, const AM_MEDIA_TYPE * pmt);
53 static HRESULT AVISplitter_QueryAccept(LPVOID iface, const AM_MEDIA_TYPE * pmt);
54 static HRESULT AVISplitter_InputPin_PreConnect(IPin * iface, IPin * pConnectPin);
55 static HRESULT AVISplitter_ChangeStart(LPVOID iface);
56 static HRESULT AVISplitter_ChangeStop(LPVOID iface);
57 static HRESULT AVISplitter_ChangeRate(LPVOID iface);
58
59 static HRESULT AVISplitter_InputPin_Construct(const PIN_INFO * pPinInfo, SAMPLEPROC pSampleProc, LPVOID pUserData, QUERYACCEPTPROC pQueryAccept, LPCRITICAL_SECTION pCritSec, IPin ** ppPin);
60
61 typedef struct AVISplitter
62 {
63     const IBaseFilterVtbl * lpVtbl;
64
65     ULONG refCount;
66     CRITICAL_SECTION csFilter;
67     FILTER_STATE state;
68     REFERENCE_TIME rtStreamStart;
69     IReferenceClock * pClock;
70     FILTER_INFO filterInfo;
71
72     PullPin * pInputPin;
73     ULONG cStreams;
74     IPin ** ppPins;
75     IMediaSample * pCurrentSample;
76     RIFFCHUNK CurrentChunk;
77     LONGLONG CurrentChunkOffset; /* in media time */
78     LONGLONG EndOfFile;
79     AVIMAINHEADER AviHeader;
80 } AVISplitter;
81
82 typedef struct AVISplitter_OutputPin
83 {
84     OutputPin pin;
85
86     AM_MEDIA_TYPE * pmt;
87     float fSamplesPerSec;
88     DWORD dwSamplesProcessed;
89     DWORD dwSampleSize;
90     DWORD dwLength;
91     MediaSeekingImpl mediaSeeking;
92 } AVISplitter_OutputPin;
93
94
95 #define _IMediaSeeking_Offset ((int)(&(((AVISplitter_OutputPin*)0)->mediaSeeking)))
96 #define ICOM_THIS_From_IMediaSeeking(impl, iface) impl* This = (impl*)(((char*)iface)-_IMediaSeeking_Offset);
97
98 HRESULT AVISplitter_create(IUnknown * pUnkOuter, LPVOID * ppv)
99 {
100     HRESULT hr;
101     PIN_INFO piInput;
102     AVISplitter * pAviSplit;
103
104     TRACE("(%p, %p)\n", pUnkOuter, ppv);
105
106     *ppv = NULL;
107
108     if (pUnkOuter)
109         return CLASS_E_NOAGGREGATION;
110     
111     pAviSplit = CoTaskMemAlloc(sizeof(AVISplitter));
112
113     pAviSplit->lpVtbl = &AVISplitter_Vtbl;
114     pAviSplit->refCount = 1;
115     InitializeCriticalSection(&pAviSplit->csFilter);
116     pAviSplit->state = State_Stopped;
117     pAviSplit->pClock = NULL;
118     pAviSplit->pCurrentSample = NULL;
119     ZeroMemory(&pAviSplit->filterInfo, sizeof(FILTER_INFO));
120
121     pAviSplit->cStreams = 0;
122     pAviSplit->ppPins = CoTaskMemAlloc(1 * sizeof(IPin *));
123
124     /* construct input pin */
125     piInput.dir = PINDIR_INPUT;
126     piInput.pFilter = (IBaseFilter *)pAviSplit;
127     strncpyW(piInput.achName, wcsInputPinName, sizeof(piInput.achName) / sizeof(piInput.achName[0]));
128
129     hr = AVISplitter_InputPin_Construct(&piInput, AVISplitter_Sample, (LPVOID)pAviSplit, AVISplitter_QueryAccept, &pAviSplit->csFilter, (IPin **)&pAviSplit->pInputPin);
130
131     if (SUCCEEDED(hr))
132     {
133         pAviSplit->ppPins[0] = (IPin *)pAviSplit->pInputPin;
134         pAviSplit->pInputPin->fnPreConnect = AVISplitter_InputPin_PreConnect;
135         *ppv = (LPVOID)pAviSplit;
136     }
137     else
138     {
139         CoTaskMemFree(pAviSplit->ppPins);
140         DeleteCriticalSection(&pAviSplit->csFilter);
141         CoTaskMemFree(pAviSplit);
142     }
143
144     return hr;
145 }
146
147 static HRESULT AVISplitter_OutputPin_Init(const PIN_INFO * pPinInfo, ALLOCATOR_PROPERTIES * props, LPVOID pUserData, QUERYACCEPTPROC pQueryAccept, const AM_MEDIA_TYPE * pmt, float fSamplesPerSec, LPCRITICAL_SECTION pCritSec, AVISplitter_OutputPin * pPinImpl)
148 {
149     pPinImpl->pmt = CoTaskMemAlloc(sizeof(AM_MEDIA_TYPE));
150     CopyMediaType(pPinImpl->pmt, pmt);
151     pPinImpl->dwSamplesProcessed = 0;
152     pPinImpl->dwSampleSize = 0;
153     pPinImpl->fSamplesPerSec = fSamplesPerSec;
154
155     MediaSeekingImpl_Init((LPVOID)pPinInfo->pFilter, AVISplitter_ChangeStop, AVISplitter_ChangeStart, AVISplitter_ChangeRate, &pPinImpl->mediaSeeking);
156     pPinImpl->mediaSeeking.lpVtbl = &AVISplitter_Seeking_Vtbl;
157
158     return OutputPin_Init(pPinInfo, props, pUserData, pQueryAccept, pCritSec, &pPinImpl->pin);
159 }
160
161 static HRESULT AVISplitter_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)
162 {
163     AVISplitter_OutputPin * pPinImpl;
164
165     *ppPin = NULL;
166
167     assert(pPinInfo->dir == PINDIR_OUTPUT);
168
169     pPinImpl = CoTaskMemAlloc(sizeof(AVISplitter_OutputPin));
170
171     if (!pPinImpl)
172         return E_OUTOFMEMORY;
173
174     if (SUCCEEDED(AVISplitter_OutputPin_Init(pPinInfo, props, pUserData, pQueryAccept, pmt, fSamplesPerSec, pCritSec, pPinImpl)))
175     {
176         pPinImpl->pin.pin.lpVtbl = &AVISplitter_OutputPin_Vtbl;
177         
178         *ppPin = (IPin *)pPinImpl;
179         return S_OK;
180     }
181     return E_FAIL;
182 }
183
184 static HRESULT WINAPI AVISplitter_QueryInterface(IBaseFilter * iface, REFIID riid, LPVOID * ppv)
185 {
186     ICOM_THIS(AVISplitter, iface);
187     TRACE("(%s, %p)\n", qzdebugstr_guid(riid), ppv);
188
189     *ppv = NULL;
190
191     if (IsEqualIID(riid, &IID_IUnknown))
192         *ppv = (LPVOID)This;
193     else if (IsEqualIID(riid, &IID_IPersist))
194         *ppv = (LPVOID)This;
195     else if (IsEqualIID(riid, &IID_IMediaFilter))
196         *ppv = (LPVOID)This;
197     else if (IsEqualIID(riid, &IID_IBaseFilter))
198         *ppv = (LPVOID)This;
199
200     if (*ppv)
201     {
202         IUnknown_AddRef((IUnknown *)(*ppv));
203         return S_OK;
204     }
205
206     FIXME("No interface for %s!\n", qzdebugstr_guid(riid));
207
208     return E_NOINTERFACE;
209 }
210
211 static ULONG WINAPI AVISplitter_AddRef(IBaseFilter * iface)
212 {
213     ICOM_THIS(AVISplitter, iface);
214     TRACE("()\n");
215     return InterlockedIncrement(&This->refCount);
216 }
217
218 static ULONG WINAPI AVISplitter_Release(IBaseFilter * iface)
219 {
220     ICOM_THIS(AVISplitter, iface);
221     TRACE("()\n");
222     if (!InterlockedDecrement(&This->refCount))
223     {
224         ULONG i;
225
226         DeleteCriticalSection(&This->csFilter);
227         IReferenceClock_Release(This->pClock);
228         
229         for (i = 0; i < This->cStreams + 1; i++)
230             IPin_Release(This->ppPins[i]);
231         
232         HeapFree(GetProcessHeap(), 0, This->ppPins);
233         This->lpVtbl = NULL;
234         
235         TRACE("Destroying AVI splitter\n");
236         CoTaskMemFree(This);
237         
238         return 0;
239     }
240     else
241         return This->refCount;
242 }
243
244 /** IPersist methods **/
245
246 static HRESULT WINAPI AVISplitter_GetClassID(IBaseFilter * iface, CLSID * pClsid)
247 {
248     TRACE("(%p)\n", pClsid);
249
250     *pClsid = CLSID_AviSplitter;
251
252     return S_OK;
253 }
254
255 /** IMediaFilter methods **/
256
257 static HRESULT WINAPI AVISplitter_Stop(IBaseFilter * iface)
258 {
259     HRESULT hr;
260     ICOM_THIS(AVISplitter, iface);
261
262     TRACE("()\n");
263
264     EnterCriticalSection(&This->csFilter);
265     {
266         hr = PullPin_StopProcessing(This->pInputPin);
267
268         This->state = State_Stopped;
269     }
270     LeaveCriticalSection(&This->csFilter);
271     
272     return hr;
273 }
274
275 static HRESULT WINAPI AVISplitter_Pause(IBaseFilter * iface)
276 {
277     HRESULT hr = S_OK;
278     BOOL bInit;
279     ICOM_THIS(AVISplitter, iface);
280     
281     TRACE("()\n");
282
283     EnterCriticalSection(&This->csFilter);
284     {
285         bInit = (This->state == State_Stopped);
286         This->state = State_Paused;
287     }
288     LeaveCriticalSection(&This->csFilter);
289
290     if (bInit)
291     {
292         unsigned int i;
293
294         hr = PullPin_Seek(This->pInputPin, This->CurrentChunkOffset, This->EndOfFile);
295
296         if (SUCCEEDED(hr))
297             hr = PullPin_InitProcessing(This->pInputPin);
298
299         if (SUCCEEDED(hr))
300         {
301             for (i = 1; i < This->cStreams + 1; i++)
302             {
303                 OutputPin_DeliverNewSegment((OutputPin *)This->ppPins[i], 0, (LONGLONG)ceil(10000000.0 * (float)((AVISplitter_OutputPin *)This->ppPins[i])->dwLength / ((AVISplitter_OutputPin *)This->ppPins[i])->fSamplesPerSec), 1.0);
304                 ((AVISplitter_OutputPin *)This->ppPins[i])->mediaSeeking.llDuration = (LONGLONG)ceil(10000000.0 * (float)((AVISplitter_OutputPin *)This->ppPins[i])->dwLength / ((AVISplitter_OutputPin *)This->ppPins[i])->fSamplesPerSec);
305                 ((AVISplitter_OutputPin *)This->ppPins[i])->mediaSeeking.llStop = (LONGLONG)ceil(10000000.0 * (float)((AVISplitter_OutputPin *)This->ppPins[i])->dwLength / ((AVISplitter_OutputPin *)This->ppPins[i])->fSamplesPerSec);
306                 OutputPin_CommitAllocator((OutputPin *)This->ppPins[i]);
307             }
308
309             /* FIXME: this is a little hacky: we have to deliver (at least?) one sample
310              * to each renderer before they will complete their transitions. We should probably
311              * seek through the stream for the first of each, rather than do it this way which is
312              * probably a bit prone to deadlocking */
313             hr = PullPin_StartProcessing(This->pInputPin);
314         }
315     }
316     /* FIXME: else pause thread */
317
318     return hr;
319 }
320
321 static HRESULT WINAPI AVISplitter_Run(IBaseFilter * iface, REFERENCE_TIME tStart)
322 {
323     HRESULT hr = S_OK;
324     ICOM_THIS(AVISplitter, iface);
325
326     TRACE("(%s)\n", wine_dbgstr_longlong(tStart));
327
328     EnterCriticalSection(&This->csFilter);
329     {
330         This->rtStreamStart = tStart;
331
332         This->state = State_Running;
333     }
334     LeaveCriticalSection(&This->csFilter);
335
336     return hr;
337 }
338
339 static HRESULT WINAPI AVISplitter_GetState(IBaseFilter * iface, DWORD dwMilliSecsTimeout, FILTER_STATE *pState)
340 {
341     ICOM_THIS(AVISplitter, iface);
342
343     TRACE("(%ld, %p)\n", dwMilliSecsTimeout, pState);
344
345     EnterCriticalSection(&This->csFilter);
346     {
347         *pState = This->state;
348     }
349     LeaveCriticalSection(&This->csFilter);
350
351     /* FIXME: this is a little bit unsafe, but I don't see that we can do this
352      * while in the critical section. Maybe we could copy the pointer and addref in the
353      * critical section and then release after this.
354      */
355     if (This->pInputPin && (PullPin_WaitForStateChange(This->pInputPin, dwMilliSecsTimeout) == S_FALSE))
356         return VFW_S_STATE_INTERMEDIATE;
357
358     return S_OK;
359 }
360
361 static HRESULT WINAPI AVISplitter_SetSyncSource(IBaseFilter * iface, IReferenceClock *pClock)
362 {
363     ICOM_THIS(AVISplitter, iface);
364
365     TRACE("(%p)\n", pClock);
366
367     EnterCriticalSection(&This->csFilter);
368     {
369         if (This->pClock)
370             IReferenceClock_Release(This->pClock);
371         This->pClock = pClock;
372         if (This->pClock)
373             IReferenceClock_AddRef(This->pClock);
374     }
375     LeaveCriticalSection(&This->csFilter);
376
377     return S_OK;
378 }
379
380 static HRESULT WINAPI AVISplitter_GetSyncSource(IBaseFilter * iface, IReferenceClock **ppClock)
381 {
382     ICOM_THIS(AVISplitter, iface);
383
384     TRACE("(%p)\n", ppClock);
385
386     EnterCriticalSection(&This->csFilter);
387     {
388         *ppClock = This->pClock;
389         IReferenceClock_AddRef(This->pClock);
390     }
391     LeaveCriticalSection(&This->csFilter);
392     
393     return S_OK;
394 }
395
396 /** IBaseFilter implementation **/
397
398 static HRESULT WINAPI AVISplitter_EnumPins(IBaseFilter * iface, IEnumPins **ppEnum)
399 {
400     ENUMPINDETAILS epd;
401     ICOM_THIS(AVISplitter, iface);
402
403     TRACE("(%p)\n", ppEnum);
404
405     epd.cPins = This->cStreams + 1; /* +1 for input pin */
406     epd.ppPins = This->ppPins;
407     return IEnumPinsImpl_Construct(&epd, ppEnum);
408 }
409
410 static HRESULT WINAPI AVISplitter_FindPin(IBaseFilter * iface, LPCWSTR Id, IPin **ppPin)
411 {
412     FIXME("AVISplitter::FindPin(...)\n");
413
414     /* FIXME: critical section */
415
416     return E_NOTIMPL;
417 }
418
419 static HRESULT WINAPI AVISplitter_QueryFilterInfo(IBaseFilter * iface, FILTER_INFO *pInfo)
420 {
421     ICOM_THIS(AVISplitter, iface);
422
423     TRACE("(%p)\n", pInfo);
424
425     strcpyW(pInfo->achName, This->filterInfo.achName);
426     pInfo->pGraph = This->filterInfo.pGraph;
427
428     if (pInfo->pGraph)
429         IFilterGraph_AddRef(pInfo->pGraph);
430     
431     return S_OK;
432 }
433
434 static HRESULT WINAPI AVISplitter_JoinFilterGraph(IBaseFilter * iface, IFilterGraph *pGraph, LPCWSTR pName)
435 {
436     HRESULT hr = S_OK;
437     ICOM_THIS(AVISplitter, iface);
438
439     TRACE("(%p, %s)\n", pGraph, debugstr_w(pName));
440
441     EnterCriticalSection(&This->csFilter);
442     {
443         if (pName)
444             strcpyW(This->filterInfo.achName, pName);
445         else
446             *This->filterInfo.achName = '\0';
447         This->filterInfo.pGraph = pGraph; /* NOTE: do NOT increase ref. count */
448     }
449     LeaveCriticalSection(&This->csFilter);
450
451     return hr;
452 }
453
454 static HRESULT WINAPI AVISplitter_QueryVendorInfo(IBaseFilter * iface, LPWSTR *pVendorInfo)
455 {
456     TRACE("(%p)\n", pVendorInfo);
457     return E_NOTIMPL;
458 }
459
460 static const IBaseFilterVtbl AVISplitter_Vtbl =
461 {
462     ICOM_MSVTABLE_COMPAT_DummyRTTIVALUE
463     AVISplitter_QueryInterface,
464     AVISplitter_AddRef,
465     AVISplitter_Release,
466     AVISplitter_GetClassID,
467     AVISplitter_Stop,
468     AVISplitter_Pause,
469     AVISplitter_Run,
470     AVISplitter_GetState,
471     AVISplitter_SetSyncSource,
472     AVISplitter_GetSyncSource,
473     AVISplitter_EnumPins,
474     AVISplitter_FindPin,
475     AVISplitter_QueryFilterInfo,
476     AVISplitter_JoinFilterGraph,
477     AVISplitter_QueryVendorInfo
478 };
479
480 static HRESULT AVISplitter_NextChunk(LONGLONG * pllCurrentChunkOffset, RIFFCHUNK * pCurrentChunk, const REFERENCE_TIME * tStart, const REFERENCE_TIME * tStop, const BYTE * pbSrcStream)
481 {
482     *pllCurrentChunkOffset += MEDIATIME_FROM_BYTES(sizeof(RIFFCHUNK) + RIFFROUND(pCurrentChunk->cb));
483
484     if (*pllCurrentChunkOffset > *tStop)
485         return S_FALSE; /* no more data - we couldn't even get the next chunk header! */
486     else if (*pllCurrentChunkOffset + MEDIATIME_FROM_BYTES(sizeof(RIFFCHUNK)) >= *tStop)
487     {
488         memcpy(pCurrentChunk, pbSrcStream + (DWORD)BYTES_FROM_MEDIATIME(*pllCurrentChunkOffset - *tStart), (DWORD)BYTES_FROM_MEDIATIME(*tStop - *pllCurrentChunkOffset));
489         return S_FALSE; /* no more data */
490     }
491     else
492         memcpy(pCurrentChunk, pbSrcStream + (DWORD)BYTES_FROM_MEDIATIME(*pllCurrentChunkOffset - *tStart), sizeof(RIFFCHUNK));
493
494     return S_OK;
495 }
496
497 static HRESULT AVISplitter_Sample(LPVOID iface, IMediaSample * pSample)
498 {
499     ICOM_THIS(AVISplitter, iface);
500     LPBYTE pbSrcStream = NULL;
501     long cbSrcStream = 0;
502     REFERENCE_TIME tStart, tStop;
503     HRESULT hr;
504     BOOL bMoreData = TRUE;
505
506     hr = IMediaSample_GetPointer(pSample, &pbSrcStream);
507
508     hr = IMediaSample_GetTime(pSample, &tStart, &tStop);
509
510     cbSrcStream = IMediaSample_GetActualDataLength(pSample);
511
512     /* trace removed for performance reasons */
513 /*  TRACE("(%p)\n", pSample); */
514
515     assert(BYTES_FROM_MEDIATIME(tStop - tStart) == cbSrcStream);
516
517     if (This->CurrentChunkOffset <= tStart && This->CurrentChunkOffset + MEDIATIME_FROM_BYTES(sizeof(RIFFCHUNK)) > tStart)
518     {
519         DWORD offset = (DWORD)BYTES_FROM_MEDIATIME(tStart - This->CurrentChunkOffset);
520         assert(offset <= sizeof(RIFFCHUNK));
521         memcpy((BYTE *)&This->CurrentChunk + offset, pbSrcStream, sizeof(RIFFCHUNK) - offset);
522     }
523     else if (This->CurrentChunkOffset > tStart)
524     {
525         DWORD offset = (DWORD)BYTES_FROM_MEDIATIME(This->CurrentChunkOffset - tStart);
526         if (offset >= (DWORD)cbSrcStream)
527         {
528             FIXME("large offset\n");
529             return S_OK;
530         }
531
532         memcpy(&This->CurrentChunk, pbSrcStream + offset, sizeof(RIFFCHUNK));
533     }
534
535     assert(This->CurrentChunkOffset + MEDIATIME_FROM_BYTES(sizeof(RIFFCHUNK)) < tStop);
536
537     while (bMoreData)
538     {
539         BYTE * pbDstStream;
540         long cbDstStream;
541         long chunk_remaining_bytes = 0;
542         long offset_src;
543         WORD streamId;
544         AVISplitter_OutputPin * pOutputPin;
545         BOOL bSyncPoint = TRUE;
546
547         if (This->CurrentChunkOffset >= tStart)
548             offset_src = (long)BYTES_FROM_MEDIATIME(This->CurrentChunkOffset - tStart) + sizeof(RIFFCHUNK);
549         else
550             offset_src = 0;
551
552         switch (TWOCCFromFOURCC(This->CurrentChunk.fcc))
553         {
554         case cktypeDIBcompressed:
555             bSyncPoint = FALSE;
556             /* fall-through */
557         case cktypeDIBbits:
558             /* FIXME: check that pin is of type video */
559             break;
560         case cktypeWAVEbytes:
561             /* FIXME: check that pin is of type audio */
562             break;
563         case cktypePALchange:
564             FIXME("handle palette change\n");
565             break;
566         case ckidJUNK:
567             /* silently ignore */
568             if (S_FALSE == AVISplitter_NextChunk(&This->CurrentChunkOffset, &This->CurrentChunk, &tStart, &tStop, pbSrcStream))
569                 bMoreData = FALSE;
570             continue;
571         default:
572             FIXME("Skipping unknown chunk type: %s at file offset 0x%lx\n", debugstr_an((LPSTR)&This->CurrentChunk.fcc, 4), (DWORD)BYTES_FROM_MEDIATIME(This->CurrentChunkOffset));
573             if (S_FALSE == AVISplitter_NextChunk(&This->CurrentChunkOffset, &This->CurrentChunk, &tStart, &tStop, pbSrcStream))
574                 bMoreData = FALSE;
575             continue;
576         }
577
578         streamId = StreamFromFOURCC(This->CurrentChunk.fcc);
579
580         if (streamId > This->cStreams)
581         {
582             ERR("Corrupted AVI file (contains stream id %d, but supposed to only have %ld streams)\n", streamId, This->cStreams);
583             return E_FAIL;
584         }
585
586         pOutputPin = (AVISplitter_OutputPin *)This->ppPins[streamId + 1];
587
588         if (!This->pCurrentSample)
589         {
590             /* cache media sample until it is ready to be despatched
591              * (i.e. we reach the end of the chunk) */
592             hr = OutputPin_GetDeliveryBuffer(&pOutputPin->pin, &This->pCurrentSample, NULL, NULL, 0);
593
594             if (SUCCEEDED(hr))
595             {
596                 hr = IMediaSample_SetActualDataLength(This->pCurrentSample, 0);
597                 assert(hr == S_OK);
598             }
599             else
600             {
601                 TRACE("Skipping sending sample for stream %02d due to error (%lx)\n", streamId, hr);
602                 This->pCurrentSample = NULL;
603                 if (S_FALSE == AVISplitter_NextChunk(&This->CurrentChunkOffset, &This->CurrentChunk, &tStart, &tStop, pbSrcStream))
604                     bMoreData = FALSE;
605                 continue;
606             }
607         }
608
609         hr = IMediaSample_GetPointer(This->pCurrentSample, &pbDstStream);
610
611         if (SUCCEEDED(hr))
612         {
613             cbDstStream = IMediaSample_GetSize(This->pCurrentSample);
614
615             chunk_remaining_bytes = (long)BYTES_FROM_MEDIATIME(This->CurrentChunkOffset + MEDIATIME_FROM_BYTES(This->CurrentChunk.cb + sizeof(RIFFCHUNK)) - tStart) - offset_src;
616         
617             assert(chunk_remaining_bytes >= 0);
618             assert(chunk_remaining_bytes <= cbDstStream - IMediaSample_GetActualDataLength(This->pCurrentSample));
619
620             /* trace removed for performance reasons */
621 /*          TRACE("chunk_remaining_bytes: 0x%lx, cbSrcStream: 0x%lx, offset_src: 0x%lx\n", chunk_remaining_bytes, cbSrcStream, offset_src); */
622         }
623
624         if (chunk_remaining_bytes <= cbSrcStream - offset_src)
625         {
626             if (SUCCEEDED(hr))
627             {
628                 memcpy(pbDstStream + IMediaSample_GetActualDataLength(This->pCurrentSample), pbSrcStream + offset_src, chunk_remaining_bytes);
629                 hr = IMediaSample_SetActualDataLength(This->pCurrentSample, chunk_remaining_bytes + IMediaSample_GetActualDataLength(This->pCurrentSample));
630                 assert(hr == S_OK);
631             }
632
633             if (SUCCEEDED(hr))
634             {
635                 REFERENCE_TIME tAviStart, tAviStop;
636
637                 /* FIXME: hack */
638                 if (pOutputPin->dwSamplesProcessed == 0)
639                     IMediaSample_SetDiscontinuity(This->pCurrentSample, TRUE);
640
641                 IMediaSample_SetSyncPoint(This->pCurrentSample, bSyncPoint);
642
643                 pOutputPin->dwSamplesProcessed++;
644
645                 if (pOutputPin->dwSampleSize)
646                     tAviStart = (LONGLONG)ceil(10000000.0 * (float)(pOutputPin->dwSamplesProcessed - 1) * (float)IMediaSample_GetActualDataLength(This->pCurrentSample) / ((float)pOutputPin->dwSampleSize * pOutputPin->fSamplesPerSec));
647                 else
648                     tAviStart = (LONGLONG)ceil(10000000.0 * (float)(pOutputPin->dwSamplesProcessed - 1) / (float)pOutputPin->fSamplesPerSec);
649                 if (pOutputPin->dwSampleSize)
650                     tAviStop = (LONGLONG)ceil(10000000.0 * (float)pOutputPin->dwSamplesProcessed * (float)IMediaSample_GetActualDataLength(This->pCurrentSample) / ((float)pOutputPin->dwSampleSize * pOutputPin->fSamplesPerSec));
651                 else
652                     tAviStop = (LONGLONG)ceil(10000000.0 * (float)pOutputPin->dwSamplesProcessed / (float)pOutputPin->fSamplesPerSec);
653
654                 IMediaSample_SetTime(This->pCurrentSample, &tAviStart, &tAviStop);
655
656
657                 hr = OutputPin_SendSample(&pOutputPin->pin, This->pCurrentSample);
658                 if (hr != S_OK && hr != VFW_E_NOT_CONNECTED)
659                     ERR("Error sending sample (%lx)\n", hr);
660             }
661
662             if (This->pCurrentSample)
663                 IMediaSample_Release(This->pCurrentSample);
664             
665             This->pCurrentSample = NULL;
666
667             if (S_FALSE == AVISplitter_NextChunk(&This->CurrentChunkOffset, &This->CurrentChunk, &tStart, &tStop, pbSrcStream))
668                 bMoreData = FALSE;
669         }
670         else
671         {
672             if (SUCCEEDED(hr))
673             {
674                 memcpy(pbDstStream + IMediaSample_GetActualDataLength(This->pCurrentSample), pbSrcStream + offset_src, cbSrcStream - offset_src);
675                 IMediaSample_SetActualDataLength(This->pCurrentSample, cbSrcStream - offset_src + IMediaSample_GetActualDataLength(This->pCurrentSample));
676             }
677             bMoreData = FALSE;
678         }
679     }
680     return hr;
681 }
682
683 static HRESULT AVISplitter_QueryAccept(LPVOID iface, const AM_MEDIA_TYPE * pmt)
684 {
685     if (IsEqualIID(&pmt->majortype, &MEDIATYPE_Stream) && IsEqualIID(&pmt->subtype, &MEDIASUBTYPE_Avi))
686         return S_OK;
687     return S_FALSE;
688 }
689
690 static HRESULT AVISplitter_ProcessStreamList(AVISplitter * This, const BYTE * pData, DWORD cb)
691 {
692     PIN_INFO piOutput;
693     RIFFCHUNK * pChunk;
694     IPin ** ppOldPins;
695     HRESULT hr;
696     AM_MEDIA_TYPE amt;
697     float fSamplesPerSec = 0.0f;
698     DWORD dwSampleSize = 0;
699     DWORD dwLength = 0;
700     ALLOCATOR_PROPERTIES props;
701     static const WCHAR wszStreamTemplate[] = {'S','t','r','e','a','m',' ','%','0','2','d',0};
702
703     props.cbAlign = 1;
704     props.cbPrefix = 0;
705     props.cbBuffer = 0x20000;
706     props.cBuffers = 2;
707     
708     ZeroMemory(&amt, sizeof(amt));
709     piOutput.dir = PINDIR_OUTPUT;
710     piOutput.pFilter = (IBaseFilter *)This;
711     wsprintfW(piOutput.achName, wszStreamTemplate, This->cStreams);
712
713     for (pChunk = (RIFFCHUNK *)pData; ((BYTE *)pChunk >= pData) && ((BYTE *)pChunk + sizeof(RIFFCHUNK) < pData + cb) && (pChunk->cb > 0); pChunk = (RIFFCHUNK *)((BYTE*)pChunk + sizeof(RIFFCHUNK) + pChunk->cb))
714     {
715         switch (pChunk->fcc)
716         {
717         case ckidSTREAMHEADER:
718             {
719                 const AVISTREAMHEADER * pStrHdr = (const AVISTREAMHEADER *)pChunk;
720                 TRACE("processing stream header\n");
721
722                 fSamplesPerSec = (float)pStrHdr->dwRate / (float)pStrHdr->dwScale;
723
724                 switch (pStrHdr->fccType)
725                 {
726                 case streamtypeVIDEO:
727                     memcpy(&amt.formattype, &FORMAT_VideoInfo, sizeof(GUID));
728                     amt.pbFormat = NULL;
729                     amt.cbFormat = 0;
730                     break;
731                 case streamtypeAUDIO:
732                     memcpy(&amt.formattype, &FORMAT_WaveFormatEx, sizeof(GUID));
733                     break;
734                 default:
735                     memcpy(&amt.formattype, &FORMAT_None, sizeof(GUID));
736                 }
737                 memcpy(&amt.majortype, &MEDIATYPE_Video, sizeof(GUID));
738                 amt.majortype.Data1 = pStrHdr->fccType;
739                 memcpy(&amt.subtype, &MEDIATYPE_Video, sizeof(GUID));
740                 amt.subtype.Data1 = pStrHdr->fccHandler;
741                 TRACE("Subtype FCC: %.04s\n", (LPSTR)&pStrHdr->fccHandler);
742                 amt.lSampleSize = pStrHdr->dwSampleSize;
743                 amt.bFixedSizeSamples = (amt.lSampleSize != 0);
744
745                 /* FIXME: Is this right? */
746                 if (!amt.lSampleSize)
747                 {
748                     amt.lSampleSize = 1;
749                     dwSampleSize = 1;
750                 }
751
752                 amt.bTemporalCompression = IsEqualGUID(&amt.majortype, &MEDIATYPE_Video); /* FIXME? */
753                 dwSampleSize = pStrHdr->dwSampleSize;
754                 dwLength = pStrHdr->dwLength;
755                 if (!dwLength)
756                     dwLength = This->AviHeader.dwTotalFrames;
757
758                 if (pStrHdr->dwSuggestedBufferSize)
759                     props.cbBuffer = pStrHdr->dwSuggestedBufferSize;
760
761                 break;
762             }
763         case ckidSTREAMFORMAT:
764             TRACE("processing stream format data\n");
765             if (IsEqualIID(&amt.formattype, &FORMAT_VideoInfo))
766             {
767                 VIDEOINFOHEADER * pvi;
768                 /* biCompression member appears to override the value in the stream header.
769                  * i.e. the stream header can say something completely contradictory to what
770                  * is in the BITMAPINFOHEADER! */
771                 if (pChunk->cb < sizeof(BITMAPINFOHEADER))
772                 {
773                     ERR("Not enough bytes for BITMAPINFOHEADER\n");
774                     return E_FAIL;
775                 }
776                 amt.cbFormat = sizeof(VIDEOINFOHEADER) - sizeof(BITMAPINFOHEADER) + pChunk->cb;
777                 amt.pbFormat = CoTaskMemAlloc(amt.cbFormat);
778                 ZeroMemory(amt.pbFormat, amt.cbFormat);
779                 pvi = (VIDEOINFOHEADER *)amt.pbFormat;
780                 pvi->AvgTimePerFrame = (LONGLONG)(10000000.0 / fSamplesPerSec);
781                 CopyMemory(&pvi->bmiHeader, (const BYTE *)(pChunk + 1), pChunk->cb);
782                 if (pvi->bmiHeader.biCompression)
783                     amt.subtype.Data1 = pvi->bmiHeader.biCompression;
784             }
785             else
786             {
787                 amt.cbFormat = pChunk->cb;
788                 amt.pbFormat = CoTaskMemAlloc(amt.cbFormat);
789                 CopyMemory(amt.pbFormat, (const BYTE *)(pChunk + 1), amt.cbFormat);
790             }
791             break;
792         case ckidSTREAMNAME:
793             TRACE("processing stream name\n");
794             /* FIXME: this doesn't exactly match native version (we omit the "##)" prefix), but hey... */
795             MultiByteToWideChar(CP_ACP, 0, (LPCSTR)(pChunk + 1), pChunk->cb, piOutput.achName, sizeof(piOutput.achName) / sizeof(piOutput.achName[0]));
796             break;
797         case ckidSTREAMHANDLERDATA:
798             FIXME("process stream handler data\n");
799             break;
800         case ckidJUNK:
801             TRACE("JUNK chunk ignored\n");
802             break;
803         default:
804             FIXME("unknown chunk type \"%.04s\" ignored\n", (LPSTR)&pChunk->fcc);
805         }
806     }
807
808     if (IsEqualGUID(&amt.formattype, &FORMAT_WaveFormatEx))
809     {
810         memcpy(&amt.subtype, &MEDIATYPE_Video, sizeof(GUID));
811         amt.subtype.Data1 = ((WAVEFORMATEX *)amt.pbFormat)->wFormatTag;
812     }
813
814     dump_AM_MEDIA_TYPE(&amt);
815     FIXME("fSamplesPerSec = %f\n", (double)fSamplesPerSec);
816     FIXME("dwSampleSize = %lx\n", dwSampleSize);
817     FIXME("dwLength = %lx\n", dwLength);
818
819     ppOldPins = This->ppPins;
820
821     This->ppPins = HeapAlloc(GetProcessHeap(), 0, (This->cStreams + 2) * sizeof(IPin *));
822     memcpy(This->ppPins, ppOldPins, (This->cStreams + 1) * sizeof(IPin *));
823
824     hr = AVISplitter_OutputPin_Construct(&piOutput, &props, NULL, AVISplitter_OutputPin_QueryAccept, &amt, fSamplesPerSec, &This->csFilter, This->ppPins + This->cStreams + 1);
825
826     if (SUCCEEDED(hr))
827     {
828         ((AVISplitter_OutputPin *)(This->ppPins[This->cStreams + 1]))->dwSampleSize = dwSampleSize;
829         ((AVISplitter_OutputPin *)(This->ppPins[This->cStreams + 1]))->dwLength = dwLength;
830         ((AVISplitter_OutputPin *)(This->ppPins[This->cStreams + 1]))->pin.pin.pUserData = (LPVOID)This->ppPins[This->cStreams + 1];
831         This->cStreams++;
832         HeapFree(GetProcessHeap(), 0, ppOldPins);
833     }
834     else
835     {
836         HeapFree(GetProcessHeap(), 0, This->ppPins);
837         This->ppPins = ppOldPins;
838         ERR("Failed with error %lx\n", hr);
839     }
840
841     return hr;
842 }
843
844 static HRESULT AVISplitter_RemoveOutputPins(AVISplitter * This)
845 {
846     /* NOTE: should be in critical section when calling this function */
847
848     ULONG i;
849     IPin ** ppOldPins = This->ppPins;
850
851     /* reduce the pin array down to 1 (just our input pin) */
852     This->ppPins = HeapAlloc(GetProcessHeap(), 0, sizeof(IPin *) * 1);
853     memcpy(This->ppPins, ppOldPins, sizeof(IPin *) * 1);
854
855     for (i = 0; i < This->cStreams; i++)
856     {
857         OutputPin_DeliverDisconnect((OutputPin *)ppOldPins[i + 1]);
858         IPin_Release(ppOldPins[i + 1]);
859     }
860
861     This->cStreams = 0;
862     HeapFree(GetProcessHeap(), 0, ppOldPins);
863
864     return S_OK;
865 }
866
867 /* FIXME: fix leaks on failure here */
868 static HRESULT AVISplitter_InputPin_PreConnect(IPin * iface, IPin * pConnectPin)
869 {
870     ICOM_THIS(PullPin, iface);
871     HRESULT hr;
872     RIFFLIST list;
873     LONGLONG pos = 0; /* in bytes */
874     BYTE * pBuffer;
875     RIFFCHUNK * pCurrentChunk;
876     AVISplitter * pAviSplit = (AVISplitter *)This->pin.pinInfo.pFilter;
877
878     hr = IAsyncReader_SyncRead(This->pReader, pos, sizeof(list), (BYTE *)&list);
879     pos += sizeof(list);
880
881     if (list.fcc != ckidRIFF)
882     {
883         ERR("Input stream not a RIFF file\n");
884         return E_FAIL;
885     }
886     if (list.cb > 1 * 1024 * 1024 * 1024) /* cannot be more than 1Gb in size */
887     {
888         ERR("Input stream violates RIFF spec\n");
889         return E_FAIL;
890     }
891     if (list.fccListType != ckidAVI)
892     {
893         ERR("Input stream not an AVI RIFF file\n");
894         return E_FAIL;
895     }
896
897     hr = IAsyncReader_SyncRead(This->pReader, pos, sizeof(list), (BYTE *)&list);
898     if (list.fcc != ckidLIST)
899     {
900         ERR("Expected LIST chunk, but got %.04s\n", (LPSTR)&list.fcc);
901         return E_FAIL;
902     }
903     if (list.fccListType != ckidHEADERLIST)
904     {
905         ERR("Header list expected. Got: %.04s\n", (LPSTR)&list.fccListType);
906         return E_FAIL;
907     }
908
909     pBuffer = HeapAlloc(GetProcessHeap(), 0, list.cb - sizeof(RIFFLIST) + sizeof(RIFFCHUNK));
910     hr = IAsyncReader_SyncRead(This->pReader, pos + sizeof(list), list.cb - sizeof(RIFFLIST) + sizeof(RIFFCHUNK), pBuffer);
911
912     pAviSplit->AviHeader.cb = 0;
913
914     for (pCurrentChunk = (RIFFCHUNK *)pBuffer; (BYTE *)pCurrentChunk + sizeof(*pCurrentChunk) < pBuffer + list.cb; pCurrentChunk = (RIFFCHUNK *)(((BYTE *)pCurrentChunk) + sizeof(*pCurrentChunk) + pCurrentChunk->cb))
915     {
916         RIFFLIST * pList;
917
918         switch (pCurrentChunk->fcc)
919         {
920         case ckidMAINAVIHEADER:
921             /* AVIMAINHEADER includes the structure that is pCurrentChunk at the moment */
922             memcpy(&pAviSplit->AviHeader, pCurrentChunk, sizeof(pAviSplit->AviHeader));
923             break;
924         case ckidLIST:
925             pList = (RIFFLIST *)pCurrentChunk;
926             switch (pList->fccListType)
927             {
928             case ckidSTREAMLIST:
929                 hr = AVISplitter_ProcessStreamList(pAviSplit, (BYTE *)pCurrentChunk + sizeof(RIFFLIST), pCurrentChunk->cb + sizeof(RIFFCHUNK) - sizeof(RIFFLIST));
930                 break;
931             case ckidODML:
932                 FIXME("process ODML header\n");
933                 break;
934             }
935             break;
936         case ckidJUNK:
937             /* ignore */
938             break;
939         default:
940             FIXME("unrecognised header list type: %.04s\n", (LPSTR)&pCurrentChunk->fcc);
941         }
942     }
943     HeapFree(GetProcessHeap(), 0, pBuffer);
944
945     if (pAviSplit->AviHeader.cb != sizeof(pAviSplit->AviHeader) - sizeof(RIFFCHUNK))
946     {
947         ERR("Avi Header wrong size!\n");
948         return E_FAIL;
949     }
950
951     pos += sizeof(RIFFCHUNK) + list.cb;
952     hr = IAsyncReader_SyncRead(This->pReader, pos, sizeof(list), (BYTE *)&list);
953
954     if (list.fcc == ckidJUNK)
955     {
956         pos += sizeof(RIFFCHUNK) + list.cb;
957         hr = IAsyncReader_SyncRead(This->pReader, pos, sizeof(list), (BYTE *)&list);
958     }
959
960     if (list.fcc != ckidLIST)
961     {
962         ERR("Expected LIST, but got %.04s\n", (LPSTR)&list.fcc);
963         return E_FAIL;
964     }
965     if (list.fccListType != ckidAVIMOVIE)
966     {
967         ERR("Expected AVI movie list, but got %.04s\n", (LPSTR)&list.fccListType);
968         return E_FAIL;
969     }
970
971     if (hr == S_OK)
972     {
973         pAviSplit->CurrentChunkOffset = MEDIATIME_FROM_BYTES(pos + sizeof(RIFFLIST));
974         pAviSplit->EndOfFile = MEDIATIME_FROM_BYTES(pos + list.cb + sizeof(RIFFLIST));
975         hr = IAsyncReader_SyncRead(This->pReader, BYTES_FROM_MEDIATIME(pAviSplit->CurrentChunkOffset), sizeof(pAviSplit->CurrentChunk), (BYTE *)&pAviSplit->CurrentChunk);
976     }
977
978     if (hr != S_OK)
979         return E_FAIL;
980
981     TRACE("AVI File ok\n");
982
983     return hr;
984 }
985
986 static HRESULT AVISplitter_ChangeStart(LPVOID iface)
987 {
988     FIXME("(%p)\n", iface);
989     return S_OK;
990 }
991
992 static HRESULT AVISplitter_ChangeStop(LPVOID iface)
993 {
994     FIXME("(%p)\n", iface);
995     return S_OK;
996 }
997
998 static HRESULT AVISplitter_ChangeRate(LPVOID iface)
999 {
1000     FIXME("(%p)\n", iface);
1001     return S_OK;
1002 }
1003
1004
1005 static HRESULT WINAPI AVISplitter_Seeking_QueryInterface(IMediaSeeking * iface, REFIID riid, LPVOID * ppv)
1006 {
1007     ICOM_THIS_From_IMediaSeeking(AVISplitter_OutputPin, iface);
1008
1009     return IUnknown_QueryInterface((IUnknown *)This, riid, ppv);
1010 }
1011
1012 static ULONG WINAPI AVISplitter_Seeking_AddRef(IMediaSeeking * iface)
1013 {
1014     ICOM_THIS_From_IMediaSeeking(AVISplitter_OutputPin, iface);
1015
1016     return IUnknown_AddRef((IUnknown *)This);
1017 }
1018
1019 static ULONG WINAPI AVISplitter_Seeking_Release(IMediaSeeking * iface)
1020 {
1021     ICOM_THIS_From_IMediaSeeking(AVISplitter_OutputPin, iface);
1022
1023     return IUnknown_Release((IUnknown *)This);
1024 }
1025
1026 static const IMediaSeekingVtbl AVISplitter_Seeking_Vtbl =
1027 {
1028     ICOM_MSVTABLE_COMPAT_DummyRTTIVALUE
1029     AVISplitter_Seeking_QueryInterface,
1030     AVISplitter_Seeking_AddRef,
1031     AVISplitter_Seeking_Release,
1032     MediaSeekingImpl_GetCapabilities,
1033     MediaSeekingImpl_CheckCapabilities,
1034     MediaSeekingImpl_IsFormatSupported,
1035     MediaSeekingImpl_QueryPreferredFormat,
1036     MediaSeekingImpl_GetTimeFormat,
1037     MediaSeekingImpl_IsUsingTimeFormat,
1038     MediaSeekingImpl_SetTimeFormat,
1039     MediaSeekingImpl_GetDuration,
1040     MediaSeekingImpl_GetStopPosition,
1041     MediaSeekingImpl_GetCurrentPosition,
1042     MediaSeekingImpl_ConvertTimeFormat,
1043     MediaSeekingImpl_SetPositions,
1044     MediaSeekingImpl_GetPositions,
1045     MediaSeekingImpl_GetAvailable,
1046     MediaSeekingImpl_SetRate,
1047     MediaSeekingImpl_GetRate,
1048     MediaSeekingImpl_GetPreroll
1049 };
1050
1051 HRESULT WINAPI AVISplitter_OutputPin_QueryInterface(IPin * iface, REFIID riid, LPVOID * ppv)
1052 {
1053     ICOM_THIS(AVISplitter_OutputPin, iface);
1054
1055     TRACE("(%s, %p)\n", qzdebugstr_guid(riid), ppv);
1056
1057     *ppv = NULL;
1058
1059     if (IsEqualIID(riid, &IID_IUnknown))
1060         *ppv = (LPVOID)iface;
1061     else if (IsEqualIID(riid, &IID_IPin))
1062         *ppv = (LPVOID)iface;
1063     else if (IsEqualIID(riid, &IID_IMediaSeeking))
1064         *ppv = (LPVOID)&This->mediaSeeking;
1065
1066     if (*ppv)
1067     {
1068         IUnknown_AddRef((IUnknown *)(*ppv));
1069         return S_OK;
1070     }
1071
1072     FIXME("No interface for %s!\n", qzdebugstr_guid(riid));
1073
1074     return E_NOINTERFACE;
1075 }
1076
1077 static ULONG WINAPI AVISplitter_OutputPin_Release(IPin * iface)
1078 {
1079     ICOM_THIS(AVISplitter_OutputPin, iface);
1080     
1081     TRACE("()\n");
1082     
1083     if (!InterlockedDecrement(&This->pin.pin.refCount))
1084     {
1085         DeleteMediaType(This->pmt);
1086         CoTaskMemFree(This->pmt);
1087         DeleteMediaType(&This->pin.pin.mtCurrent);
1088         CoTaskMemFree(This);
1089         return 0;
1090     }
1091     return This->pin.pin.refCount;
1092 }
1093
1094 static HRESULT WINAPI AVISplitter_OutputPin_EnumMediaTypes(IPin * iface, IEnumMediaTypes ** ppEnum)
1095 {
1096     ENUMMEDIADETAILS emd;
1097     ICOM_THIS(AVISplitter_OutputPin, iface);
1098
1099     TRACE("(%p)\n", ppEnum);
1100
1101     /* override this method to allow enumeration of your types */
1102     emd.cMediaTypes = 1;
1103     emd.pMediaTypes = This->pmt;
1104
1105     return IEnumMediaTypesImpl_Construct(&emd, ppEnum);
1106 }
1107
1108 static HRESULT AVISplitter_OutputPin_QueryAccept(LPVOID iface, const AM_MEDIA_TYPE * pmt)
1109 {
1110     ICOM_THIS(AVISplitter_OutputPin, iface);
1111
1112     TRACE("()\n");
1113     dump_AM_MEDIA_TYPE(pmt);
1114
1115     return (memcmp(This->pmt, pmt, sizeof(AM_MEDIA_TYPE)) == 0);
1116 }
1117
1118 static const IPinVtbl AVISplitter_OutputPin_Vtbl = 
1119 {
1120     ICOM_MSVTABLE_COMPAT_DummyRTTIVALUE
1121     AVISplitter_OutputPin_QueryInterface,
1122     IPinImpl_AddRef,
1123     AVISplitter_OutputPin_Release,
1124     OutputPin_Connect,
1125     OutputPin_ReceiveConnection,
1126     OutputPin_Disconnect,
1127     IPinImpl_ConnectedTo,
1128     IPinImpl_ConnectionMediaType,
1129     IPinImpl_QueryPinInfo,
1130     IPinImpl_QueryDirection,
1131     IPinImpl_QueryId,
1132     IPinImpl_QueryAccept,
1133     AVISplitter_OutputPin_EnumMediaTypes,
1134     IPinImpl_QueryInternalConnections,
1135     OutputPin_EndOfStream,
1136     OutputPin_BeginFlush,
1137     OutputPin_EndFlush,
1138     OutputPin_NewSegment
1139 };
1140
1141 static HRESULT AVISplitter_InputPin_Construct(const PIN_INFO * pPinInfo, SAMPLEPROC pSampleProc, LPVOID pUserData, QUERYACCEPTPROC pQueryAccept, LPCRITICAL_SECTION pCritSec, IPin ** ppPin)
1142 {
1143     PullPin * pPinImpl;
1144
1145     *ppPin = NULL;
1146
1147     if (pPinInfo->dir != PINDIR_INPUT)
1148     {
1149         ERR("Pin direction(%x) != PINDIR_INPUT\n", pPinInfo->dir);
1150         return E_INVALIDARG;
1151     }
1152
1153     pPinImpl = CoTaskMemAlloc(sizeof(*pPinImpl));
1154
1155     if (!pPinImpl)
1156         return E_OUTOFMEMORY;
1157
1158     if (SUCCEEDED(PullPin_Init(pPinInfo, pSampleProc, pUserData, pQueryAccept, pCritSec, pPinImpl)))
1159     {
1160         pPinImpl->pin.lpVtbl = &AVISplitter_InputPin_Vtbl;
1161         
1162         *ppPin = (IPin *)(&pPinImpl->pin.lpVtbl);
1163         return S_OK;
1164     }
1165     return E_FAIL;
1166 }
1167
1168 static HRESULT WINAPI AVISplitter_InputPin_Disconnect(IPin * iface)
1169 {
1170     HRESULT hr;
1171     ICOM_THIS(IPinImpl, iface);
1172
1173     TRACE("()\n");
1174
1175     EnterCriticalSection(This->pCritSec);
1176     {
1177         if (This->pConnectedTo)
1178         {
1179             FILTER_STATE state;
1180
1181             hr = IBaseFilter_GetState(This->pinInfo.pFilter, 0, &state);
1182
1183             if (SUCCEEDED(hr) && (state == State_Stopped))
1184             {
1185                 IPin_Release(This->pConnectedTo);
1186                 This->pConnectedTo = NULL;
1187                 hr = AVISplitter_RemoveOutputPins((AVISplitter *)This->pinInfo.pFilter);
1188             }
1189             else
1190                 hr = VFW_E_NOT_STOPPED;
1191         }
1192         else
1193             hr = S_FALSE;
1194     }
1195     LeaveCriticalSection(This->pCritSec);
1196     
1197     return hr;
1198 }
1199
1200 static const IPinVtbl AVISplitter_InputPin_Vtbl =
1201 {
1202     ICOM_MSVTABLE_COMPAT_DummyRTTIVALUE
1203     PullPin_QueryInterface,
1204     IPinImpl_AddRef,
1205     PullPin_Release,
1206     OutputPin_Connect,
1207     PullPin_ReceiveConnection,
1208     AVISplitter_InputPin_Disconnect,
1209     IPinImpl_ConnectedTo,
1210     IPinImpl_ConnectionMediaType,
1211     IPinImpl_QueryPinInfo,
1212     IPinImpl_QueryDirection,
1213     IPinImpl_QueryId,
1214     IPinImpl_QueryAccept,
1215     IPinImpl_EnumMediaTypes,
1216     IPinImpl_QueryInternalConnections,
1217     PullPin_EndOfStream,
1218     PullPin_BeginFlush,
1219     PullPin_EndFlush,
1220     PullPin_NewSegment
1221 };