「深度学习福利」大神带你进阶工程师，立即查看>>>
当然你可以直接用现成的虚拟摄像头软件实现这个功能。不过当初我开发这个插件的原因是，需要在Flash产品里面共享桌面，如果此时需要引导用户安装一个第三方的虚拟摄像头体验不好，所以公司希望我自己开发一个虚拟摄像头，一键安装减少用户的使用门槛。所谓的虚拟摄像头实际上在windows系统上注册了一个特殊dll，这个dll是一个COM组件。
虚拟摄像头需要用到Direct Show编程。
下载Direct Show开发代码
里面有如下的文件夹，我只需要用到第一个文件夹里面的代码—— baseclasses baseclasses capture common dmo dvd filters misc players vmr9
创建工程
打开Visual Studio ，新建一个win32 Dll项目。 打开属性页，在VC++ 目录一栏中的库目录里面添加刚才的baseclasses的路径，这样我们就能在项目中引用这个目录里的代码了。 在C/C++属性页里面的附加库目录里面也把baseclasses的路径填入。 在dll.cpp中，我们需要把Filter注册成COM组件。 分别需要调用AMovieSetupRegisterServer函数、CreateComObject函数以及IFilterMapper2接口的RegisterFilter函数完成注册。
主逻辑
在头文件中，我们需要声明两个类 class CVCamStream; class CVCam : public CSource { public: ////////////////////////////////////////////////////////////////////////// // IUnknown ////////////////////////////////////////////////////////////////////////// static CUnknown * WINAPI CreateInstance(LPUNKNOWN lpunk, HRESULT *phr); STDMETHODIMP QueryInterface(REFIID riid, void **ppv); IFilterGraph *GetGraph() {return m_pGraph;} static int cx, cy; static HANDLE SocketThread; static SOCKET ClientSocket; private: CVCam(LPUNKNOWN lpunk, HRESULT *phr); }; class CVCamStream : public CSourceStream, public IAMDroppedFrames,public IAMStreamConfig, public IKsPropertySet { public:
COM组件只需要实现CUnknown接口即可。我们继承了Direct Show的CSource类，那么就已经实现了这个接口。 CVCamStream类用来实现图像数据的输出。 在CVCam的构造函数里面我们创建CVCamStream类的实例 m_paStreams = (CSourceStream **) new CVCamStream*[1]; m_paStreams[0] = new CVCamStream(phr, this, L"Flex COM");
在实现COM接口的QueryInterface函数中，我们调用了CVCamStream类的QueryInterface HRESULT CVCam::QueryInterface(REFIID riid, void **ppv) { //Forward request for IAMStreamConfig & IKsPropertySet to the pin if(riid == _uuidof(IAMStreamConfig) || riid == _uuidof(IAMDroppedFrames) || riid == _uuidof(IKsPropertySet)) return m_paStreams[0]->QueryInterface(riid, ppv); else return CSource::QueryInterface(riid, ppv); }
定义媒体类型 HRESULT CVCamStream::GetMediaType(int iPosition, CMediaType *pmt)
在这个函数中，我们配置了媒体的具体的格式参数，比如24位RGB格式，图像的宽高等等。 另外还需要对GetStreamCaps函数进行实现，配置媒体的格式。 HRESULT STDMETHODCALLTYPE CVCamStream::GetStreamCaps(int iIndex, AM_MEDIA_TYPE **pmt, BYTE *pSCC)
捕获桌面
系统会调用FillBuffer函数，在这个函数中，我们将捕获到的数据填充到缓冲里面，Direct Show会处理剩下的事情。 HRESULT CVCamStream::FillBuffer(IMediaSample *pms)
捕获桌面只需要用到一个函数CopyScreenToBitmap HANDLE hDib = CopyScreenToBitmap(&ScreenRect, pData, (BITMAPINFO *)&(pVih->bmiHeader), m_hCursor); if (hDib) DeleteObject(hDib);
pData是我们定义的一个指针，通过下面的代码，我们的pData就指向了缓存，数据填充到pData指向的内存中。 BYTE *pData; pms->GetPointer(&pData);
进阶
实际产品会有很多需求，光实现捕获桌面是远远不够的，我们需要对这个捕获进行控制，比如捕获制定区域，停止捕获，恢复捕获等等。那么就涉及到和COM进行通讯了。 我们可以通过VS的窗口设计器创建一个windows窗口，然后提供一个用户操作界面。 如何响应这个窗口的用户操作呢？ 通过windows消息 INT_PTR CALLBACK WindowMessage(HWND hDlg, UINT message, WPARAM wParam, LPARAM lParam)
当然最关键是需要在Flash产品的程序里面唤起这个窗口，需要用的socket编程。 SOCKET Listen_Sock = socket(AF_INET, SOCK_STREAM, 0); SOCKADDR_IN serverAddr; ZeroMemory((char *)&serverAddr, sizeof(serverAddr)); serverAddr.sin_family = AF_INET; serverAddr.sin_port = htons(1234); serverAddr.sin_addr.s_addr = htonl(INADDR_ANY); bind(Listen_Sock, (struct sockaddr *)&serverAddr, sizeof(serverAddr)); listen(Listen_Sock, 5);
C++的socket编程十分的繁琐，其他语言都会进行封装，让开发变的十分便利。 在一个线程里面写个死循环进行读取socket的数据，这是比较初级的多线程阻塞式的socket编程。对于我们这个程序是绰绰有余了。毕竟不是服务器，不需要面对并发的问题。
源码
https://github.com/langhuihui/FlexCOM