OSDN Git Service

Bum version.
[lamexp/LameXP.git] / src / Dialog_CueImport.cpp
index 3a2c0fc..4f29c70 100644 (file)
@@ -1,6 +1,6 @@
 ///////////////////////////////////////////////////////////////////////////////
 // LameXP - Audio Encoder Front-End
-// Copyright (C) 2004-2011 LoRd_MuldeR <MuldeR2@GMX.de>
+// Copyright (C) 2004-2012 LoRd_MuldeR <MuldeR2@GMX.de>
 //
 // This program is free software; you can redistribute it and/or modify
 // it under the terms of the GNU General Public License as published by
 #include "Global.h"
 #include "Model_CueSheet.h"
 #include "Model_AudioFile.h"
+#include "Model_FileList.h"
 #include "Dialog_WorkingBanner.h"
 #include "Thread_FileAnalyzer.h"
+#include "Thread_CueSplitter.h"
+#include "LockedFile.h"
 
 #include <QFileInfo>
 #include <QMessageBox>
 #include <QFileDialog>
 #include <QProgressDialog>
 #include <QMenu>
+#include <QTextCodec>
+#include <QInputDialog>
 
 #define SET_FONT_BOLD(WIDGET,BOLD) { QFont _font = WIDGET->font(); _font.setBold(BOLD); WIDGET->setFont(_font); }
+#define EXPAND(STR) QString(STR).leftJustified(96, ' ')
 
 ////////////////////////////////////////////////////////////
 // Constructor & Destructor
 ////////////////////////////////////////////////////////////
 
-CueImportDialog::CueImportDialog(QWidget *parent)
+CueImportDialog::CueImportDialog(QWidget *parent, FileListModel *fileList, const QString &cueFile)
 :
