com.clarkparsia.pellet.sparqldl.engine.QuerySizeEstimator Maven / Gradle / Ivy

Go to download
// Copyright (c) 2006 - 2008, Clark & Parsia, LLC. 
// This source code is available under the terms of the Affero General Public License v3.
//
// Please see LICENSE.txt for full license terms, including the availability of proprietary exceptions.
// Questions, comments, or requests for clarification: [email protected]

package com.clarkparsia.pellet.sparqldl.engine;

import java.util.HashSet;
import java.util.Set;

import org.mindswap.pellet.utils.ATermUtils;
import org.mindswap.pellet.utils.SizeEstimate;

import com.clarkparsia.pellet.sparqldl.model.Query;
import com.clarkparsia.pellet.sparqldl.model.QueryAtom;

import aterm.ATermAppl;

/**
 * 
 * Title: Computation of size estimate for a knowledge base and a query.
 * 
 * 
 * Description:
 * 
 * 
 * Copyright: Copyright (c) 2007
 * 
 * 
 * Company: Clark & Parsia, LLC. 
 * 
 * 
 * @author Petr Kremen
 */
public class QuerySizeEstimator {

	public static void computeSizeEstimate(final Query query) {
		final SizeEstimate sizeEstimate = query.getKB().getSizeEstimate();

		final Set concepts = new HashSet();
		final Set properties = new HashSet();
//		boolean fullDone = false;
		for (final QueryAtom atom : query.getAtoms()) {
//			if (!fullDone) {
//				switch (atom.getPredicate()) {
//				case Type:
//					if (query.getDistVars()
//							.contains(atom.getArguments().get(1))) {
//						fullDone = true;
//					}
//					break;
//				case PropertyValue:
//					if (query.getDistVars()
//							.contains(atom.getArguments().get(1))) {
//						fullDone = true;
//					}
//					break;
//				case SameAs:
//				case DifferentFrom:
//					break;
//				default:
////					fullDone = true;
//					;
//				}
//				if (fullDone) {
//					sizeEstimate.computeAll();
//				}
//			}

			for (final ATermAppl argument : atom.getArguments()) {
				if (!ATermUtils.isVar(argument)) {
					if ((query.getKB().isClass(argument) || ATermUtils
							.isComplexClass(argument))
							&& !sizeEstimate.isComputed(argument)) {
						concepts.add(argument);
					}

					if (query.getKB().isProperty(argument)
							&& !sizeEstimate.isComputed(argument)) {
						properties.add(argument);
					}
				}
			}
		}

		sizeEstimate.compute(concepts, properties);
	}
}