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