]> cloud.milkyroute.net Git - dolphin.git/blob - src/kitemviews/kfileitemmodel.cpp
port Dolphin from KUrl to QUrl
[dolphin.git] / src / kitemviews / kfileitemmodel.cpp
1 /*****************************************************************************
2 * Copyright (C) 2011 by Peter Penz <peter.penz19@gmail.com> *
3 * Copyright (C) 2013 by Frank Reininghaus <frank78ac@googlemail.com> *
4 * Copyright (C) 2013 by Emmanuel Pescosta <emmanuelpescosta099@gmail.com> *
5 * *
6 * This program is free software; you can redistribute it and/or modify *
7 * it under the terms of the GNU General Public License as published by *
8 * the Free Software Foundation; either version 2 of the License, or *
9 * (at your option) any later version. *
10 * *
11 * This program is distributed in the hope that it will be useful, *
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of *
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the *
14 * GNU General Public License for more details. *
15 * *
16 * You should have received a copy of the GNU General Public License *
17 * along with this program; if not, write to the *
18 * Free Software Foundation, Inc., *
19 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA *
20 *****************************************************************************/
21
22 #include "kfileitemmodel.h"
23
24 #include <KGlobalSettings>
25 #include <KLocalizedString>
26 #include <KStringHandler>
27 #include <KDebug>
28 #include <kstringhandler_deprecated.h> //TODO: port to QCollator
29
30 #include "private/kfileitemmodelsortalgorithm.h"
31 #include "private/kfileitemmodeldirlister.h"
32
33 #include <QMimeData>
34 #include <QTimer>
35 #include <QWidget>
36
37 #include <algorithm>
38 #include <vector>
39
40 // #define KFILEITEMMODEL_DEBUG
41
42 KFileItemModel::KFileItemModel(QObject* parent) :
43 KItemModelBase("text", parent),
44 m_dirLister(0),
45 m_naturalSorting(KGlobalSettings::naturalSorting()),
46 m_sortDirsFirst(true),
47 m_sortRole(NameRole),
48 m_sortingProgressPercent(-1),
49 m_roles(),
50 m_caseSensitivity(Qt::CaseInsensitive),
51 m_itemData(),
52 m_items(),
53 m_filter(),
54 m_filteredItems(),
55 m_requestRole(),
56 m_maximumUpdateIntervalTimer(0),
57 m_resortAllItemsTimer(0),
58 m_pendingItemsToInsert(),
59 m_groups(),
60 m_expandedDirs(),
61 m_urlsToExpand()
62 {
63 m_dirLister = new KFileItemModelDirLister(this);
64 m_dirLister->setDelayedMimeTypes(true);
65
66 const QWidget* parentWidget = qobject_cast<QWidget*>(parent);
67 if (parentWidget) {
68 m_dirLister->setMainWindow(parentWidget->window());
69 }
70
71 connect(m_dirLister, &KFileItemModelDirLister::started, this, &KFileItemModel::directoryLoadingStarted);
72 connect(m_dirLister, static_cast<void(KFileItemModelDirLister::*)()>(&KFileItemModelDirLister::canceled), this, &KFileItemModel::slotCanceled);
73 connect(m_dirLister, static_cast<void(KFileItemModelDirLister::*)(const QUrl&)>(&KFileItemModelDirLister::completed), this, &KFileItemModel::slotCompleted);
74 connect(m_dirLister, &KFileItemModelDirLister::itemsAdded, this, &KFileItemModel::slotItemsAdded);
75 connect(m_dirLister, &KFileItemModelDirLister::itemsDeleted, this, &KFileItemModel::slotItemsDeleted);
76 connect(m_dirLister, &KFileItemModelDirLister::refreshItems, this, &KFileItemModel::slotRefreshItems);
77 connect(m_dirLister, static_cast<void(KFileItemModelDirLister::*)()>(&KFileItemModelDirLister::clear), this, &KFileItemModel::slotClear);
78 connect(m_dirLister, &KFileItemModelDirLister::infoMessage, this, &KFileItemModel::infoMessage);
79 connect(m_dirLister, &KFileItemModelDirLister::errorMessage, this, &KFileItemModel::errorMessage);
80 connect(m_dirLister, static_cast<void(KFileItemModelDirLister::*)(const QUrl&, const QUrl&)>(&KFileItemModelDirLister::redirection), this, &KFileItemModel::directoryRedirection);
81 connect(m_dirLister, &KFileItemModelDirLister::urlIsFileError, this, &KFileItemModel::urlIsFileError);
82
83 // Apply default roles that should be determined
84 resetRoles();
85 m_requestRole[NameRole] = true;
86 m_requestRole[IsDirRole] = true;
87 m_requestRole[IsLinkRole] = true;
88 m_roles.insert("text");
89 m_roles.insert("isDir");
90 m_roles.insert("isLink");
91
92 // For slow KIO-slaves like used for searching it makes sense to show results periodically even
93 // before the completed() or canceled() signal has been emitted.
94 m_maximumUpdateIntervalTimer = new QTimer(this);
95 m_maximumUpdateIntervalTimer->setInterval(2000);
96 m_maximumUpdateIntervalTimer->setSingleShot(true);
97 connect(m_maximumUpdateIntervalTimer, &QTimer::timeout, this, &KFileItemModel::dispatchPendingItemsToInsert);
98
99 // When changing the value of an item which represents the sort-role a resorting must be
100 // triggered. Especially in combination with KFileItemModelRolesUpdater this might be done
101 // for a lot of items within a quite small timeslot. To prevent expensive resortings the
102 // resorting is postponed until the timer has been exceeded.
103 m_resortAllItemsTimer = new QTimer(this);
104 m_resortAllItemsTimer->setInterval(500);
105 m_resortAllItemsTimer->setSingleShot(true);
106 connect(m_resortAllItemsTimer, &QTimer::timeout, this, &KFileItemModel::resortAllItems);
107
108 connect(KGlobalSettings::self(), &KGlobalSettings::naturalSortingChanged,
109 this, &KFileItemModel::slotNaturalSortingChanged);
110 }
111
112 KFileItemModel::~KFileItemModel()
113 {
114 qDeleteAll(m_itemData);
115 qDeleteAll(m_filteredItems.values());
116 qDeleteAll(m_pendingItemsToInsert);
117 }
118
119 void KFileItemModel::loadDirectory(const QUrl &url)
120 {
121 m_dirLister->openUrl(url);
122 }
123
124 void KFileItemModel::refreshDirectory(const QUrl &url)
125 {
126 // Refresh all expanded directories first (Bug 295300)
127 QHashIterator<QUrl, QUrl> expandedDirs(m_expandedDirs);
128 while (expandedDirs.hasNext()) {
129 expandedDirs.next();
130 m_dirLister->openUrl(expandedDirs.value(), KDirLister::Reload);
131 }
132
133 m_dirLister->openUrl(url, KDirLister::Reload);
134 }
135
136 QUrl KFileItemModel::directory() const
137 {
138 return m_dirLister->url();
139 }
140
141 void KFileItemModel::cancelDirectoryLoading()
142 {
143 m_dirLister->stop();
144 }
145
146 int KFileItemModel::count() const
147 {
148 return m_itemData.count();
149 }
150
151 QHash<QByteArray, QVariant> KFileItemModel::data(int index) const
152 {
153 if (index >= 0 && index < count()) {
154 ItemData* data = m_itemData.at(index);
155 if (data->values.isEmpty()) {
156 data->values = retrieveData(data->item, data->parent);
157 }
158
159 return data->values;
160 }
161 return QHash<QByteArray, QVariant>();
162 }
163
164 bool KFileItemModel::setData(int index, const QHash<QByteArray, QVariant>& values)
165 {
166 if (index < 0 || index >= count()) {
167 return false;
168 }
169
170 QHash<QByteArray, QVariant> currentValues = data(index);
171
172 // Determine which roles have been changed
173 QSet<QByteArray> changedRoles;
174 QHashIterator<QByteArray, QVariant> it(values);
175 while (it.hasNext()) {
176 it.next();
177 const QByteArray role = sharedValue(it.key());
178 const QVariant value = it.value();
179
180 if (currentValues[role] != value) {
181 currentValues[role] = value;
182 changedRoles.insert(role);
183 }
184 }
185
186 if (changedRoles.isEmpty()) {
187 return false;
188 }
189
190 m_itemData[index]->values = currentValues;
191 if (changedRoles.contains("text")) {
192 QUrl url = m_itemData[index]->item.url();
193 url = url.adjusted(QUrl::RemoveFilename);
194 url.setPath(url.path() + currentValues["text"].toString());
195 m_itemData[index]->item.setUrl(url);
196 }
197
198 emitItemsChangedAndTriggerResorting(KItemRangeList() << KItemRange(index, 1), changedRoles);
199
200 return true;
201 }
202
203 void KFileItemModel::setSortDirectoriesFirst(bool dirsFirst)
204 {
205 if (dirsFirst != m_sortDirsFirst) {
206 m_sortDirsFirst = dirsFirst;
207 resortAllItems();
208 }
209 }
210
211 bool KFileItemModel::sortDirectoriesFirst() const
212 {
213 return m_sortDirsFirst;
214 }
215
216 void KFileItemModel::setShowHiddenFiles(bool show)
217 {
218 m_dirLister->setShowingDotFiles(show);
219 m_dirLister->emitChanges();
220 if (show) {
221 dispatchPendingItemsToInsert();
222 }
223 }
224
225 bool KFileItemModel::showHiddenFiles() const
226 {
227 return m_dirLister->showingDotFiles();
228 }
229
230 void KFileItemModel::setShowDirectoriesOnly(bool enabled)
231 {
232 m_dirLister->setDirOnlyMode(enabled);
233 }
234
235 bool KFileItemModel::showDirectoriesOnly() const
236 {
237 return m_dirLister->dirOnlyMode();
238 }
239
240 QMimeData* KFileItemModel::createMimeData(const KItemSet& indexes) const
241 {
242 QMimeData* data = new QMimeData();
243
244 // The following code has been taken from KDirModel::mimeData()
245 // (kdelibs/kio/kio/kdirmodel.cpp)
246 // Copyright (C) 2006 David Faure <faure@kde.org>
247 QList<QUrl> urls;
248 QList<QUrl> mostLocalUrls;
249 bool canUseMostLocalUrls = true;
250 const ItemData* lastAddedItem = 0;
251
252 foreach (int index, indexes) {
253 const ItemData* itemData = m_itemData.at(index);
254 const ItemData* parent = itemData->parent;
255
256 while (parent && parent != lastAddedItem) {
257 parent = parent->parent;
258 }
259
260 if (parent && parent == lastAddedItem) {
261 // A parent of 'itemData' has been added already.
262 continue;
263 }
264
265 lastAddedItem = itemData;
266 const KFileItem& item = itemData->item;
267 if (!item.isNull()) {
268 urls << item.targetUrl();
269
270 bool isLocal;
271 mostLocalUrls << item.mostLocalUrl(isLocal);
272 if (!isLocal) {
273 canUseMostLocalUrls = false;
274 }
275 }
276 }
277
278 const bool different = canUseMostLocalUrls && mostLocalUrls != urls;
279 if (different) {
280 data->setUrls(mostLocalUrls);
281 } else {
282 data->setUrls(urls);
283 }
284
285 return data;
286 }
287
288 int KFileItemModel::indexForKeyboardSearch(const QString& text, int startFromIndex) const
289 {
290 startFromIndex = qMax(0, startFromIndex);
291 for (int i = startFromIndex; i < count(); ++i) {
292 if (fileItem(i).text().startsWith(text, Qt::CaseInsensitive)) {
293 return i;
294 }
295 }
296 for (int i = 0; i < startFromIndex; ++i) {
297 if (fileItem(i).text().startsWith(text, Qt::CaseInsensitive)) {
298 return i;
299 }
300 }
301 return -1;
302 }
303
304 bool KFileItemModel::supportsDropping(int index) const
305 {
306 const KFileItem item = fileItem(index);
307 return !item.isNull() && (item.isDir() || item.isDesktopFile());
308 }
309
310 QString KFileItemModel::roleDescription(const QByteArray& role) const
311 {
312 static QHash<QByteArray, QString> description;
313 if (description.isEmpty()) {
314 int count = 0;
315 const RoleInfoMap* map = rolesInfoMap(count);
316 for (int i = 0; i < count; ++i) {
317 description.insert(map[i].role, i18nc(map[i].roleTranslationContext, map[i].roleTranslation));
318 }
319 }
320
321 return description.value(role);
322 }
323
324 QList<QPair<int, QVariant> > KFileItemModel::groups() const
325 {
326 if (!m_itemData.isEmpty() && m_groups.isEmpty()) {
327 #ifdef KFILEITEMMODEL_DEBUG
328 QElapsedTimer timer;
329 timer.start();
330 #endif
331 switch (typeForRole(sortRole())) {
332 case NameRole: m_groups = nameRoleGroups(); break;
333 case SizeRole: m_groups = sizeRoleGroups(); break;
334 case DateRole: m_groups = dateRoleGroups(); break;
335 case PermissionsRole: m_groups = permissionRoleGroups(); break;
336 case RatingRole: m_groups = ratingRoleGroups(); break;
337 default: m_groups = genericStringRoleGroups(sortRole()); break;
338 }
339
340 #ifdef KFILEITEMMODEL_DEBUG
341 kDebug() << "[TIME] Calculating groups for" << count() << "items:" << timer.elapsed();
342 #endif
343 }
344
345 return m_groups;
346 }
347
348 KFileItem KFileItemModel::fileItem(int index) const
349 {
350 if (index >= 0 && index < count()) {
351 return m_itemData.at(index)->item;
352 }
353
354 return KFileItem();
355 }
356
357 KFileItem KFileItemModel::fileItem(const QUrl &url) const
358 {
359 const int indexForUrl = index(url);
360 if (indexForUrl >= 0) {
361 return m_itemData.at(indexForUrl)->item;
362 }
363 return KFileItem();
364 }
365
366 int KFileItemModel::index(const KFileItem& item) const
367 {
368 return index(item.url());
369 }
370
371 int KFileItemModel::index(const QUrl& url) const
372 {
373 const QUrl urlToFind = url.adjusted(QUrl::StripTrailingSlash);
374
375 const int itemCount = m_itemData.count();
376 int itemsInHash = m_items.count();
377
378 int index = m_items.value(urlToFind, -1);
379 while (index < 0 && itemsInHash < itemCount) {
380 // Not all URLs are stored yet in m_items. We grow m_items until either
381 // urlToFind is found, or all URLs have been stored in m_items.
382 // Note that we do not add the URLs to m_items one by one, but in
383 // larger blocks. After each block, we check if urlToFind is in
384 // m_items. We could in principle compare urlToFind with each URL while
385 // we are going through m_itemData, but comparing two QUrls will,
386 // unlike calling qHash for the URLs, trigger a parsing of the URLs
387 // which costs both CPU cycles and memory.
388 const int blockSize = 1000;
389 const int currentBlockEnd = qMin(itemsInHash + blockSize, itemCount);
390 for (int i = itemsInHash; i < currentBlockEnd; ++i) {
391 const QUrl nextUrl = m_itemData.at(i)->item.url();
392 m_items.insert(nextUrl, i);
393 }
394
395 itemsInHash = currentBlockEnd;
396 index = m_items.value(urlToFind, -1);
397 }
398
399 if (index < 0) {
400 // The item could not be found, even though all items from m_itemData
401 // should be in m_items now. We print some diagnostic information which
402 // might help to find the cause of the problem, but only once. This
403 // prevents that obtaining and printing the debugging information
404 // wastes CPU cycles and floods the shell or .xsession-errors.
405 static bool printDebugInfo = true;
406
407 if (m_items.count() != m_itemData.count() && printDebugInfo) {
408 printDebugInfo = false;
409
410 kWarning() << "The model is in an inconsistent state.";
411 kWarning() << "m_items.count() ==" << m_items.count();
412 kWarning() << "m_itemData.count() ==" << m_itemData.count();
413
414 // Check if there are multiple items with the same URL.
415 QMultiHash<QUrl, int> indexesForUrl;
416 for (int i = 0; i < m_itemData.count(); ++i) {
417 indexesForUrl.insert(m_itemData.at(i)->item.url(), i);
418 }
419
420 foreach (const QUrl& url, indexesForUrl.uniqueKeys()) {
421 if (indexesForUrl.count(url) > 1) {
422 kWarning() << "Multiple items found with the URL" << url;
423 foreach (int index, indexesForUrl.values(url)) {
424 const ItemData* data = m_itemData.at(index);
425 kWarning() << "index" << index << ":" << data->item;
426 if (data->parent) {
427 kWarning() << "parent" << data->parent->item;
428 }
429 }
430 }
431 }
432 }
433 }
434
435 return index;
436 }
437
438 KFileItem KFileItemModel::rootItem() const
439 {
440 return m_dirLister->rootItem();
441 }
442
443 void KFileItemModel::clear()
444 {
445 slotClear();
446 }
447
448 void KFileItemModel::setRoles(const QSet<QByteArray>& roles)
449 {
450 if (m_roles == roles) {
451 return;
452 }
453
454 const QSet<QByteArray> changedRoles = (roles - m_roles) + (m_roles - roles);
455 m_roles = roles;
456
457 if (count() > 0) {
458 const bool supportedExpanding = m_requestRole[ExpandedParentsCountRole];
459 const bool willSupportExpanding = roles.contains("expandedParentsCount");
460 if (supportedExpanding && !willSupportExpanding) {
461 // No expanding is supported anymore. Take care to delete all items that have an expansion level
462 // that is not 0 (and hence are part of an expanded item).
463 removeExpandedItems();
464 }
465 }
466
467 m_groups.clear();
468 resetRoles();
469
470 QSetIterator<QByteArray> it(roles);
471 while (it.hasNext()) {
472 const QByteArray& role = it.next();
473 m_requestRole[typeForRole(role)] = true;
474 }
475
476 if (count() > 0) {
477 // Update m_data with the changed requested roles
478 const int maxIndex = count() - 1;
479 for (int i = 0; i <= maxIndex; ++i) {
480 m_itemData[i]->values = retrieveData(m_itemData.at(i)->item, m_itemData.at(i)->parent);
481 }
482
483 emit itemsChanged(KItemRangeList() << KItemRange(0, count()), changedRoles);
484 }
485
486 // Clear the 'values' of all filtered items. They will be re-populated with the
487 // correct roles the next time 'values' will be accessed via data(int).
488 QHash<KFileItem, ItemData*>::iterator filteredIt = m_filteredItems.begin();
489 const QHash<KFileItem, ItemData*>::iterator filteredEnd = m_filteredItems.end();
490 while (filteredIt != filteredEnd) {
491 (*filteredIt)->values.clear();
492 ++filteredIt;
493 }
494 }
495
496 QSet<QByteArray> KFileItemModel::roles() const
497 {
498 return m_roles;
499 }
500
501 bool KFileItemModel::setExpanded(int index, bool expanded)
502 {
503 if (!isExpandable(index) || isExpanded(index) == expanded) {
504 return false;
505 }
506
507 QHash<QByteArray, QVariant> values;
508 values.insert(sharedValue("isExpanded"), expanded);
509 if (!setData(index, values)) {
510 return false;
511 }
512
513 const KFileItem item = m_itemData.at(index)->item;
514 const QUrl url = item.url();
515 const QUrl targetUrl = item.targetUrl();
516 if (expanded) {
517 m_expandedDirs.insert(targetUrl, url);
518 m_dirLister->openUrl(url, KDirLister::Keep);
519
520 const QVariantList previouslyExpandedChildren = m_itemData.at(index)->values.value("previouslyExpandedChildren").value<QVariantList>();
521 foreach (const QVariant& var, previouslyExpandedChildren) {
522 m_urlsToExpand.insert(var.toUrl());
523 }
524 } else {
525 // Note that there might be (indirect) children of the folder which is to be collapsed in
526 // m_pendingItemsToInsert. To prevent that they will be inserted into the model later,
527 // possibly without a parent, which might result in a crash, we insert all pending items
528 // right now. All new items which would be without a parent will then be removed.
529 dispatchPendingItemsToInsert();
530
531 // Check if the index of the collapsed folder has changed. If that is the case, then items
532 // were inserted before the collapsed folder, and its index needs to be updated.
533 if (m_itemData.at(index)->item != item) {
534 index = this->index(item);
535 }
536
537 m_expandedDirs.remove(targetUrl);
538 m_dirLister->stop(url);
539
540 const int parentLevel = expandedParentsCount(index);
541 const int itemCount = m_itemData.count();
542 const int firstChildIndex = index + 1;
543
544 QVariantList expandedChildren;
545
546 int childIndex = firstChildIndex;
547 while (childIndex < itemCount && expandedParentsCount(childIndex) > parentLevel) {
548 ItemData* itemData = m_itemData.at(childIndex);
549 if (itemData->values.value("isExpanded").toBool()) {
550 const QUrl targetUrl = itemData->item.targetUrl();
551 const QUrl url = itemData->item.url();
552 m_expandedDirs.remove(targetUrl);
553 m_dirLister->stop(url); // TODO: try to unit-test this, see https://bugs.kde.org/show_bug.cgi?id=332102#c11
554 expandedChildren.append(targetUrl);
555 }
556 ++childIndex;
557 }
558 const int childrenCount = childIndex - firstChildIndex;
559
560 removeFilteredChildren(KItemRangeList() << KItemRange(index, 1 + childrenCount));
561 removeItems(KItemRangeList() << KItemRange(firstChildIndex, childrenCount), DeleteItemData);
562
563 m_itemData.at(index)->values.insert("previouslyExpandedChildren", expandedChildren);
564 }
565
566 return true;
567 }
568
569 bool KFileItemModel::isExpanded(int index) const
570 {
571 if (index >= 0 && index < count()) {
572 return m_itemData.at(index)->values.value("isExpanded").toBool();
573 }
574 return false;
575 }
576
577 bool KFileItemModel::isExpandable(int index) const
578 {
579 if (index >= 0 && index < count()) {
580 // Call data (instead of accessing m_itemData directly)
581 // to ensure that the value is initialized.
582 return data(index).value("isExpandable").toBool();
583 }
584 return false;
585 }
586
587 int KFileItemModel::expandedParentsCount(int index) const
588 {
589 if (index >= 0 && index < count()) {
590 return expandedParentsCount(m_itemData.at(index));
591 }
592 return 0;
593 }
594
595 QSet<QUrl> KFileItemModel::expandedDirectories() const
596 {
597 return m_expandedDirs.values().toSet();
598 }
599
600 void KFileItemModel::restoreExpandedDirectories(const QSet<QUrl> &urls)
601 {
602 m_urlsToExpand = urls;
603 }
604
605 void KFileItemModel::expandParentDirectories(const QUrl &url)
606 {
607 const int pos = m_dirLister->url().path().length();
608
609 // Assure that each sub-path of the URL that should be
610 // expanded is added to m_urlsToExpand. KDirLister
611 // does not care whether the parent-URL has already been
612 // expanded.
613 QUrl urlToExpand = m_dirLister->url();
614 const QStringList subDirs = url.path().mid(pos).split(QDir::separator());
615 for (int i = 0; i < subDirs.count() - 1; ++i) {
616 urlToExpand.setPath(urlToExpand.path() + '/' + subDirs.at(i));
617 m_urlsToExpand.insert(urlToExpand);
618 }
619
620 // KDirLister::open() must called at least once to trigger an initial
621 // loading. The pending URLs that must be restored are handled
622 // in slotCompleted().
623 QSetIterator<QUrl> it2(m_urlsToExpand);
624 while (it2.hasNext()) {
625 const int idx = index(it2.next());
626 if (idx >= 0 && !isExpanded(idx)) {
627 setExpanded(idx, true);
628 break;
629 }
630 }
631 }
632
633 void KFileItemModel::setNameFilter(const QString& nameFilter)
634 {
635 if (m_filter.pattern() != nameFilter) {
636 dispatchPendingItemsToInsert();
637 m_filter.setPattern(nameFilter);
638 applyFilters();
639 }
640 }
641
642 QString KFileItemModel::nameFilter() const
643 {
644 return m_filter.pattern();
645 }
646
647 void KFileItemModel::setMimeTypeFilters(const QStringList& filters)
648 {
649 if (m_filter.mimeTypes() != filters) {
650 dispatchPendingItemsToInsert();
651 m_filter.setMimeTypes(filters);
652 applyFilters();
653 }
654 }
655
656 QStringList KFileItemModel::mimeTypeFilters() const
657 {
658 return m_filter.mimeTypes();
659 }
660
661
662 void KFileItemModel::applyFilters()
663 {
664 // Check which shown items from m_itemData must get
665 // hidden and hence moved to m_filteredItems.
666 QVector<int> newFilteredIndexes;
667
668 const int itemCount = m_itemData.count();
669 for (int index = 0; index < itemCount; ++index) {
670 ItemData* itemData = m_itemData.at(index);
671
672 // Only filter non-expanded items as child items may never
673 // exist without a parent item
674 if (!itemData->values.value("isExpanded").toBool()) {
675 const KFileItem item = itemData->item;
676 if (!m_filter.matches(item)) {
677 newFilteredIndexes.append(index);
678 m_filteredItems.insert(item, itemData);
679 }
680 }
681 }
682
683 const KItemRangeList removedRanges = KItemRangeList::fromSortedContainer(newFilteredIndexes);
684 removeItems(removedRanges, KeepItemData);
685
686 // Check which hidden items from m_filteredItems should
687 // get visible again and hence removed from m_filteredItems.
688 QList<ItemData*> newVisibleItems;
689
690 QHash<KFileItem, ItemData*>::iterator it = m_filteredItems.begin();
691 while (it != m_filteredItems.end()) {
692 if (m_filter.matches(it.key())) {
693 newVisibleItems.append(it.value());
694 it = m_filteredItems.erase(it);
695 } else {
696 ++it;
697 }
698 }
699
700 insertItems(newVisibleItems);
701 }
702
703 void KFileItemModel::removeFilteredChildren(const KItemRangeList& itemRanges)
704 {
705 if (m_filteredItems.isEmpty() || !m_requestRole[ExpandedParentsCountRole]) {
706 // There are either no filtered items, or it is not possible to expand
707 // folders -> there cannot be any filtered children.
708 return;
709 }
710
711 QSet<ItemData*> parents;
712 foreach (const KItemRange& range, itemRanges) {
713 for (int index = range.index; index < range.index + range.count; ++index) {
714 parents.insert(m_itemData.at(index));
715 }
716 }
717
718 QHash<KFileItem, ItemData*>::iterator it = m_filteredItems.begin();
719 while (it != m_filteredItems.end()) {
720 if (parents.contains(it.value()->parent)) {
721 delete it.value();
722 it = m_filteredItems.erase(it);
723 } else {
724 ++it;
725 }
726 }
727 }
728
729 QList<KFileItemModel::RoleInfo> KFileItemModel::rolesInformation()
730 {
731 static QList<RoleInfo> rolesInfo;
732 if (rolesInfo.isEmpty()) {
733 int count = 0;
734 const RoleInfoMap* map = rolesInfoMap(count);
735 for (int i = 0; i < count; ++i) {
736 if (map[i].roleType != NoRole) {
737 RoleInfo info;
738 info.role = map[i].role;
739 info.translation = i18nc(map[i].roleTranslationContext, map[i].roleTranslation);
740 if (map[i].groupTranslation) {
741 info.group = i18nc(map[i].groupTranslationContext, map[i].groupTranslation);
742 } else {
743 // For top level roles, groupTranslation is 0. We must make sure that
744 // info.group is an empty string then because the code that generates
745 // menus tries to put the actions into sub menus otherwise.
746 info.group = QString();
747 }
748 info.requiresBaloo = map[i].requiresBaloo;
749 info.requiresIndexer = map[i].requiresIndexer;
750 rolesInfo.append(info);
751 }
752 }
753 }
754
755 return rolesInfo;
756 }
757
758 void KFileItemModel::onGroupedSortingChanged(bool current)
759 {
760 Q_UNUSED(current);
761 m_groups.clear();
762 }
763
764 void KFileItemModel::onSortRoleChanged(const QByteArray& current, const QByteArray& previous)
765 {
766 Q_UNUSED(previous);
767 m_sortRole = typeForRole(current);
768
769 if (!m_requestRole[m_sortRole]) {
770 QSet<QByteArray> newRoles = m_roles;
771 newRoles << current;
772 setRoles(newRoles);
773 }
774
775 resortAllItems();
776 }
777
778 void KFileItemModel::onSortOrderChanged(Qt::SortOrder current, Qt::SortOrder previous)
779 {
780 Q_UNUSED(current);
781 Q_UNUSED(previous);
782 resortAllItems();
783 }
784
785 void KFileItemModel::resortAllItems()
786 {
787 m_resortAllItemsTimer->stop();
788
789 const int itemCount = count();
790 if (itemCount <= 0) {
791 return;
792 }
793
794 #ifdef KFILEITEMMODEL_DEBUG
795 QElapsedTimer timer;
796 timer.start();
797 kDebug() << "===========================================================";
798 kDebug() << "Resorting" << itemCount << "items";
799 #endif
800
801 // Remember the order of the current URLs so
802 // that it can be determined which indexes have
803 // been moved because of the resorting.
804 QList<QUrl> oldUrls;
805 oldUrls.reserve(itemCount);
806 foreach (const ItemData* itemData, m_itemData) {
807 oldUrls.append(itemData->item.url());
808 }
809
810 m_items.clear();
811 m_items.reserve(itemCount);
812
813 // Resort the items
814 sort(m_itemData.begin(), m_itemData.end());
815 for (int i = 0; i < itemCount; ++i) {
816 m_items.insert(m_itemData.at(i)->item.url(), i);
817 }
818
819 // Determine the first index that has been moved.
820 int firstMovedIndex = 0;
821 while (firstMovedIndex < itemCount
822 && firstMovedIndex == m_items.value(oldUrls.at(firstMovedIndex))) {
823 ++firstMovedIndex;
824 }
825
826 const bool itemsHaveMoved = firstMovedIndex < itemCount;
827 if (itemsHaveMoved) {
828 m_groups.clear();
829
830 int lastMovedIndex = itemCount - 1;
831 while (lastMovedIndex > firstMovedIndex
832 && lastMovedIndex == m_items.value(oldUrls.at(lastMovedIndex))) {
833 --lastMovedIndex;
834 }
835
836 Q_ASSERT(firstMovedIndex <= lastMovedIndex);
837
838 // Create a list movedToIndexes, which has the property that
839 // movedToIndexes[i] is the new index of the item with the old index
840 // firstMovedIndex + i.
841 const int movedItemsCount = lastMovedIndex - firstMovedIndex + 1;
842 QList<int> movedToIndexes;
843 movedToIndexes.reserve(movedItemsCount);
844 for (int i = firstMovedIndex; i <= lastMovedIndex; ++i) {
845 const int newIndex = m_items.value(oldUrls.at(i));
846 movedToIndexes.append(newIndex);
847 }
848
849 emit itemsMoved(KItemRange(firstMovedIndex, movedItemsCount), movedToIndexes);
850 } else if (groupedSorting()) {
851 // The groups might have changed even if the order of the items has not.
852 const QList<QPair<int, QVariant> > oldGroups = m_groups;
853 m_groups.clear();
854 if (groups() != oldGroups) {
855 emit groupsChanged();
856 }
857 }
858
859 #ifdef KFILEITEMMODEL_DEBUG
860 kDebug() << "[TIME] Resorting of" << itemCount << "items:" << timer.elapsed();
861 #endif
862 }
863
864 void KFileItemModel::slotCompleted()
865 {
866 dispatchPendingItemsToInsert();
867
868 if (!m_urlsToExpand.isEmpty()) {
869 // Try to find a URL that can be expanded.
870 // Note that the parent folder must be expanded before any of its subfolders become visible.
871 // Therefore, some URLs in m_restoredExpandedUrls might not be visible yet
872 // -> we expand the first visible URL we find in m_restoredExpandedUrls.
873 foreach (const QUrl& url, m_urlsToExpand) {
874 const int indexForUrl = index(url);
875 if (indexForUrl >= 0) {
876 m_urlsToExpand.remove(url);
877 if (setExpanded(indexForUrl, true)) {
878 // The dir lister has been triggered. This slot will be called
879 // again after the directory has been expanded.
880 return;
881 }
882 }
883 }
884
885 // None of the URLs in m_restoredExpandedUrls could be found in the model. This can happen
886 // if these URLs have been deleted in the meantime.
887 m_urlsToExpand.clear();
888 }
889
890 emit directoryLoadingCompleted();
891 }
892
893 void KFileItemModel::slotCanceled()
894 {
895 m_maximumUpdateIntervalTimer->stop();
896 dispatchPendingItemsToInsert();
897
898 emit directoryLoadingCanceled();
899 }
900
901 void KFileItemModel::slotItemsAdded(const QUrl &directoryUrl, const KFileItemList& items)
902 {
903 Q_ASSERT(!items.isEmpty());
904
905 QUrl parentUrl;
906 if (m_expandedDirs.contains(directoryUrl)) {
907 parentUrl = m_expandedDirs.value(directoryUrl);
908 } else {
909 parentUrl = directoryUrl.adjusted(QUrl::StripTrailingSlash);
910 }
911
912 if (m_requestRole[ExpandedParentsCountRole]) {
913 // If the expanding of items is enabled, the call
914 // dirLister->openUrl(url, KDirLister::Keep) in KFileItemModel::setExpanded()
915 // might result in emitting the same items twice due to the Keep-parameter.
916 // This case happens if an item gets expanded, collapsed and expanded again
917 // before the items could be loaded for the first expansion.
918 if (index(items.first().url()) >= 0) {
919 // The items are already part of the model.
920 return;
921 }
922
923 if (directoryUrl != directory()) {
924 // To be able to compare whether the new items may be inserted as children
925 // of a parent item the pending items must be added to the model first.
926 dispatchPendingItemsToInsert();
927 }
928
929 // KDirLister keeps the children of items that got expanded once even if
930 // they got collapsed again with KFileItemModel::setExpanded(false). So it must be
931 // checked whether the parent for new items is still expanded.
932 const int parentIndex = index(parentUrl);
933 if (parentIndex >= 0 && !m_itemData[parentIndex]->values.value("isExpanded").toBool()) {
934 // The parent is not expanded.
935 return;
936 }
937 }
938
939 QList<ItemData*> itemDataList = createItemDataList(parentUrl, items);
940
941 if (!m_filter.hasSetFilters()) {
942 m_pendingItemsToInsert.append(itemDataList);
943 } else {
944 // The name or type filter is active. Hide filtered items
945 // before inserting them into the model and remember
946 // the filtered items in m_filteredItems.
947 foreach (ItemData* itemData, itemDataList) {
948 if (m_filter.matches(itemData->item)) {
949 m_pendingItemsToInsert.append(itemData);
950 } else {
951 m_filteredItems.insert(itemData->item, itemData);
952 }
953 }
954 }
955
956 if (useMaximumUpdateInterval() && !m_maximumUpdateIntervalTimer->isActive()) {
957 // Assure that items get dispatched if no completed() or canceled() signal is
958 // emitted during the maximum update interval.
959 m_maximumUpdateIntervalTimer->start();
960 }
961 }
962
963 void KFileItemModel::slotItemsDeleted(const KFileItemList& items)
964 {
965 dispatchPendingItemsToInsert();
966
967 QVector<int> indexesToRemove;
968 indexesToRemove.reserve(items.count());
969
970 foreach (const KFileItem& item, items) {
971 const int indexForItem = index(item);
972 if (indexForItem >= 0) {
973 indexesToRemove.append(indexForItem);
974 } else {
975 // Probably the item has been filtered.
976 QHash<KFileItem, ItemData*>::iterator it = m_filteredItems.find(item);
977 if (it != m_filteredItems.end()) {
978 delete it.value();
979 m_filteredItems.erase(it);
980 }
981 }
982 }
983
984 std::sort(indexesToRemove.begin(), indexesToRemove.end());
985
986 if (m_requestRole[ExpandedParentsCountRole] && !m_expandedDirs.isEmpty()) {
987 // Assure that removing a parent item also results in removing all children
988 QVector<int> indexesToRemoveWithChildren;
989 indexesToRemoveWithChildren.reserve(m_itemData.count());
990
991 const int itemCount = m_itemData.count();
992 foreach (int index, indexesToRemove) {
993 indexesToRemoveWithChildren.append(index);
994
995 const int parentLevel = expandedParentsCount(index);
996 int childIndex = index + 1;
997 while (childIndex < itemCount && expandedParentsCount(childIndex) > parentLevel) {
998 indexesToRemoveWithChildren.append(childIndex);
999 ++childIndex;
1000 }
1001 }
1002
1003 indexesToRemove = indexesToRemoveWithChildren;
1004 }
1005
1006 const KItemRangeList itemRanges = KItemRangeList::fromSortedContainer(indexesToRemove);
1007 removeFilteredChildren(itemRanges);
1008 removeItems(itemRanges, DeleteItemData);
1009 }
1010
1011 void KFileItemModel::slotRefreshItems(const QList<QPair<KFileItem, KFileItem> >& items)
1012 {
1013 Q_ASSERT(!items.isEmpty());
1014 #ifdef KFILEITEMMODEL_DEBUG
1015 kDebug() << "Refreshing" << items.count() << "items";
1016 #endif
1017
1018 // Get the indexes of all items that have been refreshed
1019 QList<int> indexes;
1020 indexes.reserve(items.count());
1021
1022 QSet<QByteArray> changedRoles;
1023
1024 QListIterator<QPair<KFileItem, KFileItem> > it(items);
1025 while (it.hasNext()) {
1026 const QPair<KFileItem, KFileItem>& itemPair = it.next();
1027 const KFileItem& oldItem = itemPair.first;
1028 const KFileItem& newItem = itemPair.second;
1029 const int indexForItem = index(oldItem);
1030 if (indexForItem >= 0) {
1031 m_itemData[indexForItem]->item = newItem;
1032
1033 // Keep old values as long as possible if they could not retrieved synchronously yet.
1034 // The update of the values will be done asynchronously by KFileItemModelRolesUpdater.
1035 QHashIterator<QByteArray, QVariant> it(retrieveData(newItem, m_itemData.at(indexForItem)->parent));
1036 QHash<QByteArray, QVariant>& values = m_itemData[indexForItem]->values;
1037 while (it.hasNext()) {
1038 it.next();
1039 const QByteArray& role = it.key();
1040 if (values.value(role) != it.value()) {
1041 values.insert(role, it.value());
1042 changedRoles.insert(role);
1043 }
1044 }
1045
1046 m_items.remove(oldItem.url());
1047 m_items.insert(newItem.url(), indexForItem);
1048 indexes.append(indexForItem);
1049 } else {
1050 // Check if 'oldItem' is one of the filtered items.
1051 QHash<KFileItem, ItemData*>::iterator it = m_filteredItems.find(oldItem);
1052 if (it != m_filteredItems.end()) {
1053 ItemData* itemData = it.value();
1054 itemData->item = newItem;
1055
1056 // The data stored in 'values' might have changed. Therefore, we clear
1057 // 'values' and re-populate it the next time it is requested via data(int).
1058 itemData->values.clear();
1059
1060 m_filteredItems.erase(it);
1061 m_filteredItems.insert(newItem, itemData);
1062 }
1063 }
1064 }
1065
1066 // If the changed items have been created recently, they might not be in m_items yet.
1067 // In that case, the list 'indexes' might be empty.
1068 if (indexes.isEmpty()) {
1069 return;
1070 }
1071
1072 // Extract the item-ranges out of the changed indexes
1073 qSort(indexes);
1074 const KItemRangeList itemRangeList = KItemRangeList::fromSortedContainer(indexes);
1075 emitItemsChangedAndTriggerResorting(itemRangeList, changedRoles);
1076 }
1077
1078 void KFileItemModel::slotClear()
1079 {
1080 #ifdef KFILEITEMMODEL_DEBUG
1081 kDebug() << "Clearing all items";
1082 #endif
1083
1084 qDeleteAll(m_filteredItems.values());
1085 m_filteredItems.clear();
1086 m_groups.clear();
1087
1088 m_maximumUpdateIntervalTimer->stop();
1089 m_resortAllItemsTimer->stop();
1090
1091 qDeleteAll(m_pendingItemsToInsert);
1092 m_pendingItemsToInsert.clear();
1093
1094 const int removedCount = m_itemData.count();
1095 if (removedCount > 0) {
1096 qDeleteAll(m_itemData);
1097 m_itemData.clear();
1098 m_items.clear();
1099 emit itemsRemoved(KItemRangeList() << KItemRange(0, removedCount));
1100 }
1101
1102 m_expandedDirs.clear();
1103 }
1104
1105 void KFileItemModel::slotNaturalSortingChanged()
1106 {
1107 m_naturalSorting = KGlobalSettings::naturalSorting();
1108 resortAllItems();
1109 }
1110
1111 void KFileItemModel::dispatchPendingItemsToInsert()
1112 {
1113 if (!m_pendingItemsToInsert.isEmpty()) {
1114 insertItems(m_pendingItemsToInsert);
1115 m_pendingItemsToInsert.clear();
1116 }
1117 }
1118
1119 void KFileItemModel::insertItems(QList<ItemData*>& newItems)
1120 {
1121 if (newItems.isEmpty()) {
1122 return;
1123 }
1124
1125 #ifdef KFILEITEMMODEL_DEBUG
1126 QElapsedTimer timer;
1127 timer.start();
1128 kDebug() << "===========================================================";
1129 kDebug() << "Inserting" << newItems.count() << "items";
1130 #endif
1131
1132 m_groups.clear();
1133 prepareItemsForSorting(newItems);
1134
1135 if (m_sortRole == NameRole && m_naturalSorting) {
1136 // Natural sorting of items can be very slow. However, it becomes much
1137 // faster if the input sequence is already mostly sorted. Therefore, we
1138 // first sort 'newItems' according to the QStrings returned by
1139 // KFileItem::text() using QString::operator<(), which is quite fast.
1140 parallelMergeSort(newItems.begin(), newItems.end(), nameLessThan, QThread::idealThreadCount());
1141 }
1142
1143 sort(newItems.begin(), newItems.end());
1144
1145 #ifdef KFILEITEMMODEL_DEBUG
1146 kDebug() << "[TIME] Sorting:" << timer.elapsed();
1147 #endif
1148
1149 KItemRangeList itemRanges;
1150 const int existingItemCount = m_itemData.count();
1151 const int newItemCount = newItems.count();
1152 const int totalItemCount = existingItemCount + newItemCount;
1153
1154 if (existingItemCount == 0) {
1155 // Optimization for the common special case that there are no
1156 // items in the model yet. Happens, e.g., when entering a folder.
1157 m_itemData = newItems;
1158 itemRanges << KItemRange(0, newItemCount);
1159 } else {
1160 m_itemData.reserve(totalItemCount);
1161 for (int i = existingItemCount; i < totalItemCount; ++i) {
1162 m_itemData.append(0);
1163 }
1164
1165 // We build the new list m_itemData in reverse order to minimize
1166 // the number of moves and guarantee O(N) complexity.
1167 int targetIndex = totalItemCount - 1;
1168 int sourceIndexExistingItems = existingItemCount - 1;
1169 int sourceIndexNewItems = newItemCount - 1;
1170
1171 int rangeCount = 0;
1172
1173 while (sourceIndexNewItems >= 0) {
1174 ItemData* newItem = newItems.at(sourceIndexNewItems);
1175 if (sourceIndexExistingItems >= 0 && lessThan(newItem, m_itemData.at(sourceIndexExistingItems))) {
1176 // Move an existing item to its new position. If any new items
1177 // are behind it, push the item range to itemRanges.
1178 if (rangeCount > 0) {
1179 itemRanges << KItemRange(sourceIndexExistingItems + 1, rangeCount);
1180 rangeCount = 0;
1181 }
1182
1183 m_itemData[targetIndex] = m_itemData.at(sourceIndexExistingItems);
1184 --sourceIndexExistingItems;
1185 } else {
1186 // Insert a new item into the list.
1187 ++rangeCount;
1188 m_itemData[targetIndex] = newItem;
1189 --sourceIndexNewItems;
1190 }
1191 --targetIndex;
1192 }
1193
1194 // Push the final item range to itemRanges.
1195 if (rangeCount > 0) {
1196 itemRanges << KItemRange(sourceIndexExistingItems + 1, rangeCount);
1197 }
1198
1199 // Note that itemRanges is still sorted in reverse order.
1200 std::reverse(itemRanges.begin(), itemRanges.end());
1201 }
1202
1203 // The indexes in m_items are not correct anymore. Therefore, we clear m_items.
1204 // It will be re-populated with the updated indices if index(const QUrl&) is called.
1205 m_items.clear();
1206
1207 emit itemsInserted(itemRanges);
1208
1209 #ifdef KFILEITEMMODEL_DEBUG
1210 kDebug() << "[TIME] Inserting of" << newItems.count() << "items:" << timer.elapsed();
1211 #endif
1212 }
1213
1214 void KFileItemModel::removeItems(const KItemRangeList& itemRanges, RemoveItemsBehavior behavior)
1215 {
1216 if (itemRanges.isEmpty()) {
1217 return;
1218 }
1219
1220 m_groups.clear();
1221
1222 // Step 1: Remove the items from m_itemData, and free the ItemData.
1223 int removedItemsCount = 0;
1224 foreach (const KItemRange& range, itemRanges) {
1225 removedItemsCount += range.count;
1226
1227 for (int index = range.index; index < range.index + range.count; ++index) {
1228 if (behavior == DeleteItemData) {
1229 delete m_itemData.at(index);
1230 }
1231
1232 m_itemData[index] = 0;
1233 }
1234 }
1235
1236 // Step 2: Remove the ItemData pointers from the list m_itemData.
1237 int target = itemRanges.at(0).index;
1238 int source = itemRanges.at(0).index + itemRanges.at(0).count;
1239 int nextRange = 1;
1240
1241 const int oldItemDataCount = m_itemData.count();
1242 while (source < oldItemDataCount) {
1243 m_itemData[target] = m_itemData[source];
1244 ++target;
1245 ++source;
1246
1247 if (nextRange < itemRanges.count() && source == itemRanges.at(nextRange).index) {
1248 // Skip the items in the next removed range.
1249 source += itemRanges.at(nextRange).count;
1250 ++nextRange;
1251 }
1252 }
1253
1254 m_itemData.erase(m_itemData.end() - removedItemsCount, m_itemData.end());
1255
1256 // The indexes in m_items are not correct anymore. Therefore, we clear m_items.
1257 // It will be re-populated with the updated indices if index(const QUrl&) is called.
1258 m_items.clear();
1259
1260 emit itemsRemoved(itemRanges);
1261 }
1262
1263 QList<KFileItemModel::ItemData*> KFileItemModel::createItemDataList(const QUrl& parentUrl, const KFileItemList& items) const
1264 {
1265 if (m_sortRole == TypeRole) {
1266 // Try to resolve the MIME-types synchronously to prevent a reordering of
1267 // the items when sorting by type (per default MIME-types are resolved
1268 // asynchronously by KFileItemModelRolesUpdater).
1269 determineMimeTypes(items, 200);
1270 }
1271
1272 const int parentIndex = index(parentUrl);
1273 ItemData* parentItem = parentIndex < 0 ? 0 : m_itemData.at(parentIndex);
1274
1275 QList<ItemData*> itemDataList;
1276 itemDataList.reserve(items.count());
1277
1278 foreach (const KFileItem& item, items) {
1279 ItemData* itemData = new ItemData();
1280 itemData->item = item;
1281 itemData->parent = parentItem;
1282 itemDataList.append(itemData);
1283 }
1284
1285 return itemDataList;
1286 }
1287
1288 void KFileItemModel::prepareItemsForSorting(QList<ItemData*>& itemDataList)
1289 {
1290 switch (m_sortRole) {
1291 case PermissionsRole:
1292 case OwnerRole:
1293 case GroupRole:
1294 case DestinationRole:
1295 case PathRole:
1296 // These roles can be determined with retrieveData, and they have to be stored
1297 // in the QHash "values" for the sorting.
1298 foreach (ItemData* itemData, itemDataList) {
1299 if (itemData->values.isEmpty()) {
1300 itemData->values = retrieveData(itemData->item, itemData->parent);
1301 }
1302 }
1303 break;
1304
1305 case TypeRole:
1306 // At least store the data including the file type for items with known MIME type.
1307 foreach (ItemData* itemData, itemDataList) {
1308 if (itemData->values.isEmpty()) {
1309 const KFileItem item = itemData->item;
1310 if (item.isDir() || item.isMimeTypeKnown()) {
1311 itemData->values = retrieveData(itemData->item, itemData->parent);
1312 }
1313 }
1314 }
1315 break;
1316
1317 default:
1318 // The other roles are either resolved by KFileItemModelRolesUpdater
1319 // (this includes the SizeRole for directories), or they do not need
1320 // to be stored in the QHash "values" for sorting because the data can
1321 // be retrieved directly from the KFileItem (NameRole, SizeRole for files,
1322 // DateRole).
1323 break;
1324 }
1325 }
1326
1327 int KFileItemModel::expandedParentsCount(const ItemData* data)
1328 {
1329 // The hash 'values' is only guaranteed to contain the key "expandedParentsCount"
1330 // if the corresponding item is expanded, and it is not a top-level item.
1331 const ItemData* parent = data->parent;
1332 if (parent) {
1333 if (parent->parent) {
1334 Q_ASSERT(parent->values.contains("expandedParentsCount"));
1335 return parent->values.value("expandedParentsCount").toInt() + 1;
1336 } else {
1337 return 1;
1338 }
1339 } else {
1340 return 0;
1341 }
1342 }
1343
1344 void KFileItemModel::removeExpandedItems()
1345 {
1346 QVector<int> indexesToRemove;
1347
1348 const int maxIndex = m_itemData.count() - 1;
1349 for (int i = 0; i <= maxIndex; ++i) {
1350 const ItemData* itemData = m_itemData.at(i);
1351 if (itemData->parent) {
1352 indexesToRemove.append(i);
1353 }
1354 }
1355
1356 removeItems(KItemRangeList::fromSortedContainer(indexesToRemove), DeleteItemData);
1357 m_expandedDirs.clear();
1358
1359 // Also remove all filtered items which have a parent.
1360 QHash<KFileItem, ItemData*>::iterator it = m_filteredItems.begin();
1361 const QHash<KFileItem, ItemData*>::iterator end = m_filteredItems.end();
1362
1363 while (it != end) {
1364 if (it.value()->parent) {
1365 delete it.value();
1366 it = m_filteredItems.erase(it);
1367 } else {
1368 ++it;
1369 }
1370 }
1371 }
1372
1373 void KFileItemModel::emitItemsChangedAndTriggerResorting(const KItemRangeList& itemRanges, const QSet<QByteArray>& changedRoles)
1374 {
1375 emit itemsChanged(itemRanges, changedRoles);
1376
1377 // Trigger a resorting if necessary. Note that this can happen even if the sort
1378 // role has not changed at all because the file name can be used as a fallback.
1379 if (changedRoles.contains(sortRole()) || changedRoles.contains(roleForType(NameRole))) {
1380 foreach (const KItemRange& range, itemRanges) {
1381 bool needsResorting = false;
1382
1383 const int first = range.index;
1384 const int last = range.index + range.count - 1;
1385
1386 // Resorting the model is necessary if
1387 // (a) The first item in the range is "lessThan" its predecessor,
1388 // (b) the successor of the last item is "lessThan" the last item, or
1389 // (c) the internal order of the items in the range is incorrect.
1390 if (first > 0
1391 && lessThan(m_itemData.at(first), m_itemData.at(first - 1))) {
1392 needsResorting = true;
1393 } else if (last < count() - 1
1394 && lessThan(m_itemData.at(last + 1), m_itemData.at(last))) {
1395 needsResorting = true;
1396 } else {
1397 for (int index = first; index < last; ++index) {
1398 if (lessThan(m_itemData.at(index + 1), m_itemData.at(index))) {
1399 needsResorting = true;
1400 break;
1401 }
1402 }
1403 }
1404
1405 if (needsResorting) {
1406 m_resortAllItemsTimer->start();
1407 return;
1408 }
1409 }
1410 }
1411
1412 if (groupedSorting() && changedRoles.contains(sortRole())) {
1413 // The position is still correct, but the groups might have changed
1414 // if the changed item is either the first or the last item in a
1415 // group.
1416 // In principle, we could try to find out if the item really is the
1417 // first or last one in its group and then update the groups
1418 // (possibly with a delayed timer to make sure that we don't
1419 // re-calculate the groups very often if items are updated one by
1420 // one), but starting m_resortAllItemsTimer is easier.
1421 m_resortAllItemsTimer->start();
1422 }
1423 }
1424
1425 void KFileItemModel::resetRoles()
1426 {
1427 for (int i = 0; i < RolesCount; ++i) {
1428 m_requestRole[i] = false;
1429 }
1430 }
1431
1432 KFileItemModel::RoleType KFileItemModel::typeForRole(const QByteArray& role) const
1433 {
1434 static QHash<QByteArray, RoleType> roles;
1435 if (roles.isEmpty()) {
1436 // Insert user visible roles that can be accessed with
1437 // KFileItemModel::roleInformation()
1438 int count = 0;
1439 const RoleInfoMap* map = rolesInfoMap(count);
1440 for (int i = 0; i < count; ++i) {
1441 roles.insert(map[i].role, map[i].roleType);
1442 }
1443
1444 // Insert internal roles (take care to synchronize the implementation
1445 // with KFileItemModel::roleForType() in case if a change is done).
1446 roles.insert("isDir", IsDirRole);
1447 roles.insert("isLink", IsLinkRole);
1448 roles.insert("isExpanded", IsExpandedRole);
1449 roles.insert("isExpandable", IsExpandableRole);
1450 roles.insert("expandedParentsCount", ExpandedParentsCountRole);
1451
1452 Q_ASSERT(roles.count() == RolesCount);
1453 }
1454
1455 return roles.value(role, NoRole);
1456 }
1457
1458 QByteArray KFileItemModel::roleForType(RoleType roleType) const
1459 {
1460 static QHash<RoleType, QByteArray> roles;
1461 if (roles.isEmpty()) {
1462 // Insert user visible roles that can be accessed with
1463 // KFileItemModel::roleInformation()
1464 int count = 0;
1465 const RoleInfoMap* map = rolesInfoMap(count);
1466 for (int i = 0; i < count; ++i) {
1467 roles.insert(map[i].roleType, map[i].role);
1468 }
1469
1470 // Insert internal roles (take care to synchronize the implementation
1471 // with KFileItemModel::typeForRole() in case if a change is done).
1472 roles.insert(IsDirRole, "isDir");
1473 roles.insert(IsLinkRole, "isLink");
1474 roles.insert(IsExpandedRole, "isExpanded");
1475 roles.insert(IsExpandableRole, "isExpandable");
1476 roles.insert(ExpandedParentsCountRole, "expandedParentsCount");
1477
1478 Q_ASSERT(roles.count() == RolesCount);
1479 };
1480
1481 return roles.value(roleType);
1482 }
1483
1484 QHash<QByteArray, QVariant> KFileItemModel::retrieveData(const KFileItem& item, const ItemData* parent) const
1485 {
1486 // It is important to insert only roles that are fast to retrieve. E.g.
1487 // KFileItem::iconName() can be very expensive if the MIME-type is unknown
1488 // and hence will be retrieved asynchronously by KFileItemModelRolesUpdater.
1489 QHash<QByteArray, QVariant> data;
1490 data.insert(sharedValue("url"), item.url());
1491
1492 const bool isDir = item.isDir();
1493 if (m_requestRole[IsDirRole] && isDir) {
1494 data.insert(sharedValue("isDir"), true);
1495 }
1496
1497 if (m_requestRole[IsLinkRole] && item.isLink()) {
1498 data.insert(sharedValue("isLink"), true);
1499 }
1500
1501 if (m_requestRole[NameRole]) {
1502 data.insert(sharedValue("text"), item.text());
1503 }
1504
1505 if (m_requestRole[SizeRole] && !isDir) {
1506 data.insert(sharedValue("size"), item.size());
1507 }
1508
1509 if (m_requestRole[DateRole]) {
1510 // Don't use KFileItem::timeString() as this is too expensive when
1511 // having several thousands of items. Instead the formatting of the
1512 // date-time will be done on-demand by the view when the date will be shown.
1513 const QDateTime dateTime = item.time(KFileItem::ModificationTime);
1514 data.insert(sharedValue("date"), dateTime);
1515 }
1516
1517 if (m_requestRole[PermissionsRole]) {
1518 data.insert(sharedValue("permissions"), item.permissionsString());
1519 }
1520
1521 if (m_requestRole[OwnerRole]) {
1522 data.insert(sharedValue("owner"), item.user());
1523 }
1524
1525 if (m_requestRole[GroupRole]) {
1526 data.insert(sharedValue("group"), item.group());
1527 }
1528
1529 if (m_requestRole[DestinationRole]) {
1530 QString destination = item.linkDest();
1531 if (destination.isEmpty()) {
1532 destination = QLatin1String("-");
1533 }
1534 data.insert(sharedValue("destination"), destination);
1535 }
1536
1537 if (m_requestRole[PathRole]) {
1538 QString path;
1539 if (item.url().scheme() == QLatin1String("trash")) {
1540 path = item.entry().stringValue(KIO::UDSEntry::UDS_EXTRA);
1541 } else {
1542 // For performance reasons cache the home-path in a static QString
1543 // (see QDir::homePath() for more details)
1544 static QString homePath;
1545 if (homePath.isEmpty()) {
1546 homePath = QDir::homePath();
1547 }
1548
1549 path = item.localPath();
1550 if (path.startsWith(homePath)) {
1551 path.replace(0, homePath.length(), QLatin1Char('~'));
1552 }
1553 }
1554
1555 const int index = path.lastIndexOf(item.text());
1556 path = path.mid(0, index - 1);
1557 data.insert(sharedValue("path"), path);
1558 }
1559
1560 if (m_requestRole[IsExpandableRole] && isDir) {
1561 data.insert(sharedValue("isExpandable"), true);
1562 }
1563
1564 if (m_requestRole[ExpandedParentsCountRole]) {
1565 if (parent) {
1566 const int level = expandedParentsCount(parent) + 1;
1567 data.insert(sharedValue("expandedParentsCount"), level);
1568 }
1569 }
1570
1571 if (item.isMimeTypeKnown()) {
1572 data.insert(sharedValue("iconName"), item.iconName());
1573
1574 if (m_requestRole[TypeRole]) {
1575 data.insert(sharedValue("type"), item.mimeComment());
1576 }
1577 } else if (m_requestRole[TypeRole] && isDir) {
1578 static const QString folderMimeType = item.mimeComment();
1579 data.insert(sharedValue("type"), folderMimeType);
1580 }
1581
1582 return data;
1583 }
1584
1585 bool KFileItemModel::lessThan(const ItemData* a, const ItemData* b) const
1586 {
1587 int result = 0;
1588
1589 if (a->parent != b->parent) {
1590 const int expansionLevelA = expandedParentsCount(a);
1591 const int expansionLevelB = expandedParentsCount(b);
1592
1593 // If b has a higher expansion level than a, check if a is a parent
1594 // of b, and make sure that both expansion levels are equal otherwise.
1595 for (int i = expansionLevelB; i > expansionLevelA; --i) {
1596 if (b->parent == a) {
1597 return true;
1598 }
1599 b = b->parent;
1600 }
1601
1602 // If a has a higher expansion level than a, check if b is a parent
1603 // of a, and make sure that both expansion levels are equal otherwise.
1604 for (int i = expansionLevelA; i > expansionLevelB; --i) {
1605 if (a->parent == b) {
1606 return false;
1607 }
1608 a = a->parent;
1609 }
1610
1611 Q_ASSERT(expandedParentsCount(a) == expandedParentsCount(b));
1612
1613 // Compare the last parents of a and b which are different.
1614 while (a->parent != b->parent) {
1615 a = a->parent;
1616 b = b->parent;
1617 }
1618 }
1619
1620 if (m_sortDirsFirst || m_sortRole == SizeRole) {
1621 const bool isDirA = a->item.isDir();
1622 const bool isDirB = b->item.isDir();
1623 if (isDirA && !isDirB) {
1624 return true;
1625 } else if (!isDirA && isDirB) {
1626 return false;
1627 }
1628 }
1629
1630 result = sortRoleCompare(a, b);
1631
1632 return (sortOrder() == Qt::AscendingOrder) ? result < 0 : result > 0;
1633 }
1634
1635 /**
1636 * Helper class for KFileItemModel::sort().
1637 */
1638 class KFileItemModelLessThan
1639 {
1640 public:
1641 KFileItemModelLessThan(const KFileItemModel* model) :
1642 m_model(model)
1643 {
1644 }
1645
1646 bool operator()(const KFileItemModel::ItemData* a, const KFileItemModel::ItemData* b) const
1647 {
1648 return m_model->lessThan(a, b);
1649 }
1650
1651 private:
1652 const KFileItemModel* m_model;
1653 };
1654
1655 void KFileItemModel::sort(QList<KFileItemModel::ItemData*>::iterator begin,
1656 QList<KFileItemModel::ItemData*>::iterator end) const
1657 {
1658 KFileItemModelLessThan lessThan(this);
1659
1660 if (m_sortRole == NameRole) {
1661 // Sorting by name can be expensive, in particular if natural sorting is
1662 // enabled. Use all CPU cores to speed up the sorting process.
1663 static const int numberOfThreads = QThread::idealThreadCount();
1664 parallelMergeSort(begin, end, lessThan, numberOfThreads);
1665 } else {
1666 // Sorting by other roles is quite fast. Use only one thread to prevent
1667 // problems caused by non-reentrant comparison functions, see
1668 // https://bugs.kde.org/show_bug.cgi?id=312679
1669 mergeSort(begin, end, lessThan);
1670 }
1671 }
1672
1673 int KFileItemModel::sortRoleCompare(const ItemData* a, const ItemData* b) const
1674 {
1675 const KFileItem& itemA = a->item;
1676 const KFileItem& itemB = b->item;
1677
1678 int result = 0;
1679
1680 switch (m_sortRole) {
1681 case NameRole:
1682 // The name role is handled as default fallback after the switch
1683 break;
1684
1685 case SizeRole: {
1686 if (itemA.isDir()) {
1687 // See "if (m_sortFoldersFirst || m_sortRole == SizeRole)" in KFileItemModel::lessThan():
1688 Q_ASSERT(itemB.isDir());
1689
1690 const QVariant valueA = a->values.value("size");
1691 const QVariant valueB = b->values.value("size");
1692 if (valueA.isNull() && valueB.isNull()) {
1693 result = 0;
1694 } else if (valueA.isNull()) {
1695 result = -1;
1696 } else if (valueB.isNull()) {
1697 result = +1;
1698 } else {
1699 result = valueA.toInt() - valueB.toInt();
1700 }
1701 } else {
1702 // See "if (m_sortFoldersFirst || m_sortRole == SizeRole)" in KFileItemModel::lessThan():
1703 Q_ASSERT(!itemB.isDir());
1704 const KIO::filesize_t sizeA = itemA.size();
1705 const KIO::filesize_t sizeB = itemB.size();
1706 if (sizeA > sizeB) {
1707 result = +1;
1708 } else if (sizeA < sizeB) {
1709 result = -1;
1710 } else {
1711 result = 0;
1712 }
1713 }
1714 break;
1715 }
1716
1717 case DateRole: {
1718 const QDateTime dateTimeA = itemA.time(KFileItem::ModificationTime);
1719 const QDateTime dateTimeB = itemB.time(KFileItem::ModificationTime);
1720 if (dateTimeA < dateTimeB) {
1721 result = -1;
1722 } else if (dateTimeA > dateTimeB) {
1723 result = +1;
1724 }
1725 break;
1726 }
1727
1728 case RatingRole: {
1729 result = a->values.value("rating").toInt() - b->values.value("rating").toInt();
1730 break;
1731 }
1732
1733 case ImageSizeRole: {
1734 // Alway use a natural comparing to interpret the numbers of a string like
1735 // "1600 x 1200" for having a correct sorting.
1736 result = KStringHandler::naturalCompare(a->values.value("imageSize").toString(),
1737 b->values.value("imageSize").toString(),
1738 Qt::CaseSensitive);
1739 break;
1740 }
1741
1742 default: {
1743 const QByteArray role = roleForType(m_sortRole);
1744 result = QString::compare(a->values.value(role).toString(),
1745 b->values.value(role).toString());
1746 break;
1747 }
1748
1749 }
1750
1751 if (result != 0) {
1752 // The current sort role was sufficient to define an order
1753 return result;
1754 }
1755
1756 // Fallback #1: Compare the text of the items
1757 result = stringCompare(itemA.text(), itemB.text());
1758 if (result != 0) {
1759 return result;
1760 }
1761
1762 // Fallback #2: KFileItem::text() may not be unique in case UDS_DISPLAY_NAME is used
1763 result = stringCompare(itemA.name(m_caseSensitivity == Qt::CaseInsensitive),
1764 itemB.name(m_caseSensitivity == Qt::CaseInsensitive));
1765 if (result != 0) {
1766 return result;
1767 }
1768
1769 // Fallback #3: It must be assured that the sort order is always unique even if two values have been
1770 // equal. In this case a comparison of the URL is done which is unique in all cases
1771 // within KDirLister.
1772 return QString::compare(itemA.url().url(), itemB.url().url(), Qt::CaseSensitive);
1773 }
1774
1775 int KFileItemModel::stringCompare(const QString& a, const QString& b) const
1776 {
1777 // Taken from KDirSortFilterProxyModel (kdelibs/kfile/kdirsortfilterproxymodel.*)
1778 // Copyright (C) 2006 by Peter Penz <peter.penz@gmx.at>
1779 // Copyright (C) 2006 by Dominic Battre <dominic@battre.de>
1780 // Copyright (C) 2006 by Martin Pool <mbp@canonical.com>
1781
1782 if (m_caseSensitivity == Qt::CaseInsensitive) {
1783 const int result = m_naturalSorting ? KStringHandler::naturalCompare(a, b, Qt::CaseInsensitive)
1784 : QString::compare(a, b, Qt::CaseInsensitive);
1785 if (result != 0) {
1786 // Only return the result, if the strings are not equal. If they are equal by a case insensitive
1787 // comparison, still a deterministic sort order is required. A case sensitive
1788 // comparison is done as fallback.
1789 return result;
1790 }
1791 }
1792
1793 return m_naturalSorting ? KStringHandler::naturalCompare(a, b, Qt::CaseSensitive)
1794 : QString::compare(a, b, Qt::CaseSensitive);
1795 }
1796
1797 bool KFileItemModel::useMaximumUpdateInterval() const
1798 {
1799 return !m_dirLister->url().isLocalFile();
1800 }
1801
1802 static bool localeAwareLessThan(const QChar& c1, const QChar& c2)
1803 {
1804 return QString::localeAwareCompare(c1, c2) < 0;
1805 }
1806
1807 QList<QPair<int, QVariant> > KFileItemModel::nameRoleGroups() const
1808 {
1809 Q_ASSERT(!m_itemData.isEmpty());
1810
1811 const int maxIndex = count() - 1;
1812 QList<QPair<int, QVariant> > groups;
1813
1814 QString groupValue;
1815 QChar firstChar;
1816 for (int i = 0; i <= maxIndex; ++i) {
1817 if (isChildItem(i)) {
1818 continue;
1819 }
1820
1821 const QString name = m_itemData.at(i)->item.text();
1822
1823 // Use the first character of the name as group indication
1824 QChar newFirstChar = name.at(0).toUpper();
1825 if (newFirstChar == QLatin1Char('~') && name.length() > 1) {
1826 newFirstChar = name.at(1).toUpper();
1827 }
1828
1829 if (firstChar != newFirstChar) {
1830 QString newGroupValue;
1831 if (newFirstChar.isLetter()) {
1832 // Try to find a matching group in the range 'A' to 'Z'.
1833 static std::vector<QChar> lettersAtoZ;
1834 if (lettersAtoZ.empty()) {
1835 for (char c = 'A'; c <= 'Z'; ++c) {
1836 lettersAtoZ.push_back(QLatin1Char(c));
1837 }
1838 }
1839
1840 std::vector<QChar>::iterator it = std::lower_bound(lettersAtoZ.begin(), lettersAtoZ.end(), newFirstChar, localeAwareLessThan);
1841 if (it != lettersAtoZ.end()) {
1842 if (localeAwareLessThan(newFirstChar, *it) && it != lettersAtoZ.begin()) {
1843 // newFirstChar belongs to the group preceding *it.
1844 // Example: for an umlaut 'A' in the German locale, *it would be 'B' now.
1845 --it;
1846 }
1847 newGroupValue = *it;
1848 } else {
1849 newGroupValue = newFirstChar;
1850 }
1851 } else if (newFirstChar >= QLatin1Char('0') && newFirstChar <= QLatin1Char('9')) {
1852 // Apply group '0 - 9' for any name that starts with a digit
1853 newGroupValue = i18nc("@title:group Groups that start with a digit", "0 - 9");
1854 } else {
1855 newGroupValue = i18nc("@title:group", "Others");
1856 }
1857
1858 if (newGroupValue != groupValue) {
1859 groupValue = newGroupValue;
1860 groups.append(QPair<int, QVariant>(i, newGroupValue));
1861 }
1862
1863 firstChar = newFirstChar;
1864 }
1865 }
1866 return groups;
1867 }
1868
1869 QList<QPair<int, QVariant> > KFileItemModel::sizeRoleGroups() const
1870 {
1871 Q_ASSERT(!m_itemData.isEmpty());
1872
1873 const int maxIndex = count() - 1;
1874 QList<QPair<int, QVariant> > groups;
1875
1876 QString groupValue;
1877 for (int i = 0; i <= maxIndex; ++i) {
1878 if (isChildItem(i)) {
1879 continue;
1880 }
1881
1882 const KFileItem& item = m_itemData.at(i)->item;
1883 const KIO::filesize_t fileSize = !item.isNull() ? item.size() : ~0U;
1884 QString newGroupValue;
1885 if (!item.isNull() && item.isDir()) {
1886 newGroupValue = i18nc("@title:group Size", "Folders");
1887 } else if (fileSize < 5 * 1024 * 1024) {
1888 newGroupValue = i18nc("@title:group Size", "Small");
1889 } else if (fileSize < 10 * 1024 * 1024) {
1890 newGroupValue = i18nc("@title:group Size", "Medium");
1891 } else {
1892 newGroupValue = i18nc("@title:group Size", "Big");
1893 }
1894
1895 if (newGroupValue != groupValue) {
1896 groupValue = newGroupValue;
1897 groups.append(QPair<int, QVariant>(i, newGroupValue));
1898 }
1899 }
1900
1901 return groups;
1902 }
1903
1904 QList<QPair<int, QVariant> > KFileItemModel::dateRoleGroups() const
1905 {
1906 Q_ASSERT(!m_itemData.isEmpty());
1907
1908 const int maxIndex = count() - 1;
1909 QList<QPair<int, QVariant> > groups;
1910
1911 const QDate currentDate = QDate::currentDate();
1912
1913 QDate previousModifiedDate;
1914 QString groupValue;
1915 for (int i = 0; i <= maxIndex; ++i) {
1916 if (isChildItem(i)) {
1917 continue;
1918 }
1919
1920 const QDateTime modifiedTime = m_itemData.at(i)->item.time(KFileItem::ModificationTime);
1921 const QDate modifiedDate = modifiedTime.date();
1922 if (modifiedDate == previousModifiedDate) {
1923 // The current item is in the same group as the previous item
1924 continue;
1925 }
1926 previousModifiedDate = modifiedDate;
1927
1928 const int daysDistance = modifiedDate.daysTo(currentDate);
1929
1930 QString newGroupValue;
1931 if (currentDate.year() == modifiedDate.year() && currentDate.month() == modifiedDate.month()) {
1932 switch (daysDistance / 7) {
1933 case 0:
1934 switch (daysDistance) {
1935 case 0: newGroupValue = i18nc("@title:group Date", "Today"); break;
1936 case 1: newGroupValue = i18nc("@title:group Date", "Yesterday"); break;
1937 default: newGroupValue = modifiedTime.toString(i18nc("@title:group The week day name: %A", "%A"));
1938 }
1939 break;
1940 case 1:
1941 newGroupValue = i18nc("@title:group Date", "One Week Ago");
1942 break;
1943 case 2:
1944 newGroupValue = i18nc("@title:group Date", "Two Weeks Ago");
1945 break;
1946 case 3:
1947 newGroupValue = i18nc("@title:group Date", "Three Weeks Ago");
1948 break;
1949 case 4:
1950 case 5:
1951 newGroupValue = i18nc("@title:group Date", "Earlier this Month");
1952 break;
1953 default:
1954 Q_ASSERT(false);
1955 }
1956 } else {
1957 const QDate lastMonthDate = currentDate.addMonths(-1);
1958 if (lastMonthDate.year() == modifiedDate.year() && lastMonthDate.month() == modifiedDate.month()) {
1959 if (daysDistance == 1) {
1960 newGroupValue = modifiedTime.toString(i18nc("@title:group Date: %B is full month name in current locale, and %Y is full year number", "Yesterday (%B, %Y)"));
1961 } else if (daysDistance <= 7) {
1962 newGroupValue = modifiedTime.toString(i18nc("@title:group The week day name: %A, %B is full month name in current locale, and %Y is full year number", "%A (%B, %Y)"));
1963 } else if (daysDistance <= 7 * 2) {
1964 newGroupValue = modifiedTime.toString(i18nc("@title:group Date: %B is full month name in current locale, and %Y is full year number", "One Week Ago (%B, %Y)"));
1965 } else if (daysDistance <= 7 * 3) {
1966 newGroupValue = modifiedTime.toString(i18nc("@title:group Date: %B is full month name in current locale, and %Y is full year number", "Two Weeks Ago (%B, %Y)"));
1967 } else if (daysDistance <= 7 * 4) {
1968 newGroupValue = modifiedTime.toString(i18nc("@title:group Date: %B is full month name in current locale, and %Y is full year number", "Three Weeks Ago (%B, %Y)"));
1969 } else {
1970 newGroupValue = modifiedTime.toString(i18nc("@title:group Date: %B is full month name in current locale, and %Y is full year number", "Earlier on %B, %Y"));
1971 }
1972 } else {
1973 newGroupValue = modifiedTime.toString(i18nc("@title:group The month and year: %B is full month name in current locale, and %Y is full year number", "%B, %Y"));
1974 }
1975 }
1976
1977 if (newGroupValue != groupValue) {
1978 groupValue = newGroupValue;
1979 groups.append(QPair<int, QVariant>(i, newGroupValue));
1980 }
1981 }
1982
1983 return groups;
1984 }
1985
1986 QList<QPair<int, QVariant> > KFileItemModel::permissionRoleGroups() const
1987 {
1988 Q_ASSERT(!m_itemData.isEmpty());
1989
1990 const int maxIndex = count() - 1;
1991 QList<QPair<int, QVariant> > groups;
1992
1993 QString permissionsString;
1994 QString groupValue;
1995 for (int i = 0; i <= maxIndex; ++i) {
1996 if (isChildItem(i)) {
1997 continue;
1998 }
1999
2000 const ItemData* itemData = m_itemData.at(i);
2001 const QString newPermissionsString = itemData->values.value("permissions").toString();
2002 if (newPermissionsString == permissionsString) {
2003 continue;
2004 }
2005 permissionsString = newPermissionsString;
2006
2007 const QFileInfo info(itemData->item.url().toLocalFile());
2008
2009 // Set user string
2010 QString user;
2011 if (info.permission(QFile::ReadUser)) {
2012 user = i18nc("@item:intext Access permission, concatenated", "Read, ");
2013 }
2014 if (info.permission(QFile::WriteUser)) {
2015 user += i18nc("@item:intext Access permission, concatenated", "Write, ");
2016 }
2017 if (info.permission(QFile::ExeUser)) {
2018 user += i18nc("@item:intext Access permission, concatenated", "Execute, ");
2019 }
2020 user = user.isEmpty() ? i18nc("@item:intext Access permission, concatenated", "Forbidden") : user.mid(0, user.count() - 2);
2021
2022 // Set group string
2023 QString group;
2024 if (info.permission(QFile::ReadGroup)) {
2025 group = i18nc("@item:intext Access permission, concatenated", "Read, ");
2026 }
2027 if (info.permission(QFile::WriteGroup)) {
2028 group += i18nc("@item:intext Access permission, concatenated", "Write, ");
2029 }
2030 if (info.permission(QFile::ExeGroup)) {
2031 group += i18nc("@item:intext Access permission, concatenated", "Execute, ");
2032 }
2033 group = group.isEmpty() ? i18nc("@item:intext Access permission, concatenated", "Forbidden") : group.mid(0, group.count() - 2);
2034
2035 // Set others string
2036 QString others;
2037 if (info.permission(QFile::ReadOther)) {
2038 others = i18nc("@item:intext Access permission, concatenated", "Read, ");
2039 }
2040 if (info.permission(QFile::WriteOther)) {
2041 others += i18nc("@item:intext Access permission, concatenated", "Write, ");
2042 }
2043 if (info.permission(QFile::ExeOther)) {
2044 others += i18nc("@item:intext Access permission, concatenated", "Execute, ");
2045 }
2046 others = others.isEmpty() ? i18nc("@item:intext Access permission, concatenated", "Forbidden") : others.mid(0, others.count() - 2);
2047
2048 const QString newGroupValue = i18nc("@title:group Files and folders by permissions", "User: %1 | Group: %2 | Others: %3", user, group, others);
2049 if (newGroupValue != groupValue) {
2050 groupValue = newGroupValue;
2051 groups.append(QPair<int, QVariant>(i, newGroupValue));
2052 }
2053 }
2054
2055 return groups;
2056 }
2057
2058 QList<QPair<int, QVariant> > KFileItemModel::ratingRoleGroups() const
2059 {
2060 Q_ASSERT(!m_itemData.isEmpty());
2061
2062 const int maxIndex = count() - 1;
2063 QList<QPair<int, QVariant> > groups;
2064
2065 int groupValue = -1;
2066 for (int i = 0; i <= maxIndex; ++i) {
2067 if (isChildItem(i)) {
2068 continue;
2069 }
2070 const int newGroupValue = m_itemData.at(i)->values.value("rating", 0).toInt();
2071 if (newGroupValue != groupValue) {
2072 groupValue = newGroupValue;
2073 groups.append(QPair<int, QVariant>(i, newGroupValue));
2074 }
2075 }
2076
2077 return groups;
2078 }
2079
2080 QList<QPair<int, QVariant> > KFileItemModel::genericStringRoleGroups(const QByteArray& role) const
2081 {
2082 Q_ASSERT(!m_itemData.isEmpty());
2083
2084 const int maxIndex = count() - 1;
2085 QList<QPair<int, QVariant> > groups;
2086
2087 bool isFirstGroupValue = true;
2088 QString groupValue;
2089 for (int i = 0; i <= maxIndex; ++i) {
2090 if (isChildItem(i)) {
2091 continue;
2092 }
2093 const QString newGroupValue = m_itemData.at(i)->values.value(role).toString();
2094 if (newGroupValue != groupValue || isFirstGroupValue) {
2095 groupValue = newGroupValue;
2096 groups.append(QPair<int, QVariant>(i, newGroupValue));
2097 isFirstGroupValue = false;
2098 }
2099 }
2100
2101 return groups;
2102 }
2103
2104 void KFileItemModel::emitSortProgress(int resolvedCount)
2105 {
2106 // Be tolerant against a resolvedCount with a wrong range.
2107 // Although there should not be a case where KFileItemModelRolesUpdater
2108 // (= caller) provides a wrong range, it is important to emit
2109 // a useful progress information even if there is an unexpected
2110 // implementation issue.
2111
2112 const int itemCount = count();
2113 if (resolvedCount >= itemCount) {
2114 m_sortingProgressPercent = -1;
2115 if (m_resortAllItemsTimer->isActive()) {
2116 m_resortAllItemsTimer->stop();
2117 resortAllItems();
2118 }
2119
2120 emit directorySortingProgress(100);
2121 } else if (itemCount > 0) {
2122 resolvedCount = qBound(0, resolvedCount, itemCount);
2123
2124 const int progress = resolvedCount * 100 / itemCount;
2125 if (m_sortingProgressPercent != progress) {
2126 m_sortingProgressPercent = progress;
2127 emit directorySortingProgress(progress);
2128 }
2129 }
2130 }
2131
2132 const KFileItemModel::RoleInfoMap* KFileItemModel::rolesInfoMap(int& count)
2133 {
2134 static const RoleInfoMap rolesInfoMap[] = {
2135 // | role | roleType | role translation | group translation | requires Baloo | requires indexer
2136 { 0, NoRole, 0, 0, 0, 0, false, false },
2137 { "text", NameRole, I18N_NOOP2_NOSTRIP("@label", "Name"), 0, 0, false, false },
2138 { "size", SizeRole, I18N_NOOP2_NOSTRIP("@label", "Size"), 0, 0, false, false },
2139 { "date", DateRole, I18N_NOOP2_NOSTRIP("@label", "Date"), 0, 0, false, false },
2140 { "type", TypeRole, I18N_NOOP2_NOSTRIP("@label", "Type"), 0, 0, false, false },
2141 { "rating", RatingRole, I18N_NOOP2_NOSTRIP("@label", "Rating"), 0, 0, true, false },
2142 { "tags", TagsRole, I18N_NOOP2_NOSTRIP("@label", "Tags"), 0, 0, true, false },
2143 { "comment", CommentRole, I18N_NOOP2_NOSTRIP("@label", "Comment"), 0, 0, true, false },
2144 { "wordCount", WordCountRole, I18N_NOOP2_NOSTRIP("@label", "Word Count"), I18N_NOOP2_NOSTRIP("@label", "Document"), true, true },
2145 { "lineCount", LineCountRole, I18N_NOOP2_NOSTRIP("@label", "Line Count"), I18N_NOOP2_NOSTRIP("@label", "Document"), true, true },
2146 { "imageSize", ImageSizeRole, I18N_NOOP2_NOSTRIP("@label", "Image Size"), I18N_NOOP2_NOSTRIP("@label", "Image"), true, true },
2147 { "orientation", OrientationRole, I18N_NOOP2_NOSTRIP("@label", "Orientation"), I18N_NOOP2_NOSTRIP("@label", "Image"), true, true },
2148 { "artist", ArtistRole, I18N_NOOP2_NOSTRIP("@label", "Artist"), I18N_NOOP2_NOSTRIP("@label", "Audio"), true, true },
2149 { "album", AlbumRole, I18N_NOOP2_NOSTRIP("@label", "Album"), I18N_NOOP2_NOSTRIP("@label", "Audio"), true, true },
2150 { "duration", DurationRole, I18N_NOOP2_NOSTRIP("@label", "Duration"), I18N_NOOP2_NOSTRIP("@label", "Audio"), true, true },
2151 { "track", TrackRole, I18N_NOOP2_NOSTRIP("@label", "Track"), I18N_NOOP2_NOSTRIP("@label", "Audio"), true, true },
2152 { "path", PathRole, I18N_NOOP2_NOSTRIP("@label", "Path"), I18N_NOOP2_NOSTRIP("@label", "Other"), false, false },
2153 { "destination", DestinationRole, I18N_NOOP2_NOSTRIP("@label", "Link Destination"), I18N_NOOP2_NOSTRIP("@label", "Other"), false, false },
2154 { "copiedFrom", CopiedFromRole, I18N_NOOP2_NOSTRIP("@label", "Copied From"), I18N_NOOP2_NOSTRIP("@label", "Other"), true, false },
2155 { "permissions", PermissionsRole, I18N_NOOP2_NOSTRIP("@label", "Permissions"), I18N_NOOP2_NOSTRIP("@label", "Other"), false, false },
2156 { "owner", OwnerRole, I18N_NOOP2_NOSTRIP("@label", "Owner"), I18N_NOOP2_NOSTRIP("@label", "Other"), false, false },
2157 { "group", GroupRole, I18N_NOOP2_NOSTRIP("@label", "User Group"), I18N_NOOP2_NOSTRIP("@label", "Other"), false, false },
2158 };
2159
2160 count = sizeof(rolesInfoMap) / sizeof(RoleInfoMap);
2161 return rolesInfoMap;
2162 }
2163
2164 void KFileItemModel::determineMimeTypes(const KFileItemList& items, int timeout)
2165 {
2166 QElapsedTimer timer;
2167 timer.start();
2168 foreach (const KFileItem& item, items) { // krazy:exclude=foreach
2169 // Only determine mime types for files here. For directories,
2170 // KFileItem::determineMimeType() reads the .directory file inside to
2171 // load the icon, but this is not necessary at all if we just need the
2172 // type. Some special code for setting the correct mime type for
2173 // directories is in retrieveData().
2174 if (!item.isDir()) {
2175 item.determineMimeType();
2176 }
2177
2178 if (timer.elapsed() > timeout) {
2179 // Don't block the user interface, let the remaining items
2180 // be resolved asynchronously.
2181 return;
2182 }
2183 }
2184 }
2185
2186 QByteArray KFileItemModel::sharedValue(const QByteArray& value)
2187 {
2188 static QSet<QByteArray> pool;
2189 const QSet<QByteArray>::const_iterator it = pool.constFind(value);
2190
2191 if (it != pool.constEnd()) {
2192 return *it;
2193 } else {
2194 pool.insert(value);
2195 return value;
2196 }
2197 }
2198
2199 bool KFileItemModel::isConsistent() const
2200 {
2201 // m_items may contain less items than m_itemData because m_items
2202 // is populated lazily, see KFileItemModel::index(const QUrl& url).
2203 if (m_items.count() > m_itemData.count()) {
2204 return false;
2205 }
2206
2207 for (int i = 0; i < count(); ++i) {
2208 // Check if m_items and m_itemData are consistent.
2209 const KFileItem item = fileItem(i);
2210 if (item.isNull()) {
2211 qWarning() << "Item" << i << "is null";
2212 return false;
2213 }
2214
2215 const int itemIndex = index(item);
2216 if (itemIndex != i) {
2217 qWarning() << "Item" << i << "has a wrong index:" << itemIndex;
2218 return false;
2219 }
2220
2221 // Check if the items are sorted correctly.
2222 if (i > 0 && !lessThan(m_itemData.at(i - 1), m_itemData.at(i))) {
2223 qWarning() << "The order of items" << i - 1 << "and" << i << "is wrong:"
2224 << fileItem(i - 1) << fileItem(i);
2225 return false;
2226 }
2227
2228 // Check if all parent-child relationships are consistent.
2229 const ItemData* data = m_itemData.at(i);
2230 const ItemData* parent = data->parent;
2231 if (parent) {
2232 if (expandedParentsCount(data) != expandedParentsCount(parent) + 1) {
2233 qWarning() << "expandedParentsCount is inconsistent for parent" << parent->item << "and child" << data->item;
2234 return false;
2235 }
2236
2237 const int parentIndex = index(parent->item);
2238 if (parentIndex >= i) {
2239 qWarning() << "Index" << parentIndex << "of parent" << parent->item << "is not smaller than index" << i << "of child" << data->item;
2240 return false;
2241 }
2242 }
2243 }
2244
2245 return true;
2246 }
2247