1 files changed, 11 insertions, 2 deletions
diff --git a/seriestreewidget.h b/seriestreewidget.h
index fa84419..6e949b2 100644
--- a/seriestreewidget.h
+++ b/seriestreewidget.h
@@ -13,13 +13,18 @@
 
 class QLineEdit;
 class QPushButton;
-class QTreeView;
 class QSortFilterProxyModel;
+class SeriesTreeModel;
+class SeriesTreeView;
 
 class SeriesTreeWidget : public QWidget {
 	Q_OBJECT
 	public:
 		explicit SeriesTreeWidget(QWidget *parent = 0);
+		SeriesTreeView *seriesTree() { return mView; }
+
+	public slots:
+		void newSeries();
 
 	private slots:
 		void filter();
@@ -28,14 +33,18 @@ class SeriesTreeWidget : public QWidget {
 	private:
 		QLineEdit *mFilterEdit;
 		QPushButton *mFilter;
-		QTreeView *mView;
+		SeriesTreeView *mView;
 		QSortFilterProxyModel *mProxy;
+		SeriesTreeModel *mModel;
 };
 
 class SeriesTreeView : public QTreeView {
 	Q_OBJECT
 	public:
 		explicit SeriesTreeView(QWidget *parent = 0);
+
+	protected:
+		virtual void contextMenuEvent(QContextMenuEvent *e);
 };
 
 #endif