--- /dev/null
+/*
+ * To change this template, choose Tools | Templates
+ * and open the template in the editor.
+ */
+package org.ufcspa.simdecs.diagram;
+
+import java.util.ArrayList;
+import java.util.HashMap;
+import java.util.Map;
+import java.util.TreeMap;
+import org.ufcspa.simdecs.diagram.elements.SwimLane;
+
+/**
+ *
+ * @author Maroni
+ */
+public class DiagramManager {
+
+ private String diagramName;
+ private Map<String, SwimLane> swimLanes;
+
+ public DiagramManager(String diagramName) {
+ this.diagramName = diagramName;
+ swimLanes = new TreeMap<String, SwimLane>();
+ }
+
+ public void addSwimLane(SwimLane swimLane) {
+ swimLanes.put(swimLane.getId(), swimLane);
+ }
+
+ public final void clear() {
+ swimLanes = new HashMap<String, SwimLane>();
+ }
+
+ public String getDiagramName() {
+ return diagramName;
+ }
+
+ public void setDiagramName(String diagramName) {
+ this.diagramName = diagramName;
+ }
+
+ public ArrayList<SwimLane> getSwimLanes() {
+ return new ArrayList(swimLanes.values());
+ }
+
+ public SwimLane getSwimLane(String id) {
+ return swimLanes.get(id);
+ }
+
+ public void removeSwimLane(String id) {
+ swimLanes.remove(id);
+ }
+}