src/gui/transferlistmodel.h

   1 /*
   2  * Bittorrent Client using Qt and libtorrent.
   3  * Copyright (C) 2015-2024  Vladimir Golovnev <glassez@yandex.ru>
   4  * Copyright (C) 2010  Christophe Dumez <chris@qbittorrent.org>
   5  *
   6  * This program is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU General Public License
   8  * as published by the Free Software Foundation; either version 2
   9  * of the License, or (at your option) any later version.
  10  *
  11  * This program is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  * GNU General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU General Public License
  17  * along with this program; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
  19  *
  20  * In addition, as a special exception, the copyright holders give permission to
  21  * link this program with the OpenSSL project's "OpenSSL" library (or with
  22  * modified versions of it that use the same license as the "OpenSSL" library),
  23  * and distribute the linked executables. You must obey the GNU General Public
  24  * License in all respects for all of the code used other than "OpenSSL".  If you
  25  * modify file(s), you may extend this exception to your version of the file(s),
  26  * but you are not obligated to do so. If you do not wish to do so, delete this
  27  * exception statement from your version.
  28  */
  29
  30 #pragma once
  31
  32 #include <QAbstractListModel>
  33 #include <QColor>
  34 #include <QHash>
  35 #include <QIcon>
  36 #include <QList>
  37
  38 #include "base/bittorrent/torrent.h"
  39
  40 namespace BitTorrent
  41 {
  42     class InfoHash;
  43 }
  44
  45 class TransferListModel final : public QAbstractListModel
  46 {
  47     Q_OBJECT
  48     Q_DISABLE_COPY_MOVE(TransferListModel)
  49
  50 public:
  51     enum Column
  52     {
  53         TR_QUEUE_POSITION,
  54         TR_NAME,
  55         TR_SIZE,
  56         TR_TOTAL_SIZE,
  57         TR_PROGRESS,
  58         TR_STATUS,
  59         TR_SEEDS,
  60         TR_PEERS,
  61         TR_DLSPEED,
  62         TR_UPSPEED,
  63         TR_ETA,
  64         TR_RATIO,
  65         TR_POPULARITY,
  66         TR_CATEGORY,
  67         TR_TAGS,
  68         TR_ADD_DATE,
  69         TR_SEED_DATE,
  70         TR_TRACKER,
  71         TR_DLLIMIT,
  72         TR_UPLIMIT,
  73         TR_AMOUNT_DOWNLOADED,
  74         TR_AMOUNT_UPLOADED,
  75         TR_AMOUNT_DOWNLOADED_SESSION,
  76         TR_AMOUNT_UPLOADED_SESSION,
  77         TR_AMOUNT_LEFT,
  78         TR_TIME_ELAPSED,
  79         TR_SAVE_PATH,
  80         TR_COMPLETED,
  81         TR_RATIO_LIMIT,
  82         TR_SEEN_COMPLETE_DATE,
  83         TR_LAST_ACTIVITY,
  84         TR_AVAILABILITY,
  85         TR_DOWNLOAD_PATH,
  86         TR_INFOHASH_V1,
  87         TR_INFOHASH_V2,
  88         TR_REANNOUNCE,
  89         TR_PRIVATE,
  90
  91         NB_COLUMNS
  92     };
  93
  94     enum DataRole
  95     {
  96         UnderlyingDataRole = Qt::UserRole,
  97         AdditionalUnderlyingDataRole
  98     };
  99
 100     explicit TransferListModel(QObject *parent = nullptr);
 101
 102     int rowCount(const QModelIndex &parent = {}) const override;
 103     int columnCount(const QModelIndex &parent = {}) const override;
 104     QVariant data(const QModelIndex &index, int role = Qt::DisplayRole) const override;
 105     bool setData(const QModelIndex &index, const QVariant &value, int role) override;
 106     QVariant headerData(int section, Qt::Orientation orientation, int role) const override;
 107     Qt::ItemFlags flags(const QModelIndex &index) const override;
 108
 109     BitTorrent::Torrent *torrentHandle(const QModelIndex &index) const;
 110
 111 private slots:
 112     void addTorrents(const QList<BitTorrent::Torrent *> &torrents);
 113     void handleTorrentAboutToBeRemoved(BitTorrent::Torrent *torrent);
 114     void handleTorrentStatusUpdated(BitTorrent::Torrent *torrent);
 115     void handleTorrentsUpdated(const QList<BitTorrent::Torrent *> &torrents);
 116
 117 private:
 118     void configure();
 119     void loadUIThemeResources();
 120     QString displayValue(const BitTorrent::Torrent *torrent, int column) const;
 121     QVariant internalValue(const BitTorrent::Torrent *torrent, int column, bool alt) const;
 122     QIcon getIconByState(BitTorrent::TorrentState state) const;
 123
 124     QList<BitTorrent::Torrent *> m_torrentList;  // maps row number to torrent handle
 125     QHash<BitTorrent::Torrent *, int> m_torrentMap;  // maps torrent handle to row number
 126     const QHash<BitTorrent::TorrentState, QString> m_statusStrings;
 127     // row text colors
 128     QHash<BitTorrent::TorrentState, QColor> m_stateThemeColors;
 129
 130     enum class HideZeroValuesMode
 131     {
 132         Never,
 133         Stopped,
 134         Always
 135     };
 136
 137     HideZeroValuesMode m_hideZeroValuesMode = HideZeroValuesMode::Never;
 138
 139     // cached icons
 140     QIcon m_checkingIcon;
 141     QIcon m_completedIcon;
 142     QIcon m_downloadingIcon;
 143     QIcon m_errorIcon;
 144     QIcon m_movingIcon;
 145     QIcon m_stoppedIcon;
 146     QIcon m_queuedIcon;
 147     QIcon m_stalledDLIcon;
 148     QIcon m_stalledUPIcon;
 149     QIcon m_uploadingIcon;
 150 };
 151
 152 Q_DECLARE_METATYPE(TransferListModel::Column)