开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > 图形图像 > OpenCV > 查看源码
CameraDS.cpp
资源名称：mymfc_v2.rar　[点击查看]
上传用户：wjt888999
上传日期：2022-08-11
资源大小：5225k
文件大小：11k
源码类别：

OpenCV

开发平台：

Visual C++

CameraDS.cpp：源码内容
  1. //////////////////////////////////////////////////////////////////////
  2. // Video Capture using DirectShow
  3. // Author: Shiqi Yu (shiqi.yu@gmail.com)
  4. // Thanks to:
  5. // HardyAI@OpenCV China
  6. // flymanbox@OpenCV China (for his contribution to function CameraName, and frame width/height setting)
  7. // Last modification: April 9, 2009
  8. //////////////////////////////////////////////////////////////////////
  11. //////////////////////////////////////////////////////////////////////
  12. // 使用说明：
  13. //   1. 将CameraDS.h CameraDS.cpp以及目录DirectShow复制到你的项目中
  14. //   2. 菜单 Project->Settings->Settings for:(All configurations)->C/C++->Category(Preprocessor)->Additional include directories
  15. //      设置为 DirectShow/Include
  16. //   3. 菜单 Project->Settings->Settings for:(All configurations)->Link->Category(Input)->Additional library directories
  17. //      设置为 DirectShow/Lib
  18. //////////////////////////////////////////////////////////////////////
  20. // CameraDS.cpp: implementation of the CCameraDS class.
  21. //
  22. //////////////////////////////////////////////////////////////////////
  24. #include "stdafx.h"
  25. #include "CameraDS.h"
  27. #pragma comment(lib,"Strmiids.lib") 
  28. //////////////////////////////////////////////////////////////////////
  29. // Construction/Destruction
  30. //////////////////////////////////////////////////////////////////////
  32. CCameraDS::CCameraDS()
  33. {
  34. m_bConnected = false;
  35. m_nWidth = 0;
  36. m_nHeight = 0;
  37. m_bLock = false;
  38. m_bChanged = false;
  39. m_pFrame = NULL;
  40. m_nBufferSize = 0;
  42. m_pNullFilter = NULL;
  43. m_pMediaEvent = NULL;
  44. m_pSampleGrabberFilter = NULL;
  45. m_pGraph = NULL;
  47. CoInitialize(NULL);
  48. }
  50. CCameraDS::~CCameraDS()
  51. {
  52. CloseCamera();
  53. CoUninitialize();
  54. }
  56. void CCameraDS::CloseCamera()
  57. {
  58. if(m_bConnected)
  59. m_pMediaControl->Stop();
  61. m_pGraph = NULL;
  62. m_pDeviceFilter = NULL;
  63. m_pMediaControl = NULL;
  64. m_pSampleGrabberFilter = NULL;
  65. m_pSampleGrabber = NULL;
  66. m_pGrabberInput = NULL;
  67. m_pGrabberOutput = NULL;
  68. m_pCameraOutput = NULL;
  69. m_pMediaEvent = NULL;
  70. m_pNullFilter = NULL;
  71. m_pNullInputPin = NULL;
  74. if (m_pFrame)
  75. cvReleaseImage(&m_pFrame);
  77. m_bConnected = false;
  78. m_nWidth = 0;
  79. m_nHeight = 0;
  80. m_bLock = false;
  81. m_bChanged = false;
  82. m_nBufferSize = 0;
  83. }
  85. bool CCameraDS::OpenCamera(int nCamID, bool bDisplayProperties, int nWidth, int nHeight)
  86. {
  88. HRESULT hr = S_OK;
  90. CoInitialize(NULL);
  91. // Create the Filter Graph Manager.
  92. hr = CoCreateInstance(CLSID_FilterGraph, NULL, CLSCTX_INPROC,
  93. IID_IGraphBuilder, (void **)&m_pGraph);
  95. hr = CoCreateInstance(CLSID_SampleGrabber, NULL, CLSCTX_INPROC_SERVER, 
  96. IID_IBaseFilter, (LPVOID *)&m_pSampleGrabberFilter);
  98. hr = m_pGraph->QueryInterface(IID_IMediaControl, (void **) &m_pMediaControl);
  99. hr = m_pGraph->QueryInterface(IID_IMediaEvent, (void **) &m_pMediaEvent);
  101. hr = CoCreateInstance(CLSID_NullRenderer, NULL, CLSCTX_INPROC_SERVER,
  102. IID_IBaseFilter, (LPVOID*) &m_pNullFilter);
  105. hr = m_pGraph->AddFilter(m_pNullFilter, L"NullRenderer");
  107. hr = m_pSampleGrabberFilter->QueryInterface(IID_ISampleGrabber, (void**)&m_pSampleGrabber);
  109. AM_MEDIA_TYPE   mt;
  110. ZeroMemory(&mt, sizeof(AM_MEDIA_TYPE));
  111. mt.majortype = MEDIATYPE_Video;
  112. mt.subtype = MEDIASUBTYPE_RGB24;
  113. mt.formattype = FORMAT_VideoInfo; 
  114. hr = m_pSampleGrabber->SetMediaType(&mt);
  115. MYFREEMEDIATYPE(mt);
  117. m_pGraph->AddFilter(m_pSampleGrabberFilter, L"Grabber");
  118.  
  119. // Bind Device Filter.  We know the device because the id was passed in
  120. BindFilter(nCamID, &m_pDeviceFilter);
  121. m_pGraph->AddFilter(m_pDeviceFilter, NULL);
  123. CComPtr<IEnumPins> pEnum;
  124. m_pDeviceFilter->EnumPins(&pEnum);
  125.  
  126. hr = pEnum->Reset();
  127. hr = pEnum->Next(1, &m_pCameraOutput, NULL); 
  129. pEnum = NULL; 
  130. m_pSampleGrabberFilter->EnumPins(&pEnum);
  131. pEnum->Reset();
  132. hr = pEnum->Next(1, &m_pGrabberInput, NULL); 
  134. pEnum = NULL;
  135. m_pSampleGrabberFilter->EnumPins(&pEnum);
  136. pEnum->Reset();
  137. pEnum->Skip(1);
  138. hr = pEnum->Next(1, &m_pGrabberOutput, NULL); 
  140. pEnum = NULL;
  141. m_pNullFilter->EnumPins(&pEnum);
  142. pEnum->Reset();
  143. hr = pEnum->Next(1, &m_pNullInputPin, NULL);
  145. //SetCrossBar();
  147. if (bDisplayProperties) 
  148. {
  149. CComPtr<ISpecifyPropertyPages> pPages;
  151. HRESULT hr = m_pCameraOutput->QueryInterface(IID_ISpecifyPropertyPages, (void**)&pPages);
  152. if (SUCCEEDED(hr))
  153. {
  154. PIN_INFO PinInfo;
  155. m_pCameraOutput->QueryPinInfo(&PinInfo);
  157. CAUUID caGUID;
  158. pPages->GetPages(&caGUID);
  160. OleCreatePropertyFrame(NULL, 0, 0,
  161. L"Property Sheet", 1,
  162. (IUnknown **)&(m_pCameraOutput.p),
  163. caGUID.cElems,
  164. caGUID.pElems,
  165. 0, 0, NULL);
  166. CoTaskMemFree(caGUID.pElems);
  167. PinInfo.pFilter->Release();
  168. }
  169. pPages = NULL;
  170. }
  171. else 
  172. {
  173. //////////////////////////////////////////////////////////////////////////////
  174. // 加入由 lWidth和lHeight设置的摄像头的宽和高 的功能，默认320*240
  175. // by flymanbox @2009-01-24
  176. //////////////////////////////////////////////////////////////////////////////
  177.    int _Width = nWidth, _Height = nHeight;
  178.    IAMStreamConfig*   iconfig; 
  179.    iconfig = NULL;
  180.    hr = m_pCameraOutput->QueryInterface(IID_IAMStreamConfig,   (void**)&iconfig);   
  181.       
  182.    AM_MEDIA_TYPE* pmt;    
  183.    if(iconfig->GetFormat(&pmt) !=S_OK) 
  184.    {
  185.   //printf("GetFormat Failed ! n");
  186.   return   false;   
  187.    }
  188.       
  189.    VIDEOINFOHEADER*   phead;
  190.    if ( pmt->formattype == FORMAT_VideoInfo)   
  191.    {   
  192. phead=( VIDEOINFOHEADER*)pmt->pbFormat;   
  193. phead->bmiHeader.biWidth = _Width;   
  194. phead->bmiHeader.biHeight = _Height;   
  195. if(( hr=iconfig->SetFormat(pmt)) != S_OK )   
  196. {
  197. return   false;
  198. }
  200. }   
  202. iconfig->Release();   
  203. iconfig=NULL;   
  204. MYFREEMEDIATYPE(*pmt);
  205. }
  207. hr = m_pGraph->Connect(m_pCameraOutput, m_pGrabberInput);
  208. hr = m_pGraph->Connect(m_pGrabberOutput, m_pNullInputPin);
  210. if (FAILED(hr))
  211. {
  212. switch(hr)
  213. {
  214. case VFW_S_NOPREVIEWPIN :
  215. break;
  216. case E_FAIL :
  217. break;
  218. case E_INVALIDARG :
  219. break;
  220. case E_POINTER :
  221. break;
  222. }
  223. }
  225. m_pSampleGrabber->SetBufferSamples(TRUE);
  226. m_pSampleGrabber->SetOneShot(TRUE);
  227.     
  228. hr = m_pSampleGrabber->GetConnectedMediaType(&mt);
  229. if(FAILED(hr))
  230. return false;
  232. VIDEOINFOHEADER *videoHeader;
  233. videoHeader = reinterpret_cast<VIDEOINFOHEADER*>(mt.pbFormat);
  234. m_nWidth = videoHeader->bmiHeader.biWidth;
  235. m_nHeight = videoHeader->bmiHeader.biHeight;
  236. m_bConnected = true;
  238. pEnum = NULL;
  239. return true;
  240. }
  243. bool CCameraDS::BindFilter(int nCamID, IBaseFilter **pFilter)
  244. {
  245. if (nCamID < 0)
  246. return false;
  247.  
  248.     // enumerate all video capture devices
  249. CComPtr<ICreateDevEnum> pCreateDevEnum;
  250. HRESULT hr = CoCreateInstance(CLSID_SystemDeviceEnum, NULL, CLSCTX_INPROC_SERVER,
  251. IID_ICreateDevEnum, (void**)&pCreateDevEnum);
  252. if (hr != NOERROR)
  253. {
  254. return false;
  255. }
  257.     CComPtr<IEnumMoniker> pEm;
  258.     hr = pCreateDevEnum->CreateClassEnumerator(CLSID_VideoInputDeviceCategory,
  259.         &pEm, 0);
  260.     if (hr != NOERROR) 
  261. {
  262. return false;
  263.     }
  265.     pEm->Reset();
  266.     ULONG cFetched;
  267.     IMoniker *pM;
  268. int index = 0;
  269.     while(hr = pEm->Next(1, &pM, &cFetched), hr==S_OK, index <= nCamID)
  270.     {
  271. IPropertyBag *pBag;
  272. hr = pM->BindToStorage(0, 0, IID_IPropertyBag, (void **)&pBag);
  273. if(SUCCEEDED(hr)) 
  274. {
  275. VARIANT var;
  276. var.vt = VT_BSTR;
  277. hr = pBag->Read(L"FriendlyName", &var, NULL);
  278. if (hr == NOERROR) 
  279. {
  280. if (index == nCamID)
  281. {
  282. pM->BindToObject(0, 0, IID_IBaseFilter, (void**)pFilter);
  283. }
  284. SysFreeString(var.bstrVal);
  285. }
  286. pBag->Release();
  287. }
  288. pM->Release();
  289. index++;
  290.     }
  292. pCreateDevEnum = NULL;
  293. return true;
  294. }
  298. //将输入crossbar变成PhysConn_Video_Composite
  299. void CCameraDS::SetCrossBar()
  300. {
  301. int i;
  302. IAMCrossbar *pXBar1 = NULL;
  303. ICaptureGraphBuilder2 *pBuilder = NULL;
  305.  
  306. HRESULT hr = CoCreateInstance(CLSID_CaptureGraphBuilder2, NULL,
  307. CLSCTX_INPROC_SERVER, IID_ICaptureGraphBuilder2, 
  308. (void **)&pBuilder);
  310. if (SUCCEEDED(hr))
  311. {
  312. hr = pBuilder->SetFiltergraph(m_pGraph);
  313. }
  316. hr = pBuilder->FindInterface(&LOOK_UPSTREAM_ONLY, NULL, 
  317. m_pDeviceFilter,IID_IAMCrossbar, (void**)&pXBar1);
  319. if (SUCCEEDED(hr)) 
  320. {
  321.    long OutputPinCount;
  322. long InputPinCount;
  323. long PinIndexRelated;
  324. long PhysicalType;
  325. long inPort = 0;
  326. long outPort = 0;
  328. pXBar1->get_PinCounts(&OutputPinCount,&InputPinCount);
  329. for( i =0;i<InputPinCount;i++)
  330. {
  331. pXBar1->get_CrossbarPinInfo(TRUE,i,&PinIndexRelated,&PhysicalType);
  332. if(PhysConn_Video_Composite==PhysicalType) 
  333. {
  334. inPort = i;
  335. break;
  336. }
  337. }
  338. for( i =0;i<OutputPinCount;i++)
  339. {
  340. pXBar1->get_CrossbarPinInfo(FALSE,i,&PinIndexRelated,&PhysicalType);
  341. if(PhysConn_Video_VideoDecoder==PhysicalType) 
  342. {
  343. outPort = i;
  344. break;
  345. }
  346. }
  347.   
  348. if(S_OK==pXBar1->CanRoute(outPort,inPort))
  349. {
  350. pXBar1->Route(outPort,inPort);
  351. }
  352. pXBar1->Release();  
  353. }
  354. pBuilder->Release();
  355. }
  357. /*
  358. The returned image can not be released.
  359. */
  360. IplImage* CCameraDS::QueryFrame()
  361. {
  363. long evCode;
  364. long size = 0;
  366. m_pMediaControl->Run();
  367. m_pMediaEvent->WaitForCompletion(INFINITE, &evCode);
  368.  
  369. m_pSampleGrabber->GetCurrentBuffer(&size, NULL);
  371. //if the buffer size changed
  372. if (size != m_nBufferSize)
  373. {
  374. if (m_pFrame)
  375. cvReleaseImage(&m_pFrame);
  377. m_nBufferSize = size;
  378. m_pFrame = cvCreateImage(cvSize(m_nWidth, m_nHeight), IPL_DEPTH_8U, 3);
  379. }
  381. m_pSampleGrabber->GetCurrentBuffer(&m_nBufferSize, (long*)m_pFrame->imageData);
  382. cvFlip(m_pFrame);
  384. return m_pFrame;
  385. }
  387. int CCameraDS::CameraCount()
  388. {
  390. int count = 0;
  391.   CoInitialize(NULL);
  393.    // enumerate all video capture devices
  394. CComPtr<ICreateDevEnum> pCreateDevEnum;
  395.     HRESULT hr = CoCreateInstance(CLSID_SystemDeviceEnum, NULL, CLSCTX_INPROC_SERVER,
  396. IID_ICreateDevEnum, (void**)&pCreateDevEnum);
  398.     CComPtr<IEnumMoniker> pEm;
  399.     hr = pCreateDevEnum->CreateClassEnumerator(CLSID_VideoInputDeviceCategory,
  400.         &pEm, 0);
  401.     if (hr != NOERROR) 
  402. {
  403. return count;
  404.     }
  406.     pEm->Reset();
  407.     ULONG cFetched;
  408.     IMoniker *pM;
  409.     while(hr = pEm->Next(1, &pM, &cFetched), hr==S_OK)
  410.     {
  411. count++;
  412.     }
  414. pCreateDevEnum = NULL;
  415. pEm = NULL;
  416. return count;
  417. }
  419. int CCameraDS::CameraName(int nCamID, char* sName, int nBufferSize)
  420. {
  421. int count = 0;
  422.   CoInitialize(NULL);
  424.    // enumerate all video capture devices
  425. CComPtr<ICreateDevEnum> pCreateDevEnum;
  426.     HRESULT hr = CoCreateInstance(CLSID_SystemDeviceEnum, NULL, CLSCTX_INPROC_SERVER,
  427. IID_ICreateDevEnum, (void**)&pCreateDevEnum);
  429.     CComPtr<IEnumMoniker> pEm;
  430.     hr = pCreateDevEnum->CreateClassEnumerator(CLSID_VideoInputDeviceCategory,
  431.         &pEm, 0);
  432.     if (hr != NOERROR) return 0;
  435.     pEm->Reset();
  436.     ULONG cFetched;
  437.     IMoniker *pM;
  438.     while(hr = pEm->Next(1, &pM, &cFetched), hr==S_OK)
  439.     {
  440. if (count == nCamID)
  441. {
  442. IPropertyBag *pBag=0;
  443. hr = pM->BindToStorage(0, 0, IID_IPropertyBag, (void **)&pBag);
  444. if(SUCCEEDED(hr))
  445. {
  446. VARIANT var;
  447. var.vt = VT_BSTR;
  448. hr = pBag->Read(L"FriendlyName", &var, NULL); //还有其他属性,像描述信息等等...
  449.             if(hr == NOERROR)
  450.         {
  451.         //获取设备名称
  452. WideCharToMultiByte(CP_ACP,0,var.bstrVal,-1,sName, nBufferSize ,"",NULL);
  454.                 SysFreeString(var.bstrVal);
  455.         }
  456.     pBag->Release();
  457. }
  458. pM->Release();
  460. break;
  461. }
  462. count++;
  463.     }
  465. pCreateDevEnum = NULL;
  466. pEm = NULL;
  468. return 1;
  469. }