Allow to sort by rating, which can be quite useful in combination with the "Show...

[dolphin.git] / src / dolphinitemcategorizer.cpp

diff --git a/src/dolphinitemcategorizer.cpp b/src/dolphinitemcategorizer.cpp
index d599aad8eb4b42bc4fee233aa70e93477ca548b8..7dc19ab92fded0020ea3ee789012bf93c4729479 100644 (file)
--- a/src/dolphinitemcategorizer.cpp
+++ b/src/dolphinitemcategorizer.cpp
@@ -22,57 +22,159 @@
 
 #include "dolphinview.h"
 
-#include <klocale.h>
+#ifdef HAVE_NEPOMUK
+#include <config-nepomuk.h>
+#include <nepomuk/global.h>
+#include <nepomuk/resource.h>
+#endif
+
+#include <kdatetime.h>
 #include <kdirmodel.h>
+#include <kfileitem.h>
+#include <klocale.h>
+#include <kurl.h>
 
-#include <QSortFilterProxyModel>
+#include <QtGui/QSortFilterProxyModel>
 
 DolphinItemCategorizer::DolphinItemCategorizer() :
-        KItemCategorizer()
-{}
+    KItemCategorizer()
+{
+}
 
 DolphinItemCategorizer::~DolphinItemCategorizer()
-{}
+{
+}
 
 QString DolphinItemCategorizer::categoryForItem(const QModelIndex& index,
                                                 int sortRole)
 {
     QString retString;
 
-    if (!index.isValid()) {
+    if (!index.isValid())
+    {
         return retString;
     }
 
-    // KDirModel checks columns to know to which role are
-    // we talking about
-    QModelIndex theIndex = index.model()->index(index.row(),
-                           sortRole,
-                           index.parent());
-
-    const QSortFilterProxyModel* proxyModel = static_cast<const QSortFilterProxyModel*>(index.model());
-    const KDirModel* dirModel = static_cast<const KDirModel*>(proxyModel->sourceModel());
-
-    QVariant data = theIndex.model()->data(theIndex, Qt::DisplayRole);
-
-    QModelIndex mappedIndex = proxyModel->mapToSource(theIndex);
-    KFileItem* item = dirModel->itemForIndex(mappedIndex);
-
-    switch (sortRole) {
-    case DolphinView::SortByName:
-        retString = data.toString().toUpper().at(0);
-        break;
-    case DolphinView::SortBySize:
-        int fileSize = (item) ? item->size() : -1;
-        if (item && item->isDir()) {
-            retString = i18n("Unknown");
-        } else if (fileSize < 5242880) {
-            retString = i18n("Small");
-        } else if (fileSize < 10485760) {
-            retString = i18n("Medium");
-        } else {
-            retString = i18n("Big");
+    const KDirModel *dirModel = qobject_cast<const KDirModel*>(index.model());
+    KFileItem *item = dirModel->itemForIndex(index);
+
+    switch (sortRole)
+    {
+        case DolphinView::SortByName:
+        {
+            // KDirModel checks columns to know to which role are
+            // we talking about
+            QModelIndex theIndex = index.model()->index(index.row(),
+                                                        KDirModel::Name,
+                                                        index.parent());
+
+            if (!theIndex.isValid()) {
+                return retString;
+            }
+
+            QVariant data = theIndex.model()->data(theIndex, Qt::DisplayRole);
+            if (data.toString().size())
+            {
+                if (!item->isHidden() && data.toString().at(0).isLetter())
+                    retString = data.toString().toUpper().at(0);
+                else if (item->isHidden() && data.toString().at(0) == '.' &&
+                         data.toString().at(1).isLetter())
+                    retString = data.toString().toUpper().at(1);
+                else if (item->isHidden() && data.toString().at(0) == '.' &&
+                         !data.toString().at(1).isLetter())
+                    retString = i18n("Others");
+                else if (item->isHidden() && data.toString().at(0) != '.')
+                    retString = data.toString().toUpper().at(0);
+                else if (item->isHidden())
+                    retString = data.toString().toUpper().at(0);
+                else
+                {
+                    bool validCategory = false;
+
+                    const QString str(data.toString().toUpper());
+                    const QChar* currA = str.unicode();
+                    while (!currA->isNull() && !validCategory) {
+                        if (currA->isLetter())
+                            validCategory = true;
+                        else if (currA->isDigit())
+                            return i18n("Others");
+                        else
+                            ++currA;
+                    }
+
+                    if (!validCategory)
+                        retString = i18n("Others");
+                    else
+                        retString = *currA;
+                }
+            }
+            break;
+        }
+
+        case DolphinView::SortByDate:
+        {
+            KDateTime modifiedTime;
+            modifiedTime.setTime_t(item->time(KIO::UDS_MODIFICATION_TIME));
+            modifiedTime = modifiedTime.toLocalZone();
+
+            if (modifiedTime.daysTo(KDateTime::currentLocalDateTime()) == 0)
+                retString = i18n("Today");
+            else if (modifiedTime.daysTo(KDateTime::currentLocalDateTime()) == 1)
+                retString = i18n("Yesterday");
+            else if (modifiedTime.daysTo(KDateTime::currentLocalDateTime()) < 7)
+                retString = i18n("Less than a week");
+            else if (modifiedTime.daysTo(KDateTime::currentLocalDateTime()) < 31)
+                retString = i18n("Less than a month");
+            else if (modifiedTime.daysTo(KDateTime::currentLocalDateTime()) < 365)
+                retString = i18n("Less than a year");
+            else
+                retString = i18n("More than a year");
+            break;
+        }
+
+        case DolphinView::SortByPermissions:
+            retString = item->permissionsString();
+            break;
+
+        case DolphinView::SortByOwner:
+            retString = item->user();
+            break;
+
+        case DolphinView::SortByGroup:
+            retString = item->group();
+            break;
+
+        case DolphinView::SortBySize: {
+            const int fileSize = item ? item->size() : -1;
+            if (item && item->isDir()) {
+                retString = i18n("Folders");
+            } else if (fileSize < 5242880) {
+                retString = i18nc("Size", "Small");
+            } else if (fileSize < 10485760) {
+                retString = i18nc("Size", "Medium");
+            } else {
+                retString = i18nc("Size", "Big");
+            }
+            break;
+        }
+
+        case DolphinView::SortByType:
+            retString = item->mimeComment();
+            break;
+
+#ifdef HAVE_NEPOMUK
+        case DolphinView::SortByRating: {
+            KFileItem* item = dirModel->itemForIndex(index);
+            if (item != 0) {
+                const Nepomuk::Resource resource(item->url().url(), Nepomuk::NFO::File());
+                const quint32 rating = resource.rating();
+                retString = i18np("1 star", "%1 stars", rating);
+            }
+            break;
         }
-        break;
+        case DolphinView::SortByTags:
+            break;
+#endif
     }
 
     return retString;