make units usable in treesets-maps

JavaBWAPI · Jan 20, 2019 · e848670 · e848670
1 parent 5002031
commit e848670
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/src/main/java/bwapi/Unit.java b/src/main/java/bwapi/Unit.java
@@ -12,7 +12,7 @@
 import static bwapi.Race.Zerg;
 import static bwapi.UnitType.*;
 
-public class Unit {
+public class Unit implements Comparable<Unit>{
     private static Set<Order> gatheringGasOrders = EnumSet.of(
             Harvest1, Harvest2, MoveToGas, WaitForGas, HarvestGas, ReturnGas, ResetCollision);
     private static Set<Order> gatheringMineralOrders = EnumSet.of(
@@ -4115,17 +4115,21 @@ public boolean canPlaceCOP(TilePosition target, boolean checkCanIssueCommandType
     }
 
 
-    public boolean equals(final Object that) {
-        if (!(that instanceof Unit)) {
+    public boolean equals(final Object other) {
+        if (!(other instanceof Unit)) {
             return false;
         }
-        return getID() == ((Unit) that).getID();
+        return getID() == ((Unit) other).getID();
     }
 
     public int hashCode() {
         return getID();
     }
 
+    public int compareTo(final Unit other) {
+        return getID() - other.getID();
+    }
+
     void updateType(final int frame) {
         if (frame > lastTypeUpdate) {
             lastTypeUpdate = frame;