1 files changed, 4 insertions, 18 deletions
diff --git a/archivetreeview.cpp b/archivetreeview.cpp
index b2ba254..7587a6a 100644
--- a/archivetreeview.cpp
+++ b/archivetreeview.cpp
@@ -103,25 +103,11 @@ ArchiveTreeView::ArchiveTreeView(QWidget *parent) : QWidget(parent){
 }
 
 void ArchiveTreeView::setFileViewMode(int mode){
-	switch(mode){
-		case FilesTreeModel::Archived:
-		case FilesTreeModel::Local:
-			mSeriesWidget->setEnabled(false);
-			break;
-		case FilesTreeModel::Normal:
-			mSeriesWidget->setEnabled(true);
-			break;
-		default:
-			;
-	}
-	mFilesWidget->resetSize();
-	mFilesModel->setMode(mode);
-	if(mode == FilesTreeModel::Normal){
-		currentChanged(QItemSelection(), QItemSelection());
-	}
-	mFilesWidget->filesTree()->expandAll();
-	mFilesWidget->filesTree()->header()->resizeSections(QHeaderView::ResizeToContents);
+	SeriesTreeModel *seriesModel = static_cast<SeriesTreeModel*>(SmGlobals::instance()->model("SeriesModel"));
+	seriesModel->setRowFilter(mode);
+	mSeriesWidget->seriesProxy()->invalidate();
 	constructWindowTitle();
+	mSeriesWidget->expandItems();
 }
 
 void ArchiveTreeView::cleanDatabase(const QString &table){