1 package org.simantics.g3d.scl;
3 import java.util.ArrayList;
4 import java.util.Collection;
5 import java.util.Collections;
6 import java.util.HashSet;
9 import java.util.Stack;
11 import org.simantics.db.ReadGraph;
12 import org.simantics.db.Session;
13 import org.simantics.db.WriteGraph;
14 import org.simantics.db.common.request.ReadRequest;
15 import org.simantics.db.common.request.WriteRequest;
16 import org.simantics.db.exception.DatabaseException;
17 import org.simantics.db.layer0.util.Layer0Utils;
18 import org.simantics.g3d.ontology.G3D;
19 import org.simantics.g3d.scenegraph.NodeMap;
20 import org.simantics.g3d.scenegraph.base.INode;
21 import org.simantics.g3d.scenegraph.base.NodeListener;
22 import org.simantics.g3d.scenegraph.base.ParentNode;
23 import org.simantics.objmap.exceptions.MappingException;
24 import org.simantics.objmap.graph.IMapping;
25 import org.simantics.objmap.graph.IMappingListener;
26 import org.simantics.utils.datastructures.MapSet;
27 import org.simantics.utils.datastructures.Pair;
30 * NodeMap implementation used with SCL scripts.
32 * In practice, there are no visible objects to synchronize.
35 * This implementation is Copy-paste of AbstractVTKNodeMap with VTK references removed.
36 * There may be more sensible way to do this.
43 public abstract class ScriptNodeMap<E extends INode> implements NodeMap<Object,E>, IMappingListener, NodeListener {
45 private static final boolean DEBUG = false;
47 protected Session session;
48 protected IMapping<Object,E> mapping;
50 protected ParentNode<E> rootNode;
52 protected Set<E> nodes = new HashSet<E>();
54 private boolean dirty = false;
56 public ScriptNodeMap(Session session, IMapping<Object,E> mapping, ParentNode<E> rootNode) {
57 this.session = session;
58 this.mapping = mapping;
59 this.rootNode = rootNode;
60 mapping.addMappingListener(this);
61 rootNode.addListener(this);
66 protected abstract void addActor(E node);
67 protected abstract void removeActor(E node);
68 protected abstract void updateActor(E node,Set<String> ids);
70 public void repaint() {
74 public void populate() {
75 for (E node : rootNode.getNodes()) {
76 receiveAdd(node, node.getParentRel(),true);
82 public E getNode(Object o) {
86 @SuppressWarnings("unchecked")
88 public Collection<Object> getRenderObjects(INode node) {
89 return Collections.EMPTY_LIST;
92 @SuppressWarnings("unchecked")
94 public ParentNode<E> getRootNode() {
95 return (ParentNode<E>)rootNode;
101 public boolean isChangeTracking() {
102 return changeTracking;
106 public void setChangeTracking(boolean enabled) {
107 changeTracking = enabled;
110 private boolean changeTracking = true;
112 protected Object syncMutex = new Object();
115 private List<Pair<E,String>> added = new ArrayList<Pair<E,String>>();
116 private List<Pair<E,String>> removed = new ArrayList<Pair<E,String>>();
117 private MapSet<E, String> updated = new MapSet.Hash<E, String>();
119 private boolean rangeModified = false;
123 @SuppressWarnings("unchecked")
125 public void updateRenderObjectsFor(E node) {
130 @SuppressWarnings("unchecked")
131 private void receiveAdd(E node, String id, boolean db) {
132 if (DEBUG) System.out.println("receiveAdd " + debugString(node) + " " + id + " " + db);
133 synchronized (syncMutex) {
134 for (Pair<E, String> n : added) {
135 if (n.first.equals(node))
138 if (changeTracking) {
139 mapping.rangeModified((E)node.getParent());
141 added.add(new Pair<E, String>(node, id));
142 rangeModified = true;
147 @SuppressWarnings("unchecked")
148 private void receiveRemove(E node, String id, boolean db) {
149 if (DEBUG) System.out.println("receiveRemove " + debugString(node) + " " + id + " " + db);
150 synchronized (syncMutex) {
151 for (Pair<E, String> n : removed) {
152 if (n.first.equals(node))
155 if (changeTracking && !db)
156 mapping.rangeModified((E)node.getParent());
157 removed.add(new Pair<E, String>(node, id));
158 rangeModified = true;
163 @SuppressWarnings("unchecked")
164 private void receiveUpdate(E node, String id, boolean db) {
165 if (DEBUG) System.out.println("receiveUpdate " + debugString(node) + " " + id + " " + db);
166 synchronized (syncMutex) {
167 // for (Pair<E, String> n : updated) {
168 // if (n.first.equals(node))
171 if (changeTracking && !db)
172 mapping.rangeModified(node);
173 //updated.add(new Pair<E, String>(node, id));
174 updated.add(node, id);
175 rangeModified = true;
180 private boolean graphUpdates = false;
181 private Set<E> graphModified = new HashSet<E>();
183 private boolean requestCommit = false;
184 private String commitMessage = null;
187 public void commit(String message) {
188 requestCommit = true;
189 commitMessage = message;
192 protected void doCommit() throws DatabaseException {
193 session.syncRequest(new WriteRequest() {
196 public void perform(WriteGraph graph) throws DatabaseException {
197 if (DEBUG) System.out.println("Commit " + commitMessage);
198 if (commitMessage != null) {
199 Layer0Utils.addCommentMetadata(graph, commitMessage);
200 graph.markUndoPoint();
201 commitMessage = null;
209 protected void commit(WriteGraph graph) throws DatabaseException {
210 synchronized(syncMutex) {
211 if (DEBUG) System.out.println("Commit");
213 mapping.updateDomain(graph);
214 graphUpdates = false;
216 if (DEBUG) System.out.println("Commit done");
223 public void domainModified() {
226 if (DEBUG)System.out.println("domainModified");
227 // FIXME : this is called by IMapping id DB thread
229 // session.asyncRequest(new ReadRequest() {
231 // @SuppressWarnings("unchecked")
233 // public void run(ReadGraph graph) throws DatabaseException {
240 protected void reset(ReadGraph graph) throws MappingException {
241 if (DEBUG) System.out.println("Reset");
242 synchronized (syncMutex) {
244 mapping.getRangeModified().clear();
245 for (Object o : mapping.getDomain())
246 mapping.domainModified(o);
247 mapping.updateRange(graph);
248 graphModified.clear();
249 graphUpdates = false;
254 protected void update(ReadGraph graph) throws DatabaseException {
255 if (DEBUG) System.out.println("Graph update start");
256 synchronized (syncMutex) {
258 for (Object domainObject : mapping.getDomainModified()) {
259 E rangeObject = mapping.get(domainObject);
260 if (rangeObject != null)
261 graphModified.add(rangeObject);
263 mapping.updateRange(graph);
264 graphModified.clear();
267 graphUpdates = false;
271 //if (mapping.isRangeModified() && !runUndo) // FIXME : redo?
272 if (mapping.isRangeModified())
273 commit((String)null);
274 if (DEBUG) System.out.println("Graph update done");
278 public void rangeModified() {
279 //System.out.println("rangeModified");
283 public void update() throws DatabaseException{
289 if (requestCommit && !rangeModified) { // FIXME : not thread safe.
290 requestCommit = false;
293 session.syncRequest(new ReadRequest() {
294 @SuppressWarnings("unchecked")
296 public void run(ReadGraph graph) throws DatabaseException {
305 // Reusable containers for data synchronisation
306 List<Pair<E, String>> rem = new ArrayList<Pair<E,String>>(); // Removed objects
307 List<Pair<E, String>> add = new ArrayList<Pair<E,String>>(); // Added objects
308 MapSet<E, String> mod = new MapSet.Hash<E, String>(); // Modified objects
309 Set<E> propagation = new HashSet<E>(); // Objects with propagated changes
310 Stack<E> stack = new Stack<E>(); // Stack for handling propagation
311 Set<E> delete = Collections.synchronizedSet(new HashSet<E>()); // Objects to be completely deleted
312 Set<E> deleteUC = new HashSet<E>();
318 * When objects are removed (either from Java or Graph), after remove processing the Java objects remain in mapping cache.
319 * This causes problems with Undo and Redo, whcih the end up re-using the removed objects from mapping cache.
321 * This code here synchronizes removed and added objects to collect deletable objects. (a deletable object is one which is removed but not added).
324 protected void syncDeletes() {
326 for (Pair<E, String> n : removed) {
327 deleteUC.add(n.first);
329 for (Pair<E, String> n : added) {
330 deleteUC.remove(n.first);
332 if (DEBUG && deleteUC.size() > 0) {
333 System.out.println("Delete sync");
335 System.out.println(debugString(n));
338 delete.addAll(deleteUC);
343 * Clears deletable objects from mapping cache.
345 protected void clearDeletes() {
346 if (DEBUG && delete.size() > 0) System.out.println("Delete");
348 if (DEBUG) System.out.println(debugString(n));
349 mapping.getRange().remove(n);
354 protected String debugString(E n) {
355 return n + "@" + Integer.toHexString(n.hashCode());
358 @SuppressWarnings("unchecked")
359 protected void updateCycle() {
366 synchronized (syncMutex) {
369 for (E e : updated.getKeys()) {
370 for (String s : updated.getValues(e)) {
380 for (Pair<E, String> n : rem) {
381 stopListening(n.first);
382 removeActor(n.first);
385 for (Pair<E, String> n : add) {
390 for (E e : mod.getKeys()) {
391 Set<String> ids = mod.getValues(e);
392 if (ids.contains(G3D.URIs.hasPosition) || ids.contains(G3D.URIs.hasOrientation)) {
393 if (!propagation.contains(e))
398 if (propagation.size() > 0) {
400 stack.addAll(propagation);
402 while (!stack.isEmpty()) {
403 E node = stack.pop();
404 if (propagation.contains(node))
406 propagation.add(node);
407 for (NodeListener l : node.getListeners()) {
409 //changeTracking = false;
410 //l.propertyChanged(node, G3D.URIs.hasPosition);
411 //changeTracking = true;
413 l.propertyChanged(node, G3D.URIs.hasWorldPosition);
416 if (node instanceof ParentNode) {
417 stack.addAll(((ParentNode<E>)node).getNodes());
422 // synchronized (syncMutex) {
423 // rem.addAll(removed);
424 // add.addAll(added);
425 // //mod.addAll(updated);
426 // for (E e : updated.getKeys()) {
427 // for (String s : updated.getValues(e))
436 for (E e : mod.getKeys()) {
437 Set<String> ids = mod.getValues(e);
442 for (Pair<E, String> n : rem) {
443 for (NodeListener l : nodeListeners)
444 l.nodeRemoved(null, n.first, n.second);
446 for (Pair<E, String> n : add) {
447 for (NodeListener l : nodeListeners)
448 l.nodeAdded(n.first.getParent(), n.first, n.second);
450 // for (Pair<E, String> n : mod) {
451 // for (NodeListener l : nodeListeners)
452 // l.propertyChanged(n.first, n.second);
454 for (E e : mod.getKeys()) {
455 for (NodeListener l : nodeListeners)
456 for (String s : mod.getValues(e))
457 l.propertyChanged(e, s);
460 synchronized (syncMutex) {
461 if (added.isEmpty() && removed.isEmpty() && updated.getKeys().size() == 0)
462 rangeModified = false;
466 @SuppressWarnings("unchecked")
467 private void listen(INode node) {
468 node.addListener(this);
469 if (node instanceof ParentNode<?>) {
470 ParentNode<INode> parentNode = (ParentNode<INode>)node;
471 for (INode n : parentNode.getNodes())
476 private void stopListening(INode node) {
477 node.removeListener(this);
478 if (node instanceof ParentNode<?>) {
479 @SuppressWarnings("unchecked")
480 ParentNode<INode> parentNode = (ParentNode<INode>)node;
481 for (INode n : parentNode.getNodes())
486 @SuppressWarnings("unchecked")
488 public void propertyChanged(INode node, String id) {
489 //receiveUpdate((E)node, id, graphUpdates);
490 receiveUpdate((E)node, id, graphModified.contains(node));
494 @SuppressWarnings("unchecked")
496 public <T extends INode> void nodeAdded(ParentNode<T> node, INode child,
498 if (DEBUG) System.out.println("Node added " + child + " parent " + node);
499 //receiveAdd((E)child, rel ,graphUpdates);
500 receiveAdd((E)child, rel ,graphModified.contains(node));
504 @SuppressWarnings("unchecked")
506 public <T extends INode> void nodeRemoved(ParentNode<T> node, INode child,
508 if (DEBUG) System.out.println("Node removed " + child + " parent " + node);
509 //receiveRemove((E)child, rel, graphUpdates);
510 receiveRemove((E)child, rel, graphModified.contains(node));
512 //FIXME : sometimes removed structural models cause ObjMap to add their children again.
513 // removing the listener here prevents corruption of visual model, but better fix is needed.
514 stopListening(child);
518 public void delete() {
520 changeTracking = false;
521 mapping.removeMappingListener(this);
523 for (E node : nodes) {
524 node.removeListener(this);
532 private List<NodeListener> nodeListeners = new ArrayList<NodeListener>();
534 public void addListener(NodeListener listener) {
535 nodeListeners.add(listener);
540 public void removeListener(NodeListener listener) {
541 nodeListeners.remove(listener);
545 public IMapping<Object,E> getMapping() {