gerrit.simantics Code Review - simantics/sysdyn.git/blob

   1 /*******************************************************************************\r
   2  * Copyright (c) 2007, 2011 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.jfreechart.chart.properties.pie;\r
  13 \r
  14 import java.util.Collection;\r
  15 \r
  16 import org.eclipse.jface.layout.GridDataFactory;\r
  17 import org.eclipse.jface.layout.GridLayoutFactory;\r
  18 import org.eclipse.jface.viewers.ISelectionProvider;\r
  19 import org.eclipse.jface.viewers.IStructuredSelection;\r
  20 import org.eclipse.swt.SWT;\r
  21 import org.eclipse.swt.custom.ScrolledComposite;\r
  22 import org.eclipse.swt.events.SelectionAdapter;\r
  23 import org.eclipse.swt.events.SelectionEvent;\r
  24 import org.eclipse.swt.graphics.Point;\r
  25 import org.eclipse.swt.widgets.Composite;\r
  26 import org.eclipse.swt.widgets.Control;\r
  27 import org.eclipse.swt.widgets.Tree;\r
  28 import org.eclipse.ui.IWorkbenchSite;\r
  29 import org.simantics.browsing.ui.swt.SingleSelectionInputSource;\r
  30 import org.simantics.browsing.ui.swt.widgets.Button;\r
  31 import org.simantics.browsing.ui.swt.widgets.GraphExplorerComposite;\r
  32 import org.simantics.browsing.ui.swt.widgets.impl.SelectionListenerImpl;\r
  33 import org.simantics.browsing.ui.swt.widgets.impl.Widget;\r
  34 import org.simantics.browsing.ui.swt.widgets.impl.WidgetSupport;\r
  35 import org.simantics.browsing.ui.swt.widgets.impl.WidgetSupportImpl;\r
  36 import org.simantics.db.Resource;\r
  37 import org.simantics.db.WriteGraph;\r
  38 import org.simantics.db.common.request.PossibleObjectWithType;\r
  39 import org.simantics.db.common.utils.ListUtils;\r
  40 import org.simantics.db.exception.DatabaseException;\r
  41 import org.simantics.db.layer0.util.RemoverUtil;\r
  42 import org.simantics.db.management.ISessionContext;\r
  43 import org.simantics.jfreechart.chart.ChartUtils;\r
  44 import org.simantics.jfreechart.chart.properties.AllVariablesOfModel;\r
  45 import org.simantics.jfreechart.chart.properties.ChartVariable;\r
  46 import org.simantics.jfreechart.chart.properties.LabelPropertyTabContributor;\r
  47 import org.simantics.jfreechart.chart.properties.xyline.AxisAndVariablesExplorerComposite;\r
  48 import org.simantics.layer0.Layer0;\r
  49 import org.simantics.sysdyn.JFreeChartResource;\r
  50 import org.simantics.ui.utils.AdaptionUtils;\r
  51 import org.simantics.utils.datastructures.ArrayMap;\r
  52 import org.simantics.utils.datastructures.Pair;\r
  53 import org.simantics.utils.ui.ExceptionUtils;\r
  54 \r
  55 /**\r
  56  * Tab for modifying series in a pie chart configuration\r
  57  * @author Teemu Lempinen\r
  58  *\r
  59  */\r
  60 public class PieSeriesTab2 extends LabelPropertyTabContributor implements Widget {\r
  61 \r
  62     private GraphExplorerComposite explorer;\r
  63     private ScrolledComposite propertyContainer;\r
  64     private WidgetSupportImpl additionalSupport;\r
  65     private Button add, remove;\r
  66     private Resource chartResource;\r
  67 \r
  68     public PieSeriesTab2() {\r
  69         additionalSupport = new WidgetSupportImpl();\r
  70     }\r
  71 \r
  72     @Override\r
  73     public void createControls(Composite body, IWorkbenchSite site, final ISessionContext context, WidgetSupport support) {\r
  74         support.register(this);\r
  75         Composite composite = new Composite(body, SWT.NONE);\r
  76         GridDataFactory.fillDefaults().grab(true, true).applyTo(composite);\r
  77         GridLayoutFactory.fillDefaults().numColumns(2).margins(3, 3).applyTo(composite);\r
  78 \r
  79         // (Ontology-based) GraphExplorer displaying variables of a pie chart\r
  80         explorer = new AxisAndVariablesExplorerComposite(ArrayMap.keys(\r
  81                 "displaySelectors", "displayFilter").values(false, false), site, composite, support, SWT.FULL_SELECTION | SWT.BORDER | SWT.SINGLE);\r
  82         explorer.setBrowseContexts(JFreeChartResource.URIs.PieSeriesBrowseContext);\r
  83         explorer.setInputSource(new SingleSelectionInputSource(\r
  84                 Resource.class));\r
  85         explorer.getExplorer().setAutoExpandLevel(2); // Expand everything in the beginning\r
  86         explorer.finish();\r
  87 \r
  88         ((Tree)explorer.getExplorerControl()).addSelectionListener(new SelectionAdapter() {\r
  89             public void widgetSelected(SelectionEvent e) {\r
  90                 updateSelection(context);\r
  91             }\r
  92         });\r
  93         GridDataFactory.fillDefaults().hint(250, SWT.DEFAULT).grab(false, true).applyTo(explorer);\r
  94 \r
  95         // Scrolled composite for displaying properties of a selection in explorer\r
  96         propertyContainer = new ScrolledComposite(composite, SWT.H_SCROLL | SWT.V_SCROLL);\r
  97         GridDataFactory.fillDefaults().span(1, 2).grab(true, true).applyTo(propertyContainer);\r
  98         GridLayoutFactory.fillDefaults().applyTo(propertyContainer);\r
  99         propertyContainer.setExpandHorizontal(true);\r
 100         propertyContainer.setExpandVertical(true);\r
 101 \r
 102 \r
 103         // Buttons for adding and removing variables from a pie plot\r
 104         Composite buttonComposite = new Composite(composite, SWT.NONE);\r
 105         GridDataFactory.fillDefaults().applyTo(buttonComposite);\r
 106         GridLayoutFactory.fillDefaults().numColumns(3).applyTo(buttonComposite);\r
 107 \r
 108         add = new Button(buttonComposite, additionalSupport, SWT.NONE);\r
 109         add.setText("Add");\r
 110         add.addSelectionListener(new NewVariableListener(context));\r
 111 \r
 112         remove = new Button(buttonComposite, additionalSupport, SWT.NONE);\r
 113         remove.setText("Remove");\r
 114         remove.addSelectionListener(new RemoveListener(context));\r
 115     }\r
 116 \r
 117     /**\r
 118      * Updates the content of propertyContainer  \r
 119      * @param context\r
 120      */\r
 121     private void updateSelection(ISessionContext context) {\r
 122         ISelectionProvider selectionProvider = (ISelectionProvider)explorer.getAdapter(ISelectionProvider.class);\r
 123         IStructuredSelection selection = (IStructuredSelection)selectionProvider.getSelection();\r
 124         final Resource resource = AdaptionUtils.adaptToSingle(selection, Resource.class);\r
 125         if(resource == null)\r
 126             return;\r
 127 \r
 128         for(Control child : propertyContainer.getChildren()) {\r
 129             child.dispose();\r
 130         }\r
 131 \r
 132         try {\r
 133                 Collection<ChartVariable> variables = context.getSession().syncRequest(new AllVariablesOfModel(resource));\r
 134                 PieSeriesPropertyComposite2 spc = new PieSeriesPropertyComposite2(propertyContainer, context, additionalSupport, variables, SWT.NONE);\r
 135                 propertyContainer.setContent(spc);\r
 136                 Point size = spc.computeSize(SWT.DEFAULT, SWT.DEFAULT);\r
 137                 propertyContainer.setMinSize(size);\r
 138         \r
 139                 additionalSupport.fireInput(context, selection);\r
 140         } catch (DatabaseException e) {\r
 141                 ExceptionUtils.logAndShowError(e);\r
 142         }\r
 143     }\r
 144 \r
 145 \r
 146     /**\r
 147      * SelectionListener for adding a new variable to a plot\r
 148      * @author Teemu Lempinen\r
 149      *\r
 150      */\r
 151     private class NewVariableListener extends SelectionListenerImpl<Resource> {\r
 152 \r
 153         public NewVariableListener(ISessionContext context) {\r
 154             super(context);\r
 155         }\r
 156 \r
 157         @Override\r
 158         public void apply(WriteGraph graph, Resource input) throws DatabaseException {\r
 159             JFreeChartResource jfree = JFreeChartResource.getInstance(graph);\r
 160             Layer0 l0 = Layer0.getInstance(graph);\r
 161             Resource dataset = null;\r
 162             if(input == null) {\r
 163                 if(chartResource != null) {\r
 164                     Resource plot = graph.syncRequest(new PossibleObjectWithType(chartResource, l0.ConsistsOf, jfree.Plot));\r
 165                     if(plot != null)\r
 166                         dataset = graph.syncRequest(new PossibleObjectWithType(plot, l0.ConsistsOf, jfree.Dataset));\r
 167                 }\r
 168             } else {\r
 169                 if(graph.isInstanceOf(input, jfree.Series)) {\r
 170                     dataset = graph.getPossibleObject(input, l0.PartOf);\r
 171                 }\r
 172             }\r
 173             if(dataset != null) {\r
 174                 // Create series with no rvi\r
 175                 Resource series = ChartUtils.createSeries(graph, dataset, null);\r
 176                 graph.claimLiteral(series, jfree.Series_exploded, false);\r
 177             }\r
 178         }\r
 179     }\r
 180 \r
 181     /**\r
 182      * SelectionListener for remove button\r
 183      * @author Teemu Lempinen\r
 184      *\r
 185      */\r
 186     private class RemoveListener extends SelectionListenerImpl<Resource> {\r
 187 \r
 188         public RemoveListener(ISessionContext context) {\r
 189             super(context);\r
 190         }\r
 191 \r
 192         /**\r
 193          * Removes selected resource from explorer\r
 194          */\r
 195         @Override\r
 196         public void apply(WriteGraph graph, Resource input) throws DatabaseException {\r
 197             if(input == null)\r
 198                 return; \r
 199 \r
 200             JFreeChartResource jfree = JFreeChartResource.getInstance(graph);\r
 201             Layer0 l0 = Layer0.getInstance(graph);\r
 202             Resource list = null;\r
 203             if(graph.isInstanceOf(input, jfree.Series)) {\r
 204                 // Remove series from dataset and seriesList\r
 205                 Resource dataset = graph.getPossibleObject(input, l0.PartOf);\r
 206                 if(dataset != null)\r
 207                     list = graph.getPossibleObject(dataset, jfree.Dataset_seriesList);\r
 208 \r
 209                 if(list != null)\r
 210                     ListUtils.removeElement(graph, list, input);\r
 211                 RemoverUtil.remove(graph, input);\r
 212             }\r
 213         }\r
 214     }\r
 215     \r
 216     @Override\r
 217     public void setInput(ISessionContext context, Object input) {\r
 218         chartResource = AdaptionUtils.adaptToSingle(input, Resource.class);\r
 219     }\r
 220 \r
 221 }\r