All Downloads are FREE. Search and download functionalities are using the official Maven repository.

org.eclipse.rdf4j.query.parser.serql.QueryModelBuilder Maven / Gradle / Ivy

There is a newer version: 3.7.7
Show newest version
/*******************************************************************************
 * Copyright (c) 2015 Eclipse RDF4J contributors, Aduna, and others.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Distribution License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/org/documents/edl-v10.php.
 *******************************************************************************/
package org.eclipse.rdf4j.query.parser.serql;

import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;

import org.eclipse.rdf4j.model.IRI;
import org.eclipse.rdf4j.model.Literal;
import org.eclipse.rdf4j.model.Value;
import org.eclipse.rdf4j.model.ValueFactory;
import org.eclipse.rdf4j.model.vocabulary.RDF;
import org.eclipse.rdf4j.query.MalformedQueryException;
import org.eclipse.rdf4j.query.algebra.And;
import org.eclipse.rdf4j.query.algebra.Bound;
import org.eclipse.rdf4j.query.algebra.Compare;
import org.eclipse.rdf4j.query.algebra.Compare.CompareOp;
import org.eclipse.rdf4j.query.algebra.CompareAll;
import org.eclipse.rdf4j.query.algebra.CompareAny;
import org.eclipse.rdf4j.query.algebra.Datatype;
import org.eclipse.rdf4j.query.algebra.Difference;
import org.eclipse.rdf4j.query.algebra.Distinct;
import org.eclipse.rdf4j.query.algebra.Exists;
import org.eclipse.rdf4j.query.algebra.Extension;
import org.eclipse.rdf4j.query.algebra.ExtensionElem;
import org.eclipse.rdf4j.query.algebra.FunctionCall;
import org.eclipse.rdf4j.query.algebra.In;
import org.eclipse.rdf4j.query.algebra.Intersection;
import org.eclipse.rdf4j.query.algebra.IsBNode;
import org.eclipse.rdf4j.query.algebra.IsLiteral;
import org.eclipse.rdf4j.query.algebra.IsResource;
import org.eclipse.rdf4j.query.algebra.IsURI;
import org.eclipse.rdf4j.query.algebra.Label;
import org.eclipse.rdf4j.query.algebra.Lang;
import org.eclipse.rdf4j.query.algebra.LangMatches;
import org.eclipse.rdf4j.query.algebra.Like;
import org.eclipse.rdf4j.query.algebra.LocalName;
import org.eclipse.rdf4j.query.algebra.Namespace;
import org.eclipse.rdf4j.query.algebra.Not;
import org.eclipse.rdf4j.query.algebra.Or;
import org.eclipse.rdf4j.query.algebra.Order;
import org.eclipse.rdf4j.query.algebra.OrderElem;
import org.eclipse.rdf4j.query.algebra.Projection;
import org.eclipse.rdf4j.query.algebra.ProjectionElem;
import org.eclipse.rdf4j.query.algebra.ProjectionElemList;
import org.eclipse.rdf4j.query.algebra.Reduced;
import org.eclipse.rdf4j.query.algebra.Regex;
import org.eclipse.rdf4j.query.algebra.SameTerm;
import org.eclipse.rdf4j.query.algebra.SingletonSet;
import org.eclipse.rdf4j.query.algebra.Slice;
import org.eclipse.rdf4j.query.algebra.StatementPattern;
import org.eclipse.rdf4j.query.algebra.Str;
import org.eclipse.rdf4j.query.algebra.TupleExpr;
import org.eclipse.rdf4j.query.algebra.Union;
import org.eclipse.rdf4j.query.algebra.ValueConstant;
import org.eclipse.rdf4j.query.algebra.ValueExpr;
import org.eclipse.rdf4j.query.algebra.Var;
import org.eclipse.rdf4j.query.parser.serql.ast.ASTAnd;
import org.eclipse.rdf4j.query.parser.serql.ast.ASTBNode;
import org.eclipse.rdf4j.query.parser.serql.ast.ASTBasicPathExpr;
import org.eclipse.rdf4j.query.parser.serql.ast.ASTBasicPathExprTail;
import org.eclipse.rdf4j.query.parser.serql.ast.ASTBooleanConstant;
import org.eclipse.rdf4j.query.parser.serql.ast.ASTBooleanExpr;
import org.eclipse.rdf4j.query.parser.serql.ast.ASTBound;
import org.eclipse.rdf4j.query.parser.serql.ast.ASTCompare;
import org.eclipse.rdf4j.query.parser.serql.ast.ASTCompareAll;
import org.eclipse.rdf4j.query.parser.serql.ast.ASTCompareAny;
import org.eclipse.rdf4j.query.parser.serql.ast.ASTConstruct;
import org.eclipse.rdf4j.query.parser.serql.ast.ASTConstructQuery;
import org.eclipse.rdf4j.query.parser.serql.ast.ASTDatatype;
import org.eclipse.rdf4j.query.parser.serql.ast.ASTEdge;
import org.eclipse.rdf4j.query.parser.serql.ast.ASTExists;
import org.eclipse.rdf4j.query.parser.serql.ast.ASTFrom;
import org.eclipse.rdf4j.query.parser.serql.ast.ASTFunctionCall;
import org.eclipse.rdf4j.query.parser.serql.ast.ASTGraphIntersect;
import org.eclipse.rdf4j.query.parser.serql.ast.ASTGraphMinus;
import org.eclipse.rdf4j.query.parser.serql.ast.ASTGraphUnion;
import org.eclipse.rdf4j.query.parser.serql.ast.ASTIn;
import org.eclipse.rdf4j.query.parser.serql.ast.ASTInList;
import org.eclipse.rdf4j.query.parser.serql.ast.ASTIsBNode;
import org.eclipse.rdf4j.query.parser.serql.ast.ASTIsLiteral;
import org.eclipse.rdf4j.query.parser.serql.ast.ASTIsResource;
import org.eclipse.rdf4j.query.parser.serql.ast.ASTIsURI;
import org.eclipse.rdf4j.query.parser.serql.ast.ASTLabel;
import org.eclipse.rdf4j.query.parser.serql.ast.ASTLang;
import org.eclipse.rdf4j.query.parser.serql.ast.ASTLangMatches;
import org.eclipse.rdf4j.query.parser.serql.ast.ASTLike;
import org.eclipse.rdf4j.query.parser.serql.ast.ASTLimit;
import org.eclipse.rdf4j.query.parser.serql.ast.ASTLiteral;
import org.eclipse.rdf4j.query.parser.serql.ast.ASTLocalName;
import org.eclipse.rdf4j.query.parser.serql.ast.ASTNamespace;
import org.eclipse.rdf4j.query.parser.serql.ast.ASTNode;
import org.eclipse.rdf4j.query.parser.serql.ast.ASTNodeElem;
import org.eclipse.rdf4j.query.parser.serql.ast.ASTNot;
import org.eclipse.rdf4j.query.parser.serql.ast.ASTNull;
import org.eclipse.rdf4j.query.parser.serql.ast.ASTOffset;
import org.eclipse.rdf4j.query.parser.serql.ast.ASTOptPathExpr;
import org.eclipse.rdf4j.query.parser.serql.ast.ASTOptPathExprTail;
import org.eclipse.rdf4j.query.parser.serql.ast.ASTOr;
import org.eclipse.rdf4j.query.parser.serql.ast.ASTOrderBy;
import org.eclipse.rdf4j.query.parser.serql.ast.ASTOrderExpr;
import org.eclipse.rdf4j.query.parser.serql.ast.ASTPathExpr;
import org.eclipse.rdf4j.query.parser.serql.ast.ASTPathExprTail;
import org.eclipse.rdf4j.query.parser.serql.ast.ASTPathExprUnion;
import org.eclipse.rdf4j.query.parser.serql.ast.ASTProjectionElem;
import org.eclipse.rdf4j.query.parser.serql.ast.ASTQueryBody;
import org.eclipse.rdf4j.query.parser.serql.ast.ASTQueryContainer;
import org.eclipse.rdf4j.query.parser.serql.ast.ASTRegex;
import org.eclipse.rdf4j.query.parser.serql.ast.ASTReifiedStat;
import org.eclipse.rdf4j.query.parser.serql.ast.ASTSameTerm;
import org.eclipse.rdf4j.query.parser.serql.ast.ASTSelect;
import org.eclipse.rdf4j.query.parser.serql.ast.ASTSelectQuery;
import org.eclipse.rdf4j.query.parser.serql.ast.ASTStr;
import org.eclipse.rdf4j.query.parser.serql.ast.ASTString;
import org.eclipse.rdf4j.query.parser.serql.ast.ASTTupleIntersect;
import org.eclipse.rdf4j.query.parser.serql.ast.ASTTupleMinus;
import org.eclipse.rdf4j.query.parser.serql.ast.ASTTupleUnion;
import org.eclipse.rdf4j.query.parser.serql.ast.ASTURI;
import org.eclipse.rdf4j.query.parser.serql.ast.ASTValueExpr;
import org.eclipse.rdf4j.query.parser.serql.ast.ASTVar;
import org.eclipse.rdf4j.query.parser.serql.ast.ASTWhere;
import org.eclipse.rdf4j.query.parser.serql.ast.Node;
import org.eclipse.rdf4j.query.parser.serql.ast.VisitorException;

