bundles/org.simantics.charts/src/org/simantics/charts/ui/ChartDropActionFactory.java

   1 /*******************************************************************************\r
   2  * Copyright (c) 2007, 2014 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.charts.ui;\r
  13 \r
  14 import java.util.ArrayList;\r
  15 import java.util.Collection;\r
  16 import java.util.HashSet;\r
  17 import java.util.List;\r
  18 import java.util.Set;\r
  19 \r
  20 import org.eclipse.jface.viewers.ISelection;\r
  21 import org.simantics.Simantics;\r
  22 import org.simantics.charts.ontology.ChartResource;\r
  23 import org.simantics.charts.query.AddChartItem;\r
  24 import org.simantics.charts.query.ChartItemDescriptor;\r
  25 import org.simantics.databoard.Bindings;\r
  26 import org.simantics.databoard.type.BooleanType;\r
  27 import org.simantics.databoard.type.Datatype;\r
  28 import org.simantics.databoard.type.NumberType;\r
  29 import org.simantics.databoard.util.ObjectUtils;\r
  30 import org.simantics.db.ReadGraph;\r
  31 import org.simantics.db.Resource;\r
  32 import org.simantics.db.common.procedure.adapter.ProcedureAdapter;\r
  33 import org.simantics.db.exception.DatabaseException;\r
  34 import org.simantics.db.layer0.SelectionHints;\r
  35 import org.simantics.db.layer0.adapter.DropActionFactory;\r
  36 import org.simantics.db.layer0.request.PossibleModel;\r
  37 import org.simantics.db.layer0.variable.RVI;\r
  38 import org.simantics.db.layer0.variable.Variable;\r
  39 import org.simantics.db.layer0.variable.VariableReference;\r
  40 import org.simantics.modeling.ModelingResources;\r
  41 import org.simantics.modeling.PropertyVariables;\r
  42 import org.simantics.modeling.PropertyVariablesImpl;\r
  43 import org.simantics.modeling.utils.VariableReferences;\r
  44 import org.simantics.trend.configuration.TrendItem.Renderer;\r
  45 import org.simantics.ui.selection.WorkbenchSelectionElement;\r
  46 import org.simantics.utils.datastructures.collections.CollectionUtils;\r
  47 import org.simantics.utils.ui.ErrorLogger;\r
  48 import org.simantics.utils.ui.ISelectionUtils;\r
  49 \r
  50 /**\r
  51  * @author Tuukka Lehtonen\r
  52  */\r
  53 public class ChartDropActionFactory implements DropActionFactory {\r
  54 \r
  55     @Override\r
  56     public Runnable create(ReadGraph g, Object target, Object source, int operation) throws DatabaseException {\r
  57         //System.out.println("DROP: " + source + " -> " + target);\r
  58 \r
  59         final Resource chart = ISelectionUtils.getSinglePossibleKey(target, SelectionHints.KEY_MAIN, Resource.class);\r
  60         if (chart == null) return null;\r
  61         Resource targetModel = g.syncRequest(new PossibleModel(chart));\r
  62         if (targetModel == null) return null;\r
  63 \r
  64         if(source instanceof RVI) {\r
  65 \r
  66                 RVI rvi = (RVI)source;\r
  67                 List<VariableReference> refs = CollectionUtils.toList(new VariableReference(rvi, SubscriptionDropActionFactory.getDatatype(g, targetModel, rvi), null));\r
  68                 \r
  69                 AddVariableToChartAction a = new AddVariableToChartAction(chart, null, refs);\r
  70                 a.init(g);\r
  71                 return a;\r
  72                 \r
  73         }\r
  74         \r
  75         List<PropertyVariables> vars = ISelectionUtils.getPossibleKeys(source, SelectionHints.KEY_MAIN, PropertyVariables.class);\r
  76         if (!vars.isEmpty()) {\r
  77             // FIXME: this is a hack for indexed value support\r
  78             vars = PropertyVariablesImpl.resolve(g, vars);\r
  79             List<VariableReference> references2 = g.syncRequest(VariableReferences.toReferences(targetModel, vars));\r
  80             List<VariableReference> references = new ArrayList<VariableReference>();\r
  81             for (VariableReference ref : references2) {\r
  82                 if (ref.datatype instanceof BooleanType || ref.datatype instanceof NumberType) {\r
  83                         references.add(ref);\r
  84                 }\r
  85             }\r
  86             \r
  87                 AddVariableToChartAction a = new AddVariableToChartAction(chart, null, references);\r
  88                 a.init(g);\r
  89                 return a;\r
  90         }\r
  91         final List<Variable> vars2 = ISelectionUtils.getPossibleKeys(source, SelectionHints.KEY_MAIN, Variable.class);\r
  92         if (!vars2.isEmpty()) {\r
  93             // FIXME: this is a hack for indexed value support\r
  94             List<VariableReference> references2 = g.syncRequest(VariableReferences.variablesToReferences(targetModel, vars2));\r
  95             List<VariableReference> references = new ArrayList<VariableReference>();\r
  96             for (VariableReference ref : references2) {\r
  97                 if (ref.datatype instanceof BooleanType || ref.datatype instanceof NumberType) {\r
  98                     references.add(ref);\r
  99                 }\r
 100             }\r
 101             AddVariableToChartAction a = new AddVariableToChartAction(chart, null, references);\r
 102             a.init(g);\r
 103             return a;\r
 104         }\r
 105 \r
 106         if(source instanceof ISelection) {\r
 107             List<WorkbenchSelectionElement> wses = ISelectionUtils.filterSelection((ISelection)source, WorkbenchSelectionElement.class);\r
 108             if (!wses.isEmpty()) {\r
 109                 List<Variable> wsevars = new ArrayList<Variable>();\r
 110                 ChartVariable av = new ChartVariable(g);\r
 111                 for(WorkbenchSelectionElement wse : wses) {\r
 112                     Variable v = wse.getContent(av);\r
 113                     if(v != null) {\r
 114                         wsevars.add(v);\r
 115                     }\r
 116                 }\r
 117 \r
 118                 if (!wsevars.isEmpty()) {\r
 119                     // FIXME: this is a hack for indexed value support\r
 120                     List<VariableReference> references2 = g.syncRequest(VariableReferences.variablesToReferences(targetModel, wsevars));\r
 121                     List<VariableReference> references = new ArrayList<VariableReference>();\r
 122                     for (VariableReference ref : references2) {\r
 123                         if (ref.datatype instanceof BooleanType || ref.datatype instanceof NumberType) {\r
 124                             references.add(ref);\r
 125                         }\r
 126                     }\r
 127                     AddVariableToChartAction a = new AddVariableToChartAction(chart, null, references);\r
 128                     a.init(g);\r
 129                     return a;\r
 130                 }\r
 131             }\r
 132         }\r
 133         {\r
 134             List<Resource> srcs = ISelectionUtils.getPossibleKeys(source, SelectionHints.KEY_MAIN, Resource.class);\r
 135             ModelingResources MOD = ModelingResources.getInstance(g);\r
 136             ChartResource CHART = ChartResource.getInstance(g);\r
 137             List<ChartItemDescriptor> newItems = new ArrayList<ChartItemDescriptor>();\r
 138             Set<Resource> movedPlots = new HashSet<Resource>();\r
 139             for (Resource res : srcs) {\r
 140                 if (g.isInstanceOf(res, MOD.Subscription_Item)) {\r
 141                     Resource model = g.syncRequest(new PossibleModel(res));\r
 142                     if (ObjectUtils.objectEquals(targetModel, model)) {\r
 143                         ChartItemDescriptor desc = new ChartItemDescriptor();\r
 144                         desc.subscriptionItem = res;\r
 145 \r
 146                         Datatype datatype = g.getPossibleRelatedValue(res, MOD.Subscription_Item_Datatype, Bindings.getBindingUnchecked(Datatype.class));\r
 147                         desc.renderer = datatype instanceof BooleanType ? Renderer.Binary : Renderer.Analog;\r
 148 \r
 149                         newItems.add(desc);\r
 150                     }\r
 151                 } else if (g.isInstanceOf(res, CHART.Chart_Item)) {\r
 152                     Resource model = g.syncRequest(new PossibleModel(res));\r
 153                     if (ObjectUtils.objectEquals(targetModel, model))\r
 154                         movedPlots.add(res);\r
 155                 }\r
 156             }\r
 157             if (!newItems.isEmpty() || !movedPlots.isEmpty())\r
 158                 return addPlots(chart, newItems, movedPlots);\r
 159         }\r
 160 \r
 161         return null;\r
 162     }\r
 163 \r
 164     public static Runnable addPlots(final Resource chart, final List<ChartItemDescriptor> references, final Set<Resource> movedPlots) {\r
 165         return new Runnable() {\r
 166             @Override\r
 167             public void run() {\r
 168                 Simantics.getSession().asyncRequest(\r
 169                         AddChartItem.addAndMoveChartItems(chart, references, movedPlots),\r
 170                         new ProcedureAdapter<Collection<Resource>>() {\r
 171                             @Override\r
 172                             public void exception(Throwable e) {\r
 173                                 ErrorLogger.defaultLogError(e);\r
 174                             }\r
 175                         });\r
 176             }\r
 177         };\r
 178     }\r
 179 \r
 180 }