thrown robot collision damage change, remove flying robot that hit cat, destroy flying robots that died, don't allow flying robots to pick up cheese

Author: hannah-y-gao

engine/src/main/battlecode/world/GameWorld.java

@@ -1137,6 +1137,7 @@ public void destroyRobot(int id, boolean fromException, boolean fromDamage) {
 
             for (MapLocation robotLoc : robot.getAllPartLocations()) {
                 removeRobot(robotLoc);
+                removeFlyingRobot(robotLoc);
             }
 
             if (robot.isCarryingRobot()) {

engine/src/main/battlecode/world/InternalRobot.java

@@ -724,8 +724,8 @@ private void getThrown(Direction dir) {
 
         this.setInternalLocationOnly(this.getLocation());
 
-        this.travelFlying(true);
         this.travelFlying(false);
+        this.travelFlying(true);
     }
 
     public void getDropped(MapLocation loc) {
@@ -786,6 +786,10 @@ public void hitTarget(boolean isSecondMove) {
             InternalRobot robot = this.gameWorld.getRobot(this.getLocation().add(this.thrownDir));
             robot.addHealth(-damage);
         }
+        else if (this.gameWorld.getFlyingRobot(this.getLocation().add(this.thrownDir)) != null){
+            InternalRobot robot = this.gameWorld.getFlyingRobot(this.getLocation().add(this.thrownDir));
+            robot.remainingThrowDuration = 1; // force other robot to drop to ground as well on next turn
+        }
         this.thrownDir = null;
         this.remainingThrowDuration = 0;
 
@@ -821,6 +825,7 @@ public void travelFlying(boolean isSecondMove) {
         } else if (this.gameWorld.getRobot(newLoc) != null
                 && this.gameWorld.getRobot(newLoc).getType() == UnitType.CAT) {
             // cat feeding!
+            this.gameWorld.removeFlyingRobot(this.location);
             this.addHealth(-this.getHealth()); // rat dies :(
             // put cat to sleep
             this.gameWorld.getRobot(newLoc).sleepTimeRemaining = GameConstants.CAT_SLEEP_TIME;

engine/src/main/battlecode/world/RobotControllerImpl.java

@@ -469,6 +469,10 @@ private void assertCanPickUpCheese(MapLocation loc) throws GameActionException {
         if (myType != UnitType.BABY_RAT && myType != UnitType.RAT_KING) {
             throw new GameActionException(CANT_DO_THAT, "Only rats can pick up cheese");
         }
+
+        if (this.robot.isBeingThrown()){
+            throw new GameActionException(CANT_DO_THAT, "Flying rats cannot pick up cheese");
+        }
     }
 
     @Override
@@ -551,7 +555,8 @@ public RobotInfo senseRobotAtLocation(MapLocation loc) throws GameActionExceptio
     @Override
     public boolean canSenseRobot(int id) {
         InternalRobot sensedRobot = getRobotByID(id);
-        return sensedRobot != null && canSenseLocation(sensedRobot.getLocation());
+        Boolean isFlying = sensedRobot.isBeingThrown();
+        return sensedRobot != null && isFlying && canSenseLocation(sensedRobot.getLocation());
     }
 
     @Override