Code formatting/indentation unified in trunk
[qtrapids] / src / gui / DownloadView.h
1 /***************************************************************************
2  *   Copyright (C) 2009 by Lassi Väätämöinen   *
3  *   lassi.vaatamoinen@ixonos.com   *
4  *                                                                         *
5  *   This program is free software; you can redistribute it and/or modify  *
6  *   it under the terms of the GNU General Public License as published by  *
7  *   the Free Software Foundation; either version 2 of the License, or     *
8  *   (at your option) any later version.                                   *
9  *                                                                         *
10  *   This program is distributed in the hope that it will be useful,       *
11  *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
12  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
13  *   GNU General Public License for more details.                          *
14  *                                                                         *
15  *   You should have received a copy of the GNU General Public License     *
16  *   along with this program; if not, write to the                         *
17  *   Free Software Foundation, Inc.,                                       *
18  *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
19  ***************************************************************************/
20 #ifndef DOWNLOADVIEW_H
21 #define DOWNLOADVIEW_H
22
23 #include <map>
24
25 #include <QTreeWidget>
26
27 #include "QBittorrentSession.h"
28
29 class DownloadViewItem;
30 class QTimer;
31
32 /**
33         @class DownloadView
34         @brief A view showing all downloaded torrents
35         @author Lassi Väätämöinen <lassi.vaatamoinen@ixonos.com>
36 */
37 class DownloadView : public QTreeWidget
38 {
39     Q_OBJECT
40
41 public:
42     DownloadView(QWidget* parent);
43
44     ~DownloadView();
45
46     void newItem(qtrapids::QTorrentHandle handle);
47     void updateItem(qtrapids::QTorrentHandle handle);
48     qtrapids::QTorrentHandle removeSelected();
49     void removeItem(qtrapids::QTorrentHandle handle);
50     void setRefreshInterval(int msec);
51
52 private slots:
53     void on_itemClicked(QTreeWidgetItem * item, int column);
54     void on_timeout();
55
56 private:
57     // Maps torrent to downloadview item.
58     // Key: SHA1 info hash of torrent. Data: View item corresponding to torrent.
59     std::map<qtrapids::QTorrentHandle, DownloadViewItem*> items_;
60     QTimer *timer_;
61
62     // Private functions.
63     QString GetStatusString(qtrapids::QTorrentHandle::State const& status) const;
64     QColor GetStatusColor(qtrapids::QTorrentHandle::State const& status) const;
65     void UpdateView();
66
67
68 };
69
70
71 /**
72         @class DownloadViewItem
73         @brief Represents one item row of DownloadView
74 */
75 class DownloadViewItem : public QTreeWidgetItem
76 {
77
78 public:
79     DownloadViewItem(QTreeWidget* parent, int type) :
80             QTreeWidgetItem(parent, type = QTreeWidgetItem::UserType) {};
81
82     DownloadViewItem(const QStringList& strings,
83                      int type = QTreeWidgetItem::UserType) :
84             QTreeWidgetItem (strings, type = Type) {};
85
86
87     /// @return An item comprising of string list, suitable for QTableView
88     /// header.
89     static DownloadViewItem *getHeaderItem()
90     {
91         DownloadViewItem *item
92         = new DownloadViewItem(QStringList()
93                                << "Name"
94                                << "Size" << "Status"
95                                << "Progress" << "DL speed"
96                                << "UL speed" << "Seeds/Leechers"
97                                << "Ratio" << "ETA");
98
99         return item;
100     }
101
102     /// @todo QTorrentHandle as one hidden column
103 };
104
105 #endif