1 files changed, 6 insertions, 3 deletions
diff --git a/playerwidget.h b/playerwidget.h
index c9206cd..5a6ff95 100644
--- a/playerwidget.h
+++ b/playerwidget.h
@@ -5,7 +5,7 @@
 
 class QTreeView;
 class QStandardItemModel;
-class QSortFilterProxyModel;
+class QStandardItem;
 class QLineEdit;
 class QLabel;
 class QSlider;
@@ -16,7 +16,7 @@ class BeetPlayerProxy;
 class PlayerWidget : public QWidget {
     Q_OBJECT
     public:
-        enum ItemType { Artist, Album, Song };
+        enum ItemType { Artist, Album, Song, Genre };
         enum CustomRoles { TypeRole = Qt::UserRole + 1, IdRole = Qt::UserRole + 2, FullPathRole = Qt::UserRole + 3, GenreRole = Qt::UserRole + 4 };
         explicit PlayerWidget(QWidget *parent = 0);
 
@@ -27,11 +27,14 @@ class PlayerWidget : public QWidget {
 
     private:
         void setupGui();
+        void populateByArtist(QStandardItem *parent, const QString &filter);
+        void populateBySong(QStandardItem *parent, const QString &filter);
+        void populateByGenre(QStandardItem *parent, const QString &filter);
         QLineEdit *mFilter;
         QMediaPlayer *mPlayer;
         QTreeView *mView;
         QStandardItemModel *mViewModel;
-        BeetPlayerProxy *mViewProxy;
+        QStandardItemModel *mSearchModel;
         QLabel *mNowPlayingL;
         QSlider *mSlider;
         QTextEdit *mCurrentTE;