// Copyright 2013 The CGMath Developers. For a full listing of the authors,
// refer to the AUTHORS file at the top-level directory of this distribution.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

use std::num::{One, one, zero};

use traits::alg::*;
use traits::util::*;
use types::vector::*;

#[deriving(Clone, Eq, Zero)] pub struct Mat2<S> { x: Vec2<S>, y: Vec2<S> }
#[deriving(Clone, Eq, Zero)] pub struct Mat3<S> { x: Vec3<S>, y: Vec3<S>, z: Vec3<S> }
#[deriving(Clone, Eq, Zero)] pub struct Mat4<S> { x: Vec4<S>, y: Vec4<S>, z: Vec4<S>, w: Vec4<S> }

// Constructors

impl<S: Field> Mat2<S> {
    #[inline]
    pub fn new(c0r0: S, c0r1: S,
               c1r0: S, c1r1: S) -> Mat2<S> {
        Mat2::from_cols(Vec2::new(c0r0, c0r1),
                        Vec2::new(c1r0, c1r1))
    }

    #[inline]
    pub fn from_cols(c0: Vec2<S>, c1: Vec2<S>) -> Mat2<S> {
        Mat2 { x: c0, y: c1 }
    }
}

impl<S: Field> Mat3<S> {
    #[inline]
    pub fn new(c0r0:S, c0r1:S, c0r2:S,
               c1r0:S, c1r1:S, c1r2:S,
               c2r0:S, c2r1:S, c2r2:S) -> Mat3<S> {
        Mat3::from_cols(Vec3::new(c0r0, c0r1, c0r2),
                        Vec3::new(c1r0, c1r1, c1r2),
                        Vec3::new(c2r0, c2r1, c2r2))
    }

    #[inline]
    pub fn from_cols(c0: Vec3<S>, c1: Vec3<S>, c2: Vec3<S>) -> Mat3<S> {
        Mat3 { x: c0, y: c1, z: c2 }
    }
}

impl<S: Field> Mat4<S> {
    #[inline]
    pub fn new(c0r0: S, c0r1: S, c0r2: S, c0r3: S,
               c1r0: S, c1r1: S, c1r2: S, c1r3: S,
               c2r0: S, c2r1: S, c2r2: S, c2r3: S,
               c3r0: S, c3r1: S, c3r2: S, c3r3: S) -> Mat4<S>  {
        Mat4::from_cols(Vec4::new(c0r0, c0r1, c0r2, c0r3),
                        Vec4::new(c1r0, c1r1, c1r2, c1r3),
                        Vec4::new(c2r0, c2r1, c2r2, c2r3),
                        Vec4::new(c3r0, c3r1, c3r2, c3r3))
    }

    #[inline]
    pub fn from_cols(c0: Vec4<S>, c1: Vec4<S>, c2: Vec4<S>, c3: Vec4<S>) -> Mat4<S> {
        Mat4 { x: c0, y: c1, z: c2, w: c3 }
    }
}

// Trait impls

impl_indexable!(Mat2<T>, [Vec2<T>, ..2])
impl_indexable!(Mat3<T>, [Vec3<T>, ..3])
impl_indexable!(Mat4<T>, [Vec4<T>, ..4])

impl<S: Clone + Field> Swappable<Vec2<S>, [Vec2<S>, ..2]> for Mat2<S>;
impl<S: Clone + Field> Swappable<Vec3<S>, [Vec3<S>, ..3]> for Mat3<S>;
impl<S: Clone + Field> Swappable<Vec4<S>, [Vec4<S>, ..4]> for Mat4<S>;

impl_scalar_binop!(Mat2<S>, Mul, mul)
impl_scalar_binop!(Mat3<S>, Mul, mul)
impl_scalar_binop!(Mat4<S>, Mul, mul)

impl_scalar_binop!(Mat2<S>, Div, div)
impl_scalar_binop!(Mat3<S>, Div, div)
impl_scalar_binop!(Mat4<S>, Div, div)

impl_scalar_binop!(Mat2<S>, Rem, rem)
impl_scalar_binop!(Mat3<S>, Rem, rem)
impl_scalar_binop!(Mat4<S>, Rem, rem)

impl<S: Field> ScalarMul<S> for Mat2<S>;
impl<S: Field> ScalarMul<S> for Mat3<S>;
impl<S: Field> ScalarMul<S> for Mat4<S>;

impl_coordinate_binop!(Mat2<S>, Mat2<S>, Mat2<S>, Add, add)
impl_coordinate_binop!(Mat3<S>, Mat3<S>, Mat3<S>, Add, add)
impl_coordinate_binop!(Mat4<S>, Mat4<S>, Mat4<S>, Add, add)

impl_coordinate_binop!(Mat2<S>, Mat2<S>, Mat2<S>, Sub, sub)
impl_coordinate_binop!(Mat3<S>, Mat3<S>, Mat3<S>, Sub, sub)
impl_coordinate_binop!(Mat4<S>, Mat4<S>, Mat4<S>, Sub, sub)

