mirror
/
transmission
mirror of https://github.com/transmission/transmission
1
0
Fork 0
Code Issues Releases Wiki Activity
transmission/qt/file-tree.cc

909 lines
21 KiB
C++
Raw Blame History

/*
* This file Copyright (C) Mnemosyne LLC
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License version 2
* as published by the Free Software Foundation.
*
* http://www.gnu.org/licenses/old-licenses/gpl-2.0.html
*
* $Id$
*/
#include <cassert>
#include <iostream>
#include <QApplication>
#include <QHeaderView>
#include <QPainter>
#include <QResizeEvent>
#include <QSortFilterProxyModel>
#include <QStringList>
#include <libtransmission/transmission.h> // priorities
#include "file-tree.h"
#include "formatter.h"
#include "hig.h"
#include "torrent.h" // FileList
#include "utils.h" // mime icons
enum
{
COL_NAME,
COL_PROGRESS,
COL_WANTED,
COL_PRIORITY,
NUM_COLUMNS
};
/****
*****
****/
QHash<QString,int>&
FileTreeItem :: getMyChildRows ()
{
const size_t n = childCount();
// ensure that all the rows are hashed
while (myFirstUnhashedRow < n)
{
myChildRows.insert (myChildren[myFirstUnhashedRow]->name(), myFirstUnhashedRow);
myFirstUnhashedRow++;
}
return myChildRows;
}
FileTreeItem :: ~FileTreeItem ()
{
assert(myChildren.isEmpty());
if (myParent != 0)
{
const int pos = row();
assert ((pos>=0) && "couldn't find child in parent's lookup");
myParent->myChildren.removeAt(pos);
myParent->myChildRows.remove(name());
myParent->myFirstUnhashedRow = pos;
}
}
void
FileTreeItem :: appendChild (FileTreeItem * child)
{
const size_t n = childCount();
child->myParent = this;
myChildren.append (child);
myFirstUnhashedRow = n;
}
FileTreeItem *
FileTreeItem :: child (const QString& filename)
{
FileTreeItem * item(0);
const int row = getMyChildRows().value (filename, -1);
if (row != -1)
{
item = child (row);
assert (filename == item->name());
}
return item;
}
int
FileTreeItem :: row () const
{
int i(-1);
if(myParent)
{
i = myParent->getMyChildRows().value (name(), -1);
assert (this == myParent->myChildren[i]);
}
return i;
}
QVariant
FileTreeItem :: data (int column, int role) const
{
QVariant value;
if (role == Qt::EditRole)
{
if (column == 0)
value.setValue (name());
}
else if (role == Qt::DisplayRole)
{
switch(column)
{
case COL_NAME:
value.setValue (fileSizeName());
break;
case COL_PROGRESS:
value.setValue (progress());
break;
case COL_WANTED:
value.setValue (isSubtreeWanted());
break;
case COL_PRIORITY:
value.setValue (priorityString());
break;
}
}
return value;
}
void
FileTreeItem :: getSubtreeSize (uint64_t& have, uint64_t& total) const
{
have += myHaveSize;
total += myTotalSize;
foreach(const FileTreeItem * i, myChildren)
i->getSubtreeSize(have, total);
}
double
FileTreeItem :: progress () const
{
double d(0);
uint64_t have(0), total(0);
getSubtreeSize(have, total);
if (total)
d = have / (double)total;
return d;
}
QString
FileTreeItem :: fileSizeName () const
{
uint64_t have(0), total(0);
QString str;
getSubtreeSize(have, total);
str = QString(name() + " (%1)").arg(Formatter::sizeToString(total));
return str;
}
bool
FileTreeItem :: update (int index,
bool wanted,
int priority,
uint64_t totalSize,
uint64_t haveSize,
bool updateFields)
{
bool changed = false;
std::cerr << __FILE__ << ':' << __LINE__ << " index " << index << " wanted " << wanted << " myIndex " << myIndex << " myWanted" << myIsWanted << std::endl;
if (myIndex != index)
{
myIndex = index;
changed = true;
}
if (updateFields && myIsWanted != wanted)
{
myIsWanted = wanted;
changed = true;
std::cerr << __FILE__ << ':' << __LINE__ << " setting myIsWanted to " << myIsWanted << std::endl;
}
if (updateFields && myPriority != priority)
{
myPriority = priority;
changed = true;
}
if (myTotalSize != totalSize)
{
myTotalSize = totalSize;
changed = true;
}
if (myHaveSize != haveSize)
{
myHaveSize = haveSize;
changed = true;
}
return changed;
}
QString
FileTreeItem :: priorityString () const
{
const int i(priority());
if(i == LOW) return tr("Low");
if(i == HIGH) return tr("High");
if(i == NORMAL) return tr("Normal");
return tr("Mixed");
}
int
FileTreeItem :: priority () const
{
int i(0);
if (myChildren.isEmpty())
{
switch (myPriority)
{
case TR_PRI_LOW:
i |= LOW;
break;
case TR_PRI_HIGH:
i |= HIGH;
break;
default:
i |= NORMAL;
break;
}
}
foreach (const FileTreeItem * child, myChildren)
i |= child->priority();
return i;
}
void
FileTreeItem :: setSubtreePriority (int i, QSet<int>& ids)
{
if (myPriority != i)
{
myPriority = i;
if (myIndex >= 0)
ids.insert (myIndex);
}
foreach (FileTreeItem * child, myChildren)
child->setSubtreePriority (i, ids);
}
void
FileTreeItem :: twiddlePriority (QSet<int>& ids, int& p)
{
const int old(priority());
if (old & LOW)
p = TR_PRI_NORMAL;
else if (old & NORMAL)
p = TR_PRI_HIGH;
else
p = TR_PRI_LOW;
std::cerr << __FILE__ << ':' << __LINE__ << " twiddlePriority, p " << p << std::endl;
setSubtreePriority (p, ids);
}
int
FileTreeItem :: isSubtreeWanted () const
{
if(myChildren.isEmpty())
return myIsWanted ? Qt::Checked : Qt::Unchecked;
int wanted(-1);
foreach (const FileTreeItem * child, myChildren)
{
const int childWanted = child->isSubtreeWanted();
if (wanted == -1)
wanted = childWanted;
if (wanted != childWanted)
wanted = Qt::PartiallyChecked;
if (wanted == Qt::PartiallyChecked)
return wanted;
}
return wanted;
}
void
FileTreeItem :: setSubtreeWanted (bool b, QSet<int>& ids)
{
std::cerr << __FILE__ << ':' << __LINE__ << " twiddleWanted, wanted " << b << std::endl;
if (myIsWanted != b)
{
myIsWanted = b;
if (myIndex >= 0)
ids.insert(myIndex);
}
foreach (FileTreeItem * child, myChildren)
child->setSubtreeWanted (b, ids);
}
void
FileTreeItem :: twiddleWanted (QSet<int>& ids, bool& wanted)
{
std::cerr << __FILE__ << ':' << __LINE__ << " twiddleWanted" << std::endl;
wanted = isSubtreeWanted() != Qt::Checked;
setSubtreeWanted (wanted, ids);
}
/***
****
****
***/
FileTreeModel :: FileTreeModel (QObject *parent):
QAbstractItemModel(parent)
{
rootItem = new FileTreeItem(-1);
}
FileTreeModel :: ~FileTreeModel()
{
std::cerr << __FILE__ << ':' << __LINE__ << " dtor " << std::endl;
clear();
delete rootItem;
}
QVariant
FileTreeModel :: data (const QModelIndex &index, int role) const
{
QVariant value;
if (index.isValid())
{
FileTreeItem * i = static_cast<FileTreeItem*>(index.internalPointer());
value = i->data (index.column(), role);
}
return value;
}
Qt::ItemFlags
FileTreeModel :: flags (const QModelIndex& index) const
{
int i(Qt::ItemIsSelectable | Qt::ItemIsEnabled);
if(index.column() == COL_NAME)
i |= Qt::ItemIsEditable;
if(index.column() == COL_WANTED)
i |= Qt::ItemIsUserCheckable | Qt::ItemIsTristate;
return (Qt::ItemFlags)i;
}
bool
FileTreeModel :: setData (const QModelIndex& index, const QVariant& newname, int role)
{
if (role == Qt::EditRole)
{
QString oldpath;
QModelIndex walk = index;
FileTreeItem * item = static_cast<FileTreeItem*>(index.internalPointer());
while (item && !item->name().isEmpty())
{
if (oldpath.isEmpty())
oldpath = item->name();
else
oldpath = item->name() + "/" + oldpath;
item = item->parent ();
}
emit pathEdited (oldpath, newname.toString());
}
return false; // don't update the view until the session confirms the change
}
QVariant
FileTreeModel :: headerData (int column, Qt::Orientation orientation, int role) const
{
QVariant data;
if (orientation==Qt::Horizontal && role==Qt::DisplayRole)
{
switch (column)
{
case COL_NAME:
data.setValue (tr("File"));
break;
case COL_PROGRESS:
data.setValue (tr("Progress"));
break;
case COL_WANTED:
data.setValue (tr("Download"));
break;
case COL_PRIORITY:
data.setValue (tr("Priority"));
break;
default:
break;
}
}
return data;
}
QModelIndex
FileTreeModel :: index (int row, int column, const QModelIndex& parent) const
{
QModelIndex i;
if (!hasIndex (row, column, parent))
{
std::cerr << " I don't have this index " << std::endl;
}
else
{
FileTreeItem * parentItem;
if(!parent.isValid())
parentItem = rootItem;
else
parentItem = static_cast<FileTreeItem*>(parent.internalPointer());
FileTreeItem * childItem = parentItem->child(row);
if (childItem)
i = createIndex(row, column, childItem);
}
return i;
}
QModelIndex
FileTreeModel :: parent (const QModelIndex& child) const
{
return parent (child, 0); // QAbstractItemModel::parent() wants col 0
}
QModelIndex
FileTreeModel :: parent (const QModelIndex& child, int column) const
{
if (!child.isValid())
return QModelIndex();
FileTreeItem * childItem = static_cast<FileTreeItem*>(child.internalPointer());
return indexOf (childItem->parent(), column);
}
int
FileTreeModel :: rowCount (const QModelIndex& parent) const
{
FileTreeItem * parentItem;
if (!parent.isValid())
parentItem = rootItem;
else
parentItem = static_cast<FileTreeItem*>(parent.internalPointer());
return parentItem->childCount();
}
int
FileTreeModel :: columnCount (const QModelIndex &parent) const
{
Q_UNUSED(parent);
return 4;
}
QModelIndex
FileTreeModel :: indexOf (FileTreeItem * item, int column) const
{
if (!item || item==rootItem)
return QModelIndex();
return createIndex(item->row(), column, item);
}
void
FileTreeModel :: clearSubtree (const QModelIndex& top)
{
size_t i = rowCount (top);
std::cerr << __FILE__ << ':' << __LINE__ << " clearSubtree " << i << std::endl;
while (i > 0)
clearSubtree(index(--i, 0, top));
delete static_cast<FileTreeItem*>(top.internalPointer());
}
void
FileTreeModel :: clear ()
{
std::cerr << __FILE__ << ':' << __LINE__ << " clear" << std::endl;
clearSubtree (QModelIndex());
reset ();
}
void
FileTreeModel :: addFile (int index,
const QString & filename,
bool wanted,
int priority,
uint64_t size,
uint64_t have,
QList<QModelIndex> & rowsAdded,
bool updateFields)
{
bool added = false;
FileTreeItem * i (rootItem);
foreach (QString token, filename.split (QChar::fromAscii('/')))
{
FileTreeItem * child(i->child(token));
if (!child)
{
added = true;
QModelIndex parentIndex (indexOf(i, 0));
const int n (i->childCount());
beginInsertRows (parentIndex, n, n);
i->appendChild ((child = new FileTreeItem(-1, token)));
endInsertRows ();
rowsAdded.append (indexOf(child, 0));
}
i = child;
}
if (i != rootItem)
{
if (i->update (index, wanted, priority, size, have, added || updateFields))
{
std::cerr << __FILE__ << ':' << __LINE__ << " emitting dataChanged for row " << i << std::endl;
dataChanged (indexOf(i, 0), indexOf(i, NUM_COLUMNS-1));
}
}
}
void
FileTreeModel :: parentsChanged (const QModelIndex& index, int column)
{
QModelIndex walk = index;
for (;;)
{
walk = parent(walk, column);
if(!walk.isValid())
break;
dataChanged(walk, walk);
}
}
void
FileTreeModel :: subtreeChanged (const QModelIndex& index, int column)
{
const int childCount = rowCount (index);
if (!childCount)
return;
// tell everyone that this tier changed
dataChanged (index.child(0,column), index.child(childCount-1,column));
// walk the subtiers
for (int i=0; i<childCount; ++i)
subtreeChanged (index.child(i,column), column);
}
void
FileTreeModel :: clicked (const QModelIndex& index)
{
const int column (index.column());
if (!index.isValid())
return;
if (column == COL_WANTED)
{
bool want;
QSet<int> file_ids;
FileTreeItem * item;
item = static_cast<FileTreeItem*>(index.internalPointer());
item->twiddleWanted (file_ids, want);
emit wantedChanged (file_ids, want);
dataChanged (index, index);
parentsChanged (index, column);
subtreeChanged (index, column);
}
else if (column == COL_PRIORITY)
{
int priority;
QSet<int> file_ids;
FileTreeItem * item;
std::cerr << "clicked in COL_PRIORITY" << std::endl;
item = static_cast<FileTreeItem*>(index.internalPointer());
item->twiddlePriority (file_ids, priority);
emit priorityChanged (file_ids, priority);
dataChanged(index, index);
parentsChanged(index, column);
subtreeChanged(index, column);
}
}
/****
*****
****/
QSize
FileTreeDelegate :: sizeHint(const QStyleOptionViewItem& item, const QModelIndex& index) const
{
QSize size;
switch(index.column())
{
case COL_NAME:
{
const QFontMetrics fm(item.font);
const QString text = index.data().toString();
const int iconSize = QApplication::style()->pixelMetric(QStyle::PM_SmallIconSize);
size.rwidth() = HIG::PAD_SMALL + iconSize;
size.rheight() = std::max(iconSize, fm.height());
break;
}
case COL_PROGRESS:
case COL_WANTED:
size = QSize(20, 1);
break;
default:
{
const QFontMetrics fm(item.font);
const QString text = index.data().toString();
size = fm.size(0, text);
break;
}
}
size.rheight() += 8; // make the spacing a little nicer
return size;
}
void
FileTreeDelegate :: paint (QPainter * painter,
const QStyleOptionViewItem & option,
const QModelIndex & index) const
{
const int column(index.column());
if ((column != COL_PROGRESS) && (column != COL_WANTED) && (column != COL_NAME))
{
QItemDelegate::paint(painter, option, index);
return;
}
QStyle * style (QApplication :: style());
if (option.state & QStyle::State_Selected)
painter->fillRect (option.rect, option.palette.highlight());
painter->save();
if (option.state & QStyle::State_Selected)
painter->setBrush (option.palette.highlightedText());
if (column == COL_NAME)
{
// draw the file icon
static const int iconSize (style->pixelMetric(QStyle :: PM_SmallIconSize));
const QRect iconArea (option.rect.x(),
option.rect.y() + (option.rect.height()-iconSize)/2,
iconSize, iconSize);
QIcon icon;
if (index.model()->hasChildren(index))
{
icon = style->standardIcon(QStyle::StandardPixmap(QStyle::SP_DirOpenIcon));
}
else
{
QString name = index.data().toString();
icon = Utils :: guessMimeIcon (name.left(name.lastIndexOf(" (")));
}
icon.paint (painter, iconArea, Qt::AlignCenter, QIcon::Normal, QIcon::On);
// draw the name
QStyleOptionViewItem tmp (option);
tmp.rect.setWidth (option.rect.width() - iconArea.width() - HIG::PAD_SMALL);
tmp.rect.moveRight (option.rect.right());
QItemDelegate::paint (painter, tmp, index);
}
else if(column == COL_PROGRESS)
{
QStyleOptionProgressBar p;
p.state = option.state | QStyle::State_Small;
p.direction = QApplication::layoutDirection();
p.rect = option.rect;
p.rect.setSize (QSize(option.rect.width()-2, option.rect.height()-8));
p.rect.moveCenter (option.rect.center());
p.fontMetrics = QApplication::fontMetrics();
p.minimum = 0;
p.maximum = 100;
p.textAlignment = Qt::AlignCenter;
p.textVisible = true;
p.progress = (int)(100.0*index.data().toDouble());
p.text = QString().sprintf("%d%%", p.progress);
style->drawControl(QStyle::CE_ProgressBar, &p, painter);
}
else if(column == COL_WANTED)
{
QStyleOptionButton o;
o.state = option.state;
o.direction = QApplication::layoutDirection();
o.rect.setSize (QSize(20, option.rect.height()));
o.rect.moveCenter (option.rect.center());
o.fontMetrics = QApplication::fontMetrics();
switch (index.data().toInt())
{
case Qt::Unchecked: o.state |= QStyle::State_Off; break;
case Qt::Checked: o.state |= QStyle::State_On; break;
default: o.state |= QStyle::State_NoChange;break;
}
style->drawControl (QStyle::CE_CheckBox, &o, painter);
}
painter->restore();
}
/****
*****
*****
*****
****/
FileTreeView :: FileTreeView (QWidget * parent):
QTreeView (parent),
myModel (this),
myProxy (new QSortFilterProxyModel()),
myDelegate (this)
{
setSortingEnabled (true);
setAlternatingRowColors (true);
setSelectionBehavior (QAbstractItemView::SelectRows);
setSelectionMode (QAbstractItemView::ExtendedSelection);
myProxy->setSourceModel (&myModel);
setModel (myProxy);
setItemDelegate (&myDelegate);
setHorizontalScrollBarPolicy (Qt::ScrollBarAlwaysOff);
sortByColumn (COL_NAME, Qt::AscendingOrder);
installEventFilter (this);
for (int i=0; i<NUM_COLUMNS; ++i)
header()->setResizeMode(i, QHeaderView::Interactive);
connect (this, SIGNAL(clicked(const QModelIndex&)),
this, SLOT(onClicked(const QModelIndex&)));
connect (&myModel, SIGNAL(priorityChanged(const QSet<int>&, int)),
this, SIGNAL(priorityChanged(const QSet<int>&, int)));
connect (&myModel, SIGNAL(wantedChanged(const QSet<int>&, bool)),
this, SIGNAL(wantedChanged(const QSet<int>&, bool)));
connect (&myModel, SIGNAL(pathEdited(const QString&, const QString&)),
this, SIGNAL(pathEdited(const QString&, const QString&)));
}
FileTreeView :: ~FileTreeView ()
{
myProxy->deleteLater();
}
void
FileTreeView :: onClicked (const QModelIndex& proxyIndex)
{
const QModelIndex modelIndex = myProxy->mapToSource (proxyIndex);
std::cerr << __FILE__ << ':' << __LINE__ << " calling myModel.clicked()" << std::endl;
myModel.clicked (modelIndex);
}
bool
FileTreeView :: eventFilter (QObject * o, QEvent * event)
{
if (o != this)
return false;
// this is kind of a hack to get the last three columns be the
// right size, and to have the filename column use whatever
// space is left over...
if (event->type() == QEvent::Resize)
{
QResizeEvent * r = dynamic_cast<QResizeEvent*>(event);
int left = r->size().width();
const QFontMetrics fontMetrics(font());
for (int column=0; column<NUM_COLUMNS; ++column)
{
if (column == COL_NAME)
continue;
if (isColumnHidden (column))
continue;
const QString header = myModel.headerData (column, Qt::Horizontal).toString() + " ";
const int width = fontMetrics.size (0, header).width();
setColumnWidth (column, width);
left -= width;
}
left -= 20; // not sure why this is necessary. it works in different themes + font sizes though...
setColumnWidth(COL_NAME, std::max(left,0));
return false;
}
#if 0
// handle using the keyboard to toggle the
// wanted/unwanted state or the file priority
else if (event->type() == QEvent::KeyPress)
{
std::cerr << __FILE__ << ':' << __LINE__ << " " << qPrintable(dynamic_cast<QKeyEvent*>(event)->text()) << std::endl;
switch(dynamic_cast<QKeyEvent*>(event)->key())
{
case Qt::Key_Space:
std::cerr << __FILE__ << ':' << __LINE__ << " calling COL_WANTED.clicked()" << std::endl;
foreach (QModelIndex i, selectionModel()->selectedRows(COL_WANTED))
clicked (i);
return false;
case Qt::Key_Enter:
case Qt::Key_Return:
std::cerr << __FILE__ << ':' << __LINE__ << " calling COL_PRIORITY.clicked()" << std::endl;
foreach (QModelIndex i, selectionModel()->selectedRows(COL_PRIORITY))
clicked (i);
return false;
}
}
#endif
return false;
}
void
FileTreeView :: update (const FileList& files)
{
update (files, true);
}
void
FileTreeView :: update (const FileList& files, bool updateFields)
{
std::cerr << "update updateFields " << updateFields << std::endl;
foreach (const TrFile file, files)
{
QList<QModelIndex> added;
std::cerr << __FILE__ << ':' << __LINE__ << " add file " << qPrintable(file.filename) << " wanted " << file.wanted << std::endl;
myModel.addFile (file.index, file.filename, file.wanted, file.priority, file.size, file.have, added, updateFields);
foreach (QModelIndex i, added)
expand (myProxy->mapFromSource(i));
}
}
void
FileTreeView :: clear ()
{
std::cerr << __FILE__ << ':' << __LINE__ << " clear" << std::endl;
myModel.clear();
}
Reference in New Issue View Git Blame Copy Permalink