1 files changed, 7 insertions, 12 deletions
diff --git a/dbanalyzer.h b/dbanalyzer.h
index eec9f5a..cc88bbc 100644
--- a/dbanalyzer.h
+++ b/dbanalyzer.h
@@ -9,18 +9,13 @@
 #define DBANALYZER_H
 
 #include <QThread>
-#include <QVariant>
-#include <QList>
-#include <QHash>
 #include <QtWidgets/QDialog>
 #include <QSqlDatabase>
 #include <QMutex>
 #include <QModelIndex>
 
-class QSqlQuery;
 class QTabWidget;
-class QTreeView;
-class QPushButton;
+class SmTreeView;
 class SmTreeModel;
 class DbAnalyzer;
 class QLabel;
@@ -50,13 +45,13 @@ class DbAnalyzerDialog : public QDialog {
         void deleteMarks();
 
     private:
-        void populate(QTreeView *view ,SmTreeModel *model, const QList<QList<QVariant> > &data, const QHash<int, QVariant> &marks = QHash<int, QVariant>());
+        void populate(SmTreeView *view ,SmTreeModel *model, const QList<QList<QVariant> > &data, const QHash<int, QVariant> &marks = QHash<int, QVariant>());
         const QList<int> currentIds() const;
         QTabWidget *mTab;
-        QTreeView *mNoActorsV;
-        QTreeView *mNoCoversV;
-        QTreeView *mStrayActorsV;
-        QTreeView *mStrayGenresV;
+        SmTreeView *mNoActorsV;
+        SmTreeView *mNoCoversV;
+        SmTreeView *mStrayActorsV;
+        SmTreeView *mStrayGenresV;
         SmTreeModel *mNoActorsM;
         SmTreeModel *mNoCoversM;
         SmTreeModel *mStrayActorsM;
@@ -70,7 +65,7 @@ class DbAnalyzerDialog : public QDialog {
         QList<QString> mTotals;
         DbAnalyzer *mAnalyzer;
         int mMarkMode;
-        QTreeView *mCurrentView;
+        SmTreeView *mCurrentView;
         QStackedLayout *mButtonStack;
 };