directshow使用Sample Grabber采样

//以下代码来自陆其明所著之DirectShow开发指南,他是这方面的专家,目前市面上两本关于directshow的书都是他写的…

1、 指定你想要扑获的媒体类型
在Sample Grabber 过滤器连接到别的过滤器之前你必须配置它。首先你要调用CoCreateInstance来创建Sample Grabber ，然后调用IfilterGraph::AddFilter来加载它到过滤图形中。然后查询IsampleGrabber接口。我们使用IsampleGrabber::SetMediaType方法来设置媒体类型。这个方法指定了Sample Grabber过滤器将要连接的媒体类型。你可以仅仅指定主媒体类型；或者主类型加子类型；或者主类型，子类型和类型格式。
例如，如果你想扑获一个未压缩的视频桢，这个视频桢要求是兼容当前显示模式的，你可以设置主类型为MEDIATYPE_Video然后设置基于当前显示位深的子类型。下面的例子也许能说明问题：

// 创建 Sample Grabber.  
IBaseFilter *pF = NULL;  
ISampleGrabber *pGrabber = NULL;  
hr = CoCreateInstance(CLSID_SampleGrabber, NULL, CLSCTX_INPROC_SERVER,  
    IID_IBaseFilter, reinterpret_cast<void**>(&pF));  
hr = pF->QueryInterface(IID_ISampleGrabber,  
    reinterpret_cast<void**>(&pGrabber));  
hr = pGraph->AddFilter(pF, L"SampleGrabber");  
// 找到当前的色深  
HDC hdc = GetDC(NULL);  
int iBitDepth = GetDeviceCaps(hdc, BITSPIXEL);  
ReleaseDC(NULL, hdc);  
// 设置媒体类型  
AM_MEDIA_TYPE mt;  
ZeroMemory(&mt, sizeof(AM_MEDIA_TYPE));  
mt.majortype = MEDIATYPE_Video;  
switch (iBitDepth)  
{  
case 8:  
    mt.subtype = MEDIASUBTYPE_RGB8;  
    break;  
case 16:  
    mt.subtype = MEDIASUBTYPE_RGB555;  
    break;  
case 24:  
    mt.subtype = MEDIASUBTYPE_RGB24;  
    break;  
case 32:  
    mt.subtype = MEDIASUBTYPE_RGB32;  
    break;  
default:  
    return E_FAIL;  
}  
hr = pGrabber->SetMediaType(&mt);  

2、 建立包含Sample Grabber过滤图形

在你指定媒体类型之后，你就可以建立一个包含Sample Grabber的过滤图形了。Sample Grabber将仅连接到指定的类型，而且允许你在建立过滤图形时使用智能连接机制。例如前面的代码就指定了未压缩的视频桢。准备好了以后，你就可以调用IgraphBuilder::AddSourceFilter方法从视频文件源流中加入捕获过滤器。为了将这个过滤器插入过滤图形，我们还需要调用IgraphBuilder::Connect方法来连接源过滤器和Sample Grabber。过滤图形管理器会自动的添加需要的解码过滤器。下面的代码就是这样做的。它使用了两个帮助函数来枚举引脚。GetPin函数在过滤器中找到第一个引脚，输入或者输出引脚。ConnectFilters函数找到过滤器中的第一个输出引脚，然后连接这个引脚到另外一个过滤器的第一个输入引脚。

HRESULT GetPin(IBaseFilter *, PIN_DIRECTION, IPin **);  
HRESULT ConnectFilters(IGraphBuilder *, IBaseFilter *, IBaseFilter *);  

IBaseFilter *pSrc;  
hr = pGraph->AddSourceFilter(wszFileName, L"Source", &pSrc);  
hr = ConnectTwoFilters(pGraph, pSrc, pF);  

// Helper functions:  
HRESULT GetPin(IBaseFilter *pFilter, PIN_DIRECTION PinDir, IPin **ppPin)  
{  
    IEnumPins  *pEnum;  
    IPin       *pPin;  
    pFilter->EnumPins(&pEnum);  
    while(pEnum->Next(1, &pPin, 0) == S_OK)  
    {  
        PIN_DIRECTION PinDirThis;  
        pPin->QueryDirection(&PinDirThis);  
        if (PinDir == PinDirThis)  
        {  
            pEnum->Release();  
            *ppPin = pPin;  
            return S_OK;  
        }  
        pPin->Release();  
    }  
    pEnum->Release();  
    return E_FAIL;    
}  

