avclass.h

基于ARM9(Wince)的网络音视频通信

#pragma once
#include "winsock2.h"

//UDP支持
#include "..\UDP\UDP.h"

//音频支持
#include "WaveIn.h"
#include "waveout.h"

//G726支持
#include "g726.h"

//视频支持
#include "Gdiplus.h"
using namespace Gdiplus; 


#define VideoData_Size 1440 //每块视频数据包的大小
#define Video_Width 320 //视频宽度
#define Video_Height 240 //视频长度
#define AudioData_Size 960  //每块音频数据包的大小
#define Compr_AudioData_Size 120  //压缩后音频块的大小


//音频输入输出变量
CWaveIn *g_pIn;
CWaveOut *g_pOut;	
char pin[AudioData_Size],pout[Compr_AudioData_Size];
char wave_data[AudioData_Size];

//UDP变量
CUDP_CE m_CEUdp;


//视频输入变量
GdiplusStartupInput m_gdiPlusInPut;
ULONG_PTR m_gdiPlusToken;
char video_data[Video_Width*Video_Height];
int index;//视频数据当前索引
class AVClass
{
private:
	
public:

	
    //=====================================================================
	//	语法格式：	void InitAV(CWnd * p)
	//	实现功能：	初始化音频和视频，用于录音、播放音频，以及播放视频
	//	参数：	  	p为窗口类指针
	//	返回值：  	无
	//=====================================================================
	void InitAV(CWnd * p,int local_port,CString remote_ip,int remote_port)
	{
		//-------------------------UDP连接--------------------------//
		m_CEUdp.m_OnUdpRecv = OnUdpCERecv;
		DWORD nResult = m_CEUdp.Open(p,local_port,remote_ip,remote_port);
		if (nResult <=0) 
		{
			AfxMessageBox(_T("打开端口失败"));
			return;
		}

		//-------------------------音频--------------------------//
		g_pOut = new CWaveOut();
		g_pIn = new CWaveIn();
		g_pOut->StartPlay();
		g_pIn->StartRec(OnRecording,(DWORD)p);	
		//-------------------------视频--------------------------//
		GdiplusStartup( &m_gdiPlusToken, &m_gdiPlusInPut, NULL ); //初始化GDI+ 
		memset(video_data,0,Video_Width*Video_Height);
		index=0;

	}

	//=====================================================================
	//	语法格式：	void FreeAV()
	//	实现功能：	释放音频、视频
	//	参数：	  	无
	//	返回值：  	无
	//=====================================================================
	void FreeAV()
	{
		
		//-------------------------音频--------------------------//
		g_pOut->StopPlay();
		g_pIn->StopRec();
		delete g_pOut;
		delete g_pIn;
		//-------------------------视频--------------------------//		
		GdiplusShutdown(m_gdiPlusToken); //销毁GDI+

	    //------------------------UDP--------------------------//
		m_CEUdp.Close();

	}

	//=====================================================================
	//	语法格式：	void RecAndPlay(WPARAM wParam,LPARAM lParam,HWND hwnd)
	//	实现功能：	接收网络传来的音频，以及播放
	//	参数：	  	wParam,表示数据；lParam，表示数据长度；hwnd，表示显示视频的窗口句柄
	//	返回值：  	无
	//=====================================================================
	 static void CALLBACK OnUdpCERecv(CWnd *pWnd,char* buf,int nLen,sockaddr * addr)
	 {
		/*测试收到的数据大小
		
	    CString tmp;
		tmp.Format(L"%d",nLen);
		MessageBox(0,tmp,0,0);
        return;*/

		//-------------------------如果是音频数据--------------------------//
		if(nLen==Compr_AudioData_Size)
		{
			g726_Decode(buf,(unsigned char*)wave_data);
			g_pOut->Play(wave_data,AudioData_Size);
			return;
		}
		//-------------------------如果是视频数据--------------------------//
      
		if(nLen==VideoData_Size)//完整的视频数据块
		{
            for(int i=0;i<nLen;i++)
			{
				video_data[index]=buf[i];
				index++;
			}
			return;
		}

		//视频数据块的最后一块
		for(int i=0;i<nLen;i++)
		{
			video_data[index]=buf[i];
			index++;
		}

		//如果JPEG图像特别大，则肯定是出错，则抛弃
		if(index>Video_Width*Video_Height)
		{
			//MessageBox(0,"缓冲区出错","错误信息",0);
			return;
		}
		
	
		try{
			IPicture *pPic;
			IStream *pStm ; 
			//分配全局存储空间 
			HGLOBAL hGlobal=GlobalAlloc(GMEM_MOVEABLE,index);
			LPVOID pvData=NULL ;
			//锁定分配内存块 
			pvData=GlobalLock(hGlobal);
			//复制数据包video_data到pvData
			memcpy(pvData,video_data,index);

			GlobalUnlock(hGlobal);
			CreateStreamOnHGlobal(hGlobal,TRUE,&pStm);

			ULARGE_INTEGER        pSeek;       
			LARGE_INTEGER    dlibMove  ={  0  }  ;  
			pStm->Seek(dlibMove,STREAM_SEEK_SET  ,&pSeek);  
		//	Sleep(15);

			//装入图形文件
			if(FAILED(OleLoadPicture(pStm,index,TRUE,IID_IPicture,(LPVOID*)&pPic)))
			{//附:如果video_data这个数组包含的图像有错，则OleLoadPicture 容易产生读写内存错误
			//	pPic->Release();
			//	pStm->Release();
				return ;
			}

			Image img(pStm,0);
			Graphics mGraphics(GetDC(pWnd->m_hWnd));
			mGraphics.DrawImage(&img, 0, 0, Video_Width, Video_Height);
			img.~Image();//会出错
			mGraphics.~Graphics();
			pPic->Release();
			pStm->Release();
		}
		catch(CException * e)
		{}


		
		memset(video_data,0,Video_Width*Video_Height);
		index=0;

	 }

	//=====================================================================
	//	语法格式：	static void OnRecording(char *data,int length,DWORD userdata)
	//	实现功能：	释放音频
	//	参数：	  	data表示数据,length表示数据长度,userdata暂时没用
	//	返回值：  	无
	//=====================================================================
	static void OnRecording(char *data,int length,DWORD userdata)
	{
		memcpy(pin,g_pIn->buffer,AudioData_Size);	
		g726_Encode((unsigned char*)pin,pout);
		m_CEUdp.SendData(pout,Compr_AudioData_Size);
	}


};