]> gerrit.simantics Code Review - simantics/3d.git/blobdiff - org.simantics.plant3d/src/org/simantics/plant3d/editor/Plant3DEditor.java
Close Plant3DEditor automatically when its input disappears from DB
[simantics/3d.git] / org.simantics.plant3d / src / org / simantics / plant3d / editor / Plant3DEditor.java
index a97be312f5e3605947f3a45a80c2c2a735bce984..72f2c2066e4329bcac5c306786805ce4146eff01 100644 (file)
@@ -1,13 +1,16 @@
 package org.simantics.plant3d.editor;
 
 import java.util.ArrayList;
+import java.util.Collection;
 import java.util.HashSet;
 import java.util.List;
 import java.util.Set;
 
+import org.eclipse.jface.action.Action;
 import org.eclipse.jface.action.IMenuListener;
 import org.eclipse.jface.action.IMenuManager;
 import org.eclipse.jface.action.MenuManager;
+import org.eclipse.jface.action.Separator;
 import org.eclipse.jface.layout.GridDataFactory;
 import org.eclipse.jface.layout.GridLayoutFactory;
 import org.eclipse.jface.viewers.ISelectionChangedListener;
@@ -16,24 +19,26 @@ import org.eclipse.jface.viewers.SelectionChangedEvent;
 import org.eclipse.swt.SWT;
 import org.eclipse.swt.events.DisposeEvent;
 import org.eclipse.swt.events.DisposeListener;
-import org.eclipse.swt.layout.FillLayout;
 import org.eclipse.swt.widgets.Composite;
 import org.eclipse.swt.widgets.Menu;
 import org.eclipse.ui.views.contentoutline.IContentOutlinePage;
+import org.simantics.Simantics;
 import org.simantics.db.ReadGraph;
 import org.simantics.db.Resource;
 import org.simantics.db.Session;
+import org.simantics.db.common.request.ParametrizedRead;
 import org.simantics.db.common.request.ReadRequest;
 import org.simantics.db.exception.DatabaseException;
+import org.simantics.db.layer0.request.combinations.Combinators;
 import org.simantics.g3d.scenegraph.NodeMap;
 import org.simantics.g3d.scenegraph.RenderListener;
 import org.simantics.g3d.scenegraph.base.INode;
+import org.simantics.g3d.scenegraph.base.ParentNode;
 import org.simantics.g3d.toolbar.ToolComposite;
 import org.simantics.g3d.vtk.action.RemoveAction;
 import org.simantics.g3d.vtk.common.HoverHighlighter;
 import org.simantics.g3d.vtk.common.NodeSelectionProvider2;
 import org.simantics.g3d.vtk.common.SelectionHighlighter;
-import org.simantics.g3d.vtk.common.VTKSelectionItem;
 import org.simantics.g3d.vtk.common.VtkView;
 import org.simantics.g3d.vtk.shape.vtkShape;
 import org.simantics.g3d.vtk.swt.ContextMenuListener;
@@ -58,6 +63,7 @@ import org.simantics.plant3d.ontology.Plant3D;
 import org.simantics.plant3d.scenegraph.EndComponent;
 import org.simantics.plant3d.scenegraph.Equipment;
 import org.simantics.plant3d.scenegraph.IP3DNode;
+import org.simantics.plant3d.scenegraph.IP3DVisualNode;
 import org.simantics.plant3d.scenegraph.InlineComponent;
 import org.simantics.plant3d.scenegraph.Nozzle;
 import org.simantics.plant3d.scenegraph.P3DRootNode;
@@ -74,6 +80,7 @@ import org.simantics.selectionview.StandardPropertyPage;
 import org.simantics.ui.workbench.IPropertyPage;
 import org.simantics.ui.workbench.IResourceEditorInput;
 import org.simantics.ui.workbench.ResourceEditorPart;
+import org.simantics.ui.workbench.editor.input.InputValidationCombinators;
 import org.simantics.utils.threads.ThreadUtils;
 import org.simantics.utils.ui.ExceptionUtils;
 
@@ -82,7 +89,7 @@ import vtk.vtkCameraPass;
 import vtk.vtkDefaultPass;
 import vtk.vtkGaussianBlurPass;
 import vtk.vtkLightsPass;
