bundles/org.simantics.browsing.ui.graph.impl/src/org/simantics/browsing/ui/graph/impl/contributor/viewpoint/ViewpointContributorImpl.java

   1 /*******************************************************************************
   2  * Copyright (c) 2007, 2010 Association for Decentralized Information Management
   3  * in Industry THTH ry.
   4  * All rights reserved. This program and the accompanying materials
   5  * are made available under the terms of the Eclipse Public License v1.0
   6  * which accompanies this distribution, and is available at
   7  * http://www.eclipse.org/legal/epl-v10.html
   8  *
   9  * Contributors:
  10  *     VTT Technical Research Centre of Finland - initial API and implementation
  11  *******************************************************************************/
  12 package org.simantics.browsing.ui.graph.impl.contributor.viewpoint;
  13
  14 import java.util.Collection;
  15 import java.util.Collections;
  16
  17 import org.simantics.browsing.ui.BuiltinKeys;
  18 import org.simantics.browsing.ui.BuiltinKeys.ViewpointContributionKey;
  19 import org.simantics.browsing.ui.NodeContext;
  20 import org.simantics.browsing.ui.PrimitiveQueryUpdater;
  21 import org.simantics.browsing.ui.Tester;
  22 import org.simantics.browsing.ui.common.NodeContextUtil;
  23 import org.simantics.browsing.ui.content.Contributor;
  24 import org.simantics.browsing.ui.content.ViewpointContribution;
  25 import org.simantics.browsing.ui.content.ViewpointContributionFactory;
  26 import org.simantics.browsing.ui.graph.impl.contribution.ViewpointContributionImpl;
  27 import org.simantics.db.ReadGraph;
  28 import org.simantics.db.exception.DatabaseException;
  29 import org.simantics.db.layer0.exception.InvalidVariableException;
  30 import org.simantics.layer0.Layer0;
  31 import org.simantics.operation.Layer0X;
  32 import org.simantics.utils.ReflectionUtils;
  33
  34 abstract public class ViewpointContributorImpl<T> implements Contributor<ViewpointContributionFactory> {
  35
  36     abstract public Collection<?> getContribution(ReadGraph graph, T input) throws DatabaseException;
  37
  38     abstract public String getViewpointId();
  39
  40     final private Class<?> clazz;
  41
  42     protected Layer0 L0;
  43     protected Layer0X L0X;
  44
  45     @Override
  46     public Tester getNodeContextTester() {
  47         return null;
  48     }
  49
  50     public ViewpointContributorImpl() {
  51         clazz = ReflectionUtils.getSingleParameterType(getClass());
  52     }
  53
  54     public ViewpointContributorImpl(Class<?> clazz) {
  55         this.clazz = clazz;
  56     }
  57
  58     @Override
  59     public ViewpointContributionFactory getFactory() {
  60
  61         return new ViewpointContributionFactory() {
  62
  63             @Override
  64             public String getViewpointId() {
  65                 return ViewpointContributorImpl.this.getViewpointId();
  66             }
  67
  68             @Override
  69             public ViewpointContribution create(final PrimitiveQueryUpdater updater, NodeContext context, final ViewpointContributionKey key) {
  70
  71                 return new ViewpointContributionImpl(updater, context, key) {
  72
  73                     @SuppressWarnings("unchecked")
  74                     @Override
  75                     public Collection<NodeContext> children(ReadGraph graph, NodeContext context) throws DatabaseException {
  76
  77                         try {
  78                             T input = (T)context.getConstant(BuiltinKeys.INPUT);
  79                             L0 = Layer0.getInstance(graph);
  80                             L0X = Layer0X.getInstance(graph);
  81                             return NodeContextUtil.toContextCollectionWithInput(ViewpointContributorImpl.this, ViewpointContributorImpl.this.getContribution(graph, input));
  82                         } catch (InvalidVariableException e) {
  83                             return Collections.emptyList();
  84                         }
  85
  86                     }
  87
  88                     @Override
  89                     public Tester getNodeContextTester() {
  90                         return ViewpointContributorImpl.this.getNodeContextTester();
  91                     }
  92
  93                     @Override
  94                     public Class<?> getInputClass() {
  95                         return ViewpointContributorImpl.this.getInputClass();
  96                     }
  97
  98                 };
  99
 100             }
 101
 102         };
 103
 104     }
 105
 106     @Override
 107     public Class<?> getInputClass() {
 108         return clazz;
 109     }
 110
 111 }