4 * Copyright 2003 Robert Shearman
5 * Copyright 2004-2005 Christian Costa
6 * Copyright 2008 Maarten Lankhorst
8 * This library is free software; you can redistribute it and/or
9 * modify it under the terms of the GNU Lesser General Public
10 * License as published by the Free Software Foundation; either
11 * version 2.1 of the License, or (at your option) any later version.
13 * This library is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
16 * Lesser General Public License for more details.
18 * You should have received a copy of the GNU Lesser General Public
19 * License along with this library; if not, write to the Free Software
20 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
23 * - Reference leaks, if they still exist
24 * - Files without an index are not handled correctly yet.
25 * - When stopping/starting, a sample is lost. This should be compensated by
26 * keeping track of previous index/position.
27 * - Debugging channels are noisy at the moment, especially with thread
28 * related messages, however this is the only correct thing to do right now,
29 * since wine doesn't correctly handle all messages yet.
32 #include "quartz_private.h"
33 #include "control_private.h"
42 #include "wine/unicode.h"
43 #include "wine/debug.h"
50 #define TWOCCFromFOURCC(fcc) HIWORD(fcc)
52 /* four character codes used in AVI files */
53 #define ckidINFO mmioFOURCC('I','N','F','O')
54 #define ckidREC mmioFOURCC('R','E','C',' ')
56 WINE_DEFAULT_DEBUG_CHANNEL(quartz);
58 typedef struct StreamData
64 AVISTREAMHEADER streamheader;
66 AVISTDINDEX **stdindex;
70 /* Position, in index units */
71 DWORD pos, pos_next, index, index_next;
73 /* Packet handling: a thread is created and waits on the packet event handle
74 * On an event acquire the sample lock, addref the sample and set it to NULL,
75 * then queue a new packet.
77 HANDLE thread, packet_queued;
80 /* Amount of preroll samples for this stream */
84 typedef struct AVISplitterImpl
87 RIFFCHUNK CurrentChunk;
88 LONGLONG CurrentChunkOffset; /* in media time */
90 AVIMAINHEADER AviHeader;
91 AVIEXTHEADER ExtHeader;
93 AVIOLDINDEX *oldindex;
100 AVISplitterImpl *This;
104 /* The threading stuff cries for an explanation
106 * PullPin starts processing and calls AVISplitter_first_request
107 * AVISplitter_first_request creates a thread for each stream
108 * A stream can be audio, video, subtitles or something undefined.
110 * AVISplitter_first_request loads a single packet to each but one stream,
111 * and queues it for that last stream. This is to prevent WaitForNext to time
114 * The processing loop is entered. It calls IAsyncReader_WaitForNext in the
115 * PullPin. Every time it receives a packet, it will call AVISplitter_Sample
116 * AVISplitter_Sample will signal the relevant thread that a new sample is
117 * arrived, when that thread is ready it will read the packet and transmits
118 * it downstream with AVISplitter_Receive
120 * Threads terminate upon receiving NULL as packet or when ANY error code
121 * != S_OK occurs. This means that any error is fatal to processing.
124 static HRESULT AVISplitter_SendEndOfFile(AVISplitterImpl *This, DWORD streamnumber)
129 TRACE("End of file reached\n");
131 hr = IPin_ConnectedTo(This->Parser.ppPins[streamnumber+1], &ppin);
134 hr = IPin_EndOfStream(ppin);
137 TRACE("--> %x\n", hr);
139 /* Force the pullpin thread to stop */
143 /* Thread worker horse */
144 static HRESULT AVISplitter_next_request(AVISplitterImpl *This, DWORD streamnumber)
146 StreamData *stream = This->streams + streamnumber;
147 PullPin *pin = This->Parser.pInputPin;
148 IMediaSample *sample = NULL;
151 TRACE("(%p, %u)->()\n", This, streamnumber);
153 hr = IMemAllocator_GetBuffer(pin->pAlloc, &sample, NULL, NULL, 0);
155 ERR("... %08x?\n", hr);
159 LONGLONG rtSampleStart;
160 /* Add 4 for the next header, which should hopefully work */
161 LONGLONG rtSampleStop;
163 stream->pos = stream->pos_next;
164 stream->index = stream->index_next;
166 IMediaSample_SetDiscontinuity(sample, stream->seek);
167 stream->seek = FALSE;
171 IMediaSample_SetPreroll(sample, TRUE);
174 IMediaSample_SetPreroll(sample, FALSE);
175 IMediaSample_SetSyncPoint(sample, TRUE);
177 if (stream->stdindex)
179 AVISTDINDEX *index = stream->stdindex[stream->index];
180 AVISTDINDEX_ENTRY *entry = &index->aIndex[stream->pos];
183 rtSampleStart = index->qwBaseOffset;
184 keyframe = !(entry->dwSize >> 31);
185 rtSampleStart += entry->dwOffset;
186 rtSampleStart = MEDIATIME_FROM_BYTES(rtSampleStart);
189 if (index->nEntriesInUse == stream->pos_next)
191 stream->pos_next = 0;
192 ++stream->index_next;
195 rtSampleStop = rtSampleStart + MEDIATIME_FROM_BYTES(entry->dwSize & ~(1 << 31));
197 TRACE("offset(%u) size(%u)\n", (DWORD)BYTES_FROM_MEDIATIME(rtSampleStart), (DWORD)BYTES_FROM_MEDIATIME(rtSampleStop - rtSampleStart));
200 if (stream->index >= stream->entries)
202 ERR("END OF STREAM ON %u\n", streamnumber);
203 IMediaSample_Release(sample);
204 hr = AVISplitter_SendEndOfFile(This, streamnumber);
208 else if (This->oldindex)
210 DWORD flags = This->oldindex->aIndex[stream->pos].dwFlags;
211 DWORD size = This->oldindex->aIndex[stream->pos].dwSize;
213 keyframe = !!(flags & AVIIF_KEYFRAME);
215 rtSampleStart = MEDIATIME_FROM_BYTES(This->offset);
216 rtSampleStart += MEDIATIME_FROM_BYTES(This->oldindex->aIndex[stream->pos].dwOffset);
217 rtSampleStop = rtSampleStart + MEDIATIME_FROM_BYTES(size);
218 if (flags & AVIIF_MIDPART)
220 FIXME("Only stand alone frames are currently handled correctly!\n");
222 if (flags & AVIIF_LIST)
224 FIXME("Not sure if this is handled correctly\n");
225 rtSampleStart += MEDIATIME_FROM_BYTES(sizeof(RIFFLIST));
226 rtSampleStop += MEDIATIME_FROM_BYTES(sizeof(RIFFLIST));
230 rtSampleStart += MEDIATIME_FROM_BYTES(sizeof(RIFFCHUNK));
231 rtSampleStop += MEDIATIME_FROM_BYTES(sizeof(RIFFCHUNK));
234 /* Slow way of finding next index */
237 } while (stream->pos_next * sizeof(This->oldindex->aIndex[0]) < This->oldindex->cb
238 && StreamFromFOURCC(This->oldindex->aIndex[stream->pos_next].dwChunkId) != streamnumber);
243 IMediaSample_Release(sample);
244 stream->pos_next = 0;
245 ++stream->index_next;
246 ERR("END OF STREAM ON %u\n", streamnumber);
247 hr = AVISplitter_SendEndOfFile(This, streamnumber);
251 else /* TODO: Generate an index automagically */
253 ERR("CAN'T PLAY WITHOUT AN INDEX! SOS! SOS! SOS!\n");
257 if (rtSampleStart != rtSampleStop)
259 hr = IMediaSample_SetTime(sample, &rtSampleStart, &rtSampleStop);
261 hr = IAsyncReader_Request(pin->pReader, sample, streamnumber);
264 assert(IMediaSample_Release(sample) == 0);
268 stream->sample = sample;
269 IMediaSample_SetActualDataLength(sample, 0);
270 SetEvent(stream->packet_queued);
277 ERR("There should be no sample!\n");
278 assert(IMediaSample_Release(sample) == 0);
281 TRACE("--> %08x\n", hr);
286 static HRESULT AVISplitter_Receive(AVISplitterImpl *This, IMediaSample *sample, DWORD streamnumber)
288 Parser_OutputPin *pin = (Parser_OutputPin *)This->Parser.ppPins[1+streamnumber];
290 LONGLONG start, stop;
291 StreamData *stream = &This->streams[streamnumber];
293 start = pin->dwSamplesProcessed;
294 start *= stream->streamheader.dwScale;
296 start /= stream->streamheader.dwRate;
298 if (stream->streamheader.dwSampleSize)
300 ULONG len = IMediaSample_GetActualDataLength(sample);
301 ULONG size = stream->streamheader.dwSampleSize;
303 pin->dwSamplesProcessed += len / size;
306 ++pin->dwSamplesProcessed;
308 stop = pin->dwSamplesProcessed;
309 stop *= stream->streamheader.dwScale;
311 stop /= stream->streamheader.dwRate;
313 IMediaSample_SetTime(sample, &start, &stop);
315 hr = OutputPin_SendSample(&pin->pin, sample);
317 /* Uncomment this if you want to debug the time differences between the
318 * different streams, it is useful for that
320 FIXME("stream %u, hr: %08x, Start: %u.%03u, Stop: %u.%03u\n", streamnumber, hr,
321 (DWORD)(start / 10000000), (DWORD)((start / 10000)%1000),
322 (DWORD)(stop / 10000000), (DWORD)((stop / 10000)%1000));
327 static DWORD WINAPI AVISplitter_thread_reader(LPVOID data)
329 struct thread_args *args = data;
330 AVISplitterImpl *This = args->This;
331 DWORD streamnumber = args->stream;
336 HRESULT nexthr = S_FALSE;
337 IMediaSample *sample;
339 WaitForSingleObject(This->streams[streamnumber].packet_queued, INFINITE);
340 sample = This->streams[streamnumber].sample;
341 This->streams[streamnumber].sample = NULL;
345 nexthr = AVISplitter_next_request(This, streamnumber);
347 hr = AVISplitter_Receive(This, sample, streamnumber);
349 FIXME("Receiving error: %08x\n", hr);
351 IMediaSample_Release(sample);
354 } while (hr == S_OK);
356 FIXME("Thread %u terminated with hr %08x!\n", streamnumber, hr);
361 static HRESULT AVISplitter_Sample(LPVOID iface, IMediaSample * pSample, DWORD_PTR cookie)
363 AVISplitterImpl *This = iface;
364 StreamData *stream = This->streams + cookie;
367 if (!IMediaSample_GetActualDataLength(pSample))
369 ERR("Received empty sample\n");
373 /* Send the sample to whatever thread is appropiate
374 * That thread should also not have a sample queued at the moment
377 TRACE("(%p)->(%p size: %u, %lu)\n", This, pSample, IMediaSample_GetActualDataLength(pSample), cookie);
378 assert(cookie < This->Parser.cStreams);
379 assert(!stream->sample);
380 assert(WaitForSingleObject(stream->packet_queued, 0) == WAIT_TIMEOUT);
382 IMediaSample_AddRef(pSample);
384 stream->sample = pSample;
385 SetEvent(stream->packet_queued);
390 static HRESULT AVISplitter_done_process(LPVOID iface);
392 /* On the first request we have to be sure that (cStreams-1) samples have
393 * already been processed, because otherwise some pins might not ever finish
394 * a Pause state change
396 static HRESULT AVISplitter_first_request(LPVOID iface)
398 AVISplitterImpl *This = (AVISplitterImpl *)iface;
401 IMediaSample *sample = NULL;
402 BOOL have_sample = FALSE;
404 TRACE("(%p)->()\n", This);
406 for (x = 0; x < This->Parser.cStreams; ++x)
408 StreamData *stream = This->streams + x;
410 /* Nothing should be running at this point */
411 assert(!stream->thread);
414 /* It could be we asked the thread to terminate, and the thread
415 * already terminated before receiving the deathwish */
416 ResetEvent(stream->packet_queued);
418 stream->pos_next = stream->pos;
419 stream->index_next = stream->index;
421 /* There should be a packet queued from AVISplitter_next_request last time
422 * It needs to be done now because this is the only way to ensure that every
423 * stream will have at least 1 packet processed
424 * If this is done after the threads start it could go all awkward and we
425 * would have no guarantees that it's successful at all
430 DWORD_PTR dwUser = ~0;
431 hr = IAsyncReader_WaitForNext(This->Parser.pInputPin->pReader, 10000, &sample, &dwUser);
435 AVISplitter_Sample(iface, sample, dwUser);
436 IMediaSample_Release(sample);
439 hr = AVISplitter_next_request(This, x);
440 TRACE("-->%08x\n", hr);
442 /* Could be an EOF instead */
443 have_sample = (hr == S_OK);
448 /* FIXME: Don't do this for each pin that sent an EOF */
449 for (x = 0; x < This->Parser.cStreams && SUCCEEDED(hr); ++x)
451 struct thread_args *args;
454 if ((This->streams[x].stdindex && This->streams[x].index_next >= This->streams[x].entries) ||
455 (!This->streams[x].stdindex && This->streams[x].index_next))
457 This->streams[x].thread = NULL;
461 args = CoTaskMemAlloc(sizeof(*args));
464 This->streams[x].thread = CreateThread(NULL, 0, AVISplitter_thread_reader, args, 0, &tid);
465 FIXME("Created stream %u thread 0x%08x\n", x, tid);
469 ERR("Horsemen of the apocalypse came to bring error 0x%08x\n", hr);
474 static HRESULT AVISplitter_done_process(LPVOID iface)
476 AVISplitterImpl *This = iface;
480 for (x = 0; x < This->Parser.cStreams; ++x)
482 StreamData *stream = This->streams + x;
484 FIXME("Waiting for %u to terminate\n", x);
485 /* Make the thread return first */
486 SetEvent(stream->packet_queued);
487 assert(WaitForSingleObject(stream->thread, 100000) != WAIT_TIMEOUT);
488 CloseHandle(stream->thread);
489 stream->thread = NULL;
492 assert(IMediaSample_Release(stream->sample) == 0);
493 stream->sample = NULL;
495 ResetEvent(stream->packet_queued);
501 static HRESULT AVISplitter_QueryAccept(LPVOID iface, const AM_MEDIA_TYPE * pmt)
503 if (IsEqualIID(&pmt->majortype, &MEDIATYPE_Stream) && IsEqualIID(&pmt->subtype, &MEDIASUBTYPE_Avi))
508 static HRESULT AVISplitter_ProcessIndex(AVISplitterImpl *This, AVISTDINDEX **index, LONGLONG qwOffset, DWORD cb)
515 if (cb < sizeof(AVISTDINDEX))
517 FIXME("size %u too small\n", cb);
521 pIndex = CoTaskMemAlloc(cb);
523 return E_OUTOFMEMORY;
525 IAsyncReader_SyncRead(((PullPin *)This->Parser.ppPins[0])->pReader, qwOffset, cb, (BYTE *)pIndex);
526 pIndex = CoTaskMemRealloc(pIndex, pIndex->cb);
528 return E_OUTOFMEMORY;
530 IAsyncReader_SyncRead(((PullPin *)This->Parser.ppPins[0])->pReader, qwOffset, pIndex->cb, (BYTE *)pIndex);
531 rest = pIndex->cb - sizeof(AVISUPERINDEX) + sizeof(RIFFCHUNK) + sizeof(pIndex->aIndex[0]) * ANYSIZE_ARRAY;
533 TRACE("FOURCC: %s\n", debugstr_an((char *)&pIndex->fcc, 4));
534 TRACE("wLongsPerEntry: %hd\n", pIndex->wLongsPerEntry);
535 TRACE("bIndexSubType: %hd\n", pIndex->bIndexSubType);
536 TRACE("bIndexType: %hd\n", pIndex->bIndexType);
537 TRACE("nEntriesInUse: %u\n", pIndex->nEntriesInUse);
538 TRACE("dwChunkId: %.4s\n", (char *)&pIndex->dwChunkId);
539 TRACE("qwBaseOffset: %x%08x\n", (DWORD)(pIndex->qwBaseOffset >> 32), (DWORD)pIndex->qwBaseOffset);
540 TRACE("dwReserved_3: %u\n", pIndex->dwReserved_3);
542 if (pIndex->bIndexType != AVI_INDEX_OF_CHUNKS
543 || pIndex->wLongsPerEntry != 2
544 || rest < (pIndex->nEntriesInUse * sizeof(DWORD) * pIndex->wLongsPerEntry)
545 || (pIndex->bIndexSubType != AVI_INDEX_SUB_DEFAULT))
547 FIXME("Invalid index chunk encountered\n");
551 for (x = 0; x < pIndex->nEntriesInUse; ++x)
553 BOOL keyframe = !(pIndex->aIndex[x].dwSize >> 31);
554 DWORDLONG offset = pIndex->qwBaseOffset + pIndex->aIndex[x].dwOffset;
555 TRACE("dwOffset: %x%08x\n", (DWORD)(offset >> 32), (DWORD)offset);
556 TRACE("dwSize: %u\n", (pIndex->aIndex[x].dwSize & ~(1<<31)));
557 TRACE("Frame is a keyframe: %s\n", keyframe ? "yes" : "no");
564 static HRESULT AVISplitter_ProcessOldIndex(AVISplitterImpl *This)
566 ULONGLONG mov_pos = BYTES_FROM_MEDIATIME(This->CurrentChunkOffset) - sizeof(DWORD);
567 AVIOLDINDEX *pAviOldIndex = This->oldindex;
571 for (x = 0; x < pAviOldIndex->cb / sizeof(pAviOldIndex->aIndex[0]); ++x)
573 DWORD temp, temp2 = 0, offset, chunkid;
574 PullPin *pin = This->Parser.pInputPin;
576 offset = pAviOldIndex->aIndex[x].dwOffset;
577 chunkid = pAviOldIndex->aIndex[x].dwChunkId;
579 TRACE("dwChunkId: %.4s\n", (char *)&chunkid);
580 TRACE("dwFlags: %08x\n", pAviOldIndex->aIndex[x].dwFlags);
581 TRACE("dwOffset (%s): %08x\n", relative ? "relative" : "absolute", offset);
582 TRACE("dwSize: %08x\n", pAviOldIndex->aIndex[x].dwSize);
584 /* Only scan once, or else this will take too long */
587 IAsyncReader_SyncRead(pin->pReader, offset, sizeof(DWORD), (BYTE *)&temp);
588 relative = (chunkid != temp);
590 if (chunkid == mmioFOURCC('7','F','x','x')
591 && ((char *)&temp)[0] == 'i' && ((char *)&temp)[1] == 'x')
596 if (offset + mov_pos < BYTES_FROM_MEDIATIME(This->EndOfFile))
597 IAsyncReader_SyncRead(pin->pReader, offset + mov_pos, sizeof(DWORD), (BYTE *)&temp2);
599 if (chunkid == mmioFOURCC('7','F','x','x')
600 && ((char *)&temp2)[0] == 'i' && ((char *)&temp2)[1] == 'x')
602 /* Do nothing, all is great */
604 else if (temp2 != chunkid)
606 ERR("Faulty index or bug in handling: Wanted FCC: %s, Abs FCC: %s (@ %x), Rel FCC: %s (@ %.0x%08x)\n",
607 debugstr_an((char *)&chunkid, 4), debugstr_an((char *)&temp, 4), offset,
608 debugstr_an((char *)&temp2, 4), (DWORD)((mov_pos + offset) >> 32), (DWORD)(mov_pos + offset));
612 TRACE("Scanned dwChunkId: %s\n", debugstr_an((char *)&temp2, 4));
615 TRACE("Scanned dwChunkId: %s\n", debugstr_an((char *)&temp, 4));
617 /* Only dump one packet */
623 FIXME("Dropping index: no idea whether it is relative or absolute\n");
624 CoTaskMemFree(This->oldindex);
625 This->oldindex = NULL;
630 This->offset = (DWORD)mov_pos;
635 static HRESULT AVISplitter_ProcessStreamList(AVISplitterImpl * This, const BYTE * pData, DWORD cb, ALLOCATOR_PROPERTIES *props)
638 const RIFFCHUNK * pChunk;
641 float fSamplesPerSec = 0.0f;
642 DWORD dwSampleSize = 0;
645 static const WCHAR wszStreamTemplate[] = {'S','t','r','e','a','m',' ','%','0','2','d',0};
648 ZeroMemory(&amt, sizeof(amt));
649 piOutput.dir = PINDIR_OUTPUT;
650 piOutput.pFilter = (IBaseFilter *)This;
651 wsprintfW(piOutput.achName, wszStreamTemplate, This->Parser.cStreams);
652 This->streams = CoTaskMemRealloc(This->streams, sizeof(StreamData) * (This->Parser.cStreams+1));
653 stream = This->streams + This->Parser.cStreams;
654 ZeroMemory(stream, sizeof(*stream));
656 for (pChunk = (const RIFFCHUNK *)pData;
657 ((const BYTE *)pChunk >= pData) && ((const BYTE *)pChunk + sizeof(RIFFCHUNK) < pData + cb) && (pChunk->cb > 0);
658 pChunk = (const RIFFCHUNK *)((const BYTE*)pChunk + sizeof(RIFFCHUNK) + pChunk->cb)
663 case ckidSTREAMHEADER:
665 const AVISTREAMHEADER * pStrHdr = (const AVISTREAMHEADER *)pChunk;
666 TRACE("processing stream header\n");
667 stream->streamheader = *pStrHdr;
669 fSamplesPerSec = (float)pStrHdr->dwRate / (float)pStrHdr->dwScale;
670 CoTaskMemFree(amt.pbFormat);
674 switch (pStrHdr->fccType)
676 case streamtypeVIDEO:
677 amt.formattype = FORMAT_VideoInfo;
679 case streamtypeAUDIO:
680 amt.formattype = FORMAT_WaveFormatEx;
683 FIXME("fccType %.4s not handled yet\n", (char *)&pStrHdr->fccType);
684 amt.formattype = FORMAT_None;
686 amt.majortype = MEDIATYPE_Video;
687 amt.majortype.Data1 = pStrHdr->fccType;
688 amt.subtype = MEDIATYPE_Video;
689 amt.subtype.Data1 = pStrHdr->fccHandler;
690 TRACE("Subtype FCC: %.04s\n", (LPCSTR)&pStrHdr->fccHandler);
691 amt.lSampleSize = pStrHdr->dwSampleSize;
692 amt.bFixedSizeSamples = (amt.lSampleSize != 0);
694 /* FIXME: Is this right? */
695 if (!amt.lSampleSize)
701 amt.bTemporalCompression = IsEqualGUID(&amt.majortype, &MEDIATYPE_Video); /* FIXME? */
702 dwSampleSize = pStrHdr->dwSampleSize;
703 dwLength = pStrHdr->dwLength;
705 dwLength = This->AviHeader.dwTotalFrames;
707 if (pStrHdr->dwSuggestedBufferSize && pStrHdr->dwSuggestedBufferSize > props->cbBuffer)
708 props->cbBuffer = pStrHdr->dwSuggestedBufferSize;
712 case ckidSTREAMFORMAT:
713 TRACE("processing stream format data\n");
714 if (IsEqualIID(&amt.formattype, &FORMAT_VideoInfo))
716 VIDEOINFOHEADER * pvi;
717 /* biCompression member appears to override the value in the stream header.
718 * i.e. the stream header can say something completely contradictory to what
719 * is in the BITMAPINFOHEADER! */
720 if (pChunk->cb < sizeof(BITMAPINFOHEADER))
722 ERR("Not enough bytes for BITMAPINFOHEADER\n");
725 amt.cbFormat = sizeof(VIDEOINFOHEADER) - sizeof(BITMAPINFOHEADER) + pChunk->cb;
726 amt.pbFormat = CoTaskMemAlloc(amt.cbFormat);
727 ZeroMemory(amt.pbFormat, amt.cbFormat);
728 pvi = (VIDEOINFOHEADER *)amt.pbFormat;
729 pvi->AvgTimePerFrame = (LONGLONG)(10000000.0 / fSamplesPerSec);
731 CopyMemory(&pvi->bmiHeader, (const BYTE *)(pChunk + 1), pChunk->cb);
732 if (pvi->bmiHeader.biCompression)
733 amt.subtype.Data1 = pvi->bmiHeader.biCompression;
735 else if (IsEqualIID(&amt.formattype, &FORMAT_WaveFormatEx))
737 amt.cbFormat = pChunk->cb;
738 if (amt.cbFormat < sizeof(WAVEFORMATEX))
739 amt.cbFormat = sizeof(WAVEFORMATEX);
740 amt.pbFormat = CoTaskMemAlloc(amt.cbFormat);
741 ZeroMemory(amt.pbFormat, amt.cbFormat);
742 CopyMemory(amt.pbFormat, (const BYTE *)(pChunk + 1), pChunk->cb);
746 amt.cbFormat = pChunk->cb;
747 amt.pbFormat = CoTaskMemAlloc(amt.cbFormat);
748 CopyMemory(amt.pbFormat, (const BYTE *)(pChunk + 1), amt.cbFormat);
752 TRACE("processing stream name\n");
753 /* FIXME: this doesn't exactly match native version (we omit the "##)" prefix), but hey... */
754 MultiByteToWideChar(CP_ACP, 0, (LPCSTR)(pChunk + 1), pChunk->cb, piOutput.achName, sizeof(piOutput.achName) / sizeof(piOutput.achName[0]));
756 case ckidSTREAMHANDLERDATA:
757 FIXME("process stream handler data\n");
760 TRACE("JUNK chunk ignored\n");
762 case ckidAVISUPERINDEX:
764 const AVISUPERINDEX *pIndex = (const AVISUPERINDEX *)pChunk;
766 long rest = pIndex->cb - sizeof(AVISUPERINDEX) + sizeof(RIFFCHUNK) + sizeof(pIndex->aIndex[0]) * ANYSIZE_ARRAY;
768 if (pIndex->cb < sizeof(AVISUPERINDEX) - sizeof(RIFFCHUNK))
770 FIXME("size %u\n", pIndex->cb);
776 ERR("Stream %d got more than 1 superindex?\n", This->Parser.cStreams);
780 TRACE("wLongsPerEntry: %hd\n", pIndex->wLongsPerEntry);
781 TRACE("bIndexSubType: %hd\n", pIndex->bIndexSubType);
782 TRACE("bIndexType: %hd\n", pIndex->bIndexType);
783 TRACE("nEntriesInUse: %u\n", pIndex->nEntriesInUse);
784 TRACE("dwChunkId: %.4s\n", (char *)&pIndex->dwChunkId);
785 if (pIndex->dwReserved[0])
786 TRACE("dwReserved[0]: %u\n", pIndex->dwReserved[0]);
787 if (pIndex->dwReserved[2])
788 TRACE("dwReserved[1]: %u\n", pIndex->dwReserved[1]);
789 if (pIndex->dwReserved[2])
790 TRACE("dwReserved[2]: %u\n", pIndex->dwReserved[2]);
792 if (pIndex->bIndexType != AVI_INDEX_OF_INDEXES
793 || pIndex->wLongsPerEntry != 4
794 || rest < (pIndex->nEntriesInUse * sizeof(DWORD) * pIndex->wLongsPerEntry)
795 || (pIndex->bIndexSubType != AVI_INDEX_SUB_2FIELD && pIndex->bIndexSubType != AVI_INDEX_SUB_DEFAULT))
797 FIXME("Invalid index chunk encountered\n");
801 stream->entries = pIndex->nEntriesInUse;
802 stream->stdindex = CoTaskMemRealloc(stream->stdindex, sizeof(*stream->stdindex) * stream->entries);
803 for (x = 0; x < pIndex->nEntriesInUse; ++x)
805 TRACE("qwOffset: %x%08x\n", (DWORD)(pIndex->aIndex[x].qwOffset >> 32), (DWORD)pIndex->aIndex[x].qwOffset);
806 TRACE("dwSize: %u\n", pIndex->aIndex[x].dwSize);
807 TRACE("dwDuration: %u (unreliable)\n", pIndex->aIndex[x].dwDuration);
809 AVISplitter_ProcessIndex(This, &stream->stdindex[nstdindex-1], pIndex->aIndex[x].qwOffset, pIndex->aIndex[x].dwSize);
814 FIXME("unknown chunk type \"%.04s\" ignored\n", (LPCSTR)&pChunk->fcc);
818 if (IsEqualGUID(&amt.formattype, &FORMAT_WaveFormatEx))
820 amt.subtype = MEDIATYPE_Video;
821 amt.subtype.Data1 = ((WAVEFORMATEX *)amt.pbFormat)->wFormatTag;
824 dump_AM_MEDIA_TYPE(&amt);
825 TRACE("fSamplesPerSec = %f\n", (double)fSamplesPerSec);
826 TRACE("dwSampleSize = %x\n", dwSampleSize);
827 TRACE("dwLength = %x\n", dwLength);
829 stream->fSamplesPerSec = fSamplesPerSec;
830 stream->dwSampleSize = dwSampleSize;
831 stream->dwLength = dwLength; /* TODO: Use this for mediaseeking */
832 stream->packet_queued = CreateEventW(NULL, 0, 0, NULL);
834 hr = Parser_AddPin(&(This->Parser), &piOutput, props, &amt);
835 CoTaskMemFree(amt.pbFormat);
841 static HRESULT AVISplitter_ProcessODML(AVISplitterImpl * This, const BYTE * pData, DWORD cb)
843 const RIFFCHUNK * pChunk;
845 for (pChunk = (const RIFFCHUNK *)pData;
846 ((const BYTE *)pChunk >= pData) && ((const BYTE *)pChunk + sizeof(RIFFCHUNK) < pData + cb) && (pChunk->cb > 0);
847 pChunk = (const RIFFCHUNK *)((const BYTE*)pChunk + sizeof(RIFFCHUNK) + pChunk->cb)
852 case ckidAVIEXTHEADER:
855 const AVIEXTHEADER * pExtHdr = (const AVIEXTHEADER *)pChunk;
857 TRACE("processing extension header\n");
858 if (pExtHdr->cb != sizeof(AVIEXTHEADER) - sizeof(RIFFCHUNK))
860 FIXME("Size: %u\n", pExtHdr->cb);
863 TRACE("dwGrandFrames: %u\n", pExtHdr->dwGrandFrames);
864 for (x = 0; x < 61; ++x)
865 if (pExtHdr->dwFuture[x])
866 FIXME("dwFuture[%i] = %u (0x%08x)\n", x, pExtHdr->dwFuture[x], pExtHdr->dwFuture[x]);
867 This->ExtHeader = *pExtHdr;
871 FIXME("unknown chunk type \"%.04s\" ignored\n", (LPCSTR)&pChunk->fcc);
878 static HRESULT AVISplitter_InitializeStreams(AVISplitterImpl *This)
886 for (x = 0; x < This->Parser.cStreams; ++x)
888 This->streams[x].frames = 0;
889 This->streams[x].pos = ~0;
890 This->streams[x].index = 0;
893 nMax = This->oldindex->cb / sizeof(This->oldindex->aIndex[0]);
895 /* Ok, maybe this is more of an excercise to see if I interpret everything correctly or not, but that is useful for now. */
896 for (n = 0; n < nMax; ++n)
898 DWORD streamId = StreamFromFOURCC(This->oldindex->aIndex[n].dwChunkId);
899 if (streamId >= This->Parser.cStreams)
901 FIXME("Stream id %s ignored\n", debugstr_an((char*)&This->oldindex->aIndex[n].dwChunkId, 4));
904 if (This->streams[streamId].pos == ~0)
905 This->streams[streamId].pos = n;
907 if (This->streams[streamId].streamheader.dwSampleSize)
908 This->streams[streamId].frames += This->oldindex->aIndex[n].dwSize / This->streams[streamId].streamheader.dwSampleSize;
910 ++This->streams[streamId].frames;
913 for (x = 0; x < This->Parser.cStreams; ++x)
915 if ((DWORD)This->streams[x].frames != This->streams[x].streamheader.dwLength)
917 FIXME("stream %u: frames found: %u, frames meant to be found: %u\n", x, (DWORD)This->streams[x].frames, This->streams[x].streamheader.dwLength);
922 else if (!This->streams[0].entries)
924 for (x = 0; x < This->Parser.cStreams; ++x)
926 This->streams[x].frames = This->streams[x].streamheader.dwLength;
928 /* MS Avi splitter does seek through the whole file, we should! */
929 ERR("We should be manually seeking through the entire file to build an index, because the index is missing!!!\n");
933 /* Not much here yet */
934 for (x = 0; x < This->Parser.cStreams; ++x)
936 StreamData *stream = This->streams + x;
942 if (stream->stdindex)
946 for (y = 0; y < stream->entries; ++y)
948 if (stream->streamheader.dwSampleSize)
952 for (z = 0; z < stream->stdindex[y]->nEntriesInUse; ++z)
954 UINT len = stream->stdindex[y]->aIndex[z].dwSize & ~(1 << 31);
955 frames += len / stream->streamheader.dwSampleSize + !!(len % stream->streamheader.dwSampleSize);
959 frames += stream->stdindex[y]->nEntriesInUse;
962 else frames = stream->frames;
964 frames *= stream->streamheader.dwScale;
965 /* Keep accuracy as high as possible for duration */
966 This->Parser.mediaSeeking.llDuration = frames * 10000000;
967 This->Parser.mediaSeeking.llDuration /= stream->streamheader.dwRate;
968 This->Parser.mediaSeeking.llStop = This->Parser.mediaSeeking.llDuration;
969 This->Parser.mediaSeeking.llCurrent = 0;
971 frames /= stream->streamheader.dwRate;
973 TRACE("Duration: %d days, %d hours, %d minutes and %d seconds\n", (DWORD)(frames / 86400),
974 (DWORD)((frames % 86400) / 3600), (DWORD)((frames % 3600) / 60), (DWORD)(frames % 60));
980 static HRESULT AVISplitter_Disconnect(LPVOID iface);
982 /* FIXME: fix leaks on failure here */
983 static HRESULT AVISplitter_InputPin_PreConnect(IPin * iface, IPin * pConnectPin, ALLOCATOR_PROPERTIES *props)
985 PullPin *This = (PullPin *)iface;
988 LONGLONG pos = 0; /* in bytes */
990 RIFFCHUNK * pCurrentChunk;
991 LONGLONG total, avail;
995 AVISplitterImpl * pAviSplit = (AVISplitterImpl *)This->pin.pinInfo.pFilter;
997 hr = IAsyncReader_SyncRead(This->pReader, pos, sizeof(list), (BYTE *)&list);
1000 if (list.fcc != FOURCC_RIFF)
1002 ERR("Input stream not a RIFF file\n");
1005 if (list.fccListType != formtypeAVI)
1007 ERR("Input stream not an AVI RIFF file\n");
1011 hr = IAsyncReader_SyncRead(This->pReader, pos, sizeof(list), (BYTE *)&list);
1012 if (list.fcc != FOURCC_LIST)
1014 ERR("Expected LIST chunk, but got %.04s\n", (LPSTR)&list.fcc);
1017 if (list.fccListType != listtypeAVIHEADER)
1019 ERR("Header list expected. Got: %.04s\n", (LPSTR)&list.fccListType);
1023 pBuffer = HeapAlloc(GetProcessHeap(), 0, list.cb - sizeof(RIFFLIST) + sizeof(RIFFCHUNK));
1024 hr = IAsyncReader_SyncRead(This->pReader, pos + sizeof(list), list.cb - sizeof(RIFFLIST) + sizeof(RIFFCHUNK), pBuffer);
1026 pAviSplit->AviHeader.cb = 0;
1028 /* Stream list will set the buffer size here, so set a default and allow an override */
1029 props->cbBuffer = 0x20000;
1031 for (pCurrentChunk = (RIFFCHUNK *)pBuffer; (BYTE *)pCurrentChunk + sizeof(*pCurrentChunk) < pBuffer + list.cb; pCurrentChunk = (RIFFCHUNK *)(((BYTE *)pCurrentChunk) + sizeof(*pCurrentChunk) + pCurrentChunk->cb))
1035 switch (pCurrentChunk->fcc)
1037 case ckidMAINAVIHEADER:
1038 /* AVIMAINHEADER includes the structure that is pCurrentChunk at the moment */
1039 memcpy(&pAviSplit->AviHeader, pCurrentChunk, sizeof(pAviSplit->AviHeader));
1042 pList = (RIFFLIST *)pCurrentChunk;
1043 switch (pList->fccListType)
1045 case ckidSTREAMLIST:
1046 hr = AVISplitter_ProcessStreamList(pAviSplit, (BYTE *)pCurrentChunk + sizeof(RIFFLIST), pCurrentChunk->cb + sizeof(RIFFCHUNK) - sizeof(RIFFLIST), props);
1049 hr = AVISplitter_ProcessODML(pAviSplit, (BYTE *)pCurrentChunk + sizeof(RIFFLIST), pCurrentChunk->cb + sizeof(RIFFCHUNK) - sizeof(RIFFLIST));
1053 case ckidAVIPADDING:
1057 FIXME("unrecognised header list type: %.04s\n", (LPSTR)&pCurrentChunk->fcc);
1060 HeapFree(GetProcessHeap(), 0, pBuffer);
1062 if (pAviSplit->AviHeader.cb != sizeof(pAviSplit->AviHeader) - sizeof(RIFFCHUNK))
1064 ERR("Avi Header wrong size!\n");
1068 pos += sizeof(RIFFCHUNK) + list.cb;
1069 hr = IAsyncReader_SyncRead(This->pReader, pos, sizeof(list), (BYTE *)&list);
1071 while (list.fcc == ckidAVIPADDING || (list.fcc == FOURCC_LIST && list.fccListType == ckidINFO))
1073 pos += sizeof(RIFFCHUNK) + list.cb;
1075 hr = IAsyncReader_SyncRead(This->pReader, pos, sizeof(list), (BYTE *)&list);
1078 if (list.fcc != FOURCC_LIST)
1080 ERR("Expected LIST, but got %.04s\n", (LPSTR)&list.fcc);
1083 if (list.fccListType != listtypeAVIMOVIE)
1085 ERR("Expected AVI movie list, but got %.04s\n", (LPSTR)&list.fccListType);
1089 IAsyncReader_Length(This->pReader, &total, &avail);
1091 /* FIXME: AVIX files are extended beyond the FOURCC chunk "AVI ", and thus won't be played here,
1092 * once I get one of the files I'll try to fix it */
1095 This->rtStart = pAviSplit->CurrentChunkOffset = MEDIATIME_FROM_BYTES(pos + sizeof(RIFFLIST));
1096 pos += list.cb + sizeof(RIFFCHUNK);
1098 pAviSplit->EndOfFile = This->rtStop = MEDIATIME_FROM_BYTES(pos);
1101 ERR("File smaller (%x%08x) then EndOfFile (%x%08x)\n", (DWORD)(total >> 32), (DWORD)total, (DWORD)(pAviSplit->EndOfFile >> 32), (DWORD)pAviSplit->EndOfFile);
1105 hr = IAsyncReader_SyncRead(This->pReader, BYTES_FROM_MEDIATIME(pAviSplit->CurrentChunkOffset), sizeof(pAviSplit->CurrentChunk), (BYTE *)&pAviSplit->CurrentChunk);
1109 props->cbPrefix = 0;
1110 /* Comrades, prevent shortage of buffers, or you will feel the consequences! DA! */
1111 props->cBuffers = 2 * pAviSplit->Parser.cStreams;
1113 /* Now peek into the idx1 index, if available */
1114 if (hr == S_OK && (total - pos) > sizeof(RIFFCHUNK))
1116 memset(&list, 0, sizeof(list));
1118 hr = IAsyncReader_SyncRead(This->pReader, pos, sizeof(list), (BYTE *)&list);
1119 if (list.fcc == ckidAVIOLDINDEX)
1121 pAviSplit->oldindex = CoTaskMemRealloc(pAviSplit->oldindex, list.cb + sizeof(RIFFCHUNK));
1122 if (pAviSplit->oldindex)
1124 hr = IAsyncReader_SyncRead(This->pReader, pos, sizeof(RIFFCHUNK) + list.cb, (BYTE *)pAviSplit->oldindex);
1127 hr = AVISplitter_ProcessOldIndex(pAviSplit);
1131 CoTaskMemFree(pAviSplit->oldindex);
1132 pAviSplit->oldindex = NULL;
1140 for (x = 0; x < pAviSplit->Parser.cStreams; ++x)
1141 if (pAviSplit->streams[x].entries)
1146 CoTaskMemFree(pAviSplit->oldindex);
1147 pAviSplit->oldindex = NULL;
1148 if (indexes < pAviSplit->Parser.cStreams)
1150 /* This error could possible be survived by switching to old type index,
1151 * but I would rather find out why it doesn't find everything here
1153 ERR("%d indexes expected, but only have %d\n", indexes, pAviSplit->Parser.cStreams);
1157 else if (!indexes && pAviSplit->oldindex)
1158 indexes = pAviSplit->Parser.cStreams;
1160 if (!indexes && pAviSplit->AviHeader.dwFlags & AVIF_MUSTUSEINDEX)
1162 FIXME("No usable index was found!\n");
1166 /* Now, set up the streams */
1168 hr = AVISplitter_InitializeStreams(pAviSplit);
1172 AVISplitter_Disconnect(pAviSplit);
1176 TRACE("AVI File ok\n");
1181 static HRESULT AVISplitter_Flush(LPVOID iface)
1183 AVISplitterImpl *This = (AVISplitterImpl*)iface;
1186 ERR("(%p)->()\n", This);
1188 for (x = 0; x < This->Parser.cStreams; ++x)
1190 StreamData *stream = This->streams + x;
1193 assert(IMediaSample_Release(stream->sample) == 0);
1194 stream->sample = NULL;
1196 ResetEvent(stream->packet_queued);
1197 assert(!stream->thread);
1203 static HRESULT AVISplitter_Disconnect(LPVOID iface)
1205 AVISplitterImpl *This = iface;
1208 /* TODO: Remove other memory that's allocated during connect */
1209 CoTaskMemFree(This->oldindex);
1210 This->oldindex = NULL;
1212 for (x = 0; x < This->Parser.cStreams; ++x)
1216 StreamData *stream = &This->streams[x];
1218 for (i = 0; i < stream->entries; ++i)
1219 CoTaskMemFree(stream->stdindex[i]);
1221 CoTaskMemFree(stream->stdindex);
1222 CloseHandle(stream->packet_queued);
1224 CoTaskMemFree(This->streams);
1225 This->streams = NULL;
1229 static ULONG WINAPI AVISplitter_Release(IBaseFilter *iface)
1231 AVISplitterImpl *This = (AVISplitterImpl *)iface;
1234 ref = InterlockedDecrement(&This->Parser.refCount);
1236 TRACE("(%p)->() Release from %d\n", This, ref + 1);
1240 AVISplitter_Flush(This);
1241 Parser_Destroy(&This->Parser);
1247 static HRESULT AVISplitter_seek(IBaseFilter *iface)
1249 AVISplitterImpl *This = (AVISplitterImpl *)iface;
1250 PullPin *pPin = This->Parser.pInputPin;
1251 LONGLONG newpos, endpos;
1254 newpos = This->Parser.mediaSeeking.llCurrent;
1255 endpos = This->Parser.mediaSeeking.llDuration;
1257 if (newpos > endpos)
1259 WARN("Requesting position %x%08x beyond end of stream %x%08x\n", (DWORD)(newpos>>32), (DWORD)newpos, (DWORD)(endpos>>32), (DWORD)endpos);
1260 return E_INVALIDARG;
1263 FIXME("Moving position to %u.%03u s!\n", (DWORD)(newpos / 10000000), (DWORD)((newpos / 10000)%1000));
1265 EnterCriticalSection(&pPin->thread_lock);
1266 /* Send a flush to all output pins */
1267 IPin_BeginFlush((IPin *)pPin);
1269 /* Make sure this is done while stopped, BeginFlush takes care of this */
1270 EnterCriticalSection(&This->Parser.csFilter);
1271 for (x = 0; x < This->Parser.cStreams; ++x)
1273 Parser_OutputPin *pin = (Parser_OutputPin *)This->Parser.ppPins[1+x];
1274 StreamData *stream = This->streams + x;
1275 IPin *victim = NULL;
1276 LONGLONG wanted_frames;
1277 DWORD last_keyframe = 0, last_keyframeidx = 0, preroll = 0;
1279 wanted_frames = newpos;
1280 wanted_frames *= stream->streamheader.dwRate;
1281 wanted_frames /= 10000000;
1282 wanted_frames /= stream->streamheader.dwScale;
1284 IPin_ConnectedTo((IPin *)pin, &victim);
1287 IPin_NewSegment(victim, newpos, endpos, pPin->dRate);
1288 IPin_Release(victim);
1291 pin->dwSamplesProcessed = 0;
1295 if (stream->stdindex)
1299 for (y = 0; y < stream->entries; ++y)
1301 for (z = 0; z < stream->stdindex[y]->nEntriesInUse; ++z)
1303 if (stream->streamheader.dwSampleSize)
1305 ULONG len = stream->stdindex[y]->aIndex[z].dwSize & ~(1 << 31);
1306 ULONG size = stream->streamheader.dwSampleSize;
1308 pin->dwSamplesProcessed += len / size;
1310 ++pin->dwSamplesProcessed;
1312 else ++pin->dwSamplesProcessed;
1314 if (!(stream->stdindex[y]->aIndex[z].dwSize >> 31))
1317 last_keyframeidx = y;
1323 if (pin->dwSamplesProcessed >= wanted_frames)
1326 if (pin->dwSamplesProcessed >= wanted_frames)
1329 stream->index = last_keyframeidx;
1330 stream->pos = last_keyframe;
1335 nMax = This->oldindex->cb / sizeof(This->oldindex->aIndex[0]);
1337 for (n = 0; n < nMax; ++n)
1339 DWORD streamId = StreamFromFOURCC(This->oldindex->aIndex[n].dwChunkId);
1343 if (stream->streamheader.dwSampleSize)
1345 ULONG len = This->oldindex->aIndex[n].dwSize;
1346 ULONG size = stream->streamheader.dwSampleSize;
1348 pin->dwSamplesProcessed += len / size;
1350 ++pin->dwSamplesProcessed;
1352 else ++pin->dwSamplesProcessed;
1354 if (This->oldindex->aIndex[n].dwFlags & AVIIF_KEYFRAME)
1362 if (pin->dwSamplesProcessed >= wanted_frames)
1366 stream->pos = last_keyframe;
1369 stream->preroll = preroll;
1372 LeaveCriticalSection(&This->Parser.csFilter);
1374 TRACE("Done flushing\n");
1375 IPin_EndFlush((IPin *)pPin);
1376 LeaveCriticalSection(&pPin->thread_lock);
1381 static const IBaseFilterVtbl AVISplitterImpl_Vtbl =
1383 Parser_QueryInterface,
1385 AVISplitter_Release,
1391 Parser_SetSyncSource,
1392 Parser_GetSyncSource,
1395 Parser_QueryFilterInfo,
1396 Parser_JoinFilterGraph,
1397 Parser_QueryVendorInfo
1400 HRESULT AVISplitter_create(IUnknown * pUnkOuter, LPVOID * ppv)
1403 AVISplitterImpl * This;
1405 TRACE("(%p, %p)\n", pUnkOuter, ppv);
1410 return CLASS_E_NOAGGREGATION;
1412 /* Note: This memory is managed by the transform filter once created */
1413 This = CoTaskMemAlloc(sizeof(AVISplitterImpl));
1415 This->streams = NULL;
1416 This->oldindex = NULL;
1418 hr = Parser_Create(&(This->Parser), &AVISplitterImpl_Vtbl, &CLSID_AviSplitter, AVISplitter_Sample, AVISplitter_QueryAccept, AVISplitter_InputPin_PreConnect, AVISplitter_Flush, AVISplitter_Disconnect, AVISplitter_first_request, AVISplitter_done_process, NULL, AVISplitter_seek, NULL);
1423 *ppv = (LPVOID)This;