/**************编码Graph的创建*************/
HRESULT hr = E_FAIL;
hr = _CreateFilterGraph(&m_nDecode.pGraph, &m_nDecode.pBuilder);
if(FAILED(hr))
{
AfxMessageBox("Uninitialize COM Library!");
return -7;
}
int iResult = -7;
do
{
/***************************查询接口***************************/
hr = m_nDecode.pGraph->QueryInterface(IID_IMediaControl, (void**)&m_nDecode.pControl);
if (FAILED(hr))//查询失败
break ;
hr = m_nDecode.pGraph->QueryInterface(IID_IMediaEventEx, (void**)&m_nDecode.pEvent);
if (FAILED(hr))//查询失败
break ;
hr = m_nDecode.pGraph->QueryInterface(IID_IMediaSeeking, (void**)&m_nDecode.pSeek);
if (FAILED(hr))//查询失败
break ;
hr = m_nDecode.pGraph->QueryInterface(IID_IMediaFilter, (void**)&m_nDecode.pMedia);
if (FAILED(hr))//查询失败
break ;
/*************************辅助Filter的创建*********************/
hr = m_nDecode.pGraph->AddSourceFilter(strSourcePathName.AllocSysString(), L"Source Filter", &m_nDecodeFilter.pSrc);
if(FAILED(hr))//添加失败
break ;
hr = CoCreateInstance(CLSID_AviDest, NULL, CLSCTX_INPROC, IID_IBaseFilter, (void**)&m_nDecodeFilter.pAviMux);
if (FAILED(hr)) //创建失败
break;
hr = m_nDecodeFilter.pAviMux->QueryInterface(IID_IMediaSeeking, (void**)&m_nDecode.pSeek);
if (FAILED(hr))//查询失败
break ;
hr = CoCreateInstance(CLSID_FileWriter, NULL, CLSCTX_INPROC, IID_IFileSinkFilter2, (void**)&m_nDecodeFilter.pSink);
if (FAILED(hr)) //创建失败
break ;
hr = m_nDecodeFilter.pSink->QueryInterface(IID_IBaseFilter, (void**)&m_nDecodeFilter.pFileWrite);
if(FAILED(hr)) //创建失败
break ;
if (!m_nDecodeFilter.H264Decode) //H264Encode Filter 未被创建
break;
hr = m_nDecode.pGraph->AddFilter(m_nDecodeFilter.H264Decode, L"H264 UnCompress Filter");
/*********************创建Graph中的上游链路(H264Decode)*****************/
hr = m_nDecode.pBuilder->RenderStream(NULL, NULL, m_nDecodeFilter.pSrc, NULL, m_nDecodeFilter.H264Decode);
if (FAILED(hr)) //创建链路失败
{
AfxMessageBox("/r该编解码器不支持所选的媒体类型,请下载相应的插件。/rCreate pSrc->pAviSplitter->H264UnCompressFilter Failed!");
break;
}
/**********************第三方编码器的检测********************/
if(strCompressorName.IsEmpty()) //压缩设备名为空
m_nDecodeFilter.pEncode = NULL;
else
if(!_CreateCompressDevice(strCompressorName, &m_nDecodeFilter.pEncode)) //创建压缩设备失败
{
iResult = -3;
break ;
}
/************************添加Filter到Graph中********************/
if(m_nDecodeFilter.pEncode)
{
hr = m_nDecode.pGraph->AddFilter(m_nDecodeFilter.pEncode, L"Compress Filter");
if(FAILED(hr))
break;
}
hr = m_nDecode.pGraph->AddFilter(m_nDecodeFilter.pAviMux, L"AVI Dest Filter");
if(FAILED(hr))
break;
hr = m_nDecode.pGraph->AddFilter((IBaseFilter *)m_nDecodeFilter.pFileWrite, L"File Write Filter");
if(FAILED(hr))
break;
hr = m_nDecodeFilter.pSink->SetMode(AM_FILE_OVERWRITE);
if(FAILED(hr))
break;
hr = m_nDecodeFilter.pSink->SetFileName(strDestPathName.AllocSysString() , NULL);
if(FAILED(hr))
{
iResult = -5;
break;
}
/*********************创建Graph中的下游链路(H264Decode)*****************/
hr = m_nDecode.pBuilder->RenderStream(NULL, NULL, m_nDecodeFilter.H264Decode, m_nDecodeFilter.pEncode, m_nDecodeFilter.pFileWrite);
if (FAILED(hr))
{
AfxMessageBox("/r该编解码器不支持所选的媒体类型,请下载相应的插件。/rCreate H264 Filter->---->File Write Filter Failed!");
break;
}
hr=m_nDecode.pMedia->SetSyncSource(NULL);
hr = m_nDecode.pControl->Run(); //运行filter graph
if(SUCCEEDED(hr))
{
iResult = 1;
m_nDecodeCtl.bStopFlag = true;
}
}while(false);
return iResult;
}
int CEncode::GetProgress(bool bFlag)
{
HRESULT hr = E_FAIL;
if(bFlag) //压缩状态
{
if(m_nEncodeCtl.bCompleted) //数据传输完成
{
m_nEncodeCtl.bCompleted = false;
return 100;
}
}
else //解压状态
{
if(m_nDecodeCtl.bCompleted) //数据传输完成
{
m_nDecodeCtl.bCompleted = false;
return 100;
}
}
/********************获得数据传输的当前状态*********************/
IMediaSeeking * pSeek = bFlag? m_nEncode.pSeek : m_nDecode.pSeek;
LONGLONG lDuration = 1;
if(pSeek == NULL)
return -1;
hr = pSeek->GetDuration(&lDuration);
if(SUCCEEDED(hr)) //得到总的持续时间成功
{
LONGLONG Start = 0;
hr = pSeek->GetCurrentPosition(&Start); //得到当前的位置
if(SUCCEEDED(hr)) //获得当前的位置成功
{
int iProgress = (int)(100*(Start/((double)lDuration)));
if (bFlag) //压缩状态
{
m_nEncodeCtl.iProgress = iProgress;
return m_nEncodeCtl.iProgress;
}
else //解缩状态
{
m_nDecodeCtl.iProgress = iProgress;
return m_nDecodeCtl.iProgress;
}
}
return -1;
}
else
return -1;
}
bool CEncode::TerminiateProgress(bool bFlag)
{
HRESULT hr = E_FAIL;
if(bFlag)
{
/*********************终止编码*******************/
if(m_nEncode.pControl!=NULL&&m_nEncodeCtl.bStopFlag)
{
OAFilterState state = State_Stopped;
if (SUCCEEDED(m_nEncode.pControl->GetState(15, &state)/*得到当前状态*/))
{
if(state != State_Stopped)
hr = m_nEncode.pControl->Stop(); //停止filter graph
}
_DestroyEncodeGraph();
}
return SUCCEEDED(hr);
}
else
{
/*********************终止解码*******************/
if(m_nDecode.pControl!=NULL&&m_nDecodeCtl.bStopFlag)
{
OAFilterState state = State_Stopped;
if (SUCCEEDED(m_nDecode.pControl->GetState(15, &state)/*得到当前状态*/))
{
if(state != State_Stopped)
hr = m_nDecode.pControl->Stop(); //停止filter graph
}
_DestroyDecodeGraph();
}
return SUCCEEDED(hr);
}
}
bool CEncode::SetNotifyWindow(HWND inWindow, bool bFlag)
{
HRESULT hr = E_FAIL;
if (bFlag) //压缩过程
{
if(m_nEncode.pEvent)
{
hr = m_nEncode.pEvent->SetNotifyWindow((OAHWND)inWindow, WM_ENCODE_NOTIFY, 0);
return SUCCEEDED(hr);
}
}
else //解压过程
if(m_nDecode.pEvent)
{
hr = m_nDecode.pEvent->SetNotifyWindow((OAHWND)inWindow, WM_DECODE_NOTIFY, 0);
return SUCCEEDED(hr);
}
return false;
}
void CEncode::OnCodecNotify(WPARAM inWParam, LPARAM inLParam, bool bFlags)
{
IMediaEventEx * pEvent = bFlags? m_nEncode.pEvent:m_nDecode.pEvent;
if (pEvent)
{
LONG eventCode = 0, eventParam1 = 0, eventParam2 = 0;
while (SUCCEEDED(pEvent->GetEvent(&eventCode, &eventParam1, &eventParam2, 0)))
{
switch (eventCode)
{
case EC_COMPLETE: //数据传输完毕
TerminiateProgress(bFlags);
bFlags? (m_nEncodeCtl.bCompleted = true) : (m_nDecodeCtl.bCompleted = true);
break;
case EC_USERABORT: //运行出错
case EC_ERRORABORT:
TerminiateProgress(bFlags);
bFlags? (m_nEncodeCtl.bCompleted = true) : (m_nDecodeCtl.bCompleted = true);
break;
default:
break;
}
pEvent->FreeEventParams(eventCode, eventParam1, eventParam2);
}
}
}
int CEncode::_IsH264Encoded(CString strFileName)
{
IGraphBuilder * pGraph = NULL;
IBaseFilter * pSrc = NULL;
IBaseFilter * pAviSplitter = NULL;
HRESULT hr = E_FAIL;
int nResult = -1;
do
{
/******************创建Filter Graph Manager******************/
hr = CoCreateInstance(CLSID_FilterGraph, NULL, CLSCTX_INPROC_SERVER, IID_IGraphBuilder, (void**)&pGraph);
if(FAILED(hr)) //创建Filter Graph Manager失败
{
AfxMessageBox("Create Filter Graph Manager Failed!");
return nResult;
}
int nlen = MultiByteToWideChar(CP_ACP, 0, strFileName, -1, NULL, NULL);
LPWSTR pstrFile = new WCHAR[nlen];
MultiByteToWideChar(CP_ACP, 0, strFileName, -1, pstrFile, nlen);
/***************创建Source Filter与AVI Splitter Filter***************/
hr = pGraph->AddSourceFilter(pstrFile, L"Source Filter", &pSrc);
if(pstrFile)
delete pstrFile; //释放资源
if(FAILED(hr))//添加Source Filter失败
{
AfxMessageBox("Create Source Filter Failed!");
return nResult;
}
hr = CoCreateInstance(CLSID_AviSplitter, NULL, CLSCTX_INPROC_SERVER, IID_IBaseFilter, (void **)&pAviSplitter);
if(FAILED(hr))//创建Avi Splitter Filter 失败
{
AfxMessageBox("Add Avi Splitter Filer Failed!");
return nResult;
}
pGraph->AddFilter(pAviSplitter, L"AVI Splitter");
/****************链接Source Filter 与AVI Splitter Filter**************/
IPin * pSrcOutPin = NULL;
IPin * pAviInPin = NULL;
IPin * pAviOutPin = NULL;
if(_FindPins(pSrc, false, &pSrcOutPin)&&_FindPins(pAviSplitter, true, &pAviInPin))
{
hr = pGraph->Connect(pSrcOutPin, pAviInPin);
if(FAILED(hr)) //连接失败
{
pSrcOutPin->Release();
pAviInPin->Release();
AfxMessageBox("Connect Source Filter to Avi Splitter Failed!");
break;
}
else //连接成功
{
if(_FindPins(pAviSplitter, false, &pAviOutPin))
{
IEnumMediaTypes * pEnumType = NULL;
hr = pAviOutPin->EnumMediaTypes(&pEnumType);
if(FAILED(hr))//枚举pin的媒体类型失败
{
pAviOutPin->Release();
AfxMessageBox("Enum MediaTypes Failed!");
break;
}
else //枚举pin的媒体类型成功
{
AM_MEDIA_TYPE * mediaType;
ULONG nfeched = 0;
pEnumType->Reset();
while(SUCCEEDED(pEnumType->Next(1, &mediaType, &nfeched))&&nfeched)//枚举媒体类型
{
static const GUID MEDIASUBTYPE_DT264=
{0x52445444, 0x0000, 0x0010, {0x80, 0x00, 0x00, 0xAA, 0x00, 0x38, 0x9B, 0x71}};
if(mediaType->subtype == MEDIASUBTYPE_DT264) //为H264编码文件
{
nResult = 1;
DeleteMediaType(mediaType);
break;
}
else 不为H264编码文件
nResult = 0;
DeleteMediaType(mediaType);
}
pEnumType->Release();
}
pAviOutPin->Release();
}
pGraph->Disconnect(pSrcOutPin);
pGraph->Disconnect(pAviInPin);
pSrcOutPin->Release();
pAviInPin->Release();
}
}
else //查找Source Filter的输出pin或Avi Splitter Filter的输入pin失败
return nResult;
}while(false);
/********************释放资源*********************/
if(pAviSplitter)
{
pGraph->RemoveFilter(pAviSplitter);
SafeRelease(pAviSplitter);
}
if(pSrc)
{
pGraph->RemoveFilter(pSrc);
SafeRelease(pSrc);
}
SafeRelease(pGraph);
return nResult;
}
bool CEncode::_IsFileOpenedCorrectly(CString strFileName)
{
bool bOpened = false;
char * pFileName = (LPSTR)(LPCTSTR)strFileName;
try
{
CFile f(pFileName, CFile::modeRead|CFile::shareDenyNone);
bOpened = true;
f.Close();
}
catch(CFileException e)
{
bOpened = false;
throw;
}
return bOpened;
}
bool CEncode::_IsSuitedFileType(CString strFileName)
{
if(strFileName.IsEmpty())
return false;
if (strFileName.GetLength()<5) //文件名长度小于5
return false;
else //文件名长度大于5
{
if ((strFileName.Right(4)).Compare(".avi")!=0) //文件名后四位不为.avi
return false;
}
return true;
}