+/*******************************************************************************\r
+ * Copyright (c) 2007, 2011 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.editor;\r
+\r
+import java.awt.datatransfer.Transferable;\r
+import java.awt.datatransfer.UnsupportedFlavorException;\r
+import java.awt.dnd.DnDConstants;\r
+import java.awt.dnd.DropTargetDragEvent;\r
+import java.awt.dnd.DropTargetDropEvent;\r
+import java.awt.dnd.DropTargetEvent;\r
+import java.io.IOException;\r
+import java.util.ArrayList;\r
+import java.util.Collection;\r
+import java.util.Collections;\r
+import java.util.List;\r
+\r
+import org.eclipse.jface.viewers.ISelection;\r
+import org.simantics.Simantics;\r
+import org.simantics.charts.query.AddChartItem;\r
+import org.simantics.charts.query.ChartItemDescriptor;\r
+import org.simantics.charts.ui.AddVariableToChartAction;\r
+import org.simantics.charts.ui.ChartDropActionFactory;\r
+import org.simantics.charts.ui.ChartVariable;\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.request.UnaryRead;\r
+import org.simantics.db.exception.DatabaseException;\r
+import org.simantics.db.layer0.SelectionHints;\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.db.layer0.variable.Variables;\r
+import org.simantics.db.request.Read;\r
+import org.simantics.g2d.diagram.participant.AbstractDiagramParticipant;\r
+import org.simantics.g2d.dnd.DragItem;\r
+import org.simantics.g2d.dnd.IDnDContext;\r
+import org.simantics.g2d.dnd.IDragItem;\r
+import org.simantics.g2d.dnd.IDropTargetParticipant;\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.ui.dnd.LocalObjectTransfer;\r
+import org.simantics.ui.dnd.LocalObjectTransferable;\r
+import org.simantics.ui.selection.WorkbenchSelectionElement;\r
+import org.simantics.utils.ui.ErrorLogger;\r
+import org.simantics.utils.ui.ISelectionUtils;\r
+import org.simantics.utils.ui.dialogs.ShowMessage;\r
+\r
+/**\r
+ * @author Tuukka Lehtonen\r
+ */\r
+public class SubscriptionDropParticipant extends AbstractDiagramParticipant implements IDropTargetParticipant { \r
+\r
+ private static class SubscriptionItemDragItem extends DragItem<ChartItemDescriptor> {\r
+ public SubscriptionItemDragItem(ChartItemDescriptor obj) {\r
+ super(obj);\r
+ }\r
+ }\r
+\r
+ private static class VariableReferenceDragItem extends DragItem<VariableReference> {\r
+ public VariableReferenceDragItem(VariableReference obj) {\r
+ super(obj);\r
+ }\r
+ }\r
+ \r
+ Resource container;\r
+ Resource model;\r
+\r
+ public SubscriptionDropParticipant(Resource container) {\r
+ this.container = container;\r
+ try {\r
+ model = Simantics.getSession().sync( new PossibleModel( container ) );\r
+ } catch (DatabaseException e) {\r
+ }\r
+ }\r
+\r
+ @Override\r
+ public void dragEnter(DropTargetDragEvent dtde, IDnDContext dp) {\r
+ // The transferable doesn't know LOT\r
+ if (!dtde.isDataFlavorSupported(LocalObjectTransferable.FLAVOR)) {\r
+ dtde.rejectDrag();\r
+ return;\r
+ }\r
+\r
+ // The source cannot link, too bad\r
+ if ((dtde.getSourceActions() & DnDConstants.ACTION_LINK) == 0) {\r
+ dtde.rejectDrag();\r
+ return;\r
+ }\r
+\r
+ // Ensure the content is usable\r
+ try {\r
+ Transferable t = dtde.getTransferable();\r
+ Object data = t.getTransferData(LocalObjectTransferable.FLAVOR);\r
+ data = LocalObjectTransfer.getTransfer().getObject();\r
+\r
+ List<IDragItem> items = new ArrayList<IDragItem>();\r
+\r
+ final List<Resource> resources = ISelectionUtils.getPossibleKeys(data, SelectionHints.KEY_MAIN, Resource.class);\r
+ if (!resources.isEmpty()) {\r
+ // Support SubscriptionItem drags\r
+ items.addAll( Simantics.getSession().syncRequest(new Read<List<IDragItem>>() {\r
+ @Override\r
+ public List<IDragItem> perform(ReadGraph graph) throws DatabaseException {\r
+ List<IDragItem> result = new ArrayList<IDragItem>();\r
+ //Layer0 L0 = Layer0.getInstance(graph);\r
+ ModelingResources MOD = ModelingResources.getInstance(graph);\r
+ Resource targetModel = graph.syncRequest(new PossibleModel(container));\r
+ if (targetModel != null) {\r
+ for (Resource r : resources) {\r
+ //System.out.println( graph.getPossibleRelatedValue(r, L0.HasName) );\r
+ if (graph.isInstanceOf(r, MOD.Subscription_Item)) {\r
+ Resource model = graph.syncRequest(new PossibleModel(r));\r
+ if (ObjectUtils.objectEquals(targetModel, model))\r
+ result.add( new SubscriptionItemDragItem( AddChartItem.createDescriptor(graph, r) ) );\r
+ }\r
+ }\r
+ }\r
+ return result;\r
+ }\r
+ }) );\r
+ }\r
+\r
+ if(data instanceof RVI) {\r
+\r
+ VariableReferenceDragItem vrdi = new VariableReferenceDragItem(Simantics.getSession().sync(new UnaryRead<RVI, VariableReference>((RVI)data) {\r
+ @Override\r
+ public VariableReference perform(ReadGraph graph) throws DatabaseException {\r
+ return new VariableReference(parameter, Variables.getDatatype(graph, model, parameter), null);\r
+ }\r
+ }));\r
+ items.add( vrdi );\r
+\r
+ } else {\r
+ // Variable/PropertyVariable are mutually exclusive\r
+ List<IDragItem> varItems = null;\r
+\r
+ // 1st try Variable\r
+ final List<Variable> vars2 = ISelectionUtils.getPossibleKeys(data, SelectionHints.KEY_MAIN, Variable.class);\r
+ if (!vars2.isEmpty()) {\r
+ varItems = Simantics.getSession().syncRequest( new Read<List<IDragItem>>() {\r
+ @Override\r
+ public List<IDragItem> perform(ReadGraph graph) throws DatabaseException {\r
+ return toDragItems( graph.syncRequest(VariableReferences.variablesToReferences(model, vars2)) );\r
+ }\r
+ } );\r
+ }\r
+ if (varItems == null || varItems.isEmpty()) {\r
+ // Try legacy PropertyVariables\r
+ final List<PropertyVariables> vars = ISelectionUtils.getPossibleKeys(data, SelectionHints.KEY_MAIN, PropertyVariables.class);\r
+ if (!vars.isEmpty()) {\r
+ varItems = Simantics.getSession().syncRequest( new Read<List<IDragItem>>() {\r
+ @Override\r
+ public List<IDragItem> perform(ReadGraph graph) throws DatabaseException {\r
+ List<PropertyVariables> vars2 = PropertyVariablesImpl.resolve(graph, vars);\r
+ return toDragItems( graph.syncRequest(VariableReferences.toReferences(model, vars2)) );\r
+ }\r
+ } );\r
+ }\r
+ }\r
+ if (varItems != null)\r
+ items.addAll(varItems);\r
+\r
+ // Try WorkbenchSelectionElement\r
+ if (data instanceof ISelection) {\r
+ final List<WorkbenchSelectionElement> wses = ISelectionUtils.filterSelection((ISelection)data, WorkbenchSelectionElement.class);\r
+ if (!wses.isEmpty()) {\r
+ items.addAll( Simantics.getSession().syncRequest( new Read<List<IDragItem>>() {\r
+ @Override\r
+ public List<IDragItem> perform(ReadGraph graph) throws DatabaseException {\r
+ List<Variable> wsevars = new ArrayList<Variable>();\r
+ ChartVariable av = new ChartVariable(graph);\r
+ for(WorkbenchSelectionElement wse : wses) {\r
+ Variable v = wse.getContent(av);\r
+ if(v != null) {\r
+ wsevars.add(v);\r
+ }\r
+ }\r
+ return toDragItems( graph.syncRequest(VariableReferences.variablesToReferences(model, wsevars)) );\r
+ }\r
+ } ) );\r
+ }\r
+ }\r
+ }\r
+\r
+ if (items.isEmpty()) {\r
+ dtde.rejectDrag();\r
+ } else {\r
+ // Accept, make sure it is Link\r
+ for (IDragItem i : items)\r
+ dp.add(i);\r
+ dtde.acceptDrag( DnDConstants.ACTION_LINK );\r
+ }\r
+ } catch (UnsupportedFlavorException e) {\r
+ throw new RuntimeException(e);\r
+ } catch (IOException e) {\r
+ ErrorLogger.defaultLogError(e);\r
+ dtde.rejectDrag();\r
+ } catch (DatabaseException e) {\r
+ ErrorLogger.defaultLogError(e);\r
+ dtde.rejectDrag();\r
+ }\r
+ }\r
+\r
+ @Override\r
+ public void dragExit(DropTargetEvent dte, IDnDContext dp) {\r
+ for (IDragItem i : dp.getItemsByClass(SubscriptionItemDragItem.class))\r
+ dp.remove(i);\r
+ for (IDragItem i : dp.getItemsByClass(VariableReferenceDragItem.class))\r
+ dp.remove(i);\r
+ }\r
+\r
+ @Override\r
+ public void dragOver(DropTargetDragEvent dtde, IDnDContext dp) {\r
+ }\r
+\r
+ @Override\r
+ public void drop(DropTargetDropEvent dtde, IDnDContext dp) {\r
+ // Subscription Item\r
+ Collection<SubscriptionItemDragItem> subs = dp.getItemsByClass(SubscriptionItemDragItem.class);\r
+ if (!subs.isEmpty()) {\r
+ List<ChartItemDescriptor> cicr = new ArrayList<ChartItemDescriptor>();\r
+ for (SubscriptionItemDragItem sidi : subs) {\r
+ cicr.add(sidi.getObject());\r
+ }\r
+ ChartDropActionFactory.addPlots(container, cicr, Collections.<Resource>emptySet()).run();\r
+ dtde.dropComplete(true);\r
+ return;\r
+ }\r
+ \r
+ // Variable Reference\r
+ Collection<VariableReferenceDragItem> vrdis = dp.getItemsByClass(VariableReferenceDragItem.class);\r
+ if (!vrdis.isEmpty()) {\r
+ List<VariableReference> refs = new ArrayList<VariableReference>();\r
+ for (VariableReferenceDragItem vrdi : vrdis) {\r
+ refs.add( vrdi.getObject() );\r
+ }\r
+ Resource chart = container;\r
+ try {\r
+ AddVariableToChartAction a = new AddVariableToChartAction( chart, null, refs );\r
+ a.init();\r
+ a.run();\r
+ dtde.dropComplete(true);\r
+ } catch (DatabaseException e) {\r
+ ShowMessage.showError(e.getClass().getName(), e.getMessage());\r
+ dtde.dropComplete(true);\r
+ }\r
+ return;\r
+ }\r
+ \r
+ dtde.rejectDrop();\r
+ }\r
+\r
+ @Override\r
+ public void dropActionChanged(DropTargetDragEvent dtde, IDnDContext dp) {\r
+ dtde.acceptDrag( DnDConstants.ACTION_LINK );\r
+ }\r
+\r
+ @Override\r
+ public int getAllowedOps() {\r
+ return DnDConstants.ACTION_LINK;\r
+ }\r
+\r
+ private static List<IDragItem> toDragItems(Collection<VariableReference> references) {\r
+ List<IDragItem> result = new ArrayList<IDragItem>(references.size());\r
+ for (VariableReference vr : references) {\r
+ VariableReferenceDragItem di = new VariableReferenceDragItem( vr );\r
+ result.add( di );\r
+ }\r
+ return result;\r
+ }\r
+ \r
+ @Override\r
+ public double getPriority() {\r
+ return 10.0;\r
+ }\r
+\r
+}
\ No newline at end of file