开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > 多媒体 > IP电话/视频会议 > 查看源码
AVIOMgr.cpp
资源名称：网络视频电话系统.rar　[点击查看]
上传用户：oldpeter23
上传日期：2013-01-09
资源大小：1111k
文件大小：28k
源码类别：

IP电话/视频会议

开发平台：

Visual C++

AVIOMgr.cpp：源码内容
  1. //NetTalk
  2. /*------------------------------------------------------------------------------*
  3.  =============================
  4.    模块名称: AVIOMgr.cpp
  5.  =============================
  6.  
  7.  [版权]
  8.  
  9.    2000-2002  115软件工厂  版权所有
  10.                                               
  11. *------------------------------------------------------------------------------*/
  13. #include "WndX.h"
  14. #include <vfw.h> 
  16. #include "AVIOMgr.h"
  17. #include "UDPSocket.h"
  18. #include "g729a.h"
  19. #include "AudioPlay.h"
  20. #include <stdio.h>
  21. /*------------------------------------------------------------------------------*/
  22. CAVIOMgr* pMgrInst=0;
  25. /*------------------------------------------------------------------------------*/
  26. //声音输入线程，包括声音的压缩
  27. DWORD WINAPI CAVIOMgr::AudioInThreadProc(LPVOID lpParameter)
  28. {
  29. CAVIOMgr* pMgr=(CAVIOMgr*)lpParameter;
  30. MSG msg;
  31. while(GetMessage(&msg,0,0,0))
  32. {
  33. switch(msg.message)
  34. {
  35. case WIM_DATA:
  36. {
  37. WAVEHDR* pWH=(WAVEHDR*)msg.lParam;
  38. waveInUnprepareHeader((HWAVEIN)msg.wParam,pWH,sizeof(WAVEHDR));
  40. if(pWH->dwBytesRecorded!=SIZE_AUDIO_FRAME)
  41. break;//it's not full recorded,i think the wave recorder has ben
  42. //stopped,discard it,and don't do anything!
  43. //
  44. CopyMemory(pMgr->m_AudioLocal,pWH->lpData,SIZE_AUDIO_FRAME);
  45. pMgr->OnEncodeAudioData(pMgr->m_AudioLocal,SIZE_AUDIO_FRAME);
  47. waveInPrepareHeader((HWAVEIN)msg.wParam,pWH,sizeof(WAVEHDR));
  48. waveInAddBuffer((HWAVEIN)msg.wParam,pWH,sizeof(WAVEHDR));
  49. }
  50. break;
  51. }
  52. }
  53. return msg.wParam;
  54. }
  56. /*------------------------------------------------------------------------------*/
  57. //音频输出线程，包括音频解码
  58. DWORD WINAPI CAVIOMgr::AudioOutThreadProc(LPVOID lpParameter)
  59. {
  61. CAVIOMgr* pMgr=(CAVIOMgr*)lpParameter;
  62. MSG msg;
  63. while(GetMessage(&msg,0,0,0))
  64. {
  65. switch(msg.message)
  66. {
  67. case IOM_AUDIO:
  68. pMgr->OnDecodeAudioData((PACK_AUDIO*)msg.lParam,(int)msg.wParam);
  69. break;
  70. case WOM_DONE:
  71. {
  72. WAVEHDR* pwh=(WAVEHDR*)msg.lParam;
  73. waveOutUnprepareHeader((HWAVEOUT)msg.wParam,pwh,sizeof(WAVEHDR));
  74. pMgr->m_iAudioBuf--;
  75. delete []pwh->lpData;//删除Play调用时分配的内存
  76. delete pwh;
  77. }
  78. break;
  79. }
  80. }
  81. return msg.wParam;
  82. }
  84. /*------------------------------------------------------------------------------*/
  85. //视频流回调函数
  86. LRESULT CALLBACK CAVIOMgr::VideoStreamCallbackProc(HWND hWnd, LPVIDEOHDR lpVHdr)
  87. {
  89. CAVIOMgr* pMgr=(CAVIOMgr*)capGetUserData(hWnd);
  90. if(pMgr)
  91. {
  92. //压缩视频数据
  93. pMgr->OnEncodeVideoData((char*)lpVHdr->lpData,lpVHdr->dwBytesUsed);
  95. }
  96. return TRUE;
  97. }
  99. /*------------------------------------------------------------------------------*/
  100. //Socket线程，负责接收数据
  101. DWORD WINAPI CAVIOMgr::SockThreadProc(LPVOID lpParameter)
  102. {
  103. CAVIOMgr* pMgr=(CAVIOMgr*)lpParameter;
  105. char buf[4096];
  106. int iLen=0;
  107. while(1)
  108. {
  109. iLen=pMgr->m_Socket.RecvFrom(buf,4096,(sockaddr*)&pMgr->m_sockaddr);
  110. if(iLen>0)
  111. {
  113. switch(*((short*)buf))//check the flag
  114. {
  115. case FLAG_CMD:
  116. {
  117. //命令
  118. pMgr->OnCommand((PACK_CMD*)buf,iLen);
  119. }
  120. break;
  121. case FLAG_AUDIO:
  122. {
  123. //音频数据
  124. if(pMgr->m_ds.bAudioOut&&
  125. pMgr->m_ds.bAudioCodec&&
  126. pMgr->m_hAudioOut&&
  127. ((PACK_AUDIO*)buf)->session==pMgr->m_session)
  128. {
  129. char* p=new char[iLen];
  130. if(p)
  131. {
  132. CopyMemory(p,buf,iLen);
  133. if(!PostThreadMessage(pMgr->m_dwAudioOutId,IOM_AUDIO,iLen,(LPARAM)p))
  134. delete []p;
  135. }
  136. }
  137. }
  138. break;
  139. case FLAG_VIDEO:
  140. {
  141. //视频数据
  142. if(pMgr->m_ds.bVideoOut&&
  143. pMgr->m_ds.bVideoCodec&&
  144. ((PACK_VIDEO*)buf)->session==pMgr->m_session)
  145. {
  147. pMgr->OnDecodeVideoData((PACK_VIDEO*)buf,iLen);
  149. }
  150. }
  152. break;
  153. }
  154. }
  155. else
  156. {
  157. if(!pMgr->m_Socket.IsSocket())
  158. break;
  159. }//the socket should be closed,that is m_Socket have been
  160. //destroyed,so break the loop and end the thread
  161. }
  162. return 0;
  163. }
  165. /*------------------------------------------------------------------------------*/
  167. CAVIOMgr::CAVIOMgr()
  168. {
  169. pMgrInst=this;
  171. m_hSockThread=NULL;
  173. m_hAudioOut=0;
  174. m_hAudioIn=0;
  176. m_idCmd=0;
  178. m_bEnable=TRUE;
  179. m_bEnableBandAdjust=TRUE;
  181. }
  183. /*------------------------------------------------------------------------------*/
  184. CAVIOMgr::~CAVIOMgr()
  185. {
  186. Destroy();
  187. pMgrInst=0;
  188. }
  190. /*------------------------------------------------------------------------------*/
  191. //初始化SOCKET,指定PORT
  192. BOOL CAVIOMgr::InitSocket(UINT nPort)
  193. {
  195. DestroySocket();
  197. BOOL bRet=FALSE;
  198. if(!m_Socket.Create(nPort))
  199. goto RET;
  200. //创建socket线程(socket为block方式)
  201. m_hSockThread=CreateThread(0,0,SockThreadProc,(LPVOID)this,0,&m_dwSockThreadId);
  202. if(!m_hSockThread)
  203. goto RET;
  206. bRet=TRUE;
  207. RET:
  208. if(!bRet)
  209. DestroySocket();
  211. return bRet;
  212. }
  214. /*------------------------------------------------------------------------------*/
  216. BOOL CAVIOMgr::DestroySocket()
  217. {
  218. m_Socket.Destroy();
  219. //
  221. if(m_hSockThread)
  222. {
  223. BOOL b=FALSE;
  224. DWORD ExitCode;
  225. int Timeout = 50; 
  226. while(Timeout)//等待线程结束，如果到一定时间还没结束，就强制结束
  227. {  //因为Socket已经destroy了，所以socket线程会返回
  228. GetExitCodeThread(m_hSockThread, &ExitCode);
  230. if (ExitCode != STILL_ACTIVE)
  231. {
  232. b=TRUE;
  233. // Thread has ended.
  234. break;
  235. }
  236. else
  237. {
  238. Sleep(10);
  239. }
  241. --Timeout;
  242. }
  243. if(!b)//time out ,terminate it
  244. TerminateThread(m_hSockThread,0);
  246. }
  247. m_hSockThread=NULL;
  248. return TRUE;
  249. }
  251. /*------------------------------------------------------------------------------*/
  252. //设定目标
  253. void CAVIOMgr::SetDst(char *ip, unsigned short port)
  254. {
  255. m_dst.sin_family=AF_INET;
  256. //将计算机名或IP转化为网络地址
  257. m_dst.sin_addr.s_addr=CUDPSocket::Name2Inet(ip);
  258. m_dst.sin_port=htons(port);
  260. }
  262. /*------------------------------------------------------------------------------*/
  263. //初始化音频输入，当初始化完成，录音就开始了
  264. BOOL CAVIOMgr::InitAudioRec()
  265. {
  266. DestroyAudioRec();
  267. BOOL bRet=FALSE;
  268. //创建录音线程
  269. m_hAudioIn=CreateThread(0,0,AudioInThreadProc,this,0,&m_dwAudioInId);
  270. if(!m_hAudioIn)
  271. goto RET;
  272. if(!m_AudioRec.Create(0,m_dwAudioInId,(DWORD)this,CALLBACK_THREAD,SIZE_AUDIO_FRAME))
  273. goto RET;
  274. //开始录音
  275. if(!m_AudioRec.Start())
  276. goto RET;
  278. bRet=TRUE;
  279. RET:
  280. if(!bRet)
  281. {
  282. //如果失败，就向主窗口发送出错消息
  283. PostMessage(m_hwndMainWnd,IOM_NOTIFY,MAKEWPARAM(IOMN_ERR,IOME_AUDIO_IN),(LPARAM)this);
  284. DestroyAudioRec();
  285. }
  287. return bRet;
  289. }
  291. /*------------------------------------------------------------------------------*/
  292. //该死的waveIn函数，结束录音时老是出现死锁（XP下还是会出现这个问题，也不知道怎么解决，苦啊）
  293. BOOL CAVIOMgr::DestroyAudioRec()
  294. {
  296. m_AudioRec.Stop();
  297.     m_AudioRec.Destroy();
  298. if(m_hAudioIn)
  299. {
  300. int t=50;
  301. DWORD ExitCode;
  302. BOOL bEnd=FALSE;
  303. //向录音线程发送退出消息，并等待线程结束
  304. PostThreadMessage(m_dwAudioInId,WM_QUIT,0,0);
  305. while(t)
  306. {
  308. GetExitCodeThread(m_hAudioIn,&ExitCode);
  309. if(ExitCode!= STILL_ACTIVE)
  310. {
  311. bEnd=TRUE;
  312. break;
  313. }
  314. else
  315. Sleep(10);
  316. t--;
  317. }
  318. if(!bEnd)
  319. TerminateThread(m_hAudioIn,0);
  320. m_hAudioIn=0;
  322. }
  324. return TRUE;
  326. }
  328. /*------------------------------------------------------------------------------*/
  329. //初始化播放设备
  330. BOOL CAVIOMgr::InitAudioPlay()
  331. {
  332. BOOL bRet=FALSE;
  333. DestroyAudioPlay();
  334. m_iAudioBuf=0;
  335. m_hAudioOut=CreateThread(0,0,AudioOutThreadProc,this,0,&m_dwAudioOutId);
  336. if(!m_hAudioOut)
  337. goto RET;
  338. if(!m_AudioPlay.Create(0,m_dwAudioOutId,(DWORD)this,CALLBACK_THREAD))
  339. goto RET;
  340. bRet=TRUE;
  343. RET:
  344. if(!bRet)
  345. {
  346. //向主窗口发送出错消息
  347. PostMessage(m_hwndMainWnd,IOM_NOTIFY,MAKEWPARAM(IOMN_ERR,IOME_AUDIO_OUT),(LPARAM)this);
  348. DestroyAudioPlay();
  349. }
  351. return bRet;
  352. }
  354. /*------------------------------------------------------------------------------*/
  355. //
  356. BOOL CAVIOMgr::DestroyAudioPlay()
  357. {
  359. m_AudioPlay.Destroy();
  360. if(m_hAudioOut)
  361. {
  362. int t=50;
  363. DWORD ExitCode;
  364. BOOL bEnd=FALSE;
  365. PostThreadMessage(m_dwAudioOutId,WM_QUIT,0,0);
  366. while(t)
  367. {
  369. GetExitCodeThread(m_hAudioOut,&ExitCode);
  370. if(ExitCode!= STILL_ACTIVE)
  371. {
  372. bEnd=TRUE;
  373. break;
  374. }
  375. else
  376. Sleep(10);
  377. t--;
  378. }
  379. if(!bEnd)
  380. TerminateThread(m_hAudioOut,0);
  381. m_hAudioOut=0;
  383. }
  384. return TRUE;
  385. }
  387. /*------------------------------------------------------------------------------*/
  388. //音频解码，并将音频数据发送到想要数据的窗口
  389. void CAVIOMgr::OnDecodeAudioData(PACK_AUDIO *ppa, int len)
  390. {
  392. if(m_CodecMgr.DecodeAudioData((char*)ppa->data,SIZE_AUDIO_PACKED,m_AudioRemote,0))
  393. {
  394. //为了避免延迟过长，当累积的缓冲超过六块时抛弃即将加入的缓冲
  395. if(m_iAudioBuf<6)
  396. {
  397. m_iAudioBuf++;
  398. m_AudioPlay.Play(m_AudioRemote,SIZE_AUDIO_FRAME);
  399. }
  400. if(m_hwndRemoteAudioRcv)
  401. SendMessage(m_hwndRemoteAudioRcv,IOM_AUDIO,1,(LPARAM)this);
  402. }
  404. m_uDataRcv+=sizeof(PACK_AUDIO);
  405. delete []ppa;
  407. }
  409. /*------------------------------------------------------------------------------*/
  410. //压缩音频原始数据并发送出去
  411. void CAVIOMgr::OnEncodeAudioData(char *pa, int len)
  412. {
  414. m_AudioPack.flag=FLAG_AUDIO;
  415. m_AudioPack.session=m_session;
  417. if(m_CodecMgr.EncodeAudioData(pa,len,(char*)m_AudioPack.data,0))
  418. {
  419. m_Socket.SendTo((char*)&m_AudioPack,sizeof(PACK_AUDIO),(sockaddr*)&m_dst);
  420. m_uDataSend+=sizeof(PACK_AUDIO);
  421. }
  423. if(m_hwndLocalAudioRcv)
  424. SendMessage(m_hwndLocalAudioRcv,IOM_AUDIO,0,(LPARAM)this);
  426. }
  428. /*------------------------------------------------------------------------------*/
  429. //视频解码
  430. void CAVIOMgr::OnDecodeVideoData(PACK_VIDEO *ppv, int len)
  431. {
  433. if(m_CodecMgr.DecodeVideoData(((char*)ppv)+sizeof(PACK_VIDEO),ppv->data_size,m_VideoRemote,0,0))
  434. {
  435. m_nFrameCount++;
  437. m_nCurVid=ppv->id;
  438. //将视频数据发送到视频显示窗口
  439. if(m_hwndRemoteVideoRcv)
  440. SendMessage(m_hwndRemoteVideoRcv,IOM_VIDEO,1,(LPARAM)this);
  442. }
  443. //统计收到的数据
  444. m_uDataRcv+=sizeof(PACK_VIDEO)+ppv->data_size;
  445. }
  447. /*------------------------------------------------------------------------------*/
  448. //视频压缩并发送
  449. void CAVIOMgr::OnEncodeVideoData(char *pv, int len)
  450. {
  452. int rlen;
  454. if(m_bVideoSend)
  455. {
  456. if(m_CodecMgr.EncodeVideoData(pv,len,m_VideoPack+sizeof(PACK_VIDEO),&rlen,0))
  457. {
  459. ((PACK_VIDEO*)m_VideoPack)->data_size=(unsigned short)rlen;
  460. ((PACK_VIDEO*)m_VideoPack)->id=m_idVideo++;
  461. ((PACK_VIDEO*)m_VideoPack)->session=m_session;
  462. m_Socket.SendTo(m_VideoPack,rlen+sizeof(PACK_VIDEO),(sockaddr*)&m_dst);
  463. //统计发送的数据
  464. m_uDataSend+=rlen+sizeof(PACK_VIDEO);
  465. }
  466. }
  467. m_VideoLocal=pv;
  468. if(m_hwndLocalVideoRcv)
  469. {
  470. //将视频数据发送到视频显示窗口
  471. SendMessage(m_hwndLocalVideoRcv,IOM_VIDEO,0,(LPARAM)this);
  472. }
  475. }
  477. /*------------------------------------------------------------------------------*/
  478. //初始化，创建SOCKET，并监听
  479. BOOL CAVIOMgr::Init(UINT nPort)
  480. {
  481. Destroy();
  483. m_hwndLocalVideoRcv=0;
  484. m_hwndLocalAudioRcv=0;
  485. m_hwndRemoteVideoRcv=0;
  486. m_hwndRemoteAudioRcv=0;
  487. m_hwndMainWnd=0;
  488. m_bVideoSend=TRUE;
  490. m_iStatus=STA_FREE;
  491. BOOL bRet=FALSE;
  492. m_uDataRcv=0;
  493. m_uDataSend=0;
  494. if(!InitSocket(nPort))
  495. goto RET;
  496. //创建消息接收窗口，主要用于超时重发机制，因为要设定很多不同ID的定时器
  497. m_MsgRcvWnd.Create(0,0,0,0,CRectX(0,0,0,0),0,0);
  498. bRet=TRUE;
  499. RET:
  500. return bRet;
  501. }
  503. /*------------------------------------------------------------------------------*/
  504. //初始化视频捕捉设备
  505. BOOL CAVIOMgr::InitCap()//
  506. {
  507. HWND hCap;
  509. m_ViCap.Destroy();
  510. BOOL bRet=FALSE;
  511. if(!m_ViCap.Init())
  512. goto RET;
  513. //得到驱动数
  514. if(!m_ViCap.GetDriverNum())
  515. goto RET;
  516. //连接到驱动
  517. if(!m_ViCap.ConnectToDriver(0))
  518. goto RET;
  519. hCap=m_ViCap.GetCapWindow();
  520. //设置视频格式
  521. if(!capSetVideoFormat(hCap,&m_CodecMgr.m_BmpU,sizeof(BITMAPINFO)))
  522. goto RET;
  523. //设置视频流回调函数
  524. if(!capSetCallbackOnVideoStream(hCap,VideoStreamCallbackProc))
  525. goto RET;
  526. //将user数据设为AVIOMgr对象指针
  527. if(!capSetUserData(hCap,(DWORD)this))
  528. goto RET;
  529. //开始视频流
  530. if(!capCaptureSequenceNoFile(hCap))
  531. goto RET;
  533. bRet=TRUE;
  534. RET:
  535. if(!bRet)
  536. {
  537. PostMessage(m_hwndMainWnd,IOM_NOTIFY,MAKEWPARAM(IOMN_ERR,IOME_VIDEO_IN),(LPARAM)this);
  538. m_ViCap.Destroy();
  539. }
  540. return bRet;
  541. }
  543. /*------------------------------------------------------------------------------*/
  544. BOOL CAVIOMgr::DestroyCap()
  545. {
  547. m_ViCap.Destroy();
  548. return TRUE;
  549. }
  551. /*------------------------------------------------------------------------------*/
  552. void CAVIOMgr::Destroy()
  553. {
  554. EndTalk();
  555. DestroySocket();
  556. DestroyAudioRec();
  557. DestroyCap();
  558. DestroyAudioPlay();
  559. DestroyAudioCodec();
  560. DestroyVideoCodec();
  561. }
  563. /*------------------------------------------------------------------------------*/
  564. //
  565. BOOL CAVIOMgr::InitAudioCodec()
  566. {
  568. BOOL bRet=FALSE;
  570. if(!m_CodecMgr.InitCodecA())
  571. goto RET;
  572. //启动丢包率定时器
  573. SetTimer(m_hwndMainWnd,100,1000,DropRateCounter);
  574. bRet=TRUE;
  575. RET:
  576. if(!bRet)
  577. {
  578. PostMessage(m_hwndMainWnd,IOM_NOTIFY,MAKEWPARAM(IOMN_ERR,IOME_AUDIO_CODEC),(LPARAM)this);
  579. DestroyAudioCodec();
  580. }
  581. return bRet;
  582. }
  584. /*------------------------------------------------------------------------------*/
  585. //
  586. void CAVIOMgr::DestroyAudioCodec()
  587. {
  588. //关闭丢包率定时器
  589. KillTimer(m_hwndMainWnd,100);
  590. m_CodecMgr.DestroyCodecA();
  591. }
  593. /*------------------------------------------------------------------------------*/
  594. //初始化视频编码解码器
  595. BOOL CAVIOMgr::InitVideoCodec()
  596. {
  597. m_idVideo=0;
  598. m_nFps=0;
  599. m_nFrameCount=0;
  600. m_nLastFrameCount=0;
  601. m_nDropRate=0;
  602. m_nCurVid=0;
  603. m_nLastVid=0;
  604. BOOL bRet=FALSE;
  605. //
  606. if(!m_CodecMgr.InitCodecV())
  607. goto RET;
  609. ((PACK_VIDEO*)m_VideoPack)->flag=FLAG_VIDEO;
  612. bRet=TRUE;
  613. RET:
  614. if(!bRet)
  615. {
  616. PostMessage(m_hwndMainWnd,IOM_NOTIFY,MAKEWPARAM(IOMN_ERR,IOME_VIDEO_CODEC),(LPARAM)this);
  617. DestroyVideoCodec();
  618. }
  619. return bRet;
  620. }
  622. /*------------------------------------------------------------------------------*/
  623. void CAVIOMgr::DestroyVideoCodec()
  624. {
  628. m_CodecMgr.DestroyCodecV();
  629. }
  632. /*------------------------------------------------------------------------------*/
  633. void CAVIOMgr::GetDeviceSupport(DEVICE_SUPPORT &ds)
  634. {
  636. }
  638. /*------------------------------------------------------------------------------*/
  639. //呼叫某个IP
  640. BOOL CAVIOMgr::Call(char *ip, unsigned short port)
  641. {
  642. BOOL bRet=FALSE;
  643. if(m_iStatus!=STA_FREE)
  644. goto RET;
  645. SetDst(ip,port);
  646. PACK_CMD pc;
  647. m_session=rand()&0xff;
  648. pc.flag=FLAG_CMD;
  649. pc.cmd=CMD_CALL;
  650. pc.type=0;
  651. pc.data_size_extra=0;
  652. pc.ext=VER_AVIO&0xff;//AVIO 的版本
  653. if(!SendCmd(&pc,sizeof(pc),m_dst,TRUE))
  654. goto RET;
  656. m_iStatus=STA_BUSY;
  657. bRet=TRUE;
  658. m_uDataSend=0;
  659. m_uDataRcv=0;
  661. RET:
  662. return bRet;
  663. }
  665. /*------------------------------------------------------------------------------*/
  666. //结束通话
  667. BOOL CAVIOMgr::EndTalk()
  668. {
  669. BOOL bRet=FALSE;
  670. if(m_iStatus==STA_FREE)
  671. goto RET;
  672. if(m_iStatus==STA_BUSY)
  673. {
  674. m_iStatus=STA_FREE;
  676. PACK_CMD pc;
  677. pc.flag=FLAG_CMD;
  679. pc.cmd=CMD_END;
  680. pc.data_size_extra=0;
  681. pc.type=0;
  682. SendCmd(&pc,sizeof(pc),m_dst,FALSE);
  683. ZeroMemory(&m_dst,sizeof(m_dst));
  684. DestroyVideoCodec();
  685. DestroyAudioCodec();
  686. DestroyAudioPlay();
  687. DestroyAudioRec();
  688. DestroyCap();
  691. }
  693. bRet=TRUE;
  694. RET:
  695. return bRet;
  696. }
  698. /*------------------------------------------------------------------------------*/
  699. //接受请求
  700. BOOL CAVIOMgr::Accept()
  701. {
  702. BOOL bRet=FALSE;
  703. if(m_iStatus!=STA_BUSY)
  704. goto RET;
  705. PACK_CMD pc;
  706. pc.flag=FLAG_CMD;
  707. pc.cmd=CMD_ACCEPT;
  709. pc.data_size_extra=0;
  710. pc.type=0;
  711. SendCmd(&pc,sizeof(pc),m_dst,TRUE);
  713. ZeroMemory(&m_ds,sizeof(m_ds));
  714. m_ds.bVideoOut=true;//always ok
  715. if(InitVideoCodec())//why this must ahead InitAudioCodec??i don't know why
  716. m_ds.bVideoCodec=true;
  717. if(InitAudioCodec())
  718. m_ds.bAudioCodec=true;
  719. if(InitAudioPlay())
  720. m_ds.bAudioOut=true;
  721. if(InitAudioRec())
  722. m_ds.bAudioIn=true;
  724. if(m_ds.bVideoCodec&&InitCap())
  725. m_ds.bVideoIn=true;
  727. if(!m_ds.bAudioCodec&&!m_ds.bVideoCodec)
  728. {
  729. DestroyCap();
  730. DestroyAudioRec();
  731. DestroyAudioPlay();
  732. DestroyAudioCodec();
  733. DestroyVideoCodec();
  735. m_iStatus=STA_FREE;
  736. }
  737. bRet=TRUE;
  738. RET:
  739. return bRet;
  741. }
  743. /*------------------------------------------------------------------------------*/
  744. //拒绝请求
  745. BOOL CAVIOMgr::Refuse()
  746. {
  747. BOOL bRet=FALSE;
  748. if(m_iStatus!=STA_BUSY)
  749. goto RET;
  750. PACK_CMD pc;
  751. pc.flag=FLAG_CMD;
  752. pc.cmd=CMD_REFUSE;
  754. pc.ext=R_USER;
  755. pc.data_size_extra=0;
  756. pc.type=0;
  757. m_iStatus=STA_FREE;
  759. SendCmd(&pc,sizeof(pc),m_dst,TRUE);
  761. bRet=TRUE;
  762. RET:
  763. return bRet;
  764. }
  766. /*------------------------------------------------------------------------------*/
  767. //解释发送来的命令
  768. void CAVIOMgr::OnCommand(PACK_CMD *ppc, int len)
  769. {
  770. switch(ppc->cmd)
  771. {
  772. case CMD_CALL:
  773. {
  774. if(ppc->type==0)
  775. {
  776. if(m_bEnable)
  777. {
  778. //如果是空闲的，就保存地址，设置状态为工作状态
  779. //并向主窗口发送呼叫通知
  780. if(m_iStatus==STA_FREE)
  781. {
  782. m_dst=m_sockaddr;
  783. m_iStatus=STA_BUSY;
  784. m_session=ppc->session;
  785. PostMessage(m_hwndMainWnd,IOM_NOTIFY,MAKEWPARAM(IOMN_CALL,VER_AVIO),(LPARAM)this);
  786. //返回请求确认
  787. PACK_CMD pc;
  788. pc.flag=FLAG_CMD;
  789. pc.data_size_extra=0;
  791. pc.cmd=CMD_CALL;
  792. pc.ext=VER_AVIO&0xff;// AVIO的版本
  793. pc.type=1;
  794. pc.id=ppc->id;
  795. pc.session=m_session;
  796. m_Socket.SendTo((char*)&pc,sizeof(pc),(sockaddr*)&m_sockaddr);
  799. }
  800. else
  801. if(m_iStatus==STA_BUSY&&m_session!=ppc->session)
  802. {
  803. /*//有另一台机请求通话，拒绝！
  805. PACK_CMD pc;
  806. pc.flag=FLAG_CMD;
  807. pc.cmd=CMD_REFUSE;
  808. pc.data_size_extra=0;
  809. pc.ext=R_BUSY;
  810. pc.type=0;
  811. pc.id=ppc->id;
  812. pc.session=ppc->session;
  813. m_Socket.SendTo((char*)&pc,sizeof(pc),(sockaddr*)&m_sockaddr);*/
  814. }
  815. }
  816. }
  817. //回应
  818. else
  819. {
  820. if(ppc->session==m_session)
  821. {
  822. KillTimer(m_MsgRcvWnd,ppc->id);
  823. m_CmdQueue.Remove(ppc->id);
  824. PostMessage(m_hwndMainWnd,IOM_NOTIFY,IOMN_CONNECTED,(LPARAM)this);
  825. }
  826. }
  827. }break;
  828. case CMD_END:
  829. {
  830. if(ppc->session==m_session)
  831. {
  832. if(ppc->type==0)
  833. {
  835. if(m_iStatus==STA_BUSY)
  836. {
  837. PostMessage(m_hwndMainWnd,IOM_NOTIFY,MAKEWPARAM(IOMN_END,E_USER),(LPARAM)this);
  838. }
  839. PACK_CMD pc;
  840. pc.flag=FLAG_CMD;
  841. pc.data_size_extra=0;
  842. pc.cmd=CMD_END;
  843. pc.type=1;
  844. pc.id=ppc->id;
  845. pc.session=m_session;
  846. m_Socket.SendTo((char*)&pc,sizeof(pc),(sockaddr*)&m_sockaddr);
  848. }
  849. else
  850. {
  851. KillTimer(m_MsgRcvWnd,ppc->id);
  852. m_CmdQueue.Remove(ppc->id);
  853. }
  854. }
  856. }
  857. break;
  858. case CMD_REFUSE:
  859. {
  860. if(ppc->session==m_session)
  861. {
  863. if(ppc->type==0)
  864. {
  866. if(m_iStatus==STA_BUSY)
  867. {
  868. m_iStatus=STA_FREE;
  869. PostMessage(m_hwndMainWnd,IOM_NOTIFY,MAKEWPARAM(IOMN_REFUSE,ppc->ext),(LPARAM)this);
  870. }
  871. PACK_CMD pc;
  872. pc.flag=FLAG_CMD;
  873. pc.data_size_extra=0;
  874. pc.cmd=CMD_REFUSE;
  875. pc.type=1;
  876. pc.id=ppc->id;
  877. pc.session=m_session;
  878. m_Socket.SendTo((char*)&pc,sizeof(pc),(sockaddr*)&m_sockaddr);
  881. }
  882. else
  883. {
  884. KillTimer(m_MsgRcvWnd,ppc->id);
  885. m_CmdQueue.Remove(ppc->id);
  886. }
  887. }
  888. }
  889. break;
  890. case CMD_ACCEPT:
  891. {
  892. if(ppc->session==m_session)
  893. {
  894. if(ppc->type==0)
  895. {
  897. if(m_iStatus==STA_BUSY)
  898. {
  899. PostMessage(m_hwndMainWnd,IOM_NOTIFY,MAKEWPARAM(IOMN_ACCEPT,VER_AVIO),(LPARAM)this);
  900. }
  901. PACK_CMD pc;
  902. pc.flag=FLAG_CMD;
  903. pc.data_size_extra=0;
  904. pc.cmd=CMD_ACCEPT;
  905. pc.type=1;
  906. pc.id=ppc->id;
  907. pc.session=m_session;
  908. m_Socket.SendTo((char*)&pc,sizeof(pc),(sockaddr*)&m_sockaddr);
  910. }
  911. else
  912. {
  913. KillTimer(m_MsgRcvWnd,ppc->id);
  914. m_CmdQueue.Remove(ppc->id);
  915. }
  916. }
  918. }
  919. break;
  920. case CMD_TXT:
  921. {
  922. if(ppc->type==0)
  923. {
  924. //bug exist
  925. SendMessage(m_hwndMainWnd,IOM_NOTIFY,MAKEWPARAM(IOMN_TXT,0),(LPARAM)(((char*)ppc)+sizeof(PACK_CMD)));
  926. PACK_CMD pc;
  927. pc.flag=FLAG_CMD;
  928. pc.data_size_extra=0;
  929. pc.cmd=CMD_TXT;
  930. pc.type=1;
  931. pc.id=ppc->id;
  932. m_Socket.SendTo((char*)&pc,sizeof(pc),(sockaddr*)&m_sockaddr);
  933. }
  934. else
  935. {
  937. KillTimer(m_MsgRcvWnd,ppc->id);
  938. UINT len;char resend;sockaddr_in addr;
  939. char* p=m_CmdQueue.GetBuf(ppc->id,len,resend,addr);
  940. if(p)
  941. {
  942. SendMessage(m_hwndMainWnd,IOM_NOTIFY,MAKEWPARAM(IOMN_TXT,1),(LPARAM)(p+sizeof(PACK_CMD)));
  943. }
  944. m_CmdQueue.Remove(ppc->id);
  945. }
  946. }
  947. break;
  949. case CMD_DROPRATE:
  950. {
  951. //根据丢包率来调整视频质量
  952. if(ppc->type==0&&ppc->session&&m_bEnableBandAdjust)
  953. {
  954. SetVideoQuality(100-ppc->ext);
  955. SendMessage(m_hwndMainWnd,IOM_NOTIFY,MAKEWPARAM(IOMN_VQSET,100-ppc->ext),(LPARAM)this);
  956. }
  957. }break;
  959. }
  961. }
  963. /*------------------------------------------------------------------------------*/
  964. //以下是设置对外消息接口
  965. //设置主窗口
  966. void CAVIOMgr::SetMainWnd(HWND hwnd)
  967. {
  968. m_hwndMainWnd=hwnd;
  969. }
  971. void CAVIOMgr::SetLocalVideoRcvWnd(HWND hwnd)
  972. {
  973. m_hwndLocalVideoRcv=hwnd;
  974. }
  976. void CAVIOMgr::SetLocalAudioRcvWnd(HWND hwnd)
  977. {
  978. m_hwndLocalAudioRcv=hwnd;
  979. }
  981. void CAVIOMgr::SetRemoteVideoRcvWnd(HWND hwnd)
  982. {
  983. m_hwndRemoteVideoRcv=hwnd;
  984. }
  986. void CAVIOMgr::SetRemoteAudioRcvWnd(HWND hwnd)
  987. {
  988. m_hwndRemoteAudioRcv=hwnd;
  989. }
  991. /*------------------------------------------------------------------------------*/
  992. //得到当前状态
  993. int CAVIOMgr::GetStatus()
  994. {
  995. return m_iStatus;
  996. }
  998. /*------------------------------------------------------------------------------*/
  999. //调用以下函数要谨慎
  1000. //只有在收到相应消息时才能调用
  1001. /////////////////////////
  1002. ///
  1003. char* CAVIOMgr::GetLocalVideo(UINT &len)
  1004. {
  1005. len=m_CodecMgr.m_BmpU.bmiHeader.biSizeImage;
  1006. return m_VideoLocal;
  1007. }
  1009. char* CAVIOMgr::GetLocalAudio(UINT &len)
  1010. {
  1011. len=SIZE_AUDIO_FRAME;
  1012. return m_AudioLocal;
  1013. }
  1015. char* CAVIOMgr::GetRemoteVideo(UINT &len)
  1016. {
  1017. len=m_CodecMgr.m_BmpU.bmiHeader.biSizeImage;
  1018. return m_VideoRemote;
  1019. }
  1021. char* CAVIOMgr::GetRemoteAudio(UINT &len)
  1022. {
  1023. len=SIZE_AUDIO_FRAME;
  1024. return m_AudioRemote;
  1025. }
  1027. /*------------------------------------------------------------------------------*/
  1028. //得到原始视频格式
  1029. BITMAPINFO* CAVIOMgr::GetBitampInfo()
  1030. {
  1031. return &m_CodecMgr.m_BmpU;
  1032. }
  1034. /*------------------------------------------------------------------------------*/
  1035. //得到数据发送量
  1036. UINT CAVIOMgr::GetDataSend()
  1037. {
  1038. return m_uDataSend;
  1039. }
  1040. /*------------------------------------------------------------------------------*/
  1041. //得到数据接收量
  1042. UINT CAVIOMgr::GetDataRcv()
  1043. {
  1044. return m_uDataRcv;
  1045. }
  1046. /*------------------------------------------------------------------------------*/
  1047. //设置是否传输视频
  1048. void CAVIOMgr::VideoSend(BOOL b)
  1049. {
  1050. m_bVideoSend=b;
  1052. }
  1053. /*------------------------------------------------------------------------------*/
  1054. //得到对方地址
  1055. sockaddr_in CAVIOMgr::GetRemoteAddr()
  1056. {
  1057. return m_dst;
  1058. }
  1059. /*------------------------------------------------------------------------------*/
  1060. //设置视频质量
  1061. void CAVIOMgr::SetVideoQuality(UINT q)
  1062. {
  1063. m_CodecMgr.m_cv.lQ=q*100;
  1064. }
  1065. /*------------------------------------------------------------------------------*/
  1066. //预览
  1067. BOOL CAVIOMgr::EnablePreview(BOOL b)
  1068. {
  1069. if(m_iStatus==STA_FREE&&b)
  1070. {
  1071. return InitCap();
  1072. }
  1073. else
  1074. if(m_iStatus==STA_FREE&&!b)
  1075. {
  1076. DestroyCap();
  1078. }
  1079. return TRUE;
  1080. }
  1082. /*------------------------------------------------------------------------------*/
  1083. //当对方接收请求后，会受到Accept的消息，这时必须调用此函数，来打开各种设备
  1084. void CAVIOMgr::OnAccept()
  1085. {
  1086. if(m_iStatus==STA_FREE) return;
  1087. ZeroMemory(&m_ds,sizeof(m_ds));
  1089. m_ds.bVideoOut=true;//always ok
  1090. if(InitVideoCodec())
  1091. m_ds.bVideoCodec=true;
  1092. if(InitAudioCodec())
  1093. m_ds.bAudioCodec=true;
  1094. if(InitAudioPlay())
  1095. m_ds.bAudioOut=true;
  1096. if(InitAudioRec())
  1097. m_ds.bAudioIn=true;
  1099. if(m_ds.bVideoCodec&&InitCap())
  1100. m_ds.bVideoIn=true;
  1102. if(!m_ds.bAudioCodec&&!m_ds.bVideoCodec)
  1103. {
  1104. DestroyCap();
  1105. DestroyAudioRec();
  1106. DestroyAudioPlay();
  1107. DestroyAudioCodec();
  1108. DestroyVideoCodec();
  1110. m_iStatus=STA_FREE;
  1111. }
  1115. }
  1117. /*------------------------------------------------------------------------------*/
  1118. //得到帧数
  1119. UINT CAVIOMgr::GetFrameCount()
  1120. {
  1121. return m_nFrameCount;
  1122. }
  1124. /*------------------------------------------------------------------------------*/
  1125. //发送命令（比较可靠）
  1126. BOOL CAVIOMgr::SendCmd(PACK_CMD *pCmd, UINT nLen,sockaddr_in &addr, BOOL bFlag)
  1127. {
  1128. if(!pCmd) return FALSE;
  1129. pCmd->id=m_idCmd++;
  1130. pCmd->session=m_session;
  1132. int l=m_Socket.SendTo((char*)pCmd,nLen,(sockaddr*)&addr);
  1133. if(l>0)
  1134. {
  1135. if(bFlag)
  1136. {
  1137. m_CmdQueue.Add(pCmd->id,(char*)pCmd,nLen,addr);//把命令加入命令表
  1138. SetTimer(m_MsgRcvWnd,pCmd->id,800,CmdTimeOutProc);
  1139. }
  1140. return TRUE;
  1141. }
  1142. else
  1143. return FALSE;
  1144. }
  1146. /*------------------------------------------------------------------------------*/
  1147. //命令超时重发
  1148. void CALLBACK CAVIOMgr::CmdTimeOutProc(HWND hwnd,UINT uMsg,UINT idEvent,DWORD dwTime)
  1149. {
  1150. if(pMgrInst)
  1151. {
  1152. UINT len;char resend;sockaddr_in addr;
  1153. char* p=pMgrInst->m_CmdQueue.GetBuf(idEvent,len,resend,addr);//找出相应ID的命令(已经超时,需要重发)
  1154. if(p)
  1155. {
  1156. if(resend<5)
  1157. pMgrInst->m_Socket.SendTo(p,len,(sockaddr*)&addr);
  1158. else//当重发次数超过4次时,宣布失败
  1159. {
  1160. KillTimer(hwnd,idEvent);//
  1162. if(pMgrInst)
  1163. {
  1164. PACK_CMD* pCmd=(PACK_CMD*)p;
  1165. switch(pCmd->cmd)
  1166. {
  1167. case CMD_CALL:
  1168. {
  1169. PostMessage(pMgrInst->m_hwndMainWnd,IOM_NOTIFY,MAKEWPARAM(IOMN_REFUSE,R_TIMEOUT),(LPARAM)pMgrInst);
  1170. pMgrInst->m_iStatus=STA_FREE;
  1172. }
  1173. break;
  1174. case CMD_REFUSE:
  1175. break;
  1176. case CMD_ACCEPT:
  1177. break;
  1178. case CMD_END:
  1179. break;
  1180. case CMD_TXT:
  1181. break;
  1182. }
  1183. }
  1184. pMgrInst->m_CmdQueue.Remove(idEvent);//把命令从表中移除
  1185. }
  1186. }
  1187. else
  1188. KillTimer(hwnd,idEvent);
  1189. }
  1190. }
  1192. /*------------------------------------------------------------------------------*/
  1193. //丢包率计算(包括帧频率计算)
  1194. void CALLBACK CAVIOMgr::DropRateCounter(HWND hwnd,UINT uMsg,UINT idEvent,DWORD dwTime)
  1195. {
  1196. //计算丢包率
  1197. if(pMgrInst)
  1198. {
  1199. UINT delt;
  1200. if(pMgrInst->m_nCurVid<pMgrInst->m_nLastVid)
  1201. delt=pMgrInst->m_nCurVid+256-pMgrInst->m_nLastVid;
  1202. else
  1203. delt=pMgrInst->m_nCurVid-pMgrInst->m_nLastVid;
  1204. if(delt>0)
  1205. pMgrInst->m_nDropRate=100-(pMgrInst->m_nFrameCount-pMgrInst->m_nLastFrameCount)*100/delt;
  1206. pMgrInst->m_nLastVid=pMgrInst->m_nCurVid;
  1207. //发送丢包率
  1208. PACK_CMD pc;
  1209. pc.flag=FLAG_CMD;
  1210. pc.cmd=CMD_DROPRATE;
  1211. pc.session=pMgrInst->m_session;
  1212. pc.type=0;
  1213. pc.data_size_extra=0;
  1214. pc.ext=pMgrInst->m_nDropRate;
  1215. pMgrInst->m_Socket.SendTo((char*)&pc,sizeof(PACK_CMD),(SOCKADDR*)&pMgrInst->m_dst);
  1216. //计算帧频率
  1217. pMgrInst->m_nFps=(pMgrInst->m_nFrameCount-pMgrInst->m_nLastFrameCount);
  1220. pMgrInst->m_nLastFrameCount=pMgrInst->m_nFrameCount;
  1221. }
  1223. }
  1225. /*------------------------------------------------------------------------------*/
  1226. UINT CAVIOMgr::GetFps()
  1227. {
  1228. return m_nFps;
  1229. }
  1231. /*------------------------------------------------------------------------------*/
  1232. //发送文本信息
  1233. BOOL CAVIOMgr::SendTxt(char *ip, unsigned short port, char *txt)
  1234. {
  1235. BOOL bRet=FALSE;
  1237. PACK_CMD *ppc=(PACK_CMD*)new char[sizeof(PACK_CMD)+strlen(txt)+1];
  1238. strcpy(((char*)ppc)+sizeof(PACK_CMD),txt);
  1239. ppc->flag=FLAG_CMD;
  1240. ppc->cmd=CMD_TXT;
  1241. ppc->type=0;
  1242. ppc->data_size_extra=strlen(txt)+1;
  1243. //将ip和port转化
  1244. sockaddr_in addr;
  1245. addr.sin_family=AF_INET;
  1246. addr.sin_addr.s_addr=CUDPSocket::Name2Inet(ip);
  1247. addr.sin_port=htons(port);
  1248. //
  1249. if(!SendCmd(ppc,sizeof(PACK_CMD)+ppc->data_size_extra,addr,TRUE))
  1250. goto RET;
  1252. bRet=TRUE;
  1254. RET:
  1255. return bRet;
  1256. }
  1258. /*------------------------------------------------------------------------------*/
  1259. //得到前一次发来数据的地址
  1260. sockaddr_in& CAVIOMgr::GetCurrentAddr()
  1261. {
  1262. return m_sockaddr;
  1263. }
  1265. /*------------------------------------------------------------------------------*/
  1266. //
  1267. BOOL CAVIOMgr::Enable(BOOL bEnable)
  1268. {
  1269. BOOL t=m_bEnable;
  1270. m_bEnable=bEnable;
  1271. return t;
  1272. }
  1274. /*------------------------------------------------------------------------------*/
  1275. //是否自动调整图象质量
  1276. BOOL CAVIOMgr::EnableBandAdjust(BOOL bEnable)
  1277. {
  1278. BOOL t=m_bEnableBandAdjust;
  1279. m_bEnableBandAdjust=bEnable;
  1280. return t;
  1281. }