1 /***************************************************************************
2 * Copyright (C) 2005-08 by the Quassel Project *
3 * devel@quassel-irc.org *
5 * This program is free software; you can redistribute it and/or modify *
6 * it under the terms of the GNU General Public License as published by *
7 * the Free Software Foundation; either version 2 of the License, or *
8 * (at your option) version 3. *
10 * This program is distributed in the hope that it will be useful, *
11 * but WITHOUT ANY WARRANTY; without even the implied warranty of *
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the *
13 * GNU General Public License for more details. *
15 * You should have received a copy of the GNU General Public License *
16 * along with this program; if not, write to the *
17 * Free Software Foundation, Inc., *
18 * 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. *
19 ***************************************************************************/
21 #include "bufferviewfilter.h"
25 #include "networkmodel.h"
27 /*****************************************
28 * The Filter for the Tree View
29 *****************************************/
30 BufferViewFilter::BufferViewFilter(QAbstractItemModel *model, const Modes &filtermode, const QList<NetworkId> &nets)
31 : QSortFilterProxyModel(model),
33 networks(QSet<NetworkId>::fromList(nets))
35 setSourceModel(model);
36 setSortCaseSensitivity(Qt::CaseInsensitive);
39 Qt::ItemFlags BufferViewFilter::flags(const QModelIndex &index) const {
40 Qt::ItemFlags flags = mapToSource(index).flags();
41 if(mode & FullCustom) {
42 if(index == QModelIndex() || index.parent() == QModelIndex())
43 flags |= Qt::ItemIsDropEnabled;
48 bool BufferViewFilter::dropMimeData(const QMimeData *data, Qt::DropAction action, int row, int column, const QModelIndex &parent) {
49 // drops have to occur in the open field
50 if(parent != QModelIndex())
51 return QSortFilterProxyModel::dropMimeData(data, action, row, column, parent);
53 if(!NetworkModel::mimeContainsBufferList(data))
56 QList< QPair<NetworkId, BufferId> > bufferList = NetworkModel::mimeDataToBufferList(data);
60 for(int i = 0; i < bufferList.count(); i++) {
61 netId = bufferList[i].first;
62 bufferId = bufferList[i].second;
63 if(!networks.contains(netId)) {
71 void BufferViewFilter::addBuffer(const BufferId &bufferuid) {
72 if(!buffers.contains(bufferuid)) {
78 void BufferViewFilter::removeBuffer(const QModelIndex &index) {
79 if(!(mode & FullCustom))
80 return; // only custom buffers can be customized... obviously... :)
82 if(index.parent() == QModelIndex())
83 return; // only child elements can be deleted
85 bool lastBuffer = (rowCount(index.parent()) == 1);
86 NetworkId netId = index.data(NetworkModel::NetworkIdRole).value<NetworkId>();
87 BufferId bufferuid = index.data(NetworkModel::BufferIdRole).value<BufferId>();
89 if(buffers.contains(bufferuid)) {
90 buffers.remove(bufferuid);
93 networks.remove(netId);
94 Q_ASSERT(!networks.contains(netId));
103 bool BufferViewFilter::filterAcceptBuffer(const QModelIndex &source_bufferIndex) const {
104 BufferInfo::Type bufferType = (BufferInfo::Type) source_bufferIndex.data(NetworkModel::BufferTypeRole).toInt();
106 if((mode & NoChannels) && bufferType == BufferInfo::ChannelBuffer)
108 if((mode & NoQueries) && bufferType == BufferInfo::QueryBuffer)
110 if((mode & NoServers) && bufferType == BufferInfo::StatusBuffer)
113 // bool isActive = source_bufferIndex.data(NetworkModel::BufferActiveRole).toBool();
114 // if((mode & NoActive) && isActive)
116 // if((mode & NoInactive) && !isActive)
119 if((mode & FullCustom)) {
120 BufferId bufferuid = source_bufferIndex.data(NetworkModel::BufferIdRole).value<BufferId>();
121 return buffers.contains(bufferuid);
127 bool BufferViewFilter::filterAcceptNetwork(const QModelIndex &source_index) const {
128 NetworkId net = source_index.data(NetworkModel::NetworkIdRole).value<NetworkId>();
129 return !((mode & (SomeNets | FullCustom)) && !networks.contains(net));
132 bool BufferViewFilter::filterAcceptsRow(int source_row, const QModelIndex &source_parent) const {
133 QModelIndex child = sourceModel()->index(source_row, 0, source_parent);
135 if(!child.isValid()) {
136 qDebug() << "filterAcceptsRow has been called with an invalid Child";
140 if(source_parent == QModelIndex())
141 return filterAcceptNetwork(child);
143 return filterAcceptBuffer(child);
146 bool BufferViewFilter::lessThan(const QModelIndex &left, const QModelIndex &right) const {
147 int lefttype = left.data(NetworkModel::BufferTypeRole).toInt();
148 int righttype = right.data(NetworkModel::BufferTypeRole).toInt();
150 if(lefttype != righttype)
151 return lefttype < righttype;
153 return QSortFilterProxyModel::lessThan(left, right);
156 QVariant BufferViewFilter::data(const QModelIndex &index, int role) const {
157 if(role == Qt::ForegroundRole)
158 return foreground(index);
160 return QSortFilterProxyModel::data(index, role);
163 QVariant BufferViewFilter::foreground(const QModelIndex &index) const {
164 if(!index.data(NetworkModel::ItemActiveRole).toBool())
165 return QColor(Qt::gray);
167 BufferItem::ActivityLevel activity = (BufferItem::ActivityLevel)index.data(NetworkModel::BufferActivityRole).toInt();
169 if(activity & BufferItem::Highlight)
170 return QColor(Qt::magenta);
171 if(activity & BufferItem::NewMessage)
172 return QColor(Qt::green);
173 if(activity & BufferItem::OtherActivity)
174 return QColor(Qt::darkGreen);
176 return QColor(Qt::black);
178 // FIXME:: make colors configurable;