98 lines
2.9 KiB
C++
98 lines
2.9 KiB
C++
/*
|
|
* This file Copyright (C) 2009-2015 Mnemosyne LLC
|
|
*
|
|
* It may be used under the GNU GPL versions 2 or 3
|
|
* or any future license endorsed by Mnemosyne LLC.
|
|
*
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include <cstdint>
|
|
|
|
#include <QAbstractItemModel>
|
|
#include <QList>
|
|
#include <QMap>
|
|
#include <QSet>
|
|
|
|
class FileTreeItem;
|
|
|
|
class FileTreeModel: public QAbstractItemModel
|
|
{
|
|
Q_OBJECT
|
|
|
|
public:
|
|
enum
|
|
{
|
|
COL_NAME,
|
|
COL_SIZE,
|
|
COL_PROGRESS,
|
|
COL_WANTED,
|
|
COL_PRIORITY,
|
|
|
|
NUM_COLUMNS
|
|
};
|
|
|
|
enum Role
|
|
{
|
|
SortRole = Qt::UserRole,
|
|
FileIndexRole,
|
|
WantedRole,
|
|
CompleteRole
|
|
};
|
|
|
|
public:
|
|
FileTreeModel (QObject * parent = nullptr, bool isEditable = true);
|
|
virtual ~FileTreeModel ();
|
|
|
|
void setEditable (bool editable);
|
|
|
|
void clear ();
|
|
void addFile (int index, const QString& filename,
|
|
bool wanted, int priority,
|
|
uint64_t size, uint64_t have,
|
|
bool torrentChanged);
|
|
|
|
bool openFile (const QModelIndex& index);
|
|
|
|
void twiddleWanted (const QModelIndexList& indices);
|
|
void twiddlePriority (const QModelIndexList& indices);
|
|
|
|
void setWanted (const QModelIndexList& indices, bool wanted);
|
|
void setPriority (const QModelIndexList& indices, int priority);
|
|
|
|
QModelIndex parent (const QModelIndex& child, int column) const;
|
|
|
|
// QAbstractItemModel
|
|
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 = Qt::DisplayRole) const;
|
|
virtual QModelIndex index (int row, int column, const QModelIndex& parent = QModelIndex ()) const;
|
|
virtual QModelIndex parent (const QModelIndex& child) const;
|
|
virtual int rowCount (const QModelIndex& parent = QModelIndex ()) const;
|
|
virtual int columnCount (const QModelIndex& parent = QModelIndex ()) const;
|
|
virtual bool setData (const QModelIndex& index, const QVariant& value, int role = Qt::EditRole);
|
|
|
|
signals:
|
|
void priorityChanged (const QSet<int>& fileIndices, int);
|
|
void wantedChanged (const QSet<int>& fileIndices, bool);
|
|
void pathEdited (const QString& oldpath, const QString& newname);
|
|
void openRequested (const QString& path);
|
|
|
|
private:
|
|
void clearSubtree (const QModelIndex&);
|
|
QModelIndex indexOf (FileTreeItem *, int column) const;
|
|
void emitParentsChanged (const QModelIndex&, int firstColumn, int lastColumn, QSet<QModelIndex> * visitedParentIndices = nullptr);
|
|
void emitSubtreeChanged (const QModelIndex&, int firstColumn, int lastColumn);
|
|
FileTreeItem * findItemForFileIndex (int fileIndex) const;
|
|
FileTreeItem * itemFromIndex (const QModelIndex&) const;
|
|
QModelIndexList getOrphanIndices (const QModelIndexList& indices) const;
|
|
|
|
private:
|
|
bool myIsEditable;
|
|
|
|
FileTreeItem * myRootItem;
|
|
QMap<int, FileTreeItem *> myIndexCache;
|
|
};
|
|
|