// Copyright (c) 2018-2020 Thomas Kramer.
// SPDX-FileCopyrightText: 2018-2022 Thomas Kramer
//
// SPDX-License-Identifier: AGPL-3.0-or-later

//! This module contains data types and functions for basic polygons without holes.

use crate::CoordinateType;

use crate::edge::Edge;
use crate::point::Point;
use crate::rect::Rect;

pub use crate::traits::{DoubledOrientedArea, MapPointwise, TryBoundingBox, WindingNumber};

use crate::types::*;

use crate::traits::TryCastCoord;
use num_traits::{Num, NumCast};
use std::cmp::{Ord, PartialEq};
use std::iter::FromIterator;
use std::slice::Iter;

/// A `SimplePolygon` is a polygon defined by vertices. It does not contain holes but can be
/// self-intersecting.
///
/// TODO: Implement `Deref` for accessing the vertices.
#[derive(Clone, Debug, Hash, Eq, PartialEq)]
#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]
pub struct SimplePolygon<T> {
    /// Vertices of the polygon.
    points: Vec<Point<T>>,
    /// Set to `true` only if the vertices are normalized.
    normalized: bool,
}

/// Shorthand notation for creating a simple polygon.
/// # Example
/// ```
/// # #[macro_use]
/// # extern crate iron_shapes;
/// # fn main() {
/// use iron_shapes::prelude::*;
/// let p = simple_polygon!((0, 0), (1, 0), (1, 1));
/// assert_eq!(p, SimplePolygon::from(vec![(0, 0), (1, 0), (1, 1)]));
/// # }
/// ```
#[macro_export]
macro_rules! simple_polygon {
 ($($x:expr),*) => {SimplePolygon::new((vec![$($x.into()),*]))}
}

impl<T> SimplePolygon<T> {
    /// Create a new polygon from a list of points.
    /// The points are taken as they are, without reordering
    /// or simplification.
    pub fn new_raw(points: Vec<Point<T>>) -> Self {
        Self {
            points,
            normalized: false,
        }
    }

    /// Create empty polygon without any vertices.
    pub fn empty() -> Self {
        SimplePolygon {
            points: Vec::new(),
            normalized: true,
        }
    }

    /// Get the number of vertices.
    pub fn len(&self) -> usize {
        self.points.len()
    }

    /// Check if polygon has no vertices.
    pub fn is_empty(&self) -> bool {
        self.points.is_empty()
    }

    /// Shortcut for `self.points.iter()`.
    pub fn iter(&self) -> Iter<Point<T>> {
        self.points.iter()
    }

    /// Access the inner list of points.
    pub fn points(&self) -> &[Point<T>] {
        &self.points
    }
}

impl<T: PartialOrd> SimplePolygon<T> {
    /// Create a new polygon from a list of points.
    /// The polygon is normalized by rotating the points.
    pub fn new(points: Vec<Point<T>>) -> Self {
        Self::new_raw(points).normalized()
    }

    /// Mutably access the inner list of points.
    /// If the polygon was normalized, then the modified list of points will be normalized again.
    pub fn with_points_mut<R>(&mut self, f: impl FnOnce(&mut Vec<Point<T>>) -> R) -> R {
        let normalize = self.normalized;
        self.normalized = false;
        let r = f(&mut self.points);
        if normalize {
            self.normalize()
        }
        r
    }
}

impl<T: PartialOrd> SimplePolygon<T> {
    /// Rotate the vertices to get the lexicographically smallest polygon.
    /// Does not change the orientation.
    pub fn normalize(&mut self) {
        if !self.normalized {
            let rotated = |rot: usize| self.points.iter().cycle().skip(rot).take(self.points.len());
            let best_rot = (0..self.points.len()).skip(1).fold(0, |best_rot, rot| {
                if rotated(rot).lt(rotated(best_rot)) {
                    rot
                } else {
                    best_rot
                }
            });
            self.points.rotate_left(best_rot);
        }
        self.normalized = true;
    }

    /// Rotate the vertices to get the lexicographically smallest polygon.
    /// Does not change the orientation.
    pub fn normalized(mut self) -> Self {
        self.normalize();
        self
    }
}

impl<T: PartialEq> SimplePolygon<T> {
    /// Check if polygons can be made equal by rotating their vertices.
    pub fn normalized_eq(&self, other: &Self) -> bool {
        if self.len() != other.len() {
            false
        } else {
            (0..self.points.len()).any(|rot| {
                let rotated = self.points.iter().cycle().skip(rot).take(self.points.len());
                rotated.eq(other.points.iter())
            })
        }
    }
}

