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.layout.FillLayout;
20 import org.eclipse.swt.widgets.Composite;
21 import org.eclipse.swt.widgets.Menu;
22 import org.eclipse.ui.views.contentoutline.IContentOutlinePage;
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;
124 public void createPartControl(Composite parent) {
125 this.parent = parent;
126 //parent.setLayout (new FillLayout ());
129 // component = new SWTAWTComponent(parent,SWT.NONE) {
132 // protected Component createSwingComponent() {
133 // if (panel == null) {
134 // panel = new InteractiveVtkPanel();
135 // vtkPanelUtil.registerPanel(panel);
142 IResourceEditorInput rei = (IResourceEditorInput)getEditorInput();
143 input = rei.getResource();
145 toolComposite = new ToolComposite(parent, SWT.BORDER);
146 toolComposite.setVisible(true);
148 panel = new InteractiveVtkComposite(parent);
150 GridLayoutFactory.fillDefaults().margins(0, 0).spacing(0, 0).applyTo(parent);
151 GridDataFactory.fillDefaults().align(SWT.FILL, SWT.FILL).grab(true, true).applyTo(getPanel().getComponent());
154 GridDataFactory.fillDefaults().align(SWT.FILL, SWT.CENTER).grab(true, false).applyTo(toolComposite);
156 //IActionBars actionBars = getEditorSite().getActionBars();
162 new ContextMenuListener(panel, contextMenu);
164 cameraAction = createCameraAction();
165 panel.setDefaultAction(cameraAction);
166 panel.useDefaultAction();
167 panel.setPickType(4);
170 ControlPointFactory.preloadCache(getLibraryUri());
171 ComponentUtils.preloadCache();
172 } catch (Exception e) {
173 ExceptionUtils.logAndShowError("Cannot open Plant3D editor",e);
178 getSession().syncRequest(new ReadRequest() {
181 public void run(ReadGraph graph) throws DatabaseException {
182 //System.out.println("START PLANT3D LOAD");
183 PipingRules.setEnabled(false);
184 IMappingSchema<Resource,INode> schema = getSchema(graph);
185 mapping = Mappings.createWithListening(schema);
186 rootNode = (P3DRootNode)mapping.map(graph, input);
187 // update control points.
188 // TODO : this should be optimized.
191 P3DUtil.finalizeDBLoad(rootNode);
192 nodeMap = createNodeMap(getSession(), mapping, panel,rootNode);
194 } catch (Exception e) {
195 throw new DatabaseException(e);
198 //System.out.println("END PLANT3D LOAD");
202 if (rootNode == null)
203 throw new RuntimeException("Scenegraph loading failed.");
206 selectionProvider = createSelectionProvider();
208 cameraAction.addSelectionChangedListener(selectionProvider);
210 cameraAction.addHoverChangedListener(createHoverHighlhighter());
211 selectionProvider.addSelectionChangedListener(createSelectionHighlighter());
213 getSite().setSelectionProvider(selectionProvider);
214 getSite().getPage().addPostSelectionListener(selectionProvider);
216 //outlinePage = new ScenegraphOutlinePage(rootNode);
219 parent.addDisposeListener(new DisposeListener() {
222 public void widgetDisposed(DisposeEvent e) {
223 getSite().getPage().removePostSelectionListener(selectionProvider);
225 PipingRules.setEnabled(false);
227 PipingRules.setEnabled(true);
229 // component.dispose();
230 //panel.getComponent().dispose();
234 } catch (DatabaseException e1) {
235 ExceptionUtils.logAndShowError("Cannot open Plant3D editor",e1);
243 protected vtkCameraAndSelectorAction createCameraAction() {
244 return new vtkCameraAndSelectorAction(panel);
247 protected NodeSelectionProvider2<Resource,INode> createSelectionProvider() {
248 return new NodeSelectionProvider2<Resource,INode>(this,mapping,nodeMap);
251 protected HoverHighlighter<Resource> createHoverHighlhighter() {
252 return new HoverHighlighter<>(panel,nodeMap);
255 protected SelectionHighlighter<Resource> createSelectionHighlighter() {
256 return new SelectionHighlighter<Resource>(panel,nodeMap);
259 protected String getLibraryUri() {
260 return Plant3D.URIs.Builtin;
263 protected void createActions() {
264 focusAction = new FocusAction(panel, cameraAction);
265 translateAction = new TranslateAction(panel,nodeMap,toolComposite);
266 translateInlineAction = new TranslateInlineAction(panel, nodeMap,toolComposite);
267 translateFreeVariableLengthAction = new TranslateFreeVariableLengthAction(panel, getRootNode(), toolComposite);
268 rotateAction = new RotateAction(panel,nodeMap,toolComposite);
269 removeAction = new RemoveAction(nodeMap);
270 removeSplitAction = new RemoveAndSplitAction(nodeMap);
271 routePipeAction = new RoutePipeAction(panel,rootNode, toolComposite);
272 addComponentAction = new AddComponentAction(panel, rootNode, getLibraryUri());
273 reversePipeRunAction = new ReversePipeRunAction(nodeMap);
276 public void populate() {
277 ThreadUtils.asyncExec(panel.getThreadQueue(), new Runnable() {
282 panel.addListener(new RenderListener() {
285 public void preRender() {
290 public void postRender() {
292 P3DUtil.finalizeDBLoad2(rootNode);
293 if (nodeMap.isRangeModified());
294 nodeMap.commit("Load sync");
295 } catch (Exception e) {
296 ExceptionUtils.logAndShowError("Failed to load model correctly", e);
297 //throw new DatabaseException(e);
299 panel.removeListener(this);
309 protected IMappingSchema<Resource, INode> getSchema(ReadGraph graph) throws DatabaseException {
310 IMappingSchema<Resource,INode> schema = SchemaBuilder.getSchema(graph);
314 protected P3DNodeMap createNodeMap(Session session, IMapping<Resource, INode> mapping, VtkView panel, P3DRootNode rootNode) {
315 return new P3DNodeMap(session, mapping, panel,rootNode);
319 public void setFocus() {
320 //component.setFocus();
321 panel.getComponent().setFocus();
324 private void createScene() {
325 vtkRenderer ren1 = panel.getRenderer();
327 boolean multiPass = false;
328 boolean blur = false;
329 boolean ssaa = false;
330 //boolean sobel = true;
331 boolean mblur = false;
334 vtkLightsPass lightsPass = new vtkLightsPass();
335 vtkDefaultPass defaultPass = new vtkDefaultPass();
338 vtkRenderPassCollection passes = new vtkRenderPassCollection();
339 passes.AddItem(lightsPass);
340 passes.AddItem(defaultPass);
342 vtkSequencePass seq = new vtkSequencePass();
343 seq.SetPasses(passes);
345 vtkCameraPass cameraPass = new vtkCameraPass();
346 cameraPass.SetDelegatePass(seq);
349 vtkGaussianBlurPass blurPass = new vtkGaussianBlurPass();
350 blurPass.SetDelegatePass(cameraPass);
351 ren1.SetPass(blurPass);
353 vtkSSAAPass ssaaPass = new vtkSSAAPass();
354 ssaaPass.SetDelegatePass(cameraPass);
355 ren1.SetPass(ssaaPass);
357 vtkSimpleMotionBlurPass mBlurPass = new vtkSimpleMotionBlurPass();
358 mBlurPass.SetDelegatePass(cameraPass);
359 ren1.SetPass(mBlurPass);
360 // } else if (sobel) {
361 // vtkSobelGradientMagnitudePass sobelPass = new vtkSobelGradientMagnitudePass();
362 // sobelPass.SetDelegatePass(sobelPass);
363 // ren1.SetPass(sobelPass);
365 ren1.SetPass(cameraPass);
369 // ren1.GetRenderWindow().LineSmoothingOn();
370 // ren1.GetRenderWindow().PointSmoothingOn();
371 // ren1.GetRenderWindow().PolygonSmoothingOn();
372 // ren1.GetRenderWindow().SetMultiSamples(2);
376 ren1.SetBackground2(1,1,1); // background color white
377 ren1.SetBackground(0.9,0.9,0.9);
378 ren1.SetGradientBackground(true);
380 // vtkActor grid = vtkShape.createGridActor(8,1.0,1|2|4);
381 vtkActor grid = vtkShape.createGridActor(8,1.0, 2 );
384 panel.addDeletable(grid);
386 AxesDisplay axesDisplay = new AxesDisplay(panel);
391 protected Menu contextMenu;
393 protected void hookContextMenu() {
394 MenuManager menuMgr = new MenuManager("#PopupMenu");
395 menuMgr.setRemoveAllWhenShown(true);
396 menuMgr.addMenuListener(new IMenuListener() {
397 public void menuAboutToShow(IMenuManager manager) {
398 createContextMenu(manager);
402 contextMenu = menuMgr.createContextMenu(parent);
405 protected void createContextMenu(IMenuManager m) {
406 List<INode> selected = selectionProvider.getSelectedNodes();
407 List<VTKSelectionItem<Resource>> selectedItems = selectionProvider.getSelectedItems();
408 if (selectedItems.size() > 0) {
409 List<vtkProp> props = new ArrayList<>();
410 for (VTKSelectionItem<Resource> i : selectedItems) {
411 vtkProp p = (vtkProp)i.getAdapter(vtkProp.class);
415 if (props.size() > 0) {
416 focusAction.setProps(props);
421 if (selected.size() == 0) {
422 m.add(new AddEquipmentAction(rootNode, getLibraryUri()));
423 // for (Item eq : P3DUtil.getEquipments(getLibraryUri())) {
424 // m.add(new AddEquipmentAction(rootNode, eq));
426 } else if (selected.size() == 1) {
427 IP3DNode node = (IP3DNode)selected.get(0);
428 if (node instanceof Equipment) {
429 m.add(translateAction);
431 for (Item eq : P3DUtil.getNozzles(getLibraryUri())) {
432 AddNozzleAction add = new AddNozzleAction(rootNode, eq);
433 add.setEquipment((Equipment)node);
437 } else if (node instanceof Nozzle) {
438 Nozzle nozzle = (Nozzle)node;
439 if (!nozzle.isFixed()) {
440 m.add(translateAction);
443 m.add(routePipeAction);
444 routePipeAction.setComponent(nozzle);
445 routePipeAction.setEnabled(nozzle.getNext() == null && nozzle.getPrevious() == null);
446 m.add(addComponentAction);
447 addComponentAction.setComponent(nozzle);
449 } else if (node instanceof TurnComponent) {
450 m.add(translateAction);
451 TurnComponent component = (TurnComponent)node;
452 m.add(routePipeAction);
453 routePipeAction.setComponent(component);
454 routePipeAction.setEnabled(component.getNext() == null || component.getPrevious() == null);
455 m.add(addComponentAction);
456 addComponentAction.setComponent(component);
458 m.add(removeSplitAction);
459 removeSplitAction.setNode(node);
460 } else if (node instanceof EndComponent) {
461 m.add(translateAction);
462 m.add(addComponentAction);
463 addComponentAction.setComponent((PipelineComponent)node);
465 } else if (node instanceof InlineComponent) {
466 //m.add(translateInlineAction);
467 InlineComponent component = (InlineComponent)node;
468 if (component.isVariableLength())
469 m.add(translateFreeVariableLengthAction);
471 m.add(translateInlineAction);
472 m.add(routePipeAction);
473 routePipeAction.setComponent(component);
474 m.add(addComponentAction);
475 addComponentAction.setComponent(component);
477 m.add(removeSplitAction);
478 removeSplitAction.setNode(node);
479 } else if (node instanceof PipeRun) {
480 m.add(reversePipeRunAction);
482 reversePipeRunAction.setNode(node);
488 translateAction.setNode(node);
489 translateInlineAction.setNode(node);
490 translateFreeVariableLengthAction.setNode(node);
491 rotateAction.setNode(node);
492 removeAction.setNode(node);
495 } catch (DatabaseException e) {
496 ExceptionUtils.logAndShowError(e);
500 private IContentOutlinePage createOutline() {
501 if (rootNode == null || selectionProvider == null)
503 //IContentOutlinePage outlinePage = new VTKContentOutlinePage<Resource,Object>(rootNode, selectionProvider);
504 IContentOutlinePage outlinePage = new P3DContentOutlinePage(rootNode, selectionProvider) {
505 protected void createContextMenu(IMenuManager manager) {
506 Plant3DEditor.this.createContextMenu(manager);
509 outlinePage.addSelectionChangedListener(new ISelectionChangedListener() {
512 public void selectionChanged(SelectionChangedEvent event) {
513 selectionProvider.selectionChanged(event);
520 public <T> T getAdapter(Class<T> adapter) {
521 if (adapter.isAssignableFrom(IPropertyPage.class))
522 return adapter.cast(new StandardPropertyPage(getSite(),getPropertyContexts()));
523 if (adapter.isAssignableFrom(ISelectionProvider.class))
524 return adapter.cast(selectionProvider);
525 if (adapter.isAssignableFrom(IContentOutlinePage.class)) {
526 return adapter.cast(createOutline());
528 if (adapter.isAssignableFrom(NodeMap.class)) {
529 return adapter.cast(nodeMap);
531 if (adapter.isAssignableFrom(INode.class)) {
532 return adapter.cast(rootNode);
534 if (adapter.isAssignableFrom(IMapping.class)) {
535 return adapter.cast(mapping);
537 // if (adapter.isAssignableFrom(InteractiveVtkPanel.class)) {
538 // return adapter.cast(panel);
540 if (adapter.isAssignableFrom(VtkView.class))
541 return adapter.cast(panel);
542 return super.getAdapter(adapter);
545 public Set<String> getPropertyContexts() {
546 Set<String> result = new HashSet<String>();
547 result.add("http://www.simantics.org/Project-1.0/ProjectBrowseContext");
551 public InteractiveVtkComposite getPanel() {
555 public P3DRootNode getRootNode() {
559 public IMapping<Resource, INode> getMapping() {
563 public P3DNodeMap getNodeMap() {