HRESULT ConnectFilters(IGraphBuilder *pGraph, IBaseFilter *pFirst, IBaseFilter *pSecond)  
{  
    IPin *pOut = NULL, *pIn = NULL;  
    HRESULT hr = GetPin(pFirst, PINDIR_OUTPUT, &pOut);  
    if (FAILED(hr)) return hr;  
    hr = GetPin(pSecond, PINDIR_INPUT, &pIn);  
    if (FAILED(hr))   
    {  
        pOut->Release();  
        return E_FAIL;  
     }  
    hr = pGraph->Connect(pOut, pIn);  
    pIn->Release();  
    pOut->Release();  
    return hr;  
}  

应用程序必须连接了Sample Grabber的输出引脚。如果你想丢弃采样，那么你可以连接Null Renderer过滤器。这个过滤器将会丢弃它所收到的每一桢。

  IBaseFilter *pNull = NULL;
     hr = CoCreateInstance(CLSID_NullRenderer, NULL, CLSCTX_INPROC_SERVER,
         IID_IBaseFilter, reinterpret_cast<void**>(&pNull));
     hr = pGraph->AddFilter(pNull, L"NullRenderer");
     hr = ConnectTwoFilters(pGraph, pF, pNull);

这里不推荐连接标准Video Renerder,因为它是一个Trans_In_Place Filter ,输入和输出PIN采用的是相当的数据缓存,而且这个内存极有可能在显卡上面,而从显存中读取数据比从内存中读取数据要慢得多,这会影响效率.

3、运行过滤图形
Sample Grabber工作包含两种模式：
A、 在将采样向下传送之前产生每个采样的拷贝，然后放到其缓冲。
B、 以回调方式进行处理数据，回调由应用程序定义。
这里我们仅讨论一下缓冲模式。但是大家要注意的是回调方式会影响我们的工作效率，甚至死锁。回调函数设置我们用IsampleGrabber::SetCallback方法。
为了激活缓冲模式，我们调用IsampleGrabberr::SetBufferSample方法，参数填充TRUE。你也可以使用IsampleGrabber::SetOneShot方法，这样会导致每捕获一桢后过滤图形停止。这个特性对我们如果仅想在流里面捕获一桢的需求十分有益。我们可以搜索到想要捕获的地方运行过滤图形，进行截获。但是桢的精确度还是要靠数据源的性质决定。
下面的例子实现的就是这些：

// 设置快照和缓冲模式.  
hr = pGrabber->SetOneShot(TRUE);  //所使用回调/Video Renerder方式设置为false
hr = pGrabber->SetBufferSamples(TRUE);  //所使用回调方/Video Renerder式设置为false

IMediaFilter *pMediaFilter = NULL;  
IMediaControl *pControl = NULL;  
IMediaEventEx *pEvent = NULL;  

pMediaFilter->SetSyncSource(NULL); // 关掉参考时钟  
pControl->Run(); // 运行过滤图形.  
pEvent->WaitForCompletion(INFINITE, &evCode); // 等待直到结束  

4、 从Sample Grabber中得到缓冲采样，或者实现回调函数来获得数据。
在缓冲模式，Sample Grabber过滤器存储了它收到的每个采样的拷贝。我们要获得缓冲数据就要调用IsampleGrabber::GetCurrentBuffer方法。这个方法填充一个调用者分配好的矩阵。为了能获得缓冲区大小，调用方法的时候必须将缓冲的指针填NULL。

        long cbBuffer = 0;
         hr = pGrabber->GetCurrentBuffer(&cbBuffer, NULL);
         char *pBuffer = new char[cbBuffer];
         if (!pBuffer) 
         {
             return E_OUTOFMEMORY;
         }
         hr = pGrabber->GetCurrentBuffer(&cbBuffer, 
             reinterpret_cast<long*>(pBuffer));

使用IsampleGrabber::GetConnectedMediaType方法来获得缓冲格式。例如，如果缓冲是一个未压缩的视频桢，它的格式就是VIDEOINFOHEADER结构格式。注意，Sample Grabber不支持VIDEOINFOHEADER2结构。

        AM_MEDIA_TYPE mt;
         hr = pGrabber->GetConnectedMediaType(mt);
         VIDEOINFOHEADER *pVih;
         if (mt.formattype == FORMAT_VideoInfo) 
             pVih = reinterpret_cast<VIDEOINFOHEADER*>(mt.pbFormat);
         else 
             return VFW_E_INVALIDMEDIATYPE; // Something went wrong
         // pVih->bmiHeader 这个参数是BITMAPINFOHEADER 结构，是每一桢的图形信息.

         // 释放格式块
         FreeMediaType(mt)

;
在原书中还读到了用GDI方式画出所捕获的图形,这里不再写出,有兴趣的参与原书.