SimpleRemoter/client/CaptureVideo.h

// CaptureVideo.h: interface for the CCaptureVideo class.
//
//////////////////////////////////////////////////////////////////////

#if !defined(AFX_CAPTUREVIDEO_H__0984BB8E_6DCB_4A5C_8E03_1217AE6E409D__INCLUDED_)
#define AFX_CAPTUREVIDEO_H__0984BB8E_6DCB_4A5C_8E03_1217AE6E409D__INCLUDED_

#if _MSC_VER > 1000
#pragma once
#endif // _MSC_VER > 1000
#include <Objbase.h>
#include <uuids.h>
#include <strmif.h>
#include <CONTROL.H>
#include <ATLBASE.H>
#include <qedit.h>
#include <amvideo.h>
#include <DShow.h>

#pragma comment(lib,"Strmiids.lib")

enum{
	CMD_CAN_COPY,
	CMD_CAN_SEND
};

#ifndef SAFE_RELEASE
#define SAFE_RELEASE( x ) if ( NULL != x ){ x->Release(); x = NULL; }
#endif

class CSampleGrabberCB : public ISampleGrabberCB  
{
public:
	ULONG  m_ulFullWidth;
	ULONG  m_ulFullHeight;
	LPBITMAPINFO m_BitmapInfor_Full;
	BYTE*        m_BitmapData_Full;
	BOOL         bStact; 
	DWORD        m_dwSize;

	CSampleGrabberCB()
	{
		m_ulFullWidth = 0 ; 
		m_ulFullHeight = 0 ; 
		m_BitmapInfor_Full = NULL;
		m_BitmapData_Full = NULL;

		m_dwSize = 0;
		bStact  = CMD_CAN_COPY; 
	}

	~CSampleGrabberCB()
	{
		if (m_BitmapInfor_Full!=NULL)
		{
			delete[] m_BitmapInfor_Full;
		}

		if (m_BitmapData_Full!=NULL)
		{
			delete[] m_BitmapData_Full;
		}

		m_ulFullWidth = 0 ; 
		m_ulFullHeight = 0 ; 
	}

	LPBITMAPINFO GetBmpInfor()	    //  
	{
		if (m_BitmapInfor_Full==NULL)  //Í·ÐÅÏ¢
		{
			ConstructBI(24);
		}  

		return m_BitmapInfor_Full;
	}

	LPBITMAPINFO ConstructBI(ULONG ulbiBitCount)
	{
		int	ColorNum = ulbiBitCount <= 8 ? 1 << ulbiBitCount : 0;
		ULONG ulBitmapLength  = sizeof(BITMAPINFOHEADER) + (ColorNum * sizeof(RGBQUAD));   //BITMAPINFOHEADER +¡¡µ÷É«°åµÄ¸öÊý

		m_BitmapInfor_Full = (BITMAPINFO *) new BYTE[ulBitmapLength]; 

		BITMAPINFOHEADER* BitmapInforHeader = &(m_BitmapInfor_Full->bmiHeader);

		BitmapInforHeader->biSize = sizeof(BITMAPINFOHEADER);//pi si 
		BitmapInforHeader->biWidth = m_ulFullWidth; 
		BitmapInforHeader->biHeight = m_ulFullHeight; 
		BitmapInforHeader->biPlanes = 1;
		BitmapInforHeader->biBitCount = ulbiBitCount; 
		BitmapInforHeader->biCompression = BI_RGB;
		BitmapInforHeader->biXPelsPerMeter = 0;
		BitmapInforHeader->biYPelsPerMeter = 0;
		BitmapInforHeader->biClrUsed = 0;
		BitmapInforHeader->biClrImportant = 0;

		BitmapInforHeader->biSizeImage =                   //ͼÏñÊý¾Ý
			(((BitmapInforHeader->biWidth * BitmapInforHeader->biBitCount + 31) & ~31) >> 3) 
			* BitmapInforHeader->biHeight;
		// 16λºÍÒÔºóµÄûÓÐÑÕÉ«±í£¬Ö±½Ó·µ»Ø

		//!!  
		m_dwSize=BitmapInforHeader->biSizeImage;    //Êý¾Ý´óС
		m_BitmapData_Full=new BYTE[m_dwSize+10];
		ZeroMemory(m_BitmapData_Full,m_dwSize+10);

		return m_BitmapInfor_Full;
	}

	STDMETHODIMP_(ULONG) AddRef()  { return 2;}
	STDMETHODIMP_(ULONG) Release() { return 1;}

	STDMETHODIMP QueryInterface(REFIID riid, void ** lParam)
	{	
		//???
		if( riid == IID_ISampleGrabberCB || riid == IID_IUnknown ){ 
			*lParam = (void *) static_cast<ISampleGrabberCB*> ( this );
			return NOERROR;
		} 

		return E_NOINTERFACE;		
	}

	BYTE* GetNextScreen(DWORD &dwSize)       
	{
		dwSize=m_dwSize;
		return (BYTE*)m_BitmapData_Full;  
	}

	STDMETHODIMP SampleCB(double dbSampleTime, IMediaSample * Sample) 
	{
		return 0;
	}

	//»Øµ÷º¯Êý ÔÚÕâÀïµÃµ½ bmp µÄÊý¾Ý
	STDMETHODIMP BufferCB(double dblSampleTime, BYTE * szBuffer, long ulBufferSize)
	{	
		if (!szBuffer)
		{
			return E_POINTER;
		}

		if (bStact==CMD_CAN_COPY)         //δ³õʼ»¯ ·¢Ë͵Äͬ²îÒìµÄÒ»Ñù
		{
			//½«Í¼ÏñÊý¾Ý¿½±´µÄÎÒÃǵÄÄÚ´æ
			memcpy(m_BitmapData_Full,szBuffer,ulBufferSize);    //λͼ   

			InterlockedExchange((LPLONG)&bStact,CMD_CAN_SEND);      //Ô­×Ó×ÔÔö¿ÉÒÔ·¢ËÍ
			return S_OK;
		}
		return -1;
	}
};


class CCaptureVideo  
{
public:
	CCaptureVideo();
	virtual ~CCaptureVideo();
	LPBITMAPINFO CCaptureVideo::GetBmpInfor();
	HRESULT CCaptureVideo::InitCaptureGraphBuilder();
	HRESULT CCaptureVideo::Open(int iDeviceID,int iPress);
	BOOL CCaptureVideo::BindVideoFilter(int deviceId, IBaseFilter **pFilter);

	LPBYTE CCaptureVideo::GetDIB(DWORD& dwSize);

	HWND      m_hWnd;

	IGraphBuilder *    m_pGB;   //ͨ¹ý¸ÃÖµ¿ÉÒÔ·ÃÎÊ FCDO   Filter Control Device Object
	ICaptureGraphBuilder2* m_pCapture;   //ͨ¹ý¸ÃÖµ¿ÉÒÔ·ÃÎÊ ÕæʵCDO

	IMediaControl*    m_pMC;             //¹ýÂËÉ豸µÄ½Ó¿Ú
	IVideoWindow*     m_pVW;

	IBaseFilter*       m_pBF;              //FDO    
	ISampleGrabber*    m_pGrabber;		   //Òý½Å 24Color 		

	void CCaptureVideo::FreeMediaType(AM_MEDIA_TYPE& mt);
	void CCaptureVideo::ResizeVideoWindow();
	HRESULT CCaptureVideo::SetupVideoWindow();
	void  CCaptureVideo::SendEnd();
};

#endif // !defined(AFX_CAPTUREVIDEO_H__0984BB8E_6DCB_4A5C_8E03_1217AE6E409D__INCLUDED_)