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 "FLACencodeinputpin.h"
00034
00035 FLACEncodeInputPin::FLACEncodeInputPin(AbstractTransformFilter* inParentFilter, CCritSec* inFilterLock, AbstractTransformOutputPin* inOutputPin, vector<CMediaType*> inAcceptableMediaTypes)
00036 : AbstractTransformInputPin(inParentFilter, inFilterLock, inOutputPin, NAME("FLACEncodeInputPin"), L"PCM In", inAcceptableMediaTypes)
00037 , mTweakedHeaders(false)
00038 , mBegun(false)
00039 , mWaveFormat(NULL)
00040 , mUptoFrame(0)
00041
00042
00043 {
00044
00045 }
00046
00047 FLACEncodeInputPin::~FLACEncodeInputPin(void)
00048 {
00049
00050 DestroyCodec();
00051 }
00052
00053
00054
00055 HRESULT FLACEncodeInputPin::TransformData(unsigned char* inBuf, long inNumBytes) {
00056
00057 if (mBegun == false) {
00058
00059
00060 mBegun = true;
00061 init();
00062 }
00063 FLAC__int32* locFLACBuff = NULL;
00064 FLACEncodeFilter* locParentFilter = (FLACEncodeFilter*)mParentFilter;
00065 unsigned long locFLACBuffSize = (inNumBytes * 8) / locParentFilter->mFLACFormatBlock.numBitsPerSample;
00066 unsigned long locNumSamplesPerChannel = locFLACBuffSize / locParentFilter->mFLACFormatBlock.numChannels;
00067
00068 locFLACBuff = new FLAC__int32[locFLACBuffSize];
00069
00070
00071
00072
00073
00074
00075 short locTempShort = 0;
00076 for (int i = 0; i < inNumBytes; i += 2) {
00077 locTempShort = *((short*)(inBuf + i));
00078 locFLACBuff[i/2] = locTempShort;
00079 }
00080
00081 bool locRetVal = process_interleaved(locFLACBuff, locNumSamplesPerChannel);
00082
00083 if (locRetVal == true) {
00084 return 0;
00085 } else {
00086 return -1;
00087 }
00088
00089 }
00090 bool FLACEncodeInputPin::ConstructCodec()
00091 {
00092
00093 set_channels(mWaveFormat->nChannels);
00094 set_sample_rate(mWaveFormat->nSamplesPerSec);
00095 set_bits_per_sample(mWaveFormat->wBitsPerSample);
00096
00097 FLACEncodeFilter* locParentFilter = (FLACEncodeFilter*)mParentFilter;
00098 locParentFilter->mFLACFormatBlock.numBitsPerSample = mWaveFormat->wBitsPerSample;
00099 locParentFilter->mFLACFormatBlock.numChannels = mWaveFormat->nChannels;
00100 locParentFilter->mFLACFormatBlock.sampleRate = mWaveFormat->nSamplesPerSec;
00101
00102
00103
00104
00105
00107 return true;
00108 }
00109 void FLACEncodeInputPin::DestroyCodec()
00110 {
00111
00112
00113 }
00114
00115
00116
00117
00118 ::FLAC__StreamEncoderWriteStatus FLACEncodeInputPin::write_callback(const FLAC__byte inBuffer[], unsigned inNumBytes, unsigned inNumSamples, unsigned inCurrentFrame) {
00119
00120
00121
00122
00123
00124
00125 LONGLONG locFrameStart = 0;
00126 LONGLONG locFrameEnd = 0;
00127
00128
00129 if (!mTweakedHeaders) {
00130
00131
00132 unsigned char* locBuf = new unsigned char[inNumBytes];
00133 memcpy((void*)locBuf, (const void*) inBuffer, inNumBytes);
00134
00135 FLACHeaderTweaker::eFLACAcceptHeaderResult locResult = mHeaderTweaker.acceptHeader(new OggPacket(locBuf, inNumBytes, false, false));
00136
00137 if (locResult == FLACHeaderTweaker::LAST_HEADER_ACCEPTED) {
00138
00139
00140 mTweakedHeaders = true;
00141
00142 for (unsigned long i = 0; i < mHeaderTweaker.numNewHeaders(); i++) {
00143
00144
00145
00146
00147 IMediaSample* locSample;
00148 HRESULT locHR = mOutputPin->GetDeliveryBuffer(&locSample, NULL, NULL, NULL);
00149
00150 if (FAILED(locHR)) {
00151
00152
00153 return FLAC__STREAM_ENCODER_WRITE_STATUS_FATAL_ERROR;
00154 }
00155
00156 BYTE* locBuffer = NULL;
00157
00158
00159 locSample->GetPointer(&locBuffer);
00160
00161 memcpy((void*)locBuffer, (const void*)mHeaderTweaker.getHeader(i)->packetData(), mHeaderTweaker.getHeader(i)->packetSize());
00162
00163
00164 SetSampleParams(locSample, mHeaderTweaker.getHeader(i)->packetSize(), &locFrameStart, &locFrameEnd);
00165
00166 {
00167 CAutoLock locLock(m_pLock);
00168
00169
00170 HRESULT locHR = ((FLACEncodeOutputPin*)(mOutputPin))->mDataQueue->Receive(locSample);
00171 if (locHR != S_OK) {
00172
00173 } else {
00174
00175 }
00176 }
00177
00178
00179
00180
00181 }
00182
00183 return FLAC__STREAM_ENCODER_WRITE_STATUS_OK;
00184 } else if (locResult == FLACHeaderTweaker::HEADER_ACCEPTED) {
00185
00186
00187 return FLAC__STREAM_ENCODER_WRITE_STATUS_OK;
00188 } else {
00189
00190 return FLAC__STREAM_ENCODER_WRITE_STATUS_FATAL_ERROR;
00191 }
00192
00193 }
00194
00195 locFrameStart = mUptoFrame;
00196 if (inNumSamples != 0) {
00197 mUptoFrame += inNumSamples;
00198 }
00199 locFrameEnd = mUptoFrame;
00200
00201
00202
00203 IMediaSample* locSample;
00204 HRESULT locHR = mOutputPin->GetDeliveryBuffer(&locSample, &locFrameStart, &locFrameEnd, NULL);
00205
00206 if (FAILED(locHR)) {
00207
00208
00209 return FLAC__STREAM_ENCODER_WRITE_STATUS_FATAL_ERROR;
00210 }
00211
00212 BYTE* locBuffer = NULL;
00213
00214
00215 locSample->GetPointer(&locBuffer);
00216
00217
00218 if (locSample->GetSize() >= inNumBytes) {
00219
00220 memcpy((void*)locBuffer, (const void*)inBuffer, inNumBytes);
00221
00222
00223 SetSampleParams(locSample, inNumBytes, &locFrameStart, &locFrameEnd);
00224
00225 {
00226 CAutoLock locLock(m_pLock);
00227
00228
00229 HRESULT locHR = ((FLACEncodeOutputPin*)(mOutputPin))->mDataQueue->Receive(locSample);
00230 if (locHR != S_OK) {
00231
00232 } else {
00233
00234 }
00235 }
00236
00237 return FLAC__STREAM_ENCODER_WRITE_STATUS_OK;
00238 } else {
00239 throw 0;
00240 }
00241
00242
00243 }
00244 void FLACEncodeInputPin::metadata_callback(const ::FLAC__StreamMetadata *metadata)
00245 {
00246
00247
00248 }
00249
00250 STDMETHODIMP FLACEncodeInputPin::EndOfStream(void) {
00251
00252 finish();
00253 return AbstractTransformInputPin::EndOfStream();
00254 }
00255
00256 HRESULT FLACEncodeInputPin::SetMediaType(const CMediaType* inMediaType)
00257 {
00258
00259 if ( (inMediaType->subtype == MEDIASUBTYPE_PCM) &&
00260 (inMediaType->formattype == FORMAT_WaveFormatEx)) {
00261
00262 mWaveFormat = (WAVEFORMATEX*)inMediaType->pbFormat;
00263
00264 } else {
00265
00266 throw 0;
00267 }
00268
00269
00270
00271 ConstructCodec();
00272
00273 return CBaseInputPin::SetMediaType(inMediaType);
00274
00275 }