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         }
  95
  96         private void updateTransform(IP3DNode node) {
  97                 if (DEBUG) System.out.println("P3DNodeMap update Transform " + debugString(node));
  98
  99                 node.update(view.getRenderer());
 100
 101                 if (node instanceof ParentNode<?>) {
 102                         ParentNode<IP3DNode> p = (ParentNode<IP3DNode>)node;
 103                         for (IP3DNode n : p.getNodes())
 104                                 updateTransform(n);
 105                 }
 106         }
 107
 108         @Override
 109         protected Collection<vtkProp> getActors(INode n) {
 110                 List<vtkProp> props = new ArrayList<vtkProp>();
 111                 if (!(n instanceof IP3DVisualNode))
 112                         return props;
 113                 IP3DVisualNode node = (IP3DVisualNode)n;
 114                 for (vtkProp3D p : ((IP3DVisualNode)node).getActors())
 115                         props.add(p);
 116
 117                 return props;
 118         }
 119
 120         @Override
 121         protected void removeActor(INode n) {
 122                 if (DEBUG) System.out.println("P3DNodeMap.removeActor " + debugString(n));
 123                 if (!(n instanceof IP3DVisualNode))
 124                         return;
 125                 IP3DVisualNode node = (IP3DVisualNode)n;
 126                 remActor(node);
 127
 128                 if (node instanceof P3DParentNode<?>) {
 129                         for (IP3DNode n2 : ((P3DParentNode<?>)node).getNodes())
 130                                 if (n2 instanceof IP3DVisualNode)
 131                                         removeActor((IP3DVisualNode)n2);
 132                 }
 133         }
 134
 135         @Override
 136         protected void addActor(INode n) {
 137                 if (DEBUG) System.out.println("P3DNodeMap.addActor " + debugString(n));
 138                 if (!(n instanceof IP3DVisualNode))
 139                         return;
 140                 IP3DVisualNode node = (IP3DVisualNode)n;
 141
 142                 if (hasActor(node)) {
 143                     if (DEBUG) System.out.println("Node already has an actor");
 144                         return;
 145                 }
 146                 if (Thread.currentThread() != view.getThreadQueue().getThread())
 147                         throw new RuntimeException("Illegal thread.");
 148
 149                 view.lock();
 150
 151                 node.visualize(view);
 152
 153                 for (vtkProp3D act : node.getActors()) {
 154                         nodeToActor.add(node, act);
 155                         actorToNode.put(act, node);
 156                 }
 157                 if (DEBUG) System.out.println("Added " + node.getActors().size() + " actors");
 158
 159                 if (node instanceof P3DParentNode<?>) {
 160                         for (IP3DNode n2 : ((P3DParentNode<?>)node).getNodes())
 161                                 if (n2 instanceof IP3DVisualNode)
 162                                         addActor((IP3DVisualNode)n2);
 163                 }
 164
 165                 updateTransform(node);
 166
 167                 view.unlock();
 168
 169         }
 170
 171
 172
 173         private boolean hasActor(IP3DVisualNode node) {
 174                 List<vtkProp> list = nodeToActor.getValues(node);
 175                 if (list == null || list.size() == 0)
 176                         return false;
 177                 return true;
 178         }
 179
 180         private void remActor(IP3DVisualNode node) {
 181                 if (Thread.currentThread() != view.getThreadQueue().getThread())
 182                         throw new RuntimeException("Illegal thread.");
 183
 184                 List<vtkProp> list = nodeToActor.getValues(node);
 185                 if (list != null) {
 186                         for (vtkProp obj : list) {
 187                                 actorToNode.remove(obj);
 188                         }
 189                         nodeToActor.remove(node);
 190                         view.lock();
 191
 192                         node.stopVisualize();
 193
 194                         view.unlock();
 195                 }
 196         }
 197
 198         @Override
 199         protected void update(ReadGraph graph) throws DatabaseException {
 200                 validate();
 201 //              System.out.println("Graph updates");
 202                 super.update(graph);
 203                 validate();
 204         }
 205
 206         @Override
 207         public void commit(String commitMessage) {
 208                 validate();
 209 //              System.out.println("Graph commit");
 210                 super.commit(commitMessage);
 211
 212         }
 213
 214         @Override
 215         protected void commit(WriteGraph graph) throws DatabaseException {
 216             super.commit(graph);
 217             validateGraph(graph);
 218         }
 219         @Override
 220         protected void doCommit() {
 221 //              System.out.println("Do commit");
 222                 validate();
 223                 super.doCommit();
 224         }
 225
 226         private void validate() {
 227                 for (INode node : rootNode.getNodes()) {
 228                         if (node instanceof PipeRun)
 229                                 PipingRules.validate((PipeRun)node);
 230                 }
 231         }
 232
 233         @Override
 234         public synchronized void preRender() {
 235 //              System.out.println("P3DNodeMap preRender");
 236 //              super.preRender();
 237                 try {
 238 //                      boolean b = false;
 239 //                      synchronized (syncMutex) {
 240 //                              b = PipingRules.update();
 241 //                      }
 242 //                      if (b)
 243 //                              super.preRender();
 244                         boolean b = true;
 245                         while (b) {
 246                                 updateCycle();
 247                                 b = PipingRules.update();
 248                         }
 249                 } catch (Exception e) {
 250                         e.printStackTrace();
 251                 }
 252
 253         }
 254
 255         protected void validateGraph(ReadGraph graph) throws DatabaseException {
 256             Plant3D P3D = Plant3D.getInstance(graph);
 257             Resource root = (Resource)mapping.inverseGet(rootNode);
 258             if (!graph.isInstanceOf(root, P3D.Plant))
 259                 throw new DatabaseException("Root is not a Plant");
 260             Set<Resource> equipment = new HashSet<>();
 261             Set<Resource> pipeRuns = new HashSet<>();
 262             for (Resource r : graph.getObjects(root, P3D.children)) {
 263                 if (graph.isInstanceOf(r, P3D.Equipment)) {
 264                     equipment.add(r);
 265                 } else if (graph.isInstanceOf(r, P3D.PipeRun)) {
 266                     pipeRuns.add(r);
 267                 } else {
 268                     throw new DatabaseException("Unknown resource " + r+ " " + mapping.get(r));
 269                 }
 270             }
 271             Set<Resource> nozzles = new HashSet<>();
 272             for (Resource e : equipment) {
 273                 for (Resource n : graph.getObjects(e, P3D.HasNozzle)) {
 274                     if (graph.isInstanceOf(n, P3D.Nozzle)) {
 275                         nozzles.add(n);
 276                     } else {
 277                         throw new DatabaseException("Unknown nozzle resource " + n+ " " + mapping.get(n));
 278                     }
 279                 }
 280             }
 281
 282             Set<Resource> components = new HashSet<>();
 283         for (Resource run : pipeRuns) {
 284             for (Resource c : graph.getObjects(run, P3D.children)) {
 285                 if (graph.isInstanceOf(c, P3D.PipelineComponent)) {
 286                     components.add(c);
 287                 } else {
 288                     throw new DatabaseException("Unknown component resource " + c + " " + mapping.get(c));
 289                 }
 290             }
 291         }
 292
 293         for (Resource c : components) {
 294             for (Resource connected : graph.getObjects(c, P3D.Connects)) {
 295                 if (!components.contains(connected) && !nozzles.contains(connected)) {
 296                     throw new DatabaseException("Unbrowsable component resource " + connected + " " + mapping.get(connected) +" connected to " + c + " " + mapping.get(c));
 297                 }
 298                 Collection<Resource> connectedConnected = graph.getObjects(connected, P3D.Connects);
 299                 if (!connectedConnected.contains(c)) {
 300                     throw new DatabaseException("Component resource " + c + " " + mapping.get(c) + " is connected to " + connected + " " + mapping.get(connected) +", but its has no connection back.");
 301                 }
 302             }
 303         }
 304         }
 305
 306 }