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