[轉] 從攝像頭流中捕捉一張圖片

視頻問題，從攝像頭流中捕捉一張圖片。用ISampleGrabber方法 （by Atomictry(天影)）

 

//函數說明 /* * 本地顯示函數 */ HRESULT StartDisplay(HWND hwnd); /* * 建立鏈路 */ HRESULT BuilderGraph() /* * 拍照 */ HRESULT SnapStill() /* * 修改攝像頭分辨率，色深，頻率 */ BOOL SetFormat(ICaptureGraphBuilder2* pBuilder, IBaseFilter* pCap, long lWidth, long lHeight, unsigned short iColorBit, __int64 iRate ) HRESULT StartDisplay(HWND hwnd) { HRESULT hr = S_OK; hr = BuilderGraph(); if(hr==S_FALSE) return S_FALSE; //Create Display Windows pSendWindow->put_Owner((OAHWND)hwnd); pSendWindow->put_WindowStyle(WS_CHILD | WS_CLIPSIBLINGS | WS_CLIPCHILDREN); pSendWindow->SetWindowPosition(0,0,250,250); pSendWindow->put_Visible(OATRUE); hr = pGrabber->SetOneShot(FALSE); hr = pGrabber->SetBufferSamples(TRUE); //Begin display hr=pSendControl->Run(); Sleep(500); SnapStill(); return S_OK; } HRESULT BuilderGraph() { HRESULT hr=S_OK; pSendGraph=NULL; pCaputerFilter=NULL; pCaputerBuilder=NULL; pSendWindow=NULL; pSendControl=NULL; pSendEvent=NULL; pGrabberSample = NULL; //1.Builder Filter Graph hr=CoCreateInstance((REFCLSID)CLSID_FilterGraph,NULL, CLSCTX_INPROC_SERVER, (REFIID)IID_IGraphBuilder, (void**)&pSendGraph); if(FAILED(hr)) return S_FALSE; //2.Builder Caputer Filter hr=CoCreateInstance(CLSID_CaptureGraphBuilder2,NULL, CLSCTX_INPROC,IID_ICaptureGraphBuilder2, (void**)&pCaputerBuilder); if(FAILED(hr)) return S_FALSE; //3.Builder SampleGrabber Filter hr = CoCreateInstance(CLSID_SampleGrabber, NULL, CLSCTX_INPROC_SERVER, IID_IBaseFilter, (void**)&pGrabberSample); if(FAILED(hr)) return S_FALSE; hr=pCaputerBuilder->SetFiltergraph(pSendGraph); if(FAILED(hr))return S_FALSE; //Find Caputer Filter hr=FindCaputerDevice(); if(FAILED(hr)) { SAFE_RELEASE(pCaputerBuilder); SAFE_RELEASE(pSendGraph); return S_FALSE; } hr =pSendGraph->AddFilter(pNetSend,NULL); hr =pSendGraph->AddFilter(pCaputerFilter,L"Caputer Filter"); hr =pSendGraph->AddFilter(pCompressor,L"Compressor Filter"); hr = pSendGraph->AddFilter(pGrabberSample, L"Sample Grabber"); hr=pSendGraph->QueryInterface(IID_IMediaControl,(void**)&pSendControl); if(FAILED(hr)) return S_FALSE; hr=pSendGraph->QueryInterface(IID_IVideoWindow,(void**)&pSendWindow); if(FAILED(hr)) return S_FALSE; hr=pSendGraph->QueryInterface(IID_IMediaEvent,(void**)&pSendEvent); if(FAILED(hr)) return S_FALSE; hr=pGrabberSample->QueryInterface(IID_ISampleGrabber, (void**)&pGrabber); if(FAILED(hr)) return S_FALSE; // 修改分辨率 SetFormat(pCaputerBuilder, pCaputerFilter, 640, 480, 24, 30); //Display local video hr=pCaputerBuilder->RenderStream(&PIN_CATEGORY_PREVIEW,NULL,pCaputerFilter,pGrabberSample,NULL); if(FAILED(hr)) return S_FALSE; return S_OK; } HRESULT SnapStill() { HRESULT hr; long cbBuffer = 0; hr = pGrabber->GetCurrentBuffer(&cbBuffer, NULL); if(FAILED(hr)) return E_FAIL; char *pBuffer = new char[cbBuffer]; if (!pBuffer) { // Deal Out of memory. Return an error code. } hr = pGrabber->GetCurrentBuffer(&cbBuffer, (long*)pBuffer); if(FAILED(hr)) return E_FAIL; //生成Bitmap AM_MEDIA_TYPE mt; hr = pGrabber->GetConnectedMediaType(&mt); if (FAILED(hr)) return E_FAIL; VIDEOINFOHEADER *pVideoHeader = (VIDEOINFOHEADER*)mt.pbFormat; if(pVideoHeader==NULL) return E_FAIL; BITMAPINFO BitmapInfo; ZeroMemory(&BitmapInfo, sizeof(BitmapInfo)); CopyMemory(&BitmapInfo.bmiHeader, &(pVideoHeader->bmiHeader), sizeof(BITMAPINFOHEADER)); HBITMAP hBitmap; hBitmap = ::CreateDIBitmap(::GetDC(NULL), &(pVideoHeader->bmiHeader), CBM_INIT, pBuffer, &BitmapInfo, DIB_RGB_COLORS); if(hBitmap==NULL) return E_FAIL; CString strSaveFileName=""; CFileDialog filedlg(FALSE,_T("bmp"),_T(""),OFN_HIDEREADONLY|OFN_OVERWRITEPROMPT,_T("BMP(*.bmp)")); if(filedlg.DoModal()==IDOK){ strSaveFileName=filedlg.GetPathName(); SaveBitmapToFile(hBitmap,strSaveFileName.GetBuffer(0)); } return hr; } BOOL SetFormat(ICaptureGraphBuilder2* pBuilder, IBaseFilter* pCap, long lWidth, long lHeight, unsigned short iColorBit, __int64 iRate ) { VIDEOINFOHEADER* phead; IAMStreamConfig* iconfig; HRESULT hr; hr = pBuilder -> FindInterface( &PIN_CATEGORY_CAPTURE, &MEDIATYPE_Interleaved, pCap, IID_IAMStreamConfig, (void **)&iconfig ); if ( hr != NOERROR ) { hr = pBuilder -> FindInterface( &PIN_CATEGORY_CAPTURE, &MEDIATYPE_Video, pCap, IID_IAMStreamConfig, (void **)&iconfig); } if ( FAILED( hr ) ) return FALSE; AM_MEDIA_TYPE* pmt; if ( iconfig -> GetFormat( &pmt ) != S_OK ) return FALSE; if ( pmt -> formattype == FORMAT_VideoInfo) { phead = ( VIDEOINFOHEADER* )pmt -> pbFormat; phead -> bmiHeader.biBitCount = iColorBit; phead -> bmiHeader.biWidth = lWidth; phead -> bmiHeader.biHeight = lHeight; phead -> bmiHeader.biSizeImage = lWidth * lHeight * iColorBit / 8; phead -> AvgTimePerFrame = iRate; if ( ( hr = iconfig -> SetFormat( pmt ) ) != S_OK ) return FALSE; } iconfig -> Release(); iconfig = NULL; FreeMediaType( *pmt ); return TRUE; }