1 files changed, 3 insertions, 7 deletions
diff --git a/newpicsdialog.h b/newpicsdialog.h
index 148795a..87db4f3 100644
--- a/newpicsdialog.h
+++ b/newpicsdialog.h
@@ -10,17 +10,13 @@
 
 #include <QtWidgets/QDialog>
 #include <QSqlDatabase>
-#include <QtWidgets/QTreeView>
 
 #include "smtreemodel.h"
 
 class QTabWidget;
-class QSqlDatabase;
-class QSqlQuery;
-class QTreeView;
+class SmTreeView;
 class NewPicFilesModel;
 class QSortFilterProxyModel;
-class QPushButton;
 class MappingTreeWidget;
 class MappingEditWidget;
 
@@ -30,12 +26,12 @@ class NewPicsDialog : public QDialog {
     Q_OBJECT
     public:
         explicit NewPicsDialog(QWidget *parent = 0, Qt::WindowFlags f = 0);
-        explicit NewPicsDialog(const QStringList &files, QWidget *parent = 0, Qt::WindowFlags f = 0);
 
     public slots:
         void selectFiles();
         void removeFiles();
         void addFiles(const QStringList &files);
+        void clearFiles();
         void accept();
 
     private:
@@ -43,7 +39,7 @@ class NewPicsDialog : public QDialog {
         QTabWidget *mTab;
         QWidget *mFilesWidget;
         MappingEditWidget *mMappingEditWidget;
-        QTreeView *mFilesV;
+        SmTreeView *mFilesV;
         NewPicFilesModel *mFilesModel;
         QSortFilterProxyModel *mFilesProxy;
         QPushButton *mAddFiles;