class QueryModelBuilder extends AbstractASTVisitor {

	public static TupleExpr buildQueryModel(ASTQueryContainer node, ValueFactory valueFactory)
			throws MalformedQueryException {
		try {
			QueryModelBuilder qmBuilder = new QueryModelBuilder(valueFactory);
			return (TupleExpr) node.jjtAccept(qmBuilder, null);
		} catch (VisitorException e) {
			throw new MalformedQueryException(e.getMessage(), e);
		}
	}

	/*-----------*
	 * Variables *
	 *-----------*/

	private final ValueFactory valueFactory;

	private int constantVarID = 1;

	private GraphPattern graphPattern;

	/*--------------*
	 * Constructors *
	 *--------------*/

	public QueryModelBuilder(ValueFactory valueFactory) {
		this.valueFactory = valueFactory;
	}

	/*---------*
	 * Methods *
	 *---------*/

	private Var createConstantVar(Value value) {
		Var var = new Var("-const-" + constantVarID++);
		var.setAnonymous(true);
		var.setValue(value);
		return var;
	}

	private GraphPattern parseGraphPattern(Node node) throws VisitorException {
		graphPattern = new GraphPattern(graphPattern);
		try {
			node.jjtAccept(this, null);
			return graphPattern;
		} finally {
			graphPattern = graphPattern.getParent();
		}
	}