impl<T: Copy> SimplePolygon<T> {
    /// Create a new simple polygon from a rectangle.
    pub fn from_rect(rect: &Rect<T>) -> Self {
        Self {
            points: vec![
                rect.lower_left(),
                rect.lower_right(),
                rect.upper_right(),
                rect.upper_left(),
            ],
            normalized: true,
        }
    }
}

#[test]
fn test_from_rect() {
    let r = Rect::new((1, 2), (3, 4));
    assert_eq!(
        SimplePolygon::from_rect(&r),
        SimplePolygon::from_rect(&r).normalized()
    );
}

impl<T> SimplePolygon<T> {
    /// Get index of previous vertex.
    fn prev(&self, i: usize) -> usize {
        match i {
            0 => self.points.len() - 1,
            x => x - 1,
        }
    }

    /// Get index of next vertex.
    fn next(&self, i: usize) -> usize {
        match i {
            _ if i == self.points.len() - 1 => 0,
            x => x + 1,
        }
    }
}

impl<T: Copy> SimplePolygon<T> {
    /// Get an iterator over the polygon points.
    /// Point 0 is appended to the end to close the cycle.
    fn iter_cycle(&self) -> impl Iterator<Item = &Point<T>> {
        self.points.iter().cycle().take(self.points.len() + 1)
    }

    /// Get all exterior edges of the polygon.
    /// # Examples
    ///
    /// ```
    /// use iron_shapes::simple_polygon::SimplePolygon;
    /// use iron_shapes::edge::Edge;
    /// let coords = vec![(0, 0), (1, 0)];
    ///
    /// let poly = SimplePolygon::from(coords);
    ///
    /// assert_eq!(poly.edges(), vec![Edge::new((0, 0), (1, 0)), Edge::new((1, 0), (0, 0))]);
    ///
    /// ```
    pub fn edges(&self) -> Vec<Edge<T>> {
        self.edges_iter().collect()
    }

    /// Iterate over all edges.
    pub fn edges_iter(&self) -> impl Iterator<Item = Edge<T>> + '_ {
        self.iter()
            .zip(self.iter_cycle().skip(1))
            .map(|(a, b)| Edge::new(a, b))
    }
}

impl<T: CoordinateType> SimplePolygon<T> {
    /// Normalize the points of the polygon such that they are arranged counter-clock-wise.
    ///
    /// After normalizing, `SimplePolygon::area_doubled_oriented()` will return a semi-positive value.
    ///
    /// For self-intersecting polygons, the orientation is not clearly defined. For example an `8` shape
    /// has not orientation.
    /// Here, the oriented area is used to define the orientation.
    pub fn normalize_orientation<Area>(&mut self)
    where
        Area: Num + PartialOrd + From<T>,
    {
        if self.orientation::<Area>() != Orientation::CounterClockWise {
            self.points.reverse();
        }
    }

    /// Call `normalize_orientation()` while taking ownership and returning the result.
    pub fn normalized_orientation<Area>(mut self) -> Self
    where
        Area: Num + PartialOrd + From<T>,
    {
        self.normalize_orientation::<Area>();
        self
    }

    /// Get the orientation of the polygon.
    /// The orientation is defined by the oriented area. A polygon with a positive area
    /// is oriented counter-clock-wise, otherwise it is oriented clock-wise.
    ///
    /// # Examples
    ///
    /// ```
    /// use iron_shapes::simple_polygon::SimplePolygon;
    /// use iron_shapes::point::Point;
    /// use iron_shapes::types::Orientation;
    /// let coords = vec![(0, 0), (3, 0), (3, 1)];
    ///
    /// let poly = SimplePolygon::from(coords);
    ///
    /// assert_eq!(poly.orientation::<i64>(), Orientation::CounterClockWise);
    ///
    /// ```
    pub fn orientation<Area>(&self) -> Orientation
    where
        Area: Num + From<T> + PartialOrd,
    {
        // Find the orientation based the polygon area.
        let area2: Area = self.area_doubled_oriented();

        if area2 > Area::zero() {
            Orientation::CounterClockWise
        } else if area2 < Area::zero() {
            Orientation::ClockWise
        } else {
            debug_assert!(area2 == Area::zero());
            Orientation::Straight
        }
    }

    /// Get the convex hull of the polygon.
    ///
    /// Implements Andrew's Monotone Chain algorithm.
    /// See: <http://geomalgorithms.com/a10-_hull-1.html>
    pub fn convex_hull(&self) -> SimplePolygon<T>
    where
        T: Ord,
    {
        crate::algorithms::convex_hull::convex_hull(self.points.clone())
    }

    /// Test if all edges are parallel to the x or y axis.
    pub fn is_rectilinear(&self) -> bool {
        self.edges_iter().all(|e| e.is_rectilinear())
    }

