ai.libs.jaicore.search.probleminputs.AMDP Maven / Gradle / Ivy

package ai.libs.jaicore.search.probleminputs;

import java.util.Map;
import java.util.Random;
import java.util.stream.Collectors;

import ai.libs.jaicore.basic.sets.SetUtil;

public abstract class AMDP> implements IMDP {

	private final N initState;

	protected AMDP(final N initState) {
		super();
		this.initState = initState;
	}

	@Override
	public N getInitState() {
		return this.initState;
	}

	@Override
	public double getProb(final N state, final A action, final N successor) throws InterruptedException {
		return this.getProb(state, action, successor, false);
	}

	public double getProb(final N state, final A action, final N successor, final boolean setProbabilityOfUndefinedToZero) throws InterruptedException {
		Map dist = this.getProb(state, action);
		if (!dist.containsKey(successor)) {
			if (setProbabilityOfUndefinedToZero) {
				return 0.0;
			}
			else {
				throw new IllegalArgumentException("No probability defined for the following triplet:\n\tFrom state: " + state + "\n\tUsed action: " + action  + "\n\tTo state: " + successor + ".\nDistribution is: " + dist.entrySet().stream().map(e -> "\n\t" + e.toString()).collect(Collectors.joining()));
			}
		}
		return dist.get(successor);
	}

	@Override
	public boolean isTerminalState(final N state) throws InterruptedException {
		return this.getApplicableActions(state).isEmpty();
	}

	@Override
	public A getUniformlyRandomApplicableAction(final N state, final Random random) throws InterruptedException {
		return SetUtil.getRandomElement(this.getApplicableActions(state), random);
	}

	@Override
	public boolean isActionApplicableInState(final N state, final A action) throws InterruptedException {
		return this.getApplicableActions(state).contains(action);
	}
}