bundles/org.simantics.browsing.ui.graph.impl/src/org/simantics/browsing/ui/graph/impl/contributor/labeler/LabelerContributorImpl.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.contributor.labeler;\r
  13 \r
  14 import java.util.Map;\r
  15 \r
  16 import org.simantics.browsing.ui.BuiltinKeys;\r
  17 import org.simantics.browsing.ui.BuiltinKeys.LabelerKey;\r
  18 import org.simantics.browsing.ui.NodeContext;\r
  19 import org.simantics.browsing.ui.PrimitiveQueryUpdater;\r
  20 import org.simantics.browsing.ui.Tester;\r
  21 import org.simantics.browsing.ui.common.ColumnKeys;\r
  22 import org.simantics.browsing.ui.content.Contributor;\r
  23 import org.simantics.browsing.ui.content.Labeler;\r
  24 import org.simantics.browsing.ui.content.Labeler.Modifier;\r
  25 import org.simantics.browsing.ui.content.LabelerFactory;\r
  26 import org.simantics.browsing.ui.graph.impl.contribution.LabelerContributionImpl;\r
  27 import org.simantics.db.ReadGraph;\r
  28 import org.simantics.db.UndoContext;\r
  29 import org.simantics.db.exception.DatabaseException;\r
  30 import org.simantics.utils.ReflectionUtils;\r
  31 import org.simantics.utils.datastructures.ArrayMap;\r
  32 \r
  33 abstract public class LabelerContributorImpl<T> implements Contributor<LabelerFactory> {\r
  34 \r
  35     abstract public String getLabel(ReadGraph graph, T input) throws DatabaseException;\r
  36 \r
  37     public int getCategory(ReadGraph graph, T input) throws DatabaseException {\r
  38         return 0;\r
  39     }\r
  40 \r
  41     public Modifier getModifier(ReadGraph graph, T input, String columnKey) throws DatabaseException {\r
  42         return null;\r
  43     }\r
  44 \r
  45     final private Class<?> clazz;\r
  46 \r
  47     @Override\r
  48     public Tester getNodeContextTester() {\r
  49         return null;\r
  50     }\r
  51 \r
  52     public LabelerContributorImpl() {\r
  53         clazz = ReflectionUtils.getSingleParameterType(getClass());\r
  54     }\r
  55 \r
  56     public LabelerContributorImpl(Class<?> clazz) {\r
  57         this.clazz = clazz;\r
  58     }\r
  59 \r
  60     @Override\r
  61     public LabelerFactory getFactory() {\r
  62 \r
  63         return new LabelerFactory() {\r
  64 \r
  65             @Override\r
  66             public Labeler create(final PrimitiveQueryUpdater updater, NodeContext context, final LabelerKey key) {\r
  67 \r
  68                 return new LabelerContributionImpl(updater, context, key) {\r
  69 \r
  70                     @SuppressWarnings("unchecked")\r
  71                     @Override\r
  72                     public Map<String, String> labels(ReadGraph graph, NodeContext context) throws DatabaseException {\r
  73 \r
  74                         T input = (T)context.getConstant(BuiltinKeys.INPUT);\r
  75 \r
  76                         String value = LabelerContributorImpl.this.getLabel(graph, input);\r
  77 \r
  78                         return new ArrayMap<String, String>(ColumnKeys.KEYS_SINGLE,\r
  79                                 new String[] { value });\r
  80 \r
  81                     }\r
  82 \r
  83                     @SuppressWarnings("unchecked")\r
  84                     @Override\r
  85                     public Modifier getModifier(ReadGraph graph, UndoContext undoContext, NodeContext context, String columnKey) throws DatabaseException {\r
  86 \r
  87                         T input = (T)context.getConstant(BuiltinKeys.INPUT);\r
  88 \r
  89                         return LabelerContributorImpl.this.getModifier(graph, input, columnKey);\r
  90 \r
  91                     }\r
  92 \r
  93                     @SuppressWarnings("unchecked")\r
  94                     @Override\r
  95                     public int category(ReadGraph graph, NodeContext context) throws DatabaseException {\r
  96 \r
  97                         T input = (T)context.getConstant(BuiltinKeys.INPUT);\r
  98 \r
  99                         return LabelerContributorImpl.this.getCategory(graph, input);\r
 100 \r
 101                     }\r
 102 \r
 103                     @Override\r
 104                     public String toString() {\r
 105                         return LabelerContributorImpl.this.toString();\r
 106                     }\r
 107 \r
 108                 };\r
 109 \r
 110             }\r
 111 \r
 112         };\r
 113 \r
 114     }\r
 115 \r
 116     @Override\r
 117     public Class<?> getInputClass() {\r
 118         return clazz;\r
 119     }\r
 120 \r
 121 }\r