一个最简单的源Filter的编写步骤

最新推荐文章于 2023-01-19 20:31:11 发布

zhoujiaxq

最新推荐文章于 2023-01-19 20:31:11 发布

阅读量1.6k

点赞数

分类专栏： windows DSHOW c++

windows 同时被 3 个专栏收录

87 篇文章 0 订阅

订阅专栏

c++

14 篇文章 0 订阅

订阅专栏

DSHOW

7 篇文章 0 订阅

订阅专栏

1.创建一个空的Dll工程，添加5个空文件分别名为：MyOutputPin.h、 MySourceFilter.h、MyOutputPin.cpp、MySourceFilter.cpp和MySourceFilter.def。

2.声明两个类，一个是Filter的实现类，一个是输出Pin的实现类，由于是最简单的源Filter，因此Filter只有一个输出Pin。实现的功能是从本地磁盘读取三个图片文件，轮流显示这三张图片，效果是模拟一个视频流。这两个类的声明代码：

view plain copy to clipboard print ?
//MySourceFilter.h
class  CMySourceFilter
    //从SDK库中的CSource类派生
    :   public  CSource
{
public :
    //实例化接口
    static  CUnknown * WINAPI CreateInstance(LPUNKNOWN lpunk,  HRESULT  *phr);
private :
    //构造函数
    CMySourceFilter(LPUNKNOWN lpunk, HRESULT  *phr);
};
view plain copy to clipboard print ?
//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类。这个类只有两个函数需要编写，很简单。
view plain copy to clipboard print ?
//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。
view plain copy to clipboard print ?
//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函数的实现
view plain copy to clipboard print ?
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。
view plain copy to clipboard print ?
//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文件的内容
view plain copy to clipboard print ?
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，效果出来了，三幅图片轮流显示在窗口中。