Added build script of Debian
[qtrapids] / src / client / DownloadView.cpp
index d097234..cf04e51 100644 (file)
@@ -1,11 +1,9 @@
 /***************************************************************************
- *   Copyright (C) 2009 by Lassi Väätämöinen   *
- *   lassi.vaatamoinen@ixonos.com   *
+ *   Copyright (C) 2010 by Ixonos Plc   *
  *                                                                         *
  *   This program 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 2 of the License, or     *
- *   (at your option) any later version.                                   *
+ *   the Free Software Foundation; version 2 of the License.               *
  *                                                                         *
  *   This program is distributed in the hope that it will be useful,       *
  *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
@@ -29,14 +27,15 @@ namespace qtrapids
 {
 
 DownloadView::DownloadView(QWidget* parent) :
-        QTreeWidget(parent),
-        items_()
+               QTreeWidget(parent),
+               items_(),
+               settings_()
 {
-    setRootIsDecorated(false); // Hide branch lines, making one-level treeview (similar to list)
-    setHeaderItem(DownloadViewItem::getHeaderItem());
+       setRootIsDecorated(false); // Hide branch lines, making one-level treeview (similar to list)
+       setHeaderItem(DownloadViewItem::getHeaderItem());
 
-    connect(this, SIGNAL(itemPressed(QTreeWidgetItem*, int)),
-            this, SLOT(on_itemClicked(QTreeWidgetItem*, int)));
+       connect(this, SIGNAL(itemPressed(QTreeWidgetItem*, int)),
+               this, SLOT(on_itemClicked(QTreeWidgetItem*, int)));
 
 }
 
@@ -47,177 +46,199 @@ DownloadView::~DownloadView()
 
 void DownloadView::updateItem(TorrentState const& info, ParamsMap_t other_info)
 {
-    DownloadItems_t::iterator p = items_.find(info.hash);
-    switch (info.action)
-    {
-    case TorrentState::action_add :
-        if (p == items_.end())
-        {
-            addItem_(info, other_info);
-        }
-        else
-        {
-            qWarning() << "item with similar info hash marked as added";
-            updateItem_(p.value(), info, other_info);
-        }
-        break;
-    case TorrentState::action_update :
-        if (p != items_.end())
-        {
-            updateItem_(p.value(), info, other_info);
-        }
-        else
-        {
-            qWarning() << "item does not exist in list but information update arrived";
-        }
-        break;
-    case TorrentState::action_remove :
-        if (p != items_.end())
-        {
-            removeItem_(p.value(), info);
-        }
-        else
-        {
-            qWarning() << "item removal request arrived but there is no such item";
-        }
-        break;
-    default:
-        qWarning() << "unknown action requested: " << info.action;
-        break;
-    }
+       DownloadItems_t::iterator p = items_.find(info.hash);
+       switch (info.action) {
+       case TorrentState::action_add :
+               if (p == items_.end()) {
+                       addItem_(info, other_info);
+               } else {
+                       qWarning() << "item with similar info hash marked as added";
+                       updateItem_(p.value(), info, other_info);
+               }
+               break;
+       case TorrentState::action_update :
+               if (p != items_.end()) {
+                       updateItem_(p.value(), info, other_info);
+               } else {
+                       qWarning() << "item does not exist in list but information update arrived";
+               }
+               break;
+       case TorrentState::action_remove :
+               if (p != items_.end()) {
+                       removeItem_(p.value(), info);
+               } else {
+                       qWarning() << "item removal request arrived but there is no such item";
+               }
+               break;
+       default:
+               qWarning() << "unknown action requested: " << info.action;
+               break;
+       }
 }
 
 void DownloadView::removeItem_(DownloadViewItem *item, TorrentState const& info)
 {
-    QString hash = item->getHash();
+       QString hash = item->getHash();
 
-    int removed = items_.remove(hash);
-    if (!removed)
-        qDebug() << "Inconsistent download view state on item removal";
+       int removed = items_.remove(hash);
+       if (!removed)
+               qDebug() << "Inconsistent download view state on item removal";
 
-    int index = indexOfTopLevelItem(item);
-    if (index >= 0)
-    {
-        takeTopLevelItem(index);
-    }
+       int index = indexOfTopLevelItem(item);
+       if (index >= 0) {
+               takeTopLevelItem(index);
+       }
 }
 
 void DownloadView::addItem_(TorrentState const& info, ParamsMap_t)
 {
-    DownloadViewItem *item = new DownloadViewItem
-    ( info.hash,
-      QStringList()
-      << info.name
-      << formatSize(info.total_size)
-      << GetStatusString((TorrentStatus::Id)info.state)
-      << formatProgress(info.progress)
-      << QString::number(info.down_rate, 'f', 2)
-      << QString::number(info.up_rate, 'f', 2)
-      << QString::number(info.seeds) + "/" + QString::number(info.leeches)
-      << QString::number(info.ratio)
-      << "ETA" );
-
-    QBrush brushTmp(GetStatusColor((TorrentStatus::Id)info.state));
-    item->setForeground(2, brushTmp);
-
-    addTopLevelItem(item);
-    items_[info.hash] = item;
+       DownloadViewItem *item = new DownloadViewItem
+       ( info.hash,
+         QStringList()
+         << info.name
+         << formatSize(info.total_size)
+         << GetStatusString((TorrentStatus::Id)info.state)
+         << formatProgress(info.progress)
+         << formatSize(info.down_rate)
+         << formatSize(info.up_rate)
+         << QString::number(info.seeds) + "/" + QString::number(info.leeches)
+         << QString::number(info.ratio)
+         << "ETA" );
+
+       QBrush brushTmp(GetStatusColor((TorrentStatus::Id)info.state));
+       item->setForeground(2, brushTmp);
+
+       addTopLevelItem(item);
+       items_[info.hash] = item;
 }
 
 
 void DownloadView::updateItem_(DownloadViewItem *item
                                , TorrentState const& info, ParamsMap_t)
 {
-    item->setData(2, Qt::DisplayRole,
-                  QVariant(GetStatusString((TorrentStatus::Id)info.state)));
-    item->setData(3, Qt::DisplayRole,
-                  QVariant(formatProgress(info.progress)));
-    item->setData(4, Qt::DisplayRole,
-                  QVariant(QString::number(info.down_rate)));
-    item->setData(5, Qt::DisplayRole,
-                  QVariant(QString::number(info.up_rate)));
-    item->setData(6, Qt::DisplayRole,
-                  QString::number(info.seeds) + "/" + QString::number(info.leeches));
-
-    QBrush brushTmp(GetStatusColor((TorrentStatus::Id)info.state));
-    item->setForeground(2, brushTmp);
+       item->setData(2, Qt::DisplayRole,
+                     QVariant(GetStatusString((TorrentStatus::Id)info.state)));
+       item->setData(3, Qt::DisplayRole,
+                     QVariant(formatProgress(info.progress)));
+       item->setData(4, Qt::DisplayRole,
+                     QVariant(formatSize(info.down_rate)));
+       item->setData(5, Qt::DisplayRole,
+                     QVariant(formatSize(info.up_rate)));
+       item->setData(6, Qt::DisplayRole,
+                     QString::number(info.seeds) + "/" + QString::number(info.leeches));
+       item->setData(7, Qt::DisplayRole, QString::number(info.ratio));
+       
+       // Calculate ETA
+       if (info.down_rate > 0) {
+               qulonglong eta = (info.total_size - info.total_done) / info.down_rate;
+               item->setData(8, Qt::DisplayRole, formatElapsedTime(eta));
+       }       else {
+               item->setData(8, Qt::DisplayRole, "N/A");
+       }
+       
+       // Set color for status text
+       QBrush brushTmp(GetStatusColor((TorrentStatus::Id)info.state));
+       item->setForeground(2, brushTmp);
 }
 
 
 QString DownloadView::prepareRemoveSelected()
 {
-    qDebug() << "DownloadView::removeSelected() " <<  topLevelItemCount() ;
+       qDebug() << "DownloadView::removeSelected() " <<  topLevelItemCount() ;
 
-    DownloadViewItem *item = dynamic_cast<DownloadViewItem*> (currentItem());
-    QString hash = item->getHash();
+       DownloadViewItem *item = dynamic_cast<DownloadViewItem*> (currentItem());
+       QString hash = item->getHash();
 
-    item->setDisabled(true);
+       item->setDisabled(true);
 
-    qDebug() << "DownloadView::removeSelected() " <<  topLevelItemCount() ;
+       qDebug() << "DownloadView::removeSelected() " <<  topLevelItemCount() ;
 
-    return hash;
+       return hash;
 }
 
 
 void DownloadView::on_itemClicked(QTreeWidgetItem * , int)
 {
-    /*
-      qDebug() << "DownloadView::on_itemClicked(()" << item << "," << column;
-      qDebug() << "current item" << currentItem();
-
-      if (item == currentItem() && item->isSelected()) {
-      item->setSelected(false);
-      }
-    */
+       /*
+         qDebug() << "DownloadView::on_itemClicked(()" << item << "," << column;
+         qDebug() << "current item" << currentItem();
+
+         if (item == currentItem() && item->isSelected()) {
+         item->setSelected(false);
+         }
+       */
+}
+
+
+void DownloadView::saveView()
+{
+               QTreeWidgetItem *item = headerItem();
+               QList<QVariant> columns;
+               
+               for (int i = 0; i < item->columnCount(); ++i) {
+                       isColumnHidden(i) ? columns.push_back(QVariant(false)) : columns.push_back(QVariant(true));
+               }
+               
+       settings_.setValue("downloadview_columns", QVariant(columns));
+}
+
+
+void DownloadView::restoreView()
+{
+       QTreeWidgetItem *item = headerItem();
+       QVariant columns(settings_.value("downloadview_columns"));
+       QList<QVariant> columnList = columns.toList();
+       
+       for (int i = 0; i < columnList.size(); ++i) {
+               columnList.at(i).toBool() ? setColumnHidden(i, false) : setColumnHidden(i, true);
+       }
 }
 
 
 QString DownloadView::GetStatusString(TorrentStatus::Id status)
 {
-    switch (status)
-    {
-    case TorrentStatus::QUEUED_FOR_CHECKING :
-        return tr("Queued");
-    case TorrentStatus::CHECKING_FILES :
-        return tr("Checking");
-    case TorrentStatus::DOWNLOADING_METADATA :
-        return tr("DL meta");
-    case TorrentStatus::DOWNLOADING :
-        return tr("Downloading");
-    case TorrentStatus::FINISHED :
-        return tr("Finished");
-    case TorrentStatus::SEEDING :
-        return tr("Seeding");
-    case TorrentStatus::ALLOCATING :
-        return tr("Allocating");
-    case TorrentStatus::CHECKING_RESUME_DATA :
-        return tr("Checking resume");
-    default:
-        return tr("N/A");
-    }
+       switch (status) {
+               case TorrentStatus::QUEUED_FOR_CHECKING :
+                       return tr("Queued");
+               case TorrentStatus::CHECKING_FILES :
+                       return tr("Checking");
+               case TorrentStatus::DOWNLOADING_METADATA :
+                       return tr("DL meta");
+               case TorrentStatus::DOWNLOADING :
+                       return tr("Downloading");
+               case TorrentStatus::FINISHED :
+                       return tr("Finished");
+               case TorrentStatus::SEEDING :
+                       return tr("Seeding");
+               case TorrentStatus::ALLOCATING :
+                       return tr("Allocating");
+               case TorrentStatus::CHECKING_RESUME_DATA :
+                       return tr("Checking resume");
+               default:
+                       return tr("N/A");
+       }
 }
 
 
 QColor DownloadView::GetStatusColor(TorrentStatus::Id status)
 {
-    QColor green(40,205,40);
-    QColor yellow(255,174,0);
-
-    switch (status)
-    {
-    case TorrentStatus::QUEUED_FOR_CHECKING :
-    case TorrentStatus::CHECKING_FILES :
-    case TorrentStatus::DOWNLOADING_METADATA :
-    case TorrentStatus::ALLOCATING :
-    case TorrentStatus::CHECKING_RESUME_DATA:
-        return yellow;
-    case TorrentStatus::DOWNLOADING :
-    case TorrentStatus::FINISHED :
-    case TorrentStatus::SEEDING :
-        return green;
-    default:
-        return QColor();
-    }
+       QColor green(40,205,40);
+       QColor yellow(255,174,0);
+
+       switch (status) {
+               case TorrentStatus::QUEUED_FOR_CHECKING :
+               case TorrentStatus::CHECKING_FILES :
+               case TorrentStatus::DOWNLOADING_METADATA :
+               case TorrentStatus::ALLOCATING :
+               case TorrentStatus::CHECKING_RESUME_DATA:
+                       return yellow;
+               case TorrentStatus::DOWNLOADING :
+               case TorrentStatus::FINISHED :
+               case TorrentStatus::SEEDING :
+                       return green;
+               default:
+                       return QColor();
+       }
 }
 
 } // namespace qtrapids