Expand the level when filtering

E.g., expand the artist to albums.

2 files changed, 14 insertions, 1 deletions

diff --git a/collectionwidget.cpp b/collectionwidget.cpp
index 96c46fc..127c6a6 100644
--- a/collectionwidget.cpp
+++ b/collectionwidget.cpp
@@ -28,7 +28,10 @@ CollectionWidget::CollectionWidget(QWidget *parent) : QWidget(parent){
     connect(clearFilter, &QAction::triggered, [this, proxy] { mFilter->clear(); proxy->setFilterRegExp("");} );
     filterTB->addAction(clearFilter);
     QAction *activateFilter = new QAction(QIcon(":/stomp.png"), tr("Filter!"), this);
-    connect(activateFilter, &QAction::triggered, [this, proxy] { proxy->setFilterRegExp(mFilter->text());} );
+    connect(activateFilter, &QAction::triggered, [this, proxy] {
+        proxy->setFilterRegExp(mFilter->text());
+        expandFirst(proxy);
+    } );
     connect(mFilter, &QLineEdit::returnPressed, activateFilter, &QAction::trigger);
     filterTB->addAction(activateFilter);
     filterGBL->addWidget(filterTB);
@@ -43,3 +46,12 @@ void CollectionWidget::enableSorting(int column, Qt::SortOrder sortOrder){
     view()->setSortingEnabled(true);
     view()->sortByColumn(column, sortOrder);
 }
+
+void CollectionWidget::expandFirst(QSortFilterProxyModel *proxy){
+    int row = 0;
+    QModelIndex idx = proxy->index(0, 0);
+    while(idx.isValid()){
+        mView->expand(idx);
+        idx = proxy->index(++row, 0);
+    }
+}
diff --git a/collectionwidget.h b/collectionwidget.h
index d464054..fe4a086 100644
--- a/collectionwidget.h
+++ b/collectionwidget.h
@@ -26,6 +26,7 @@ class CollectionWidget : public QWidget {
 
     public slots:
         virtual void populate() { return; }
+        virtual void expandFirst(QSortFilterProxyModel *proxy);
 
     private:
         BeetView *mView;