00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00022
00023
00024
00025
00026
00027
00028
00029
00030
00031 #include "stdafx.h"
00032 #include "oggmuxinputpin.h"
00033
00034 OggMuxInputPin::OggMuxInputPin(OggMuxFilter* inParentFilter, CCritSec* inFilterLock, HRESULT* inHR, OggMuxStream* inMuxStream)
00035 : CBaseInputPin(NAME("OggMuxInputPin"), inParentFilter, inFilterLock, inHR, L"Ogg Packet In")
00036 , mParentFilter(inParentFilter)
00037 , mMuxStream(inMuxStream)
00038 , mNeedsFLACHeaderTweak(false)
00039 , mNeedsFLACHeaderCount(false)
00040
00041 {
00042
00043
00044 OggPaginatorSettings* locSettings = new OggPaginatorSettings;
00045 locSettings->mMinPageSize = 4096;
00046 locSettings->mMaxPageSize = 8192;
00047
00048 LARGE_INTEGER locTicks;
00049 QueryPerformanceCounter(&locTicks);
00050 srand((unsigned int)locTicks.LowPart);
00051 locSettings->mSerialNo = ((unsigned long)(rand() + 1)) * ((unsigned long)(rand() + 1));
00052
00053
00054
00055
00056
00057
00058
00059
00060
00061 mPaginator.setParameters(locSettings);
00062 mPaginator.setPageCallback(mMuxStream);
00063
00064
00065 }
00066
00067 OggMuxInputPin::~OggMuxInputPin(void)
00068 {
00069
00070 }
00071
00072 STDMETHODIMP OggMuxInputPin::NonDelegatingQueryInterface(REFIID riid, void **ppv)
00073 {
00074 if (riid == IID_IMediaSeeking) {
00075 *ppv = (IMediaSeeking*)this;
00076 ((IUnknown*)*ppv)->AddRef();
00077 return NOERROR;
00078 }
00079
00080 return CBaseInputPin::NonDelegatingQueryInterface(riid, ppv);
00081 }
00082
00083
00084 HRESULT OggMuxInputPin::SetMediaType(const CMediaType* inMediaType) {
00085
00086
00087 if ((inMediaType->majortype == MEDIATYPE_Video) && (inMediaType->subtype == MEDIASUBTYPE_Theora)) {
00088
00089
00090 sTheoraFormatBlock* locTheora = (sTheoraFormatBlock*)inMediaType->pbFormat;
00091
00092
00093 mMuxStream->setConversionParams(locTheora->frameRateNumerator, locTheora->frameRateDenominator, 10000000, locTheora->maxKeyframeInterval);
00094 mMuxStream->setNumHeaders(3);
00095 mPaginator.setNumHeaders(3);
00096 } else if (inMediaType->majortype == MEDIATYPE_Audio) {
00097 if (inMediaType->subtype == MEDIASUBTYPE_Vorbis) {
00098
00099 sVorbisFormatBlock* locVorbis = (sVorbisFormatBlock*)inMediaType->pbFormat;
00100
00101 mMuxStream->setConversionParams(locVorbis->samplesPerSec, 1, 10000000);
00102 mMuxStream->setNumHeaders(3);
00103 mPaginator.setNumHeaders(3);
00104
00105 } else if (inMediaType->subtype == MEDIASUBTYPE_Speex) {
00106
00107 sSpeexFormatBlock* locSpeex = (sSpeexFormatBlock*)inMediaType->pbFormat;
00108 mMuxStream->setConversionParams(locSpeex->samplesPerSec, 1, 10000000);
00109 mMuxStream->setNumHeaders(2);
00110 mPaginator.setNumHeaders(2);
00111 } else if (inMediaType->subtype == MEDIASUBTYPE_OggFLAC_1_0) {
00112
00113 sFLACFormatBlock* locFLAC = (sFLACFormatBlock*)inMediaType->pbFormat;
00114 mMuxStream->setConversionParams(locFLAC->samplesPerSec, 1, 10000000);
00115
00116
00117 mNeedsFLACHeaderCount = true;
00118 } else if (inMediaType->subtype == MEDIASUBTYPE_FLAC) {
00119
00120
00121 sFLACFormatBlock* locFLAC = (sFLACFormatBlock*)inMediaType->pbFormat;
00122 mMuxStream->setConversionParams(locFLAC->samplesPerSec, 1, 10000000);
00123
00124 mNeedsFLACHeaderTweak = true;
00125 }
00126
00127
00128 } else if (inMediaType->majortype == MEDIATYPE_Text) {
00129 if (inMediaType->subtype == MEDIASUBTYPE_CMML) {
00130 sCMMLFormatBlock* locCMML = (sCMMLFormatBlock*)inMediaType->pbFormat;
00131 mMuxStream->setConversionParams(locCMML->granuleNumerator,locCMML->granuleDenominator, 10000000);
00132 mMuxStream->setNumHeaders(1);
00133 mPaginator.setNumHeaders(1);
00134
00135 }
00136
00137 }
00138 return S_OK;
00139 }
00140
00141 HRESULT OggMuxInputPin::GetMediaType(int inPosition, CMediaType* outMediaType) {
00142 switch(inPosition) {
00143 case 0:
00144 outMediaType->majortype = MEDIATYPE_Video;
00145 outMediaType->subtype = MEDIASUBTYPE_Theora;
00146 return S_OK;
00147 case 1:
00148 outMediaType->majortype = MEDIATYPE_Audio;
00149 outMediaType->subtype = MEDIASUBTYPE_Vorbis;
00150 return S_OK;
00151 case 2:
00152 outMediaType->majortype = MEDIATYPE_Audio;
00153 outMediaType->subtype = MEDIASUBTYPE_Speex;
00154 return S_OK;
00155 case 3:
00156 outMediaType->majortype = MEDIATYPE_Audio;
00157 outMediaType->subtype = MEDIASUBTYPE_OggFLAC_1_0;
00158 return S_OK;
00159 case 4:
00160 outMediaType->majortype = MEDIATYPE_Audio;
00161 outMediaType->subtype = MEDIASUBTYPE_FLAC;
00162 return S_OK;
00163
00164 case 5:
00165 outMediaType->majortype = MEDIATYPE_Text;
00166 outMediaType->subtype = MEDIASUBTYPE_CMML;
00167 return S_OK;
00168
00169
00170 default:
00171 return VFW_S_NO_MORE_ITEMS;
00172
00173 }
00174 }
00175 HRESULT OggMuxInputPin::CheckMediaType(const CMediaType* inMediaType) {
00176 if ( (inMediaType->majortype == MEDIATYPE_Video
00177 && inMediaType->subtype == MEDIASUBTYPE_Theora
00178 && inMediaType->formattype == FORMAT_Theora)
00179 ||
00180 (inMediaType->majortype == MEDIATYPE_Audio
00181 && inMediaType->subtype == MEDIASUBTYPE_Vorbis
00182 && inMediaType->formattype == FORMAT_Vorbis)
00183 ||
00184 (inMediaType->majortype == MEDIATYPE_Audio
00185 && inMediaType->subtype == MEDIASUBTYPE_Speex
00186 && inMediaType->formattype == FORMAT_Speex)
00187 ||
00188 (inMediaType->majortype == MEDIATYPE_Audio
00189 && inMediaType->subtype == MEDIASUBTYPE_OggFLAC_1_0
00190 && inMediaType->formattype == FORMAT_FLAC)
00191 ||
00192 (inMediaType->majortype == MEDIATYPE_Audio
00193 && inMediaType->subtype == MEDIASUBTYPE_FLAC
00194 && inMediaType->formattype == FORMAT_FLAC)
00195 ||
00196 (inMediaType->majortype == MEDIATYPE_Text
00197 && inMediaType->subtype == MEDIASUBTYPE_CMML
00198 && inMediaType->formattype == FORMAT_CMML)
00199
00200 ) {
00201 return S_OK;
00202 } else {
00203 return E_FAIL;
00204 }
00205 }
00206
00207 STDMETHODIMP OggMuxInputPin::Receive(IMediaSample* inSample) {
00208 CAutoLock locLock(mParentFilter->mStreamLock);
00209 LONGLONG locStart = 0;
00210 LONGLONG locEnd = 0;
00211 BYTE* locSampleBuff;
00212 inSample->GetPointer(&locSampleBuff);
00213 HRESULT locHR = inSample->GetTime(&locStart, &locEnd);
00214
00215
00216 long locBuffSize = inSample->GetActualDataLength();
00217 unsigned char* locBuff = new unsigned char[locBuffSize];
00218 memcpy((void*)locBuff, (const void*)locSampleBuff, inSample->GetActualDataLength());
00219 StampedOggPacket* locPacket = new StampedOggPacket(locBuff, inSample->GetActualDataLength(), false, false, locStart, locEnd, StampedOggPacket::OGG_END_ONLY);
00220
00221 if (mNeedsFLACHeaderCount) {
00222 mNeedsFLACHeaderCount = false;
00223
00224 mPaginator.setNumHeaders( (locPacket->packetData()[8]) + 1 );
00225 }
00226 if ((mNeedsFLACHeaderTweak)) {
00227
00228
00229
00230
00231
00232
00233
00234
00235
00236
00237
00238
00239
00240
00241
00242
00243
00244
00245
00246
00247
00248
00249
00250 FLACMetadataSplitter* locFLACSplitter = new FLACMetadataSplitter;
00251
00252
00253 locFLACSplitter->loadMetadata(locPacket->clone());
00254
00255
00256
00257 for (unsigned long i = 0; i < locFLACSplitter->numHeaders(); i++) {
00258
00259
00260 if (i == 0) {
00261
00262 StampedOggPacket* locHeadPack = locFLACSplitter->getHeader(i);
00263 mPaginator.setNumHeaders((locHeadPack->packetData()[8]) + 1);
00264 delete locHeadPack;
00265 }
00266 mPaginator.acceptStampedOggPacket(locFLACSplitter->getHeader(i));
00267
00268 }
00269 mNeedsFLACHeaderTweak = false;
00270
00271 delete locFLACSplitter;
00272
00273
00274 } else {
00275
00276
00277
00278 mPaginator.acceptStampedOggPacket(locPacket);
00279 }
00280
00281 return S_OK;
00282
00283
00284
00285 }
00286
00287 HRESULT OggMuxInputPin::CompleteConnect(IPin* inReceivePin) {
00288
00289
00290 IMediaSeeking* locSeeker = NULL;
00291 inReceivePin->QueryInterface(IID_IMediaSeeking, (void**)&locSeeker);
00292 SetDelegate(locSeeker);
00293
00294
00295 mMuxStream->setIsActive(true);
00296 return mParentFilter->addAnotherPin();
00297
00298
00299 }
00300
00301 HRESULT OggMuxInputPin::BreakConnect()
00302 {
00303 CAutoLock locLock(m_pLock);
00304
00305 ReleaseDelegate();
00306 return CBaseInputPin::BreakConnect();
00307 }
00308
00309
00310
00311
00312 STDMETHODIMP OggMuxInputPin::EndOfStream(void) {
00313 CAutoLock locLock(mParentFilter->mStreamLock);
00314 mPaginator.finishStream();
00315 mMuxStream->setIsEOS(true);
00316
00317
00318
00319 return S_OK;
00320
00321 }