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