import edu.mit.six825.bn.inputs.*;
import edu.mit.six825.bn.bayesnet.*;
import edu.mit.six825.bn.functiontable.*;

public class example {

    public static void main(String[] args) {

	BayesNet bn = Nets.getBurglary();

	FunctionVariable [] vars = new FunctionVariable[2];
	Comparable       [] vals = new Comparable[2];

	vars[0] = bn.nodes.getNode("JohnCalls").var;
	vars[1] = bn.nodes.getNode("MaryCalls").var;
	vals[0] = ComparableBoolean.TRUE;
	vals[1] = ComparableBoolean.TRUE;

	Assignment evidence = new Assignment(new FunctionVariableSet(vars),
					     vals);

	Solver enumSolver = new EnumerationSolver(bn);

	enumSolver.setEvidence(evidence);

	System.out.println(enumSolver.query(bn.nodes.getNode("Burglary")));
    }
}