package edu.brown.cs.h2r.burlapcraft.domaingenerator.propositionalfunction;


import burlap.mdp.core.oo.propositional.PropositionalFunction;
import burlap.mdp.core.oo.state.OOState;
import edu.brown.cs.h2r.burlapcraft.helper.HelperActions;
import edu.brown.cs.h2r.burlapcraft.state.BCBlock;

public class PFBlockIsShape extends PropositionalFunction {

	protected String shape;

	public PFBlockIsShape(String name, String[] parameterClasses, String shape) {
		super(name, parameterClasses);
		this.shape = shape;
	}

	@Override
	public boolean isTrue(OOState s, String... params) {
		BCBlock block = (BCBlock)s.object(params[0]);
		return HelperActions.blockShapeMap.get(block.type).equals(this.shape);
	}
	
}