bundles/org.simantics.browsing.ui.model/src/org/simantics/browsing/ui/model/dnd/DndBrowseContext.java

   1 /*******************************************************************************\r
   2  * Copyright (c) 2010, 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.browsing.ui.model.dnd;\r
  13 \r
  14 import java.util.ArrayList;\r
  15 import java.util.Collection;\r
  16 import java.util.HashSet;\r
  17 \r
  18 import org.eclipse.swt.dnd.DND;\r
  19 import org.simantics.browsing.ui.BuiltinKeys;\r
  20 import org.simantics.browsing.ui.NodeContext;\r
  21 import org.simantics.browsing.ui.model.InvalidContribution;\r
  22 import org.simantics.browsing.ui.model.nodetypes.EntityNodeType;\r
  23 import org.simantics.browsing.ui.model.nodetypes.NodeType;\r
  24 import org.simantics.browsing.ui.model.nodetypes.OrderedNodeTypeMultiMap;\r
  25 import org.simantics.db.ReadGraph;\r
  26 import org.simantics.db.Resource;\r
  27 import org.simantics.db.exception.DatabaseException;\r
  28 import org.simantics.viewpoint.ontology.ViewpointResource;\r
  29 \r
  30 public class DndBrowseContext {\r
  31     OrderedNodeTypeMultiMap<DropActionContribution> dropActionContributions = new OrderedNodeTypeMultiMap<DropActionContribution>();\r
  32     \r
  33     private DndBrowseContext() {\r
  34     }\r
  35     \r
  36     public static DndBrowseContext create(ReadGraph g, Collection<Resource> browseContextResources) throws DatabaseException, InvalidContribution {\r
  37         ViewpointResource vr = ViewpointResource.getInstance(g);\r
  38         DndBrowseContext browseContext = new DndBrowseContext();\r
  39         for(Resource browseContextResource : findSubcontexts(g, browseContextResources)) {\r
  40             for(Resource actionContributionResource : \r
  41                 g.getObjects(browseContextResource, vr.BrowseContext_HasDropActionContribution)) {\r
  42                 DropActionContribution.load(g, actionContributionResource,\r
  43                         browseContext.dropActionContributions\r
  44                         );\r
  45             }\r
  46         }\r
  47         return browseContext;\r
  48     }\r
  49     \r
  50     private static Collection<Resource> findSubcontexts(ReadGraph g,\r
  51             Collection<Resource> browseContexts) throws DatabaseException {\r
  52         ViewpointResource vr = ViewpointResource.getInstance(g);\r
  53         HashSet<Resource> result = new HashSet<Resource>(browseContexts);\r
  54         ArrayList<Resource> stack = new ArrayList<Resource>(browseContexts);\r
  55         while(!stack.isEmpty()) {\r
  56             Resource cur = stack.remove(stack.size()-1);\r
  57             for(Resource sc : g.getObjects(cur, vr.BrowseContext_Includes))\r
  58                 if(result.add(sc))\r
  59                     stack.add(sc);\r
  60         }\r
  61         return result;\r
  62     }    \r
  63     \r
  64     private static NodeType getNodeType(ReadGraph graph, NodeContext parent) throws DatabaseException {\r
  65         NodeType nodeType = parent.getConstant(NodeType.TYPE);\r
  66         if(nodeType == null) {            \r
  67             // TODO remove this code when root of model browser is fixed\r
  68             Object input = parent.getConstant(BuiltinKeys.INPUT);\r
  69             if(input instanceof Resource)\r
  70                 nodeType = EntityNodeType.getNodeTypeFor(graph, (Resource)input);\r
  71         }\r
  72         return nodeType;\r
  73     }\r
  74     \r
  75     /**\r
  76      * Returns drop copy action.\r
  77      */\r
  78     public Runnable getAction(ReadGraph graph, NodeContext target, Object source) throws DatabaseException {\r
  79         return getAction(graph, target, source, DND.DROP_COPY);\r
  80     }\r
  81     \r
  82     /**\r
  83      * Return drop operation for given operation. @See org.eclipse.swt.dnd.DND\r
  84      */\r
  85     public Runnable getAction(ReadGraph graph, NodeContext target, Object source, int operation) throws DatabaseException {\r
  86         NodeType nodeType = getNodeType(graph, target);\r
  87         if(nodeType == null)\r
  88             return null;\r
  89         for(DropActionContribution contribution : dropActionContributions.get(graph, nodeType)) {\r
  90             Runnable action = contribution.getAction(graph, target, source, operation);\r
  91             if(action != null)\r
  92                 return action;\r
  93         }\r
  94         return null;\r
  95     }\r
  96 }\r