aviomgr.cpp

一个实时音频或视频通讯程序

//NetTalk
/*------------------------------------------------------------------------------*\
 =============================
   模块名称: AVIOMgr.cpp
 =============================
 
 [版权]
 
   2000-2001  115软件工厂  版权所有
                                              
\*------------------------------------------------------------------------------*/
#include "WndX.h"
#include <vfw.h>

#include "AVIOMgr.h"
#include "UDPSocket.h"
#include "g729a.h"
#include "AudioPlay.h"
#include <stdio.h>

CAVIOMgr* pMgr=0;
//audio in thread proc,include encode
DWORD WINAPI CAVIOMgr::AudioInThreadProc(LPVOID lpParameter)
{
	CAVIOMgr* pMgr=(CAVIOMgr*)lpParameter;
	MSG msg;
	while(GetMessage(&msg,0,0,0))
	{
		switch(msg.message)
		{
		case WIM_DATA:
			{
				WAVEHDR* pWH=(WAVEHDR*)msg.lParam;
				waveInUnprepareHeader((HWAVEIN)msg.wParam,pWH,sizeof(WAVEHDR));
							
				if(pWH->dwBytesRecorded!=SIZE_AUDIO_FRAME)
					break;//it's not full recorded,i think the wave recorder has ben
				//stopped,discard it,and don't do anything!
				//
				CopyMemory(pMgr->m_paL,pWH->lpData,SIZE_AUDIO_FRAME);
				pMgr->OnEncodeAudioData(pMgr->m_paL,SIZE_AUDIO_FRAME);
				
				waveInPrepareHeader((HWAVEIN)msg.wParam,pWH,sizeof(WAVEHDR));
				waveInAddBuffer((HWAVEIN)msg.wParam,pWH,sizeof(WAVEHDR));
			}
			break;
		}
	}	
	return msg.wParam;
}
//音频输出线程，包括音频解码
DWORD WINAPI CAVIOMgr::AudioOutThreadProc(LPVOID lpParameter)
{
	
	CAVIOMgr* pMgr=(CAVIOMgr*)lpParameter;
	MSG msg;
	while(GetMessage(&msg,0,0,0))
	{
		switch(msg.message)
		{
		case IOM_AUDIO:
			pMgr->OnDecodeAudioData((PACK_AUDIO*)msg.lParam,(int)msg.wParam);
			break;
		case WOM_DONE:
			{
				WAVEHDR* pwh=(WAVEHDR*)msg.lParam;
				waveOutUnprepareHeader((HWAVEOUT)msg.wParam,pwh,sizeof(WAVEHDR));
				pMgr->m_iAudioBuf--;
				delete []pwh->lpData;//删除Play调用时分配的内存
				delete pwh;
			}
			break;
		}
	}
	return msg.wParam;
}

//视频流回调函数
LRESULT CALLBACK CAVIOMgr::VideoStreamCallbackProc(HWND hWnd, LPVIDEOHDR lpVHdr)
{
	
	CAVIOMgr* pMgr=(CAVIOMgr*)capGetUserData(hWnd);
	if(pMgr)
	{
		//压缩视频数据
		pMgr->OnEncodeVideoData((char*)lpVHdr->lpData,lpVHdr->dwBytesUsed);
		
	}	
	return TRUE;
}

//Socket线程，负责接收数据
DWORD WINAPI CAVIOMgr::SockThreadProc(LPVOID lpParameter)
{
	CAVIOMgr* pMgr=(CAVIOMgr*)lpParameter;

	char buf[4096];
	int iLen=0;
	while(1)
	{
		iLen=pMgr->m_Socket.RecvFrom(buf,4096,(sockaddr*)&pMgr->m_sockaddr);
		if(iLen>0)
		{
			switch(*((short*)buf))//check the flag
			{
			case FLAG_CMD:
				//命令
				pMgr->OnCommand((PACK_CMD*)buf,iLen);
				break;
			case FLAG_AUDIO:
				{
					//音频数据
					if(pMgr->m_ds.bAudioOut&&pMgr->m_ds.bAudioCodec&&pMgr->m_hAudioOut)
					{
						char* p=new char[iLen];
						if(p)
						{
							CopyMemory(p,buf,iLen);
							if(!PostThreadMessage(pMgr->m_dwAudioOutId,IOM_AUDIO,iLen,(LPARAM)p))
								delete []p;
						}
					}
				}
				break;
			case FLAG_VIDEO:
				{
					//视频数据
					if(pMgr->m_ds.bVideoOut&&pMgr->m_ds.bVideoCodec)
					{
						
						pMgr->OnDecodeVideoData((PACK_VIDEO*)buf,iLen);			
						
					}
				}
				
				break;
			}
		}
		else
			break;//the socket should be closed,that is m_Socket have been
		//destroyed,so break the loop and end the thread
	}
	return 0;
}

