--- /dev/null
+/*******************************************************************************\r
+ * Copyright (c) 2007, 2014 Association for Decentralized Information Management in\r
+ * Industry THTH ry.\r
+ * All rights reserved. This program and the accompanying materials\r
+ * are made available under the terms of the Eclipse Public License v1.0\r
+ * which accompanies this distribution, and is available at\r
+ * http://www.eclipse.org/legal/epl-v10.html\r
+ *\r
+ * Contributors:\r
+ * VTT Technical Research Centre of Finland - initial API and implementation\r
+ *******************************************************************************/\r
+package org.simantics.charts.ui;\r
+\r
+import java.util.ArrayList;\r
+import java.util.Collection;\r
+import java.util.HashSet;\r
+import java.util.List;\r
+import java.util.Set;\r
+\r
+import org.eclipse.jface.viewers.ISelection;\r
+import org.simantics.Simantics;\r
+import org.simantics.charts.ontology.ChartResource;\r
+import org.simantics.charts.query.AddChartItem;\r
+import org.simantics.charts.query.ChartItemDescriptor;\r
+import org.simantics.databoard.Bindings;\r
+import org.simantics.databoard.type.BooleanType;\r
+import org.simantics.databoard.type.Datatype;\r
+import org.simantics.databoard.type.NumberType;\r
+import org.simantics.databoard.util.ObjectUtils;\r
+import org.simantics.db.ReadGraph;\r
+import org.simantics.db.Resource;\r
+import org.simantics.db.common.procedure.adapter.ProcedureAdapter;\r
+import org.simantics.db.exception.DatabaseException;\r
+import org.simantics.db.layer0.SelectionHints;\r
+import org.simantics.db.layer0.adapter.DropActionFactory;\r
+import org.simantics.db.layer0.request.PossibleModel;\r
+import org.simantics.db.layer0.variable.RVI;\r
+import org.simantics.db.layer0.variable.Variable;\r
+import org.simantics.db.layer0.variable.VariableReference;\r
+import org.simantics.modeling.ModelingResources;\r
+import org.simantics.modeling.PropertyVariables;\r
+import org.simantics.modeling.PropertyVariablesImpl;\r
+import org.simantics.modeling.utils.VariableReferences;\r
+import org.simantics.trend.configuration.TrendItem.Renderer;\r
+import org.simantics.ui.selection.WorkbenchSelectionElement;\r
+import org.simantics.utils.datastructures.collections.CollectionUtils;\r
+import org.simantics.utils.ui.ErrorLogger;\r
+import org.simantics.utils.ui.ISelectionUtils;\r
+\r
+/**\r
+ * @author Tuukka Lehtonen\r
+ */\r
+public class ChartDropActionFactory implements DropActionFactory {\r
+\r
+ @Override\r
+ public Runnable create(ReadGraph g, Object target, Object source, int operation) throws DatabaseException {\r
+ //System.out.println("DROP: " + source + " -> " + target);\r
+\r
+ final Resource chart = ISelectionUtils.getSinglePossibleKey(target, SelectionHints.KEY_MAIN, Resource.class);\r
+ if (chart == null) return null;\r
+ Resource targetModel = g.syncRequest(new PossibleModel(chart));\r
+ if (targetModel == null) return null;\r
+\r
+ if(source instanceof RVI) {\r
+\r
+ RVI rvi = (RVI)source;\r
+ List<VariableReference> refs = CollectionUtils.toList(new VariableReference(rvi, SubscriptionDropActionFactory.getDatatype(g, targetModel, rvi), null));\r
+ \r
+ AddVariableToChartAction a = new AddVariableToChartAction(chart, null, refs);\r
+ a.init(g);\r
+ return a;\r
+ \r
+ }\r
+ \r
+ List<PropertyVariables> vars = ISelectionUtils.getPossibleKeys(source, SelectionHints.KEY_MAIN, PropertyVariables.class);\r
+ if (!vars.isEmpty()) {\r
+ // FIXME: this is a hack for indexed value support\r
+ vars = PropertyVariablesImpl.resolve(g, vars);\r
+ List<VariableReference> references2 = g.syncRequest(VariableReferences.toReferences(targetModel, vars));\r
+ List<VariableReference> references = new ArrayList<VariableReference>();\r
+ for (VariableReference ref : references2) {\r
+ if (ref.datatype instanceof BooleanType || ref.datatype instanceof NumberType) {\r
+ references.add(ref);\r
+ }\r
+ }\r
+ \r
+ AddVariableToChartAction a = new AddVariableToChartAction(chart, null, references);\r
+ a.init(g);\r
+ return a;\r
+ }\r
+ final List<Variable> vars2 = ISelectionUtils.getPossibleKeys(source, SelectionHints.KEY_MAIN, Variable.class);\r
+ if (!vars2.isEmpty()) {\r
+ // FIXME: this is a hack for indexed value support\r
+ List<VariableReference> references2 = g.syncRequest(VariableReferences.variablesToReferences(targetModel, vars2));\r
+ List<VariableReference> references = new ArrayList<VariableReference>();\r
+ for (VariableReference ref : references2) {\r
+ if (ref.datatype instanceof BooleanType || ref.datatype instanceof NumberType) {\r
+ references.add(ref);\r
+ }\r
+ }\r
+ AddVariableToChartAction a = new AddVariableToChartAction(chart, null, references);\r
+ a.init(g);\r
+ return a;\r
+ }\r
+\r
+ if(source instanceof ISelection) {\r
+ List<WorkbenchSelectionElement> wses = ISelectionUtils.filterSelection((ISelection)source, WorkbenchSelectionElement.class);\r
+ if (!wses.isEmpty()) {\r
+ List<Variable> wsevars = new ArrayList<Variable>();\r
+ ChartVariable av = new ChartVariable(g);\r
+ for(WorkbenchSelectionElement wse : wses) {\r
+ Variable v = wse.getContent(av);\r
+ if(v != null) {\r
+ wsevars.add(v);\r
+ }\r
+ }\r
+\r
+ if (!wsevars.isEmpty()) {\r
+ // FIXME: this is a hack for indexed value support\r
+ List<VariableReference> references2 = g.syncRequest(VariableReferences.variablesToReferences(targetModel, wsevars));\r
+ List<VariableReference> references = new ArrayList<VariableReference>();\r
+ for (VariableReference ref : references2) {\r
+ if (ref.datatype instanceof BooleanType || ref.datatype instanceof NumberType) {\r
+ references.add(ref);\r
+ }\r
+ }\r
+ AddVariableToChartAction a = new AddVariableToChartAction(chart, null, references);\r
+ a.init(g);\r
+ return a;\r
+ }\r
+ }\r
+ }\r
+ {\r
+ List<Resource> srcs = ISelectionUtils.getPossibleKeys(source, SelectionHints.KEY_MAIN, Resource.class);\r
+ ModelingResources MOD = ModelingResources.getInstance(g);\r
+ ChartResource CHART = ChartResource.getInstance(g);\r
+ List<ChartItemDescriptor> newItems = new ArrayList<ChartItemDescriptor>();\r
+ Set<Resource> movedPlots = new HashSet<Resource>();\r
+ for (Resource res : srcs) {\r
+ if (g.isInstanceOf(res, MOD.Subscription_Item)) {\r
+ Resource model = g.syncRequest(new PossibleModel(res));\r
+ if (ObjectUtils.objectEquals(targetModel, model)) {\r
+ ChartItemDescriptor desc = new ChartItemDescriptor();\r
+ desc.subscriptionItem = res;\r
+\r
+ Datatype datatype = g.getPossibleRelatedValue(res, MOD.Subscription_Item_Datatype, Bindings.getBindingUnchecked(Datatype.class));\r
+ desc.renderer = datatype instanceof BooleanType ? Renderer.Binary : Renderer.Analog;\r
+\r
+ newItems.add(desc);\r
+ }\r
+ } else if (g.isInstanceOf(res, CHART.Chart_Item)) {\r
+ Resource model = g.syncRequest(new PossibleModel(res));\r
+ if (ObjectUtils.objectEquals(targetModel, model))\r
+ movedPlots.add(res);\r
+ }\r
+ }\r
+ if (!newItems.isEmpty() || !movedPlots.isEmpty())\r
+ return addPlots(chart, newItems, movedPlots);\r
+ }\r
+\r
+ return null;\r
+ }\r
+\r
+ public static Runnable addPlots(final Resource chart, final List<ChartItemDescriptor> references, final Set<Resource> movedPlots) {\r
+ return new Runnable() {\r
+ @Override\r
+ public void run() {\r
+ Simantics.getSession().asyncRequest(\r
+ AddChartItem.addAndMoveChartItems(chart, references, movedPlots),\r
+ new ProcedureAdapter<Collection<Resource>>() {\r
+ @Override\r
+ public void exception(Throwable e) {\r
+ ErrorLogger.defaultLogError(e);\r
+ }\r
+ });\r
+ }\r
+ };\r
+ }\r
+\r
+}
\ No newline at end of file