#!/usr/bin/env python
# coding:utf-8
# Author:  mozman <me@mozman.at>
# Purpose: svg shapes
# Created: 08.09.2010
# Copyright (C) 2010, Manfred Moitzi
# License: MIT License

from svgwrite.base import BaseElement
from svgwrite.mixins import Presentation, Markers, Transform


class Line(BaseElement, Transform, Presentation, Markers):
    """The **line** element defines a line segment that starts at one point
    and ends at another.
    """

    elementname = "line"

    def __init__(self, start=(0, 0), end=(0, 0), **extra):
        """
        :param 2-tuple start: start point (**x1**, **y1**)
        :param 2-tuple end: end point (**x2**, **y2**)
        :param extra: additional SVG attributes as keyword-arguments

        """
        super(Line, self).__init__(**extra)
        x1, y1 = start
        x2, y2 = end
        self["x1"] = x1
        self["y1"] = y1
        self["x2"] = x2
        self["y2"] = y2


class Rect(BaseElement, Transform, Presentation):
    """The **rect** element defines a rectangle which is axis-aligned with the current
    user coordinate system. Rounded rectangles can be achieved by setting appropriate
    values for attributes **rx** and **ry**.
    """

    elementname = "rect"

    def __init__(self, insert=(0, 0), size=(1, 1), rx=None, ry=None, **extra):
        """
        :param 2-tuple insert: insert point (**x**, **y**), left-upper point
        :param 2-tuple size: (**width**, **height**)
        :param <length> rx: corner x-radius
        :param <length> ry: corner y-radius
        :param extra: additional SVG attributes as keyword-arguments

        """
        super(Rect, self).__init__(**extra)
        x, y = insert
        width, height = size
        self["x"] = x
        self["y"] = y
        self["width"] = width
        self["height"] = height
        if rx is not None:
            self["rx"] = rx
        if ry is not None:
            self["ry"] = ry


class Circle(BaseElement, Transform, Presentation):
    """The **circle** element defines a circle based on a center point and a radius."""

    elementname = "circle"

    def __init__(self, center=(0, 0), r=1, **extra):
        """
        :param 2-tuple center: circle center point (**cx**, **cy**)
        :param length r: circle-radius **r**
        :param extra: additional SVG attributes as keyword-arguments

        """
        super(Circle, self).__init__(**extra)
        cx, cy = center
        self["cx"] = cx
        self["cy"] = cy
        self["r"] = r


class Ellipse(BaseElement, Transform, Presentation):
    """The **ellipse** element defines an ellipse which is axis-aligned with the
    current user coordinate system based on a center point and two radii.
    """

    elementname = "ellipse"

    def __init__(self, center=(0, 0), r=(1, 1), **extra):
        """
        :param 2-tuple center: ellipse center point (**cx**, **cy**)
        :param 2-tuple r: ellipse radii (**rx**, **ry**)
        :param extra: additional SVG attributes as keyword-arguments

        """
        super(Ellipse, self).__init__(**extra)
        cx, cy = center
        rx, ry = r
        self["cx"] = cx
        self["cy"] = cy
        self["rx"] = rx
        self["ry"] = ry


class Polyline(BaseElement, Transform, Presentation, Markers):
    """The **polyline** element defines a set of connected straight line
    segments. Typically, **polyline** elements define open shapes.
    """

    elementname = "polyline"

    def __init__(self, points=[], **extra):
        """
        :param `iterable` points: `iterable` of points (points are `2-tuples`)
        :param extra: additional SVG attributes as keyword-arguments

        """
        super(Polyline, self).__init__(**extra)
        self.points = list(points)
        if self.debug:
            for point in self.points:
                x, y = point
                self.validator.check_svg_type(x, "coordinate")
                self.validator.check_svg_type(y, "coordinate")

    def get_xml(self):
        self.attribs["points"] = self.points_to_string(self.points)
        return super(Polyline, self).get_xml()

    def points_to_string(self, points):
        """
        Convert a `list` of points `2-tuples` to a `string` ``'p1x,p1y p2x,p2y ...'``.

        """
        strings = []
        for point in points:
            if len(point) != 2:
                raise TypeError("got %s values, but expected 2 values." % len(point))
            x, y = point
            if self.debug:
                self.validator.check_svg_type(x, "coordinate")
                self.validator.check_svg_type(y, "coordinate")
            if self.profile == "tiny":
                if isinstance(x, float):
                    x = round(x, 4)
                if isinstance(y, float):
                    y = round(y, 4)
            point = "%s,%s" % (x, y)
            strings.append(point)
        return " ".join(strings)


class Polygon(Polyline):
    """The **polygon** element defines a closed shape consisting of a set of
    connected straight line segments.

    Same as :class:`~svgwrite.shapes.Polyline` but closed.
    """

    elementname = "polygon"
