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