git.oblomov.eu Git - wine/blob - dlls/quartz/mpgparse.c

   1 /*
   2  * Implements MPEG-1 / MPEG-2 Parser(Splitter).
   3  *
   4  *      FIXME - no splitter implementation.
   5  *      FIXME - no packet support (implemented payload only)
   6  *      FIXME - no seeking
   7  *
   8  * Copyright (C) 2002 Hidenori TAKESHIMA <hidenori@a2.ctktv.ne.jp>
   9  *
  10  * This library is free software; you can redistribute it and/or
  11  * modify it under the terms of the GNU Lesser General Public
  12  * License as published by the Free Software Foundation; either
  13  * version 2.1 of the License, or (at your option) any later version.
  14  *
  15  * This library is distributed in the hope that it will be useful,
  16  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  18  * Lesser General Public License for more details.
  19  *
  20  * You should have received a copy of the GNU Lesser General Public
  21  * License along with this library; if not, write to the Free Software
  22  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  23  */
  24
  25 #include "config.h"
  26
  27 #include "windef.h"
  28 #include "winbase.h"
  29 #include "wingdi.h"
  30 #include "winuser.h"
  31 #include "mmsystem.h"
  32 #include "vfw.h"
  33 #include "winerror.h"
  34 #include "strmif.h"
  35 #include "control.h"
  36 #include "vfwmsgs.h"
  37 #include "amvideo.h"
  38 #include "mmreg.h"
  39 #include "uuids.h"
  40 #include "dvdmedia.h"
  41
  42 #include "wine/debug.h"
  43 WINE_DEFAULT_DEBUG_CHANNEL(quartz);
  44
  45 #include "quartz_private.h"
  46 #include "parser.h"
  47 #include "mtype.h"
  48
  49 static const WCHAR QUARTZ_MPEG1Parser_Name[] =
  50 { 'M','P','E','G','-','I',' ','S','p','l','i','t','t','e','r',0 };
  51 static const WCHAR QUARTZ_MPEG2Parser_Name[] =
  52 { 'M','P','E','G','-','2',' ','S','p','l','i','t','t','e','r',0 };
  53 static const WCHAR QUARTZ_MPGParserInPin_Name[] =
  54 { 'I','n',0 };
  55 static const WCHAR QUARTZ_MPGParserOutPin_VideoPinName[] =
  56 { 'V','i','d','e','o',0 };
  57 static const WCHAR QUARTZ_MPGParserOutPin_AudioPinName[] =
  58 { 'A','u','d','i','o',0 };
  59         /* FIXME */
  60 static const WCHAR QUARTZ_MPGParserOutPin_UnknownTypePinName[] =
  61 { 'O','u','t',0 };
  62
  63
  64
  65 static DWORD bitratesl1[16] =
  66         {  0, 32, 64, 96,128,160,192,224,256,288,320,352,384,416,448,  0};
  67 static DWORD bitratesl2[16] =
  68         {  0, 32, 48, 56, 64, 80, 96,112,128,160,192,224,256,320,384,  0};
  69 static DWORD bitratesl3[16] =
  70         {  0, 32, 40, 48, 56, 64, 80, 96,112,128,160,192,224,256,320,  0};
  71
  72
  73 /****************************************************************************
  74  *
  75  *      CMPGParseImpl
  76  */
  77
  78
  79 typedef struct CMPGParseImpl CMPGParseImpl;
  80 typedef struct CMPGParsePayload CMPGParsePayload;
  81
  82 enum MPGPayloadType
  83 {
  84         MPGPayload_Video = 0xe0,
  85         MPGPayload_Audio = 0xc0,
  86 };
  87
  88 struct CMPGParseImpl
  89 {
  90         LONGLONG        llPosNext;
  91         BOOL    bRawPayload;
  92         DWORD   dwPayloadBlockSizeMax;
  93         DWORD   cPayloads;
  94         CMPGParsePayload*       pPayloads;
  95 };
  96
  97 struct CMPGParsePayload
  98 {
  99         enum MPGPayloadType     payloadtype;
 100         BOOL    bDataDiscontinuity;
 101 };
 102
 103
 104 static HRESULT CMPGParseImpl_SyncReadPayload(
 105         CParserImpl* pImpl, CMPGParseImpl* This,
 106         enum MPGPayloadType payloadtype,
 107         LONGLONG llPosStart, LONG lLength, BYTE* pbBuf )
 108 {
 109         if ( This == NULL || This->pPayloads == NULL )
 110                 return E_UNEXPECTED;
 111
 112         if ( This->bRawPayload )
 113         {
 114                 if ( payloadtype != This->pPayloads[0].payloadtype )
 115                         return E_UNEXPECTED;
 116                 return IAsyncReader_SyncRead( pImpl->m_pReader, llPosStart, lLength, pbBuf );
 117         }
 118         else
 119         {
 120                 FIXME( "not implemented\n" );
 121         }
 122
 123         return E_NOTIMPL;
 124 }
 125
 126
 127
 128 static HRESULT CMPGParseImpl_InitParser( CParserImpl* pImpl, ULONG* pcStreams )
 129 {
 130         CMPGParseImpl*  This = NULL;
 131         HRESULT hr;
 132         BYTE    hdrbuf[8];
 133
 134         TRACE("(%p,%p)\n",pImpl,pcStreams);
 135
 136         This = (CMPGParseImpl*)QUARTZ_AllocMem( sizeof(CMPGParseImpl) );
 137         if ( This == NULL )
 138                 return E_OUTOFMEMORY;
 139         pImpl->m_pUserData = This;
 140         ZeroMemory( This, sizeof(CMPGParseImpl) );
 141
 142         hr = IAsyncReader_SyncRead( pImpl->m_pReader, 0, 8, hdrbuf );
 143         if ( FAILED(hr) )
 144                 return hr;
 145         if ( hr != S_OK )
 146                 return E_FAIL;
 147
 148         if ( hdrbuf[0] == 0x00 && hdrbuf[1] == 0x00 &&
 149                  hdrbuf[2] == 0x01 && hdrbuf[3] == 0xba )
 150         {
 151                 This->bRawPayload = FALSE;
 152                 This->dwPayloadBlockSizeMax = 0;
 153
 154                 FIXME( "no mpeg/system support\n" );
 155                 return E_FAIL;
 156         }
 157         else
 158         if ( hdrbuf[0] == 0x00 && hdrbuf[1] == 0x00 &&
 159                  hdrbuf[2] == 0x01 && hdrbuf[3] == 0xb3 )
 160         {
 161                 TRACE( "mpeg/video payload\n" );
 162                 This->llPosNext = 0;
 163                 This->bRawPayload = TRUE;
 164                 This->dwPayloadBlockSizeMax = 0x4000;
 165                 This->cPayloads = 1;
 166                 This->pPayloads = (CMPGParsePayload*)QUARTZ_AllocMem( sizeof(CMPGParsePayload) );
 167                 if ( This->pPayloads == NULL )
 168                         return E_OUTOFMEMORY;
 169                 *pcStreams = 1;
 170                 This->pPayloads[0].payloadtype = MPGPayload_Video;
 171                 This->pPayloads[0].bDataDiscontinuity = TRUE;
 172         }
 173         else
 174         if ( hdrbuf[0] == 0xff && (hdrbuf[1]&0xf0) == 0xf0 )
 175         {
 176                 TRACE( "mpeg/audio payload\n" );
 177                 This->llPosNext = 0;
 178                 This->bRawPayload = TRUE;
 179                 This->dwPayloadBlockSizeMax = 0;
 180                 This->cPayloads = 1;
 181                 This->pPayloads = (CMPGParsePayload*)QUARTZ_AllocMem( sizeof(CMPGParsePayload) );
 182                 if ( This->pPayloads == NULL )
 183                         return E_OUTOFMEMORY;
 184                 *pcStreams = 1;
 185                 This->pPayloads[0].payloadtype = MPGPayload_Audio;
 186                 This->pPayloads[0].bDataDiscontinuity = TRUE;
 187         }
 188         else
 189         {
 190                 return E_FAIL;
 191         }
 192
 193         return S_OK;
 194 }
 195
 196 static HRESULT CMPGParseImpl_UninitParser( CParserImpl* pImpl )
 197 {
 198         CMPGParseImpl*  This = (CMPGParseImpl*)pImpl->m_pUserData;
 199         ULONG   nIndex;
 200
 201         TRACE("(%p)\n",This);
 202
 203         if ( This == NULL )
 204                 return NOERROR;
 205
 206         /* destruct */
 207
 208         if ( This->pPayloads != NULL )
 209         {
 210                 for ( nIndex = 0; nIndex < This->cPayloads; nIndex++ )
 211                 {
 212                         /* release this stream */
 213
 214
 215                 }
 216                 QUARTZ_FreeMem( This->pPayloads );
 217                 This->pPayloads = NULL;
 218         }
 219
 220         QUARTZ_FreeMem( This );
 221         pImpl->m_pUserData = NULL;
 222
 223         return NOERROR;
 224 }
 225
 226 static LPCWSTR CMPGParseImpl_GetOutPinName( CParserImpl* pImpl, ULONG nStreamIndex )
 227 {
 228         CMPGParseImpl*  This = (CMPGParseImpl*)pImpl->m_pUserData;
 229
 230         TRACE("(%p,%lu)\n",This,nStreamIndex);
 231
 232         if ( This == NULL || nStreamIndex >= This->cPayloads )
 233                 return NULL;
 234
 235         switch ( This->pPayloads[nStreamIndex].payloadtype )
 236         {
 237         case MPGPayload_Video:
 238                 return QUARTZ_MPGParserOutPin_VideoPinName;
 239         case MPGPayload_Audio:
 240                 return QUARTZ_MPGParserOutPin_AudioPinName;
 241         default:
 242                 FIXME("mpeg - unknown stream type %02x\n",This->pPayloads[nStreamIndex].payloadtype);
 243         }
 244
 245         /* FIXME */
 246         return QUARTZ_MPGParserOutPin_UnknownTypePinName;
 247 }
 248
 249 static HRESULT CMPGParseImpl_GetStreamType( CParserImpl* pImpl, ULONG nStreamIndex, AM_MEDIA_TYPE* pmt )
 250 {
 251         CMPGParseImpl*  This = (CMPGParseImpl*)pImpl->m_pUserData;
 252         HRESULT hr;
 253         BYTE    hdrbuf[140+10];
 254         UINT    seqhdrlen;
 255         MPEG1VIDEOINFO* pmpg1vi;
 256         MPEG2VIDEOINFO* pmpg2vi;
 257         MPEG1WAVEFORMAT*        pmpg1wav;
 258         enum MPGPayloadType     payloadtype;
 259         DWORD   dwPayloadBlockSize;
 260
 261         TRACE("(%p,%lu,%p)\n",This,nStreamIndex,pmt);
 262
 263         if ( This == NULL )
 264                 return E_UNEXPECTED;
 265         if ( nStreamIndex >= This->cPayloads )
 266                 return E_INVALIDARG;
 267
 268         ZeroMemory( pmt, sizeof(AM_MEDIA_TYPE) );
 269
 270         payloadtype = This->pPayloads[nStreamIndex].payloadtype;
 271         switch ( payloadtype )
 272         {
 273         case MPGPayload_Video:
 274                 hr = CMPGParseImpl_SyncReadPayload(
 275                         pImpl, This, payloadtype, 0, 140+10, hdrbuf );
 276                 if ( FAILED(hr) )
 277                         return hr;
 278                 if ( hr != S_OK )
 279                         return E_FAIL;
 280
 281                 memcpy( &pmt->majortype, &MEDIATYPE_Video, sizeof(GUID) );
 282                 seqhdrlen = 12;
 283                 if ( hdrbuf[seqhdrlen-1] & 0x2 )
 284                         seqhdrlen += 64;
 285                 if ( hdrbuf[seqhdrlen-1] & 0x1 )
 286                         seqhdrlen += 64;
 287                 if ( hdrbuf[seqhdrlen  ] == 0x00 && hdrbuf[seqhdrlen+1] == 0x00 &&
 288                          hdrbuf[seqhdrlen+2] == 0x01 && hdrbuf[seqhdrlen+3] == 0xb5 )
 289                 {
 290                         /* video MPEG-2 */
 291                         FIXME("video MPEG-2\n");
 292                         if ( (hdrbuf[seqhdrlen+4]&0xf0) != 0x1 )
 293                                 return E_FAIL;
 294                         memcpy( &pmt->subtype, &MEDIASUBTYPE_MPEG2_VIDEO, sizeof(GUID) );
 295                         memcpy( &pmt->formattype, &FORMAT_MPEG2_VIDEO, sizeof(GUID) );
 296                         pmt->bFixedSizeSamples = 0;
 297                         pmt->bTemporalCompression = 1;
 298                         pmt->lSampleSize = 0;
 299                         pmt->cbFormat = sizeof(MPEG2VIDEOINFO);
 300                         pmt->pbFormat = (BYTE*)CoTaskMemAlloc( sizeof(MPEG2VIDEOINFO) );
 301                         if ( pmt->pbFormat == NULL )
 302                                 return E_OUTOFMEMORY;
 303                         ZeroMemory( pmt->pbFormat, sizeof(MPEG2VIDEOINFO) );
 304                         pmpg2vi = (MPEG2VIDEOINFO*)pmt->pbFormat;
 305                         pmpg2vi->hdr.bmiHeader.biSize = sizeof(BITMAPINFOHEADER);
 306                         pmpg2vi->hdr.bmiHeader.biWidth = ((UINT)hdrbuf[4] << 4) | ((UINT)hdrbuf[5] >> 4); /* FIXME! */
 307                         pmpg2vi->hdr.bmiHeader.biHeight = (((UINT)hdrbuf[5] & 0xf) << 8) | ((UINT)hdrbuf[6]); /* FIXME! */
 308                         pmpg2vi->hdr.dwInterlaceFlags = AMINTERLACE_FieldPatBothRegular; /* FIXME? */
 309                         pmpg2vi->hdr.dwCopyProtectFlags = AMCOPYPROTECT_RestrictDuplication; /* FIXME? */
 310                         pmpg2vi->hdr.dwPictAspectRatioX = 1; /* FIXME? */
 311                         pmpg2vi->hdr.dwPictAspectRatioY = 1; /* FIXME? */
 312                         pmpg2vi->dwStartTimeCode = 0;
 313                         pmpg2vi->cbSequenceHeader = seqhdrlen + 10;
 314                         switch ( hdrbuf[seqhdrlen+4] & 0xf )
 315                         {
 316                         case 5: pmpg2vi->dwProfile = AM_MPEG2Profile_Simple; break;
 317                         case 4: pmpg2vi->dwProfile = AM_MPEG2Profile_Main; break;
 318                         case 3: pmpg2vi->dwProfile = AM_MPEG2Profile_SNRScalable; break;
 319                         case 2: pmpg2vi->dwProfile = AM_MPEG2Profile_SpatiallyScalable; break;
 320                         case 1: pmpg2vi->dwProfile = AM_MPEG2Profile_High; break;
 321                         default: return E_FAIL;
 322                         }
 323                         switch ( hdrbuf[seqhdrlen+5] >> 4 )
 324                         {
 325                         case 10: pmpg2vi->dwLevel = AM_MPEG2Level_Low; break;
 326                         case  8: pmpg2vi->dwLevel = AM_MPEG2Level_Main; break;
 327                         case  6: pmpg2vi->dwLevel = AM_MPEG2Level_High1440; break;
 328                         case  4: pmpg2vi->dwLevel = AM_MPEG2Level_High; break;
 329                         default: return E_FAIL;
 330                         }
 331                         pmpg2vi->dwFlags = 0; /* FIXME? */
 332                         memcpy( pmpg2vi->dwSequenceHeader, hdrbuf, seqhdrlen + 10 );
 333
 334                         return S_OK;
 335                 }
 336                 else
 337                 {
 338                         /* MPEG-1 */
 339                         memcpy( &pmt->subtype, &MEDIASUBTYPE_MPEG1Payload, sizeof(GUID) );
 340                         memcpy( &pmt->formattype, &FORMAT_MPEGVideo, sizeof(GUID) );
 341                         pmt->bFixedSizeSamples = 0;
 342                         pmt->bTemporalCompression = 1;
 343                         pmt->lSampleSize = 0;
 344                         pmt->cbFormat = sizeof(MPEG1VIDEOINFO);
 345                         pmt->pbFormat = (BYTE*)CoTaskMemAlloc( sizeof(MPEG1VIDEOINFO) );
 346                         if ( pmt->pbFormat == NULL )
 347                                 return E_OUTOFMEMORY;
 348                         ZeroMemory( pmt->pbFormat, sizeof(MPEG1VIDEOINFO) );
 349                         pmpg1vi = (MPEG1VIDEOINFO*)pmt->pbFormat;
 350                         pmpg1vi->hdr.bmiHeader.biSize = sizeof(BITMAPINFOHEADER);
 351                         pmpg1vi->hdr.bmiHeader.biWidth = ((UINT)hdrbuf[4] << 4) | ((UINT)hdrbuf[5] >> 4);
 352                         pmpg1vi->hdr.bmiHeader.biHeight = (((UINT)hdrbuf[5] & 0xf) << 8) | ((UINT)hdrbuf[6]);
 353                         pmpg1vi->hdr.bmiHeader.biPlanes = 1;
 354                         pmpg1vi->dwStartTimeCode = 0;
 355                         pmpg1vi->cbSequenceHeader = seqhdrlen;
 356                         memcpy( pmpg1vi->bSequenceHeader, hdrbuf, seqhdrlen );
 357                 }
 358
 359                 return S_OK;
 360         case MPGPayload_Audio:
 361                 hr = CMPGParseImpl_SyncReadPayload(
 362                         pImpl, This, payloadtype, 0, 4, hdrbuf );
 363                 if ( FAILED(hr) )
 364                         return hr;
 365                 if ( hr != S_OK )
 366                         return E_FAIL;
 367
 368                 memcpy( &pmt->majortype, &MEDIATYPE_Audio, sizeof(GUID) );
 369                 memcpy( &pmt->formattype, &FORMAT_WaveFormatEx, sizeof(GUID) );
 370
 371                 if ( !( hdrbuf[1] & 0x8 ) )
 372                 {
 373                         /* FIXME!!! */
 374                         FIXME("audio not MPEG-1\n");
 375                         return E_FAIL;
 376                 }
 377                 else
 378                 {
 379                         /* MPEG-1 */
 380                         memcpy( &pmt->subtype, &MEDIASUBTYPE_MPEG1AudioPayload, sizeof(GUID) );
 381                         pmt->bFixedSizeSamples = 0;
 382                         pmt->bTemporalCompression = 1;
 383                         pmt->lSampleSize = 0;
 384                         pmt->cbFormat = sizeof(MPEG1WAVEFORMAT);
 385                         pmt->pbFormat = (BYTE*)CoTaskMemAlloc( sizeof(MPEG1WAVEFORMAT) );
 386                         if ( pmt->pbFormat == NULL )
 387                                 return E_OUTOFMEMORY;
 388                         ZeroMemory( pmt->pbFormat, sizeof(MPEG1WAVEFORMAT) );
 389                         pmpg1wav = (MPEG1WAVEFORMAT*)pmt->pbFormat;
 390                         switch ( hdrbuf[1] & 0x6 )
 391                         {
 392                         case 0x6: pmpg1wav->fwHeadLayer = ACM_MPEG_LAYER1;
 393                         case 0x4: pmpg1wav->fwHeadLayer = ACM_MPEG_LAYER2;
 394                         case 0x2: pmpg1wav->fwHeadLayer = ACM_MPEG_LAYER3;
 395                         default: return E_FAIL;
 396                         }
 397
 398                         switch ( pmpg1wav->fwHeadLayer )
 399                         {
 400                         case ACM_MPEG_LAYER1:
 401                                 pmpg1wav->dwHeadBitrate = bitratesl1[hdrbuf[2]>>4]*1000;
 402                                 break;
 403                         case ACM_MPEG_LAYER2:
 404                                 pmpg1wav->dwHeadBitrate = bitratesl2[hdrbuf[2]>>4]*1000;
 405                                 break;
 406                         case ACM_MPEG_LAYER3:
 407                                 pmpg1wav->dwHeadBitrate = bitratesl3[hdrbuf[2]>>4]*1000;
 408                                 break;
 409                         }
 410                         if ( pmpg1wav->dwHeadBitrate == 0 )
 411                                 return E_FAIL;
 412
 413                         switch ( hdrbuf[3] & 0xc0 )
 414                         {
 415                         case 0x00: pmpg1wav->fwHeadMode = ACM_MPEG_STEREO;
 416                         case 0x40: pmpg1wav->fwHeadMode = ACM_MPEG_JOINTSTEREO;
 417                         case 0x80: pmpg1wav->fwHeadMode = ACM_MPEG_DUALCHANNEL;
 418                         case 0xc0: pmpg1wav->fwHeadMode = ACM_MPEG_SINGLECHANNEL;
 419                         }
 420
 421                         pmpg1wav->fwHeadModeExt = (hdrbuf[3] & 0x30) >> 4; /* FIXME?? */
 422                         pmpg1wav->wHeadEmphasis = (hdrbuf[3] & 0x03); /* FIXME?? */
 423                         pmpg1wav->fwHeadFlags = ACM_MPEG_ID_MPEG1;
 424                         if ( hdrbuf[1] & 0x1 )
 425                                 pmpg1wav->fwHeadFlags |= ACM_MPEG_PROTECTIONBIT;
 426                         if ( hdrbuf[2] & 0x1 )
 427                                 pmpg1wav->fwHeadFlags |= ACM_MPEG_PRIVATEBIT;
 428                         if ( hdrbuf[3] & 0x8 )
 429                                 pmpg1wav->fwHeadFlags |= ACM_MPEG_COPYRIGHT;
 430                         if ( hdrbuf[3] & 0x4 )
 431                                 pmpg1wav->fwHeadFlags |= ACM_MPEG_ORIGINALHOME;
 432                         pmpg1wav->dwPTSLow = 0;
 433                         pmpg1wav->dwPTSHigh = 0;
 434
 435                         pmpg1wav->wfx.wFormatTag = WAVE_FORMAT_MPEG;
 436                         pmpg1wav->wfx.nChannels = (pmpg1wav->fwHeadMode != ACM_MPEG_SINGLECHANNEL) ? 2 : 1;
 437                         switch ( hdrbuf[2] & 0x0c )
 438                         {
 439                         case 0x00: pmpg1wav->wfx.nSamplesPerSec = 44100;
 440                         case 0x01: pmpg1wav->wfx.nSamplesPerSec = 48000;
 441                         case 0x02: pmpg1wav->wfx.nSamplesPerSec = 32000;
 442                         default: return E_FAIL;
 443                         }
 444                         pmpg1wav->wfx.nAvgBytesPerSec = pmpg1wav->dwHeadBitrate >> 3;
 445                         switch ( pmpg1wav->fwHeadLayer )
 446                         {
 447                         case ACM_MPEG_LAYER1:
 448                                 pmpg1wav->wfx.nBlockAlign = (384>>3) * pmpg1wav->dwHeadBitrate / pmpg1wav->wfx.nSamplesPerSec;
 449                                 break;
 450                         case ACM_MPEG_LAYER2:
 451                                 pmpg1wav->wfx.nBlockAlign = (1152>>3) * pmpg1wav->dwHeadBitrate / pmpg1wav->wfx.nSamplesPerSec;
 452                                 break;
 453                         case ACM_MPEG_LAYER3:
 454                                 pmpg1wav->wfx.nBlockAlign = 1;
 455                                 break;
 456                         }
 457                         pmpg1wav->wfx.wBitsPerSample = 0;
 458                         pmpg1wav->wfx.cbSize = sizeof(MPEG1WAVEFORMAT) - sizeof(WAVEFORMATEX);
 459                         if ( pmpg1wav->fwHeadLayer != ACM_MPEG_LAYER3 )
 460                         {
 461                                 pmt->bFixedSizeSamples = 1;
 462                                 pmt->lSampleSize = pmpg1wav->wfx.nBlockAlign;
 463                         }
 464                         dwPayloadBlockSize = (pmpg1wav->wfx.nAvgBytesPerSec + pmpg1wav->wfx.nBlockAlign - 1) / pmpg1wav->wfx.nBlockAlign;
 465                         if ( dwPayloadBlockSize > This->dwPayloadBlockSizeMax )
 466                                 This->dwPayloadBlockSizeMax = dwPayloadBlockSize;
 467                 }
 468
 469                 return S_OK;
 470         default:
 471                 FIXME("mpeg - unknown stream type %02x\n",This->pPayloads[nStreamIndex].payloadtype);
 472                 break;
 473         }
 474
 475         FIXME("stub\n");
 476         return E_NOTIMPL;
 477 }
 478
 479 static HRESULT CMPGParseImpl_CheckStreamType( CParserImpl* pImpl, ULONG nStreamIndex, const AM_MEDIA_TYPE* pmt )
 480 {
 481         CMPGParseImpl*  This = (CMPGParseImpl*)pImpl->m_pUserData;
 482         HRESULT hr;
 483         AM_MEDIA_TYPE   mt;
 484         MPEG1VIDEOINFO* pmpg1vi;
 485         MPEG1VIDEOINFO* pmpg1viCheck;
 486         MPEG2VIDEOINFO* pmpg2vi;
 487         MPEG2VIDEOINFO* pmpg2viCheck;
 488         WAVEFORMATEX*   pwfx;
 489         WAVEFORMATEX*   pwfxCheck;
 490         enum MPGPayloadType     payloadtype;
 491
 492         TRACE("(%p,%lu,%p)\n",This,nStreamIndex,pmt);
 493
 494         if ( This == NULL )
 495                 return E_UNEXPECTED;
 496         if ( nStreamIndex >= This->cPayloads )
 497                 return E_INVALIDARG;
 498
 499         hr = CMPGParseImpl_GetStreamType( pImpl, nStreamIndex, &mt );
 500         if ( FAILED(hr) )
 501                 return hr;
 502         if ( !IsEqualGUID( &pmt->majortype, &mt.majortype ) ||
 503                  !IsEqualGUID( &pmt->subtype, &mt.subtype ) ||
 504                  !IsEqualGUID( &pmt->formattype, &mt.formattype ) )
 505         {
 506                 hr = E_FAIL;
 507                 goto end;
 508         }
 509
 510         TRACE("check format\n");
 511         hr = S_OK;
 512
 513         payloadtype = This->pPayloads[nStreamIndex].payloadtype;
 514         switch ( payloadtype )
 515         {
 516         case MPGPayload_Video:
 517                 if ( IsEqualGUID( &mt.formattype, &FORMAT_MPEGVideo ) )
 518                 {
 519                         /* MPEG-1 Video */
 520                         if ( pmt->cbFormat != mt.cbFormat ||
 521                                  pmt->pbFormat == NULL )
 522                         {
 523                                 hr = E_FAIL;
 524                                 goto end;
 525                         }
 526                         pmpg1vi = (MPEG1VIDEOINFO*)mt.pbFormat;
 527                         pmpg1viCheck = (MPEG1VIDEOINFO*)pmt->pbFormat;
 528                         if ( memcmp( pmpg1vi, pmpg1viCheck, sizeof(MPEG1VIDEOINFO) ) != 0 )
 529                         {
 530                                 hr = E_FAIL;
 531                                 goto end;
 532                         }
 533                 }
 534                 else
 535                 if ( IsEqualGUID( &mt.formattype, &FORMAT_MPEG2_VIDEO ) )
 536                 {
 537                         /* MPEG-2 Video */
 538                         if ( pmt->cbFormat != mt.cbFormat ||
 539                                  pmt->pbFormat == NULL )
 540                         {
 541                                 hr = E_FAIL;
 542                                 goto end;
 543                         }
 544                         pmpg2vi = (MPEG2VIDEOINFO*)mt.pbFormat;
 545                         pmpg2viCheck = (MPEG2VIDEOINFO*)pmt->pbFormat;
 546                         if ( memcmp( pmpg2vi, pmpg2viCheck, sizeof(MPEG2VIDEOINFO) ) != 0 )
 547                         {
 548                                 hr = E_FAIL;
 549                                 goto end;
 550                         }
 551                 }
 552                 else
 553                 {
 554                         hr = E_FAIL;
 555                         goto end;
 556                 }
 557                 break;
 558         case MPGPayload_Audio:
 559                 if ( IsEqualGUID( &mt.formattype, &FORMAT_WaveFormatEx ) )
 560                 {
 561                         if ( mt.cbFormat != pmt->cbFormat ||
 562                                  pmt->pbFormat == NULL )
 563                         {
 564                                 hr = E_FAIL;
 565                                 goto end;
 566                         }
 567                         pwfx = (WAVEFORMATEX*)mt.pbFormat;
 568                         pwfxCheck = (WAVEFORMATEX*)pmt->pbFormat;
 569
 570                         if ( memcmp( pwfx, pwfxCheck, sizeof(WAVEFORMATEX) ) != 0 )
 571                         {
 572                                 hr = E_FAIL;
 573                                 goto end;
 574                         }
 575                 }
 576                 else
 577                 {
 578                         hr = E_FAIL;
 579                         goto end;
 580                 }
 581
 582                 break;
 583         default:
 584                 FIXME( "unsupported payload type\n" );
 585                 hr = E_FAIL;
 586                 goto end;
 587         }
 588
 589         hr = S_OK;
 590 end:
 591         QUARTZ_MediaType_Free( &mt );
 592
 593         TRACE("%08lx\n",hr);
 594
 595         return hr;
 596 }
 597
 598 static HRESULT CMPGParseImpl_GetAllocProp( CParserImpl* pImpl, ALLOCATOR_PROPERTIES* pReqProp )
 599 {
 600         CMPGParseImpl*  This = (CMPGParseImpl*)pImpl->m_pUserData;
 601
 602         TRACE("(%p,%p)\n",This,pReqProp);
 603
 604         if ( This == NULL )
 605                 return E_UNEXPECTED;
 606
 607         ZeroMemory( pReqProp, sizeof(ALLOCATOR_PROPERTIES) );
 608         pReqProp->cBuffers = This->cPayloads;
 609         pReqProp->cbBuffer = This->dwPayloadBlockSizeMax;
 610
 611         return S_OK;
 612 }
 613
 614 static HRESULT CMPGParseImpl_GetNextRequest( CParserImpl* pImpl, ULONG* pnStreamIndex, LONGLONG* pllStart, LONG* plLength, REFERENCE_TIME* prtStart, REFERENCE_TIME* prtStop, DWORD* pdwSampleFlags )
 615 {
 616         CMPGParseImpl*  This = (CMPGParseImpl*)pImpl->m_pUserData;
 617
 618         if ( This == NULL )
 619                 return E_UNEXPECTED;
 620         *pdwSampleFlags = AM_SAMPLE_SPLICEPOINT;
 621
 622         TRACE("(%p)\n",This);
 623
 624         if ( This->bRawPayload )
 625         {
 626                 if ( This->dwPayloadBlockSizeMax == 0 ||
 627                          This->cPayloads != 1 || This->pPayloads == NULL )
 628                         return E_UNEXPECTED;
 629                 *pnStreamIndex = 0;
 630                 *pllStart = This->llPosNext;
 631                 *plLength = This->dwPayloadBlockSizeMax;
 632                 *prtStart = 0;
 633                 *prtStop = 0;
 634                 if ( This->pPayloads[0].bDataDiscontinuity )
 635                 {
 636                         *pdwSampleFlags |= AM_SAMPLE_DATADISCONTINUITY;
 637                         This->pPayloads[0].bDataDiscontinuity = FALSE;
 638                 }
 639         }
 640         else
 641         {
 642                 FIXME("stub\n");
 643                 return E_NOTIMPL;
 644         }
 645
 646         return S_OK;
 647 }
 648
 649 static HRESULT CMPGParseImpl_ProcessSample( CParserImpl* pImpl, ULONG nStreamIndex, LONGLONG llStart, LONG lLength, IMediaSample* pSample )
 650 {
 651         CMPGParseImpl*  This = (CMPGParseImpl*)pImpl->m_pUserData;
 652         HRESULT hr;
 653
 654         TRACE("(%p,%lu,%ld,%ld,%p)\n",This,nStreamIndex,(long)llStart,lLength,pSample);
 655
 656         if ( This == NULL )
 657                 return E_UNEXPECTED;
 658
 659         if ( This->bRawPayload )
 660         {
 661                 hr = IMediaSample_SetTime(pSample,NULL,NULL);
 662                 if ( FAILED(hr) )
 663                         return hr;
 664         }
 665
 666         return NOERROR;
 667 }
 668
 669
 670 static const struct ParserHandlers CMPGParseImpl_Handlers =
 671 {
 672         CMPGParseImpl_InitParser,
 673         CMPGParseImpl_UninitParser,
 674         CMPGParseImpl_GetOutPinName,
 675         CMPGParseImpl_GetStreamType,
 676         CMPGParseImpl_CheckStreamType,
 677         CMPGParseImpl_GetAllocProp,
 678         CMPGParseImpl_GetNextRequest,
 679         CMPGParseImpl_ProcessSample,
 680
 681         /* for IQualityControl */
 682         NULL, /* pQualityNotify */
 683
 684         /* for seeking */
 685         NULL, /* pGetSeekingCaps */
 686         NULL, /* pIsTimeFormatSupported */
 687         NULL, /* pGetCurPos */
 688         NULL, /* pSetCurPos */
 689         NULL, /* pGetDuration */
 690         NULL, /* pGetStopPos */
 691         NULL, /* pSetStopPos */
 692         NULL, /* pGetPreroll */
 693 };
 694
 695 HRESULT QUARTZ_CreateMPEG1Splitter(IUnknown* punkOuter,void** ppobj)
 696 {
 697         return QUARTZ_CreateParser(
 698                 punkOuter,ppobj,
 699                 &CLSID_MPEG1Splitter,
 700                 QUARTZ_MPEG1Parser_Name,
 701                 QUARTZ_MPGParserInPin_Name,
 702                 &CMPGParseImpl_Handlers );
 703 }
 704
 705 HRESULT QUARTZ_CreateMPEG2Splitter(IUnknown* punkOuter,void** ppobj)
 706 {
 707         return QUARTZ_CreateParser(
 708                 punkOuter,ppobj,
 709                 &CLSID_MMSPLITTER,
 710                 QUARTZ_MPEG2Parser_Name,
 711                 QUARTZ_MPGParserInPin_Name,
 712                 &CMPGParseImpl_Handlers );
 713 }
 714
 715