	@Override
	public TupleExpr visit(ASTQueryContainer node, Object data) throws VisitorException {
		// Skip the namespace declarations, any information it contains should
		// already have been processed
		return (TupleExpr) node.getQuery().jjtAccept(this, null);
	}

	@Override
	public TupleExpr visit(ASTTupleUnion node, Object data) throws VisitorException {
		TupleExpr leftArg = (TupleExpr) node.getLeftArg().jjtAccept(this, null);
		TupleExpr rightArg = (TupleExpr) node.getRightArg().jjtAccept(this, null);

		TupleExpr result = new Union(leftArg, rightArg);

		if (node.isDistinct()) {
			result = new Distinct(result);
		}

		return result;
	}

	@Override
	public TupleExpr visit(ASTTupleMinus node, Object data) throws VisitorException {
		TupleExpr leftArg = (TupleExpr) node.getLeftArg().jjtAccept(this, null);
		TupleExpr rightArg = (TupleExpr) node.getRightArg().jjtAccept(this, null);

		return new Difference(leftArg, rightArg);
	}

	@Override
	public TupleExpr visit(ASTTupleIntersect node, Object data) throws VisitorException {
		TupleExpr leftArg = (TupleExpr) node.getLeftArg().jjtAccept(this, null);
		TupleExpr rightArg = (TupleExpr) node.getRightArg().jjtAccept(this, null);

		return new Intersection(leftArg, rightArg);
	}

	@Override
	public TupleExpr visit(ASTGraphUnion node, Object data) throws VisitorException {
		TupleExpr leftArg = (TupleExpr) node.getLeftArg().jjtAccept(this, null);
		TupleExpr rightArg = (TupleExpr) node.getRightArg().jjtAccept(this, null);

		TupleExpr result = new Union(leftArg, rightArg);

		if (node.isDistinct()) {
			result = new Distinct(result);
		}

		return result;
	}

	@Override
	public TupleExpr visit(ASTGraphMinus node, Object data) throws VisitorException {
		TupleExpr leftArg = (TupleExpr) node.getLeftArg().jjtAccept(this, null);
		TupleExpr rightArg = (TupleExpr) node.getRightArg().jjtAccept(this, null);

		return new Difference(leftArg, rightArg);
	}

	@Override
	public TupleExpr visit(ASTGraphIntersect node, Object data) throws VisitorException {
		TupleExpr leftArg = (TupleExpr) node.getLeftArg().jjtAccept(this, null);
		TupleExpr rightArg = (TupleExpr) node.getRightArg().jjtAccept(this, null);

		return new Intersection(leftArg, rightArg);
	}

