our_dick/include/vec.tpp
2020-08-16 08:20:47 -07:00

131 lines
3.9 KiB
C++

#ifndef REXY_VEC_TPP
#define REXY_VEC_TPP
namespace math{
template<typename T, size_t R>
template<typename U>
constexpr vector<T,R>& vector<T,R>::operator=(const vector<U,R>& m){
base::operator=(m);
return *this;
}
template<typename T, size_t R>
constexpr auto vector<T,R>::operator[](size_type i) -> reference{
return this->m_data[i];
}
template<typename T, size_t R>
constexpr auto vector<T,R>::operator[](size_type i)const -> const_reference{
return this->m_data[i];
}
template<typename T, typename U, size_t C, size_t R>
constexpr auto operator*(const matrix<U,R,C>& left, const vector<T,C>& right){
using res_t = decltype(std::declval<T>() * std::declval<U>());
vector<res_t,R> res(zero_initialize);
size_t index = 0;
//columns == rows
for(size_t i = 0;i < R;++i){
for(size_t k = 0;k < C;++k){
res.get(index) += left[i][k] * right[k];
}
++index;
}
return res;
}
template<typename T, typename U, size_t R>
constexpr auto operator*(const vector<T,R>& left, const vector<U,R>& right){
using res_t = decltype(std::declval<T>() * std::declval<U>());
res_t res = 0;
for(size_t i = 0;i < R;++i){
res += left[i] * right[i];
}
return res;
}
template<typename T, typename U, size_t R, std::enable_if_t<!is_matrix<U>::value,int>>
constexpr auto operator*(const vector<T,R>& left, U&& right){
using res_t = decltype(std::declval<T>() * std::declval<U>());
vector<res_t,R> res(zero_initialize);
for(size_t i = 0;i < R;++i){
res[i] = left[i] * std::forward<U>(right);
}
return res;
}
template<typename T, typename U, size_t R, std::enable_if_t<!is_matrix<U>::value,int>>
constexpr auto operator*(U&& left, const vector<T,R>& right){
using res_t = decltype(std::declval<U>() * std::declval<T>());
vector<res_t,R> res(zero_initialize);
for(size_t i = 0;i < R;++i){
res[i] = std::forward<U>(right) * left[i];
}
return res;
}
template<typename T, typename U, size_t R, std::enable_if_t<!is_matrix<U>::value,int>>
constexpr auto operator/(const vector<T,R>& left, U&& right){
using res_t = decltype(std::declval<T>() / std::declval<U>());
vector<res_t,R> res(zero_initialize);
for(size_t i = 0;i < R;++i){
res[i] = left[i] / std::forward<U>(right);
}
return res;
}
template<typename T, typename U, size_t R>
constexpr auto operator+(const vector<T,R>& left, const vector<U,R>& right){
using res_t = decltype(std::declval<T>() + std::declval<U>());
vector<res_t,R> res(zero_initialize);
for(size_t i = 0;i < R;++i){
res[i] = left[i] + right[i];
}
return res;
}
template<typename T, typename U, size_t R>
constexpr auto operator-(const vector<T,R>& left, const vector<U,R>& right){
using res_t = decltype(std::declval<T>() - std::declval<U>());
vector<res_t,R> res(zero_initialize);
for(size_t i = 0;i < R;++i){
res[i] = left[i] - right[i];
}
return res;
}
template<typename T, typename U, size_t R>
constexpr auto operator-(const vector<T,R>& left){
using res_t = decltype(-std::declval<U>());
vector<res_t,R> res(zero_initialize);
for(size_t i = 0;i < R;++i){
res[i] = -left[i];
}
return res;
}
template<typename T, typename U, size_t R, std::enable_if_t<!is_matrix<U>::value,int>>
constexpr decltype(auto) operator*=(vector<T,R>& left, U&& right){
for(size_t i = 0;i < R;++i){
left[i] *= right;
}
return left;
}
template<typename T, typename U, size_t R, std::enable_if_t<!is_matrix<U>::value,int>>
constexpr decltype(auto) operator/=(vector<T,R>& left, U&& right){
for(size_t i = 0;i < R;++i){
left[i] /= right;
}
return left;
}
template<typename T, typename U, size_t R>
constexpr decltype(auto) operator+=(vector<T,R>& left, const vector<U,R>& right){
for(size_t i = 0;i < R;++i){
left[i] += right[i];
}
return left;
}
template<typename T, typename U, size_t R>
constexpr decltype(auto) operator-=(vector<T,R>& left, const vector<U,R>& right){
for(size_t i = 0;i < R;++i){
left[i] -= right[i];
}
return left;
}
}
#endif