X-Git-Url: https://gerrit.simantics.org/r/gitweb?a=blobdiff_plain;f=org.simantics.g3d.csg%2Fsrc%2Forg%2Fsimantics%2Fg3d%2Fcsg%2Feditor%2FCSGNodeMap.java;fp=org.simantics.g3d.csg%2Fsrc%2Forg%2Fsimantics%2Fg3d%2Fcsg%2Feditor%2FCSGNodeMap.java;h=1776694178f99a164dc61fd8387e93f553806674;hb=87b3241ec277ba3d8e414b26186a032c9cdcaeed;hp=0000000000000000000000000000000000000000;hpb=1f0bcd66274375f2278d2e6c486cb28257a5f7b2;p=simantics%2F3d.git

diff --git a/org.simantics.g3d.csg/src/org/simantics/g3d/csg/editor/CSGNodeMap.java b/org.simantics.g3d.csg/src/org/simantics/g3d/csg/editor/CSGNodeMap.java
new file mode 100644
index 00000000..17766941
--- /dev/null
+++ b/org.simantics.g3d.csg/src/org/simantics/g3d/csg/editor/CSGNodeMap.java
@@ -0,0 +1,119 @@
+package org.simantics.g3d.csg.editor;
+
+import java.util.ArrayList;
+import java.util.Collection;
+import java.util.List;
+import java.util.Set;
+
+import org.simantics.db.Session;
+import org.simantics.g3d.csg.scenegraph2.CSGrootNode;
+import org.simantics.g3d.csg.scenegraph2.ICSGnode;
+import org.simantics.g3d.scenegraph.base.ParentNode;
+import org.simantics.g3d.vtk.common.AbstractVTKNodeMap;
+import org.simantics.g3d.vtk.common.InteractiveVtkPanel;
+import org.simantics.objmap.graph.IMapping;
+import org.simantics.utils.threads.AWTThread;
+
+import vtk.vtkProp;
+import vtk.vtkProp3D;
+import vtk.vtkRenderer;
+
+public class CSGNodeMap extends AbstractVTKNodeMap<ICSGnode> {
+	
+
+	
+	public CSGNodeMap(Session session, IMapping mapping, InteractiveVtkPanel panel, CSGrootNode rootNode) {
+		super(session, mapping, panel, rootNode);
+		rootNode.setNodeMap(this);
+	}
+	
+
+	@SuppressWarnings("unchecked")
+	protected void updateActor(ICSGnode node, Set<String> ids) {
+		//System.out.println("CSGNodeMap.updateActor " + node);
+		if (node.getParent() instanceof ICSGnode) {
+			ICSGnode parent = (ICSGnode) node.getParent();
+			if (!"child".equals(node.getParentRel())) {
+				updateActor(parent,null);
+				return;
+			}
+		}
+		
+		if (node instanceof ParentNode<?>) {
+			ParentNode<ICSGnode> p = (ParentNode<ICSGnode>)node;
+			for (ICSGnode n : p.getNodes())
+				remActor(n);
+		}
+		
+		remActor(node);
+		addActor(node);
+
+	}
+
+	@Override
+	protected Collection<vtkProp> getActors(ICSGnode node) {
+		List<vtkProp> props = new ArrayList<vtkProp>();
+		for (vtkProp3D p : ((ICSGnode)node).getActors())
+			props.add(p);
+		return props;
+	}
+	
+	protected void removeActor(ICSGnode node) {
+		//System.out.println("CSGNodeMap.removeActor " + node);
+		remActor(node);
+		
+		if (!"child".equals(node.getParentRel())) {
+			if (node.getParent() instanceof ICSGnode)
+				updateActor((ICSGnode)node.getParent(),null);
+		}
+	}
+	
+	protected void addActor(ICSGnode node) {
+		//System.out.println("CSGNodeMap.addActor " + node);
+		if (hasActor(node))
+			return;
+		if (Thread.currentThread() != AWTThread.getThreadAccess().getThread())
+			throw new RuntimeException("Illegal thread.");
+		
+		panel.lock();
+		
+		node.visualize(panel);
+
+		for (vtkProp3D act : node.getActors()) {
+			nodeToActor.add(node, act);
+            actorToNode.put(act, node);
+		}
+
+        panel.unlock();
+
+	}
+	
+	
+	
+	private boolean hasActor(ICSGnode node) {
+		List<vtkProp> list = nodeToActor.getValues(node);
+		if (list == null || list.size() == 0)
+			return false;
+		return true;
+	}
+	
+	private void remActor(ICSGnode node) {
+		if (Thread.currentThread() != AWTThread.getThreadAccess().getThread())
+			throw new RuntimeException("Illegal thread.");
+
+		List<vtkProp> list = nodeToActor.getValues(node);
+		if (list != null) {
+			for (vtkProp obj : list) {
+				actorToNode.remove(obj);	
+			}
+			nodeToActor.remove(node);
+			panel.lock();
+			
+			node.stopVisualize();
+			
+			panel.unlock();
+		}
+	}
+	
+
+}