3D framework (Simca 2012)

[simantics/3d.git] / org.simantics.g3d.csg / src / org / simantics / g3d / csg / editor / CSGNodeMap.java

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