	@Override
	public TupleExpr visit(ASTSelectQuery node, Object data) throws VisitorException {
		TupleExpr tupleExpr;

		ASTQueryBody queryBodyNode = node.getQueryBody();

		if (queryBodyNode != null) {
			// Build tuple expression for query body
			tupleExpr = (TupleExpr) queryBodyNode.jjtAccept(this, null);
		} else {
			tupleExpr = new SingletonSet();
		}

		// Apply result ordering
		ASTOrderBy orderByNode = node.getOrderBy();
		if (orderByNode != null) {
			List orderElemements = (List) orderByNode.jjtAccept(this, null);
			tupleExpr = new Order(tupleExpr, orderElemements);
		}

		// Apply projection
		tupleExpr = (TupleExpr) node.getSelectClause().jjtAccept(this, tupleExpr);

		// process limit and offset clauses, if present.
		ASTLimit limitNode = node.getLimit();
		int limit = -1;
		if (limitNode != null) {
			limit = (Integer) limitNode.jjtAccept(this, null);
		}

		ASTOffset offsetNode = node.getOffset();
		int offset = -1;
		if (offsetNode != null) {
			offset = (Integer) offsetNode.jjtAccept(this, null);
		}

		if (offset >= 1 || limit >= 0) {
			tupleExpr = new Slice(tupleExpr, offset, limit);
		}
		return tupleExpr;
	}

	@Override
	public TupleExpr visit(ASTSelect node, Object data) throws VisitorException {
		TupleExpr result = (TupleExpr) data;

		Extension extension = new Extension();
		ProjectionElemList projElemList = new ProjectionElemList();

		for (ASTProjectionElem projElemNode : node.getProjectionElemList()) {
			ValueExpr valueExpr = (ValueExpr) projElemNode.getValueExpr().jjtAccept(this, null);

			String alias = projElemNode.getAlias();
			if (alias != null) {
				// aliased projection element
				extension.addElement(new ExtensionElem(valueExpr, alias));
				projElemList.addElement(new ProjectionElem(alias));
			} else if (valueExpr instanceof Var) {
				// unaliased variable
				Var projVar = (Var) valueExpr;
				projElemList.addElement(new ProjectionElem(projVar.getName()));
			} else {
				throw new IllegalStateException("required alias for non-Var projection elements not found");
			}
		}

		if (!extension.getElements().isEmpty()) {
			extension.setArg(result);
			result = extension;
		}

		result = new Projection(result, projElemList);

		if (node.isDistinct()) {
			result = new Distinct(result);
		} else if (node.isReduced()) {
			result = new Reduced(result);
		}

		return result;
	}

	@Override
	public TupleExpr visit(ASTConstructQuery node, Object data) throws VisitorException {
		TupleExpr tupleExpr;

		if (node.hasQueryBody()) {
			// Build tuple expression for query body
			tupleExpr = (TupleExpr) node.getQueryBody().jjtAccept(this, null);
		} else {
			tupleExpr = new SingletonSet();
		}

		// Apply result ordering
		ASTOrderBy orderByNode = node.getOrderBy();
		if (orderByNode != null) {
			List orderElemements = (List) orderByNode.jjtAccept(this, null);
			tupleExpr = new Order(tupleExpr, orderElemements);
		}

		// Create constructor
		ConstructorBuilder cb = new ConstructorBuilder();
		ASTConstruct constructNode = node.getConstructClause();

		if (!constructNode.isWildcard()) {
			TupleExpr constructExpr = (TupleExpr) constructNode.jjtAccept(this, null);
			tupleExpr = cb.buildConstructor(tupleExpr, constructExpr, constructNode.isDistinct(),
					constructNode.isReduced());
		} else if (node.hasQueryBody()) {
			tupleExpr = cb.buildConstructor(tupleExpr, constructNode.isDistinct(), constructNode.isReduced());
		}
		// else: "construct *" without query body, just return the SingletonSet

		// process limit and offset clauses, if present.
		ASTLimit limitNode = node.getLimit();
		int limit = -1;
		if (limitNode != null) {
			limit = (Integer) limitNode.jjtAccept(this, null);
		}

		ASTOffset offsetNode = node.getOffset();
		int offset = -1;

		if (offsetNode != null) {
			offset = (Integer) offsetNode.jjtAccept(this, null);
		}

		if (offset >= 1 || limit >= 0) {
			tupleExpr = new Slice(tupleExpr, offset, limit);
		}

		return tupleExpr;
	}

	@Override
	public TupleExpr visit(ASTConstruct node, Object data) throws VisitorException {
		assert !node.isWildcard() : "Cannot build constructor for wildcards";

		return parseGraphPattern(node.getPathExpr()).buildTupleExpr();
	}

