dll/directx/quartz/mpegsplit.c

   1 /*
   2  * MPEG Splitter Filter
   3  *
   4  * Copyright 2003 Robert Shearman
   5  * Copyright 2004-2005 Christian Costa
   6  * Copyright 2007 Chris Robinson
   7  * Copyright 2008 Maarten Lankhorst
   8  *
   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.
  13  *
  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.
  18  *
  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
  22  */
  23
  24 #include <assert.h>
  25 //#include <math.h>
  26
  27 #include "quartz_private.h"
  28 #include "control_private.h"
  29 #include "pin.h"
  30
  31 //#include "uuids.h"
  32 #include <mmreg.h>
  33 //#include "mmsystem.h"
  34
  35 //#include "winternl.h"
  36
  37 //#include "wine/unicode.h"
  38 #include <wine/debug.h>
  39
  40 #include "parser.h"
  41
  42 WINE_DEFAULT_DEBUG_CHANNEL(quartz);
  43
  44 #define SEQUENCE_HEADER_CODE     0xB3
  45 #define PACK_START_CODE          0xBA
  46
  47 #define SYSTEM_START_CODE        0xBB
  48 #define AUDIO_ELEMENTARY_STREAM  0xC0
  49 #define VIDEO_ELEMENTARY_STREAM  0xE0
  50
  51 #define MPEG_SYSTEM_HEADER 3
  52 #define MPEG_VIDEO_HEADER 2
  53 #define MPEG_AUDIO_HEADER 1
  54 #define MPEG_NO_HEADER 0
  55
  56 #define SEEK_INTERVAL (ULONGLONG)(10 * 10000000) /* Add an entry every 10 seconds */
  57
  58 struct seek_entry {
  59     ULONGLONG bytepos;
  60     ULONGLONG timepos;
  61 };
  62
  63 typedef struct MPEGSplitterImpl
  64 {
  65     ParserImpl Parser;
  66     LONGLONG EndOfFile;
  67     LONGLONG duration;
  68     LONGLONG position;
  69     DWORD begin_offset;
  70     BYTE header[4];
  71
  72     /* Whether we just seeked (or started playing) */
  73     BOOL seek;
  74
  75     /* Seeking cache */
  76     ULONG seek_entries;
  77     struct seek_entry *seektable;
  78 } MPEGSplitterImpl;
  79
  80 static int MPEGSplitter_head_check(const BYTE *header)
  81 {
  82     /* If this is a possible start code, check for a system or video header */
  83     if (header[0] == 0 && header[1] == 0 && header[2] == 1)
  84     {
  85         /* Check if we got a system or elementary stream start code */
  86         if (header[3] == PACK_START_CODE ||
  87             header[3] == VIDEO_ELEMENTARY_STREAM ||
  88             header[3] == AUDIO_ELEMENTARY_STREAM)
  89             return MPEG_SYSTEM_HEADER;
  90
  91         /* Check for a MPEG video sequence start code */
  92         if (header[3] == SEQUENCE_HEADER_CODE)
  93             return MPEG_VIDEO_HEADER;
  94     }
  95
  96     /* This should give a good guess if we have an MPEG audio header */
  97     if(header[0] == 0xff && ((header[1]>>5)&0x7) == 0x7 &&
  98        ((header[1]>>1)&0x3) != 0 && ((header[2]>>4)&0xf) != 0xf &&
  99        ((header[2]>>2)&0x3) != 0x3)
 100         return MPEG_AUDIO_HEADER;
 101
 102     /* Nothing yet.. */
 103     return MPEG_NO_HEADER;
 104 }
 105
 106 static const WCHAR wszAudioStream[] = {'A','u','d','i','o',0};
 107 static const WCHAR wszVideoStream[] = {'V','i','d','e','o',0};
 108
 109 static const DWORD freqs[10] = { 44100, 48000, 32000, 22050, 24000, 16000, 11025, 12000,  8000, 0 };
 110
 111 static const DWORD tabsel_123[2][3][16] = {
 112     { {0,32,64,96,128,160,192,224,256,288,320,352,384,416,448,},
 113       {0,32,48,56, 64, 80, 96,112,128,160,192,224,256,320,384,},
 114       {0,32,40,48, 56, 64, 80, 96,112,128,160,192,224,256,320,} },
 115
 116     { {0,32,48,56,64,80,96,112,128,144,160,176,192,224,256,},
 117       {0,8,16,24,32,40,48,56,64,80,96,112,128,144,160,},
 118       {0,8,16,24,32,40,48,56,64,80,96,112,128,144,160,} }
 119 };
 120
 121 static HRESULT parse_header(BYTE *header, LONGLONG *plen, LONGLONG *pduration)
 122 {
 123     LONGLONG duration;
 124
 125     int bitrate_index, freq_index, lsf = 1, mpeg1, layer, padding, bitrate, length;
 126
 127     if (!(header[0] == 0xff && ((header[1]>>5)&0x7) == 0x7 &&
 128        ((header[1]>>1)&0x3) != 0 && ((header[2]>>4)&0xf) != 0xf &&
 129        ((header[2]>>2)&0x3) != 0x3))
 130     {
 131         FIXME("Not a valid header: %02x:%02x\n", header[0], header[1]);
 132         return E_INVALIDARG;
 133     }
 134
 135     mpeg1 = (header[1]>>4)&0x1;
 136     if (mpeg1)
 137         lsf = ((header[1]>>3)&0x1)^1;
 138
 139     layer = 4-((header[1]>>1)&0x3);
 140     bitrate_index = ((header[2]>>4)&0xf);
 141     freq_index = ((header[2]>>2)&0x3) + (mpeg1?(lsf*3):6);
 142     padding = ((header[2]>>1)&0x1);
 143
 144     bitrate = tabsel_123[lsf][layer-1][bitrate_index] * 1000;
 145     if (!bitrate || layer != 3)
 146     {
 147         FIXME("Not a valid header: %02x:%02x:%02x:%02x\n", header[0], header[1], header[2], header[3]);
 148         return E_INVALIDARG;
 149     }
 150
 151
 152     if (layer == 3 || layer == 2)
 153         length = 144 * bitrate / freqs[freq_index] + padding;
 154     else
 155         length = 4 * (12 * bitrate / freqs[freq_index] + padding);
 156
 157     duration = (ULONGLONG)10000000 * (ULONGLONG)(length) / (ULONGLONG)(bitrate/8);
 158     *plen = length;
 159     if (pduration)
 160         *pduration += duration;
 161     return S_OK;
 162 }
 163
 164 static HRESULT FillBuffer(MPEGSplitterImpl *This, IMediaSample *pCurrentSample)
 165 {
 166     Parser_OutputPin * pOutputPin = (Parser_OutputPin*)This->Parser.ppPins[1];
 167     LONGLONG length = 0;
 168     LONGLONG pos = BYTES_FROM_MEDIATIME(This->Parser.pInputPin->rtNext);
 169     LONGLONG time = This->position;
 170     HRESULT hr;
 171     BYTE *fbuf = NULL;
 172     DWORD len = IMediaSample_GetActualDataLength(pCurrentSample);
 173
 174     TRACE("Source length: %u\n", len);
 175     IMediaSample_GetPointer(pCurrentSample, &fbuf);
 176
 177     /* Find the next valid header.. it <SHOULD> be right here */
 178     assert(parse_header(fbuf, &length, &This->position) == S_OK);
 179     IMediaSample_SetActualDataLength(pCurrentSample, length);
 180
 181     /* Queue the next sample */
 182     if (length + 4 == len)
 183     {
 184         PullPin *pin = This->Parser.pInputPin;
 185         LONGLONG stop = BYTES_FROM_MEDIATIME(pin->rtStop);
 186
 187         hr = S_OK;
 188         memcpy(This->header, fbuf + length, 4);
 189         while (FAILED(hr = parse_header(This->header, &length, NULL)))
 190         {
 191             memmove(This->header, This->header+1, 3);
 192             if (pos + 4 >= stop)
 193                 break;
 194             IAsyncReader_SyncRead(pin->pReader, ++pos, 1, This->header + 3);
 195         }
 196         pin->rtNext = MEDIATIME_FROM_BYTES(pos);
 197
 198         if (SUCCEEDED(hr))
 199         {
 200             /* Remove 4 for the last header, which should hopefully work */
 201             IMediaSample *sample = NULL;
 202             LONGLONG rtSampleStart = pin->rtNext - MEDIATIME_FROM_BYTES(4);
 203             LONGLONG rtSampleStop = rtSampleStart + MEDIATIME_FROM_BYTES(length + 4);
 204
 205             if (rtSampleStop > pin->rtStop)
 206                 rtSampleStop = MEDIATIME_FROM_BYTES(ALIGNUP(BYTES_FROM_MEDIATIME(pin->rtStop), pin->cbAlign));
 207
 208             hr = IMemAllocator_GetBuffer(pin->pAlloc, &sample, NULL, NULL, 0);
 209             if (SUCCEEDED(hr))
 210             {
 211                 IMediaSample_SetTime(sample, &rtSampleStart, &rtSampleStop);
 212                 IMediaSample_SetPreroll(sample, 0);
 213                 IMediaSample_SetDiscontinuity(sample, 0);
 214                 IMediaSample_SetSyncPoint(sample, 1);
 215                 pin->rtCurrent = rtSampleStart;
 216                 pin->rtNext = rtSampleStop;
 217                 hr = IAsyncReader_Request(pin->pReader, sample, 0);
 218             }
 219             if (FAILED(hr))
 220                 FIXME("o_Ox%08x\n", hr);
 221         }
 222     }
 223     /* If not, we're presumably at the end of file */
 224
 225     TRACE("Media time : %u.%03u\n", (DWORD)(This->position/10000000), (DWORD)((This->position/10000)%1000));
 226
 227     IMediaSample_SetTime(pCurrentSample, &time, &This->position);
 228
 229     hr = OutputPin_SendSample(&pOutputPin->pin, pCurrentSample);
 230
 231     if (hr != S_OK)
 232     {
 233         if (hr != S_FALSE)
 234             TRACE("Error sending sample (%x)\n", hr);
 235         else
 236             TRACE("S_FALSE (%d), holding\n", IMediaSample_GetActualDataLength(pCurrentSample));
 237     }
 238
 239     return hr;
 240 }
 241
 242
 243 static HRESULT MPEGSplitter_process_sample(LPVOID iface, IMediaSample * pSample, DWORD_PTR cookie)
 244 {
 245     MPEGSplitterImpl *This = iface;
 246     BYTE *pbSrcStream;
 247     DWORD cbSrcStream = 0;
 248     REFERENCE_TIME tStart, tStop, tAviStart = This->position;
 249     HRESULT hr;
 250
 251     hr = IMediaSample_GetTime(pSample, &tStart, &tStop);
 252     if (SUCCEEDED(hr))
 253     {
 254         cbSrcStream = IMediaSample_GetActualDataLength(pSample);
 255         hr = IMediaSample_GetPointer(pSample, &pbSrcStream);
 256     }
 257
 258     /* Flush occurring */
 259     if (cbSrcStream == 0)
 260     {
 261         FIXME(".. Why do I need you?\n");
 262         return S_OK;
 263     }
 264
 265     /* trace removed for performance reasons */
 266     /* TRACE("(%p), %llu -> %llu\n", pSample, tStart, tStop); */
 267
 268     /* Now, try to find a new header */
 269     hr = FillBuffer(This, pSample);
 270     if (hr != S_OK)
 271     {
 272         WARN("Failed with hres: %08x!\n", hr);
 273
 274         /* Unset progression if denied! */
 275         if (hr == VFW_E_WRONG_STATE || hr == S_FALSE)
 276         {
 277             memcpy(This->header, pbSrcStream, 4);
 278             This->Parser.pInputPin->rtCurrent = tStart;
 279             This->position = tAviStart;
 280         }
 281     }
 282
 283     if (BYTES_FROM_MEDIATIME(tStop) >= This->EndOfFile || This->position >= This->Parser.mediaSeeking.llStop)
 284     {
 285         unsigned int i;
 286
 287         TRACE("End of file reached\n");
 288
 289         for (i = 0; i < This->Parser.cStreams; i++)
 290         {
 291             IPin* ppin;
 292
 293             hr = IPin_ConnectedTo(This->Parser.ppPins[i+1], &ppin);
 294             if (SUCCEEDED(hr))
 295             {
 296                 hr = IPin_EndOfStream(ppin);
 297                 IPin_Release(ppin);
 298             }
 299             if (FAILED(hr))
 300                 WARN("Error sending EndOfStream to pin %u (%x)\n", i, hr);
 301         }
 302
 303         /* Force the pullpin thread to stop */
 304         hr = S_FALSE;
 305     }
 306
 307     return hr;
 308 }
 309
 310
 311 static HRESULT MPEGSplitter_query_accept(LPVOID iface, const AM_MEDIA_TYPE *pmt)
 312 {
 313     if (!IsEqualIID(&pmt->majortype, &MEDIATYPE_Stream))
 314         return S_FALSE;
 315
 316     if (IsEqualIID(&pmt->subtype, &MEDIASUBTYPE_MPEG1Audio))
 317         return S_OK;
 318
 319     if (IsEqualIID(&pmt->subtype, &MEDIASUBTYPE_MPEG1Video))
 320         FIXME("MPEG-1 video streams not yet supported.\n");
 321     else if (IsEqualIID(&pmt->subtype, &MEDIASUBTYPE_MPEG1System))
 322         FIXME("MPEG-1 system streams not yet supported.\n");
 323     else if (IsEqualIID(&pmt->subtype, &MEDIASUBTYPE_MPEG1VideoCD))
 324         FIXME("MPEG-1 VideoCD streams not yet supported.\n");
 325
 326     return S_FALSE;
 327 }
 328
 329
 330 static HRESULT MPEGSplitter_init_audio(MPEGSplitterImpl *This, const BYTE *header, PIN_INFO *ppiOutput, AM_MEDIA_TYPE *pamt)
 331 {
 332     WAVEFORMATEX *format;
 333     int bitrate_index;
 334     int freq_index;
 335     int mode_ext;
 336     int emphasis;
 337     int lsf = 1;
 338     int mpeg1;
 339     int layer;
 340     int mode;
 341
 342     ZeroMemory(pamt, sizeof(*pamt));
 343     ppiOutput->dir = PINDIR_OUTPUT;
 344     ppiOutput->pFilter = (IBaseFilter*)This;
 345     wsprintfW(ppiOutput->achName, wszAudioStream);
 346
 347     pamt->formattype = FORMAT_WaveFormatEx;
 348     pamt->majortype = MEDIATYPE_Audio;
 349     pamt->subtype = MEDIASUBTYPE_MPEG1AudioPayload;
 350
 351     pamt->lSampleSize = 0;
 352     pamt->bFixedSizeSamples = FALSE;
 353     pamt->bTemporalCompression = 0;
 354
 355     mpeg1 = (header[1]>>4)&0x1;
 356     if (mpeg1)
 357         lsf = ((header[1]>>3)&0x1)^1;
 358
 359     layer         = 4-((header[1]>>1)&0x3);
 360     bitrate_index =   ((header[2]>>4)&0xf);
 361     freq_index    =   ((header[2]>>2)&0x3) + (mpeg1?(lsf*3):6);
 362     mode          =   ((header[3]>>6)&0x3);
 363     mode_ext      =   ((header[3]>>4)&0x3);
 364     emphasis      =   ((header[3]>>0)&0x3);
 365
 366     if (!bitrate_index)
 367     {
 368         /* Set to highest bitrate so samples will fit in for sure */
 369         FIXME("Variable-bitrate audio not fully supported.\n");
 370         bitrate_index = 15;
 371     }
 372
 373     pamt->cbFormat = ((layer==3)? sizeof(MPEGLAYER3WAVEFORMAT) :
 374                                   sizeof(MPEG1WAVEFORMAT));
 375     pamt->pbFormat = CoTaskMemAlloc(pamt->cbFormat);
 376     if (!pamt->pbFormat)
 377         return E_OUTOFMEMORY;
 378     ZeroMemory(pamt->pbFormat, pamt->cbFormat);
 379     format = (WAVEFORMATEX*)pamt->pbFormat;
 380
 381     format->wFormatTag      = ((layer == 3) ? WAVE_FORMAT_MPEGLAYER3 :
 382                                               WAVE_FORMAT_MPEG);
 383     format->nChannels       = ((mode == 3) ? 1 : 2);
 384     format->nSamplesPerSec  = freqs[freq_index];
 385     format->nAvgBytesPerSec = tabsel_123[lsf][layer-1][bitrate_index] * 1000 / 8;
 386
 387     if (layer == 3)
 388         format->nBlockAlign = format->nAvgBytesPerSec * 8 * 144 /
 389                               (format->nSamplesPerSec<<lsf) + 1;
 390     else if (layer == 2)
 391         format->nBlockAlign = format->nAvgBytesPerSec * 8 * 144 /
 392                               format->nSamplesPerSec + 1;
 393     else
 394         format->nBlockAlign = 4 * (format->nAvgBytesPerSec * 8 * 12 / format->nSamplesPerSec + 1);
 395
 396     format->wBitsPerSample = 0;
 397
 398     if (layer == 3)
 399     {
 400         MPEGLAYER3WAVEFORMAT *mp3format = (MPEGLAYER3WAVEFORMAT*)format;
 401
 402         format->cbSize = MPEGLAYER3_WFX_EXTRA_BYTES;
 403
 404         mp3format->wID = MPEGLAYER3_ID_MPEG;
 405         mp3format->fdwFlags = MPEGLAYER3_FLAG_PADDING_ON;
 406         mp3format->nBlockSize = format->nBlockAlign;
 407         mp3format->nFramesPerBlock = 1;
 408
 409         /* Beware the evil magic numbers. This struct is apparently horribly
 410          * under-documented, and the only references I could find had it being
 411          * set to this with no real explanation. It works fine though, so I'm
 412          * not complaining (yet).
 413          */
 414         mp3format->nCodecDelay = 1393;
 415     }
 416     else
 417     {
 418         MPEG1WAVEFORMAT *mpgformat = (MPEG1WAVEFORMAT*)format;
 419
 420         format->cbSize = 22;
 421
 422         mpgformat->fwHeadLayer   = ((layer == 1) ? ACM_MPEG_LAYER1 :
 423                                     ((layer == 2) ? ACM_MPEG_LAYER2 :
 424                                      ACM_MPEG_LAYER3));
 425         mpgformat->dwHeadBitrate = format->nAvgBytesPerSec * 8;
 426         mpgformat->fwHeadMode    = ((mode == 3) ? ACM_MPEG_SINGLECHANNEL :
 427                                     ((mode == 2) ? ACM_MPEG_DUALCHANNEL :
 428                                      ((mode == 1) ? ACM_MPEG_JOINTSTEREO :
 429                                       ACM_MPEG_STEREO)));
 430         mpgformat->fwHeadModeExt = ((mode == 1) ? 0x0F : (1<<mode_ext));
 431         mpgformat->wHeadEmphasis = emphasis + 1;
 432         mpgformat->fwHeadFlags   = ACM_MPEG_ID_MPEG1;
 433     }
 434     pamt->subtype.Data1 = format->wFormatTag;
 435
 436     TRACE("MPEG audio stream detected:\n"
 437           "\tLayer %d (%#x)\n"
 438           "\tFrequency: %d\n"
 439           "\tChannels: %d (%d)\n"
 440           "\tBytesPerSec: %d\n",
 441           layer, format->wFormatTag, format->nSamplesPerSec,
 442           format->nChannels, mode, format->nAvgBytesPerSec);
 443
 444     dump_AM_MEDIA_TYPE(pamt);
 445
 446     return S_OK;
 447 }
 448
 449
 450 static HRESULT MPEGSplitter_pre_connect(IPin *iface, IPin *pConnectPin, ALLOCATOR_PROPERTIES *props)
 451 {
 452     PullPin *pPin = (PullPin *)iface;
 453     MPEGSplitterImpl *This = (MPEGSplitterImpl*)pPin->pin.pinInfo.pFilter;
 454     HRESULT hr;
 455     LONGLONG pos = 0; /* in bytes */
 456     BYTE header[10];
 457     int streamtype = 0;
 458     LONGLONG total, avail;
 459     AM_MEDIA_TYPE amt;
 460     PIN_INFO piOutput;
 461
 462     IAsyncReader_Length(pPin->pReader, &total, &avail);
 463     This->EndOfFile = total;
 464
 465     hr = IAsyncReader_SyncRead(pPin->pReader, pos, 4, header);
 466     if (SUCCEEDED(hr))
 467         pos += 4;
 468
 469     /* Skip ID3 v2 tag, if any */
 470     if (SUCCEEDED(hr) && !memcmp("ID3", header, 3))
 471     do {
 472         UINT length;
 473         hr = IAsyncReader_SyncRead(pPin->pReader, pos, 6, header + 4);
 474         if (FAILED(hr))
 475             break;
 476         pos += 6;
 477         TRACE("Found ID3 v2.%d.%d\n", header[3], header[4]);
 478         length  = (header[6] & 0x7F) << 21;
 479         length += (header[7] & 0x7F) << 14;
 480         length += (header[8] & 0x7F) << 7;
 481         length += (header[9] & 0x7F);
 482         TRACE("Length: %u\n", length);
 483         pos += length;
 484
 485         /* Read the real header for the mpeg splitter */
 486         hr = IAsyncReader_SyncRead(pPin->pReader, pos, 4, header);
 487         if (SUCCEEDED(hr))
 488             pos += 4;
 489         TRACE("%x:%x:%x:%x\n", header[0], header[1], header[2], header[3]);
 490     } while (0);
 491
 492     while(SUCCEEDED(hr) && !(streamtype=MPEGSplitter_head_check(header)))
 493     {
 494         TRACE("%x:%x:%x:%x\n", header[0], header[1], header[2], header[3]);
 495         /* No valid header yet; shift by a byte and check again */
 496         memmove(header, header+1, 3);
 497         hr = IAsyncReader_SyncRead(pPin->pReader, pos++, 1, header + 3);
 498     }
 499     if (FAILED(hr))
 500         return hr;
 501     pos -= 4;
 502     This->begin_offset = pos;
 503     memcpy(This->header, header, 4);
 504
 505     This->seektable[0].bytepos = pos;
 506     This->seektable[0].timepos = 0;
 507
 508     switch(streamtype)
 509     {
 510         case MPEG_AUDIO_HEADER:
 511         {
 512             LONGLONG duration = 0;
 513             DWORD last_entry = 0;
 514
 515             DWORD ticks = GetTickCount();
 516
 517             hr = MPEGSplitter_init_audio(This, header, &piOutput, &amt);
 518             if (SUCCEEDED(hr))
 519             {
 520                 WAVEFORMATEX *format = (WAVEFORMATEX*)amt.pbFormat;
 521
 522                 props->cbAlign = 1;
 523                 props->cbPrefix = 0;
 524                 /* Make the output buffer a multiple of the frame size */
 525                 props->cbBuffer = 0x4000 / format->nBlockAlign *
 526                                  format->nBlockAlign;
 527                 props->cBuffers = 3;
 528                 hr = Parser_AddPin(&(This->Parser), &piOutput, props, &amt);
 529             }
 530
 531             if (FAILED(hr))
 532             {
 533                 if (amt.pbFormat)
 534                     CoTaskMemFree(amt.pbFormat);
 535                 ERR("Could not create pin for MPEG audio stream (%x)\n", hr);
 536                 break;
 537             }
 538
 539             /* Check for idv1 tag, and remove it from stream if found */
 540             hr = IAsyncReader_SyncRead(pPin->pReader, This->EndOfFile-128, 3, header+4);
 541             if (FAILED(hr))
 542                 break;
 543             if (!strncmp((char*)header+4, "TAG", 3))
 544                 This->EndOfFile -= 128;
 545             This->Parser.pInputPin->rtStop = MEDIATIME_FROM_BYTES(This->EndOfFile);
 546             This->Parser.pInputPin->rtStart = This->Parser.pInputPin->rtCurrent = MEDIATIME_FROM_BYTES(This->begin_offset);
 547
 548             /* http://mpgedit.org/mpgedit/mpeg_format/mpeghdr.htm has a whole read up on audio headers */
 549             while (pos + 3 < This->EndOfFile)
 550             {
 551                 LONGLONG length = 0;
 552                 hr = IAsyncReader_SyncRead(pPin->pReader, pos, 4, header);
 553                 if (hr != S_OK)
 554                     break;
 555                 while (parse_header(header, &length, &duration))
 556                 {
 557                     /* No valid header yet; shift by a byte and check again */
 558                     memmove(header, header+1, 3);
 559                     hr = IAsyncReader_SyncRead(pPin->pReader, pos++, 1, header + 3);
 560                     if (hr != S_OK || This->EndOfFile - pos < 4)
 561                        break;
 562                 }
 563                 pos += length;
 564
 565                 if (This->seektable && (duration / SEEK_INTERVAL) > last_entry)
 566                 {
 567                     if (last_entry + 1 > duration / SEEK_INTERVAL)
 568                     {
 569                         ERR("Somehow skipped %d interval lengths instead of 1\n", (DWORD)(duration/SEEK_INTERVAL) - (last_entry + 1));
 570                     }
 571                     ++last_entry;
 572
 573                     TRACE("Entry: %u\n", last_entry);
 574                     if (last_entry >= This->seek_entries)
 575                     {
 576                         This->seek_entries += 64;
 577                         This->seektable = CoTaskMemRealloc(This->seektable, (This->seek_entries)*sizeof(struct seek_entry));
 578                     }
 579                     This->seektable[last_entry].bytepos = pos;
 580                     This->seektable[last_entry].timepos = duration;
 581                 }
 582
 583                 TRACE("Pos: %x%08x/%x%08x\n", (DWORD)(pos >> 32), (DWORD)pos, (DWORD)(This->EndOfFile>>32), (DWORD)This->EndOfFile);
 584             }
 585             hr = S_OK;
 586             TRACE("Duration: %d seconds\n", (DWORD)(duration / 10000000));
 587             TRACE("Parsing took %u ms\n", GetTickCount() - ticks);
 588             This->duration = duration;
 589
 590             This->Parser.mediaSeeking.llCurrent = 0;
 591             This->Parser.mediaSeeking.llDuration = duration;
 592             This->Parser.mediaSeeking.llStop = duration;
 593             break;
 594         }
 595         case MPEG_VIDEO_HEADER:
 596             FIXME("MPEG video processing not yet supported!\n");
 597             hr = E_FAIL;
 598             break;
 599         case MPEG_SYSTEM_HEADER:
 600             FIXME("MPEG system streams not yet supported!\n");
 601             hr = E_FAIL;
 602             break;
 603
 604         default:
 605             break;
 606     }
 607     This->position = 0;
 608
 609     return hr;
 610 }
 611
 612 static HRESULT MPEGSplitter_cleanup(LPVOID iface)
 613 {
 614     MPEGSplitterImpl *This = iface;
 615
 616     TRACE("(%p)\n", This);
 617
 618     return S_OK;
 619 }
 620
 621 static HRESULT MPEGSplitter_seek(IBaseFilter *iface)
 622 {
 623     MPEGSplitterImpl *This = (MPEGSplitterImpl*)iface;
 624     PullPin *pPin = This->Parser.pInputPin;
 625     LONGLONG newpos, timepos, bytepos;
 626     HRESULT hr = S_OK;
 627     BYTE header[4];
 628
 629     newpos = This->Parser.mediaSeeking.llCurrent;
 630
 631     if (newpos > This->duration)
 632     {
 633         WARN("Requesting position %x%08x beyond end of stream %x%08x\n", (DWORD)(newpos>>32), (DWORD)newpos, (DWORD)(This->duration>>32), (DWORD)This->duration);
 634         return E_INVALIDARG;
 635     }
 636
 637     if (This->position/1000000 == newpos/1000000)
 638     {
 639         TRACE("Requesting position %x%08x same as current position %x%08x\n", (DWORD)(newpos>>32), (DWORD)newpos, (DWORD)(This->position>>32), (DWORD)This->position);
 640         return S_OK;
 641     }
 642
 643     /* Position, cached */
 644     bytepos = This->seektable[newpos / SEEK_INTERVAL].bytepos;
 645     timepos = This->seektable[newpos / SEEK_INTERVAL].timepos;
 646
 647     hr = IAsyncReader_SyncRead(pPin->pReader, bytepos, 4, header);
 648     while (bytepos + 3 < This->EndOfFile)
 649     {
 650         LONGLONG length = 0;
 651         hr = IAsyncReader_SyncRead(pPin->pReader, bytepos, 4, header);
 652         if (hr != S_OK || timepos >= newpos)
 653             break;
 654
 655         while (parse_header(header, &length, &timepos) && bytepos + 3 < This->EndOfFile)
 656         {
 657             /* No valid header yet; shift by a byte and check again */
 658             memmove(header, header+1, 3);
 659             hr = IAsyncReader_SyncRead(pPin->pReader, ++bytepos, 1, header + 3);
 660             if (hr != S_OK)
 661                 break;
 662          }
 663          bytepos += length;
 664          TRACE("Pos: %x%08x/%x%08x\n", (DWORD)(bytepos >> 32), (DWORD)bytepos, (DWORD)(This->EndOfFile>>32), (DWORD)This->EndOfFile);
 665     }
 666
 667     if (SUCCEEDED(hr))
 668     {
 669         PullPin *pin = This->Parser.pInputPin;
 670         IPin *victim = NULL;
 671
 672         TRACE("Moving sound to %08u bytes!\n", (DWORD)bytepos);
 673
 674         EnterCriticalSection(&pin->thread_lock);
 675         IPin_BeginFlush((IPin *)pin);
 676
 677         /* Make sure this is done while stopped, BeginFlush takes care of this */
 678         EnterCriticalSection(&This->Parser.csFilter);
 679         memcpy(This->header, header, 4);
 680         IPin_ConnectedTo(This->Parser.ppPins[1], &victim);
 681         if (victim)
 682         {
 683             IPin_NewSegment(victim, newpos, This->duration, pin->dRate);
 684             IPin_Release(victim);
 685         }
 686
 687         pin->rtStart = pin->rtCurrent = MEDIATIME_FROM_BYTES(bytepos);
 688         pin->rtStop = MEDIATIME_FROM_BYTES((REFERENCE_TIME)This->EndOfFile);
 689         This->seek = TRUE;
 690         This->position = newpos;
 691         LeaveCriticalSection(&This->Parser.csFilter);
 692
 693         TRACE("Done flushing\n");
 694         IPin_EndFlush((IPin *)pin);
 695         LeaveCriticalSection(&pin->thread_lock);
 696     }
 697     return hr;
 698 }
 699
 700 static HRESULT MPEGSplitter_disconnect(LPVOID iface)
 701 {
 702     /* TODO: Find memory leaks etc */
 703     return S_OK;
 704 }
 705
 706 static HRESULT MPEGSplitter_first_request(LPVOID iface)
 707 {
 708     MPEGSplitterImpl *This = iface;
 709     PullPin *pin = This->Parser.pInputPin;
 710     HRESULT hr;
 711     LONGLONG length;
 712     IMediaSample *sample;
 713
 714     TRACE("Seeking? %d\n", This->seek);
 715     assert(parse_header(This->header, &length, NULL) == S_OK);
 716
 717     if (pin->rtCurrent >= pin->rtStop)
 718     {
 719         /* Last sample has already been queued, request nothing more */
 720         FIXME("Done!\n");
 721         return S_OK;
 722     }
 723
 724     hr = IMemAllocator_GetBuffer(pin->pAlloc, &sample, NULL, NULL, 0);
 725
 726     pin->rtNext = pin->rtCurrent;
 727     if (SUCCEEDED(hr))
 728     {
 729         LONGLONG rtSampleStart = pin->rtNext;
 730         /* Add 4 for the next header, which should hopefully work */
 731         LONGLONG rtSampleStop = rtSampleStart + MEDIATIME_FROM_BYTES(length + 4);
 732
 733         if (rtSampleStop > pin->rtStop)
 734             rtSampleStop = MEDIATIME_FROM_BYTES(ALIGNUP(BYTES_FROM_MEDIATIME(pin->rtStop), pin->cbAlign));
 735
 736         hr = IMediaSample_SetTime(sample, &rtSampleStart, &rtSampleStop);
 737
 738         pin->rtCurrent = pin->rtNext;
 739         pin->rtNext = rtSampleStop;
 740
 741         IMediaSample_SetPreroll(sample, FALSE);
 742         IMediaSample_SetDiscontinuity(sample, This->seek);
 743         IMediaSample_SetSyncPoint(sample, 1);
 744         This->seek = 0;
 745
 746         hr = IAsyncReader_Request(pin->pReader, sample, 0);
 747     }
 748     if (FAILED(hr))
 749         ERR("Horsemen of the apocalypse came to bring error 0x%08x\n", hr);
 750
 751     return hr;
 752 }
 753
 754 static const IBaseFilterVtbl MPEGSplitter_Vtbl =
 755 {
 756     Parser_QueryInterface,
 757     Parser_AddRef,
 758     Parser_Release,
 759     Parser_GetClassID,
 760     Parser_Stop,
 761     Parser_Pause,
 762     Parser_Run,
 763     Parser_GetState,
 764     Parser_SetSyncSource,
 765     Parser_GetSyncSource,
 766     Parser_EnumPins,
 767     Parser_FindPin,
 768     Parser_QueryFilterInfo,
 769     Parser_JoinFilterGraph,
 770     Parser_QueryVendorInfo
 771 };
 772
 773 HRESULT MPEGSplitter_create(IUnknown * pUnkOuter, LPVOID * ppv)
 774 {
 775     MPEGSplitterImpl *This;
 776     HRESULT hr = E_FAIL;
 777
 778     TRACE("(%p, %p)\n", pUnkOuter, ppv);
 779
 780     *ppv = NULL;
 781
 782     if (pUnkOuter)
 783         return CLASS_E_NOAGGREGATION;
 784
 785     This = CoTaskMemAlloc(sizeof(MPEGSplitterImpl));
 786     if (!This)
 787         return E_OUTOFMEMORY;
 788
 789     ZeroMemory(This, sizeof(MPEGSplitterImpl));
 790     This->seektable = CoTaskMemAlloc(sizeof(struct seek_entry) * 64);
 791     if (!This->seektable)
 792     {
 793         CoTaskMemFree(This);
 794         return E_OUTOFMEMORY;
 795     }
 796     This->seek_entries = 64;
 797
 798     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);
 799     if (FAILED(hr))
 800     {
 801         CoTaskMemFree(This);
 802         return hr;
 803     }
 804     This->seek = 1;
 805
 806     /* Note: This memory is managed by the parser filter once created */
 807     *ppv = This;
 808
 809     return hr;
 810 }