use num::{One, Zero};
use point::TypedPoint2D;
use rect::TypedRect;
use size::TypedSize2D;
use length::UnknownUnit;
use std::ops::{Add, Mul, Sub};
use std::marker::PhantomData;
define_matrix! {
#[derive(Clone, Copy, Eq, Hash, PartialEq)]
pub struct TypedMatrix2D<T, Src, Dst> {
pub m11: T, pub m12: T,
pub m21: T, pub m22: T,
pub m31: T, pub m32: T,
}
}
pub type Matrix2D<T> = TypedMatrix2D<T, UnknownUnit, UnknownUnit>;
impl<T, Src, Dst> TypedMatrix2D<T, Src, Dst> {
pub fn new(m11: T, m12: T, m21: T, m22: T, m31: T, m32: T) -> TypedMatrix2D<T, Src, Dst> {
TypedMatrix2D {
m11: m11, m12: m12,
m21: m21, m22: m22,
m31: m31, m32: m32,
_unit: PhantomData,
}
}
}
impl<T: Copy, Src, Dst> TypedMatrix2D<T, Src, Dst> {
pub fn to_array(&self) -> [T; 6] {
[
self.m11, self.m12,
self.m21, self.m22,
self.m31, self.m32
]
}
}
impl<T, Src, Dst> TypedMatrix2D<T, Src, Dst>
where T: Copy + Clone +
Add<T, Output=T> +
Mul<T, Output=T> +
Sub<T, Output=T> +
PartialOrd +
One + Zero {
pub fn mul<NewSrc>(&self, m: &TypedMatrix2D<T, NewSrc, Src>) -> TypedMatrix2D<T, NewSrc, Dst> {
TypedMatrix2D::new(
m.m11*self.m11 + m.m12*self.m21,
m.m11*self.m12 + m.m12*self.m22,
m.m21*self.m11 + m.m22*self.m21,
m.m21*self.m12 + m.m22*self.m22,
m.m31*self.m11 + m.m32*self.m21 + self.m31,
m.m31*self.m12 + m.m32*self.m22 + self.m32
)
}
pub fn translate(&self, x: T, y: T) -> TypedMatrix2D<T, Src, Dst> {
let (_0, _1): (T, T) = (Zero::zero(), One::one());
let matrix = TypedMatrix2D::new(_1, _0,
_0, _1,
x, y);
self.mul(&matrix)
}
pub fn scale(&self, x: T, y: T) -> TypedMatrix2D<T, Src, Dst> {
TypedMatrix2D::new(self.m11 * x, self.m12,
self.m21, self.m22 * y,
self.m31, self.m32)
}
pub fn identity() -> TypedMatrix2D<T, Src, Dst> {
let (_0, _1) = (Zero::zero(), One::one());
TypedMatrix2D::new(_1, _0,
_0, _1,
_0, _0)
}
#[inline]
pub fn transform_point(&self, point: &TypedPoint2D<T, Src>) -> TypedPoint2D<T, Dst> {
TypedPoint2D::new(point.x * self.m11 + point.y * self.m21 + self.m31,
point.x * self.m12 + point.y * self.m22 + self.m32)
}
#[inline]
pub fn transform_rect(&self, rect: &TypedRect<T, Src>) -> TypedRect<T, Dst> {
let top_left = self.transform_point(&rect.origin);
let top_right = self.transform_point(&rect.top_right());
let bottom_left = self.transform_point(&rect.bottom_left());
let bottom_right = self.transform_point(&rect.bottom_right());
let (mut min_x, mut min_y) = (top_left.x, top_left.y);
let (mut max_x, mut max_y) = (min_x, min_y);
for point in &[top_right, bottom_left, bottom_right] {
if point.x < min_x {
min_x = point.x
}
if point.x > max_x {
max_x = point.x
}
if point.y < min_y {
min_y = point.y
}
if point.y > max_y {
max_y = point.y
}
}
TypedRect::new(TypedPoint2D::new(min_x, min_y),
TypedSize2D::new(max_x - min_x, max_y - min_y))
}
}
impl<T: Copy, Src, Dst> TypedMatrix2D<T, Src, Dst> {
pub fn to_untyped(&self) -> Matrix2D<T> {
Matrix2D::new(
self.m11, self.m12,
self.m21, self.m22,
self.m31, self.m32
)
}
pub fn from_untyped(p: &Matrix2D<T>) -> TypedMatrix2D<T, Src, Dst> {
TypedMatrix2D::new(
p.m11, p.m12,
p.m21, p.m22,
p.m31, p.m32
)
}
}