1 files changed, 3 insertions, 1 deletions
diff --git a/collectionwidget.h b/collectionwidget.h
index 86135bb..c76582c 100644
--- a/collectionwidget.h
+++ b/collectionwidget.h
@@ -2,8 +2,8 @@
 #define COLLECTIONWIDGET_H
 
 #include <QWidget>
+#include <QTreeView>
 
-class QTreeView;
 class QStandardItemModel;
 class QSortFilterProxyModel;
 
@@ -18,6 +18,8 @@ class CollectionWidget : public QWidget {
         QSortFilterProxyModel *proxy() { return mProxy; }
         void setHeaders(const QStringList headers) { mHeaders = headers; }
         const QStringList headers() const { return mHeaders; }
+        void disableSorting() { mView->setSortingEnabled(false); }
+        void enableSorting(int column = 0, Qt::SortOrder sortOrder = Qt::AscendingOrder);
 
     signals: