Christophe Dumez
13 years ago
12 changed files with 1005 additions and 696 deletions
@ -0,0 +1,98 @@
@@ -0,0 +1,98 @@
|
||||
/*
|
||||
* Bittorrent Client using Qt4 and libtorrent. |
||||
* Copyright (C) 2006-2012 Christophe Dumez |
||||
* |
||||
* 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. |
||||
* |
||||
* This program is distributed in the hope that it will be useful, |
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
||||
* GNU General Public License for more details. |
||||
* |
||||
* You should have received a copy of the GNU General Public License |
||||
* along with this program; if not, write to the Free Software |
||||
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. |
||||
* |
||||
* In addition, as a special exception, the copyright holders give permission to |
||||
* link this program with the OpenSSL project's "OpenSSL" library (or with |
||||
* modified versions of it that use the same license as the "OpenSSL" library), |
||||
* and distribute the linked executables. You must obey the GNU General Public |
||||
* License in all respects for all of the code used other than "OpenSSL". If you |
||||
* modify file(s), you may extend this exception to your version of the file(s), |
||||
* but you are not obligated to do so. If you do not wish to do so, delete this |
||||
* exception statement from your version. |
||||
* |
||||
* Contact : chris@qbittorrent.org |
||||
*/ |
||||
|
||||
#include "torrentcontentfiltermodel.h" |
||||
#include "torrentcontentmodel.h" |
||||
|
||||
TorrentContentFilterModel::TorrentContentFilterModel(QObject *parent): |
||||
QSortFilterProxyModel(parent), m_model(new TorrentContentModel(this)) |
||||
{ |
||||
connect(m_model, SIGNAL(filteredFilesChanged()), this, SIGNAL(filteredFilesChanged())); |
||||
setSourceModel(m_model); |
||||
// Filter settings
|
||||
setFilterCaseSensitivity(Qt::CaseInsensitive); |
||||
setFilterKeyColumn(TorrentContentModelItem::COL_NAME); |
||||
setFilterRole(Qt::DisplayRole); |
||||
setDynamicSortFilter(true); |
||||
setSortCaseSensitivity(Qt::CaseInsensitive); |
||||
} |
||||
|
||||
TorrentContentFilterModel::~TorrentContentFilterModel() |
||||
{ |
||||
delete m_model; |
||||
} |
||||
|
||||
TorrentContentModel* TorrentContentFilterModel::model() const |
||||
{ |
||||
return m_model; |
||||
} |
||||
|
||||
TorrentContentModelItem::FileType TorrentContentFilterModel::getType(const QModelIndex& index) const |
||||
{ |
||||
return m_model->getType(mapToSource(index)); |
||||
} |
||||
|
||||
int TorrentContentFilterModel::getFileIndex(const QModelIndex& index) const |
||||
{ |
||||
return m_model->getFileIndex(mapToSource(index)); |
||||
} |
||||
|
||||
QModelIndex TorrentContentFilterModel::parent(const QModelIndex& child) const |
||||
{ |
||||
if (!child.isValid()) return QModelIndex(); |
||||
QModelIndex sourceParent = m_model->parent(mapToSource(child)); |
||||
if (!sourceParent.isValid()) return QModelIndex(); |
||||
return mapFromSource(sourceParent); |
||||
} |
||||
|
||||
bool TorrentContentFilterModel::filterAcceptsRow(int source_row, const QModelIndex& source_parent) const |
||||
{ |
||||
if (m_model->getType(m_model->index(source_row, 0, source_parent)) == TorrentContentModelItem::FOLDER) { |
||||
// always accept folders, since we want to filter their children
|
||||
return true; |
||||
} |
||||
return QSortFilterProxyModel::filterAcceptsRow(source_row, source_parent); |
||||
} |
||||
|
||||
void TorrentContentFilterModel::selectAll() |
||||
{ |
||||
for (int i=0; i<rowCount(); ++i) { |
||||
setData(index(i, 0), Qt::Checked, Qt::CheckStateRole); |
||||
} |
||||
emit dataChanged(index(0,0), index(rowCount(), columnCount())); |
||||
} |
||||
|
||||
void TorrentContentFilterModel::selectNone() |
||||
{ |
||||
for (int i=0; i<rowCount(); ++i) { |
||||
setData(index(i, 0), Qt::Unchecked, Qt::CheckStateRole); |
||||
} |
||||
emit dataChanged(index(0,0), index(rowCount(), columnCount())); |
||||
} |
@ -0,0 +1,65 @@
@@ -0,0 +1,65 @@
|
||||
/*
|
||||
* Bittorrent Client using Qt4 and libtorrent. |
||||
* Copyright (C) 2006-2012 Christophe Dumez |
||||
* |
||||
* 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. |
||||
* |
||||
* This program is distributed in the hope that it will be useful, |
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
||||
* GNU General Public License for more details. |
||||
* |
||||
* You should have received a copy of the GNU General Public License |
||||
* along with this program; if not, write to the Free Software |
||||
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. |
||||
* |
||||
* In addition, as a special exception, the copyright holders give permission to |
||||
* link this program with the OpenSSL project's "OpenSSL" library (or with |
||||
* modified versions of it that use the same license as the "OpenSSL" library), |
||||
* and distribute the linked executables. You must obey the GNU General Public |
||||
* License in all respects for all of the code used other than "OpenSSL". If you |
||||
* modify file(s), you may extend this exception to your version of the file(s), |
||||
* but you are not obligated to do so. If you do not wish to do so, delete this |
||||
* exception statement from your version. |
||||
* |
||||
* Contact : chris@qbittorrent.org |
||||
*/ |
||||
|
||||
#ifndef TORRENTCONTENTFILTERMODEL_H |
||||
#define TORRENTCONTENTFILTERMODEL_H |
||||
|
||||
#include <QSortFilterProxyModel> |
||||
#include "torrentcontentmodelitem.h" |
||||
|
||||
class TorrentContentModel; |
||||
|
||||
class TorrentContentFilterModel: public QSortFilterProxyModel { |
||||
Q_OBJECT |
||||
|
||||
public: |
||||
TorrentContentFilterModel(QObject *parent = 0); |
||||
virtual ~TorrentContentFilterModel(); |
||||
|
||||
TorrentContentModel* model() const; |
||||
TorrentContentModelItem::FileType getType(const QModelIndex& index) const; |
||||
int getFileIndex(const QModelIndex& index) const; |
||||
virtual QModelIndex parent(const QModelIndex& child) const; |
||||
|
||||
signals: |
||||
void filteredFilesChanged(); |
||||
|
||||
protected: |
||||
virtual bool filterAcceptsRow(int source_row, const QModelIndex& source_parent) const; |
||||
|
||||
public slots: |
||||
void selectAll(); |
||||
void selectNone(); |
||||
|
||||
private: |
||||
TorrentContentModel *m_model; |
||||
}; |
||||
|
||||
#endif // TORRENTCONTENTFILTERMODEL_H
|
@ -0,0 +1,325 @@
@@ -0,0 +1,325 @@
|
||||
/*
|
||||
* Bittorrent Client using Qt4 and libtorrent. |
||||
* Copyright (C) 2006-2012 Christophe Dumez |
||||
* |
||||
* 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. |
||||
* |
||||
* This program is distributed in the hope that it will be useful, |
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
||||
* GNU General Public License for more details. |
||||
* |
||||
* You should have received a copy of the GNU General Public License |
||||
* along with this program; if not, write to the Free Software |
||||
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. |
||||
* |
||||
* In addition, as a special exception, the copyright holders give permission to |
||||
* link this program with the OpenSSL project's "OpenSSL" library (or with |
||||
* modified versions of it that use the same license as the "OpenSSL" library), |
||||
* and distribute the linked executables. You must obey the GNU General Public |
||||
* License in all respects for all of the code used other than "OpenSSL". If you |
||||
* modify file(s), you may extend this exception to your version of the file(s), |
||||
* but you are not obligated to do so. If you do not wish to do so, delete this |
||||
* exception statement from your version. |
||||
* |
||||
* Contact : chris@qbittorrent.org |
||||
*/ |
||||
|
||||
#include "iconprovider.h" |
||||
#include "misc.h" |
||||
#include "torrentcontentmodel.h" |
||||
#include "torrentcontentmodelitem.h" |
||||
#include <QDir> |
||||
|
||||
TorrentContentModel::TorrentContentModel(QObject *parent): |
||||
QAbstractItemModel(parent), files_index(0) |
||||
{ |
||||
QList<QVariant> rootData; |
||||
rootData << tr("Name") << tr("Size") << tr("Progress") << tr("Priority"); |
||||
rootItem = new TorrentContentModelItem(rootData); |
||||
} |
||||
|
||||
TorrentContentModel::~TorrentContentModel() |
||||
{ |
||||
if (files_index) |
||||
delete [] files_index; |
||||
delete rootItem; |
||||
} |
||||
|
||||
void TorrentContentModel::updateFilesProgress(const std::vector<libtorrent::size_type>& fp) |
||||
{ |
||||
emit layoutAboutToBeChanged(); |
||||
for (unsigned int i=0; i<fp.size(); ++i) { |
||||
Q_ASSERT(fp[i] >= 0); |
||||
files_index[i]->setProgress(fp[i]); |
||||
} |
||||
emit dataChanged(index(0,0), index(rowCount(), columnCount())); |
||||
} |
||||
|
||||
void TorrentContentModel::updateFilesPriorities(const std::vector<int> &fprio) |
||||
{ |
||||
emit layoutAboutToBeChanged(); |
||||
for (unsigned int i=0; i<fprio.size(); ++i) { |
||||
//qDebug("Called updateFilesPriorities with %d", fprio[i]);
|
||||
files_index[i]->setPriority(fprio[i]); |
||||
} |
||||
emit dataChanged(index(0,0), index(rowCount(), columnCount())); |
||||
} |
||||
|
||||
std::vector<int> TorrentContentModel::getFilesPriorities(unsigned int nbFiles) const |
||||
{ |
||||
std::vector<int> prio; |
||||
for (unsigned int i=0; i<nbFiles; ++i) { |
||||
//qDebug("Called getFilesPriorities: %d", files_index[i]->getPriority());
|
||||
prio.push_back(files_index[i]->getPriority()); |
||||
} |
||||
return prio; |
||||
} |
||||
|
||||
bool TorrentContentModel::allFiltered() const |
||||
{ |
||||
for (int i=0; i<rootItem->childCount(); ++i) { |
||||
if (rootItem->child(i)->getPriority() != prio::IGNORED) |
||||
return false; |
||||
} |
||||
return true; |
||||
} |
||||
|
||||
int TorrentContentModel::columnCount(const QModelIndex& parent) const |
||||
{ |
||||
if (parent.isValid()) |
||||
return static_cast<TorrentContentModelItem*>(parent.internalPointer())->columnCount(); |
||||
else |
||||
return rootItem->columnCount(); |
||||
} |
||||
|
||||
bool TorrentContentModel::setData(const QModelIndex& index, const QVariant& value, int role) |
||||
{ |
||||
if (!index.isValid()) |
||||
return false; |
||||
|
||||
if (index.column() == 0 && role == Qt::CheckStateRole) { |
||||
TorrentContentModelItem *item = static_cast<TorrentContentModelItem*>(index.internalPointer()); |
||||
qDebug("setData(%s, %d", qPrintable(item->getName()), value.toInt()); |
||||
if (item->getPriority() != value.toInt()) { |
||||
if (value.toInt() == Qt::PartiallyChecked) |
||||
item->setPriority(prio::PARTIAL); |
||||
else if (value.toInt() == Qt::Unchecked) |
||||
item->setPriority(prio::IGNORED); |
||||
else |
||||
item->setPriority(prio::NORMAL); |
||||
emit dataChanged(this->index(0,0), this->index(rowCount(), columnCount())); |
||||
emit filteredFilesChanged(); |
||||
} |
||||
return true; |
||||
} |
||||
|
||||
if (role == Qt::EditRole) { |
||||
TorrentContentModelItem *item = static_cast<TorrentContentModelItem*>(index.internalPointer()); |
||||
switch(index.column()) { |
||||
case TorrentContentModelItem::COL_NAME: |
||||
item->setName(value.toString()); |
||||
break; |
||||
case TorrentContentModelItem::COL_SIZE: |
||||
item->setSize(value.toULongLong()); |
||||
break; |
||||
case TorrentContentModelItem::COL_PROGRESS: |
||||
item->setProgress(value.toDouble()); |
||||
break; |
||||
case TorrentContentModelItem::COL_PRIO: |
||||
item->setPriority(value.toInt()); |
||||
break; |
||||
default: |
||||
return false; |
||||
} |
||||
emit dataChanged(index, index); |
||||
return true; |
||||
} |
||||
return false; |
||||
} |
||||
|
||||
TorrentContentModelItem::FileType TorrentContentModel::getType(const QModelIndex& index) const |
||||
{ |
||||
const TorrentContentModelItem *item = static_cast<const TorrentContentModelItem*>(index.internalPointer()); |
||||
return item->getType(); |
||||
} |
||||
|
||||
int TorrentContentModel::getFileIndex(const QModelIndex& index) |
||||
{ |
||||
TorrentContentModelItem *item = static_cast<TorrentContentModelItem*>(index.internalPointer()); |
||||
return item->getFileIndex(); |
||||
} |
||||
|
||||
QVariant TorrentContentModel::data(const QModelIndex &index, int role) const |
||||
{ |
||||
if (!index.isValid()) |
||||
return QVariant(); |
||||
|
||||
TorrentContentModelItem *item = static_cast<TorrentContentModelItem*>(index.internalPointer()); |
||||
if (index.column() == 0 && role == Qt::DecorationRole) { |
||||
if (item->isFolder()) |
||||
return IconProvider::instance()->getIcon("inode-directory"); |
||||
else |
||||
return IconProvider::instance()->getIcon("text-plain"); |
||||
} |
||||
if (index.column() == 0 && role == Qt::CheckStateRole) { |
||||
if (item->data(TorrentContentModelItem::COL_PRIO).toInt() == prio::IGNORED) |
||||
return Qt::Unchecked; |
||||
if (item->data(TorrentContentModelItem::COL_PRIO).toInt() == prio::PARTIAL) |
||||
return Qt::PartiallyChecked; |
||||
return Qt::Checked; |
||||
} |
||||
if (role != Qt::DisplayRole) |
||||
return QVariant(); |
||||
|
||||
return item->data(index.column()); |
||||
} |
||||
|
||||
Qt::ItemFlags TorrentContentModel::flags(const QModelIndex& index) const |
||||
{ |
||||
if (!index.isValid()) |
||||
return 0; |
||||
|
||||
if (getType(index) == TorrentContentModelItem::FOLDER) |
||||
return Qt::ItemIsEditable | Qt::ItemIsEnabled | Qt::ItemIsSelectable | Qt::ItemIsUserCheckable | Qt::ItemIsTristate; |
||||
return Qt::ItemIsEditable | Qt::ItemIsEnabled | Qt::ItemIsSelectable | Qt::ItemIsUserCheckable; |
||||
} |
||||
|
||||
QVariant TorrentContentModel::headerData(int section, Qt::Orientation orientation, int role) const |
||||
{ |
||||
if (orientation == Qt::Horizontal && role == Qt::DisplayRole) |
||||
return rootItem->data(section); |
||||
|
||||
return QVariant(); |
||||
} |
||||
|
||||
QModelIndex TorrentContentModel::index(int row, int column, const QModelIndex& parent) const |
||||
{ |
||||
if (parent.isValid() && parent.column() != 0) |
||||
return QModelIndex(); |
||||
|
||||
if (column >= TorrentContentModelItem::NB_COL) |
||||
return QModelIndex(); |
||||
|
||||
TorrentContentModelItem *parentItem; |
||||
|
||||
if (!parent.isValid()) |
||||
parentItem = rootItem; |
||||
else |
||||
parentItem = static_cast<TorrentContentModelItem*>(parent.internalPointer()); |
||||
Q_ASSERT(parentItem); |
||||
if (row >= parentItem->childCount()) |
||||
return QModelIndex(); |
||||
|
||||
TorrentContentModelItem *childItem = parentItem->child(row); |
||||
if (childItem) { |
||||
return createIndex(row, column, childItem); |
||||
} else { |
||||
return QModelIndex(); |
||||
} |
||||
} |
||||
|
||||
QModelIndex TorrentContentModel::parent(const QModelIndex& index) const |
||||
{ |
||||
if (!index.isValid()) |
||||
return QModelIndex(); |
||||
|
||||
TorrentContentModelItem *childItem = static_cast<TorrentContentModelItem*>(index.internalPointer()); |
||||
if (!childItem) return QModelIndex(); |
||||
TorrentContentModelItem *parentItem = childItem->parent(); |
||||
|
||||
if (parentItem == rootItem) |
||||
return QModelIndex(); |
||||
|
||||
return createIndex(parentItem->row(), 0, parentItem); |
||||
} |
||||
|
||||
int TorrentContentModel::rowCount(const QModelIndex& parent) const |
||||
{ |
||||
TorrentContentModelItem *parentItem; |
||||
|
||||
if (parent.column() > 0) |
||||
return 0; |
||||
|
||||
if (!parent.isValid()) |
||||
parentItem = rootItem; |
||||
else |
||||
parentItem = static_cast<TorrentContentModelItem*>(parent.internalPointer()); |
||||
|
||||
return parentItem->childCount(); |
||||
} |
||||
|
||||
void TorrentContentModel::clear() |
||||
{ |
||||
qDebug("clear called"); |
||||
beginResetModel(); |
||||
if (files_index) { |
||||
delete [] files_index; |
||||
files_index = 0; |
||||
} |
||||
rootItem->deleteAllChildren(); |
||||
endResetModel(); |
||||
} |
||||
|
||||
void TorrentContentModel::setupModelData(const libtorrent::torrent_info &t) |
||||
{ |
||||
qDebug("setup model data called"); |
||||
if (t.num_files() == 0) |
||||
return; |
||||
|
||||
emit layoutAboutToBeChanged(); |
||||
// Initialize files_index array
|
||||
qDebug("Torrent contains %d files", t.num_files()); |
||||
files_index = new TorrentContentModelItem*[t.num_files()]; |
||||
|
||||
TorrentContentModelItem *parent = this->rootItem; |
||||
TorrentContentModelItem *root_folder = parent; |
||||
TorrentContentModelItem *current_parent; |
||||
|
||||
// Iterate over files
|
||||
for (int i=0; i<t.num_files(); ++i) { |
||||
libtorrent::file_entry fentry = t.file_at(i); |
||||
current_parent = root_folder; |
||||
#if LIBTORRENT_VERSION_MINOR >= 16 |
||||
QString path = QDir::cleanPath(misc::toQStringU(fentry.path)).replace("\\", "/"); |
||||
#else |
||||
QString path = QDir::cleanPath(misc::toQStringU(fentry.path.string())).replace("\\", "/"); |
||||
#endif |
||||
// Iterate of parts of the path to create necessary folders
|
||||
QStringList pathFolders = path.split("/"); |
||||
pathFolders.removeAll(".unwanted"); |
||||
pathFolders.takeLast(); |
||||
foreach (const QString &pathPart, pathFolders) { |
||||
TorrentContentModelItem *new_parent = current_parent->childWithName(pathPart); |
||||
if (!new_parent) { |
||||
new_parent = new TorrentContentModelItem(pathPart, current_parent); |
||||
} |
||||
current_parent = new_parent; |
||||
} |
||||
// Actually create the file
|
||||
TorrentContentModelItem *f = new TorrentContentModelItem(t, fentry, current_parent, i); |
||||
files_index[i] = f; |
||||
} |
||||
emit layoutChanged(); |
||||
} |
||||
|
||||
void TorrentContentModel::selectAll() |
||||
{ |
||||
for (int i=0; i<rootItem->childCount(); ++i) { |
||||
TorrentContentModelItem *child = rootItem->child(i); |
||||
if (child->getPriority() == prio::IGNORED) |
||||
child->setPriority(prio::NORMAL); |
||||
} |
||||
emit dataChanged(index(0,0), index(rowCount(), columnCount())); |
||||
} |
||||
|
||||
void TorrentContentModel::selectNone() |
||||
{ |
||||
for (int i=0; i<rootItem->childCount(); ++i) { |
||||
rootItem->child(i)->setPriority(prio::IGNORED); |
||||
} |
||||
emit dataChanged(index(0,0), index(rowCount(), columnCount())); |
||||
} |
@ -0,0 +1,76 @@
@@ -0,0 +1,76 @@
|
||||
/*
|
||||
* Bittorrent Client using Qt4 and libtorrent. |
||||
* Copyright (C) 2006-2012 Christophe Dumez |
||||
* |
||||
* 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. |
||||
* |
||||
* This program is distributed in the hope that it will be useful, |
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
||||
* GNU General Public License for more details. |
||||
* |
||||
* You should have received a copy of the GNU General Public License |
||||
* along with this program; if not, write to the Free Software |
||||
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. |
||||
* |
||||
* In addition, as a special exception, the copyright holders give permission to |
||||
* link this program with the OpenSSL project's "OpenSSL" library (or with |
||||
* modified versions of it that use the same license as the "OpenSSL" library), |
||||
* and distribute the linked executables. You must obey the GNU General Public |
||||
* License in all respects for all of the code used other than "OpenSSL". If you |
||||
* modify file(s), you may extend this exception to your version of the file(s), |
||||
* but you are not obligated to do so. If you do not wish to do so, delete this |
||||
* exception statement from your version. |
||||
* |
||||
* Contact : chris@qbittorrent.org |
||||
*/ |
||||
|
||||
#ifndef TORRENTCONTENTMODEL_H |
||||
#define TORRENTCONTENTMODEL_H |
||||
|
||||
#include <QAbstractItemModel> |
||||
#include <QModelIndex> |
||||
#include <QVariant> |
||||
#include <libtorrent/torrent_info.hpp> |
||||
#include "torrentcontentmodelitem.h" |
||||
|
||||
class TorrentContentModel: public QAbstractItemModel { |
||||
Q_OBJECT |
||||
|
||||
private: |
||||
TorrentContentModelItem *rootItem; |
||||
TorrentContentModelItem **files_index; |
||||
|
||||
public: |
||||
TorrentContentModel(QObject *parent = 0); |
||||
~TorrentContentModel(); |
||||
|
||||
void updateFilesProgress(const std::vector<libtorrent::size_type>& fp); |
||||
void updateFilesPriorities(const std::vector<int> &fprio); |
||||
std::vector<int> getFilesPriorities(unsigned int nbFiles) const; |
||||
bool allFiltered() const; |
||||
virtual int columnCount(const QModelIndex &parent=QModelIndex()) const; |
||||
virtual bool setData(const QModelIndex& index, const QVariant& value, int role = Qt::EditRole); |
||||
TorrentContentModelItem::FileType getType(const QModelIndex& index) const; |
||||
int getFileIndex(const QModelIndex& index); |
||||
virtual QVariant data(const QModelIndex& index, int role = Qt::DisplayRole) const; |
||||
virtual Qt::ItemFlags flags(const QModelIndex& index) const; |
||||
virtual QVariant headerData(int section, Qt::Orientation orientation, int role) const; |
||||
virtual QModelIndex index(int row, int column, const QModelIndex& parent = QModelIndex()) const; |
||||
virtual QModelIndex parent(const QModelIndex& index) const; |
||||
virtual int rowCount(const QModelIndex& parent = QModelIndex()) const; |
||||
void clear(); |
||||
void setupModelData(const libtorrent::torrent_info& t); |
||||
|
||||
signals: |
||||
void filteredFilesChanged(); |
||||
|
||||
public slots: |
||||
void selectAll(); |
||||
void selectNone(); |
||||
}; |
||||
|
||||
#endif // TORRENTCONTENTMODEL_H
|
@ -0,0 +1,320 @@
@@ -0,0 +1,320 @@
|
||||
/*
|
||||
* Bittorrent Client using Qt4 and libtorrent. |
||||
* Copyright (C) 2006-2012 Christophe Dumez |
||||
* |
||||
* 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. |
||||
* |
||||
* This program is distributed in the hope that it will be useful, |
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
||||
* GNU General Public License for more details. |
||||
* |
||||
* You should have received a copy of the GNU General Public License |
||||
* along with this program; if not, write to the Free Software |
||||
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. |
||||
* |
||||
* In addition, as a special exception, the copyright holders give permission to |
||||
* link this program with the OpenSSL project's "OpenSSL" library (or with |
||||
* modified versions of it that use the same license as the "OpenSSL" library), |
||||
* and distribute the linked executables. You must obey the GNU General Public |
||||
* License in all respects for all of the code used other than "OpenSSL". If you |
||||
* modify file(s), you may extend this exception to your version of the file(s), |
||||
* but you are not obligated to do so. If you do not wish to do so, delete this |
||||
* exception statement from your version. |
||||
* |
||||
* Contact : chris@qbittorrent.org |
||||
*/ |
||||
|
||||
#include "misc.h" |
||||
#include "torrentcontentmodelitem.h" |
||||
#include <QDebug> |
||||
|
||||
TorrentContentModelItem::TorrentContentModelItem(const libtorrent::torrent_info &t, |
||||
const libtorrent::file_entry &f, |
||||
TorrentContentModelItem *parent, |
||||
int file_index): |
||||
m_parentItem(parent), m_type(TFILE), m_fileIndex(file_index), m_totalDone(0) |
||||
{ |
||||
Q_ASSERT(parent); |
||||
|
||||
#if LIBTORRENT_VERSION_MINOR >= 16 |
||||
QString name = misc::fileName(misc::toQStringU(t.files().file_path(f))); |
||||
#else |
||||
Q_UNUSED(t); |
||||
QString name = misc::toQStringU(f.path.filename()); |
||||
#endif |
||||
// Do not display incomplete extensions
|
||||
if (name.endsWith(".!qB")) |
||||
name.chop(4); |
||||
|
||||
m_itemData << name |
||||
<< QVariant((qulonglong)f.size) |
||||
<< 0. |
||||
<< prio::NORMAL; |
||||
|
||||
/* Update parent */ |
||||
m_parentItem->appendChild(this); |
||||
m_parentItem->updateSize(); |
||||
} |
||||
|
||||
TorrentContentModelItem::TorrentContentModelItem(QString name, TorrentContentModelItem *parent): |
||||
m_parentItem(parent), m_type(FOLDER), m_totalDone(0) |
||||
{ |
||||
// Do not display incomplete extensions
|
||||
if (name.endsWith(".!qB")) |
||||
name.chop(4); |
||||
m_itemData << name |
||||
<< 0. |
||||
<< 0. |
||||
<< prio::NORMAL; |
||||
|
||||
/* Update parent */ |
||||
m_parentItem->appendChild(this); |
||||
} |
||||
|
||||
TorrentContentModelItem::TorrentContentModelItem(const QList<QVariant>& data): |
||||
m_parentItem(0), m_type(ROOT), m_itemData(data), m_totalDone(0) |
||||
{ |
||||
Q_ASSERT(data.size() == 4); |
||||
} |
||||
|
||||
TorrentContentModelItem::~TorrentContentModelItem() |
||||
{ |
||||
qDeleteAll(m_childItems); |
||||
} |
||||
|
||||
TorrentContentModelItem::FileType TorrentContentModelItem::getType() const |
||||
{ |
||||
return m_type; |
||||
} |
||||
|
||||
int TorrentContentModelItem::getFileIndex() const |
||||
{ |
||||
Q_ASSERT(m_type == TFILE); |
||||
return m_fileIndex; |
||||
} |
||||
|
||||
void TorrentContentModelItem::deleteAllChildren() |
||||
{ |
||||
Q_ASSERT(m_type == ROOT); |
||||
qDeleteAll(m_childItems); |
||||
m_childItems.clear(); |
||||
} |
||||
|
||||
const QList<TorrentContentModelItem*>& TorrentContentModelItem::children() const |
||||
{ |
||||
return m_childItems; |
||||
} |
||||
|
||||
QString TorrentContentModelItem::getName() const |
||||
{ |
||||
return m_itemData.at(COL_NAME).toString(); |
||||
} |
||||
|
||||
void TorrentContentModelItem::setName(const QString& name) |
||||
{ |
||||
Q_ASSERT(m_type != ROOT); |
||||
m_itemData.replace(COL_NAME, name); |
||||
} |
||||
|
||||
qulonglong TorrentContentModelItem::getSize() const |
||||
{ |
||||
return m_itemData.value(COL_SIZE).toULongLong(); |
||||
} |
||||
|
||||
void TorrentContentModelItem::setSize(qulonglong size) |
||||
{ |
||||
Q_ASSERT (m_type != ROOT); |
||||
if (getSize() == size) |
||||
return; |
||||
m_itemData.replace(COL_SIZE, (qulonglong)size); |
||||
m_parentItem->updateSize(); |
||||
} |
||||
|
||||
void TorrentContentModelItem::updateSize() |
||||
{ |
||||
if (m_type == ROOT) |
||||
return; |
||||
Q_ASSERT(m_type == FOLDER); |
||||
qulonglong size = 0; |
||||
foreach (TorrentContentModelItem* child, m_childItems) { |
||||
if (child->getPriority() != prio::IGNORED) |
||||
size += child->getSize(); |
||||
} |
||||
setSize(size); |
||||
} |
||||
|
||||
void TorrentContentModelItem::setProgress(qulonglong done) |
||||
{ |
||||
Q_ASSERT (m_type != ROOT); |
||||
if (getPriority() == 0) return; |
||||
m_totalDone = done; |
||||
qulonglong size = getSize(); |
||||
Q_ASSERT(m_totalDone <= size); |
||||
qreal progress; |
||||
if (size > 0) |
||||
progress = m_totalDone/(float)size; |
||||
else |
||||
progress = 1.; |
||||
Q_ASSERT(progress >= 0. && progress <= 1.); |
||||
m_itemData.replace(COL_PROGRESS, progress); |
||||
m_parentItem->updateProgress(); |
||||
} |
||||
|
||||
qulonglong TorrentContentModelItem::getTotalDone() const |
||||
{ |
||||
return m_totalDone; |
||||
} |
||||
|
||||
float TorrentContentModelItem::getProgress() const |
||||
{ |
||||
Q_ASSERT (m_type != ROOT); |
||||
if (getPriority() == 0) |
||||
return -1; |
||||
qulonglong size = getSize(); |
||||
if (size > 0) |
||||
return m_totalDone / (float) getSize(); |
||||
return 1.; |
||||
} |
||||
|
||||
void TorrentContentModelItem::updateProgress() |
||||
{ |
||||
if (m_type == ROOT) return; |
||||
Q_ASSERT(m_type == FOLDER); |
||||
m_totalDone = 0; |
||||
foreach (TorrentContentModelItem* child, m_childItems) { |
||||
if (child->getPriority() > 0) |
||||
m_totalDone += child->getTotalDone(); |
||||
} |
||||
//qDebug("Folder: total_done: %llu/%llu", total_done, getSize());
|
||||
Q_ASSERT(m_totalDone <= getSize()); |
||||
setProgress(m_totalDone); |
||||
} |
||||
|
||||
int TorrentContentModelItem::getPriority() const |
||||
{ |
||||
return m_itemData.value(COL_PRIO).toInt(); |
||||
} |
||||
|
||||
void TorrentContentModelItem::setPriority(int new_prio, bool update_parent) |
||||
{ |
||||
Q_ASSERT(new_prio != prio::PARTIAL || m_type == FOLDER); // PARTIAL only applies to folders
|
||||
const int old_prio = getPriority(); |
||||
if (old_prio == new_prio) return; |
||||
qDebug("setPriority(%s, %d)", qPrintable(getName()), new_prio); |
||||
// Reset progress if priority is 0
|
||||
if (new_prio == 0) { |
||||
setProgress(0); |
||||
} |
||||
|
||||
m_itemData.replace(COL_PRIO, new_prio); |
||||
|
||||
// Update parent
|
||||
if (update_parent && m_parentItem) { |
||||
qDebug("Updating parent item"); |
||||
m_parentItem->updateSize(); |
||||
m_parentItem->updateProgress(); |
||||
m_parentItem->updatePriority(); |
||||
} |
||||
|
||||
// Update children
|
||||
if (new_prio != prio::PARTIAL && !m_childItems.empty()) { |
||||
qDebug("Updating children items"); |
||||
foreach (TorrentContentModelItem* child, m_childItems) { |
||||
// Do not update the parent since
|
||||
// the parent is causing the update
|
||||
child->setPriority(new_prio, false); |
||||
} |
||||
} |
||||
if (m_type == FOLDER) { |
||||
updateSize(); |
||||
updateProgress(); |
||||
} |
||||
} |
||||
|
||||
// Only non-root folders use this function
|
||||
void TorrentContentModelItem::updatePriority() |
||||
{ |
||||
if (m_type == ROOT) return; |
||||
Q_ASSERT(m_type == FOLDER); |
||||
if (m_childItems.isEmpty()) return; |
||||
// If all children have the same priority
|
||||
// then the folder should have the same
|
||||
// priority
|
||||
const int prio = m_childItems.first()->getPriority(); |
||||
for (int i=1; i<m_childItems.size(); ++i) { |
||||
if (m_childItems.at(i)->getPriority() != prio) { |
||||
setPriority(prio::PARTIAL); |
||||
return; |
||||
} |
||||
} |
||||
// All child items have the same priorrity
|
||||
// Update mine if necessary
|
||||
if (prio != getPriority()) |
||||
setPriority(prio); |
||||
} |
||||
|
||||
TorrentContentModelItem* TorrentContentModelItem::childWithName(const QString& name) const |
||||
{ |
||||
foreach (TorrentContentModelItem *child, m_childItems) { |
||||
if (child->getName() == name) |
||||
return child; |
||||
} |
||||
return 0; |
||||
} |
||||
|
||||
bool TorrentContentModelItem::isFolder() const |
||||
{ |
||||
return (m_type==FOLDER); |
||||
} |
||||
|
||||
void TorrentContentModelItem::appendChild(TorrentContentModelItem *item) |
||||
{ |
||||
Q_ASSERT(item); |
||||
Q_ASSERT(m_type != TFILE); |
||||
int i=0; |
||||
for (i=0; i<m_childItems.size(); ++i) { |
||||
QString newchild_name = item->getName(); |
||||
if (QString::localeAwareCompare(newchild_name, m_childItems.at(i)->getName()) < 0) |
||||
break; |
||||
} |
||||
m_childItems.insert(i, item); |
||||
} |
||||
|
||||
TorrentContentModelItem* TorrentContentModelItem::child(int row) |
||||
{ |
||||
//Q_ASSERT(row >= 0 && row < childItems.size());
|
||||
return m_childItems.value(row, 0); |
||||
} |
||||
|
||||
int TorrentContentModelItem::childCount() const |
||||
{ |
||||
return m_childItems.count(); |
||||
} |
||||
|
||||
int TorrentContentModelItem::columnCount() const |
||||
{ |
||||
return m_itemData.count(); |
||||
} |
||||
|
||||
QVariant TorrentContentModelItem::data(int column) const |
||||
{ |
||||
if (column == COL_PROGRESS && m_type != ROOT) |
||||
return getProgress(); |
||||
return m_itemData.value(column); |
||||
} |
||||
|
||||
int TorrentContentModelItem::row() const |
||||
{ |
||||
if (m_parentItem) |
||||
return m_parentItem->children().indexOf(const_cast<TorrentContentModelItem*>(this)); |
||||
return 0; |
||||
} |
||||
|
||||
TorrentContentModelItem* TorrentContentModelItem::parent() |
||||
{ |
||||
return m_parentItem; |
||||
} |
@ -0,0 +1,102 @@
@@ -0,0 +1,102 @@
|
||||
/*
|
||||
* Bittorrent Client using Qt4 and libtorrent. |
||||
* Copyright (C) 2006-2012 Christophe Dumez |
||||
* |
||||
* 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. |
||||
* |
||||
* This program is distributed in the hope that it will be useful, |
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
||||
* GNU General Public License for more details. |
||||
* |
||||
* You should have received a copy of the GNU General Public License |
||||
* along with this program; if not, write to the Free Software |
||||
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. |
||||
* |
||||
* In addition, as a special exception, the copyright holders give permission to |
||||
* link this program with the OpenSSL project's "OpenSSL" library (or with |
||||
* modified versions of it that use the same license as the "OpenSSL" library), |
||||
* and distribute the linked executables. You must obey the GNU General Public |
||||
* License in all respects for all of the code used other than "OpenSSL". If you |
||||
* modify file(s), you may extend this exception to your version of the file(s), |
||||
* but you are not obligated to do so. If you do not wish to do so, delete this |
||||
* exception statement from your version. |
||||
* |
||||
* Contact : chris@qbittorrent.org |
||||
*/ |
||||
|
||||
#ifndef TORRENTCONTENTMODELITEM_H |
||||
#define TORRENTCONTENTMODELITEM_H |
||||
|
||||
#include <QList> |
||||
#include <QVariant> |
||||
#include <libtorrent/torrent_info.hpp> |
||||
|
||||
namespace prio { |
||||
enum FilePriority {IGNORED=0, NORMAL=1, HIGH=2, MAXIMUM=7, PARTIAL=-1}; |
||||
} |
||||
|
||||
class TorrentContentModelItem { |
||||
public: |
||||
enum TreeItemColumns {COL_NAME, COL_SIZE, COL_PROGRESS, COL_PRIO, NB_COL}; |
||||
enum FileType {TFILE, FOLDER, ROOT}; |
||||
|
||||
// File Construction
|
||||
TorrentContentModelItem(const libtorrent::torrent_info &t, |
||||
const libtorrent::file_entry &f, |
||||
TorrentContentModelItem *parent, |
||||
int file_index); |
||||
// Folder constructor
|
||||
TorrentContentModelItem(QString name, TorrentContentModelItem *parent = 0); |
||||
// Invisible root item constructor
|
||||
TorrentContentModelItem(const QList<QVariant>& data); |
||||
|
||||
~TorrentContentModelItem(); |
||||
|
||||
FileType getType() const; |
||||
|
||||
int getFileIndex() const; |
||||
|
||||
QString getName() const; |
||||
void setName(const QString& name); |
||||
|
||||
qulonglong getSize() const; |
||||
void setSize(qulonglong size); |
||||
void updateSize(); |
||||
qulonglong getTotalDone() const; |
||||
|
||||
void setProgress(qulonglong done); |
||||
float getProgress() const; |
||||
void updateProgress(); |
||||
|
||||
int getPriority() const; |
||||
void setPriority(int new_prio, bool update_parent=true); |
||||
void updatePriority(); |
||||
|
||||
TorrentContentModelItem* childWithName(const QString& name) const; |
||||
bool isFolder() const; |
||||
|
||||
void appendChild(TorrentContentModelItem *item); |
||||
TorrentContentModelItem *child(int row); |
||||
int childCount() const; |
||||
int columnCount() const; |
||||
QVariant data(int column) const; |
||||
int row() const; |
||||
|
||||
TorrentContentModelItem *parent(); |
||||
void deleteAllChildren(); |
||||
const QList<TorrentContentModelItem*>& children() const; |
||||
|
||||
private: |
||||
TorrentContentModelItem *m_parentItem; |
||||
FileType m_type; |
||||
QList<TorrentContentModelItem*> m_childItems; |
||||
QList<QVariant> m_itemData; |
||||
int m_fileIndex; |
||||
qulonglong m_totalDone; |
||||
}; |
||||
|
||||
#endif // TORRENTCONTENTMODELITEM_H
|
@ -1,684 +0,0 @@
@@ -1,684 +0,0 @@
|
||||
/*
|
||||
* Bittorrent Client using Qt4 and libtorrent. |
||||
* Copyright (C) 2006 Christophe Dumez |
||||
* |
||||
* 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. |
||||
* |
||||
* This program is distributed in the hope that it will be useful, |
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
||||
* GNU General Public License for more details. |
||||
* |
||||
* You should have received a copy of the GNU General Public License |
||||
* along with this program; if not, write to the Free Software |
||||
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. |
||||
* |
||||
* In addition, as a special exception, the copyright holders give permission to |
||||
* link this program with the OpenSSL project's "OpenSSL" library (or with |
||||
* modified versions of it that use the same license as the "OpenSSL" library), |
||||
* and distribute the linked executables. You must obey the GNU General Public |
||||
* License in all respects for all of the code used other than "OpenSSL". If you |
||||
* modify file(s), you may extend this exception to your version of the file(s), |
||||
* but you are not obligated to do so. If you do not wish to do so, delete this |
||||
* exception statement from your version. |
||||
* |
||||
* Contact : chris@qbittorrent.org |
||||
*/ |
||||
|
||||
#ifndef TORRENTFILESMODEL_H |
||||
#define TORRENTFILESMODEL_H |
||||
|
||||
#include <QAbstractItemModel> |
||||
#include <QModelIndex> |
||||
#include <QVariant> |
||||
#include <QStringList> |
||||
#include <QString> |
||||
#include <QDir> |
||||
#include <QDebug> |
||||
#include <QSortFilterProxyModel> |
||||
#include <libtorrent/torrent_info.hpp> |
||||
#include "proplistdelegate.h" |
||||
#include "iconprovider.h" |
||||
|
||||
namespace prio { |
||||
enum FilePriority {IGNORED=0, NORMAL=1, HIGH=2, MAXIMUM=7, PARTIAL=-1}; |
||||
} |
||||
|
||||
class TorrentFileItem { |
||||
|
||||
public: |
||||
enum TreeItemColumns {COL_NAME, COL_SIZE, COL_PROGRESS, COL_PRIO, NB_COL}; |
||||
enum FileType {TFILE, FOLDER, ROOT}; |
||||
|
||||
private: |
||||
QList<TorrentFileItem*> childItems; |
||||
QList<QVariant> itemData; |
||||
TorrentFileItem *parentItem; |
||||
FileType m_type; |
||||
qulonglong total_done; |
||||
int file_index; |
||||
|
||||
public: |
||||
// File Construction
|
||||
TorrentFileItem(const libtorrent::torrent_info &t, const libtorrent::file_entry &f, TorrentFileItem *parent, int _file_index) { |
||||
Q_ASSERT(parent); |
||||
parentItem = parent; |
||||
m_type = TFILE; |
||||
file_index = _file_index; |
||||
#if LIBTORRENT_VERSION_MINOR >= 16 |
||||
QString name = misc::fileName(misc::toQStringU(t.files().file_path(f))); |
||||
#else |
||||
Q_UNUSED(t); |
||||
QString name = misc::toQStringU(f.path.filename()); |
||||
#endif |
||||
// Do not display incomplete extensions
|
||||
if (name.endsWith(".!qB")) |
||||
name.chop(4); |
||||
itemData << name; |
||||
//qDebug("Created a TreeItem file with name %s", qPrintable(getName()));
|
||||
//qDebug("parent is %s", qPrintable(parent->getName()));
|
||||
itemData << QVariant((qulonglong)f.size); |
||||
total_done = 0; |
||||
itemData << 0.; // Progress;
|
||||
itemData << prio::NORMAL; // Priority
|
||||
if (parent) { |
||||
parent->appendChild(this); |
||||
parent->updateSize(); |
||||
} |
||||
} |
||||
|
||||
// Folder constructor
|
||||
TorrentFileItem(QString name, TorrentFileItem *parent=0) { |
||||
parentItem = parent; |
||||
m_type = FOLDER; |
||||
// Do not display incomplete extensions
|
||||
if (name.endsWith(".!qB")) |
||||
name.chop(4); |
||||
itemData << name; |
||||
itemData << 0.; // Size
|
||||
itemData << 0.; // Progress;
|
||||
total_done = 0; |
||||
itemData << prio::NORMAL; // Priority
|
||||
if (parent) { |
||||
parent->appendChild(this); |
||||
} |
||||
} |
||||
|
||||
TorrentFileItem(const QList<QVariant>& data) { |
||||
parentItem = 0; |
||||
m_type = ROOT; |
||||
Q_ASSERT(data.size() == 4); |
||||
itemData = data; |
||||
total_done = 0; |
||||
} |
||||
|
||||
~TorrentFileItem() { |
||||
qDebug("Deleting item: %s", qPrintable(getName())); |
||||
qDeleteAll(childItems); |
||||
} |
||||
|
||||
FileType getType() const { |
||||
return m_type; |
||||
} |
||||
|
||||
int getFileIndex() const { |
||||
Q_ASSERT(m_type ==TFILE); |
||||
return file_index; |
||||
} |
||||
|
||||
void deleteAllChildren() { |
||||
Q_ASSERT(m_type==ROOT); |
||||
qDeleteAll(childItems); |
||||
childItems.clear(); |
||||
Q_ASSERT(childItems.empty()); |
||||
} |
||||
|
||||
QList<TorrentFileItem*> children() const { |
||||
return childItems; |
||||
} |
||||
|
||||
QString getName() const { |
||||
//Q_ASSERT(type != ROOT);
|
||||
return itemData.at(COL_NAME).toString(); |
||||
} |
||||
|
||||
void setName(QString name) { |
||||
Q_ASSERT(m_type != ROOT); |
||||
itemData.replace(COL_NAME, name); |
||||
} |
||||
|
||||
qulonglong getSize() const { |
||||
return itemData.value(COL_SIZE).toULongLong(); |
||||
} |
||||
|
||||
void setSize(qulonglong size) { |
||||
if (getSize() == size) return; |
||||
itemData.replace(COL_SIZE, (qulonglong)size); |
||||
if (parentItem) |
||||
parentItem->updateSize(); |
||||
} |
||||
|
||||
void updateSize() { |
||||
if (m_type == ROOT) return; |
||||
Q_ASSERT(m_type == FOLDER); |
||||
qulonglong size = 0; |
||||
foreach (TorrentFileItem* child, childItems) { |
||||
if (child->getPriority() != prio::IGNORED) |
||||
size += child->getSize(); |
||||
} |
||||
setSize(size); |
||||
} |
||||
|
||||
void setProgress(qulonglong done) { |
||||
Q_ASSERT (m_type != ROOT); |
||||
if (getPriority() == 0) return; |
||||
total_done = done; |
||||
qulonglong size = getSize(); |
||||
Q_ASSERT(total_done <= size); |
||||
qreal progress; |
||||
if (size > 0) |
||||
progress = total_done/(float)size; |
||||
else |
||||
progress = 1.; |
||||
Q_ASSERT(progress >= 0. && progress <= 1.); |
||||
itemData.replace(COL_PROGRESS, progress); |
||||
parentItem->updateProgress(); |
||||
} |
||||
|
||||
qulonglong getTotalDone() const { |
||||
return total_done; |
||||
} |
||||
|
||||
qreal getProgress() const { |
||||
Q_ASSERT (m_type != ROOT); |
||||
if (getPriority() == 0) |
||||
return -1; |
||||
qulonglong size = getSize(); |
||||
if (size > 0) |
||||
return total_done/(float)getSize(); |
||||
return 1.; |
||||
} |
||||
|
||||
void updateProgress() { |
||||
if (m_type == ROOT) return; |
||||
Q_ASSERT(m_type == FOLDER); |
||||
total_done = 0; |
||||
foreach (TorrentFileItem* child, childItems) { |
||||
if (child->getPriority() > 0) |
||||
total_done += child->getTotalDone(); |
||||
} |
||||
//qDebug("Folder: total_done: %llu/%llu", total_done, getSize());
|
||||
Q_ASSERT(total_done <= getSize()); |
||||
setProgress(total_done); |
||||
} |
||||
|
||||
int getPriority() const { |
||||
return itemData.value(COL_PRIO).toInt(); |
||||
} |
||||
|
||||
void setPriority(int new_prio, bool update_parent=true) { |
||||
Q_ASSERT(new_prio != prio::PARTIAL || m_type == FOLDER); // PARTIAL only applies to folders
|
||||
const int old_prio = getPriority(); |
||||
if (old_prio == new_prio) return; |
||||
qDebug("setPriority(%s, %d)", qPrintable(getName()), new_prio); |
||||
// Reset progress if priority is 0
|
||||
if (new_prio == 0) { |
||||
setProgress(0); |
||||
} |
||||
|
||||
itemData.replace(COL_PRIO, new_prio); |
||||
|
||||
// Update parent
|
||||
if (update_parent && parentItem) { |
||||
qDebug("Updating parent item"); |
||||
parentItem->updateSize(); |
||||
parentItem->updateProgress(); |
||||
parentItem->updatePriority(); |
||||
} |
||||
|
||||
// Update children
|
||||
if (new_prio != prio::PARTIAL && !childItems.empty()) { |
||||
qDebug("Updating children items"); |
||||
foreach (TorrentFileItem* child, childItems) { |
||||
// Do not update the parent since
|
||||
// the parent is causing the update
|
||||
child->setPriority(new_prio, false); |
||||
} |
||||
} |
||||
if (m_type == FOLDER) { |
||||
updateSize(); |
||||
updateProgress(); |
||||
} |
||||
} |
||||
|
||||
// Only non-root folders use this function
|
||||
void updatePriority() { |
||||
if (m_type == ROOT) return; |
||||
Q_ASSERT(m_type == FOLDER); |
||||
if (childItems.isEmpty()) return; |
||||
// If all children have the same priority
|
||||
// then the folder should have the same
|
||||
// priority
|
||||
const int prio = childItems.first()->getPriority(); |
||||
for (int i=1; i<childItems.size(); ++i) { |
||||
if (childItems.at(i)->getPriority() != prio) { |
||||
setPriority(prio::PARTIAL); |
||||
return; |
||||
} |
||||
} |
||||
// All child items have the same priorrity
|
||||
// Update mine if necessary
|
||||
if (prio != getPriority()) |
||||
setPriority(prio); |
||||
} |
||||
|
||||
TorrentFileItem* childWithName(QString name) const { |
||||
foreach (TorrentFileItem *child, childItems) { |
||||
if (child->getName() == name) return child; |
||||
} |
||||
return 0; |
||||
} |
||||
|
||||
bool isFolder() const { |
||||
return (m_type==FOLDER); |
||||
} |
||||
|
||||
void appendChild(TorrentFileItem *item) { |
||||
Q_ASSERT(item); |
||||
//Q_ASSERT(!childWithName(item->getName()));
|
||||
Q_ASSERT(m_type != TFILE); |
||||
int i=0; |
||||
for (i=0; i<childItems.size(); ++i) { |
||||
QString newchild_name = item->getName(); |
||||
if (QString::localeAwareCompare(newchild_name, childItems.at(i)->getName()) < 0) break; |
||||
} |
||||
childItems.insert(i, item); |
||||
//childItems.append(item);
|
||||
//Q_ASSERT(type != ROOT || childItems.size() == 1);
|
||||
} |
||||
|
||||
TorrentFileItem *child(int row) { |
||||
//Q_ASSERT(row >= 0 && row < childItems.size());
|
||||
return childItems.value(row, 0); |
||||
} |
||||
|
||||
int childCount() const { |
||||
return childItems.count(); |
||||
} |
||||
|
||||
int columnCount() const { |
||||
return itemData.count(); |
||||
} |
||||
|
||||
QVariant data(int column) const { |
||||
if (column == COL_PROGRESS && m_type != ROOT) |
||||
return getProgress(); |
||||
return itemData.value(column); |
||||
} |
||||
|
||||
int row() const { |
||||
if (parentItem) { |
||||
return parentItem->children().indexOf(const_cast<TorrentFileItem*>(this)); |
||||
} |
||||
return 0; |
||||
} |
||||
|
||||
TorrentFileItem *parent() { |
||||
return parentItem; |
||||
} |
||||
}; |
||||
|
||||
|
||||
class TorrentFilesModel: public QAbstractItemModel { |
||||
Q_OBJECT |
||||
|
||||
private: |
||||
TorrentFileItem *rootItem; |
||||
TorrentFileItem **files_index; |
||||
|
||||
public: |
||||
TorrentFilesModel(QObject *parent=0): QAbstractItemModel(parent) { |
||||
files_index = 0; |
||||
QList<QVariant> rootData; |
||||
rootData << tr("Name") << tr("Size") << tr("Progress") << tr("Priority"); |
||||
rootItem = new TorrentFileItem(rootData); |
||||
} |
||||
|
||||
~TorrentFilesModel() { |
||||
qDebug() << Q_FUNC_INFO << "ENTER"; |
||||
delete [] files_index; |
||||
delete rootItem; |
||||
qDebug() << Q_FUNC_INFO << "EXIT"; |
||||
} |
||||
|
||||
void updateFilesProgress(std::vector<libtorrent::size_type> fp) { |
||||
emit layoutAboutToBeChanged(); |
||||
for (unsigned int i=0; i<fp.size(); ++i) { |
||||
Q_ASSERT(fp[i] >= 0); |
||||
files_index[i]->setProgress(fp[i]); |
||||
} |
||||
emit dataChanged(index(0,0), index(rowCount(), columnCount())); |
||||
} |
||||
|
||||
void updateFilesPriorities(const std::vector<int> &fprio) { |
||||
emit layoutAboutToBeChanged(); |
||||
for (unsigned int i=0; i<fprio.size(); ++i) { |
||||
//qDebug("Called updateFilesPriorities with %d", fprio[i]);
|
||||
files_index[i]->setPriority(fprio[i]); |
||||
} |
||||
emit dataChanged(index(0,0), index(rowCount(), columnCount())); |
||||
} |
||||
|
||||
std::vector<int> getFilesPriorities(unsigned int nbFiles) const { |
||||
std::vector<int> prio; |
||||
for (unsigned int i=0; i<nbFiles; ++i) { |
||||
//qDebug("Called getFilesPriorities: %d", files_index[i]->getPriority());
|
||||
prio.push_back(files_index[i]->getPriority()); |
||||
} |
||||
return prio; |
||||
} |
||||
|
||||
bool allFiltered() const { |
||||
for (int i=0; i<rootItem->childCount(); ++i) { |
||||
if (rootItem->child(i)->getPriority() != prio::IGNORED) |
||||
return false; |
||||
} |
||||
return true; |
||||
} |
||||
|
||||
int columnCount(const QModelIndex &parent=QModelIndex()) const { |
||||
if (parent.isValid()) |
||||
return static_cast<TorrentFileItem*>(parent.internalPointer())->columnCount(); |
||||
else |
||||
return rootItem->columnCount(); |
||||
} |
||||
|
||||
bool setData(const QModelIndex & index, const QVariant & value, int role = Qt::EditRole) { |
||||
if (!index.isValid()) return false; |
||||
if (index.column() == 0 && role == Qt::CheckStateRole) { |
||||
TorrentFileItem *item = static_cast<TorrentFileItem*>(index.internalPointer()); |
||||
qDebug("setData(%s, %d", qPrintable(item->getName()), value.toInt()); |
||||
if (item->getPriority() != value.toInt()) { |
||||
if (value.toInt() == Qt::PartiallyChecked) |
||||
item->setPriority(prio::PARTIAL); |
||||
else if (value.toInt() == Qt::Unchecked) |
||||
item->setPriority(prio::IGNORED); |
||||
else |
||||
item->setPriority(prio::NORMAL); |
||||
emit dataChanged(this->index(0,0), this->index(rowCount(), columnCount())); |
||||
emit filteredFilesChanged(); |
||||
} |
||||
return true; |
||||
} |
||||
if (role == Qt::EditRole) { |
||||
TorrentFileItem *item = static_cast<TorrentFileItem*>(index.internalPointer()); |
||||
switch(index.column()) { |
||||
case TorrentFileItem::COL_NAME: |
||||
item->setName(value.toString()); |
||||
break; |
||||
case TorrentFileItem::COL_SIZE: |
||||
item->setSize(value.toULongLong()); |
||||
break; |
||||
case TorrentFileItem::COL_PROGRESS: |
||||
item->setProgress(value.toDouble()); |
||||
break; |
||||
case TorrentFileItem::COL_PRIO: |
||||
item->setPriority(value.toInt()); |
||||
break; |
||||
default: |
||||
return false; |
||||
} |
||||
emit dataChanged(index, index); |
||||
return true; |
||||
} |
||||
return false; |
||||
} |
||||
|
||||
TorrentFileItem::FileType getType(const QModelIndex &index) const { |
||||
const TorrentFileItem *item = static_cast<const TorrentFileItem*>(index.internalPointer()); |
||||
return item->getType(); |
||||
} |
||||
|
||||
int getFileIndex(const QModelIndex &index) { |
||||
TorrentFileItem *item = static_cast<TorrentFileItem*>(index.internalPointer()); |
||||
return item->getFileIndex(); |
||||
} |
||||
|
||||
QVariant data(const QModelIndex &index, int role=Qt::DisplayRole) const { |
||||
if (!index.isValid()) |
||||
return QVariant(); |
||||
TorrentFileItem *item = static_cast<TorrentFileItem*>(index.internalPointer()); |
||||
if (index.column() == 0 && role == Qt::DecorationRole) { |
||||
if (item->isFolder()) |
||||
return IconProvider::instance()->getIcon("inode-directory"); |
||||
else |
||||
return IconProvider::instance()->getIcon("text-plain"); |
||||
} |
||||
if (index.column() == 0 && role == Qt::CheckStateRole) { |
||||
if (item->data(TorrentFileItem::COL_PRIO).toInt() == prio::IGNORED) |
||||
return Qt::Unchecked; |
||||
if (item->data(TorrentFileItem::COL_PRIO).toInt() == prio::PARTIAL) |
||||
return Qt::PartiallyChecked; |
||||
return Qt::Checked; |
||||
} |
||||
if (role != Qt::DisplayRole) |
||||
return QVariant(); |
||||
|
||||
return item->data(index.column()); |
||||
} |
||||
|
||||
Qt::ItemFlags flags(const QModelIndex &index) const { |
||||
if (!index.isValid()) |
||||
return 0; |
||||
if (getType(index) == TorrentFileItem::FOLDER) |
||||
return Qt::ItemIsEditable | Qt::ItemIsEnabled | Qt::ItemIsSelectable | Qt::ItemIsUserCheckable | Qt::ItemIsTristate; |
||||
return Qt::ItemIsEditable | Qt::ItemIsEnabled | Qt::ItemIsSelectable | Qt::ItemIsUserCheckable; |
||||
} |
||||
|
||||
QVariant headerData(int section, Qt::Orientation orientation, int role) const { |
||||
if (orientation == Qt::Horizontal && role == Qt::DisplayRole) |
||||
return rootItem->data(section); |
||||
|
||||
return QVariant(); |
||||
} |
||||
|
||||
QModelIndex index(int row, int column, const QModelIndex &parent=QModelIndex()) const { |
||||
if (parent.isValid() && parent.column() != 0) |
||||
return QModelIndex(); |
||||
if (column >= TorrentFileItem::NB_COL) |
||||
return QModelIndex(); |
||||
|
||||
TorrentFileItem *parentItem; |
||||
|
||||
if (!parent.isValid()) |
||||
parentItem = rootItem; |
||||
else |
||||
parentItem = static_cast<TorrentFileItem*>(parent.internalPointer()); |
||||
Q_ASSERT(parentItem); |
||||
if (row >= parentItem->childCount()) |
||||
return QModelIndex(); |
||||
|
||||
TorrentFileItem *childItem = parentItem->child(row); |
||||
if (childItem) { |
||||
return createIndex(row, column, childItem); |
||||
} else { |
||||
return QModelIndex(); |
||||
} |
||||
} |
||||
|
||||
QModelIndex parent(const QModelIndex &index) const { |
||||
if (!index.isValid()) |
||||
return QModelIndex(); |
||||
|
||||
TorrentFileItem *childItem = static_cast<TorrentFileItem*>(index.internalPointer()); |
||||
if (!childItem) return QModelIndex(); |
||||
TorrentFileItem *parentItem = childItem->parent(); |
||||
|
||||
if (parentItem == rootItem) |
||||
return QModelIndex(); |
||||
|
||||
return createIndex(parentItem->row(), 0, parentItem); |
||||
} |
||||
|
||||
int rowCount(const QModelIndex &parent=QModelIndex()) const { |
||||
TorrentFileItem *parentItem; |
||||
|
||||
if (parent.column() > 0) |
||||
return 0; |
||||
|
||||
if (!parent.isValid()) |
||||
parentItem = rootItem; |
||||
else |
||||
parentItem = static_cast<TorrentFileItem*>(parent.internalPointer()); |
||||
|
||||
return parentItem->childCount(); |
||||
} |
||||
|
||||
void clear() { |
||||
qDebug("clear called"); |
||||
beginResetModel(); |
||||
if (files_index) { |
||||
delete [] files_index; |
||||
files_index = 0; |
||||
} |
||||
rootItem->deleteAllChildren(); |
||||
endResetModel(); |
||||
} |
||||
|
||||
void setupModelData(const libtorrent::torrent_info &t) { |
||||
qDebug("setup model data called"); |
||||
if (t.num_files() == 0) return; |
||||
emit layoutAboutToBeChanged(); |
||||
// Initialize files_index array
|
||||
qDebug("Torrent contains %d files", t.num_files()); |
||||
files_index = new TorrentFileItem*[t.num_files()]; |
||||
|
||||
TorrentFileItem *parent = this->rootItem; |
||||
TorrentFileItem *root_folder = parent; |
||||
TorrentFileItem *current_parent; |
||||
|
||||
// Iterate over files
|
||||
for (int i=0; i<t.num_files(); ++i) { |
||||
libtorrent::file_entry fentry = t.file_at(i); |
||||
current_parent = root_folder; |
||||
#if LIBTORRENT_VERSION_MINOR >= 16 |
||||
QString path = QDir::cleanPath(misc::toQStringU(fentry.path)).replace("\\", "/"); |
||||
#else |
||||
QString path = QDir::cleanPath(misc::toQStringU(fentry.path.string())).replace("\\", "/"); |
||||
#endif |
||||
// Iterate of parts of the path to create necessary folders
|
||||
QStringList pathFolders = path.split("/"); |
||||
pathFolders.removeAll(".unwanted"); |
||||
pathFolders.takeLast(); |
||||
foreach (const QString &pathPart, pathFolders) { |
||||
TorrentFileItem *new_parent = current_parent->childWithName(pathPart); |
||||
if (!new_parent) { |
||||
new_parent = new TorrentFileItem(pathPart, current_parent); |
||||
} |
||||
current_parent = new_parent; |
||||
} |
||||
// Actually create the file
|
||||
TorrentFileItem *f = new TorrentFileItem(t, fentry, current_parent, i); |
||||
files_index[i] = f; |
||||
} |
||||
emit layoutChanged(); |
||||
} |
||||
|
||||
public slots: |
||||
void selectAll() { |
||||
for (int i=0; i<rootItem->childCount(); ++i) { |
||||
TorrentFileItem *child = rootItem->child(i); |
||||
if (child->getPriority() == prio::IGNORED) |
||||
child->setPriority(prio::NORMAL); |
||||
} |
||||
emit dataChanged(index(0,0), index(rowCount(), columnCount())); |
||||
} |
||||
|
||||
void selectNone() { |
||||
for (int i=0; i<rootItem->childCount(); ++i) { |
||||
rootItem->child(i)->setPriority(prio::IGNORED); |
||||
} |
||||
emit dataChanged(index(0,0), index(rowCount(), columnCount())); |
||||
} |
||||
|
||||
signals: |
||||
void filteredFilesChanged(); |
||||
}; |
||||
|
||||
class TorrentFilesFilterModel: public QSortFilterProxyModel { |
||||
Q_OBJECT |
||||
|
||||
public: |
||||
TorrentFilesFilterModel(QObject *parent = 0): QSortFilterProxyModel(parent) { |
||||
m_model = new TorrentFilesModel(this); |
||||
connect(m_model, SIGNAL(filteredFilesChanged()), this, SIGNAL(filteredFilesChanged())); |
||||
setSourceModel(m_model); |
||||
// Filter settings
|
||||
setFilterCaseSensitivity(Qt::CaseInsensitive); |
||||
setFilterKeyColumn(TorrentFileItem::COL_NAME); |
||||
setFilterRole(Qt::DisplayRole); |
||||
setDynamicSortFilter(true); |
||||
setSortCaseSensitivity(Qt::CaseInsensitive); |
||||
} |
||||
|
||||
~TorrentFilesFilterModel() { |
||||
qDebug() << Q_FUNC_INFO << "ENTER"; |
||||
delete m_model; |
||||
qDebug() << Q_FUNC_INFO << "EXIT"; |
||||
} |
||||
|
||||
TorrentFilesModel* model() const { |
||||
return m_model; |
||||
} |
||||
|
||||
TorrentFileItem::FileType getType(const QModelIndex &index) const { |
||||
return m_model->getType(mapToSource(index)); |
||||
} |
||||
|
||||
int getFileIndex(const QModelIndex &index) { |
||||
return m_model->getFileIndex(mapToSource(index)); |
||||
} |
||||
|
||||
QModelIndex parent(const QModelIndex & child) const { |
||||
if (!child.isValid()) return QModelIndex(); |
||||
QModelIndex sourceParent = m_model->parent(mapToSource(child)); |
||||
if (!sourceParent.isValid()) return QModelIndex(); |
||||
return mapFromSource(sourceParent); |
||||
} |
||||
|
||||
signals: |
||||
void filteredFilesChanged(); |
||||
|
||||
protected: |
||||
bool filterAcceptsRow(int source_row, const QModelIndex &source_parent) const { |
||||
if (m_model->getType(m_model->index(source_row, 0, source_parent)) == TorrentFileItem::FOLDER) { |
||||
// always accept folders, since we want to filter their children
|
||||
return true; |
||||
} |
||||
return QSortFilterProxyModel::filterAcceptsRow(source_row, source_parent); |
||||
} |
||||
|
||||
public slots: |
||||
void selectAll() { |
||||
for (int i=0; i<rowCount(); ++i) { |
||||
setData(index(i, 0), Qt::Checked, Qt::CheckStateRole); |
||||
} |
||||
emit dataChanged(index(0,0), index(rowCount(), columnCount())); |
||||
} |
||||
|
||||
void selectNone() { |
||||
for (int i=0; i<rowCount(); ++i) { |
||||
setData(index(i, 0), Qt::Unchecked, Qt::CheckStateRole); |
||||
} |
||||
emit dataChanged(index(0,0), index(rowCount(), columnCount())); |
||||
} |
||||
|
||||
private: |
||||
TorrentFilesModel *m_model; |
||||
}; |
||||
|
||||
|
||||
#endif // TORRENTFILESMODEL_H
|
Loading…
Reference in new issue