use num::Float;
use std::ops::{Index, IndexMut, AddAssign, SubAssign, MulAssign, DivAssign, Add, Neg, Sub, Mul, Div};
use std::fmt::{Debug, Formatter, Result, Display};
pub struct Vector2<T: Float> {
pub x: T,
pub y: T,
}
pub type Vector2F = Vector2<f32>;
pub type Vector2D = Vector2<f64>;
impl<T: Float> Vector2<T> {
pub fn new_default() -> Vector2<T> {
return Vector2 {
x: T::zero(),
y: T::zero(),
};
}
pub fn new(x_: T, y_: T) -> Vector2<T> {
return Vector2 {
x: x_,
y: y_,
};
}
pub fn new_lst(lst: [T; 2]) -> Vector2<T> {
return Vector2 {
x: lst[0],
y: lst[1],
};
}
}
impl<T: Float> Vector2<T> {
pub fn set_scalar(&mut self, s: T) {
self.x = s;
self.y = s;
}
pub fn set_scalar2(&mut self, x: T, y: T) {
self.x = x;
self.y = y;
}
pub fn set_lst(&mut self, lst: [T; 2]) {
self.x = lst[0];
self.y = lst[1];
}
pub fn set_self(&mut self, pt: Vector2<T>) {
self.x = pt.x;
self.y = pt.y;
}
pub fn set_zero(&mut self) {
self.x = T::zero();
self.y = T::zero();
}
pub fn normalize(&mut self) {
let l = self.length();
self.x = T::div(self.x, l);
self.y = T::div(self.y, l);
}
}
impl<T: Float> Vector2<T> {
pub fn add_scalar(&self, v: T) -> Vector2<T> {
return Vector2::new(self.x + v, self.y + v);
}
pub fn add_vec(&self, v: Vector2<T>) -> Vector2<T> {
return Vector2::new(self.x + v.x, self.y + v.y);
}
pub fn sub_scalar(&self, v: T) -> Vector2<T> {
return Vector2::new(self.x - v, self.y - v);
}
pub fn sub_vec(&self, v: Vector2<T>) -> Vector2<T> {
return Vector2::new(self.x - v.x, self.y - v.y);
}
pub fn mul_scalar(&self, v: T) -> Vector2<T> {
return Vector2::new(self.x * v, self.y * v);
}
pub fn mul_vec(&self, v: Vector2<T>) -> Vector2<T> {
return Vector2::new(self.x * v.x, self.y * v.y);
}
pub fn div_scalar(&self, v: T) -> Vector2<T> {
return Vector2::new(self.x / v, self.y / v);
}
pub fn div_vec(&self, v: Vector2<T>) -> Vector2<T> {
return Vector2::new(self.x / v.x, self.y / v.y);
}
pub fn dot(&self, v: &Vector2<T>) -> T {
return self.x * v.x + self.y * v.y;
}
pub fn cross(&self, v: &Vector2<T>) -> T {
return self.x * v.y - v.x * self.y;
}
}
impl<T: Float> Vector2<T> {
pub fn rsub_scalar(&self, v: T) -> Vector2<T> {
return Vector2::new(v - self.x, v - self.y);
}
pub fn rsub_vec(&self, v: Vector2<T>) -> Vector2<T> {
return Vector2::new(v.x - self.x, v.y - self.y);
}
pub fn rdiv_scalar(&self, v: T) -> Vector2<T> {
return Vector2::new(v / self.x, v / self.y);
}
pub fn rdiv_vec(&self, v: Vector2<T>) -> Vector2<T> {
return Vector2::new(v.x / self.x, v.y / self.y);
}
pub fn rcross(&self, v: Vector2<T>) -> T {
return v.x * self.y - self.x * v.y;
}
}
impl<T: Float> Vector2<T> {
pub fn iadd_scalar(&mut self, v: T) {
self.x = T::add(self.x, v);
self.y = T::add(self.y, v);
}
pub fn iadd_vec(&mut self, v: Vector2<T>) {
self.x = T::add(self.x, v.x);
self.y = T::add(self.y, v.y);
}
pub fn isub_scalar(&mut self, v: T) {
self.x = T::sub(self.x, v);
self.y = T::sub(self.y, v);
}
pub fn isub_vec(&mut self, v: Vector2<T>) {
self.x = T::sub(self.x, v.x);
self.y = T::sub(self.y, v.y);
}
pub fn imul_scalar(&mut self, v: T) {
self.x = T::mul(self.x, v);
self.y = T::mul(self.y, v);
}
pub fn imul_vec(&mut self, v: Vector2<T>) {
self.x = T::mul(self.x, v.x);
self.y = T::mul(self.y, v.y);
}
pub fn idiv_scalar(&mut self, v: T) {
self.x = T::div(self.x, v);
self.y = T::div(self.y, v);
}
pub fn idiv_vec(&mut self, v: Vector2<T>) {
self.x = T::div(self.x, v.x);
self.y = T::div(self.y, v.y);
}
}
impl<T: Float> Vector2<T> {
pub fn at(&self, i: usize) -> &T {
match i {
0 => return &self.x,
1 => return &self.y,
_ => { panic!() }
}
}
pub fn at_mut(&mut self, i: usize) -> &mut T {
match i {
0 => return &mut self.x,
1 => return &mut self.y,
_ => { panic!() }
}
}
pub fn sum(&self) -> T {
return self.x + self.y;
}
pub fn avg(&self) -> T {
return (self.x + self.y) / T::from(2.0).unwrap();
}
pub fn min(&self) -> T {
return self.x.min(self.y);
}
pub fn max(&self) -> T {
return self.x.max(self.y);
}
pub fn absmin(&self) -> T {
return crate::math_utils::absmin(self.x, self.y);
}
pub fn absmax(&self) -> T {
return crate::math_utils::absmax(self.x, self.y);
}
pub fn dominant_axis(&self) -> usize {
match self.x.abs() > self.y.abs() {
true => 0,
false => 1
}
}
pub fn subminant_axis(&self) -> usize {
match self.x.abs() < self.y.abs() {
true => 0,
false => 1
}
}
pub fn normalized(&self) -> Vector2<T> {
let l = self.length();
return Vector2::new(self.x / l, self.y / l);
}
pub fn length(&self) -> T {
return T::sqrt(self.x * self.x + self.y * self.y);
}
pub fn length_squared(&self) -> T {
return self.x * self.x + self.y * self.y;
}
pub fn distance_to(&self, other: Vector2<T>) -> T {
return self.sub_vec(other).length();
}
pub fn distance_squared_to(&self, other: Vector2<T>) -> T {
return self.sub_vec(other).length_squared();
}
pub fn reflected(&self, normal: &Vector2<T>) -> Vector2<T> {
return self.sub_vec(normal.mul_scalar(T::mul(T::from(2.0).unwrap(), self.dot(normal))));
}
pub fn projected(&self, normal: &Vector2<T>) -> Vector2<T> {
return self.sub_vec(normal.mul_scalar(self.dot(normal)));
}
pub fn tangential(&self) -> Vector2<T> {
return Vector2::new(-self.y, self.x);
}
pub fn is_equal(&self, other: &Vector2<T>) -> bool {
return self.x == other.x && self.y == other.y;
}
pub fn is_similar(&self, other: &Vector2<T>, epsilon: Option<T>) -> bool {
return ((self.x - other.x).abs() < epsilon.unwrap_or(T::epsilon())) &&
((self.y - other.y).abs() < epsilon.unwrap_or(T::epsilon()));
}
}
impl<T: Float> Clone for Vector2<T> {
fn clone(&self) -> Self {
return Vector2 {
x: self.x,
y: self.y,
};
}
}
impl<T: Float> Copy for Vector2<T> {}
impl<T: Float> Index<usize> for Vector2<T> {
type Output = T;
fn index(&self, index: usize) -> &Self::Output {
return self.at(index);
}
}
impl<T: Float> IndexMut<usize> for Vector2<T> {
fn index_mut(&mut self, index: usize) -> &mut Self::Output {
return self.at_mut(index);
}
}
impl<T: Float> AddAssign<T> for Vector2<T> {
fn add_assign(&mut self, rhs: T) {
self.iadd_scalar(rhs);
}
}
impl<T: Float> AddAssign for Vector2<T> {
fn add_assign(&mut self, rhs: Self) {
self.iadd_vec(rhs);
}
}
impl<T: Float> SubAssign<T> for Vector2<T> {
fn sub_assign(&mut self, rhs: T) {
self.isub_scalar(rhs);
}
}
impl<T: Float> SubAssign for Vector2<T> {
fn sub_assign(&mut self, rhs: Self) {
self.isub_vec(rhs);
}
}
impl<T: Float> MulAssign<T> for Vector2<T> {
fn mul_assign(&mut self, rhs: T) {
self.imul_scalar(rhs);
}
}
impl<T: Float> MulAssign for Vector2<T> {
fn mul_assign(&mut self, rhs: Self) {
self.imul_vec(rhs);
}
}
impl<T: Float> DivAssign<T> for Vector2<T> {
fn div_assign(&mut self, rhs: T) {
self.idiv_scalar(rhs);
}
}
impl<T: Float> DivAssign for Vector2<T> {
fn div_assign(&mut self, rhs: Self) {
self.idiv_vec(rhs);
}
}
impl<T: Float> PartialEq for Vector2<T> {
fn eq(&self, other: &Self) -> bool {
return self.is_equal(other);
}
}
impl<T: Float> Eq for Vector2<T> {}
impl<T: Float> Neg for Vector2<T> {
type Output = Vector2<T>;
fn neg(self) -> Self::Output {
return Vector2::new(-self.x, -self.y);
}
}
impl<T: Float> Add<T> for Vector2<T> {
type Output = Vector2<T>;
fn add(self, rhs: T) -> Self::Output {
return self.add_scalar(rhs);
}
}
impl<T: Float> Add for Vector2<T> {
type Output = Vector2<T>;
fn add(self, rhs: Self) -> Self::Output {
return self.add_vec(rhs);
}
}
impl<T: Float> Sub<T> for Vector2<T> {
type Output = Vector2<T>;
fn sub(self, rhs: T) -> Self::Output {
return self.sub_scalar(rhs);
}
}
impl<T: Float> Sub for Vector2<T> {
type Output = Vector2<T>;
fn sub(self, rhs: Self) -> Self::Output {
return self.sub_vec(rhs);
}
}
impl<T: Float> Mul<T> for Vector2<T> {
type Output = Vector2<T>;
fn mul(self, rhs: T) -> Self::Output {
return self.mul_scalar(rhs);
}
}
impl<T: Float> Mul for Vector2<T> {
type Output = Vector2<T>;
fn mul(self, rhs: Self) -> Self::Output {
return self.mul_vec(rhs);
}
}
impl<T: Float> Div<T> for Vector2<T> {
type Output = Vector2<T>;
fn div(self, rhs: T) -> Self::Output {
return self.div_scalar(rhs);
}
}
impl<T: Float> Div for Vector2<T> {
type Output = Vector2<T>;
fn div(self, rhs: Self) -> Self::Output {
return self.div_vec(rhs);
}
}
impl<T: Float + Debug> Debug for Vector2<T> {
fn fmt(&self, f: &mut Formatter<'_>) -> Result {
f.debug_tuple("")
.field(&self.x)
.field(&self.y)
.finish()
}
}
impl<T: Float + Display> Display for Vector2<T> {
fn fmt(&self, f: &mut Formatter<'_>) -> Result {
write!(f, "{} {}", self.x, self.y)
}
}
pub fn min<T: Float>(a: &Vector2<T>, b: &Vector2<T>) -> Vector2<T> {
return Vector2::new(T::min(a.x, b.x), T::min(a.y, b.y));
}
pub fn max<T: Float>(a: &Vector2<T>, b: &Vector2<T>) -> Vector2<T> {
return Vector2::new(T::max(a.x, b.x), T::max(a.y, b.y));
}
pub fn clamp<T: Float>(v: &Vector2<T>, low: &Vector2<T>, high: &Vector2<T>) -> Vector2<T> {
return Vector2::new(crate::math_utils::clamp(v.x, low.x, high.x),
crate::math_utils::clamp(v.y, low.y, high.y));
}
pub fn ceil<T: Float>(a: &Vector2<T>) -> Vector2<T> {
return Vector2::new((a.x).ceil(), (a.y).ceil());
}
pub fn floor<T: Float>(a: &Vector2<T>) -> Vector2<T> {
return Vector2::new((a.x).floor(), (a.y).floor());
}
pub fn monotonic_catmull_rom<T: Float>(v0: Vector2<T>, v1: Vector2<T>,
v2: Vector2<T>, v3: Vector2<T>, f: T) -> Vector2<T> {
let two = T::from(2.0).unwrap();
let three = T::from(3.0).unwrap();
let mut d1 = (v2 - v0) / two;
let mut d2 = (v3 - v1) / two;
let d_diff = v2 - v1;
if d_diff.x.abs() < T::epsilon() ||
crate::math_utils::sign(d_diff.x) != crate::math_utils::sign(d1.x) ||
crate::math_utils::sign(d_diff.x) != crate::math_utils::sign(d2.x) {
d1.x = T::zero();
d2.x = T::zero();
}
if d_diff.y.abs() < T::epsilon() ||
crate::math_utils::sign(d_diff.y) != crate::math_utils::sign(d1.y) ||
crate::math_utils::sign(d_diff.y) != crate::math_utils::sign(d2.y) {
d1.y = T::zero();
d2.y = T::zero();
}
let a3 = d1 + d2 - d_diff * two;
let a2 = d_diff * three - d1 * two - d2;
let a1 = d1;
let a0 = v1;
return a3 * crate::math_utils::cubic(f) + a2 * crate::math_utils::square(f) + a1 * f + a0;
}