dlls/quartz/waveparser.c

   1 /*
   2  * WAVE Parser Filter
   3  *
   4  * Copyright 2005 Christian Costa
   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., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  19  */
  20
  21 #include "quartz_private.h"
  22 #include "pin.h"
  23
  24 #include "uuids.h"
  25 #include "aviriff.h"
  26 #include "vfwmsgs.h"
  27 #include "mmsystem.h"
  28
  29 #include "wine/unicode.h"
  30 #include "wine/debug.h"
  31
  32 #include <math.h>
  33 #include <assert.h>
  34
  35 #include "parser.h"
  36
  37 WINE_DEFAULT_DEBUG_CHANNEL(quartz);
  38
  39 static const WCHAR wcsOutputPinName[] = {'o','u','t','p','u','t',' ','p','i','n',0};
  40
  41 typedef struct WAVEParserImpl
  42 {
  43     ParserImpl Parser;
  44     LONGLONG StartOfFile; /* in media time */
  45     LONGLONG EndOfFile;
  46     DWORD nAvgBytesPerSec;
  47     DWORD nBlockAlign;
  48 } WAVEParserImpl;
  49
  50 static inline WAVEParserImpl *impl_from_IMediaSeeking( IMediaSeeking *iface )
  51 {
  52     return CONTAINING_RECORD(iface, WAVEParserImpl, Parser.sourceSeeking.IMediaSeeking_iface);
  53 }
  54
  55 static inline WAVEParserImpl *impl_from_IBaseFilter( IBaseFilter *iface )
  56 {
  57     return CONTAINING_RECORD(iface, WAVEParserImpl, Parser.filter.IBaseFilter_iface);
  58 }
  59
  60 static LONGLONG bytepos_to_duration(WAVEParserImpl *This, LONGLONG bytepos)
  61 {
  62     LONGLONG duration = BYTES_FROM_MEDIATIME(bytepos - This->StartOfFile);
  63     duration *= 10000000;
  64     duration /= This->nAvgBytesPerSec;
  65
  66     return duration;
  67 }
  68
  69 static LONGLONG duration_to_bytepos(WAVEParserImpl *This, LONGLONG duration)
  70 {
  71     LONGLONG bytepos;
  72
  73     bytepos = This->nAvgBytesPerSec;
  74     bytepos *= duration;
  75     bytepos /= 10000000;
  76     bytepos -= bytepos % This->nBlockAlign;
  77     bytepos += BYTES_FROM_MEDIATIME(This->StartOfFile);
  78
  79     return MEDIATIME_FROM_BYTES(bytepos);
  80 }
  81
  82 static HRESULT WAVEParser_Sample(LPVOID iface, IMediaSample * pSample, DWORD_PTR cookie)
  83 {
  84     WAVEParserImpl *This = iface;
  85     LPBYTE pbSrcStream = NULL;
  86     ULONG cbSrcStream = 0;
  87     REFERENCE_TIME tStart, tStop;
  88     HRESULT hr;
  89     IMediaSample *newsample = NULL;
  90     Parser_OutputPin *pOutputPin;
  91     PullPin *pin = This->Parser.pInputPin;
  92
  93     IMediaSample_GetPointer(pSample, &pbSrcStream);
  94     hr = IMediaSample_GetTime(pSample, &tStart, &tStop);
  95
  96     cbSrcStream = IMediaSample_GetActualDataLength(pSample);
  97
  98     /* Flush occurring */
  99     if (cbSrcStream == 0)
 100     {
 101         TRACE(".. Why do I need you?\n");
 102         return S_OK;
 103     }
 104
 105     pOutputPin = unsafe_impl_Parser_OutputPin_from_IPin(This->Parser.ppPins[1]);
 106
 107     if (SUCCEEDED(hr))
 108         hr = IMemAllocator_GetBuffer(pin->pAlloc, &newsample, NULL, NULL, 0);
 109
 110     if (SUCCEEDED(hr))
 111     {
 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));
 115
 116         if (rtSampleStop > pin->rtStop)
 117             rtSampleStop = MEDIATIME_FROM_BYTES(ALIGNUP(BYTES_FROM_MEDIATIME(pin->rtStop), pin->cbAlign));
 118
 119         IMediaSample_SetTime(newsample, &rtSampleStart, &rtSampleStop);
 120
 121         pin->rtCurrent = pin->rtNext;
 122         pin->rtNext = rtSampleStop;
 123
 124         IMediaSample_SetPreroll(newsample, FALSE);
 125         IMediaSample_SetDiscontinuity(newsample, FALSE);
 126         IMediaSample_SetSyncPoint(newsample, TRUE);
 127
 128         hr = IAsyncReader_Request(pin->pReader, newsample, 0);
 129     }
 130
 131     if (SUCCEEDED(hr))
 132     {
 133         REFERENCE_TIME tAviStart, tAviStop;
 134
 135         IMediaSample_SetSyncPoint(pSample, TRUE);
 136         pOutputPin->dwSamplesProcessed++;
 137
 138         tAviStart = bytepos_to_duration(This, tStart);
 139         tAviStop = bytepos_to_duration(This, tStart + MEDIATIME_FROM_BYTES(IMediaSample_GetActualDataLength(pSample)));
 140
 141         IMediaSample_SetTime(pSample, &tAviStart, &tAviStop);
 142
 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);
 146         else if (hr != S_OK)
 147             /* Unset progression if denied! */
 148             This->Parser.pInputPin->rtCurrent = tStart;
 149     }
 150
 151     if (tStop >= This->EndOfFile || (bytepos_to_duration(This, tStop) >= This->Parser.sourceSeeking.llStop) || hr == VFW_E_NOT_CONNECTED)
 152     {
 153         unsigned int i;
 154
 155         TRACE("End of file reached\n");
 156
 157         for (i = 0; i < This->Parser.cStreams; i++)
 158         {
 159             IPin* ppin;
 160             HRESULT hr;
 161
 162             TRACE("Send End Of Stream to output pin %u\n", i);
 163
 164             hr = IPin_ConnectedTo(This->Parser.ppPins[i+1], &ppin);
 165             if (SUCCEEDED(hr))
 166             {
 167                 hr = IPin_EndOfStream(ppin);
 168                 IPin_Release(ppin);
 169             }
 170             if (FAILED(hr))
 171             {
 172                 ERR("%x\n", hr);
 173                 break;
 174             }
 175         }
 176
 177         /* Force the pullpin thread to stop */
 178         hr = S_FALSE;
 179     }
 180
 181     return hr;
 182 }
 183
 184 static HRESULT WAVEParser_QueryAccept(LPVOID iface, const AM_MEDIA_TYPE * pmt)
 185 {
 186     if (!IsEqualIID(&pmt->majortype, &MEDIATYPE_Stream))
 187         return S_FALSE;
 188     if (IsEqualIID(&pmt->subtype, &MEDIASUBTYPE_WAVE))
 189         return S_OK;
 190     if (IsEqualIID(&pmt->subtype, &MEDIASUBTYPE_AU) || IsEqualIID(&pmt->subtype, &MEDIASUBTYPE_AIFF))
 191         FIXME("AU and AIFF files not supported yet!\n");
 192     return S_FALSE;
 193 }
 194
 195 static HRESULT WINAPI WAVEParserImpl_seek(IMediaSeeking *iface)
 196 {
 197     WAVEParserImpl *This = impl_from_IMediaSeeking(iface);
 198     PullPin *pPin = This->Parser.pInputPin;
 199     IPin *victim = NULL;
 200     LONGLONG newpos, curpos, endpos, bytepos;
 201
 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);
 206
 207     if (newpos > endpos)
 208     {
 209         WARN("Requesting position %x%08x beyond end of stream %x%08x\n", (DWORD)(newpos>>32), (DWORD)newpos, (DWORD)(endpos>>32), (DWORD)endpos);
 210         return E_INVALIDARG;
 211     }
 212
 213     if (curpos/1000000 == newpos/1000000)
 214     {
 215         TRACE("Requesting position %x%08x same as current position %x%08x\n", (DWORD)(newpos>>32), (DWORD)newpos, (DWORD)(curpos>>32), (DWORD)curpos);
 216         return S_OK;
 217     }
 218
 219     TRACE("Moving sound to %08u bytes!\n", (DWORD)BYTES_FROM_MEDIATIME(bytepos));
 220
 221     EnterCriticalSection(&pPin->thread_lock);
 222     IPin_BeginFlush(&pPin->pin.IPin_iface);
 223
 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);
 227     if (victim)
 228     {
 229         IPin_NewSegment(victim, newpos, endpos, pPin->dRate);
 230         IPin_Release(victim);
 231     }
 232
 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);
 236
 237     TRACE("Done flushing\n");
 238     IPin_EndFlush(&pPin->pin.IPin_iface);
 239     LeaveCriticalSection(&pPin->thread_lock);
 240
 241     return S_OK;
 242 }
 243
 244 static HRESULT WAVEParser_InputPin_PreConnect(IPin * iface, IPin * pConnectPin, ALLOCATOR_PROPERTIES *props)
 245 {
 246     PullPin *This = impl_PullPin_from_IPin(iface);
 247     HRESULT hr;
 248     RIFFLIST list;
 249     RIFFCHUNK chunk;
 250     LONGLONG pos = 0; /* in bytes */
 251     PIN_INFO piOutput;
 252     AM_MEDIA_TYPE amt;
 253     WAVEParserImpl * pWAVEParser = impl_from_IBaseFilter(This->pin.pinInfo.pFilter);
 254
 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]));
 258
 259     hr = IAsyncReader_SyncRead(This->pReader, pos, sizeof(list), (BYTE *)&list);
 260     pos += sizeof(list);
 261
 262    if (list.fcc != FOURCC_RIFF)
 263     {
 264         ERR("Input stream not a RIFF file\n");
 265         return E_FAIL;
 266     }
 267     if (list.cb > 1 * 1024 * 1024 * 1024) /* cannot be more than 1Gb in size */
 268     {
 269         ERR("Input stream violates RIFF spec\n");
 270         return E_FAIL;
 271     }
 272     if (list.fccListType != mmioFOURCC('W','A','V','E'))
 273     {
 274         ERR("Input stream not an WAVE RIFF file\n");
 275         return E_FAIL;
 276     }
 277
 278     hr = IAsyncReader_SyncRead(This->pReader, pos, sizeof(chunk), (BYTE *)&chunk);
 279     pos += sizeof(chunk);
 280     if (chunk.fcc != mmioFOURCC('f','m','t',' '))
 281     {
 282         ERR("Expected 'fmt ' chunk, but got %.04s\n", (LPSTR)&chunk.fcc);
 283         return E_FAIL;
 284     }
 285
 286     amt.majortype = MEDIATYPE_Audio;
 287     amt.formattype = FORMAT_WaveFormatEx;
 288     amt.cbFormat = chunk.cb;
 289     amt.pbFormat = CoTaskMemAlloc(amt.cbFormat);
 290     amt.pUnk = NULL;
 291     IAsyncReader_SyncRead(This->pReader, pos, amt.cbFormat, amt.pbFormat);
 292     amt.subtype = MEDIATYPE_Audio;
 293     amt.subtype.Data1 = ((WAVEFORMATEX*)amt.pbFormat)->wFormatTag;
 294
 295     pos += chunk.cb;
 296     hr = IAsyncReader_SyncRead(This->pReader, pos, sizeof(chunk), (BYTE *)&chunk);
 297     if (chunk.fcc == mmioFOURCC('f','a','c','t'))
 298     {
 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);
 302     }
 303     if (chunk.fcc != mmioFOURCC('d','a','t','a'))
 304     {
 305         ERR("Expected 'data' chunk, but got %.04s\n", (LPSTR)&chunk.fcc);
 306         return E_FAIL;
 307     }
 308
 309     if (hr == S_OK)
 310     {
 311         pWAVEParser->StartOfFile = MEDIATIME_FROM_BYTES(pos + sizeof(RIFFCHUNK));
 312         pWAVEParser->EndOfFile = MEDIATIME_FROM_BYTES(pos + chunk.cb + sizeof(RIFFCHUNK));
 313     }
 314
 315     if (hr != S_OK)
 316         return E_FAIL;
 317
 318     props->cbAlign = ((WAVEFORMATEX*)amt.pbFormat)->nBlockAlign;
 319     props->cbPrefix = 0;
 320     props->cbBuffer = 4096;
 321     props->cBuffers = 3;
 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);
 326
 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));
 330
 331     This->rtStop = pWAVEParser->EndOfFile;
 332     This->rtStart = pWAVEParser->StartOfFile;
 333
 334     TRACE("WAVE File ok\n");
 335
 336     return hr;
 337 }
 338
 339 static HRESULT WAVEParser_Cleanup(LPVOID iface)
 340 {
 341     WAVEParserImpl *This = iface;
 342
 343     TRACE("(%p)->()\n", This);
 344
 345     return S_OK;
 346 }
 347
 348 static HRESULT WAVEParser_first_request(LPVOID iface)
 349 {
 350     WAVEParserImpl *This = iface;
 351     PullPin *pin = This->Parser.pInputPin;
 352     HRESULT hr;
 353     IMediaSample *sample;
 354
 355     if (pin->rtCurrent >= pin->rtStop)
 356     {
 357         /* Last sample has already been queued, request nothing more */
 358         TRACE("Done!\n");
 359         return S_OK;
 360     }
 361
 362     hr = IMemAllocator_GetBuffer(pin->pAlloc, &sample, NULL, NULL, 0);
 363
 364     pin->rtNext = pin->rtCurrent;
 365     if (SUCCEEDED(hr))
 366     {
 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]);
 371
 372         if (rtSampleStop > pin->rtStop)
 373             rtSampleStop = MEDIATIME_FROM_BYTES(ALIGNUP(BYTES_FROM_MEDIATIME(pin->rtStop), pin->cbAlign));
 374
 375         IMediaSample_SetTime(sample, &rtSampleStart, &rtSampleStop);
 376
 377         pin->rtCurrent = pin->rtNext;
 378         pin->rtNext = rtSampleStop;
 379
 380         IMediaSample_SetPreroll(sample, FALSE);
 381         if (!outpin->dwSamplesProcessed++)
 382             IMediaSample_SetDiscontinuity(sample, TRUE);
 383         else
 384             IMediaSample_SetDiscontinuity(sample, FALSE);
 385
 386         hr = IAsyncReader_Request(pin->pReader, sample, 0);
 387     }
 388     if (FAILED(hr))
 389         ERR("Horsemen of the apocalypse came to bring error 0x%08x %p\n", hr, sample);
 390
 391     return hr;
 392 }
 393
 394 static HRESULT WAVEParser_disconnect(LPVOID iface)
 395 {
 396     /* TODO: Find and plug memory leaks */
 397     return S_OK;
 398 }
 399
 400 static const IBaseFilterVtbl WAVEParser_Vtbl =
 401 {
 402     Parser_QueryInterface,
 403     Parser_AddRef,
 404     Parser_Release,
 405     Parser_GetClassID,
 406     Parser_Stop,
 407     Parser_Pause,
 408     Parser_Run,
 409     Parser_GetState,
 410     Parser_SetSyncSource,
 411     Parser_GetSyncSource,
 412     Parser_EnumPins,
 413     Parser_FindPin,
 414     Parser_QueryFilterInfo,
 415     Parser_JoinFilterGraph,
 416     Parser_QueryVendorInfo
 417 };
 418
 419 HRESULT WAVEParser_create(IUnknown * pUnkOuter, LPVOID * ppv)
 420 {
 421     HRESULT hr;
 422     WAVEParserImpl * This;
 423
 424     TRACE("(%p, %p)\n", pUnkOuter, ppv);
 425
 426     *ppv = NULL;
 427
 428     if (pUnkOuter)
 429         return CLASS_E_NOAGGREGATION;
 430
 431     /* Note: This memory is managed by the transform filter once created */
 432     This = CoTaskMemAlloc(sizeof(WAVEParserImpl));
 433
 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);
 435
 436     if (FAILED(hr))
 437         return hr;
 438
 439     *ppv = &This->Parser.filter.IBaseFilter_iface;
 440
 441     return hr;
 442 }