【重传】vc实现虚拟摄像头2例

#include "Filter.h" #include "DibHelper.h" #include "MyConfig.h" #include "resource.h" CUnknown * WINAPI CVCam::CreateInstance(LPUNKNOWN lpunk, HRESULT *phr) { ASSERT(phr); CUnknown *punk = new CVCam(lpunk, phr); return punk; } CVCam::CVCam(LPUNKNOWN lpunk, HRESULT *phr) : CSource(NAME("Virtual Camera"), lpunk, CLSID_VirtualCamera) { ASSERT(phr); CAutoLock cAutoLock(&m_cStateLock); m_paStreams = (CSourceStream **) new CVCamPin*[1]; m_paStreams[0] = new CVCamPin(phr, this); } STDMETHODIMP CVCam::QueryInterface(REFIID riid, void **ppv) { if(riid == IID_IAMStreamConfig || riid == IID_IKsPropertySet || riid == IID_IMyConfig || riid == IID_ISpecifyPropertyPages) return m_paStreams[0]->QueryInterface(riid, ppv); else return CSource::QueryInterface(riid, ppv); } STDMETHODIMP CVCam::NonDelegatingQueryInterface(REFIID riid, void **ppv) { if(riid == IID_IAMStreamConfig || riid == IID_IKsPropertySet || riid == IID_ISpecifyPropertyPages || riid == IID_IMyConfig) return m_paStreams[0]->NonDelegatingQueryInterface(riid, ppv); else return CSource::NonDelegatingQueryInterface(riid, ppv); } STDMETHODIMP CVCamPin::GetPages(CAUUID *pPages){ CheckPointer(pPages,E_POINTER); pPages->cElems = 1; pPages->pElems = (GUID *) CoTaskMemAlloc(sizeof(GUID)); if (pPages->pElems == NULL) return E_OUTOFMEMORY; *(pPages->pElems) = CLSID_MyPropertyPage; return NOERROR; } // GetPages CVCamPin::CVCamPin(HRESULT *phr, CSource *pFilter) : CSourceStream(NAME("Push Source Desktop"), phr, pFilter, L"Out"), m_FramesWritten(0), m_bZeroMemory(0), m_iFrameNumber(0), m_rtFrameLength(FPS_20), // Capture and display desktop 30 times per second m_nCurrentBitDepth(32) { m_pParent = (CVCam*)pFilter; m_hCursor = LoadCursor(0, MAKEINTRESOURCE(IDC_ARROW)); // Get the device context of the main display HDC hDC; hDC = CreateDC(TEXT("DISPLAY"), NULL, NULL, NULL); HKEY hKEY; DWORD dwDisposition = REG_OPENED_EXISTING_KEY; long ret = RegCreateKeyEx(HKEY_LOCAL_MACHINE, "SOFTWARE\\Virtual Camera", 0, TEXT(""), 0, KEY_ALL_ACCESS, NULL, &hKEY, &dwDisposition); DWORD type = REG_DWORD; DWORD type2 = REG_SZ; DWORD size = sizeof(DWORD); DWORD size2; DWORD dwWidth = 320; DWORD dwHeight = 240; ret = RegQueryValueEx(hKEY, "top", NULL, &type, (LPBYTE)&m_rScreen.top, &size); if(ret != ERROR_SUCCESS) m_rScreen.top = 0; ret = RegQueryValueEx(hKEY, "left", NULL, &type, (LPBYTE)&m_rScreen.left, &size); if(ret != ERROR_SUCCESS) m_rScreen.left = 0; ret = RegQueryValueEx(hKEY, "width", NULL, &type, (LPBYTE)&dwWidth, &size); m_rScreen.right = m_rScreen.left + dwWidth; ret = RegQueryValueEx(hKEY, "height", NULL, &type, (LPBYTE)&dwHeight, &size); m_rScreen.bottom = m_rScreen.top + dwHeight; ret = RegQueryValueEx(hKEY, "auto", NULL, &type, (LPBYTE)&m_bAutoSetRect, &size); if(ret != ERROR_SUCCESS) m_bAutoSetRect = true; ret = RegQueryValueEx(hKEY, "Text", NULL, &type2, (LPBYTE)&m_szText, &size2); if(ret != ERROR_SUCCESS) strcpy(m_szText, ""); // Save dimensions for later use in FillBuffer() m_iImageWidth = 320; m_iImageHeight = 240; // Release the device context DeleteDC(hDC); } CVCamPin::~CVCamPin() { } ////////////////////////////////////////////////////////////////////////// // --- IMyConfig --- ////////////////////////////////////////////////////////////////////////// STDMETHODIMP CVCamPin::put_Rect(RECT *rt) { CAutoLock l(&m_cSharedState); m_rScreen.left = rt->left; m_rScreen.top = rt->top; m_rScreen.right = rt->right; m_rScreen.bottom = rt->bottom; HKEY hKEY; DWORD dwDisposition = REG_OPENED_EXISTING_KEY; RegCreateKeyEx(HKEY_LOCAL_MACHINE, "SOFTWARE\\Virtual Camera", 0, "", 0, KEY_ALL_ACCESS, NULL, &hKEY, &dwDisposition); DWORD dwWidth = rt->right-rt->left; DWORD dwHeight = rt->bottom-rt->top; RegSetValueEx(hKEY, "top", NULL, REG_DWORD, (const BYTE* )&rt->top, 4); RegSetValueEx(hKEY, "left", NULL, REG_DWORD, (const BYTE* )&rt->left, 4); RegSetValueEx(hKEY, "width", NULL, REG_DWORD, (const BYTE* )&dwWidth, 4); RegSetValueEx(hKEY, "height", NULL, REG_DWORD, (const BYTE* )&dwHeight, 4); return S_OK; } STDMETHODIMP CVCamPin::get_Rect(RECT *rt) { CAutoLock l(&m_cSharedState); *rt = m_rScreen; return S_OK; } STDMETHODIMP CVCamPin::put_AutoSetRect(bool *bSet) { CAutoLock l(&m_cSharedState); m_bAutoSetRect = *bSet; HKEY hKEY; DWORD dwDisposition = REG_OPENED_EXISTING_KEY; RegCreateKeyEx(HKEY_LOCAL_MACHINE, "SOFTWARE\\Virtual Camera", 0, "", 0, KEY_ALL_ACCESS, NULL, &hKEY, &dwDisposition); DWORD dwFlag = m_bAutoSetRect ? 1:0; RegSetValueEx(hKEY, "auto", NULL, REG_DWORD, (const BYTE* )&dwFlag, 4); return S_OK; } STDMETHODIMP CVCamPin::get_AutoSetRect(bool *bSet) { CAutoLock l(&m_cSharedState); *bSet = m_bAutoSetRect; return S_OK; } STDMETHODIMP CVCamPin::put_Text(char *pszText) { CAutoLock l(&m_cSharedState); strcpy(m_szText, pszText); HKEY hKEY; DWORD dwDisposition = REG_OPENED_EXISTING_KEY; RegCreateKeyEx(HKEY_LOCAL_MACHINE, "SOFTWARE\\Virtual Camera", 0, "", 0, KEY_ALL_ACCESS, NULL, &hKEY, &dwDisposition); RegSetValueEx(hKEY, "text", NULL, REG_SZ, (const BYTE* )&m_szText, strlen(m_szText)); return S_OK; } STDMETHODIMP CVCamPin::get_Text(char *pszText) { CAutoLock l(&m_cSharedState); strcpy(pszText, m_szText); return S_OK; } // // GetMediaType // HRESULT CVCamPin::GetMediaType(CMediaType *pmt) { CheckPointer(pmt,E_POINTER); CAutoLock cAutoLock(m_pFilter->pStateLock()); VIDEOINFO *pvi = (VIDEOINFO *) pmt->AllocFormatBuffer(sizeof(VIDEOINFO)); if(NULL == pvi) return(E_OUTOFMEMORY); // Initialize the VideoInfo structure before configuring its members ZeroMemory(pvi, sizeof(VIDEOINFO)); pvi->bmiHeader.biCompression = BI_RGB; pvi->bmiHeader.biBitCount = 32; // Adjust the parameters common to all formats pvi->bmiHeader.biSize = sizeof(BITMAPINFOHEADER); pvi->bmiHeader.biWidth = m_iImageWidth; pvi->bmiHeader.biHeight = m_iImageHeight; pvi->bmiHeader.biPlanes = 1; pvi->bmiHeader.biSizeImage = GetBitmapSize(&pvi->bmiHeader); pvi->bmiHeader.biClrImportant = 0; pvi->AvgTimePerFrame = FPS_20; SetRectEmpty(&(pvi->rcSource)); // we want the whole image area rendered. SetRectEmpty(&(pvi->rcTarget)); // no particular destination rectangle pmt->SetType(&MEDIATYPE_Video); pmt->SetFormatType(&FORMAT_VideoInfo); pmt->SetTemporalCompression(FALSE); // Work out the GUID for the subtype from the header info. const GUID SubTypeGUID = GetBitmapSubtype(&pvi->bmiHeader); pmt->SetSubtype(&SubTypeGUID); pmt->SetSampleSize(pvi->bmiHeader.biSizeImage); return NOERROR; } // GetMediaType // // CheckMediaType // HRESULT CVCamPin::CheckMediaType(const CMediaType *pMediaType) { CheckPointer(pMediaType,E_POINTER); CAutoLock cAutoLock(m_pFilter->pStateLock()); if((*(pMediaType->Type()) != MEDIATYPE_Video) || // we only output video !(pMediaType->IsFixedSize())) // in fixed size samples { return E_INVALIDARG; } // Check for the subtypes we support const GUID *SubType = pMediaType->Subtype(); if (SubType == NULL) return E_INVALIDARG; if(/*(*SubType != MEDIASUBTYPE_RGB8) && (*SubType != MEDIASUBTYPE_RGB565) && (*SubType != MEDIASUBTYPE_RGB555) && (*SubType != MEDIASUBTYPE_RGB24) && */(*SubType != MEDIASUBTYPE_RGB32)) { return E_INVALIDARG; } // Get the format area of the media type VIDEOINFO *pvi = (VIDEOINFO *) pMediaType->Format(); if(pvi == NULL) return E_INVALIDARG; // Check if the image width & height have changed if(pvi->bmiHeader.biWidth != m_iImageWidth || abs(pvi->bmiHeader.biHeight) != m_iImageHeight) { // If the image width/height is changed, fail CheckMediaType() to force // the renderer to resize the image. retu