1 files changed, 7 insertions, 2 deletions
diff --git a/delegates.h b/delegates.h
index 0db3bf9..d24095a 100644
--- a/delegates.h
+++ b/delegates.h
@@ -13,6 +13,8 @@
 #include <QStyleOptionViewItem>
 #include <QStyledItemDelegate>
 
+class ArchiveBrowserModelProxy;
+
 class FileNoDelegate : public QStyledItemDelegate {
     Q_OBJECT
     public:
@@ -35,8 +37,11 @@ class FileTypeDelegate : public QStyledItemDelegate {
 class GenreDelegate : public QStyledItemDelegate {
     Q_OBJECT
     public:
-    explicit GenreDelegate(QObject *parent = nullptr) : QStyledItemDelegate(parent) {}
-    virtual QString displayText(const QVariant &value, const QLocale &locale) const;
+    explicit GenreDelegate(QObject *parent = nullptr, ArchiveBrowserModelProxy *proxy = nullptr) : QStyledItemDelegate(parent), mProxy(proxy) {};
+        virtual QString displayText(const QVariant &value, const QLocale &locale) const;
+
+    private:
+        ArchiveBrowserModelProxy *mProxy;
 };
 
 class DvdNoDelegate : public QStyledItemDelegate {