- Added session-level upload/download-rate limit preferences

[qtrapids] / src / gui / PreferencesDialog.cpp

diff --git a/src/gui/PreferencesDialog.cpp b/src/gui/PreferencesDialog.cpp
index e541e90..099755d 100644 (file)
--- a/src/gui/PreferencesDialog.cpp
+++ b/src/gui/PreferencesDialog.cpp
@@ -22,32 +22,58 @@
 #include <QLineEdit>
 #include <QLabel>
 #include <QPushButton>
+#include <QSpinBox>
 #include <QDialogButtonBox>
 #include <QAbstractButton>
 #include <QFileDialog>
 
+#include "QBittorrentSession.h"
 #include "PreferencesDialog.h"
 
-PreferencesDialog::PreferencesDialog(QWidget* parent, Qt::WindowFlags f) :
+PreferencesDialog::PreferencesDialog(QWidget* parent, Qt::WindowFlags f,
+                                                                                                                                                qtrapids::QBittorrentSession *const btSession) :
                QDialog(parent, f), // Superclass
                dirLineEdit_(NULL),
                dialogButtons_(NULL),
+               uploadRateSpinBox_(NULL),
+               downloadRateSpinBox_(NULL),
+               btSession_(btSession),
                settings_()
 {
        setWindowTitle("Preferences");
 
        QBoxLayout *verticalBox = new QBoxLayout(QBoxLayout::TopToBottom);
-       QBoxLayout *horizontalBox1 = new QBoxLayout(QBoxLayout::LeftToRight);
+       //QBoxLayout *horizontalBox1 = new QBoxLayout(QBoxLayout::LeftToRight);
+       QGridLayout *grid = new QGridLayout;
        setLayout(verticalBox);
-       verticalBox->addLayout(horizontalBox1);
+       verticalBox->addLayout(grid);
 
        QLabel *dirLabel = new QLabel(tr("Download directory: "));
        dirLineEdit_ = new QLineEdit(this);
        QPushButton *browseDirButton = new QPushButton(tr("Browse.."));
-
-       horizontalBox1->addWidget(dirLabel);
-       horizontalBox1->addWidget(dirLineEdit_);
-       horizontalBox1->addWidget(browseDirButton);
+       
+       QLabel *uploadLabel = new QLabel(tr("Max. upload rate: "));
+       QLabel *downloadLabel = new QLabel(tr("Max. download rate: "));
+       uploadRateSpinBox_ = new QSpinBox(this);
+       downloadRateSpinBox_ = new QSpinBox(this);
+       
+       grid->addWidget(dirLabel, 0, 0);
+       grid->addWidget(dirLineEdit_, 0, 1);
+       grid->addWidget(browseDirButton, 0, 2);
+       
+       grid->addWidget(uploadLabel, 1, 0);
+       grid->addWidget(uploadRateSpinBox_, 1, 1);
+       grid->addWidget(downloadLabel, 2, 0);
+       grid->addWidget(downloadRateSpinBox_, 2, 1);
+       
+       uploadRateSpinBox_->setRange(0, 1000);
+       uploadRateSpinBox_->setSuffix(" kB/s");
+       downloadRateSpinBox_->setRange(0, 1000);
+       downloadRateSpinBox_->setSuffix(" kB/s");
+       
+//     horizontalBox1->addWidget(dirLabel);
+//     horizontalBox1->addWidget(dirLineEdit_);
+//     horizontalBox1->addWidget(browseDirButton);
 
        connect(browseDirButton, SIGNAL(clicked()),
                this, SLOT(on_browseDirButtonClicked()));
@@ -95,7 +121,7 @@ void PreferencesDialog::on_buttonClicked(QAbstractButton* button)
        case QDialogButtonBox::AcceptRole :
                qDebug() << "PreferencesDialog: OK";
                WriteSettings();
-               close();
+               done(QDialog::Accepted);
                break;
        case QDialogButtonBox::ApplyRole :
                qDebug() << "PreferencesDialog: APPLY";
@@ -103,7 +129,7 @@ void PreferencesDialog::on_buttonClicked(QAbstractButton* button)
                break;
        case QDialogButtonBox::RejectRole :
                qDebug() << "PreferencesDialog: CANCEL";
-               close();
+               done(QDialog::Rejected);
                break;
        default:
                return;
@@ -117,6 +143,7 @@ void PreferencesDialog::on_downloadDirectorySelected(const QString& directory)
        if (directory == "")
                return;
 
+       dirLineEdit_->clear();
        dirLineEdit_->insert(directory);
 
        /// @todo check that user has privileges to write to this directory.
@@ -126,16 +153,30 @@ void PreferencesDialog::on_downloadDirectorySelected(const QString& directory)
 // ========================= Private functions ==========================
 void PreferencesDialog::WriteSettings()
 {
+       int ulRate = 1000*uploadRateSpinBox_->value();
+       int dlRate = 1000*downloadRateSpinBox_->value();
+       
        settings_.setValue("download/directory", dirLineEdit_->text());
-
+       settings_.setValue("network/uploadRate", ulRate);
+       settings_.setValue("network/downloadRate", dlRate);
+       
        // NOTE: We might need to call QSettigns::sync() here to instantly write settings.
        // settings are written also by QSettings() destructor and by event loop at regular interval.
+       
+       // If session pointer was given, apply settings immediately.
+       if (btSession_) {
+               btSession_->setUploadRateLimit(ulRate);
+               btSession_->setDownloadrateLimit(dlRate);
+       }
+       
 }
 
 void PreferencesDialog::ReadSettings()
 {
        dirLineEdit_->insert(settings_.value("download/directory").toString());
-
+       uploadRateSpinBox_->setValue(settings_.value("network/uploadRate").toInt());
+       downloadRateSpinBox_->setValue(settings_.value("network/downloadRate").toInt());
+       
        // NOTE: We might need to call QSettigns::sync() here to instantly write settings.
        // settings are written also by QSettings() destructor and by event loop at regular interval.
 }