// pathfinder/geometry/src/basic/point.rs
//
// Copyright © 2019 The Pathfinder Project Developers.
//
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
// option. This file may not be copied, modified, or distributed
// except according to those terms.

//! A SIMD-optimized point type.

use pathfinder_simd::default::{F32x4, I32x4};
use std::ops::{Add, AddAssign, Mul, Neg, Sub};

/// 2D points with 32-bit floating point coordinates.
#[derive(Clone, Copy, Debug, Default)]
pub struct Point2DF32(pub F32x4);

impl Point2DF32 {
    #[inline]
    pub fn new(x: f32, y: f32) -> Point2DF32 {
        Point2DF32(F32x4::new(x, y, 0.0, 0.0))
    }

    #[inline]
    pub fn splat(value: f32) -> Point2DF32 {
        Point2DF32(F32x4::splat(value))
    }

    #[inline]
    pub fn to_3d(self) -> Point3DF32 {
        Point3DF32(self.0.concat_xy_xy(F32x4::new(0.0, 1.0, 0.0, 0.0)))
    }

    #[inline]
    pub fn x(&self) -> f32 {
        self.0[0]
    }

    #[inline]
    pub fn y(&self) -> f32 {
        self.0[1]
    }

    #[inline]
    pub fn set_x(&mut self, x: f32) {
        self.0[0] = x;
    }

    #[inline]
    pub fn set_y(&mut self, y: f32) {
        self.0[1] = y;
    }

    #[inline]
    pub fn min(&self, other: Point2DF32) -> Point2DF32 {
        Point2DF32(self.0.min(other.0))
    }

    #[inline]
    pub fn max(&self, other: Point2DF32) -> Point2DF32 {
        Point2DF32(self.0.max(other.0))
    }

    #[inline]
    pub fn clamp(&self, min_val: Point2DF32, max_val: Point2DF32) -> Point2DF32 {
        self.max(min_val).min(max_val)
    }

    #[inline]
    pub fn det(&self, other: Point2DF32) -> f32 {
        self.x() * other.y() - self.y() * other.x()
    }

    #[inline]
    pub fn dot(&self, other: Point2DF32) -> f32 {
        let xy = self.0 * other.0;
        xy.x() + xy.y()
    }

    #[inline]
    pub fn scale(&self, x: f32) -> Point2DF32 {
        Point2DF32(self.0 * F32x4::splat(x))
    }

    #[inline]
    pub fn scale_xy(&self, factors: Point2DF32) -> Point2DF32 {
        Point2DF32(self.0 * factors.0)
    }

    #[inline]
    pub fn floor(&self) -> Point2DF32 {
        Point2DF32(self.0.floor())
    }

    #[inline]
    pub fn ceil(&self) -> Point2DF32 {
        Point2DF32(self.0.ceil())
    }

    /// Treats this point as a vector and calculates its squared length.
    #[inline]
    pub fn square_length(&self) -> f32 {
        let squared = self.0 * self.0;
        squared[0] + squared[1]
    }

    /// Treats this point as a vector and calculates its length.
    #[inline]
    pub fn length(&self) -> f32 {
        f32::sqrt(self.square_length())
    }

    /// Treats this point as a vector and normalizes it.
    #[inline]
    pub fn normalize(&self) -> Point2DF32 {
        self.scale(1.0 / self.length())
    }

    /// Swaps y and x.
    #[inline]
    pub fn yx(&self) -> Point2DF32 {
        Point2DF32(self.0.yxwz())
    }

    #[inline]
    pub fn is_zero(&self) -> bool {
        *self == Point2DF32::default()
    }

    #[inline]
    pub fn lerp(&self, other: Point2DF32, t: f32) -> Point2DF32 {
        *self + (other - *self).scale(t)
    }
}

impl PartialEq for Point2DF32 {
    #[inline]
    fn eq(&self, other: &Point2DF32) -> bool {
        let results = self.0.packed_eq(other.0);
        results[0] != 0 && results[1] != 0
    }
}

impl Add<Point2DF32> for Point2DF32 {
    type Output = Point2DF32;
    #[inline]
    fn add(self, other: Point2DF32) -> Point2DF32 {
        Point2DF32(self.0 + other.0)
    }
}

impl Sub<Point2DF32> for Point2DF32 {
    type Output = Point2DF32;
    #[inline]
    fn sub(self, other: Point2DF32) -> Point2DF32 {
        Point2DF32(self.0 - other.0)
    }
}

impl Mul<Point2DF32> for Point2DF32 {
    type Output = Point2DF32;
    #[inline]
    fn mul(self, other: Point2DF32) -> Point2DF32 {
        Point2DF32(self.0 * other.0)
    }
}

impl Neg for Point2DF32 {
    type Output = Point2DF32;
    #[inline]
    fn neg(self) -> Point2DF32 {
        Point2DF32(-self.0)
    }
}

/// 2D points with 32-bit signed integer coordinates.
#[derive(Clone, Copy, Debug, Default)]
pub struct Point2DI32(pub I32x4);