	@Override
	public TupleExpr visit(ASTQueryBody node, Object data) throws VisitorException {
		graphPattern = new GraphPattern(graphPattern);
		try {
			super.visit(node, data);
			return graphPattern.buildTupleExpr();
		} finally {
			graphPattern = graphPattern.getParent();
		}
	}

	@Override
	public Object visit(ASTFrom node, Object data) throws VisitorException {
		StatementPattern.Scope scope = StatementPattern.Scope.DEFAULT_CONTEXTS;
		Var contextVar = null;

		if (node.hasContextID()) {
			scope = StatementPattern.Scope.NAMED_CONTEXTS;
			ValueExpr contextID = (ValueExpr) node.getContextID().jjtAccept(this, null);

			if (contextID instanceof Var) {
				contextVar = (Var) contextID;
			} else if (contextID instanceof ValueConstant) {
				ValueConstant vc = (ValueConstant) contextID;
				contextVar = createConstantVar(vc.getValue());
			} else {
				throw new IllegalArgumentException("Unexpected contextID result type: " + contextID.getClass());
			}
		}

		graphPattern.setStatementPatternScope(scope);
		graphPattern.setContextVar(contextVar);

		node.getPathExpr().jjtAccept(this, null);

		return null;
	}

	@Override
	public Object visit(ASTWhere node, Object data) throws VisitorException {
		ValueExpr valueExpr = (ValueExpr) node.getCondition().jjtAccept(this, null);
		graphPattern.addConstraint(valueExpr);
		return null;
	}

	@Override
	public List visit(ASTOrderBy node, Object data) throws VisitorException {
		List orderExprList = node.getOrderExprList();

		List elements = new ArrayList<>(orderExprList.size());

		for (ASTOrderExpr orderExpr : orderExprList) {
			elements.add((OrderElem) orderExpr.jjtAccept(this, null));
		}

		return elements;
	}

	@Override
	public OrderElem visit(ASTOrderExpr node, Object data) throws VisitorException {
		ValueExpr valueExpr = (ValueExpr) node.getValueExpr().jjtAccept(this, null);
		return new OrderElem(valueExpr, node.isAscending());
	}

	@Override
	public Integer visit(ASTLimit node, Object data) throws VisitorException {
		return node.getValue();
	}

	@Override
	public Integer visit(ASTOffset node, Object data) throws VisitorException {
		return node.getValue();
	}

	@Override
	public Object visit(ASTPathExprUnion node, Object data) throws VisitorException {
		Iterator args = node.getPathExprList().iterator();

		// Create new sub-graph pattern for optional path expressions
		TupleExpr unionExpr = parseGraphPattern(args.next()).buildTupleExpr();

		while (args.hasNext()) {
			TupleExpr argExpr = parseGraphPattern(args.next()).buildTupleExpr();
			unionExpr = new Union(unionExpr, argExpr);
		}

		graphPattern.addRequiredTE(unionExpr);

		return null;
	}

	@Override
	public Object visit(ASTBasicPathExpr node, Object data) throws VisitorException {
		// process subject node
		List subjVars = (List) node.getHead().jjtAccept(this, null);

		// supply subject vars to tail segment
		node.getTail().jjtAccept(this, subjVars);

		return null;
	}

	@Override
	public Object visit(ASTOptPathExpr node, Object data) throws VisitorException {
		// Create new sub-graph pattern for optional path expressions
		graphPattern = new GraphPattern(graphPattern);

		super.visit(node, data);

		graphPattern.getParent().addOptionalTE(graphPattern);
		graphPattern = graphPattern.getParent();

		return null;
	}

	@Override
	public Object visit(ASTBasicPathExprTail tailNode, Object data) throws VisitorException {
		List subjVars = (List) data;
		Var predVar = (Var) tailNode.getEdge().jjtAccept(this, null);
		List objVars = (List) tailNode.getNode().jjtAccept(this, null);

		Var contextVar = graphPattern.getContextVar();
		StatementPattern.Scope spScope = graphPattern.getStatementPatternScope();

		for (Var subjVar : subjVars) {
			for (Var objVar : objVars) {
				StatementPattern sp = new StatementPattern(spScope, subjVar, predVar, objVar, contextVar);
				graphPattern.addRequiredTE(sp);
			}
		}

		// Process next tail segment
		ASTPathExprTail nextTailNode = tailNode.getNextTail();
		if (nextTailNode != null) {
			List joinVars = nextTailNode.isBranch() ? subjVars : objVars;
			nextTailNode.jjtAccept(this, joinVars);
		}

		return null;
	}

