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

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