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