package burlap.domain.singleagent.graphdefined;

import burlap.mdp.core.state.MutableState;
import burlap.mdp.core.state.State;
import burlap.mdp.core.state.annotations.DeepCopyState;

import java.util.ArrayList;
import java.util.List;

/**
 * @author James MacGlashan.
 */
@DeepCopyState
public class GraphStateNode implements MutableState {

	protected int id;

	protected static List<Object> keys;

	public GraphStateNode() {
		if(keys == null) {
			keys = new ArrayList<Object>();
			keys.add("node");
		}
	}

	public GraphStateNode(int id) {
		super();
		this.id = id;
	}

	@Override
	public MutableState set(Object variableKey, Object value) {
		if(value instanceof Number) {
			id = ((Number) value).intValue();
		} else if(value instanceof String) {
			try {
				id = Integer.parseInt((String) value);
			} catch(Exception e) {
				throw new RuntimeException("Could not parse string value " + value + " into an int. Cannot set GraphState value.");
			}
		} else {
			throw new RuntimeException("Cannot set graph value to value of type " + value.getClass().getName() + ". Use int or String.");
		}
		return this;
	}

	@Override
	public List<Object> variableKeys() {
		return keys;
	}

	@Override
	public Object get(Object variableKey) {
		return id;
	}

	@Override
	public State copy() {
		return new GraphStateNode(id);
	}

	public int getId() {
		return id;
	}

	public void setId(int id) {
		this.id = id;
	}

	@Override
	public String toString() {
		return String.valueOf(id);
	}
}

@@ -494,14 +494,30 @@
 add(
-%22node%22
+GraphDefinedDomain.VAR
 );%0A%09
@@ -562,13 +562,12 @@
 %7B%0A%09%09
-super
+this
 ();%0A

Fix initialization of GraphStateNode variable keys.