1 files changed, 10 insertions, 3 deletions
diff --git a/archivetreeview.h b/archivetreeview.h
index 2f07d2f..613bdc6 100644
--- a/archivetreeview.h
+++ b/archivetreeview.h
@@ -9,10 +9,12 @@
 #define ARCHIVETREEVIEW_H
 
 #include <QWidget>
+#include <QList>
 
 class SmUberModel;
-class QTreeView;
 class SeriesTreeWidget;
+class FilesTreeWidget;
+class FilesTreeModel;
 
 class ArchiveTreeView : public QWidget
 {
@@ -20,12 +22,17 @@ class ArchiveTreeView : public QWidget
 	public:
 		explicit ArchiveTreeView(QWidget *parent = 0);
 		SeriesTreeWidget *seriesWidget() { return mSeriesWidget; }
+		FilesTreeWidget *filesWidget() { return mFilesWidget; }
+
+	private slots:
+		void currentChanged();
 
 	private:
 		//widgets
 		SeriesTreeWidget *mSeriesWidget;
-		QTreeView *mFilesView;
-
+		FilesTreeWidget *mFilesWidget;
+		QList<int> mSelectedPartIds;
+		FilesTreeModel *mFilesModel;
 };
 
 #endif