org.simantics.plant3d/src/org/simantics/plant3d/editor/P3DNodeMap.java

   1 package org.simantics.plant3d.editor;
   2
   3 import java.util.ArrayList;
   4 import java.util.Collection;
   5 import java.util.HashSet;
   6 import java.util.List;
   7 import java.util.Set;
   8
   9 import org.simantics.db.ReadGraph;
  10 import org.simantics.db.Resource;
  11 import org.simantics.db.Session;
  12 import org.simantics.db.WriteGraph;
  13 import org.simantics.db.exception.DatabaseException;
  14 import org.simantics.g3d.ontology.G3D;
  15 import org.simantics.g3d.scenegraph.base.INode;
  16 import org.simantics.g3d.scenegraph.base.ParentNode;
  17 import org.simantics.g3d.vtk.common.AbstractVTKNodeMap;
  18 import org.simantics.g3d.vtk.common.VtkView;
  19 import org.simantics.objmap.graph.IMapping;
  20 import org.simantics.plant3d.ontology.Plant3D;
  21 import org.simantics.plant3d.scenegraph.IP3DNode;
  22 import org.simantics.plant3d.scenegraph.IP3DVisualNode;
  23 import org.simantics.plant3d.scenegraph.P3DParentNode;
  24 import org.simantics.plant3d.scenegraph.P3DRootNode;
  25 import org.simantics.plant3d.scenegraph.ParameterizedNode;
  26 import org.simantics.plant3d.scenegraph.PipeRun;
  27 import org.simantics.plant3d.scenegraph.controlpoint.PipeControlPoint;
  28 import org.simantics.plant3d.scenegraph.controlpoint.PipingRules;
  29
  30 import vtk.vtkProp;
  31 import vtk.vtkProp3D;
  32
  33 public class P3DNodeMap extends AbstractVTKNodeMap<Resource,INode> {
  34
  35         private static final boolean DEBUG = false;
  36
  37         public P3DNodeMap(Session session, IMapping mapping, VtkView panel, P3DRootNode rootNode) {
  38                 super(session, mapping, panel, rootNode);
  39                 rootNode.setNodeMap(this);
  40         }
  41         @Override
  42         protected void updateActor(INode n, Set<String> ids) {
  43                 if (DEBUG) System.out.println("P3DNodeMap update " + debugString(n));
  44                 if (!(n instanceof IP3DVisualNode)) {
  45                         if (n instanceof PipeControlPoint) {
  46                                 n = ((PipeControlPoint)n).getPipelineComponent();
  47                                 if (n == null)
  48                                         return;
  49                         } else {
  50                                 return;
  51                         }
  52                 }
  53
  54                 IP3DVisualNode node = (IP3DVisualNode)n;
  55
  56                 if (DEBUG) {
  57                         System.out.print("P3DNodeMap update " + debugString(node));
  58                         for (String s : ids)
  59                                 System.out.print(" " + s);
  60                         System.out.println();
  61                 }
  62
  63                 if (ids.contains(Plant3D.URIs.hasGeometry)) {
  64                         node.visualize(view);
  65                         updateRenderObjectsFor(node);
  66                         updateTransform(node);
  67                 }
  68                 if (n instanceof ParameterizedNode) {
  69                         ParameterizedNode geom = (ParameterizedNode)n;
  70                         for (String id : geom.getParameterMap().keySet()) {
  71                                 if (ids.contains(id)) {
  72                                         node.visualize(view);
  73                                         updateRenderObjectsFor(node);
  74                                         //updateTransform(node);
  75                                         break;
  76                                 }
  77                         }
  78                 } else if (n instanceof PipeRun) {
  79                         // FIXME: may require rule based update!
  80                         PipeRun run = (PipeRun)n;
  81                         Set<String> ids2 = new HashSet<String>();
  82                         ids2.add(Plant3D.URIs.hasGeometry);
  83                         for (PipeControlPoint pcp : run.getControlPoints()) {
  84                                 updateActor(pcp, ids2);
  85                         }
  86                 }
  87
  88                 if (ids.contains(G3D.URIs.hasPosition) ||
  89                         ids.contains(G3D.URIs.hasOrientation) ||
  90                         ids.contains(G3D.URIs.hasWorldPosition) ||
  91                         ids.contains(G3D.URIs.hasWorldOrientation)) {
  92                         updateTransform(node);
  93                 }
  94                 if (ids.contains(Plant3D.URIs.HasTurnRadiusIndex)) {
  95                     node.visualize(view);
  96                     updateRenderObjectsFor(node);
  97                 }
  98         }
  99
 100         private void updateTransform(IP3DNode node) {
 101                 if (DEBUG) System.out.println("P3DNodeMap update Transform " + debugString(node));
 102
 103                 node.update(view.getRenderer());
 104
 105                 if (node instanceof ParentNode<?>) {
 106                         ParentNode<IP3DNode> p = (ParentNode<IP3DNode>)node;
 107                         for (IP3DNode n : p.getNodes())
 108                                 updateTransform(n);
 109                 }
 110         }
 111
 112         @Override
 113         protected Collection<vtkProp> getActors(INode n) {
 114                 List<vtkProp> props = new ArrayList<vtkProp>();
 115                 if (!(n instanceof IP3DVisualNode))
 116                         return props;
 117                 IP3DVisualNode node = (IP3DVisualNode)n;
 118                 for (vtkProp3D p : ((IP3DVisualNode)node).getActors())
 119                         props.add(p);
 120
 121                 return props;
 122         }
 123
 124         @Override
 125         protected void removeActor(INode n) {
 126                 if (DEBUG) System.out.println("P3DNodeMap.removeActor " + debugString(n));
 127                 if (!(n instanceof IP3DVisualNode))
 128                         return;
 129                 IP3DVisualNode node = (IP3DVisualNode)n;
 130                 remActor(node);
 131
 132                 if (node instanceof P3DParentNode<?>) {
 133                         for (IP3DNode n2 : ((P3DParentNode<?>)node).getNodes())
 134                                 if (n2 instanceof IP3DVisualNode)
 135                                         removeActor((IP3DVisualNode)n2);
 136                 }
 137         }
 138
 139         @Override
 140         protected void addActor(INode n) {
 141                 if (DEBUG) System.out.println("P3DNodeMap.addActor " + debugString(n));
 142                 if (!(n instanceof IP3DVisualNode))
 143                         return;
 144                 IP3DVisualNode node = (IP3DVisualNode)n;
 145
 146                 if (hasActor(node)) {
 147                     if (DEBUG) System.out.println("Node already has an actor");
 148                         return;
 149                 }
 150                 if (Thread.currentThread() != view.getThreadQueue().getThread())
 151                         throw new RuntimeException("Illegal thread.");
 152
 153                 view.lock();
 154
 155                 node.visualize(view);
 156
 157                 map(node, node.getActors());
 158
 159                 if (DEBUG) System.out.println("Added " + node.getActors().size() + " actors");
 160
 161                 if (node instanceof P3DParentNode<?>) {
 162                         for (IP3DNode n2 : ((P3DParentNode<?>)node).getNodes())
 163                                 if (n2 instanceof IP3DVisualNode)
 164                                         addActor((IP3DVisualNode)n2);
 165                 }
 166
 167                 updateTransform(node);
 168
 169                 view.unlock();
 170
 171         }
 172
 173
 174
 175         private boolean hasActor(IP3DVisualNode node) {
 176                 Collection<vtkProp> list = getRenderObjects(node);
 177                 if (list == null || list.size() == 0)
 178                         return false;
 179                 return true;
 180         }
 181
 182         private void remActor(IP3DVisualNode node) {
 183                 if (Thread.currentThread() != view.getThreadQueue().getThread())
 184                         throw new RuntimeException("Illegal thread.");
 185
 186                 Collection<vtkProp> list = getRenderObjects(node);
 187                 if (list.size() > 0) {
 188                     removeMap(node);
 189                     view.lock();
 190                     node.stopVisualize();
 191                     view.unlock();
 192                 }
 193         }
 194
 195         @Override
 196         protected void update(ReadGraph graph) throws DatabaseException {
 197                 validate();
 198 //              System.out.println("Graph updates");
 199                 super.update(graph);
 200                 validate();
 201         }
 202
 203         @Override
 204         public void commit(String commitMessage) {
 205                 validate();
 206 //              System.out.println("Graph commit");
 207                 super.commit(commitMessage);
 208
 209         }
 210
 211         @Override
 212         protected void commit(WriteGraph graph) throws DatabaseException {
 213             super.commit(graph);
 214             validateGraph(graph);
 215         }
 216         @Override
 217         protected void doCommit() {
 218 //              System.out.println("Do commit");
 219                 validate();
 220                 super.doCommit();
 221         }
 222
 223         private void validate() {
 224                 for (INode node : rootNode.getNodes()) {
 225                         if (node instanceof PipeRun)
 226                                 PipingRules.validate((PipeRun)node);
 227                 }
 228         }
 229
 230         @Override
 231         public synchronized void preRender() {
 232 //              System.out.println("P3DNodeMap preRender");
 233 //              super.preRender();
 234                 try {
 235 //                      boolean b = false;
 236 //                      synchronized (syncMutex) {
 237 //                              b = PipingRules.update();
 238 //                      }
 239 //                      if (b)
 240 //                              super.preRender();
 241                         boolean b = true;
 242                         while (b) {
 243                                 updateCycle();
 244                                 b = PipingRules.update();
 245                         }
 246                 } catch (Exception e) {
 247                         e.printStackTrace();
 248                 }
 249
 250         }
 251
 252         protected void validateGraph(ReadGraph graph) throws DatabaseException {
 253             Plant3D P3D = Plant3D.getInstance(graph);
 254             Resource root = (Resource)mapping.inverseGet(rootNode);
 255             if (!graph.isInstanceOf(root, P3D.Plant))
 256                 throw new DatabaseException("Root is not a Plant");
 257             Set<Resource> equipment = new HashSet<>();
 258             Set<Resource> pipeRuns = new HashSet<>();
 259             for (Resource r : graph.getObjects(root, P3D.children)) {
 260                 if (graph.isInstanceOf(r, P3D.Equipment)) {
 261                     equipment.add(r);
 262                 } else if (graph.isInstanceOf(r, P3D.PipeRun)) {
 263                     pipeRuns.add(r);
 264                 } else {
 265                     throw new DatabaseException("Unknown resource " + r+ " " + mapping.get(r));
 266                 }
 267             }
 268             Set<Resource> nozzles = new HashSet<>();
 269             for (Resource e : equipment) {
 270                 for (Resource n : graph.getObjects(e, P3D.HasNozzle)) {
 271                     if (graph.isInstanceOf(n, P3D.Nozzle)) {
 272                         nozzles.add(n);
 273                     } else {
 274                         throw new DatabaseException("Unknown nozzle resource " + n+ " " + mapping.get(n));
 275                     }
 276                 }
 277             }
 278
 279             Set<Resource> components = new HashSet<>();
 280         for (Resource run : pipeRuns) {
 281             for (Resource c : graph.getObjects(run, P3D.children)) {
 282                 if (graph.isInstanceOf(c, P3D.PipelineComponent)) {
 283                     components.add(c);
 284                 } else {
 285                     throw new DatabaseException("Unknown component resource " + c + " " + mapping.get(c));
 286                 }
 287             }
 288         }
 289
 290         for (Resource c : components) {
 291             for (Resource connected : graph.getObjects(c, P3D.Connects)) {
 292                 if (!components.contains(connected) && !nozzles.contains(connected)) {
 293                     throw new DatabaseException("Unbrowsable component resource " + connected + " " + mapping.get(connected) +" connected to " + c + " " + mapping.get(c));
 294                 }
 295                 Collection<Resource> connectedConnected = graph.getObjects(connected, P3D.Connects);
 296                 if (!connectedConnected.contains(c)) {
 297                     throw new DatabaseException("Component resource " + c + " " + mapping.get(c) + " is connected to " + connected + " " + mapping.get(connected) +", but its has no connection back.");
 298                 }
 299             }
 300         }
 301         }
 302
 303 }