/*
 * 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);
    }
}