package aima.gui.swing.applications.agent;

import java.awt.Color;
import java.awt.Graphics;
import java.awt.Graphics2D;
import java.util.ArrayList;
import java.util.Hashtable;
import java.util.List;

import aima.core.agent.Action;
import aima.core.agent.Agent;
import aima.core.agent.Environment;
import aima.core.agent.Percept;
import aima.core.agent.impl.DynamicAction;
import aima.core.environment.vacuum.VacuumEnvironment;
import aima.gui.swing.framework.EmptyEnvironmentView;

/**
 * Displays the informations provided by a <code>VacuumEnvironment</code> on a
 * panel using 2D-graphics.
 * 
 * @author Ruediger Lunde
 */
public class VacuumView extends EmptyEnvironmentView {

	private static final long serialVersionUID = 1L;
	private Hashtable<Agent, Action> lastActions = new Hashtable<Agent, Action>();

	@Override
	public void agentActed(Agent agent, Percept percept, Action action, Environment source) {
		lastActions.put(agent, action);
		String prefix = "";
		if (env.getAgents().size() > 1)
			prefix = "A" + env.getAgents().indexOf(agent) + ": ";
		notify(prefix + action.toString());
		super.agentActed(agent, percept, action, source);
	}

	protected VacuumEnvironment getVacuumEnv() {
		return (VacuumEnvironment) env;
	}

	/**
	 * Creates a 2D-graphics showing the agent in its environment. Locations are
	 * represented as rectangles, dirt as grey background color, and the agent
	 * as red Pacman.
	 */
	@Override
	public void paintComponent(Graphics g) {
		List<String> locations = getLocations();
		adjustTransformation(0, 0, 11 * locations.size() - 1, 10);
		Graphics2D g2 = (Graphics2D) g;
		g2.setBackground(Color.white);
		g2.clearRect(0, 0, getWidth(), getHeight());
		for (int i = 0; i < locations.size(); i++) {
			String location = locations.get(i);
			Agent agent = getAgent(location);
			if (isDirty(location)) {
				g2.setColor(Color.LIGHT_GRAY);
				g2.fillRect(x(11 * i), y(0), scale(10), scale(10));
			}
			g2.setColor(Color.black);
			g2.drawRect(x(11 * i), y(0), scale(10), scale(10));
			g2.drawString(location.toString(), x(11 * i) + 10, y(0) + 20);
			if (agent != null) {
				Action action = lastActions.get(agent);
				g2.setColor(Color.RED);
				if (action == null || !((DynamicAction) action).getAttribute("name").equals("Suck"))
					g2.fillArc(x(11 * i + 2), y(2), scale(6), scale(6), 200, 320);
				else
					g2.fillOval(x(11 * i + 2), y(2), scale(6), scale(6));
			}
		}
	}

	/** Returns the names of all locations used. */
	protected List<String> getLocations() {
		List<String> result = new ArrayList<String>();
		if (env != null) {
			result.add(VacuumEnvironment.LOCATION_A);
			result.add(VacuumEnvironment.LOCATION_B);
		}
		return result;
	}

	/** Checks whether the specified location is dirty. */
	protected boolean isDirty(String location) {
		return VacuumEnvironment.LocationState.Dirty == getVacuumEnv().getLocationState(location);
	}

	/**
	 * Checks whether an agent is currently at the specified location and
	 * returns one of them if any.
	 */
	protected Agent getAgent(Object location) {
		VacuumEnvironment e = getVacuumEnv();
		for (Agent a : e.getAgents())
			if (location.equals(e.getAgentLocation(a)))
				return a;
		return null;
	}
}