Clementine-audio-player-Mac.../src/transcoder/transcodedialog.cpp

362 lines
11 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;
}
}
// 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_);
}
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();
}
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_;
}