#5127: Add ResourceTreeViewToolbar to EntityClassChooser

codereader · Jan 3, 2021 · 8cf632f · 8cf632f
1 parent 3e34b69
commit 8cf632f
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/libs/wxutil/EntityClassChooser.cpp b/libs/wxutil/EntityClassChooser.cpp
@@ -1,7 +1,9 @@
 #include "EntityClassChooser.h"
+
 #include "dataview/TreeModel.h"
 #include "dataview/TreeViewItemStyle.h"
 #include "dataview/ThreadedResourceTreePopulator.h"
+#include "dataview/ResourceTreeViewToolbar.h"
 #include "dataview/VFSTreePopulator.h"
 
 #include "i18n.h"
@@ -307,7 +309,10 @@ void EntityClassChooser::setupTreeView()
     _treeView->AppendIconTextColumn(_("Classname"), _columns.iconAndName.getColumnIndex(),
         wxDATAVIEW_CELL_INERT, wxCOL_WIDTH_AUTOSIZE, wxALIGN_NOT, wxDATAVIEW_COL_SORTABLE);
 
+    auto* toolbar = new wxutil::ResourceTreeViewToolbar(parent, _treeView);
+
     parent->GetSizer()->Prepend(_treeView, 1, wxEXPAND | wxBOTTOM | wxRIGHT, 6);
+    parent->GetSizer()->Prepend(toolbar, 0, wxALIGN_LEFT | wxBOTTOM | wxLEFT | wxRIGHT, 6);
 }
 
 // Update the usage information