org.simantics.g3d.vtk/src/org/simantics/g3d/vtk/common/VTKContentOutlinePage.java

   1 /*******************************************************************************\r
   2  * Copyright (c) 2012, 2013 Association for Decentralized Information Management in\r
   3  * Industry THTH ry.\r
   4  * All rights reserved. This program and the accompanying materials\r
   5  * are made available under the terms of the Eclipse Public License v1.0\r
   6  * which accompanies this distribution, and is available at\r
   7  * http://www.eclipse.org/legal/epl-v10.html\r
   8  *\r
   9  * Contributors:\r
  10  *     VTT Technical Research Centre of Finland - initial API and implementation\r
  11  *******************************************************************************/\r
  12 package org.simantics.g3d.vtk.common;\r
  13 \r
  14 import java.util.ArrayList;\r
  15 import java.util.Collection;\r
  16 import java.util.List;\r
  17 \r
  18 import org.eclipse.jface.viewers.ISelection;\r
  19 import org.eclipse.jface.viewers.ISelectionChangedListener;\r
  20 import org.eclipse.jface.viewers.SelectionChangedEvent;\r
  21 import org.eclipse.jface.viewers.StructuredSelection;\r
  22 import org.eclipse.swt.widgets.Composite;\r
  23 import org.simantics.g3d.scenegraph.IG3DNode;\r
  24 import org.simantics.g3d.scenegraph.base.INode;\r
  25 import org.simantics.g3d.scenegraph.base.ParentNode;\r
  26 import org.simantics.g3d.tools.AdaptationUtils;\r
  27 import org.simantics.g3d.ui.ScenegraphOutlinePage;\r
  28 \r
  29 public class VTKContentOutlinePage<DBObject,JavaObject> extends ScenegraphOutlinePage{\r
  30         private NodeSelectionProvider2<DBObject,JavaObject> provider;\r
  31         \r
  32         public VTKContentOutlinePage(ParentNode<? extends INode> rootNode, NodeSelectionProvider2<DBObject,JavaObject> provider) {\r
  33                 super(rootNode);\r
  34                 this.provider = provider;\r
  35         }\r
  36         \r
  37         private boolean outsideSelection = false;\r
  38         \r
  39         @Override\r
  40         public void createControl(Composite parent) {\r
  41 \r
  42                 super.createControl(parent);\r
  43 \r
  44                 provider.addSelectionChangedListener(new ISelectionChangedListener() {\r
  45                         \r
  46                         @Override\r
  47                         public void selectionChanged(SelectionChangedEvent event) {\r
  48                                 Object source = event.getSource();\r
  49                                 if (source == VTKContentOutlinePage.this)\r
  50                                         return;\r
  51                                 ISelection s = event.getSelection();\r
  52                                 Collection<INode> nodes = AdaptationUtils.adaptToCollection(s, INode.class);\r
  53                                 outsideSelection = true;\r
  54                                 getTreeViewer().setSelection(new StructuredSelection(nodes.toArray()),true);\r
  55                                 outsideSelection = false;\r
  56                                 \r
  57                         }\r
  58                 });\r
  59         }\r
  60         \r
  61         List<INode> selectedNodes = new ArrayList<INode>();\r
  62         \r
  63         @SuppressWarnings("unchecked")\r
  64         @Override\r
  65         protected void fireSelectionChanged(ISelection selection) {\r
  66                 if (outsideSelection)\r
  67                         return;\r
  68                 // TreeViewer provides selected object in tree order, not in selected order.\r
  69                 Collection<INode> selectedUnsortedNodes = AdaptationUtils.adaptToCollection(selection, INode.class);\r
  70                 List<INode> toRemove = new ArrayList<INode>();\r
  71                 for (INode node : selectedNodes) {\r
  72                         if (!selectedUnsortedNodes.contains(node))\r
  73                                 toRemove.add(node);\r
  74                 }\r
  75                 for (INode node : toRemove)\r
  76                         selectedNodes.remove(node);\r
  77                 for (INode node : selectedUnsortedNodes) {\r
  78                         if (!selectedNodes.contains(node))\r
  79                                 selectedNodes.add(node);\r
  80                 }\r
  81                 \r
  82                 List<VTKSelectionItem<DBObject>> selectedItems = new ArrayList<VTKSelectionItem<DBObject>>();\r
  83                 for (INode node : selectedNodes) {\r
  84                         DBObject r = provider.mapping.inverseGet((JavaObject)node);\r
  85                         selectedItems.add(new VTKSelectionItem<DBObject>(null, (IG3DNode)node, r));\r
  86                 }\r
  87                 \r
  88                 super.fireSelectionChanged(new StructuredSelection(selectedItems));\r
  89         }\r
  90         \r
  91 \r
  92 }\r