Angelica/src/main/java/kroppeb/stareval/function/BasicFunctionContext.java

package kroppeb.stareval.function;

import it.unimi.dsi.fastutil.objects.Object2ObjectOpenHashMap;
import kroppeb.stareval.expression.Expression;
import kroppeb.stareval.expression.VariableExpression;

import java.util.Map;

public class BasicFunctionContext implements FunctionContext {
	final private Map<String, Expression> variables = new Object2ObjectOpenHashMap<>();

	public void setVariable(String name, Expression value) {
		variables.put(name, value);
	}

	public void setIntVariable(String name, int value) {
		setVariable(name, (VariableExpression) (c, r) -> r.intReturn = value);
	}

	public void setFloatVariable(String name, float value) {
		setVariable(name, (VariableExpression) (c, r) -> r.floatReturn = value);
	}

	@Override
	public Expression getVariable(String name) {
		Expression expression = variables.get(name);
		if (expression == null)
			throw new RuntimeException("Variable hasn't been set: " + name);
		return expression;
	}

	@Override
	public boolean hasVariable(String name) {
		return variables.containsKey(name);
	}
}