package maps.convert.osm2gml;

import java.awt.Color;

import java.util.ArrayList;
import java.util.Collection;
import java.util.Collections;

import rescuecore2.misc.gui.ShapeDebugFrame;

import rescuecore2.misc.geometry.Line2D;

/**
   A ShapeInfo that knows how to draw DirectedEdges.
*/
public class DirectedEdgeShapeInfo extends ShapeDebugFrame.Line2DShapeInfo {
    private Collection<DirectedEdge> edges;

    /**
       Create a new DirectedEdgeShapeInfo.
       @param edge The edge to draw.
       @param name The name of the edge.
       @param colour The colour to draw the edge.
       @param thick Whether to draw the edge thick or not.
     */
    public DirectedEdgeShapeInfo(DirectedEdge edge, String name, Color colour, boolean thick)  {
        this(Collections.singleton(edge), name, colour, thick);
    }

    /**
       Create a new DirectedEdgeShapeInfo.
       @param edges The edges to draw.
       @param name The name of the edge.
       @param colour The colour to draw the edge.
       @param thick Whether to draw the edge thick or not.
     */
    public DirectedEdgeShapeInfo(Collection<DirectedEdge> edges, String name, Color colour, boolean thick)  {
        super(makeLines(edges), name, colour, thick, true);
        this.edges = edges;
    }

    @Override
    public Object getObject() {
        return edges;
    }

    private static Collection<Line2D> makeLines(Collection<DirectedEdge> edges) {
        if (edges == null) {
            return null;
        }
        Collection<Line2D> result = new ArrayList<Line2D>();
        for (DirectedEdge next : edges) {
            result.add(next.getLine());
        }
        return result;
    }
}