File indexing completed on 2024-04-28 04:18:52

0001 // vim: set tabstop=4 shiftwidth=4 expandtab:
0002 /*
0003 Gwenview: an image viewer
0004 Copyright 2012 Aurélien Gâteau <agateau@kde.org>
0005 
0006 This program is free software; you can redistribute it and/or
0007 modify it under the terms of the GNU General Public License
0008 as published by the Free Software Foundation; either version 2
0009 of the License, or (at your option) any later version.
0010 
0011 This program is distributed in the hope that it will be useful,
0012 but WITHOUT ANY WARRANTY; without even the implied warranty of
0013 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
0014 GNU General Public License for more details.
0015 
0016 You should have received a copy of the GNU General Public License
0017 along with this program; if not, write to the Free Software
0018 Foundation, Inc., 51 Franklin Street, Fifth Floor, Cambridge, MA 02110-1301, USA.
0019 
0020 */
0021 // Self
0022 #include "kindproxymodel.h"
0023 
0024 // Local
0025 
0026 // KF
0027 #include <KDirModel>
0028 #include <KFileItem>
0029 
0030 // Qt
0031 
0032 namespace Gwenview
0033 {
0034 struct KindProxyModelPrivate {
0035     MimeTypeUtils::Kinds mKindFilter;
0036 };
0037 
0038 KindProxyModel::KindProxyModel(QObject *parent)
0039     : QSortFilterProxyModel(parent)
0040     , d(new KindProxyModelPrivate)
0041 {
0042 }
0043 
0044 KindProxyModel::~KindProxyModel()
0045 {
0046     delete d;
0047 }
0048 
0049 void KindProxyModel::setKindFilter(MimeTypeUtils::Kinds filter)
0050 {
0051     if (d->mKindFilter != filter) {
0052         d->mKindFilter = filter;
0053         invalidateFilter();
0054     }
0055 }
0056 
0057 MimeTypeUtils::Kinds KindProxyModel::kindFilter() const
0058 {
0059     return d->mKindFilter;
0060 }
0061 
0062 bool KindProxyModel::filterAcceptsRow(int sourceRow, const QModelIndex &sourceParent) const
0063 {
0064     if (d->mKindFilter == MimeTypeUtils::Kinds()) {
0065         return true;
0066     }
0067     const QModelIndex index = sourceModel()->index(sourceRow, 0, sourceParent);
0068     const KFileItem fileItem = index.data(KDirModel::FileItemRole).value<KFileItem>();
0069     if (fileItem.isNull()) {
0070         return false;
0071     }
0072     const MimeTypeUtils::Kinds kind = MimeTypeUtils::fileItemKind(fileItem);
0073     return d->mKindFilter & kind;
0074 }
0075 
0076 } // namespace
0077 
0078 #include "moc_kindproxymodel.cpp"