astra.reasoner.node.ModuleFormulaReasonerNode Maven / Gradle / Ivy

Go to download
package astra.reasoner.node;

import java.util.Map;
import java.util.Stack;

import astra.formula.Formula;
import astra.formula.ModuleFormula;
import astra.reasoner.NewReasoner;
import astra.reasoner.Reasoner;
import astra.reasoner.util.BindingsEvaluateVisitor;
import astra.term.Term;

public class ModuleFormulaReasonerNode extends ReasonerNode {
    ModuleFormula formula;
    int state = 0;
    ReasonerNode child;

    public ModuleFormulaReasonerNode(ReasonerNode parent, ModuleFormula formula, Map initial, boolean singleResult) {
        super(parent, singleResult);

        this.formula = formula;
        this.initial = initial;
    }

    @Override
    public boolean solve(Reasoner reasoner, Stack stack) {
        switch (state) {
            case 0:
                Formula f = formula.adaptor().invoke(
                    new BindingsEvaluateVisitor(initial, reasoner.agent()), 
                    formula.predicate());
                child = ((NewReasoner) reasoner).createReasonerNode(this, f, initial, singleResult);
                stack.push(child);
                state++;
                break;
            case 1:
                if (child.isFailed()) {
                    finished = true;
                    failed = true;
                    return false;
                }
                solutions = child.solutions();
                finished = true;
                return true;            
        }       

        return true;
    }
}