1 package org.simantics.g3d.vtk.common;
\r
3 import java.util.ArrayList;
\r
4 import java.util.Collection;
\r
5 import java.util.HashMap;
\r
6 import java.util.HashSet;
\r
7 import java.util.List;
\r
8 import java.util.Map;
\r
9 import java.util.Set;
\r
10 import java.util.Stack;
\r
12 import org.simantics.db.ReadGraph;
\r
13 import org.simantics.db.Resource;
\r
14 import org.simantics.db.Session;
\r
15 import org.simantics.db.WriteGraph;
\r
16 import org.simantics.db.common.request.ReadRequest;
\r
17 import org.simantics.db.common.request.WriteRequest;
\r
18 import org.simantics.db.exception.DatabaseException;
\r
19 import org.simantics.g3d.ontology.G3D;
\r
20 import org.simantics.g3d.scenegraph.IG3DNode;
\r
21 import org.simantics.g3d.scenegraph.RenderListener;
\r
22 import org.simantics.g3d.scenegraph.base.INode;
\r
23 import org.simantics.g3d.scenegraph.base.NodeListener;
\r
24 import org.simantics.g3d.scenegraph.base.ParentNode;
\r
25 import org.simantics.objmap.graph.IMapping;
\r
26 import org.simantics.objmap.graph.IMappingListener;
\r
27 import org.simantics.utils.datastructures.Callback;
\r
28 import org.simantics.utils.datastructures.MapList;
\r
29 import org.simantics.utils.datastructures.MapSet;
\r
30 import org.simantics.utils.datastructures.Pair;
\r
31 import org.simantics.utils.ui.ExceptionUtils;
\r
35 public abstract class AbstractVTKNodeMap<E extends INode> implements VTKNodeMap, IMappingListener, RenderListener, NodeListener{
\r
37 private static final boolean DEBUG = false;
\r
39 protected Session session;
\r
40 protected IMapping<Object,E> mapping;
\r
41 protected InteractiveVtkPanel panel;
\r
43 protected MapList<E, vtkProp> nodeToActor = new MapList<E, vtkProp>();
\r
44 protected Map<vtkProp,E> actorToNode = new HashMap<vtkProp, E>();
\r
46 protected ParentNode<E> rootNode;
\r
48 public AbstractVTKNodeMap(Session session, IMapping<Object,E> mapping, InteractiveVtkPanel panel, ParentNode<E> rootNode) {
\r
49 this.session = session;
\r
50 this.mapping = mapping;
\r
52 this.rootNode = rootNode;
\r
53 panel.addListener(this);
\r
54 mapping.addMappingListener(this);
\r
55 rootNode.addListener(this);
\r
59 protected abstract void addActor(E node);
\r
60 protected abstract void removeActor(E node);
\r
61 protected abstract void updateActor(E node,Set<String> ids);
\r
63 public void repaint() {
\r
67 public void populate() {
\r
68 for (E node : rootNode.getNodes()) {
\r
69 receiveAdd(node, node.getParentRel(),true);
\r
75 public INode getNode(vtkProp prop) {
\r
76 return actorToNode.get(prop);
\r
79 @SuppressWarnings("unchecked")
\r
81 public Collection<vtkProp> getRenderObjects(INode node) {
\r
82 return nodeToActor.getValues((E)node);
\r
85 @SuppressWarnings("unchecked")
\r
87 public ParentNode<IG3DNode> getRootNode() {
\r
88 return (ParentNode<IG3DNode>)rootNode;
\r
94 public boolean isChangeTracking() {
\r
95 return changeTracking;
\r
99 public void setChangeTracking(boolean enabled) {
\r
100 changeTracking = enabled;
\r
103 private boolean changeTracking = true;
\r
105 protected Object syncMutex = new Object();
\r
108 private List<Pair<E,String>> added = new ArrayList<Pair<E,String>>();
\r
109 private List<Pair<E,String>> removed = new ArrayList<Pair<E,String>>();
\r
110 //private List<Pair<E,String>> updated = new ArrayList<Pair<E,String>>();
\r
111 private MapSet<E, String> updated = new MapSet.Hash<E, String>();
\r
113 private boolean rangeModified = false;
\r
115 @SuppressWarnings("unchecked")
\r
117 public void updateRenderObjectsFor(INode node) {
\r
118 List<vtkProp> toDelete = new ArrayList<vtkProp>();
\r
119 for (vtkProp prop : nodeToActor.getValues((E)node)) {
\r
120 if (prop.GetVTKId() != 0) {
\r
121 panel.GetRenderer().RemoveActor(prop);
\r
123 toDelete.add(prop);
\r
125 actorToNode.remove(prop);
\r
127 nodeToActor.remove((E)node);
\r
128 Collection<vtkProp> coll = getActors((E)node);
\r
131 for (vtkProp prop : coll) {
\r
132 nodeToActor.add((E)node,prop);
\r
133 actorToNode.put(prop, (E)node);
\r
134 toDelete.remove(prop);
\r
136 for (vtkProp p : toDelete)
\r
140 protected abstract Collection<vtkProp> getActors(E node);
\r
142 @SuppressWarnings("unchecked")
\r
143 private void receiveAdd(E node, String id, boolean db) {
\r
144 if (DEBUG) System.out.println("receiveAdd " + node + " " + id + " " + db);
\r
145 synchronized (syncMutex) {
\r
146 for (Pair<E, String> n : added) {
\r
147 if (n.first.equals(node))
\r
150 if (changeTracking) {
\r
151 mapping.rangeModified((E)node.getParent());
\r
153 added.add(new Pair<E, String>(node, id));
\r
154 rangeModified = true;
\r
159 @SuppressWarnings("unchecked")
\r
160 private void receiveRemove(E node, String id, boolean db) {
\r
161 if (DEBUG) System.out.println("receiveRemove " + node + " " + id + " " + db);
\r
162 synchronized (syncMutex) {
\r
163 for (Pair<E, String> n : removed) {
\r
164 if (n.first.equals(node))
\r
167 if (changeTracking && !db)
\r
168 mapping.rangeModified((E)node.getParent());
\r
169 removed.add(new Pair<E, String>(node, id));
\r
170 rangeModified = true;
\r
175 @SuppressWarnings("unchecked")
\r
176 private void receiveUpdate(E node, String id, boolean db) {
\r
177 if (DEBUG) System.out.println("receiveUpdate " + node + " " + id + " " + db);
\r
178 synchronized (syncMutex) {
\r
179 // for (Pair<E, String> n : updated) {
\r
180 // if (n.first.equals(node))
\r
183 if (changeTracking && !db)
\r
184 mapping.rangeModified(node);
\r
185 //updated.add(new Pair<E, String>(node, id));
\r
186 updated.add(node, id);
\r
187 rangeModified = true;
\r
192 private boolean graphUpdates = false;
\r
193 private Set<E> graphModified = new HashSet<E>();
\r
195 private boolean requestCommit = false;
\r
198 public void commit() {
\r
199 requestCommit = true;
\r
202 protected void doCommit() {
\r
203 session.asyncRequest(new WriteRequest() {
\r
206 public void perform(WriteGraph graph) throws DatabaseException {
\r
210 }, new Callback<DatabaseException>() {
\r
213 public void run(DatabaseException parameter) {
\r
214 if (parameter != null)
\r
215 ExceptionUtils.logAndShowError("Cannot commit editor changes", parameter);
\r
220 protected void commit(WriteGraph graph) throws DatabaseException {
\r
221 synchronized(syncMutex) {
\r
222 if (DEBUG) System.out.println("Commit");
\r
223 graphUpdates = true;
\r
224 mapping.updateDomain(graph);
\r
225 graphUpdates = false;
\r
230 public void domainModified() {
\r
233 if (DEBUG)System.out.println("domainModified");
\r
234 session.asyncRequest(new ReadRequest() {
\r
236 @SuppressWarnings("unchecked")
\r
238 public void run(ReadGraph graph) throws DatabaseException {
\r
245 protected void update(ReadGraph graph) throws DatabaseException {
\r
246 synchronized (syncMutex) {
\r
247 graphUpdates = true;
\r
248 for (Object domainObject : mapping.getDomainModified()) {
\r
249 E rangeObject = mapping.get(domainObject);
\r
250 if (rangeObject != null)
\r
251 graphModified.add(rangeObject);
\r
253 mapping.updateRange(graph);
\r
254 graphModified.clear();
\r
255 graphUpdates = false;
\r
258 if (mapping.isRangeModified())
\r
263 public void rangeModified() {
\r
264 //System.out.println("rangeModified");
\r
269 public void postRender() {
\r
270 // Commit changes if
\r
271 // 1. Commit has been requested
\r
272 // 2. There are no pending changes that should be processed in preRender()
\r
273 if (requestCommit && !rangeModified) { // FIXME : not thread safe.
\r
274 requestCommit = false;
\r
279 List<Pair<E, String>> rem = new ArrayList<Pair<E,String>>();
\r
280 List<Pair<E, String>> add = new ArrayList<Pair<E,String>>();
\r
281 MapSet<E, String> mod = new MapSet.Hash<E, String>();
\r
282 Set<E> propagation = new HashSet<E>();
\r
283 Stack<E> stack = new Stack<E>();
\r
287 public synchronized void preRender() {
\r
291 @SuppressWarnings("unchecked")
\r
292 protected void updateCycle() {
\r
296 propagation.clear();
\r
298 synchronized (syncMutex) {
\r
299 rem.addAll(removed);
\r
301 for (E e : updated.getKeys()) {
\r
302 for (String s : updated.getValues(e)) {
\r
312 for (Pair<E, String> n : rem) {
\r
313 stopListening(n.first);
\r
314 removeActor(n.first);
\r
318 for (Pair<E, String> n : add) {
\r
323 for (E e : mod.getKeys()) {
\r
324 Set<String> ids = mod.getValues(e);
\r
325 if (ids.contains(G3D.URIs.hasPosition) || ids.contains(G3D.URIs.hasOrientation)) {
\r
326 if (!propagation.contains(e))
\r
327 propagation.add(e);
\r
331 if (propagation.size() > 0) {
\r
333 stack.addAll(propagation);
\r
334 propagation.clear();
\r
335 while (!stack.isEmpty()) {
\r
336 E node = stack.pop();
\r
337 if (propagation.contains(node))
\r
339 propagation.add(node);
\r
340 for (NodeListener l : node.getListeners()) {
\r
342 //changeTracking = false;
\r
343 //l.propertyChanged(node, G3D.URIs.hasPosition);
\r
344 //changeTracking = true;
\r
346 l.propertyChanged(node, G3D.URIs.hasWorldPosition);
\r
349 if (node instanceof ParentNode) {
\r
350 stack.addAll(((ParentNode<E>)node).getNodes());
\r
355 // synchronized (syncMutex) {
\r
356 // rem.addAll(removed);
\r
357 // add.addAll(added);
\r
358 // //mod.addAll(updated);
\r
359 // for (E e : updated.getKeys()) {
\r
360 // for (String s : updated.getValues(e))
\r
364 // removed.clear();
\r
366 // updated.clear();
\r
369 for (E e : mod.getKeys()) {
\r
370 Set<String> ids = mod.getValues(e);
\r
371 updateActor(e,ids);
\r
375 for (Pair<E, String> n : rem) {
\r
376 for (NodeListener l : nodeListeners)
\r
377 l.nodeRemoved(null, n.first, n.second);
\r
379 for (Pair<E, String> n : add) {
\r
380 for (NodeListener l : nodeListeners)
\r
381 l.nodeAdded(n.first.getParent(), n.first, n.second);
\r
383 // for (Pair<E, String> n : mod) {
\r
384 // for (NodeListener l : nodeListeners)
\r
385 // l.propertyChanged(n.first, n.second);
\r
387 for (E e : mod.getKeys()) {
\r
388 for (NodeListener l : nodeListeners)
\r
389 for (String s : mod.getValues(e))
\r
390 l.propertyChanged(e, s);
\r
392 synchronized (syncMutex) {
\r
393 if (added.isEmpty() && removed.isEmpty() && updated.getKeys().size() == 0)
\r
394 rangeModified = false;
\r
398 @SuppressWarnings("unchecked")
\r
399 private void listen(INode node) {
\r
400 node.addListener(this);
\r
401 if (node instanceof ParentNode<?>) {
\r
402 ParentNode<INode> parentNode = (ParentNode<INode>)node;
\r
403 for (INode n : parentNode.getNodes())
\r
408 private void stopListening(INode node) {
\r
409 node.removeListener(this);
\r
410 if (node instanceof ParentNode<?>) {
\r
411 @SuppressWarnings("unchecked")
\r
412 ParentNode<INode> parentNode = (ParentNode<INode>)node;
\r
413 for (INode n : parentNode.getNodes())
\r
418 @SuppressWarnings("unchecked")
\r
420 public void propertyChanged(INode node, String id) {
\r
421 //receiveUpdate((E)node, id, graphUpdates);
\r
422 receiveUpdate((E)node, id, graphModified.contains(node));
\r
426 @SuppressWarnings("unchecked")
\r
428 public <T extends INode> void nodeAdded(ParentNode<T> node, INode child,
\r
430 if (DEBUG) System.out.println("Node added " + child + " parent " + node);
\r
431 //receiveAdd((E)child, rel ,graphUpdates);
\r
432 receiveAdd((E)child, rel ,graphModified.contains(node));
\r
436 @SuppressWarnings("unchecked")
\r
438 public <T extends INode> void nodeRemoved(ParentNode<T> node, INode child,
\r
440 if (DEBUG) System.out.println("Node removed " + child + " parent " + node);
\r
441 //receiveRemove((E)child, rel, graphUpdates);
\r
442 receiveRemove((E)child, rel, graphModified.contains(node));
\r
444 //FIXME : sometimes removed structural models cause ObjMap to add their children again.
\r
445 // removing the listener here prevents corruption of visual model, but better fix is needed.
\r
446 stopListening(child);
\r
450 public void delete() {
\r
451 changeTracking = false;
\r
452 panel.removeListener(this);
\r
453 mapping.removeMappingListener(this);
\r
455 List<E> nodes = new ArrayList<E>(nodeToActor.getKeySize());
\r
456 nodes.addAll(nodeToActor.getKeys());
\r
457 for (E node : nodes) {
\r
458 node.removeListener(this);
\r
462 for (vtkProp prop : actorToNode.keySet()) {
\r
463 if (prop.GetVTKId() != 0)
\r
466 actorToNode.clear();
\r
467 nodeToActor.clear();
\r
472 private List<NodeListener> nodeListeners = new ArrayList<NodeListener>();
\r
474 public void addListener(NodeListener listener) {
\r
475 nodeListeners.add(listener);
\r
480 public void removeListener(NodeListener listener) {
\r
481 nodeListeners.remove(listener);
\r
485 public IMapping<Object,E> getMapping() {
\r