get number cat and rat traps

Author: hannah-y-gao

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

@@ -62,6 +62,24 @@ public interface RobotController {
      */
     Team getBackstabbingTeam();
 
+    /**
+     * Returns the number of active cat traps for the team.
+     *
+     * @return the number of active cat traps this team has.
+     *
+     * @battlecode.doc.costlymethod
+     */
+    int getNumberCatTraps();
+
+    /**
+     * Returns the number of active rat traps for the team.
+     *
+     * @return the number of active rat traps this team has.
+     *
+     * @battlecode.doc.costlymethod
+     */
+    int getNumberRatTraps();
+
     // *********************************
     // ****** UNIT QUERY METHODS *******
     // *********************************

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

@@ -165,6 +165,16 @@ public int getDirt() {
         return this.gameWorld.getTeamInfo().getDirt(getTeam());
     }
 
+    @Override 
+    public int getNumberCatTraps(){
+        return this.gameWorld.getTrapCount(TrapType.CAT_TRAP, this.getTeam());
+    }
+
+    @Override 
+    public int getNumberRatTraps(){
+        return this.gameWorld.getTrapCount(TrapType.RAT_TRAP, this.getTeam());
+    }
+
     @Override
     public UnitType getType() {
         return this.robot.getType();