-import vtk.vtkProp;
+import vtk.vtkProp3D;
 import vtk.vtkRenderPassCollection;
 import vtk.vtkRenderer;
 import vtk.vtkSSAAPass;
@@ -120,24 +127,27 @@ public class Plant3DEditor extends ResourceEditorPart {
        
        private P3DNodeMap nodeMap;
        
+       /** Constants for selecting the up-direction */
+       public static final int X = 0, Y = 1, Z = 2;
+       
+       protected int upDirection = 1;
+
+       ParametrizedRead<IResourceEditorInput, Boolean> INPUT_VALIDATOR =
+                       Combinators.compose(
+                                       InputValidationCombinators.hasURI(),
+                                       InputValidationCombinators.extractInputResource()
+                                       );
+
+       @Override
+       protected ParametrizedRead<IResourceEditorInput, Boolean> getInputValidator() {
+               return INPUT_VALIDATOR;
+       }
+
        @Override
        public void createPartControl(Composite parent) {
                this.parent = parent;
-               //parent.setLayout (new FillLayout ());
-               
-        
-//             component = new SWTAWTComponent(parent,SWT.NONE) {
-//                     
-//                     @Override
-//                     protected Component createSwingComponent() {
-//                             if (panel == null) {
-//                                     panel = new InteractiveVtkPanel();
-//                                     vtkPanelUtil.registerPanel(panel);
-//                                     createScene();
-//                             }
-//                             return panel;
-//                     }
-//             };
+
+               activateValidation();
 
                IResourceEditorInput rei = (IResourceEditorInput)getEditorInput();
                input = rei.getResource();
@@ -162,13 +172,25 @@ public class Plant3DEditor extends ResourceEditorPart {
                new ContextMenuListener(panel, contextMenu);
                
                cameraAction = createCameraAction();
+               switch (upDirection) {
+               case 0:
+                       cameraAction.setUpDirection(new double[] { 1, 0, 0 });
+                       break;
+               case 1:
+                       cameraAction.setUpDirection(new double[] { 0, 1, 0 });
+                       break;
+               case 2:
+                       cameraAction.setUpDirection(new double[] { 0, 0, 1 });
+                       break;
+               }
+               
                panel.setDefaultAction(cameraAction);
                panel.useDefaultAction();
                panel.setPickType(4);
                
                try {
-                       ControlPointFactory.preloadCache(getLibraryUri());
-                       ComponentUtils.preloadCache();
+                       ControlPointFactory.preloadCache(Simantics.getSession(), getLibraryUri());
+                       ComponentUtils.preloadCache(Simantics.getSession());
                } catch (Exception e) {
                        ExceptionUtils.logAndShowError("Cannot open Plant3D editor",e);
                        return;
@@ -237,9 +259,12 @@ public class Plant3DEditor extends ResourceEditorPart {
                }
                
                createActions();
-               
        }
        
+       public void setUpDirection(int upDirection) {
+               this.upDirection = upDirection;
+       }
+
        protected vtkCameraAndSelectorAction createCameraAction() {
            return new vtkCameraAndSelectorAction(panel); 
        }
@@ -288,19 +313,21 @@ public class Plant3DEditor extends ResourceEditorPart {
                     
                     @Override
                     public void postRender() {
+                        panel.removeListener(this);
+                        
                         try {
                             P3DUtil.finalizeDBLoad2(rootNode);
                             if (nodeMap.isRangeModified());
                                 nodeMap.commit("Load sync");
                         } catch (Exception e) {
                             ExceptionUtils.logAndShowError("Failed to load model correctly", e);
-                            //throw new DatabaseException(e);
                         }
-                        panel.removeListener(this);  
-                        
+                                       
+                                       List<vtkProp3D> props = new ArrayList<>();
+                                       collectProps(rootNode, props);
+                                       fitToWindow(props);
                     }
                 });
-                               
                        }
                });
                
@@ -321,7 +348,7 @@ public class Plant3DEditor extends ResourceEditorPart {
                panel.getComponent().setFocus();
        }
        
