1 package org.simantics.plant3d.editor;
3 import java.util.ArrayList;
4 import java.util.Collection;
5 import java.util.HashSet;
9 import org.eclipse.jface.action.Action;
10 import org.eclipse.jface.action.IMenuListener;
11 import org.eclipse.jface.action.IMenuManager;
12 import org.eclipse.jface.action.MenuManager;
13 import org.eclipse.jface.action.Separator;
14 import org.eclipse.jface.layout.GridDataFactory;
15 import org.eclipse.jface.layout.GridLayoutFactory;
16 import org.eclipse.jface.viewers.ISelectionChangedListener;
17 import org.eclipse.jface.viewers.ISelectionProvider;
18 import org.eclipse.jface.viewers.SelectionChangedEvent;
19 import org.eclipse.swt.SWT;
20 import org.eclipse.swt.events.DisposeEvent;
21 import org.eclipse.swt.events.DisposeListener;
22 import org.eclipse.swt.widgets.Composite;
23 import org.eclipse.swt.widgets.Menu;
24 import org.eclipse.ui.views.contentoutline.IContentOutlinePage;
25 import org.simantics.Simantics;
26 import org.simantics.db.ReadGraph;
27 import org.simantics.db.Resource;
28 import org.simantics.db.Session;
29 import org.simantics.db.common.request.ReadRequest;
30 import org.simantics.db.exception.DatabaseException;
31 import org.simantics.g3d.scenegraph.NodeMap;
32 import org.simantics.g3d.scenegraph.RenderListener;
33 import org.simantics.g3d.scenegraph.base.INode;
34 import org.simantics.g3d.scenegraph.base.ParentNode;
35 import org.simantics.g3d.toolbar.ToolComposite;
36 import org.simantics.g3d.vtk.action.RemoveAction;
37 import org.simantics.g3d.vtk.common.HoverHighlighter;
38 import org.simantics.g3d.vtk.common.NodeSelectionProvider2;
39 import org.simantics.g3d.vtk.common.SelectionHighlighter;
40 import org.simantics.g3d.vtk.common.VtkView;
41 import org.simantics.g3d.vtk.shape.vtkShape;
42 import org.simantics.g3d.vtk.swt.ContextMenuListener;
43 import org.simantics.g3d.vtk.swt.FocusAction;
44 import org.simantics.g3d.vtk.swt.InteractiveVtkComposite;
45 import org.simantics.g3d.vtk.swt.RotateAction;
46 import org.simantics.g3d.vtk.swt.TranslateAction;
47 import org.simantics.g3d.vtk.swt.vtkCameraAndSelectorAction;
48 import org.simantics.g3d.vtk.utils.AxesDisplay;
49 import org.simantics.objmap.graph.IMapping;
50 import org.simantics.objmap.graph.Mappings;
51 import org.simantics.objmap.graph.schema.IMappingSchema;
52 import org.simantics.plant3d.actions.AddComponentAction;
53 import org.simantics.plant3d.actions.AddEquipmentAction;
54 import org.simantics.plant3d.actions.AddNozzleAction;
55 import org.simantics.plant3d.actions.RemoveAndSplitAction;
56 import org.simantics.plant3d.actions.ReversePipeRunAction;
57 import org.simantics.plant3d.actions.RoutePipeAction;
58 import org.simantics.plant3d.actions.TranslateFreeVariableLengthAction;
59 import org.simantics.plant3d.actions.TranslateInlineAction;
60 import org.simantics.plant3d.ontology.Plant3D;
61 import org.simantics.plant3d.scenegraph.EndComponent;
62 import org.simantics.plant3d.scenegraph.Equipment;
63 import org.simantics.plant3d.scenegraph.IP3DNode;
64 import org.simantics.plant3d.scenegraph.IP3DVisualNode;
65 import org.simantics.plant3d.scenegraph.InlineComponent;
66 import org.simantics.plant3d.scenegraph.Nozzle;
67 import org.simantics.plant3d.scenegraph.P3DRootNode;
68 import org.simantics.plant3d.scenegraph.PipeRun;
69 import org.simantics.plant3d.scenegraph.PipelineComponent;
70 import org.simantics.plant3d.scenegraph.SchemaBuilder;
71 import org.simantics.plant3d.scenegraph.TurnComponent;
72 import org.simantics.plant3d.scenegraph.controlpoint.ControlPointFactory;
73 import org.simantics.plant3d.scenegraph.controlpoint.PipingRules;
74 import org.simantics.plant3d.utils.ComponentUtils;
75 import org.simantics.plant3d.utils.Item;
76 import org.simantics.plant3d.utils.P3DUtil;
77 import org.simantics.selectionview.StandardPropertyPage;
78 import org.simantics.ui.workbench.IPropertyPage;
79 import org.simantics.ui.workbench.IResourceEditorInput;
80 import org.simantics.ui.workbench.ResourceEditorPart;
81 import org.simantics.utils.threads.ThreadUtils;
82 import org.simantics.utils.ui.ExceptionUtils;
85 import vtk.vtkCameraPass;
86 import vtk.vtkDefaultPass;
87 import vtk.vtkGaussianBlurPass;
88 import vtk.vtkLightsPass;
90 import vtk.vtkRenderPassCollection;
91 import vtk.vtkRenderer;
92 import vtk.vtkSSAAPass;
93 import vtk.vtkSequencePass;
94 import vtk.vtkSimpleMotionBlurPass;
97 public class Plant3DEditor extends ResourceEditorPart {
99 private Composite parent;
100 protected ToolComposite toolComposite;
101 private Resource input;
102 // private InteractiveVtkPanel panel;
103 // private SWTAWTComponent component;
104 private InteractiveVtkComposite panel;
107 private P3DRootNode rootNode;
108 private IMapping<Resource,INode> mapping;
110 protected NodeSelectionProvider2<Resource, INode> selectionProvider;
113 protected vtkCameraAndSelectorAction cameraAction;
114 protected FocusAction focusAction;
115 protected TranslateAction translateAction;
116 protected TranslateInlineAction translateInlineAction;
117 protected TranslateFreeVariableLengthAction translateFreeVariableLengthAction;
118 protected RotateAction rotateAction;
119 protected RemoveAction removeAction;
120 protected RemoveAndSplitAction removeSplitAction;
121 protected RoutePipeAction routePipeAction;
122 protected AddComponentAction addComponentAction;
123 protected ReversePipeRunAction reversePipeRunAction;
125 private P3DNodeMap nodeMap;
127 /** Constants for selecting the up-direction */
128 public static final int X = 0, Y = 1, Z = 2;
130 protected int upDirection = 1;
133 public void createPartControl(Composite parent) {
134 this.parent = parent;
135 //parent.setLayout (new FillLayout ());
138 // component = new SWTAWTComponent(parent,SWT.NONE) {
141 // protected Component createSwingComponent() {
142 // if (panel == null) {
143 // panel = new InteractiveVtkPanel();
144 // vtkPanelUtil.registerPanel(panel);
151 IResourceEditorInput rei = (IResourceEditorInput)getEditorInput();
152 input = rei.getResource();
154 toolComposite = new ToolComposite(parent, SWT.BORDER);
155 toolComposite.setVisible(true);
157 panel = new InteractiveVtkComposite(parent);
159 GridLayoutFactory.fillDefaults().margins(0, 0).spacing(0, 0).applyTo(parent);
160 GridDataFactory.fillDefaults().align(SWT.FILL, SWT.FILL).grab(true, true).applyTo(getPanel().getComponent());
163 GridDataFactory.fillDefaults().align(SWT.FILL, SWT.CENTER).grab(true, false).applyTo(toolComposite);
165 //IActionBars actionBars = getEditorSite().getActionBars();
171 new ContextMenuListener(panel, contextMenu);
173 cameraAction = createCameraAction();
174 switch (upDirection) {
176 cameraAction.setUpDirection(new double[] { 1, 0, 0 });
179 cameraAction.setUpDirection(new double[] { 0, 1, 0 });
182 cameraAction.setUpDirection(new double[] { 0, 0, 1 });
186 panel.setDefaultAction(cameraAction);
187 panel.useDefaultAction();
188 panel.setPickType(4);
191 ControlPointFactory.preloadCache(Simantics.getSession(), getLibraryUri());
192 ComponentUtils.preloadCache(Simantics.getSession());
193 } catch (Exception e) {
194 ExceptionUtils.logAndShowError("Cannot open Plant3D editor",e);
199 getSession().syncRequest(new ReadRequest() {
202 public void run(ReadGraph graph) throws DatabaseException {
203 //System.out.println("START PLANT3D LOAD");
204 PipingRules.setEnabled(false);
205 IMappingSchema<Resource,INode> schema = getSchema(graph);
206 mapping = Mappings.createWithListening(schema);
207 rootNode = (P3DRootNode)mapping.map(graph, input);
208 // update control points.
209 // TODO : this should be optimized.
212 P3DUtil.finalizeDBLoad(rootNode);
213 nodeMap = createNodeMap(getSession(), mapping, panel,rootNode);
215 } catch (Exception e) {
216 throw new DatabaseException(e);
219 //System.out.println("END PLANT3D LOAD");
223 if (rootNode == null)
224 throw new RuntimeException("Scenegraph loading failed.");
227 selectionProvider = createSelectionProvider();
229 cameraAction.addSelectionChangedListener(selectionProvider);
231 cameraAction.addHoverChangedListener(createHoverHighlhighter());
232 selectionProvider.addSelectionChangedListener(createSelectionHighlighter());
234 getSite().setSelectionProvider(selectionProvider);
235 getSite().getPage().addPostSelectionListener(selectionProvider);
237 //outlinePage = new ScenegraphOutlinePage(rootNode);
240 parent.addDisposeListener(new DisposeListener() {
243 public void widgetDisposed(DisposeEvent e) {
244 getSite().getPage().removePostSelectionListener(selectionProvider);
246 PipingRules.setEnabled(false);
248 PipingRules.setEnabled(true);
250 // component.dispose();
251 //panel.getComponent().dispose();
255 } catch (DatabaseException e1) {
256 ExceptionUtils.logAndShowError("Cannot open Plant3D editor",e1);
263 public void setUpDirection(int upDirection) {
264 this.upDirection = upDirection;
267 protected vtkCameraAndSelectorAction createCameraAction() {
268 return new vtkCameraAndSelectorAction(panel);
271 protected NodeSelectionProvider2<Resource,INode> createSelectionProvider() {
272 return new NodeSelectionProvider2<Resource,INode>(this,mapping,nodeMap);
275 protected HoverHighlighter<Resource> createHoverHighlhighter() {
276 return new HoverHighlighter<>(panel,nodeMap);
279 protected SelectionHighlighter<Resource> createSelectionHighlighter() {
280 return new SelectionHighlighter<Resource>(panel,nodeMap);
283 protected String getLibraryUri() {
284 return Plant3D.URIs.Builtin;
287 protected void createActions() {
288 focusAction = new FocusAction(panel, cameraAction);
289 translateAction = new TranslateAction(panel,nodeMap,toolComposite);
290 translateInlineAction = new TranslateInlineAction(panel, nodeMap,toolComposite);
291 translateFreeVariableLengthAction = new TranslateFreeVariableLengthAction(panel, getRootNode(), toolComposite);
292 rotateAction = new RotateAction(panel,nodeMap,toolComposite);
293 removeAction = new RemoveAction(nodeMap);
294 removeSplitAction = new RemoveAndSplitAction(nodeMap);
295 routePipeAction = new RoutePipeAction(panel,rootNode, toolComposite);
296 addComponentAction = new AddComponentAction(panel, rootNode, getLibraryUri());
297 reversePipeRunAction = new ReversePipeRunAction(nodeMap);
300 public void populate() {
301 ThreadUtils.asyncExec(panel.getThreadQueue(), new Runnable() {
306 panel.addListener(new RenderListener() {
309 public void preRender() {
314 public void postRender() {
315 panel.removeListener(this);
318 P3DUtil.finalizeDBLoad2(rootNode);
319 if (nodeMap.isRangeModified());
320 nodeMap.commit("Load sync");
321 } catch (Exception e) {
322 ExceptionUtils.logAndShowError("Failed to load model correctly", e);
325 List<vtkProp3D> props = new ArrayList<vtkProp3D>();
326 collectProps(rootNode, props);
335 protected IMappingSchema<Resource, INode> getSchema(ReadGraph graph) throws DatabaseException {
336 IMappingSchema<Resource,INode> schema = SchemaBuilder.getSchema(graph);
340 protected P3DNodeMap createNodeMap(Session session, IMapping<Resource, INode> mapping, VtkView panel, P3DRootNode rootNode) {
341 return new P3DNodeMap(session, mapping, panel,rootNode);
345 public void setFocus() {
346 //component.setFocus();
347 panel.getComponent().setFocus();
350 protected void createScene() {
351 vtkRenderer ren1 = panel.getRenderer();
353 boolean multiPass = false;
354 boolean blur = false;
355 boolean ssaa = false;
356 //boolean sobel = true;
357 boolean mblur = false;
360 vtkLightsPass lightsPass = new vtkLightsPass();
361 vtkDefaultPass defaultPass = new vtkDefaultPass();
364 vtkRenderPassCollection passes = new vtkRenderPassCollection();
365 passes.AddItem(lightsPass);
366 passes.AddItem(defaultPass);
368 vtkSequencePass seq = new vtkSequencePass();
369 seq.SetPasses(passes);
371 vtkCameraPass cameraPass = new vtkCameraPass();
372 cameraPass.SetDelegatePass(seq);
375 vtkGaussianBlurPass blurPass = new vtkGaussianBlurPass();
376 blurPass.SetDelegatePass(cameraPass);
377 ren1.SetPass(blurPass);
379 vtkSSAAPass ssaaPass = new vtkSSAAPass();
380 ssaaPass.SetDelegatePass(cameraPass);
381 ren1.SetPass(ssaaPass);
383 vtkSimpleMotionBlurPass mBlurPass = new vtkSimpleMotionBlurPass();
384 mBlurPass.SetDelegatePass(cameraPass);
385 ren1.SetPass(mBlurPass);
386 // } else if (sobel) {
387 // vtkSobelGradientMagnitudePass sobelPass = new vtkSobelGradientMagnitudePass();
388 // sobelPass.SetDelegatePass(sobelPass);
389 // ren1.SetPass(sobelPass);
391 ren1.SetPass(cameraPass);
395 // ren1.GetRenderWindow().LineSmoothingOn();
396 // ren1.GetRenderWindow().PointSmoothingOn();
397 // ren1.GetRenderWindow().PolygonSmoothingOn();
398 // ren1.GetRenderWindow().SetMultiSamples(2);
402 ren1.SetBackground2(1,1,1); // background color white
403 ren1.SetBackground(0.9,0.9,0.9);
404 ren1.SetGradientBackground(true);
406 // vtkActor grid = vtkShape.createGridActor(8,1.0,1|2|4);
407 int dir = 1 << upDirection;
408 vtkActor grid = vtkShape.createGridActor(8, 1.0, dir);
411 panel.addDeletable(grid);
413 AxesDisplay axesDisplay = new AxesDisplay(panel);
417 protected Menu contextMenu;
419 protected void hookContextMenu() {
420 MenuManager menuMgr = new MenuManager("#PopupMenu");
421 menuMgr.setRemoveAllWhenShown(true);
422 menuMgr.addMenuListener(new IMenuListener() {
423 public void menuAboutToShow(IMenuManager manager) {
424 createContextMenu(manager);
428 contextMenu = menuMgr.createContextMenu(parent);
431 protected void createContextMenu(IMenuManager m) {
432 List<INode> selected = selectionProvider.getSelectedNodes();
434 createFocusMenu(m, selected);
436 m.add(new Separator());
439 if (selected.size() == 0) {
440 m.add(new AddEquipmentAction(rootNode, getLibraryUri()));
441 // for (Item eq : P3DUtil.getEquipments(getLibraryUri())) {
442 // m.add(new AddEquipmentAction(rootNode, eq));
444 } else if (selected.size() == 1) {
445 IP3DNode node = (IP3DNode)selected.get(0);
446 if (node instanceof Equipment) {
447 m.add(translateAction);
449 for (Item eq : P3DUtil.getNozzles(Simantics.getSession(), getLibraryUri())) {
450 AddNozzleAction add = new AddNozzleAction(rootNode, eq);
451 add.setEquipment((Equipment)node);
455 } else if (node instanceof Nozzle) {
456 Nozzle nozzle = (Nozzle)node;
457 if (!nozzle.isFixed()) {
458 m.add(translateAction);
461 m.add(routePipeAction);
462 routePipeAction.setComponent(nozzle);
463 routePipeAction.setEnabled(nozzle.getNext() == null && nozzle.getPrevious() == null);
464 m.add(addComponentAction);
465 addComponentAction.setComponent(nozzle);
467 } else if (node instanceof TurnComponent) {
468 m.add(translateAction);
469 TurnComponent component = (TurnComponent)node;
470 m.add(routePipeAction);
471 routePipeAction.setComponent(component);
472 routePipeAction.setEnabled(component.getNext() == null || component.getPrevious() == null);
473 m.add(addComponentAction);
474 addComponentAction.setComponent(component);
476 m.add(removeSplitAction);
477 removeSplitAction.setNode(node);
478 } else if (node instanceof EndComponent) {
479 m.add(translateAction);
480 m.add(addComponentAction);
481 addComponentAction.setComponent((PipelineComponent)node);
483 } else if (node instanceof InlineComponent) {
484 //m.add(translateInlineAction);
485 InlineComponent component = (InlineComponent)node;
486 if (component.isVariableLength())
487 m.add(translateFreeVariableLengthAction);
489 m.add(translateInlineAction);
490 m.add(routePipeAction);
491 routePipeAction.setComponent(component);
492 m.add(addComponentAction);
493 addComponentAction.setComponent(component);
495 m.add(removeSplitAction);
496 removeSplitAction.setNode(node);
497 } else if (node instanceof PipeRun) {
498 m.add(reversePipeRunAction);
500 reversePipeRunAction.setNode(node);
506 translateAction.setNode(node);
507 translateInlineAction.setNode(node);
508 translateFreeVariableLengthAction.setNode(node);
509 rotateAction.setNode(node);
510 removeAction.setNode(node);
513 } catch (DatabaseException e) {
514 ExceptionUtils.logAndShowError(e);
518 protected void createFocusMenu(IMenuManager m, List<INode> selected) {
519 m.add(new Action("Fit to Window") {
522 List<vtkProp3D> props = new ArrayList<>();
523 final Collection<INode> collection = !selected.isEmpty() ? selected : getRootNode().getChild();
524 for (INode n : collection)
525 collectProps(n, props);
528 getPanel().refresh();
532 if (!selected.isEmpty()) {
533 List<vtkProp3D> actors = new ArrayList<>();
534 for (INode n : selected)
535 collectProps(n, actors);
536 if (actors.size() > 0) {
537 focusAction.setProps(new ArrayList<>(actors));
543 private IContentOutlinePage createOutline() {
544 if (rootNode == null || selectionProvider == null)
546 //IContentOutlinePage outlinePage = new VTKContentOutlinePage<Resource,Object>(rootNode, selectionProvider);
547 IContentOutlinePage outlinePage = new P3DContentOutlinePage(rootNode, selectionProvider) {
548 protected void createContextMenu(IMenuManager manager) {
549 Plant3DEditor.this.createContextMenu(manager);
552 outlinePage.addSelectionChangedListener(new ISelectionChangedListener() {
555 public void selectionChanged(SelectionChangedEvent event) {
556 selectionProvider.selectionChanged(event);
563 public <T> T getAdapter(Class<T> adapter) {
564 if (adapter.isAssignableFrom(IPropertyPage.class))
565 return adapter.cast(new StandardPropertyPage(getSite(),getPropertyContexts()));
566 if (adapter.isAssignableFrom(ISelectionProvider.class))
567 return adapter.cast(selectionProvider);
568 if (adapter.isAssignableFrom(IContentOutlinePage.class)) {
569 return adapter.cast(createOutline());
571 if (adapter.isAssignableFrom(NodeMap.class)) {
572 return adapter.cast(nodeMap);
574 if (adapter.isAssignableFrom(INode.class)) {
575 return adapter.cast(rootNode);
577 if (adapter.isAssignableFrom(IMapping.class)) {
578 return adapter.cast(mapping);
580 // if (adapter.isAssignableFrom(InteractiveVtkPanel.class)) {
581 // return adapter.cast(panel);
583 if (adapter.isAssignableFrom(VtkView.class))
584 return adapter.cast(panel);
585 return super.getAdapter(adapter);
588 public Set<String> getPropertyContexts() {
589 Set<String> result = new HashSet<String>();
590 result.add("http://www.simantics.org/Project-1.0/ProjectBrowseContext");
594 public InteractiveVtkComposite getPanel() {
598 public P3DRootNode getRootNode() {
602 public IMapping<Resource, INode> getMapping() {
606 public P3DNodeMap getNodeMap() {
610 public void fitToWindow(Collection<vtkProp3D> props) {
611 cameraAction.fitToView(props);
614 protected static void collectProps(INode node, List<vtkProp3D> props) {
615 if (node instanceof IP3DVisualNode)
616 props.addAll(((IP3DVisualNode) node).getActors());
618 if (node instanceof ParentNode)
619 for (INode n : ((ParentNode<?>) node).getNodes()) {
620 collectProps(n, props);