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

源码开发语言/平台
当前位置：首页 > 源码/资料 > Windows编程 > 多媒体编程 > 查看源码
asf2mkvDlg.cpp
资源名称：fullMPCcode.rar　[点击查看]
上传用户：xjjlds
上传日期：2015-12-05
资源大小：22823k
文件大小：15k
源码类别：

多媒体编程

开发平台：

Visual C++

asf2mkvDlg.cpp：源码内容
  1. /* 
  2.  * Copyright (C) 2003-2005 Gabest
  3.  * http://www.gabest.org
  4.  *
  5.  *  This Program is free software; you can redistribute it and/or modify
  6.  *  it under the terms of the GNU General Public License as published by
  7.  *  the Free Software Foundation; either version 2, or (at your option)
  8.  *  any later version.
  9.  *   
  10.  *  This Program is distributed in the hope that it will be useful,
  11.  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  12.  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  13.  *  GNU General Public License for more details.
  14.  *   
  15.  *  You should have received a copy of the GNU General Public License
  16.  *  along with GNU Make; see the file COPYING.  If not, write to
  17.  *  the Free Software Foundation, 675 Mass Ave, Cambridge, MA 02139, USA. 
  18.  *  http://www.gnu.org/copyleft/gpl.html
  19.  *
  20.  */
  22. // asf2mkvDlg.cpp : implementation file
  23. //
  25. #include "stdafx.h"
  26. #include "asf2mkv.h"
  27. #include "asf2mkvDlg.h"
  29. #ifdef _DEBUG
  30. #define new DEBUG_NEW
  31. #endif
  33. #include <initguid.h>
  34. #include ".asf2mkvdlg.h"
  36. //  {6B6D0800-9ADA-11d0-A520-00A0D10129C0}
  37. DEFINE_GUID(CLSID_NetShowSource, 
  38. 0x6b6d0800, 0x9ada, 0x11d0, 0xa5, 0x20, 0x0, 0xa0, 0xd1, 0x1, 0x29, 0xc0);
  40. // CAboutDlg dialog used for App About
  42. class CAboutDlg : public CDialog
  43. {
  44. public:
  45. CAboutDlg();
  47. // Dialog Data
  48. enum { IDD = IDD_ABOUTBOX };
  50. protected:
  51. virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV support
  53. // Implementation
  54. protected:
  55. DECLARE_MESSAGE_MAP()
  56. };
  58. CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
  59. {
  60. }
  62. void CAboutDlg::DoDataExchange(CDataExchange* pDX)
  63. {
  64. CDialog::DoDataExchange(pDX);
  65. }
  67. BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
  68. END_MESSAGE_MAP()
  70. // CUrlDropTarget
  72. BEGIN_MESSAGE_MAP(CUrlDropTarget, COleDropTarget)
  73. END_MESSAGE_MAP()
  75. DROPEFFECT CUrlDropTarget::OnDragEnter(CWnd* pWnd, COleDataObject* pDataObject, DWORD dwKeyState, CPoint point)
  76. {
  77. return DROPEFFECT_NONE;
  78. }
  80. DROPEFFECT CUrlDropTarget::OnDragOver(CWnd* pWnd, COleDataObject* pDataObject, DWORD dwKeyState, CPoint point)
  81. {
  82. UINT CF_URL = RegisterClipboardFormat(_T("UniformResourceLocator"));
  83. return pDataObject->IsDataAvailable(CF_URL) ? DROPEFFECT_COPY : DROPEFFECT_NONE;
  84. }
  86. BOOL CUrlDropTarget::OnDrop(CWnd* pWnd, COleDataObject* pDataObject, DROPEFFECT dropEffect, CPoint point)
  87. {
  88. UINT CF_URL = RegisterClipboardFormat(_T("UniformResourceLocator"));
  90. BOOL bResult = FALSE;
  92. if(pDataObject->IsDataAvailable(CF_URL)) 
  93. {
  94. FORMATETC fmt = {CF_URL, NULL, DVASPECT_CONTENT, -1, TYMED_HGLOBAL};
  95. if(HGLOBAL hGlobal = pDataObject->GetGlobalData(CF_URL, &fmt))
  96. {
  97. LPCSTR pText = (LPCSTR)GlobalLock(hGlobal);
  98. if(AfxIsValidString(pText))
  99. {
  100. AfxGetMainWnd()->SendMessage(WM_OPENURL, 0, (LPARAM)pText);
  101. GlobalUnlock(hGlobal);
  102. bResult = TRUE;
  103. }
  104. }
  105. }
  107. return bResult;
  108. }
  110. DROPEFFECT CUrlDropTarget::OnDropEx(CWnd* pWnd, COleDataObject* pDataObject, DROPEFFECT dropDefault, DROPEFFECT dropList, CPoint point)
  111. {
  112. return (DROPEFFECT)-1;
  113. }
  115. void CUrlDropTarget::OnDragLeave(CWnd* pWnd)
  116. {
  117. }
  119. DROPEFFECT CUrlDropTarget::OnDragScroll(CWnd* pWnd, DWORD dwKeyState, CPoint point)
  120. {
  121. return DROPEFFECT_NONE;
  122. }
  124. // Casf2mkvDlg dialog
  126. #define WM_GRAPHNOTIFY (WM_APP+1)
  128. Casf2mkvDlg::Casf2mkvDlg(CWnd* pParent /*=NULL*/)
  129. : CResizableDialog(Casf2mkvDlg::IDD, pParent)
  130. , m_fRecording(false)
  131. , m_mru(0, _T("MRU"), _T("file%d"), 20, 10)
  132. {
  133. m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
  134. }
  136. void Casf2mkvDlg::DoDataExchange(CDataExchange* pDX)
  137. {
  138. __super::DoDataExchange(pDX);
  139. DDX_Control(pDX, IDC_COMBO1, m_combo);
  140. DDX_Control(pDX, IDC_STATIC1, m_video);
  141. }
  144. void Casf2mkvDlg::SetupCombo()
  145. {
  146. m_combo.ResetContent();
  147. for(int i = 0; i < m_mru.GetSize(); i++)
  148. if(!m_mru[i].IsEmpty())
  149. m_combo.AddString(m_mru[i]);
  150. }
  152. void Casf2mkvDlg::SetVideoRect()
  153. {
  154. if(pVW)
  155. {
  156. CRect r;
  157. m_video.GetWindowRect(r);
  158. r -= r.TopLeft();
  159. pVW->SetWindowPosition(r.left, r.top, r.Width(), r.Height());
  160. }
  161. }
  163. BEGIN_MESSAGE_MAP(Casf2mkvDlg, CResizableDialog)
  164. ON_WM_SYSCOMMAND()
  165. ON_WM_PAINT()
  166. ON_WM_QUERYDRAGICON()
  167. ON_MESSAGE(WM_GRAPHNOTIFY, OnGraphNotify)
  168. ON_BN_CLICKED(IDC_BUTTON1, OnRecord)
  169. ON_UPDATE_COMMAND_UI(IDC_BUTTON1, OnUpdateRecord)
  170. ON_UPDATE_COMMAND_UI(IDC_COMBO1, OnUpdateFileName)
  171. ON_UPDATE_COMMAND_UI(IDC_BUTTON2, OnUpdateSettings)
  172. ON_UPDATE_COMMAND_UI(IDC_CHECK2, OnUpdateSettings)
  173. ON_WM_SIZE()
  174. ON_WM_TIMER()
  175. ON_BN_CLICKED(IDC_BUTTON2, OnBnClickedButton2)
  176. ON_MESSAGE(WM_APP, OnUrlOpen)
  177. END_MESSAGE_MAP()
  180. // Casf2mkvDlg message handlers
  182. BOOL Casf2mkvDlg::OnInitDialog()
  183. {
  184. __super::OnInitDialog();
  186. // Add "About..." menu item to system menu.
  188. // IDM_ABOUTBOX must be in the system command range.
  189. ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
  190. ASSERT(IDM_ABOUTBOX < 0xF000);
  192. CMenu* pSysMenu = GetSystemMenu(FALSE);
  193. if (pSysMenu != NULL)
  194. {
  195. CString strAboutMenu;
  196. strAboutMenu.LoadString(IDS_ABOUTBOX);
  197. if (!strAboutMenu.IsEmpty())
  198. {
  199. pSysMenu->AppendMenu(MF_SEPARATOR);
  200. pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
  201. }
  202. }
  204. // Set the icon for this dialog.  The framework does this automatically
  205. //  when the application's main window is not a dialog
  206. SetIcon(m_hIcon, TRUE); // Set big icon
  207. SetIcon(m_hIcon, FALSE); // Set small icon
  209. // TODO: Add extra initialization here
  211. AddAnchor(IDC_STATIC1, TOP_LEFT, BOTTOM_RIGHT);
  212. AddAnchor(IDC_COMBO1, BOTTOM_LEFT, BOTTOM_RIGHT);
  213. AddAnchor(IDC_BUTTON1, BOTTOM_RIGHT);
  214. AddAnchor(IDC_BUTTON2, BOTTOM_RIGHT);
  216. m_mru.ReadList();
  217. SetupCombo();
  219. m_video.ModifyStyle(0, WS_CLIPCHILDREN);
  221. SetWindowText(ResStr(IDS_TITLE));
  223. m_urlDropTarget.Register(this);
  225. return TRUE;  // return TRUE  unless you set the focus to a control
  226. }
  228. BOOL Casf2mkvDlg::DestroyWindow()
  229. {
  230. m_urlDropTarget.Revoke();
  232. return __super::DestroyWindow();
  233. }
  235. void Casf2mkvDlg::OnSysCommand(UINT nID, LPARAM lParam)
  236. {
  237. if ((nID & 0xFFF0) == IDM_ABOUTBOX)
  238. {
  239. CAboutDlg dlgAbout;
  240. dlgAbout.DoModal();
  241. }
  242. else
  243. {
  244. __super::OnSysCommand(nID, lParam);
  245. }
  246. }
  248. // If you add a minimize button to your dialog, you will need the code below
  249. //  to draw the icon.  For MFC applications using the document/view model,
  250. //  this is automatically done for you by the framework.
  252. void Casf2mkvDlg::OnPaint() 
  253. {
  254. if (IsIconic())
  255. {
  256. CPaintDC dc(this); // device context for painting
  258. SendMessage(WM_ICONERASEBKGND, reinterpret_cast<WPARAM>(dc.GetSafeHdc()), 0);
  260. // Center icon in client rectangle
  261. int cxIcon = GetSystemMetrics(SM_CXICON);
  262. int cyIcon = GetSystemMetrics(SM_CYICON);
  263. CRect rect;
  264. GetClientRect(&rect);
  265. int x = (rect.Width() - cxIcon + 1) / 2;
  266. int y = (rect.Height() - cyIcon + 1) / 2;
  268. // Draw the icon
  269. dc.DrawIcon(x, y, m_hIcon);
  270. }
  271. else
  272. {
  273. __super::OnPaint();
  274. }
  275. }
  277. // The system calls this function to obtain the cursor to display while the user drags
  278. //  the minimized window.
  279. HCURSOR Casf2mkvDlg::OnQueryDragIcon()
  280. {
  281. return static_cast<HCURSOR>(m_hIcon);
  282. }
  285. LRESULT Casf2mkvDlg::OnGraphNotify(WPARAM wParam, LPARAM lParam)
  286. {
  287.     HRESULT hr = S_OK;
  289. LONG evCode, evParam1, evParam2;
  290.     while(pME && SUCCEEDED(pME->GetEvent(&evCode, (LONG_PTR*)&evParam1, (LONG_PTR*)&evParam2, 0)))
  291.     {
  292. hr = pME->FreeEventParams(evCode, evParam1, evParam2);
  294.         if(EC_COMPLETE == evCode)
  295.         {
  296. if(m_fRecording)
  297. {
  298. OnRecord();
  299. break;
  300. }
  301.         }
  302. }
  304.     return hr;
  305. }
  307. void Casf2mkvDlg::OnRecord()
  308. {
  309. UpdateData();
  311. HRESULT hr;
  313. if(!m_fRecording)
  314. {
  315. m_fRecording = true;
  317. UpdateDialogControls(this, FALSE);
  319. hr = E_FAIL;
  321. do
  322. {
  323. // i/o
  325. CString src;
  326. m_combo.GetWindowText(src);
  328. CFileDialog fd(
  329. FALSE, _T("mkv"), m_dst, 
  330. OFN_HIDEREADONLY|OFN_OVERWRITEPROMPT|OFN_EXPLORER|OFN_ENABLESIZING, 
  331. _T("Matroska file (*.mkv;*.mka)|*.mkv;*.mka||"), 
  332. this);
  333. if(fd.DoModal() != IDOK) break;
  334. m_dst = fd.GetPathName();
  336. m_mru.Add(src);
  337. m_mru.WriteList();
  338. SetupCombo();
  339. m_combo.SetWindowText(src);
  341. if(src.Trim().IsEmpty() || m_dst.Trim().IsEmpty()) 
  342. break;
  344. // filer graph
  346. if(FAILED(hr = pGB.CoCreateInstance(CLSID_FilterGraph))) 
  347. break;
  349. pMC = pGB; pME = pGB; pMS = pGB; pVW = pGB; pBV = pGB;
  350. if(!pMC || !pME || !pMS || !pVW || !pBV) 
  351. break;
  353. if(FAILED(hr = pME->SetNotifyWindow((OAHWND)m_hWnd, WM_GRAPHNOTIFY, 0))) 
  354. break;
  356. // windows media source filter
  358. CComPtr<IBaseFilter> pSrc;
  359. if(FAILED(hr = pSrc.CoCreateInstance(CLSID_NetShowSource))
  360. || FAILED(hr = pGB->AddFilter(pSrc, CStringW(src)))
  361. || FAILED(hr = CComQIPtr<IFileSourceFilter>(pSrc)->Load(CStringW(src), NULL)))
  362. break;
  364. // matroska muxer
  366. CComPtr<IBaseFilter> pMux;
  367. if(FAILED(hr = pMux.CoCreateInstance(__uuidof(CMatroskaMuxerFilter)))
  368. && !(pMux = new CMatroskaMuxerFilter(NULL, NULL))
  369. || FAILED(hr = pGB->AddFilter(pMux, L"Matroska Muxer")))
  370. break;
  372. BeginEnumPins(pSrc, pEP, pPin)
  373. if(FAILED(hr = pGB->Connect(pPin, GetFirstDisconnectedPin(pMux, PINDIR_INPUT)))) 
  374. break;
  375. EndEnumPins
  377. if(FAILED(hr))
  378. break;
  380. // file writer
  382. CComPtr<IBaseFilter> pFW;
  383. if(FAILED(hr = pFW.CoCreateInstance(CLSID_FileWriter))
  384. || FAILED(hr = pGB->AddFilter(pFW, CStringW(m_dst)))
  385. || FAILED(hr = CComQIPtr<IFileSinkFilter2>(pFW)->SetFileName(CStringW(m_dst), NULL))
  386. || FAILED(hr = CComQIPtr<IFileSinkFilter2>(pFW)->SetMode(AM_FILE_OVERWRITE))
  387. || FAILED(hr = pGB->Connect(GetFirstDisconnectedPin(pMux, PINDIR_OUTPUT), GetFirstDisconnectedPin(pFW, PINDIR_INPUT))))
  388. break;
  390. // insert inf. pin tee
  392. BeginEnumPins(pMux, pEP, pPin)
  393. {
  394. PIN_DIRECTION dir;
  395. CComPtr<IPin> pPinTo;
  396. CMediaType mt;
  397. if(FAILED(pPin->QueryDirection(&dir)) || dir != PINDIR_INPUT
  398. || FAILED(pPin->ConnectedTo(&pPinTo))
  399. || FAILED(pPin->ConnectionMediaType(&mt)))
  400. continue;
  402. // FIXME: the inf pin tee filter makes the video messed up, like when seeking 
  403. // onto a non-keyframe and starting to decode from that point. (audio seems to be ok)
  405. CComPtr<IBaseFilter> pInfPinTee;
  406. if(mt.majortype == MEDIATYPE_Video 
  407. || mt.majortype == MEDIATYPE_Audio)
  408. {
  409. if(FAILED(pGB->Disconnect(pPin))
  410. || FAILED(pGB->Disconnect(pPinTo))
  411. || FAILED(pInfPinTee.CoCreateInstance(CLSID_InfTee))
  412. || FAILED(pGB->AddFilter(pInfPinTee, L"Infinite Pin Tee")))
  413. continue;
  415. if(FAILED(pGB->Connect(pPinTo, GetFirstDisconnectedPin(pInfPinTee, PINDIR_INPUT)))
  416. || FAILED(pGB->Connect(GetFirstDisconnectedPin(pInfPinTee, PINDIR_OUTPUT), pPin)))
  417. {
  418. pGB->RemoveFilter(pInfPinTee);
  419. pGB->ConnectDirect(pPinTo, pPin, NULL);
  420. continue;
  421. }
  422. }
  424. pPin = GetFirstDisconnectedPin(pInfPinTee, PINDIR_OUTPUT);
  426. CComPtr<IBaseFilter> pRenderer;
  428. if(mt.majortype == MEDIATYPE_Video)
  429. {
  430. if(SUCCEEDED(pRenderer.CoCreateInstance(CLSID_VideoRendererDefault))
  431. || SUCCEEDED(pRenderer.CoCreateInstance(CLSID_VideoRenderer)))
  432. pPinTo = ::GetFirstDisconnectedPin(pRenderer, PINDIR_INPUT);
  433. }
  434. else if(mt.majortype == MEDIATYPE_Audio)
  435. {
  436. if(SUCCEEDED(pRenderer.CoCreateInstance(CLSID_DSoundRender)))
  437. pPinTo = ::GetFirstDisconnectedPin(pRenderer, PINDIR_INPUT);
  438. }
  440. if(pPin && pPinTo && pRenderer)
  441. {
  442. if(SUCCEEDED(pGB->AddFilter(pRenderer, L"Renderer"))
  443. && FAILED(pGB->Connect(pPin, pPinTo)))
  444. pGB->RemoveFilter(pRenderer);
  445. }
  446. }
  447. EndEnumPins
  449. // setup video window
  451. if(SUCCEEDED(pVW->put_Owner((OAHWND)m_video.m_hWnd))
  452. && SUCCEEDED(pVW->put_WindowStyle(WS_CHILD|WS_CLIPSIBLINGS|WS_CLIPCHILDREN))
  453. && SUCCEEDED(pVW->put_MessageDrain((OAHWND)m_hWnd)))
  454. {
  455. SetVideoRect();
  456. }
  458. // timer for polling the position
  460. SetTimer(1, 500, NULL);
  462. // go!
  464. if(FAILED(hr = pMC->Run()))
  465. break;
  467. hr = S_OK;
  468. }
  469. while(false);
  471. if(FAILED(hr)) OnRecord();
  472. }
  473. else
  474. {
  475. if(pMC) pMC->Stop();
  477. pMC.Release(); pME.Release(); pMS.Release();
  478. pVW.Release(); pBV.Release();
  479. pGB.Release();
  481. m_fRecording = false;
  483. SetWindowText(ResStr(IDS_TITLE));
  484. }
  485. }
  487. void Casf2mkvDlg::OnUpdateRecord(CCmdUI* pCmdUI)
  488. {
  489. CString url;
  490. m_combo.GetWindowText(url);
  491. url.Trim();
  492. pCmdUI->Enable(!url.IsEmpty());
  493. pCmdUI->SetText(ResStr(!m_fRecording ? IDS_RECORD : IDS_STOP));
  494. }
  496. void Casf2mkvDlg::OnUpdateFileName(CCmdUI* pCmdUI)
  497. {
  498. pCmdUI->Enable(!m_fRecording);
  499. }
  501. void Casf2mkvDlg::OnUpdateSettings(CCmdUI* pCmdUI)
  502. {
  503. pCmdUI->Enable(!m_fRecording);
  504. }
  506. void Casf2mkvDlg::OnSize(UINT nType, int cx, int cy)
  507. {
  508. CResizableDialog::OnSize(nType, cx, cy);
  510. SetVideoRect();
  511. }
  513. void Casf2mkvDlg::OnTimer(UINT nIDEvent)
  514. {
  515. if(nIDEvent == 1)
  516. {
  517. if(pMS)
  518. {
  519. REFERENCE_TIME rtPos = 0, rtDur = 0;
  520. pMS->GetCurrentPosition(&rtPos);
  521. pMS->GetDuration(&rtDur);
  523. CString title;
  524. if(rtDur > 0) title.Format(_T("%s (progress: %I64d%%)"), ResStr(IDS_TITLE), 100i64 * rtPos / rtDur);
  525. else title = ResStr(IDS_TITLE) + _T(" (live)");
  527. BeginEnumFilters(pGB, pEF, pBF)
  528. {
  529. if(CComQIPtr<IAMNetworkStatus, &IID_IAMNetworkStatus> pAMNS = pBF)
  530. {
  531. long BufferingProgress = 0;
  532. if(SUCCEEDED(pAMNS->get_BufferingProgress(&BufferingProgress)) && BufferingProgress > 0)
  533. {
  534. CString str;
  535. str.Format(_T(" (buffer: %d%%)"), BufferingProgress);
  536. title += str;
  537. }
  538. break;
  539. }
  540. }
  541. EndEnumFilters
  543. SetWindowText(title);
  544. }
  545. }
  547. CResizableDialog::OnTimer(nIDEvent);
  548. }
  550. void Casf2mkvDlg::OnBnClickedButton2()
  551. {
  552. CComPtr<IBaseFilter> pBF;
  553. pBF.CoCreateInstance(CLSID_NetShowSource);
  554. if(pBF) ShowPPage(pBF, m_hWnd);
  555. }
  557. LRESULT Casf2mkvDlg::OnUrlOpen(WPARAM wParam, LPARAM lParam)
  558. {
  559. m_combo.SetWindowText(CString((char*)lParam));
  560. return 0;
  561. }
  563. //////////////////
  565. Casf2mkvDlg::CRecentFileAndURLList::CRecentFileAndURLList(UINT nStart, LPCTSTR lpszSection,
  566. LPCTSTR lpszEntryFormat, int nSize,
  567. int nMaxDispLen) 
  568. : CRecentFileList(nStart, lpszSection, lpszEntryFormat, nSize, nMaxDispLen)
  569. {
  570. }
  572. //#include <afximpl.h>
  573. extern BOOL AFXAPI AfxFullPath(LPTSTR lpszPathOut, LPCTSTR lpszFileIn);
  574. extern BOOL AFXAPI AfxComparePath(LPCTSTR lpszPath1, LPCTSTR lpszPath2);
  576. void Casf2mkvDlg::CRecentFileAndURLList::Add(LPCTSTR lpszPathName)
  577. {
  578. ASSERT(m_arrNames != NULL);
  579. ASSERT(lpszPathName != NULL);
  580. ASSERT(AfxIsValidString(lpszPathName));
  582. if(CString(lpszPathName).MakeLower().Find(_T("@device:")) >= 0)
  583. return;
  585. bool fURL = (CString(lpszPathName).Find(_T("://")) >= 0);
  587. // fully qualify the path name
  588. TCHAR szTemp[_MAX_PATH];
  589. if(fURL) _tcscpy(szTemp, lpszPathName);
  590. else AfxFullPath(szTemp, lpszPathName);
  592. // update the MRU list, if an existing MRU string matches file name
  593. int iMRU;
  594. for (iMRU = 0; iMRU < m_nSize-1; iMRU++)
  595. {
  596. if((fURL && !_tcscmp(m_arrNames[iMRU], szTemp))
  597. || AfxComparePath(m_arrNames[iMRU], szTemp))
  598. break;      // iMRU will point to matching entry
  599. }
  600. // move MRU strings before this one down
  601. for (; iMRU > 0; iMRU--)
  602. {
  603. ASSERT(iMRU > 0);
  604. ASSERT(iMRU < m_nSize);
  605. m_arrNames[iMRU] = m_arrNames[iMRU-1];
  606. }
  607. // place this one at the beginning
  608. m_arrNames[0] = szTemp;
  609. }