From 7cc94057a1e6a61ce38d08c1862e2af38f8d3a99 Mon Sep 17 00:00:00 2001
From: Anuken <arnukren@gmail.com>
Date: Tue, 28 Jul 2020 15:27:43 -0400
Subject: [PATCH] Display modes for waves

---
 core/assets/bundles/bundle.properties         |   6 +
 core/src/mindustry/content/UnitTypes.java     |   4 +-
 core/src/mindustry/editor/WaveGraph.java      | 114 ++++++++++++++----
 core/src/mindustry/editor/WaveInfoDialog.java |  70 +++++++----
 gradle.properties                             |   2 +-
 5 files changed, 146 insertions(+), 50 deletions(-)

diff --git a/core/assets/bundles/bundle.properties b/core/assets/bundles/bundle.properties
index 9ba28d903d..90cfe73dcb 100644
--- a/core/assets/bundles/bundle.properties
+++ b/core/assets/bundles/bundle.properties
@@ -340,6 +340,12 @@ waves.load = Load from Clipboard
 waves.invalid = Invalid waves in clipboard.
 waves.copied = Waves copied.
 waves.none = No enemies defined.\nNote that empty wave layouts will automatically be replaced with the default layout.
+
+#these are intentionally in lower case
+wavemode.counts = counts
+wavemode.totals = totals
+wavemode.health = health
+
 editor.default = [lightgray]<Default>
 details = Details...
 edit = Edit...
