Fix diagram profiles to work with latest DB changes
[simantics/platform.git] / bundles / org.simantics.diagram / src / org / simantics / diagram / adapter / TypeGroup.java
1 /*******************************************************************************
2  * Copyright (c) 2007, 2018 Association for Decentralized Information Management
3  * in Industry THTH ry.
4  * All rights reserved. This program and the accompanying materials
5  * are made available under the terms of the Eclipse Public License v1.0
6  * which accompanies this distribution, and is available at
7  * http://www.eclipse.org/legal/epl-v10.html
8  *
9  * Contributors:
10  *     VTT Technical Research Centre of Finland - initial API and implementation
11  *******************************************************************************/
12 package org.simantics.diagram.adapter;
13
14 import java.util.Arrays;
15 import java.util.Collection;
16 import java.util.Collections;
17 import java.util.HashSet;
18 import java.util.Set;
19
20 import org.simantics.databoard.Bindings;
21 import org.simantics.db.ReadGraph;
22 import org.simantics.db.RequestProcessor;
23 import org.simantics.db.Resource;
24 import org.simantics.db.common.primitiverequest.OrderedSet;
25 import org.simantics.db.common.procedure.wrapper.SetListenerToSingleSetListener;
26 import org.simantics.db.common.request.BinaryRead;
27 import org.simantics.db.exception.DatabaseException;
28 import org.simantics.db.procedure.SetListener;
29 import org.simantics.diagram.stubs.DiagramResource;
30 import org.simantics.layer0.Layer0;
31 import org.simantics.scenegraph.profile.Group;
32
33 /**
34  * @author Tuukka Lehtonen
35  */
36 public class TypeGroup implements Group {
37
38     private final Collection<Resource> types;
39     private final String               name;
40
41     public TypeGroup(String name, Resource type) {
42         this(name, new Resource[] { type });
43     }
44
45     public TypeGroup(String name, Resource... types) {
46         this.types = Arrays.asList(types);
47         this.name = name;
48     }
49
50     @Override
51     public int hashCode() {
52         final int prime = 31;
53         int result = 1;
54         result = prime * result + types.hashCode();
55         return result;
56     }
57
58     @Override
59     public boolean equals(Object obj) {
60         if (this == obj)
61             return true;
62         if (obj == null)
63             return false;
64         if (getClass() != obj.getClass())
65             return false;
66         TypeGroup other = (TypeGroup) obj;
67         return types.equals(other.types);
68     }
69
70     /**
71      * Initialize TypeGroup from a DIAGRAM.Group instance.
72      * 
73      * @param graph
74      * @param group
75      * @throws DatabaseException
76      */
77     public TypeGroup(ReadGraph graph, final Resource group) throws DatabaseException {
78         Layer0 L0 = Layer0.getInstance(graph);
79         DiagramResource DIA = DiagramResource.getInstance(graph);
80
81         this.name = graph.getPossibleRelatedValue(group, L0.HasName, Bindings.STRING);
82         this.types = graph.getObjects(group, DIA.TypeGroup_HasType);
83     }
84
85     @Override
86     public void trackItems(RequestProcessor processor, final Resource runtimeDiagram, final SetListener<Resource> listener) throws DatabaseException {
87         if (types.isEmpty()) {
88             System.out.println("TypeGroup has no types!");
89             return;
90         }
91
92         processor.syncRequest(new BinaryRead<Resource, Collection<Resource>, Collection<Resource>>(runtimeDiagram, types) {
93
94             @Override
95             public Set<Resource> perform(ReadGraph graph) throws DatabaseException {
96                 HashSet<Resource> result = new HashSet<Resource>();
97
98                 Resource realDiagram = graph.getPossibleObject(runtimeDiagram, DiagramResource.getInstance(graph).RuntimeDiagram_HasConfiguration);
99                 if (realDiagram == null)
100                     return result;
101
102 //                System.out.println("looking for diagram elements of type:");
103 //                for (Resource t : types)
104 //                    System.out.println("\t" + NameUtils.getSafeName(graph, t, true));
105
106                 Collection<Resource> elements = graph.syncRequest(new OrderedSet(realDiagram));
107                 for (Resource element : elements) {
108 //                    System.out.println("checking element " + NameUtils.getSafeName(graph, element, true));
109                     Collection<Resource> elementTypes = graph.getTypes(element);
110                     if (!Collections.disjoint(types, elementTypes))
111                         result.add(element);
112                 }
113
114                 return result;
115             }
116
117         }, new SetListenerToSingleSetListener<Resource>(listener));
118     }
119
120     @Override
121     public String toString() {
122         return "every '" + name + "'";
123     }
124
125 }