From 34d2a0d3a041535d2ed43e0f2118cbe245bc99c6 Mon Sep 17 00:00:00 2001
From: TranquillyUnpleasant
 <62061444+TranquillyUnpleasant@users.noreply.github.com>
Date: Tue, 7 Sep 2021 22:38:43 +0500
Subject: [PATCH] Hide all button for units (#5953)

* Hide all button for units

* Add bundle property
---
 core/assets/bundles/bundle.properties    |  1 +
 core/src/mindustry/editor/WaveGraph.java | 11 +++++++++++
 2 files changed, 12 insertions(+)

diff --git a/core/assets/bundles/bundle.properties b/core/assets/bundles/bundle.properties
index 4ae1258e7d..b16bdbdcfa 100644
--- a/core/assets/bundles/bundle.properties
+++ b/core/assets/bundles/bundle.properties
@@ -401,6 +401,7 @@ waves.sort.reverse = Reverse Sort
 waves.sort.begin = Begin
 waves.sort.health = Health
 waves.sort.type = Type
+waves.units.hide = Hide All
 
 #these are intentionally in lower case
 wavemode.counts = counts
diff --git a/core/src/mindustry/editor/WaveGraph.java b/core/src/mindustry/editor/WaveGraph.java
index 76bbc82d2b..dc7dc6f112 100644
--- a/core/src/mindustry/editor/WaveGraph.java
+++ b/core/src/mindustry/editor/WaveGraph.java
@@ -181,6 +181,17 @@ public class WaveGraph extends Table{
 
         colors.clear();
         colors.left();
+        colors.button("@waves.units.hide", Styles.fullTogglet, () -> {
+            if(hidden.size == usedCopy.size){
+                hidden.clear();
+            }else{
+                hidden.addAll(usedCopy);
+            }
+
+            used.clear();
+            used.addAll(usedCopy);
+            for(UnitType o : hidden) used.remove(o);
+        }).update(b -> b.setChecked(hidden.size == usedCopy.size)).height(32f).width(130f);
         colors.pane(t -> {
             t.left();
             for(UnitType type : used){