1 files changed, 76 insertions, 0 deletions
diff --git a/src/iconfetcher.h b/src/iconfetcher.h
new file mode 100644
index 00000000..030bfb79
--- /dev/null
+++ b/src/iconfetcher.h
@@ -0,0 +1,76 @@
+#ifndef MODORGANIZER_ICONFETCHER_INCLUDED
+#define MODORGANIZER_ICONFETCHER_INCLUDED
+
+#include <QFileIconProvider>
+#include <mutex>
+
+class IconFetcher
+{
+public:
+  IconFetcher();
+  ~IconFetcher();
+
+  void stop();
+
+  QVariant icon(const QString& path) const;
+  QPixmap genericFileIcon() const;
+  QPixmap genericDirectoryIcon() const;
+
+private:
+  struct QuickCache
+  {
+    QPixmap file;
+    QPixmap directory;
+  };
+
+  struct Cache
+  {
+    std::map<QString, QPixmap, std::less<>> map;
+    std::mutex mapMutex;
+
+    std::set<QString> queue;
+    std::mutex queueMutex;
+  };
+
+  class Waiter
+  {
+  public:
+    void wait();
+    void wakeUp();
+
+  private:
+    mutable std::mutex m_wakeUpMutex;
+    std::condition_variable m_wakeUp;
+    bool m_queueAvailable = false;
+  };
+
+
+  const int m_iconSize;
+  QFileIconProvider m_provider;
+  std::thread m_thread;
+  std::atomic<bool> m_stop;
+
+  mutable QuickCache m_quickCache;
+  mutable Cache m_extensionCache;
+  mutable Cache m_fileCache;
+  mutable Waiter m_waiter;
+
+
+  bool hasOwnIcon(const QString& path) const;
+
+  template <class T>
+  QPixmap getPixmapIcon(T&& t) const
+  {
+    return m_provider.icon(t).pixmap({m_iconSize, m_iconSize});
+  }
+
+  void threadFun();
+
+  void checkCache(Cache& cache);
+  void queue(Cache& cache, QString path) const;
+
+  QVariant fileIcon(const QString& path) const;
+  QVariant extensionIcon(const QStringRef& ext) const;
+};
+
+#endif // MODORGANIZER_ICONFETCHER_INCLUDED