X-Git-Url: https://gerrit.simantics.org/r/gitweb?a=blobdiff_plain;f=org.simantics.plant3d%2Fsrc%2Forg%2Fsimantics%2Fplant3d%2Feditor%2FP3DNodeMap.java;fp=org.simantics.plant3d%2Fsrc%2Forg%2Fsimantics%2Fplant3d%2Feditor%2FP3DNodeMap.java;h=b098951346a114f6b1566ee798a6c20d358d88b1;hb=a460e609147d064dd3da464bcf1626845e0f93b4;hp=0000000000000000000000000000000000000000;hpb=3e2205a22ddee674f7936da114f5d520bd83e0ff;p=simantics%2F3d.git

diff --git a/org.simantics.plant3d/src/org/simantics/plant3d/editor/P3DNodeMap.java b/org.simantics.plant3d/src/org/simantics/plant3d/editor/P3DNodeMap.java
new file mode 100644
index 00000000..b0989513
--- /dev/null
+++ b/org.simantics.plant3d/src/org/simantics/plant3d/editor/P3DNodeMap.java
@@ -0,0 +1,246 @@
+package org.simantics.plant3d.editor;
+
+import java.util.ArrayList;
+import java.util.Collection;
+import java.util.HashSet;
+import java.util.List;
+import java.util.Set;
+
+import org.simantics.db.ReadGraph;
+import org.simantics.db.Resource;
+import org.simantics.db.Session;
+import org.simantics.db.exception.DatabaseException;
+import org.simantics.g3d.ontology.G3D;
+import org.simantics.g3d.scenegraph.base.INode;
+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.plant3d.ontology.Plant3D;
+import org.simantics.plant3d.scenegraph.IP3DNode;
+import org.simantics.plant3d.scenegraph.IP3DVisualNode;
+import org.simantics.plant3d.scenegraph.P3DParentNode;
+import org.simantics.plant3d.scenegraph.P3DRootNode;
+import org.simantics.plant3d.scenegraph.ParameterizedNode;
+import org.simantics.plant3d.scenegraph.PipeRun;
+import org.simantics.plant3d.scenegraph.controlpoint.PipeControlPoint;
+import org.simantics.plant3d.scenegraph.controlpoint.PipingRules;
+import org.simantics.utils.threads.AWTThread;
+
+import vtk.vtkProp;
+import vtk.vtkProp3D;
+
+public class P3DNodeMap extends AbstractVTKNodeMap<INode> {
+	
+	private static final boolean DEBUG = false;
+
+	public P3DNodeMap(Session session, IMapping mapping, InteractiveVtkPanel panel, P3DRootNode rootNode) {
+		super(session, mapping, panel, rootNode);
+		rootNode.setNodeMap(this);
+	}
+	@Override
+	protected void updateActor(INode n, Set<String> ids) {
+		if (DEBUG) System.out.println("P3DNodeMap update " + n);
+		if (!(n instanceof IP3DVisualNode)) {
+			if (n instanceof PipeControlPoint) {
+				n = ((PipeControlPoint)n).getPipelineComponent();
+				if (n == null)
+					return;
+			} else {
+				return;
+			}
+		}
+		
+		IP3DVisualNode node = (IP3DVisualNode)n;
+		
+		if (DEBUG) {
+			System.out.print("P3DNodeMap update " + node);
+			for (String s : ids)
+				System.out.print(" " + s);
+			System.out.println();
+		}
+		
+		if (ids.contains(Plant3D.URIs.hasGeometry)) {
+			node.visualize(panel);
+			updateRenderObjectsFor(node);
+			updateTransform(node);
+		} 
+		if (n instanceof ParameterizedNode) {
+			ParameterizedNode geom = (ParameterizedNode)n;
+			for (String id : geom.getParameterMap().keySet()) {
+				if (ids.contains(id)) {
+					node.visualize(panel);
+					updateRenderObjectsFor(node);
+					updateTransform(node);
+					break;
+				}
+			}
+		} else if (n instanceof PipeRun) {
+			// FIXME: may require rule based update!
+			PipeRun run = (PipeRun)n;
+			Set<String> ids2 = new HashSet<String>();
+			ids2.add(Plant3D.URIs.hasGeometry);
+			for (PipeControlPoint pcp : run.getControlPoints()) {
+				updateActor(pcp, ids2);
+			}
+		}
+		
+		if (ids.contains(G3D.URIs.hasPosition) || 
+			ids.contains(G3D.URIs.hasOrientation) ||
+			ids.contains(G3D.URIs.hasWorldPosition) ||
+			ids.contains(G3D.URIs.hasWorldOrientation)) {
+			updateTransform(node);
+		}
+	}
+	
+	private void updateTransform(IP3DNode node) {
+		if (DEBUG) System.out.println("P3DNodeMap update Transform " + node);
+
+		node.update(panel.GetRenderer());
+		
+		if (node instanceof ParentNode<?>) {
+			ParentNode<IP3DNode> p = (ParentNode<IP3DNode>)node;
+			for (IP3DNode n : p.getNodes())
+				updateTransform(n);
+		}
+	}
+
+	@Override
+	protected Collection<vtkProp> getActors(INode n) {
+		List<vtkProp> props = new ArrayList<vtkProp>();
+		if (!(n instanceof IP3DVisualNode))
+			return props;
+		IP3DVisualNode node = (IP3DVisualNode)n;
+		for (vtkProp3D p : ((IP3DVisualNode)node).getActors())
+			props.add(p);
+		
+		return props;
+	}
+	
+	@Override
+	protected void removeActor(INode n) {
+		if (DEBUG) System.out.println("P3DNodeMap.removeActor " + n);
+		if (!(n instanceof IP3DVisualNode))
+			return;
+		IP3DVisualNode node = (IP3DVisualNode)n;
+		remActor(node);
+		
+		if (node instanceof P3DParentNode<?>) {
+			for (IP3DNode n2 : ((P3DParentNode<?>)node).getNodes())
+				if (n2 instanceof IP3DVisualNode)
+					removeActor((IP3DVisualNode)n2);
+		}
+	}
+	
+	@Override
+	protected void addActor(INode n) {
+		if (DEBUG) System.out.println("P3DNodeMap.addActor " + n);
+		if (!(n instanceof IP3DVisualNode))
+			return;
+		IP3DVisualNode node = (IP3DVisualNode)n;
+		
+		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);
+		}
+		
+		if (node instanceof P3DParentNode<?>) {
+			for (IP3DNode n2 : ((P3DParentNode<?>)node).getNodes())
+				if (n2 instanceof IP3DVisualNode)
+					addActor((IP3DVisualNode)n2);
+		}
+		
+		updateTransform(node);
+
+        panel.unlock();
+
+	}
+	
+	
+	
+	private boolean hasActor(IP3DVisualNode node) {
+		List<vtkProp> list = nodeToActor.getValues(node);
+		if (list == null || list.size() == 0)
+			return false;
+		return true;
+	}
+	
+	private void remActor(IP3DVisualNode 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();
+		}
+	}
+	
+	@Override
+	protected void update(ReadGraph graph) throws DatabaseException {
+		validate();
+//		System.out.println("Graph updates");
+		super.update(graph);
+		validate();
+	}
+	
+	@Override
+	public void commit() {
+		validate();
+//		System.out.println("Graph commit");
+		super.commit();
+		
+	}
+	@Override
+	protected void doCommit() {
+//		System.out.println("Do commit");
+		validate();
+		super.doCommit();
+	}
+	
+	private void validate() {
+		for (INode node : rootNode.getNodes()) {
+			if (node instanceof PipeRun)
+				PipingRules.validate((PipeRun)node);
+		}
+	}
+	
+	@Override
+	public synchronized void preRender() {
+//		System.out.println("P3DNodeMap preRender");
+//		super.preRender();
+		try {
+//			boolean b = false;
+//			synchronized (syncMutex) {
+//				b = PipingRules.update();
+//			}
+//			if (b)
+//				super.preRender();
+			boolean b = true;
+			while (b) {
+				updateCycle();
+				b = PipingRules.update();
+			}
+		} catch (Exception e) {
+			e.printStackTrace();
+		}
+		
+	}
+
+}