-       private void createScene() {
+       protected void createScene() {
                vtkRenderer ren1 = panel.getRenderer();
                
                boolean multiPass = false;
@@ -378,14 +405,14 @@ public class Plant3DEditor extends ResourceEditorPart {
                ren1.SetGradientBackground(true);
 
                // vtkActor grid = vtkShape.createGridActor(8,1.0,1|2|4);
-               vtkActor grid = vtkShape.createGridActor(8,1.0, 2 );
+               int dir = 1 << upDirection;
+               vtkActor grid = vtkShape.createGridActor(8, 1.0, dir);
                grid.SetPickable(0);
                ren1.AddActor(grid);
                panel.addDeletable(grid);
                
                AxesDisplay axesDisplay = new AxesDisplay(panel);
                axesDisplay.show();
-               
        }
        
        protected Menu contextMenu;
@@ -404,19 +431,11 @@ public class Plant3DEditor extends ResourceEditorPart {
        
        protected void createContextMenu(IMenuManager m) {
                List<INode> selected = selectionProvider.getSelectedNodes();
-               List<VTKSelectionItem<Resource>> selectedItems = selectionProvider.getSelectedItems();
-               if (selectedItems.size() > 0) {
-                   List<vtkProp> props = new ArrayList<>();
-                   for (VTKSelectionItem<Resource> i : selectedItems) {
-                       vtkProp p = (vtkProp)i.getAdapter(vtkProp.class);
-                       if (p != null)
-                           props.add(p);
-                   }
-                   if (props.size() > 0) {
-                       focusAction.setProps(props);
-                       m.add(focusAction);
-                   }
-               }
+               
+               createFocusMenu(m, selected);
+
+               m.add(new Separator());
+               
                try {
                        if (selected.size() == 0) {
                            m.add(new AddEquipmentAction(rootNode, getLibraryUri()));
@@ -428,7 +447,7 @@ public class Plant3DEditor extends ResourceEditorPart {
                                if (node instanceof Equipment) {
                                        m.add(translateAction);
                                        m.add(rotateAction);
-                                       for (Item eq : P3DUtil.getNozzles(getLibraryUri())) {
+                                       for (Item eq : P3DUtil.getNozzles(Simantics.getSession(), getLibraryUri())) {
                                                AddNozzleAction add = new AddNozzleAction(rootNode, eq);
                                                add.setEquipment((Equipment)node);
                                                m.add(add);
@@ -496,6 +515,39 @@ public class Plant3DEditor extends ResourceEditorPart {
                        ExceptionUtils.logAndShowError(e);
                }
        }
+
+       private class FitToWindow extends Action {
+               private List<INode> selected;
+               public FitToWindow(List<INode> selected) {
+                       super("Fit to Window");
+                       this.selected = selected;
+                       //setAccelerator('1');
+               }
+               @Override
+               public void run() {
+                       List<vtkProp3D> props = new ArrayList<>();
+                       final Collection<INode> collection = !selected.isEmpty() ? selected : getRootNode().getChild();
+                       for (INode n : collection)
+                               collectProps(n, props);
+
+                       fitToWindow(props);
+                       getPanel().refresh();
+               }
+       }
+
+       protected void createFocusMenu(IMenuManager m, List<INode> selected) {
+               m.add(new FitToWindow(selected));
+
+               if (!selected.isEmpty()) {
+                       List<vtkProp3D> actors = new ArrayList<>();
+                       for (INode n : selected)
+                               collectProps(n, actors);
+                       if (actors.size() > 0) {
+                               focusAction.setProps(new ArrayList<>(actors));
+                               m.add(focusAction);
+                       }
+               }
+       }
        
        private IContentOutlinePage createOutline() {
                if (rootNode == null || selectionProvider == null)
@@ -563,4 +615,18 @@ public class Plant3DEditor extends ResourceEditorPart {
        public P3DNodeMap getNodeMap() {
         return nodeMap;
     }
+
+       public void fitToWindow(Collection<vtkProp3D> props) {
+               cameraAction.fitToView(props);
+       }
+
+       protected static void collectProps(INode node, List<vtkProp3D> props) {
+               if (node instanceof IP3DVisualNode)
+                       props.addAll(((IP3DVisualNode) node).getActors());
+               
+               if (node instanceof ParentNode)
+               for (INode n : ((ParentNode<?>) node).getNodes()) {
+                               collectProps(n, props);
+               }
+       }
 }