码迷，mamicode.com
首页 > 其他好文 > 详细

CCapture directshow 视频捕获类

时间：2014-05-01 10:31:47      阅读：702      评论：0      收藏：0      [点我收藏+]

标签：style   blog   class   code   tar   ext   width   strong   color   string   get   

 

[cpp] view plaincopy
 
  1. // Capture.h for class CCapture  
  2.   
  3. #include <dshow.h>  
  4. #include <qedit.h>  
  5. #include <atlbase.h>  
  6. #if !defined(CAPTURE_H_________)  
  7. #define CAPTURE_H_________  
  8.   
  9. // image size: 160*120  176*144   320*240  640*480  1024*1806  
  10. #define IMG_WIDTH 320  
  11. #define IMG_HEIGHT 240  
  12.   
  13. typedef void (*capCallBackFunc)(LPVOID lParam);  
  14. enum DeviceType{DTypeVideo, DTypeAudio};  
  15. class CSampleGrabberCB; // 用于不会帧数据保存图片的接口  
  16. class CCapture  
  17. {  
  18.     friend class CSampleGrabberCB;  
  19. public:  
  20.     // 设置回调函数 用于处理获取的图片帧数据  
  21.     CDialog *m_dlgParent;  
  22.     capCallBackFunc calFunc;  
  23.     void SetCallBKFun(capCallBackFunc f);  
  24.     /////////////////////////////////  
  25.     CCapture();  
  26.     virtual ~CCapture();  
  27.     int EnumDevice(HWND hCmbList, DeviceType deviceType); // 设备枚举  
  28. //  void SaveGraph(TCHAR *wFileName);   // 保存滤波器链表  
  29.     void SetCameraFormat(HWND hwndParent);  // 设置摄像头的视频格式  
  30.     void SetCameraFilter(HWND hwndParent);  // 设置摄像头的图像参数  
  31.     HRESULT CaptureVideo(CString inFileName);   // 捕获保存视频  
  32.     HRESULT CaptureImage(CString inFileName);   // 抓取保存图片  
  33.     HRESULT CaptureImage(); // 抓取图片并显示  
  34.     HRESULT Preview(int iDevVideoID, HWND hVideo, int iDevAudioID = 0, HWND hAudio = NULL); // 采集预览视频  
  35.     HRESULT InitCaptureGraphBuilder();  // 创建滤波器管理器，查询其各种控制接口  
  36.     void StopCapture();  // 停止捕获  
  37.     void FreeMediaType(AM_MEDIA_TYPE &mt);  // 释放对象内存  
  38.   
  39.     void SetOnShot(BOOL bFlag);   // 设置是否捕获帧数据  
  40.     void SetParent(CDialog *pdlg);  
  41. protected:  
  42.     bool BindFilter(int iDeviceID, IBaseFilter **pOutFilter, DeviceType deviceType); // 把指定的设备滤波器捆绑到链表中  
  43.     void ResizeVideoWindow();           // 更改视频显示窗口  
  44.     HRESULT SetupVideoWindow();         // 设置视频显示窗口的特性  
  45.     static UINT ThreadFunDrawText(LPVOID lParam);  
  46. private:  
  47.     HWND m_hWnd;            // 视频显示窗口的句柄  
  48.     IBaseFilter *m_pVideoCap;       // 视频捕获滤波器  
  49.     IBaseFilter *m_pAudioCap;       // 音频捕获滤波器  
  50.     CComPtr<ISampleGrabber> m_pGrabber;       // 抓取图片滤波器  
  51.     IBaseFilter *m_pMux;    // 写文件滤波器  
  52.     ICaptureGraphBuilder2 *m_pCapGB;    // 增强型捕获滤波器链表管理  
  53.     IGraphBuilder *m_pGB;   // 滤波链表管理器  
  54.     IVideoWindow *m_pVW;    // 视频显示窗口接口  
  55.     IMediaControl *m_pMC;   // 媒体控制接口  
  56.     static bool m_bRecording;       // 录制视频标志  
  57.   
  58.     IBaseFilter *m_pXviDCodec;   //mpeg4 滤波器  
  59. };  
  60.   
  61.   
  62. #endif  


 

