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

源码开发语言/平台
当前位置：首页 > 电子书籍 > VC书籍 > 查看源码
backDlg.cpp
资源名称：VC++视频传输.rar　[点击查看]
上传用户：hxb_1234
上传日期：2010-03-30
资源大小：8328k
文件大小：10k
源码类别：

VC书籍

开发平台：

Visual C++

backDlg.cpp：源码内容
  1. // backDlg.cpp : implementation file
  2. //
  4. #include "stdafx.h"
  5. #include "back.h"
  6. #include "backDlg.h"
  8. #include "....includeav8api.h"
  9. #include "....includeDefine.h"
  11. #ifdef _DEBUG
  12. #define new DEBUG_NEW
  13. #undef THIS_FILE
  14. static char THIS_FILE[] = __FILE__;
  15. #endif
  17. static CHAR caFileName[STRSIZE] = "temp.mpg";
  19. typedef WORD (CALLBACK *USER_CALLBACK) (HDRVR hdrvr, UINT msg, DWORD dwUser, DWORD dwParam1, DWORD dwParam2);
  21. WORD CALLBACK _loadds MyCallBack(HDRVR hdrvr, UINT msg, DWORD dwUser, DWORD dwParam1, DWORD dwParam2);
  23. static HMMIO hmmioInput;
  24. static HMMIO hmmioOutput;
  26. /////////////////////////////////////////////////////////////////////////////
  27. // CAboutDlg dialog used for App About
  29. class CAboutDlg : public CDialog
  30. {
  32. public:
  33. CAboutDlg();
  35. // Dialog Data
  36. //{{AFX_DATA(CAboutDlg)
  37. enum { IDD = IDD_ABOUTBOX };
  38. //}}AFX_DATA
  40. // ClassWizard generated virtual function overrides
  41. //{{AFX_VIRTUAL(CAboutDlg)
  42. protected:
  43. virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV support
  44. //}}AFX_VIRTUAL
  46. // Implementation
  47. protected:
  48. //{{AFX_MSG(CAboutDlg)
  49. //}}AFX_MSG
  50. DECLARE_MESSAGE_MAP()
  51. };
  55. CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
  57. {
  58. //{{AFX_DATA_INIT(CAboutDlg)
  59. //}}AFX_DATA_INIT
  60. }
  64. void CAboutDlg::DoDataExchange(CDataExchange* pDX)
  66. {
  67. CDialog::DoDataExchange(pDX);
  68. //{{AFX_DATA_MAP(CAboutDlg)
  69. //}}AFX_DATA_MAP
  70. }
  72. BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
  73. //{{AFX_MSG_MAP(CAboutDlg)
  74. // No message handlers
  75. //}}AFX_MSG_MAP
  76. END_MESSAGE_MAP()
  78. /////////////////////////////////////////////////////////////////////////////
  79. // CBackDlg dialog
  81. CBackDlg::CBackDlg(CWnd* pParent /*=NULL*/)
  82. : CDialog(CBackDlg::IDD, pParent), m_hVFDrv(0), m_bStream(0)
  84. {
  85. //{{AFX_DATA_INIT(CBackDlg)
  86. // NOTE: the ClassWizard will add member initialization here
  87. //}}AFX_DATA_INIT
  88. // Note that LoadIcon does not require a subsequent DestroyIcon in Win32
  89. m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
  90. }
  94. CBackDlg::~CBackDlg()
  96. {
  97.     // make sure that on destruct, OnStop() is called to 
  98.     // stop/close/unlock.  This prevents a system crash (blue screen)
  99.     OnStop();
  100. }
  104. void CBackDlg::DoDataExchange(CDataExchange* pDX)
  106. {
  107. CDialog::DoDataExchange(pDX);
  108. //{{AFX_DATA_MAP(CBackDlg)
  109. // NOTE: the ClassWizard will add DDX and DDV calls here
  110. //}}AFX_DATA_MAP
  111. }
  113. BEGIN_MESSAGE_MAP(CBackDlg, CDialog)
  114. //{{AFX_MSG_MAP(CBackDlg)
  115. ON_WM_SYSCOMMAND()
  116. ON_WM_PAINT()
  117. ON_WM_QUERYDRAGICON()
  118. ON_BN_CLICKED(IDC_RECORD, OnRecord)
  119. ON_BN_CLICKED(IDC_STOP, OnStop)
  120. //}}AFX_MSG_MAP
  121. END_MESSAGE_MAP()
  123. /////////////////////////////////////////////////////////////////////////////
  124. // CBackDlg message handlers
  128. BOOL CBackDlg::OnInitDialog()
  130. {
  131. CDialog::OnInitDialog();
  133. // Add "About..." menu item to system menu.
  135. // IDM_ABOUTBOX must be in the system command range.
  136. ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
  137. ASSERT(IDM_ABOUTBOX < 0xF000);
  139. CMenu* pSysMenu = GetSystemMenu(FALSE);
  140. CString strAboutMenu;
  141. strAboutMenu.LoadString(IDS_ABOUTBOX);
  142. if (!strAboutMenu.IsEmpty())
  143. {
  144. pSysMenu->AppendMenu(MF_SEPARATOR);
  145. pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
  146. }
  148. // Set the icon for this dialog.  The framework does this automatically
  149. //  when the application's main window is not a dialog
  150. SetIcon(m_hIcon, TRUE); // Set big icon
  151. SetIcon(m_hIcon, FALSE); // Set small icon
  153. // TODO: Add extra initialization here
  154. cbSystem=(CButton *)GetDlgItem(IDC_SYSTEM);
  155. cbVideo=(CButton *)GetDlgItem(IDC_VIDEO);
  156. cbAudio=(CButton *)GetDlgItem(IDC_AUDIO);
  157. cbSystem->SetCheck(TRUE);
  159.     return TRUE;  // return TRUE  unless you set the focus to a control
  161. }
  165. void CBackDlg::OnSysCommand(UINT nID, LPARAM lParam)
  167. {
  168. if ((nID & 0xFFF0) == IDM_ABOUTBOX)
  169. {
  170. CAboutDlg dlgAbout;
  171. dlgAbout.DoModal();
  172. }
  173. else
  174. {
  175. CDialog::OnSysCommand(nID, lParam);
  176. }
  177. }
  179. // If you add a minimize button to your dialog, you will need the code below
  180. //  to draw the icon.  For MFC applications using the document/view model,
  181. //  this is automatically done for you by the framework.
  183. void CBackDlg::OnPaint() 
  184. {
  185. if (IsIconic())
  186. {
  187. CPaintDC dc(this); // device context for painting
  189. SendMessage(WM_ICONERASEBKGND, (WPARAM) dc.GetSafeHdc(), 0);
  191. // Center icon in client rectangle
  192. int cxIcon = GetSystemMetrics(SM_CXICON);
  193. int cyIcon = GetSystemMetrics(SM_CYICON);
  194. CRect rect;
  195. GetClientRect(&rect);
  196. int x = (rect.Width() - cxIcon + 1) / 2;
  197. int y = (rect.Height() - cyIcon + 1) / 2;
  199. // Draw the icon
  200. dc.DrawIcon(x, y, m_hIcon);
  201. }
  202. else
  203. {
  204. CDialog::OnPaint();
  205. }
  206. }
  208. // The system calls this to obtain the cursor to display while the user drags
  209. //  the minimized window.
  210. HCURSOR CBackDlg::OnQueryDragIcon()
  211. {
  212. return (HCURSOR) m_hIcon;
  213. }
  216. /////////////////////////////////////////////////////////////////////////////
  217. // CAspenApp commands
  219. WORD CALLBACK _loadds MyCallBack(HDRVR hdrvr, UINT msg, DWORD dwUser, DWORD dwParam1, DWORD dwParam2)
  220. {
  221. DWORD dwFlags = 0;
  223. TRACE0("MyCallBack");
  224. switch (msg)
  225. {
  226. case VF_MSGUSER_BUF_WRITE:
  227.     {
  228.         LONG lRet;
  229.         LPVF_BUFWRITE_STRUCT lpBufWrite = (LPVF_BUFWRITE_STRUCT)dwParam1;
  230.         lRet = mmioWrite(hmmioOutput, (LPSTR)lpBufWrite->lpBuffer, (LONG)lpBufWrite->dwBufferWrite);
  231.         if (lRet == -1L)
  232.             {
  233.         return (FALSE);
  234.             }
  235.         break;
  236.     }
  238. case VF_MSGUSER_BUF_CREATE:
  239.     {
  240.         LPVF_CALLBACK_STRUCT lpCallback = (LPVF_CALLBACK_STRUCT)dwParam1;
  242.         if (lpCallback->wSubMsg == VF_SUBMSG_OUTPUT)
  243.     {
  244.                 dwFlags = MMIO_CREATE | MMIO_WRITE;
  245.                 hmmioOutput = mmioOpen((LPTSTR)caFileName, (LPMMIOINFO)NULL, dwFlags);
  246.     }
  247.         else
  248.             {
  249.                 dwFlags = MMIO_READ | MMIO_DENYWRITE;
  250.                 hmmioInput = mmioOpen((LPTSTR)caFileName, (LPMMIOINFO)NULL, dwFlags);
  251.             }
  252.         break;
  253.     }
  255. case VF_MSGUSER_BUF_CLOSE:
  256.     {
  257.         LPVF_CALLBACK_STRUCT lpCallback = (LPVF_CALLBACK_STRUCT)dwParam1;
  259.         if (lpCallback->wSubMsg == VF_SUBMSG_OUTPUT)
  260.     {
  261.         if (hmmioOutput)
  262.     {
  263.         mmioClose(hmmioOutput, NULL);
  264.         hmmioOutput = NULL;
  265.     }
  266.     }
  267.         else
  268.     {
  269.         if (hmmioInput)
  270.         {
  271.             mmioClose(hmmioInput, NULL);
  272.             hmmioInput = NULL;
  273.         }
  274.     }
  275.         break;
  276.     }
  277.     } // switch
  279. return 1;
  280. }
  284. VOID CBackDlg::OnRecord() 
  286. {
  287.     // TODO: Add your control notification handler code here
  288.     WORD flags;
  290.     m_hVFDrv = OpenDriver (L"av8api.dll", NULL, NULL);
  291.     if (!m_hVFDrv)
  292.     {
  293.         AfxMessageBox(_TEXT("Can't OpenDriver()."));
  294.         return ;
  295.     }
  297.     if (!HVFAssign(m_hVFDrv, 0))
  298.     {
  299.         AfxMessageBox(_TEXT("Can't Assign Dev!"));
  300.         return ;
  301.     }
  303.     if (!HVFLock(m_hVFDrv, VF_CAP_ALL))
  304.     {
  305.         AfxMessageBox(_TEXT("Can't Lock Dev!"));
  306.         return ;
  307.     }
  309.     flags = VF_FLAG_MPEG | VF_FLAG_ENCODE | VF_FLAG_OUTBUF;
  310.     m_bStream = static_cast<BYTE>
  311.         (HVFOpen(m_hVFDrv, 
  312.                  flags, 
  313.                  reinterpret_cast<unsigned long>(MyCallBack)));
  315.     if(cbSystem->GetCheck())
  316.     {
  317.         InitEncodeVideoVxD();
  318.         InitEncodeAudioVxD();
  319.     } 
  321.     else 
  322.     {
  323.         if(cbVideo->GetCheck())
  324.         {
  325.             InitEncodeVideoVxD();
  326.         }
  328.         else 
  329.         {
  330.             InitEncodeAudioVxD();
  331.         }
  332.     }
  334.     InitEncodeSystemVxD();
  335.     if (m_bStream) 
  336.     {
  337.         HVFRecord(m_hVFDrv, m_bStream, NULL, NULL);
  338.     }
  339. } // CBackDlg::OnRecord()
  343. VOID CBackDlg::OnStop() 
  345. {
  346.     // TODO: Add your control notification handler code here
  347.     m_bStream = 0;
  348.     if (m_hVFDrv)
  349.     {
  350.         HVFStop(m_hVFDrv, m_bStream);
  351.         HVFClose(m_hVFDrv, m_bStream);
  352.         HVFUnlock(m_hVFDrv, VF_CAP_ALL);
  353.         CloseDriver(m_hVFDrv, NULL, NULL);
  354.         m_hVFDrv = 0;
  355.     }
  356. } // CBackDlg::OnStop()
  360. BOOL CBackDlg::InitEncodeVideoVxD(void)
  362. {
  363.     DWORD dwValue;
  364.     WORD  wWidth, wHeight;
  366.     wWidth=352;
  367.     wHeight=240;
  368.     dwValue=MAKELONG(wWidth, wHeight);
  370.     HVFSet(m_hVFDrv, m_bStream, VF_INDEX_VID_OUTPUTSIZE, dwValue);
  372.     /* SDL - also set video width and height to scale images */
  373.     wWidth = (wWidth * 45) / 44; // try to maintain standard MPEG SIF aspect ratio
  374.     dwValue = MAKELONG(wWidth, wHeight /* [sic] */);
  375.     HVFSet(m_hVFDrv, m_bStream, VF_INDEX_VID_INPUTSIZE, dwValue);
  377.     HVFSet(m_hVFDrv, m_bStream, VF_INDEX_VID_BITRATE, BITRATE_DEFAULT);
  379.     HVFSet(m_hVFDrv, m_bStream, VF_INDEX_VID_RATE, VF_FLAG_VID_25);
  381.     HVFSet(m_hVFDrv, m_bStream, VF_INDEX_VID_IINTERVAL, IINTERVAL_DEFAULT);
  383.     HVFSet(m_hVFDrv, m_bStream, VF_INDEX_VID_BINTERVAL, PINTERVAL_DEFAULT);
  385.     HVFSet(m_hVFDrv, m_bStream, VF_INDEX_VID_MODE, VF_FLAG_VID_PAL);
  387.     HVFSet(m_hVFDrv, m_bStream, VF_INDEX_VID_SOURCE, VF_FLAG_VID_COMPOSITE);
  389.     HVFSet(m_hVFDrv, m_bStream, VF_INDEX_VID_ALGORITHM, VF_FLAG_VID_MPEG);
  391.     HVFSet(m_hVFDrv, m_bStream, VF_INDEX_VID_BRIGHTNESS, BRIGHTNESS_DEFAULT);
  393.     HVFSet(m_hVFDrv, m_bStream, VF_INDEX_VID_CONTRAST, CONTRAST_DEFAULT);    
  395.     HVFSet(m_hVFDrv, m_bStream, VF_INDEX_VID_HUE, HUE_DEFAULT);
  397.     HVFSet(m_hVFDrv, m_bStream, VF_INDEX_VID_SATURATION, SATURATION_DEFAULT);
  399.     return TRUE;
  400. }
  404. BOOL CBackDlg::InitEncodeAudioVxD(void)
  406. {
  407.   DWORD dwValue;
  409. HVFSet(m_hVFDrv, m_bStream, VF_INDEX_AUD_SAMPLE, SAMPLE_RATE_DEFAULT);
  410. HVFSet(m_hVFDrv, m_bStream, VF_INDEX_AUD_BITRATE, BIT_RATE_DEFAULT);
  412. dwValue=VF_FLAG_AUD_MPEG; // set to MPEG algorithm
  413. dwValue=(dwValue<<16)+VF_FLAG_AUD_NONE;
  414. HVFSet(m_hVFDrv, m_bStream, VF_INDEX_AUD_ALGORITHM, dwValue);
  415. HVFSet(m_hVFDrv, m_bStream, VF_INDEX_AUD_MODE, MODE_DEFAULT);
  416. HVFSet(m_hVFDrv, m_bStream, VF_INDEX_AUD_VOLUME, VOLUME_MIN);
  417. HVFSet(m_hVFDrv, m_bStream, VF_INDEX_AUD_VOLUME, VOLUME_DEFAULT);
  418. HVFSet(m_hVFDrv, m_bStream, VF_INDEX_AUD_GAIN, GAIN_DEFAULT);
  420. return TRUE;
  421. }
  425. BOOL CBackDlg::InitEncodeSystemVxD(void)
  427. {
  428. if(cbSystem->GetCheck()){
  429. HVFSet(m_hVFDrv, m_bStream, VF_INDEX_STM_TYPE, VF_FLAG_MPGSYS);
  430. } else {
  431. if(cbVideo->GetCheck()){
  432. HVFSet(m_hVFDrv, m_bStream, VF_INDEX_STM_TYPE, VF_FLAG_MPGVID);
  433. } else {
  434. HVFSet(m_hVFDrv, m_bStream, VF_INDEX_STM_TYPE, VF_FLAG_MPGAUD);
  435. }
  436. }
  438. HVFSet(m_hVFDrv, m_bStream, VF_INDEX_STM_ENCODEMODE, VF_FLAG_STM_NORMAL);
  439. return TRUE;
  440. }