use std::mem;
use std::ops::{Add, AddAssign, Mul, Neg};
use crate::algebra::{Force2, Velocity2};
use na::{self, Isometry2, Matrix1, Matrix3, RealField, Vector3};
#[derive(Clone, Copy, Debug)]
pub struct Inertia2<N: RealField + Copy> {
pub linear: N,
pub angular: N,
}
impl<N: RealField + Copy> Inertia2<N> {
pub fn new(linear: N, angular: N) -> Self {
Inertia2 { linear, angular }
}
pub fn new_with_angular_matrix(linear: N, angular: Matrix1<N>) -> Self {
Self::new(linear, angular.x)
}
pub fn mass(&self) -> N {
self.linear
}
pub fn inv_mass(&self) -> N {
if self.linear.is_zero() {
N::zero()
} else {
self.linear
}
}
pub fn zero() -> Self {
Inertia2::new(na::zero(), na::zero())
}
#[inline]
pub fn angular_matrix(&self) -> &Matrix1<N> {
unsafe { mem::transmute(&self.angular) }
}
pub fn to_matrix(&self) -> Matrix3<N> {
let diag = Vector3::new(self.linear, self.linear, self.angular);
Matrix3::from_diagonal(&diag)
}
pub fn transformed(&self, _: &Isometry2<N>) -> Self {
*self
}
pub fn inverse(&self) -> Self {
let inv_mass = if self.linear.is_zero() {
N::zero()
} else {
N::one() / self.linear
};
let inv_angular = if self.angular.is_zero() {
N::zero()
} else {
N::one() / self.angular
};
Inertia2::new(inv_mass, inv_angular)
}
}
impl<N: RealField + Copy> Neg for Inertia2<N> {
type Output = Self;
#[inline]
fn neg(self) -> Self {
Self::new(-self.linear, -self.angular)
}
}
impl<N: RealField + Copy> Add<Inertia2<N>> for Inertia2<N> {
type Output = Inertia2<N>;
#[inline]
fn add(self, rhs: Inertia2<N>) -> Inertia2<N> {
Inertia2::new(self.linear + rhs.linear, self.angular + rhs.angular)
}
}
impl<N: RealField + Copy> AddAssign<Inertia2<N>> for Inertia2<N> {
#[inline]
fn add_assign(&mut self, rhs: Inertia2<N>) {
self.linear += rhs.linear;
self.angular += rhs.angular;
}
}
impl<N: RealField + Copy> Mul<Velocity2<N>> for Inertia2<N> {
type Output = Force2<N>;
#[inline]
fn mul(self, rhs: Velocity2<N>) -> Force2<N> {
Force2::new(rhs.linear * self.linear, self.angular * rhs.angular)
}
}
impl<N: RealField + Copy> Mul<Force2<N>> for Inertia2<N> {
type Output = Velocity2<N>;
#[inline]
fn mul(self, rhs: Force2<N>) -> Velocity2<N> {
Velocity2::new(rhs.linear * self.linear, self.angular * rhs.angular)
}
}