	@Override
	public Object visit(ASTOptPathExprTail tailNode, Object data) throws VisitorException {
		List subjVars = (List) data;

		// Create new sub-graph pattern for optional path expressions
		graphPattern = new GraphPattern(graphPattern);

		// optional path expression tail
		tailNode.getOptionalTail().jjtAccept(this, subjVars);

		ASTWhere whereNode = tailNode.getWhereClause();
		if (whereNode != null) {
			// boolean contraint on optional path expression tail
			whereNode.jjtAccept(this, null);
		}

		graphPattern.getParent().addOptionalTE(graphPattern);
		graphPattern = graphPattern.getParent();

		ASTPathExprTail nextTailNode = tailNode.getNextTail();
		if (nextTailNode != null) {
			// branch after optional path expression tail
			nextTailNode.jjtAccept(this, subjVars);
		}

		return null;
	}

	@Override
	public Var visit(ASTEdge node, Object data) throws VisitorException {
		ValueExpr arg = (ValueExpr) node.getValueExpr().jjtAccept(this, null);

		if (arg instanceof Var) {
			return (Var) arg;
		} else if (arg instanceof ValueConstant) {
			ValueConstant vc = (ValueConstant) arg;
			return createConstantVar(vc.getValue());
		} else {
			throw new IllegalArgumentException("Unexpected edge argument type: " + arg.getClass());
		}
	}

	@Override
	public List visit(ASTNode node, Object data) throws VisitorException {
		List nodeVars = new ArrayList<>();

		for (ASTNodeElem nodeElem : node.getNodeElemList()) {
			Var nodeVar = (Var) nodeElem.jjtAccept(this, null);
			nodeVars.add(nodeVar);
		}

		// Create any implicit unequalities
		for (int i = 0; i < nodeVars.size() - 1; i++) {
			Var var1 = nodeVars.get(i);

			for (int j = i + 1; j < nodeVars.size(); j++) {
				Var var2 = nodeVars.get(j);

				// At least one of the variables should be non-constant
				// for the unequality to make any sense:
				if (!var1.hasValue() || !var2.hasValue()) {
					graphPattern.addConstraint(new Not(new SameTerm(var1, var2)));
				}
			}
		}

		return nodeVars;
	}

	@Override
	public Var visit(ASTNodeElem node, Object data) throws VisitorException {
		ValueExpr valueExpr = (ValueExpr) node.getChild().jjtAccept(this, null);

		if (valueExpr instanceof Var) {
			return (Var) valueExpr;
		} else if (valueExpr instanceof ValueConstant) {
			ValueConstant vc = (ValueConstant) valueExpr;
			return createConstantVar(vc.getValue());
		} else {
			throw new IllegalArgumentException("Unexpected node element result type: " + valueExpr.getClass());
		}
	}

	@Override
	public Var visit(ASTReifiedStat node, Object data) throws VisitorException {
		assert node.getID() != null : "ID variable not set";

		Var subjVar = (Var) node.getSubject().jjtAccept(this, null);
		Var predVar = (Var) node.getPredicate().jjtAccept(this, null);
		Var objVar = (Var) node.getObject().jjtAccept(this, null);
		Var idVar = (Var) node.getID().jjtAccept(this, null);

		Var contextVar = graphPattern.getContextVar();
		StatementPattern.Scope spScope = graphPattern.getStatementPatternScope();

		Var rdfType = new Var("_rdfType", RDF.TYPE);
		Var rdfStatement = new Var("_rdfStatement", RDF.STATEMENT);
		Var rdfSubject = new Var("_rdfSubject", RDF.SUBJECT);
		Var rdfPredicate = new Var("_rdfPredicate", RDF.PREDICATE);
		Var rdfObject = new Var("_rdfObject", RDF.OBJECT);

		graphPattern.addRequiredTE(new StatementPattern(spScope, idVar, rdfType, rdfStatement, contextVar));
		graphPattern.addRequiredTE(new StatementPattern(spScope, idVar, rdfSubject, subjVar, contextVar));
		graphPattern.addRequiredTE(new StatementPattern(spScope, idVar, rdfPredicate, predVar, contextVar));
		graphPattern.addRequiredTE(new StatementPattern(spScope, idVar, rdfObject, objVar, contextVar));

		return idVar;
	}

