1 ///////////////////////////////////////////////////////////////////////////////
2 // LameXP - Audio Encoder Front-End
3 // Copyright (C) 2004-2020 LoRd_MuldeR <MuldeR2@GMX.de>
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; always including the non-optional
9 // LAMEXP GNU GENERAL PUBLIC LICENSE ADDENDUM. See "License.txt" file!
11 // This program is distributed in the hope that it will be useful,
12 // but WITHOUT ANY WARRANTY; without even the implied warranty of
13 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 // GNU General Public License for more details.
16 // You should have received a copy of the GNU General Public License along
17 // with this program; if not, write to the Free Software Foundation, Inc.,
18 // 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
20 // http://www.gnu.org/licenses/gpl-2.0.txt
21 ///////////////////////////////////////////////////////////////////////////////
23 #include "Dialog_CueImport.h"
26 #include "UIC_CueSheetImport.h"
30 #include "Model_CueSheet.h"
31 #include "Model_AudioFile.h"
32 #include "Model_FileList.h"
33 #include "Dialog_WorkingBanner.h"
34 #include "Thread_FileAnalyzer.h"
35 #include "Thread_CueSplitter.h"
36 #include "Registry_Decoder.h"
37 #include "LockedFile.h"
40 #include <MUtils/Global.h>
41 #include <MUtils/OSSupport.h>
42 #include <MUtils/GUI.h>
46 #include <QMessageBox>
48 #include <QFileDialog>
49 #include <QProgressDialog>
52 #include <QInputDialog>
54 #define SET_FONT_BOLD(WIDGET,BOLD) { QFont _font = WIDGET->font(); _font.setBold(BOLD); WIDGET->setFont(_font); }
55 #define EXPAND(STR) QString(STR).leftJustified(96, ' ')
57 ////////////////////////////////////////////////////////////
58 // Constructor & Destructor
59 ////////////////////////////////////////////////////////////
61 CueImportDialog::CueImportDialog(QWidget
*parent
, FileListModel
*fileList
, const QString
&cueFile
, const SettingsModel
*settings
)
64 ui(new Ui::CueSheetImport
),
66 m_cueFileName(cueFile
),
69 //Init the dialog, from the .ui file
73 setMinimumSize(this->size());
74 setMaximumHeight(this->height());
77 m_model
= new CueSheetModel();
78 connect(m_model
, SIGNAL(modelReset()), this, SLOT(modelChanged()));
81 ui
->treeView
->setModel(m_model
);
82 ui
->treeView
->header()->setStretchLastSection(false);
83 ui
->treeView
->header()->setResizeMode(QHeaderView::ResizeToContents
);
84 ui
->treeView
->header()->setResizeMode(1, QHeaderView::Stretch
);
85 ui
->treeView
->header()->setMovable(false);
86 ui
->treeView
->setItemsExpandable(false);
88 //Enable up/down button
89 connect(ui
->imprtButton
, SIGNAL(clicked()), this, SLOT(importButtonClicked()));
90 connect(ui
->browseButton
, SIGNAL(clicked()), this, SLOT(browseButtonClicked()));
91 connect(ui
->loadOtherButton
, SIGNAL(clicked()), this, SLOT(loadOtherButtonClicked()));
94 ui
->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.")));
97 CueImportDialog::~CueImportDialog(void)
99 MUTILS_DELETE(m_model
);
103 ////////////////////////////////////////////////////////////
105 ////////////////////////////////////////////////////////////
107 void CueImportDialog::showEvent(QShowEvent
*event
)
109 QDialog::showEvent(event
);
113 ////////////////////////////////////////////////////////////
115 ////////////////////////////////////////////////////////////
117 int CueImportDialog::exec(void)
119 WorkingBanner
*progress
= new WorkingBanner(dynamic_cast<QWidget
*>(parent()));
120 progress
->show(tr("Loading Cue Sheet file, please be patient..."));
122 QFileInfo
cueFileInfo(m_cueFileName
);
123 if(!cueFileInfo
.exists() || !cueFileInfo
.isFile())
125 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("-", "−");
126 QMessageBox::warning(progress
, tr("Cue Sheet Error"), text
);
128 MUTILS_DELETE(progress
);
129 return CueSheetModel::ErrorIOFailure
;
132 //----------------------//
134 QTextCodec
*codec
= NULL
;
136 QFile
cueFile(cueFileInfo
.canonicalFilePath());
137 cueFile
.open(QIODevice::ReadOnly
);
138 QByteArray bomCheck
= cueFile
.isOpen() ? cueFile
.peek(16) : QByteArray();
140 if((!bomCheck
.isEmpty()) && bomCheck
.startsWith("\xef\xbb\xbf"))
142 codec
= QTextCodec::codecForName("UTF-8");
144 else if((!bomCheck
.isEmpty()) && bomCheck
.startsWith("\xff\xfe"))
146 codec
= QTextCodec::codecForName("UTF-16LE");
148 else if((!bomCheck
.isEmpty()) && bomCheck
.startsWith("\xfe\xff"))
150 codec
= QTextCodec::codecForName("UTF-16BE");
154 const QString systemDefault
= tr("(System Default)");
156 QStringList codecList
;
157 codecList
.append(systemDefault
);
158 codecList
.append(MUtils::available_codepages());
160 QInputDialog
*input
= new QInputDialog(progress
);
161 input
->setLabelText(EXPAND(tr("Select ANSI Codepage for Cue Sheet file:")));
162 input
->setOkButtonText(tr("OK"));
163 input
->setCancelButtonText(tr("Cancel"));
164 input
->setTextEchoMode(QLineEdit::Normal
);
165 input
->setComboBoxItems(codecList
);
167 if(input
->exec() < 1)
170 MUTILS_DELETE(input
);
171 MUTILS_DELETE(progress
);
175 if(input
->textValue().compare(systemDefault
, Qt::CaseInsensitive
))
177 qDebug("User-selected codec is: %s", input
->textValue().toLatin1().constData());
178 codec
= QTextCodec::codecForName(input
->textValue().toLatin1().constData());
182 qDebug("Going to use the system's default codec!");
183 codec
= QTextCodec::codecForName("System");
186 MUTILS_DELETE(input
);
191 //----------------------//
193 QString baseName
= cueFileInfo
.completeBaseName().simplified();
194 while(baseName
.endsWith(".") || baseName
.endsWith(" ")) baseName
.chop(1);
195 if(baseName
.isEmpty()) baseName
= tr("New Folder");
197 m_outputDir
= QString("%1/%2").arg(cueFileInfo
.canonicalPath(), baseName
);
198 for(int n
= 2; QDir(m_outputDir
).exists() || QFileInfo(m_outputDir
).exists(); n
++)
200 m_outputDir
= QString("%1/%2 (%3)").arg(cueFileInfo
.canonicalPath(), baseName
, QString::number(n
));
203 setWindowTitle(QString("%1: %2").arg(windowTitle().split(":", QString::SkipEmptyParts
).first().trimmed(), cueFileInfo
.fileName()));
205 int iResult
= m_model
->loadCueSheet(m_cueFileName
, QApplication::instance(), codec
);
206 if(iResult
!= CueSheetModel::ErrorSuccess
)
208 QString errorMsg
= tr("An unknown error has occured!");
212 case CueSheetModel::ErrorIOFailure
:
213 errorMsg
= tr("The file could not be opened for reading. Make sure you have the required rights!");
215 case CueSheetModel::ErrorBadFile
:
216 errorMsg
= tr("The provided file does not look like a valid Cue Sheet disc image file!");
218 case CueSheetModel::ErrorUnsupported
:
219 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."));
221 case CueSheetModel::ErrorInconsistent
:
222 errorMsg
= tr("The selected Cue Sheet file contains inconsistent information. Take care!");
226 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("-", "−");
227 QMessageBox::warning(progress
, tr("Cue Sheet Error"), text
);
229 MUTILS_DELETE(progress
);
234 MUTILS_DELETE(progress
);
235 return QDialog::exec();
238 void CueImportDialog::modelChanged(void)
240 ui
->treeView
->expandAll();
241 ui
->editOutputDir
->setText(QDir::toNativeSeparators(m_outputDir
));
242 if(const AudioFileModel_MetaInfo
*albumInfo
= m_model
->getAlbumInfo())
244 ui
->labelArtist
->setText(albumInfo
->artist().isEmpty() ? tr("Unknown Artist") : albumInfo
->artist());
245 ui
->labelAlbum
->setText(albumInfo
->album().isEmpty() ? tr("Unknown Album") : albumInfo
->album());
249 void CueImportDialog::browseButtonClicked(void)
251 QString currentDir
= QDir::fromNativeSeparators(m_outputDir
);
252 while(!QDir(currentDir
).exists())
254 const int pos
= currentDir
.lastIndexOf(QChar('/'));
257 currentDir
= currentDir
.left(pos
);
264 if(MUtils::GUI::themes_enabled())
266 newOutDir
= QFileDialog::getExistingDirectory(this, tr("Choose Output Directory"), currentDir
);
270 QFileDialog
dialog(this, tr("Choose Output Directory"));
271 dialog
.setFileMode(QFileDialog::DirectoryOnly
);
272 dialog
.setDirectory(currentDir
);
275 newOutDir
= dialog
.selectedFiles().first();
279 if(!newOutDir
.isEmpty())
281 m_outputDir
= newOutDir
;
286 void CueImportDialog::importButtonClicked(void)
288 static const unsigned __int64 oneGigabyte
= 1073741824ui
64;
289 static const unsigned __int64 minimumFreeDiskspaceMultiplier
= 2ui
64;
290 static const char *writeTestBuffer
= "LAMEXP_WRITE_TEST";
292 QDir
outputDir(m_outputDir
);
293 outputDir
.mkpath(".");
294 if(!(outputDir
.exists() && outputDir
.isReadable()))
296 QMessageBox::warning(this, tr("LameXP"), QString("<nobr>%2</nobr>").arg(tr("Error: The selected output directory could not be created!")));
300 QFile
writeTest(QString("%1/~%2.txt").arg(m_outputDir
, MUtils::next_rand_str()));
301 if(!(writeTest
.open(QIODevice::ReadWrite
) && (writeTest
.write(writeTestBuffer
) == strlen(writeTestBuffer
))))
303 QMessageBox::warning(this, tr("LameXP"), QString("<nobr>%2</nobr>").arg(tr("Error: The selected output directory is not writable!")));
312 quint64 currentFreeDiskspace
= 0;
313 if(MUtils::OS::free_diskspace(m_outputDir
, currentFreeDiskspace
))
315 if(currentFreeDiskspace
< (oneGigabyte
* minimumFreeDiskspaceMultiplier
))
317 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!")));
326 void CueImportDialog::loadOtherButtonClicked(void)
331 void CueImportDialog::analyzedFile(const AudioFileModel
&file
)
333 qDebug("Received result: <%s> <%s/%s>", file
.filePath().toLatin1().constData(), file
.techInfo().containerType().toLatin1().constData(), file
.techInfo().audioType().toLatin1().constData());
337 ////////////////////////////////////////////////////////////
339 ////////////////////////////////////////////////////////////
341 void CueImportDialog::importCueSheet(void)
344 QList
<LockedFile
*> locks
;
346 //Fetch all files that are referenced in the Cue Sheet and lock them
347 int nFiles
= m_model
->getFileCount();
348 for(int i
= 0; i
< nFiles
; i
++)
352 LockedFile
*temp
= new LockedFile(m_model
->getFileName(i
));
354 files
<< temp
->filePath();
356 catch(const std::exception
&error
)
358 qWarning("Failed to lock file:\n%s\n", error
.what());
362 qWarning("Failed to lock file!");
366 //Check if all files could be locked
367 if(files
.count() < m_model
->getFileCount())
369 if(QMessageBox::warning(this, tr("Cue Sheet Error"), tr("Warning: Some of the required input files could not be found!"), tr("Continue Anyway"), tr("Abort")) == 1)
375 //Process all avialble input files
376 if(files
.count() > 0)
378 //Analyze all source files first
379 if(analyzeFiles(files
))
381 //Now split files according to Cue Sheet
387 while(!locks
.isEmpty())
389 delete locks
.takeFirst();
393 bool CueImportDialog::analyzeFiles(QStringList
&files
)
396 bool bSuccess
= true;
398 WorkingBanner
*progress
= new WorkingBanner(this);
399 FileAnalyzer
*analyzer
= new FileAnalyzer(files
);
401 connect(analyzer
, SIGNAL(fileSelected(QString
)), progress
, SLOT(setText(QString
)), Qt::QueuedConnection
);
402 connect(analyzer
, SIGNAL(fileAnalyzed(AudioFileModel
)), this, SLOT(analyzedFile(AudioFileModel
)), Qt::QueuedConnection
);
403 connect(progress
, SIGNAL(userAbort()), analyzer
, SLOT(abortProcess()), Qt::DirectConnection
);
405 progress
->show(tr("Analyzing file(s), please wait..."), analyzer
);
408 if(analyzer
->filesAccepted() < static_cast<unsigned int>(files
.count()))
410 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)
416 MUTILS_DELETE(progress
);
417 MUTILS_DELETE(analyzer
);
422 void CueImportDialog::splitFiles(void)
424 QString baseName
= QFileInfo(m_cueFileName
).completeBaseName().replace(".", " ").simplified();
426 QScopedPointer
<WorkingBanner
> progress(new WorkingBanner(this));
427 QScopedPointer
<CueSplitter
> splitter(new CueSplitter(m_outputDir
, baseName
, m_model
, m_fileInfo
));
429 connect(splitter
.data(), SIGNAL(fileSelected(QString
)), progress
.data(), SLOT(setText(QString
)), Qt::QueuedConnection
);
430 connect(splitter
.data(), SIGNAL(fileSplit(AudioFileModel
)), m_fileList
, SLOT(addFile(AudioFileModel
)), Qt::QueuedConnection
);
431 connect(splitter
.data(), SIGNAL(progressValChanged(unsigned int)), progress
.data(), SLOT(setProgressVal(unsigned int)), Qt::QueuedConnection
);
432 connect(splitter
.data(), SIGNAL(progressMaxChanged(unsigned int)), progress
.data(), SLOT(setProgressMax(unsigned int)), Qt::QueuedConnection
);
433 connect(progress
.data(), SIGNAL(userAbort()), splitter
.data(), SLOT(abortProcess()), Qt::DirectConnection
);
435 DecoderRegistry::configureDecoders(m_settings
);
437 progress
->show(tr("Splitting file(s), please wait..."), splitter
.data());
440 if(splitter
->getAborted())
442 QMessageBox::warning(this, tr("Cue Sheet Error"), tr("Process was aborted by the user after %n track(s)!", "", splitter
->getTracksSuccess()));
444 else if(!splitter
->getSuccess())
446 QMessageBox::warning(this, tr("Cue Sheet Error"), tr("An unexpected error has occured while splitting the Cue Sheet!"));
450 QString text
= QString("<nobr>%1 %2</nobr>").arg(tr("Imported %n track(s) from the Cue Sheet.", "", splitter
->getTracksSuccess()), tr("Skipped %n track(s).", "", splitter
->getTracksSkipped()));
451 QMessageBox::information(this, tr("Cue Sheet Completed"), text
);