complexify the PlayTestListener

Author: JasperGeurtz

src/main/java/bwapi/Unit.java

@@ -433,7 +433,9 @@ public Set<Unit> getInterceptors() {
         if (getType() != Protoss_Carrier && getType() != Hero_Gantrithor) {
             return new HashSet<>();
         }
-        return connectedUnits;
+        return game.getAllUnits().stream()
+                .filter(u -> equals(u.getCarrier()))
+                .collect(Collectors.toSet());
     }
 
     public Unit getHatchery() {
@@ -444,7 +446,9 @@ public Set<Unit> getLarva() {
         if (!getType().producesLarva()) {
             return new HashSet<>();
         }
-        return connectedUnits;
+        return game.getAllUnits().stream()
+                .filter(u -> equals(u.getHatchery()))
+                .collect(Collectors.toSet());
     }
 
     //TODO
@@ -502,6 +506,10 @@ public boolean isBurrowed() {
         return unitData.isBurrowed();
     }
 
+    public boolean isCarrying() {
+        return isCarryingGas() || isCarryingMinerals();
+    }
+
     public boolean isCarryingGas() {
         return unitData.carryResourceType() == 1;
     }

src/test/java/PlayTestListener.java

@@ -1,4 +1,7 @@
 import bwapi.*;
+import bwapi.point.Position;
+import bwapi.point.TilePosition;
+import bwapi.types.Race;
 import bwapi.types.UnitType;
 
 import java.util.Comparator;
@@ -22,25 +25,58 @@ public void onStart() {
 
     public void onFrame() {
         final Player self = game.self();
+        final Race race = self.getRace();
+
         // find the depot
         final Unit depot = self.getUnits().stream()
                 .filter(u -> u.getType().isResourceDepot())
                 .findFirst().get();
 
         // train workers
         if (depot.isIdle()) {
-            depot.train(self.getRace().getWorker());
+            depot.train(race.getWorker());
         }
 
-
-        //find accessible minerals
+        // find accessible minerals
         final PriorityQueue<Unit> minerals = new PriorityQueue<>(Comparator.comparingInt(a -> a.getDistance(depot)));
 
-        minerals.addAll(game.getMinerals());
+        Position avgMineralPos = Position.Origin;
+        int count = 0;
+        for (final Unit mineral : game.getMinerals()) {
+            if (mineral.isVisible()) {
+                avgMineralPos = avgMineralPos.add(mineral.getInitialPosition());
+                count += 1;
+                minerals.add(mineral);
+            }
+        }
+        avgMineralPos = avgMineralPos.divide(count);
+        System.out.println(avgMineralPos);
 
+        // make workers gather minerals if idle
         self.getUnits().stream()
                 .filter(u -> u.getType().isWorker() && u.isIdle())
                 .forEach(u -> u.gather(minerals.poll()));
+
+
+        if (self.supplyTotal() - self.supplyUsed() <= 2) {
+            final UnitType supplyProvider = race.getSupplyProvider();
+            if (self.minerals() >= supplyProvider.mineralPrice()) {
+                final TilePosition depotTP = depot.getTilePosition();
+                final TilePosition avgMinTP = avgMineralPos.toTilePosition();
+                final UnitType builderType = supplyProvider.whatBuilds().getKey();
+                self.getUnits().stream()
+                        .filter(u -> u.getType().equals(builderType) && !u.isCarrying())
+                        .findFirst().ifPresent(u -> {
+                            if (u.getType().isWorker()) {
+                                u.build(supplyProvider, depotTP.add(depotTP.subtract(avgMinTP)));
+                            }
+                            else {
+                                u.train(supplyProvider);
+                            }
+                });
+            }
+        }
+
     }
 
     public static void main(String[] args) {