src/java/org/ufcspa/simdecs/diagram/DiagramManager.java

   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 }