4 * Copyright 2005 Christian Costa
6 * This library is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
11 * This library is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with this library; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
21 #include "quartz_private.h"
29 #include "wine/unicode.h"
30 #include "wine/debug.h"
37 WINE_DEFAULT_DEBUG_CHANNEL(quartz
);
39 static const WCHAR wcsOutputPinName
[] = {'o','u','t','p','u','t',' ','p','i','n',0};
41 typedef struct WAVEParserImpl
44 LONGLONG StartOfFile
; /* in media time */
46 DWORD nAvgBytesPerSec
;
50 static inline WAVEParserImpl
*impl_from_IMediaSeeking( IMediaSeeking
*iface
)
52 return CONTAINING_RECORD(iface
, WAVEParserImpl
, Parser
.sourceSeeking
.IMediaSeeking_iface
);
55 static inline WAVEParserImpl
*impl_from_IBaseFilter( IBaseFilter
*iface
)
57 return CONTAINING_RECORD(iface
, WAVEParserImpl
, Parser
.filter
.IBaseFilter_iface
);
60 static LONGLONG
bytepos_to_duration(WAVEParserImpl
*This
, LONGLONG bytepos
)
62 LONGLONG duration
= BYTES_FROM_MEDIATIME(bytepos
- This
->StartOfFile
);
64 duration
/= This
->nAvgBytesPerSec
;
69 static LONGLONG
duration_to_bytepos(WAVEParserImpl
*This
, LONGLONG duration
)
73 bytepos
= This
->nAvgBytesPerSec
;
76 bytepos
-= bytepos
% This
->nBlockAlign
;
77 bytepos
+= BYTES_FROM_MEDIATIME(This
->StartOfFile
);
79 return MEDIATIME_FROM_BYTES(bytepos
);
82 static HRESULT
WAVEParser_Sample(LPVOID iface
, IMediaSample
* pSample
, DWORD_PTR cookie
)
84 WAVEParserImpl
*This
= iface
;
85 LPBYTE pbSrcStream
= NULL
;
86 ULONG cbSrcStream
= 0;
87 REFERENCE_TIME tStart
, tStop
;
89 IMediaSample
*newsample
= NULL
;
90 Parser_OutputPin
*pOutputPin
;
91 PullPin
*pin
= This
->Parser
.pInputPin
;
93 IMediaSample_GetPointer(pSample
, &pbSrcStream
);
94 hr
= IMediaSample_GetTime(pSample
, &tStart
, &tStop
);
96 cbSrcStream
= IMediaSample_GetActualDataLength(pSample
);
101 TRACE(".. Why do I need you?\n");
105 pOutputPin
= unsafe_impl_Parser_OutputPin_from_IPin(This
->Parser
.ppPins
[1]);
108 hr
= IMemAllocator_GetBuffer(pin
->pAlloc
, &newsample
, NULL
, NULL
, 0);
112 LONGLONG rtSampleStart
= pin
->rtNext
;
113 /* Add 4 for the next header, which should hopefully work */
114 LONGLONG rtSampleStop
= rtSampleStart
+ MEDIATIME_FROM_BYTES(IMediaSample_GetSize(newsample
));
116 if (rtSampleStop
> pin
->rtStop
)
117 rtSampleStop
= MEDIATIME_FROM_BYTES(ALIGNUP(BYTES_FROM_MEDIATIME(pin
->rtStop
), pin
->cbAlign
));
119 IMediaSample_SetTime(newsample
, &rtSampleStart
, &rtSampleStop
);
121 pin
->rtCurrent
= pin
->rtNext
;
122 pin
->rtNext
= rtSampleStop
;
124 IMediaSample_SetPreroll(newsample
, FALSE
);
125 IMediaSample_SetDiscontinuity(newsample
, FALSE
);
126 IMediaSample_SetSyncPoint(newsample
, TRUE
);
128 hr
= IAsyncReader_Request(pin
->pReader
, newsample
, 0);
133 REFERENCE_TIME tAviStart
, tAviStop
;
135 IMediaSample_SetSyncPoint(pSample
, TRUE
);
136 pOutputPin
->dwSamplesProcessed
++;
138 tAviStart
= bytepos_to_duration(This
, tStart
);
139 tAviStop
= bytepos_to_duration(This
, tStart
+ MEDIATIME_FROM_BYTES(IMediaSample_GetActualDataLength(pSample
)));
141 IMediaSample_SetTime(pSample
, &tAviStart
, &tAviStop
);
143 hr
= BaseOutputPinImpl_Deliver(&pOutputPin
->pin
, pSample
);
144 if (hr
!= S_OK
&& hr
!= S_FALSE
&& hr
!= VFW_E_WRONG_STATE
)
145 ERR("Error sending sample (%x)\n", hr
);
147 /* Unset progression if denied! */
148 This
->Parser
.pInputPin
->rtCurrent
= tStart
;
151 if (tStop
>= This
->EndOfFile
|| (bytepos_to_duration(This
, tStop
) >= This
->Parser
.sourceSeeking
.llStop
) || hr
== VFW_E_NOT_CONNECTED
)
155 TRACE("End of file reached\n");
157 for (i
= 0; i
< This
->Parser
.cStreams
; i
++)
162 TRACE("Send End Of Stream to output pin %u\n", i
);
164 hr
= IPin_ConnectedTo(This
->Parser
.ppPins
[i
+1], &ppin
);
167 hr
= IPin_EndOfStream(ppin
);
177 /* Force the pullpin thread to stop */
184 static HRESULT
WAVEParser_QueryAccept(LPVOID iface
, const AM_MEDIA_TYPE
* pmt
)
186 if (!IsEqualIID(&pmt
->majortype
, &MEDIATYPE_Stream
))
188 if (IsEqualIID(&pmt
->subtype
, &MEDIASUBTYPE_WAVE
))
190 if (IsEqualIID(&pmt
->subtype
, &MEDIASUBTYPE_AU
) || IsEqualIID(&pmt
->subtype
, &MEDIASUBTYPE_AIFF
))
191 FIXME("AU and AIFF files not supported yet!\n");
195 static HRESULT WINAPI
WAVEParserImpl_seek(IMediaSeeking
*iface
)
197 WAVEParserImpl
*This
= impl_from_IMediaSeeking(iface
);
198 PullPin
*pPin
= This
->Parser
.pInputPin
;
200 LONGLONG newpos
, curpos
, endpos
, bytepos
;
202 newpos
= This
->Parser
.sourceSeeking
.llCurrent
;
203 curpos
= bytepos_to_duration(This
, pPin
->rtCurrent
);
204 endpos
= bytepos_to_duration(This
, This
->EndOfFile
);
205 bytepos
= duration_to_bytepos(This
, newpos
);
209 WARN("Requesting position %x%08x beyond end of stream %x%08x\n", (DWORD
)(newpos
>>32), (DWORD
)newpos
, (DWORD
)(endpos
>>32), (DWORD
)endpos
);
213 if (curpos
/1000000 == newpos
/1000000)
215 TRACE("Requesting position %x%08x same as current position %x%08x\n", (DWORD
)(newpos
>>32), (DWORD
)newpos
, (DWORD
)(curpos
>>32), (DWORD
)curpos
);
219 TRACE("Moving sound to %08u bytes!\n", (DWORD
)BYTES_FROM_MEDIATIME(bytepos
));
221 EnterCriticalSection(&pPin
->thread_lock
);
222 IPin_BeginFlush(&pPin
->pin
.IPin_iface
);
224 /* Make sure this is done while stopped, BeginFlush takes care of this */
225 EnterCriticalSection(&This
->Parser
.filter
.csFilter
);
226 IPin_ConnectedTo(This
->Parser
.ppPins
[1], &victim
);
229 IPin_NewSegment(victim
, newpos
, endpos
, pPin
->dRate
);
230 IPin_Release(victim
);
233 pPin
->rtStart
= pPin
->rtCurrent
= bytepos
;
234 unsafe_impl_Parser_OutputPin_from_IPin(This
->Parser
.ppPins
[1])->dwSamplesProcessed
= 0;
235 LeaveCriticalSection(&This
->Parser
.filter
.csFilter
);
237 TRACE("Done flushing\n");
238 IPin_EndFlush(&pPin
->pin
.IPin_iface
);
239 LeaveCriticalSection(&pPin
->thread_lock
);
244 static HRESULT
WAVEParser_InputPin_PreConnect(IPin
* iface
, IPin
* pConnectPin
, ALLOCATOR_PROPERTIES
*props
)
246 PullPin
*This
= impl_PullPin_from_IPin(iface
);
250 LONGLONG pos
= 0; /* in bytes */
253 WAVEParserImpl
* pWAVEParser
= impl_from_IBaseFilter(This
->pin
.pinInfo
.pFilter
);
255 piOutput
.dir
= PINDIR_OUTPUT
;
256 piOutput
.pFilter
= &pWAVEParser
->Parser
.filter
.IBaseFilter_iface
;
257 lstrcpynW(piOutput
.achName
, wcsOutputPinName
, sizeof(piOutput
.achName
) / sizeof(piOutput
.achName
[0]));
259 hr
= IAsyncReader_SyncRead(This
->pReader
, pos
, sizeof(list
), (BYTE
*)&list
);
262 if (list
.fcc
!= FOURCC_RIFF
)
264 ERR("Input stream not a RIFF file\n");
267 if (list
.cb
> 1 * 1024 * 1024 * 1024) /* cannot be more than 1Gb in size */
269 ERR("Input stream violates RIFF spec\n");
272 if (list
.fccListType
!= mmioFOURCC('W','A','V','E'))
274 ERR("Input stream not an WAVE RIFF file\n");
278 hr
= IAsyncReader_SyncRead(This
->pReader
, pos
, sizeof(chunk
), (BYTE
*)&chunk
);
279 pos
+= sizeof(chunk
);
280 if (chunk
.fcc
!= mmioFOURCC('f','m','t',' '))
282 ERR("Expected 'fmt ' chunk, but got %.04s\n", (LPSTR
)&chunk
.fcc
);
286 amt
.majortype
= MEDIATYPE_Audio
;
287 amt
.formattype
= FORMAT_WaveFormatEx
;
288 amt
.cbFormat
= chunk
.cb
;
289 amt
.pbFormat
= CoTaskMemAlloc(amt
.cbFormat
);
291 IAsyncReader_SyncRead(This
->pReader
, pos
, amt
.cbFormat
, amt
.pbFormat
);
292 amt
.subtype
= MEDIATYPE_Audio
;
293 amt
.subtype
.Data1
= ((WAVEFORMATEX
*)amt
.pbFormat
)->wFormatTag
;
296 hr
= IAsyncReader_SyncRead(This
->pReader
, pos
, sizeof(chunk
), (BYTE
*)&chunk
);
297 if (chunk
.fcc
== mmioFOURCC('f','a','c','t'))
299 FIXME("'fact' chunk not supported yet\n");
300 pos
+= sizeof(chunk
) + chunk
.cb
;
301 hr
= IAsyncReader_SyncRead(This
->pReader
, pos
, sizeof(chunk
), (BYTE
*)&chunk
);
303 if (chunk
.fcc
!= mmioFOURCC('d','a','t','a'))
305 ERR("Expected 'data' chunk, but got %.04s\n", (LPSTR
)&chunk
.fcc
);
311 pWAVEParser
->StartOfFile
= MEDIATIME_FROM_BYTES(pos
+ sizeof(RIFFCHUNK
));
312 pWAVEParser
->EndOfFile
= MEDIATIME_FROM_BYTES(pos
+ chunk
.cb
+ sizeof(RIFFCHUNK
));
318 props
->cbAlign
= ((WAVEFORMATEX
*)amt
.pbFormat
)->nBlockAlign
;
320 props
->cbBuffer
= 4096;
322 pWAVEParser
->nBlockAlign
= ((WAVEFORMATEX
*)amt
.pbFormat
)->nBlockAlign
;
323 pWAVEParser
->nAvgBytesPerSec
= ((WAVEFORMATEX
*)amt
.pbFormat
)->nAvgBytesPerSec
;
324 hr
= Parser_AddPin(&(pWAVEParser
->Parser
), &piOutput
, props
, &amt
);
325 CoTaskMemFree(amt
.pbFormat
);
327 pWAVEParser
->Parser
.sourceSeeking
.llCurrent
= 0;
328 pWAVEParser
->Parser
.sourceSeeking
.llStop
= pWAVEParser
->Parser
.sourceSeeking
.llDuration
= bytepos_to_duration(pWAVEParser
, pWAVEParser
->EndOfFile
);
329 TRACE("Duration: %u seconds\n", (DWORD
)(pWAVEParser
->Parser
.sourceSeeking
.llDuration
/ (LONGLONG
)10000000));
331 This
->rtStop
= pWAVEParser
->EndOfFile
;
332 This
->rtStart
= pWAVEParser
->StartOfFile
;
334 TRACE("WAVE File ok\n");
339 static HRESULT
WAVEParser_Cleanup(LPVOID iface
)
341 WAVEParserImpl
*This
= iface
;
343 TRACE("(%p)->()\n", This
);
348 static HRESULT
WAVEParser_first_request(LPVOID iface
)
350 WAVEParserImpl
*This
= iface
;
351 PullPin
*pin
= This
->Parser
.pInputPin
;
353 IMediaSample
*sample
;
355 if (pin
->rtCurrent
>= pin
->rtStop
)
357 /* Last sample has already been queued, request nothing more */
362 hr
= IMemAllocator_GetBuffer(pin
->pAlloc
, &sample
, NULL
, NULL
, 0);
364 pin
->rtNext
= pin
->rtCurrent
;
367 LONGLONG rtSampleStart
= pin
->rtNext
;
368 /* Add 4 for the next header, which should hopefully work */
369 LONGLONG rtSampleStop
= rtSampleStart
+ MEDIATIME_FROM_BYTES(IMediaSample_GetSize(sample
));
370 Parser_OutputPin
*outpin
= unsafe_impl_Parser_OutputPin_from_IPin(This
->Parser
.ppPins
[1]);
372 if (rtSampleStop
> pin
->rtStop
)
373 rtSampleStop
= MEDIATIME_FROM_BYTES(ALIGNUP(BYTES_FROM_MEDIATIME(pin
->rtStop
), pin
->cbAlign
));
375 IMediaSample_SetTime(sample
, &rtSampleStart
, &rtSampleStop
);
377 pin
->rtCurrent
= pin
->rtNext
;
378 pin
->rtNext
= rtSampleStop
;
380 IMediaSample_SetPreroll(sample
, FALSE
);
381 if (!outpin
->dwSamplesProcessed
++)
382 IMediaSample_SetDiscontinuity(sample
, TRUE
);
384 IMediaSample_SetDiscontinuity(sample
, FALSE
);
386 hr
= IAsyncReader_Request(pin
->pReader
, sample
, 0);
389 ERR("Horsemen of the apocalypse came to bring error 0x%08x %p\n", hr
, sample
);
394 static HRESULT
WAVEParser_disconnect(LPVOID iface
)
396 /* TODO: Find and plug memory leaks */
400 static const IBaseFilterVtbl WAVEParser_Vtbl
=
402 Parser_QueryInterface
,
410 Parser_SetSyncSource
,
411 Parser_GetSyncSource
,
414 Parser_QueryFilterInfo
,
415 Parser_JoinFilterGraph
,
416 Parser_QueryVendorInfo
419 HRESULT
WAVEParser_create(IUnknown
* pUnkOuter
, LPVOID
* ppv
)
422 WAVEParserImpl
* This
;
424 TRACE("(%p, %p)\n", pUnkOuter
, ppv
);
429 return CLASS_E_NOAGGREGATION
;
431 /* Note: This memory is managed by the transform filter once created */
432 This
= CoTaskMemAlloc(sizeof(WAVEParserImpl
));
434 hr
= Parser_Create(&(This
->Parser
), &WAVEParser_Vtbl
, &CLSID_WAVEParser
, WAVEParser_Sample
, WAVEParser_QueryAccept
, WAVEParser_InputPin_PreConnect
, WAVEParser_Cleanup
, WAVEParser_disconnect
, WAVEParser_first_request
, NULL
, NULL
, WAVEParserImpl_seek
, NULL
);
439 *ppv
= &This
->Parser
.filter
.IBaseFilter_iface
;