Merge pull request #3 from NiChrosia/master

Remove reset on WorldLoadEvent

src/UnitInfo/core/HudUi.java

@@ -59,6 +59,19 @@ public class HudUi {
     int maxwave;
     int coreamount;
 
+    public void reset(){
+        bars.each(bar -> {
+            weapon = new Table();
+            core = new Table();
+            wave = new Table();
+            waveTable = new Table();
+
+            bar.visible = false;
+            bar.clear();
+            bar.remove();
+        });
+    }
+
     public Unit getUnit(){
         Seq<Unit> units = Groups.unit.intersect(Core.input.mouseWorldX(), Core.input.mouseWorldY(), 4, 4);
         if(units.size <= 0) return player.unit();

src/UnitInfo/core/Main.java

@@ -3,29 +3,29 @@ package UnitInfo.core;
 import UnitInfo.ui.FreeBar;
 import arc.Core;
 import arc.Events;
-import mindustry.Vars;
 import mindustry.game.EventType.*;
 import mindustry.gen.Groups;
 import mindustry.mod.Mod;
 
 public class Main extends Mod {
     public static Setting settingAdder = new Setting();
+    public static HudUi hud = new HudUi();
 
-    public Main(){
+    @Override
+    public void init(){
         Events.on(ClientLoadEvent.class, e -> {
-            HudUi hud = new HudUi();
+            hud.reset();
+
+            hud = new HudUi();
             settingAdder.init();
             hud.addTable();
             hud.addCoreTable();
         });
 
-        Events.on(WorldLoadEvent.class, e -> {
-            HudUi hud = new HudUi();
-            hud.addWaveTable();
-        });
-
         Events.on(ResetEvent.class, e -> {
-            HudUi hud = new HudUi();
+            hud.reset();
+
+            hud = new HudUi();
             hud.addWaveTable();
         });
 
@@ -45,10 +45,6 @@ public class Main extends Mod {
         */
     }
 
-    @Override
-    public void init(){
-    }
-
     @Override
     public void loadContent(){