com.clarkparsia.pellet.sparqldl.model.CoreNewImpl 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.model;

import java.util.ArrayList;
import java.util.Collection;
import java.util.List;
import java.util.Set;

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

import aterm.ATermAppl;

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

/**
 * 
 * Title: Implementation of the Core of undistinguished variables.
 * 
 * 
 * Description:
 * 
 * 
 * Copyright: Copyright (c) 2007
 * 
 * 
 * Company: Clark & Parsia, LLC. 
 * 
 * 
 * @author Petr Kremen
 */
public class CoreNewImpl implements QueryAtom, Core {

	private final QueryAtom	atom;

	private final Query		query;

	public CoreNewImpl(final Collection atoms, final Collection uv,
			final KnowledgeBase kb) {
		query = new QueryImpl( kb, false );

		final List signature = new ArrayList();

		for( final QueryAtom atom : atoms ) {
			query.add( atom );

			// this is nasty - remodeling will be fine
			switch ( atom.getPredicate() ) {
			case PropertyValue:
				final ATermAppl a1 = atom.getArguments().get( 1 );
				addI( atom.getArguments().get( 0 ), signature, uv );
				addI( atom.getArguments().get( 2 ), signature, uv );
				if( ATermUtils.isVar( a1 ) ) {
					if( !uv.contains( a1 ) ) {
						query.addDistVar( a1, VarType.PROPERTY );
					}
				}
				break;
			case Type:
				final ATermAppl aa1 = atom.getArguments().get( 1 );
				addI( atom.getArguments().get( 0 ), signature, uv );
				if( ATermUtils.isVar( aa1 ) ) {
					if( !uv.contains( aa1 ) ) {
						query.addDistVar( aa1, VarType.CLASS );
					}
				}
				break;
			default:
				throw new IllegalArgumentException( "Atom type " + atom.getPredicate()
						+ " is not supported in a core." );
			}
		}

		atom = new QueryAtomImpl( QueryPredicate.UndistVarCore, signature );
	}

	private CoreNewImpl(final Query query, final QueryAtom atom) {
		this.atom = atom;
		this.query = query;
	}

	private void addI(ATermAppl aa0, final List signature, final Collection uv) {
		if( ATermUtils.isVar( aa0 ) ) {
			if( !uv.contains( aa0 ) ) {
				query.addDistVar( aa0, VarType.INDIVIDUAL );
				signature.add( aa0 );
			}
		}
		else {
			signature.add( aa0 );
		}
	}

	/**
	 * {@inheritDoc}
	 */
	public CoreNewImpl apply(final ResultBinding binding) {
		return new CoreNewImpl( query.apply( binding ), atom.apply( binding ) );
	}

	@Override
	public int hashCode() {
		return atom.hashCode() + 7 * query.hashCode();
	}

	@Override
	public boolean equals(Object obj) {
		if( this == obj )
			return true;
		if( obj == null )
			return false;
		if( getClass() != obj.getClass() )
			return false;
		final CoreNewImpl other = (CoreNewImpl) obj;

		return atom.equals( other.atom ) && query.equals( other.query );
	}

	public Set getConstants() {
		return query.getConstants();
	}

	public Set getDistVars() {
		return query.getDistVars();
	}

	public Set getUndistVars() {
		return query.getUndistVars();
	}

	public List getArguments() {
		return atom.getArguments();
	}

	public QueryPredicate getPredicate() {
		return atom.getPredicate();
	}

	public Query getQuery() {
		return query;
	}

	public boolean isGround() {
		return atom.isGround();
	}

	@Override
	public String toString() {
		return atom.toString();
	}
}