tests/auto/xmlpatternsview/view/TreeSortFilter.cpp

   1 /****************************************************************************
   2 **
   3 ** Copyright (C) 2012 Nokia Corporation and/or its subsidiary(-ies).
   4 ** Contact: http://www.qt-project.org/
   5 **
   6 ** This file is part of the test suite of the Qt Toolkit.
   7 **
   8 ** $QT_BEGIN_LICENSE:LGPL$
   9 ** GNU Lesser General Public License Usage
  10 ** This file may be used under the terms of the GNU Lesser General Public
  11 ** License version 2.1 as published by the Free Software Foundation and
  12 ** appearing in the file LICENSE.LGPL included in the packaging of this
  13 ** file. Please review the following information to ensure the GNU Lesser
  14 ** General Public License version 2.1 requirements will be met:
  15 ** http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html.
  16 **
  17 ** In addition, as a special exception, Nokia gives you certain additional
  18 ** rights. These rights are described in the Nokia Qt LGPL Exception
  19 ** version 1.1, included in the file LGPL_EXCEPTION.txt in this package.
  20 **
  21 ** GNU General Public License Usage
  22 ** Alternatively, this file may be used under the terms of the GNU General
  23 ** Public License version 3.0 as published by the Free Software Foundation
  24 ** and appearing in the file LICENSE.GPL included in the packaging of this
  25 ** file. Please review the following information to ensure the GNU General
  26 ** Public License version 3.0 requirements will be met:
  27 ** http://www.gnu.org/copyleft/gpl.html.
  28 **
  29 ** Other Usage
  30 ** Alternatively, this file may be used in accordance with the terms and
  31 ** conditions contained in a signed written agreement between you and Nokia.
  32 **
  33 **
  34 **
  35 **
  36 **
  37 **
  38 ** $QT_END_LICENSE$
  39 **
  40 ****************************************************************************/
  41
  42 #include <QtDebug>
  43
  44 #include "TreeSortFilter.h"
  45
  46 using namespace QPatternistSDK;
  47
  48 TreeSortFilter::TreeSortFilter(QObject *p) : QSortFilterProxyModel(p)
  49 {
  50     Q_ASSERT(p);
  51 }
  52
  53 bool TreeSortFilter::lessThan(const QModelIndex &left,
  54                               const QModelIndex &right) const
  55 {
  56     const QVariant leftData(sourceModel()->data(left));
  57     const QVariant rightData(sourceModel()->data(right));
  58
  59     return numericLessThan(leftData.toString(), rightData.toString());
  60 }
  61
  62 bool TreeSortFilter::numericLessThan(const QString &l, const QString &r) const
  63 {
  64     QString ls(l);
  65     QString rs(r);
  66     const int len = (l.length() > r.length() ? r.length() : l.length());
  67
  68     for(int i = 0;i < len; ++i)
  69     {
  70         const QChar li(l.at(i));
  71         const QChar ri(r.at(i));
  72
  73         if(li >= QLatin1Char('0') &&
  74            li <= QLatin1Char('9') &&
  75            ri >= QLatin1Char('0') &&
  76            ri <= QLatin1Char('9'))
  77         {
  78             ls = l.mid(i);
  79             rs = r.mid(i);
  80             break;
  81         }
  82         else if(li != ri)
  83             break;
  84     }
  85
  86     const int ld = ls.toInt();
  87     const int rd = rs.toInt();
  88
  89     if(ld == rd)
  90         return ls.localeAwareCompare(rs) < 0;
  91     else
  92         return ld < rd;
  93 }
  94
  95 bool TreeSortFilter::filterAcceptsRow(int sourceRow, const QModelIndex &sourceParent) const
  96 {
  97     if(filterRegExp().isEmpty())
  98         return true;
  99
 100     QModelIndex current(sourceModel()->index(sourceRow, filterKeyColumn(), sourceParent));
 101
 102     if(sourceModel()->hasChildren(current))
 103     {
 104         bool atLeastOneValidChild = false;
 105         int i = 0;
 106         while(!atLeastOneValidChild)
 107         {
 108             const QModelIndex child(current.child(i, current.column()));
 109             if(!child.isValid())
 110                 // No valid child
 111                 break;
 112
 113             atLeastOneValidChild = filterAcceptsRow(i, current);
 114             i++;
 115         }
 116         return atLeastOneValidChild;
 117     }
 118
 119     return sourceModel()->data(current).toString().contains(filterRegExp());
 120 }
 121
 122 // vim: et:ts=4:sw=4:sts=4