/*
* $Id$
*
* (C) 2003-2006 Gabest
* (C) 2006-2010 see AUTHORS
*
* This file is part of mplayerc.
*
* Mplayerc is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 3 of the License, or
* (at your option) any later version.
*
* Mplayerc is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see .
*
*/
#include "stdafx.h"
#include "mplayerc.h"
#include "SaveDlg.h"
#include "../../filters/filters.h"
#include "internal_filter_config.h"
// CSaveDlg dialog
IMPLEMENT_DYNAMIC(CSaveDlg, CCmdUIDialog)
CSaveDlg::CSaveDlg(CString in, CString out, CWnd* pParent /*=NULL*/)
: CCmdUIDialog(CSaveDlg::IDD, pParent)
, m_in(in), m_out(out)
, m_nIDTimerEvent((UINT_PTR)-1)
{
}
CSaveDlg::~CSaveDlg()
{
}
void CSaveDlg::DoDataExchange(CDataExchange* pDX)
{
CCmdUIDialog::DoDataExchange(pDX);
DDX_Control(pDX, IDC_ANIMATE1, m_anim);
DDX_Control(pDX, IDC_PROGRESS1, m_progress);
DDX_Control(pDX, IDC_REPORT, m_report);
DDX_Control(pDX, IDC_FROMTO, m_fromto);
}
BEGIN_MESSAGE_MAP(CSaveDlg, CCmdUIDialog)
ON_BN_CLICKED(IDCANCEL, OnBnClickedCancel)
ON_MESSAGE(WM_GRAPHNOTIFY, OnGraphNotify)
ON_WM_TIMER()
END_MESSAGE_MAP()
// CSaveDlg message handlers
BOOL CSaveDlg::OnInitDialog()
{
CCmdUIDialog::OnInitDialog();
m_anim.Open(IDR_AVI_FILECOPY);
m_anim.Play(0, (UINT)-1, (UINT)-1);
CString str, in = m_in, out = m_out;
if(in.GetLength() > 60) in = in.Left(17) + _T("..") + in.Right(43);
if(out.GetLength() > 60) out = out.Left(17) + _T("..") + out.Right(43);
str.Format(_T("%s\r\n%s"), in, out);
m_fromto.SetWindowText(str);
m_progress.SetRange(0, 100);
if(FAILED(pGB.CoCreateInstance(CLSID_FilterGraph)) || !(pMC = pGB) || !(pME = pGB) || !(pMS = pGB)
|| FAILED(pME->SetNotifyWindow((OAHWND)m_hWnd, WM_GRAPHNOTIFY, 0)))
{
m_report.SetWindowText(_T("Error"));
return FALSE;
}
HRESULT hr;
CStringW fnw = m_in;
CComPtr pReader;
#if INTERNAL_SOURCEFILTER_CDDA
if(!pReader && m_in.Mid(m_in.ReverseFind('.')+1).MakeLower() == _T("cda"))
{
hr = S_OK;
CComPtr pUnk = (IUnknown*)(INonDelegatingUnknown*)DNew CCDDAReader(NULL, &hr);
if(FAILED(hr) || !(pReader = pUnk) || FAILED(pReader->Load(fnw, NULL)))
pReader.Release();
}
#endif
#if INTERNAL_SOURCEFILTER_CDXA
if(!pReader)
{
hr = S_OK;
CComPtr pUnk = (IUnknown*)(INonDelegatingUnknown*)DNew CCDXAReader(NULL, &hr);
if(FAILED(hr) || !(pReader = pUnk) || FAILED(pReader->Load(fnw, NULL)))
pReader.Release();
}
#endif
#if INTERNAL_SOURCEFILTER_VTS
if(!pReader /*&& ext == _T("ifo")*/)
{
hr = S_OK;
CComPtr pUnk = (IUnknown*)(INonDelegatingUnknown*)DNew CVTSReader(NULL, &hr);
if(FAILED(hr) || !(pReader = pUnk) || FAILED(pReader->Load(fnw, NULL)))
pReader.Release();
else
{
CPath pout(m_out);
pout.RenameExtension(_T(".ifo"));
CopyFile(m_in, pout, FALSE);
}
}
#endif
if(!pReader)
{
hr = S_OK;
CComPtr pUnk;
pUnk.CoCreateInstance(CLSID_AsyncReader);
if(FAILED(hr) || !(pReader = pUnk) || FAILED(pReader->Load(fnw, NULL)))
pReader.Release();
}
if(!pReader)
{
hr = S_OK;
CComPtr pUnk;
pUnk.CoCreateInstance(CLSID_URLReader);
if(CComQIPtr pSrc = pUnk) // url reader has to be in the graph to load the file
{
pGB->AddFilter(pSrc, fnw);
if(FAILED(hr) || !(pReader = pUnk) || FAILED(hr = pReader->Load(fnw, NULL)))
{
pReader.Release();
pGB->RemoveFilter(pSrc);
}
}
}
CComQIPtr pSrc = pReader;
if(FAILED(pGB->AddFilter(pSrc, fnw)))
{
m_report.SetWindowText(_T("Sorry, can't save this file, press cancel"));
return FALSE;
}
CComQIPtr pMid = DNew CStreamDriveThruFilter(NULL, &hr);
if(FAILED(pGB->AddFilter(pMid, L"StreamDriveThru")))
{
m_report.SetWindowText(_T("Error"));
return FALSE;
}
CComQIPtr pDst;
pDst.CoCreateInstance(CLSID_FileWriter);
CComQIPtr pFSF = pDst;
pFSF->SetFileName(CStringW(m_out), NULL);
pFSF->SetMode(AM_FILE_OVERWRITE);
if(FAILED(pGB->AddFilter(pDst, L"File Writer")))
{
m_report.SetWindowText(_T("Error"));
return FALSE;
}
hr = pGB->Connect(
GetFirstPin((pSrc), PINDIR_OUTPUT),
GetFirstPin((pMid), PINDIR_INPUT));
hr = pGB->Connect(
GetFirstPin((pMid), PINDIR_OUTPUT),
GetFirstPin((pDst), PINDIR_INPUT));
pMS = pMid;
pMC->Run();
m_nIDTimerEvent = SetTimer(1, 1000, NULL);
return TRUE; // return TRUE unless you set the focus to a control
// EXCEPTION: OCX Property Pages should return FALSE
}
void CSaveDlg::OnBnClickedCancel()
{
if(pMC) pMC->Stop();
OnCancel();
}
LRESULT CSaveDlg::OnGraphNotify(WPARAM wParam, LPARAM lParam)
{
LONG evCode, evParam1, evParam2;
while(pME && SUCCEEDED(pME->GetEvent(&evCode, (LONG_PTR*)&evParam1, (LONG_PTR*)&evParam2, 0)))
{
HRESULT hr = pME->FreeEventParams(evCode, evParam1, evParam2);
UNUSED_ALWAYS(hr);
if(EC_COMPLETE == evCode)
{
EndDialog(IDOK);
}
else if(EC_ERRORABORT == evCode)
{
TRACE(_T("CSaveDlg::OnGraphNotify / EC_ERRORABORT, hr = %08x\n"), (HRESULT)evParam1);
m_report.SetWindowText(_T("Copying unexpectedly terminated!"));
}
}
return 0;
}
void CSaveDlg::OnTimer(UINT_PTR nIDEvent)
{
if(nIDEvent == m_nIDTimerEvent && pGB)
{
if(pMS)
{
CString str;
REFERENCE_TIME pos = 0, dur = 0;
pMS->GetCurrentPosition(&pos);
pMS->GetDuration(&dur);
REFERENCE_TIME time = 0;
CComQIPtr(pGB)->GetCurrentPosition(&time);
REFERENCE_TIME speed = time > 0 ? pos*10000000/time / 1024 : 0i64;
str.Format(_T("%I64d/%I64d KB, %I64d KB/s, %I64d s"),
pos/1024, dur/1024, speed, speed > 0 ? (dur-pos)/1024 / speed : 0);
m_report.SetWindowText(str);
m_progress.SetPos(dur > 0 ? (int)(100*pos/dur) : 0);
}
}
CCmdUIDialog::OnTimer(nIDEvent);
}