add PaintType.isEnemy() UnitType.getBaseType() and RobotController.canPaint() methods

Author: Ian Gonzalez Hermosillo

engine/src/main/battlecode/common/PaintType.java

@@ -11,6 +11,10 @@ public boolean isAlly() {
         return this == ALLY_PRIMARY || this == ALLY_SECONDARY;
     }
 
+    public boolean isEnemy(){
+        return this == ENEMY_PRIMARY || this == ENEMY_SECONDARY;
+    }
+
     public boolean isSecondary() {
         return this == ALLY_SECONDARY || this == ENEMY_SECONDARY;
     }

engine/src/main/battlecode/common/RobotController.java

@@ -676,6 +676,16 @@ public interface RobotController {
     // ***** ATTACK / HEAL ********
     // ****************************
 
+    /**
+     * Tests whether this robot can paint the given location. 
+     * 
+     * @param loc target location to paint
+     * @return true if rc.attack(loc) will paint the given location 
+     * 
+     * @battlecode.doc.costlymethod
+     */
+    boolean canPaint(MapLocation loc);
+
     /**
      * Tests whether this robot can attack the given location. Types of
      * attacks for specific units determine whether or not towers, other
@@ -857,6 +867,7 @@ public interface RobotController {
      * @param red   the red component of the dot's color
      * @param green the green component of the dot's color
      * @param blue  the blue component of the dot's color
+     * @throws GameActionException if the location is off the map
      *
      * @battlecode.doc.costlymethod
      */
@@ -870,6 +881,7 @@ public interface RobotController {
      * @param red      the red component of the line's color
      * @param green    the green component of the line's color
      * @param blue     the blue component of the line's color
+     * @throws GameActionException if any location is off the map
      *
      * @battlecode.doc.costlymethod
      */

engine/src/main/battlecode/common/UnitType.java

@@ -78,6 +78,18 @@ public UnitType getNextLevel(){
         }
     }
 
+    public UnitType getBaseType(){
+        switch (this){
+            case LEVEL_TWO_DEFENSE_TOWER: return LEVEL_ONE_DEFENSE_TOWER;
+            case LEVEL_THREE_DEFENSE_TOWER: return LEVEL_ONE_DEFENSE_TOWER;
+            case LEVEL_TWO_PAINT_TOWER: return LEVEL_ONE_PAINT_TOWER;
+            case LEVEL_THREE_PAINT_TOWER: return LEVEL_ONE_PAINT_TOWER;
+            case LEVEL_TWO_MONEY_TOWER: return LEVEL_ONE_MONEY_TOWER;
+            case LEVEL_THREE_MONEY_TOWER: return LEVEL_ONE_MONEY_TOWER;
+            default: return this;
+        }
+    }
+
     UnitType(int paintCost, int moneyCost, int attackCost, int health, int level, int paintCapacity, int actionCooldown, int actionRadiusSquared, int attackStrength, int aoeAttackStrength, int paintPerTurn, int moneyPerTurn) {
         this.paintCost = paintCost;
         this.moneyCost = moneyCost;

engine/src/main/battlecode/instrumenter/bytecode/resources/MethodCosts.txt

@@ -26,6 +26,7 @@ battlecode/common/RobotController/getResourcePattern                    2    tru
 battlecode/common/RobotController/getTowerPattern                       2    true
 battlecode/common/RobotController/adjacentLocation                      1    true
 battlecode/common/RobotController/attack                                0    true
+battlecode/common/RobotController/canPaint                              10   true
 battlecode/common/RobotController/canAttack                             10   true
 battlecode/common/RobotController/canMopSwing                           10   true 
 battlecode/common/RobotController/mopSwing                              0    true            
@@ -111,11 +112,13 @@ battlecode/common/Message/getBytes                                      3    fal
 battlecode/common/Message/toString                                      5    false 
 battlecode/common/Message/copy                                          5    false
 battlecode/common/PaintType/isAlly                                      2    false
+battlecode/common/PaintType/isEnemy                                     2    false
 battlecode/common/PaintType/isSecondary                                 2    false
 battlecode/common/UnitType/isRobotType                                  3    false
 battlecode/common/UnitType/isTowerType                                  3    false
 battlecode/common/UnitType/canUpgradeType                               3    false
 battlecode/common/UnitType/getNextLevel                                 3    false
+battlecode/common/UnitType/getBaseType                                  3    false
 java/lang/Math/IEEEremainder            1  false
 java/lang/Math/abs                      1  false
 java/lang/Math/acos                     1  false

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

@@ -832,6 +832,27 @@ public void completeResourcePattern(MapLocation loc) throws GameActionException
     // ****** ATTACK / HEAL ********
     // *****************************
 
+    @Override
+    public boolean canPaint(MapLocation loc){
+        assertNotNull(loc);
+        if (!onTheMap(loc))
+            return false;
+        // towers and moppers cannot paint tiles
+        if (getType().isTowerType() || getType() == UnitType.MOPPER){
+            return false;
+        }
+        if (getType() == UnitType.SOLDIER){
+            if (loc.distanceSquaredTo(this.robot.getLocation()) > UnitType.SOLDIER.actionRadiusSquared)
+                return false;
+            return this.gameWorld.isPaintable(loc) && this.gameWorld.teamFromPaint(this.gameWorld.getPaint(loc)) != getTeam().opponent();
+        }
+        else{
+            if (loc.distanceSquaredTo(this.robot.getLocation()) > UnitType.SPLASHER.actionRadiusSquared)
+                return false;
+            return this.gameWorld.isPaintable(loc);
+        }
+    }
+
     private void assertCanAttackSoldier(MapLocation loc) throws GameActionException {
         assertIsActionReady();
         assertCanActLocation(loc, UnitType.SOLDIER.actionRadiusSquared);