source: trunk/qt/file-tree.h @ 13810

Last change on this file since 13810 was 13810, checked in by jordan, 8 years ago

(trunk) #1220 'change top folder names' -- add file-renaming to the Qt client

  • Property svn:keywords set to Date Rev Author Id
File size: 5.4 KB
Line 
1/*
2 * This file Copyright (C) Mnemosyne LLC
3 *
4 * This program is free software; you can redistribute it and/or modify
5 * it under the terms of the GNU General Public License version 2
6 * as published by the Free Software Foundation.
7 *
8 * http://www.gnu.org/licenses/old-licenses/gpl-2.0.html
9 *
10 * $Id: file-tree.h 13810 2013-01-20 01:31:58Z jordan $
11 */
12
13#ifndef QTR_TREE_FILE_MODEL
14#define QTR_TREE_FILE_MODEL
15
16#include <QAbstractItemModel>
17#include <QObject>
18#include <QItemDelegate>
19#include <QList>
20#include <QHash>
21#include <QSet>
22#include <QSize>
23#include <QString>
24#include <QTreeView>
25#include <QVariant>
26
27class QSortFilterProxyModel;
28class QStyle;
29
30#include "torrent.h" // FileList
31
32/****
33*****
34****/
35
36class FileTreeItem: public QObject
37{
38        Q_OBJECT;
39
40        enum { LOW=(1<<0), NORMAL=(1<<1), HIGH=(1<<2) };
41
42    public:
43        virtual ~FileTreeItem( );
44        FileTreeItem( int fileIndex, const QString& name="" ):
45            myIndex(fileIndex), myParent(0), myName(name),
46            myPriority(0), myIsWanted(0),
47            myHaveSize(0), myTotalSize(0),
48            myFirstUnhashedRow(0) { }
49
50    public:
51        void appendChild( FileTreeItem *child );
52        FileTreeItem * child( const QString& filename );
53        FileTreeItem * child( int row ) { return myChildren.at( row ); }
54        int childCount( ) const { return myChildren.size( ); }
55        FileTreeItem * parent( ) { return myParent; }
56        const FileTreeItem * parent( ) const { return myParent; }
57        int row( ) const;
58        const QString& name( ) const { return myName; }
59        QVariant data( int column ) const;
60        bool update( int index, bool want, int priority, uint64_t total, uint64_t have, bool torrentChanged );
61        void twiddleWanted( QSet<int>& fileIds, bool& );
62        void twiddlePriority( QSet<int>& fileIds, int& );
63
64    private:
65        void setSubtreePriority( int priority, QSet<int>& fileIds );
66        void setSubtreeWanted( bool, QSet<int>& fileIds );
67        QString priorityString( ) const;
68        void getSubtreeSize( uint64_t& have, uint64_t& total ) const;
69        QString fileSizeName( ) const;
70        double progress( ) const;
71        int priority( ) const;
72        int isSubtreeWanted( ) const;
73
74        int myIndex;
75        FileTreeItem * myParent;
76        QList<FileTreeItem*> myChildren;
77        QHash<QString,int> myChildRows;
78        QHash<QString,int>& getMyChildRows();
79        const QString myName;
80        int myPriority;
81        bool myIsWanted;
82        uint64_t myHaveSize;
83        uint64_t myTotalSize;
84        size_t myFirstUnhashedRow;
85};
86
87class FileTreeModel: public QAbstractItemModel
88{
89        Q_OBJECT
90
91    public:
92        FileTreeModel( QObject *parent = 0);
93        ~FileTreeModel( );
94
95    public:
96        QVariant data( const QModelIndex &index, int role = Qt::DisplayRole ) const;
97        Qt::ItemFlags flags( const QModelIndex& index ) const;
98        QVariant headerData( int section, Qt::Orientation orientation, int role = Qt::DisplayRole ) const;
99        QModelIndex index( int row, int column, const QModelIndex& parent = QModelIndex() ) const;
100        QModelIndex parent( const QModelIndex& child ) const;
101        QModelIndex parent( const QModelIndex& child, int column ) const;
102        int rowCount( const QModelIndex& parent = QModelIndex( ) ) const;
103        int columnCount( const QModelIndex &parent = QModelIndex( ) ) const;
104        virtual bool setData ( const QModelIndex & index, const QVariant & value, int role = Qt::EditRole );
105
106    signals:
107        void priorityChanged( const QSet<int>& fileIndices, int );
108        void wantedChanged( const QSet<int>& fileIndices, bool );
109        void pathEdited (const QString& oldpath, const QString& newname);
110
111    public:
112        void clear( );
113        void addFile( int index, const QString& filename,
114                      bool wanted, int priority,
115                      uint64_t size, uint64_t have,
116                      QList<QModelIndex>& rowsAdded,
117                      bool torrentChanged );
118
119    private:
120        void clearSubtree( const QModelIndex & );
121        QModelIndex indexOf( FileTreeItem *, int column ) const;
122        void parentsChanged( const QModelIndex &, int column );
123        void subtreeChanged( const QModelIndex &, int column );
124
125    private:
126        FileTreeItem * rootItem;
127
128    public slots:
129        void clicked ( const QModelIndex & index );
130};
131
132class FileTreeDelegate: public QItemDelegate
133{
134        Q_OBJECT
135
136    public:
137        FileTreeDelegate( QObject * parent=0 ): QItemDelegate( parent ) { }
138        virtual ~FileTreeDelegate( ) { }
139
140    public:
141        virtual QSize sizeHint(const QStyleOptionViewItem&, const QModelIndex&) const;
142        virtual void paint(QPainter*, const QStyleOptionViewItem&, const QModelIndex&) const;
143};
144
145class FileTreeView: public QTreeView
146{
147        Q_OBJECT
148
149    public:
150        FileTreeView( QWidget * parent=0 );
151        virtual ~FileTreeView( );
152        void clear( );
153        void update( const FileList& files );
154        void update( const FileList& files, bool torrentChanged );
155
156    signals:
157        void priorityChanged( const QSet<int>& fileIndices, int );
158        void wantedChanged( const QSet<int>& fileIndices, bool );
159        void pathEdited (const QString& oldpath, const QString& newname);
160
161    protected:
162        bool eventFilter( QObject *, QEvent * );
163
164    private:
165        FileTreeModel myModel;
166        QSortFilterProxyModel * myProxy;
167        FileTreeDelegate myDelegate;
168
169    public slots:
170        void onClicked ( const QModelIndex & index );
171};
172
173#endif
Note: See TracBrowser for help on using the repository browser.