CAVIOMgr::CAVIOMgr()
{
	
	m_ppvL=0;
	m_ppaL=0;
	
	m_paR=0;
	m_pvR=0;
	m_paL=0;
	m_pvL=0;
	m_hSockThread=NULL;

	m_hAudioOut=0;
	m_hAudioIn=0;
	
}

CAVIOMgr::~CAVIOMgr()
{
	Destroy();
}
//初始化SOCKET
BOOL CAVIOMgr::InitSocket()
{
	
	DestroySocket();	
	
	BOOL bRet=FALSE;	
	if(!m_Socket.Create(PORT))
		goto RET;
	m_hSockThread=CreateThread(0,0,SockThreadProc,(LPVOID)this,0,&m_dwSockThreadId);
	if(!m_hSockThread)	
		goto RET;
	
	
	bRet=TRUE;
RET:
	if(!bRet)
		DestroySocket();
		
	return bRet;
}

BOOL CAVIOMgr::DestroySocket()
{
	m_Socket.Destroy();
	//
	
	if(m_hSockThread)
	{
		BOOL b=FALSE;
		DWORD ExitCode;
		int Timeout = 50; 
		while(Timeout)//等待线程结束，如果到一定时间海没结束，就强制结束
		{
			GetExitCodeThread(m_hSockThread, &ExitCode);
			
			if (ExitCode != STILL_ACTIVE)
			{
				b=TRUE;
				// Thread has ended.
				break;
			}
			else
			{
				Sleep(10);
			}
			
			--Timeout;
		}
		if(!b)//time out ,terminate it
			TerminateThread(m_hSockThread,0);
				
	}
	m_hSockThread=NULL;
	return TRUE;
}

//
void CAVIOMgr::SetDst(char *ip, unsigned short port)
{
	m_dst.sin_family=AF_INET;
	m_dst.sin_addr.s_addr=CUDPSocket::Name2Inet(ip);
	m_dst.sin_port=htons(port);
	
}

//初始化音频输入，当初始化完成，录音就开始了
BOOL CAVIOMgr::InitAudioRec()
{
	DestroyAudioRec();
	BOOL bRet=FALSE;
	//创建录音线程
	m_hAudioIn=CreateThread(0,0,AudioInThreadProc,this,0,&m_dwAudioInId);
	if(!m_hAudioIn)
		goto RET;
	if(!m_AudioRec.Create(0,m_dwAudioInId,(DWORD)this,CALLBACK_THREAD,SIZE_AUDIO_FRAME))
		goto RET;
	if(!m_AudioRec.Start())
		goto RET;
	m_paL=new char[SIZE_AUDIO_FRAME];
	if(!m_paL)
		goto RET;
	bRet=TRUE;
RET:
	if(!bRet)
	{
		//如果失败，就向主窗口发送出错消息
		PostMessage(m_hwndMainWnd,IOM_NOTIFY,MAKEWPARAM(IOMN_ERR,IOME_AUDIO_IN),(LPARAM)this);
		DestroyAudioRec();
	}
	return bRet;

}

BOOL CAVIOMgr::DestroyAudioRec()
{
	if(m_hAudioIn)
	{
		int t=50;
		DWORD ExitCode;
		BOOL bEnd=FALSE;
		PostThreadMessage(m_dwAudioInId,WM_QUIT,0,0);
		while(t)
		{
			
			GetExitCodeThread(m_hAudioIn,&ExitCode);
			if(ExitCode!= STILL_ACTIVE)
			{
				bEnd=TRUE;
				break;
			}
			else
				Sleep(10);
			t--;
		}
		if(!bEnd)
			TerminateThread(m_hAudioIn,0);
		m_hAudioIn=0;
		
	}
	m_AudioRec.Stop();
    m_AudioRec.Destroy();
	
	if(m_paL)
	{
		delete []m_paL;
		m_paL=0;
	}
	return TRUE;

}

BOOL CAVIOMgr::InitAudioPlay()
{
	BOOL bRet=FALSE;
	DestroyAudioPlay();
	m_iAudioBuf=0;
	m_hAudioOut=CreateThread(0,0,AudioOutThreadProc,this,0,&m_dwAudioOutId);
	if(!m_hAudioOut)
		goto RET;
	if(!m_AudioPlay.Create(0,m_dwAudioOutId,(DWORD)this,CALLBACK_THREAD))
		goto RET;
	bRet=TRUE;
	
	
RET:
	if(!bRet)
	{
		PostMessage(m_hwndMainWnd,IOM_NOTIFY,MAKEWPARAM(IOMN_ERR,IOME_AUDIO_OUT),(LPARAM)this);
		DestroyAudioPlay();
	}
	return bRet;
}

