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