Directshow 视频捕捉 三

如何捕捉视频流并保存到文件（Capture video to File）

1 将视频流保存到AVI文件

AVI Mux filter接收从capture pin过来的视频流，然后将其打包成AVI流。音频流也可以连接到AVI Mux Filter上，这样mux filter就将视频流和视频流合成AVI流。File writer将AVI流写入到文件中。
可以像下面这样构建graph图

Cpp代码  

1. IBaseFilter *pMux;  
2. hr = pBuild->SetOutputFileName(&MEDIASUBTYPE_Avi,            //Specifies AVI for the target file  
3.                                 L"C:\\Example.avi",             //File name  
4.                                &pMux,                                   //Receives a pointer to the mux  
5.                                NULL);       //(Optional)Receives a pointer to the file sink  

 

 

第一个参数表明文件的类型，这里表明是AVI，第二个参数是制定文件的名称。对于AVI文件，SetOutputFileName函数会创建一个AVI mux Filter 和一个 File writer Filter ，并且将两个filter添加到graph图中，在这个函数中，通过File Writer Filter 请求IFileSinkFilter接口，然后调用IFileSinkFilter::SetFileName方法，设置文件的名称。然后将两个filter连接起来。第三个参数返回一个指向 AVI Mux的指针，同时，它也通过第四个参数返回一个IFileSinkFilter参数，如果你不需要这个参数，你可以将这个参数设置成NULL。
然后，你应该调用下面的函数将capture filter 和AVI Mux连接起来。

Cpp代码  

1. hr = pBuild->RenderStream(&PIN_CATEGORY_CAPTURE,         //Pin category  
2.                           &MEDIATYPE_Video,         //Media type  
3.                           pCap,         //Capture filter  
4.                           NULL,         //Intermediate filter(optional)  
5.                           pMux);         //Mux or file sink filter  
6. //Release the mux filter  
7. pMux->Release();  

 

 

第5个参数就是使用的上面函数返回的pMux指针。
当捕捉音频的时候，媒体类型要设置为MEDIATYPE_Audio，如果你从两个不同的设备捕捉视频和音频，你最好将音频设置成主流，这样可以防止两个数据流间drift，因为avi mux filter为同步音频，会调整视频的播放速度的。为了设置master 流，调用IConfigAviMux::SetMasterStream方法，可以采用如下的代码：

Cpp代码  

1. IConfigAviMux *pConfigMux = NULL;  
2. hr = pMux->QueryInterface(IID_IConfigAviMux, (void**)&pConfigMux);  
3. if(SUCCEEDED(hr))  
4. {  
5.     pConfigMux->SetMasterStream(1);  
6.     pConfigMux->Release();  
7. }  

 

 

SetMasterStream的参数指的是数据流的数目，这个是由调用RenderStream的次序决定的。例如，如果你调用RenderStream首先用于视频流，然后是音频，那么视频流就是0，音频流就是1。
添加编码filter

Cpp代码  

1. IBaseFilter *pEncoder;  
2. //Add it to the filter graph  
3. pGraph->AddFilter(pEncoder, L"Encode");  
4. //Render the stream  
5. hr = pBuild->RenderStream(&PIN_CATEGORY_CAPTURE, &MEDIATYPE_Video,   
6.                           pCap, pEncoder, pMux);  
7. pEncoder->Release();  

 

 

2 将视频流保存成wmv格式的文件

为了将视频流保存成并编码成windows media video （WMV）格式的文件，将capture pin连到WM ASF Writer filter。

构建graph图最简单的方法就是将在ICaptureGraphBuilder2::SetOutputFileName方法中指定MEDIASUBTYPE_Asf的filter。如下

Cpp代码  

1. IBaseFilter *pASFWriter = 0;  
2. hr = pBuild->SetOutputFileName(&MEDIASUBTYPE_Asf,        //Create a windows media file  
3.                                L"C:\\VidCap.wmv",               //File name  
4.                                &pASFWriter,             //Receives a pointer to the filter  
5.                                NULL);               //Receives an IFileSinkFilter interface pointer(optional)  

 

 

参数MEDIASUBTYPE_Asf 告诉graph builder，要使用wm asf writer作为文件接收器，于是，pbuild 就创建这个filter，将其添加到graph图中，然后调用IFileSinkFilter::SetFileName来设置输出文件的名字。第三个参数用来返回一个ASF writer指针，第四个参数用来返回文件的指针。

在将任何pin连接到WM ASF Writer之前，一定要对WM ASF Writer进行一下设置，你可以同过WM ASF Writer的IConfigAsfWriter接口指针来进行设置。

Cpp代码  

1. IConfigAsfWriter *pConfig = 0;  
2. hr = pASFWriter->QueryInterface(IID_IConfigAsfWriter, (void**)&pConfig);  
3. if(SUCCEEDED(hr))  
4. {  
5.     //Configure the ASF Writer filter  
6.     pConfig->Release();  
7. }  

 

 

然后调用ICaptureGraphBuilder2::RenderStream将capture Filter 和 ASF writer连接起来：

Cpp代码  

1. hr = pBuild->RenderStream(&PIN_CATEGORY_CAPTURE,         //Capture pin  
2.                           &MEDIATYPE_Video,             //Video. Use MEDIATYPE_Audio for audio  
3.                           pCap,         //Pointer to the capture filter  
4.                           0,   
5.                           pASFWriter);      //Pointer to the sink filter(ASF Filter)  

 

3保存成自定义的文件格式
如果你想将文件保存成自己的格式，你必须有自己的 file writer。看下面的代码：

Cpp代码  

1. IBaseFilter *pMux = 0;  
2. IFileSinkFilter *pSink = 0;  
3. hr = pBuild->SetOutputFileName(&CLSID_MyCustomMuxFilter,     //开发自己的Filter  
4.                                L"C:\\VidCap.avi", &pMux, &pSink);  

 

 

 

4如何将视频流保存进多个文件
当你将视频流保存进一个文件后，如果你想开始保存第二个文件，这时，你应该首先将graph停止，然后通过IFileSinkFilter::SetFileName改变 File Writer 的文件名称。注意，IFileSinkFilter指针你可以在SetOutputFileName时通过第四个参数返回的。
看看保存多个文件的代码：

Cpp代码  

1. IBaseFilter *pMux = 0;  
2. IFileSinkFilter *pSink = 0;  
3. hr = pBuild->SetOutputFileName(&MEDIASUBTYPE_Avi,   
4.                                L"C:\\YourFileName.avi", &pMux, &pSink);  
5. if(SUCCEEDED(hr))  
6. {  
7.     hr = pBuild->RenderStream(&PIN_CATEGORY_CAPTURE, &MEDIATYPE_Video,  
8.                                                 pCap, NULL, pMux);  
9.     if(SUCCEEDED(hr))  
10.     {  
11.         pControl->Run();  
12.         pControl->Stop();  
13.         //Change the file name and run the graph again  
14.         pSink->SetFileName(L"YourFileName02.avi", 0);  
15.         pControl->Run();  
16.     }  
17.   
18.     pMux->Release();  
19.     pSink->Release();  
20. }