	@Override
	public ValueExpr visit(ASTOr node, Object data) throws VisitorException {
		Iterator iter = node.getOperandList().iterator();

		ValueExpr result = (ValueExpr) iter.next().jjtAccept(this, null);

		while (iter.hasNext()) {
			ValueExpr operand = (ValueExpr) iter.next().jjtAccept(this, null);
			result = new Or(result, operand);
		}

		return result;
	}

	@Override
	public ValueExpr visit(ASTAnd node, Object data) throws VisitorException {
		Iterator iter = node.getOperandList().iterator();

		ValueExpr result = (ValueExpr) iter.next().jjtAccept(this, null);

		while (iter.hasNext()) {
			ValueExpr operand = (ValueExpr) iter.next().jjtAccept(this, null);
			result = new And(result, operand);
		}

		return result;
	}

	@Override
	public ValueConstant visit(ASTBooleanConstant node, Object data) throws VisitorException {
		return new ValueConstant(valueFactory.createLiteral(node.getValue()));
	}

	@Override
	public Not visit(ASTNot node, Object data) throws VisitorException {
		return new Not((ValueExpr) super.visit(node, data));
	}

	@Override
	public Bound visit(ASTBound node, Object data) throws VisitorException {
		return new Bound((Var) super.visit(node, data));
	}

	@Override
	public IsResource visit(ASTIsResource node, Object data) throws VisitorException {
		return new IsResource((ValueExpr) super.visit(node, data));
	}

	@Override
	public IsLiteral visit(ASTIsLiteral node, Object data) throws VisitorException {
		return new IsLiteral((ValueExpr) super.visit(node, data));
	}

	@Override
	public IsURI visit(ASTIsURI node, Object data) throws VisitorException {
		return new IsURI((ValueExpr) super.visit(node, data));
	}

	@Override
	public IsBNode visit(ASTIsBNode node, Object data) throws VisitorException {
		return new IsBNode((ValueExpr) super.visit(node, data));
	}

	@Override
	public LangMatches visit(ASTLangMatches node, Object data) throws VisitorException {
		ValueExpr tag = (ValueExpr) node.getLanguageTag().jjtAccept(this, null);
		ValueExpr range = (ValueExpr) node.getLanguageRange().jjtAccept(this, null);
		return new LangMatches(tag, range);
	}

	@Override
	public Exists visit(ASTExists node, Object data) throws VisitorException {
		return new Exists((TupleExpr) super.visit(node, data));
	}

	@Override
	public SameTerm visit(ASTSameTerm node, Object data) throws VisitorException {
		ValueExpr leftArg = (ValueExpr) node.getLeftOperand().jjtAccept(this, null);
		ValueExpr rightArg = (ValueExpr) node.getRightOperand().jjtAccept(this, null);
		return new SameTerm(leftArg, rightArg);
	}

	@Override
	public Compare visit(ASTCompare node, Object data) throws VisitorException {
		ValueExpr leftArg = (ValueExpr) node.getLeftOperand().jjtAccept(this, null);
		ValueExpr rightArg = (ValueExpr) node.getRightOperand().jjtAccept(this, null);
		CompareOp operator = node.getOperator().getValue();

		return new Compare(leftArg, rightArg, operator);
	}

	@Override
	public CompareAny visit(ASTCompareAny node, Object data) throws VisitorException {
		ValueExpr valueExpr = (ValueExpr) node.getLeftOperand().jjtAccept(this, null);
		TupleExpr tupleExpr = (TupleExpr) node.getRightOperand().jjtAccept(this, null);
		CompareOp op = node.getOperator().getValue();

		return new CompareAny(valueExpr, tupleExpr, op);
	}

	@Override
	public CompareAll visit(ASTCompareAll node, Object data) throws VisitorException {
		ValueExpr valueExpr = (ValueExpr) node.getLeftOperand().jjtAccept(this, null);
		TupleExpr tupleExpr = (TupleExpr) node.getRightOperand().jjtAccept(this, null);
		CompareOp op = node.getOperator().getValue();

		return new CompareAll(valueExpr, tupleExpr, op);
	}

	@Override
	public Like visit(ASTLike node, Object data) throws VisitorException {
		ValueExpr expr = (ValueExpr) node.getValueExpr().jjtAccept(this, null);
		String pattern = (String) node.getPattern().jjtAccept(this, null);
		boolean caseSensitive = !node.ignoreCase();

		return new Like(expr, pattern, caseSensitive);
	}

