4 * Copyright 2003 Robert Shearman
5 * Copyright 2004-2005 Christian Costa
6 * Copyright 2007 Chris Robinson
7 * Copyright 2008 Maarten Lankhorst
9 * This library is free software; you can redistribute it and/or
10 * modify it under the terms of the GNU Lesser General Public
11 * License as published by the Free Software Foundation; either
12 * version 2.1 of the License, or (at your option) any later version.
14 * This library is distributed in the hope that it will be useful,
15 * but WITHOUT ANY WARRANTY; without even the implied warranty of
16 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
17 * Lesser General Public License for more details.
19 * You should have received a copy of the GNU Lesser General Public
20 * License along with this library; if not, write to the Free Software
21 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
27 #include "quartz_private.h"
36 #include "wine/unicode.h"
37 #include "wine/debug.h"
41 WINE_DEFAULT_DEBUG_CHANNEL(quartz
);
43 #define SEQUENCE_HEADER_CODE 0xB3
44 #define PACK_START_CODE 0xBA
46 #define SYSTEM_START_CODE 0xBB
47 #define AUDIO_ELEMENTARY_STREAM 0xC0
48 #define VIDEO_ELEMENTARY_STREAM 0xE0
50 #define MPEG_SYSTEM_HEADER 3
51 #define MPEG_VIDEO_HEADER 2
52 #define MPEG_AUDIO_HEADER 1
53 #define MPEG_NO_HEADER 0
55 typedef struct MPEGSplitterImpl
58 IAMStreamSelect IAMStreamSelect_iface
;
64 /* Whether we just seeked (or started playing) */
68 static inline MPEGSplitterImpl
*impl_from_IBaseFilter( IBaseFilter
*iface
)
70 return CONTAINING_RECORD(iface
, MPEGSplitterImpl
, Parser
.filter
.IBaseFilter_iface
);
73 static inline MPEGSplitterImpl
*impl_from_IMediaSeeking( IMediaSeeking
*iface
)
75 return CONTAINING_RECORD(iface
, MPEGSplitterImpl
, Parser
.sourceSeeking
.IMediaSeeking_iface
);
78 static inline MPEGSplitterImpl
*impl_from_IAMStreamSelect( IAMStreamSelect
*iface
)
80 return CONTAINING_RECORD(iface
, MPEGSplitterImpl
, IAMStreamSelect_iface
);
83 static int MPEGSplitter_head_check(const BYTE
*header
)
85 /* If this is a possible start code, check for a system or video header */
86 if (header
[0] == 0 && header
[1] == 0 && header
[2] == 1)
88 /* Check if we got a system or elementary stream start code */
89 if (header
[3] == PACK_START_CODE
||
90 header
[3] == VIDEO_ELEMENTARY_STREAM
||
91 header
[3] == AUDIO_ELEMENTARY_STREAM
)
92 return MPEG_SYSTEM_HEADER
;
94 /* Check for a MPEG video sequence start code */
95 if (header
[3] == SEQUENCE_HEADER_CODE
)
96 return MPEG_VIDEO_HEADER
;
99 /* This should give a good guess if we have an MPEG audio header */
100 if(header
[0] == 0xff && ((header
[1]>>5)&0x7) == 0x7 &&
101 ((header
[1]>>1)&0x3) != 0 && ((header
[2]>>4)&0xf) != 0xf &&
102 ((header
[2]>>2)&0x3) != 0x3)
103 return MPEG_AUDIO_HEADER
;
106 return MPEG_NO_HEADER
;
109 static const WCHAR wszAudioStream
[] = {'A','u','d','i','o',0};
111 static const DWORD freqs
[10] = { 44100, 48000, 32000, 22050, 24000, 16000, 11025, 12000, 8000, 0 };
113 static const DWORD tabsel_123
[2][3][16] = {
114 { {0,32,64,96,128,160,192,224,256,288,320,352,384,416,448,},
115 {0,32,48,56, 64, 80, 96,112,128,160,192,224,256,320,384,},
116 {0,32,40,48, 56, 64, 80, 96,112,128,160,192,224,256,320,} },
118 { {0,32,48,56,64,80,96,112,128,144,160,176,192,224,256,},
119 {0,8,16,24,32,40,48,56,64,80,96,112,128,144,160,},
120 {0,8,16,24,32,40,48,56,64,80,96,112,128,144,160,} }
123 static HRESULT
parse_header(BYTE
*header
, LONGLONG
*plen
, LONGLONG
*pduration
)
125 int bitrate_index
, freq_index
, lsf
= 1, mpeg1
, layer
, padding
, bitrate
, length
;
128 if (MPEGSplitter_head_check(header
) != MPEG_AUDIO_HEADER
)
130 FIXME("Not a valid header: %02x:%02x:%02x:%02x\n", header
[0], header
[1], header
[2], header
[3]);
134 mpeg1
= (header
[1]>>4)&0x1;
136 lsf
= ((header
[1]>>3)&0x1)^1;
138 layer
= 4-((header
[1]>>1)&0x3);
139 bitrate_index
= ((header
[2]>>4)&0xf);
140 freq_index
= ((header
[2]>>2)&0x3) + (mpeg1
?(lsf
*3):6);
141 padding
= ((header
[2]>>1)&0x1);
143 bitrate
= tabsel_123
[lsf
][layer
-1][bitrate_index
] * 1000;
146 FIXME("Not a valid header: %02x:%02x:%02x:%02x\n", header
[0], header
[1], header
[2], header
[3]);
151 length
= 4 * (12 * bitrate
/ freqs
[freq_index
] + padding
);
153 length
= 144 * bitrate
/ freqs
[freq_index
] + padding
;
155 length
= 144 * bitrate
/ (freqs
[freq_index
]<<lsf
) + padding
;
158 ERR("Impossible layer %d\n", layer
);
162 duration
= (ULONGLONG
)10000000 * (ULONGLONG
)(length
) / (ULONGLONG
)(bitrate
/8);
165 *pduration
+= duration
;
169 static HRESULT
FillBuffer(MPEGSplitterImpl
*This
, IMediaSample
*pCurrentSample
)
171 Parser_OutputPin
* pOutputPin
= unsafe_impl_Parser_OutputPin_from_IPin(This
->Parser
.ppPins
[1]);
173 LONGLONG pos
= BYTES_FROM_MEDIATIME(This
->Parser
.pInputPin
->rtNext
);
174 LONGLONG time
= This
->position
, rtstop
, rtstart
;
177 DWORD len
= IMediaSample_GetActualDataLength(pCurrentSample
);
179 TRACE("Source length: %u\n", len
);
180 IMediaSample_GetPointer(pCurrentSample
, &fbuf
);
182 /* Find the next valid header.. it <SHOULD> be right here */
183 hr
= parse_header(fbuf
, &length
, &This
->position
);
185 IMediaSample_SetActualDataLength(pCurrentSample
, length
);
187 /* Queue the next sample */
188 if (length
+ 4 == len
)
190 PullPin
*pin
= This
->Parser
.pInputPin
;
191 LONGLONG stop
= BYTES_FROM_MEDIATIME(pin
->rtStop
);
194 memcpy(This
->header
, fbuf
+ length
, 4);
195 while (FAILED(hr
= parse_header(This
->header
, &length
, NULL
)))
197 memmove(This
->header
, This
->header
+1, 3);
200 IAsyncReader_SyncRead(pin
->pReader
, ++pos
, 1, This
->header
+ 3);
202 pin
->rtNext
= MEDIATIME_FROM_BYTES(pos
);
206 /* Remove 4 for the last header, which should hopefully work */
207 IMediaSample
*sample
= NULL
;
208 LONGLONG rtSampleStart
= pin
->rtNext
- MEDIATIME_FROM_BYTES(4);
209 LONGLONG rtSampleStop
= rtSampleStart
+ MEDIATIME_FROM_BYTES(length
+ 4);
211 if (rtSampleStop
> pin
->rtStop
)
212 rtSampleStop
= MEDIATIME_FROM_BYTES(ALIGNUP(BYTES_FROM_MEDIATIME(pin
->rtStop
), pin
->cbAlign
));
214 hr
= IMemAllocator_GetBuffer(pin
->pAlloc
, &sample
, NULL
, NULL
, 0);
217 IMediaSample_SetTime(sample
, &rtSampleStart
, &rtSampleStop
);
218 IMediaSample_SetPreroll(sample
, FALSE
);
219 IMediaSample_SetDiscontinuity(sample
, FALSE
);
220 IMediaSample_SetSyncPoint(sample
, TRUE
);
221 hr
= IAsyncReader_Request(pin
->pReader
, sample
, 0);
224 pin
->rtCurrent
= rtSampleStart
;
225 pin
->rtNext
= rtSampleStop
;
228 IMediaSample_Release(sample
);
231 FIXME("o_Ox%08x\n", hr
);
234 /* If not, we're presumably at the end of file */
236 TRACE("Media time : %u.%03u\n", (DWORD
)(This
->position
/10000000), (DWORD
)((This
->position
/10000)%1000));
238 if (IMediaSample_IsDiscontinuity(pCurrentSample
) == S_OK
) {
240 EnterCriticalSection(&This
->Parser
.filter
.csFilter
);
241 pOutputPin
->pin
.pin
.tStart
= time
;
242 pOutputPin
->pin
.pin
.dRate
= This
->Parser
.sourceSeeking
.dRate
;
243 hr
= IPin_ConnectedTo(&pOutputPin
->pin
.pin
.IPin_iface
, &victim
);
246 hr
= IPin_NewSegment(victim
, time
, This
->Parser
.sourceSeeking
.llStop
,
247 This
->Parser
.sourceSeeking
.dRate
);
249 FIXME("NewSegment returns %08x\n", hr
);
250 IPin_Release(victim
);
252 LeaveCriticalSection(&This
->Parser
.filter
.csFilter
);
256 rtstart
= (double)(time
- pOutputPin
->pin
.pin
.tStart
) / pOutputPin
->pin
.pin
.dRate
;
257 rtstop
= (double)(This
->position
- pOutputPin
->pin
.pin
.tStart
) / pOutputPin
->pin
.pin
.dRate
;
258 IMediaSample_SetTime(pCurrentSample
, &rtstart
, &rtstop
);
259 IMediaSample_SetMediaTime(pCurrentSample
, &time
, &This
->position
);
261 hr
= BaseOutputPinImpl_Deliver(&pOutputPin
->pin
, pCurrentSample
);
266 TRACE("Error sending sample (%x)\n", hr
);
268 TRACE("S_FALSE (%d), holding\n", IMediaSample_GetActualDataLength(pCurrentSample
));
275 static HRESULT
MPEGSplitter_process_sample(LPVOID iface
, IMediaSample
* pSample
, DWORD_PTR cookie
)
277 MPEGSplitterImpl
*This
= iface
;
279 DWORD cbSrcStream
= 0;
280 REFERENCE_TIME tStart
, tStop
, tAviStart
= This
->position
;
283 hr
= IMediaSample_GetTime(pSample
, &tStart
, &tStop
);
286 cbSrcStream
= IMediaSample_GetActualDataLength(pSample
);
287 hr
= IMediaSample_GetPointer(pSample
, &pbSrcStream
);
290 /* Flush occurring */
291 if (cbSrcStream
== 0)
293 FIXME(".. Why do I need you?\n");
297 /* trace removed for performance reasons */
298 /* TRACE("(%p), %llu -> %llu\n", pSample, tStart, tStop); */
300 /* Now, try to find a new header */
301 hr
= FillBuffer(This
, pSample
);
304 WARN("Failed with hres: %08x!\n", hr
);
306 /* Unset progression if denied! */
307 if (hr
== VFW_E_WRONG_STATE
|| hr
== S_FALSE
)
309 memcpy(This
->header
, pbSrcStream
, 4);
310 This
->Parser
.pInputPin
->rtCurrent
= tStart
;
311 This
->position
= tAviStart
;
315 if (BYTES_FROM_MEDIATIME(tStop
) >= This
->EndOfFile
|| This
->position
>= This
->Parser
.sourceSeeking
.llStop
)
319 TRACE("End of file reached\n");
321 for (i
= 0; i
< This
->Parser
.cStreams
; i
++)
325 hr
= IPin_ConnectedTo(This
->Parser
.ppPins
[i
+1], &ppin
);
328 hr
= IPin_EndOfStream(ppin
);
332 WARN("Error sending EndOfStream to pin %u (%x)\n", i
, hr
);
335 /* Force the pullpin thread to stop */
343 static HRESULT
MPEGSplitter_query_accept(LPVOID iface
, const AM_MEDIA_TYPE
*pmt
)
345 if (!IsEqualIID(&pmt
->majortype
, &MEDIATYPE_Stream
))
348 if (IsEqualIID(&pmt
->subtype
, &MEDIASUBTYPE_MPEG1Audio
))
351 if (IsEqualIID(&pmt
->subtype
, &MEDIASUBTYPE_MPEG1Video
))
352 FIXME("MPEG-1 video streams not yet supported.\n");
353 else if (IsEqualIID(&pmt
->subtype
, &MEDIASUBTYPE_MPEG1System
))
354 FIXME("MPEG-1 system streams not yet supported.\n");
355 else if (IsEqualIID(&pmt
->subtype
, &MEDIASUBTYPE_MPEG1VideoCD
))
356 FIXME("MPEG-1 VideoCD streams not yet supported.\n");
357 else FIXME("%s\n", debugstr_guid(&pmt
->subtype
));
363 static HRESULT
MPEGSplitter_init_audio(MPEGSplitterImpl
*This
, const BYTE
*header
, PIN_INFO
*ppiOutput
, AM_MEDIA_TYPE
*pamt
)
365 WAVEFORMATEX
*format
;
376 ZeroMemory(pamt
, sizeof(*pamt
));
377 ppiOutput
->dir
= PINDIR_OUTPUT
;
378 ppiOutput
->pFilter
= &This
->Parser
.filter
.IBaseFilter_iface
;
379 wsprintfW(ppiOutput
->achName
, wszAudioStream
);
381 pamt
->formattype
= FORMAT_WaveFormatEx
;
382 pamt
->majortype
= MEDIATYPE_Audio
;
383 pamt
->subtype
= MEDIASUBTYPE_MPEG1AudioPayload
;
385 pamt
->lSampleSize
= 0;
386 pamt
->bFixedSizeSamples
= FALSE
;
387 pamt
->bTemporalCompression
= 0;
389 mpeg1
= (header
[1]>>4)&0x1;
391 lsf
= ((header
[1]>>3)&0x1)^1;
393 layer
= 4-((header
[1]>>1)&0x3);
394 bitrate_index
= ((header
[2]>>4)&0xf);
395 padding
= ((header
[2]>>1)&0x1);
396 freq_index
= ((header
[2]>>2)&0x3) + (mpeg1
?(lsf
*3):6);
397 mode
= ((header
[3]>>6)&0x3);
398 mode_ext
= ((header
[3]>>4)&0x3);
399 emphasis
= ((header
[3]>>0)&0x3);
403 /* Set to highest bitrate so samples will fit in for sure */
404 FIXME("Variable-bitrate audio not fully supported.\n");
408 pamt
->cbFormat
= ((layer
==3)? sizeof(MPEGLAYER3WAVEFORMAT
) :
409 sizeof(MPEG1WAVEFORMAT
));
410 pamt
->pbFormat
= CoTaskMemAlloc(pamt
->cbFormat
);
412 return E_OUTOFMEMORY
;
413 ZeroMemory(pamt
->pbFormat
, pamt
->cbFormat
);
414 format
= (WAVEFORMATEX
*)pamt
->pbFormat
;
416 format
->wFormatTag
= ((layer
== 3) ? WAVE_FORMAT_MPEGLAYER3
:
418 format
->nChannels
= ((mode
== 3) ? 1 : 2);
419 format
->nSamplesPerSec
= freqs
[freq_index
];
420 format
->nAvgBytesPerSec
= tabsel_123
[lsf
][layer
-1][bitrate_index
] * 1000 / 8;
423 format
->nBlockAlign
= format
->nAvgBytesPerSec
* 8 * 144 /
424 (format
->nSamplesPerSec
<<lsf
) + padding
;
426 format
->nBlockAlign
= format
->nAvgBytesPerSec
* 8 * 144 /
427 format
->nSamplesPerSec
+ padding
;
429 format
->nBlockAlign
= 4 * (format
->nAvgBytesPerSec
* 8 * 12 / format
->nSamplesPerSec
+ padding
);
431 format
->wBitsPerSample
= 0;
435 MPEGLAYER3WAVEFORMAT
*mp3format
= (MPEGLAYER3WAVEFORMAT
*)format
;
437 format
->cbSize
= MPEGLAYER3_WFX_EXTRA_BYTES
;
439 mp3format
->wID
= MPEGLAYER3_ID_MPEG
;
440 mp3format
->fdwFlags
= MPEGLAYER3_FLAG_PADDING_ON
;
441 mp3format
->nBlockSize
= format
->nBlockAlign
;
442 mp3format
->nFramesPerBlock
= 1;
444 /* Beware the evil magic numbers. This struct is apparently horribly
445 * under-documented, and the only references I could find had it being
446 * set to this with no real explanation. It works fine though, so I'm
447 * not complaining (yet).
449 mp3format
->nCodecDelay
= 1393;
453 MPEG1WAVEFORMAT
*mpgformat
= (MPEG1WAVEFORMAT
*)format
;
457 mpgformat
->fwHeadLayer
= ((layer
== 1) ? ACM_MPEG_LAYER1
:
458 ((layer
== 2) ? ACM_MPEG_LAYER2
:
460 mpgformat
->dwHeadBitrate
= format
->nAvgBytesPerSec
* 8;
461 mpgformat
->fwHeadMode
= ((mode
== 3) ? ACM_MPEG_SINGLECHANNEL
:
462 ((mode
== 2) ? ACM_MPEG_DUALCHANNEL
:
463 ((mode
== 1) ? ACM_MPEG_JOINTSTEREO
:
465 mpgformat
->fwHeadModeExt
= ((mode
== 1) ? 0x0F : (1<<mode_ext
));
466 mpgformat
->wHeadEmphasis
= emphasis
+ 1;
467 mpgformat
->fwHeadFlags
= ACM_MPEG_ID_MPEG1
;
469 pamt
->subtype
.Data1
= format
->wFormatTag
;
471 TRACE("MPEG audio stream detected:\n"
474 "\tChannels: %d (%d)\n"
475 "\tBytesPerSec: %d\n",
476 layer
, format
->wFormatTag
, format
->nSamplesPerSec
,
477 format
->nChannels
, mode
, format
->nAvgBytesPerSec
);
479 dump_AM_MEDIA_TYPE(pamt
);
485 static HRESULT
MPEGSplitter_pre_connect(IPin
*iface
, IPin
*pConnectPin
, ALLOCATOR_PROPERTIES
*props
)
487 PullPin
*pPin
= impl_PullPin_from_IPin(iface
);
488 MPEGSplitterImpl
*This
= (MPEGSplitterImpl
*)pPin
->pin
.pinInfo
.pFilter
;
490 LONGLONG pos
= 0; /* in bytes */
493 LONGLONG total
, avail
;
497 IAsyncReader_Length(pPin
->pReader
, &total
, &avail
);
498 This
->EndOfFile
= total
;
500 hr
= IAsyncReader_SyncRead(pPin
->pReader
, pos
, 4, header
);
504 /* Skip ID3 v2 tag, if any */
505 if (SUCCEEDED(hr
) && !memcmp("ID3", header
, 3))
508 hr
= IAsyncReader_SyncRead(pPin
->pReader
, pos
, 6, header
+ 4);
512 TRACE("Found ID3 v2.%d.%d\n", header
[3], header
[4]);
513 if(header
[3] <= 4 && header
[4] != 0xff &&
514 (header
[5]&0x0f) == 0 && (header
[6]&0x80) == 0 &&
515 (header
[7]&0x80) == 0 && (header
[8]&0x80) == 0 &&
516 (header
[9]&0x80) == 0)
518 length
= (header
[6]<<21) | (header
[7]<<14) |
519 (header
[8]<< 7) | (header
[9] );
522 TRACE("Length: %u\n", length
);
526 /* Read the real header for the mpeg splitter */
527 hr
= IAsyncReader_SyncRead(pPin
->pReader
, pos
, 4, header
);
534 TRACE("Testing header %x:%x:%x:%x\n", header
[0], header
[1], header
[2], header
[3]);
536 streamtype
= MPEGSplitter_head_check(header
);
537 if (streamtype
== MPEG_AUDIO_HEADER
)
540 if (parse_header(header
, &length
, NULL
) == S_OK
)
543 /* Ensure we have a valid header by seeking for the next frame, some bad
544 * encoded ID3v2 may have an incorrect length and we end up finding bytes
545 * like FF FE 00 28 which are nothing more than a Unicode BOM followed by
546 * ')' character from inside a ID3v2 tag. Unfortunately that sequence
547 * matches like a valid mpeg audio header.
549 hr
= IAsyncReader_SyncRead(pPin
->pReader
, pos
+ length
- 4, 4, next_header
);
552 if (parse_header(next_header
, &length
, NULL
) == S_OK
)
554 TRACE("%x:%x:%x:%x is a fake audio header, looking for next...\n",
555 header
[0], header
[1], header
[2], header
[3]);
558 else if (streamtype
) /* Video or System stream */
561 /* No valid header yet; shift by a byte and check again */
562 memmove(header
, header
+1, 3);
563 hr
= IAsyncReader_SyncRead(pPin
->pReader
, pos
++, 1, header
+ 3);
568 This
->begin_offset
= pos
;
569 memcpy(This
->header
, header
, 4);
573 case MPEG_AUDIO_HEADER
:
575 LONGLONG duration
= 0;
576 WAVEFORMATEX
*format
;
578 hr
= MPEGSplitter_init_audio(This
, header
, &piOutput
, &amt
);
581 format
= (WAVEFORMATEX
*)amt
.pbFormat
;
585 /* Make the output buffer a multiple of the frame size */
586 props
->cbBuffer
= 0x4000 / format
->nBlockAlign
*
589 hr
= Parser_AddPin(&(This
->Parser
), &piOutput
, props
, &amt
);
594 CoTaskMemFree(amt
.pbFormat
);
595 ERR("Could not create pin for MPEG audio stream (%x)\n", hr
);
599 /* Check for idv1 tag, and remove it from stream if found */
600 hr
= IAsyncReader_SyncRead(pPin
->pReader
, This
->EndOfFile
-128, 3, header
);
603 if (!strncmp((char*)header
, "TAG", 3))
604 This
->EndOfFile
-= 128;
605 This
->Parser
.pInputPin
->rtStop
= MEDIATIME_FROM_BYTES(This
->EndOfFile
);
606 This
->Parser
.pInputPin
->rtStart
= This
->Parser
.pInputPin
->rtCurrent
= MEDIATIME_FROM_BYTES(This
->begin_offset
);
608 duration
= (This
->EndOfFile
-This
->begin_offset
) * 10000000 / format
->nAvgBytesPerSec
;
609 TRACE("Duration: %d seconds\n", (DWORD
)(duration
/ 10000000));
611 This
->Parser
.sourceSeeking
.llCurrent
= 0;
612 This
->Parser
.sourceSeeking
.llDuration
= duration
;
613 This
->Parser
.sourceSeeking
.llStop
= duration
;
616 case MPEG_VIDEO_HEADER
:
617 FIXME("MPEG video processing not yet supported!\n");
620 case MPEG_SYSTEM_HEADER
:
621 FIXME("MPEG system streams not yet supported!\n");
633 static HRESULT
MPEGSplitter_cleanup(LPVOID iface
)
635 MPEGSplitterImpl
*This
= iface
;
637 TRACE("(%p)\n", This
);
642 static HRESULT WINAPI
MPEGSplitter_seek(IMediaSeeking
*iface
)
644 MPEGSplitterImpl
*This
= impl_from_IMediaSeeking(iface
);
645 PullPin
*pPin
= This
->Parser
.pInputPin
;
646 LONGLONG newpos
, timepos
, bytepos
;
647 HRESULT hr
= E_INVALIDARG
;
650 newpos
= This
->Parser
.sourceSeeking
.llCurrent
;
651 if (This
->position
/1000000 == newpos
/1000000)
653 TRACE("Requesting position %x%08x same as current position %x%08x\n", (DWORD
)(newpos
>>32), (DWORD
)newpos
, (DWORD
)(This
->position
>>32), (DWORD
)This
->position
);
657 bytepos
= This
->begin_offset
;
659 /* http://mpgedit.org/mpgedit/mpeg_format/mpeghdr.htm has a whole read up on audio headers */
660 while (bytepos
+ 3 < This
->EndOfFile
)
662 LONGLONG duration
= timepos
;
664 hr
= IAsyncReader_SyncRead(pPin
->pReader
, bytepos
, 4, header
);
667 while ((hr
=parse_header(header
, &length
, &duration
)) != S_OK
&&
668 bytepos
+ 4 < This
->EndOfFile
)
670 /* No valid header yet; shift by a byte and check again */
671 memmove(header
, header
+1, 3);
672 hr
= IAsyncReader_SyncRead(pPin
->pReader
, ++bytepos
+ 3, 1, header
+ 3);
676 if (hr
!= S_OK
|| duration
> newpos
)
684 PullPin
*pin
= This
->Parser
.pInputPin
;
686 TRACE("Moving sound to %08u bytes!\n", (DWORD
)bytepos
);
688 EnterCriticalSection(&pin
->thread_lock
);
689 IPin_BeginFlush(&pin
->pin
.IPin_iface
);
691 /* Make sure this is done while stopped, BeginFlush takes care of this */
692 EnterCriticalSection(&This
->Parser
.filter
.csFilter
);
693 memcpy(This
->header
, header
, 4);
695 pin
->rtStart
= pin
->rtCurrent
= MEDIATIME_FROM_BYTES(bytepos
);
696 pin
->rtStop
= MEDIATIME_FROM_BYTES((REFERENCE_TIME
)This
->EndOfFile
);
698 This
->position
= newpos
;
699 LeaveCriticalSection(&This
->Parser
.filter
.csFilter
);
701 TRACE("Done flushing\n");
702 IPin_EndFlush(&pin
->pin
.IPin_iface
);
703 LeaveCriticalSection(&pin
->thread_lock
);
708 static HRESULT
MPEGSplitter_disconnect(LPVOID iface
)
710 /* TODO: Find memory leaks etc */
714 static HRESULT
MPEGSplitter_first_request(LPVOID iface
)
716 MPEGSplitterImpl
*This
= iface
;
717 PullPin
*pin
= This
->Parser
.pInputPin
;
720 IMediaSample
*sample
;
722 TRACE("Seeking? %d\n", This
->seek
);
724 hr
= parse_header(This
->header
, &length
, NULL
);
727 if (pin
->rtCurrent
>= pin
->rtStop
)
729 /* Last sample has already been queued, request nothing more */
734 hr
= IMemAllocator_GetBuffer(pin
->pAlloc
, &sample
, NULL
, NULL
, 0);
736 pin
->rtNext
= pin
->rtCurrent
;
739 LONGLONG rtSampleStart
= pin
->rtNext
;
740 /* Add 4 for the next header, which should hopefully work */
741 LONGLONG rtSampleStop
= rtSampleStart
+ MEDIATIME_FROM_BYTES(length
+ 4);
743 if (rtSampleStop
> pin
->rtStop
)
744 rtSampleStop
= MEDIATIME_FROM_BYTES(ALIGNUP(BYTES_FROM_MEDIATIME(pin
->rtStop
), pin
->cbAlign
));
746 IMediaSample_SetTime(sample
, &rtSampleStart
, &rtSampleStop
);
747 IMediaSample_SetPreroll(sample
, FALSE
);
748 IMediaSample_SetDiscontinuity(sample
, TRUE
);
749 IMediaSample_SetSyncPoint(sample
, 1);
752 hr
= IAsyncReader_Request(pin
->pReader
, sample
, 0);
755 pin
->rtCurrent
= pin
->rtNext
;
756 pin
->rtNext
= rtSampleStop
;
759 IMediaSample_Release(sample
);
762 ERR("Horsemen of the apocalypse came to bring error 0x%08x\n", hr
);
767 static HRESULT WINAPI
MPEGSplitter_QueryInterface(IBaseFilter
*iface
, REFIID riid
, void **ppv
)
769 MPEGSplitterImpl
*This
= impl_from_IBaseFilter(iface
);
770 TRACE("(%s, %p)\n", qzdebugstr_guid(riid
), ppv
);
774 if ( IsEqualIID(riid
, &IID_IUnknown
)
775 || IsEqualIID(riid
, &IID_IPersist
)
776 || IsEqualIID(riid
, &IID_IMediaFilter
)
777 || IsEqualIID(riid
, &IID_IBaseFilter
) )
779 else if ( IsEqualIID(riid
, &IID_IAMStreamSelect
) )
780 *ppv
= &This
->IAMStreamSelect_iface
;
784 IBaseFilter_AddRef(iface
);
788 if (!IsEqualIID(riid
, &IID_IPin
) && !IsEqualIID(riid
, &IID_IVideoWindow
))
789 FIXME("No interface for %s!\n", qzdebugstr_guid(riid
));
791 return E_NOINTERFACE
;
794 static const IBaseFilterVtbl MPEGSplitter_Vtbl
=
796 MPEGSplitter_QueryInterface
,
804 Parser_SetSyncSource
,
805 Parser_GetSyncSource
,
807 BaseFilterImpl_FindPin
,
808 Parser_QueryFilterInfo
,
809 Parser_JoinFilterGraph
,
810 Parser_QueryVendorInfo
813 static HRESULT WINAPI
AMStreamSelect_QueryInterface(IAMStreamSelect
*iface
, REFIID riid
, void **ppv
)
815 MPEGSplitterImpl
*This
= impl_from_IAMStreamSelect(iface
);
817 return IBaseFilter_QueryInterface(&This
->Parser
.filter
.IBaseFilter_iface
, riid
, ppv
);
820 static ULONG WINAPI
AMStreamSelect_AddRef(IAMStreamSelect
*iface
)
822 MPEGSplitterImpl
*This
= impl_from_IAMStreamSelect(iface
);
824 return IBaseFilter_AddRef(&This
->Parser
.filter
.IBaseFilter_iface
);
827 static ULONG WINAPI
AMStreamSelect_Release(IAMStreamSelect
*iface
)
829 MPEGSplitterImpl
*This
= impl_from_IAMStreamSelect(iface
);
831 return IBaseFilter_Release(&This
->Parser
.filter
.IBaseFilter_iface
);
834 static HRESULT WINAPI
AMStreamSelect_Count(IAMStreamSelect
*iface
, DWORD
*streams
)
836 MPEGSplitterImpl
*This
= impl_from_IAMStreamSelect(iface
);
838 FIXME("(%p/%p)->(%p) stub!\n", This
, iface
, streams
);
843 static HRESULT WINAPI
AMStreamSelect_Info(IAMStreamSelect
*iface
, LONG index
, AM_MEDIA_TYPE
**media_type
, DWORD
*flags
, LCID
*lcid
, DWORD
*group
, WCHAR
**name
, IUnknown
**object
, IUnknown
**unknown
)
845 MPEGSplitterImpl
*This
= impl_from_IAMStreamSelect(iface
);
847 FIXME("(%p/%p)->(%d,%p,%p,%p,%p,%p,%p,%p) stub!\n", This
, iface
, index
, media_type
, flags
, lcid
, group
, name
, object
, unknown
);
852 static HRESULT WINAPI
AMStreamSelect_Enable(IAMStreamSelect
*iface
, LONG index
, DWORD flags
)
854 MPEGSplitterImpl
*This
= impl_from_IAMStreamSelect(iface
);
856 FIXME("(%p/%p)->(%d,%x) stub!\n", This
, iface
, index
, flags
);
861 static const IAMStreamSelectVtbl AMStreamSelectVtbl
=
863 AMStreamSelect_QueryInterface
,
864 AMStreamSelect_AddRef
,
865 AMStreamSelect_Release
,
866 AMStreamSelect_Count
,
868 AMStreamSelect_Enable
871 HRESULT
MPEGSplitter_create(IUnknown
* pUnkOuter
, LPVOID
* ppv
)
873 MPEGSplitterImpl
*This
;
876 TRACE("(%p, %p)\n", pUnkOuter
, ppv
);
881 return CLASS_E_NOAGGREGATION
;
883 This
= CoTaskMemAlloc(sizeof(MPEGSplitterImpl
));
885 return E_OUTOFMEMORY
;
887 ZeroMemory(This
, sizeof(MPEGSplitterImpl
));
888 hr
= Parser_Create(&(This
->Parser
), &MPEGSplitter_Vtbl
, &CLSID_MPEG1Splitter
, MPEGSplitter_process_sample
, MPEGSplitter_query_accept
, MPEGSplitter_pre_connect
, MPEGSplitter_cleanup
, MPEGSplitter_disconnect
, MPEGSplitter_first_request
, NULL
, NULL
, MPEGSplitter_seek
, NULL
);
894 This
->IAMStreamSelect_iface
.lpVtbl
= &AMStreamSelectVtbl
;
897 /* Note: This memory is managed by the parser filter once created */
898 *ppv
= &This
->Parser
.filter
.IBaseFilter_iface
;