#![allow(clippy::just_underscores_and_digits)]
use super::{Angle, UnknownUnit};
use crate::approxeq::ApproxEq;
use crate::box2d::Box2D;
use crate::num::{One, Zero};
use crate::point::{point2, Point2D};
use crate::rect::Rect;
use crate::transform3d::Transform3D;
use crate::trig::Trig;
use crate::vector::{vec2, Vector2D};
use core::cmp::{Eq, PartialEq};
use core::fmt;
use core::hash::Hash;
use core::marker::PhantomData;
use core::ops::{Add, Div, Mul, Sub};
#[cfg(feature = "bytemuck")]
use bytemuck::{Pod, Zeroable};
#[cfg(feature = "mint")]
use mint;
use num_traits::NumCast;
#[cfg(feature = "serde")]
use serde::{Deserialize, Serialize};
#[repr(C)]
#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]
#[cfg_attr(
feature = "serde",
serde(bound(serialize = "T: Serialize", deserialize = "T: Deserialize<'de>"))
)]
#[rustfmt::skip]
pub struct Transform2D<T, Src, Dst> {
pub m11: T, pub m12: T,
pub m21: T, pub m22: T,
pub m31: T, pub m32: T,
#[doc(hidden)]
pub _unit: PhantomData<(Src, Dst)>,
}
#[cfg(feature = "arbitrary")]
impl<'a, T, Src, Dst> arbitrary::Arbitrary<'a> for Transform2D<T, Src, Dst>
where
T: arbitrary::Arbitrary<'a>,
{
fn arbitrary(u: &mut arbitrary::Unstructured<'a>) -> arbitrary::Result<Self> {
let (m11, m12, m21, m22, m31, m32) = arbitrary::Arbitrary::arbitrary(u)?;
Ok(Transform2D {
m11,
m12,
m21,
m22,
m31,
m32,
_unit: PhantomData,
})
}
}
#[cfg(feature = "bytemuck")]
unsafe impl<T: Zeroable, Src, Dst> Zeroable for Transform2D<T, Src, Dst> {}
#[cfg(feature = "bytemuck")]
unsafe impl<T: Pod, Src: 'static, Dst: 'static> Pod for Transform2D<T, Src, Dst> {}
impl<T: Copy, Src, Dst> Copy for Transform2D<T, Src, Dst> {}
impl<T: Clone, Src, Dst> Clone for Transform2D<T, Src, Dst> {
fn clone(&self) -> Self {
Transform2D {
m11: self.m11.clone(),
m12: self.m12.clone(),
m21: self.m21.clone(),
m22: self.m22.clone(),
m31: self.m31.clone(),
m32: self.m32.clone(),
_unit: PhantomData,
}
}
}
impl<T, Src, Dst> Eq for Transform2D<T, Src, Dst> where T: Eq {}
impl<T, Src, Dst> PartialEq for Transform2D<T, Src, Dst>
where
T: PartialEq,
{
fn eq(&self, other: &Self) -> bool {
self.m11 == other.m11
&& self.m12 == other.m12
&& self.m21 == other.m21
&& self.m22 == other.m22
&& self.m31 == other.m31
&& self.m32 == other.m32
}
}
impl<T, Src, Dst> Hash for Transform2D<T, Src, Dst>
where
T: Hash,
{
fn hash<H: core::hash::Hasher>(&self, h: &mut H) {
self.m11.hash(h);
self.m12.hash(h);
self.m21.hash(h);
self.m22.hash(h);
self.m31.hash(h);
self.m32.hash(h);
}
}
impl<T, Src, Dst> Transform2D<T, Src, Dst> {
#[rustfmt::skip]
pub const fn new(m11: T, m12: T, m21: T, m22: T, m31: T, m32: T) -> Self {
Transform2D {
m11, m12,
m21, m22,
m31, m32,
_unit: PhantomData,
}
}
#[inline]
pub fn approx_eq(&self, other: &Self) -> bool
where
T: ApproxEq<T>,
{
<Self as ApproxEq<T>>::approx_eq(&self, &other)
}
#[inline]
pub fn approx_eq_eps(&self, other: &Self, eps: &T) -> bool
where
T: ApproxEq<T>,
{
<Self as ApproxEq<T>>::approx_eq_eps(&self, &other, &eps)
}
}
impl<T: Copy, Src, Dst> Transform2D<T, Src, Dst> {
#[inline]
#[rustfmt::skip]
pub fn to_array(&self) -> [T; 6] {
[
self.m11, self.m12,
self.m21, self.m22,
self.m31, self.m32
]
}
#[inline]
#[rustfmt::skip]
pub fn to_array_transposed(&self) -> [T; 6] {
[
self.m11, self.m21, self.m31,
self.m12, self.m22, self.m32
]
}
#[inline]
pub fn to_arrays(&self) -> [[T; 2]; 3] {
[
[self.m11, self.m12],
[self.m21, self.m22],
[self.m31, self.m32],
]
}
#[inline]
#[rustfmt::skip]
pub fn from_array(array: [T; 6]) -> Self {
Self::new(
array[0], array[1],
array[2], array[3],
array[4], array[5],
)
}
#[inline]
#[rustfmt::skip]
pub fn from_arrays(array: [[T; 2]; 3]) -> Self {
Self::new(
array[0][0], array[0][1],
array[1][0], array[1][1],
array[2][0], array[2][1],
)
}
#[inline]
#[rustfmt::skip]
pub fn to_untyped(&self) -> Transform2D<T, UnknownUnit, UnknownUnit> {
Transform2D::new(
self.m11, self.m12,
self.m21, self.m22,
self.m31, self.m32
)
}
#[inline]
#[rustfmt::skip]
pub fn from_untyped(p: &Transform2D<T, UnknownUnit, UnknownUnit>) -> Self {
Transform2D::new(
p.m11, p.m12,
p.m21, p.m22,
p.m31, p.m32
)
}
#[inline]
#[rustfmt::skip]
pub fn with_source<NewSrc>(&self) -> Transform2D<T, NewSrc, Dst> {
Transform2D::new(
self.m11, self.m12,
self.m21, self.m22,
self.m31, self.m32,
)
}
#[inline]
#[rustfmt::skip]
pub fn with_destination<NewDst>(&self) -> Transform2D<T, Src, NewDst> {
Transform2D::new(
self.m11, self.m12,
self.m21, self.m22,
self.m31, self.m32,
)
}
pub fn to_3d(&self) -> Transform3D<T, Src, Dst>
where
T: Zero + One,
{
Transform3D::new_2d(self.m11, self.m12, self.m21, self.m22, self.m31, self.m32)
}
}
impl<T: NumCast + Copy, Src, Dst> Transform2D<T, Src, Dst> {
#[inline]
pub fn cast<NewT: NumCast>(&self) -> Transform2D<NewT, Src, Dst> {
self.try_cast().unwrap()
}
#[rustfmt::skip]
pub fn try_cast<NewT: NumCast>(&self) -> Option<Transform2D<NewT, Src, Dst>> {
match (NumCast::from(self.m11), NumCast::from(self.m12),
NumCast::from(self.m21), NumCast::from(self.m22),
NumCast::from(self.m31), NumCast::from(self.m32)) {
(Some(m11), Some(m12),
Some(m21), Some(m22),
Some(m31), Some(m32)) => {
Some(Transform2D::new(
m11, m12,
m21, m22,
m31, m32
))
},
_ => None
}
}
}
impl<T, Src, Dst> Transform2D<T, Src, Dst>
where
T: Zero + One,
{
#[inline]
pub fn identity() -> Self {
Self::translation(T::zero(), T::zero())
}
fn is_identity(&self) -> bool
where
T: PartialEq,
{
*self == Self::identity()
}
}
impl<T, Src, Dst> Transform2D<T, Src, Dst>
where
T: Copy + Add<Output = T> + Mul<Output = T>,
{
#[must_use]
#[rustfmt::skip]
pub fn then<NewDst>(&self, mat: &Transform2D<T, Dst, NewDst>) -> Transform2D<T, Src, NewDst> {
Transform2D::new(
self.m11 * mat.m11 + self.m12 * mat.m21,
self.m11 * mat.m12 + self.m12 * mat.m22,
self.m21 * mat.m11 + self.m22 * mat.m21,
self.m21 * mat.m12 + self.m22 * mat.m22,
self.m31 * mat.m11 + self.m32 * mat.m21 + mat.m31,
self.m31 * mat.m12 + self.m32 * mat.m22 + mat.m32,
)
}
}
impl<T, Src, Dst> Transform2D<T, Src, Dst>
where
T: Zero + One,
{
#[inline]
#[rustfmt::skip]
pub fn translation(x: T, y: T) -> Self {
let _0 = || T::zero();
let _1 = || T::one();
Self::new(
_1(), _0(),
_0(), _1(),
x, y,
)
}
#[inline]
#[must_use]
pub fn then_translate(&self, v: Vector2D<T, Dst>) -> Self
where
T: Copy + Add<Output = T> + Mul<Output = T>,
{
self.then(&Transform2D::translation(v.x, v.y))
}
#[inline]
#[must_use]
pub fn pre_translate(&self, v: Vector2D<T, Src>) -> Self
where
T: Copy + Add<Output = T> + Mul<Output = T>,
{
Transform2D::translation(v.x, v.y).then(self)
}
}
impl<T, Src, Dst> Transform2D<T, Src, Dst>
where
T: Copy + Add<Output = T> + Sub<Output = T> + Mul<Output = T> + Zero + Trig,
{
#[inline]
#[rustfmt::skip]
pub fn rotation(theta: Angle<T>) -> Self {
let _0 = Zero::zero();
let cos = theta.get().cos();
let sin = theta.get().sin();
Transform2D::new(
cos, sin,
_0 - sin, cos,
_0, _0
)
}
#[inline]
#[must_use]
pub fn then_rotate(&self, theta: Angle<T>) -> Self {
self.then(&Transform2D::rotation(theta))
}
#[inline]
#[must_use]
pub fn pre_rotate(&self, theta: Angle<T>) -> Self {
Transform2D::rotation(theta).then(self)
}
}
impl<T, Src, Dst> Transform2D<T, Src, Dst> {
#[inline]
#[rustfmt::skip]
pub fn scale(x: T, y: T) -> Self
where
T: Zero,
{
let _0 = || Zero::zero();
Self::new(
x, _0(),
_0(), y,
_0(), _0(),
)
}
#[inline]
#[must_use]
pub fn then_scale(&self, x: T, y: T) -> Self
where
T: Copy + Add<Output = T> + Mul<Output = T> + Zero,
{
self.then(&Transform2D::scale(x, y))
}
#[inline]
#[must_use]
#[rustfmt::skip]
pub fn pre_scale(&self, x: T, y: T) -> Self
where
T: Copy + Mul<Output = T>,
{
Transform2D::new(
self.m11 * x, self.m12 * x,
self.m21 * y, self.m22 * y,
self.m31, self.m32
)
}
}
impl<T, Src, Dst> Transform2D<T, Src, Dst>
where
T: Copy + Add<Output = T> + Mul<Output = T>,
{
#[inline]
#[must_use]
pub fn transform_point(&self, point: Point2D<T, Src>) -> Point2D<T, Dst> {
Point2D::new(
point.x * self.m11 + point.y * self.m21 + self.m31,
point.x * self.m12 + point.y * self.m22 + self.m32,
)
}
#[inline]
#[must_use]
pub fn transform_vector(&self, vec: Vector2D<T, Src>) -> Vector2D<T, Dst> {
vec2(
vec.x * self.m11 + vec.y * self.m21,
vec.x * self.m12 + vec.y * self.m22,
)
}
#[inline]
#[must_use]
pub fn outer_transformed_rect(&self, rect: &Rect<T, Src>) -> Rect<T, Dst>
where
T: Sub<Output = T> + Zero + PartialOrd,
{
let min = rect.min();
let max = rect.max();
Rect::from_points(&[
self.transform_point(min),
self.transform_point(max),
self.transform_point(point2(max.x, min.y)),
self.transform_point(point2(min.x, max.y)),
])
}
#[inline]
#[must_use]
pub fn outer_transformed_box(&self, b: &Box2D<T, Src>) -> Box2D<T, Dst>
where
T: Sub<Output = T> + Zero + PartialOrd,
{
Box2D::from_points(&[
self.transform_point(b.min),
self.transform_point(b.max),
self.transform_point(point2(b.max.x, b.min.y)),
self.transform_point(point2(b.min.x, b.max.y)),
])
}
}
impl<T, Src, Dst> Transform2D<T, Src, Dst>
where
T: Copy + Sub<Output = T> + Mul<Output = T> + Div<Output = T> + PartialEq + Zero + One,
{
pub fn determinant(&self) -> T {
self.m11 * self.m22 - self.m12 * self.m21
}
#[inline]
pub fn is_invertible(&self) -> bool {
self.determinant() != Zero::zero()
}
#[must_use]
pub fn inverse(&self) -> Option<Transform2D<T, Dst, Src>> {
let det = self.determinant();
let _0: T = Zero::zero();
let _1: T = One::one();
if det == _0 {
return None;
}
let inv_det = _1 / det;
Some(Transform2D::new(
inv_det * self.m22,
inv_det * (_0 - self.m12),
inv_det * (_0 - self.m21),
inv_det * self.m11,
inv_det * (self.m21 * self.m32 - self.m22 * self.m31),
inv_det * (self.m31 * self.m12 - self.m11 * self.m32),
))
}
}
impl<T, Src, Dst> Default for Transform2D<T, Src, Dst>
where
T: Zero + One,
{
fn default() -> Self {
Self::identity()
}
}
impl<T: ApproxEq<T>, Src, Dst> ApproxEq<T> for Transform2D<T, Src, Dst> {
#[inline]
fn approx_epsilon() -> T {
T::approx_epsilon()
}
fn approx_eq_eps(&self, other: &Self, eps: &T) -> bool {
self.m11.approx_eq_eps(&other.m11, eps)
&& self.m12.approx_eq_eps(&other.m12, eps)
&& self.m21.approx_eq_eps(&other.m21, eps)
&& self.m22.approx_eq_eps(&other.m22, eps)
&& self.m31.approx_eq_eps(&other.m31, eps)
&& self.m32.approx_eq_eps(&other.m32, eps)
}
}
impl<T, Src, Dst> fmt::Debug for Transform2D<T, Src, Dst>
where
T: Copy + fmt::Debug + PartialEq + One + Zero,
{
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
if self.is_identity() {
write!(f, "[I]")
} else {
self.to_array().fmt(f)
}
}
}
#[cfg(feature = "mint")]
impl<T, Src, Dst> From<mint::RowMatrix3x2<T>> for Transform2D<T, Src, Dst> {
#[rustfmt::skip]
fn from(m: mint::RowMatrix3x2<T>) -> Self {
Transform2D {
m11: m.x.x, m12: m.x.y,
m21: m.y.x, m22: m.y.y,
m31: m.z.x, m32: m.z.y,
_unit: PhantomData,
}
}
}
#[cfg(feature = "mint")]
impl<T, Src, Dst> From<Transform2D<T, Src, Dst>> for mint::RowMatrix3x2<T> {
fn from(t: Transform2D<T, Src, Dst>) -> Self {
mint::RowMatrix3x2 {
x: mint::Vector2 { x: t.m11, y: t.m12 },
y: mint::Vector2 { x: t.m21, y: t.m22 },
z: mint::Vector2 { x: t.m31, y: t.m32 },
}
}
}
#[cfg(test)]
mod test {
use super::*;
use crate::approxeq::ApproxEq;
use crate::default;
#[cfg(feature = "mint")]
use mint;
use core::f32::consts::FRAC_PI_2;
type Mat = default::Transform2D<f32>;
fn rad(v: f32) -> Angle<f32> {
Angle::radians(v)
}
#[test]
pub fn test_translation() {
let t1 = Mat::translation(1.0, 2.0);
let t2 = Mat::identity().pre_translate(vec2(1.0, 2.0));
let t3 = Mat::identity().then_translate(vec2(1.0, 2.0));
assert_eq!(t1, t2);
assert_eq!(t1, t3);
assert_eq!(
t1.transform_point(Point2D::new(1.0, 1.0)),
Point2D::new(2.0, 3.0)
);
assert_eq!(t1.then(&t1), Mat::translation(2.0, 4.0));
}
#[test]
pub fn test_rotation() {
let r1 = Mat::rotation(rad(FRAC_PI_2));
let r2 = Mat::identity().pre_rotate(rad(FRAC_PI_2));
let r3 = Mat::identity().then_rotate(rad(FRAC_PI_2));
assert_eq!(r1, r2);
assert_eq!(r1, r3);
assert!(r1
.transform_point(Point2D::new(1.0, 2.0))
.approx_eq(&Point2D::new(-2.0, 1.0)));
assert!(r1.then(&r1).approx_eq(&Mat::rotation(rad(FRAC_PI_2 * 2.0))));
}
#[test]
pub fn test_scale() {
let s1 = Mat::scale(2.0, 3.0);
let s2 = Mat::identity().pre_scale(2.0, 3.0);
let s3 = Mat::identity().then_scale(2.0, 3.0);
assert_eq!(s1, s2);
assert_eq!(s1, s3);
assert!(s1
.transform_point(Point2D::new(2.0, 2.0))
.approx_eq(&Point2D::new(4.0, 6.0)));
}
#[test]
pub fn test_pre_then_scale() {
let m = Mat::rotation(rad(FRAC_PI_2)).then_translate(vec2(6.0, 7.0));
let s = Mat::scale(2.0, 3.0);
assert_eq!(m.then(&s), m.then_scale(2.0, 3.0));
}
#[test]
pub fn test_inverse_simple() {
let m1 = Mat::identity();
let m2 = m1.inverse().unwrap();
assert!(m1.approx_eq(&m2));
}
#[test]
pub fn test_inverse_scale() {
let m1 = Mat::scale(1.5, 0.3);
let m2 = m1.inverse().unwrap();
assert!(m1.then(&m2).approx_eq(&Mat::identity()));
assert!(m2.then(&m1).approx_eq(&Mat::identity()));
}
#[test]
pub fn test_inverse_translate() {
let m1 = Mat::translation(-132.0, 0.3);
let m2 = m1.inverse().unwrap();
assert!(m1.then(&m2).approx_eq(&Mat::identity()));
assert!(m2.then(&m1).approx_eq(&Mat::identity()));
}
#[test]
fn test_inverse_none() {
assert!(Mat::scale(2.0, 0.0).inverse().is_none());
assert!(Mat::scale(2.0, 2.0).inverse().is_some());
}
#[test]
pub fn test_pre_post() {
let m1 = default::Transform2D::identity()
.then_scale(1.0, 2.0)
.then_translate(vec2(1.0, 2.0));
let m2 = default::Transform2D::identity()
.pre_translate(vec2(1.0, 2.0))
.pre_scale(1.0, 2.0);
assert!(m1.approx_eq(&m2));
let r = Mat::rotation(rad(FRAC_PI_2));
let t = Mat::translation(2.0, 3.0);
let a = Point2D::new(1.0, 1.0);
assert!(r
.then(&t)
.transform_point(a)
.approx_eq(&Point2D::new(1.0, 4.0)));
assert!(t
.then(&r)
.transform_point(a)
.approx_eq(&Point2D::new(-4.0, 3.0)));
assert!(t
.then(&r)
.transform_point(a)
.approx_eq(&r.transform_point(t.transform_point(a))));
}
#[test]
fn test_size_of() {
use core::mem::size_of;
assert_eq!(size_of::<default::Transform2D<f32>>(), 6 * size_of::<f32>());
assert_eq!(size_of::<default::Transform2D<f64>>(), 6 * size_of::<f64>());
}
#[test]
pub fn test_is_identity() {
let m1 = default::Transform2D::identity();
assert!(m1.is_identity());
let m2 = m1.then_translate(vec2(0.1, 0.0));
assert!(!m2.is_identity());
}
#[test]
pub fn test_transform_vector() {
let m1 = Mat::translation(1.0, 1.0);
let v1 = vec2(10.0, -10.0);
assert_eq!(v1, m1.transform_vector(v1));
}
#[cfg(feature = "mint")]
#[test]
pub fn test_mint() {
let m1 = Mat::rotation(rad(FRAC_PI_2));
let mm: mint::RowMatrix3x2<_> = m1.into();
let m2 = Mat::from(mm);
assert_eq!(m1, m2);
}
}