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