diff -r 338ebdc9b947 -r 14f2b1b15242 treeeditor.h
--- a/treeeditor.h	Fri Oct 02 14:31:03 2009 +0000
+++ b/treeeditor.h	Fri Nov 13 08:32:03 2009 +0000
@@ -2,7 +2,7 @@
 #define TREEEDITOR_H
 
 #include <QTreeView>
-#include <QSortFilterProxyModel>
+#include "mysortfilterproxymodel.h"
 
 class VymModel;
 
@@ -16,7 +16,8 @@
 public:
     TreeEditor(VymModel *m);
 	~TreeEditor();
-	QSortFilterProxyModel *getProxyModel();
+	MySortFilterProxyModel *getProxyModel();
+	QModelIndex getSelectedIndex();
 
 public slots:
 	void setSortFilter (QString f);
@@ -27,7 +28,7 @@
 
 private:
 	VymModel *model;
-	QSortFilterProxyModel *proxyModel;
+	MySortFilterProxyModel *proxyModel;
 };
 
 #endif