| 1 | /* |
| 2 | * To change this template, choose Tools | Templates |
| 3 | * and open the template in the editor. |
| 4 | */ |
| 5 | package org.ufcspa.simdecs.diagram; |
| 6 | |
| 7 | import java.util.ArrayList; |
| 8 | import java.util.HashMap; |
| 9 | import java.util.Map; |
| 10 | import java.util.TreeMap; |
| 11 | import org.ufcspa.simdecs.diagram.elements.SwimLane; |
| 12 | |
| 13 | /** |
| 14 | * |
| 15 | * @author Maroni |
| 16 | */ |
| 17 | public class DiagramManager { |
| 18 | |
| 19 | private String diagramName; |
| 20 | private Map<String, SwimLane> swimLanes; |
| 21 | |
| 22 | public DiagramManager(String diagramName) { |
| 23 | this.diagramName = diagramName; |
| 24 | swimLanes = new TreeMap<String, SwimLane>(); |
| 25 | } |
| 26 | |
| 27 | public void addSwimLane(SwimLane swimLane) { |
| 28 | swimLanes.put(swimLane.getId(), swimLane); |
| 29 | } |
| 30 | |
| 31 | public final void clear() { |
| 32 | swimLanes = new HashMap<String, SwimLane>(); |
| 33 | } |
| 34 | |
| 35 | public String getDiagramName() { |
| 36 | return diagramName; |
| 37 | } |
| 38 | |
| 39 | public void setDiagramName(String diagramName) { |
| 40 | this.diagramName = diagramName; |
| 41 | } |
| 42 | |
| 43 | public ArrayList<SwimLane> getSwimLanes() { |
| 44 | return new ArrayList(swimLanes.values()); |
| 45 | } |
| 46 | |
| 47 | public SwimLane getSwimLane(String id) { |
| 48 | return swimLanes.get(id); |
| 49 | } |
| 50 | |
| 51 | public void removeSwimLane(String id) { |
| 52 | swimLanes.remove(id); |
| 53 | } |
| 54 | } |