package aima.core.environment.vacuum;

import aima.core.agent.Action;
import aima.core.agent.AgentProgram;
import aima.core.agent.Percept;
import aima.core.agent.impl.AbstractAgent;
import aima.core.agent.impl.NoOpAction;

import java.util.Objects;

/**
 * Artificial Intelligence A Modern Approach (3rd Edition): Figure 2.8, page 48.<br>
 * <br>
 * 
 * <pre>
 * function REFLEX-VACUUM-AGENT([location, status]) returns an action
 *   
 *   if status = Dirty then return Suck
 *   else if location = A then return Right
 *   else if location = B then return Left
 * </pre>
 * 
 * Figure 2.8 The agent program for a simple reflex agent in the two-state
 * vacuum environment. This program implements the action function tabulated in
 * Figure 2.3.
 * 
 * @author Ciaran O'Reilly
 * 
 */
public class ReflexVacuumAgent extends AbstractAgent {

	public ReflexVacuumAgent() {
		super(new AgentProgram() {
			// function REFLEX-VACUUM-AGENT([location, status]) returns an
			// action
			public Action execute(Percept percept) {
				LocalVacuumEnvironmentPercept vep = (LocalVacuumEnvironmentPercept) percept;

				// if status = Dirty then return Suck
				if (VacuumEnvironment.LocationState.Dirty == vep
						.getLocationState()) {
					return VacuumEnvironment.ACTION_SUCK;
					// else if location = A then return Right
				} else if (Objects.equals(VacuumEnvironment.LOCATION_A, vep
                        .getAgentLocation())) {
					return VacuumEnvironment.ACTION_MOVE_RIGHT;
				} else if (Objects.equals(VacuumEnvironment.LOCATION_B, vep
						.getAgentLocation())) {
					// else if location = B then return Left
					return VacuumEnvironment.ACTION_MOVE_LEFT;
				}

				// Note: This should not be returned if the
				// environment is correct
				return NoOpAction.NO_OP;
			}
		});
	}
}