fr.lirmm.boreal.util.evaluator.MultiEvaluator Maven / Gradle / Ivy

Go to download
package fr.lirmm.boreal.util.evaluator;

import java.util.Collection;
import java.util.function.BiFunction;
import java.util.function.Function;

import org.slf4j.Logger;

public abstract class MultiEvaluator {

	protected static Logger LOG;

	private final Collection input;
	private Function batchFunction;
	private Function lazyFunction;

	private Integer timeout;

	private BiFunction outputIfTimeout;

	public MultiEvaluator(Collection input, Function function, Integer timeout,
			BiFunction outputIfTimeout) {

		this.input = input;
		this.batchFunction = function;
		this.lazyFunction = function;
		this.outputIfTimeout = outputIfTimeout;

		if (timeout == null) {
			this.timeout = EvaluatorConstants.DEFAULT_TIMEOUT;
		} else {
			this.timeout = timeout;
		}

	}

	public MultiEvaluator(Collection input, Function batchFunction,
			Function lazyFunction, Integer timeout,
			BiFunction outputIfTimeout) {

		this.input = input;
		this.batchFunction = batchFunction;
		this.lazyFunction = lazyFunction;
		this.timeout = timeout;
		this.outputIfTimeout = outputIfTimeout;

	}

	/**
	 * Evaluates all queries in a batch and returns a list of QueryEvaluationOutput.
	 *
	 * @return A list of QueryEvaluationOutput, each containing a query and its
	 *         corresponding iterator of substitutions.
	 */
	public final Iterable batchEvaluate() {

		return new BatchProcessor(input, batchFunction, timeout, outputIfTimeout).processBatch();

	}

	/**
	 * Provides lazy evaluation for the queries, yielding QueryEvaluationOutput for
	 * each query.
	 *
	 * @return An iterable for lazy evaluation of query results, each as a
	 *         QueryEvaluationOutput.
	 */
	public final Iterable lazyEvaluate() {

		return new LazyIterator<>(input, lazyFunction);

	}

}