1 files changed, 9 insertions, 16 deletions
diff --git a/newpicsdialog.cpp b/newpicsdialog.cpp
index 37a9cba..36fdfe1 100644
--- a/newpicsdialog.cpp
+++ b/newpicsdialog.cpp
@@ -5,36 +5,24 @@
   2 of the License, or (at your option) any later version.
 */
 
-#include <QFileInfo>
-#include <QSqlDatabase>
 #include <QSqlQuery>
-#include <QtWidgets/QLabel>
-#include <QtWidgets/QTreeView>
 #include <QSortFilterProxyModel>
 #include <QtWidgets/QPushButton>
 #include <QSettings>
-#include <QDir>
 #include <QtWidgets/QFileDialog>
-#include <QPersistentModelIndex>
-#include <QtWidgets/QHBoxLayout>
 #include <QtWidgets/QVBoxLayout>
-#include <QApplication>
 
 #include "newpicsdialog.h"
 #include "smtreeitem.h"
 #include "helper.h"
 #include "mappingtreewidget.h"
 #include "mappingtreemodel.h"
+#include "smglobals.h"
 
 NewPicsDialog::NewPicsDialog(QWidget *parent, Qt::WindowFlags f) : QDialog(parent, f) {
     setupDlg();
 }
 
-NewPicsDialog::NewPicsDialog(const QStringList &files, QWidget *parent, Qt::WindowFlags f) : QDialog(parent, f){
-    setupDlg();
-    addFiles(files);
-}
-
 void NewPicsDialog::setupDlg(){
     //setup database
     mDb = QSqlDatabase::database("treedb");
@@ -45,16 +33,14 @@ void NewPicsDialog::setupDlg(){
 
     //files widget
     mFilesWidget = new QWidget;
-    mFilesV = new QTreeView;
+    mFilesV = new SmTreeView;
     mFilesProxy = new QSortFilterProxyModel(this);
     mFilesModel = new NewPicFilesModel(QStringList() << tr("Filename") << tr("Size") << tr("Val.") << tr("Md5") << tr("Mime type") << tr("Full path"), this);
     mFilesProxy->setSourceModel(mFilesModel);
     mFilesV->setModel(mFilesProxy);
     mFilesV->setSortingEnabled(true);
-    mFilesV->setAlternatingRowColors(true);
     mFilesV->setColumnHidden(2, true);
     mFilesV->setAlternatingRowColors(true);
-    mFilesV->setPalette(qApp->palette());
     mAddFiles = new QPushButton(tr("Select..."));
     connect(mAddFiles, SIGNAL(clicked()), this, SLOT(selectFiles()));
     mRemoveFiles = new QPushButton(tr("Remove"));
@@ -118,6 +104,13 @@ void NewPicsDialog::addFiles(const QStringList &files){
     }
 }
 
+void NewPicsDialog::clearFiles(){
+    mFilesV->setSortingEnabled(false);
+    SmTreeItem *rootItem = new SmTreeItem(NewPicFilesModel::NumFields);
+    mFilesModel->setRoot(rootItem);
+    mFilesV->setSortingEnabled(true);
+}
+
 void NewPicsDialog::removeFiles(){
     QModelIndexList selected = mFilesV->selectionModel()->selectedRows();
     if(selected.isEmpty()){