1 files changed, 82 insertions, 0 deletions
diff --git a/src/net/sourceforge/plantuml/nwdiag/DiagGroup.java b/src/net/sourceforge/plantuml/nwdiag/DiagGroup.java
new file mode 100644
index 0000000..58c8cd9
--- /dev/null
+++ b/src/net/sourceforge/plantuml/nwdiag/DiagGroup.java
@@ -0,0 +1,82 @@
+/* ========================================================================
+ * PlantUML : a free UML diagram generator
+ * ========================================================================
+ *
+ * (C) Copyright 2009-2017, Arnaud Roques
+ *
+ * Project Info:  http://plantuml.com
+ * 
+ * If you like this project or if you find it useful, you can support us at:
+ * 
+ * http://plantuml.com/patreon (only 1$ per month!)
+ * http://plantuml.com/paypal
+ * 
+ * This file is part of PlantUML.
+ *
+ * PlantUML is free software; you can redistribute it and/or modify it
+ * under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * PlantUML distributed in the hope that it will be useful, but
+ * WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
+ * or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public
+ * License for more details.
+ *
+ * You should have received a copy of the GNU General Public
+ * License along with this library; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301,
+ * USA.
+ *
+ *
+ * Original Author:  Arnaud Roques
+ *
+ */
+package net.sourceforge.plantuml.nwdiag;
+
+import java.util.HashSet;
+import java.util.Set;
+
+import net.sourceforge.plantuml.graphic.HtmlColor;
+
+public class DiagGroup {
+
+	private final String name;
+	private final Network network;
+	private final Set<String> elements = new HashSet<String>();
+	private HtmlColor color;
+
+	@Override
+	public String toString() {
+		return name + " " + network + " " + elements;
+	}
+
+	public DiagGroup(String name, Network network) {
+		this.name = name;
+		this.network = network;
+	}
+
+	public final String getName() {
+		return name;
+	}
+
+	public void addElement(String name) {
+		this.elements.add(name);
+	}
+
+	public boolean matches(LinkedElement tested) {
+		if (network != null && network != tested.getNetwork()) {
+			return false;
+		}
+		return elements.contains(tested.getElement().getName());
+	}
+
+	public final HtmlColor getColor() {
+		return color;
+	}
+
+	public final void setColor(HtmlColor color) {
+		this.color = color;
+	}
+
+}