	@Override
	public Regex visit(ASTRegex node, Object data) throws VisitorException {
		ValueExpr text = (ValueExpr) node.getText().jjtAccept(this, null);
		ValueExpr pattern = (ValueExpr) node.getPattern().jjtAccept(this, null);
		ValueExpr flags = null;
		if (node.hasFlags()) {
			flags = (ValueExpr) node.getFlags().jjtAccept(this, null);
		}

		return new Regex(text, pattern, flags);
	}

	@Override
	public In visit(ASTIn node, Object data) throws VisitorException {
		ValueExpr valueExpr = (ValueExpr) node.getLeftOperand().jjtAccept(this, null);
		TupleExpr tupleExpr = (TupleExpr) node.getRightOperand().jjtAccept(this, null);
		return new In(valueExpr, tupleExpr);
	}

	@Override
	public ValueExpr visit(ASTInList node, Object data) throws VisitorException {
		ValueExpr leftArg = (ValueExpr) node.getValueExpr().jjtAccept(this, null);

		ValueExpr result = null;

		for (ASTValueExpr argExpr : node.getArgList().getElements()) {
			ValueExpr rightArg = (ValueExpr) argExpr.jjtAccept(this, null);

			if (result == null) {
				// First argument
				result = new SameTerm(leftArg, rightArg);
			} else {
				SameTerm sameTerm = new SameTerm(leftArg.clone(), rightArg);
				result = new Or(result, sameTerm);
			}
		}

		assert result != null;

		return result;
	}

	@Override
	public Var visit(ASTVar node, Object data) throws VisitorException {
		Var var = new Var(node.getName());
		var.setAnonymous(node.isAnonymous());
		return var;
	}

	@Override
	public Datatype visit(ASTDatatype node, Object data) throws VisitorException {
		return new Datatype((ValueExpr) super.visit(node, data));
	}

	@Override
	public Lang visit(ASTLang node, Object data) throws VisitorException {
		return new Lang((ValueExpr) super.visit(node, data));
	}

	@Override
	public Label visit(ASTLabel node, Object data) throws VisitorException {
		return new Label((ValueExpr) super.visit(node, data));
	}

	@Override
	public Namespace visit(ASTNamespace node, Object data) throws VisitorException {
		return new Namespace((ValueExpr) super.visit(node, data));
	}

	@Override
	public LocalName visit(ASTLocalName node, Object data) throws VisitorException {
		return new LocalName((ValueExpr) super.visit(node, data));
	}

	@Override
	public Str visit(ASTStr node, Object data) throws VisitorException {
		return new Str((ValueExpr) super.visit(node, data));
	}

	@Override
	public FunctionCall visit(ASTFunctionCall node, Object data) throws VisitorException {
		ValueConstant vc = (ValueConstant) node.getURI().jjtAccept(this, null);
		assert vc.getValue() instanceof IRI;

		FunctionCall functionCall = new FunctionCall(vc.getValue().toString());

		for (ASTValueExpr argExpr : node.getArgList().getElements()) {
			functionCall.addArg((ValueExpr) argExpr.jjtAccept(this, null));
		}

		return functionCall;
	}

	@Override
	public Object visit(ASTNull node, Object data) throws VisitorException {
		throw new VisitorException("Use of NULL values in SeRQL queries has been deprecated, use BOUND(...) instead");
	}

	@Override
	public ValueConstant visit(ASTURI node, Object data) throws VisitorException {
		return new ValueConstant(valueFactory.createIRI(node.getValue()));
	}

	@Override
	public ValueConstant visit(ASTBNode node, Object data) throws VisitorException {
		return new ValueConstant(valueFactory.createBNode(node.getID()));
	}

	@Override
	public ValueConstant visit(ASTLiteral litNode, Object data) throws VisitorException {
		IRI datatype = null;

		// Get datatype URI from child URI node, if present
		ASTValueExpr dtNode = litNode.getDatatypeNode();
		if (dtNode instanceof ASTURI) {
			datatype = valueFactory.createIRI(((ASTURI) dtNode).getValue());
		} else if (dtNode != null) {
			throw new IllegalArgumentException("Unexpected datatype type: " + dtNode.getClass());
		}

		Literal literal;
		if (datatype != null) {
			literal = valueFactory.createLiteral(litNode.getLabel(), datatype);
		} else if (litNode.hasLang()) {
			literal = valueFactory.createLiteral(litNode.getLabel(), litNode.getLang());
		} else {
			literal = valueFactory.createLiteral(litNode.getLabel());
		}

		return new ValueConstant(literal);
	}

	@Override
	public String visit(ASTString node, Object data) throws VisitorException {
		return node.getValue();
	}
}




© 2015 - 2025 Weber Informatics LLC | Privacy Policy