]> cloud.milkyroute.net Git - dolphin.git/blob - src/kitemviews/kfileitemmodelrolesupdater.cpp
Fix icon resize animation
[dolphin.git] / src / kitemviews / kfileitemmodelrolesupdater.cpp
1 /*
2 * SPDX-FileCopyrightText: 2011 Peter Penz <peter.penz19@gmail.com>
3 *
4 * SPDX-License-Identifier: GPL-2.0-or-later
5 */
6
7 #include "kfileitemmodelrolesupdater.h"
8
9 #include "dolphindebug.h"
10 #include "kfileitemmodel.h"
11 #include "private/kdirectorycontentscounter.h"
12 #include "private/kpixmapmodifier.h"
13
14 #include <KConfig>
15 #include <KConfigGroup>
16 #include <KIO/JobUiDelegate>
17 #include <KIO/PreviewJob>
18 #include <KIconLoader>
19 #include <KJobWidgets>
20 #include <KOverlayIconPlugin>
21 #include <KPluginMetaData>
22 #include <KSharedConfig>
23
24 #ifdef HAVE_BALOO
25 #include "private/kbaloorolesprovider.h"
26 #include <Baloo/File>
27 #include <Baloo/FileMonitor>
28 #endif
29
30 #include <QApplication>
31 #include <QIcon>
32 #include <QPainter>
33 #include <QPluginLoader>
34 #include <QElapsedTimer>
35 #include <QTimer>
36
37 // #define KFILEITEMMODELROLESUPDATER_DEBUG
38
39 namespace {
40 // Maximum time in ms that the KFileItemModelRolesUpdater
41 // may perform a blocking operation
42 const int MaxBlockTimeout = 200;
43
44 // If the number of items is smaller than ResolveAllItemsLimit,
45 // the roles of all items will be resolved.
46 const int ResolveAllItemsLimit = 500;
47
48 // Not only the visible area, but up to ReadAheadPages before and after
49 // this area will be resolved.
50 const int ReadAheadPages = 5;
51 }
52
53 KFileItemModelRolesUpdater::KFileItemModelRolesUpdater(KFileItemModel* model, QObject* parent) :
54 QObject(parent),
55 m_state(Idle),
56 m_previewChangedDuringPausing(false),
57 m_iconSizeChangedDuringPausing(false),
58 m_rolesChangedDuringPausing(false),
59 m_previewShown(false),
60 m_enlargeSmallPreviews(true),
61 m_clearPreviews(false),
62 m_finishedItems(),
63 m_model(model),
64 m_iconSize(),
65 m_firstVisibleIndex(0),
66 m_lastVisibleIndex(-1),
67 m_maximumVisibleItems(50),
68 m_roles(),
69 m_resolvableRoles(),
70 m_enabledPlugins(),
71 m_localFileSizePreviewLimit(0),
72 m_scanDirectories(true),
73 m_pendingSortRoleItems(),
74 m_pendingIndexes(),
75 m_pendingPreviewItems(),
76 m_previewJob(),
77 m_hoverSequenceItem(),
78 m_hoverSequenceIndex(0),
79 m_hoverSequencePreviewJob(nullptr),
80 m_hoverSequenceNumSuccessiveFailures(0),
81 m_recentlyChangedItemsTimer(nullptr),
82 m_recentlyChangedItems(),
83 m_changedItems(),
84 m_directoryContentsCounter(nullptr)
85 #ifdef HAVE_BALOO
86 , m_balooFileMonitor(nullptr)
87 #endif
88 {
89 Q_ASSERT(model);
90
91 const KConfigGroup globalConfig(KSharedConfig::openConfig(), "PreviewSettings");
92 m_enabledPlugins = globalConfig.readEntry("Plugins", KIO::PreviewJob::defaultPlugins());
93 m_localFileSizePreviewLimit = static_cast<qulonglong>(globalConfig.readEntry("MaximumSize", 0));
94
95 connect(m_model, &KFileItemModel::itemsInserted,
96 this, &KFileItemModelRolesUpdater::slotItemsInserted);
97 connect(m_model, &KFileItemModel::itemsRemoved,
98 this, &KFileItemModelRolesUpdater::slotItemsRemoved);
99 connect(m_model, &KFileItemModel::itemsChanged,
100 this, &KFileItemModelRolesUpdater::slotItemsChanged);
101 connect(m_model, &KFileItemModel::itemsMoved,
102 this, &KFileItemModelRolesUpdater::slotItemsMoved);
103 connect(m_model, &KFileItemModel::sortRoleChanged,
104 this, &KFileItemModelRolesUpdater::slotSortRoleChanged);
105
106 // Use a timer to prevent that each call of slotItemsChanged() results in a synchronous
107 // resolving of the roles. Postpone the resolving until no update has been done for 100 ms.
108 m_recentlyChangedItemsTimer = new QTimer(this);
109 m_recentlyChangedItemsTimer->setInterval(100);
110 m_recentlyChangedItemsTimer->setSingleShot(true);
111 connect(m_recentlyChangedItemsTimer, &QTimer::timeout, this, &KFileItemModelRolesUpdater::resolveRecentlyChangedItems);
112
113 m_resolvableRoles.insert("size");
114 m_resolvableRoles.insert("type");
115 m_resolvableRoles.insert("isExpandable");
116 #ifdef HAVE_BALOO
117 m_resolvableRoles += KBalooRolesProvider::instance().roles();
118 #endif
119
120 m_directoryContentsCounter = new KDirectoryContentsCounter(m_model, this);
121 connect(m_directoryContentsCounter, &KDirectoryContentsCounter::result,
122 this, &KFileItemModelRolesUpdater::slotDirectoryContentsCountReceived);
123
124 const auto plugins = KPluginMetaData::findPlugins(QStringLiteral("kf5/overlayicon"));
125 for (const KPluginMetaData &data : plugins) {
126 auto instance = QPluginLoader(data.fileName()).instance();
127 auto plugin = qobject_cast<KOverlayIconPlugin *>(instance);
128 if (plugin) {
129 m_overlayIconsPlugin.append(plugin);
130 connect(plugin, &KOverlayIconPlugin::overlaysChanged, this, &KFileItemModelRolesUpdater::slotOverlaysChanged);
131 } else {
132 // not our/valid plugin, so delete the created object
133 delete instance;
134 }
135 }
136 }
137
138 KFileItemModelRolesUpdater::~KFileItemModelRolesUpdater()
139 {
140 killPreviewJob();
141 }
142
143 void KFileItemModelRolesUpdater::setIconSize(const QSize& size)
144 {
145 if (size != m_iconSize) {
146 m_iconSize = size;
147 if (m_state == Paused) {
148 m_iconSizeChangedDuringPausing = true;
149 } else if (m_previewShown) {
150 // An icon size change requires the regenerating of
151 // all previews
152 m_finishedItems.clear();
153 startUpdating();
154 }
155 }
156 }
157
158 QSize KFileItemModelRolesUpdater::iconSize() const
159 {
160 return m_iconSize;
161 }
162
163 void KFileItemModelRolesUpdater::setVisibleIndexRange(int index, int count)
164 {
165 if (index < 0) {
166 index = 0;
167 }
168 if (count < 0) {
169 count = 0;
170 }
171
172 if (index == m_firstVisibleIndex && count == m_lastVisibleIndex - m_firstVisibleIndex + 1) {
173 // The range has not been changed
174 return;
175 }
176
177 m_firstVisibleIndex = index;
178 m_lastVisibleIndex = qMin(index + count - 1, m_model->count() - 1);
179
180 startUpdating();
181 }
182
183 void KFileItemModelRolesUpdater::setMaximumVisibleItems(int count)
184 {
185 m_maximumVisibleItems = count;
186 }
187
188 void KFileItemModelRolesUpdater::setPreviewsShown(bool show)
189 {
190 if (show == m_previewShown) {
191 return;
192 }
193
194 m_previewShown = show;
195 if (!show) {
196 m_clearPreviews = true;
197 }
198
199 updateAllPreviews();
200 }
201
202 bool KFileItemModelRolesUpdater::previewsShown() const
203 {
204 return m_previewShown;
205 }
206
207 void KFileItemModelRolesUpdater::setEnlargeSmallPreviews(bool enlarge)
208 {
209 if (enlarge != m_enlargeSmallPreviews) {
210 m_enlargeSmallPreviews = enlarge;
211 if (m_previewShown) {
212 updateAllPreviews();
213 }
214 }
215 }
216
217 bool KFileItemModelRolesUpdater::enlargeSmallPreviews() const
218 {
219 return m_enlargeSmallPreviews;
220 }
221
222 void KFileItemModelRolesUpdater::setEnabledPlugins(const QStringList& list)
223 {
224 if (m_enabledPlugins != list) {
225 m_enabledPlugins = list;
226 if (m_previewShown) {
227 updateAllPreviews();
228 }
229 }
230 }
231
232 void KFileItemModelRolesUpdater::setPaused(bool paused)
233 {
234 if (paused == (m_state == Paused)) {
235 return;
236 }
237
238 if (paused) {
239 m_state = Paused;
240 killPreviewJob();
241 } else {
242 const bool updatePreviews = (m_iconSizeChangedDuringPausing && m_previewShown) ||
243 m_previewChangedDuringPausing;
244 const bool resolveAll = updatePreviews || m_rolesChangedDuringPausing;
245 if (resolveAll) {
246 m_finishedItems.clear();
247 }
248
249 m_iconSizeChangedDuringPausing = false;
250 m_previewChangedDuringPausing = false;
251 m_rolesChangedDuringPausing = false;
252
253 if (!m_pendingSortRoleItems.isEmpty()) {
254 m_state = ResolvingSortRole;
255 resolveNextSortRole();
256 } else {
257 m_state = Idle;
258 }
259
260 startUpdating();
261 }
262 }
263
264 void KFileItemModelRolesUpdater::setRoles(const QSet<QByteArray>& roles)
265 {
266 if (m_roles != roles) {
267 m_roles = roles;
268
269 #ifdef HAVE_BALOO
270 // Check whether there is at least one role that must be resolved
271 // with the help of Baloo. If this is the case, a (quite expensive)
272 // resolving will be done in KFileItemModelRolesUpdater::rolesData() and
273 // the role gets watched for changes.
274 const KBalooRolesProvider& rolesProvider = KBalooRolesProvider::instance();
275 bool hasBalooRole = false;
276 QSetIterator<QByteArray> it(roles);
277 while (it.hasNext()) {
278 const QByteArray& role = it.next();
279 if (rolesProvider.roles().contains(role)) {
280 hasBalooRole = true;
281 break;
282 }
283 }
284
285 if (hasBalooRole && m_balooConfig.fileIndexingEnabled() && !m_balooFileMonitor) {
286 m_balooFileMonitor = new Baloo::FileMonitor(this);
287 connect(m_balooFileMonitor, &Baloo::FileMonitor::fileMetaDataChanged,
288 this, &KFileItemModelRolesUpdater::applyChangedBalooRoles);
289 } else if (!hasBalooRole && m_balooFileMonitor) {
290 delete m_balooFileMonitor;
291 m_balooFileMonitor = nullptr;
292 }
293 #endif
294
295 if (m_state == Paused) {
296 m_rolesChangedDuringPausing = true;
297 } else {
298 startUpdating();
299 }
300 }
301 }
302
303 QSet<QByteArray> KFileItemModelRolesUpdater::roles() const
304 {
305 return m_roles;
306 }
307
308 bool KFileItemModelRolesUpdater::isPaused() const
309 {
310 return m_state == Paused;
311 }
312
313 QStringList KFileItemModelRolesUpdater::enabledPlugins() const
314 {
315 return m_enabledPlugins;
316 }
317
318 void KFileItemModelRolesUpdater::setLocalFileSizePreviewLimit(const qlonglong size)
319 {
320 m_localFileSizePreviewLimit = size;
321 }
322
323 qlonglong KFileItemModelRolesUpdater::localFileSizePreviewLimit() const
324 {
325 return m_localFileSizePreviewLimit;
326 }
327
328 void KFileItemModelRolesUpdater::setScanDirectories(bool enabled)
329 {
330 m_scanDirectories = enabled;
331 }
332
333 bool KFileItemModelRolesUpdater::scanDirectories() const
334 {
335 return m_scanDirectories;
336 }
337
338 void KFileItemModelRolesUpdater::setHoverSequenceState(const QUrl& itemUrl, int seqIdx)
339 {
340 const KFileItem item = m_model->fileItem(itemUrl);
341
342 if (item != m_hoverSequenceItem) {
343 killHoverSequencePreviewJob();
344 }
345
346 m_hoverSequenceItem = item;
347 m_hoverSequenceIndex = seqIdx;
348
349 if (!m_previewShown) {
350 return;
351 }
352
353 m_hoverSequenceNumSuccessiveFailures = 0;
354
355 loadNextHoverSequencePreview();
356 }
357
358 void KFileItemModelRolesUpdater::slotItemsInserted(const KItemRangeList& itemRanges)
359 {
360 QElapsedTimer timer;
361 timer.start();
362
363 // Determine the sort role synchronously for as many items as possible.
364 if (m_resolvableRoles.contains(m_model->sortRole())) {
365 int insertedCount = 0;
366 for (const KItemRange& range : itemRanges) {
367 const int lastIndex = insertedCount + range.index + range.count - 1;
368 for (int i = insertedCount + range.index; i <= lastIndex; ++i) {
369 if (timer.elapsed() < MaxBlockTimeout) {
370 applySortRole(i);
371 } else {
372 m_pendingSortRoleItems.insert(m_model->fileItem(i));
373 }
374 }
375 insertedCount += range.count;
376 }
377
378 applySortProgressToModel();
379
380 // If there are still items whose sort role is unknown, check if the
381 // asynchronous determination of the sort role is already in progress,
382 // and start it if that is not the case.
383 if (!m_pendingSortRoleItems.isEmpty() && m_state != ResolvingSortRole) {
384 killPreviewJob();
385 m_state = ResolvingSortRole;
386 resolveNextSortRole();
387 }
388 }
389
390 startUpdating();
391 }
392
393 void KFileItemModelRolesUpdater::slotItemsRemoved(const KItemRangeList& itemRanges)
394 {
395 Q_UNUSED(itemRanges)
396
397 const bool allItemsRemoved = (m_model->count() == 0);
398
399 #ifdef HAVE_BALOO
400 if (m_balooFileMonitor) {
401 // Don't let the FileWatcher watch for removed items
402 if (allItemsRemoved) {
403 m_balooFileMonitor->clear();
404 } else {
405 QStringList newFileList;
406 const QStringList oldFileList = m_balooFileMonitor->files();
407 for (const QString& file : oldFileList) {
408 if (m_model->index(QUrl::fromLocalFile(file)) >= 0) {
409 newFileList.append(file);
410 }
411 }
412 m_balooFileMonitor->setFiles(newFileList);
413 }
414 }
415 #endif
416
417 if (allItemsRemoved) {
418 m_state = Idle;
419
420 m_finishedItems.clear();
421 m_pendingSortRoleItems.clear();
422 m_pendingIndexes.clear();
423 m_pendingPreviewItems.clear();
424 m_recentlyChangedItems.clear();
425 m_recentlyChangedItemsTimer->stop();
426 m_changedItems.clear();
427 m_hoverSequenceLoadedItems.clear();
428
429 killPreviewJob();
430 } else {
431 // Only remove the items from m_finishedItems. They will be removed
432 // from the other sets later on.
433 QSet<KFileItem>::iterator it = m_finishedItems.begin();
434 while (it != m_finishedItems.end()) {
435 if (m_model->index(*it) < 0) {
436 it = m_finishedItems.erase(it);
437 } else {
438 ++it;
439 }
440 }
441
442 // Removed items won't have hover previews loaded anymore.
443 for (const KItemRange& itemRange : itemRanges) {
444 int index = itemRange.index;
445 for (int count = itemRange.count; count > 0; --count) {
446 const KFileItem item = m_model->fileItem(index);
447 m_hoverSequenceLoadedItems.remove(item);
448 ++index;
449 }
450 }
451
452 // The visible items might have changed.
453 startUpdating();
454 }
455 }
456
457 void KFileItemModelRolesUpdater::slotItemsMoved(const KItemRange& itemRange, const QList<int> &movedToIndexes)
458 {
459 Q_UNUSED(itemRange)
460 Q_UNUSED(movedToIndexes)
461
462 // The visible items might have changed.
463 startUpdating();
464 }
465
466 void KFileItemModelRolesUpdater::slotItemsChanged(const KItemRangeList& itemRanges,
467 const QSet<QByteArray>& roles)
468 {
469 Q_UNUSED(roles)
470
471 // Find out if slotItemsChanged() has been done recently. If that is the
472 // case, resolving the roles is postponed until a timer has exceeded
473 // to prevent expensive repeated updates if files are updated frequently.
474 const bool itemsChangedRecently = m_recentlyChangedItemsTimer->isActive();
475
476 QSet<KFileItem>& targetSet = itemsChangedRecently ? m_recentlyChangedItems : m_changedItems;
477
478 for (const KItemRange& itemRange : itemRanges) {
479 int index = itemRange.index;
480 for (int count = itemRange.count; count > 0; --count) {
481 const KFileItem item = m_model->fileItem(index);
482 targetSet.insert(item);
483 ++index;
484 }
485 }
486
487 m_recentlyChangedItemsTimer->start();
488
489 if (!itemsChangedRecently) {
490 updateChangedItems();
491 }
492 }
493
494 void KFileItemModelRolesUpdater::slotSortRoleChanged(const QByteArray& current,
495 const QByteArray& previous)
496 {
497 Q_UNUSED(current)
498 Q_UNUSED(previous)
499
500 if (m_resolvableRoles.contains(current)) {
501 m_pendingSortRoleItems.clear();
502 m_finishedItems.clear();
503
504 const int count = m_model->count();
505 QElapsedTimer timer;
506 timer.start();
507
508 // Determine the sort role synchronously for as many items as possible.
509 for (int index = 0; index < count; ++index) {
510 if (timer.elapsed() < MaxBlockTimeout) {
511 applySortRole(index);
512 } else {
513 m_pendingSortRoleItems.insert(m_model->fileItem(index));
514 }
515 }
516
517 applySortProgressToModel();
518
519 if (!m_pendingSortRoleItems.isEmpty()) {
520 // Trigger the asynchronous determination of the sort role.
521 killPreviewJob();
522 m_state = ResolvingSortRole;
523 resolveNextSortRole();
524 }
525 } else {
526 m_state = Idle;
527 m_pendingSortRoleItems.clear();
528 applySortProgressToModel();
529 }
530 }
531
532 void KFileItemModelRolesUpdater::slotGotPreview(const KFileItem& item, const QPixmap& pixmap)
533 {
534 if (m_state != PreviewJobRunning) {
535 return;
536 }
537
538 m_changedItems.remove(item);
539
540 const int index = m_model->index(item);
541 if (index < 0) {
542 return;
543 }
544
545 QPixmap scaledPixmap = transformPreviewPixmap(pixmap);
546
547 QHash<QByteArray, QVariant> data = rolesData(item);
548
549 const QStringList overlays = data["iconOverlays"].toStringList();
550 // Strangely KFileItem::overlays() returns empty string-values, so
551 // we need to check first whether an overlay must be drawn at all.
552 // It is more efficient to do it here, as KIconLoader::drawOverlays()
553 // assumes that an overlay will be drawn and has some additional
554 // setup time.
555 if (!scaledPixmap.isNull()) {
556 for (const QString& overlay : overlays) {
557 if (!overlay.isEmpty()) {
558 // There is at least one overlay, draw all overlays above m_pixmap
559 // and cancel the check
560 KIconLoader::global()->drawOverlays(overlays, scaledPixmap, KIconLoader::Desktop);
561 break;
562 }
563 }
564 }
565
566 data.insert("iconPixmap", scaledPixmap);
567
568 disconnect(m_model, &KFileItemModel::itemsChanged,
569 this, &KFileItemModelRolesUpdater::slotItemsChanged);
570 m_model->setData(index, data);
571 connect(m_model, &KFileItemModel::itemsChanged,
572 this, &KFileItemModelRolesUpdater::slotItemsChanged);
573
574 m_finishedItems.insert(item);
575 }
576
577 void KFileItemModelRolesUpdater::slotPreviewFailed(const KFileItem& item)
578 {
579 if (m_state != PreviewJobRunning) {
580 return;
581 }
582
583 m_changedItems.remove(item);
584
585 const int index = m_model->index(item);
586 if (index >= 0) {
587 QHash<QByteArray, QVariant> data;
588 data.insert("iconPixmap", QPixmap());
589
590 disconnect(m_model, &KFileItemModel::itemsChanged,
591 this, &KFileItemModelRolesUpdater::slotItemsChanged);
592 m_model->setData(index, data);
593 connect(m_model, &KFileItemModel::itemsChanged,
594 this, &KFileItemModelRolesUpdater::slotItemsChanged);
595
596 applyResolvedRoles(index, ResolveAll);
597 m_finishedItems.insert(item);
598 }
599 }
600
601 void KFileItemModelRolesUpdater::slotPreviewJobFinished()
602 {
603 m_previewJob = nullptr;
604
605 if (m_state != PreviewJobRunning) {
606 return;
607 }
608
609 m_state = Idle;
610
611 if (!m_pendingPreviewItems.isEmpty()) {
612 startPreviewJob();
613 } else {
614 if (!m_changedItems.isEmpty()) {
615 updateChangedItems();
616 }
617 }
618 }
619
620 void KFileItemModelRolesUpdater::slotHoverSequenceGotPreview(const KFileItem& item, const QPixmap& pixmap)
621 {
622 const int index = m_model->index(item);
623 if (index < 0) {
624 return;
625 }
626
627 QHash<QByteArray, QVariant> data = m_model->data(index);
628 QVector<QPixmap> pixmaps = data["hoverSequencePixmaps"].value<QVector<QPixmap>>();
629 const int loadedIndex = pixmaps.size();
630
631 float wap = m_hoverSequencePreviewJob->sequenceIndexWraparoundPoint();
632 if (!m_hoverSequencePreviewJob->handlesSequences()) {
633 wap = 1.0f;
634 }
635 if (wap >= 0.0f) {
636 data["hoverSequenceWraparoundPoint"] = wap;
637 m_model->setData(index, data);
638 }
639
640 // For hover sequence previews we never load index 0, because that's just the regular preview
641 // in "iconPixmap". But that means we'll load index 1 even for thumbnailers that don't support
642 // sequences, in which case we can just throw away the preview because it's the same as for
643 // index 0. Unfortunately we can't find it out earlier :(
644 if (wap < 0.0f || loadedIndex < static_cast<int>(wap)) {
645 // Add the preview to the model data
646
647 const QPixmap scaledPixmap = transformPreviewPixmap(pixmap);
648
649 pixmaps.append(scaledPixmap);
650 data["hoverSequencePixmaps"] = QVariant::fromValue(pixmaps);
651
652 m_model->setData(index, data);
653
654 const auto loadedIt = std::find(m_hoverSequenceLoadedItems.begin(),
655 m_hoverSequenceLoadedItems.end(), item);
656 if (loadedIt == m_hoverSequenceLoadedItems.end()) {
657 m_hoverSequenceLoadedItems.push_back(item);
658 trimHoverSequenceLoadedItems();
659 }
660 }
661
662 m_hoverSequenceNumSuccessiveFailures = 0;
663 }
664
665 void KFileItemModelRolesUpdater::slotHoverSequencePreviewFailed(const KFileItem& item)
666 {
667 const int index = m_model->index(item);
668 if (index < 0) {
669 return;
670 }
671
672 static const int numRetries = 2;
673
674 QHash<QByteArray, QVariant> data = m_model->data(index);
675 QVector<QPixmap> pixmaps = data["hoverSequencePixmaps"].value<QVector<QPixmap>>();
676
677 qCDebug(DolphinDebug).nospace()
678 << "Failed to generate hover sequence preview #" << pixmaps.size()
679 << " for file " << item.url().toString()
680 << " (attempt " << (m_hoverSequenceNumSuccessiveFailures+1)
681 << "/" << (numRetries+1) << ")";
682
683 if (m_hoverSequenceNumSuccessiveFailures >= numRetries) {
684 // Give up and simply duplicate the previous sequence image (if any)
685
686 pixmaps.append(pixmaps.empty() ? QPixmap() : pixmaps.last());
687 data["hoverSequencePixmaps"] = QVariant::fromValue(pixmaps);
688
689 if (!data.contains("hoverSequenceWraparoundPoint")) {
690 // hoverSequenceWraparoundPoint is only available when PreviewJob succeeds, so unless
691 // it has previously succeeded, it's best to assume that it just doesn't handle
692 // sequences instead of trying to load the next image indefinitely.
693 data["hoverSequenceWraparoundPoint"] = 1.0f;
694 }
695
696 m_model->setData(index, data);
697
698 m_hoverSequenceNumSuccessiveFailures = 0;
699 } else {
700 // Retry
701
702 m_hoverSequenceNumSuccessiveFailures++;
703 }
704
705 // Next image in the sequence (or same one if the retry limit wasn't reached yet) will be
706 // loaded automatically, because slotHoverSequencePreviewJobFinished() will be triggered
707 // even when PreviewJob fails.
708 }
709
710 void KFileItemModelRolesUpdater::slotHoverSequencePreviewJobFinished()
711 {
712 const int index = m_model->index(m_hoverSequenceItem);
713 if (index < 0) {
714 m_hoverSequencePreviewJob = nullptr;
715 return;
716 }
717
718 // Since a PreviewJob can only have one associated sequence index, we can only generate
719 // one sequence image per job, so we have to start another one for the next index.
720
721 // Load the next image in the sequence
722 m_hoverSequencePreviewJob = nullptr;
723 loadNextHoverSequencePreview();
724 }
725
726 void KFileItemModelRolesUpdater::resolveNextSortRole()
727 {
728 if (m_state != ResolvingSortRole) {
729 return;
730 }
731
732 QSet<KFileItem>::iterator it = m_pendingSortRoleItems.begin();
733 while (it != m_pendingSortRoleItems.end()) {
734 const KFileItem item = *it;
735 const int index = m_model->index(item);
736
737 // Continue if the sort role has already been determined for the
738 // item, and the item has not been changed recently.
739 if (!m_changedItems.contains(item) && m_model->data(index).contains(m_model->sortRole())) {
740 it = m_pendingSortRoleItems.erase(it);
741 continue;
742 }
743
744 applySortRole(index);
745 m_pendingSortRoleItems.erase(it);
746 break;
747 }
748
749 if (!m_pendingSortRoleItems.isEmpty()) {
750 applySortProgressToModel();
751 QTimer::singleShot(0, this, &KFileItemModelRolesUpdater::resolveNextSortRole);
752 } else {
753 m_state = Idle;
754
755 // Prevent that we try to update the items twice.
756 disconnect(m_model, &KFileItemModel::itemsMoved,
757 this, &KFileItemModelRolesUpdater::slotItemsMoved);
758 applySortProgressToModel();
759 connect(m_model, &KFileItemModel::itemsMoved,
760 this, &KFileItemModelRolesUpdater::slotItemsMoved);
761 startUpdating();
762 }
763 }
764
765 void KFileItemModelRolesUpdater::resolveNextPendingRoles()
766 {
767 if (m_state != ResolvingAllRoles) {
768 return;
769 }
770
771 while (!m_pendingIndexes.isEmpty()) {
772 const int index = m_pendingIndexes.takeFirst();
773 const KFileItem item = m_model->fileItem(index);
774
775 if (m_finishedItems.contains(item)) {
776 continue;
777 }
778
779 applyResolvedRoles(index, ResolveAll);
780 m_finishedItems.insert(item);
781 m_changedItems.remove(item);
782 break;
783 }
784
785 if (!m_pendingIndexes.isEmpty()) {
786 QTimer::singleShot(0, this, &KFileItemModelRolesUpdater::resolveNextPendingRoles);
787 } else {
788 m_state = Idle;
789
790 if (m_clearPreviews) {
791 // Only go through the list if there are items which might still have previews.
792 if (m_finishedItems.count() != m_model->count()) {
793 QHash<QByteArray, QVariant> data;
794 data.insert("iconPixmap", QPixmap());
795 data.insert("hoverSequencePixmaps", QVariant::fromValue(QVector<QPixmap>()));
796
797 disconnect(m_model, &KFileItemModel::itemsChanged,
798 this, &KFileItemModelRolesUpdater::slotItemsChanged);
799 for (int index = 0; index <= m_model->count(); ++index) {
800 if (m_model->data(index).contains("iconPixmap") ||
801 m_model->data(index).contains("hoverSequencePixmaps"))
802 {
803 m_model->setData(index, data);
804 }
805 }
806 connect(m_model, &KFileItemModel::itemsChanged,
807 this, &KFileItemModelRolesUpdater::slotItemsChanged);
808
809 }
810 m_clearPreviews = false;
811 }
812
813 if (!m_changedItems.isEmpty()) {
814 updateChangedItems();
815 }
816 }
817 }
818
819 void KFileItemModelRolesUpdater::resolveRecentlyChangedItems()
820 {
821 m_changedItems += m_recentlyChangedItems;
822 m_recentlyChangedItems.clear();
823 updateChangedItems();
824 }
825
826 void KFileItemModelRolesUpdater::applyChangedBalooRoles(const QString& file)
827 {
828 #ifdef HAVE_BALOO
829 const KFileItem item = m_model->fileItem(QUrl::fromLocalFile(file));
830
831 if (item.isNull()) {
832 // itemUrl is not in the model anymore, probably because
833 // the corresponding file has been deleted in the meantime.
834 return;
835 }
836 applyChangedBalooRolesForItem(item);
837 #else
838 Q_UNUSED(file)
839 #endif
840 }
841
842 void KFileItemModelRolesUpdater::applyChangedBalooRolesForItem(const KFileItem &item)
843 {
844 #ifdef HAVE_BALOO
845 Baloo::File file(item.localPath());
846 file.load();
847
848 const KBalooRolesProvider& rolesProvider = KBalooRolesProvider::instance();
849 QHash<QByteArray, QVariant> data;
850
851 const auto roles = rolesProvider.roles();
852 for (const QByteArray& role : roles) {
853 // Overwrite all the role values with an empty QVariant, because the roles
854 // provider doesn't overwrite it when the property value list is empty.
855 // See bug 322348
856 data.insert(role, QVariant());
857 }
858
859 QHashIterator<QByteArray, QVariant> it(rolesProvider.roleValues(file, m_roles));
860 while (it.hasNext()) {
861 it.next();
862 data.insert(it.key(), it.value());
863 }
864
865 disconnect(m_model, &KFileItemModel::itemsChanged,
866 this, &KFileItemModelRolesUpdater::slotItemsChanged);
867 const int index = m_model->index(item);
868 m_model->setData(index, data);
869 connect(m_model, &KFileItemModel::itemsChanged,
870 this, &KFileItemModelRolesUpdater::slotItemsChanged);
871 #else
872 #ifndef Q_CC_MSVC
873 Q_UNUSED(item)
874 #endif
875 #endif
876 }
877
878 void KFileItemModelRolesUpdater::slotDirectoryContentsCountReceived(const QString& path, int count, long size)
879 {
880 const bool getSizeRole = m_roles.contains("size");
881 const bool getIsExpandableRole = m_roles.contains("isExpandable");
882
883 if (getSizeRole || getIsExpandableRole) {
884 const int index = m_model->index(QUrl::fromLocalFile(path));
885 if (index >= 0) {
886 QHash<QByteArray, QVariant> data;
887
888 if (getSizeRole) {
889 data.insert("count", count);
890 data.insert("size", QVariant::fromValue(size));
891 }
892 if (getIsExpandableRole) {
893 data.insert("isExpandable", count > 0);
894 }
895
896 disconnect(m_model, &KFileItemModel::itemsChanged,
897 this, &KFileItemModelRolesUpdater::slotItemsChanged);
898 m_model->setData(index, data);
899 connect(m_model, &KFileItemModel::itemsChanged,
900 this, &KFileItemModelRolesUpdater::slotItemsChanged);
901 }
902 }
903 }
904
905 void KFileItemModelRolesUpdater::startUpdating()
906 {
907 if (m_state == Paused) {
908 return;
909 }
910
911 if (m_finishedItems.count() == m_model->count()) {
912 // All roles have been resolved already.
913 m_state = Idle;
914 return;
915 }
916
917 // Terminate all updates that are currently active.
918 killPreviewJob();
919 m_pendingIndexes.clear();
920
921 QElapsedTimer timer;
922 timer.start();
923
924 // Determine the icons for the visible items synchronously.
925 updateVisibleIcons();
926
927 // A detailed update of the items in and near the visible area
928 // only makes sense if sorting is finished.
929 if (m_state == ResolvingSortRole) {
930 return;
931 }
932
933 // Start the preview job or the asynchronous resolving of all roles.
934 QList<int> indexes = indexesToResolve();
935
936 if (m_previewShown) {
937 m_pendingPreviewItems.clear();
938 m_pendingPreviewItems.reserve(indexes.count());
939
940 for (int index : qAsConst(indexes)) {
941 const KFileItem item = m_model->fileItem(index);
942 if (!m_finishedItems.contains(item)) {
943 m_pendingPreviewItems.append(item);
944 }
945 }
946
947 startPreviewJob();
948 } else {
949 m_pendingIndexes = indexes;
950 // Trigger the asynchronous resolving of all roles.
951 m_state = ResolvingAllRoles;
952 QTimer::singleShot(0, this, &KFileItemModelRolesUpdater::resolveNextPendingRoles);
953 }
954 }
955
956 void KFileItemModelRolesUpdater::updateVisibleIcons()
957 {
958 int lastVisibleIndex = m_lastVisibleIndex;
959 if (lastVisibleIndex <= 0) {
960 // Guess a reasonable value for the last visible index if the view
961 // has not told us about the real value yet.
962 lastVisibleIndex = qMin(m_firstVisibleIndex + m_maximumVisibleItems, m_model->count() - 1);
963 if (lastVisibleIndex <= 0) {
964 lastVisibleIndex = qMin(200, m_model->count() - 1);
965 }
966 }
967
968 QElapsedTimer timer;
969 timer.start();
970
971 // Try to determine the final icons for all visible items.
972 int index;
973 for (index = m_firstVisibleIndex; index <= lastVisibleIndex && timer.elapsed() < MaxBlockTimeout; ++index) {
974 applyResolvedRoles(index, ResolveFast);
975 }
976
977 // KFileItemListView::initializeItemListWidget(KItemListWidget*) will load
978 // preliminary icons (i.e., without mime type determination) for the
979 // remaining items.
980 }
981
982 void KFileItemModelRolesUpdater::startPreviewJob()
983 {
984 m_state = PreviewJobRunning;
985
986 if (m_pendingPreviewItems.isEmpty()) {
987 QTimer::singleShot(0, this, &KFileItemModelRolesUpdater::slotPreviewJobFinished);
988 return;
989 }
990
991 // PreviewJob internally caches items always with the size of
992 // 128 x 128 pixels or 256 x 256 pixels. A (slow) downscaling is done
993 // by PreviewJob if a smaller size is requested. For images KFileItemModelRolesUpdater must
994 // do a downscaling anyhow because of the frame, so in this case only the provided
995 // cache sizes are requested.
996 const QSize cacheSize = (m_iconSize.width() > 128) || (m_iconSize.height() > 128)
997 ? QSize(256, 256) : QSize(128, 128);
998
999 // KIO::filePreview() will request the MIME-type of all passed items, which (in the
1000 // worst case) might block the application for several seconds. To prevent such
1001 // a blocking, we only pass items with known mime type to the preview job.
1002 const int count = m_pendingPreviewItems.count();
1003 KFileItemList itemSubSet;
1004 itemSubSet.reserve(count);
1005
1006 if (m_pendingPreviewItems.first().isMimeTypeKnown()) {
1007 // Some mime types are known already, probably because they were
1008 // determined when loading the icons for the visible items. Start
1009 // a preview job for all items at the beginning of the list which
1010 // have a known mime type.
1011 do {
1012 itemSubSet.append(m_pendingPreviewItems.takeFirst());
1013 } while (!m_pendingPreviewItems.isEmpty() && m_pendingPreviewItems.first().isMimeTypeKnown());
1014 } else {
1015 // Determine mime types for MaxBlockTimeout ms, and start a preview
1016 // job for the corresponding items.
1017 QElapsedTimer timer;
1018 timer.start();
1019
1020 do {
1021 const KFileItem item = m_pendingPreviewItems.takeFirst();
1022 item.determineMimeType();
1023 itemSubSet.append(item);
1024 } while (!m_pendingPreviewItems.isEmpty() && timer.elapsed() < MaxBlockTimeout);
1025 }
1026
1027 KIO::PreviewJob* job = new KIO::PreviewJob(itemSubSet, cacheSize, &m_enabledPlugins);
1028
1029 job->setIgnoreMaximumSize(itemSubSet.first().isLocalFile() && !itemSubSet.first().isSlow() && m_localFileSizePreviewLimit <= 0);
1030 if (job->uiDelegate()) {
1031 KJobWidgets::setWindow(job, qApp->activeWindow());
1032 }
1033
1034 connect(job, &KIO::PreviewJob::gotPreview,
1035 this, &KFileItemModelRolesUpdater::slotGotPreview);
1036 connect(job, &KIO::PreviewJob::failed,
1037 this, &KFileItemModelRolesUpdater::slotPreviewFailed);
1038 connect(job, &KIO::PreviewJob::finished,
1039 this, &KFileItemModelRolesUpdater::slotPreviewJobFinished);
1040
1041 m_previewJob = job;
1042 }
1043
1044 QPixmap KFileItemModelRolesUpdater::transformPreviewPixmap(const QPixmap& pixmap)
1045 {
1046 QPixmap scaledPixmap = pixmap;
1047
1048 if (!pixmap.hasAlpha() && !pixmap.isNull()
1049 && m_iconSize.width() > KIconLoader::SizeSmallMedium
1050 && m_iconSize.height() > KIconLoader::SizeSmallMedium) {
1051 if (m_enlargeSmallPreviews) {
1052 KPixmapModifier::applyFrame(scaledPixmap, m_iconSize);
1053 } else {
1054 // Assure that small previews don't get enlarged. Instead they
1055 // should be shown centered within the frame.
1056 const QSize contentSize = KPixmapModifier::sizeInsideFrame(m_iconSize);
1057 const bool enlargingRequired = scaledPixmap.width() < contentSize.width() &&
1058 scaledPixmap.height() < contentSize.height();
1059 if (enlargingRequired) {
1060 QSize frameSize = scaledPixmap.size() / scaledPixmap.devicePixelRatio();
1061 frameSize.scale(m_iconSize, Qt::KeepAspectRatio);
1062
1063 QPixmap largeFrame(frameSize);
1064 largeFrame.fill(Qt::transparent);
1065
1066 KPixmapModifier::applyFrame(largeFrame, frameSize);
1067
1068 QPainter painter(&largeFrame);
1069 painter.drawPixmap((largeFrame.width() - scaledPixmap.width() / scaledPixmap.devicePixelRatio()) / 2,
1070 (largeFrame.height() - scaledPixmap.height() / scaledPixmap.devicePixelRatio()) / 2,
1071 scaledPixmap);
1072 scaledPixmap = largeFrame;
1073 } else {
1074 // The image must be shrunk as it is too large to fit into
1075 // the available icon size
1076 KPixmapModifier::applyFrame(scaledPixmap, m_iconSize);
1077 }
1078 }
1079 } else if (!pixmap.isNull()) {
1080 KPixmapModifier::scale(scaledPixmap, m_iconSize * qApp->devicePixelRatio());
1081 scaledPixmap.setDevicePixelRatio(qApp->devicePixelRatio());
1082 }
1083
1084 return scaledPixmap;
1085 }
1086
1087 void KFileItemModelRolesUpdater::loadNextHoverSequencePreview()
1088 {
1089 if (m_hoverSequenceItem.isNull() || m_hoverSequencePreviewJob) {
1090 return;
1091 }
1092
1093 const int index = m_model->index(m_hoverSequenceItem);
1094 if (index < 0) {
1095 return;
1096 }
1097
1098 // We generate the next few sequence indices in advance (buffering)
1099 const int maxSeqIdx = m_hoverSequenceIndex+5;
1100
1101 QHash<QByteArray, QVariant> data = m_model->data(index);
1102
1103 if (!data.contains("hoverSequencePixmaps")) {
1104 // The pixmap at index 0 isn't used ("iconPixmap" will be used instead)
1105 data.insert("hoverSequencePixmaps", QVariant::fromValue(QVector<QPixmap>() << QPixmap()));
1106 m_model->setData(index, data);
1107 }
1108
1109 const QVector<QPixmap> pixmaps = data["hoverSequencePixmaps"].value<QVector<QPixmap>>();
1110
1111 const int loadSeqIdx = pixmaps.size();
1112
1113 float wap = -1.0f;
1114 if (data.contains("hoverSequenceWraparoundPoint")) {
1115 wap = data["hoverSequenceWraparoundPoint"].toFloat();
1116 }
1117 if (wap >= 1.0f && loadSeqIdx >= static_cast<int>(wap)) {
1118 // Reached the wraparound point -> no more previews to load.
1119 return;
1120 }
1121
1122 if (loadSeqIdx > maxSeqIdx) {
1123 // Wait until setHoverSequenceState() is called with a higher sequence index.
1124 return;
1125 }
1126
1127 // PreviewJob internally caches items always with the size of
1128 // 128 x 128 pixels or 256 x 256 pixels. A (slow) downscaling is done
1129 // by PreviewJob if a smaller size is requested. For images KFileItemModelRolesUpdater must
1130 // do a downscaling anyhow because of the frame, so in this case only the provided
1131 // cache sizes are requested.
1132 const QSize cacheSize = (m_iconSize.width() > 128) || (m_iconSize.height() > 128)
1133 ? QSize(256, 256) : QSize(128, 128);
1134
1135 KIO::PreviewJob* job = new KIO::PreviewJob({m_hoverSequenceItem}, cacheSize, &m_enabledPlugins);
1136
1137 job->setSequenceIndex(loadSeqIdx);
1138 job->setIgnoreMaximumSize(m_hoverSequenceItem.isLocalFile() && !m_hoverSequenceItem.isSlow() && m_localFileSizePreviewLimit <= 0);
1139 if (job->uiDelegate()) {
1140 KJobWidgets::setWindow(job, qApp->activeWindow());
1141 }
1142
1143 connect(job, &KIO::PreviewJob::gotPreview,
1144 this, &KFileItemModelRolesUpdater::slotHoverSequenceGotPreview);
1145 connect(job, &KIO::PreviewJob::failed,
1146 this, &KFileItemModelRolesUpdater::slotHoverSequencePreviewFailed);
1147 connect(job, &KIO::PreviewJob::finished,
1148 this, &KFileItemModelRolesUpdater::slotHoverSequencePreviewJobFinished);
1149
1150 m_hoverSequencePreviewJob = job;
1151 }
1152
1153 void KFileItemModelRolesUpdater::killHoverSequencePreviewJob()
1154 {
1155 if (m_hoverSequencePreviewJob) {
1156 disconnect(m_hoverSequencePreviewJob, &KIO::PreviewJob::gotPreview,
1157 this, &KFileItemModelRolesUpdater::slotHoverSequenceGotPreview);
1158 disconnect(m_hoverSequencePreviewJob, &KIO::PreviewJob::failed,
1159 this, &KFileItemModelRolesUpdater::slotHoverSequencePreviewFailed);
1160 disconnect(m_hoverSequencePreviewJob, &KIO::PreviewJob::finished,
1161 this, &KFileItemModelRolesUpdater::slotHoverSequencePreviewJobFinished);
1162 m_hoverSequencePreviewJob->kill();
1163 m_hoverSequencePreviewJob = nullptr;
1164 }
1165 }
1166
1167 void KFileItemModelRolesUpdater::updateChangedItems()
1168 {
1169 if (m_state == Paused) {
1170 return;
1171 }
1172
1173 if (m_changedItems.isEmpty()) {
1174 return;
1175 }
1176
1177 m_finishedItems -= m_changedItems;
1178
1179 if (m_resolvableRoles.contains(m_model->sortRole())) {
1180 m_pendingSortRoleItems += m_changedItems;
1181
1182 if (m_state != ResolvingSortRole) {
1183 // Stop the preview job if necessary, and trigger the
1184 // asynchronous determination of the sort role.
1185 killPreviewJob();
1186 m_state = ResolvingSortRole;
1187 QTimer::singleShot(0, this, &KFileItemModelRolesUpdater::resolveNextSortRole);
1188 }
1189
1190 return;
1191 }
1192
1193 QList<int> visibleChangedIndexes;
1194 QList<int> invisibleChangedIndexes;
1195 visibleChangedIndexes.reserve(m_changedItems.size());
1196 invisibleChangedIndexes.reserve(m_changedItems.size());
1197
1198 auto changedItemsIt = m_changedItems.begin();
1199 while (changedItemsIt != m_changedItems.end()) {
1200 const auto& item = *changedItemsIt;
1201 const int index = m_model->index(item);
1202
1203 if (index < 0) {
1204 changedItemsIt = m_changedItems.erase(changedItemsIt);
1205 continue;
1206 }
1207 ++changedItemsIt;
1208
1209 if (index >= m_firstVisibleIndex && index <= m_lastVisibleIndex) {
1210 visibleChangedIndexes.append(index);
1211 } else {
1212 invisibleChangedIndexes.append(index);
1213 }
1214 }
1215
1216 std::sort(visibleChangedIndexes.begin(), visibleChangedIndexes.end());
1217
1218 if (m_previewShown) {
1219 for (int index : qAsConst(visibleChangedIndexes)) {
1220 m_pendingPreviewItems.append(m_model->fileItem(index));
1221 }
1222
1223 for (int index : qAsConst(invisibleChangedIndexes)) {
1224 m_pendingPreviewItems.append(m_model->fileItem(index));
1225 }
1226
1227 if (!m_previewJob) {
1228 startPreviewJob();
1229 }
1230 } else {
1231 const bool resolvingInProgress = !m_pendingIndexes.isEmpty();
1232 m_pendingIndexes = visibleChangedIndexes + m_pendingIndexes + invisibleChangedIndexes;
1233 if (!resolvingInProgress) {
1234 // Trigger the asynchronous resolving of the changed roles.
1235 m_state = ResolvingAllRoles;
1236 QTimer::singleShot(0, this, &KFileItemModelRolesUpdater::resolveNextPendingRoles);
1237 }
1238 }
1239 }
1240
1241 void KFileItemModelRolesUpdater::applySortRole(int index)
1242 {
1243 QHash<QByteArray, QVariant> data;
1244 const KFileItem item = m_model->fileItem(index);
1245
1246 if (m_model->sortRole() == "type") {
1247 if (!item.isMimeTypeKnown()) {
1248 item.determineMimeType();
1249 }
1250
1251 data.insert("type", item.mimeComment());
1252 } else if (m_model->sortRole() == "size" && item.isLocalFile() && item.isDir()) {
1253 const QString path = item.localPath();
1254 if (m_scanDirectories) {
1255 m_directoryContentsCounter->scanDirectory(path);
1256 }
1257 } else {
1258 // Probably the sort role is a baloo role - just determine all roles.
1259 data = rolesData(item);
1260 }
1261
1262 disconnect(m_model, &KFileItemModel::itemsChanged,
1263 this, &KFileItemModelRolesUpdater::slotItemsChanged);
1264 m_model->setData(index, data);
1265 connect(m_model, &KFileItemModel::itemsChanged,
1266 this, &KFileItemModelRolesUpdater::slotItemsChanged);
1267 }
1268
1269 void KFileItemModelRolesUpdater::applySortProgressToModel()
1270 {
1271 // Inform the model about the progress of the resolved items,
1272 // so that it can give an indication when the sorting has been finished.
1273 const int resolvedCount = m_model->count() - m_pendingSortRoleItems.count();
1274 m_model->emitSortProgress(resolvedCount);
1275 }
1276
1277 bool KFileItemModelRolesUpdater::applyResolvedRoles(int index, ResolveHint hint)
1278 {
1279 const KFileItem item = m_model->fileItem(index);
1280 const bool resolveAll = (hint == ResolveAll);
1281
1282 bool iconChanged = false;
1283 if (!item.isMimeTypeKnown() || !item.isFinalIconKnown()) {
1284 item.determineMimeType();
1285 iconChanged = true;
1286 } else if (!m_model->data(index).contains("iconName")) {
1287 iconChanged = true;
1288 }
1289
1290 if (iconChanged || resolveAll || m_clearPreviews) {
1291 if (index < 0) {
1292 return false;
1293 }
1294
1295 QHash<QByteArray, QVariant> data;
1296 if (resolveAll) {
1297 data = rolesData(item);
1298 }
1299
1300 if (!item.iconName().isEmpty()) {
1301 data.insert("iconName", item.iconName());
1302 }
1303
1304 if (m_clearPreviews) {
1305 data.insert("iconPixmap", QPixmap());
1306 data.insert("hoverSequencePixmaps", QVariant::fromValue(QVector<QPixmap>()));
1307 }
1308
1309 disconnect(m_model, &KFileItemModel::itemsChanged,
1310 this, &KFileItemModelRolesUpdater::slotItemsChanged);
1311 m_model->setData(index, data);
1312 connect(m_model, &KFileItemModel::itemsChanged,
1313 this, &KFileItemModelRolesUpdater::slotItemsChanged);
1314 return true;
1315 }
1316
1317 return false;
1318 }
1319
1320 QHash<QByteArray, QVariant> KFileItemModelRolesUpdater::rolesData(const KFileItem& item)
1321 {
1322 QHash<QByteArray, QVariant> data;
1323
1324 const bool getSizeRole = m_roles.contains("size");
1325 const bool getIsExpandableRole = m_roles.contains("isExpandable");
1326
1327 if ((getSizeRole || getIsExpandableRole) && item.isDir()) {
1328 if (item.isLocalFile()) {
1329 // Tell m_directoryContentsCounter that we want to count the items
1330 // inside the directory. The result will be received in slotDirectoryContentsCountReceived.
1331 if (m_scanDirectories) {
1332 const QString path = item.localPath();
1333 m_directoryContentsCounter->scanDirectory(path);
1334 }
1335 } else if (getSizeRole) {
1336 data.insert("size", -1); // -1 indicates an unknown number of items
1337 }
1338 }
1339
1340 if (m_roles.contains("type")) {
1341 data.insert("type", item.mimeComment());
1342 }
1343
1344 QStringList overlays = item.overlays();
1345 for (KOverlayIconPlugin *it : qAsConst(m_overlayIconsPlugin)) {
1346 overlays.append(it->getOverlays(item.url()));
1347 }
1348 data.insert("iconOverlays", overlays);
1349
1350 #ifdef HAVE_BALOO
1351 if (m_balooFileMonitor) {
1352 m_balooFileMonitor->addFile(item.localPath());
1353 applyChangedBalooRolesForItem(item);
1354 }
1355 #endif
1356 return data;
1357 }
1358
1359 void KFileItemModelRolesUpdater::slotOverlaysChanged(const QUrl& url, const QStringList &)
1360 {
1361 const KFileItem item = m_model->fileItem(url);
1362 if (item.isNull()) {
1363 return;
1364 }
1365 const int index = m_model->index(item);
1366 QHash<QByteArray, QVariant> data = m_model->data(index);
1367 QStringList overlays = item.overlays();
1368 for (KOverlayIconPlugin *it : qAsConst(m_overlayIconsPlugin)) {
1369 overlays.append(it->getOverlays(url));
1370 }
1371 data.insert("iconOverlays", overlays);
1372 m_model->setData(index, data);
1373 }
1374
1375 void KFileItemModelRolesUpdater::updateAllPreviews()
1376 {
1377 if (m_state == Paused) {
1378 m_previewChangedDuringPausing = true;
1379 } else {
1380 m_finishedItems.clear();
1381 startUpdating();
1382 }
1383 }
1384
1385 void KFileItemModelRolesUpdater::killPreviewJob()
1386 {
1387 if (m_previewJob) {
1388 disconnect(m_previewJob, &KIO::PreviewJob::gotPreview,
1389 this, &KFileItemModelRolesUpdater::slotGotPreview);
1390 disconnect(m_previewJob, &KIO::PreviewJob::failed,
1391 this, &KFileItemModelRolesUpdater::slotPreviewFailed);
1392 disconnect(m_previewJob, &KIO::PreviewJob::finished,
1393 this, &KFileItemModelRolesUpdater::slotPreviewJobFinished);
1394 m_previewJob->kill();
1395 m_previewJob = nullptr;
1396 m_pendingPreviewItems.clear();
1397 }
1398 }
1399
1400 QList<int> KFileItemModelRolesUpdater::indexesToResolve() const
1401 {
1402 const int count = m_model->count();
1403
1404 QList<int> result;
1405 result.reserve(qMin(count, (m_lastVisibleIndex - m_firstVisibleIndex + 1) +
1406 ResolveAllItemsLimit +
1407 (2 * m_maximumVisibleItems)));
1408
1409 // Add visible items.
1410 // Resolve files first, their previews are quicker.
1411 QList<int> visibleDirs;
1412 for (int i = m_firstVisibleIndex; i <= m_lastVisibleIndex; ++i) {
1413 const KFileItem item = m_model->fileItem(i);
1414 if (item.isDir()) {
1415 visibleDirs.append(i);
1416 } else {
1417 result.append(i);
1418 }
1419 }
1420
1421 result.append(visibleDirs);
1422
1423 // We need a reasonable upper limit for number of items to resolve after
1424 // and before the visible range. m_maximumVisibleItems can be quite large
1425 // when using Compact View.
1426 const int readAheadItems = qMin(ReadAheadPages * m_maximumVisibleItems, ResolveAllItemsLimit / 2);
1427
1428 // Add items after the visible range.
1429 const int endExtendedVisibleRange = qMin(m_lastVisibleIndex + readAheadItems, count - 1);
1430 for (int i = m_lastVisibleIndex + 1; i <= endExtendedVisibleRange; ++i) {
1431 result.append(i);
1432 }
1433
1434 // Add items before the visible range in reverse order.
1435 const int beginExtendedVisibleRange = qMax(0, m_firstVisibleIndex - readAheadItems);
1436 for (int i = m_firstVisibleIndex - 1; i >= beginExtendedVisibleRange; --i) {
1437 result.append(i);
1438 }
1439
1440 // Add items on the last page.
1441 const int beginLastPage = qMax(endExtendedVisibleRange + 1, count - m_maximumVisibleItems);
1442 for (int i = beginLastPage; i < count; ++i) {
1443 result.append(i);
1444 }
1445
1446 // Add items on the first page.
1447 const int endFirstPage = qMin(beginExtendedVisibleRange, m_maximumVisibleItems);
1448 for (int i = 0; i < endFirstPage; ++i) {
1449 result.append(i);
1450 }
1451
1452 // Continue adding items until ResolveAllItemsLimit is reached.
1453 int remainingItems = ResolveAllItemsLimit - result.count();
1454
1455 for (int i = endExtendedVisibleRange + 1; i < beginLastPage && remainingItems > 0; ++i) {
1456 result.append(i);
1457 --remainingItems;
1458 }
1459
1460 for (int i = beginExtendedVisibleRange - 1; i >= endFirstPage && remainingItems > 0; --i) {
1461 result.append(i);
1462 --remainingItems;
1463 }
1464
1465 return result;
1466 }
1467
1468 void KFileItemModelRolesUpdater::trimHoverSequenceLoadedItems()
1469 {
1470 static const size_t maxLoadedItems = 20;
1471
1472 size_t loadedItems = m_hoverSequenceLoadedItems.size();
1473 while (loadedItems > maxLoadedItems) {
1474 const KFileItem item = m_hoverSequenceLoadedItems.front();
1475
1476 m_hoverSequenceLoadedItems.pop_front();
1477 loadedItems--;
1478
1479 const int index = m_model->index(item);
1480 if (index >= 0) {
1481 QHash<QByteArray, QVariant> data = m_model->data(index);
1482 data["hoverSequencePixmaps"] = QVariant::fromValue(QVector<QPixmap>() << QPixmap());
1483 m_model->setData(index, data);
1484 }
1485 }
1486 }
1487