1 files changed, 6 insertions, 3 deletions
diff --git a/playerwidget.h b/playerwidget.h
index 5a6ff95..22cd2c5 100644
--- a/playerwidget.h
+++ b/playerwidget.h
@@ -3,7 +3,6 @@
 
 #include <QWidget>
 
-class QTreeView;
 class QStandardItemModel;
 class QStandardItem;
 class QLineEdit;
@@ -11,7 +10,9 @@ class QLabel;
 class QSlider;
 class QTextEdit;
 class QMediaPlayer;
+class QToolBar;
 class BeetPlayerProxy;
+class BeetView;
 
 class PlayerWidget : public QWidget {
     Q_OBJECT
@@ -27,19 +28,21 @@ class PlayerWidget : public QWidget {
 
     private:
         void setupGui();
+        void createActions();
         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;
+        BeetView *mView;
         QStandardItemModel *mViewModel;
         QStandardItemModel *mSearchModel;
         QLabel *mNowPlayingL;
         QSlider *mSlider;
         QTextEdit *mCurrentTE;
-        QTreeView *mPlayListView;
+        BeetView *mPlayListView;
         QStandardItemModel *mPlayListModel;
+        QToolBar *mToolBar;
 };
 
 #endif // PLAYERWIDGET_H