// 创建sink writer. 返回流索引.
HRESULT VideoEncoder::CreateSinkWriter(IMFSinkWriter** ppSinkWriter, DWORD* pStreamIndex)
{
HRESULT hr = S_OK;
if (this->m_outputFile == L"")
{
return ERROR_FILE_INVALID;
}
// 创建sink writer.
*ppSinkWriter = nullptr;
IMFSinkWriter *pSinkWriter = nullptr;
IMFMediaType* pOutputMediaType = nullptr;
IMFMediaType *pInMediaType = nullptr;
CheckHR(MFCreateSinkWriterFromURL(this->m_outputFile.c_str(), nullptr, nullptr, &pSinkWriter));
// 创建和配置输出媒体类型.
CheckHR(MFCreateMediaType(&pOutputMediaType));
CheckHR(pOutputMediaType->SetGUID(MF_MT_MAJOR_TYPE, MFMediaType_Video));
CheckHR(pOutputMediaType->SetGUID(MF_MT_SUBTYPE, this->m_outputVideoFormat));
CheckHR(pOutputMediaType->SetUINT32(MF_MT_AVG_BITRATE, this->m_videoBitRate));
CheckHR(pOutputMediaType->SetUINT32(MF_MT_INTERLACE_MODE, MFVideoInterlace_Progressive));
CheckHR(MFSetAttributeSize(pOutputMediaType, MF_MT_FRAME_SIZE, this->m_frameWidth, this->m_frameHeight));
CheckHR(MFSetAttributeRatio(pOutputMediaType, MF_MT_FRAME_RATE, (UINT32)this->m_fps, 1));
CheckHR(MFSetAttributeRatio(pOutputMediaType, MF_MT_PIXEL_ASPECT_RATIO, 1, 1));
DWORD streamIndex;
CheckHR(pSinkWriter->AddStream(pOutputMediaType, &streamIndex));
// 设置输入的媒体类型.
CheckHR(MFCreateMediaType(&pInMediaType));
CheckHR(pInMediaType->SetGUID(MF_MT_MAJOR_TYPE, MFMediaType_Video));
CheckHR(pInMediaType->SetGUID(MF_MT_SUBTYPE, MFVideoFormat_RGB32));
CheckHR(pInMediaType->SetUINT32(MF_MT_INTERLACE_MODE, MFVideoInterlace_Progressive));
// 输入的步幅信息不为所有输出编码解码器需要.但某些编解码器需要它，如 H.264.
// 如果步幅是去掉，或设置为负值，H.264 将从下到上处理图像.
CheckHR(pInMediaType->SetUINT32(MF_MT_DEFAULT_STRIDE, this->m_frameStride));
CheckHR(MFSetAttributeSize(pInMediaType, MF_MT_FRAME_SIZE, this->m_frameWidth, this->m_frameHeight));
CheckHR(MFSetAttributeRatio(pInMediaType, MF_MT_FRAME_RATE, (UINT32)this->m_fps, 1));
CheckHR(MFSetAttributeRatio(pInMediaType, MF_MT_PIXEL_ASPECT_RATIO, 1, 1));
CheckHR(pSinkWriter->SetInputMediaType(streamIndex, pInMediaType, nullptr));
// 开始编写.
CheckHR(pSinkWriter->BeginWriting());
*ppSinkWriter = pSinkWriter;
(*ppSinkWriter)->AddRef();
*pStreamIndex = streamIndex;
cleanup:
if (!SUCCEEDED(hr))
{
DWORD error = GetLastError();
this->m_logFileStream << "意外错误: " << error << endl;
}
SafeRelease(&pSinkWriter);
SafeRelease(&pOutputMediaType);
return hr;
}
//-------------------------------------------------------------------
// Write the sample
//
HRESULT VidWriter::writeFrame(BYTE *pData)
{
HRESULT hr;
IMFSample *pSample = NULL;
const DWORD cbBuffer = 4 * m_width * m_height;
// Unlock the buffer
if (m_pBuffer) m_pBuffer->Unlock();
// Set the data length of the buffer
hr = m_pBuffer->SetCurrentLength(cbBuffer);
if (FAILED(hr)) goto done;
// Create a media sample and add the buffer to it
hr = MFCreateSample(&pSample);
if (FAILED(hr)) goto done;
hr = pSample->AddBuffer(m_pBuffer);
if (FAILED(hr)) goto done;
// Set the time stamp and the duration
hr = pSample->SetSampleTime(m_rtStart);
if (FAILED(hr)) goto done;
hr = pSample->SetSampleDuration(m_frametime);
if (FAILED(hr)) goto done;
// increment the time stamp
m_rtStart += m_frametime;
// Send the sample to the Sink Writer
hr = m_pWriter->WriteSample(m_streamIndex, pSample);
done:
SafeRelease(&pSample);
return hr;
}
//-------------------------------------------------------------------
// Finalise the sink writer
//
HRESULT VidWriter::finalise()
{
if (!m_pWriter) return E_ABORT;
HRESULT hr = m_pWriter->Finalize();
SafeRelease(&m_pWriter);
return hr;
}
//-------------------------------------------------------------------
// Open a file for writing and prepare the sink writer
//
HRESULT VidWriter::initSinkWriter(WCHAR *filename)
{
HRESULT hr = S_OK;
IMFMediaType *pMediaTypeOut = NULL;
IMFMediaType *pMediaTypeIn = NULL;
// Create the sink writer
SafeRelease(&m_pWriter);
hr = MFCreateSinkWriterFromURL(filename, NULL, NULL, &m_pWriter);
if (FAILED(hr)) goto done;
// Create the output media type
hr = MFCreateMediaType(&pMediaTypeOut);
if (FAILED(hr)) goto done;
// Configure it
hr = configureOutput(pMediaTypeOut);
if (FAILED(hr)) goto done;
// Add it to the sink writer
hr = m_pWriter->AddStream(pMediaTypeOut, &m_streamIndex);
if (FAILED(hr)) goto done;
// Create the input media type
hr = MFCreateMediaType(&pMediaTypeIn);
if (FAILED(hr)) goto done;
// Configure it
hr = configureInput(pMediaTypeIn);
if (FAILED(hr)) goto done;
// Add it to the sink writer
hr = m_pWriter->SetInputMediaType(m_streamIndex, pMediaTypeIn, NULL);
if (FAILED(hr)) goto done;
// Tell the sink writer to start accepting data
hr = m_pWriter->BeginWriting();
// Reset the frame timer
m_rtStart = 0;
done:
SafeRelease(&pMediaTypeOut);
SafeRelease(&pMediaTypeIn);
return hr;
}
