src/kitemviews/kfileitemmodel.h

   1 /***************************************************************************
   2  *   Copyright (C) 2011 by Peter Penz <peter.penz19@gmail.com>             *
   3  *                                                                         *
   4  *   This program is free software; you can redistribute it and/or modify  *
   5  *   it under the terms of the GNU General Public License as published by  *
   6  *   the Free Software Foundation; either version 2 of the License, or     *
   7  *   (at your option) any later version.                                   *
   8  *                                                                         *
   9  *   This program is distributed in the hope that it will be useful,       *
  10  *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
  11  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
  12  *   GNU General Public License for more details.                          *
  13  *                                                                         *
  14  *   You should have received a copy of the GNU General Public License     *
  15  *   along with this program; if not, write to the                         *
  16  *   Free Software Foundation, Inc.,                                       *
  17  *   51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA            *
  18  ***************************************************************************/
  19
  20 #ifndef KFILEITEMMODEL_H
  21 #define KFILEITEMMODEL_H
  22
  23 #include <libdolphin_export.h>
  24 #include <KFileItemList>
  25 #include <KUrl>
  26 #include <kitemviews/kitemmodelbase.h>
  27 #include <kitemviews/private/kfileitemmodelfilter.h>
  28
  29 #include <QHash>
  30
  31 class KFileItemModelDirLister;
  32 class QTimer;
  33
  34 /**
  35  * @brief KItemModelBase implementation for KFileItems.
  36  *
  37  * Allows to load items of a directory. Sorting and grouping of
  38  * items are supported. Roles that are not part of KFileItem can
  39  * be added with KFileItemModel::setData().
  40  *
  41  * Recursive expansion of sub-directories is supported by
  42  * KFileItemModel::setExpanded().
  43  */
  44 class LIBDOLPHINPRIVATE_EXPORT KFileItemModel : public KItemModelBase
  45 {
  46     Q_OBJECT
  47
  48 public:
  49     explicit KFileItemModel(QObject* parent = 0);
  50     virtual ~KFileItemModel();
  51
  52     /**
  53      * Loads the directory specified by \a url. The signals
  54      * directoryLoadingStarted(), directoryLoadingProgress() and directoryLoadingCompleted()
  55      * indicate the current state of the loading process. The items
  56      * of the directory are added after the loading has been completed.
  57      */
  58     void loadDirectory(const KUrl& url);
  59
  60     /**
  61      * Throws away all currently loaded items and refreshes the directory
  62      * by reloading all items again.
  63      */
  64     void refreshDirectory(const KUrl& url);
  65
  66     /**
  67      * @return Parent directory of the items that are shown. In case
  68      *         if a directory tree is shown, KFileItemModel::dir() returns
  69      *         the root-parent of all items.
  70      * @see rootItem()
  71      */
  72     KUrl directory() const;
  73
  74     /**
  75      * Cancels the loading of a directory which has been started by either
  76      * loadDirectory() or refreshDirectory().
  77      */
  78     void cancelDirectoryLoading();
  79
  80     virtual int count() const;
  81     virtual QHash<QByteArray, QVariant> data(int index) const;
  82     virtual bool setData(int index, const QHash<QByteArray, QVariant>& values);
  83
  84     /**
  85      * Sets a separate sorting with directories first (true) or a mixed
  86      * sorting of files and directories (false).
  87      */
  88     void setSortDirectoriesFirst(bool dirsFirst);
  89     bool sortDirectoriesFirst() const;
  90
  91     void setShowHiddenFiles(bool show);
  92     bool showHiddenFiles() const;
  93
  94     /**
  95      * If set to true, only directories are shown as items of the model. Files
  96      * are ignored.
  97      */
  98     void setShowDirectoriesOnly(bool enabled);
  99     bool showDirectoriesOnly() const;
 100
 101     /** @reimp */
 102     virtual QMimeData* createMimeData(const QSet<int>& indexes) const;
 103
 104     /** @reimp */
 105     virtual int indexForKeyboardSearch(const QString& text, int startFromIndex = 0) const;
 106
 107     /** @reimp */
 108     virtual bool supportsDropping(int index) const;
 109
 110     /** @reimp */
 111     virtual QString roleDescription(const QByteArray& role) const;
 112
 113     /** @reimp */
 114     virtual QList<QPair<int, QVariant> > groups() const;
 115
 116     /**
 117      * @return The file-item for the index \a index. If the index is in a valid
 118      *         range it is assured that the file-item is not null. The runtime
 119      *         complexity of this call is O(1).
 120      */
 121     KFileItem fileItem(int index) const;
 122
 123     /**
 124      * @return The file-item for the url \a url. If no file-item with the given
 125      *         URL is found KFileItem::isNull() will be true for the returned
 126      *         file-item. The runtime complexity of this call is O(1).
 127      */
 128     KFileItem fileItem(const KUrl& url) const;
 129
 130     /**
 131      * @return The index for the file-item \a item. -1 is returned if no file-item
 132      *         is found or if the file-item is null. The runtime
 133      *         complexity of this call is O(1).
 134      */
 135     int index(const KFileItem& item) const;
 136
 137     /**
 138      * @return The index for the URL \a url. -1 is returned if no file-item
 139      *         is found. The runtime complexity of this call is O(1).
 140      */
 141     int index(const KUrl& url) const;
 142
 143     /**
 144      * @return Root item of all items representing the item
 145      *         for KFileItemModel::dir().
 146      */
 147     KFileItem rootItem() const;
 148
 149     /**
 150      * Clears all items of the model.
 151      */
 152     void clear();
 153
 154     /**
 155      * Sets the roles that should be shown for each item.
 156      */
 157     void setRoles(const QSet<QByteArray>& roles);
 158     QSet<QByteArray> roles() const;
 159
 160     virtual bool setExpanded(int index, bool expanded);
 161     virtual bool isExpanded(int index) const;
 162     virtual bool isExpandable(int index) const;
 163     virtual int expandedParentsCount(int index) const;
 164
 165     QSet<KUrl> expandedDirectories() const;
 166
 167     /**
 168      * Marks the URLs in \a urls as sub-directories which were expanded previously.
 169      * After calling loadDirectory() or refreshDirectory() the marked sub-directories
 170      * will be expanded step-by-step.
 171      */
 172     void restoreExpandedDirectories(const QSet<KUrl>& urls);
 173
 174     /**
 175      * Expands all parent-directories of the item \a url.
 176      */
 177     void expandParentDirectories(const KUrl& url);
 178
 179     void setNameFilter(const QString& nameFilter);
 180     QString nameFilter() const;
 181
 182     void setMimeTypeFilters(const QStringList& filters);
 183     QStringList mimeTypeFilters() const;
 184
 185     struct RoleInfo
 186     {   QByteArray role;
 187         QString translation;
 188         QString group;
 189         bool requiresNepomuk;
 190         bool requiresIndexer;
 191     };
 192
 193     /**
 194      * @return Provides static information for all available roles that
 195      *         are supported by KFileItemModel. Some roles can only be
 196      *         determined if Nepomuk is enabled and/or the Nepomuk
 197      *         indexing is enabled.
 198      */
 199     static QList<RoleInfo> rolesInformation();
 200
 201 signals:
 202     /**
 203      * Is emitted if the loading of a directory has been started. It is
 204      * assured that a signal directoryLoadingCompleted() will be send after
 205      * the loading has been finished. For tracking the loading progress
 206      * the signal directoryLoadingProgress() gets emitted in between.
 207      */
 208     void directoryLoadingStarted();
 209
 210     /**
 211      * Is emitted after the loading of a directory has been completed or new
 212      * items have been inserted to an already loaded directory. Usually
 213      * one or more itemsInserted() signals are emitted before loadingCompleted()
 214      * (the only exception is loading an empty directory, where only a
 215      * loadingCompleted() signal gets emitted).
 216      */
 217     void directoryLoadingCompleted();
 218
 219     /**
 220      * Is emitted after the loading of a directory has been canceled.
 221      */
 222     void directoryLoadingCanceled();
 223
 224     /**
 225      * Informs about the progress in percent when loading a directory. It is assured
 226      * that the signal directoryLoadingStarted() has been emitted before.
 227      */
 228     void directoryLoadingProgress(int percent);
 229
 230     /**
 231      * Is emitted if the sort-role gets resolved asynchronously and provides
 232      * the progress-information of the sorting in percent. It is assured
 233      * that the last sortProgress-signal contains 100 as value.
 234      */
 235     void directorySortingProgress(int percent);
 236
 237     /**
 238      * Is emitted if an information message (e.g. "Connecting to host...")
 239      * should be shown.
 240      */
 241     void infoMessage(const QString& message);
 242
 243     /**
 244      * Is emitted if an error message (e.g. "Unknown location")
 245      * should be shown.
 246      */
 247     void errorMessage(const QString& message);
 248
 249     /**
 250      * Is emitted if a redirection from the current URL \a oldUrl
 251      * to the new URL \a newUrl has been done.
 252      */
 253     void directoryRedirection(const KUrl& oldUrl, const KUrl& newUrl);
 254
 255     /**
 256      * Is emitted when the URL passed by KFileItemModel::setUrl() represents a file.
 257      * In this case no signal errorMessage() will be emitted.
 258      */
 259     void urlIsFileError(const KUrl& url);
 260
 261 protected:
 262     virtual void onGroupedSortingChanged(bool current);
 263     virtual void onSortRoleChanged(const QByteArray& current, const QByteArray& previous);
 264     virtual void onSortOrderChanged(Qt::SortOrder current, Qt::SortOrder previous);
 265
 266 private slots:
 267     /**
 268      * Resorts all items dependent on the set sortRole(), sortOrder()
 269      * and foldersFirst() settings.
 270      */
 271     void resortAllItems();
 272
 273     void slotCompleted();
 274     void slotCanceled();
 275     void slotNewItems(const KFileItemList& items);
 276     void slotItemsDeleted(const KFileItemList& items);
 277     void slotRefreshItems(const QList<QPair<KFileItem, KFileItem> >& items);
 278     void slotClear();
 279     void slotClear(const KUrl& url);
 280     void slotNaturalSortingChanged();
 281
 282     void dispatchPendingItemsToInsert();
 283
 284 private:
 285     enum RoleType {
 286         // User visible roles:
 287         NoRole, NameRole, SizeRole, DateRole, PermissionsRole, OwnerRole,
 288         GroupRole, TypeRole, DestinationRole, PathRole,
 289         // User visible roles available with Nepomuk:
 290         CommentRole, TagsRole, RatingRole, ImageSizeRole, OrientationRole,
 291         WordCountRole, LineCountRole, ArtistRole, AlbumRole, DurationRole, TrackRole,
 292         CopiedFromRole,
 293         // Non-visible roles:
 294         IsDirRole, IsLinkRole, IsExpandedRole, IsExpandableRole, ExpandedParentsCountRole,
 295         // Mandatory last entry:
 296         RolesCount
 297     };
 298
 299     struct ItemData
 300     {
 301         KFileItem item;
 302         QHash<QByteArray, QVariant> values;
 303         ItemData* parent;
 304     };
 305
 306     void insertItems(const KFileItemList& items);
 307     void removeItems(const KFileItemList& items);
 308
 309     /**
 310      * Helper method for insertItems() and removeItems(): Creates
 311      * a list of ItemData elements based on the given items.
 312      * Note that the ItemData instances are created dynamically and
 313      * must be deleted by the caller.
 314      */
 315     QList<ItemData*> createItemDataList(const KFileItemList& items) const;
 316
 317     void removeExpandedItems();
 318
 319     /**
 320      * Resets all values from m_requestRole to false.
 321      */
 322     void resetRoles();
 323
 324     /**
 325      * @return Role-type for the given role.
 326      *         Runtime complexity is O(1).
 327      */
 328     RoleType typeForRole(const QByteArray& role) const;
 329
 330     /**
 331      * @return Role-byte-array for the given role-type.
 332      *         Runtime complexity is O(1).
 333      */
 334     QByteArray roleForType(RoleType roleType) const;
 335
 336     QHash<QByteArray, QVariant> retrieveData(const KFileItem& item) const;
 337
 338     /**
 339      * @return True if the item-data \a a should be ordered before the item-data
 340      *         \b. The item-data may have different parent-items.
 341      */
 342     bool lessThan(const ItemData* a, const ItemData* b) const;
 343
 344     /**
 345      * Helper method for lessThan() and expandedParentsCountCompare(): Compares
 346      * the passed item-data using m_sortRole as criteria. Both items must
 347      * have the same parent item, otherwise the comparison will be wrong.
 348      */
 349     int sortRoleCompare(const ItemData* a, const ItemData* b) const;
 350
 351     int stringCompare(const QString& a, const QString& b) const;
 352
 353     /**
 354      * Compares the expansion level of both items. The "expansion level" is defined
 355      * by the number of parent directories. However simply comparing just the numbers
 356      * is not sufficient, it is also important to check the hierarchy for having
 357      * a correct order like shown in a tree.
 358      */
 359     int expandedParentsCountCompare(const ItemData* a, const ItemData* b) const;
 360
 361     /**
 362      * Helper method for expandedParentsCountCompare().
 363      */
 364     QString subPath(const KFileItem& item,
 365                     const QString& itemPath,
 366                     int start,
 367                     bool* isDir) const;
 368
 369     bool useMaximumUpdateInterval() const;
 370
 371     QList<QPair<int, QVariant> > nameRoleGroups() const;
 372     QList<QPair<int, QVariant> > sizeRoleGroups() const;
 373     QList<QPair<int, QVariant> > dateRoleGroups() const;
 374     QList<QPair<int, QVariant> > permissionRoleGroups() const;
 375     QList<QPair<int, QVariant> > ratingRoleGroups() const;
 376     QList<QPair<int, QVariant> > genericStringRoleGroups(const QByteArray& typeForRole) const;
 377
 378     /**
 379      * Helper method for all xxxRoleGroups() methods to check whether the
 380      * item with the given index is a child-item. A child-item is defined
 381      * as item having an expansion-level > 0. All xxxRoleGroups() methods
 382      * should skip the grouping if the item is a child-item (although
 383      * KItemListView would be capable to show sub-groups in groups this
 384      * results in visual clutter for most usecases).
 385      */
 386     bool isChildItem(int index) const;
 387
 388     /**
 389      * @return Recursive list of child items that have \a item as upper most parent.
 390      */
 391     KFileItemList childItems(const KFileItem& item) const;
 392
 393     /**
 394      * Is invoked by KFileItemModelRolesUpdater and results in emitting the
 395      * sortProgress signal with a percent-value of the progress.
 396      */
 397     void emitSortProgress(int resolvedCount);
 398
 399     /**
 400      * Applies the filters set through @ref setNameFilter and @ref setMimeTypeFilters.
 401      */
 402     void applyFilters();
 403
 404     /**
 405      * Maps the QByteArray-roles to RoleTypes and provides translation- and
 406      * group-contexts.
 407      */
 408     struct RoleInfoMap
 409     {
 410         const char* const role;
 411         const RoleType roleType;
 412         const char* const roleTranslationContext;
 413         const char* const roleTranslation;
 414         const char* const groupTranslationContext;
 415         const char* const groupTranslation;
 416         const bool requiresNepomuk;
 417         const bool requiresIndexer;
 418     };
 419
 420     /**
 421      * @return Map of user visible roles that are accessible by KFileItemModel::rolesInformation().
 422      */
 423     static const RoleInfoMap* rolesInfoMap(int& count);
 424
 425     /**
 426      * Determines the MIME-types of all items that can be done within
 427      * the given timeout.
 428      */
 429     static void determineMimeTypes(const KFileItemList& items, int timeout);
 430
 431 private:
 432     KFileItemModelDirLister* m_dirLister;
 433
 434     bool m_naturalSorting;
 435     bool m_sortDirsFirst;
 436
 437     RoleType m_sortRole;
 438     int m_sortingProgressPercent; // Value of directorySortingProgress() signal
 439     QSet<QByteArray> m_roles;
 440     Qt::CaseSensitivity m_caseSensitivity;
 441
 442     QList<ItemData*> m_itemData;
 443     QHash<KUrl, int> m_items; // Allows O(1) access for KFileItemModel::index(const KFileItem& item)
 444
 445     KFileItemModelFilter m_filter;
 446     QSet<KFileItem> m_filteredItems; // Items that got hidden by KFileItemModel::setNameFilter()
 447
 448     bool m_requestRole[RolesCount];
 449
 450     QTimer* m_maximumUpdateIntervalTimer;
 451     QTimer* m_resortAllItemsTimer;
 452     KFileItemList m_pendingItemsToInsert;
 453
 454     // Cache for KFileItemModel::groups()
 455     mutable QList<QPair<int, QVariant> > m_groups;
 456
 457     // Stores the smallest expansion level of the root-URL. Is required to calculate
 458     // the "expandedParentsCount" role in an efficient way. A value < 0 indicates a
 459     // special meaning:
 460     enum ExpandedParentsCountRootTypes
 461     {
 462         // m_expandedParentsCountRoot is uninitialized and must be determined by checking
 463         // the root URL from the KDirLister.
 464         UninitializedExpandedParentsCountRoot = -1,
 465         // All items should be forced to get an expanded parents count of 0 even if they
 466         // represent child items. This is useful for slaves that provide no parent items
 467         // for child items like e.g. the search IO slaves.
 468         ForceExpandedParentsCountRoot = -2
 469     };
 470     mutable int m_expandedParentsCountRoot;
 471
 472     // Stores the URLs of the expanded directories.
 473     QSet<KUrl> m_expandedDirs;
 474
 475     // URLs that must be expanded. The expanding is initially triggered in setExpanded()
 476     // and done step after step in slotCompleted().
 477     QSet<KUrl> m_urlsToExpand;
 478
 479     friend class KFileItemModelSortAlgorithm;  // Accesses lessThan() method
 480     friend class KFileItemModelRolesUpdater;   // Accesses emitSortProgress() method
 481     friend class KFileItemModelTest;           // For unit testing
 482     friend class KFileItemListViewTest;        // For unit testing
 483     friend class DolphinPart;                  // Accesses m_dirLister
 484 };
 485
 486 inline bool KFileItemModel::isChildItem(int index) const
 487 {
 488     return m_requestRole[ExpandedParentsCountRole] && m_itemData.at(index)->values.value("expandedParentsCount").toInt() > 0;
 489 }
 490
 491 #endif
 492
 493