bundles/org.simantics.browsing.ui.graph.impl/src/org/simantics/browsing/ui/graph/impl/InstanceOfViewpointFactory.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;
  13
  14 import java.util.Collection;
  15
  16 import org.simantics.browsing.ui.NodeContext;
  17 import org.simantics.browsing.ui.PrimitiveQueryUpdater;
  18 import org.simantics.browsing.ui.BuiltinKeys.ViewpointKey;
  19 import org.simantics.browsing.ui.content.Viewpoint;
  20 import org.simantics.browsing.ui.content.ViewpointFactory;
  21 import org.simantics.db.ReadGraph;
  22 import org.simantics.db.Resource;
  23 import org.simantics.db.exception.DatabaseException;
  24 import org.simantics.layer0.Layer0;
  25
  26 public class InstanceOfViewpointFactory implements ViewpointFactory {
  27
  28     @Override
  29     public Viewpoint create(final PrimitiveQueryUpdater updater, final NodeContext context, final ViewpointKey key) {
  30         assert(updater != null);
  31         assert(context != null);
  32         return new LazyViewpoint(updater, context, key) {
  33             @Override
  34             public NodeContext[] children(ReadGraph graph) throws DatabaseException {
  35                 Layer0 L0 = Layer0.getInstance(graph);
  36                 Collection<Resource> children = graph.getObjects(getInput(Resource.class), L0.InstanceOf);
  37                 return toContextsWithInput(children);
  38             }
  39             @Override
  40             public Boolean hasChildren(ReadGraph graph) throws DatabaseException {
  41                 Layer0 L0 = Layer0.getInstance(graph);
  42                 Collection<Resource> children = graph.getObjects(getInput(Resource.class), L0.InstanceOf);
  43                 return Boolean.valueOf(children.size() > 0);
  44             }
  45         };
  46     }
  47
  48     @Override
  49     public String toString() {
  50         return "Types";
  51     }
  52
  53 }