-       QDialog(parent)
+       QDialog(parent),
+       m_cueFileName(cueFile),
+       m_fileList(fileList)
 {
        //Init the dialog, from the .ui file
        setupUi(this);
@@ -66,6 +74,7 @@ CueImportDialog::CueImportDialog(QWidget *parent)
        //Enable up/down button
        connect(imprtButton, SIGNAL(clicked()), this, SLOT(importButtonClicked()));
        connect(browseButton, SIGNAL(clicked()), this, SLOT(browseButtonClicked()));
+       connect(loadOtherButton, SIGNAL(clicked()), this, SLOT(loadOtherButtonClicked()));
 
        //Translate
        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.")));
@@ -90,26 +99,95 @@ void CueImportDialog::showEvent(QShowEvent *event)
 // Slots
 ////////////////////////////////////////////////////////////
 
-int CueImportDialog::exec(const QString &cueFile)
+int CueImportDialog::exec(void)
 {
        WorkingBanner *progress = new WorkingBanner(dynamic_cast<QWidget*>(parent()));
        progress->show(tr("Loading Cue Sheet file, please be patient..."));
-       
-       QFileInfo cueFileInfo(cueFile);
-       m_outputDir = QFileInfo(cueFile).canonicalPath();
-
-       setWindowTitle(QString("%1: %2").arg(windowTitle().split(":", QString::SkipEmptyParts).first().trimmed(), cueFileInfo.fileName()));
 
-       if(!cueFileInfo.exists() || !cueFileInfo.isFile() || m_outputDir.isEmpty())
+       QFileInfo cueFileInfo(m_cueFileName);
+       if(!cueFileInfo.exists() || !cueFileInfo.isFile())
        {
-               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(cueFile), tr("The specified file could not be found!")).replace("-", "&minus;");
+               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;");
                QMessageBox::warning(progress, tr("Cue Sheet Error"), text);
                progress->close();
                LAMEXP_DELETE(progress);
                return CueSheetModel::ErrorIOFailure;
        }
 
-       int iResult = m_model->loadCueSheet(cueFile, QApplication::instance());
+       //----------------------//
+
+       QTextCodec *codec = NULL;
+
+       QFile cueFile(cueFileInfo.canonicalFilePath());
+       cueFile.open(QIODevice::ReadOnly);
+       QByteArray bomCheck = cueFile.isOpen() ? cueFile.peek(16) : QByteArray();
+
+       if((!bomCheck.isEmpty()) && bomCheck.startsWith("\xef\xbb\xbf"))
+       {
+               codec = QTextCodec::codecForName("UTF-8");
+       }
+       else if((!bomCheck.isEmpty()) && bomCheck.startsWith("\xff\xfe"))
+       {
+               codec = QTextCodec::codecForName("UTF-16LE");
+       }
+       else if((!bomCheck.isEmpty()) && bomCheck.startsWith("\xfe\xff"))
+       {
+               codec = QTextCodec::codecForName("UTF-16BE");
+       }
+       else
+       {
+               const QString systemDefault = tr("(System Default)");
+
+               QStringList codecList;
+               codecList.append(systemDefault);
+               codecList.append(lamexp_available_codepages());
+
+               QInputDialog *input = new QInputDialog(progress);
+               input->setLabelText(EXPAND(tr("Select ANSI Codepage for Cue Sheet file:")));
+               input->setOkButtonText(tr("OK"));
+               input->setCancelButtonText(tr("Cancel"));
+               input->setTextEchoMode(QLineEdit::Normal);
+               input->setComboBoxItems(codecList);
+       
+               if(input->exec() < 1)
+               {
+                       progress->close();
+                       LAMEXP_DELETE(input);
+                       LAMEXP_DELETE(progress);
+                       return Rejected;
+               }
+       
+               if(input->textValue().compare(systemDefault, Qt::CaseInsensitive))
+               {
+                       qDebug("User-selected codec is: %s", input->textValue().toLatin1().constData());
+                       codec = QTextCodec::codecForName(input->textValue().toLatin1().constData());
+               }
+               else
+               {
+                       qDebug("Going to use the system's default codec!");
+                       codec = QTextCodec::codecForName("System");
+               }
+
+               LAMEXP_DELETE(input);
+       }
+
+       bomCheck.clear();
+
+       //----------------------//
+
+       QString baseName = cueFileInfo.completeBaseName().simplified();
+       while(baseName.endsWith(".") || baseName.endsWith(" ")) baseName.chop(1);
+       if(baseName.isEmpty()) baseName = tr("New Folder");
+
+       m_outputDir = QString("%1/%2").arg(cueFileInfo.canonicalPath(), baseName);
+       for(int n = 2; QDir(m_outputDir).exists() || QFileInfo(m_outputDir).exists(); n++)
+       {
+               m_outputDir = QString("%1/%2 (%3)").arg(cueFileInfo.canonicalPath(), baseName, QString::number(n));
+       }
+
+       setWindowTitle(QString("%1: %2").arg(windowTitle().split(":", QString::SkipEmptyParts).first().trimmed(), cueFileInfo.fileName()));
+
+       int iResult = m_model->loadCueSheet(m_cueFileName, QApplication::instance(), codec);
        if(iResult != CueSheetModel::ErrorSuccess)
        {
                QString errorMsg = tr("An unknown error has occured!");
@@ -130,7 +208,7 @@ int CueImportDialog::exec(const QString &cueFile)
                        break;
                }
                
-               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(cueFile), errorMsg).replace("-", "&minus;");
+               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;");
                QMessageBox::warning(progress, tr("Cue Sheet Error"), text);
                progress->close();
                LAMEXP_DELETE(progress);
