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