381 lines
9.0 KiB
C++
381 lines
9.0 KiB
C++
/* Copyright (C) 2017 Eric Wasylishen
|
|
|
|
This program is free software; you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation; either version 2 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 General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with this program; if not, write to the Free Software
|
|
Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
|
|
|
See file, 'COPYING', for details.
|
|
*/
|
|
|
|
#ifndef __COMMON_QVEC_HH__
|
|
#define __COMMON_QVEC_HH__
|
|
|
|
#include <initializer_list>
|
|
#include <cassert>
|
|
#include <cmath>
|
|
|
|
#ifndef qmax // FIXME: Remove this ifdef
|
|
#define qmax(a,b) (((a)>(b)) ? (a) : (b))
|
|
#define qmin(a,b) (((a)>(b)) ? (b) : (a))
|
|
#endif
|
|
|
|
template <int N, class T>
|
|
class qvec {
|
|
protected:
|
|
T v[N];
|
|
|
|
public:
|
|
qvec() {
|
|
for (int i=0; i<N; i++)
|
|
v[i] = 0;
|
|
}
|
|
|
|
qvec(const T &a) {
|
|
for (int i=0; i<N; i++)
|
|
v[i] = a;
|
|
}
|
|
|
|
qvec(const T &a, const T &b) {
|
|
v[0] = a;
|
|
if (1 < N)
|
|
v[1] = b;
|
|
for (int i=2; i<N; i++)
|
|
v[i] = 0;
|
|
}
|
|
|
|
qvec(const T &a, const T &b, const T &c) {
|
|
v[0] = a;
|
|
if (1 < N)
|
|
v[1] = b;
|
|
if (2 < N)
|
|
v[2] = c;
|
|
for (int i=3; i<N; i++)
|
|
v[i] = 0;
|
|
}
|
|
|
|
qvec(const T &a, const T &b, const T &c, const T &d) {
|
|
v[0] = a;
|
|
if (1 < N)
|
|
v[1] = b;
|
|
if (2 < N)
|
|
v[2] = c;
|
|
if (3 < N)
|
|
v[3] = d;
|
|
for (int i=4; i<N; i++)
|
|
v[i] = 0;
|
|
}
|
|
|
|
/**
|
|
* Casting from another vector type of the same length
|
|
*/
|
|
template <class T2>
|
|
qvec(const qvec<N, T2> &other) {
|
|
for (int i=0; i<N; i++)
|
|
v[i] = static_cast<T>(other[i]);
|
|
}
|
|
|
|
template <int N2>
|
|
qvec(const qvec<N2, T> &other) {
|
|
const int minSize = qmin(N,N2);
|
|
|
|
// truncates if `other` is longer than `this`
|
|
for (int i=0; i<minSize; i++)
|
|
v[i] = other[i];
|
|
|
|
// zero-fill if `other` is smaller than `this`
|
|
for (int i=minSize; i<N; i++)
|
|
v[i] = 0;
|
|
}
|
|
|
|
bool operator==(const qvec<N,T> &other) const {
|
|
for (int i=0; i<N; i++)
|
|
if (v[i] != other.v[i])
|
|
return false;
|
|
return true;
|
|
}
|
|
|
|
T operator[](const size_t idx) const {
|
|
assert(idx < N);
|
|
return v[idx];
|
|
}
|
|
|
|
T &operator[](const size_t idx) {
|
|
assert(idx < N);
|
|
return v[idx];
|
|
}
|
|
|
|
void operator+=(const qvec<N,T> &other) {
|
|
for (int i=0; i<N; i++)
|
|
v[i] += other.v[i];
|
|
}
|
|
void operator-=(const qvec<N,T> &other) {
|
|
for (int i=0; i<N; i++)
|
|
v[i] -= other.v[i];
|
|
}
|
|
void operator*=(const T &scale) {
|
|
for (int i=0; i<N; i++)
|
|
v[i] *= scale;
|
|
}
|
|
void operator/=(const T &scale) {
|
|
for (int i=0; i<N; i++)
|
|
v[i] /= scale;
|
|
}
|
|
|
|
qvec<N,T> operator+(const qvec<N,T> &other) const {
|
|
qvec<N,T> res(*this);
|
|
res += other;
|
|
return res;
|
|
}
|
|
|
|
qvec<N,T> operator-(const qvec<N,T> &other) const {
|
|
qvec<N,T> res(*this);
|
|
res -= other;
|
|
return res;
|
|
}
|
|
|
|
qvec<N,T> operator*(const T &scale) const {
|
|
qvec<N,T> res(*this);
|
|
res *= scale;
|
|
return res;
|
|
}
|
|
|
|
qvec<N,T> operator/(const T &scale) const {
|
|
qvec<N,T> res(*this);
|
|
res /= scale;
|
|
return res;
|
|
}
|
|
};
|
|
|
|
namespace qv {
|
|
template <class T>
|
|
qvec<3,T> cross(const qvec<3,T> &v1, const qvec<3,T> &v2) {
|
|
return qvec<3,T>(v1[1] * v2[2] - v1[2] * v2[1],
|
|
v1[2] * v2[0] - v1[0] * v2[2],
|
|
v1[0] * v2[1] - v1[1] * v2[0]);
|
|
}
|
|
|
|
template <int N, class T>
|
|
T dot(const qvec<N,T> &v1, const qvec<N,T> &v2) {
|
|
T result = 0;
|
|
for (int i=0; i<N; i++) {
|
|
result += v1[i] * v2[i];
|
|
}
|
|
return result;
|
|
}
|
|
|
|
template <int N, class T>
|
|
qvec<N,T> floor(const qvec<N,T> &v1) {
|
|
qvec<N,T> res;
|
|
for (int i=0; i<N; i++) {
|
|
res[i] = std::floor(v1[i]);
|
|
}
|
|
return res;
|
|
}
|
|
|
|
template <int N, class T>
|
|
T length2(const qvec<N,T> &v1) {
|
|
T len2 = 0;
|
|
for (int i=0; i<N; i++) {
|
|
len2 += (v1[i] * v1[i]);
|
|
}
|
|
return len2;
|
|
}
|
|
|
|
template <int N, class T>
|
|
T length(const qvec<N,T> &v1) {
|
|
return std::sqrt(length2(v1));
|
|
}
|
|
|
|
template <int N, class T>
|
|
qvec<N,T> normalize(const qvec<N,T> &v1) {
|
|
return v1 / length(v1);
|
|
}
|
|
|
|
template <int N, class T>
|
|
T distance(const qvec<N,T> &v1, const qvec<N,T> &v2) {
|
|
return length(v2 - v1);
|
|
}
|
|
};
|
|
|
|
|
|
using qvec2f = qvec<2, float>;
|
|
using qvec3f = qvec<3, float>;
|
|
using qvec4f = qvec<4, float>;
|
|
|
|
using qvec2d = qvec<2, double>;
|
|
using qvec3d = qvec<3, double>;
|
|
using qvec4d = qvec<4, double>;
|
|
|
|
using qvec2i = qvec<2, int>;
|
|
|
|
template <class T>
|
|
class qplane3 {
|
|
private:
|
|
qvec<3, T> m_normal;
|
|
T m_dist;
|
|
|
|
public:
|
|
qplane3(const qvec<3, T> &normal, const T &dist)
|
|
: m_normal(normal),
|
|
m_dist(dist) {}
|
|
|
|
T distAbove(const qvec<3, T> &pt) const { return qv::dot(pt, m_normal) - m_dist; }
|
|
const qvec<3, T> &normal() const { return m_normal; }
|
|
const T dist() const { return m_dist; }
|
|
};
|
|
|
|
using qplane3f = qplane3<float>;
|
|
using qplane3d = qplane3<double>;
|
|
|
|
/**
|
|
* M row, N column matrix
|
|
*/
|
|
template <int M, int N, class T>
|
|
class qmat {
|
|
protected:
|
|
qvec<M, T> m_cols[N];
|
|
|
|
public:
|
|
/**
|
|
* Identity matrix if square, otherwise fill with 0
|
|
*/
|
|
qmat() {
|
|
for (int i=0; i<N; i++)
|
|
m_cols[i] = qvec<M, T>(0);
|
|
|
|
if (M == N) {
|
|
// identity matrix
|
|
for (int i=0; i<N; i++) {
|
|
m_cols[i][i] = 1;
|
|
}
|
|
}
|
|
}
|
|
|
|
// copy constructor
|
|
qmat(const qmat<M,N,T> &other) {
|
|
for (int i=0; i<N; i++)
|
|
this->m_cols[i] = other.m_cols[i];
|
|
}
|
|
|
|
/**
|
|
* Casting from another matrix type of the same size
|
|
*/
|
|
template <class T2>
|
|
qmat(const qmat<M, N, T2> &other) {
|
|
for (int i=0; i<N; i++)
|
|
m_cols[i] = qvec<M, T>(other[i]);
|
|
}
|
|
|
|
// initializer list
|
|
qmat(std::initializer_list<T> list) {
|
|
assert(list.size() == M*N);
|
|
const T *listPtr = list.begin();
|
|
|
|
for (int j=0; j<N; j++) {
|
|
for (int i=0; i<M; i++) {
|
|
this->m_cols[j][i] = listPtr[(j*M)+i];
|
|
}
|
|
}
|
|
}
|
|
|
|
bool operator==(const qmat<M,N,T> &other) const {
|
|
for (int i=0; i<N; i++)
|
|
if (this->m_cols[i] != other.m_cols[i])
|
|
return false;
|
|
return true;
|
|
}
|
|
|
|
// access to columns
|
|
|
|
qvec<M, T> operator[](const size_t idx) const {
|
|
assert(idx >= 0 && idx < N);
|
|
return m_cols[idx];
|
|
}
|
|
|
|
qvec<M, T> &operator[](const size_t idx) {
|
|
assert(idx >= 0 && idx < N);
|
|
return m_cols[idx];
|
|
}
|
|
|
|
// multiplication by a vector
|
|
|
|
qvec<M,T> operator*(const qvec<N, T> &vec) const {
|
|
qvec<M,T> res;
|
|
for (int j=0; j<N; j++) {
|
|
res += this->m_cols[j] * vec[j];
|
|
}
|
|
return res;
|
|
}
|
|
|
|
// multiplication by a matrix
|
|
|
|
template<int P>
|
|
qmat<M,P,T> operator*(const qmat<N, P, T> &other) const {
|
|
qmat<M,P,T> res;
|
|
for (int i=0; i<M; i++) {
|
|
for (int j=0; j<P; j++) {
|
|
T val = 0;
|
|
for (int k=0; k<N; k++) {
|
|
val += (*this)[i][k] * other[k][j];
|
|
}
|
|
res[i][j] = val;
|
|
}
|
|
}
|
|
return res;
|
|
}
|
|
|
|
// multiplication by a scalar
|
|
|
|
qmat<M,N,T> operator*(const T scalar) const {
|
|
qmat<M,N,T> res;
|
|
for (int j=0; j<N; j++) {
|
|
res[j] = this->m_cols[j] * scalar;
|
|
}
|
|
return res;
|
|
}
|
|
};
|
|
|
|
using qmat2x2f = qmat<2, 2, float>;
|
|
using qmat2x3f = qmat<2, 3, float>;
|
|
using qmat2x4f = qmat<2, 4, float>;
|
|
|
|
using qmat3x2f = qmat<3, 2, float>;
|
|
using qmat3x3f = qmat<3, 3, float>;
|
|
using qmat3x4f = qmat<3, 4, float>;
|
|
|
|
using qmat4x2f = qmat<4, 2, float>;
|
|
using qmat4x3f = qmat<4, 3, float>;
|
|
using qmat4x4f = qmat<4, 4, float>;
|
|
|
|
|
|
using qmat2x2d = qmat<2, 2, double>;
|
|
using qmat2x3d = qmat<2, 3, double>;
|
|
using qmat2x4d = qmat<2, 4, double>;
|
|
|
|
using qmat3x2d = qmat<3, 2, double>;
|
|
using qmat3x3d = qmat<3, 3, double>;
|
|
using qmat3x4d = qmat<3, 4, double>;
|
|
|
|
using qmat4x2d = qmat<4, 2, double>;
|
|
using qmat4x3d = qmat<4, 3, double>;
|
|
using qmat4x4d = qmat<4, 4, double>;
|
|
|
|
namespace qv {
|
|
qmat4x4f invert(const qmat4x4f &input, bool *ok);
|
|
qmat4x4d invert(const qmat4x4d &input, bool *ok);
|
|
|
|
qmat2x2f invert(const qmat2x2f &input, bool *ok);
|
|
};
|
|
|
|
#endif /* __COMMON_QVEC_HH__ */
|