Homec4science
Diffusion tamaas (features/declarative) Browse

tamaas/src/core/static_types.hh
f31c9478eeb0features/declarative

Last Change
Raw File

static_types.hh
View Options

/**
* @file
*
* @author Lucas Frérot <lucas.frerot@epfl.ch>
*
* @section LICENSE
*
* Copyright (©) 2017 EPFL (Ecole Polytechnique Fédérale de
* Lausanne) Laboratory (LSMS - Laboratoire de Simulation en Mécanique des
* Solides)
*
* Tamaas is free software: you can redistribute it and/or modify it under the
* terms of the GNU Lesser General Public License as published by the Free
* Software Foundation, either version 3 of the License, or (at your option) any
* later version.
*
* Tamaas 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 Lesser General Public License for more
* details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with Tamaas. If not, see <http://www.gnu.org/licenses/>.
*
*/
/* -------------------------------------------------------------------------- */
#ifndef __STATIC_TYPES_HH__
#define __STATIC_TYPES_HH__
/* -------------------------------------------------------------------------- */
#include "tamaas.hh"
__BEGIN_TAMAAS__
template <typename T, UInt n, UInt m>
class StaticMatrix;
/// Vector class with size determined at compile-time
template <typename T, UInt n>
class StaticVector {
public:
/// Implicit wrap constructor
StaticVector(T & spot):_mem(&spot) {}
public:
/// Access operator
T & operator()(UInt i) {
TAMAAS_ASSERT(i < n, "Access out of bounds");
return _mem[i];
}
/// Access operator
const T & operator()(UInt i) const {
TAMAAS_ASSERT(i < n, "Access out of bounds");
return _mem[i];
}
/// L2 norm squared
T l2squared() const {
T res = 0;
for (UInt i = 0 ; i < n ; ++i) res += _mem[i]*_mem[i];
return res;
}
/// L2 norm
T l2norm() const { return std::sqrt(l2squared()); }
/// Matrix-vector product
template <UInt m>
void mul(const StaticMatrix<T, n, m> & mat, const StaticVector<T, m> & vec) {
*this = 0;
for (UInt i = 0 ; i < n ; ++i)
for (UInt j = 0 ; j < m ; ++j)
(*this)(i) += mat(i, j) * vec(j);
}
#define VECTOR_OP(op) \
template <typename T1> \
void operator op(const StaticVector<T1, n> & o) { \
for (UInt i = 0 ; i < n ; ++i) (*this)(i) op o(i); \
}
VECTOR_OP(+=);
VECTOR_OP(-=);
VECTOR_OP(*=);
VECTOR_OP(/=);
#undef VECTOR_OP
#define SCALAR_OP(op) \
StaticVector & operator op(const T & x) { \
for (UInt i = 0 ; i < n ; ++i) (*this)(i) op x; \
return *this; \
}
SCALAR_OP(+=);
SCALAR_OP(-=);
SCALAR_OP(*=);
SCALAR_OP(/=);
SCALAR_OP(=);
#undef SCALAR_OP
protected:
T * _mem;
};
/// Matrix class with size determined at compile time
template <typename T, UInt n, UInt m>
class StaticMatrix : public StaticVector<T, n*m> {
public:
using StaticVector<T, n*m>::StaticVector;
template <UInt p>
void mul(const StaticMatrix<T, n, p> & mat, const StaticVector<T, p> & vec) = delete;
public:
/// Access operator
T & operator()(UInt i, UInt j) {
UInt index = i * m + j;
TAMAAS_ASSERT(index < n, "Access out of bounds");
return this->_mem[index];
}
/// Access operator
const T & operator()(UInt i, UInt j) const {
UInt index = i * m + j;
TAMAAS_ASSERT(index < n, "Access out of bounds");
return this->_mem[index];
}
};
__END_TAMAAS__
#endif // __STATIC_TYPES_HH__
c4science · Help