videocapture.cpp

局域网视频聊天

#include "Stdafx.h"
#include "PTPHelp.h"
#include "PTPHelpDlg.h"
#include "VideoCapture.h"


#include <afxmt.h>


#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif


#pragma comment(lib,"vfw32")
#pragma comment(lib,"winmm")

//////////////////////////////////////////////////////////////////////
// Construction/Destruction
//////////////////////////////////////////////////////////////////////

VideoCapture::VideoCapture()
{
	m_capwnd=NULL;
	isOver=FALSE;	
	log.Open("videocapture.log",CFile::modeCreate | CFile::modeWrite);
}





VideoCapture::~VideoCapture()
{
}





void VideoCapture::SetDialog(CDialog *dialog)
{
	dlg=dialog;
}





BOOL VideoCapture::Initialize()
{
	char devname[100],devversion[100];
	char str[200];
	int index=0;
	m_capwnd = capCreateCaptureWindow("Capture",WS_POPUP,0,0,1,1,0,0);//创建捕捉窗口
	if(m_capwnd==NULL)
	{
		log.WriteString("\n Unable to create capture window");
		return FALSE;
	}

	capSetUserData(m_capwnd,this);
	//Change destroy functions also........
    capSetCallbackOnVideoStream(m_capwnd,OnCaptureVideo);//调用视频流回调函数
 
	capGetDriverDescription(index,devname,100,devversion,100);//获取设备信息
	sprintf(str,"\n Driver name = %s version = %s ",devname,devversion);
	log.WriteString(str);

	// 连接到视频捕捉设备
	if( ! capDriverConnect(m_capwnd,index) )
	{
		AfxMessageBox("没有可用的频捕捉设备！");
		log.WriteString("\n Unable to connect driver to the window");
		m_capwnd=NULL;
		return FALSE;
	}

	// 设置视频捕捉参数
	if(SetCapturePara()==FALSE)
	{
		log.WriteString("\n Setting capture parameters failed");
		capDriverDisconnect(m_capwnd);//断开连接
		return FALSE;
	}	
return TRUE;
}





BOOL VideoCapture::StartCapture()//开始捕捉
{
	//开始捕捉，并不把捕捉的数据写入AVI文件
	if(capCaptureSequenceNoFile(m_capwnd)==FALSE)
	{
		log.WriteString("\n 视频捕捉失败！");
		return FALSE;
	}	
	return TRUE;
}


BOOL VideoCapture::StopCapture()//停止捕捉
{
	
	capCaptureStop(m_capwnd);//关闭捕捉窗口
	capCaptureAbort(m_capwnd);//销毁捕捉窗口
	Sleep(500);	
	return TRUE;
}

/**
*   Stop the catpure process and disconnect the driver
*
*/
void VideoCapture::Destroy()
{

	if(m_capwnd==NULL) return;   
    // Stop the capturing process    
	capCaptureAbort(m_capwnd);       
    // Disable the callback function..
	capSetCallbackOnVideoStream(m_capwnd, NULL);
	Sleep(300);		// This delay is important...		
	// Finally disconnect the driver
    capDriverDisconnect(m_capwnd);
}


BOOL  VideoCapture::SetCapturePara()//设置视频捕捉参数
{
	CAPTUREPARMS CapParms={0};	
	capCaptureGetSetup(m_capwnd,&CapParms,sizeof(CapParms));//
	CapParms.fAbortLeftMouse = FALSE;
	CapParms.fAbortRightMouse = FALSE;
	CapParms.fYield = TRUE;
	CapParms.fCaptureAudio = FALSE;
	CapParms.wPercentDropForError = 50;

	if(capCaptureSetSetup(m_capwnd,&CapParms,sizeof(CapParms))==FALSE)
	{
	log.WriteString("\n Failed to set the capture parameters ");
	return FALSE;
	}
	// Set Video Format 
	capGetVideoFormat(m_capwnd,&m_bmpinfo,sizeof(m_bmpinfo));
	m_bmpinfo.bmiHeader.biWidth=IMAGE_WIDTH;
	m_bmpinfo.bmiHeader.biHeight=IMAGE_HEIGHT;
	BOOL ret=capSetVideoFormat(m_capwnd,&m_bmpinfo,sizeof(m_bmpinfo));
	if(ret==TRUE)
	log.WriteString("\n Video parameters set properly");
	return TRUE;
}




/**
*
*  Allocate Memory for DIB image buffer
* 
*/

int VideoCapture::AllocateMemory(PBITMAPINFO &bmpinfo)
{
int size1,size2,size;
BITMAPINFO tbmp;
char  str[200];

	capGetVideoFormat(m_capwnd,&tbmp,sizeof(tbmp));

	size1 = getFormatSize ( tbmp );
	size2 = getImageSize ( tbmp );
	size = size1 + size2;


	sprintf(str,"\n Formatsize = %d imagesize = %d , fun_size = %d ",
			 size1,size2, capGetVideoFormatSize(m_capwnd));
	log.WriteString(str);

	bmpinfo=(BITMAPINFO *) new BYTE[size];

	
	if(bmpinfo==NULL)
	{
	AfxMessageBox("Unable to allocate memory");
    return -1;
	}

	memset(bmpinfo,0,sizeof(*bmpinfo));

	capGetVideoFormat(m_capwnd,bmpinfo,sizeof(*bmpinfo));
	return size1;


}




/**
*
*   Calculates the Format Size for DIB image
*
*/

int VideoCapture::getFormatSize(BITMAPINFO bmp)
{
int size;
	size=(bmp.bmiHeader.biSize!=0)?bmp.bmiHeader.biSize :sizeof(BITMAPINFOHEADER);

	//return (size+ bmp.bmiHeader.biClrUsed *sizeof(RGBQUAD));

	return size; //RGBQUAD is absent for 24 bit bmp image.
}






/**
*	Calculates the Size of Image
*
*/

int VideoCapture::getImageSize(BITMAPINFO bmp)
{
int size;
BITMAPINFOHEADER head=bmp.bmiHeader;


	if( head.biSizeImage==0 )
	{
		size=( head.biWidth * head.biHeight * head.biBitCount)/8;
	}
	else
	size = head.biSizeImage;

return size;

}


LRESULT CALLBACK OnCaptureVideo(HWND mwnd,LPVIDEOHDR lphdr)//视频流采集回调函数
{

	VideoCapture *vidcap=(VideoCapture *)capGetUserData(mwnd);//获取视频数据
	if(vidcap!=NULL )
	((CPTPHelpDlg*) (vidcap->dlg))->SendVideo(lphdr->lpData,lphdr->dwBytesUsed);//显示视频
	return TRUE;

}