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 "oggstream.h"
00033
00034 OggStream::OggStream(OggPage* inBOSPage, OggDemuxSourceFilter* inOwningFilter, bool inAllowSeek)
00035 : mCodecHeaders(NULL)
00036 , mOwningFilter(inOwningFilter)
00037 , mAllowSeek(inAllowSeek)
00038 , mSerialNo(0)
00039 , mSourcePin(NULL)
00040 , mNumHeadersNeeded(0)
00041 , mStreamReady(false)
00042
00043 , mFirstRun(true)
00044 , mSendExcess(true)
00045 , mLastEndGranulePos(0)
00046 , mLastStartGranulePos(0)
00047 , mStreamLock(NULL)
00048 , mAllowDispatch(false)
00049 {
00050
00051
00052 mSerialNo = inBOSPage->header()->StreamSerialNo();
00053
00054
00055 mStreamLock = new CCritSec;
00056
00057
00058
00059
00060
00061 mPacketiser.setPacketSink(this);
00062
00063 }
00064
00065 OggStream::~OggStream(void)
00066 {
00067
00068
00069 delete mSourcePin;
00070
00071
00072 delete mCodecHeaders;
00073
00074 delete mStreamLock;
00075
00076 }
00077
00078 bool OggStream::streamReady() {
00079 return mStreamReady;
00080
00081 }
00082
00083 unsigned long OggStream::serialNo() {
00084 return mSerialNo;
00085 }
00086 void OggStream::setAllowDispatch(bool inAllowDispatch) {
00087 mAllowDispatch = inAllowDispatch;
00088 }
00089 bool OggStream::acceptStampedOggPacket(StampedOggPacket* inPacket) {
00090 if (!mStreamReady) {
00091
00092
00093
00094 processHeaderPacket(inPacket);
00095
00096 if(mNumHeadersNeeded <= 0) {
00097
00098 AddPin();
00099
00100 }
00101
00102
00103 } else {
00104
00105 if (mAllowDispatch) {
00106 if (mFirstRun) {
00107 mFirstRun = false;
00108
00109
00110 deliverCodecHeaders();
00111 }
00112
00113 processDataPacket(inPacket);
00114 delete inPacket;
00115 } else {
00116 delete inPacket;
00117 return false;
00118 }
00119 }
00120
00121 return true;
00122 }
00123
00124
00125
00126 bool OggStream::processHeaderPacket(StampedOggPacket* inPacket) {
00127
00128
00129
00130
00131 if (inPacket != NULL) {
00132
00133
00134 mCodecHeaders->addPacket(inPacket);
00135 mNumHeadersNeeded--;
00136
00137 }
00138 return true;
00139 }
00140
00141 bool OggStream::processDataPacket(StampedOggPacket* inPacket) {
00142
00143
00144 if (inPacket != NULL) {
00145 dispatchPacket(inPacket);
00146
00147 inPacket = NULL;
00148 }
00149
00150 return true;
00151 }
00152
00153
00154 OggDemuxSourcePin* OggStream::getPin() {
00155 return mSourcePin;
00156 }
00157
00158
00159 CMediaType* OggStream::createMediaType(GUID inMajorType, GUID inSubType, GUID inFormatType, unsigned long inFormatSize, BYTE* inFormatBlock) {
00160 AM_MEDIA_TYPE locAMMediaType;
00161 locAMMediaType.majortype = inMajorType;
00162
00163 locAMMediaType.subtype = inSubType;
00164 locAMMediaType.formattype = inFormatType;
00165 locAMMediaType.cbFormat = inFormatSize;
00166 locAMMediaType.pbFormat = inFormatBlock;
00167 locAMMediaType.pUnk = NULL;
00168
00169
00170
00171 return new CMediaType(locAMMediaType);
00172 }
00173
00174 unsigned long OggStream::numCodecHeaders() {
00175
00176
00177 return mCodecHeaders->numPackets();
00178 }
00179 void OggStream::flush() {
00180 CAutoLock locLock(mStreamLock);
00181
00182
00183
00184 mPacketiser.reset();
00185 }
00186
00187 void OggStream::flush(unsigned short inNumPacketsToIgnore) {
00188 CAutoLock locLock(mStreamLock);
00189
00190
00191
00192
00193 mPacketiser.reset();
00194 mPacketiser.setNumIgnorePackets(inNumPacketsToIgnore);
00195 }
00196
00197 bool OggStream::AddPin() {
00198 createFormatBlock();
00199 CMediaType* locMediaType = createMediaType( getMajorTypeGUID(),
00200 getSubtypeGUID(),
00201 getFormatGUID(),
00202 getFormatBlockSize(),
00203 getFormatBlock());
00204
00205
00206 OggDemuxSourcePin* locSourcePin = new OggDemuxSourcePin( NAME("Ogg Source Pin"),
00207 mOwningFilter,
00208 mOwningFilter->m_pLock,
00209 mCodecHeaders,
00210 locMediaType,
00211 getPinName(),
00212 mAllowSeek,
00213 getNumBuffers(),
00214 getBufferSize()
00215 );
00216 mStreamReady = true;
00217 mSourcePin = locSourcePin;
00218
00219 return true;
00220
00221 }
00222
00223 void OggStream::setLastEndGranPos(__int64 inGranPos) {
00224
00225
00226 mLastEndGranulePos = inGranPos;
00227 }
00228 bool OggStream::acceptOggPage(OggPage* inOggPage) {
00229
00230
00231
00232
00233
00234
00235
00236 if ( (!mAllowSeek) && (inOggPage->header()->isBOS() )) {
00237
00238
00239
00240
00241
00242 mOwningFilter->mSeekTimeBase = -1;
00243
00244 }
00245
00246
00247
00248
00249
00251 mLastStartGranulePos = mLastEndGranulePos;
00252 mLastEndGranulePos = inOggPage->header()->GranulePos();
00253
00254
00255 return mPacketiser.acceptOggPage(inOggPage);
00256
00257 }
00258
00259 void OggStream::setSendExcess(bool inSendExcess) {
00260 mSendExcess = inSendExcess;
00261 }
00262
00263 bool OggStream::deliverCodecHeaders() {
00264 StampedOggPacket* locPacket = NULL;
00265 for (unsigned long i = 0; i < mCodecHeaders->numPackets(); i++) {
00266 locPacket = mCodecHeaders->getPacket(i);
00267
00268 dispatchPacket(locPacket);
00269 }
00270 return true;
00271 }
00272
00273
00274 bool OggStream::dispatchPacket(StampedOggPacket* inPacket) {
00275
00276 return mSourcePin->deliverOggPacket(inPacket);
00277 }
00278
00279 void OggStream::setSerialNo(unsigned long inSerialNo) {
00280 mSerialNo = inSerialNo;
00281 }
00282