用DirectShow实现视频采集

http://blog.csdn.net/dansin/archive/2005/03/27/331854.aspx

DirectShow作为DirectX的一个子集，它为用户提供了强大、方便的多媒体开接口，并且它拥有直接操作硬件的能力，这使得它的效率远胜于用GDI等图形方式编写的多媒体程序。前面一篇文章已经对DirectShow作了粗略的介绍，阐述了它的原理及一些编程方法。这里结合实践中运用DirectShow实现视频采集（WIN32）来加深对DirectShow的理解和操作能力。

1．系统环境及开发环境

  

l       系统支持DirectX（Win 2K以上系统）

l       VC++  6.0安装有DirectX  SDK（最好与系统支持的DirectX版本相同）

l       视频采集设备（如USB摄像头，本文以USB PC Camera 310P为例）

2．基本思想

DirectShow的基本原理是多媒体数据在过滤器图表（Filter Graph）中流动，通过过滤器图表中各过滤器（Filter）实现在功能，最终实现多媒体数据在渲染过滤器（Vendering Filters）中的显示和回放。

前面我们已经知道，一般过滤器可分为三类：源过滤器（Source Filters）、转换过滤器（Transform Filters）、渲染过滤器（Vendering Filters）。它们分别完成数据提供、数据格式转换（压缩编码等）和数据渲染和回放功能。所以，为了实现在WIN32系统下的视频采集，我们首先要构造出一个适当的过滤器图表，然后通过应用程序对过滤器图表的管理来完成视频采集的功能。 

这里我们一般需要2至3个过滤器。为什么这个数字会不准确呢？那是因为一方面系统采集设备的驱动模型是不确定的（一般有WDM和VFW两种）；另一方面同一采集设备它们的Filter会由于驱动程序的差异造成Filter中引脚（Pin）的不一致；还有就是不同总线的采集设备（PCI、USB、AGP）它们的Filter也是不一致的。比如：同为USB摄像头，有些Filter有两个输出引脚（Capture和Preview）；而有些Filter则只有一个输出引脚（Capture）。这里Preview引脚用来将做视频预览，Capture引脚用来将输入数据以供编码、保存等用处。

这几个过滤器分别是：

l       Video Capture Filter 采集设备Filter

l       Smart Tee Filter 将没有Preview引脚Filter的Capture引脚分为两支数据流（可选）

l       Video Venderer  视频渲染及回放Filter

通过上面3个过滤器，我们可以构造出一个完整的视频采集过滤器图表（如图1） 

   

                                                                    图1

我们也可以对上面的过滤器图表稍做修改，将它变为一个既可以预览视频，又可以将视频保存为媒体文件的图表（如图2）。

 

                                                                   图2

图表构造出来后，接下来就午剩下具体的实现了，我们只需依次构造每个Filter，然后将各信Filter的Pin按序相连即可完成图表的构造。最后，我们通过应用程序向图表发送命令（通过图表管理器完成）来控制整个视频采集的流程。

3．具体实现

首先我们需要创建几个接口全局变量。

IGraphBuilder *pGraph;   //过滤器图表管理器

ICaptureGraphBuilder2 *pBuild;  //视频采集过滤器图表

IBaseFilter *pCap;   //Video Capture Filter

IBaseFilter *pSmartTee;   //Smart Tee Filter

IBaseFilter *pRender;  //Video Renderer Filter

IMediaControl *pControl;  //用户命令接口，用来控制过滤器图表

IMediaEvent *pEvent;     //过滤器图表事件接口

1）  采集设备枚举

在构造Video Capture Filter前，我们必须列举出系统的所有采集设备，然后才能根据列举的设备名称创建Video Capture Filter。列举设备的函数实现如下

bool ListCaptureDevices()

