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