1 files changed, 6 insertions, 1 deletions
diff --git a/archivecontroller.h b/archivecontroller.h
index b863f29..66068dc 100644
--- a/archivecontroller.h
+++ b/archivecontroller.h
@@ -10,7 +10,9 @@
 
 #include <QObject>
 #include <QItemSelection>
+#include <QStringList>
 
+class ArchiveView;
 class ArchiveTree;
 class ArchiveProxy;
 class ArchiveFiles;
@@ -24,13 +26,15 @@ class ArchiveController : public QObject {
     Q_OBJECT
     public:
         explicit ArchiveController(QObject *parent = 0);
-        void setArchiveView(ArchiveTree *atree, ArchiveProxy *aproxy);
+        void setArchiveView(ArchiveView *view);
+        void setArchiveTree(ArchiveTree *atree, ArchiveProxy *aproxy);
         void setArchiveFiles(ArchiveFiles *afiles, ArchiveFilesProxy *afilesproxy);
         void setModels(ArchiveModel *amodel, ArchiveFilesModel *afilesmodel);
         ArchiveFilesModel *archiveFilesModel() { return mArchiveFilesModel; }
         ArchiveFiles *archiveFiles() { return mArchiveFiles; }
         ArchiveModel *archiveTreeModel() { return mArchiveModel; }
         ArchiveTree *archiveTree() { return mArchiveTree; }
+        ArchiveView *archiveView() { return mArchiveView; }
         void init();
 
     public slots:
@@ -46,6 +50,7 @@ class ArchiveController : public QObject {
 
     private:
         QModelIndexList mapToSource(const QSortFilterProxyModel *proxy, const QModelIndexList idxs) const;
+        ArchiveView *mArchiveView;
         ArchiveTree *mArchiveTree;
         ArchiveProxy *mArchiveProxy;
         QItemSelectionModel *mArchiveSelection;