{

         ICreateDevEnum *pDevEnum = NULL;  //设备枚举器Interface

         IEnumMoniker *pEnum = NULL;       //名称枚举Interface

         // Create the System Device Enumerator.

         HRESULT hr = CoCreateInstance(CLSID_SystemDeviceEnum, NULL,

    CLSCTX_INPROC_SERVER, IID_ICreateDevEnum,

    reinterpret_cast<void**>(&pDevEnum));  //创建设备枚举COM对象

   

         if (SUCCEEDED(hr))

         {

                   // Create an enumerator for the video capture category.

                   hr = pDevEnum->CreateClassEnumerator(

                              CLSID_VideoInputDeviceCategory,

                              &pEnum, 0);       //创建视频采集设备枚举COM对象

         }

         ////////////////////////////////////////////////////////////

         IMoniker *pMoniker = NULL;

         if(pEnum == NULL)

         {

                   return false;  //如果没有设备，返回

         } 

         while (pEnum->Next(1, &pMoniker, NULL) == S_OK)  //依次枚举，直至为空

         {

      

                   IPropertyBag *pPropBag;

                   hr = pMoniker->BindToStorage(0, 0, IID_IPropertyBag,

                       (void**)(&pPropBag));

                   if (FAILED(hr))

                   {

                       pMoniker->Release();

                       continue;  // Skip this one, maybe the next one will work.

                   }

                    // Find the description or friendly name.

                   VARIANT varName;

                   VariantInit(&varName);

                   hr = pPropBag->Read(L"Description", &varName, 0);

                   if (FAILED(hr))

                   {

                            hr = pPropBag->Read(L"FriendlyName", &varName, 0);  //设备友好名称

                   }

                   if (SUCCEEDED(hr))

                   {

                    // Add it to the application's list box.

                            char displayName[1024];

                            WideCharToMultiByte(CP_ACP,0,varName.bstrVal,-1,displayName,1024,"",NULL);

                            m_nList.AddString(displayName);  //字符转换，枚举名称均为UNICODE码 

                            VariantClear(&varName);

                   }

                   pPropBag->Release();

                   pMoniker->Release();

         }

         return true;

}

2）创建Video Capture Filter

根据枚举出来的设备友好名称（FriendlyName）创建Video Capture Filter。

bool CTest_capDlg::CreateHardwareFilter(const char * friendlyName)

{   //将friendlyName与所有的设备名称依次对比，如果相同，则创建Filter

         ICreateDevEnum * enumHardware = NULL;

  HRESULT hr = CoCreateInstance(CLSID_SystemDeviceEnum,NULL,CLSCTX_ALL

,IID_ICreateDevEnum,(void **)&enumHardware); 

         if( FAILED(hr) )

         {

                   return false;

         }

         IEnumMoniker * enumMoniker = NULL; 

         hr = enumHardware->CreateClassEnumerator(CLSID_VideoInputDeviceCategory,&enumMoniker,0); 

         if(enumMoniker)

         {

                   enumMoniker->Reset();

                   ULONG fetched = 0;

                   IMoniker * moniker = NULL;

                   char friendlyName[256];

                   while(!pCap && SUCCEEDED(enumMoniker->Next(1,&moniker,&fetched)) && fetched)

                   {

                            if(moniker)

                            {

                                     IPropertyBag * propertyBag = NULL;

                                     VARIANT name;

                                     friendlyName[0]=0;

                                     hr=moniker->BindToStorage(0,0,IID_IPropertyBag,(void **)&propertyBag);           

                                     if(SUCCEEDED(hr))

                                     {

                                               name.vt=VT_BSTR;

                                               hr = propertyBag->Read(L"FriendlyName",&name,NULL);

                                     }

                                     else

                                               return false;            

                                     if(SUCCEEDED(hr))

                                     {                

                                               WideCharToMultiByte(CP_ACP,0,name.bstrVal,-1,friendlyName,256,NULL,NULL);                                       

                                               moniker->BindToObject(0,0,IID_IBaseFilter,(void **)&pCap);

                                     }

                                     else

                                               return false;

                                     if(propertyBag)

                                     {

                                               propertyBag->Release();

                                               propertyBag=NULL;

                                     }

                                     moniker->Release();

                            }

                   }

                   enumMoniker->Release();

         }

         enumHardware->Release();

         return true;

}

3）创建视频采集过滤器图表

DirectX较高版本中一般都为开发者提供了一个ICaptureGraphBuilder2接口，开发者可以通过它方便地创建视频采集过滤器图表，然后再将它添加到IGraphBuilder图表管理器中（如图3）。                                                             

                        

                  图3

bool InitCaptureGraphBuilder()

