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

   1 package org.simantics.plant3d.editor;
   2
   3 import java.util.HashSet;
   4 import java.util.List;
   5 import java.util.Set;
   6
   7 import org.eclipse.jface.action.IMenuListener;
   8 import org.eclipse.jface.action.IMenuManager;
   9 import org.eclipse.jface.action.MenuManager;
  10 import org.eclipse.jface.viewers.ISelectionChangedListener;
  11 import org.eclipse.jface.viewers.ISelectionProvider;
  12 import org.eclipse.jface.viewers.SelectionChangedEvent;
  13 import org.eclipse.swt.events.DisposeEvent;
  14 import org.eclipse.swt.events.DisposeListener;
  15 import org.eclipse.swt.layout.FillLayout;
  16 import org.eclipse.swt.widgets.Composite;
  17 import org.eclipse.swt.widgets.Menu;
  18 import org.eclipse.ui.views.contentoutline.IContentOutlinePage;
  19 import org.simantics.db.ReadGraph;
  20 import org.simantics.db.Resource;
  21 import org.simantics.db.Session;
  22 import org.simantics.db.common.request.ReadRequest;
  23 import org.simantics.db.exception.DatabaseException;
  24 import org.simantics.g3d.scenegraph.IG3DNode;
  25 import org.simantics.g3d.scenegraph.NodeMap;
  26 import org.simantics.g3d.scenegraph.RenderListener;
  27 import org.simantics.g3d.scenegraph.base.INode;
  28 import org.simantics.g3d.vtk.action.RemoveAction;
  29 import org.simantics.g3d.vtk.common.HoverHighlighter;
  30 import org.simantics.g3d.vtk.common.NodeSelectionProvider2;
  31 import org.simantics.g3d.vtk.common.SelectionHighlighter;
  32 import org.simantics.g3d.vtk.common.VtkView;
  33 import org.simantics.g3d.vtk.shape.vtkShape;
  34 import org.simantics.g3d.vtk.swt.ContextMenuListener;
  35 import org.simantics.g3d.vtk.swt.InteractiveVtkComposite;
  36 import org.simantics.g3d.vtk.swt.RotateAction;
  37 import org.simantics.g3d.vtk.swt.TranslateAction;
  38 import org.simantics.g3d.vtk.swt.vtkCameraAndSelectorAction;
  39 import org.simantics.g3d.vtk.utils.AxesDisplay;
  40 import org.simantics.objmap.graph.IMapping;
  41 import org.simantics.objmap.graph.Mappings;
  42 import org.simantics.objmap.graph.schema.IMappingSchema;
  43 import org.simantics.plant3d.actions.AddComponentAction;
  44 import org.simantics.plant3d.actions.AddEquipmentAction;
  45 import org.simantics.plant3d.actions.AddNozzleAction;
  46 import org.simantics.plant3d.actions.RemoveAndSplitAction;
  47 import org.simantics.plant3d.actions.RoutePipeAction;
  48 import org.simantics.plant3d.actions.TranslateFreeVariableLengthAction;
  49 import org.simantics.plant3d.actions.TranslateInlineAction;
  50 import org.simantics.plant3d.ontology.Plant3D;
  51 import org.simantics.plant3d.scenegraph.EndComponent;
  52 import org.simantics.plant3d.scenegraph.Equipment;
  53 import org.simantics.plant3d.scenegraph.IP3DNode;
  54 import org.simantics.plant3d.scenegraph.InlineComponent;
  55 import org.simantics.plant3d.scenegraph.Nozzle;
  56 import org.simantics.plant3d.scenegraph.P3DRootNode;
  57 import org.simantics.plant3d.scenegraph.PipelineComponent;
  58 import org.simantics.plant3d.scenegraph.SchemaBuilder;
  59 import org.simantics.plant3d.scenegraph.TurnComponent;
  60 import org.simantics.plant3d.scenegraph.controlpoint.ControlPointFactory;
  61 import org.simantics.plant3d.scenegraph.controlpoint.PipingRules;
  62 import org.simantics.plant3d.utils.ComponentUtils;
  63 import org.simantics.plant3d.utils.Item;
  64 import org.simantics.plant3d.utils.P3DUtil;
  65 import org.simantics.selectionview.StandardPropertyPage;
  66 import org.simantics.ui.workbench.IPropertyPage;
  67 import org.simantics.ui.workbench.IResourceEditorInput;
  68 import org.simantics.ui.workbench.ResourceEditorPart;
  69 import org.simantics.utils.threads.ThreadUtils;
  70 import org.simantics.utils.ui.ExceptionUtils;
  71
  72 import vtk.vtkActor;
  73 import vtk.vtkCameraPass;
  74 import vtk.vtkDefaultPass;
  75 import vtk.vtkLightsPass;
  76 import vtk.vtkRenderPassCollection;
  77 import vtk.vtkRenderer;
  78 import vtk.vtkSequencePass;
  79
  80
  81 public class Plant3DEditor extends ResourceEditorPart {
  82
  83         private Composite parent;
  84         private Resource input;
  85 //      private InteractiveVtkPanel panel;
  86 //      private SWTAWTComponent component;
  87         private InteractiveVtkComposite panel;
  88
  89         private P3DRootNode rootNode;
  90         private IMapping<Resource,Object> mapping;
  91
  92         private NodeSelectionProvider2<Resource,Object> selectionProvider;
  93
  94         protected vtkCameraAndSelectorAction cameraAction;
  95         protected TranslateAction translateAction;
  96         protected TranslateInlineAction translateInlineAction;
  97         protected TranslateFreeVariableLengthAction translateFreeVariableLengthAction;
  98         protected RotateAction rotateAction;
  99         protected RemoveAction removeAction;
 100         protected RemoveAndSplitAction removeSplitAction;
 101         protected RoutePipeAction routePipeAction;
 102         protected AddComponentAction addComponentAction;
 103
 104         private P3DNodeMap nodeMap;
 105
 106         @Override
 107         public void createPartControl(Composite parent) {
 108                 this.parent = parent;
 109                 parent.setLayout (new FillLayout ());
 110 //              component = new SWTAWTComponent(parent,SWT.NONE) {
 111 //
 112 //                      @Override
 113 //                      protected Component createSwingComponent() {
 114 //                              if (panel == null) {
 115 //                                      panel = new InteractiveVtkPanel();
 116 //                                      vtkPanelUtil.registerPanel(panel);
 117 //                                      createScene();
 118 //                              }
 119 //                              return panel;
 120 //                      }
 121 //              };
 122
 123                 IResourceEditorInput rei = (IResourceEditorInput)getEditorInput();
 124                 input = rei.getResource();
 125
 126                 panel = new InteractiveVtkComposite(parent);
 127
 128
 129                 //IActionBars actionBars = getEditorSite().getActionBars();
 130
 131                 hookContextMenu();
 132
 133                 createScene();
 134
 135                 new ContextMenuListener(panel, contextMenu);
 136
 137                 cameraAction = createCameraAction();
 138                 panel.setDefaultAction(cameraAction);
 139                 panel.useDefaultAction();
 140                 panel.setPickType(4);
 141
 142                 try {
 143                         ControlPointFactory.preloadCache(getLibraryUri());
 144                         ComponentUtils.preloadCache();
 145                 } catch (Exception e) {
 146                         ExceptionUtils.logAndShowError("Cannot open Plant3D editor",e);
 147                         return;
 148                 }
 149
 150                 try {
 151                         getSession().syncRequest(new ReadRequest() {
 152
 153                                 @SuppressWarnings({ "rawtypes", "unchecked" })
 154                                 @Override
 155                                 public void run(ReadGraph graph) throws DatabaseException {
 156                                     System.out.println("START PLANT3D LOAD");
 157                                         PipingRules.setEnabled(false);
 158                                         IMappingSchema<Resource, Object> schema = getSchema(graph);
 159                                         mapping = Mappings.createWithListening(schema);
 160                                         rootNode = (P3DRootNode)mapping.map(graph, input);
 161                                         // update control points.
 162                                         // TODO : this should be optimized.
 163
 164                                         try {
 165                                             P3DUtil.finalizeDBLoad(rootNode);
 166                                             nodeMap = createNodeMap(getSession(), mapping, panel,rootNode);
 167
 168                     } catch (Exception e) {
 169                         throw new DatabaseException(e);
 170                     }
 171
 172                                         System.out.println("END PLANT3D LOAD");
 173                                 }
 174                         });
 175
 176                         if (rootNode == null)
 177                                 throw new RuntimeException("Scenegraph loading failed.");
 178                         populate();
 179
 180                         selectionProvider = createSelectionProvider();
 181
 182                         cameraAction.addSelectionChangedListener(selectionProvider);
 183
 184                         cameraAction.addHoverChangedListener(createHoverHighlhighter());
 185                         selectionProvider.addSelectionChangedListener(createSelectionHighlighter());
 186
 187                         getSite().setSelectionProvider(selectionProvider);
 188                         getSite().getPage().addPostSelectionListener(selectionProvider);
 189
 190                         //outlinePage = new ScenegraphOutlinePage(rootNode);
 191
 192
 193                         parent.addDisposeListener(new DisposeListener() {
 194
 195                                 @Override
 196                                 public void widgetDisposed(DisposeEvent e) {
 197                                         getSite().getPage().removePostSelectionListener(selectionProvider);
 198
 199                                         PipingRules.setEnabled(false);
 200                                         nodeMap.delete();
 201                                         PipingRules.setEnabled(true);
 202                                         mapping.dispose();
 203 //                                      component.dispose();
 204                                         //panel.getComponent().dispose();
 205
 206                                 }
 207                         });
 208                 } catch (DatabaseException e1) {
 209                         ExceptionUtils.logAndShowError("Cannot open Plant3D editor",e1);
 210                         return;
 211                 }
 212
 213                 createActions();
 214
 215         }
 216
 217         protected vtkCameraAndSelectorAction createCameraAction() {
 218             return new vtkCameraAndSelectorAction(panel);
 219         }
 220
 221         protected NodeSelectionProvider2<Resource,Object> createSelectionProvider() {
 222             return new NodeSelectionProvider2<Resource,Object>(this,mapping,nodeMap);
 223         }
 224
 225         protected HoverHighlighter createHoverHighlhighter() {
 226             return new HoverHighlighter(panel,nodeMap);
 227         }
 228
 229         protected SelectionHighlighter<Resource,INode> createSelectionHighlighter() {
 230             return new SelectionHighlighter<Resource,INode>(panel,nodeMap);
 231         }
 232
 233         protected String getLibraryUri() {
 234             return Plant3D.URIs.Builtin;
 235         }
 236
 237         protected void createActions() {
 238                 translateAction = new TranslateAction(panel,nodeMap);
 239                 translateInlineAction = new TranslateInlineAction(panel, nodeMap);
 240                 translateFreeVariableLengthAction = new TranslateFreeVariableLengthAction(panel, getRootNode());
 241                 rotateAction = new RotateAction(panel,nodeMap);
 242                 removeAction = new RemoveAction(nodeMap) {
 243                         public void setNode(IG3DNode node) {
 244                                 super.setNode(node);
 245                         }
 246                 };
 247
 248                 removeSplitAction = new RemoveAndSplitAction(nodeMap);
 249                 routePipeAction = new RoutePipeAction(panel,rootNode);
 250                 addComponentAction = new AddComponentAction(panel, rootNode, getLibraryUri());
 251         }
 252
 253         public void populate() {
 254                 ThreadUtils.asyncExec(panel.getThreadQueue(), new Runnable() {
 255
 256                         @Override
 257                         public void run() {
 258                                 nodeMap.populate();
 259                                 panel.addListener(new RenderListener() {
 260
 261                     @Override
 262                     public void preRender() {
 263
 264                     }
 265
 266                     @Override
 267                     public void postRender() {
 268                         try {
 269                             P3DUtil.finalizeDBLoad2(rootNode);
 270                             if (nodeMap.isRangeModified());
 271                                 nodeMap.commit("Load sync");
 272                         } catch (Exception e) {
 273                             //throw new DatabaseException(e);
 274                         }
 275                         panel.removeListener(this);
 276
 277                     }
 278                 });
 279
 280                         }
 281                 });
 282
 283         }
 284
 285         protected IMappingSchema<Resource, Object> getSchema(ReadGraph graph) throws DatabaseException {
 286                 IMappingSchema<Resource,Object> schema = SchemaBuilder.getSchema(graph);
 287                 return schema;
 288         }
 289
 290         protected P3DNodeMap createNodeMap(Session session, IMapping<Resource, Object> mapping, VtkView panel, P3DRootNode rootNode) {
 291                 return new P3DNodeMap(session, mapping, panel,rootNode);
 292         }
 293
 294         @Override
 295         public void setFocus() {
 296                 //component.setFocus();
 297                 panel.getComponent().setFocus();
 298         }
 299
 300         private void createScene() {
 301                 vtkRenderer ren1 = panel.getRenderer();
 302
 303                 boolean multiPass = false;
 304                 if (multiPass) {
 305
 306                         vtkLightsPass lightsPass = new vtkLightsPass();
 307                         vtkDefaultPass defaultPass = new vtkDefaultPass();
 308
 309
 310                         vtkRenderPassCollection passes = new vtkRenderPassCollection();
 311                         passes.AddItem(lightsPass);
 312                         passes.AddItem(defaultPass);
 313
 314                         vtkSequencePass seq = new vtkSequencePass();
 315                         seq.SetPasses(passes);
 316
 317                         vtkCameraPass cameraPass = new vtkCameraPass();
 318                         cameraPass.SetDelegatePass(seq);
 319
 320                         ren1.SetPass(cameraPass);
 321
 322                 }
 323 //              ren1.GetRenderWindow().LineSmoothingOn();
 324 //              ren1.GetRenderWindow().PointSmoothingOn();
 325 //              ren1.GetRenderWindow().PolygonSmoothingOn();
 326 //              ren1.GetRenderWindow().SetMultiSamples(2);
 327
 328
 329
 330                 ren1.SetBackground2(1,1,1); // background color white
 331                 ren1.SetBackground(0.9,0.9,0.9);
 332                 ren1.SetGradientBackground(true);
 333
 334                 // vtkActor grid = vtkShape.createGridActor(8,1.0,1|2|4);
 335                 vtkActor grid = vtkShape.createGridActor(8,1.0, 2 );
 336                 grid.SetPickable(0);
 337                 ren1.AddActor(grid);
 338                 panel.addDeletable(grid);
 339
 340                 AxesDisplay axesDisplay = new AxesDisplay(panel);
 341                 axesDisplay.show();
 342
 343         }
 344
 345         protected Menu contextMenu;
 346
 347         protected void hookContextMenu() {
 348                 MenuManager menuMgr = new MenuManager("#PopupMenu");
 349                 menuMgr.setRemoveAllWhenShown(true);
 350                 menuMgr.addMenuListener(new IMenuListener() {
 351                         public void menuAboutToShow(IMenuManager manager) {
 352                                 createContextMenu(manager);
 353                         }
 354                 });
 355
 356                 contextMenu = menuMgr.createContextMenu(parent);
 357         }
 358
 359         protected void createContextMenu(IMenuManager m) {
 360                 List<IG3DNode> selected = selectionProvider.getSelectedNodes();
 361                 try {
 362                         if (selected.size() == 0) {
 363                                 for (Item eq : P3DUtil.getEquipments(getLibraryUri())) {
 364                                         m.add(new AddEquipmentAction(rootNode, eq));
 365                                 }
 366                         } else if (selected.size() == 1) {
 367                                 IP3DNode node = (IP3DNode)selected.get(0);
 368                                 if (node instanceof Equipment) {
 369                                         m.add(translateAction);
 370                                         m.add(rotateAction);
 371                                         for (Item eq : P3DUtil.getNozzles(getLibraryUri())) {
 372                                                 AddNozzleAction add = new AddNozzleAction(rootNode, eq);
 373                                                 add.setEquipment((Equipment)node);
 374                                                 m.add(add);
 375                                         }
 376                                         m.add(removeAction);
 377                                 } else if (node instanceof Nozzle) {
 378                                         Nozzle nozzle = (Nozzle)node;
 379                                         if (!nozzle.isFixed()) {
 380                                                 m.add(translateAction);
 381                                                 m.add(rotateAction);
 382                                         }
 383                                         m.add(routePipeAction);
 384                                         routePipeAction.setComponent(nozzle);
 385                                         routePipeAction.setEnabled(nozzle.getNext() == null && nozzle.getPrevious() == null);
 386                                         m.add(addComponentAction);
 387                                         addComponentAction.setComponent(nozzle);
 388                                         m.add(removeAction);
 389                                 } else if (node instanceof TurnComponent) {
 390                                         m.add(translateAction);
 391                                         TurnComponent component = (TurnComponent)node;
 392                                         m.add(routePipeAction);
 393                                         routePipeAction.setComponent(component);
 394                                         routePipeAction.setEnabled(component.getNext() == null || component.getPrevious() == null);
 395                                         m.add(addComponentAction);
 396                                         addComponentAction.setComponent(component);
 397                                         m.add(removeAction);
 398                                         m.add(removeSplitAction);
 399                                         removeSplitAction.setNode(node);
 400                                 } else if (node instanceof EndComponent) {
 401                                         m.add(translateAction);
 402                                         m.add(addComponentAction);
 403                                         addComponentAction.setComponent((PipelineComponent)node);
 404                                         m.add(removeAction);
 405                                 } else if (node instanceof InlineComponent) {
 406                                         //m.add(translateInlineAction);
 407                                         InlineComponent component = (InlineComponent)node;
 408                                         if (component.isVariableLength())
 409                                             m.add(translateFreeVariableLengthAction);
 410                                         else
 411                                             m.add(translateInlineAction);
 412                                         m.add(routePipeAction);
 413                                         routePipeAction.setComponent(component);
 414                                         m.add(addComponentAction);
 415                                         addComponentAction.setComponent(component);
 416                                         m.add(removeAction);
 417                                         m.add(removeSplitAction);
 418                                         removeSplitAction.setNode(node);
 419                                 } else {
 420                                     m.add(removeAction);
 421                                 }
 422
 423
 424                                 translateAction.setNode(node);
 425                                 translateInlineAction.setNode(node);
 426                                 translateFreeVariableLengthAction.setNode(node);
 427                                 rotateAction.setNode(node);
 428                                 removeAction.setNode(node);
 429
 430                         }
 431                 } catch (DatabaseException e) {
 432                         ExceptionUtils.logAndShowError(e);
 433                 }
 434         }
 435
 436         private IContentOutlinePage createOutline() {
 437                 if (rootNode == null || selectionProvider == null)
 438                         return null;
 439                 //IContentOutlinePage outlinePage = new VTKContentOutlinePage<Resource,Object>(rootNode, selectionProvider);
 440                 IContentOutlinePage outlinePage = new P3DContentOutlinePage(rootNode, selectionProvider) {
 441                         protected void createContextMenu(IMenuManager manager) {
 442                                 Plant3DEditor.this.createContextMenu(manager);
 443                         };
 444                 };
 445                 outlinePage.addSelectionChangedListener(new ISelectionChangedListener() {
 446
 447                         @Override
 448                         public void selectionChanged(SelectionChangedEvent event) {
 449                                 selectionProvider.selectionChanged(event);
 450                         }
 451                 });
 452                 return outlinePage;
 453         }
 454
 455         @SuppressWarnings("rawtypes")
 456         @Override
 457         public Object getAdapter(Class adapter) {
 458                 if (IPropertyPage.class.equals(adapter))
 459                         return new StandardPropertyPage(getSite(),getPropertyContexts());
 460                 if (IContentOutlinePage.class.equals(adapter)) {
 461                         return createOutline();
 462                 }
 463                 if (NodeMap.class.equals(adapter)) {
 464                         return nodeMap;
 465                 }
 466                 if (INode.class.equals(adapter)) {
 467                         return rootNode;
 468                 }
 469                 if (IMapping.class.equals(adapter)) {
 470                         return mapping;
 471                 }
 472 //              if (InteractiveVtkPanel.class.equals(adapter)) {
 473 //                      return panel;
 474 //              }
 475                 if (VtkView.class.equals(adapter))
 476                         return panel;
 477                 if (ISelectionProvider.class.equals(adapter))
 478                         return selectionProvider;
 479                 return super.getAdapter(adapter);
 480         }
 481
 482         public Set<String> getPropertyContexts() {
 483                 Set<String> result = new HashSet<String>();
 484                 result.add("http://www.simantics.org/Project-1.0/ProjectBrowseContext");
 485                 return result;
 486         }
 487
 488         public InteractiveVtkComposite getPanel() {
 489         return panel;
 490     }
 491
 492         public P3DRootNode getRootNode() {
 493                 return rootNode;
 494         }
 495
 496         public IMapping<Resource, Object> getMapping() {
 497                 return mapping;
 498         }
 499
 500         public P3DNodeMap getNodeMap() {
 501         return nodeMap;
 502     }
 503 }