1 /*******************************************************************************
\r
2 * Copyright (c) 2007, 2010 Association for Decentralized Information Management
\r
3 * in Industry THTH ry.
\r
4 * All rights reserved. This program and the accompanying materials
\r
5 * are made available under the terms of the Eclipse Public License v1.0
\r
6 * which accompanies this distribution, and is available at
\r
7 * http://www.eclipse.org/legal/epl-v10.html
\r
10 * VTT Technical Research Centre of Finland - initial API and implementation
\r
11 *******************************************************************************/
\r
12 package org.simantics.structural2.queries;
\r
14 import gnu.trove.set.hash.THashSet;
\r
16 import java.util.Collection;
\r
17 import java.util.Collections;
\r
18 import java.util.Set;
\r
20 import org.simantics.db.ReadGraph;
\r
21 import org.simantics.db.Resource;
\r
22 import org.simantics.db.exception.DatabaseException;
\r
23 import org.simantics.db.service.CollectionSupport;
\r
24 import org.simantics.structural.stubs.StructuralResource2;
\r
26 public class ConnectionSet {
\r
28 final Set<Resource> connections;
\r
29 THashSet<Resource> joins;
\r
30 THashSet<Resource> excluded;
\r
31 THashSet<Resource> excludedJoins;
\r
33 StructuralResource2 sr;
\r
35 // public ConnectionSet() {
\r
39 public ConnectionSet(ReadGraph graph) {
\r
40 sr = StructuralResource2.getInstance(graph);
\r
41 connections = graph.getService(CollectionSupport.class).createSet();
\r
44 public boolean excludeConnection(Resource connection) {
\r
45 if (excluded == null)
\r
46 excluded = new THashSet<Resource>();
\r
47 return excluded.add(connection);
\r
50 public boolean excludeJoin(Resource join) {
\r
51 if (excludedJoins == null)
\r
52 excludedJoins = new THashSet<Resource>();
\r
53 return excludedJoins.add(join);
\r
56 public boolean excludeConnections(Collection<Resource> connections) {
\r
57 if (connections.isEmpty())
\r
59 if (excluded == null)
\r
60 excluded = new THashSet<Resource>();
\r
61 return excluded.addAll(connections);
\r
64 public boolean isExcluded(Resource connection) {
\r
65 return excluded != null && excluded.contains(connection);
\r
68 public boolean isJoinExcluded(Resource join) {
\r
69 return excludedJoins != null && excludedJoins.contains(join);
\r
72 public void addConnection(ReadGraph g, Resource connection) throws DatabaseException {
\r
73 if(isExcluded(connection))
\r
75 if(connections.add(connection)) {
\r
76 if(sr == null) sr = StructuralResource2.getInstance(g);
\r
77 for(Resource join : g.getObjects(connection, sr.IsJoinedBy))
\r
82 public void addJoin(ReadGraph g, Resource join) throws DatabaseException {
\r
83 if(isJoinExcluded(join))
\r
86 joins = new THashSet<Resource>();
\r
87 if(joins.add(join)) {
\r
88 if(sr == null) sr = StructuralResource2.getInstance(g);
\r
89 for(Resource connection : g.getObjects(join, sr.Joins))
\r
90 addConnection(g, connection);
\r
94 public Set<Resource> getConnections() {
\r
98 public Set<Resource> getJoins() {
\r
99 if(joins == null) return Collections.emptySet();
\r