fr.lirmm.boreal.util.converter.QueryConverter Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of integraal-util Show documentation

Util objects for integraal

There is a newer version: 1.6.3

package fr.lirmm.boreal.util.converter;

import com.google.common.collect.BiMap;
import com.google.common.collect.HashBiMap;
import fr.boreal.model.logicalElements.api.Variable;
import fr.boreal.model.query.api.FOQuery;
import fr.boreal.model.query.factory.FOQueryFactory;
import fr.lirmm.graphik.integraal.api.core.ConjunctiveQuery;

import java.util.stream.Collectors;

/**
 * Converts FOQuery to and from Graal's ConjunctiveQuery
 * 

 * This is used for compatibility with Graal 1.3 version.
 *
 * @author Florent Tornil
 *
 */
public class QueryConverter {

	private static final BiMap, ConjunctiveQuery> cache = HashBiMap.create();

	/**
	 * Converts the given FOQuery into a Graal ConjunctiveQuery
	 *
	 * @param query the query to convert
	 * @return the converted query
	 */
	public static ConjunctiveQuery convert(FOQuery query) {
		ConjunctiveQuery q = cache.get(query);
		if (q == null) {
			q = fr.lirmm.graphik.integraal.core.factory.DefaultConjunctiveQueryFactory.instance().create(
					AtomSetConverter.convert(query.getFormula()),
					query.getAnswerVariables().stream().map(TermConverter::convert).collect(Collectors.toList()));
			cache.put(query, q);
		}
		return q;

	}

	/**
	 * Converts the given FOQuery into a Graal ConjunctiveQuery
	 *
	 * @param query the query to convert
	 * @return the converted query
	 */
	public static fr.lirmm.graphik.graal.api.core.ConjunctiveQuery convert2(FOQuery query) {

		return fr.lirmm.graphik.graal.core.factory.DefaultConjunctiveQueryFactory.instance().create(
				AtomSetConverter.convert2(query.getFormula()),
				query.getAnswerVariables().stream().map(TermConverter::convert2).collect(Collectors.toList()));

	}

	/**
	 * Converts the given ConjunctiveQuery into a FOQuery
	 *
	 * @param query the query to convert
	 * @return the converted query
	 */
	public static FOQuery reverse(ConjunctiveQuery query) {
		FOQuery q = cache.inverse().get(query);
		if (q == null) {
			q = FOQueryFactory.instance().createOrGetQuery(
					AtomSetConverter.reverse(query.getAtomSet()), query.getAnswerVariables().stream()
							.map(TermConverter::reverse).map(t -> (Variable) t).collect(Collectors.toList()),
					null);
		}
		return q;
	}

}