Commander unit component

core/src/mindustry/entities/def/CommanderComp.java

@@ -0,0 +1,83 @@
+package mindustry.entities.def;
+
+import arc.struct.*;
+import arc.util.ArcAnnotate.*;
+import arc.util.*;
+import mindustry.ai.formations.*;
+import mindustry.ai.types.*;
+import mindustry.annotations.Annotations.*;
+import mindustry.entities.units.*;
+import mindustry.gen.*;
+
+/** A unit that can command other units. */
+@Component
+abstract class CommanderComp implements Unitc{
+    private static final Array<FormationMember> members = new Array<>();
+
+    @Import float x, y, rotation;
+
+    transient @Nullable Formation formation;
+    transient Array<Unitc> controlling = new Array<>();
+
+    @Override
+    public void update(){
+        if(formation != null){
+            formation.anchor.set(x, y, rotation);
+            formation.updateSlots();
+        }
+    }
+
+    @Override
+    public void remove(){
+        clearCommand();
+    }
+
+    @Override
+    public void killed(){
+        clearCommand();
+    }
+
+    //make sure to reset command state when the controller is switched
+    @Override
+    public void controller(UnitController unitController){
+        clearCommand();
+    }
+
+    void command(Formation formation, Array<Unitc> units){
+        clearCommand();
+
+        controlling.addAll(units);
+        for(Unitc unit : units){
+            unit.controller(new FormationAI(this, formation));
+        }
+        this.formation = formation;
+
+        members.clear();
+        for(Unitc u : units){
+            members.add((FormationAI)u.controller());
+        }
+
+        Log.info(members);
+        Log.info(members.size);
+
+
+        //TODO doesn't handle units that don't fit a formation
+        formation.addMembers(members);
+    }
+
+    boolean isCommanding(){
+        return formation != null;
+    }
+
+    void clearCommand(){
+        //reset controlled units
+        for(Unitc unit : controlling){
+            if(unit.controller().isBeingControlled(this)){
+                unit.controller(unit.type().createController());
+            }
+        }
+
+        controlling.clear();
+        formation = null;
+    }
+}