1 package org.simantics.plant3d.editor;
3 import java.util.HashSet;
7 import org.eclipse.jface.action.IMenuListener;
8 import org.eclipse.jface.action.IMenuManager;
9 import org.eclipse.jface.action.MenuManager;
10 import org.eclipse.jface.viewers.ISelectionChangedListener;
11 import org.eclipse.jface.viewers.ISelectionProvider;
12 import org.eclipse.jface.viewers.SelectionChangedEvent;
13 import org.eclipse.swt.events.DisposeEvent;
14 import org.eclipse.swt.events.DisposeListener;
15 import org.eclipse.swt.layout.FillLayout;
16 import org.eclipse.swt.widgets.Composite;
17 import org.eclipse.swt.widgets.Menu;
18 import org.eclipse.ui.views.contentoutline.IContentOutlinePage;
19 import org.simantics.db.ReadGraph;
20 import org.simantics.db.Resource;
21 import org.simantics.db.Session;
22 import org.simantics.db.common.request.ReadRequest;
23 import org.simantics.db.exception.DatabaseException;
24 import org.simantics.g3d.scenegraph.IG3DNode;
25 import org.simantics.g3d.scenegraph.NodeMap;
26 import org.simantics.g3d.scenegraph.RenderListener;
27 import org.simantics.g3d.scenegraph.base.INode;
28 import org.simantics.g3d.vtk.action.RemoveAction;
29 import org.simantics.g3d.vtk.common.HoverHighlighter;
30 import org.simantics.g3d.vtk.common.NodeSelectionProvider2;
31 import org.simantics.g3d.vtk.common.SelectionHighlighter;
32 import org.simantics.g3d.vtk.common.VtkView;
33 import org.simantics.g3d.vtk.shape.vtkShape;
34 import org.simantics.g3d.vtk.swt.ContextMenuListener;
35 import org.simantics.g3d.vtk.swt.InteractiveVtkComposite;
36 import org.simantics.g3d.vtk.swt.RotateAction;
37 import org.simantics.g3d.vtk.swt.TranslateAction;
38 import org.simantics.g3d.vtk.swt.vtkCameraAndSelectorAction;
39 import org.simantics.g3d.vtk.utils.AxesDisplay;
40 import org.simantics.objmap.graph.IMapping;
41 import org.simantics.objmap.graph.Mappings;
42 import org.simantics.objmap.graph.schema.IMappingSchema;
43 import org.simantics.plant3d.actions.AddComponentAction;
44 import org.simantics.plant3d.actions.AddEquipmentAction;
45 import org.simantics.plant3d.actions.AddNozzleAction;
46 import org.simantics.plant3d.actions.RemoveAndSplitAction;
47 import org.simantics.plant3d.actions.RoutePipeAction;
48 import org.simantics.plant3d.actions.TranslateFreeVariableLengthAction;
49 import org.simantics.plant3d.actions.TranslateInlineAction;
50 import org.simantics.plant3d.ontology.Plant3D;
51 import org.simantics.plant3d.scenegraph.EndComponent;
52 import org.simantics.plant3d.scenegraph.Equipment;
53 import org.simantics.plant3d.scenegraph.IP3DNode;
54 import org.simantics.plant3d.scenegraph.InlineComponent;
55 import org.simantics.plant3d.scenegraph.Nozzle;
56 import org.simantics.plant3d.scenegraph.P3DRootNode;
57 import org.simantics.plant3d.scenegraph.PipelineComponent;
58 import org.simantics.plant3d.scenegraph.SchemaBuilder;
59 import org.simantics.plant3d.scenegraph.TurnComponent;
60 import org.simantics.plant3d.scenegraph.controlpoint.ControlPointFactory;
61 import org.simantics.plant3d.scenegraph.controlpoint.PipingRules;
62 import org.simantics.plant3d.utils.ComponentUtils;
63 import org.simantics.plant3d.utils.Item;
64 import org.simantics.plant3d.utils.P3DUtil;
65 import org.simantics.selectionview.StandardPropertyPage;
66 import org.simantics.ui.workbench.IPropertyPage;
67 import org.simantics.ui.workbench.IResourceEditorInput;
68 import org.simantics.ui.workbench.ResourceEditorPart;
69 import org.simantics.utils.threads.ThreadUtils;
70 import org.simantics.utils.ui.ExceptionUtils;
73 import vtk.vtkCameraPass;
74 import vtk.vtkDefaultPass;
75 import vtk.vtkLightsPass;
76 import vtk.vtkRenderPassCollection;
77 import vtk.vtkRenderer;
78 import vtk.vtkSequencePass;
81 public class Plant3DEditor extends ResourceEditorPart {
83 private Composite parent;
84 private Resource input;
85 // private InteractiveVtkPanel panel;
86 // private SWTAWTComponent component;
87 private InteractiveVtkComposite panel;
89 private P3DRootNode rootNode;
90 private IMapping<Resource,Object> mapping;
92 protected NodeSelectionProvider2<Resource,Object> selectionProvider;
94 protected vtkCameraAndSelectorAction cameraAction;
95 protected TranslateAction translateAction;
96 protected TranslateInlineAction translateInlineAction;
97 protected TranslateFreeVariableLengthAction translateFreeVariableLengthAction;
98 protected RotateAction rotateAction;
99 protected RemoveAction removeAction;
100 protected RemoveAndSplitAction removeSplitAction;
101 protected RoutePipeAction routePipeAction;
102 protected AddComponentAction addComponentAction;
104 private P3DNodeMap nodeMap;
107 public void createPartControl(Composite parent) {
108 this.parent = parent;
109 parent.setLayout (new FillLayout ());
110 // component = new SWTAWTComponent(parent,SWT.NONE) {
113 // protected Component createSwingComponent() {
114 // if (panel == null) {
115 // panel = new InteractiveVtkPanel();
116 // vtkPanelUtil.registerPanel(panel);
123 IResourceEditorInput rei = (IResourceEditorInput)getEditorInput();
124 input = rei.getResource();
126 panel = new InteractiveVtkComposite(parent);
129 //IActionBars actionBars = getEditorSite().getActionBars();
135 new ContextMenuListener(panel, contextMenu);
137 cameraAction = createCameraAction();
138 panel.setDefaultAction(cameraAction);
139 panel.useDefaultAction();
140 panel.setPickType(4);
143 ControlPointFactory.preloadCache(getLibraryUri());
144 ComponentUtils.preloadCache();
145 } catch (Exception e) {
146 ExceptionUtils.logAndShowError("Cannot open Plant3D editor",e);
151 getSession().syncRequest(new ReadRequest() {
153 @SuppressWarnings({ "rawtypes", "unchecked" })
155 public void run(ReadGraph graph) throws DatabaseException {
156 System.out.println("START PLANT3D LOAD");
157 PipingRules.setEnabled(false);
158 IMappingSchema<Resource, Object> schema = getSchema(graph);
159 mapping = Mappings.createWithListening(schema);
160 rootNode = (P3DRootNode)mapping.map(graph, input);
161 // update control points.
162 // TODO : this should be optimized.
165 P3DUtil.finalizeDBLoad(rootNode);
166 nodeMap = createNodeMap(getSession(), mapping, panel,rootNode);
168 } catch (Exception e) {
169 throw new DatabaseException(e);
172 System.out.println("END PLANT3D LOAD");
176 if (rootNode == null)
177 throw new RuntimeException("Scenegraph loading failed.");
180 selectionProvider = createSelectionProvider();
182 cameraAction.addSelectionChangedListener(selectionProvider);
184 cameraAction.addHoverChangedListener(createHoverHighlhighter());
185 selectionProvider.addSelectionChangedListener(createSelectionHighlighter());
187 getSite().setSelectionProvider(selectionProvider);
188 getSite().getPage().addPostSelectionListener(selectionProvider);
190 //outlinePage = new ScenegraphOutlinePage(rootNode);
193 parent.addDisposeListener(new DisposeListener() {
196 public void widgetDisposed(DisposeEvent e) {
197 getSite().getPage().removePostSelectionListener(selectionProvider);
199 PipingRules.setEnabled(false);
201 PipingRules.setEnabled(true);
203 // component.dispose();
204 //panel.getComponent().dispose();
208 } catch (DatabaseException e1) {
209 ExceptionUtils.logAndShowError("Cannot open Plant3D editor",e1);
217 protected vtkCameraAndSelectorAction createCameraAction() {
218 return new vtkCameraAndSelectorAction(panel);
221 protected NodeSelectionProvider2<Resource,Object> createSelectionProvider() {
222 return new NodeSelectionProvider2<Resource,Object>(this,mapping,nodeMap);
225 protected HoverHighlighter createHoverHighlhighter() {
226 return new HoverHighlighter(panel,nodeMap);
229 protected SelectionHighlighter<Resource,INode> createSelectionHighlighter() {
230 return new SelectionHighlighter<Resource,INode>(panel,nodeMap);
233 protected String getLibraryUri() {
234 return Plant3D.URIs.Builtin;
237 protected void createActions() {
238 translateAction = new TranslateAction(panel,nodeMap);
239 translateInlineAction = new TranslateInlineAction(panel, nodeMap);
240 translateFreeVariableLengthAction = new TranslateFreeVariableLengthAction(panel, getRootNode());
241 rotateAction = new RotateAction(panel,nodeMap);
242 removeAction = new RemoveAction(nodeMap) {
243 public void setNode(IG3DNode node) {
248 removeSplitAction = new RemoveAndSplitAction(nodeMap);
249 routePipeAction = new RoutePipeAction(panel,rootNode);
250 addComponentAction = new AddComponentAction(panel, rootNode, getLibraryUri());
253 public void populate() {
254 ThreadUtils.asyncExec(panel.getThreadQueue(), new Runnable() {
259 panel.addListener(new RenderListener() {
262 public void preRender() {
267 public void postRender() {
269 P3DUtil.finalizeDBLoad2(rootNode);
270 if (nodeMap.isRangeModified());
271 nodeMap.commit("Load sync");
272 } catch (Exception e) {
273 //throw new DatabaseException(e);
275 panel.removeListener(this);
285 protected IMappingSchema<Resource, Object> getSchema(ReadGraph graph) throws DatabaseException {
286 IMappingSchema<Resource,Object> schema = SchemaBuilder.getSchema(graph);
290 protected P3DNodeMap createNodeMap(Session session, IMapping<Resource, Object> mapping, VtkView panel, P3DRootNode rootNode) {
291 return new P3DNodeMap(session, mapping, panel,rootNode);
295 public void setFocus() {
296 //component.setFocus();
297 panel.getComponent().setFocus();
300 private void createScene() {
301 vtkRenderer ren1 = panel.getRenderer();
303 boolean multiPass = false;
306 vtkLightsPass lightsPass = new vtkLightsPass();
307 vtkDefaultPass defaultPass = new vtkDefaultPass();
310 vtkRenderPassCollection passes = new vtkRenderPassCollection();
311 passes.AddItem(lightsPass);
312 passes.AddItem(defaultPass);
314 vtkSequencePass seq = new vtkSequencePass();
315 seq.SetPasses(passes);
317 vtkCameraPass cameraPass = new vtkCameraPass();
318 cameraPass.SetDelegatePass(seq);
320 ren1.SetPass(cameraPass);
323 // ren1.GetRenderWindow().LineSmoothingOn();
324 // ren1.GetRenderWindow().PointSmoothingOn();
325 // ren1.GetRenderWindow().PolygonSmoothingOn();
326 // ren1.GetRenderWindow().SetMultiSamples(2);
330 ren1.SetBackground2(1,1,1); // background color white
331 ren1.SetBackground(0.9,0.9,0.9);
332 ren1.SetGradientBackground(true);
334 // vtkActor grid = vtkShape.createGridActor(8,1.0,1|2|4);
335 vtkActor grid = vtkShape.createGridActor(8,1.0, 2 );
338 panel.addDeletable(grid);
340 AxesDisplay axesDisplay = new AxesDisplay(panel);
345 protected Menu contextMenu;
347 protected void hookContextMenu() {
348 MenuManager menuMgr = new MenuManager("#PopupMenu");
349 menuMgr.setRemoveAllWhenShown(true);
350 menuMgr.addMenuListener(new IMenuListener() {
351 public void menuAboutToShow(IMenuManager manager) {
352 createContextMenu(manager);
356 contextMenu = menuMgr.createContextMenu(parent);
359 protected void createContextMenu(IMenuManager m) {
360 List<INode> selected = selectionProvider.getSelectedNodes();
362 if (selected.size() == 0) {
363 for (Item eq : P3DUtil.getEquipments(getLibraryUri())) {
364 m.add(new AddEquipmentAction(rootNode, eq));
366 } else if (selected.size() == 1) {
367 IP3DNode node = (IP3DNode)selected.get(0);
368 if (node instanceof Equipment) {
369 m.add(translateAction);
371 for (Item eq : P3DUtil.getNozzles(getLibraryUri())) {
372 AddNozzleAction add = new AddNozzleAction(rootNode, eq);
373 add.setEquipment((Equipment)node);
377 } else if (node instanceof Nozzle) {
378 Nozzle nozzle = (Nozzle)node;
379 if (!nozzle.isFixed()) {
380 m.add(translateAction);
383 m.add(routePipeAction);
384 routePipeAction.setComponent(nozzle);
385 routePipeAction.setEnabled(nozzle.getNext() == null && nozzle.getPrevious() == null);
386 m.add(addComponentAction);
387 addComponentAction.setComponent(nozzle);
389 } else if (node instanceof TurnComponent) {
390 m.add(translateAction);
391 TurnComponent component = (TurnComponent)node;
392 m.add(routePipeAction);
393 routePipeAction.setComponent(component);
394 routePipeAction.setEnabled(component.getNext() == null || component.getPrevious() == null);
395 m.add(addComponentAction);
396 addComponentAction.setComponent(component);
398 m.add(removeSplitAction);
399 removeSplitAction.setNode(node);
400 } else if (node instanceof EndComponent) {
401 m.add(translateAction);
402 m.add(addComponentAction);
403 addComponentAction.setComponent((PipelineComponent)node);
405 } else if (node instanceof InlineComponent) {
406 //m.add(translateInlineAction);
407 InlineComponent component = (InlineComponent)node;
408 if (component.isVariableLength())
409 m.add(translateFreeVariableLengthAction);
411 m.add(translateInlineAction);
412 m.add(routePipeAction);
413 routePipeAction.setComponent(component);
414 m.add(addComponentAction);
415 addComponentAction.setComponent(component);
417 m.add(removeSplitAction);
418 removeSplitAction.setNode(node);
424 translateAction.setNode(node);
425 translateInlineAction.setNode(node);
426 translateFreeVariableLengthAction.setNode(node);
427 rotateAction.setNode(node);
428 removeAction.setNode(node);
431 } catch (DatabaseException e) {
432 ExceptionUtils.logAndShowError(e);
436 private IContentOutlinePage createOutline() {
437 if (rootNode == null || selectionProvider == null)
439 //IContentOutlinePage outlinePage = new VTKContentOutlinePage<Resource,Object>(rootNode, selectionProvider);
440 IContentOutlinePage outlinePage = new P3DContentOutlinePage(rootNode, selectionProvider) {
441 protected void createContextMenu(IMenuManager manager) {
442 Plant3DEditor.this.createContextMenu(manager);
445 outlinePage.addSelectionChangedListener(new ISelectionChangedListener() {
448 public void selectionChanged(SelectionChangedEvent event) {
449 selectionProvider.selectionChanged(event);
455 @SuppressWarnings("rawtypes")
457 public Object getAdapter(Class adapter) {
458 if (IPropertyPage.class.equals(adapter))
459 return new StandardPropertyPage(getSite(),getPropertyContexts());
460 if (IContentOutlinePage.class.equals(adapter)) {
461 return createOutline();
463 if (NodeMap.class.equals(adapter)) {
466 if (INode.class.equals(adapter)) {
469 if (IMapping.class.equals(adapter)) {
472 // if (InteractiveVtkPanel.class.equals(adapter)) {
475 if (VtkView.class.equals(adapter))
477 if (ISelectionProvider.class.equals(adapter))
478 return selectionProvider;
479 return super.getAdapter(adapter);
482 public Set<String> getPropertyContexts() {
483 Set<String> result = new HashSet<String>();
484 result.add("http://www.simantics.org/Project-1.0/ProjectBrowseContext");
488 public InteractiveVtkComposite getPanel() {
492 public P3DRootNode getRootNode() {
496 public IMapping<Resource, Object> getMapping() {
500 public P3DNodeMap getNodeMap() {