impl_coordinate_op!(Mat2<S>, Mat2<S>, Neg, neg)
impl_coordinate_op!(Mat3<S>, Mat3<S>, Neg, neg)
impl_coordinate_op!(Mat4<S>, Mat4<S>, Neg, neg)

impl<S: Field> Module<S> for Mat2<S>;
impl<S: Field> Module<S> for Mat3<S>;
impl<S: Field> Module<S> for Mat4<S>;

impl<S: Field> One for Mat2<S> {
    fn one() -> Mat2<S> {
        Mat2::new(one(), zero(),
                  zero(), one())
    }
}

impl<S: Field> One for Mat3<S> {
    fn one() -> Mat3<S> {
        Mat3::new(one(), zero(), zero(),
                  zero(), one(), zero(),
                  zero(), zero(), one())
    }
}

impl<S: Field> One for Mat4<S> {
    fn one() -> Mat4<S> {
        Mat4::new(one(), zero(), zero(), zero(),
                  zero(), one(), zero(), zero(),
                  zero(), zero(), one(), zero(),
                  zero(), zero(), zero(), one())
    }
}

impl<S: Field> Ring<S> for Mat2<S>;
impl<S: Field> Ring<S> for Mat3<S>;
impl<S: Field> Ring<S> for Mat4<S>;

impl
<S: Clone + Field>
Matrix
<
    S,
    Vec2<S>, [S, ..2], [Vec2<S>, ..2],
    Vec2<S>, [S, ..2], [Vec2<S>, ..2],
    Mat2<S>
>
for Mat2<S>
{
    #[inline]
    fn r(&self, r: uint) -> Vec2<S> {
        Vec2::new(self.i(0).i(r).clone(),
                  self.i(1).i(r).clone())
    }

    #[inline]
    fn swap_r(&mut self, a: uint, b: uint) {
        self.mut_c(0).swap(a, b);
        self.mut_c(1).swap(a, b);
    }

    fn transpose(&self) -> Mat2<S> {
        Mat2::new(self.cr(0, 0).clone(), self.cr(1, 0).clone(),
                  self.cr(0, 1).clone(), self.cr(1, 1).clone())
    }
}

impl
<S: Clone + Field>
Matrix
<
    S,
    Vec3<S>, [S, ..3], [Vec3<S>, ..3],
    Vec3<S>, [S, ..3], [Vec3<S>, ..3],
    Mat3<S>
>
for Mat3<S>
{
    #[inline]
    fn r(&self, r: uint) -> Vec3<S> {
        Vec3::new(self.i(0).i(r).clone(),
                  self.i(1).i(r).clone(),
                  self.i(2).i(r).clone())
    }

    #[inline]
    fn swap_r(&mut self, a: uint, b: uint) {
        self.mut_c(0).swap(a, b);
        self.mut_c(1).swap(a, b);
        self.mut_c(2).swap(a, b);
    }

    fn transpose(&self) -> Mat3<S> {
        Mat3::new(self.cr(0, 0).clone(), self.cr(1, 0).clone(), self.cr(2, 0).clone(),
                  self.cr(0, 1).clone(), self.cr(1, 1).clone(), self.cr(2, 1).clone(),
                  self.cr(0, 2).clone(), self.cr(1, 2).clone(), self.cr(2, 2).clone())
    }
}

impl
<S: Clone + Field>
Matrix
<
    S,
    Vec4<S>, [S, ..4], [Vec4<S>, ..4],
    Vec4<S>, [S, ..4], [Vec4<S>, ..4],
    Mat4<S>
>
for Mat4<S>
{
    #[inline]
    fn r(&self, r: uint) -> Vec4<S> {
        Vec4::new(self.i(0).i(r).clone(),
                  self.i(1).i(r).clone(),
                  self.i(2).i(r).clone(),
                  self.i(2).i(r).clone())
    }

    #[inline]
    fn swap_r(&mut self, a: uint, b: uint) {
        self.mut_c(0).swap(a, b);
        self.mut_c(1).swap(a, b);
        self.mut_c(2).swap(a, b);
        self.mut_c(3).swap(a, b);
    }

    fn transpose(&self) -> Mat4<S> {
        Mat4::new(self.cr(0, 0).clone(), self.cr(1, 0).clone(), self.cr(2, 0).clone(), self.cr(3, 0).clone(),
                  self.cr(0, 1).clone(), self.cr(1, 1).clone(), self.cr(2, 1).clone(), self.cr(3, 1).clone(),
                  self.cr(0, 2).clone(), self.cr(1, 2).clone(), self.cr(2, 2).clone(), self.cr(3, 2).clone(),
                  self.cr(0, 3).clone(), self.cr(1, 3).clone(), self.cr(2, 3).clone(), self.cr(3, 3).clone())
    }
}