SizeOverloadHandicap.java

package nl.tudelft.bw4t.server.model.robots.handicap;

import nl.tudelft.bw4t.server.model.doors.Door;
import nl.tudelft.bw4t.server.model.robots.MoveType;
import nl.tudelft.bw4t.server.model.zone.Corridor;
import nl.tudelft.bw4t.server.model.zone.Room;
import nl.tudelft.bw4t.server.model.zone.Zone;

/**
 * Creates the Size overload handicap for a robot.
 */
public class SizeOverloadHandicap extends AbstractRobotDecorator {

	/**
	 * This variable indicates the size of the robot.
	 */
	private final int size;

	/**
	 * Calls the super method on p, Sets the handicap to active, Adds the
	 * handicap to the robot handicap storage.
	 * 
	 * @param p
	 *            HandicapInterface the SizeOverloadHandicap wraps around.
	 * @param s
	 *            Size the bot is gonna be.
	 */
	public SizeOverloadHandicap(IRobot p, int s) {
		super(p);
		size = s;
		robot.setSize(s);
		robot.getHandicapsList().add("SizeOverload");
	}

	/**
	 * Override of the checkZoneAccess method, The robot cannot access zones
	 * anymore.
	 * 
	 * @param startzone
	 *            Startzone
	 * @param endzone
	 *            Endzone
	 * @param door
	 *            Door
	 * @return MoveType.SAME_AREA
	 */
	@Override
	public MoveType checkZoneAccess(Zone startzone, Zone endzone, Door door) {
		if (robot.getSize() >= 4) {
			if (startzone == endzone) {
				return MoveType.SAME_AREA;
			} else if (endzone instanceof Corridor) {
				if (!robot.isOneBotPerZone()
						|| endzone.containsMeOrNothing(robot)) {
					return MoveType.ENTER_CORRIDOR;
				}
				return MoveType.HIT_OCCUPIED_ZONE;
			} else if (endzone instanceof Room) {
				return MoveType.HIT_CLOSED_DOOR;
			}
			return MoveType.ENTERING_FREESPACE;
		} else {
			return super.checkZoneAccess(startzone, endzone, door);
		}
	}

	@Override
	public int getSize() {
		return this.size;
	}
}