org.coode.oppl.search.AssignableValueExtractor Maven / Gradle / Ivy

Go to download
package org.coode.oppl.search;

import static org.coode.oppl.utils.ArgCheck.checkNotNull;

import java.util.Collections;
import java.util.Iterator;
import java.util.Set;
import java.util.regex.Pattern;

import org.coode.oppl.VariableVisitorEx;
import org.coode.oppl.function.ValueComputationParameters;
import org.coode.oppl.generated.GeneratedVariable;
import org.coode.oppl.generated.RegexpGeneratedVariable;
import org.coode.oppl.rendering.ManchesterSyntaxRenderer;
import org.coode.oppl.variabletypes.InputVariable;
import org.coode.oppl.variabletypes.VariableTypeVisitorEx;
import org.semanticweb.owlapi.model.OWLObject;

/** @author Luigi Iannone */
public final class AssignableValueExtractor implements
        VariableVisitorEx> {
    private final VariableTypeVisitorEx> assignableValuesVisitor;
    private final ValueComputationParameters parameters;

    /** @param assignableValuesVisitor
     *            assignableValuesVisitor
     * @param parameters
     *            parameters */
    public AssignableValueExtractor(
            VariableTypeVisitorEx> assignableValuesVisitor,
            ValueComputationParameters parameters) {
        this.assignableValuesVisitor = checkNotNull(assignableValuesVisitor,
                "assignableValuesVisitor");
        this.parameters = checkNotNull(parameters, "parameters");
    }

    @Override
    public  Set visit(InputVariable v) {
        return v.getType().accept(assignableValuesVisitor);
    }

    @Override
    public  Set visit(
            RegexpGeneratedVariable v) {
        Set toReturn = v.getType().accept(assignableValuesVisitor);
        Iterator iterator = toReturn.iterator();
        while (iterator.hasNext()) {
            OWLObject owlObject = iterator.next();
            ManchesterSyntaxRenderer renderer = getParameters().getConstraintSystem()
                    .getOPPLFactory()
                    .getManchesterSyntaxRenderer(getParameters().getConstraintSystem());
            owlObject.accept(renderer);
            Pattern pattern = v.getPatternGeneratingOPPLFunction().compute(
                    getParameters());
            if (!pattern.matcher(renderer.toString()).matches()) {
                iterator.remove();
            }
        }
        return toReturn;
    }

    @Override
    public  Set visit(GeneratedVariable v) {
        return Collections.emptySet();
    }

    /** @return the parameters */
    public ValueComputationParameters getParameters() {
        return parameters;
    }
}