DirectShow基礎編程 最簡單的源Filter的編寫步驟 (轉)

轉自: http://blog.csdn.net/bwmwm/article/details/5463852

 

 

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，效果出來了，三幅圖片輪流顯示在窗口中。