revert making getPosition lazy

Author: JasperGeurtz

src/main/java/bwapi/EventHandler.java

@@ -5,6 +5,7 @@ class EventHandler implements Client.EventHandler {
     private final BWEventListener eventListener;
     private final Game game;
     private final Client client;
+    private int frames;
 
     public EventHandler(final BWEventListener eventListener, final Client client) {
         this.eventListener = eventListener;
@@ -17,6 +18,7 @@ public void operation(final ClientData.Event event) {
         final Unit u;
         switch (event.getType()) {
             case MatchStart:
+                frames = 0;
                 game.init();
                 game.setLatCom(false);
                 eventListener.onStart();
@@ -25,10 +27,11 @@ public void operation(final ClientData.Event event) {
                 eventListener.onEnd(event.getV1() != 0);
                 break;
             case MatchFrame:
-                game.onFrame();
+                game.onFrame(frames);
                 eventListener.onFrame();
+                frames += 1;
                 break;
-            //case MenuFrame:
+            //case 3: //MenuFrame
             case SendText:
                 eventListener.onSendText(client.eventString(event.getV1()));
                 break;
@@ -47,19 +50,22 @@ public void operation(final ClientData.Event event) {
             case UnitDiscover:
                 game.unitCreate(event.getV1());
                 u = game.getUnit(event.getV1());
-                u.updateType();
-                u.updatePlayer();
+                u.updatePosition(frames);
+                u.updateType(frames);
+                u.updatePlayer(frames);
                 eventListener.onUnitDiscover(u);
                 break;
             case UnitEvade:
                 u = game.getUnit(event.getV1());
+                u.updatePosition(frames);
                 eventListener.onUnitEvade(u);
                 break;
             case UnitShow:
                 game.unitShow(event.getV1());
                 u = game.getUnit(event.getV1());
-                u.updateType();
-                u.updatePlayer();
+                u.updatePosition(frames);
+                u.updateType(frames);
+                u.updatePlayer(frames);
                 eventListener.onUnitShow(u);
                 break;
             case UnitHide:
@@ -70,8 +76,9 @@ public void operation(final ClientData.Event event) {
             case UnitCreate:
                 game.unitCreate(event.getV1());
                 u = game.getUnit(event.getV1());
-                u.updateType();
-                u.updatePlayer();
+                u.updatePosition(frames);
+                u.updateType(frames);
+                u.updatePlayer(frames);
                 eventListener.onUnitCreate(u);
                 break;
             case UnitDestroy:
@@ -81,18 +88,19 @@ public void operation(final ClientData.Event event) {
                 break;
             case UnitMorph:
                 u = game.getUnit(event.getV1());
-                u.updateType();
+                u.updatePosition(frames);
+                u.updateType(frames);
                 eventListener.onUnitMorph(u);
                 break;
             case UnitRenegade:
                 u = game.getUnit(event.getV1());
-                u.updatePlayer();
+                u.updatePlayer(frames);
                 eventListener.onUnitRenegade(u);
                 break;
             case UnitComplete:
                 game.unitCreate(event.getV1());
                 u = game.getUnit(event.getV1());
-                u.updatePlayer();
+                u.updatePlayer(frames);
                 eventListener.onUnitComplete(u);
                 break;
         }

src/main/java/bwapi/Game.java

@@ -262,12 +262,13 @@ void unitHide(final int id) {
         visibleUnits.remove(id);
     }
 
-    void onFrame() {
-        if (getFrameCount() > 0) {
+    void onFrame(final int frame) {
+        if (frame > 0) {
             allUnits = Collections.unmodifiableList(visibleUnits.stream()
                 .map(i -> units[i])
                 .collect(Collectors.toList()));
         }
+        getAllUnits().forEach(u -> u.updatePosition(frame));
     }
 
     void addUnitCommand(final int type, final int unit, final int target, final int x, final int y, final int extra) {

src/main/java/bwapi/Unit.java

@@ -42,8 +42,9 @@ public class Unit {
         this.unitData = unitData;
         this.game = game;
 
-        updateType();
-        updatePlayer();
+        updateType(0);
+        updatePlayer(0);
+        updatePosition(0);
 
         initialType = getType();
         initialResources = getResources();
@@ -86,11 +87,6 @@ public UnitType getType() {
     }
 
     public Position getPosition() {
-        final int frame = game.getFrameCount();
-        if (frame > lastPositionUpdate) {
-            lastPositionUpdate = frame;
-            position = new Position(unitData.getPositionX(), unitData.getPositionY());
-        }
         return position;
     }
 
@@ -4130,20 +4126,25 @@ public int hashCode() {
         return getID();
     }
 
-    void updateType() {
-        final int frame = game.getFrameCount();
+    void updateType(final int frame) {
         if (frame > lastTypeUpdate) {
             lastTypeUpdate = frame;
             position = new Position(unitData.getPositionX(), unitData.getPositionY());
         }
         unitType = UnitType.idToEnum[unitData.getType()];
     }
 
-    void updatePlayer() {
-        final int frame = game.getFrameCount();
+    void updatePlayer(final int frame) {
         if (frame > lastPlayerUpdate) {
             lastPlayerUpdate = frame;
             player = game.getPlayer(unitData.getPlayer());
         }
     }
+
+    void updatePosition(final int frame) {
+        if (frame > lastPositionUpdate) {
+            lastPositionUpdate = frame;
+            position = new Position(unitData.getPositionX(), unitData.getPositionY());
+        }
+    }
 }