1 package org.simantics.plant3d.editor;
3 import java.util.ArrayList;
4 import java.util.HashSet;
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;
81 import vtk.vtkCameraPass;
82 import vtk.vtkDefaultPass;
83 import vtk.vtkGaussianBlurPass;
84 import vtk.vtkLightsPass;
86 import vtk.vtkRenderPassCollection;
87 import vtk.vtkRenderer;
88 import vtk.vtkSSAAPass;
89 import vtk.vtkSequencePass;
90 import vtk.vtkSimpleMotionBlurPass;
93 public class Plant3DEditor extends ResourceEditorPart {
95 private Composite parent;
96 protected ToolComposite toolComposite;
97 private Resource input;
98 // private InteractiveVtkPanel panel;
99 // private SWTAWTComponent component;
100 private InteractiveVtkComposite panel;
103 private P3DRootNode rootNode;
104 private IMapping<Resource,INode> mapping;
106 protected NodeSelectionProvider2<Resource, INode> selectionProvider;
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;
121 private P3DNodeMap nodeMap;
123 /** Constants for selecting the up-direction */
124 public static final int X = 0, Y = 1, Z = 2;
126 protected int upDirection = 1;
129 public void createPartControl(Composite parent) {
130 this.parent = parent;
131 //parent.setLayout (new FillLayout ());
134 // component = new SWTAWTComponent(parent,SWT.NONE) {
137 // protected Component createSwingComponent() {
138 // if (panel == null) {
139 // panel = new InteractiveVtkPanel();
140 // vtkPanelUtil.registerPanel(panel);
147 IResourceEditorInput rei = (IResourceEditorInput)getEditorInput();
148 input = rei.getResource();
150 toolComposite = new ToolComposite(parent, SWT.BORDER);
151 toolComposite.setVisible(true);
153 panel = new InteractiveVtkComposite(parent);
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());
159 GridDataFactory.fillDefaults().align(SWT.FILL, SWT.CENTER).grab(true, false).applyTo(toolComposite);
161 //IActionBars actionBars = getEditorSite().getActionBars();
167 new ContextMenuListener(panel, contextMenu);
169 cameraAction = createCameraAction();
170 switch (upDirection) {
172 cameraAction.setUpDirection(new double[] { 1, 0, 0 });
175 cameraAction.setUpDirection(new double[] { 0, 1, 0 });
178 cameraAction.setUpDirection(new double[] { 0, 0, 1 });
182 panel.setDefaultAction(cameraAction);
183 panel.useDefaultAction();
184 panel.setPickType(4);
187 ControlPointFactory.preloadCache(Simantics.getSession(), getLibraryUri());
188 ComponentUtils.preloadCache(Simantics.getSession());
189 } catch (Exception e) {
190 ExceptionUtils.logAndShowError("Cannot open Plant3D editor",e);
195 getSession().syncRequest(new ReadRequest() {
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.
208 P3DUtil.finalizeDBLoad(rootNode);
209 nodeMap = createNodeMap(getSession(), mapping, panel,rootNode);
211 } catch (Exception e) {
212 throw new DatabaseException(e);
215 //System.out.println("END PLANT3D LOAD");
219 if (rootNode == null)
220 throw new RuntimeException("Scenegraph loading failed.");
223 selectionProvider = createSelectionProvider();
225 cameraAction.addSelectionChangedListener(selectionProvider);
227 cameraAction.addHoverChangedListener(createHoverHighlhighter());
228 selectionProvider.addSelectionChangedListener(createSelectionHighlighter());
230 getSite().setSelectionProvider(selectionProvider);
231 getSite().getPage().addPostSelectionListener(selectionProvider);
233 //outlinePage = new ScenegraphOutlinePage(rootNode);
236 parent.addDisposeListener(new DisposeListener() {
239 public void widgetDisposed(DisposeEvent e) {
240 getSite().getPage().removePostSelectionListener(selectionProvider);
242 PipingRules.setEnabled(false);
244 PipingRules.setEnabled(true);
246 // component.dispose();
247 //panel.getComponent().dispose();
251 } catch (DatabaseException e1) {
252 ExceptionUtils.logAndShowError("Cannot open Plant3D editor",e1);
260 public void setUpDirection(int upDirection) {
261 this.upDirection = upDirection;
264 protected vtkCameraAndSelectorAction createCameraAction() {
265 return new vtkCameraAndSelectorAction(panel);
268 protected NodeSelectionProvider2<Resource,INode> createSelectionProvider() {
269 return new NodeSelectionProvider2<Resource,INode>(this,mapping,nodeMap);
272 protected HoverHighlighter<Resource> createHoverHighlhighter() {
273 return new HoverHighlighter<>(panel,nodeMap);
276 protected SelectionHighlighter<Resource> createSelectionHighlighter() {
277 return new SelectionHighlighter<Resource>(panel,nodeMap);
280 protected String getLibraryUri() {
281 return Plant3D.URIs.Builtin;
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);
297 public void populate() {
298 ThreadUtils.asyncExec(panel.getThreadQueue(), new Runnable() {
303 panel.addListener(new RenderListener() {
306 public void preRender() {
311 public void postRender() {
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);
320 panel.removeListener(this);
330 protected IMappingSchema<Resource, INode> getSchema(ReadGraph graph) throws DatabaseException {
331 IMappingSchema<Resource,INode> schema = SchemaBuilder.getSchema(graph);
335 protected P3DNodeMap createNodeMap(Session session, IMapping<Resource, INode> mapping, VtkView panel, P3DRootNode rootNode) {
336 return new P3DNodeMap(session, mapping, panel,rootNode);
340 public void setFocus() {
341 //component.setFocus();
342 panel.getComponent().setFocus();
345 private void createScene() {
346 vtkRenderer ren1 = panel.getRenderer();
348 boolean multiPass = false;
349 boolean blur = false;
350 boolean ssaa = false;
351 //boolean sobel = true;
352 boolean mblur = false;
355 vtkLightsPass lightsPass = new vtkLightsPass();
356 vtkDefaultPass defaultPass = new vtkDefaultPass();
359 vtkRenderPassCollection passes = new vtkRenderPassCollection();
360 passes.AddItem(lightsPass);
361 passes.AddItem(defaultPass);
363 vtkSequencePass seq = new vtkSequencePass();
364 seq.SetPasses(passes);
366 vtkCameraPass cameraPass = new vtkCameraPass();
367 cameraPass.SetDelegatePass(seq);
370 vtkGaussianBlurPass blurPass = new vtkGaussianBlurPass();
371 blurPass.SetDelegatePass(cameraPass);
372 ren1.SetPass(blurPass);
374 vtkSSAAPass ssaaPass = new vtkSSAAPass();
375 ssaaPass.SetDelegatePass(cameraPass);
376 ren1.SetPass(ssaaPass);
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);
386 ren1.SetPass(cameraPass);
390 // ren1.GetRenderWindow().LineSmoothingOn();
391 // ren1.GetRenderWindow().PointSmoothingOn();
392 // ren1.GetRenderWindow().PolygonSmoothingOn();
393 // ren1.GetRenderWindow().SetMultiSamples(2);
397 ren1.SetBackground2(1,1,1); // background color white
398 ren1.SetBackground(0.9,0.9,0.9);
399 ren1.SetGradientBackground(true);
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);
406 panel.addDeletable(grid);
408 AxesDisplay axesDisplay = new AxesDisplay(panel);
413 protected Menu contextMenu;
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);
424 contextMenu = menuMgr.createContextMenu(parent);
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);
437 if (props.size() > 0) {
438 focusAction.setProps(props);
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));
448 } else if (selected.size() == 1) {
449 IP3DNode node = (IP3DNode)selected.get(0);
450 if (node instanceof Equipment) {
451 m.add(translateAction);
453 for (Item eq : P3DUtil.getNozzles(Simantics.getSession(), getLibraryUri())) {
454 AddNozzleAction add = new AddNozzleAction(rootNode, eq);
455 add.setEquipment((Equipment)node);
459 } else if (node instanceof Nozzle) {
460 Nozzle nozzle = (Nozzle)node;
461 if (!nozzle.isFixed()) {
462 m.add(translateAction);
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);
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);
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);
487 } else if (node instanceof InlineComponent) {
488 //m.add(translateInlineAction);
489 InlineComponent component = (InlineComponent)node;
490 if (component.isVariableLength())
491 m.add(translateFreeVariableLengthAction);
493 m.add(translateInlineAction);
494 m.add(routePipeAction);
495 routePipeAction.setComponent(component);
496 m.add(addComponentAction);
497 addComponentAction.setComponent(component);
499 m.add(removeSplitAction);
500 removeSplitAction.setNode(node);
501 } else if (node instanceof PipeRun) {
502 m.add(reversePipeRunAction);
504 reversePipeRunAction.setNode(node);
510 translateAction.setNode(node);
511 translateInlineAction.setNode(node);
512 translateFreeVariableLengthAction.setNode(node);
513 rotateAction.setNode(node);
514 removeAction.setNode(node);
517 } catch (DatabaseException e) {
518 ExceptionUtils.logAndShowError(e);
522 private IContentOutlinePage createOutline() {
523 if (rootNode == null || selectionProvider == 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);
531 outlinePage.addSelectionChangedListener(new ISelectionChangedListener() {
534 public void selectionChanged(SelectionChangedEvent event) {
535 selectionProvider.selectionChanged(event);
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());
550 if (adapter.isAssignableFrom(NodeMap.class)) {
551 return adapter.cast(nodeMap);
553 if (adapter.isAssignableFrom(INode.class)) {
554 return adapter.cast(rootNode);
556 if (adapter.isAssignableFrom(IMapping.class)) {
557 return adapter.cast(mapping);
559 // if (adapter.isAssignableFrom(InteractiveVtkPanel.class)) {
560 // return adapter.cast(panel);
562 if (adapter.isAssignableFrom(VtkView.class))
563 return adapter.cast(panel);
564 return super.getAdapter(adapter);
567 public Set<String> getPropertyContexts() {
568 Set<String> result = new HashSet<String>();
569 result.add("http://www.simantics.org/Project-1.0/ProjectBrowseContext");
573 public InteractiveVtkComposite getPanel() {
577 public P3DRootNode getRootNode() {
581 public IMapping<Resource, INode> getMapping() {
585 public P3DNodeMap getNodeMap() {