impl Point2DI32 {
    #[inline]
    pub fn new(x: i32, y: i32) -> Point2DI32 {
        Point2DI32(I32x4::new(x, y, 0, 0))
    }

    #[inline]
    pub fn splat(value: i32) -> Point2DI32 {
        Point2DI32(I32x4::splat(value))
    }

    #[inline]
    pub fn x(&self) -> i32 {
        self.0[0]
    }

    #[inline]
    pub fn y(&self) -> i32 {
        self.0[1]
    }

    #[inline]
    pub fn set_x(&mut self, x: i32) {
        self.0[0] = x;
    }

    #[inline]
    pub fn set_y(&mut self, y: i32) {
        self.0[1] = y;
    }

    #[inline]
    pub fn scale(&self, factor: i32) -> Point2DI32 {
        Point2DI32(self.0 * I32x4::splat(factor))
    }

    #[inline]
    pub fn to_f32(&self) -> Point2DF32 {
        Point2DF32(self.0.to_f32x4())
    }
}

impl Add<Point2DI32> for Point2DI32 {
    type Output = Point2DI32;
    #[inline]
    fn add(self, other: Point2DI32) -> Point2DI32 {
        Point2DI32(self.0 + other.0)
    }
}

impl AddAssign<Point2DI32> for Point2DI32 {
    #[inline]
    fn add_assign(&mut self, other: Point2DI32) {
        self.0 += other.0
    }
}

impl Sub<Point2DI32> for Point2DI32 {
    type Output = Point2DI32;
    #[inline]
    fn sub(self, other: Point2DI32) -> Point2DI32 {
        Point2DI32(self.0 - other.0)
    }
}

impl PartialEq for Point2DI32 {
    #[inline]
    fn eq(&self, other: &Point2DI32) -> bool {
        let results = self.0.packed_eq(other.0);
        results[0] != 0 && results[1] != 0
    }
}

/// 3D homogeneous points.
#[derive(Clone, Copy, Debug, PartialEq)]
pub struct Point3DF32(pub F32x4);

impl Point3DF32 {
    #[inline]
    pub fn new(x: f32, y: f32, z: f32, w: f32) -> Point3DF32 {
        Point3DF32(F32x4::new(x, y, z, w))
    }

    #[inline]
    pub fn splat(value: f32) -> Point3DF32 {
        Point3DF32(F32x4::splat(value))
    }

    #[inline]
    pub fn to_2d(self) -> Point2DF32 {
        Point2DF32(self.0)
    }

    #[inline]
    pub fn x(self) -> f32 {
        self.0[0]
    }

    #[inline]
    pub fn y(self) -> f32 {
        self.0[1]
    }

    #[inline]
    pub fn z(self) -> f32 {
        self.0[2]
    }

    #[inline]
    pub fn w(self) -> f32 {
        self.0[3]
    }

    #[inline]
    pub fn scale(&self, x: f32) -> Point3DF32 {
        let mut factors = F32x4::splat(x);
        factors[3] = 1.0;
        Point3DF32(self.0 * factors)
    }

    #[inline]
    pub fn set_x(&mut self, x: f32) {
        self.0[0] = x
    }

    #[inline]
    pub fn set_y(&mut self, y: f32) {
        self.0[1] = y
    }

    #[inline]
    pub fn set_z(&mut self, z: f32) {
        self.0[2] = z
    }

    #[inline]
    pub fn set_w(&mut self, w: f32) {
        self.0[3] = w
    }

    #[inline]
    pub fn perspective_divide(self) -> Point3DF32 {
        Point3DF32(self.0 * F32x4::splat(1.0 / self.w()))
    }

    #[inline]
    pub fn approx_eq(&self, other: &Point3DF32, epsilon: f32) -> bool {
        self.0.approx_eq(other.0, epsilon)
    }

    /// Checks to see whether this *homogeneous* coordinate equals zero.
    ///
    /// Note that since this treats the coordinate as a homogeneous coordinate, the `w` is ignored.
    // TODO(pcwalton): Optimize with SIMD.
    #[inline]
    pub fn is_zero(self) -> bool {
        self.x() == 0.0 && self.y() == 0.0 && self.z() == 0.0
    }

    #[inline]
    pub fn lerp(self, other: Point3DF32, t: f32) -> Point3DF32 {
        Point3DF32(self.0 + (other.0 - self.0) * F32x4::splat(t))
    }
}

impl Add<Point3DF32> for Point3DF32 {
    type Output = Point3DF32;
    #[inline]
    fn add(self, other: Point3DF32) -> Point3DF32 {
        Point3DF32(self.0 + other.0)
    }
}

impl AddAssign for Point3DF32 {
    #[inline]
    fn add_assign(&mut self, other: Point3DF32) {
        self.0 += other.0
    }
}

impl Mul<Point3DF32> for Point3DF32 {
    type Output = Point3DF32;
    #[inline]
    fn mul(self, other: Point3DF32) -> Point3DF32 {
        Point3DF32(self.0 * other.0)
    }
}

impl Default for Point3DF32 {
    #[inline]
    fn default() -> Point3DF32 {
        let mut point = F32x4::default();
        point.set_w(1.0);
        Point3DF32(point)
    }
}