267 lines
9.2 KiB
C++
267 lines
9.2 KiB
C++
/**
|
|
This file is a part of our_dick
|
|
Copyright (C) 2020 rexy712
|
|
|
|
This program is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU Affero General Public License as published by
|
|
the Free Software Foundation, either version 3 of the License, or
|
|
(at your option) any later version.
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU Affero General Public License for more details.
|
|
|
|
You should have received a copy of the GNU Affero General Public License
|
|
along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#ifndef REXY_MAT_HPP
|
|
#define REXY_MAT_HPP
|
|
|
|
#include <cstdlib> //size_t
|
|
#include <utility> //integer_sequence
|
|
#include <type_traits> //decay_t, is_same, integral_constant
|
|
#include "math_common.hpp"
|
|
#include "fwd_declare.hpp"
|
|
#include <cstdio>
|
|
|
|
namespace math{
|
|
|
|
//Common stuff shared by all types of matrices
|
|
template<typename T, size_t R, size_t C>
|
|
class matrix_base
|
|
{
|
|
static_assert(C > 0, "Cannot have 0 columns matrix");
|
|
static_assert(R > 0, "Cannot have 0 rows matrix");
|
|
public:
|
|
using value_type = T;
|
|
using size_type = size_t;
|
|
using pointer = value_type*;
|
|
using const_pointer = const value_type*;
|
|
using reference = value_type&;
|
|
using const_reference = const value_type&;
|
|
|
|
static constexpr size_type Columns = C;
|
|
static constexpr size_type Rows = R;
|
|
|
|
protected:
|
|
value_type m_data[R*C];
|
|
|
|
protected:
|
|
template<size_t... Ss>
|
|
constexpr matrix_base(std::integer_sequence<size_type,Ss...>);
|
|
public:
|
|
//Default construct as identity when square, zero otherwise
|
|
constexpr matrix_base();
|
|
|
|
//Range initializing constructors
|
|
constexpr explicit matrix_base(detail::zero_initialize_t);
|
|
constexpr explicit matrix_base(detail::no_initialize_t);
|
|
|
|
//Value initializing constructors
|
|
constexpr explicit matrix_base(value_type v);
|
|
template<typename... Args>
|
|
constexpr matrix_base(Args&&... args);
|
|
|
|
//Copying constructors
|
|
constexpr matrix_base(const matrix_base&) = default;
|
|
constexpr matrix_base(matrix_base&&) = default;
|
|
template<typename U>
|
|
constexpr matrix_base(const matrix_base<U,Columns,Rows>& m);
|
|
~matrix_base() = default;
|
|
|
|
constexpr matrix_base& operator=(const matrix_base&) = default;
|
|
constexpr matrix_base& operator=(matrix_base&&) = default;
|
|
template<typename U>
|
|
constexpr matrix_base& operator=(const matrix_base<U,Columns,Rows>& m);
|
|
|
|
//Getters/Setters
|
|
constexpr auto operator[](size_type x);
|
|
constexpr auto operator[](size_type x)const;
|
|
constexpr reference get(size_type x, size_type y);
|
|
constexpr const_reference get(size_type x, size_type y)const;
|
|
constexpr reference get(size_type i);
|
|
constexpr const_reference get(size_type i)const;
|
|
|
|
constexpr size_type columns()const;
|
|
constexpr size_type rows()const;
|
|
constexpr size_type size()const;
|
|
|
|
constexpr pointer raw();
|
|
constexpr const_pointer raw()const;
|
|
constexpr operator pointer();
|
|
constexpr operator const_pointer()const;
|
|
};
|
|
|
|
//Non square matrices
|
|
template<typename T, size_t R, size_t C>
|
|
class matrix : public matrix_base<T,R,C>
|
|
{
|
|
private:
|
|
using base = matrix_base<T,R,C>;
|
|
public:
|
|
using value_type = typename base::value_type;
|
|
using size_type = typename base::size_type;
|
|
using pointer = typename base::pointer;
|
|
using const_pointer = typename base::const_pointer;
|
|
using reference = typename base::reference;
|
|
using const_reference = typename base::const_reference;
|
|
|
|
public:
|
|
using base::base;
|
|
|
|
constexpr matrix(const matrix&) = default;
|
|
constexpr matrix(matrix&&) = default;
|
|
~matrix() = default;
|
|
|
|
//Assignement
|
|
constexpr matrix& operator=(const matrix&) = default;
|
|
constexpr matrix& operator=(matrix&&) = default;
|
|
template<typename U>
|
|
constexpr matrix& operator=(const matrix<U,R,C>& m);
|
|
};
|
|
|
|
//Square matrices
|
|
template<typename T, size_t R>
|
|
class matrix<T,R,R> : public matrix_base<T,R,R>
|
|
{
|
|
private:
|
|
using base = matrix_base<T,R,R>;
|
|
public:
|
|
using value_type = typename base::value_type;
|
|
using size_type = typename base::size_type;
|
|
using pointer = typename base::pointer;
|
|
using const_pointer = typename base::const_pointer;
|
|
using reference = typename base::reference;
|
|
using const_reference = typename base::const_reference;
|
|
|
|
public:
|
|
using base::base;
|
|
|
|
constexpr matrix(const matrix&) = default;
|
|
constexpr matrix(matrix&&) = default;
|
|
constexpr explicit matrix(detail::id_initialize_t);
|
|
~matrix() = default;
|
|
|
|
//Assignement
|
|
constexpr matrix& operator=(const matrix&) = default;
|
|
constexpr matrix& operator=(matrix&&) = default;
|
|
template<typename U>
|
|
constexpr matrix& operator=(const matrix<U,R,R>& m);
|
|
|
|
//square matrix arithmetic operations
|
|
constexpr value_type determinate()const;
|
|
constexpr value_type trace()const;
|
|
constexpr matrix transpose()const;
|
|
constexpr matrix inverse()const;
|
|
};
|
|
|
|
template<typename T, size_t R>
|
|
constexpr T determinate(const matrix<T,R,R>& m);
|
|
template<typename T, size_t R>
|
|
constexpr matrix<T,R,R> inverse(const matrix<T,R,R>& m);
|
|
|
|
template<typename T>
|
|
matrix<T,2,2> rotation2d_pure(T angle);
|
|
template<typename T>
|
|
constexpr matrix<T,2,2> rotation2d_pure(T sin, T cos);
|
|
template<typename T>
|
|
constexpr matrix<T,2,2> scale2d(T x, T y);
|
|
|
|
template<typename T>
|
|
matrix<T,3,3> rotation2d(T angle);
|
|
template<typename T>
|
|
constexpr matrix<T,3,3> rotation2d(T sin, T cos);
|
|
template<typename T>
|
|
matrix<T,3,3> rotation2d(T x, T y, T z);
|
|
|
|
template<typename T>
|
|
matrix<T,4,4> fov_projection(T fov, T asp, T near, T far);
|
|
template<typename T>
|
|
matrix<T,4,4> fov_asymetric_projection(T fovl, T fovr, T fovb, T fovt, T asp, T near, T far);
|
|
template<typename T>
|
|
matrix<T,4,4> ortho_projection(T w, T h, T n, T f);
|
|
template<typename T>
|
|
matrix<T,4,4> ortho_asymetric_projection(T l, T r, T b, T t, T n, T f);
|
|
template<typename T>
|
|
constexpr matrix<T,4,4> rotation3d(T angle_x, T angle_y, T angle_z);
|
|
template<typename T>
|
|
constexpr matrix<T,4,4> translation3d(T x, T y, T z);
|
|
template<typename T>
|
|
constexpr matrix<T,4,4> scale3d(T x, T y, T z);
|
|
|
|
namespace detail{
|
|
|
|
template<typename T>
|
|
struct is_matrix_helper {
|
|
template<typename U, size_t R, size_t C>
|
|
static std::true_type test(matrix_base<U,R,C>*);
|
|
static std::false_type test(void*);
|
|
|
|
static constexpr bool value = std::is_same<std::true_type,decltype(test(static_cast<std::decay_t<T>*>(nullptr)))>::value;
|
|
};
|
|
|
|
}
|
|
|
|
//Determine if a given list of tyes are all matrix types
|
|
template<typename... Ms>
|
|
struct is_matrix {
|
|
static constexpr bool value = (detail::is_matrix_helper<Ms>::value && ...);
|
|
};
|
|
|
|
namespace detail{
|
|
|
|
template<typename M1, typename M2>
|
|
struct are_same_size_matrix {
|
|
using l = std::decay_t<M1>;
|
|
using r = std::decay_t<M2>;
|
|
static constexpr bool value = is_matrix<M1,M2>::value && l::Columns == r::Columns && l::Rows == r::Rows;
|
|
};
|
|
template<typename... Ms>
|
|
using enable_if_matrix = std::enable_if_t<is_matrix<Ms...>::value,int>;
|
|
|
|
template<typename M1, typename M2>
|
|
using enable_if_eq_matrix = std::enable_if_t<are_same_size_matrix<M1,M2>::value,int>;
|
|
}
|
|
|
|
//Logic operators
|
|
template<typename T, typename U, size_t R, size_t C>
|
|
constexpr bool operator==(const matrix_base<T,R,C>& left, const matrix_base<U,R,C> right);
|
|
template<typename T, typename U, size_t R, size_t C>
|
|
constexpr bool operator!=(const matrix_base<T,R,C>& left, const matrix_base<U,R,C> right);
|
|
|
|
//Arithmetic operators
|
|
template<typename T, typename U, size_t R1, size_t C1, size_t R2>
|
|
constexpr auto operator*(const matrix<T,R1,C1>& left, const matrix<U,C1,R2>& right);
|
|
template<typename T, typename U, size_t C, size_t R, std::enable_if_t<std::is_arithmetic_v<std::decay_t<U>>,int> = 0>
|
|
constexpr auto operator*(const matrix<T,R,C>& left, U&& right);
|
|
template<typename T, typename U, size_t C, size_t R, std::enable_if_t<std::is_arithmetic_v<std::decay_t<U>>,int> = 0>
|
|
constexpr auto operator*(U&& left, const matrix<T,R,C>& right);
|
|
template<typename T, typename U, size_t C, size_t R, std::enable_if_t<std::is_arithmetic_v<std::decay_t<U>>,int> = 0>
|
|
constexpr auto operator/(const matrix<T,R,C>& left, U&& right);
|
|
template<typename T, typename U, size_t C, size_t R>
|
|
constexpr auto operator+(const matrix<T,R,C>& left, const matrix<U,R,C>& right);
|
|
template<typename T, typename U, size_t C, size_t R>
|
|
constexpr auto operator-(const matrix<T,R,C>& left, const matrix<U,R,C>& right);
|
|
template<typename T, typename U, size_t C, size_t R>
|
|
constexpr auto operator-(const matrix<T,R,C>& left);
|
|
|
|
//Arithmetic assignment operators
|
|
template<typename T, typename U, size_t R>
|
|
constexpr decltype(auto) operator*=(matrix<T,R,R>& left, const matrix<U,R,R>& right);
|
|
template<typename T, typename U, size_t C, size_t R, std::enable_if_t<std::is_arithmetic_v<std::decay_t<U>>,int> = 0>
|
|
constexpr decltype(auto) operator*=(matrix<T,R,C>& left, U&& right);
|
|
template<typename T, typename U, size_t C, size_t R, std::enable_if_t<std::is_arithmetic_v<std::decay_t<U>>,int> = 0>
|
|
constexpr decltype(auto) operator/=(matrix<T,R,C>& left, U&& right);
|
|
template<typename T, typename U, size_t C, size_t R>
|
|
constexpr decltype(auto) operator+=(matrix<T,R,C>& left, const matrix<U,R,C>& right);
|
|
template<typename T, typename U, size_t C, size_t R>
|
|
constexpr decltype(auto) operator-=(matrix<T,R,C>& left, const matrix<U,R,C>& right);
|
|
|
|
}
|
|
|
|
#include "mat.tpp"
|
|
|
|
#endif
|