@@ -146,11 +224,35 @@ void CueImportDialog::modelChanged(void)
 {
        treeView->expandAll();
        editOutputDir->setText(QDir::toNativeSeparators(m_outputDir));
+       labelArtist->setText(m_model->getAlbumPerformer().isEmpty() ? tr("Unknown Artist") : m_model->getAlbumPerformer());
+       labelAlbum->setText(m_model->getAlbumTitle().isEmpty() ? tr("Unknown Album") : m_model->getAlbumTitle());
 }
 
 void CueImportDialog::browseButtonClicked(void)
 {
-       QString newOutDir = QFileDialog::getExistingDirectory(this, tr("Choose Output Directory"));
+       QString newOutDir, currentDir = m_outputDir;
+       
+       while(QDir(currentDir).exists())
+       {
+               int pos = qMax(currentDir.lastIndexOf(QChar('\\')), currentDir.lastIndexOf(QChar('/')));
+               if(pos > 0) currentDir.left(pos - 1); else break;
+       }
+
+       if(lamexp_themes_enabled() || ((QSysInfo::windowsVersion() & QSysInfo::WV_NT_based) < QSysInfo::WV_XP))
+       {
+               newOutDir = QFileDialog::getExistingDirectory(this, tr("Choose Output Directory"), currentDir);
+       }
+       else
+       {
+               QFileDialog dialog(this, tr("Choose Output Directory"));
+               dialog.setFileMode(QFileDialog::DirectoryOnly);
+               dialog.setDirectory(currentDir);
+               if(dialog.exec())
+               {
+                       newOutDir = dialog.selectedFiles().first();
+               }
+       }
+
        if(!newOutDir.isEmpty())
        {
                m_outputDir = newOutDir;
@@ -160,10 +262,18 @@ void CueImportDialog::browseButtonClicked(void)
 
 void CueImportDialog::importButtonClicked(void)
 {
-       static const __int64 oneGigabyte = 1073741824i64; 
-       static const __int64 minimumFreeDiskspaceMultiplier = 2i64;
+       static const unsigned __int64 oneGigabyte = 1073741824ui64; 
+       static const unsigned __int64 minimumFreeDiskspaceMultiplier = 2ui64;
        static const char *writeTestBuffer = "LAMEXP_WRITE_TEST";
        
+       QDir outputDir(m_outputDir);
+       outputDir.mkpath(".");
+       if(!(outputDir.exists() && outputDir.isReadable()))
+       {
+               QMessageBox::warning(this, tr("LameXP"), QString("<nobr>%2</nobr>").arg(tr("Error: The selected output directory could not be created!")));
+               return;
+       }
+
        QFile writeTest(QString("%1/~%2.txt").arg(m_outputDir, lamexp_rand_str()));
        if(!(writeTest.open(QIODevice::ReadWrite) && (writeTest.write(writeTestBuffer) == strlen(writeTestBuffer))))
        {
@@ -176,51 +286,128 @@ void CueImportDialog::importButtonClicked(void)
                writeTest.remove();
        }
 
-       qint64 currentFreeDiskspace = lamexp_free_diskspace(m_outputDir);
-       if(currentFreeDiskspace < (oneGigabyte * minimumFreeDiskspaceMultiplier))
+       bool ok = false;
+       unsigned __int64 currentFreeDiskspace = lamexp_free_diskspace(m_outputDir, &ok);
+
+       if(ok && (currentFreeDiskspace < (oneGigabyte * minimumFreeDiskspaceMultiplier)))
        {
                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!")));
                return;
        }
 
        importCueSheet();
+       accept();
+}
+
+void CueImportDialog::loadOtherButtonClicked(void)
+{
+       done(-1);
 }
 
 void CueImportDialog::analyzedFile(const AudioFileModel &file)
 {
-       qWarning("Received results for: %s", file.filePath().toLatin1().constData());
-       m_fileInfo.insert(file.filePath(), file);
+       qDebug("Received result: <%s> <%s/%s>", file.filePath().toLatin1().constData(), file.formatContainerType().toLatin1().constData(), file.formatAudioType().toLatin1().constData());
+       m_fileInfo << file;
 }
 
 ////////////////////////////////////////////////////////////
-// Private FUnctions
+// Private Functions
 ////////////////////////////////////////////////////////////
 
 void CueImportDialog::importCueSheet(void)
 {
        QStringList files;
-       int nFiles = m_model->getFileCount();
 
-       //Fetch all files that are referenced in the Cue Sheet
+       //Fetch all files that are referenced in the Cue Sheet and lock them
+       int nFiles = m_model->getFileCount();
        for(int i = 0; i < nFiles; i++)
        {
-               files << m_model->getFileName(i);
+               QString temp = m_model->getFileName(i);
+               try
+               {
+                       m_locks << new LockedFile(temp);
+               }
+               catch(char *err)
+               {
+                       qWarning("Failed to lock file: %s", err);
+                       continue;
+               }
+               files << temp;
+       }
+       
+       //Analyze all source files first
+       if(analyzeFiles(files))
+       {
+               //Now split files according to Cue Sheet
+               splitFiles();
        }
        
-       //Analyze all source files
-       analyzeFiles(files);
+       //Release locks
+       while(!m_locks.isEmpty())
+       {
+               delete m_locks.takeFirst();
+       }
 }
 
-void CueImportDialog::analyzeFiles(QStringList &files)
+bool CueImportDialog::analyzeFiles(QStringList &files)
 {
        m_fileInfo.clear();
+       bool bSuccess = true;
 
-       WorkingBanner *progress = new WorkingBanner(dynamic_cast<QWidget*>(parent()));
+       WorkingBanner *progress = new WorkingBanner(this);
        FileAnalyzer *analyzer = new FileAnalyzer(files);
+       
        connect(analyzer, SIGNAL(fileSelected(QString)), progress, SLOT(setText(QString)), Qt::QueuedConnection);
        connect(analyzer, SIGNAL(fileAnalyzed(AudioFileModel)), this, SLOT(analyzedFile(AudioFileModel)), Qt::QueuedConnection);
+       connect(progress, SIGNAL(userAbort()), analyzer, SLOT(abortProcess()), Qt::DirectConnection);
+
+       progress->show(tr("Analyzing file(s), please wait..."), analyzer);
+       progress->close();
+
+       if(analyzer->filesAccepted() < static_cast<unsigned int>(files.count()))
+       {
+               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)
+               {
+                       bSuccess = false;
+               }
+       }
+
+       LAMEXP_DELETE(progress);
+       LAMEXP_DELETE(analyzer);
+
+       return bSuccess;
+}
+
+void CueImportDialog::splitFiles(void)
+{
+       QString baseName = QFileInfo(m_cueFileName).completeBaseName().replace(".", " ").simplified();
 
-       progress->show(tr("Adding file(s), please wait..."), analyzer);
+       WorkingBanner *progress = new WorkingBanner(this);
+       CueSplitter *splitter  = new CueSplitter(m_outputDir, baseName, m_model, m_fileInfo);
+
+       connect(splitter, SIGNAL(fileSelected(QString)), progress, SLOT(setText(QString)), Qt::QueuedConnection);
+       connect(splitter, SIGNAL(fileSplit(AudioFileModel)), m_fileList, SLOT(addFile(AudioFileModel)), Qt::QueuedConnection);
+       connect(splitter, SIGNAL(progressValChanged(unsigned int)), progress, SLOT(setProgressVal(unsigned int)), Qt::QueuedConnection);
+       connect(splitter, SIGNAL(progressMaxChanged(unsigned int)), progress, SLOT(setProgressMax(unsigned int)), Qt::QueuedConnection);
+       connect(progress, SIGNAL(userAbort()), splitter, SLOT(abortProcess()), Qt::DirectConnection);
+
+       progress->show(tr("Splitting file(s), please wait..."), splitter);
        progress->close();
+
+       if(splitter->getAborted())      
+       {
+               QMessageBox::warning(this, tr("Cue Sheet Error"), tr("Process was aborted by the user after %1 track(s)!").arg(QString::number(splitter->getTracksSuccess())));
+       }
+       else if(!splitter->getSuccess())
+       {
+               QMessageBox::warning(this, tr("Cue Sheet Error"), tr("An unexpected error has occured while splitting the Cue Sheet!"));
+       }
+       else
+       {
+               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*/)));
+               QMessageBox::information(this, tr("Cue Sheet Completed"), text);
+       }
+
+       LAMEXP_DELETE(splitter);
        LAMEXP_DELETE(progress);
 }