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 "abstractaudioencodeinputpin.h"
00034
00035
00036 AbstractAudioEncodeInputPin::AbstractAudioEncodeInputPin(AbstractAudioEncodeFilter* inParentFilter, CCritSec* inFilterLock, AbstractAudioEncodeOutputPin* inOutputPin, CHAR* inObjectName, LPCWSTR inPinDisplayName)
00037 : CBaseInputPin(inObjectName, inParentFilter, inFilterLock, &mHR, inPinDisplayName),
00038 mOutputPin(inOutputPin),
00039 mUptoFrame(0),
00040 mBegun(false),
00041 mParentFilter(inParentFilter)
00042
00043 {
00044
00045
00046
00047
00048 IMediaSeeking* locSeeker = NULL;
00049 this->NonDelegatingQueryInterface(IID_IMediaSeeking, (void**)&locSeeker);
00050 mOutputPin->SetDelegate(locSeeker);
00051
00052 }
00053
00054 STDMETHODIMP AbstractAudioEncodeInputPin::NonDelegatingQueryInterface(REFIID riid, void **ppv)
00055 {
00056 if (riid == IID_IMediaSeeking) {
00057 *ppv = (IMediaSeeking*)this;
00058 ((IUnknown*)*ppv)->AddRef();
00059 return NOERROR;
00060 }
00061
00062 return CBaseInputPin::NonDelegatingQueryInterface(riid, ppv);
00063 }
00064
00065 HRESULT AbstractAudioEncodeInputPin::CompleteConnect (IPin *inReceivePin) {
00066
00067
00068 IMediaSeeking* locSeeker = NULL;
00069 inReceivePin->QueryInterface(IID_IMediaSeeking, (void**)&locSeeker);
00070 SetDelegate(locSeeker);
00071 return CBaseInputPin::CompleteConnect(inReceivePin);
00072 }
00073 AbstractAudioEncodeInputPin::~AbstractAudioEncodeInputPin(void)
00074 {
00075
00076
00077 }
00078
00079
00080 void AbstractAudioEncodeInputPin::ResetFrameCount() {
00081 mUptoFrame = 0;
00082
00083 }
00084 bool AbstractAudioEncodeInputPin::SetSampleParams(IMediaSample* outMediaSample, unsigned long inDataSize, REFERENCE_TIME* inStartTime, REFERENCE_TIME* inEndTime)
00085 {
00086 outMediaSample->SetTime(inStartTime, inEndTime);
00087 outMediaSample->SetMediaTime(NULL, NULL);
00088 outMediaSample->SetActualDataLength(inDataSize);
00089 outMediaSample->SetPreroll(FALSE);
00090 outMediaSample->SetDiscontinuity(FALSE);
00091 outMediaSample->SetSyncPoint(TRUE);
00092 return true;
00093 }
00094
00095
00096 STDMETHODIMP AbstractAudioEncodeInputPin::Receive(IMediaSample* inSample) {
00097
00098
00099 HRESULT locHR;
00100 BYTE* locBuff = NULL;
00101 locHR = inSample->GetPointer(&locBuff);
00102
00103 if (FAILED(locHR)) {
00104
00105 return locHR;
00106 } else {
00107
00108 long locResult = encodeData(locBuff, inSample->GetActualDataLength());
00109 if (locResult >= 0) {
00110
00111 return S_OK;
00112 } else {
00113
00114 return S_FALSE;
00115 }
00116 }
00117
00118
00119 return S_OK;
00120 }
00121
00122 HRESULT AbstractAudioEncodeInputPin::CheckMediaType(const CMediaType *inMediaType) {
00123
00124
00125 if ( (inMediaType->majortype == MEDIATYPE_Audio) &&
00126 (inMediaType->subtype == MEDIASUBTYPE_PCM) &&
00127 (inMediaType->formattype == FORMAT_WaveFormatEx)
00128 )
00129 {
00130 return S_OK;
00131 } else {
00132 return S_FALSE;
00133 }
00134 }
00135
00136 STDMETHODIMP AbstractAudioEncodeInputPin::EndOfStream(void) {
00137 return mParentFilter->mOutputPin->DeliverEndOfStream();
00138 }
00139
00140 STDMETHODIMP AbstractAudioEncodeInputPin::BeginFlush() {
00141 CAutoLock locLock(m_pLock);
00142 CBaseInputPin::BeginFlush();
00143 return mParentFilter->mOutputPin->DeliverBeginFlush();
00144 }
00145 STDMETHODIMP AbstractAudioEncodeInputPin::EndFlush() {
00146 CAutoLock locLock(m_pLock);
00147 CBaseInputPin::EndFlush();
00148 return mParentFilter->mOutputPin->DeliverEndFlush();
00149
00150 }
00151
00152 STDMETHODIMP AbstractAudioEncodeInputPin::NewSegment(REFERENCE_TIME tStart, REFERENCE_TIME tStop, double dRate) {
00153 return mParentFilter->mOutputPin->DeliverNewSegment(tStart, tStop, dRate);
00154 }
00155
00156 HRESULT AbstractAudioEncodeInputPin::SetMediaType(const CMediaType* inMediaType) {
00157
00158
00159
00160 if (inMediaType->subtype == MEDIASUBTYPE_PCM) {
00161 mWaveFormat = (WAVEFORMATEX*)inMediaType->pbFormat;
00162
00163 } else {
00164
00165 throw 0;
00166 }
00167 return CBaseInputPin::SetMediaType(inMediaType);
00168 }
00169
00170 HRESULT AbstractAudioEncodeInputPin::GetMediaType(int inPosition, CMediaType *outMediaType) {
00171
00172 if (inPosition < 0) {
00173 return E_INVALIDARG;
00174 }
00175
00176 switch (inPosition) {
00177 case 0:
00178
00179 outMediaType->SetType(&MEDIATYPE_Audio);
00180 outMediaType->SetSubtype(&MEDIASUBTYPE_PCM);
00181
00182 return S_OK;
00183 default:
00184 return VFW_S_NO_MORE_ITEMS;
00185 }
00186 }