bundles/org.simantics.browsing.ui.graph/src/org/simantics/browsing/ui/graph/tester/InheritsSomeTypeTester.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.tester;
  13
  14 import java.util.Set;
  15
  16 import org.simantics.browsing.ui.NodeContext;
  17 import org.simantics.browsing.ui.NodeQueryManager;
  18 import org.simantics.browsing.ui.Tester;
  19 import org.simantics.browsing.ui.graph.impl.CommonKeys;
  20 import org.simantics.db.Resource;
  21 import org.simantics.utils.Container;
  22
  23 public class InheritsSomeTypeTester implements Tester {
  24
  25     private final Resource[] test;
  26
  27     public InheritsSomeTypeTester(Resource... types) {
  28         this.test = types;
  29     }
  30
  31     @Override
  32     public boolean test(NodeQueryManager manager, NodeContext context) {
  33         Container<Set<Resource>> types = manager.query(context, CommonKeys.INHERITS);
  34         for (Resource r : test)
  35             if (types.get().contains(r))
  36                 return true;
  37         return false;
  38     }
  39
  40 }