1 files changed, 8 insertions, 2 deletions
diff --git a/searchdialog.h b/searchdialog.h
index 19d0d09..09ed259 100644
--- a/searchdialog.h
+++ b/searchdialog.h
@@ -10,6 +10,8 @@
 
 #include <QDialog>
 
+#include "smview.h"
+
 class QCheckBox;
 class QLineEdit;
 class QPushButton;
@@ -57,6 +59,10 @@ class ActorsAndMore : public QWidget {
         void doSearch();
         void doData(const QModelIndex &cur, const QModelIndex &prev);
         void dataDoubleClicked(const QModelIndex &index);
+        void collapseAllResult() { mResultView->collapseAll(); }
+        void expandAllResult() { mResultView->expandAll(); }
+        void collapseAllData() { mDataView->collapseAll(); }
+        void expandAllData() { mDataView->expandAll(); }
 
     private:
         void searchActor(const QString &actor);
@@ -68,8 +74,8 @@ class ActorsAndMore : public QWidget {
         QLineEdit *mSearch;
         QStandardItemModel *mResultModel;
         QStandardItemModel *mDataModel;
-        QTreeView *mResultView;
-        QTreeView *mDataView;
+        SmView *mResultView;
+        SmView *mDataView;
 };
 
 class SearchDialog : public QDialog {