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