Migrated source code from Simantics SVN

[simantics/platform.git] / bundles / org.simantics.browsing.ui.model / src / org / simantics / browsing / ui / model / dnd / DndBrowseContext.java

diff --git a/bundles/org.simantics.browsing.ui.model/src/org/simantics/browsing/ui/model/dnd/DndBrowseContext.java b/bundles/org.simantics.browsing.ui.model/src/org/simantics/browsing/ui/model/dnd/DndBrowseContext.java
new file mode 100644 (file)
index 0000000..afd3773
--- /dev/null
+++ b/bundles/org.simantics.browsing.ui.model/src/org/simantics/browsing/ui/model/dnd/DndBrowseContext.java
@@ -0,0 +1,96 @@
+/*******************************************************************************\r
+ * Copyright (c) 2010, 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.browsing.ui.model.dnd;\r
+\r
+import java.util.ArrayList;\r
+import java.util.Collection;\r
+import java.util.HashSet;\r
+\r
+import org.eclipse.swt.dnd.DND;\r
+import org.simantics.browsing.ui.BuiltinKeys;\r
+import org.simantics.browsing.ui.NodeContext;\r
+import org.simantics.browsing.ui.model.InvalidContribution;\r
+import org.simantics.browsing.ui.model.nodetypes.EntityNodeType;\r
+import org.simantics.browsing.ui.model.nodetypes.NodeType;\r
+import org.simantics.browsing.ui.model.nodetypes.OrderedNodeTypeMultiMap;\r
+import org.simantics.db.ReadGraph;\r
+import org.simantics.db.Resource;\r
+import org.simantics.db.exception.DatabaseException;\r
+import org.simantics.viewpoint.ontology.ViewpointResource;\r
+\r
+public class DndBrowseContext {\r
+    OrderedNodeTypeMultiMap<DropActionContribution> dropActionContributions = new OrderedNodeTypeMultiMap<DropActionContribution>();\r
+    \r
+    private DndBrowseContext() {\r
+    }\r
+    \r
+    public static DndBrowseContext create(ReadGraph g, Collection<Resource> browseContextResources) throws DatabaseException, InvalidContribution {\r
+        ViewpointResource vr = ViewpointResource.getInstance(g);\r
+        DndBrowseContext browseContext = new DndBrowseContext();\r
+        for(Resource browseContextResource : findSubcontexts(g, browseContextResources)) {\r
+            for(Resource actionContributionResource : \r
+                g.getObjects(browseContextResource, vr.BrowseContext_HasDropActionContribution)) {\r
+                DropActionContribution.load(g, actionContributionResource,\r
+                        browseContext.dropActionContributions\r
+                        );\r
+            }\r
+        }\r
+        return browseContext;\r
+    }\r
+    \r
+    private static Collection<Resource> findSubcontexts(ReadGraph g,\r
+            Collection<Resource> browseContexts) throws DatabaseException {\r
+        ViewpointResource vr = ViewpointResource.getInstance(g);\r
+        HashSet<Resource> result = new HashSet<Resource>(browseContexts);\r
+        ArrayList<Resource> stack = new ArrayList<Resource>(browseContexts);\r
+        while(!stack.isEmpty()) {\r
+            Resource cur = stack.remove(stack.size()-1);\r
+            for(Resource sc : g.getObjects(cur, vr.BrowseContext_Includes))\r
+                if(result.add(sc))\r
+                    stack.add(sc);\r
+        }\r
+        return result;\r
+    }    \r
+    \r
+    private static NodeType getNodeType(ReadGraph graph, NodeContext parent) throws DatabaseException {\r
+        NodeType nodeType = parent.getConstant(NodeType.TYPE);\r
+        if(nodeType == null) {            \r
+            // TODO remove this code when root of model browser is fixed\r
+            Object input = parent.getConstant(BuiltinKeys.INPUT);\r
+            if(input instanceof Resource)\r
+                nodeType = EntityNodeType.getNodeTypeFor(graph, (Resource)input);\r
+        }\r
+        return nodeType;\r
+    }\r
+    \r
+    /**\r
+     * Returns drop copy action.\r
+     */\r
+    public Runnable getAction(ReadGraph graph, NodeContext target, Object source) throws DatabaseException {\r
+       return getAction(graph, target, source, DND.DROP_COPY);\r
+    }\r
+    \r
+    /**\r
+     * Return drop operation for given operation. @See org.eclipse.swt.dnd.DND\r
+     */\r
+    public Runnable getAction(ReadGraph graph, NodeContext target, Object source, int operation) throws DatabaseException {\r
+        NodeType nodeType = getNodeType(graph, target);\r
+        if(nodeType == null)\r
+            return null;\r
+        for(DropActionContribution contribution : dropActionContributions.get(graph, nodeType)) {\r
+            Runnable action = contribution.getAction(graph, target, source, operation);\r
+            if(action != null)\r
+                return action;\r
+        }\r
+        return null;\r
+    }\r
+}\r