/******************************************************************************* * Copyright (c) 2007, 2010 Association for Decentralized Information Management * in Industry THTH ry. * All rights reserved. This program and the accompanying materials * are made available under the terms of the Eclipse Public License v1.0 * which accompanies this distribution, and is available at * http://www.eclipse.org/legal/epl-v10.html * * Contributors: * VTT Technical Research Centre of Finland - initial API and implementation *******************************************************************************/ package org.simantics.mapping.constraint.instructions; import gnu.trove.set.hash.TIntHashSet; import java.util.Collection; import java.util.Iterator; import org.simantics.db.ReadGraph; import org.simantics.db.Resource; import org.simantics.db.common.utils.NameUtils; import org.simantics.db.exception.DatabaseException; import org.simantics.layer0.utils.binaryPredicates.IBinaryPredicate; public class BinaryPredicateObjectQuery extends BinaryPredicateInstruction { public BinaryPredicateObjectQuery(int variable0, int variable1, IBinaryPredicate predicate) { super(variable0, variable1, predicate); } @Override public Object query(ReadGraph g, Object[] bindings) throws DatabaseException { Resource r0 = (Resource)bindings[variable0]; // ITask task = ThreadLogger.getInstance().begin("g"); Collection result = predicate.getObjects(g, r0); if(DEBUG) { System.out.println("getObjects(" + NameUtils.getSafeName(g, r0) + ", " + predicate.toString(g) + ") returned " + result.size() + " objects" ); } // task.finish(); if(result.isEmpty()) return IInstruction.FAILURE; Iterator it = result.iterator(); bindings[variable1] = it.next(); if(it.hasNext()) return it; else return null; } @SuppressWarnings("unchecked") @Override public Object next(ReadGraph g, Object[] bindings, Object continuation) { Iterator it = (Iterator)continuation; if(it.hasNext()) { bindings[variable1] = it.next(); if(it.hasNext()) return it; else return null; } else return IInstruction.FAILURE; } @Override public void collectVariables(TIntHashSet reads, TIntHashSet writes) { super.collectVariables(reads, writes); writes.add(variable1); } }