{

         HRESULT hr = CoCreateInstance(CLSID_CaptureGraphBuilder2, NULL,

        CLSCTX_INPROC_SERVER, IID_ICaptureGraphBuilder2, (void**)&pBuild); 

         if(FAILED(hr))

                   return false;

         hr = CoCreateInstance(CLSID_FilterGraph, 0, CLSCTX_INPROC_SERVER,

            IID_IGraphBuilder, (void**)&pGraph);

         if(FAILED(hr))

         {

                   pBuild->Release();

                   return false;

          }

         pBuild->SetFiltergraph(pGraph);        ///////////////////// 过滤器图表添加到管理器中 

         pGraph->QueryInterface(IID_IMediaControl,(void **)&pControl);

         pGraph->QueryInterface(IID_IMediaEvent,(void **)&pEvent);

         return true;

}

4）创建剩余的Smart Tee和Video Renderer Filter并连接成完整的图表 

在创建完Video Capture Filter后，我们需要将Filter添加到过滤器图表中。

pGraph->AddFilter(pCap,L"Capture Filter");

然后，我们创建剩余的Filter并相连即可，值得注意的是：ICaptureGraphBuilder2为用户提供了一个RenderStream函数，它可以自动构建Smart Tee和Video Renderer Filter并将它们连接成一个完整的图表，从而完成视频采集的功能。

pBuild->RenderStream(&PIN_CATEGORY_PREVIEW, &MEDIATYPE_Video,

    pCap, NULL, NULL);

为了说明整个过程，这里我们按部就搬，依次创建各个Filter。

Smart Tee

CoCreateInstance(CLSID_SmartTee,NULL,CLSCTX_INPROC_SERVER,IID_IBaseFilter,(void **)&pSmartTee);

Video Renderer Filter

CoCreateInstance(CLSID_VideoRenderer,NULL,CLSCTX_INPROC_SERVER,IID_IBaseFilter,(void **)&pRender); 

创建好各个Filter后，我们依次取得它们的引脚（Pin），将它们按序相连即可。

IPin * GetSmartTeeInputPin()  //取得Smart Tee 输入引脚

{

         if(pSmartTee)

         {

                   IPin * pPin;

                   HRESULT hr = pSmartTee->FindPin(L"Input",&pPin);

                   if(SUCCEEDED(hr))

                   {

                            pPin->Release();

                            return pPin;

                   }

         }

         return NULL;

}

IPin * GetSmartTeeCapturePin()  //取得Smart Tee Capture引脚

{

         if(pSmartTee)

         {

                   IPin * pPin;

                   HRESULT hr = pSmartTee->FindPin(L"Capture",&pPin);

                   if(SUCCEEDED(hr))

                   {

                            pPin->Release();

                            return pPin;

                   }

         }

         return NULL;

}

IPin * GetSmartTeePreviewPin()  //取得Smart Tee Preview引脚

{

         if(pSmartTee)

         {

                   IPin * pPin;

                   HRESULT hr = pSmartTee->FindPin(L"Preview",&pPin);

                   if(SUCCEEDED(hr))

                   {

                            pPin->Release();

                            return pPin;

                   }

         }

         return NULL;

}

IPin * GetRendererPin()  //取得Video Renderer Filter的输入Pin

{

         if(pBuild)

         {

                   IPin * pPin;

                   HRESULT hr = pBuild->FindPin(pRender,PINDIR_INPUT,NULL,NULL,FALSE,0,&pPin);

                   if(SUCCEEDED(hr))

                   {

                            pPin->Release();

                            return pPin;

                   }

         }

         return NULL;

} 

将各个引脚按序连接：

IPin * pOut = FindVideoPin(&PIN_CATEGORY_CAPTURE);

IPin * pIn = GetSmartTeeInputPin();

pGraph->Connect(pOut,pIn);  //Video Capture Filter’ Capture Pin à Smart Tee’Input Pin

IPin * mOut = GetSmartTeePreviewPin(); 

IPin * mIn = GetRendererPin();                    

   

      

pGraph->Connect(mOut,mIn); //Smart Tee’s Preview Pin à Video Renderer Filter’s Input Pin

这样，一个完整的视频采集图表管理器就构造完成了。

5）开始视频采集

通过用户命令接口，我们可以方便的完成开始，暂停，停止视频采集。

pControl->Run();

pControl->Stop(); 

4．小结

通过上述视频采集过程的实现，不难发现DirectShow是一个流程清晰，开发容易的多媒体开发工具。我们在使用DirectX为我们提供的Filter构建多媒体功能的同时，也可以自己着手创建具备特定功能的Filter。总之，Direct系统还是一个巨大的宝藏，等待着我们去发掘和开采。