383 lines
12 KiB
C++
383 lines
12 KiB
C++
/* This file is part of Clementine.
|
|
Copyright 2010, David Sansome <me@davidsansome.com>
|
|
|
|
Clementine 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.
|
|
|
|
Clementine 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 Clementine. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include "transcodedialog.h"
|
|
|
|
#include <QAction>
|
|
#include <QDateTime>
|
|
#include <QDirIterator>
|
|
#include <QFileDialog>
|
|
#include <QPushButton>
|
|
#include <QSettings>
|
|
#include <algorithm>
|
|
|
|
#include "core/application.h"
|
|
#include "transcoder.h"
|
|
#include "transcoderoptionsdialog.h"
|
|
#include "ui/iconloader.h"
|
|
#include "ui/mainwindow.h"
|
|
#include "ui_transcodedialog.h"
|
|
#include "ui_transcodelogdialog.h"
|
|
#include "widgets/fileview.h"
|
|
|
|
// winspool.h defines this :(
|
|
#ifdef AddJob
|
|
#undef AddJob
|
|
#endif
|
|
|
|
const char* TranscodeDialog::kSettingsGroup = "Transcoder";
|
|
const int TranscodeDialog::kProgressInterval = 500;
|
|
const int TranscodeDialog::kMaxDestinationItems = 10;
|
|
|
|
static bool ComparePresetsByName(const TranscoderPreset& left,
|
|
const TranscoderPreset& right) {
|
|
return left.name_ < right.name_;
|
|
}
|
|
|
|
TranscodeDialog::TranscodeDialog(QWidget* parent)
|
|
: QDialog(parent),
|
|
ui_(new Ui_TranscodeDialog),
|
|
details_ui_(new Ui_TranscodeLogDialog),
|
|
details_dialog_(new QDialog(this)),
|
|
transcoder_(new Transcoder(this)),
|
|
queued_(0),
|
|
finished_success_(0),
|
|
finished_failed_(0) {
|
|
ui_->setupUi(this);
|
|
ui_->files->header()->setSectionResizeMode(QHeaderView::ResizeToContents);
|
|
|
|
details_ui_->setupUi(details_dialog_);
|
|
details_ui_->pipelines->setPipelineModel(transcoder_->model());
|
|
QPushButton* clear_button = details_ui_->buttonBox->addButton(
|
|
tr("Clear"), QDialogButtonBox::ResetRole);
|
|
connect(clear_button, SIGNAL(clicked()), details_ui_->log, SLOT(clear()));
|
|
|
|
if (Application::DebugFeaturesEnabled()) {
|
|
QAction* dump_action = new QAction(tr("Dump Graph"));
|
|
details_ui_->pipelines->addAction(dump_action);
|
|
connect(dump_action, SIGNAL(triggered(bool)),
|
|
SLOT(PipelineDumpAction(bool)));
|
|
}
|
|
|
|
// Get presets
|
|
QList<TranscoderPreset> presets = Transcoder::GetAllPresets();
|
|
std::sort(presets.begin(), presets.end(), ComparePresetsByName);
|
|
for (const TranscoderPreset& preset : presets) {
|
|
ui_->format->addItem(
|
|
QString("%1 (.%2)").arg(preset.name_, preset.extension_),
|
|
QVariant::fromValue(preset));
|
|
}
|
|
|
|
// Load settings
|
|
QSettings s;
|
|
s.beginGroup(kSettingsGroup);
|
|
last_add_dir_ = s.value("last_add_dir", QDir::homePath()).toString();
|
|
last_import_dir_ = s.value("last_import_dir", QDir::homePath()).toString();
|
|
|
|
QString last_output_format =
|
|
s.value("last_output_format", "audio/x-vorbis").toString();
|
|
for (int i = 0; i < ui_->format->count(); ++i) {
|
|
if (last_output_format ==
|
|
ui_->format->itemData(i).value<TranscoderPreset>().codec_mimetype_) {
|
|
ui_->format->setCurrentIndex(i);
|
|
break;
|
|
}
|
|
}
|
|
|
|
ui_->remove_original->setChecked(
|
|
s.value("overwrite_existing", false).toBool());
|
|
|
|
// Add a start button
|
|
start_button_ = ui_->button_box->addButton(tr("Start transcoding"),
|
|
QDialogButtonBox::ActionRole);
|
|
cancel_button_ = ui_->button_box->button(QDialogButtonBox::Cancel);
|
|
close_button_ = ui_->button_box->button(QDialogButtonBox::Close);
|
|
|
|
close_button_->setShortcut(QKeySequence::Close);
|
|
|
|
// Hide elements
|
|
cancel_button_->hide();
|
|
ui_->progress_group->hide();
|
|
|
|
// Connect stuff
|
|
connect(ui_->add, SIGNAL(clicked()), SLOT(Add()));
|
|
connect(ui_->import, SIGNAL(clicked()), SLOT(Import()));
|
|
connect(ui_->remove, SIGNAL(clicked()), SLOT(Remove()));
|
|
connect(start_button_, SIGNAL(clicked()), SLOT(Start()));
|
|
connect(cancel_button_, SIGNAL(clicked()), SLOT(Cancel()));
|
|
connect(close_button_, SIGNAL(clicked()), SLOT(hide()));
|
|
connect(ui_->details, SIGNAL(clicked()), details_dialog_, SLOT(show()));
|
|
connect(ui_->options, SIGNAL(clicked()), SLOT(Options()));
|
|
connect(ui_->select, SIGNAL(clicked()), SLOT(AddDestination()));
|
|
|
|
connect(transcoder_, SIGNAL(JobComplete(QUrl, QString, bool)),
|
|
SLOT(JobComplete(QUrl, QString, bool)));
|
|
connect(transcoder_, SIGNAL(LogLine(QString)), SLOT(LogLine(QString)));
|
|
connect(transcoder_, SIGNAL(AllJobsComplete()), SLOT(AllJobsComplete()));
|
|
}
|
|
|
|
TranscodeDialog::~TranscodeDialog() {}
|
|
|
|
void TranscodeDialog::SetWorking(bool working) {
|
|
start_button_->setVisible(!working);
|
|
cancel_button_->setVisible(working);
|
|
close_button_->setVisible(!working);
|
|
ui_->input_group->setEnabled(!working);
|
|
ui_->output_group->setEnabled(!working);
|
|
ui_->progress_group->setVisible(true);
|
|
|
|
if (working)
|
|
progress_timer_.start(kProgressInterval, this);
|
|
else
|
|
progress_timer_.stop();
|
|
}
|
|
|
|
void TranscodeDialog::Start() {
|
|
QAbstractItemModel* file_model = ui_->files->model();
|
|
const int count = file_model->rowCount();
|
|
|
|
if (count == 0) {
|
|
// Nothing to process.
|
|
return;
|
|
}
|
|
|
|
SetWorking(true);
|
|
|
|
TranscoderPreset preset = ui_->format->itemData(ui_->format->currentIndex())
|
|
.value<TranscoderPreset>();
|
|
|
|
// Add jobs to the transcoder
|
|
for (int i = 0; i < count; ++i) {
|
|
QFileInfo input_fileinfo(
|
|
file_model->index(i, 0).data(Qt::UserRole).toString());
|
|
QString output_filename = GetOutputFileName(input_fileinfo, preset);
|
|
transcoder_->AddJob(QUrl::fromLocalFile(input_fileinfo.filePath()), preset,
|
|
output_filename);
|
|
}
|
|
|
|
// Set up the progressbar
|
|
ui_->progress_bar->setValue(0);
|
|
ui_->progress_bar->setMaximum(count * 100);
|
|
|
|
// Reset the UI
|
|
queued_ = count;
|
|
finished_success_ = 0;
|
|
finished_failed_ = 0;
|
|
UpdateStatusText();
|
|
|
|
// Start transcoding
|
|
transcoder_->Start();
|
|
|
|
// Save the last output format
|
|
QSettings s;
|
|
s.beginGroup(kSettingsGroup);
|
|
s.setValue("last_output_format", preset.codec_mimetype_);
|
|
s.setValue("overwrite_existing", ui_->remove_original->isChecked());
|
|
}
|
|
|
|
void TranscodeDialog::Cancel() {
|
|
transcoder_->Cancel();
|
|
SetWorking(false);
|
|
}
|
|
|
|
void TranscodeDialog::PipelineDumpAction(bool checked) {
|
|
for (int i : details_ui_->pipelines->GetSelectedIds()) {
|
|
transcoder_->DumpGraph(i);
|
|
}
|
|
}
|
|
|
|
void TranscodeDialog::JobComplete(const QUrl& input, const QString& output,
|
|
bool success) {
|
|
if (success)
|
|
finished_success_++;
|
|
else
|
|
finished_failed_++;
|
|
queued_--;
|
|
|
|
UpdateStatusText();
|
|
UpdateProgress();
|
|
|
|
bool overwrite_existing = ui_->remove_original->isChecked();
|
|
|
|
if (success && overwrite_existing && input.isLocalFile()) {
|
|
QFileInfo input_fileinfo(input.toLocalFile());
|
|
QFileInfo output_fileinfo(output);
|
|
|
|
bool same_extension = input_fileinfo.suffix() == output_fileinfo.suffix();
|
|
bool same_path =
|
|
input_fileinfo.absolutePath() == output_fileinfo.absolutePath();
|
|
|
|
QFile(input_fileinfo.absoluteFilePath()).remove();
|
|
if (same_path && same_extension) {
|
|
QFile(output_fileinfo.absoluteFilePath())
|
|
.rename(input_fileinfo.fileName());
|
|
}
|
|
}
|
|
}
|
|
|
|
void TranscodeDialog::UpdateProgress() {
|
|
int progress = (finished_success_ + finished_failed_) * 100;
|
|
|
|
QMap<QUrl, float> current_jobs = transcoder_->GetProgress();
|
|
for (float value : current_jobs.values()) {
|
|
progress += qBound(0, int(value * 100), 99);
|
|
}
|
|
|
|
ui_->progress_bar->setValue(progress);
|
|
}
|
|
|
|
void TranscodeDialog::UpdateStatusText() {
|
|
QStringList sections;
|
|
|
|
if (queued_) {
|
|
sections << "<font color=\"#3467c8\">" + tr("%n remaining", "", queued_) +
|
|
"</font>";
|
|
}
|
|
|
|
if (finished_success_) {
|
|
sections << "<font color=\"#02b600\">" +
|
|
tr("%n finished", "", finished_success_) + "</font>";
|
|
}
|
|
|
|
if (finished_failed_) {
|
|
sections << "<font color=\"#b60000\">" +
|
|
tr("%n failed", "", finished_failed_) + "</font>";
|
|
}
|
|
|
|
ui_->progress_text->setText(sections.join(", "));
|
|
}
|
|
|
|
void TranscodeDialog::AllJobsComplete() { SetWorking(false); }
|
|
|
|
void TranscodeDialog::Add() {
|
|
QStringList filenames = QFileDialog::getOpenFileNames(
|
|
this, tr("Add files to transcode"), last_add_dir_,
|
|
QString("%1 (%2);;%3")
|
|
.arg(tr("Music"), FileView::kFileFilter,
|
|
tr(MainWindow::kAllFilesFilterSpec)));
|
|
|
|
if (filenames.isEmpty()) return;
|
|
|
|
SetFilenames(filenames);
|
|
|
|
last_add_dir_ = filenames[0];
|
|
QSettings s;
|
|
s.beginGroup(kSettingsGroup);
|
|
s.setValue("last_add_dir", last_add_dir_);
|
|
}
|
|
|
|
void TranscodeDialog::Import() {
|
|
QString path = QFileDialog::getExistingDirectory(
|
|
this, tr("Open a directory to import music from"), last_import_dir_,
|
|
QFileDialog::ShowDirsOnly);
|
|
|
|
if (path.isEmpty()) return;
|
|
|
|
QStringList filenames;
|
|
QStringList audioTypes =
|
|
QString(FileView::kFileFilter).split(" ", QString::SkipEmptyParts);
|
|
QDirIterator files(path, audioTypes, QDir::Files | QDir::Readable,
|
|
QDirIterator::Subdirectories);
|
|
|
|
while (files.hasNext()) {
|
|
filenames << files.next();
|
|
}
|
|
|
|
SetFilenames(filenames);
|
|
|
|
last_import_dir_ = path;
|
|
QSettings settings;
|
|
settings.beginGroup(kSettingsGroup);
|
|
settings.setValue("last_import_dir", last_import_dir_);
|
|
}
|
|
|
|
void TranscodeDialog::SetFilenames(const QStringList& filenames) {
|
|
for (const QString& filename : filenames) {
|
|
QString name = filename.section('/', -1, -1);
|
|
QString path = filename.section('/', 0, -2);
|
|
|
|
QTreeWidgetItem* item =
|
|
new QTreeWidgetItem(ui_->files, QStringList() << name << path);
|
|
item->setData(0, Qt::UserRole, filename);
|
|
}
|
|
}
|
|
|
|
void TranscodeDialog::Remove() { qDeleteAll(ui_->files->selectedItems()); }
|
|
|
|
void TranscodeDialog::LogLine(const QString& message) {
|
|
QString date(QDateTime::currentDateTime().toString(Qt::TextDate));
|
|
details_ui_->log->appendPlainText(QString("%1: %2").arg(date, message));
|
|
}
|
|
|
|
void TranscodeDialog::timerEvent(QTimerEvent* e) {
|
|
QDialog::timerEvent(e);
|
|
|
|
if (e->timerId() == progress_timer_.timerId()) {
|
|
UpdateProgress();
|
|
}
|
|
}
|
|
|
|
void TranscodeDialog::Options() {
|
|
TranscoderPreset preset = ui_->format->itemData(ui_->format->currentIndex())
|
|
.value<TranscoderPreset>();
|
|
|
|
TranscoderOptionsDialog dialog(preset, this);
|
|
dialog.exec();
|
|
}
|
|
|
|
// Adds a folder to the destination box.
|
|
void TranscodeDialog::AddDestination() {
|
|
int index = ui_->destination->currentIndex();
|
|
QString initial_dir = (!ui_->destination->itemData(index).isNull()
|
|
? ui_->destination->itemData(index).toString()
|
|
: QDir::homePath());
|
|
QString dir =
|
|
QFileDialog::getExistingDirectory(this, tr("Add folder"), initial_dir);
|
|
|
|
if (!dir.isEmpty()) {
|
|
// Keep only a finite number of items in the box.
|
|
while (ui_->destination->count() >= kMaxDestinationItems) {
|
|
ui_->destination->removeItem(1); // Remove the oldest folder item.
|
|
}
|
|
|
|
QIcon icon = IconLoader::Load("folder", IconLoader::Base);
|
|
QVariant data = QVariant::fromValue(dir);
|
|
// Do not insert duplicates.
|
|
int duplicate_index = ui_->destination->findData(data);
|
|
if (duplicate_index == -1) {
|
|
ui_->destination->addItem(icon, dir, data);
|
|
ui_->destination->setCurrentIndex(ui_->destination->count() - 1);
|
|
} else {
|
|
ui_->destination->setCurrentIndex(duplicate_index);
|
|
}
|
|
}
|
|
}
|
|
|
|
QString TranscodeDialog::GetOutputFileName(
|
|
const QFileInfo& input, const TranscoderPreset& preset) const {
|
|
QFileInfo path(
|
|
ui_->destination->itemData(ui_->destination->currentIndex()).toString());
|
|
QString output_path;
|
|
if (path.isDir()) {
|
|
output_path = path.filePath();
|
|
} else {
|
|
// Keep the original path.
|
|
output_path = input.path();
|
|
}
|
|
return output_path + '/' + input.completeBaseName() + '.' + preset.extension_;
|
|
}
|