1 files changed, 5 insertions, 7 deletions
diff --git a/filestreewidget.h b/filestreewidget.h
index 5068b91..6369300 100644
--- a/filestreewidget.h
+++ b/filestreewidget.h
@@ -9,11 +9,12 @@
 #define FILESTREEWIDGET_H
 
 #include <QtWidgets/QWidget>
-#include <QtWidgets/QTreeView>
 #include <QtWidgets/QDialog>
 #include <QPersistentModelIndex>
 #include <QSortFilterProxyModel>
 
+#include "smtreeview.h"
+
 typedef QList<QPersistentModelIndex> PersistenModelIndexList;
 
 class FilesTreeView;
@@ -64,17 +65,14 @@ class FilesTreeWidget : public QWidget {
 		qint64 mSelectedSize;
 };
 
-class FilesTreeView : public QTreeView {
+class FilesTreeView : public SmTreeView {
 	Q_OBJECT
 	public:
 		explicit FilesTreeView(QWidget *parent = 0);
-		virtual void setModel(QAbstractItemModel *model);
 
 	public slots:
-		void readSettings();
-		void readHeaderConfig();
-		void writeHeaderConfig();
-		void toggleHeader(QObject *action);
+        void readSettings();
+        void writeSettings();
 
 	protected:
 		virtual void contextMenuEvent(QContextMenuEvent *event);