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

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

多媒体编程

开发平台：

Visual C++

ConvertDlg.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. // ConvertDlg.cpp : implementation file
  23. //
  25. #include "stdafx.h"
  26. #include "mplayerc.h"
  27. #include "....filtersfilters.h"
  28. #include "......includematroskamatroska.h"
  29. #include "GraphBuilder.h"
  30. #include "ConvertPropsDlg.h"
  31. #include "ConvertResDlg.h"
  32. #include "ConvertChapDlg.h"
  33. #include "ConvertDlg.h"
  35. // TODO: subtitle source filter
  37. // CConvertDlg dialog
  39. CConvertDlg::CConvertDlg(CWnd* pParent /*=NULL*/)
  40. : CResizableDialog(CConvertDlg::IDD, pParent)
  41. , m_dwRegister(0)
  42. , m_fn(_T(""))
  43. {
  44. }
  46. CConvertDlg::~CConvertDlg()
  47. {
  48. }
  50. void CConvertDlg::AddFile(CString fn)
  51. {
  52. CPath path(fn);
  53. path.StripPath();
  55. CGraphBuilder gb(m_pGB, NULL);
  57. CComPtr<IBaseFilter> pBF;
  58. if(FAILED(gb.AddSourceFilter(fn, &pBF)))
  59. return;
  61. int cnt = 0;
  62. while(S_OK == m_pCGB->RenderStream(NULL, NULL, pBF, NULL, m_pMux)) cnt++;
  63. if(!cnt) {MessageBeep(-1); DeleteFilter(pBF); return;}
  65. if(m_tree.GetCount() == 0)
  66. {
  67. if(CComQIPtr<IDSMPropertyBag> pPB = m_pMux)
  68. pPB->DelAllProperties();
  70. if(CString(_T(".dsm")).CompareNoCase(path.GetExtension()) != 0)
  71. {
  72. CPath p(fn);
  73. p.RemoveExtension();
  74. SetOutputFile(CString((LPCTSTR)p) + _T(".dsm"));
  75. }
  76. }
  78. CTreeItemFile* t = new CTreeItemFile(fn, pBF, m_tree, NULL);
  80. AddFilter(*t, pBF);
  82. m_tree.Expand(*t, TVE_EXPAND);
  83. m_tree.EnsureVisible(*t);
  84. }
  86. bool CConvertDlg::SetOutputFile(LPCTSTR fn)
  87. {
  88. if(!m_pGB || !m_pMux)
  89. return false;
  91. NukeDownstream(m_pMux, m_pGB);
  93. CComPtr<IBaseFilter> pFW;
  94. CComQIPtr<IFileSinkFilter2> pFSF = m_pMux;
  95. if(pFSF) {pFW = m_pMux;}
  96. else {pFW.CoCreateInstance(CLSID_FileWriter); pFSF = pFW;}
  98. if(!pFSF
  99. || FAILED(m_pGB->AddFilter(pFW, NULL))
  100. || FAILED(pFSF->SetFileName(CStringW(fn), NULL))
  101. || FAILED(pFSF->SetMode(AM_FILE_OVERWRITE))
  102. || FAILED(m_pCGB->RenderStream(NULL, NULL, m_pMux, NULL, pFW)))
  103. {
  104. m_pGB->RemoveFilter(pFW);
  105. return false;
  106. }
  108. m_fn = fn;
  109. UpdateData(FALSE);
  111. return true;
  112. }
  114. void CConvertDlg::AddFilter(HTREEITEM hTIParent, IBaseFilter* pBFParent)
  115. {
  116. BeginEnumPins(pBFParent, pEP, pPin)
  117. {
  118. PIN_DIRECTION dir;
  119. CComPtr<IPin> pPinTo;
  120. CComPtr<IBaseFilter> pBF;
  121. if(FAILED(pPin->QueryDirection(&dir)) || dir != PINDIR_OUTPUT
  122. || FAILED(pPin->ConnectedTo(&pPinTo)) || !pPinTo
  123. || !(pBF = GetFilterFromPin(pPinTo)))
  124. continue;
  126. CTreeItem* t = NULL;
  128. if(pBF == m_pMux)
  129. {
  130. t = new CTreeItemPin(pPin, m_tree, hTIParent);
  131. }
  132. else
  133. {
  134. t = new CTreeItemFilter(pBF, m_tree, hTIParent);
  135. AddFilter(*t, pBF);
  136. }
  137. }
  138. EndEnumPins
  140. if(CComQIPtr<IDSMPropertyBag> pPB = pBFParent)
  141. {
  142. ULONG props;
  143. if(FAILED(pPB->CountProperties(&props)))
  144. props = 0;
  146. for(ULONG i = 0; i < props; i++)
  147. {
  148. PROPBAG2 PropBag;
  149. memset(&PropBag, 0, sizeof(PropBag));
  150. ULONG cPropertiesReturned = 0;
  151. if(FAILED(pPB->GetPropertyInfo(i, 1, &PropBag, &cPropertiesReturned)))
  152. continue;
  154. HRESULT hr;
  155. CComVariant var;
  156. if(SUCCEEDED(pPB->Read(1, &PropBag, NULL, &var, &hr)) && SUCCEEDED(hr))
  157. {
  158. CComQIPtr<IDSMPropertyBag> pPBMux = m_pMux;
  159. CComBSTR value;
  160. if(pPBMux && FAILED(pPBMux->GetProperty(PropBag.pstrName, &value)))
  161. pPBMux->SetProperty(PropBag.pstrName, var.bstrVal);
  162. }
  164. CoTaskMemFree(PropBag.pstrName);
  165. }
  166. }
  168. CTreeItem* t2 = new CTreeItemResourceFolder(m_tree, hTIParent);
  169. if(CComQIPtr<IDSMResourceBag> pRB = pBFParent)
  170. {
  171. for(DWORD i = 0, cnt = pRB->ResGetCount(); i < cnt; i++)
  172. {
  173. CComBSTR name, mime, desc;
  174. BYTE* pData = NULL;
  175. DWORD len = 0;
  176. if(FAILED(pRB->ResGet(i, &name, &desc, &mime, &pData, &len, NULL)))
  177. continue;
  179. if(len > 0)
  180. {
  181. m_pTIs.AddTail(new CTreeItemResource(CDSMResource(name, desc, mime, pData, len), m_tree, *t2));
  182. }
  184. CoTaskMemFree(pData);
  185. }
  186. }
  187. m_tree.Expand(*t2, TVE_EXPAND);
  189. CTreeItem* t3 = new CTreeItemChapterFolder(m_tree, hTIParent);
  190. if(CComQIPtr<IDSMChapterBag> pCB = pBFParent)
  191. {
  192. for(DWORD i = 0, cnt = pCB->ChapGetCount(); i < cnt; i++)
  193. {
  194. REFERENCE_TIME rt;
  195. CComBSTR name;
  196. if(FAILED(pCB->ChapGet(i, &rt, &name)))
  197. continue;
  199. m_pTIs.AddTail(new CTreeItemChapter(CDSMChapter(rt, name), m_tree, *t3));
  200. }
  201. }
  202. m_tree.Expand(*t3, TVE_EXPAND);
  204. m_tree.Expand(hTIParent, TVE_EXPAND);
  205. }
  207. void CConvertDlg::DeleteFilter(IBaseFilter* pBF)
  208. {
  209. BeginEnumPins(pBF, pEP, pPin)
  210. {
  211. PIN_DIRECTION dir;
  212. CComPtr<IPin> pPinTo;
  213. CComPtr<IBaseFilter> pBF;
  214. if(FAILED(pPin->QueryDirection(&dir)) || dir != PINDIR_OUTPUT
  215. || FAILED(pPin->ConnectedTo(&pPinTo)) || !pPinTo
  216. || !(pBF = GetFilterFromPin(pPinTo)))
  217. continue;
  219. if(pBF != m_pMux) DeleteFilter(pBF);
  220. }
  221. EndEnumPins
  223. m_pGB->RemoveFilter(pBF);
  224. }
  226. void CConvertDlg::DeleteItem(HTREEITEM hTI)
  227. {
  228. if(!hTI) return;
  230. DeleteChildren(hTI);
  232. CTreeItem* t = (CTreeItem*)m_tree.GetItemData(hTI);
  233. if(POSITION pos = m_pTIs.Find(t)) m_pTIs.RemoveAt(pos);
  234. delete t;
  235. m_tree.DeleteItem(hTI);
  236. }
  238. void CConvertDlg::DeleteChildren(HTREEITEM hTI)
  239. {
  240. if(!hTI) return;
  242. if(m_tree.ItemHasChildren(hTI))
  243. {
  244. HTREEITEM hChildItem = m_tree.GetChildItem(hTI);
  246. while(hChildItem != NULL)
  247. {
  248. HTREEITEM hNextItem = m_tree.GetNextItem(hChildItem, TVGN_NEXT);
  249. DeleteItem(hChildItem);
  250. hChildItem = hNextItem;
  251. }
  252. }
  253. }
  255. HTREEITEM CConvertDlg::HitTest(CPoint& sp, CPoint& cp)
  256. {
  257. sp = CPoint((LPARAM)GetMessagePos());
  258. cp = sp;
  259. m_tree.ScreenToClient(&cp);
  260. UINT flags = 0;
  261. HTREEITEM hTI = m_tree.HitTest(cp, &flags);
  262. return hTI && (flags&TVHT_ONITEM) ? hTI : NULL;
  263. }
  265. void CConvertDlg::ShowPopup(CPoint p)
  266. {
  267. CMenu m;
  268. m.CreatePopupMenu();
  270. int i = 1;
  271. m.AppendMenu(MF_STRING, i++, ResStr(IDS_CONVERT_ADDFILE));
  272. m.AppendMenu(MF_SEPARATOR);
  273. m.AppendMenu(MF_STRING, i++, ResStr(IDS_CONVERT_PROPERTIES));
  275. switch((int)m.TrackPopupMenu(TPM_LEFTBUTTON|TPM_RETURNCMD, p.x, p.y, this))
  276. {
  277. case 1:
  278. {
  279. CFileDialog fd(TRUE, NULL, m_fn, 
  280. OFN_EXPLORER|OFN_ENABLESIZING|OFN_HIDEREADONLY, 
  281. _T("Media files|*.*||"), this, 0);
  282. if(fd.DoModal() == IDOK) AddFile(fd.GetPathName());
  283. }
  284. break;
  285. case 2:
  286. EditProperties(CComQIPtr<IDSMPropertyBag>(m_pMux));
  287. break;
  288. }
  289. }
  291. void CConvertDlg::ShowFilePopup(HTREEITEM hTI, CPoint p)
  292. {
  293. CTreeItemFile* t = dynamic_cast<CTreeItemFile*>((CTreeItem*)m_tree.GetItemData(hTI));
  294. ASSERT(t);
  296. CMenu m;
  297. m.CreatePopupMenu();
  299. int i = 1;
  300. m.AppendMenu(MF_STRING, i++, ResStr(IDS_CONVERT_REMOVE));
  302. switch((int)m.TrackPopupMenu(TPM_LEFTBUTTON|TPM_RETURNCMD, p.x, p.y, this))
  303. {
  304. case 1:
  305. DeleteFilter(t->m_pBF);
  306. DeleteItem(hTI);
  307. break;
  308. }
  309. }
  311. void CConvertDlg::ShowPinPopup(HTREEITEM hTI, CPoint p)
  312. {
  313. CTreeItemPin* t = dynamic_cast<CTreeItemPin*>((CTreeItem*)m_tree.GetItemData(hTI));
  314. ASSERT(t);
  316. if(!t->m_pPin) return;
  318. CComPtr<IPin> pPinTo;
  319. t->m_pPin->ConnectedTo(&pPinTo);
  321. CMediaType mtconn;
  322. if(pPinTo) t->m_pPin->ConnectionMediaType(&mtconn);
  324. CArray<CMediaType> mts;
  325. BeginEnumMediaTypes(t->m_pPin, pEMT, pmt)
  326. mts.Add(*pmt);
  327. EndEnumMediaTypes(pmt)
  329. CMenu m;
  330. m.CreatePopupMenu();
  332. int i = 1, mtbase = 1000, mti = mtbase;
  334. m.AppendMenu(MF_STRING, i++, !pPinTo ? ResStr(IDS_CONVERT_ENABLESTREAM) : ResStr(IDS_CONVERT_DISABLESTREAM));
  336. if(mts.GetCount() > 1)
  337. {
  338. m.AppendMenu(MF_SEPARATOR);
  339. for(int i = 0; i < mts.GetCount(); i++)
  340. m.AppendMenu(MF_STRING | (mts[i] == mtconn ? MF_CHECKED : 0), mti++, CMediaTypeEx(mts[i]).ToString());
  341. }
  343. m.AppendMenu(MF_SEPARATOR);
  344. m.AppendMenu(MF_STRING | (!pPinTo ? MF_GRAYED : 0), i++, ResStr(IDS_CONVERT_PINPROPERTIES));
  346. switch(i = (int)m.TrackPopupMenu(TPM_LEFTBUTTON|TPM_RETURNCMD, p.x, p.y, this))
  347. {
  348. case 1:
  349. if(pPinTo) {m_pGB->Disconnect(pPinTo); m_pGB->Disconnect(t->m_pPin);}
  350. else if(pPinTo = GetFirstDisconnectedPin(m_pMux, PINDIR_INPUT)) m_pGB->ConnectDirect(t->m_pPin, pPinTo, NULL);
  351. t->Update();
  352. break;
  353. case 2:
  354. EditProperties(CComQIPtr<IDSMPropertyBag>(pPinTo));
  355. break;
  356. default:
  357. i -= mtbase;
  358. if(i >= 0 && i < mts.GetCount())
  359. {
  360. if(pPinTo) {m_pGB->Disconnect(pPinTo); m_pGB->Disconnect(t->m_pPin);}
  361. else {pPinTo = GetFirstDisconnectedPin(m_pMux, PINDIR_INPUT);}
  362. HRESULT hr = m_pGB->ConnectDirect(t->m_pPin, pPinTo, &mts[i]);
  363. if(FAILED(hr))
  364. {
  365. AfxMessageBox(_T("Reconnection attempt failed!"));
  366. if(mtconn.majortype != GUID_NULL) 
  367. hr = m_pGB->ConnectDirect(t->m_pPin, pPinTo, &mtconn);
  368. }
  369. t->Update();
  370. }
  371. break;
  372. }
  373. }
  375. void CConvertDlg::ShowResourceFolderPopup(HTREEITEM hTI, CPoint p)
  376. {
  377. CTreeItemResourceFolder* t = dynamic_cast<CTreeItemResourceFolder*>((CTreeItem*)m_tree.GetItemData(hTI));
  378. ASSERT(t);
  380. CMenu m;
  381. m.CreatePopupMenu();
  383. int i = 1;
  384. m.AppendMenu(MF_STRING, i++, ResStr(IDS_CONVERT_ADDRESOURCE));
  385. if(m_tree.ItemHasChildren(*t))
  386. {
  387. m.AppendMenu(MF_SEPARATOR);
  388. m.AppendMenu(MF_STRING, i++, ResStr(IDS_CONVERT_ADDRESOURCE));
  389. }
  391. switch((int)m.TrackPopupMenu(TPM_LEFTBUTTON|TPM_RETURNCMD, p.x, p.y, this))
  392. {
  393. case 1:
  394. {
  395. CFileDialog fd(TRUE, NULL, NULL, 
  396. OFN_EXPLORER|OFN_ENABLESIZING|OFN_HIDEREADONLY, 
  397. _T("All files|*.*||"), this, 0);
  398. if(fd.DoModal() == IDOK) 
  399. {
  400. CString fn = fd.GetPathName();
  401. if(FILE* f = _tfopen(fn, _T("rb")))
  402. {
  403. CDSMResource res;
  405. CPath path(fn);
  406. path.StripPath();
  407. res.name = (LPCTSTR)path;
  409. CRegKey key;
  410. TCHAR mime[256];
  411. ULONG len = countof(mime);
  412. if(ERROR_SUCCESS == key.Open(HKEY_CLASSES_ROOT, path.GetExtension().MakeLower(), KEY_READ)
  413. && ERROR_SUCCESS == key.QueryStringValue(_T("Content Type"), mime, &len))
  414. res.mime = mime;
  416. CTreeItemResource* t = new CTreeItemResource(res, m_tree, hTI);
  417. m_pTIs.AddTail(t);
  419. if(EditResource(t))
  420. {
  421. fseek(f, 0, 2);
  422. long size = ftell(f);
  423. fseek(f, 0, 0);
  424. t->m_res.data.SetSize(size);
  425. for(BYTE* ptr = t->m_res.data.GetData(),* end = ptr + size; 
  426. size > 0 && end - ptr >= size && fread(ptr, min(size, 1024), 1, f) > 0; 
  427. ptr += 1024, size -= 1024);
  428. fclose(f);
  429. }
  430. else
  431. {
  432. DeleteItem(*t);
  433. }
  434. }
  435. else
  436. {
  437. AfxMessageBox(_T("Cannot open file!"));
  438. }
  439. }
  440. }
  441. break;
  442. case 2:
  443. DeleteChildren(hTI);
  444. break;
  445. }
  446. }
  448. void CConvertDlg::ShowResourcePopup(HTREEITEM hTI, CPoint p)
  449. {
  450. CTreeItemResource* t = dynamic_cast<CTreeItemResource*>((CTreeItem*)m_tree.GetItemData(hTI));
  451. ASSERT(t);
  453. CMenu m;
  454. m.CreatePopupMenu();
  456. int i = 1;
  457. m.AppendMenu(MF_STRING, i++, ResStr(IDS_CONVERT_REMOVE));
  458. m.AppendMenu(MF_STRING, i++, ResStr(IDS_CONVERT_SAVEAS));
  459. if(AfxGetAppSettings().fEnableWebServer) m.AppendMenu(MF_STRING, 1000, ResStr(IDS_CONVERT_LAUNCHINBROWSER));
  460. m.AppendMenu(MF_SEPARATOR);
  461. m.AppendMenu(MF_STRING, i++, ResStr(IDS_CONVERT_RESOURCEPROPERTIES));
  463. switch((int)m.TrackPopupMenu(TPM_LEFTBUTTON|TPM_RETURNCMD, p.x, p.y, this))
  464. {
  465. case 1:
  466. DeleteItem(*t);
  467. break;
  468. case 2:
  469. {
  470. CFileDialog fd(FALSE, NULL, CString(t->m_res.name), 
  471. OFN_EXPLORER|OFN_ENABLESIZING|OFN_HIDEREADONLY|OFN_OVERWRITEPROMPT,
  472. _T("All files|*.*||"), this, 0);
  473. if(fd.DoModal() == IDOK)
  474. {
  475. if(FILE* f = _tfopen(fd.GetPathName(), _T("wb")))
  476. {
  477. fwrite(t->m_res.data.GetData(), 1, t->m_res.data.GetSize(), f);
  478. fclose(f);
  479. }
  480. }
  481. }
  482. break;
  483. case 3:
  484. EditResource(t);
  485. break;
  486. case 1000:
  487. {
  488. CString url;
  489. url.Format(_T("http://localhost:%d/convres.html?id=%x"), AfxGetAppSettings().nWebServerPort, (DWORD)&t->m_res);
  490. ShellExecute(NULL, _T("open"), url, NULL, NULL, SW_SHOWDEFAULT);
  491. }
  492. break;
  493. }
  494. }
  496. bool CConvertDlg::EditProperties(IDSMPropertyBag* pPB)
  497. {
  498. CConvertPropsDlg dlg(!!CComQIPtr<IPin>(pPB), this);
  500. ULONG props;
  501. if(FAILED(pPB->CountProperties(&props)))
  502. props = 0;
  504. for(ULONG i = 0; i < props; i++)
  505. {
  506. PROPBAG2 PropBag;
  507. memset(&PropBag, 0, sizeof(PropBag));
  508. ULONG cPropertiesReturned = 0;
  509. if(FAILED(pPB->GetPropertyInfo(i, 1, &PropBag, &cPropertiesReturned)))
  510. continue;
  512. HRESULT hr;
  513. CComVariant var;
  514. if(SUCCEEDED(pPB->Read(1, &PropBag, NULL, &var, &hr)) && SUCCEEDED(hr))
  515. dlg.m_props[CString(PropBag.pstrName)] = CString(var);
  517. CoTaskMemFree(PropBag.pstrName);
  518. }
  520. if(IDOK != dlg.DoModal())
  521. return false;
  523. pPB->DelAllProperties();
  525. POSITION pos = dlg.m_props.GetStartPosition();
  526. while(pos)
  527. {
  528. CString key, value;
  529. dlg.m_props.GetNextAssoc(pos, key, value);
  530. pPB->SetProperty(CStringW(key), CStringW(value));
  531. }
  533. return true;
  534. }
  536. bool CConvertDlg::EditResource(CTreeItemResource* t)
  537. {
  538. CConvertResDlg dlg(this);
  540. dlg.m_name = t->m_res.name;
  541. dlg.m_mime = t->m_res.mime;
  542. dlg.m_desc = t->m_res.desc;
  544. if(IDOK != dlg.DoModal())
  545. return false;
  547. t->m_res.name = dlg.m_name;
  548. t->m_res.mime = dlg.m_mime;
  549. t->m_res.desc = dlg.m_desc;
  551. t->Update();
  553. return true;
  554. }
  556. void CConvertDlg::ShowChapterFolderPopup(HTREEITEM hTI, CPoint p)
  557. {
  558. CTreeItemChapterFolder* t = dynamic_cast<CTreeItemChapterFolder*>((CTreeItem*)m_tree.GetItemData(hTI));
  559. ASSERT(t);
  561. CMenu m;
  562. m.CreatePopupMenu();
  564. int i = 1;
  565. m.AppendMenu(MF_STRING, i++, ResStr(IDS_CONVERT_ADDCHAPTER));
  566. if(m_tree.ItemHasChildren(*t))
  567. {
  568. m.AppendMenu(MF_SEPARATOR);
  569. m.AppendMenu(MF_STRING, i++, ResStr(IDS_CONVERT_REMOVEALL));
  570. }
  572. switch((int)m.TrackPopupMenu(TPM_LEFTBUTTON|TPM_RETURNCMD, p.x, p.y, this))
  573. {
  574. case 1:
  575. {
  576. CDSMChapter chap;
  577. CTreeItemChapter* t = new CTreeItemChapter(CDSMChapter(0, L""), m_tree, hTI);
  578. m_pTIs.AddTail(t);
  579. if(!EditChapter(t)) 
  580. DeleteItem(*t);
  581. }
  582. break;
  583. case 2:
  584. DeleteChildren(hTI);
  585. break;
  586. }
  587. }
  589. void CConvertDlg::ShowChapterPopup(HTREEITEM hTI, CPoint p)
  590. {
  591. CTreeItemChapter* t = dynamic_cast<CTreeItemChapter*>((CTreeItem*)m_tree.GetItemData(hTI));
  592. ASSERT(t);
  594. CMenu m;
  595. m.CreatePopupMenu();
  597. int i = 1;
  598. m.AppendMenu(MF_STRING, i++, ResStr(IDS_CONVERT_REMOVE));
  599. m.AppendMenu(MF_SEPARATOR);
  600. m.AppendMenu(MF_STRING, i++, ResStr(IDS_CONVERT_CHAPTERPROPERTIES));
  602. switch((int)m.TrackPopupMenu(TPM_LEFTBUTTON|TPM_RETURNCMD, p.x, p.y, this))
  603. {
  604. case 1:
  605. DeleteItem(hTI);
  606. break;
  607. case 2:
  608. EditChapter(t);
  609. break;
  610. }
  611. }
  613. bool CConvertDlg::EditChapter(CTreeItemChapter* t)
  614. {
  615. CConvertChapDlg dlg(this);
  617. int h = (int)(t->m_chap.rt/10000000/60/60);
  618. int m = (int)(t->m_chap.rt/10000000/60%60);
  619. int s = (int)(t->m_chap.rt/10000000%60);
  620. int ms = (int)(t->m_chap.rt/10000%1000);
  622. dlg.m_name = t->m_chap.name;
  623. dlg.m_time.Format(_T("%02d:%02d:%02d.%03d"), h, m, s, ms);
  625. if(IDOK != dlg.DoModal())
  626. return false;
  628. TCHAR c;
  629. if(_stscanf(dlg.m_time, _T("%d%c%d%c%d%c%d"), &h, &c, &m, &c, &s, &c, &ms) != 7)
  630. return false;
  632. t->m_chap.name = dlg.m_name;
  633. t->m_chap.rt = ((((__int64)h*60+m)*60+s)*1000+ms)*10000;
  635. t->Update();
  637. return true;
  638. }
  640. void CConvertDlg::DoDataExchange(CDataExchange* pDX)
  641. {
  642. __super::DoDataExchange(pDX);
  643. DDX_Control(pDX, IDC_TREE1, m_tree);
  644. DDX_Text(pDX, IDC_EDIT1, m_fn);
  645. }
  647. BOOL CConvertDlg::PreTranslateMessage(MSG* pMsg)
  648. {
  649. if(pMsg->message == WM_KEYDOWN && pMsg->wParam == VK_ESCAPE)
  650. return TRUE;
  652. return __super::PreTranslateMessage(pMsg);
  653. }
  655. BOOL CConvertDlg::OnInitDialog()
  656. {
  657. __super::OnInitDialog();
  659. SetIcon(AfxGetApp()->LoadIcon(IDR_MAINFRAME), TRUE);
  660. SetIcon(AfxGetApp()->LoadIcon(IDR_MAINFRAME), FALSE);
  662. AddAnchor(IDC_TREE1, TOP_LEFT, BOTTOM_RIGHT);
  663. AddAnchor(IDC_EDIT1, BOTTOM_LEFT, BOTTOM_RIGHT);
  664. AddAnchor(IDC_BUTTON1, BOTTOM_RIGHT);
  665. AddAnchor(IDC_HLINE, BOTTOM_LEFT, BOTTOM_RIGHT);
  666. AddAnchor(IDC_BUTTON2, BOTTOM_CENTER);
  667. AddAnchor(IDC_BUTTON3, BOTTOM_CENTER);
  668. AddAnchor(IDC_BUTTON4, BOTTOM_CENTER);
  670. CSize s(400, 200);
  671. SetMinTrackSize(s);
  673. m_streamtypesbm.LoadBitmap(IDB_STREAMTYPES);
  674. m_streamtypes.Create(16, 18, ILC_MASK|ILC_COLOR32, 0, 4);
  675. m_streamtypes.Add(&m_streamtypesbm, 0xffffff);
  676. m_tree.SetImageList(&m_streamtypes, TVSIL_NORMAL);
  678. GetWindowText(m_title);
  679. m_nIDEventStatus = SetTimer(1, 1000, NULL);
  681. HRESULT hr;
  682. m_pMux = new CDSMMuxerFilter(NULL, &hr, false, false);
  684. if(FAILED(m_pCGB.CoCreateInstance(CLSID_CaptureGraphBuilder2))
  685. || FAILED(m_pGB.CoCreateInstance(CLSID_FilterGraph))
  686. || FAILED(m_pCGB->SetFiltergraph(m_pGB))
  687. || FAILED(m_pGB->AddFilter(m_pMux, L"Mux"))
  688. || !(m_pMC = m_pGB) || !(m_pME = m_pGB) || !(m_pMS = m_pMux)
  689. || FAILED(m_pME->SetNotifyWindow((OAHWND)m_hWnd, WM_GRAPHNOTIFY, 0))) 
  690. {
  691. MessageBeep(-1);
  692. SendMessage(WM_CLOSE);
  693. return TRUE;
  694. }
  696. AddToRot(m_pGB, &m_dwRegister);
  698. return TRUE;  // return TRUE unless you set the focus to a control
  699. // EXCEPTION: OCX Property Pages should return FALSE
  700. }
  702. void CConvertDlg::OnOK()
  703. {
  704. }
  706. BEGIN_MESSAGE_MAP(CConvertDlg, CResizableDialog)
  707. ON_MESSAGE(WM_GRAPHNOTIFY, OnGraphNotify)
  708. ON_WM_DROPFILES()
  709. ON_WM_CLOSE()
  710. ON_NOTIFY(NM_CLICK, IDC_TREE1, OnNMClickTree1)
  711. ON_NOTIFY(NM_RCLICK, IDC_TREE1, OnNMRclickTree1)
  712. ON_NOTIFY(NM_DBLCLK, IDC_TREE1, OnNMDblclkTree1)
  713. ON_BN_CLICKED(IDC_BUTTON1, OnBnClickedButton1)
  714. ON_UPDATE_COMMAND_UI(IDC_BUTTON1, OnUpdateButton1)
  715. ON_WM_TIMER()
  716. ON_BN_CLICKED(IDC_BUTTON2, OnBnClickedButton2)
  717. ON_UPDATE_COMMAND_UI(IDC_BUTTON2, OnUpdateButton2)
  718. ON_BN_CLICKED(IDC_BUTTON3, OnBnClickedButton3)
  719. ON_UPDATE_COMMAND_UI(IDC_BUTTON3, OnUpdateButton3)
  720. ON_BN_CLICKED(IDC_BUTTON4, OnBnClickedButton4)
  721. ON_UPDATE_COMMAND_UI(IDC_BUTTON4, OnUpdateButton4)
  722. END_MESSAGE_MAP()
  724. // CConvertDlg message handlers
  726. LRESULT CConvertDlg::OnGraphNotify(WPARAM wParam, LPARAM lParam)
  727. {
  728.     HRESULT hr = S_OK;
  730. LONG evCode, evParam1, evParam2;
  731.     while(m_pME && SUCCEEDED(m_pME->GetEvent(&evCode, (LONG_PTR*)&evParam1, (LONG_PTR*)&evParam2, 0)))
  732.     {
  733. hr = m_pME->FreeEventParams(evCode, evParam1, evParam2);
  735. bool fStop = false;
  737.         if(EC_COMPLETE == evCode)
  738.         {
  739. fStop = true;
  740. }
  741. else if(EC_ERRORABORT == evCode)
  742. {
  743. fStop = true;
  745. CString errmsg;
  746. LPVOID lpMsgBuf;
  747. if(FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER|FORMAT_MESSAGE_FROM_SYSTEM|FORMAT_MESSAGE_IGNORE_INSERTS,
  748. NULL, hr, MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), (LPTSTR)&lpMsgBuf, 0, NULL))
  749. {
  750. errmsg = (LPCTSTR)lpMsgBuf;
  751. LocalFree(lpMsgBuf);
  752. }
  754. CString str;
  755. str.Format(_T("Could not complete conversion, the output file is most likely unusable.nnError code: 0x%08x"), evParam1);
  756. if(!errmsg.IsEmpty()) str += _T(" (") + errmsg + _T(")");
  757. AfxMessageBox(str, MB_OK);
  758. }
  760. if(fStop && m_pMC)
  761. {
  762. m_pMC->Stop();
  763. m_tree.EnableWindow(TRUE);
  764. }
  765. }
  767. return hr;
  768. }
  770. void CConvertDlg::OnDropFiles(HDROP hDropInfo)
  771. {
  772. for(int i = 0, j = DragQueryFile(hDropInfo, 0xffffffff, 0, 0); i < j; i++)
  773. {
  774. CString fn;
  775. fn.ReleaseBufferSetLength(DragQueryFile(hDropInfo, i, fn.GetBuffer(MAX_PATH), MAX_PATH));
  777. AddFile(fn);
  778. }
  780. __super::OnDropFiles(hDropInfo);
  781. }
  783. void CConvertDlg::OnClose()
  784. {
  785. HTREEITEM hTI = m_tree.GetRootItem();
  786. while(hTI)
  787. {
  788. HTREEITEM hTINext = m_tree.GetNextSiblingItem(hTI);
  789. DeleteItem(hTI);
  790. hTI = hTINext;
  791. }
  793. if(m_dwRegister) RemoveFromRot(m_dwRegister);
  795. __super::OnClose();
  796. }
  798. void CConvertDlg::OnNMClickTree1(NMHDR* pNMHDR, LRESULT* pResult)
  799. {
  800. CPoint sp, cp;
  801. HTREEITEM hTI = HitTest(sp, cp);
  802. if(!hTI) return;
  803. m_tree.SelectItem(hTI);
  805. *pResult = 0;
  806. }
  808. void CConvertDlg::OnNMRclickTree1(NMHDR* pNMHDR, LRESULT* pResult)
  809. {
  810. CPoint sp, cp;
  811. HTREEITEM hTI = HitTest(sp, cp);
  813. if(hTI)
  814. {
  815. m_tree.SelectItem(hTI);
  817. CTreeItem* t = (CTreeItem*)m_tree.GetItemData(hTI);
  819. if(dynamic_cast<CTreeItemPin*>(t))
  820. ShowPinPopup(hTI, sp);
  821. else if(dynamic_cast<CTreeItemFile*>(t))
  822. ShowFilePopup(hTI, sp);
  823. else if(dynamic_cast<CTreeItemResourceFolder*>(t))
  824. ShowResourceFolderPopup(hTI, sp);
  825. else if(dynamic_cast<CTreeItemResource*>(t))
  826. ShowResourcePopup(hTI, sp);
  827. else if(dynamic_cast<CTreeItemChapterFolder*>(t))
  828. ShowChapterFolderPopup(hTI, sp);
  829. else if(dynamic_cast<CTreeItemChapter*>(t))
  830. ShowChapterPopup(hTI, sp);
  831. }
  832. else
  833. {
  834. ShowPopup(sp);
  835. }
  837. *pResult = 0;
  838. }
  840. void CConvertDlg::OnNMDblclkTree1(NMHDR *pNMHDR, LRESULT *pResult)
  841. {
  842. CPoint sp, cp;
  843. HTREEITEM hTI = HitTest(sp, cp);
  845. if(hTI)
  846. {
  847. CTreeItem* t = (CTreeItem*)m_tree.GetItemData(hTI);
  849. if(CTreeItemPin* t2 = dynamic_cast<CTreeItemPin*>(t))
  850. {
  851. CComPtr<IPin> pPinTo;
  852. t2->m_pPin->ConnectedTo(&pPinTo);
  854. if(CComQIPtr<IDSMPropertyBag> pPB = pPinTo)
  855. EditProperties(pPB);
  856. }
  857. else if(CTreeItemResource* t2 = dynamic_cast<CTreeItemResource*>(t))
  858. {
  859. EditResource(t2);
  860. }
  861. else if(CTreeItemChapter* t2 = dynamic_cast<CTreeItemChapter*>(t))
  862. {
  863. EditChapter(t2);
  864. }
  865. }
  867. *pResult = 0;
  868. }
  870. void CConvertDlg::OnBnClickedButton1()
  871. {
  872. UpdateData();
  874. CFileDialog fd(FALSE, _T(".dsm"), m_fn, 
  875. OFN_EXPLORER|OFN_ENABLESIZING|OFN_HIDEREADONLY|OFN_OVERWRITEPROMPT, 
  876. _T("DirectShow Media file|*.dsm|All files|*.*|"), this, 0);
  878. if(fd.DoModal() != IDOK) return;
  880. if(!SetOutputFile(fd.GetPathName()))
  881. {
  882. AfxMessageBox(_T("Could not set output file"));
  883. }
  884. }
  886. void CConvertDlg::OnUpdateButton1(CCmdUI* pCmdUI)
  887. {
  888. OAFilterState fs;
  889. pCmdUI->Enable(m_pMC && SUCCEEDED(m_pMC->GetState(0, &fs)) && fs == State_Stopped);
  890. }
  892. void CConvertDlg::OnTimer(UINT nIDEvent)
  893. {
  894. if(nIDEvent == m_nIDEventStatus && m_pMS && m_pMC)
  895. {
  896. OAFilterState fs;
  897. if(SUCCEEDED(m_pMC->GetState(0, &fs)) && fs != State_Stopped)
  898. {
  899. GUID tf;
  900. m_pMS->GetTimeFormat(&tf);
  902. REFERENCE_TIME rtCur, rtDur;
  903. HRESULT hr = m_pMS->GetDuration(&rtDur);
  904. m_pMS->GetCurrentPosition(&rtCur);
  906. CString str;
  907. if(hr == S_OK) str.Format(_T("%.2f%%"), 1.0 * (rtCur * 100) / rtDur);
  908. else if(tf == TIME_FORMAT_BYTE) str.Format(_T("%.2fKB"), 1.0 * rtCur / 1024);
  909. else if(tf == TIME_FORMAT_MEDIA_TIME) str.Format(_T("%02d:%02d:%02d"), int(rtCur/3600000000)%60, int(rtCur/60000000)%60, int(rtCur/1000000)%60);
  910. else str = _T("Please Wait");
  912. SetWindowText(_T("Converting - ") + str);
  913. }
  914. else
  915. {
  916. SetWindowText(m_title);
  917. }
  918. }
  920. __super::OnTimer(nIDEvent);
  921. }
  923. void CConvertDlg::OnBnClickedButton2()
  924. {
  925. OAFilterState fs;
  926. if(FAILED(m_pMC->GetState(0, &fs)))
  927. return;
  929. if(fs == State_Stopped)
  930. {
  931. if(m_pMS)
  932. {
  933. LONGLONG pos = 0;
  934. m_pMS->SetPositions(&pos, AM_SEEKING_AbsolutePositioning, NULL, AM_SEEKING_NoPositioning);
  935. }
  937. if(CComQIPtr<IDSMPropertyBag> pPB = m_pMux)
  938. {
  939. pPB->SetProperty(L"APPL", L"Media Player Classic");
  940. }
  942. if(CComQIPtr<IDSMResourceBag> pRB = m_pMux)
  943. {
  944. pRB->ResRemoveAll(0);
  945. POSITION pos = m_pTIs.GetHeadPosition();
  946. while(pos)
  947. {
  948. if(CTreeItemResource* t2 = dynamic_cast<CTreeItemResource*>((CTreeItem*)m_pTIs.GetNext(pos)))
  949. pRB->ResAppend(
  950. t2->m_res.name, t2->m_res.desc, t2->m_res.mime, 
  951. t2->m_res.data.GetData(), t2->m_res.data.GetSize(), 
  952. NULL);
  953. }
  954. }
  956. if(CComQIPtr<IDSMChapterBag> pCB = m_pMux)
  957. {
  958. pCB->ChapRemoveAll();
  959. POSITION pos = m_pTIs.GetHeadPosition();
  960. while(pos)
  961. {
  962. if(CTreeItemChapter* t2 = dynamic_cast<CTreeItemChapter*>((CTreeItem*)m_pTIs.GetNext(pos)))
  963. pCB->ChapAppend(t2->m_chap.rt, t2->m_chap.name);
  964. }
  965. }
  966. }
  968. if(m_pMC) 
  969. {
  970. if(SUCCEEDED(m_pMC->Run()))
  971. m_tree.EnableWindow(FALSE);
  972. }
  973. }
  975. void CConvertDlg::OnUpdateButton2(CCmdUI* pCmdUI)
  976. {
  977. int nIn, nOut, nInC, nOutC;
  978. CountPins(m_pMux, nIn, nOut, nInC, nOutC);
  980. OAFilterState fs;
  981. pCmdUI->Enable(nInC > 0 && nOutC > 0 && GetDlgItem(IDC_EDIT1)->GetWindowTextLength() > 0
  982.         && m_pMS && m_pMC && SUCCEEDED(m_pMC->GetState(0, &fs)) && fs != State_Running);
  983. }
  985. void CConvertDlg::OnBnClickedButton3()
  986. {
  987. if(m_pMC) m_pMC->Pause();
  988. }
  990. void CConvertDlg::OnUpdateButton3(CCmdUI* pCmdUI)
  991. {
  992. OAFilterState fs;
  993. pCmdUI->Enable(m_pMC && SUCCEEDED(m_pMC->GetState(0, &fs)) && fs == State_Running);
  994. }
  996. void CConvertDlg::OnBnClickedButton4()
  997. {
  998. if(m_pMC) m_pMC->Stop();
  999. m_tree.EnableWindow(TRUE);
  1000. }
  1002. void CConvertDlg::OnUpdateButton4(CCmdUI* pCmdUI)
  1003. {
  1004. OAFilterState fs;
  1005. pCmdUI->Enable(m_pMC && SUCCEEDED(m_pMC->GetState(0, &fs)) && fs != State_Stopped);
  1006. }
  1008. //
  1009. // CFilterTreeCtrl
  1010. //
  1012. CFilterTreeCtrl::CFilterTreeCtrl()
  1013. {
  1014. }
  1016. void CFilterTreeCtrl::PreSubclassWindow()
  1017. {
  1018. EnableToolTips(TRUE);
  1020. __super::PreSubclassWindow();
  1021. }
  1023. INT_PTR CFilterTreeCtrl::OnToolHitTest(CPoint p, TOOLINFO* pTI) const
  1024. {
  1025. UINT nFlags;
  1026. HTREEITEM hTI = HitTest(p, &nFlags);
  1027. if(nFlags & TVHT_ONITEM)
  1028. {
  1029. CRect r;
  1030. GetItemRect(hTI, r, TRUE);
  1031. pTI->hwnd = m_hWnd;
  1032. pTI->uId = (UINT)hTI;
  1033. pTI->lpszText = LPSTR_TEXTCALLBACK;
  1034. pTI->rect = r;
  1035. return pTI->uId;
  1036. }
  1038. return -1;
  1039. }
  1041. BEGIN_MESSAGE_MAP(CFilterTreeCtrl, CTreeCtrl)
  1042. ON_NOTIFY_EX_RANGE(TTN_NEEDTEXTW, 0, 0xFFFF, OnToolTipText)
  1043. ON_NOTIFY_EX_RANGE(TTN_NEEDTEXTA, 0, 0xFFFF, OnToolTipText)
  1044. END_MESSAGE_MAP()
  1046. BOOL CFilterTreeCtrl::OnToolTipText(UINT id, NMHDR* pNMHDR, LRESULT* pResult)
  1047. {
  1048. TOOLTIPTEXTA* pTTTA = (TOOLTIPTEXTA*)pNMHDR;
  1049. TOOLTIPTEXTW* pTTTW = (TOOLTIPTEXTW*)pNMHDR;
  1051. UINT nID = pNMHDR->idFrom;
  1053. if(nID == (UINT)m_hWnd
  1054. && (pNMHDR->code == TTN_NEEDTEXTA && (pTTTA->uFlags & TTF_IDISHWND)
  1055. || pNMHDR->code == TTN_NEEDTEXTW && (pTTTW->uFlags & TTF_IDISHWND)))
  1056. return FALSE;
  1058. ::SendMessage(pNMHDR->hwndFrom, TTM_SETMAXTIPWIDTH, 0, (LPARAM)(INT)1000);
  1060. HTREEITEM hTI = (HTREEITEM)nID;
  1062. CString str;
  1063. static CStringA m_strTipTextA;
  1064. static CStringW m_strTipTextW;
  1066. CConvertDlg::CTreeItem* t = (CConvertDlg::CTreeItem*)GetItemData(hTI);
  1067. if(!t || !t->ToolTip(str)) return FALSE;
  1069. m_strTipTextA = str;
  1070. m_strTipTextW = str;
  1072. if(pNMHDR->code == TTN_NEEDTEXTA) pTTTA->lpszText = (LPSTR)(LPCSTR)m_strTipTextA;
  1073. else pTTTW->lpszText = (LPWSTR)(LPCWSTR)m_strTipTextW;
  1075. *pResult = 0;
  1077. return TRUE;    // message was handled
  1078. }
  1080. //
  1081. // CConvertDlg::CTreeItem*
  1082. //
  1084. CConvertDlg::CTreeItem::CTreeItem(CTreeCtrl& tree, HTREEITEM hTIParent) 
  1085. : m_tree(tree)
  1086. {
  1087. m_hTI = m_tree.InsertItem(_T(""), hTIParent);
  1088. m_tree.SetItemData(m_hTI, (DWORD_PTR)this);
  1089. Update();
  1090. }
  1092. CConvertDlg::CTreeItem::~CTreeItem()
  1093. {
  1094. }
  1096. void CConvertDlg::CTreeItem::SetLabel(LPCTSTR label)
  1097. {
  1098. m_tree.SetItemText(m_hTI, label);
  1099. }
  1101. void CConvertDlg::CTreeItem::SetImage(int nImage, int nSelectedImage)
  1102. {
  1103. m_tree.SetItemImage(m_hTI, nImage, nSelectedImage);
  1104. }
  1106. // 
  1108. CConvertDlg::CTreeItemFilter::CTreeItemFilter(IBaseFilter* pBF, CTreeCtrl& tree, HTREEITEM hTIParent) 
  1109. : CTreeItem(tree, hTIParent)
  1110. , m_pBF(pBF)
  1111. {
  1112. Update();
  1113. }
  1115. void CConvertDlg::CTreeItemFilter::Update()
  1116. {
  1117. SetLabel(CString(GetFilterName(m_pBF)));
  1118. }
  1120. //
  1122. CConvertDlg::CTreeItemFile::CTreeItemFile(CString fn, IBaseFilter* pBF, CTreeCtrl& tree, HTREEITEM hTIParent)
  1123. : CTreeItemFilter(pBF, tree, hTIParent)
  1124. , m_fn(fn)
  1125. {
  1126. Update();
  1127. }
  1129. void CConvertDlg::CTreeItemFile::Update()
  1130. {
  1131. CPath path = m_fn;
  1132. path.StripPath();
  1133. SetLabel(path);
  1134. }
  1136. bool CConvertDlg::CTreeItemFile::ToolTip(CString& str)
  1137. {
  1138. str = m_fn;
  1139. return true;
  1140. }
  1142. //
  1144. CConvertDlg::CTreeItemPin::CTreeItemPin(IPin* pPin, CTreeCtrl& tree, HTREEITEM hTIParent)
  1145. : CTreeItem(tree, hTIParent)
  1146. , m_pPin(pPin)
  1147. {
  1148. Update();
  1149. }
  1151. void CConvertDlg::CTreeItemPin::Update()
  1152. {
  1153. if(!m_pPin) {ASSERT(0); return;}
  1155. CString label = GetPinName(m_pPin);
  1156. if(!IsConnected()) label = _T("[D] ") + label;
  1157. SetLabel(label);
  1159. CMediaType mt;
  1160. if(S_OK == m_pPin->ConnectionMediaType(&mt))
  1161. {
  1162. if(mt.majortype == MEDIATYPE_Video) SetImage(1, 1);
  1163. else if(mt.majortype == MEDIATYPE_Audio) SetImage(2, 2);
  1164. else if(mt.majortype == MEDIATYPE_Text || mt.majortype == MEDIATYPE_Subtitle) SetImage(3, 3);
  1165. }
  1166. }
  1168. bool CConvertDlg::CTreeItemPin::ToolTip(CString& str)
  1169. {
  1170. CMediaTypeEx mt;
  1171. if(FAILED(m_pPin->ConnectionMediaType(&mt))) return false;
  1172. str = mt.ToString(m_pPin);
  1173. return true;
  1174. }
  1176. bool CConvertDlg::CTreeItemPin::IsConnected()
  1177. {
  1178. CComPtr<IPin> pPinTo;
  1179. return m_pPin && SUCCEEDED(m_pPin->ConnectedTo(&pPinTo)) && pPinTo;
  1180. }
  1182. //
  1184. CConvertDlg::CTreeItemResourceFolder::CTreeItemResourceFolder(CTreeCtrl& tree, HTREEITEM hTIParent)
  1185. : CTreeItem(tree, hTIParent)
  1186. {
  1187. Update();
  1188. }
  1190. void CConvertDlg::CTreeItemResourceFolder::Update()
  1191. {
  1192. SetLabel(_T("Resources"));
  1193. }
  1195. bool CConvertDlg::CTreeItemResourceFolder::ToolTip(CString& str)
  1196. {
  1197. if(!m_tree.ItemHasChildren(m_hTI))
  1198. return false;
  1200. int files = 0;
  1201. float size = 0;
  1203. HTREEITEM hChildItem = m_tree.GetChildItem(m_hTI);
  1205. while(hChildItem != NULL)
  1206. {
  1207. HTREEITEM hNextItem = m_tree.GetNextItem(hChildItem, TVGN_NEXT);
  1208. if(CTreeItemResource* t = dynamic_cast<CTreeItemResource*>((CTreeItem*)m_tree.GetItemData(hChildItem)))
  1209. size += t->m_res.data.GetSize(), files++;
  1210. hChildItem = hNextItem;
  1211. }
  1213. size /= 1024;
  1214. if(size < 1024) str.Format(_T("%d file(s), %.2f KB"), files, size);
  1215. else str.Format(_T("%d file(s), %.2f MB"), files, size/1024);
  1217. return true;
  1218. }
  1220. //
  1222. CConvertDlg::CTreeItemResource::CTreeItemResource(const CDSMResource& res, CTreeCtrl& tree, HTREEITEM hTIParent)
  1223. : CTreeItem(tree, hTIParent)
  1224. {
  1225. m_res = res;
  1226. Update();
  1227. }
  1229. CConvertDlg::CTreeItemResource::~CTreeItemResource()
  1230. {
  1231. }
  1233. void CConvertDlg::CTreeItemResource::Update()
  1234. {
  1235. SetLabel(CString(m_res.name));
  1237. CStringW mime = m_res.mime;
  1238. mime.Trim();
  1239. mime.MakeLower();
  1240. if(mime == L"application/x-truetype-font") SetImage(4, 4);
  1241. else if(mime.Find(L"text/") == 0) SetImage(5, 5);
  1242. else SetImage(6, 6);
  1243. }
  1245. bool CConvertDlg::CTreeItemResource::ToolTip(CString& str)
  1246. {
  1247. if(!m_res.mime.IsEmpty()) str = CString(m_res.mime) + _T("rnrn");
  1248. if(!m_res.desc.IsEmpty()) str += CString(m_res.desc);
  1249. str.Trim();
  1250. return true;
  1251. }
  1253. //
  1255. CConvertDlg::CTreeItemChapterFolder::CTreeItemChapterFolder(CTreeCtrl& tree, HTREEITEM hTIParent)
  1256. : CTreeItem(tree, hTIParent)
  1257. {
  1258. Update();
  1259. }
  1261. void CConvertDlg::CTreeItemChapterFolder::Update()
  1262. {
  1263. SetLabel(_T("Chapters"));
  1264. }
  1266. //
  1268. CConvertDlg::CTreeItemChapter::CTreeItemChapter(const CDSMChapter& chap, CTreeCtrl& tree, HTREEITEM hTIParent)
  1269. : CTreeItem(tree, hTIParent)
  1270. {
  1271. m_chap = chap;
  1272. Update();
  1273. }
  1275. void CConvertDlg::CTreeItemChapter::Update()
  1276. {
  1277. REFERENCE_TIME rt = m_chap.rt;
  1278. rt /= 10000;
  1279. int ms = (int)(rt%1000);
  1280. rt /= 1000;
  1281. int s = (int)(rt%60);
  1282. rt /= 60;
  1283. int m = (int)(rt%60);
  1284. rt /= 60;
  1285. int h = (int)(rt);
  1287. CString label;
  1288. label.Format(_T("%02d:%02d:%02d.%03d - %s"), h, m, s, ms, CString(m_chap.name));
  1289. SetLabel(label);
  1291. SetImage(7, 7);
  1292. }