#[cfg(feature = "serde")]
use serde::{Deserialize, Serialize};
use {
num_traits::{AsPrimitive, CheckedDiv},
std::ops::{Add, AddAssign, Div, DivAssign, Mul, MulAssign, Neg, Sub, SubAssign},
};
#[repr(C)]
#[derive(Clone, PartialEq, Eq, Debug, Copy, Default)]
#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]
pub struct Vector2<T> {
pub x: T,
pub y: T,
}
pub type Vector2i = Vector2<i32>;
pub type Vector2u = Vector2<u32>;
pub type Vector2f = Vector2<f32>;
impl<T> Vector2<T> {
pub const fn new(x: T, y: T) -> Self {
Self { x, y }
}
pub fn into_other<U>(self) -> Vector2<U>
where
T: Into<U>,
{
Vector2 {
x: self.x.into(),
y: self.y.into(),
}
}
pub fn try_into_other<U>(self) -> Result<Vector2<U>, T::Error>
where
T: TryInto<U>,
{
Ok(Vector2 {
x: self.x.try_into()?,
y: self.y.try_into()?,
})
}
pub fn as_other<U: 'static + Copy>(self) -> Vector2<U>
where
T: AsPrimitive<U>,
{
Vector2 {
x: self.x.as_(),
y: self.y.as_(),
}
}
}
impl<T: Mul<Output = T> + Add<Output = T> + Copy> Vector2<T> {
pub fn dot(self, rhs: Self) -> T {
self.x * rhs.x + self.y * rhs.y
}
pub fn length_sq(self) -> T {
self.dot(self)
}
}
impl<T: Mul<Output = T> + Sub<Output = T> + Copy> Vector2<T> {
pub fn cross(self, rhs: Self) -> T {
self.x * rhs.y - self.y * rhs.x
}
}
impl<T: Mul<Output = T>> Vector2<T> {
pub fn cwise_mul(self, rhs: Self) -> Vector2<T> {
Self {
x: self.x * rhs.x,
y: self.y * rhs.y,
}
}
}
impl<T: Div<Output = T>> Vector2<T> {
pub fn cwise_div(self, rhs: Self) -> Vector2<T> {
Vector2 {
x: self.x / rhs.x,
y: self.y / rhs.y,
}
}
}
impl<T: Div<Output = T> + CheckedDiv> Vector2<T> {
pub fn cwise_checked_div(self, rhs: Self) -> Option<Vector2<T>> {
let x = self.x.checked_div(&rhs.x)?;
let y = self.y.checked_div(&rhs.y)?;
Some(Vector2 { x, y })
}
}
impl<T: Neg<Output = T>> Vector2<T> {
pub fn perpendicular(self) -> Vector2<T> {
Vector2::new(-self.y, self.x)
}
}
impl<T> From<(T, T)> for Vector2<T> {
fn from(src: (T, T)) -> Self {
Self { x: src.0, y: src.1 }
}
}
impl<T> From<Vector2<T>> for (T, T) {
fn from(vec: Vector2<T>) -> Self {
(vec.x, vec.y)
}
}
impl<T> From<[T; 2]> for Vector2<T> {
fn from([x, y]: [T; 2]) -> Self {
Self { x, y }
}
}
impl<T> From<Vector2<T>> for [T; 2] {
fn from(vec: Vector2<T>) -> Self {
[vec.x, vec.y]
}
}
impl<T: Clone> From<T> for Vector2<T> {
fn from(src: T) -> Self {
Self {
x: src.clone(),
y: src,
}
}
}
impl<T: Add> Add<Vector2<T>> for Vector2<T> {
type Output = Vector2<T::Output>;
fn add(self, rhs: Vector2<T>) -> Vector2<T::Output> {
Vector2 {
x: self.x + rhs.x,
y: self.y + rhs.y,
}
}
}
impl<T: AddAssign> AddAssign for Vector2<T> {
fn add_assign(&mut self, rhs: Self) {
self.x += rhs.x;
self.y += rhs.y;
}
}
impl<T: Sub> Sub<Vector2<T>> for Vector2<T> {
type Output = Vector2<T::Output>;
fn sub(self, rhs: Vector2<T>) -> Vector2<T::Output> {
Vector2 {
x: self.x - rhs.x,
y: self.y - rhs.y,
}
}
}
impl<T: SubAssign> SubAssign for Vector2<T> {
fn sub_assign(&mut self, rhs: Self) {
self.x -= rhs.x;
self.y -= rhs.y;
}
}
impl<T: Mul + Copy> Mul<T> for Vector2<T> {
type Output = Vector2<T::Output>;
fn mul(self, rhs: T) -> Vector2<T::Output> {
Vector2 {
x: self.x * rhs,
y: self.y * rhs,
}
}
}
impl<T: MulAssign + Copy> MulAssign<T> for Vector2<T> {
fn mul_assign(&mut self, rhs: T) {
self.x *= rhs;
self.y *= rhs;
}
}
impl<T: Div + Copy> Div<T> for Vector2<T> {
type Output = Vector2<T::Output>;
fn div(self, rhs: T) -> Vector2<T::Output> {
Vector2 {
x: self.x / rhs,
y: self.y / rhs,
}
}
}
impl<T: DivAssign + Copy> DivAssign<T> for Vector2<T> {
fn div_assign(&mut self, rhs: T) {
self.x /= rhs;
self.y /= rhs;
}
}
impl<T: CheckedDiv> Vector2<T> {
pub fn checked_div(self, rhs: T) -> Option<Vector2<T>> {
let x = self.x.checked_div(&rhs)?;
let y = self.y.checked_div(&rhs)?;
Some(Vector2 { x, y })
}
}
impl<T: Neg<Output = T>> Neg for Vector2<T> {
type Output = Self;
fn neg(self) -> Self {
Vector2 {
x: -self.x,
y: -self.y,
}
}
}