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

源码开发语言/平台
当前位置：首页 > 源码/资料 > Internet/网络编程 > P2P编程 > 查看源码
GVCaptureDlg.cpp
资源名称：p2p_vod.rar　[点击查看]
上传用户：liguizhu
上传日期：2015-11-01
资源大小：2422k
文件大小：27k
源码类别：

P2P编程

开发平台：

Visual C++

GVCaptureDlg.cpp：源码内容
  1. /*
  2.  *  Openmysee
  3.  *
  4.  *  This program is free software; you can redistribute it and/or modify
  5.  *  it under the terms of the GNU General Public License as published by
  6.  *  the Free Software Foundation; either version 2 of the License, or
  7.  *  (at your option) any later version.
  8.  *
  9.  *  This program is distributed in the hope that it will be useful,
  10.  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  11.  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12.  *  GNU General Public License for more details.
  13.  *
  14.  *  You should have received a copy of the GNU General Public License
  15.  *  along with this program; if not, write to the Free Software
  16.  *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  17.  *
  18.  */
  19. //
  20. //xiezhouwei 2005-08-15 添加配置pin函数
  22. #include "stdafx.h"
  23. #include "GVCapture.h"
  24. #include "GVCaptureDlg.h"
  25. #include "utilities.h"
  26. #include "afxdlgs.h"
  27. #include "VideoEncParams.h"
  28. #include "SetV9Dlg.h"
  29. #include ".gvcapturedlg.h"
  32. #ifdef _DEBUG
  33. #define new DEBUG_NEW
  34. #undef THIS_FILE
  35. static char THIS_FILE[] = __FILE__;
  36. #endif
  39. CVideoEncParams g_VideoEncodingParams;
  41. /////////////////////////////////////////////////////////////////////////////
  42. // CAboutDlg dialog used for App About
  44. class CAboutDlg : public CDialog
  45. {
  46. public:
  47. CAboutDlg();
  49. // Dialog Data
  50. //{{AFX_DATA(CAboutDlg)
  51. enum { IDD = IDD_ABOUTBOX };
  52. //}}AFX_DATA
  54. // ClassWizard generated virtual function overrides
  55. //{{AFX_VIRTUAL(CAboutDlg)
  56. protected:
  57. virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV support
  58. //}}AFX_VIRTUAL
  60. // Implementation
  61. protected:
  62. //{{AFX_MSG(CAboutDlg)
  63. //}}AFX_MSG
  64. DECLARE_MESSAGE_MAP()
  65. };
  67. CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
  68. {
  69. //{{AFX_DATA_INIT(CAboutDlg)
  70. //}}AFX_DATA_INIT
  71. }
  73. void CAboutDlg::DoDataExchange(CDataExchange* pDX)
  74. {
  75. CDialog::DoDataExchange(pDX);
  76. //{{AFX_DATA_MAP(CAboutDlg)
  77. //}}AFX_DATA_MAP
  78. }
  80. BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
  81. //{{AFX_MSG_MAP(CAboutDlg)
  82. // No message handlers
  83. //}}AFX_MSG_MAP
  84. END_MESSAGE_MAP()
  86. /////////////////////////////////////////////////////////////////////////////
  87. // CGVCaptureDlg dialog
  89. CGVCaptureDlg::CGVCaptureDlg(CWnd* pParent /*=NULL*/)
  90. : CDialog(CGVCaptureDlg::IDD, pParent)
  91. {
  92. //{{AFX_DATA_INIT(CGVCaptureDlg)
  93. mstrSavePath = _T("");
  94. //}}AFX_DATA_INIT
  95. // Note that LoadIcon does not require a subsequent DestroyIcon in Win32
  96. m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
  98. m_grfCapture.Initialize();
  99. m_dwEscape = 0;
  100. m_bPause = FALSE;
  101. mpTemAudioAMT = NULL;
  103. ZeroMemory(mszAVIFilePath,sizeof(mszAVIFilePath));
  104. }
  106. void CGVCaptureDlg::DoDataExchange(CDataExchange* pDX)
  107. {
  108.     CDialog::DoDataExchange(pDX);
  109.     //{{AFX_DATA_MAP(CGVCaptureDlg)
  110.     DDX_Control(pDX, IDC_CHK_OVERLAY, mchkNeedOverlay);
  111.     DDX_Control(pDX, IDC_CHK_SAVE, mbtnSaveFile);
  112.     DDX_Control(pDX, IDC_CHK_ONLY_AUDIO, m_chkOnlyAudio);
  113.     DDX_Control(pDX, IDC_CHK_PREVIEW, m_chkPreview);
  114.     DDX_Control(pDX, IDC_CMB_VIDEO_COMPRESSOR, m_comboxVideoEncoder);
  115.     DDX_Control(pDX, IDC_CMB_VIDEO_CAPTURE, m_comboxVideoCap);
  116.     DDX_Control(pDX, IDC_CMB_AUDIO_CAPTURE, m_comboxAudioCap);
  117.     DDX_Control(pDX, IDC_CMB_AUDIO_COMPRESSOR, m_comboxAudioEncoder);
  118.     DDX_Text(pDX, IDC_EDIT_SAVEPATH, mstrSavePath);
  119.     //}}AFX_DATA_MAP
  120.     DDX_Control(pDX, IDC_CHK_CROSSBAR, m_bChkCrossBar);
  121. }
  123. BEGIN_MESSAGE_MAP(CGVCaptureDlg, CDialog)
  124. //{{AFX_MSG_MAP(CGVCaptureDlg)
  125. ON_WM_SYSCOMMAND()
  126. ON_WM_PAINT()
  127. ON_WM_ERASEBKGND()
  128. ON_WM_QUERYDRAGICON()
  129. ON_BN_CLICKED(IDC_BTN_VIDEOCAP_CONFIG, OnBtnVideocapConfig)
  130. ON_BN_CLICKED(IDC_BTN_VIDEO_COMPRESSOR_CONFIG, OnBtnVideoCompressorConfig)
  131. ON_BN_CLICKED(IDC_BTN_AUDIO_COMPRESSOR_CONFIG, OnBtnAudioCompressorConfig)
  132. ON_BN_CLICKED(IDC_BTN_BEGIN_CAP, OnBtnBeginCap)
  133. ON_BN_CLICKED(IDC_BTN_PAUSE_CAP, OnBtnPauseCap)
  134. ON_BN_CLICKED(IDC_BTN_STOP_CAP, OnBtnStopCap)
  135. ON_BN_CLICKED(IDC_CHK_PREVIEW, OnChkPreview)
  136. ON_WM_TIMER()
  137. ON_WM_CLOSE()
  138. ON_CBN_SELCHANGE(IDC_CMB_AUDIO_COMPRESSOR, OnSelchangeCmbAudioCompressor)
  139.     ON_CBN_SELCHANGE(IDC_CMB_VIDEO_CAPTURE, OnSelchangeCmbVideoCapture)
  140.     ON_CBN_SELCHANGE(IDC_CMB_AUDIO_CAPTURE, OnCbnSelchangeCmbAudioCapture)
  141. ON_CBN_SELCHANGE(IDC_CMB_VIDEO_COMPRESSOR, OnSelchangeCmbVideoCompressor)
  142. ON_BN_CLICKED(IDC_CHK_ONLY_AUDIO, OnCheckOnlyAudio)
  143. ON_BN_CLICKED(IDC_BTN_SYSCOF, OnBtnSyscof)
  144. ON_BN_CLICKED(IDC_BTN_VIDEOCAP_INI_CONFIG, OnBtnVideocapIniConfig)
  145. ON_BN_CLICKED(IDC_CHK_SAVE, OnChkSave)
  146. ON_BN_CLICKED(IDC_BTN_SETOVERLAY, OnBtnSetoverlay)
  147. ON_BN_CLICKED(IDC_CHK_OVERLAY, OnChkOverlay)
  148. ON_BN_CLICKED(IDC_CHK_CROSSBAR, OnBnClickedChkCrossbar)
  149. //}}AFX_MSG_MAP
  153.     ON_BN_CLICKED(IDC_BTN_AUDIO_CAP_CONFIG, OnBnClickedBtnAudioCapConfig)
  154. END_MESSAGE_MAP()
  156. /////////////////////////////////////////////////////////////////////////////
  157. // CGVCaptureDlg message handlers
  159. BOOL CGVCaptureDlg::OnInitDialog()
  160. {
  161. CDialog::OnInitDialog();
  163. // Add "About..." menu item to system menu.
  165. // IDM_ABOUTBOX must be in the system command range.
  166. ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
  167. ASSERT(IDM_ABOUTBOX < 0xF000);
  169. CMenu* pSysMenu = GetSystemMenu(FALSE);
  170. if (pSysMenu != NULL)
  171. {
  172. CString strAboutMenu;
  173. strAboutMenu.LoadString(IDS_ABOUTBOX);
  174. if (!strAboutMenu.IsEmpty())
  175. {
  176. pSysMenu->AppendMenu(MF_SEPARATOR);
  177. pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
  178. }
  179. }
  181. // Set the icon for this dialog.  The framework does this automatically
  182. //  when the application's main window is not a dialog
  183. SetIcon(m_hIcon, TRUE); // Set big icon
  184. SetIcon(m_hIcon, FALSE); // Set small icon
  186. //mbtnSaveFile.SetCheck(BST_CHECKED);
  187. m_grfCapture.EnumVideoCapDevices(m_strArrayVideoCapDevice);
  188. for(int i = 0; i < m_strArrayVideoCapDevice.GetSize(); i++)
  189.         m_comboxVideoCap.AddString(m_strArrayVideoCapDevice.GetAt(i));
  191.     //mbtnSaveFile.SetCheck(BST_CHECKED);
  192.     m_grfCapture.EnumAudioCapDevices(m_strArrayAudioCapDevice);
  193.     for(int i = 0; i < m_strArrayAudioCapDevice.GetSize(); i++)
  194.         m_comboxAudioCap.AddString(m_strArrayAudioCapDevice.GetAt(i));
  196. m_grfCapture.EnumVideoEncoders(m_strArrayVideoEncoders);
  197. for(i = 0; i < m_strArrayVideoEncoders.GetSize(); i++)
  198. m_comboxVideoEncoder.AddString(m_strArrayVideoEncoders.GetAt(i));
  199. m_comboxVideoEncoder.AddString("MicroSoft Media 9 DMO");
  201. m_grfCapture.EnumAudioEncoders(m_strArrayAudioEncoders);
  202. for(i = 0; i < m_strArrayAudioEncoders.GetSize(); i++)
  203. m_comboxAudioEncoder.AddString(m_strArrayAudioEncoders.GetAt(i));
  205. //得到文件中的设置数据
  206. LoadFilterConfig();
  208. CString strCurSel;
  209. if (0 != m_strCurVideoEncoder.GetBuffer(0)[0])
  210. {
  211. m_comboxVideoEncoder.SelectString(-1, m_strCurVideoEncoder);
  212. if ("MicroSoft Media 9 DMO" == m_strCurVideoEncoder)
  213. {
  214. g_VideoEncodingParams.DefaultParams();
  215. }
  217. if(FAILED(m_grfCapture.CreateVideoEncoderFilterByName((TCHAR*)((LPCTSTR)m_strCurVideoEncoder))))
  218. AfxMessageBox("Cant Create filter!!");
  219. }
  220. else
  221. {
  222. m_comboxVideoEncoder.SetCurSel(0);
  223. if((i = m_comboxVideoEncoder.GetCurSel()) != CB_ERR)
  224. {
  225. m_comboxVideoEncoder.GetLBText(i, strCurSel);
  226. m_strCurVideoEncoder = strCurSel;
  227. if ("MicroSoft Media 9 DMO" == m_strCurVideoEncoder)
  228. {
  229. g_VideoEncodingParams.DefaultParams();
  230. }
  231. if(FAILED(m_grfCapture.CreateVideoEncoderFilterByName((TCHAR*)((LPCTSTR)strCurSel))))
  232. AfxMessageBox("Cant Create filter!!");
  233. }
  234. }
  236.     if (0 != m_strCurVideoCapDevice.GetBuffer(0)[0])
  237.     {
  238.         m_comboxVideoCap.SelectString(-1, m_strCurVideoCapDevice.GetBuffer(0));
  239.         if(FAILED(m_grfCapture.CreateVideoCapDeviceFilterByName((TCHAR*)((LPCTSTR)m_strCurVideoCapDevice))))
  240.             AfxMessageBox("Cant Create filter!!");
  241.     }
  242.     else
  243.     {
  244.         m_comboxVideoCap.SetCurSel(0);
  245.         strCurSel.Empty();
  246.         if((i = m_comboxVideoCap.GetCurSel()) != CB_ERR)
  247.         {
  248.             m_comboxVideoCap.GetLBText(i, strCurSel);
  249.             m_strCurVideoCapDevice = strCurSel;
  250.             if(FAILED(m_grfCapture.CreateVideoCapDeviceFilterByName((TCHAR*)((LPCTSTR)strCurSel))))
  251.                 AfxMessageBox("Cant Create video capture device filter!!");
  252.         }
  253.     }
  254.     if (0 != m_strCurAudioCapDevice.GetBuffer(0)[0])
  255.     {
  256.         m_comboxAudioCap.SelectString(-1, m_strCurAudioCapDevice.GetBuffer(0));
  257.         if(FAILED(m_grfCapture.CreateAudioCapDeviceFilterByName((TCHAR*)((LPCTSTR)m_strCurAudioCapDevice))))
  258.             AfxMessageBox("Cant Create audio capture device filter!!");
  259.     }
  260.     else
  261.     {
  262.         m_comboxAudioCap.SetCurSel(0);
  263.         strCurSel.Empty();
  264.         if((i = m_comboxAudioCap.GetCurSel()) != CB_ERR)
  265.         {
  266.             m_comboxAudioCap.GetLBText(i, strCurSel);
  267.             m_strCurAudioCapDevice = strCurSel;
  268.             if(FAILED(m_grfCapture.CreateAudioCapDeviceFilterByName((TCHAR*)((LPCTSTR)strCurSel))))
  269.                 AfxMessageBox("Cant Create audio capture device filter!!");
  270.         }
  271.     }
  272. if (0 != m_strCurAudioEncoder.GetBuffer(0)[0])
  273. {
  274. m_comboxAudioEncoder.SelectString(-1, m_strCurAudioEncoder.GetBuffer(0));
  275. if(FAILED(m_grfCapture.CreateAudioEncoderFilterByName((TCHAR*)((LPCTSTR)m_strCurAudioEncoder))))
  276. AfxMessageBox("Cant Create filter!!");
  277. }
  278. else
  279. {
  280. m_comboxAudioEncoder.SetCurSel(0);
  281. strCurSel.Empty();
  282. if((i = m_comboxAudioEncoder.GetCurSel()) != CB_ERR)
  283. {
  284. m_comboxAudioEncoder.GetLBText(i, strCurSel);
  285. m_strCurAudioEncoder = strCurSel;
  286. if(FAILED(m_grfCapture.CreateAudioEncoderFilterByName((TCHAR*)((LPCTSTR)strCurSel))))
  287. AfxMessageBox("Cant Create filter!!");
  288. }
  289. }
  291. m_grfCapture.SetCurMediaType(mpTemAudioAMT);
  293. mchkNeedOverlay.SetCheck(BST_UNCHECKED);
  294. OnChkOverlay();
  296. m_bChkCrossBar.SetCheck(BST_UNCHECKED);
  298. m_grfCapture.SetOwner(GetDlgItem(IDC_STATIC_PREVIEW)->GetSafeHwnd());
  299. GetDlgItem(IDC_STATIC_PREVIEW)->ModifyStyle(0, WS_CLIPCHILDREN | WS_CLIPSIBLINGS);
  300. m_chkPreview.SetCheck(BST_CHECKED);
  302. if(m_strArrayVideoCapDevice.GetSize() == 0)
  303. {
  304. m_chkOnlyAudio.SetCheck(BST_CHECKED);
  305. m_comboxVideoCap.EnableWindow(FALSE);
  306. m_comboxVideoEncoder.EnableWindow(FALSE);
  307. m_chkOnlyAudio.EnableWindow(FALSE);
  308. }
  309. m_chkOnlyAudio.SetCheck(BST_UNCHECKED);
  311. return TRUE;  // return TRUE  unless you set the focus to a control
  312. }
  314. void CGVCaptureDlg::OnSysCommand(UINT nID, LPARAM lParam)
  315. {
  316. if ((nID & 0xFFF0) == IDM_ABOUTBOX)
  317. {
  318. CAboutDlg dlgAbout;
  319. dlgAbout.DoModal();
  320. }
  321. else
  322. {
  323. CDialog::OnSysCommand(nID, lParam);
  324. }
  325. }
  327. // If you add a minimize button to your dialog, you will need the code below
  328. //  to draw the icon.  For MFC applications using the document/view model,
  329. //  this is automatically done for you by the framework.
  331. void CGVCaptureDlg::OnPaint() 
  332. {
  333. if (IsIconic())
  334. {
  335. CPaintDC dc(this); // device context for painting
  337. SendMessage(WM_ICONERASEBKGND, (WPARAM) dc.GetSafeHdc(), 0);
  339. // Center icon in client rectangle
  340. int cxIcon = GetSystemMetrics(SM_CXICON);
  341. int cyIcon = GetSystemMetrics(SM_CYICON);
  342. CRect rect;
  343. GetClientRect(&rect);
  344. int x = (rect.Width() - cxIcon + 1) / 2;
  345. int y = (rect.Height() - cyIcon + 1) / 2;
  347. // Draw the icon
  348. dc.DrawIcon(x, y, m_hIcon);
  349. }
  350. else
  351. {
  352. CDialog::OnPaint();
  353. }
  354. }
  356. // The system calls this to obtain the cursor to display while the user drags
  357. //  the minimized window.
  358. HCURSOR CGVCaptureDlg::OnQueryDragIcon()
  359. {
  360. return (HCURSOR) m_hIcon;
  361. }
  363. void CGVCaptureDlg::OnBtnVideocapConfig() 
  364. {
  365. if (0 == GetSafeHwnd())
  366. {
  367. MessageBox("父窗口句柄不能为空","错误");
  368. }
  369. if(FAILED(m_grfCapture.ConfigVideoCapDeviceByName(this->GetSafeHwnd())))
  370. AfxMessageBox("Cant set property for this filter!!");
  371. }
  373. void CGVCaptureDlg::OnBtnVideoCompressorConfig() 
  374. {
  375. if (0 == GetSafeHwnd())
  376. {
  377. MessageBox("父窗口句柄不能为空","错误");
  378. }
  379. if ("MicroSoft Media 9 DMO" == m_strCurVideoEncoder)
  380. {
  381. CSetV9Dlg dlg;
  382. if(IDOK == dlg.DoModal())
  383. {
  384. if(FAILED(m_grfCapture.ConfigVideoEncoderByName(this->GetSafeHwnd())))
  385. AfxMessageBox("Cant set property for this filter!!");
  386. }
  387. return;
  388. }
  389. if(FAILED(m_grfCapture.ConfigVideoEncoderByName(this->GetSafeHwnd())))
  390. AfxMessageBox("Cant set property for this filter!!");
  391. }
  393. void CGVCaptureDlg::OnBnClickedBtnAudioCapConfig()
  394. {
  395.     if (0 == GetSafeHwnd())
  396.     {
  397.         MessageBox("父窗口句柄不能为空","错误");
  398.     }
  399.     if(FAILED(m_grfCapture.ConfigAudioCapDeviceByName(this->GetSafeHwnd())))
  400.         AfxMessageBox("Cant set property for this filter!!");
  401. }
  403. void CGVCaptureDlg::OnBtnAudioCompressorConfig() 
  404. {
  405. if (0 == GetSafeHwnd())
  406. {
  407. MessageBox("父窗口句柄不能为空","错误");
  408. }
  409. if(FAILED(m_grfCapture.ConfigAudioEncoderByName(this->GetSafeHwnd())))
  410. AfxMessageBox("Cant set property for this filter!!");
  411. }
  414. //xiezhouwei 05-08-17 添加设置字幕按钮
  415. void CGVCaptureDlg::OnBtnSetoverlay() 
  416. {
  417. if (FAILED(m_grfCapture.ConfigVideoOverlay(GetSafeHwnd())))
  418. {
  419. AfxMessageBox("Cant set property for this filter!!");
  420. }
  421. }
  423. void CGVCaptureDlg::OnChkOverlay() 
  424. {
  425. if (BST_UNCHECKED == mchkNeedOverlay.GetCheck())
  426. {
  427. m_grfCapture.ReleaseUideoOverlay();
  428. GetDlgItem(IDC_BTN_SETOVERLAY)->EnableWindow(FALSE);
  429. }
  430. if (BST_CHECKED == mchkNeedOverlay.GetCheck())
  431. {
  432. if (FAILED(m_grfCapture.CreateUideoOverlay()))
  433. {
  434. m_grfCapture.ReleaseUideoOverlay();
  435. MessageBox("字幕组件加载失败","注意");
  436. mchkNeedOverlay.SetCheck(BST_UNCHECKED);
  437. return ;
  438. }
  439. GetDlgItem(IDC_BTN_SETOVERLAY)->EnableWindow(TRUE);
  440. }
  441. }
  443. //系统设置框
  444. void CGVCaptureDlg::OnBtnSyscof() 
  445. {
  446. if(FAILED(m_grfCapture.ConfigSysCfg(this->GetSafeHwnd())))
  447. AfxMessageBox("Cant set System for this filter!!");
  449. }
  451. void CGVCaptureDlg::OnBtnBeginCap() 
  452. {
  453. if(SUCCEEDED(m_grfCapture.Run()))
  454. {
  455. GetDlgItem(IDC_BTN_VIDEO_COMPRESSOR_CONFIG)->EnableWindow(FALSE);
  456. GetDlgItem(IDC_BTN_AUDIO_COMPRESSOR_CONFIG)->EnableWindow(FALSE);
  457. m_bChkCrossBar.EnableWindow(FALSE);
  458. m_chkOnlyAudio.EnableWindow(FALSE);
  459. mchkNeedOverlay.EnableWindow(FALSE);
  460. GetDlgItem(IDC_BTN_SETOVERLAY)->EnableWindow(FALSE);
  461. mbtnSaveFile.EnableWindow(FALSE);
  462. m_comboxVideoEncoder.EnableWindow(FALSE);
  463. m_comboxVideoCap.EnableWindow(FALSE);
  464. m_comboxAudioEncoder.EnableWindow(FALSE);
  465. m_dwTimer = SetTimer(100, 1000, NULL);
  466. }
  467. m_bPause = FALSE;
  468. }
  470. void CGVCaptureDlg::OnBtnPauseCap() 
  471. {
  472. m_grfCapture.Pause();
  473. m_bPause = TRUE;
  474. }
  476. void CGVCaptureDlg::OnBtnStopCap() 
  477. {
  478. KillTimer(m_dwTimer);
  479. m_bPause = FALSE;
  480. m_dwEscape = 0;
  482. TCHAR strTime[10];
  483. wsprintf(strTime, "00:00:00");
  484. SetDlgItemText(IDC_STATIC_TIMETOCAP, strTime);
  486. SetDlgItemText(IDC_STATIC_BITRATE, "0.0");
  487. if (SUCCEEDED(m_grfCapture.Stop()))
  488. {
  489. GetDlgItem(IDC_BTN_SYSCOF)->EnableWindow();
  490. GetDlgItem(IDC_BTN_VIDEOCAP_INI_CONFIG)->EnableWindow();
  491. GetDlgItem(IDC_BTN_VIDEOCAP_CONFIG)->EnableWindow();
  492. GetDlgItem(IDC_BTN_VIDEO_COMPRESSOR_CONFIG)->EnableWindow();
  493. GetDlgItem(IDC_BTN_AUDIO_COMPRESSOR_CONFIG)->EnableWindow();
  495. m_bChkCrossBar.EnableWindow();
  496. m_chkOnlyAudio.EnableWindow();
  497. mchkNeedOverlay.EnableWindow();
  498. if (BST_CHECKED == mchkNeedOverlay.GetCheck())
  499. {
  500. GetDlgItem(IDC_BTN_SETOVERLAY)->EnableWindow();
  501. }
  502. mbtnSaveFile.EnableWindow();
  503. if(FALSE == m_grfCapture.IsOnlyAudio())
  504. {
  505. m_comboxVideoEncoder.EnableWindow();
  506. m_comboxVideoCap.EnableWindow();
  507. }
  508. m_comboxAudioEncoder.EnableWindow();
  509. }
  510. }
  512. void CGVCaptureDlg::OnChkPreview() 
  513. {
  514. m_grfCapture.SetPreview(!m_grfCapture.IsPreview());
  515. }
  517. void CGVCaptureDlg::OnTimer(UINT nIDEvent) 
  518. {
  519. if(m_bPause)
  520. return;
  521. m_dwEscape ++;
  522. // change time display
  523. int h = m_dwEscape/3600;
  524. int m = (m_dwEscape%3600)/60;
  525. int s = (m_dwEscape%3600)%60;
  527. TCHAR strH[3], strM[3], strS[3];
  528. if(h < 10)
  529. wsprintf(strH, "0%d", h);
  530. else
  531. wsprintf(strH, "%d", h);
  533. if(m < 10)
  534. wsprintf(strM, "0%d", m);
  535. else
  536. wsprintf(strM, "%d", m);
  538. if(s < 10)
  539. wsprintf(strS, "0%d", s);
  540. else
  541. wsprintf(strS, "%d", s);
  543. TCHAR strTime[10];
  544. wsprintf(strTime, "%s:%s:%s", strH, strM ,strS);
  545. SetDlgItemText(IDC_STATIC_TIMETOCAP, strTime);
  547. TCHAR strDataRata[20];
  548. sprintf(strDataRata, "%f", m_grfCapture.GetCompressRata());
  549. SetDlgItemText(IDC_STATIC_BITRATE, strDataRata);
  551. CDialog::OnTimer(nIDEvent);
  552. }
  554. void CGVCaptureDlg::OnClose() 
  555. {
  556. OnBtnStopCap();
  557. CDialog::OnClose();
  558. SaveFilterConfig();
  559. }
  561. void CGVCaptureDlg::OnCbnSelchangeCmbAudioCapture()
  562. {
  563.     CString strCurSel;
  564.     m_comboxAudioCap.GetLBText(m_comboxAudioCap.GetCurSel(), strCurSel);
  566.     if(m_strCurAudioCapDevice == strCurSel)
  567.         return;
  568.     m_strCurAudioCapDevice = strCurSel;
  570.     if(FAILED(m_grfCapture.CreateAudioCapDeviceFilterByName((TCHAR*)((LPCTSTR)strCurSel))))
  571.         AfxMessageBox("Cant Create filter!!");
  572. }
  574. void CGVCaptureDlg::OnSelchangeCmbAudioCompressor() 
  575. {
  576. CString strCurSel;
  577. m_comboxAudioEncoder.GetLBText(m_comboxAudioEncoder.GetCurSel(), strCurSel);
  579. if(m_strCurAudioEncoder == strCurSel ||((LPCTSTR)strCurSel)[0] == 0)
  580. {
  581. AfxMessageBox("当前读出字符串为空");
  582. return;
  583. }
  584. m_strCurAudioEncoder = strCurSel;
  586. if(FAILED(m_grfCapture.CreateAudioEncoderFilterByName((TCHAR*)((LPCTSTR)strCurSel))))
  587. AfxMessageBox("Cant Create filter!!");
  588. }
  590. void CGVCaptureDlg::OnSelchangeCmbVideoCapture() 
  591. {
  592. CString strCurSel;
  593. m_comboxVideoCap.GetLBText(m_comboxVideoCap.GetCurSel(), strCurSel);
  595. if(m_strCurVideoCapDevice == strCurSel)
  596. return;
  597. m_strCurVideoCapDevice = strCurSel;
  599. if(FAILED(m_grfCapture.CreateVideoCapDeviceFilterByName((TCHAR*)((LPCTSTR)strCurSel))))
  600. AfxMessageBox("Cant Create filter!!");
  601. }
  603. void CGVCaptureDlg::OnSelchangeCmbVideoCompressor() 
  604. {
  605. CString strCurSel;
  606. m_comboxVideoEncoder.GetLBText(m_comboxVideoEncoder.GetCurSel(), strCurSel);
  608. if(m_strCurVideoEncoder == strCurSel || ((LPCTSTR)strCurSel)[0] == 0)
  609. {
  610. AfxMessageBox("当前读出字符串为空");
  611. return;
  612. }
  614. m_strCurVideoEncoder = strCurSel;
  615. if ("MicroSoft Media 9 DMO" == m_strCurVideoEncoder)
  616. {
  617. g_VideoEncodingParams.DefaultParams();
  618. }
  619. if(FAILED(m_grfCapture.CreateVideoEncoderFilterByName((TCHAR*)((LPCTSTR)strCurSel))))
  620. AfxMessageBox("Cant Create filter!!");
  621. }
  623. void CGVCaptureDlg::OnNotifyStop()
  624. {
  625. OnBtnStopCap();
  626. }
  628. void CGVCaptureDlg::OnNotifyPlay()
  629. {
  630. OnBtnBeginCap();
  631. }
  633. void CGVCaptureDlg::OnCheckOnlyAudio() 
  634. {
  635. m_grfCapture.SetOnlyAudio(!m_grfCapture.IsOnlyAudio());
  637. if(m_grfCapture.IsOnlyAudio())
  638. {
  639. m_comboxVideoCap.EnableWindow(FALSE);
  640. m_comboxVideoEncoder.EnableWindow(FALSE);
  641. }
  642. else
  643. {
  644. m_comboxVideoCap.EnableWindow(TRUE);
  645. m_comboxVideoEncoder.EnableWindow(TRUE);
  646. }
  647. }
  651. bool CGVCaptureDlg::SaveFilterConfig()
  652. {
  653. char lszTotalPath[DEF_TEMP_LENGTH];
  654. int result = 0;
  655. FILE* lpFp = NULL;
  656. GetTotalPath("ConfigFilter.dat", lszTotalPath);
  657.     lpFp = fopen(lszTotalPath, "wb");
  659.     //写m_strCurAudioCapDevice
  660.     int liLen = strlen(m_strCurAudioCapDevice.GetBuffer(0));
  661.     result = fwrite(&liLen, sizeof(int), 1,lpFp);
  662.     if (1 != result)
  663.     {
  664.         fclose(lpFp);
  665.         return false;
  666.     }
  667.     result = fwrite(m_strCurAudioCapDevice.GetBuffer(0), sizeof(char), liLen,lpFp);
  668.     if (liLen != result)
  669.     {
  670.         fclose(lpFp);
  671.         return false;
  672.     }
  674. //写m_strCurAudioEncoder
  675. liLen = strlen(m_strCurAudioEncoder.GetBuffer(0));
  677. result = fwrite(&liLen, sizeof(int), 1,lpFp);
  678. if (1 != result)
  679. {
  680. fclose(lpFp);
  681. return false;
  682. }
  683. result = fwrite(m_strCurAudioEncoder.GetBuffer(0), sizeof(char), liLen,lpFp);
  684. if (liLen != result)
  685. {
  686. fclose(lpFp);
  687. return false;
  688. }
  690. //写m_strCurVideoCapDevice
  691. liLen = strlen(m_strCurVideoCapDevice.GetBuffer(0));
  692. result = fwrite(&liLen, sizeof(int), 1,lpFp);
  693. if (1 != result)
  694. {
  695. fclose(lpFp);
  696. return false;
  697. }
  698. result = fwrite(m_strCurVideoCapDevice.GetBuffer(0), sizeof(char), liLen,lpFp);
  699. if (liLen != result)
  700. {
  701. fclose(lpFp);
  702. return false;
  703. }
  705. //写m_strCurVideoEncoder
  706. liLen = strlen(m_strCurVideoEncoder.GetBuffer(0));
  707. result = fwrite(&liLen, sizeof(int), 1,lpFp);
  708. if (1 != result)
  709. {
  710. fclose(lpFp);
  711. return false;
  712. }
  713. result = fwrite(m_strCurVideoEncoder.GetBuffer(0), sizeof(char), liLen,lpFp);
  714. if (liLen != result)
  715. {
  716. fclose(lpFp);
  717. return false;
  718. }
  720. //写媒体类型
  721. AM_MEDIA_TYPE* lpCurAudioAMT = NULL;
  722. int ii = sizeof(AM_MEDIA_TYPE) + sizeof(WAVEFORMATEX);
  723. int iii = sizeof(AM_MEDIA_TYPE);
  724. if (!m_grfCapture.GetCurMediaType(&lpCurAudioAMT))
  725. {//如果没有，还必须将一个表示媒体类型为零的写入
  726. liLen = 0;
  727. fwrite(&liLen, sizeof(int), 1, lpFp);
  728. }
  730. //如果媒体格式不是音频则退出，这是一个保护机制。
  731. if (NULL != lpCurAudioAMT)
  732. {
  733. if (FORMAT_WaveFormatEx != lpCurAudioAMT->formattype)
  734. {
  735. fclose(lpFp);
  736. return false;
  737. }
  738. liLen = sizeof(AM_MEDIA_TYPE) + lpCurAudioAMT->cbFormat;
  739. result = fwrite(&liLen, sizeof(int), 1,lpFp);
  740. if (1 != result)
  741. {
  742. fclose(lpFp);
  743. return false;
  744. }
  745. result = fwrite(lpCurAudioAMT, liLen, 1,lpFp);
  746. if (1 != result)
  747. {
  748. fclose(lpFp);
  749. return false;
  750. }
  752. HeapFree(GetProcessHeap(), 0, lpCurAudioAMT);
  753. lpCurAudioAMT = NULL;
  755. }
  756. fclose(lpFp);
  757. return true;
  758. }
  760. bool CGVCaptureDlg::LoadFilterConfig()
  761. {
  762. char lszTotalPath[DEF_TEMP_LENGTH];
  763. int result = 0;
  764. FILE* lpFp = NULL;
  765. GetTotalPath("ConfigFilter.dat", lszTotalPath);
  766. lpFp = fopen(lszTotalPath, "rb");
  767. if (!lpFp)
  768. {
  769. return false;
  770.     }
  771.     //读m_strCurAudioCapDevice
  772.     int liLen = 0;
  773.     result = fread(&liLen, sizeof(int), 1,lpFp);
  774.     if (1 != result)
  775.     {
  776.         AfxMessageBox("读取配置文件出错");
  777.         fclose(lpFp);
  778.         DeleteFile(lszTotalPath);
  779.         return false;
  780.     }
  781.     if (liLen)
  782.     {
  783.         result = fread(m_strCurAudioCapDevice.GetBuffer(liLen), sizeof(char), liLen,lpFp);
  784.         if (liLen != result)
  785.         {
  786.             m_strCurAudioCapDevice.ReleaseBuffer(liLen);
  787.             AfxMessageBox("读取配置文件出错");
  788.             fclose(lpFp);
  789.             DeleteFile(lszTotalPath);
  790.             return false;
  791.         }
  792.         m_strCurAudioCapDevice.ReleaseBuffer(liLen);
  793.     }
  794. //读m_strCurAudioEncoder
  795. liLen = 0;
  796. result = fread(&liLen, sizeof(int), 1,lpFp);
  797. if (1 != result)
  798. {
  799. AfxMessageBox("读取配置文件出错");
  800. fclose(lpFp);
  801. DeleteFile(lszTotalPath);
  802. return false;
  803. }
  804. if (liLen)
  805. {
  806. result = fread(m_strCurAudioEncoder.GetBuffer(liLen), sizeof(char), liLen,lpFp);
  807. if (liLen != result)
  808. {
  809. m_strCurAudioEncoder.ReleaseBuffer(liLen);
  810. AfxMessageBox("读取配置文件出错");
  811. fclose(lpFp);
  812. DeleteFile(lszTotalPath);
  813. return false;
  814. }
  815. m_strCurAudioEncoder.ReleaseBuffer(liLen);
  816. }
  817. //读m_strCurVideoCapDevice
  818. liLen = 0;
  819. result = fread(&liLen, sizeof(int), 1,lpFp);
  820. if (1 != result)
  821. {
  822. AfxMessageBox("读取配置文件出错");
  823. fclose(lpFp);
  824. DeleteFile(lszTotalPath);
  825. return false;
  826. }
  827. if (liLen)
  828. {
  829. result = fread(m_strCurVideoCapDevice.GetBuffer(liLen), sizeof(char), liLen,lpFp);
  830. if (liLen != result)
  831. {
  832. m_strCurVideoCapDevice.ReleaseBuffer(liLen);
  833. AfxMessageBox("读取配置文件出错");
  834. fclose(lpFp);
  835. DeleteFile(lszTotalPath);
  836. return false;
  837. }
  838. m_strCurVideoCapDevice.ReleaseBuffer(liLen);
  839. }
  840. //读m_strCurVideoEncoder
  841. liLen = 0;
  842. result = fread(&liLen, sizeof(int), 1,lpFp);
  843. if (1 != result)
  844. {
  845. AfxMessageBox("读取配置文件出错");
  846. fclose(lpFp);
  847. DeleteFile(lszTotalPath);
  848. return false;
  849. }
  850. if (liLen)
  851. {
  852. result = fread(m_strCurVideoEncoder.GetBuffer(liLen), sizeof(char), liLen,lpFp);
  853. if (liLen != result)
  854. {
  855. m_strCurVideoEncoder.ReleaseBuffer(liLen);
  856. AfxMessageBox("读取配置文件出错");
  857. fclose(lpFp);
  858. DeleteFile(lszTotalPath);
  859. return false;
  860. }
  861. m_strCurVideoEncoder.ReleaseBuffer(liLen);
  862. }
  864. //读媒体类型
  865. liLen = 0;
  866. result = fread(&liLen, sizeof(int), 1,lpFp);
  867. if (1 != result)
  868. {
  869. AfxMessageBox("读取配置文件出错");
  870. fclose(lpFp);
  871. DeleteFile(lszTotalPath);
  872. return false;
  873. }
  874. if (liLen)//必须加，如果不加，则读出零个字节时就会返回false，此时应返回true，因为
  875. //只是代表没有。
  876. {
  877. //Arthur:解决m_pCurAudioAMT尽管有值为零，但是却还是分配了空间，只要有值时才给
  878. //mpTemAudioAMT分配空间就可。
  879. mpTemAudioAMT = (AM_MEDIA_TYPE *)HeapAlloc(GetProcessHeap(), 
  880. HEAP_NO_SERIALIZE|HEAP_ZERO_MEMORY, 
  881. sizeof(AM_MEDIA_TYPE) + sizeof(WAVEFORMATEX));
  883. result = fread(mpTemAudioAMT, liLen, 1,lpFp);
  884. if (1 != result)
  885. {
  886. HeapFree(GetProcessHeap(), 0, mpTemAudioAMT);
  887. mpTemAudioAMT = NULL;
  888. AfxMessageBox("读取配置文件出错");
  889. fclose(lpFp);
  890. DeleteFile(lszTotalPath);
  891. return false;
  892. }
  893. mpTemAudioAMT->pbFormat = (LPBYTE)(mpTemAudioAMT + 1);
  895. //如果媒体格式不是音频则退出，这是一个保护机制。
  896. if (FORMAT_WaveFormatEx != mpTemAudioAMT->formattype)
  897. {
  898. AfxMessageBox("读取配置文件出错");
  899. fclose(lpFp);
  900. DeleteFile(lszTotalPath);
  901. return false;
  902. }
  903. }
  904. fclose(lpFp);
  905. return true;
  906. }
  908. //求得完整路径
  909. bool CGVCaptureDlg::GetTotalPath(char *aszFileName, char *aszTotalPath)
  910. {
  911. GetModuleFileName(NULL, aszTotalPath, MAX_PATH);
  913. int liLengthPath = strlen(aszTotalPath);
  915. while (aszTotalPath[liLengthPath - 1] != '\')
  916. {
  917. --liLengthPath;
  918. }
  920. if (liLengthPath > 0)
  921. {
  922. aszTotalPath[liLengthPath] = '';
  923. if (NULL != aszFileName)
  924. {
  925. strcat(aszTotalPath, aszFileName);
  926. }
  927. return true;
  928. }
  929. return false;
  930. }
  931. //xiezhouwei 2005-08-15 添加配置pin函数
  934. void CGVCaptureDlg::OnBtnVideocapIniConfig() 
  935. {
  936. // TODO: Add your control notification handler code here
  937. m_grfCapture.ShowPinSetting(m_hWnd);
  938. return;
  939. }
  941. void CGVCaptureDlg::OnChkSave() 
  942. {
  943. char lszPath[MAX_PATH] = {0};
  944. UpdateData();
  945. if (BST_CHECKED == mbtnSaveFile.GetCheck())
  946. {
  947. GetTotalPath(NULL, lszPath);
  949. CFileDialog loFileDlg(FALSE, "avi" , NULL, OFN_EXPLORER|OFN_HIDEREADONLY | OFN_OVERWRITEPROMPT, 
  950. "AVI Files (*.avi)|*.avi|All Files (*.*)|*.*||", NULL);
  952. loFileDlg.m_ofn.lStructSize = sizeof(loFileDlg.m_ofn);
  954. if (IDOK  == loFileDlg.DoModal())
  955. {
  956. ZeroMemory(mszAVIFilePath, sizeof(mszAVIFilePath));
  958. strcpy(mszAVIFilePath, loFileDlg.GetPathName().GetBuffer(0));
  959. mstrSavePath = mszAVIFilePath;
  960. m_grfCapture.SetAVIFile(mszAVIFilePath);
  961. if ('' == mszAVIFilePath[0])
  962. {
  963. mbtnSaveFile.SetCheck(BST_UNCHECKED);
  964. }
  965. }
  966. }
  967. else
  968. {
  969. mstrSavePath.Empty();
  970. }
  971. UpdateData(false);
  972. }
  976. BOOL CGVCaptureDlg::OnEraseBkgnd(CDC* pDC)
  977. {
  978. // Intercept background erasing for the movie window, since the
  979. // video renderer will keep the screen painted.  Without this code,
  980. // your video window might get painted over with gray (the default
  981. // background brush) when it is obscured by another window and redrawn.
  982. CRect rc;
  983. // Get the bounding rectangle for the movie screen
  984. //m_VideoWindow.GetWindowRect(&rc);
  985. GetDlgItem(IDC_STATIC_PREVIEW)->GetWindowRect(&rc);
  986. ScreenToClient(&rc);
  987. // Exclude the clipping region occupied by our movie screen
  988. pDC->ExcludeClipRect(&rc);
  989. GetDlgItem(IDC_STATIC_PREVIEW)->GetWindowRect(&rc);
  990. ScreenToClient(&rc);
  991. pDC->ExcludeClipRect(&rc);
  993. // Erase the remainder of the dialog as usual
  994. return CDialog::OnEraseBkgnd(pDC);
  996. }
  998. //用户选择是否需要CrossBar输入
  999. void CGVCaptureDlg::OnBnClickedChkCrossbar()
  1000. {
  1001. if (BST_CHECKED == m_bChkCrossBar.GetCheck())
  1002. {
  1003. m_grfCapture.SetCrossBar(TRUE);
  1004. return;
  1005. }
  1006. m_grfCapture.SetCrossBar(FALSE);
  1007. }