OSDN Git Service

Bum version.
[lamexp/LameXP.git] / src / Dialog_CueImport.cpp
index 7f14ac7..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 <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
@@ -71,6 +74,7 @@ CueImportDialog::CueImportDialog(QWidget *parent, FileListModel *fileList, const
        //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.")));
@@ -110,15 +114,80 @@ int CueImportDialog::exec(void)
                return CueSheetModel::ErrorIOFailure;
        }
 
-       m_outputDir = QString("%1/%2").arg(cueFileInfo.canonicalPath(), cueFileInfo.completeBaseName());
-       for(int n = 2; QDir(m_outputDir).exists(); n++)
+       //----------------------//
+
+       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(), cueFileInfo.completeBaseName(), QString::number(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());
+       int iResult = m_model->loadCueSheet(m_cueFileName, QApplication::instance(), codec);
        if(iResult != CueSheetModel::ErrorSuccess)
        {
                QString errorMsg = tr("An unknown error has occured!");
@@ -155,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;
@@ -169,8 +262,8 @@ 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);
@@ -193,8 +286,10 @@ 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;
@@ -204,6 +299,11 @@ void CueImportDialog::importButtonClicked(void)
        accept();
 }
 
+void CueImportDialog::loadOtherButtonClicked(void)
+{
+       done(-1);
+}
+
 void CueImportDialog::analyzedFile(const AudioFileModel &file)
 {
        qDebug("Received result: <%s> <%s/%s>", file.filePath().toLatin1().constData(), file.formatContainerType().toLatin1().constData(), file.formatAudioType().toLatin1().constData());
@@ -241,8 +341,7 @@ void CueImportDialog::importCueSheet(void)
                //Now split files according to Cue Sheet
                splitFiles();
        }
-
-
+       
        //Release locks
        while(!m_locks.isEmpty())
        {
@@ -281,13 +380,15 @@ bool CueImportDialog::analyzeFiles(QStringList &files)
 
 void CueImportDialog::splitFiles(void)
 {
-       QString baseName = QFileInfo(m_cueFileName).completeBaseName().replace(".", " ").left(42).trimmed();
+       QString baseName = QFileInfo(m_cueFileName).completeBaseName().replace(".", " ").simplified();
 
        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);