BOOL CAVIOMgr::DestroyAudioPlay()
{
	m_AudioPlay.Destroy();
	if(m_hAudioOut)
	{
		int t=50;
		DWORD ExitCode;
		BOOL bEnd=FALSE;
		PostThreadMessage(m_dwAudioOutId,WM_QUIT,0,0);
		while(t)
		{
			
			GetExitCodeThread(m_hAudioOut,&ExitCode);
			if(ExitCode!= STILL_ACTIVE)
			{
				bEnd=TRUE;
				break;
			}
			else
				Sleep(10);
			t--;
		}
		if(!bEnd)
			TerminateThread(m_hAudioOut,0);
		m_hAudioOut=0;
		
	}
	return TRUE;
}
//音频解码，并将音频数据发送到想要数据的窗口
void CAVIOMgr::OnDecodeAudioData(PACK_AUDIO *ppa, int len)
{
	
	if(m_paR)
	{
		
		if(m_CodecMgr.DecodeAudioData((char*)ppa->data,SIZE_AUDIO_PACKED,m_paR,0))
		{	
			//位了避免延迟过长
			if(m_iAudioBuf<6)
			{
				m_iAudioBuf++;
				m_AudioPlay.Play(m_paR,SIZE_AUDIO_FRAME);
			}
			if(m_hwndRemoteAudioRcv)
				SendMessage(m_hwndRemoteAudioRcv,IOM_AUDIO,1,(LPARAM)this);
		}
	}	
	m_uDataRcv+=sizeof(PACK_AUDIO);
	delete []ppa;

}
//压缩音频原始数据
void CAVIOMgr::OnEncodeAudioData(char *pa, int len)
{
	if(m_ppaL)
	{
		m_ppaL->flag=FLAG_AUDIO;
		if(m_CodecMgr.EncodeAudioData(pa,len,(char*)m_ppaL->data,0))
		{			
			m_Socket.SendTo((char*)m_ppaL,sizeof(PACK_AUDIO),(sockaddr*)&m_dst);
			m_uDataSend+=sizeof(PACK_AUDIO);
		}
	}
	if(m_hwndLocalAudioRcv)
		SendMessage(m_hwndLocalAudioRcv,IOM_AUDIO,0,(LPARAM)this);
		
}
//
void CAVIOMgr::OnDecodeVideoData(PACK_VIDEO *ppv, int len)
{	
	
	if(m_pvR)
	{
		
		if(m_CodecMgr.DecodeVideoData(((char*)ppv)+sizeof(PACK_VIDEO),ppv->data_size,m_pvR,0,0))
		{
			m_nFrameCount++;
			if(m_hwndRemoteVideoRcv)
				SendMessage(m_hwndRemoteVideoRcv,IOM_VIDEO,1,(LPARAM)this);
				
		}
	}
	m_uDataRcv+=sizeof(PACK_VIDEO)+ppv->data_size;
}


//
void CAVIOMgr::OnEncodeVideoData(char *pv, int len)
{
	
	int rlen;
	
	if(m_ppvL&&m_bVideoSend)
	{			
		if(m_CodecMgr.EncodeVideoData(pv,len,((char*)m_ppvL)+sizeof(PACK_VIDEO),&rlen,0))
		{
			
			m_ppvL->data_size=(unsigned short)rlen;
			
			m_Socket.SendTo((char*)m_ppvL,rlen+sizeof(PACK_VIDEO),(sockaddr*)&m_dst);
			
			m_uDataSend+=rlen+sizeof(PACK_VIDEO);
		}
	}
	m_pvL=pv;
	if(m_hwndLocalVideoRcv)
	{
		SendMessage(m_hwndLocalVideoRcv,IOM_VIDEO,0,(LPARAM)this);
	}

	
}
//初始化，创建SOCKET，并监听
BOOL CAVIOMgr::Init()
{
	Destroy();
	pMgr=this;
	m_timerWait=0;
	m_hwndLocalVideoRcv=0;
	m_hwndLocalAudioRcv=0;
	m_hwndRemoteVideoRcv=0;
	m_hwndRemoteAudioRcv=0;
	m_hwndMainWnd=0;
	m_bVideoSend=TRUE;
	
	m_iStatus=STA_FREE;
	BOOL bRet=FALSE;
	m_uDataRcv=0;
	m_uDataSend=0;
	if(!InitSocket())
		goto RET;
	bRet=TRUE;
RET:
	return bRet;
}
//初始化视频捕捉设备
BOOL CAVIOMgr::InitCap()//
{
	HWND hCap;
	m_nFrameCount=0;
	m_ViCap.Destroy();
	BOOL bRet=FALSE;
	if(!m_ViCap.Init())
		goto RET;
	if(!m_ViCap.GetDriverNum())
		goto RET;
	if(!m_ViCap.ConnectToDriver(0))
		goto RET;
	hCap=m_ViCap.GetCapWindow();
	if(!capSetVideoFormat(hCap,&m_CodecMgr.m_BmpU,sizeof(BITMAPINFO)))
		goto RET;
	if(!capSetCallbackOnVideoStream(hCap,VideoStreamCallbackProc))
		goto RET;