[cpp] view plaincopy
 
    1. /// Capture.cpp for class CCapture implement  
    2. //  
    3. ///  
    4. ////////////////////////////////////////  
    5.   
    6. #include "StdAfx.h"  
    7. #include "Capture.h"  
    8. #include <atlconv.h>  
    9. #include "VideoChatDlg.h"  
    10. #include "yuv2bmp.h"  
    11.   
    12. #ifndef srelease  
    13. #define srelease(x) if (NULL != x)\  
    14. {\  
    15.     x->Release();\  
    16.     x = NULL;\  
    17. }  
    18. #endif  
    19.   
    20. #ifndef MAX_PATH  
    21. #define  MAX_PATH 1024  
    22. #endif  
    23. BOOL bOneShot = FALSE; // 全局变量  
    24. capCallBackFunc fun;  
    25.   
    26. class CSampleGrabberCB : public ISampleGrabberCB  
    27. {  
    28. public:  
    29.     long lWidth;  
    30.     long lHeight;  
    31.     CCapture *pCap;  
    32.     TCHAR m_szFileName[MAX_PATH]; // 位图文件名称  
    33.     CSampleGrabberCB(){  
    34.         strcpy(m_szFileName, ".\\sample.bmp");  
    35.     }  
    36.     STDMETHODIMP_(ULONG) AddRef() { return 2; }   
    37.     STDMETHODIMP_(ULONG) Release() { return 1; }   
    38.     STDMETHODIMP QueryInterface(REFIID riid, void ** ppv){   
    39.         if( riid == IID_ISampleGrabberCB || riid == IID_IUnknown ){    
    40.         *ppv = (void *) static_cast<ISampleGrabberCB*> ( this );   
    41.             return NOERROR;   
    42.         }    
    43.         return E_NOINTERFACE;   
    44.     }   
    45.     STDMETHODIMP SampleCB( double SampleTime, IMediaSample * pSample ){   
    46.         return 0;   
    47.     }   
    48.     STDMETHODIMP BufferCB( double dblSampleTime, BYTE * pBuffer, long lBufferSize ){   
    49.         if( !bOneShot )  
    50.             return 0;  
    51.         if (!pBuffer)  
    52.         {  
    53.             AfxMessageBox(_T("Save Bmp File Failure!"));  
    54.             return E_POINTER;   
    55.         }  
    56.         if (pBuffer != NULL && pCap)  
    57.         {  
    58. //          BYTE *rgb = new BYTE[lWidth*lHeight*3];   
    59. //          YUV422_C_RGB(pBuffer,rgb, (int)lHeight, (int)lWidth);  
    60. //          outBmpBuf(pBuffer, pCap);  // 将一帧图像数据传给显示函数  
    61. //          ((CVideoNetDlg *)pCap->m_dlgParent)->SendVideo(pBuffer, (int)lBufferSize);  
    62.         }  
    63.     //  SaveBitmap(pBuffer, lBufferSize);  // 保存成位图文件  
    64. //      bOneShot = FALSE; // 停止捕获图像  
    65. //      AfxMessageBox(_T("Get bmp data success."));  
    66.         return 0;   
    67.     }  
    68.     void outBmpBuf(BYTE *buf, CCapture* cap)  
    69.     {  
    70.         cap->calFunc(buf);  
    71.     }  
    72.     // 创建位图文件  
    73.     BOOL SaveBitmap(BYTE *pBuffer, long lBufferLen)  
    74.     {  
    75.         HANDLE hf = CreateFile(m_szFileName, GENERIC_WRITE,   
    76.             FILE_SHARE_READ, NULL, CREATE_ALWAYS, NULL, NULL);  
    77.         if (hf == INVALID_HANDLE_VALUE) return 0;  
    78.         // 写文件头  
    79.         BITMAPFILEHEADER fileheader;  
    80.         ZeroMemory(&fileheader, sizeof(BITMAPFILEHEADER));  
    81.         fileheader.bfType = ‘MB‘;  
    82.         fileheader.bfSize = sizeof(BITMAPFILEHEADER)+sizeof(BITMAPINFOHEADER)+lBufferLen;  
    83.         fileheader.bfOffBits = sizeof(BITMAPFILEHEADER)+sizeof(BITMAPINFOHEADER);  
    84.         DWORD dwWritter = 0;  
    85.         WriteFile(hf, &fileheader, sizeof(BITMAPFILEHEADER), &dwWritter, NULL);  
    86.         // 写文图格式  
    87.         BITMAPINFOHEADER infoHeader;  
    88.         ZeroMemory(&infoHeader, sizeof(BITMAPINFOHEADER));  
    89.         infoHeader.biSize = sizeof(BITMAPINFOHEADER);  
    90.         infoHeader.biSizeImage = lBufferLen;  
    91.         infoHeader.biWidth = lWidth;  
    92.         infoHeader.biHeight = lHeight;  
    93.         infoHeader.biBitCount = 24;  
    94.         WriteFile(hf, &infoHeader, sizeof(BITMAPINFOHEADER), &dwWritter, NULL);  
    95.         // 写位图数据  
    96.         WriteFile(hf, pBuffer, lBufferLen, &dwWritter, NULL);  
    97.         CloseHandle(hf);  
    98.         MessageBox(NULL, _T("Save bmp file succeed!"), _T("warn"), MB_OK|MB_ICONINFORMATION);  
    99.         return 0;  
    100.     }  
    101. };  
    102.   
    103. /////////////////////////////////////////////  
    104. /// for class CCapture‘s Function  
    105. ///  
    106. ////////////////////////////////////////////////  
    107. CSampleGrabberCB samCB;  
    108. CCapture::CCapture()  
    109. {  
    110.     CoInitialize(NULL);  // 初始化COM库  
    111.     m_hWnd = NULL;  
    112.     m_pVideoCap = NULL;  
    113.     m_pAudioCap = NULL;  
    114.     m_pCapGB = NULL;  
    115.     m_pGB = NULL;  
    116.     m_pMC = NULL;  
    117.     m_pMux = NULL;  
    118.     m_pVW = NULL;  
    119.     m_pGrabber = NULL;  
    120.     m_dlgParent = NULL;  
    121. }  
    122. bool CCapture::m_bRecording = false;  
    123.   
    124. CCapture::~CCapture()  
    125. {  
    126.     if (m_pMC) m_pMC->Stop();  
    127.     if (m_pVW)  
    128.     {  
    129.         m_pVW->put_Owner(NULL);  
    130.         m_pVW->put_Visible(OAFALSE);  
    131.     }  
    132.     m_hWnd = NULL;  
    133.     srelease(m_pVideoCap);  
    134.     srelease(m_pGB);  
    135.     srelease(m_pCapGB);  
    136.     srelease(m_pMC);  
    137.     srelease(m_pVW);  
    138.     m_bRecording = false;  
    139.     CoUninitialize(); // 释放COM库  
    140. }  
    141.   
    142. int CCapture::EnumDevice( HWND hCmbList, DeviceType deviceType )  
    143. {  
    144.     if (hCmbList == NULL) return -1;  
    145.     int id = 0;  
    146.     /////枚举捕获设备  
    147.     ICreateDevEnum *pCreateDevEnum;  
    148.     HRESULT hr = CoCreateInstance(CLSID_SystemDeviceEnum,  
    149.         NULL,  
    150.         CLSCTX_INPROC_SERVER,  
    151.         IID_ICreateDevEnum,  
    152.         (LPVOID *)&pCreateDevEnum);  
    153.     if ( hr != NOERROR) return -1;  
    154.     //////// 获取视频类的枚举器  
    155.     IEnumMoniker *pEm;           //枚举监控器接口  
    156.     if (deviceType == DTypeVideo)  
    157.         hr = pCreateDevEnum->CreateClassEnumerator(CLSID_VideoInputDeviceCategory, &pEm, 0);  
    158.     // 如果获取音频类的枚举器 用下面的代码  
    159.     else   
    160.          hr = pCreateDevEnum->CreateClassEnumerator(CLSID_AudioInputDeviceCategory, &pEm, 0);  
    161. //  if (hr != NOERROR) return -1;  
    162.     if (!pEm || FAILED(hr)) return -1;  
    163.     ////////////////////////  
    164.     pEm->Reset();   // 类型枚举器复位  
    165.     ULONG cFetched;  
    166.     IMoniker *pM;    // 监控器接口指针  
    167.     while(hr = pEm->Next(1, &pM, &cFetched), hr == S_OK)  
    168.     {  
    169.         IPropertyBag *pBag;   // 属性页接口指针  
    170.         hr = pM->BindToStorage(0, 0, IID_IPropertyBag, (void **)&pBag);  
    171.         // 获取设备属性页  
    172.         if (SUCCEEDED(hr))  
    173.         {  
    174.             VARIANT var;  
    175.             var.vt = VT_BSTR;       // 保存的是二进制数据  
    176.             // 获取firendlyName 形式的信息  
    177.             hr = pBag->Read(L"FriendlyName", &var, NULL);  
    178.             if (hr == NOERROR) // 获取成功  
    179.             {  
    180.                 id++;  
    181.                 char szDeviceName[256] = {0};  
    182.                 WideCharToMultiByte(CP_ACP, 0, var.bstrVal, -1, szDeviceName,80, NULL, NULL);  // 字符串编码转换UNICODE TO ANSI  
    183.                 ::SendMessage(hCmbList, CB_ADDSTRING, 0, (LPARAM)szDeviceName);//添加到组合列表框  
    184.                 SysFreeString(var.bstrVal);  //释放资源，特别要注意  
    185.             }  
    186.             pBag->Release();  
    187.         }  
    188.         pM->Release();  
    189.     }  
    190.     return 0;  
    191. }  
    192.   
    193. void CCapture::ResizeVideoWindow()  
    194. {  
    195.     if (m_pVW)  
    196.     {  
    197.         // 让图像充满整个指定窗口  
    198.         CRect rc;  
    199.         ::GetClientRect(m_hWnd, &rc);  
    200.         m_pVW->SetWindowPosition(0, 0, rc.right, rc.bottom);  
    201.     }  
    202. }  
    203.   
    204. HRESULT CCapture::SetupVideoWindow()  
    205. {  
    206.     HRESULT hr;  
    207.     //m_hWnd为类CCapture的成员变量，在使用该函数前须初始化  
    208.     hr = m_pVW->put_Visible(OAFALSE);  // 视频窗口不可见  
    209.     hr = m_pVW->put_Owner((OAHWND)m_hWnd);  // 设置视频窗口  
    210.     if (FAILED(hr)) return hr;  
    211.     hr = m_pVW->put_WindowStyle(WS_CHILD | WS_CLIPCHILDREN); //设置窗口类型  
    212.     if (FAILED(hr)) return hr;  
    213.     ResizeVideoWindow();   // 更改窗口大小  
    214.     hr = m_pVW->put_Visible(OATRUE);  // 显示视频窗口  
    215.     return hr;  
    216. }  
    217.   
    218. HRESULT CCapture::InitCaptureGraphBuilder()  
    219. {  
    220.     HRESULT hr;  
    221.     //创建IGraphBuilder接口(滤波器链表管理器) m_pGB  
    222.     hr = CoCreateInstance(CLSID_FilterGraph, NULL,  
    223.         CLSCTX_INPROC, IID_IGraphBuilder, (void **)&m_pGB);  
    224.     if (FAILED(hr)) return hr;  
    225.     //创建ICaptureGraphBuilder2接口（增强型捕获滤波器链表管理器）m_pCapGB  
    226.     hr = CoCreateInstance(CLSID_CaptureGraphBuilder2, NULL,  
    227.         CLSCTX_INPROC, IID_ICaptureGraphBuilder2, (void **)&m_pCapGB);  
    228.     if (FAILED(hr)) return hr;  
    229.     // 创建抓取图片滤波器  
    230.     if (m_pGrabber){  
    231.         m_pGrabber.Release();  
    232.         m_pGrabber = NULL;  
    233.     }  
    234.   
    235.     hr = CoCreateInstance(CLSID_SampleGrabber, NULL,CLSCTX_INPROC_SERVER, IID_ISampleGrabber, (void **)&m_pGrabber);  
    236. //  hr = m_pGrabber.CoCreateInstance( CLSID_SampleGrabber );   
    237.     if (FAILED(hr)) return hr;  
    238.     // 初始化滤波器链表管理器IGraphBuilder  
    239.     m_pCapGB->SetFiltergraph(m_pGB);  
    240.     // 查询媒体控制接口  
    241.     hr = m_pGB->QueryInterface(IID_IMediaControl, (void **)&m_pMC);  
    242.     if (FAILED(hr))  return hr;  
    243.     // 查询视频窗口接口  
    244.     hr = m_pGB->QueryInterface(IID_IVideoWindow, (LPVOID *)&m_pVW);  
    245.     if (FAILED(hr)) return hr;  
    246.     /////  
    247.     return hr;  
    248. }  
    249.   
    250. HRESULT CCapture::Preview( int iDevVideoID, HWND hVideo, int iDevAudioID /*= 0*/, HWND hAudio /*= NULL*/ )  
    251. {  
    252.     HRESULT hr;  
    253.     if (m_pMC)  
    254.         m_pMC->Stop();  
    255.     m_bRecording = false;  
    256.     // 初始化视频捕获滤波器链表管理器  
    257.     hr = InitCaptureGraphBuilder();  
    258.     if (FAILED(hr)) return hr;  
    259.   
    260.     // 把指定的视频采集设备与滤波器捆绑  
    261.     if (BindFilter(iDevVideoID, &m_pVideoCap, DTypeVideo))   
    262.     {  
    263.         // 把滤波器添加到滤波器链表中  
    264.         hr = m_pGB->AddFilter(m_pVideoCap, L"Video Capture Filter");  
    265.         if (FAILED(hr)) return hr;  
    266.     }  
    267.     else return FALSE;  
    268.   
    269.     if (BindFilter(iDevAudioID, &m_pAudioCap, DTypeAudio))  
    270.     {  
    271.         hr = m_pGB->AddFilter(m_pAudioCap, L"Audio Capture Filter");  
    272.         if (FAILED(hr))  
    273.         {  
    274.             MessageBox(NULL, _T("绑定音频设备失败!"), _T("系统提示"), MB_OK|MB_ICONINFORMATION);  
    275.         //  return hr;  
    276.         }  
    277.     }  
    278.     else  
    279.     {  
    280.         MessageBox(NULL, _T("绑定音频设备失败!"), _T("系统提示"), MB_OK|MB_ICONINFORMATION);  
    281.     //  return FALSE;  
    282.     }  
    283.     // 如果我们想抓取24位的RGB图片，如下设置媒体图片类型  
    284.     CComQIPtr<IBaseFilter, &IID_IBaseFilter> pGrabBase(m_pGrabber);  
    285.   
    286.     AM_MEDIA_TYPE mediaType;  
    287.     VIDEOINFOHEADER vih;  
    288.   
    289.     IAMStreamConfig* pConfig = NULL;  
    290.     m_pCapGB->FindInterface(&PIN_CATEGORY_CAPTURE, &MEDIATYPE_Video, m_pVideoCap, IID_IAMStreamConfig, (void**)&pConfig);  
    291.     // 设置视频格式  
    292.     ZeroMemory(&mediaType, sizeof(AM_MEDIA_TYPE));  
    293.     vih.bmiHeader.biWidth = IMG_WIDTH;  
    294.     vih.bmiHeader.biHeight = IMG_HEIGHT;  
    295.     vih.bmiHeader.biSizeImage = IMG_HEIGHT*IMG_WIDTH*3;  
    296.     mediaType.pbFormat = (BYTE *)(&vih);  
    297.     mediaType.cbFormat = sizeof(VIDEOINFOHEADER);  
    298.     mediaType.subtype = MEDIASUBTYPE_YUY2;  
    299.     mediaType.majortype = MEDIATYPE_Video;  
    300.     mediaType.formattype = FORMAT_VideoInfo;  
    301.     hr = pConfig->SetFormat(&mediaType);  
    302.     hr = m_pGrabber->SetMediaType(&mediaType);   
    303.     if( FAILED( hr ) ){   
    304.         AfxMessageBox("Fail to set media type!");   
    305.         return hr;   
    306.     }  
    307.       
    308.     hr = m_pGB->AddFilter(pGrabBase, L"SampleGrabber");  
    309.     if (FAILED(hr)) return hr;  
    310.     // 渲染媒体， 把链表中滤波器链接起来  
    311.     hr = m_pCapGB->RenderStream(&PIN_CATEGORY_PREVIEW, &MEDIATYPE_Audio, m_pAudioCap, NULL, NULL);  
    312.     hr = m_pCapGB->RenderStream(&PIN_CATEGORY_PREVIEW, &MEDIATYPE_Video, m_pVideoCap, pGrabBase, NULL);  
    313.     if (FAILED(hr))  
    314.         hr = m_pCapGB->RenderStream(&PIN_CATEGORY_CAPTURE, &MEDIATYPE_Video, m_pVideoCap, pGrabBase, NULL);  
    315.     if( FAILED( hr ) ){  
    316.         AfxMessageBox(_T("Can’t build the graph"));  
    317.         return hr;  
    318.     }  
    319.   
    320.     ////////// 设置图片捕获数据  
    321.     hr = m_pGrabber->GetConnectedMediaType( &mediaType );  
    322.     if ( FAILED( hr) ){  
    323.         AfxMessageBox(_T("Failt to read the connected media type"));  
    324.         return hr;  
    325.     }  
    326.     VIDEOINFOHEADER * pVih = (VIDEOINFOHEADER*)mediaType.pbFormat;  
    327.     samCB.lWidth = pVih->bmiHeader.biWidth;  
    328.     samCB.lHeight = pVih->bmiHeader.biHeight;  
    329.     samCB.pCap = (CCapture *)this;  
    330.     FreeMediaType(mediaType);  
    331.     hr = m_pGrabber->SetBufferSamples( TRUE );  // 如果此处为false 第一次抓取图片时失败(不用回调方式)  
    332.     hr = m_pGrabber->SetOneShot( FALSE );       
    333.     hr = m_pGrabber->SetCallback( &samCB, 1 );  
    334.     SetOnShot(TRUE);// ture 时开始捕获视频帧数据  
    335.     // 设置视频显示窗口  
    336.     m_hWnd = hVideo;  
    337.     SetupVideoWindow(); // 设置显示窗口  
    338.     hr = m_pMC->Run();  // 开始采集、预览视频，并在指定窗口显示  
    339.     if (FAILED(hr))   
    340.     {  
    341.         MessageBox(NULL, _T("请检查该设备是否被占用!"), _T("系统提示"), MB_OK|MB_ICONINFORMATION);  
    342.         return hr;  
    343.     }  
    344.     return S_OK;  
    345. }  
    346.   
    347. #if 1  // avi video format  
    348. HRESULT CCapture::CaptureVideo( CString inFileName )  // 录制视频  
    349. {  
    350.     HRESULT hr = 0;  
    351.     DWORD dwId;  
    352.     HANDLE hThread;  
    353.     m_bRecording = false;  
    354.     m_pMC->Stop();   // 先停止视频采集  
    355.   
    356.     // 设置文件名，注意第二个参数类型  
    357.     hr = m_pCapGB->SetOutputFileName(&MEDIASUBTYPE_Avi, inFileName.AllocSysString(), &m_pMux, NULL);  
    358.   
    359.     //渲染媒体 连接捕获器和AVI Muxer过滤器   
    360.     hr = m_pCapGB->RenderStream(&PIN_CATEGORY_CAPTURE, &MEDIATYPE_Video, m_pVideoCap, NULL, m_pMux);  
    361.     hr = m_pCapGB->RenderStream(&PIN_CATEGORY_CAPTURE, &MEDIATYPE_Audio, m_pAudioCap, NULL, m_pMux);  
    362.     //设置音频流为主流   
    363.     IConfigAviMux *pConfigMux;  
    364.     m_pMux->QueryInterface(IID_IConfigAviMux, (void **)&pConfigMux);  
    365.     hr = pConfigMux->SetMasterStream(1);   // 0 为视频  1为音频  
    366.   
    367.     pConfigMux->Release();  
    368.     m_pMux->Release();  
    369.     m_bRecording = true;  
    370.     hThread = CreateThread(NULL, 0,  
    371.         (LPTHREAD_START_ROUTINE)ThreadFunDrawText,  
    372.         (LPVOID)m_hWnd,  
    373.         0, &dwId);  
    374.     m_pMC->Run();  // 恢复视频采集，同时写入文件  
    375.       
    376.     return hr;  
    377. }  
    378. #else  // mpeg4 format video  
    379. HRESULT CCapture::CaptureVideo(CString inFileName)  
    380. {  
    381.     HRESULT hr=0;  
    382.       
    383.     m_pMC->Stop();  
    384.       
    385.     m_pGB->AddFilter(m_pXviDCodec,L"99 Xvid MPEG-4 Codec");  
    386.     m_pXviDCodec->Release();  
    387.       
    388.     hr = m_pCapGB->SetOutputFileName(&MEDIASUBTYPE_Avi, inFileName.AllocSysString(), &m_pMux, NULL );  
    389.     hr = ConnectFilters(m_pGB,m_pSmartTee_1,m_pXviDCodec, 0);    //0，连接capture引脚  
    390.     hr = ConnectFilters(m_pGB,m_pXviDCodec,m_pMux, 2);    //2，默认自然连接  
    391.     m_pMux->Release();  
    392.       
    393.     m_pMC->Run();  
    394.       
    395.     return hr;  
    396. }  
    397. #endif  
    398.   
    399. ///////////////////////////////  
    400.   
    401. HRESULT CCapture::CaptureImage() // 抓取并显示图像  
    402. { // 采用CB接口回调函数存储图片  
    403.     bOneShot = TRUE;  
    404.     return 0;  
    405. }  
    406.   
    407. HRESULT CCapture::CaptureImage( CString inFileName ) // 抓取图像  
    408. {  
    409.     HRESULT hr;  
    410.     AM_MEDIA_TYPE mediaType;  
    411.     hr = m_pGrabber->GetConnectedMediaType(&mediaType);  
    412.     if (FAILED(hr))  return hr;  
    413.     VIDEOINFOHEADER *pVih;  
    414.     if (mediaType.formattype == FORMAT_VideoInfo &&   
    415.         (mediaType.cbFormat >= sizeof(VIDEOINFOHEADER)) &&  
    416.         mediaType.pbFormat != NULL)  
    417.     {  
    418.         pVih = (VIDEOINFOHEADER *)mediaType.pbFormat;  
    419.     }  
    420.     else       
    421.         return VFW_E_INVALIDMEDIATYPE;  
    422.   
    423. //  hr = m_pGrabber->SetOneShot(TRUE);  
    424.     if (SUCCEEDED(m_pGrabber->SetBufferSamples(TRUE)) )  // 设置为缓冲形式)  
    425.     {  
    426.         long cbBuffer = 0;  
    427.         hr = m_pGrabber->GetCurrentBuffer(&cbBuffer, NULL);  
    428.         BYTE *pBuffer = new BYTE[cbBuffer];  
    429.         if (!pBuffer) return -1;  
    430.         // 获取一帧媒体的数据  
    431.         hr = m_pGrabber->GetCurrentBuffer(&cbBuffer, (long *)pBuffer);  
    432.         if (FAILED(hr))  return hr;  
    433. //          if (pBuffer != NULL)  
    434. //          {  
    435. //              calFunc(pBuffer);   // 将一帧图像数据传给显示函数  
    436. //      }  
    437.   
    438.         ///-------------------------测试所得数据是rgb格式还是yuv格式--------  
    439.         long n1,n2;  
    440.         int datalen = IMG_WIDTH*IMG_HEIGHT*3;  
    441.         BYTE *rgb = new BYTE[datalen];  
    442.           
    443.         YUV422_C_RGB(pBuffer,rgb, IMG_HEIGHT, IMG_WIDTH);  
    444.         n1 = strlen((char *)pBuffer);  
    445.         n2 = strlen((char *)rgb);  
    446. //      ((CVideoNetDlg *)(m_dlgParent))->SendVideo((BYTE *)pBuffer, (int)cbBuffer);  
    447.         ///------------------------------------------------------------------  
    448. ///////////////////////////////////////////////////////////////  
    449.         // Create a file to hold the bitmap  
    450.         HANDLE hf = CreateFile(inFileName, GENERIC_WRITE, FILE_SHARE_READ,    
    451.             NULL, CREATE_ALWAYS, NULL, NULL );  
    452.           
    453.         if( hf == INVALID_HANDLE_VALUE ){  
    454.             MessageBox(NULL, _T("Create bmp file failure!"), _T(""), MB_OK|MB_ICONINFORMATION);  
    455.             return 0;  
    456.         }  
    457.           
    458.         // Write out the file header  
    459.         //  
    460.         // 信息头  
    461.         BITMAPFILEHEADER bfh;  
    462.         memset( &bfh, 0, sizeof( bfh ) );  
    463.         bfh.bfType = ‘MB‘;  
    464.         bfh.bfSize = sizeof( bfh ) + cbBuffer + sizeof( BITMAPINFOHEADER );  
    465.         bfh.bfOffBits = sizeof( BITMAPINFOHEADER ) + sizeof( BITMAPFILEHEADER );  
    466.           
    467.         DWORD Written = 0;  
    468.         WriteFile( hf, &bfh, sizeof( bfh ), &Written, NULL );  
    469.           
    470.         // Write the bitmap format  
    471.         //文件头  
    472.         BITMAPINFOHEADER bih;  
    473.         memset( &bih, 0, sizeof( bih ) );  
    474.         bih.biSize = sizeof( bih );  
    475.         bih.biWidth = pVih->bmiHeader.biWidth;  
    476.         bih.biHeight = pVih->bmiHeader.biHeight;  
    477.         bih.biPlanes = 1;  
    478.         bih.biBitCount = 24;  
    479.           
    480.         Written = 0;  
    481.         WriteFile( hf, &bih, sizeof( bih ), &Written, NULL );  
    482.           
    483.         // Write the bitmap bits  
    484.         //  
    485.         Written = 0;  
    486.         WriteFile( hf, rgb, datalen, &Written, NULL );       
    487.         CloseHandle( hf );  
    488.         delete pBuffer;  
    489.         MessageBox(NULL, _T("Save photo succeeded!"), _T("抓取图片提示"), MB_OK|MB_ICONINFORMATION);  
    490.     }  
    491.     m_pGrabber->SetOneShot(FALSE);  
    492.     m_pGrabber->SetBufferSamples(FALSE);  
    493.     FreeMediaType(mediaType);  
    494.     return 0;  
    495. }  
    496.   
    497. bool CCapture::BindFilter( int iDeviceID, IBaseFilter **pOutFilter, DeviceType deviceType )  
    498. {  
    499.     if (iDeviceID < 0) return false;  
    500.     // 枚举所有的视频设备  
    501.     ICreateDevEnum *pCreateDevEnum;  
    502.     //生成设备枚举器pCreateDevEnum  
    503.     HRESULT hr = CoCreateInstance(CLSID_SystemDeviceEnum,  
    504.         NULL, CLSCTX_INPROC_SERVER, IID_ICreateDevEnum, (void **)&pCreateDevEnum);  
    505.     if (hr != NOERROR) return false;  
    506.     IEnumMoniker *pEM;  
    507.     // 创建视频输入设备类枚举器  
    508.     if (deviceType == DTypeVideo)  
    509.         hr = pCreateDevEnum->CreateClassEnumerator(CLSID_VideoInputDeviceCategory, &pEM, 0);  
    510.     // 音频设备枚举器  
    511.     else  
    512.         hr = pCreateDevEnum->CreateClassEnumerator(CLSID_AudioInputDeviceCategory, &pEM, 0);  
    513.     if (hr != NOERROR) return false;  
    514.     pEM->Reset();  // 复位该设备  
    515.     ULONG cFetched;  
    516.     IMoniker *pM;  
    517.     int indexDev = 0;  
    518.     // 获取设备  
    519.     while(hr = pEM->Next(1, &pM, &cFetched), hr == S_OK, indexDev <= iDeviceID)  
    520.     {  
    521.         IPropertyBag *pBag;  
    522.         // 获取该设备属性集  
    523.         hr = pM->BindToStorage(0,0,IID_IPropertyBag,(void **)&pBag);  
    524.         if (SUCCEEDED(hr))  
    525.         {  
    526.             VARIANT var;  
    527.             var.vt = VT_BSTR;  
    528.             hr = pBag->Read(L"FriendlyName", &var, NULL);  
    529.             if (hr == NOERROR)  
    530.             {  
    531.                 // 采集设备与捕获滤波器捆绑  
    532.                 if (indexDev == iDeviceID) pM->BindToObject(0, 0, IID_IBaseFilter, (void **)pOutFilter);  
    533.                 SysFreeString(var.bstrVal);  
    534.             }  
    535.             pBag->Release();  
    536.         }  
    537.         pM->Release();  
    538.         indexDev++;  
    539.     }  
    540.     return true;  
    541. }  
    542.   
    543. void CCapture::SetCameraFormat( HWND hwndParent ) // 设置视频格式  
    544. {  
    545.     HRESULT hr;  
    546.     IAMStreamConfig *pSC; // 流配置接口  
    547.     ISpecifyPropertyPages *pSpec; //属性页接口  
    548.     m_pMC->Stop();  // 只有停止后才能进行引脚属性的设置  
    549.     m_bRecording = false;  
    550.     // 首先查询捕获CAPTURE、视频Video接口  
    551.     hr = m_pCapGB->FindInterface(&PIN_CATEGORY_CAPTURE, &MEDIATYPE_Video,  
    552.         m_pVideoCap, IID_IAMStreamConfig, (void **)&pSC);  
    553.   
    554.     CAUUID cauuid; // 所有属性页结构体  
    555.     hr = pSC->QueryInterface(IID_ISpecifyPropertyPages, (void **)&pSpec);  
    556.     if (hr == S_OK)  
    557.     {  
    558.         // 显示属性页窗口  
    559.         hr = pSpec->GetPages(&cauuid);  // 获取所有属性页  
    560.         hr = OleCreatePropertyFrame(hwndParent, 30, 30, NULL, 1,  
    561.             (IUnknown **)&pSC, cauuid.cElems, (GUID *)cauuid.pElems, 0, 0, NULL);  
    562.         // 释放内存资源  
    563.         CoTaskMemFree(cauuid.pElems);  
    564.         pSpec->Release();  
    565.         pSC->Release();  
    566.     }  
    567.     // 恢复运行  
    568.     m_pMC->Run();          
    569. }  
    570.   
    571. void CCapture::SetCameraFilter( HWND hwndParent ) // 设置图像各参数设置  
    572. {  
    573.     HRESULT hr = 0;  
    574.     ISpecifyPropertyPages *pSpec;  
    575.     hr = m_pVideoCap->QueryInterface(IID_ISpecifyPropertyPages, (void **)&pSpec);  
    576.     if (SUCCEEDED(hr))  
    577.     {  
    578.         // 获取滤波器名称和IUnknown 接口指针  
    579.         FILTER_INFO FilterInfo;  
    580.         hr = m_pVideoCap->QueryFilterInfo(&FilterInfo);  
    581.         IUnknown *pFilterUnk;  
    582.         m_pVideoCap->QueryInterface(IID_IUnknown, (void **)&pFilterUnk);  
    583.         // 显示该页  
    584.         CAUUID caGUID;  
    585.         pSpec->GetPages(&caGUID);  
    586.         OleCreatePropertyFrame(hwndParent,  
    587.             0, 0,  
    588.             FilterInfo.achName,  
    589.             1,  
    590.             &pFilterUnk,  
    591.             caGUID.cElems,  
    592.             caGUID.pElems,  
    593.             0,  
    594.             0, NULL);  
    595.         // 释放内存资源  
    596.         CoTaskMemFree(caGUID.pElems);  
    597.         pFilterUnk->Release();  
    598.         FilterInfo.pGraph->Release();  
    599.         pSpec->Release();  
    600.     }  
    601. }  
    602.   
    603. void CCapture::StopCapture()  
    604. {  
    605.     m_pMC->Stop();  
    606. }  
    607.   
    608. UINT CCapture::ThreadFunDrawText( LPVOID lParam )  
    609. {  
    610.     HWND hwnd = (HWND)lParam;  
    611.     if (hwnd == NULL) return -1;  
    612.     HDC hdc = GetDC(hwnd);  
    613.     CRect rcDraw, rcTime;  
    614.     CTime time, time0;  
    615.     CTimeSpan timespan;  
    616.     CString strTime;  
    617.     CBrush br;  
    618.     time0 = CTime::GetCurrentTime();  
    619.     br.CreateSolidBrush(RGB(255,0,0));  
    620.     GetClientRect(hwnd, &rcDraw);  
    621.     rcTime = rcDraw;  
    622.     rcTime.bottom = rcTime.top + 30;  
    623.     rcDraw.top = rcDraw.bottom - 30;  
    624.     SelectObject(hdc, &br);  
    625.     SetTextColor(hdc, 0x0000ff);  
    626.     SetBkMode(hdc, TRANSPARENT);  
    627.     while(m_bRecording)  
    628.     {  
    629.         time = CTime::GetCurrentTime();  
    630.         timespan = time - time0;  
    631.         strTime = time.Format(_T(" %Y-%m-%d 星期%w %H:%M:%S"));  
    632.         DrawText(hdc, strTime, strTime.GetLength(), &rcTime, DT_VCENTER|DT_LEFT|DT_SINGLELINE);  
    633.         strTime = timespan.Format(_T("%H:%M:%S "));  
    634.         strTime = _T("●录制 ") + strTime;  
    635.         DrawText(hdc, strTime, strTime.GetLength(), &rcDraw, DT_VCENTER|DT_RIGHT|DT_SINGLELINE);  
    636.     }  
    637.     return 0;  
    638. }  
    639.   
    640. void CCapture::FreeMediaType(AM_MEDIA_TYPE &mt)  
    641. {  
    642.     if (mt.cbFormat != 0)  
    643.     {  
    644.         CoTaskMemFree((PVOID)mt.pbFormat);  
    645.         mt.cbFormat = 0;  
    646.         mt.pbFormat = NULL;  
    647.     }  
    648.     if (mt.pUnk != NULL)  
    649.     {  
    650.         mt.pUnk->Release();  
    651.         mt.pUnk = NULL;  
    652.     }  
    653. }  
    654.   
    655. void CCapture::SetOnShot( BOOL bFlag )  
    656. {  
    657.     bOneShot = bFlag;  
    658. }  
    659.   
    660. void CCapture::SetCallBKFun( capCallBackFunc f )  
    661. {  
    662.     this->calFunc = f;  
    663.     samCB.pCap = static_cast<CCapture *>(this);  
    664. }  
    665.   
    666. void CCapture::SetParent( CDialog *pdlg )  
    667. {  
    668.     m_dlgParent = pdlg;  
    669. }  

CCapture directshow 视频捕获类,码迷,mamicode.com

CCapture directshow 视频捕获类

标签：style   blog   class   code   tar   ext   width   strong   color   string   get   

原文地址：http://www.cnblogs.com/lidabo/p/3701654.html
踩
(0)
赞
(0)
   
举报
评论 一句话评论（0）
登录后才能评论！
分享档案
更多>
2021年07月29日 (22)
2021年07月28日 (40)
2021年07月27日 (32)
2021年07月26日 (79)
2021年07月23日 (29)
2021年07月22日 (30)
2021年07月21日 (42)
2021年07月20日 (16)
2021年07月19日 (90)
2021年07月16日 (35)
周排行
mamicode.com排行更多图片
更多
友情链接
兰亭集智   国之画   百度统计   站长统计   阿里云   chrome插件   新版天听网
关于我们 - 联系我们 - 留言反馈
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码！