src/kitemviews/private/kfileitemmodelfilter.cpp

   1 /***************************************************************************
   2  *   Copyright (C) 2011 by Janardhan Reddy                                 *
   3  *   <annapareddyjanardhanreddy@gmail.com>                                 *
   4  *                                                                         *
   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) any later version.                                   *
   9  *                                                                         *
  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.                          *
  14  *                                                                         *
  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  *   51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA            *
  19  ***************************************************************************/
  20
  21 #include "kfileitemmodelfilter.h"
  22
  23 #include <KFileItem>
  24 #include <QRegExp>
  25
  26
  27 KFileItemModelFilter::KFileItemModelFilter() :
  28     m_useRegExp(false),
  29     m_regExp(0),
  30     m_lowerCasePattern(),
  31     m_pattern()
  32 {
  33 }
  34
  35 KFileItemModelFilter::~KFileItemModelFilter()
  36 {
  37     delete m_regExp;
  38     m_regExp = 0;
  39 }
  40
  41 void KFileItemModelFilter::setPattern(const QString& filter)
  42 {
  43     m_pattern = filter;
  44     m_lowerCasePattern = filter.toLower();
  45
  46     if (filter.contains('*') || filter.contains('?') || filter.contains('[')) {
  47         if (!m_regExp) {
  48             m_regExp = new QRegExp();
  49             m_regExp->setCaseSensitivity(Qt::CaseInsensitive);
  50             m_regExp->setMinimal(false);
  51             m_regExp->setPatternSyntax(QRegExp::WildcardUnix);
  52         }
  53         m_regExp->setPattern(filter);
  54         m_useRegExp = m_regExp->isValid();
  55     } else {
  56         m_useRegExp = false;
  57     }
  58 }
  59
  60 QString KFileItemModelFilter::pattern() const
  61 {
  62     return m_pattern;
  63 }
  64
  65 void KFileItemModelFilter::setMimeTypes(const QStringList& types)
  66 {
  67     m_mimeTypes = types;
  68 }
  69
  70 QStringList KFileItemModelFilter::mimeTypes() const
  71 {
  72     return m_mimeTypes;
  73 }
  74
  75 bool KFileItemModelFilter::hasSetFilters() const
  76 {
  77     return (!m_pattern.isEmpty() || !m_mimeTypes.isEmpty());
  78 }
  79
  80
  81 bool KFileItemModelFilter::matches(const KFileItem& item) const
  82 {
  83     const bool hasPatternFilter = !m_pattern.isEmpty();
  84     const bool hasMimeTypesFilter = !m_mimeTypes.isEmpty();
  85
  86     // If no filter is set, return true.
  87     if (!hasPatternFilter && !hasMimeTypesFilter) {
  88         return true;
  89     }
  90
  91     // If both filters are set, return true when both filters are matched
  92     if (hasPatternFilter && hasMimeTypesFilter) {
  93         return (matchesPattern(item) && matchesType(item));
  94     }
  95
  96     // If only one filter is set, return true when that filter is matched
  97     if (hasPatternFilter) {
  98         return matchesPattern(item);
  99     }
 100
 101     return matchesType(item);
 102 }
 103
 104 bool KFileItemModelFilter::matchesPattern(const KFileItem& item) const
 105 {
 106     if (m_useRegExp) {
 107         return m_regExp->exactMatch(item.text());
 108     } else {
 109         return item.text().toLower().contains(m_lowerCasePattern);
 110     }
 111 }
 112
 113 bool KFileItemModelFilter::matchesType(const KFileItem& item) const
 114 {
 115     foreach (const QString& mimeType, m_mimeTypes) {
 116         if (item.mimetype() == mimeType) {
 117             return true;
 118         }
 119     }
 120
 121     return m_mimeTypes.isEmpty();
 122 }