src/apps/mplayerc/ConvertDlg.cpp

   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 }