- Search plugin interface changed: enum constants to define widget types to allow...
[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                 DownloadViewItem *item
91                 = new DownloadViewItem(QStringList()
92                                        << "Name"
93                                        << "Size" << "Status"
94                                        << "Progress" << "DL speed"
95                                        << "UL speed" << "Seeds/Leechers"
96                                        << "Ratio" << "ETA");
97
98                 return item;
99         }
100
101         /// @todo QTorrentHandle as one hidden column
102 };
103
104 #endif