    /// Get the vertex with lowest x-coordinate. Prefer lower y-coordinates to break ties.
    ///
    /// # Examples
    ///
    /// ```
    /// use iron_shapes::simple_polygon::SimplePolygon;
    /// use iron_shapes::point::Point;
    /// let coords = vec![(0, 0), (1, 0), (-1, 2), (-1, 1)];
    ///
    /// let poly = SimplePolygon::from(coords);
    ///
    /// assert_eq!(poly.lower_left_vertex(), Point::new(-1, 1));
    ///
    /// ```
    pub fn lower_left_vertex(&self) -> Point<T> {
        debug_assert!(!self.points.is_empty());

        self.lower_left_vertex_with_index().1
    }

    /// Get the vertex with lowest x-coordinate and its index.
    /// Prefer lower y-coordinates to break ties.
    fn lower_left_vertex_with_index(&self) -> (usize, Point<T>) {
        debug_assert!(!self.points.is_empty());

        // Find minimum.
        let min = self
            .points
            .iter()
            .enumerate()
            .min_by(|(_, &p1), (_, &p2)| p1.partial_cmp(&p2).unwrap());
        let (idx, point) = min.unwrap();

        (idx, *point)
    }
}

impl<T> WindingNumber<T> for SimplePolygon<T>
where
    T: CoordinateType,
{
    /// Calculate the winding number of the polygon around this point.
    ///
    /// TODO: Define how point on edges and vertices is handled.
    ///
    /// See: <http://geomalgorithms.com/a03-_inclusion.html>
    fn winding_number(&self, point: Point<T>) -> isize {
        let edges = self.edges();
        let mut winding_number = 0isize;

        // Edge Crossing Rules
        //
        // 1. an upward edge includes its starting endpoint, and excludes its final endpoint;
        // 2. a downward edge excludes its starting endpoint, and includes its final endpoint;
        // 3. horizontal edges are excluded
        // 4. the edge-ray intersection point must be strictly right of the point P.

        for e in edges {
            if e.start.y <= point.y {
                // Crosses upward?
                if e.end.y > point.y {
                    // Crosses really upward?
                    // Yes, crosses upward.
                    if e.side_of(point) == Side::Left {
                        winding_number += 1;
                    }
                }
            } else if e.end.y <= point.y {
                // Crosses downward?
                // Yes, crosses downward.
                // `e.start.y > point.y` needs not to be checked anymore.
                if e.side_of(point) == Side::Right {
                    winding_number -= 1;
                }
            }
        }

        winding_number
    }
}

/// Create a polygon from a type that is convertible into an iterator of values convertible to `Point`s.
impl<I, T, P> From<I> for SimplePolygon<T>
where
    T: Copy + PartialOrd,
    I: IntoIterator<Item = P>,
    Point<T>: From<P>,
{
    fn from(iter: I) -> Self {
        let points: Vec<Point<T>> = iter.into_iter().map(|x| x.into()).collect();

        SimplePolygon::new(points)
    }
}

// impl<T: CoordinateType> From<&Rect<T>> for SimplePolygon<T> {
//     fn from(rect: &Rect<T>) -> Self {
//         Self::new(
//             vec![rect.lower_left(), rect.lower_right(),
//                  rect.upper_right(), rect.upper_left()]
//         )
//     }
// }

//
// /// Create a polygon from a `Vec` of values convertible to `Point`s.
// impl<'a, T, P> From<&'a Vec<P>> for SimplePolygon<T>
//     where T: CoordinateType,
//           Point<T>: From<&'a P>
// {
//     fn from(vec: &'a Vec<P>) -> Self {
//         let points: Vec<Point<T>> = vec.into_iter().map(
//             |x| x.into()
//         ).collect();
//
//         SimplePolygon { points }
//     }
// }
//
// /// Create a polygon from a `Vec` of values convertible to `Point`s.
// impl<T, P> From<Vec<P>> for SimplePolygon<T>
//     where T: CoordinateType,
//           Point<T>: From<P>
// {
//     fn from(vec: Vec<P>) -> Self {
//         let points: Vec<Point<T>> = vec.into_iter().map(
//             |x| x.into()
//         ).collect();
//
//         SimplePolygon { points }
//     }
// }

/// Create a polygon from a iterator of values convertible to `Point`s.
impl<T, P> FromIterator<P> for SimplePolygon<T>
where
    T: Copy,
    P: Into<Point<T>>,
{
    fn from_iter<I>(iter: I) -> Self
    where
        I: IntoIterator<Item = P>,
    {
        let points: Vec<Point<T>> = iter.into_iter().map(|x| x.into()).collect();

        assert!(
            points.len() >= 2,
            "A polygon needs to have at least two points."
        );

        Self::new_raw(points)
    }
}

