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.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;
80 import vtk.vtkCameraPass;
81 import vtk.vtkDefaultPass;
82 import vtk.vtkGaussianBlurPass;
83 import vtk.vtkLightsPass;
85 import vtk.vtkRenderPassCollection;
86 import vtk.vtkRenderer;
87 import vtk.vtkSSAAPass;
88 import vtk.vtkSequencePass;
89 import vtk.vtkSimpleMotionBlurPass;
92 public class Plant3DEditor extends ResourceEditorPart {
94 private Composite parent;
95 protected ToolComposite toolComposite;
96 private Resource input;
97 // private InteractiveVtkPanel panel;
98 // private SWTAWTComponent component;
99 private InteractiveVtkComposite panel;
102 private P3DRootNode rootNode;
103 private IMapping<Resource,INode> mapping;
105 protected NodeSelectionProvider2<Resource, INode> selectionProvider;
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;
120 private P3DNodeMap nodeMap;
122 /** Constants for selecting the up-direction */
123 public static final int X = 0, Y = 1, Z = 2;
125 protected int upDirection = 1;
128 public void createPartControl(Composite parent) {
129 this.parent = parent;
130 //parent.setLayout (new FillLayout ());
133 // component = new SWTAWTComponent(parent,SWT.NONE) {
136 // protected Component createSwingComponent() {
137 // if (panel == null) {
138 // panel = new InteractiveVtkPanel();
139 // vtkPanelUtil.registerPanel(panel);
146 IResourceEditorInput rei = (IResourceEditorInput)getEditorInput();
147 input = rei.getResource();
149 toolComposite = new ToolComposite(parent, SWT.BORDER);
150 toolComposite.setVisible(true);
152 panel = new InteractiveVtkComposite(parent);
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());
158 GridDataFactory.fillDefaults().align(SWT.FILL, SWT.CENTER).grab(true, false).applyTo(toolComposite);
160 //IActionBars actionBars = getEditorSite().getActionBars();
166 new ContextMenuListener(panel, contextMenu);
168 cameraAction = createCameraAction();
169 switch (upDirection) {
171 cameraAction.setUpDirection(new double[] { 1, 0, 0 });
174 cameraAction.setUpDirection(new double[] { 0, 1, 0 });
177 cameraAction.setUpDirection(new double[] { 0, 0, 1 });
181 panel.setDefaultAction(cameraAction);
182 panel.useDefaultAction();
183 panel.setPickType(4);
186 ControlPointFactory.preloadCache(getLibraryUri());
187 ComponentUtils.preloadCache();
188 } catch (Exception e) {
189 ExceptionUtils.logAndShowError("Cannot open Plant3D editor",e);
194 getSession().syncRequest(new ReadRequest() {
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.
207 P3DUtil.finalizeDBLoad(rootNode);
208 nodeMap = createNodeMap(getSession(), mapping, panel,rootNode);
210 } catch (Exception e) {
211 throw new DatabaseException(e);
214 //System.out.println("END PLANT3D LOAD");
218 if (rootNode == null)
219 throw new RuntimeException("Scenegraph loading failed.");
222 selectionProvider = createSelectionProvider();
224 cameraAction.addSelectionChangedListener(selectionProvider);
226 cameraAction.addHoverChangedListener(createHoverHighlhighter());
227 selectionProvider.addSelectionChangedListener(createSelectionHighlighter());
229 getSite().setSelectionProvider(selectionProvider);
230 getSite().getPage().addPostSelectionListener(selectionProvider);
232 //outlinePage = new ScenegraphOutlinePage(rootNode);
235 parent.addDisposeListener(new DisposeListener() {
238 public void widgetDisposed(DisposeEvent e) {
239 getSite().getPage().removePostSelectionListener(selectionProvider);
241 PipingRules.setEnabled(false);
243 PipingRules.setEnabled(true);
245 // component.dispose();
246 //panel.getComponent().dispose();
250 } catch (DatabaseException e1) {
251 ExceptionUtils.logAndShowError("Cannot open Plant3D editor",e1);
259 public void setUpDirection(int upDirection) {
260 this.upDirection = upDirection;
263 protected vtkCameraAndSelectorAction createCameraAction() {
264 return new vtkCameraAndSelectorAction(panel);
267 protected NodeSelectionProvider2<Resource,INode> createSelectionProvider() {
268 return new NodeSelectionProvider2<Resource,INode>(this,mapping,nodeMap);
271 protected HoverHighlighter<Resource> createHoverHighlhighter() {
272 return new HoverHighlighter<>(panel,nodeMap);
275 protected SelectionHighlighter<Resource> createSelectionHighlighter() {
276 return new SelectionHighlighter<Resource>(panel,nodeMap);
279 protected String getLibraryUri() {
280 return Plant3D.URIs.Builtin;
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);
296 public void populate() {
297 ThreadUtils.asyncExec(panel.getThreadQueue(), new Runnable() {
302 panel.addListener(new RenderListener() {
305 public void preRender() {
310 public void postRender() {
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);
319 panel.removeListener(this);
329 protected IMappingSchema<Resource, INode> getSchema(ReadGraph graph) throws DatabaseException {
330 IMappingSchema<Resource,INode> schema = SchemaBuilder.getSchema(graph);
334 protected P3DNodeMap createNodeMap(Session session, IMapping<Resource, INode> mapping, VtkView panel, P3DRootNode rootNode) {
335 return new P3DNodeMap(session, mapping, panel,rootNode);
339 public void setFocus() {
340 //component.setFocus();
341 panel.getComponent().setFocus();
344 private void createScene() {
345 vtkRenderer ren1 = panel.getRenderer();
347 boolean multiPass = false;
348 boolean blur = false;
349 boolean ssaa = false;
350 //boolean sobel = true;
351 boolean mblur = false;
354 vtkLightsPass lightsPass = new vtkLightsPass();
355 vtkDefaultPass defaultPass = new vtkDefaultPass();
358 vtkRenderPassCollection passes = new vtkRenderPassCollection();
359 passes.AddItem(lightsPass);
360 passes.AddItem(defaultPass);
362 vtkSequencePass seq = new vtkSequencePass();
363 seq.SetPasses(passes);
365 vtkCameraPass cameraPass = new vtkCameraPass();
366 cameraPass.SetDelegatePass(seq);
369 vtkGaussianBlurPass blurPass = new vtkGaussianBlurPass();
370 blurPass.SetDelegatePass(cameraPass);
371 ren1.SetPass(blurPass);
373 vtkSSAAPass ssaaPass = new vtkSSAAPass();
374 ssaaPass.SetDelegatePass(cameraPass);
375 ren1.SetPass(ssaaPass);
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);
385 ren1.SetPass(cameraPass);
389 // ren1.GetRenderWindow().LineSmoothingOn();
390 // ren1.GetRenderWindow().PointSmoothingOn();
391 // ren1.GetRenderWindow().PolygonSmoothingOn();
392 // ren1.GetRenderWindow().SetMultiSamples(2);
396 ren1.SetBackground2(1,1,1); // background color white
397 ren1.SetBackground(0.9,0.9,0.9);
398 ren1.SetGradientBackground(true);
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);
405 panel.addDeletable(grid);
407 AxesDisplay axesDisplay = new AxesDisplay(panel);
412 protected Menu contextMenu;
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);
423 contextMenu = menuMgr.createContextMenu(parent);
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);
436 if (props.size() > 0) {
437 focusAction.setProps(props);
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));
447 } else if (selected.size() == 1) {
448 IP3DNode node = (IP3DNode)selected.get(0);
449 if (node instanceof Equipment) {
450 m.add(translateAction);
452 for (Item eq : P3DUtil.getNozzles(getLibraryUri())) {
453 AddNozzleAction add = new AddNozzleAction(rootNode, eq);
454 add.setEquipment((Equipment)node);
458 } else if (node instanceof Nozzle) {
459 Nozzle nozzle = (Nozzle)node;
460 if (!nozzle.isFixed()) {
461 m.add(translateAction);
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);
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);
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);
486 } else if (node instanceof InlineComponent) {
487 //m.add(translateInlineAction);
488 InlineComponent component = (InlineComponent)node;
489 if (component.isVariableLength())
490 m.add(translateFreeVariableLengthAction);
492 m.add(translateInlineAction);
493 m.add(routePipeAction);
494 routePipeAction.setComponent(component);
495 m.add(addComponentAction);
496 addComponentAction.setComponent(component);
498 m.add(removeSplitAction);
499 removeSplitAction.setNode(node);
500 } else if (node instanceof PipeRun) {
501 m.add(reversePipeRunAction);
503 reversePipeRunAction.setNode(node);
509 translateAction.setNode(node);
510 translateInlineAction.setNode(node);
511 translateFreeVariableLengthAction.setNode(node);
512 rotateAction.setNode(node);
513 removeAction.setNode(node);
516 } catch (DatabaseException e) {
517 ExceptionUtils.logAndShowError(e);
521 private IContentOutlinePage createOutline() {
522 if (rootNode == null || selectionProvider == 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);
530 outlinePage.addSelectionChangedListener(new ISelectionChangedListener() {
533 public void selectionChanged(SelectionChangedEvent event) {
534 selectionProvider.selectionChanged(event);
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());
549 if (adapter.isAssignableFrom(NodeMap.class)) {
550 return adapter.cast(nodeMap);
552 if (adapter.isAssignableFrom(INode.class)) {
553 return adapter.cast(rootNode);
555 if (adapter.isAssignableFrom(IMapping.class)) {
556 return adapter.cast(mapping);
558 // if (adapter.isAssignableFrom(InteractiveVtkPanel.class)) {
559 // return adapter.cast(panel);
561 if (adapter.isAssignableFrom(VtkView.class))
562 return adapter.cast(panel);
563 return super.getAdapter(adapter);
566 public Set<String> getPropertyContexts() {
567 Set<String> result = new HashSet<String>();
568 result.add("http://www.simantics.org/Project-1.0/ProjectBrowseContext");
572 public InteractiveVtkComposite getPanel() {
576 public P3DRootNode getRootNode() {
580 public IMapping<Resource, INode> getMapping() {
584 public P3DNodeMap getNodeMap() {