--- /dev/null
+/*******************************************************************************\r
+ * Copyright (c) 2007, 2010 Association for Decentralized Information Management\r
+ * in Industry THTH ry.\r
+ * All rights reserved. This program and the accompanying materials\r
+ * are made available under the terms of the Eclipse Public License v1.0\r
+ * which accompanies this distribution, and is available at\r
+ * http://www.eclipse.org/legal/epl-v10.html\r
+ *\r
+ * Contributors:\r
+ * VTT Technical Research Centre of Finland - initial API and implementation\r
+ *******************************************************************************/\r
+package org.simantics.diagram.handler;\r
+\r
+import java.util.ArrayList;\r
+import java.util.Collection;\r
+import java.util.EnumSet;\r
+import java.util.HashSet;\r
+import java.util.List;\r
+import java.util.Set;\r
+\r
+import org.simantics.g2d.connection.handler.ConnectionHandler;\r
+import org.simantics.g2d.element.ElementClass;\r
+import org.simantics.g2d.element.IElement;\r
+import org.simantics.g2d.element.handler.BendsHandler;\r
+import org.simantics.g2d.element.handler.TerminalTopology;\r
+import org.simantics.g2d.elementclass.BranchPoint;\r
+import org.simantics.g2d.elementclass.FlagHandler;\r
+import org.simantics.g2d.elementclass.MonitorHandler;\r
+import org.simantics.g2d.elementclass.NonCopyable;\r
+import org.simantics.g2d.elementclass.ReferenceElement;\r
+\r
+/**\r
+ * @author Tuukka Lehtonen\r
+ */\r
+public class ElementAssortment implements IElementAssortment {\r
+\r
+ public Set<IElement> all;\r
+\r
+ public final List<IElement> nodeList = new ArrayList<IElement>();\r
+ public final Set<IElement> nodes = new HashSet<IElement>();\r
+ public final Set<IElement> connections = new HashSet<IElement>();\r
+ public final Set<IElement> edges = new HashSet<IElement>();\r
+ public final Set<IElement> branchPoints = new HashSet<IElement>();\r
+ public final Set<IElement> flags = new HashSet<IElement>();\r
+ public final Set<IElement> references = new HashSet<IElement>();\r
+ public final Set<IElement> monitors = new HashSet<IElement>();\r
+ public final Set<IElement> others = new HashSet<IElement>();\r
+ public final Set<IElement> noncopyables = new HashSet<IElement>();\r
+\r
+ EnumSet<ElementType> contents;\r
+\r
+ /**\r
+ * @param set all the elements to initially sort out. This object assumes\r
+ * ownership of the set.\r
+ */\r
+ public ElementAssortment(Set<IElement> set) {\r
+ if (set == null)\r
+ throw new IllegalArgumentException("null element set");\r
+ this.all = new HashSet<IElement>(set);\r
+ sort(set);\r
+ }\r
+\r
+ private void sort(Set<IElement> set) {\r
+ EnumSet<ElementType> es = EnumSet.noneOf(ElementType.class);\r
+ contents = es;\r
+ for (IElement el : set) {\r
+ ElementClass ec = el.getElementClass();\r
+ NonCopyable nc = ec.getAtMostOneItemOfClass(NonCopyable.class);\r
+ ReferenceElement re = ec.getAtMostOneItemOfClass(ReferenceElement.class);\r
+ MonitorHandler mh = ec.getAtMostOneItemOfClass(MonitorHandler.class);\r
+ FlagHandler fh = ec.getAtMostOneItemOfClass(FlagHandler.class);\r
+ TerminalTopology tt = ec.getAtMostOneItemOfClass(TerminalTopology.class);\r
+ BendsHandler bh = ec.getAtMostOneItemOfClass(BendsHandler.class);\r
+ BranchPoint bp = ec.getAtMostOneItemOfClass(BranchPoint.class);\r
+ ConnectionHandler ch = ec.getAtMostOneItemOfClass(ConnectionHandler.class);\r
+ if (nc != null) {\r
+ noncopyables.add(el);\r
+ es.add(ElementType.NonCopyable);\r
+ // Don't count inherently non-copyables in "all" elements.\r
+ all.remove(el);\r
+ } else if (mh != null) {\r
+ monitors.add(el);\r
+ es.add(ElementType.Monitor);\r
+ } else if (re != null) {\r
+ references.add(el);\r
+ es.add(ElementType.Reference);\r
+ } else if (fh != null) {\r
+ flags.add(el);\r
+ es.add(ElementType.Flag);\r
+ } else if (ch != null) {\r
+ connections.add(el);\r
+ es.add(ElementType.Connection);\r
+ } else if (bp != null) {\r
+ branchPoints.add(el);\r
+ es.add(ElementType.BranchPoint);\r
+ } else if (tt != null) {\r
+ nodes.add(el);\r
+ nodeList.add(el);\r
+ es.add(ElementType.Node);\r
+ } else if (bh != null) {\r
+ edges.add(el);\r
+ es.add(ElementType.Edge);\r
+ } else {\r
+ others.add(el);\r
+ es.add(ElementType.Other);\r
+ }\r
+ }\r
+ }\r
+\r
+ public Set<IElement> getAll() {\r
+ return new HashSet<IElement>(all);\r
+ }\r
+\r
+ public boolean contains(ElementType et) {\r
+ return contents.contains(et);\r
+ }\r
+\r
+ public boolean contains(Collection<ElementType> ets) {\r
+ return contents.containsAll(ets);\r
+ }\r
+\r
+ public boolean containsAny(Collection<ElementType> ets) {\r
+ for (ElementType et : ets)\r
+ if (contents.contains(et))\r
+ return true;\r
+ return false;\r
+ }\r
+\r
+ @Override\r
+ public int count(ElementType et) {\r
+ switch (et) {\r
+ case BranchPoint: return branchPoints.size();\r
+ case Connection: return connections.size();\r
+ case Edge: return edges.size();\r
+ case Flag: return flags.size();\r
+ case Monitor: return monitors.size();\r
+ case Reference: return references.size();\r
+ case Node: return nodes.size();\r
+ case Other: return others.size();\r
+ case NonCopyable: return noncopyables.size();\r
+ default: return 0;\r
+ }\r
+ }\r
+\r
+ boolean add(ElementType et) {\r
+ return contents.add(et);\r
+ }\r
+\r
+ boolean remove(ElementType et) {\r
+ return contents.remove(et);\r
+ }\r
+\r
+ private boolean addType(Set<IElement> set, IElement e, ElementType et) {\r
+ if (all.add(e)) {\r
+ boolean ret = set.add(e);\r
+ if (ret)\r
+ contents.add(et);\r
+ return ret;\r
+ }\r
+ return false;\r
+ }\r
+\r
+ private boolean removeType(Set<IElement> set, IElement e, ElementType et) {\r
+ if (all.remove(e)) {\r
+ boolean ret = set.remove(e);\r
+ if (set.isEmpty())\r
+ contents.remove(et);\r
+ return ret;\r
+ }\r
+ return false;\r
+ }\r
+\r
+ private int removeAllType(Set<IElement> set, Collection<IElement> es, ElementType et) {\r
+ int result = 0;\r
+ for (IElement e : es)\r
+ if (removeType(set, e, et))\r
+ ++result;\r
+ return result;\r
+ }\r
+\r
+ private int clearType(Set<IElement> set, ElementType et) {\r
+ int ret = set.size();\r
+ all.removeAll(set);\r
+ contents.remove(et);\r
+ return ret;\r
+ }\r
+\r
+ public int clear(ElementType et) {\r
+ switch (et) {\r
+ case BranchPoint: return clearType(branchPoints, et);\r
+ case Connection: return clearType(connections, et);\r
+ case Edge: return clearType(edges, et);\r
+ case Flag: return clearType(flags, et);\r
+ case Monitor: return clearType(monitors, et);\r
+ case Reference: return clearType(references, et);\r
+ case Node: return clearType(nodes, et);\r
+ case Other: return clearType(others, et);\r
+ case NonCopyable: return clearType(noncopyables, et);\r
+ default: return 0;\r
+ }\r
+ }\r
+\r
+ public boolean add(ElementType et, IElement e) {\r
+ switch (et) {\r
+ case BranchPoint: return addType(branchPoints, e, et);\r
+ case Connection: return addType(connections, e, et);\r
+ case Edge: return addType(edges, e, et);\r
+ case Flag: return addType(flags, e, et);\r
+ case Monitor: return addType(monitors, e, et);\r
+ case Reference: return addType(references, e, et);\r
+ case Node: return addType(nodes, e, et);\r
+ case Other: return addType(others, e, et);\r
+ case NonCopyable: return addType(noncopyables, e, et);\r
+ default: return false;\r
+ }\r
+ }\r
+\r
+ boolean remove(ElementType et, IElement e) {\r
+ switch (et) {\r
+ case BranchPoint: return removeType(branchPoints, e, et);\r
+ case Connection: return removeType(connections, e, et);\r
+ case Edge: return removeType(edges, e, et);\r
+ case Flag: return removeType(flags, e, et);\r
+ case Monitor: return removeType(monitors, e, et);\r
+ case Reference: return removeType(references, e, et);\r
+ case Node: return removeType(nodes, e, et);\r
+ case Other: return removeType(others, e, et);\r
+ case NonCopyable: return removeType(noncopyables, e, et);\r
+ default: return false;\r
+ }\r
+ }\r
+\r
+ public int removeAll(ElementType et, Collection<IElement> es) {\r
+ if (es.isEmpty())\r
+ return 0;\r
+ switch (et) {\r
+ case BranchPoint: return removeAllType(branchPoints, es, et);\r
+ case Connection: return removeAllType(connections, es, et);\r
+ case Edge: return removeAllType(edges, es, et);\r
+ case Flag: return removeAllType(flags, es, et);\r
+ case Monitor: return removeAllType(monitors, es, et);\r
+ case Reference: return removeAllType(references, es, et);\r
+ case Node: return removeAllType(nodes, es, et);\r
+ case Other: return removeAllType(others, es, et);\r
+ case NonCopyable: return removeAllType(noncopyables, es, et);\r
+ default: return 0;\r
+ }\r
+ }\r
+\r
+ @Override\r
+ public String toString() {\r
+ StringBuilder b = new StringBuilder();\r
+ b.append("ElementAssortment:\n");\r
+ b.append("\t CONTAINS: ");\r
+ b.append(contents);\r
+ b.append("\n");\r
+ for(IElement e : nodes) b.append("\t-node " + e + "\n");\r
+ for(IElement e : connections) b.append("\t-connection " + e + "\n");\r
+ for(IElement e : edges) b.append("\t-edge " + e + "\n");\r
+ for(IElement e : branchPoints) b.append("\t-branch " + e + "\n");\r
+ for(IElement e : flags) b.append("\t-flag " + e + "\n");\r
+ for(IElement e : references) b.append("\t-reference " + e + "\n");\r
+ for(IElement e : monitors) b.append("\t-monitor " + e + "\n");\r
+ for(IElement e : others) b.append("\t-other " + e + "\n");\r
+ for(IElement e : noncopyables) b.append("\t-non-copyable " + e + "\n");\r
+ return b.toString();\r
+ }\r
+\r
+ public boolean isEmpty() {\r
+ return all.isEmpty();\r
+ }\r
+\r
+}\r