diff --git a/core/src/mindustry/content/UnitTypes.java b/core/src/mindustry/content/UnitTypes.java
index ac070898f2..414eddac83 100644
--- a/core/src/mindustry/content/UnitTypes.java
+++ b/core/src/mindustry/content/UnitTypes.java
@@ -520,7 +520,7 @@ public class UnitTypes implements ContentList{
             rotateSpeed = 3.5f;
             flying = true;
             lowAltitude = true;
-            health = 7000;
+            health = 9000;
             armor = 9f;
             engineOffset = 21;
             engineSize = 5.3f;
@@ -542,7 +542,7 @@ public class UnitTypes implements ContentList{
             rotateSpeed = 2.5f;
             flying = true;
             lowAltitude = true;
-            health = 13000;
+            health = 18000;
             engineOffset = 38;
             engineSize = 7.3f;
             hitsize = 58f;
diff --git a/core/src/mindustry/editor/WaveGraph.java b/core/src/mindustry/editor/WaveGraph.java
index 143a5839df..eb2ac64359 100644
--- a/core/src/mindustry/editor/WaveGraph.java
+++ b/core/src/mindustry/editor/WaveGraph.java
@@ -3,6 +3,7 @@ package mindustry.editor;
 import arc.graphics.*;
 import arc.graphics.g2d.*;
 import arc.math.*;
+import arc.scene.ui.*;
 import arc.scene.ui.layout.*;
 import arc.struct.*;
 import arc.util.*;
@@ -10,6 +11,7 @@ import arc.util.pooling.*;
 import mindustry.*;
 import mindustry.game.*;
 import mindustry.gen.*;
+import mindustry.graphics.*;
 import mindustry.type.*;
 import mindustry.ui.*;
 
@@ -17,9 +19,11 @@ public class WaveGraph extends Table{
     public Seq<SpawnGroup> groups = new Seq<>();
     public int from, to = 20;
 
+    private Mode mode = Mode.counts;
     private int[][] values;
     private OrderedSet<UnitType> used = new OrderedSet<>();
-    private int max;
+    private int max, maxTotal;
+    private float maxHealth;
     private Table colors;
     private ObjectSet<UnitType> hidden = new ObjectSet<>();
 
@@ -30,29 +34,64 @@ public class WaveGraph extends Table{
             Lines.stroke(Scl.scl(3f));
             Lines.precise(true);
 
-            float offsetX = Scl.scl(30f), offsetY = Scl.scl(20f);
-
-            float graphX = x + offsetX, graphY = y + offsetY, graphW = width - offsetX, graphH = height - offsetY;
-            float spacing = graphW / (values.length - 1);
-
-            for(UnitType type : used){
-                Draw.color(color(type));
-                Draw.alpha(parentAlpha);
-                Lines.beginLine();
-                for(int i = 0; i < values.length; i++){
-                    int val = values[i][type.id];
-
-                    float cx = graphX + i*spacing, cy = 2f + graphY + val * (graphH - 4f) / max;
-                    Lines.linePoint(cx, cy);
-                }
-                Lines.endLine();
-            }
-
             GlyphLayout lay = Pools.obtain(GlyphLayout.class, GlyphLayout::new);
             BitmapFont font = Fonts.outline;
 
             lay.setText(font, "1");
 
+            float fh = lay.height;
+            float offsetX = Scl.scl(30f), offsetY = Scl.scl(22f) + fh + Scl.scl(5f);
+
+            float graphX = x + offsetX, graphY = y + offsetY, graphW = width - offsetX, graphH = height - offsetY;
+            float spacing = graphW / (values.length - 1);
+
+            if(mode == Mode.counts){
+                for(UnitType type : used.orderedItems()){
+                    Draw.color(color(type));
+                    Draw.alpha(parentAlpha);
+
+                    Lines.beginLine();
+
+                    for(int i = 0; i < values.length; i++){
+                        int val = values[i][type.id];
+                        float cx = graphX + i*spacing, cy = 2f + graphY + val * (graphH - 4f) / max;
+                        Lines.linePoint(cx, cy);
+                    }
+
+                    Lines.endLine();
+                }
+            }else if(mode == Mode.totals){
+                Lines.beginLine();
+
+                Draw.color(Pal.accent);
+                for(int i = 0; i < values.length; i++){
+                    int sum = 0;
+                    for(UnitType type : used.orderedItems()){
+                        sum += values[i][type.id];
+                    }
+
+                    float cx = graphX + i*spacing, cy = 2f + graphY + sum * (graphH - 4f) / maxTotal;
+                    Lines.linePoint(cx, cy);
+                }
+
+                Lines.endLine();
+            }else if(mode == Mode.health){
+                Lines.beginLine();
+
+                Draw.color(Pal.health);
+                for(int i = 0; i < values.length; i++){
+                    float sum = 0;
+                    for(UnitType type : used.orderedItems()){
+                        sum += type.health * values[i][type.id];
+                    }
+
+                    float cx = graphX + i*spacing, cy = 2f + graphY + sum * (graphH - 4f) / maxHealth;
+                    Lines.linePoint(cx, cy);
+                }
+
+                Lines.endLine();
+            }
+
             //how many numbers can fit here
             float totalMarks = (height - offsetY - getMarginBottom() *2f - 1f) / (lay.height * 2);
 
@@ -68,13 +107,18 @@ public class WaveGraph extends Table{
                 font.draw("" + i, cx, cy + lay.height/2f - Scl.scl(3f), Align.right);
             }
 
-            float len = 4f;
+            float len = Scl.scl(4f);
+            font.setColor(Color.lightGray);
 
             for(int i = 0; i < values.length; i++){
-                float cy = y, cx = x + graphW / (values.length - 1) * i + offsetX;
+                float cy = y + fh, cx = x + graphW / (values.length - 1) * i + offsetX;
 
                 Lines.line(cx, cy, cx, cy + len);
+                if(i == values.length/2){
+                    font.draw("" + (i + from), cx, cy - 2f, Align.center);
+                }
             }
+            font.setColor(Color.white);
 
             Pools.free(lay);
 
@@ -85,15 +129,31 @@ public class WaveGraph extends Table{
         row();
 
         table(t -> colors = t).growX();
+
+        row();
+
+        table(t -> {
+            t.left();
+            ButtonGroup<Button> group = new ButtonGroup<>();
+
+            for(Mode m : Mode.all){
+                t.button("$wavemode." + m.name(), Styles.fullTogglet, () -> {
+                    mode = m;
+                }).group(group).height(32f).update(b -> b.setChecked(m == mode)).width(100f);
+            }
+        }).growX();
     }
 
     public void rebuild(){
         values = new int[to - from + 1][Vars.content.units().size];
         used.clear();
-        max = 1;
+        max = maxTotal = 1;
+        maxHealth = 1f;
 
         for(int i = from; i <= to; i++){
             int index = i - from;
+            float healthsum = 0f;
+            int sum = 0;
 
             for(SpawnGroup spawn : groups){
                 int spawned = spawn.getUnitsSpawned(i);
@@ -102,7 +162,11 @@ public class WaveGraph extends Table{
                     used.add(spawn.type);
                 }
                 max = Math.max(max, values[index][spawn.type.id]);
+                healthsum += spawned * spawn.type.health;
+                sum += spawned;
             }
+            maxTotal = Math.max(maxTotal, sum);
+            maxHealth = Math.max(maxHealth,healthsum);
         }
 
         ObjectSet<UnitType> usedCopy = new ObjectSet<>(used);
@@ -129,10 +193,14 @@ public class WaveGraph extends Table{
         for(UnitType type : hidden){
             used.remove(type);
         }
-
     }
 
     Color color(UnitType type){
         return Tmp.c1.fromHsv(type.id / (float)Vars.content.units().size * 360f, 0.7f, 1f);
     }
+
+    enum Mode{
+        counts, totals, health;
+        static Mode[] all = values();
+    }
 }
diff --git a/core/src/mindustry/editor/WaveInfoDialog.java b/core/src/mindustry/editor/WaveInfoDialog.java
index 028494fb2a..9ce81a7cfa 100644
--- a/core/src/mindustry/editor/WaveInfoDialog.java
+++ b/core/src/mindustry/editor/WaveInfoDialog.java
@@ -22,7 +22,7 @@ import static mindustry.Vars.*;
 import static mindustry.game.SpawnGroup.*;
 
 public class WaveInfoDialog extends BaseDialog{
-    private static final int displayed = 20;
+    private int displayed = 20;
     private Seq<SpawnGroup> groups = new Seq<>();
 
     private Table table;
@@ -47,28 +47,6 @@ public class WaveInfoDialog extends BaseDialog{
 
         onResize(this::setup);
         addCloseButton();
-        buttons.button("<<", () -> {
-        }).update(t -> {
-            if(t.getClickListener().isPressed()){
-                updateTimer += Time.delta;
-                if(updateTimer >= updatePeriod){
-                    start = Math.max(start - 1, 0);
-                    updateTimer = 0f;
-                    updateWaves();
-                }
-            }
-        });
-        buttons.button(">>", () -> {
-        }).update(t -> {
-            if(t.getClickListener().isPressed()){
-                updateTimer += Time.delta;
-                if(updateTimer >= updatePeriod){
-                    start++;
-                    updateTimer = 0f;
-                    updateWaves();
-                }
-            }
-        });
 
         buttons.button("$waves.edit", () -> {
             BaseDialog dialog = new BaseDialog("$waves.edit");
@@ -99,6 +77,50 @@ public class WaveInfoDialog extends BaseDialog{
             }));
             dialog.show();
         }).size(270f, 64f);
+
+        buttons.defaults().width(60f);
+
+        buttons.button("<", () -> {}).update(t -> {
+            if(t.getClickListener().isPressed()){
+                shift(-1);
+            }
+        });
+        buttons.button(">", () -> {}).update(t -> {
+            if(t.getClickListener().isPressed()){
+                shift(1);
+            }
+        });
+
+        buttons.button("-", () -> {}).update(t -> {
+            if(t.getClickListener().isPressed()){
+                view(-1);
+            }
+        });
+        buttons.button("+", () -> {}).update(t -> {
+            if(t.getClickListener().isPressed()){
+                view(1);
+            }
+        });
+    }
+
+    void view(int amount){
+        updateTimer += Time.delta;
+        if(updateTimer >= updatePeriod){
+            displayed += amount;
+            if(displayed < 5) displayed = 5;
+            updateTimer = 0f;
+            updateWaves();
+        }
+    }
+
+    void shift(int amount){
+        updateTimer += Time.delta;
+        if(updateTimer >= updatePeriod){
+            start += amount;
+            if(start < 0) start = 0;
+            updateTimer = 0f;
+            updateWaves();
+        }
     }
 
     void setup(){
@@ -120,7 +142,7 @@ public class WaveInfoDialog extends BaseDialog{
             setAlignment(Align.center, Align.center);
         }}).width(390f).growY();
 
-        cont.add(graph).grow();
+        cont.add(graph = new WaveGraph()).grow();
 
         buildGroups();
     }
diff --git a/gradle.properties b/gradle.properties
index 25e9dede97..5323a204b9 100644
--- a/gradle.properties
+++ b/gradle.properties
@@ -1,3 +1,3 @@
 org.gradle.daemon=true
 org.gradle.jvmargs=-Xms256m -Xmx1024m
-archash=9947972b4ee3f60df35ba3d8be42ad5c287c9a43
+archash=e731db5f9be2004fab0b3b23206f6c2211d01176