#5532: Toggle stage visibility button

codereader · Mar 20, 2021 · 0c70138 · 0c70138
1 parent 4f4f22d
commit 0c70138
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 2 deletions.
diff --git a/radiant/ui/materials/MaterialEditor.cpp b/radiant/ui/materials/MaterialEditor.cpp
@@ -398,6 +398,7 @@ void MaterialEditor::setupMaterialStageView()
 
     getControl<wxButton>("MaterialEditorAddStageButton")->Bind(wxEVT_BUTTON, &MaterialEditor::_onAddStage, this);
     getControl<wxButton>("MaterialEditorRemoveStageButton")->Bind(wxEVT_BUTTON, &MaterialEditor::_onRemoveStage, this);
+    getControl<wxButton>("MaterialEditorToggleStageButton")->Bind(wxEVT_BUTTON, &MaterialEditor::_onToggleStage, this);
 }
 
 void MaterialEditor::setupStageFlag(const std::string& controlName, int flags)
@@ -836,7 +837,6 @@ void MaterialEditor::_onStageListValueChanged(wxDataViewEvent& ev)
 
     wxutil::TreeModel::Row row(ev.GetItem(), *_stageList);
     stage->setEnabled(row[STAGE_COLS().enabled].getBool());
-    int index = row[STAGE_COLS().index].getInteger();
 }
 
 void MaterialEditor::updateControlsFromMaterial()
@@ -1625,6 +1625,23 @@ void MaterialEditor::_onRemoveStage(wxCommandEvent& ev)
     updateStageListFromMaterial();
 }
 
+void MaterialEditor::_onToggleStage(wxCommandEvent& ev)
+{
+    auto item = _stageView->GetSelection();
+    if (!_material || !item.IsOk()) return;
+
+    auto row = wxutil::TreeModel::Row(item, *_stageList);
+    row[STAGE_COLS().enabled] = !row[STAGE_COLS().enabled].getBool();
+
+    row.SendItemChanged();
+
+    auto stage = getEditableStageForSelection();
+    if (!stage) return;
+
+    stage->setEnabled(row[STAGE_COLS().enabled].getBool());
+    onMaterialChanged();
+}
+
 void MaterialEditor::updateNameOfSelectedStage()
 {
     auto item = _stageView->GetSelection();
@@ -1633,7 +1650,6 @@ void MaterialEditor::updateNameOfSelectedStage()
     auto row = wxutil::TreeModel::Row(item, *_stageList);
 
     row[STAGE_COLS().name] = getNameForLayer(*getSelectedStage());
-
     row.SendItemChanged();
 }
 

diff --git a/radiant/ui/materials/MaterialEditor.h b/radiant/ui/materials/MaterialEditor.h
@@ -143,6 +143,7 @@ class MaterialEditor :
     void _onStageBlendTypeChanged(wxCommandEvent& ev);
     void _onAddStage(wxCommandEvent& ev);
     void _onRemoveStage(wxCommandEvent& ev);
+    void _onToggleStage(wxCommandEvent& ev);
 
     void onMaterialChanged();