1 files changed, 44 insertions, 0 deletions
diff --git a/libs/installer_bsplugins/src/GUI/IconDelegate.h b/libs/installer_bsplugins/src/GUI/IconDelegate.h
new file mode 100644
index 0000000..057cd4e
--- /dev/null
+++ b/libs/installer_bsplugins/src/GUI/IconDelegate.h
@@ -0,0 +1,44 @@
+#ifndef GUI_ICONDELEGATE_H
+#define GUI_ICONDELEGATE_H
+
+#include <QAbstractItemView>
+#include <QAbstractProxyModel>
+#include <QStyledItemDelegate>
+#include <QTreeView>
+
+namespace GUI
+{
+
+class IconDelegate : public QStyledItemDelegate
+{
+  Q_OBJECT
+
+public:
+  explicit IconDelegate(QTreeView* view, int column = -1, int compactSize = 100);
+
+  void paint(QPainter* painter, const QStyleOptionViewItem& option,
+             const QModelIndex& index) const override;
+
+  QSize sizeHint(const QStyleOptionViewItem& option,
+                 const QModelIndex& index) const override;
+
+protected:
+  // check if icons should be compacted or not
+  //
+  [[nodiscard]] bool compact() const { return m_Compact; }
+
+  static void paintIcons(QPainter* painter, const QStyleOptionViewItem& option,
+                         const QModelIndex& index, const QList<QString>& icons);
+
+  [[nodiscard]] virtual QList<QString> getIcons(const QModelIndex& index) const = 0;
+  [[nodiscard]] virtual int getNumIcons(const QModelIndex& index) const         = 0;
+
+private:
+  int m_Column;
+  int m_CompactSize;
+  bool m_Compact;
+};
+
+}  // namespace GUI
+
+#endif  // GUI_ICONDELEGATE_H