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
00032 #include "stdafx.h"
00033 #include "flacdecodeinputpin.h"
00034
00035 FLACDecodeInputPin::FLACDecodeInputPin(AbstractTransformFilter* inParentFilter, CCritSec* inFilterLock, AbstractTransformOutputPin* inOutputPin, vector<CMediaType*> inAcceptableMediaTypes)
00036 : AbstractTransformInputPin(inParentFilter, inFilterLock, inOutputPin, NAME("FLACDecodeInputPin"), L"FLAC In", inAcceptableMediaTypes)
00037 , mGotMetaData(false)
00038 , mCodecLock(NULL)
00039
00040 , mUptoFrame(0)
00041
00042 {
00043
00044 mCodecLock = new CCritSec;
00045 ConstructCodec();
00046 }
00047
00048 FLACDecodeInputPin::~FLACDecodeInputPin(void)
00049 {
00050
00051 delete mCodecLock;
00052
00053 }
00054
00055 STDMETHODIMP FLACDecodeInputPin::NonDelegatingQueryInterface(REFIID riid, void **ppv)
00056 {
00057 if (riid == IID_IMediaSeeking) {
00058 *ppv = (IMediaSeeking*)this;
00059 ((IUnknown*)*ppv)->AddRef();
00060 return NOERROR;
00061 }
00062
00063 return CBaseInputPin::NonDelegatingQueryInterface(riid, ppv);
00064 }
00065 bool FLACDecodeInputPin::ConstructCodec()
00066 {
00067 mFLACDecoder.initCodec();
00068
00069 return true;
00070 }
00071 void FLACDecodeInputPin::DestroyCodec()
00072 {
00073
00074 }
00075 STDMETHODIMP FLACDecodeInputPin::NewSegment(REFERENCE_TIME inStartTime, REFERENCE_TIME inStopTime, double inRate)
00076 {
00077 CAutoLock locLock(mStreamLock);
00078
00079 mUptoFrame = 0;
00080 return AbstractTransformInputPin::NewSegment(inStartTime, inStopTime, inRate);
00081
00082 }
00083 HRESULT FLACDecodeInputPin::TransformData(BYTE* inBuf, long inNumBytes)
00084 {
00085
00086
00087
00088
00089
00090
00091 if(!m_bFlushing) {
00092 unsigned char* locBuff = new unsigned char[inNumBytes];
00093 memcpy((void*)locBuff, (const void*)inBuf, inNumBytes);
00094
00095 OggPacket* locPacket = new OggPacket(locBuff, inNumBytes, false, false);
00096
00097 if (mGotMetaData) {
00098 StampedOggPacket* locStamped = NULL;
00099 {
00100 CAutoLock locCodecLock(mCodecLock);
00101
00102 locStamped = (StampedOggPacket*)mFLACDecoder.decodeFLAC(locPacket)->clone();
00103 }
00104
00105 if (locStamped != NULL) {
00106
00107
00108 IMediaSample* locSample;
00109
00110 HRESULT locHR = mOutputPin->GetDeliveryBuffer(&locSample, NULL, NULL, NULL);
00111
00112 if (FAILED(locHR)) {
00113
00114
00115 delete locStamped;
00116 return S_FALSE;
00117 }
00118
00119
00120 BYTE* locBuffer = NULL;
00121
00122
00123
00124 locSample->GetPointer(&locBuffer);
00125
00126
00127
00128 if (locSample->GetSize() >= locStamped->packetSize()) {
00129 REFERENCE_TIME locFrameStart = (((__int64)(mUptoFrame * UNITS)) / mFLACDecoder.mSampleRate);
00130
00131
00132
00133 mUptoFrame += (unsigned long)locStamped->endTime();
00134
00135
00136
00137 REFERENCE_TIME locFrameEnd = (((__int64)(mUptoFrame * UNITS)) / mFLACDecoder.mSampleRate);
00138
00139 memcpy((void*)locBuffer, (const void*)locStamped->packetData(), locStamped->packetSize());
00140 SetSampleParams(locSample, locStamped->packetSize(), &locFrameStart, &locFrameEnd);
00141 HRESULT locHR = ((FLACDecodeOutputPin*)(mOutputPin))->mDataQueue->Receive(locSample);
00142 if (locHR != S_OK) {
00143
00144 } else {
00145
00146 }
00147 } else {
00148 delete locStamped;
00149 throw 0;
00150 }
00151
00152
00153 delete locStamped;
00154 return S_OK;
00155 } else {
00156 return S_FALSE;
00157 }
00158 } else {
00159 {
00160 CAutoLock locCodecLock(mCodecLock);
00161 mGotMetaData = mFLACDecoder.acceptMetadata(locPacket);
00162 }
00163 if (mGotMetaData) {
00164 return S_OK;
00165 } else {
00166 return S_FALSE;
00167 }
00168 }
00169
00170 } else {
00171
00172 return S_FALSE;
00173 }
00174
00175
00176 }
00177
00178
00179 STDMETHODIMP FLACDecodeInputPin::BeginFlush() {
00180 CAutoLock locLock(m_pLock);
00181
00182
00183
00184 HRESULT locHR = AbstractTransformInputPin::BeginFlush();
00185 {
00186 CAutoLock locCodecLock(mCodecLock);
00187 mFLACDecoder.flushCodec();
00188 }
00189 return locHR;
00190
00191 }
00192
00193 STDMETHODIMP FLACDecodeInputPin::EndOfStream(void) {
00194 CAutoLock locStreamLock(mStreamLock);
00195 {
00196 CAutoLock locCodecLock(mCodecLock);
00197 mFLACDecoder.flushCodec();
00198 }
00199
00200 return AbstractTransformInputPin::EndOfStream();
00201 }
00202
00203 HRESULT FLACDecodeInputPin::SetMediaType(const CMediaType* inMediaType) {
00204
00205
00206
00207 if (inMediaType->subtype == MEDIASUBTYPE_FLAC) {
00208
00209
00210
00211 ((FLACDecodeFilter*)mParentFilter)->setFLACFormatBlock((sFLACFormatBlock*)inMediaType->pbFormat);
00212
00213 } else {
00214 throw 0;
00215 }
00216 return CBaseInputPin::SetMediaType(inMediaType);
00217 }
00218