From 294023cf2d93a1a8bc0df4e56830cc88420e23ab Mon Sep 17 00:00:00 2001
From: Mattias Andersson <mandersson444@gmail.com>
Date: Sat, 11 Jan 2014 01:04:59 +0100
Subject: [PATCH] Add the ability to choose destination folder in the
 transcoder dialog.

---
 src/transcoder/transcodedialog.cpp | 47 +++++++++++++++++++++++++++++-
 src/transcoder/transcodedialog.h   |  6 ++++
 2 files changed, 52 insertions(+), 1 deletion(-)

diff --git a/src/transcoder/transcodedialog.cpp b/src/transcoder/transcodedialog.cpp
index d2e11b5e1..9886f4ba2 100644
--- a/src/transcoder/transcodedialog.cpp
+++ b/src/transcoder/transcodedialog.cpp
@@ -20,6 +20,7 @@
 #include "transcoderoptionsdialog.h"
 #include "ui_transcodedialog.h"
 #include "ui_transcodelogdialog.h"
+#include "ui/iconloader.h"
 #include "ui/mainwindow.h"
 #include "widgets/fileview.h"
 
@@ -95,6 +96,12 @@ TranscodeDialog::TranscodeDialog(QWidget *parent)
   cancel_button_->hide();
   ui_->progress_group->hide();
 
+  // Add a destination.
+  QIcon icon = IconLoader::Load("folder");
+  QVariant data = QVariant::fromValue(QDir::homePath());
+  QString text = TrimPath(data.toString());
+  ui_->destination->addItem(icon, text, data);
+
   // Connect stuff
   connect(ui_->add, SIGNAL(clicked()), SLOT(Add()));
   connect(ui_->remove, SIGNAL(clicked()), SLOT(Remove()));
@@ -103,6 +110,8 @@ TranscodeDialog::TranscodeDialog(QWidget *parent)
   connect(close_button_, SIGNAL(clicked()), SLOT(hide()));
   connect(ui_->details, SIGNAL(clicked()), log_dialog_, SLOT(show()));
   connect(ui_->options, SIGNAL(clicked()), SLOT(Options()));
+  connect(ui_->destination, SIGNAL(activated(int)),
+          SLOT(SetDestination(int)));
 
   connect(transcoder_, SIGNAL(JobComplete(QString,bool)), SLOT(JobComplete(QString,bool)));
   connect(transcoder_, SIGNAL(LogLine(QString)), SLOT(LogLine(QString)));
@@ -138,7 +147,8 @@ void TranscodeDialog::Start() {
   // Add jobs to the transcoder
   for (int i=0 ; i<file_model->rowCount() ; ++i) {
     QString filename = file_model->index(i, 0).data(Qt::UserRole).toString();
-    transcoder_->AddJob(filename, preset);
+    QString outfilename = SetOutputFileName(filename, preset);
+    transcoder_->AddJob(filename, preset, outfilename);
   }
 
   // Set up the progressbar
@@ -265,3 +275,38 @@ void TranscodeDialog::Options() {
     dialog.exec();
   }
 }
+
+void TranscodeDialog::SetDestination(int index) {
+  if (!ui_->destination->itemData(index).isNull()) {
+    QString dir = QFileDialog::getExistingDirectory(
+        this, tr("Choose folder"),
+        ui_->destination->itemData(index).toString());
+    // Do not update the QComboBox if the user canceled the dialog.
+    if (!dir.isEmpty()) {
+      QVariant data = QVariant::fromValue(dir);
+      QString text = TrimPath(dir);
+      ui_->destination->setItemText(index, text);
+      ui_->destination->setItemData(index, data);
+    }
+  }
+}
+
+// Returns the rightmost non-empty part of 'path'.
+QString TranscodeDialog::TrimPath(const QString& path) {
+  return path.section('/', -1, -1, QString::SectionSkipEmpty);
+}
+
+QString TranscodeDialog::SetOutputFileName(const QString& input,
+                                           const TranscoderPreset &preset) {
+  QString path = ui_->destination->itemData(
+      ui_->destination->currentIndex()).toString();
+  if (path.isEmpty()) {
+    // Keep the original path.
+    return input.section('.', 0, -2) + '.' + preset.extension_;
+  }
+  else {
+    QString file_name = TrimPath(input);
+    file_name = file_name.section('.', 0, -2);
+    return path + '/' + file_name + '.' + preset.extension_;
+  }
+}
diff --git a/src/transcoder/transcodedialog.h b/src/transcoder/transcodedialog.h
index ebf70d204..a88097ef7 100644
--- a/src/transcoder/transcodedialog.h
+++ b/src/transcoder/transcodedialog.h
@@ -25,6 +25,8 @@ class Transcoder;
 class Ui_TranscodeDialog;
 class Ui_TranscodeLogDialog;
 
+struct TranscoderPreset;
+
 class TranscodeDialog : public QDialog {
   Q_OBJECT
 
@@ -49,11 +51,15 @@ class TranscodeDialog : public QDialog {
   void LogLine(const QString& message);
   void AllJobsComplete();
   void Options();
+  void SetDestination(int index);
 
  private:
   void SetWorking(bool working);
   void UpdateStatusText();
   void UpdateProgress();
+  QString TrimPath(const QString& path);
+  QString SetOutputFileName(const QString& input,
+                            const TranscoderPreset& preset);
 
  private:
   Ui_TranscodeDialog* ui_;