src/Dialog_CueImport.cpp

   1 ///////////////////////////////////////////////////////////////////////////////
   2 // LameXP - Audio Encoder Front-End
   3 // Copyright (C) 2004-2011 LoRd_MuldeR <MuldeR2@GMX.de>
   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 of the License, or
   8 // (at your option) 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 along
  16 // with this program; if not, write to the Free Software Foundation, Inc.,
  17 // 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  18 //
  19 // http://www.gnu.org/licenses/gpl-2.0.txt
  20 ///////////////////////////////////////////////////////////////////////////////
  21
  22 #include "Dialog_CueImport.h"
  23
  24 #include "Global.h"
  25 #include "Model_CueSheet.h"
  26 #include "Model_AudioFile.h"
  27 #include "Model_FileList.h"
  28 #include "Dialog_WorkingBanner.h"
  29 #include "Thread_FileAnalyzer.h"
  30 #include "Thread_CueSplitter.h"
  31 #include "LockedFile.h"
  32
  33 #include <QFileInfo>
  34 #include <QMessageBox>
  35 #include <QTimer>
  36 #include <QFileDialog>
  37 #include <QProgressDialog>
  38 #include <QMenu>
  39
  40 #define SET_FONT_BOLD(WIDGET,BOLD) { QFont _font = WIDGET->font(); _font.setBold(BOLD); WIDGET->setFont(_font); }
  41
  42 ////////////////////////////////////////////////////////////
  43 // Constructor & Destructor
  44 ////////////////////////////////////////////////////////////
  45
  46 CueImportDialog::CueImportDialog(QWidget *parent, FileListModel *fileList, const QString &cueFile)
  47 :
  48         QDialog(parent),
  49         m_cueFileName(cueFile),
  50         m_fileList(fileList)
  51 {
  52         //Init the dialog, from the .ui file
  53         setupUi(this);
  54
  55         //Fix size
  56         setMinimumSize(this->size());
  57         setMaximumHeight(this->height());
  58
  59         //Create model
  60         m_model = new CueSheetModel();
  61         connect(m_model, SIGNAL(modelReset()), this, SLOT(modelChanged()));
  62
  63         //Setup table view
  64         treeView->setModel(m_model);
  65         treeView->header()->setStretchLastSection(false);
  66         treeView->header()->setResizeMode(QHeaderView::ResizeToContents);
  67         treeView->header()->setResizeMode(1, QHeaderView::Stretch);
  68         treeView->header()->setMovable(false);
  69         treeView->setItemsExpandable(false);
  70
  71         //Enable up/down button
  72         connect(imprtButton, SIGNAL(clicked()), this, SLOT(importButtonClicked()));
  73         connect(browseButton, SIGNAL(clicked()), this, SLOT(browseButtonClicked()));
  74
  75         //Translate
  76         labelHeaderText->setText(QString("<b>%1</b><br>%2").arg(tr("Import Cue Sheet"), tr("The following Cue Sheet will be split and imported into LameXP.")));
  77 }
  78
  79 CueImportDialog::~CueImportDialog(void)
  80 {
  81         LAMEXP_DELETE(m_model);
  82 }
  83
  84 ////////////////////////////////////////////////////////////
  85 // EVENTS
  86 ////////////////////////////////////////////////////////////
  87
  88 void CueImportDialog::showEvent(QShowEvent *event)
  89 {
  90         QDialog::showEvent(event);
  91         modelChanged();
  92 }
  93
  94 ////////////////////////////////////////////////////////////
  95 // Slots
  96 ////////////////////////////////////////////////////////////
  97
  98 int CueImportDialog::exec(void)
  99 {
 100         WorkingBanner *progress = new WorkingBanner(dynamic_cast<QWidget*>(parent()));
 101         progress->show(tr("Loading Cue Sheet file, please be patient..."));
 102
 103         QFileInfo cueFileInfo(m_cueFileName);
 104         if(!cueFileInfo.exists() || !cueFileInfo.isFile())
 105         {
 106                 QString text = QString("<nobr>%1</nobr><br><nobr>%2</nobr><br><br><nobr>%3</nobr>").arg(tr("Failed to load the Cue Sheet file:"), QDir::toNativeSeparators(m_cueFileName), tr("The specified file could not be found!")).replace("-", "&minus;");
 107                 QMessageBox::warning(progress, tr("Cue Sheet Error"), text);
 108                 progress->close();
 109                 LAMEXP_DELETE(progress);
 110                 return CueSheetModel::ErrorIOFailure;
 111         }
 112
 113         m_outputDir = QString("%1/%2").arg(cueFileInfo.canonicalPath(), cueFileInfo.completeBaseName());
 114         for(int n = 2; QDir(m_outputDir).exists(); n++)
 115         {
 116                 m_outputDir = QString("%1/%2 (%3)").arg(cueFileInfo.canonicalPath(), cueFileInfo.completeBaseName(), QString::number(n));
 117         }
 118
 119         setWindowTitle(QString("%1: %2").arg(windowTitle().split(":", QString::SkipEmptyParts).first().trimmed(), cueFileInfo.fileName()));
 120
 121         int iResult = m_model->loadCueSheet(m_cueFileName, QApplication::instance());
 122         if(iResult != CueSheetModel::ErrorSuccess)
 123         {
 124                 QString errorMsg = tr("An unknown error has occured!");
 125
 126                 switch(iResult)
 127                 {
 128                 case CueSheetModel::ErrorIOFailure:
 129                         errorMsg = tr("The file could not be opened for reading. Make sure you have the required rights!");
 130                         break;
 131                 case CueSheetModel::ErrorBadFile:
 132                         errorMsg = tr("The provided file does not look like a valid Cue Sheet disc image file!");
 133                         break;
 134                 case CueSheetModel::ErrorUnsupported:
 135                         errorMsg = QString("%1<br>%2").arg(tr("Could not find any supported audio track in the Cue Sheet image!"), tr("Note that LameXP can not handle \"binary\" Cue Sheet images."));
 136                         break;
 137                 case CueSheetModel::ErrorInconsistent:
 138                         errorMsg = tr("The selected Cue Sheet file contains inconsistent information. Take care!");
 139                         break;
 140                 }
 141
 142                 QString text = QString("<nobr>%1</nobr><br><nobr>%2</nobr><br><br><nobr>%3</nobr>").arg(tr("Failed to load the Cue Sheet file:"), QDir::toNativeSeparators(m_cueFileName), errorMsg).replace("-", "&minus;");
 143                 QMessageBox::warning(progress, tr("Cue Sheet Error"), text);
 144                 progress->close();
 145                 LAMEXP_DELETE(progress);
 146                 return iResult;
 147         }
 148
 149         progress->close();
 150         LAMEXP_DELETE(progress);
 151         return QDialog::exec();
 152 }
 153
 154 void CueImportDialog::modelChanged(void)
 155 {
 156         treeView->expandAll();
 157         editOutputDir->setText(QDir::toNativeSeparators(m_outputDir));
 158 }
 159
 160 void CueImportDialog::browseButtonClicked(void)
 161 {
 162         QString newOutDir, currentDir = m_outputDir;
 163
 164         while(QDir(currentDir).exists())
 165         {
 166                 int pos = max(currentDir.lastIndexOf(QChar('\\')), currentDir.lastIndexOf(QChar('/')));
 167                 if(pos > 0) currentDir.left(pos - 1); else break;
 168         }
 169
 170         if(lamexp_themes_enabled() || ((QSysInfo::windowsVersion() & QSysInfo::WV_NT_based) < QSysInfo::WV_XP))
 171         {
 172                 newOutDir = QFileDialog::getExistingDirectory(this, tr("Choose Output Directory"), currentDir);
 173         }
 174         else
 175         {
 176                 QFileDialog dialog(this, tr("Choose Output Directory"));
 177                 dialog.setFileMode(QFileDialog::DirectoryOnly);
 178                 dialog.setDirectory(currentDir);
 179                 if(dialog.exec())
 180                 {
 181                         newOutDir = dialog.selectedFiles().first();
 182                 }
 183         }
 184
 185         if(!newOutDir.isEmpty())
 186         {
 187                 m_outputDir = newOutDir;
 188                 modelChanged();
 189         }
 190 }
 191
 192 void CueImportDialog::importButtonClicked(void)
 193 {
 194         static const __int64 oneGigabyte = 1073741824i64;
 195         static const __int64 minimumFreeDiskspaceMultiplier = 2i64;
 196         static const char *writeTestBuffer = "LAMEXP_WRITE_TEST";
 197
 198         QDir outputDir(m_outputDir);
 199         outputDir.mkpath(".");
 200         if(!(outputDir.exists() && outputDir.isReadable()))
 201         {
 202                 QMessageBox::warning(this, tr("LameXP"), QString("<nobr>%2</nobr>").arg(tr("Error: The selected output directory could not be created!")));
 203                 return;
 204         }
 205
 206         QFile writeTest(QString("%1/~%2.txt").arg(m_outputDir, lamexp_rand_str()));
 207         if(!(writeTest.open(QIODevice::ReadWrite) && (writeTest.write(writeTestBuffer) == strlen(writeTestBuffer))))
 208         {
 209                 QMessageBox::warning(this, tr("LameXP"), QString("<nobr>%2</nobr>").arg(tr("Error: The selected output directory is not writable!")));
 210                 return;
 211         }
 212         else
 213         {
 214                 writeTest.close();
 215                 writeTest.remove();
 216         }
 217
 218         qint64 currentFreeDiskspace = lamexp_free_diskspace(m_outputDir);
 219         if(currentFreeDiskspace < (oneGigabyte * minimumFreeDiskspaceMultiplier))
 220         {
 221                 QMessageBox::warning(this, tr("Low Diskspace Warning"), QString("<nobr>%1</nobr><br><nobr>%2</nobr>").arg(tr("There are less than %1 GB of free diskspace available in the selected output directory.").arg(QString::number(minimumFreeDiskspaceMultiplier)), tr("It is highly recommend to free up more diskspace before proceeding with the import!")));
 222                 return;
 223         }
 224
 225         importCueSheet();
 226         accept();
 227 }
 228
 229 void CueImportDialog::analyzedFile(const AudioFileModel &file)
 230 {
 231         qDebug("Received result: <%s> <%s/%s>", file.filePath().toLatin1().constData(), file.formatContainerType().toLatin1().constData(), file.formatAudioType().toLatin1().constData());
 232         m_fileInfo << file;
 233 }
 234
 235 ////////////////////////////////////////////////////////////
 236 // Private Functions
 237 ////////////////////////////////////////////////////////////
 238
 239 void CueImportDialog::importCueSheet(void)
 240 {
 241         QStringList files;
 242
 243         //Fetch all files that are referenced in the Cue Sheet and lock them
 244         int nFiles = m_model->getFileCount();
 245         for(int i = 0; i < nFiles; i++)
 246         {
 247                 QString temp = m_model->getFileName(i);
 248                 try
 249                 {
 250                         m_locks << new LockedFile(temp);
 251                 }
 252                 catch(char *err)
 253                 {
 254                         qWarning("Failed to lock file: %s", err);
 255                         continue;
 256                 }
 257                 files << temp;
 258         }
 259
 260         //Analyze all source files first
 261         if(analyzeFiles(files))
 262         {
 263                 //Now split files according to Cue Sheet
 264                 splitFiles();
 265         }
 266
 267
 268         //Release locks
 269         while(!m_locks.isEmpty())
 270         {
 271                 delete m_locks.takeFirst();
 272         }
 273 }
 274
 275 bool CueImportDialog::analyzeFiles(QStringList &files)
 276 {
 277         m_fileInfo.clear();
 278         bool bSuccess = true;
 279
 280         WorkingBanner *progress = new WorkingBanner(this);
 281         FileAnalyzer *analyzer = new FileAnalyzer(files);
 282
 283         connect(analyzer, SIGNAL(fileSelected(QString)), progress, SLOT(setText(QString)), Qt::QueuedConnection);
 284         connect(analyzer, SIGNAL(fileAnalyzed(AudioFileModel)), this, SLOT(analyzedFile(AudioFileModel)), Qt::QueuedConnection);
 285         connect(progress, SIGNAL(userAbort()), analyzer, SLOT(abortProcess()), Qt::DirectConnection);
 286
 287         progress->show(tr("Analyzing file(s), please wait..."), analyzer);
 288         progress->close();
 289
 290         if(analyzer->filesAccepted() < static_cast<unsigned int>(files.count()))
 291         {
 292                 if(QMessageBox::warning(this, tr("Analysis Failed"), tr("Warning: The format of some of the input files could not be determined!"), tr("Continue Anyway"), tr("Abort")) == 1)
 293                 {
 294                         bSuccess = false;
 295                 }
 296         }
 297
 298         LAMEXP_DELETE(progress);
 299         LAMEXP_DELETE(analyzer);
 300
 301         return bSuccess;
 302 }
 303
 304 void CueImportDialog::splitFiles(void)
 305 {
 306         QString baseName = QFileInfo(m_cueFileName).completeBaseName().replace(".", " ").simplified();
 307
 308         WorkingBanner *progress = new WorkingBanner(this);
 309         CueSplitter *splitter  = new CueSplitter(m_outputDir, baseName, m_model, m_fileInfo);
 310
 311         connect(splitter, SIGNAL(fileSelected(QString)), progress, SLOT(setText(QString)), Qt::QueuedConnection);
 312         connect(splitter, SIGNAL(fileSplit(AudioFileModel)), m_fileList, SLOT(addFile(AudioFileModel)), Qt::QueuedConnection);
 313         connect(progress, SIGNAL(userAbort()), splitter, SLOT(abortProcess()), Qt::DirectConnection);
 314
 315         progress->show(tr("Splitting file(s), please wait..."), splitter);
 316         progress->close();
 317
 318         if(splitter->getAborted())
 319         {
 320                 QMessageBox::warning(this, tr("Cue Sheet Error"), tr("Process was aborted by the user after %1 track(s)!").arg(QString::number(splitter->getTracksSuccess())));
 321         }
 322         else if(!splitter->getSuccess())
 323         {
 324                 QMessageBox::warning(this, tr("Cue Sheet Error"), tr("An unexpected error has occured while splitting the Cue Sheet!"));
 325         }
 326         else
 327         {
 328                 QString text = QString("<nobr>%1</nobr>").arg(tr("Imported %1 track(s) from the Cue Sheet and skipped %2 track(s).").arg(QString::number(splitter->getTracksSuccess()), QString::number(splitter->getTracksSkipped() /*+ nTracksSkipped*/)));
 329                 QMessageBox::information(this, tr("Cue Sheet Completed"), text);
 330         }
 331
 332         LAMEXP_DELETE(splitter);
 333         LAMEXP_DELETE(progress);
 334 }