bundles/org.simantics.modeling/src/org/simantics/modeling/requests/CollectionResult.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.modeling.requests;\r
  13 \r
  14 import java.util.Collection;\r
  15 import java.util.Comparator;\r
  16 import java.util.List;\r
  17 import java.util.Map;\r
  18 import java.util.Set;\r
  19 import java.util.Vector;\r
  20 import java.util.concurrent.ConcurrentHashMap;\r
  21 import java.util.concurrent.ConcurrentSkipListSet;\r
  22 \r
  23 import org.eclipse.jface.viewers.IFilter;\r
  24 import org.simantics.db.Resource;\r
  25 import org.simantics.scl.runtime.function.Function1;\r
  26 \r
  27 /**\r
  28  * @author Tuukka Lehtonen\r
  29  */\r
  30 public class CollectionResult {\r
  31 \r
  32     public class DiagramFilter implements IFilter {\r
  33 \r
  34         private final IFilter proxy;\r
  35 \r
  36         public DiagramFilter(IFilter proxy) {\r
  37             this.proxy = proxy;\r
  38         }\r
  39 \r
  40         @Override\r
  41         public boolean select(Object node) {\r
  42             return diagramSet.contains(node) && (proxy == null || proxy.select(node));\r
  43         }\r
  44 \r
  45     }\r
  46 \r
  47     final public Set<Node>           roots       = new ConcurrentSkipListSet<Node>();\r
  48     final private Set<Node>          diagramSet  = new ConcurrentSkipListSet<Node>();\r
  49     final public List<Node>          diagramList = new Vector<Node>();\r
  50     final public Map<Resource, Node> diagrams    = new ConcurrentHashMap<Resource, Node>();\r
  51 \r
  52     public void addDiagram(Resource r, Node n) {\r
  53         diagramList.add(n);\r
  54         diagrams.put(r, n);\r
  55         diagramSet.add(n);\r
  56     }\r
  57 \r
  58     public Collection<Node> breadthFirstFlatten() {\r
  59         return breadthFirstFlatten(null);\r
  60     }\r
  61 \r
  62     public Collection<Node> breadthFirstFlatten(IFilter filter) {\r
  63         return Nodes.breadthFirstFlatten(new DiagramFilter(filter), roots);\r
  64     }\r
  65 \r
  66     public Collection<Node> depthFirstFlatten() {\r
  67         return depthFirstFlatten(null, null);\r
  68     }\r
  69 \r
  70     public Collection<Node> depthFirstFlatten(IFilter filter, Comparator<? super Node> comparator) {\r
  71         return Nodes.depthFirstFlatten(new DiagramFilter(filter), roots, comparator);\r
  72     }\r
  73 \r
  74     /**\r
  75      * @param f\r
  76      *            function that takes the walked Node as argument and returns a\r
  77      *            boolean to describe whether to continue the walk or cancel the\r
  78      *            walk. The returned value cannot be <code>null</code>.\r
  79      * @return <code>true</code> if the walk was completed or <code>false</code>\r
  80      *         if the walk was cancelled\r
  81      */\r
  82     public boolean walkTree(Function1<Node, Boolean> f) {\r
  83         return Nodes.walkTree(f, roots);\r
  84    }\r
  85 \r
  86 }