?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
X64 transport [Part 5] (Update plugins.cpp)
[xy_vsfilter.git] / src / apps / mplayerc / ConvertDlg.cpp
blob258854500f96bbcb96e04fe1dbbc9fe990095cc8
1 /*
2 * Copyright (C) 2003-2006 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 */
21
22 // ConvertDlg.cpp : implementation file
23 //
24
25 #include "stdafx.h"
26 #include <mmreg.h>
27 #include "mplayerc.h"
28 #include "..\..\filters\filters.h"
29 #include "..\..\..\include\moreuuids.h"
30 #include "FGManager.h"
31 #include "ConvertPropsDlg.h"
32 #include "ConvertResDlg.h"
33 #include "ConvertChapDlg.h"
34 #include "ConvertDlg.h"
35
36 // TODO: subtitle source filter for vobsub
37
38 // CConvertDlg dialog
39
40 CConvertDlg::CConvertDlg(CWnd* pParent /*=NULL*/)
41 : CResizableDialog(CConvertDlg::IDD, pParent)
42 , m_fn(_T(""))
43 {
44 }
45
46 CConvertDlg::~CConvertDlg()
47 {
48 }
49
50 void CConvertDlg::AddFile(CString fn)
51 {
52 CString protocol;
53
54 int i = fn.Find(_T("://"));
55 if(i > 0)
56 {
57 CString url = fn.Mid(i);
58 CPath path(fn.Left(i));
59 path.StripPath();
60 protocol = (LPCTSTR)path;
61 fn = (LPCTSTR)path + url;
62 }
63
64 CComPtr<IBaseFilter> pBF;
65 if(FAILED(m_pGB->AddSourceFilter(CStringW(fn), CStringW(fn), &pBF)))
66 return;
67
68 int nConnected = 0;
69 BeginEnumPins(pBF, pEP, pPin)
70 if(S_OK == m_pGB->ConnectFilter(pPin, m_pMux)) nConnected++;
71 EndEnumPins
72 if(!nConnected) {MessageBeep(-1); DeleteFilter(pBF); return;}
73
74 if(m_tree.GetCount() == 0)
75 {
76 if(CComQIPtr<IDSMPropertyBag> pPB = m_pMux)
77 pPB->DelAllProperties();
78
79 CString ext(_T(".dsm"));
80
81 if(!protocol.IsEmpty())
82 {
83 m_fn = protocol + ext;
84 }
85 else
86 {
87 CPath p(fn);
88 if(ext.CompareNoCase(p.GetExtension()) == 0)
89 ext = _T(" (remuxed)") + ext;
90 p.RemoveExtension();
91 m_fn = (LPCTSTR)p + ext;
92 }
93
94 UpdateData(FALSE);
95 }
96
97 CTreeItemFile* t = new CTreeItemFile(fn, pBF, m_tree, NULL);
98
99 AddFilter(*t, pBF);
100
101 m_tree.Expand(*t, TVE_EXPAND);
102 m_tree.EnsureVisible(*t);
103 }
104
105 bool CConvertDlg::ConvertFile(LPCTSTR fn, IPin* pPin)
106 {
107 OAFilterState fs;
108 if(!m_pMC || FAILED(m_pMC->GetState(0, &fs)) || fs != State_Stopped)
109 return false;
110
111 m_pGB->NukeDownstream(m_pMux);
112
113 CComPtr<IBaseFilter> pFW;
114 pFW.CoCreateInstance(CLSID_FileWriter);
115 CComQIPtr<IFileSinkFilter2> pFSF = pFW;
116
117 if(pPin)
118 {
119 CComQIPtr<IBaseMuxerRelatedPin> pRP = pPin;
120 if(!pRP) return false;
121
122 pPin = pRP->GetRelatedPin();
123 }
124 else
125 {
126 pPin = GetFirstPin(m_pMux, PINDIR_OUTPUT);
127 }
128
129 if(!pPin || !pFSF
130 || FAILED(m_pGB->AddFilter(pFW, NULL))
131 || FAILED(pFSF->SetFileName(CStringW(fn), NULL))
132 || FAILED(pFSF->SetMode(AM_FILE_OVERWRITE))
133 || FAILED(m_pGB->ConnectDirect(pPin, GetFirstPin(pFW), NULL)))
134 {
135 m_pGB->RemoveFilter(pFW);
136 return false;
137 }
138
139 if(m_pMS)
140 {
141 LONGLONG pos = 0;
142 m_pMS->SetPositions(&pos, AM_SEEKING_AbsolutePositioning, NULL, AM_SEEKING_NoPositioning);
143 }
144
145 if(CComQIPtr<IDSMPropertyBag> pPB = m_pMux)
146 {
147 pPB->SetProperty(L"APPL", L"Media Player Classic");
148 }
149
150 if(CComQIPtr<IDSMResourceBag> pRB = m_pMux)
151 {
152 pRB->ResRemoveAll(0);
153 POSITION pos = m_pTIs.GetHeadPosition();
154 while(pos)
155 {
156 if(CTreeItemResource* t2 = dynamic_cast<CTreeItemResource*>((CTreeItem*)m_pTIs.GetNext(pos)))
157 pRB->ResAppend(
158 t2->m_res.name, t2->m_res.desc, t2->m_res.mime,
159 t2->m_res.data.GetData(), t2->m_res.data.GetCount(),
160 NULL);
161 }
162 }
163
164 if(CComQIPtr<IDSMChapterBag> pCB = m_pMux)
165 {
166 pCB->ChapRemoveAll();
167 POSITION pos = m_pTIs.GetHeadPosition();
168 while(pos)
169 {
170 if(CTreeItemChapter* t2 = dynamic_cast<CTreeItemChapter*>((CTreeItem*)m_pTIs.GetNext(pos)))
171 pCB->ChapAppend(t2->m_chap.rt, t2->m_chap.name);
172 }
173 }
174
175 if(FAILED(m_pMC->Run()))
176 return false;
177
178 m_tree.EnableWindow(FALSE);
179
180 return true;
181 }
182
183 void CConvertDlg::AddFilter(HTREEITEM hTIParent, IBaseFilter* pBFParent)
184 {
185 BeginEnumPins(pBFParent, pEP, pPin)
186 {
187 CComPtr<IPin> pPinTo;
188 CComPtr<IBaseFilter> pBF;
189 if(S_OK != m_pGB->IsPinDirection(pPin, PINDIR_OUTPUT)
190 || FAILED(pPin->ConnectedTo(&pPinTo)) || !pPinTo
191 || !(pBF = GetFilterFromPin(pPinTo)))
192 continue;
193
194 CTreeItem* t = NULL;
195
196 if(pBF == m_pMux)
197 {
198 t = new CTreeItemPin(pPin, m_tree, hTIParent);
199 }
200 else
201 {
202 t = new CTreeItemFilter(pBF, m_tree, hTIParent);
203 AddFilter(*t, pBF);
204 }
205 }
206 EndEnumPins
207
208 if(CComQIPtr<IDSMPropertyBag> pPB = pBFParent)
209 {
210 ULONG props;
211 if(FAILED(pPB->CountProperties(&props)))
212 props = 0;
213
214 for(ULONG i = 0; i < props; i++)
215 {
216 PROPBAG2 PropBag;
217 memset(&PropBag, 0, sizeof(PropBag));
218 ULONG cPropertiesReturned = 0;
219 if(FAILED(pPB->GetPropertyInfo(i, 1, &PropBag, &cPropertiesReturned)))
220 continue;
221
222 HRESULT hr;
223 CComVariant var;
224 if(SUCCEEDED(pPB->Read(1, &PropBag, NULL, &var, &hr)) && SUCCEEDED(hr))
225 {
226 CComQIPtr<IDSMPropertyBag> pPBMux = m_pMux;
227 CComBSTR value;
228 if(pPBMux && FAILED(pPBMux->GetProperty(PropBag.pstrName, &value)))
229 pPBMux->SetProperty(PropBag.pstrName, var.bstrVal);
230 }
231
232 CoTaskMemFree(PropBag.pstrName);
233 }
234 }
235
236 CTreeItem* t2 = new CTreeItemResourceFolder(m_tree, hTIParent);
237 if(CComQIPtr<IDSMResourceBag> pRB = pBFParent)
238 {
239 for(DWORD i = 0, cnt = pRB->ResGetCount(); i < cnt; i++)
240 {
241 CComBSTR name, mime, desc;
242 BYTE* pData = NULL;
243 DWORD len = 0;
244 if(FAILED(pRB->ResGet(i, &name, &desc, &mime, &pData, &len, NULL)))
245 continue;
246
247 if(len > 0)
248 {
249 m_pTIs.AddTail(new CTreeItemResource(CDSMResource(name, desc, mime, pData, len), m_tree, *t2));
250 }
251
252 CoTaskMemFree(pData);
253 }
254 }
255 m_tree.Expand(*t2, TVE_EXPAND);
256
257 CTreeItem* t3 = new CTreeItemChapterFolder(m_tree, hTIParent);
258 if(CComQIPtr<IDSMChapterBag> pCB = pBFParent)
259 {
260 for(DWORD i = 0, cnt = pCB->ChapGetCount(); i < cnt; i++)
261 {
262 REFERENCE_TIME rt;
263 CComBSTR name;
264 if(FAILED(pCB->ChapGet(i, &rt, &name)))
265 continue;
266
267 m_pTIs.AddTail(new CTreeItemChapter(CDSMChapter(rt, name), m_tree, *t3));
268 }
269 }
270 m_tree.Expand(*t3, TVE_EXPAND);
271
272 m_tree.Expand(hTIParent, TVE_EXPAND);
273 }
274
275 void CConvertDlg::DeleteFilter(IBaseFilter* pBF)
276 {
277 BeginEnumPins(pBF, pEP, pPin)
278 {
279 CComPtr<IPin> pPinTo;
280 CComPtr<IBaseFilter> pBF;
281 if(S_OK != m_pGB->IsPinDirection(pPin, PINDIR_OUTPUT)
282 || FAILED(pPin->ConnectedTo(&pPinTo)) || !pPinTo
283 || !(pBF = GetFilterFromPin(pPinTo)))
284 continue;
285
286 if(pBF != m_pMux) DeleteFilter(pBF);
287 }
288 EndEnumPins
289
290 m_pGB->RemoveFilter(pBF);
291 }
292
293 void CConvertDlg::DeleteItem(HTREEITEM hTI)
294 {
295 if(!hTI) return;
296
297 DeleteChildren(hTI);
298
299 CTreeItem* t = (CTreeItem*)m_tree.GetItemData(hTI);
300 if(POSITION pos = m_pTIs.Find(t)) m_pTIs.RemoveAt(pos);
301 delete t;
302 m_tree.DeleteItem(hTI);
303 }
304
305 void CConvertDlg::DeleteChildren(HTREEITEM hTI)
306 {
307 if(!hTI) return;
308
309 if(m_tree.ItemHasChildren(hTI))
310 {
311 HTREEITEM hChildItem = m_tree.GetChildItem(hTI);
312
313 while(hChildItem != NULL)
314 {
315 HTREEITEM hNextItem = m_tree.GetNextItem(hChildItem, TVGN_NEXT);
316 DeleteItem(hChildItem);
317 hChildItem = hNextItem;
318 }
319 }
320 }
321
322 HTREEITEM CConvertDlg::HitTest(CPoint& sp, CPoint& cp)
323 {
324 sp = CPoint((LPARAM)GetMessagePos());
325 cp = sp;
326 m_tree.ScreenToClient(&cp);
327 UINT flags = 0;
328 HTREEITEM hTI = m_tree.HitTest(cp, &flags);
329 return hTI && (flags&TVHT_ONITEM) ? hTI : NULL;
330 }
331
332 void CConvertDlg::ShowPopup(CPoint p)
333 {
334 CMenu m;
335 m.CreatePopupMenu();
336
337 int i = 1;
338 m.AppendMenu(MF_STRING, i++, ResStr(IDS_CONVERT_ADDFILE));
339 m.AppendMenu(MF_SEPARATOR);
340 m.AppendMenu(MF_STRING, i++, ResStr(IDS_CONVERT_PROPERTIES));
341
342 switch((int)m.TrackPopupMenu(TPM_LEFTBUTTON|TPM_RETURNCMD, p.x, p.y, this))
343 {
344 case 1:
345 {
346 CFileDialog fd(TRUE, NULL, m_fn,
347 OFN_EXPLORER|OFN_ENABLESIZING|OFN_HIDEREADONLY|OFN_NOVALIDATE,
348 _T("Media files|*.*||"), this, 0);
349 if(fd.DoModal() == IDOK) AddFile(fd.GetPathName());
350 }
351 break;
352 case 2:
353 EditProperties(CComQIPtr<IDSMPropertyBag>(m_pMux));
354 break;
355 }
356 }
357
358 void CConvertDlg::ShowFilePopup(HTREEITEM hTI, CPoint p)
359 {
360 CTreeItemFile* t = dynamic_cast<CTreeItemFile*>((CTreeItem*)m_tree.GetItemData(hTI));
361 ASSERT(t);
362
363 CMenu m;
364 m.CreatePopupMenu();
365
366 int i = 1;
367 m.AppendMenu(MF_STRING, i++, ResStr(IDS_CONVERT_REMOVE));
368
369 switch((int)m.TrackPopupMenu(TPM_LEFTBUTTON|TPM_RETURNCMD, p.x, p.y, this))
370 {
371 case 1:
372 DeleteFilter(t->m_pBF);
373 DeleteItem(hTI);
374 break;
375 }
376 }
377
378 void CConvertDlg::ShowPinPopup(HTREEITEM hTI, CPoint p)
379 {
380 CTreeItemPin* t = dynamic_cast<CTreeItemPin*>((CTreeItem*)m_tree.GetItemData(hTI));
381 ASSERT(t);
382
383 if(!t->m_pPin) return;
384
385 CComPtr<IPin> pPinTo;
386 t->m_pPin->ConnectedTo(&pPinTo);
387
388 CMediaType mt;
389 if(pPinTo) t->m_pPin->ConnectionMediaType(&mt);
390
391 CAtlArray<CMediaType> mts;
392 BeginEnumMediaTypes(t->m_pPin, pEMT, pmt)
393 mts.Add(*pmt);
394 EndEnumMediaTypes(pmt)
395
396 CMenu m;
397 m.CreatePopupMenu();
398
399 int i = 1, mtbase = 1000, mti = mtbase;
400
401 m.AppendMenu(MF_STRING, i++, !pPinTo ? ResStr(IDS_CONVERT_ENABLESTREAM) : ResStr(IDS_CONVERT_DISABLESTREAM));
402 m.AppendMenu(MF_STRING | (!pPinTo ? MF_GRAYED : 0), i++, ResStr(IDS_CONVERT_DEMUXSTREAM));
403
404 if(mts.GetCount() > 1)
405 {
406 m.AppendMenu(MF_SEPARATOR);
407 for(int i = 0; i < mts.GetCount(); i++)
408 m.AppendMenu(MF_STRING | (mts[i] == mt ? MF_CHECKED : 0), mti++, CMediaTypeEx(mts[i]).ToString());
409 }
410
411 m.AppendMenu(MF_SEPARATOR);
412 m.AppendMenu(MF_STRING | (!pPinTo ? MF_GRAYED : 0), i++, ResStr(IDS_CONVERT_PINPROPERTIES));
413
414 switch(i = (int)m.TrackPopupMenu(TPM_LEFTBUTTON|TPM_RETURNCMD, p.x, p.y, this))
415 {
416 case 1:
417 if(pPinTo) {m_pGB->Disconnect(pPinTo); m_pGB->Disconnect(t->m_pPin);}
418 else if(pPinTo = GetFirstDisconnectedPin(m_pMux, PINDIR_INPUT)) m_pGB->ConnectDirect(t->m_pPin, pPinTo, NULL);
419 t->Update();
420 break;
421 case 2:
422 {
423 UpdateData();
424
425 CString ext = _T("raw");
426
427 if(mt.subtype == MEDIASUBTYPE_AAC) ext = _T("aac");
428 else if(mt.subtype == MEDIASUBTYPE_MP3) ext = _T("mp3");
429 else if(mt.subtype == FOURCCMap(WAVE_FORMAT_MPEG)) ext = _T("m1a");
430 else if(mt.subtype == MEDIASUBTYPE_MPEG2_AUDIO) ext = _T("m2a");
431 else if(mt.subtype == MEDIASUBTYPE_WAVE_DOLBY_AC3 || mt.subtype == MEDIASUBTYPE_DOLBY_AC3) ext = _T("ac3");
432 else if(mt.subtype == MEDIASUBTYPE_WAVE_DTS || mt.subtype == MEDIASUBTYPE_DTS) ext = _T("dts");
433 else if((mt.subtype == FOURCCMap('1CVA') || mt.subtype == FOURCCMap('1cva')) && mt.formattype == FORMAT_MPEG2_VIDEO) ext = _T("h264");
434 else if(mt.subtype == FOURCCMap('GEPJ') || mt.subtype == FOURCCMap('gepj')) ext = _T("jpg");
435 else if(mt.majortype == MEDIATYPE_Video && mt.subtype == MEDIASUBTYPE_MPEG2_VIDEO) ext = _T("m2v");
436 else if(mt.majortype == MEDIATYPE_Video && mt.subtype == MEDIASUBTYPE_MPEG1Payload) ext = _T("m1v");
437 else if(mt.subtype == MEDIASUBTYPE_UTF8 || mt.majortype == MEDIATYPE_Text) ext = _T("srt");
438 else if(mt.subtype == MEDIASUBTYPE_SSA) ext = _T("ssa");
439 else if(mt.subtype == MEDIASUBTYPE_ASS || mt.subtype == MEDIASUBTYPE_ASS2) ext = _T("ass");
440 else if(mt.subtype == MEDIASUBTYPE_SSF) ext = _T("ssf");
441 else if(mt.subtype == MEDIASUBTYPE_VOBSUB) ext = _T("sub");
442 else if(mt.subtype == MEDIASUBTYPE_PCM || mt.subtype == MEDIASUBTYPE_DVD_LPCM_AUDIO || mt.subtype == FOURCCMap(WAVE_FORMAT_EXTENSIBLE) || mt.subtype == FOURCCMap(WAVE_FORMAT_IEEE_FLOAT)) ext = _T("wav");
443 // TODO: else if...
444
445 CPath path(m_fn);
446 path.RenameExtension('.' + ext);
447
448 CFileDialog fd(FALSE, NULL, (LPCTSTR)path,
449 OFN_EXPLORER|OFN_ENABLESIZING|OFN_HIDEREADONLY,
450 _T("Media files|*.*||"), this, 0);
451 if(fd.DoModal() == IDOK)
452 {
453 if(!ConvertFile(fd.GetPathName(), pPinTo))
454 {
455 AfxMessageBox(_T("Failed to start conversion"));
456 }
457 }
458 }
459 break;
460 case 3:
461 EditProperties(CComQIPtr<IDSMPropertyBag>(pPinTo));
462 break;
463 default:
464 i -= mtbase;
465 if(i >= 0 && i < mts.GetCount())
466 {
467 if(pPinTo) {m_pGB->Disconnect(pPinTo); m_pGB->Disconnect(t->m_pPin);}
468 else {pPinTo = GetFirstDisconnectedPin(m_pMux, PINDIR_INPUT);}
469 HRESULT hr = m_pGB->ConnectDirect(t->m_pPin, pPinTo, &mts[i]);
470 if(FAILED(hr))
471 {
472 AfxMessageBox(_T("Reconnection attempt failed!"));
473 if(mt.majortype != GUID_NULL)
474 hr = m_pGB->ConnectDirect(t->m_pPin, pPinTo, &mt);
475 }
476 t->Update();
477 }
478 break;
479 }
480 }
481
482 void CConvertDlg::ShowResourceFolderPopup(HTREEITEM hTI, CPoint p)
483 {
484 CTreeItemResourceFolder* t = dynamic_cast<CTreeItemResourceFolder*>((CTreeItem*)m_tree.GetItemData(hTI));
485 ASSERT(t);
486
487 CMenu m;
488 m.CreatePopupMenu();
489
490 int i = 1;
491 m.AppendMenu(MF_STRING, i++, ResStr(IDS_CONVERT_ADDRESOURCE));
492 if(m_tree.ItemHasChildren(*t))
493 {
494 m.AppendMenu(MF_SEPARATOR);
495 m.AppendMenu(MF_STRING, i++, ResStr(IDS_CONVERT_REMOVEALL));
496 }
497
498 switch((int)m.TrackPopupMenu(TPM_LEFTBUTTON|TPM_RETURNCMD, p.x, p.y, this))
499 {
500 case 1:
501 {
502 CFileDialog fd(TRUE, NULL, NULL,
503 OFN_EXPLORER|OFN_ENABLESIZING|OFN_HIDEREADONLY,
504 _T("All files|*.*||"), this, 0);
505 if(fd.DoModal() == IDOK)
506 {
507 CString fn = fd.GetPathName();
508 if(FILE* f = _tfopen(fn, _T("rb")))
509 {
510 CDSMResource res;
511
512 CPath path(fn);
513 path.StripPath();
514 res.name = (LPCTSTR)path;
515
516 CRegKey key;
517 TCHAR mime[256];
518 ULONG len = countof(mime);
519 if(ERROR_SUCCESS == key.Open(HKEY_CLASSES_ROOT, path.GetExtension().MakeLower(), KEY_READ)
520 && ERROR_SUCCESS == key.QueryStringValue(_T("Content Type"), mime, &len))
521 res.mime = mime;
522
523 CTreeItemResource* t = new CTreeItemResource(res, m_tree, hTI);
524 m_pTIs.AddTail(t);
525
526 if(EditResource(t))
527 {
528 fseek(f, 0, 2);
529 long size = ftell(f);
530 fseek(f, 0, 0);
531 t->m_res.data.SetCount(size);
532 for(BYTE* ptr = t->m_res.data.GetData(),* end = ptr + size;
533 size > 0 && end - ptr >= size && fread(ptr, min(size, 1024), 1, f) > 0;
534 ptr += 1024, size -= 1024);
535 fclose(f);
536 }
537 else
538 {
539 DeleteItem(*t);
540 }
541 }
542 else
543 {
544 AfxMessageBox(_T("Cannot open file!"));
545 }
546 }
547 }
548 break;
549 case 2:
550 DeleteChildren(hTI);
551 break;
552 }
553 }
554
555 void CConvertDlg::ShowResourcePopup(HTREEITEM hTI, CPoint p)
556 {
557 CTreeItemResource* t = dynamic_cast<CTreeItemResource*>((CTreeItem*)m_tree.GetItemData(hTI));
558 ASSERT(t);
559
560 CMenu m;
561 m.CreatePopupMenu();
562
563 int i = 1;
564 m.AppendMenu(MF_STRING, i++, ResStr(IDS_CONVERT_REMOVE));
565 m.AppendMenu(MF_STRING, i++, ResStr(IDS_CONVERT_SAVEAS));
566 if(AfxGetAppSettings().fEnableWebServer) m.AppendMenu(MF_STRING, 1000, ResStr(IDS_CONVERT_LAUNCHINBROWSER));
567 m.AppendMenu(MF_SEPARATOR);
568 m.AppendMenu(MF_STRING, i++, ResStr(IDS_CONVERT_RESOURCEPROPERTIES));
569
570 switch((int)m.TrackPopupMenu(TPM_LEFTBUTTON|TPM_RETURNCMD, p.x, p.y, this))
571 {
572 case 1:
573 DeleteItem(*t);
574 break;
575 case 2:
576 {
577 CFileDialog fd(FALSE, NULL, CString(t->m_res.name),
578 OFN_EXPLORER|OFN_ENABLESIZING|OFN_HIDEREADONLY|OFN_OVERWRITEPROMPT,
579 _T("All files|*.*||"), this, 0);
580 if(fd.DoModal() == IDOK)
581 {
582 if(FILE* f = _tfopen(fd.GetPathName(), _T("wb")))
583 {
584 fwrite(t->m_res.data.GetData(), 1, t->m_res.data.GetCount(), f);
585 fclose(f);
586 }
587 }
588 }
589 break;
590 case 3:
591 EditResource(t);
592 break;
593 case 1000:
594 {
595 CString url;
596 url.Format(_T("http://localhost:%d/convres.html?id=%x"), AfxGetAppSettings().nWebServerPort, (DWORD)&t->m_res);
597 ShellExecute(NULL, _T("open"), url, NULL, NULL, SW_SHOWDEFAULT);
598 }
599 break;
600 }
601 }
602
603 bool CConvertDlg::EditProperties(IDSMPropertyBag* pPB)
604 {
605 CConvertPropsDlg dlg(!!CComQIPtr<IPin>(pPB), this);
606
607 ULONG props;
608 if(FAILED(pPB->CountProperties(&props)))
609 props = 0;
610
611 for(ULONG i = 0; i < props; i++)
612 {
613 PROPBAG2 PropBag;
614 memset(&PropBag, 0, sizeof(PropBag));
615 ULONG cPropertiesReturned = 0;
616 if(FAILED(pPB->GetPropertyInfo(i, 1, &PropBag, &cPropertiesReturned)))
617 continue;
618
619 HRESULT hr;
620 CComVariant var;
621 if(SUCCEEDED(pPB->Read(1, &PropBag, NULL, &var, &hr)) && SUCCEEDED(hr))
622 dlg.m_props[CString(PropBag.pstrName)] = CString(var);
623
624 CoTaskMemFree(PropBag.pstrName);
625 }
626
627 if(IDOK != dlg.DoModal())
628 return false;
629
630 pPB->DelAllProperties();
631
632 POSITION pos = dlg.m_props.GetStartPosition();
633 while(pos)
634 {
635 CString key, value;
636 dlg.m_props.GetNextAssoc(pos, key, value);
637 pPB->SetProperty(CStringW(key), CStringW(value));
638 }
639
640 return true;
641 }
642
643 bool CConvertDlg::EditResource(CTreeItemResource* t)
644 {
645 CConvertResDlg dlg(this);
646
647 dlg.m_name = t->m_res.name;
648 dlg.m_mime = t->m_res.mime;
649 dlg.m_desc = t->m_res.desc;
650
651 if(IDOK != dlg.DoModal())
652 return false;
653
654 t->m_res.name = dlg.m_name;
655 t->m_res.mime = dlg.m_mime;
656 t->m_res.desc = dlg.m_desc;
657
658 t->Update();
659
660 return true;
661 }
662
663 void CConvertDlg::ShowChapterFolderPopup(HTREEITEM hTI, CPoint p)
664 {
665 CTreeItemChapterFolder* t = dynamic_cast<CTreeItemChapterFolder*>((CTreeItem*)m_tree.GetItemData(hTI));
666 ASSERT(t);
667
668 CMenu m;
669 m.CreatePopupMenu();
670
671 int i = 1;
672 m.AppendMenu(MF_STRING, i++, ResStr(IDS_CONVERT_ADDCHAPTER));
673 if(m_tree.ItemHasChildren(*t))
674 {
675 m.AppendMenu(MF_SEPARATOR);
676 m.AppendMenu(MF_STRING, i++, ResStr(IDS_CONVERT_REMOVEALL));
677 }
678
679 switch((int)m.TrackPopupMenu(TPM_LEFTBUTTON|TPM_RETURNCMD, p.x, p.y, this))
680 {
681 case 1:
682 {
683 CDSMChapter chap;
684 CTreeItemChapter* t = new CTreeItemChapter(CDSMChapter(0, L""), m_tree, hTI);
685 m_pTIs.AddTail(t);
686 if(!EditChapter(t))
687 DeleteItem(*t);
688 }
689 break;
690 case 2:
691 DeleteChildren(hTI);
692 break;
693 }
694 }
695
696 void CConvertDlg::ShowChapterPopup(HTREEITEM hTI, CPoint p)
697 {
698 CTreeItemChapter* t = dynamic_cast<CTreeItemChapter*>((CTreeItem*)m_tree.GetItemData(hTI));
699 ASSERT(t);
700
701 CMenu m;
702 m.CreatePopupMenu();
703
704 int i = 1;
705 m.AppendMenu(MF_STRING, i++, ResStr(IDS_CONVERT_REMOVE));
706 m.AppendMenu(MF_SEPARATOR);
707 m.AppendMenu(MF_STRING, i++, ResStr(IDS_CONVERT_CHAPTERPROPERTIES));
708
709 switch((int)m.TrackPopupMenu(TPM_LEFTBUTTON|TPM_RETURNCMD, p.x, p.y, this))
710 {
711 case 1:
712 DeleteItem(hTI);
713 break;
714 case 2:
715 EditChapter(t);
716 break;
717 }
718 }
719
720 bool CConvertDlg::EditChapter(CTreeItemChapter* t)
721 {
722 CConvertChapDlg dlg(this);
723
724 int h = (int)(t->m_chap.rt/10000000/60/60);
725 int m = (int)(t->m_chap.rt/10000000/60%60);
726 int s = (int)(t->m_chap.rt/10000000%60);
727 int ms = (int)(t->m_chap.rt/10000%1000);
728
729 dlg.m_name = t->m_chap.name;
730 dlg.m_time.Format(_T("%02d:%02d:%02d.%03d"), h, m, s, ms);
731
732 if(IDOK != dlg.DoModal())
733 return false;
734
735 TCHAR c;
736 if(_stscanf(dlg.m_time, _T("%d%c%d%c%d%c%d"), &h, &c, &m, &c, &s, &c, &ms) != 7)
737 return false;
738
739 t->m_chap.name = dlg.m_name;
740 t->m_chap.rt = ((((__int64)h*60+m)*60+s)*1000+ms)*10000;
741
742 t->Update();
743
744 return true;
745 }
746
747 void CConvertDlg::DoDataExchange(CDataExchange* pDX)
748 {
749 __super::DoDataExchange(pDX);
750 DDX_Control(pDX, IDC_TREE1, m_tree);
751 DDX_Text(pDX, IDC_EDIT1, m_fn);
752 }
753
754 BOOL CConvertDlg::PreTranslateMessage(MSG* pMsg)
755 {
756 if(pMsg->message == WM_KEYDOWN && pMsg->wParam == VK_ESCAPE)
757 return TRUE;
758
759 return __super::PreTranslateMessage(pMsg);
760 }
761
762 BOOL CConvertDlg::OnInitDialog()
763 {
764 __super::OnInitDialog();
765
766 SetIcon(AfxGetApp()->LoadIcon(IDR_MAINFRAME), TRUE);
767 SetIcon(AfxGetApp()->LoadIcon(IDR_MAINFRAME), FALSE);
768
769 AddAnchor(IDC_TREE1, TOP_LEFT, BOTTOM_RIGHT);
770 AddAnchor(IDC_EDIT1, BOTTOM_LEFT, BOTTOM_RIGHT);
771 AddAnchor(IDC_BUTTON1, BOTTOM_RIGHT);
772 AddAnchor(IDC_HLINE, BOTTOM_LEFT, BOTTOM_RIGHT);
773 AddAnchor(IDC_BUTTON2, BOTTOM_CENTER);
774 AddAnchor(IDC_BUTTON3, BOTTOM_CENTER);
775 AddAnchor(IDC_BUTTON4, BOTTOM_CENTER);
776
777 CSize s(400, 200);
778 SetMinTrackSize(s);
779
780 m_streamtypesbm.LoadBitmap(IDB_STREAMTYPES);
781 m_streamtypes.Create(16, 18, ILC_MASK|ILC_COLOR32, 0, 4);
782 m_streamtypes.Add(&m_streamtypesbm, 0xffffff);
783 m_tree.SetImageList(&m_streamtypes, TVSIL_NORMAL);
784
785 GetWindowText(m_title);
786 m_nIDEventStatus = SetTimer(1, 1000, NULL);
787
788 HRESULT hr;
789 m_pMux = new CDSMMuxerFilter(NULL, &hr, false, false);
790
791 m_pGB = new CFGManagerMuxer(_T("CFGManagerMuxer"), NULL);
792 m_pGB->AddToROT();
793
794 if(FAILED(m_pGB->AddFilter(m_pMux, L"Mux"))
795 || !(m_pMC = m_pGB) || !(m_pME = m_pGB) || !(m_pMS = m_pMux)
796 || FAILED(m_pME->SetNotifyWindow((OAHWND)m_hWnd, WM_GRAPHNOTIFY, 0)))
797 {
798 MessageBeep(-1);
799 SendMessage(WM_CLOSE);
800 return TRUE;
801 }
802
803 return TRUE; // return TRUE unless you set the focus to a control
804 // EXCEPTION: OCX Property Pages should return FALSE
805 }
806
807 void CConvertDlg::OnOK()
808 {
809 }
810
811 BEGIN_MESSAGE_MAP(CConvertDlg, CResizableDialog)
812 ON_MESSAGE(WM_GRAPHNOTIFY, OnGraphNotify)
813 ON_WM_DROPFILES()
814 ON_WM_CLOSE()
815 ON_NOTIFY(NM_CLICK, IDC_TREE1, OnNMClickTree1)
816 ON_NOTIFY(NM_RCLICK, IDC_TREE1, OnNMRclickTree1)
817 ON_NOTIFY(NM_DBLCLK, IDC_TREE1, OnNMDblclkTree1)
818 ON_BN_CLICKED(IDC_BUTTON1, OnBnClickedButton1)
819 ON_UPDATE_COMMAND_UI(IDC_BUTTON1, OnUpdateButton1)
820 ON_WM_TIMER()
821 ON_BN_CLICKED(IDC_BUTTON2, OnBnClickedButton2)
822 ON_UPDATE_COMMAND_UI(IDC_BUTTON2, OnUpdateButton2)
823 ON_BN_CLICKED(IDC_BUTTON3, OnBnClickedButton3)
824 ON_UPDATE_COMMAND_UI(IDC_BUTTON3, OnUpdateButton3)
825 ON_BN_CLICKED(IDC_BUTTON4, OnBnClickedButton4)
826 ON_UPDATE_COMMAND_UI(IDC_BUTTON4, OnUpdateButton4)
827 END_MESSAGE_MAP()
828
829 // CConvertDlg message handlers
830
831 LRESULT CConvertDlg::OnGraphNotify(WPARAM wParam, LPARAM lParam)
832 {
833 HRESULT hr = S_OK;
834
835 LONG evCode, evParam1, evParam2;
836 while(m_pME && SUCCEEDED(m_pME->GetEvent(&evCode, (LONG_PTR*)&evParam1, (LONG_PTR*)&evParam2, 0)))
837 {
838 hr = m_pME->FreeEventParams(evCode, evParam1, evParam2);
839
840 bool fStop = false;
841
842 if(EC_COMPLETE == evCode)
843 {
844 fStop = true;
845 }
846 else if(EC_ERRORABORT == evCode)
847 {
848 fStop = true;
849
850 CString errmsg;
851 LPVOID lpMsgBuf;
852 if(FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER|FORMAT_MESSAGE_FROM_SYSTEM|FORMAT_MESSAGE_IGNORE_INSERTS,
853 NULL, hr, MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), (LPTSTR)&lpMsgBuf, 0, NULL))
854 {
855 errmsg = (LPCTSTR)lpMsgBuf;
856 LocalFree(lpMsgBuf);
857 }
858
859 CString str;
860 str.Format(_T("Could not complete conversion, the output file is most likely unusable.\n\nError code: 0x%08x"), evParam1);
861 if(!errmsg.IsEmpty()) str += _T(" (") + errmsg + _T(")");
862 AfxMessageBox(str, MB_OK);
863 }
864
865 if(fStop && m_pMC)
866 {
867 m_pMC->Stop();
868 m_tree.EnableWindow(TRUE);
869 }
870 }
871
872 return hr;
873 }
874
875 void CConvertDlg::OnDropFiles(HDROP hDropInfo)
876 {
877 for(int i = 0, j = DragQueryFile(hDropInfo, 0xffffffff, 0, 0); i < j; i++)
878 {
879 CString fn;
880 fn.ReleaseBufferSetLength(DragQueryFile(hDropInfo, i, fn.GetBuffer(MAX_PATH), MAX_PATH));
881
882 AddFile(fn);
883 }
884
885 __super::OnDropFiles(hDropInfo);
886 }
887
888 void CConvertDlg::OnClose()
889 {
890 HTREEITEM hTI = m_tree.GetRootItem();
891 while(hTI)
892 {
893 HTREEITEM hTINext = m_tree.GetNextSiblingItem(hTI);
894 DeleteItem(hTI);
895 hTI = hTINext;
896 }
897
898 m_pGB->RemoveFromROT();
899 m_pGB = NULL;
900
901 __super::OnClose();
902 }
903
904 void CConvertDlg::OnNMClickTree1(NMHDR* pNMHDR, LRESULT* pResult)
905 {
906 CPoint sp, cp;
907 HTREEITEM hTI = HitTest(sp, cp);
908 if(!hTI) return;
909 m_tree.SelectItem(hTI);
910
911 *pResult = 0;
912 }
913
914 void CConvertDlg::OnNMRclickTree1(NMHDR* pNMHDR, LRESULT* pResult)
915 {
916 CPoint sp, cp;
917 HTREEITEM hTI = HitTest(sp, cp);
918
919 if(hTI)
920 {
921 m_tree.SelectItem(hTI);
922
923 CTreeItem* t = (CTreeItem*)m_tree.GetItemData(hTI);
924
925 if(dynamic_cast<CTreeItemPin*>(t))
926 ShowPinPopup(hTI, sp);
927 else if(dynamic_cast<CTreeItemFile*>(t))
928 ShowFilePopup(hTI, sp);
929 else if(dynamic_cast<CTreeItemResourceFolder*>(t))
930 ShowResourceFolderPopup(hTI, sp);
931 else if(dynamic_cast<CTreeItemResource*>(t))
932 ShowResourcePopup(hTI, sp);
933 else if(dynamic_cast<CTreeItemChapterFolder*>(t))
934 ShowChapterFolderPopup(hTI, sp);
935 else if(dynamic_cast<CTreeItemChapter*>(t))
936 ShowChapterPopup(hTI, sp);
937 }
938 else
939 {
940 ShowPopup(sp);
941 }
942
943 *pResult = 0;
944 }
945
946 void CConvertDlg::OnNMDblclkTree1(NMHDR *pNMHDR, LRESULT *pResult)
947 {
948 CPoint sp, cp;
949 HTREEITEM hTI = HitTest(sp, cp);
950
951 if(hTI)
952 {
953 CTreeItem* t = (CTreeItem*)m_tree.GetItemData(hTI);
954
955 if(CTreeItemPin* t2 = dynamic_cast<CTreeItemPin*>(t))
956 {
957 CComPtr<IPin> pPinTo;
958 t2->m_pPin->ConnectedTo(&pPinTo);
959
960 if(CComQIPtr<IDSMPropertyBag> pPB = pPinTo)
961 EditProperties(pPB);
962 }
963 else if(CTreeItemResource* t2 = dynamic_cast<CTreeItemResource*>(t))
964 {
965 EditResource(t2);
966 }
967 else if(CTreeItemChapter* t2 = dynamic_cast<CTreeItemChapter*>(t))
968 {
969 EditChapter(t2);
970 }
971 }
972
973 *pResult = 0;
974 }
975
976 void CConvertDlg::OnBnClickedButton1()
977 {
978 UpdateData();
979
980 CFileDialog fd(FALSE, _T(".dsm"), m_fn,
981 OFN_EXPLORER|OFN_ENABLESIZING|OFN_HIDEREADONLY|OFN_OVERWRITEPROMPT,
982 _T("DirectShow Media file|*.dsm|All files|*.*|"), this, 0);
983
984 if(fd.DoModal() == IDOK)
985 {
986 m_fn = fd.GetPathName();
987 UpdateData(FALSE);
988 }
989 }
990
991 void CConvertDlg::OnUpdateButton1(CCmdUI* pCmdUI)
992 {
993 OAFilterState fs;
994 pCmdUI->Enable(m_pMC && SUCCEEDED(m_pMC->GetState(0, &fs)) && fs == State_Stopped);
995 }
996
997 void CConvertDlg::OnTimer(UINT nIDEvent)
998 {
999 if(nIDEvent == m_nIDEventStatus && m_pMS && m_pMC)
1000 {
1001 OAFilterState fs;
1002 if(SUCCEEDED(m_pMC->GetState(0, &fs)) && fs != State_Stopped)
1003 {
1004 GUID tf;
1005 m_pMS->GetTimeFormat(&tf);
1006
1007 REFERENCE_TIME rtCur, rtDur;
1008 HRESULT hr = m_pMS->GetDuration(&rtDur);
1009 m_pMS->GetCurrentPosition(&rtCur);
1010
1011 CString str;
1012 if(hr == S_OK && rtDur != 0) str.Format(_T("%.2f%%"), 1.0 * (rtCur * 100) / rtDur);
1013 else if(hr == S_OK && rtDur == 0) str = _T("Live");
1014 else if(tf == TIME_FORMAT_BYTE) str.Format(_T("%.2fKB"), 1.0 * rtCur / 1024);
1015 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);
1016 else str = _T("Please Wait");
1017
1018 SetWindowText(_T("Converting - ") + str);
1019 }
1020 else
1021 {
1022 SetWindowText(m_title);
1023 }
1024 }
1025
1026 __super::OnTimer(nIDEvent);
1027 }
1028
1029 void CConvertDlg::OnBnClickedButton2()
1030 {
1031 OAFilterState fs;
1032 if(FAILED(m_pMC->GetState(0, &fs)))
1033 return;
1034
1035 if(fs != State_Stopped)
1036 {
1037 m_pMC->Run();
1038 return;
1039 }
1040
1041 UpdateData();
1042
1043 if(!ConvertFile(m_fn))
1044 {
1045 AfxMessageBox(_T("Failed to start conversion"));
1046 }
1047 }
1048
1049 void CConvertDlg::OnUpdateButton2(CCmdUI* pCmdUI)
1050 {
1051 int nIn, nOut, nInC, nOutC;
1052 CountPins(m_pMux, nIn, nOut, nInC, nOutC);
1053
1054 OAFilterState fs;
1055 pCmdUI->Enable(nInC > 0 && GetDlgItem(IDC_EDIT1)->GetWindowTextLength() > 0
1056 && m_pMS && m_pMC && SUCCEEDED(m_pMC->GetState(0, &fs)) && fs != State_Running);
1057 }
1058
1059 void CConvertDlg::OnBnClickedButton3()
1060 {
1061 if(m_pMC) m_pMC->Pause();
1062 }
1063
1064 void CConvertDlg::OnUpdateButton3(CCmdUI* pCmdUI)
1065 {
1066 OAFilterState fs;
1067 pCmdUI->Enable(m_pMC && SUCCEEDED(m_pMC->GetState(0, &fs)) && fs == State_Running);
1068 }
1069
1070 void CConvertDlg::OnBnClickedButton4()
1071 {
1072 if(m_pMC) m_pMC->Stop();
1073 m_tree.EnableWindow(TRUE);
1074 }
1075
1076 void CConvertDlg::OnUpdateButton4(CCmdUI* pCmdUI)
1077 {
1078 OAFilterState fs;
1079 pCmdUI->Enable(m_pMC && SUCCEEDED(m_pMC->GetState(0, &fs)) && fs != State_Stopped);
1080 }
1081
1082 //
1083 // CFilterTreeCtrl
1084 //
1085
1086 CFilterTreeCtrl::CFilterTreeCtrl()
1087 {
1088 }
1089
1090 void CFilterTreeCtrl::PreSubclassWindow()
1091 {
1092 EnableToolTips(TRUE);
1093
1094 __super::PreSubclassWindow();
1095 }
1096
1097 INT_PTR CFilterTreeCtrl::OnToolHitTest(CPoint p, TOOLINFO* pTI) const
1098 {
1099 UINT nFlags;
1100 HTREEITEM hTI = HitTest(p, &nFlags);
1101 if(nFlags & TVHT_ONITEM)
1102 {
1103 CRect r;
1104 GetItemRect(hTI, r, TRUE);
1105 pTI->hwnd = m_hWnd;
1106 pTI->uId = (UINT)hTI;
1107 pTI->lpszText = LPSTR_TEXTCALLBACK;
1108 pTI->rect = r;
1109 return pTI->uId;
1110 }
1111
1112 return -1;
1113 }
1114
1115 BEGIN_MESSAGE_MAP(CFilterTreeCtrl, CTreeCtrl)
1116 ON_NOTIFY_EX_RANGE(TTN_NEEDTEXTW, 0, 0xFFFF, OnToolTipText)
1117 ON_NOTIFY_EX_RANGE(TTN_NEEDTEXTA, 0, 0xFFFF, OnToolTipText)
1118 END_MESSAGE_MAP()
1119
1120 BOOL CFilterTreeCtrl::OnToolTipText(UINT id, NMHDR* pNMHDR, LRESULT* pResult)
1121 {
1122 TOOLTIPTEXTA* pTTTA = (TOOLTIPTEXTA*)pNMHDR;
1123 TOOLTIPTEXTW* pTTTW = (TOOLTIPTEXTW*)pNMHDR;
1124
1125 UINT nID = pNMHDR->idFrom;
1126
1127 if(nID == (UINT)m_hWnd
1128 && (pNMHDR->code == TTN_NEEDTEXTA && (pTTTA->uFlags & TTF_IDISHWND)
1129 || pNMHDR->code == TTN_NEEDTEXTW && (pTTTW->uFlags & TTF_IDISHWND)))
1130 return FALSE;
1131
1132 ::SendMessage(pNMHDR->hwndFrom, TTM_SETMAXTIPWIDTH, 0, (LPARAM)(INT)1000);
1133
1134 HTREEITEM hTI = (HTREEITEM)nID;
1135
1136 CString str;
1137 static CStringA m_strTipTextA;
1138 static CStringW m_strTipTextW;
1139
1140 CConvertDlg::CTreeItem* t = (CConvertDlg::CTreeItem*)GetItemData(hTI);
1141 if(!t || !t->ToolTip(str)) return FALSE;
1142
1143 m_strTipTextA = str;
1144 m_strTipTextW = str;
1145
1146 if(pNMHDR->code == TTN_NEEDTEXTA) pTTTA->lpszText = (LPSTR)(LPCSTR)m_strTipTextA;
1147 else pTTTW->lpszText = (LPWSTR)(LPCWSTR)m_strTipTextW;
1148
1149 *pResult = 0;
1150
1151 return TRUE; // message was handled
1152 }
1153
1154 //
1155 // CConvertDlg::CTreeItem*
1156 //
1157
1158 CConvertDlg::CTreeItem::CTreeItem(CTreeCtrl& tree, HTREEITEM hTIParent)
1159 : m_tree(tree)
1160 {
1161 m_hTI = m_tree.InsertItem(_T(""), hTIParent);
1162 m_tree.SetItemData(m_hTI, (DWORD_PTR)this);
1163 Update();
1164 }
1165
1166 CConvertDlg::CTreeItem::~CTreeItem()
1167 {
1168 }
1169
1170 void CConvertDlg::CTreeItem::SetLabel(LPCTSTR label)
1171 {
1172 m_tree.SetItemText(m_hTI, label);
1173 }
1174
1175 void CConvertDlg::CTreeItem::SetImage(int nImage, int nSelectedImage)
1176 {
1177 m_tree.SetItemImage(m_hTI, nImage, nSelectedImage);
1178 }
1179
1180 //
1181
1182 CConvertDlg::CTreeItemFilter::CTreeItemFilter(IBaseFilter* pBF, CTreeCtrl& tree, HTREEITEM hTIParent)
1183 : CTreeItem(tree, hTIParent)
1184 , m_pBF(pBF)
1185 {
1186 Update();
1187 }
1188
1189 void CConvertDlg::CTreeItemFilter::Update()
1190 {
1191 SetLabel(CString(GetFilterName(m_pBF)));
1192 }
1193
1194 //
1195
1196 CConvertDlg::CTreeItemFile::CTreeItemFile(CString fn, IBaseFilter* pBF, CTreeCtrl& tree, HTREEITEM hTIParent)
1197 : CTreeItemFilter(pBF, tree, hTIParent)
1198 , m_fn(fn)
1199 {
1200 Update();
1201 }
1202
1203 void CConvertDlg::CTreeItemFile::Update()
1204 {
1205 CPath path = m_fn;
1206 path.StripPath();
1207 SetLabel(path);
1208 }
1209
1210 bool CConvertDlg::CTreeItemFile::ToolTip(CString& str)
1211 {
1212 str = m_fn;
1213 return true;
1214 }
1215
1216 //
1217
1218 CConvertDlg::CTreeItemPin::CTreeItemPin(IPin* pPin, CTreeCtrl& tree, HTREEITEM hTIParent)
1219 : CTreeItem(tree, hTIParent)
1220 , m_pPin(pPin)
1221 {
1222 Update();
1223 }
1224
1225 void CConvertDlg::CTreeItemPin::Update()
1226 {
1227 if(!m_pPin) {ASSERT(0); return;}
1228
1229 CString label = GetPinName(m_pPin);
1230 if(!IsConnected()) label = _T("[D] ") + label;
1231 SetLabel(label);
1232
1233 CMediaType mt;
1234 if(S_OK == m_pPin->ConnectionMediaType(&mt))
1235 {
1236 if(mt.majortype == MEDIATYPE_Video) SetImage(1, 1);
1237 else if(mt.majortype == MEDIATYPE_Audio) SetImage(2, 2);
1238 else if(mt.majortype == MEDIATYPE_Text || mt.majortype == MEDIATYPE_Subtitle) SetImage(3, 3);
1239 }
1240 }
1241
1242 bool CConvertDlg::CTreeItemPin::ToolTip(CString& str)
1243 {
1244 CMediaTypeEx mt;
1245 if(FAILED(m_pPin->ConnectionMediaType(&mt))) return false;
1246 str = mt.ToString(m_pPin);
1247 return true;
1248 }
1249
1250 bool CConvertDlg::CTreeItemPin::IsConnected()
1251 {
1252 CComPtr<IPin> pPinTo;
1253 return m_pPin && SUCCEEDED(m_pPin->ConnectedTo(&pPinTo)) && pPinTo;
1254 }
1255
1256 //
1257
1258 CConvertDlg::CTreeItemResourceFolder::CTreeItemResourceFolder(CTreeCtrl& tree, HTREEITEM hTIParent)
1259 : CTreeItem(tree, hTIParent)
1260 {
1261 Update();
1262 }
1263
1264 void CConvertDlg::CTreeItemResourceFolder::Update()
1265 {
1266 SetLabel(_T("Resources"));
1267 }
1268
1269 bool CConvertDlg::CTreeItemResourceFolder::ToolTip(CString& str)
1270 {
1271 if(!m_tree.ItemHasChildren(m_hTI))
1272 return false;
1273
1274 int files = 0;
1275 float size = 0;
1276
1277 HTREEITEM hChildItem = m_tree.GetChildItem(m_hTI);
1278
1279 while(hChildItem != NULL)
1280 {
1281 HTREEITEM hNextItem = m_tree.GetNextItem(hChildItem, TVGN_NEXT);
1282 if(CTreeItemResource* t = dynamic_cast<CTreeItemResource*>((CTreeItem*)m_tree.GetItemData(hChildItem)))
1283 size += t->m_res.data.GetCount(), files++;
1284 hChildItem = hNextItem;
1285 }
1286
1287 size /= 1024;
1288 if(size < 1024) str.Format(_T("%d file(s), %.2f KB"), files, size);
1289 else str.Format(_T("%d file(s), %.2f MB"), files, size/1024);
1290
1291 return true;
1292 }
1293
1294 //
1295
1296 CConvertDlg::CTreeItemResource::CTreeItemResource(const CDSMResource& res, CTreeCtrl& tree, HTREEITEM hTIParent)
1297 : CTreeItem(tree, hTIParent)
1298 {
1299 m_res = res;
1300 Update();
1301 }
1302
1303 CConvertDlg::CTreeItemResource::~CTreeItemResource()
1304 {
1305 }
1306
1307 void CConvertDlg::CTreeItemResource::Update()
1308 {
1309 SetLabel(CString(m_res.name));
1310
1311 CStringW mime = m_res.mime;
1312 mime.Trim();
1313 mime.MakeLower();
1314 if(mime == L"application/x-truetype-font") SetImage(4, 4);
1315 else if(mime.Find(L"text/") == 0) SetImage(5, 5);
1316 else SetImage(6, 6);
1317 }
1318
1319 bool CConvertDlg::CTreeItemResource::ToolTip(CString& str)
1320 {
1321 if(!m_res.mime.IsEmpty()) str = CString(m_res.mime) + _T("\r\n\r\n");
1322 if(!m_res.desc.IsEmpty()) str += CString(m_res.desc);
1323 str.Trim();
1324 return true;
1325 }
1326
1327 //
1328
1329 CConvertDlg::CTreeItemChapterFolder::CTreeItemChapterFolder(CTreeCtrl& tree, HTREEITEM hTIParent)
1330 : CTreeItem(tree, hTIParent)
1331 {
1332 Update();
1333 }
1334
1335 void CConvertDlg::CTreeItemChapterFolder::Update()
1336 {
1337 SetLabel(_T("Chapters"));
1338 }
1339
1340 //
1341
1342 CConvertDlg::CTreeItemChapter::CTreeItemChapter(const CDSMChapter& chap, CTreeCtrl& tree, HTREEITEM hTIParent)
1343 : CTreeItem(tree, hTIParent)
1344 {
1345 m_chap = chap;
1346 Update();
1347 }
1348
1349 void CConvertDlg::CTreeItemChapter::Update()
1350 {
1351 REFERENCE_TIME rt = m_chap.rt;
1352 rt /= 10000;
1353 int ms = (int)(rt%1000);
1354 rt /= 1000;
1355 int s = (int)(rt%60);
1356 rt /= 60;
1357 int m = (int)(rt%60);
1358 rt /= 60;
1359 int h = (int)(rt);
1360
1361 CString label;
1362 label.Format(_T("%02d:%02d:%02d.%03d - %s"), h, m, s, ms, CString(m_chap.name));
1363 SetLabel(label);
1364
1365 SetImage(7, 7);
1366 }
vsfilter mod based on vsfilter 2.39