bundles/org.simantics.browsing.ui.graph.impl/src/org/simantics/browsing/ui/graph/impl/RelatedObjectsQueryProcessor.java

   1 /*******************************************************************************\r
   2  * Copyright (c) 2007, 2010 Association for Decentralized Information Management\r
   3  * in 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.graph.impl;\r
  13 \r
  14 import java.util.Collection;\r
  15 import java.util.Collections;\r
  16 \r
  17 import org.eclipse.core.runtime.IAdaptable;\r
  18 import org.simantics.browsing.ui.BuiltinKeys;\r
  19 import org.simantics.browsing.ui.NodeContext;\r
  20 import org.simantics.browsing.ui.NodeContext.PrimitiveQueryKey;\r
  21 import org.simantics.browsing.ui.graph.impl.CommonKeys.RelatedObjectsKey;\r
  22 import org.simantics.db.ReadGraph;\r
  23 import org.simantics.db.Resource;\r
  24 import org.simantics.db.exception.DatabaseException;\r
  25 import org.simantics.utils.Container;\r
  26 \r
  27 /**\r
  28  * @author Tuukka Lehtonen\r
  29  */\r
  30 public class RelatedObjectsQueryProcessor extends LazyGraphQueryProcessor<Collection<Resource>> {\r
  31 \r
  32     @Override\r
  33     public String toString() {\r
  34         return "RelatedObjectsProcessor";\r
  35     }\r
  36 \r
  37     @Override\r
  38     public Object getIdentifier() {\r
  39         return CommonKeys.RelatedObjectsKey.class;\r
  40     }\r
  41 \r
  42     @Override\r
  43     protected Collection<Resource> initial() {\r
  44         return Collections.emptySet();\r
  45     }\r
  46 \r
  47     @Override\r
  48     protected Collection<Resource> compute(ReadGraph graph, NodeContext context, PrimitiveQueryKey<Container<Collection<Resource>>> key) throws DatabaseException {\r
  49         Object input = context.getConstant(BuiltinKeys.INPUT);\r
  50         Resource subject = adaptToSingle(input, Resource.class);\r
  51 \r
  52         RelatedObjectsKey k = (RelatedObjectsKey) key;\r
  53         final Resource relation = k.getParameter(0);\r
  54         if (relation == null)\r
  55             return Collections.emptyList();\r
  56 \r
  57         return graph.getObjects(subject, relation);\r
  58     }\r
  59 \r
  60     @SuppressWarnings("unchecked")\r
  61     public static <T> T adaptToSingle(Object o, Class<T> clazz) {\r
  62         if (clazz.isInstance(o)) {\r
  63             return (T) o;\r
  64         } else if (o instanceof IAdaptable) {\r
  65             IAdaptable a = (IAdaptable) o;\r
  66             return (T) a.getAdapter(clazz);\r
  67         } else if (o instanceof Container<?>) {\r
  68             Object obj = ((Container<?>) o).get();\r
  69             if (obj == o)\r
  70                 return null;\r
  71             return adaptToSingle(obj, clazz);\r
  72         }\r
  73         return null;\r
  74     }\r
  75 \r
  76 }\r