bundles/org.simantics.browsing.ui.common/src/org/simantics/browsing/ui/common/contributor/AbstractViewpointContributionContributor.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.common.contributor;
  13
  14 import java.util.Collection;
  15 import java.util.Collections;
  16
  17 import org.simantics.browsing.ui.BuiltinKeys.ViewpointContributionKey;
  18 import org.simantics.browsing.ui.NodeContext;
  19 import org.simantics.browsing.ui.PrimitiveQueryUpdater;
  20 import org.simantics.browsing.ui.Tester;
  21 import org.simantics.browsing.ui.content.Contributor;
  22 import org.simantics.browsing.ui.content.ViewpointContribution;
  23 import org.simantics.browsing.ui.content.ViewpointContributionFactory;
  24 import org.simantics.utils.ReflectionUtils;
  25
  26 /**
  27  * @author Antti Villberg
  28  *
  29  * @param <T> the input type accepted by this contributor
  30  */
  31 public abstract class AbstractViewpointContributionContributor<T> implements Contributor<ViewpointContributionFactory> {
  32
  33     private final Class<?> clazz;
  34
  35     public AbstractViewpointContributionContributor() {
  36         clazz = ReflectionUtils.getSingleParameterType(getClass());
  37     }
  38
  39     public AbstractViewpointContributionContributor(Class<?> clazz) {
  40         this.clazz = clazz;
  41     }
  42
  43     @Override
  44     public Class<?> getInputClass() {
  45         return clazz;
  46     }
  47
  48     @Override
  49     public ViewpointContributionFactory getFactory() {
  50
  51         return new ViewpointContributionFactory() {
  52
  53             @Override
  54             public ViewpointContribution create(final PrimitiveQueryUpdater updater,
  55                     final NodeContext context, final ViewpointContributionKey key) {
  56
  57                 return new ViewpointContribution() {
  58
  59                         Collection<NodeContext> contribution = Collections.emptySet();
  60
  61                         @Override
  62                         public Collection<NodeContext> getContribution() {
  63
  64                                 Collection<NodeContext> current = AbstractViewpointContributionContributor.this.getContribution(updater, context, key);
  65
  66                                 for(NodeContext c : current) updater.incRef(c);
  67                                 for(NodeContext c : contribution) updater.decRef(c);
  68                                 contribution = current;
  69
  70                                 return current;
  71
  72                         }
  73
  74                     @Override
  75                     public Class<?> getInputClass() {
  76                         return clazz;
  77                     }
  78
  79                     @Override
  80                     public Tester getNodeContextTester() {
  81                         return AbstractViewpointContributionContributor.this.getNodeContextTester();
  82                     }
  83
  84                 };
  85
  86             }
  87
  88             @Override
  89             public String getViewpointId() {
  90                 return AbstractViewpointContributionContributor.this.getViewpointId();
  91             }
  92
  93         };
  94
  95     }
  96
  97     abstract public Collection<NodeContext> getContribution(PrimitiveQueryUpdater updater,
  98             NodeContext context, ViewpointContributionKey key);
  99
 100     abstract public String getViewpointId();
 101
 102 }