一个最简单的源Filter的编写步骤 收藏
1.创建一个空的Dll工程,添加5个空文件分别名为:MyOutputPin.h、 MySourceFilter.h、MyOutputPin.cpp、MySourceFilter.cpp和MySourceFilter.def。
2.声明两个类,一个是Filter的实现类,一个是输出Pin的实现类,由于是最简单的源Filter,因此Filter只有一个输出Pin。实 现的功能是从本地磁盘读取三个图片文件,轮流显示这三张图片,效果是模拟一个视频流。这两个类的声明代码:
- //MySourceFilter.h
- class CMySourceFilter
- //从SDK库中的CSource类派生
- : public CSource
- {
- public :
- //实例化接口
- static CUnknown * WINAPI CreateInstance(LPUNKNOWN lpunk, HRESULT *phr);
- private :
- //构造函数
- CMySourceFilter(LPUNKNOWN lpunk, HRESULT *phr);
- };
- //MyOutputPin.h
- class CMyOutputPin
- //CSource的派生类都使用 CSourceStream的派生类做为pin
- :public CSourceStream
- {
- public :
- CMyOutputPin(HRESULT *phr, CSource *pFilter);
- ~CMyOutputPin(void );
- //填充样本函数
- //参数pMediaSample就是要传递到下一个 Filter输入pin的样本
- //把数据填充到pMediaSample中就是这个函数的功能
- HRESULT FillBuffer(IMediaSample *pMediaSample);
- //协商每个CMediaSample数据块的大小
- HRESULT DecideBufferSize(IMemAllocator *pIMemAlloc,
- ALLOCATOR_PROPERTIES *pProperties);
- //获得媒体类型
- //在枚举器中枚举支持的媒体类型时调用此函数得到PIN支持的媒体类型
- //此函数设置pmt的各个成员,因此,由此函数的内容觉 得PIN支持什么媒体类型
- HRESULT GetMediaType( int iPosition, CMediaType *pmt);
- //检测是否支持参数传入的媒体类型
- HRESULT CheckMediaType( const CMediaType *pMediaType);
- //这是质量控制接口,最简单的源Filter不需要质量控制
- STDMETHODIMP Notify(IBaseFilter *pSelf, Quality q)
- {
- return E_FAIL;
- }
- private :
- BYTE * m_pData[3]; //存储图片数据
- int m_nWidth; //图片的宽
- int m_nHeight; //图片的高
- int m_nImageSize; //图片数据的大小
- int m_nCount; //计数器,用来切换图片数据的缓冲区
- };
3.实现CMySourceFilter类。这个类只有两个函数需要编写,很简单。
- //CMySourceFilter.cpp
- CUnknown* CMySourceFilter::CreateInstance(LPUNKNOWN lpunk, HRESULT *phr)
- {
- //实例化函数的工作就是实例化一个源Filter的对象
- CUnknown *punk = new CMySourceFilter(lpunk,phr);
- if (punk == NULL)
- {
- *phr = E_OUTOFMEMORY;
- }
- return punk;
- }
- CMySourceFilter::CMySourceFilter(LPUNKNOWN lpunk, HRESULT *phr)
- : CSource(L"MyFilter" ,lpunk,CLSID_MyFilter,phr)
- {
- //创建一个pin的对象实例
- //在CSourceStream的构造函数中,会把pin 添加到Filter中
- CMyOutputPin* pOutPin = new CMyOutputPin(phr, this );
- if (FAILED(*phr))
- {
- //因此,在创建失败的时候,要将这个pin从Filter中移除
- RemovePin(pOutPin);
- pOutPin->Release();
- }
- }
4.实现CMyOutputPin类,编写Filter主要就是写pin。
- //MyOutputPin.cpp
- //构造函数
- CMyOutputPin::CMyOutputPin(HRESULT *phr, CSource *pFilter)
- : CSourceStream(L"MyFilter" ,phr,pFilter,L "Out" )
- , m_nWidth(0)
- , m_nHeight(0)
- , m_nImageSize(0)
- , m_nCount(0)
- {
- //把图片读到内存中,准备好数据
- m_pData[0] = LoadBitmapFileToMemory(L"E://DirectShow//MySourceFilter//1.bmp" ,
- m_nWidth,m_nHeight,m_nImageSize);
- m_pData[1] = LoadBitmapFileToMemory(L"E://DirectShow//MySourceFilter//2.bmp" ,
- m_nWidth,m_nHeight,m_nImageSize);
- m_pData[2] = LoadBitmapFileToMemory(L"E://DirectShow//MySourceFilter//3.bmp" ,
- m_nWidth,m_nHeight,m_nImageSize);
- }
- //析构函数
- CMyOutputPin::~CMyOutputPin(void )
- {
- //释放内存
- delete []m_pData[0];
- delete []m_pData[1];
- delete []m_pData[2];
- }
- //获取媒体类型
- //填充pmt
- //最简单的源Filter,因此只支持一种类型,所以 iPosition为0
- HRESULT CMyOutputPin::GetMediaType( int iPosition, CMediaType *pmt)
- {
- CheckPointer(pmt,E_POINTER);
- CAutoLock cAutoLock(m_pFilter->pStateLock());
- if (iPosition < 0)
- {
- return E_INVALIDARG;
- }
- // Have we run off the end of types?
- if (iPosition > 0)
- {
- return VFW_S_NO_MORE_ITEMS;
- }
- //给媒体类型申请Format的空间
- //填充每一个对象,主要是BITMAPINFOHEADER结 构
- VIDEOINFO *pvi = (VIDEOINFO *) pmt->AllocFormatBuffer(sizeof (VIDEOINFO));
- if (NULL == pvi)
- return (E_OUTOFMEMORY);
- ZeroMemory(pvi, sizeof (VIDEOINFO));
- pvi->bmiHeader.biBitCount = 24;
- pvi->bmiHeader.biHeight = m_nHeight;
- pvi->bmiHeader.biWidth = m_nWidth;
- pvi->bmiHeader.biSizeImage = m_nImageSize;
- pvi->bmiHeader.biPlanes = 1;
- pvi->bmiHeader.biSize = sizeof (BITMAPINFOHEADER);
- pvi->bmiHeader.biCompression = BI_RGB;
- pvi->bmiHeader.biClrImportant = 0;
- SetRectEmpty(&pvi->rcSource);
- SetRectEmpty(&pvi->rcTarget);
- pmt->SetType(&MEDIATYPE_Video);//设置主媒体类型
- pmt->SetSubtype(&MEDIASUBTYPE_RGB24);//设置子媒体类型
- pmt->SetFormatType(&FORMAT_VideoInfo);//设置详细格式类型
- pmt->SetSampleSize(m_nImageSize);//设置Sample的大小
- pmt->SetTemporalCompression(FALSE);
- return NOERROR;
- }
- //检查媒体类型
- //主要是对GetMediaType中设置的各个参数进行比较
- HRESULT CMyOutputPin::CheckMediaType( const CMediaType *pMediaType)
- {
- CheckPointer(pMediaType,E_POINTER);
- if (*(pMediaType->Type()) != MEDIATYPE_Video
- || !(pMediaType->IsFixedSize()))
- {
- return E_INVALIDARG;
- }
- const GUID *SubType = pMediaType->Subtype();
- if (SubType == NULL)
- {
- return E_INVALIDARG;
- }
- if (*SubType != MEDIASUBTYPE_RGB24)
- {
- return E_INVALIDARG;
- }
- const GUID* FormatType = pMediaType->FormatType();
- if (FormatType == NULL)
- {
- return E_INVALIDARG;
- }
- if (*FormatType != FORMAT_VideoInfo)
- {
- return E_INVALIDARG;
- }
- VIDEOINFO* pvi = (VIDEOINFO*)pMediaType->Format();
- if (pvi == NULL)
- {
- return E_INVALIDARG;
- }
- if (pvi->bmiHeader.biBitCount != 24 ||
- pvi->bmiHeader.biWidth != m_nWidth ||
- pvi->bmiHeader.biHeight != m_nHeight)
- {
- return E_INVALIDARG;
- }
- return S_OK;
- }
- //协商Sample的大小
- HRESULT CMyOutputPin::DecideBufferSize(IMemAllocator *pIMemAlloc, ALLOCATOR_PROPERTIES *pProperties)
- {
- CheckPointer(pIMemAlloc,E_POINTER);
- CheckPointer(pProperties,E_POINTER);
- CAutoLock cAutoLock(m_pFilter->pStateLock());
- HRESULT hr = NOERROR;
- VIDEOINFO *pvi = (VIDEOINFO *) m_mt.Format();
- //确定只有一个buffer
- pProperties->cBuffers = 1;
- //设置buffer的大小
- pProperties->cbBuffer = m_nImageSize;
- ASSERT(pProperties->cbBuffer);
- //设置属性页
- ALLOCATOR_PROPERTIES Actual;
- hr = pIMemAlloc->SetProperties(pProperties,&Actual);
- if (FAILED(hr))
- {
- return hr;
- }
- if (Actual.cbBuffer < pProperties->cbBuffer)
- {
- return E_FAIL;
- }
- ASSERT(Actual.cBuffers == 1);
- return NOERROR;
- }
- //填充Sample
- HRESULT CMyOutputPin::FillBuffer(IMediaSample *pMediaSample)
- {
- CheckPointer(pMediaSample,E_POINTER);
- BYTE * pData = NULL;
- long lDataSize = 0;
- //获得Sample中存放数据的地址
- pMediaSample->GetPointer(&pData);
- //取得Sample分配的内存大小
- lDataSize = pMediaSample->GetSize();
- ZeroMemory(pData,lDataSize);
- //把当前需要显示的数据拷贝到内存中
- CopyMemory(pData,m_pData[m_nCount%3],m_nImageSize);
- //设置时间戳
- REFERENCE_TIME start = TS_ONE * m_nCount;
- REFERENCE_TIME stop = TS_ONE + start;
- pMediaSample->SetTime(&start,&stop);
- //准备下一帧数据
- m_nCount++;
- pMediaSample->SetSyncPoint(TRUE);
- return NOERROR;
- }
LoadBitmapFileToMemory函数的实现
- BYTE * LoadBitmapFileToMemory( TCHAR * pFileName, int & nWidth, int & nHeight, int & nImageDataSize)
- {
- HBITMAP hBitmap = ( HBITMAP )LoadImage( NULL, pFileName, IMAGE_BITMAP, 0, 0,
- LR_CREATEDIBSECTION | LR_DEFAULTSIZE | LR_LOADFROMFILE );
- if (hBitmap == NULL)
- return NULL;
- HDC hDC = CreateCompatibleDC(NULL);
- HBITMAP hOldBitmap = ( HBITMAP )SelectObject(hDC, hBitmap);
- BITMAP bmp;
- GetObject(hBitmap, sizeof (bmp), &bmp);
- BITMAPINFOHEADER bih = {0};//位图信息头
- bih.biBitCount = bmp.bmBitsPixel;//每个像素字节大小
- bih.biCompression = BI_RGB;
- bih.biHeight = bmp.bmHeight;//高 度
- bih.biPlanes = 1;
- bih.biSize = sizeof (BITMAPINFOHEADER);
- bih.biSizeImage = bmp.bmWidthBytes * bmp.bmHeight;//图像数据大小
- bih.biWidth = bmp.bmWidth;//宽度
- nImageDataSize = bmp.bmWidthBytes * bmp.bmHeight;
- byte * p = new byte[nImageDataSize]; //申请内存保存位图数据
- GetDIBits(hDC, hBitmap, 0, bmp.bmHeight, p,
- (LPBITMAPINFO) &bih, DIB_RGB_COLORS);//获取位图数据
- SelectObject(hDC, hOldBitmap);
- DeleteObject(hBitmap);
- DeleteDC(hDC);
- nWidth = bmp.bmWidth;
- nHeight = bmp.bmHeight;
- return p;
- }
5.主要的工作已经做完了,功能已经实现,接下来就是生成Filter。
- //MySourceFilter.h
- //动态库工程自然也要有入口函数(固定格式)
- extern "C" BOOL WINAPI DllEntryPoint( HINSTANCE , ULONG , LPVOID );
- BOOL APIENTRY DllMain( HANDLE hModule,
- DWORD dwReason,
- LPVOID lpReserved)
- {
- return DllEntryPoint(( HINSTANCE )(hModule), dwReason, lpReserved);
- }
- //组件就少不了注册与注销函数(固定格式)
- STDAPI DllRegisterServer()
- {
- return AMovieDllRegisterServer2(TRUE);
- }
- STDAPI DllUnregisterServer()
- {
- return AMovieDllRegisterServer2(FALSE);
- }
- //组件,就要有GUID(通过工具创建)
- DEFINE_GUID(CLSID_MyFilter,
- 0x159386e0, 0x5193, 0x48ac, 0x8a, 0x57, 0x17, 0x88, 0xc7, 0x33, 0x40, 0xc1);
- //以下是注册信息的模版,写了注释的地方是我们需要填写的,其他的采用默认
- const AMOVIESETUP_MEDIATYPE sudOpPinTypes =
- {
- &MEDIATYPE_Video, // Major type
- &MEDIASUBTYPE_NULL // Minor type
- };
- const AMOVIESETUP_PIN sudOpPin =
- {
- L"Output" ,
- FALSE,
- TRUE,
- FALSE,
- FALSE,
- &CLSID_NULL,
- NULL,
- 1,
- &sudOpPinTypes };
- const AMOVIESETUP_FILTER sudBallax =
- {
- &CLSID_MyFilter, // 自定义的GUID
- L"MyFilter" , // Filter的名字
- MERIT_DO_NOT_USE,
- 1,
- &sudOpPin
- };
- // COM global table of objects in this dll
- CFactoryTemplate g_Templates[] = {
- { L"MyFilter" //Filter的名字
- , &CLSID_MyFilter//自定义的 GUID
- , CMySourceFilter::CreateInstance//Filter的实例化接口
- , NULL
- , &sudBallax }
- };
- int g_cTemplates = sizeof (g_Templates) / sizeof (g_Templates[0]);
6.MySourceFilter.def文件的内容
- LIBRARY "MySourceFilter.ax"
- EXPORTS
- DllMain PRIVATE
- DllGetClassObject PRIVATE
- DllCanUnloadNow PRIVATE
- DllRegisterServer PRIVATE
- DllUnregisterServer PRIVATE
7.注意
1)包含头文件 #include <initguid.h>,否则有可能提示 error LNK2001: 无法解析的外部符号 _CLSID_MyFilter
2)包含导出库#pragma comment(lib, "winmm")
3)包含导入库#pragma comment(lib, "strmbase.lib"),Debug下包含#pragma comment(lib, "strmbasd.lib")
8.大功告成。调用regsvr32注册Filter。使用GraphEdit调试Filter。(VS2005)
在工程的属性中选择调试,在命令中填入GraphEdit的完整路径,把Filter的工程作为启动项。按下F5,在运行的GraphEdit中选 择我们的Filter,Render pin,就可以看到一条完整的链路,然后run,效果出来了,三幅图片轮流显示在窗口中。