1 files changed, 13 insertions, 1 deletions
diff --git a/seriestreewidget.h b/seriestreewidget.h
index 4f23afb..38c111f 100644
--- a/seriestreewidget.h
+++ b/seriestreewidget.h
@@ -10,12 +10,14 @@
 
 #include <QWidget>
 #include <QTreeView>
+#include <QSortFilterProxyModel>
 
 class QLineEdit;
 class QPushButton;
 class QSortFilterProxyModel;
 class SeriesTreeModel;
 class SeriesTreeView;
+class SeriesTreeSortModel;
 
 class SeriesTreeWidget : public QWidget {
 	Q_OBJECT
@@ -35,7 +37,7 @@ class SeriesTreeWidget : public QWidget {
 		QLineEdit *mFilterEdit;
 		QPushButton *mFilter;
 		SeriesTreeView *mView;
-		QSortFilterProxyModel *mProxy;
+		SeriesTreeSortModel *mProxy;
 		SeriesTreeModel *mModel;
 };
 
@@ -48,4 +50,14 @@ class SeriesTreeView : public QTreeView {
 		virtual void contextMenuEvent(QContextMenuEvent *e);
 };
 
+class SeriesTreeSortModel : public QSortFilterProxyModel {
+	Q_OBJECT
+	public:
+		SeriesTreeSortModel(QObject *parent = 0);
+		~SeriesTreeSortModel() {}
+
+	protected:
+		virtual bool lessThan(const QModelIndex &left, const QModelIndex &right) const;
+};
+
 #endif