mirror of
https://bitbucket.org/smil3y/kde-extraapps.git
synced 2025-02-25 03:12:53 +00:00
111 lines
4.5 KiB
C++
111 lines
4.5 KiB
C++
/***************************************************************************
|
|
* Copyright (C) 2007 by Joris Guisson and Ivan Vasic *
|
|
* joris.guisson@gmail.com *
|
|
* ivasic@gmail.com *
|
|
* *
|
|
* 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. *
|
|
***************************************************************************/
|
|
#ifndef KTTORRENTFILETREEMODEL_H
|
|
#define KTTORRENTFILETREEMODEL_H
|
|
|
|
#include "torrentfilemodel.h"
|
|
#include <util/bitset.h>
|
|
|
|
#include <QSortFilterProxyModel>
|
|
|
|
namespace bt
|
|
{
|
|
class BEncoder;
|
|
class BNode;
|
|
}
|
|
|
|
namespace kt
|
|
{
|
|
|
|
/**
|
|
* Model for displaying file trees of a torrent
|
|
* @author Joris Guisson
|
|
*/
|
|
class TorrentFileTreeModel : public TorrentFileModel
|
|
{
|
|
Q_OBJECT
|
|
protected:
|
|
struct Node
|
|
{
|
|
Node* parent;
|
|
bt::TorrentFileInterface* file; // file (0 if this is a directory)
|
|
QString name; // name or directory
|
|
QList<Node*> children; // child dirs
|
|
bt::Uint64 size;
|
|
bt::BitSet chunks;
|
|
bool chunks_set;
|
|
float percentage;
|
|
|
|
Node(Node* parent,bt::TorrentFileInterface* file,const QString & name,
|
|
bt::Uint32 total_chunks);
|
|
Node(Node* parent,const QString & name, bt::Uint32 total_chunks);
|
|
~Node();
|
|
|
|
void insert(const QString & path,bt::TorrentFileInterface* file, bt::Uint32 num_chunks);
|
|
int row();
|
|
bt::Uint64 fileSize(const bt::TorrentInterface* tc);
|
|
bt::Uint64 bytesToDownload(const bt::TorrentInterface* tc);
|
|
Qt::CheckState checkState(const bt::TorrentInterface* tc) const;
|
|
QString path();
|
|
void fillChunks();
|
|
void updatePercentage(const bt::BitSet & havechunks);
|
|
void initPercentage(const bt::TorrentInterface* tc,const bt::BitSet & havechunks);
|
|
|
|
void saveExpandedState(const QModelIndex & index,QSortFilterProxyModel* pm,QTreeView* tv,bt::BEncoder* enc);
|
|
void loadExpandedState(const QModelIndex & index,QSortFilterProxyModel* pm,QTreeView* tv,bt::BNode* node);
|
|
};
|
|
public:
|
|
TorrentFileTreeModel(bt::TorrentInterface* tc,DeselectMode mode,QObject* parent);
|
|
virtual ~TorrentFileTreeModel();
|
|
|
|
virtual int rowCount(const QModelIndex & parent) const;
|
|
virtual int columnCount(const QModelIndex & parent) const;
|
|
virtual QVariant headerData(int section, Qt::Orientation orientation,int role) const;
|
|
virtual QVariant data(const QModelIndex & index, int role) const;
|
|
virtual QModelIndex parent(const QModelIndex & index) const;
|
|
virtual QModelIndex index(int row,int column,const QModelIndex & parent) const;
|
|
virtual bool setData(const QModelIndex & index, const QVariant & value, int role);
|
|
virtual void checkAll();
|
|
virtual void uncheckAll();
|
|
virtual void invertCheck();
|
|
virtual bt::Uint64 bytesToDownload();
|
|
virtual QByteArray saveExpandedState(QSortFilterProxyModel* pm,QTreeView* tv);
|
|
virtual void loadExpandedState(QSortFilterProxyModel* pm,QTreeView* tv,const QByteArray & state);
|
|
virtual bt::TorrentFileInterface* indexToFile(const QModelIndex & idx);
|
|
virtual QString dirPath(const QModelIndex & idx);
|
|
virtual void changePriority(const QModelIndexList & indexes,bt::Priority newpriority);
|
|
virtual void onCodecChange();
|
|
private:
|
|
void constructTree();
|
|
void invertCheck(const QModelIndex & idx);
|
|
bool setCheckState(const QModelIndex & index, Qt::CheckState state);
|
|
bool setName(const QModelIndex & index,const QString & name);
|
|
void modifyPathOfFiles(Node* n,const QString & path);
|
|
|
|
|
|
protected:
|
|
Node* root;
|
|
bool emit_check_state_change;
|
|
};
|
|
|
|
}
|
|
|
|
#endif
|