impl<'a, T> IntoIterator for &'a SimplePolygon<T> {
    type Item = &'a Point<T>;

    type IntoIter = std::slice::Iter<'a, Point<T>>;

    fn into_iter(self) -> Self::IntoIter {
        self.points.iter()
    }
}
// impl<T> IntoIterator for SimplePolygon<T> {
//     type Item = Point<T>;

//     type IntoIter = std::vec::IntoIter<Point<T>>;

//     fn into_iter(self) -> Self::IntoIter {
//         self.points.into_iter()
//     }
// }

impl<T> TryBoundingBox<T> for SimplePolygon<T>
where
    T: Copy + PartialOrd,
{
    fn try_bounding_box(&self) -> Option<Rect<T>> {
        if !self.is_empty() {
            let mut x_min = self.points[0].x;
            let mut x_max = x_min;
            let mut y_min = self.points[0].y;
            let mut y_max = y_min;

            for p in self.iter().skip(1) {
                if p.x < x_min {
                    x_min = p.x;
                }
                if p.x > x_max {
                    x_max = p.x;
                }
                if p.y < y_min {
                    y_min = p.y;
                }
                if p.y > y_max {
                    y_max = p.y;
                }
            }

            Some(Rect::new((x_min, y_min), (x_max, y_max)))
        } else {
            None
        }
    }
}

impl<T> MapPointwise<T> for SimplePolygon<T>
where
    T: CoordinateType,
{
    fn transform<F: Fn(Point<T>) -> Point<T>>(&self, tf: F) -> Self {
        let points = self.points.iter().map(|&p| tf(p)).collect();

        let mut new = SimplePolygon::new_raw(points);

        // Make sure the polygon is oriented the same way as before.
        // TODO: Could be done more efficiently if the magnification/mirroring of the transformation is known.
        if new.orientation::<T>() != self.orientation::<T>() {
            new.points.reverse()
        }

        new.normalized()
    }
}

impl<A, T> DoubledOrientedArea<A> for SimplePolygon<T>
where
    T: CoordinateType,
    A: Num + From<T>,
{
    /// Calculates the doubled oriented area.
    ///
    /// Using doubled area allows to compute in the integers because the area
    /// of a polygon with integer coordinates is either integer or half-integer.
    ///
    /// The area will be positive if the vertices are listed counter-clockwise,
    /// negative otherwise.
    ///
    /// Complexity: O(n)
    ///
    /// # Examples
    ///
    /// ```
    /// use iron_shapes::traits::DoubledOrientedArea;
    /// use iron_shapes::simple_polygon::SimplePolygon;
    /// let coords = vec![(0, 0), (3, 0), (3, 1)];
    ///
    /// let poly = SimplePolygon::from(coords);
    ///
    /// let area: i64 = poly.area_doubled_oriented();
    /// assert_eq!(area, 3);
    ///
    /// ```
    fn area_doubled_oriented(&self) -> A {
        let mut sum = A::zero();
        let ps = &self.points;
        for i in 0..ps.len() {
            let dy = ps[self.next(i)].y - ps[self.prev(i)].y;
            let x = ps[i].x;
            sum = sum + A::from(x) * A::from(dy);
        }
        sum
    }
}

impl<T: CoordinateType + NumCast, Dst: CoordinateType + NumCast> TryCastCoord<T, Dst>
    for SimplePolygon<T>
{
    type Output = SimplePolygon<Dst>;

    fn try_cast(&self) -> Option<Self::Output> {
        let new_points: Option<Vec<_>> = self.points.iter().map(|p| p.try_cast()).collect();

        new_points.map(|p| SimplePolygon::new(p))
    }
}

/// Two simple polygons should be the same even if points are shifted cyclical.
#[test]
fn test_normalized_eq() {
    let p1 = simple_polygon!((0, 0), (0, 1), (1, 1), (1, 0));
    let p2 = simple_polygon!((0, 0), (0, 1), (1, 1), (1, 0));
    assert!(p1.normalized_eq(&p2));

    let p2 = simple_polygon!((0, 1), (1, 1), (1, 0), (0, 0));
    assert!(p1.normalized_eq(&p2));
}

#[test]
fn test_normalize() {
    let p1 = simple_polygon!((0, 0), (0, 1), (1, 1), (1, 0));
    let p2 = simple_polygon!((0, 1), (1, 1), (1, 0), (0, 0));

    assert_eq!(p1.normalized(), p2.normalized());
}

/// Simple sanity check for computation of bounding box.
#[test]
fn test_bounding_box() {
    let p = simple_polygon!((0, 0), (0, 1), (1, 1));
    assert_eq!(p.try_bounding_box(), Some(Rect::new((0, 0), (1, 1))));
}