You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
89 lines
3.8 KiB
89 lines
3.8 KiB
#ifndef MEDUSA_BITS_OPERATORS_IMPLICITOPERATORS_HPP_
|
|
#define MEDUSA_BITS_OPERATORS_IMPLICITOPERATORS_HPP_
|
|
|
|
/**
|
|
* @file
|
|
* Implementations of implicit operators.
|
|
*/
|
|
|
|
#include "ImplicitOperators_fwd.hpp"
|
|
#include "ShapeStorage_fwd.hpp"
|
|
|
|
namespace mm {
|
|
|
|
template <class shape_storage_type, class matrix_type, class rhs_type>
|
|
ImplicitOperators<shape_storage_type, matrix_type, rhs_type>::ImplicitOperators(
|
|
const shape_storage_t& ss, matrix_t& M, rhs_t& rhs, int row_offset, int col_offset) :
|
|
ss(&ss), M(&M), rhs(&rhs), row_offset(row_offset), col_offset(col_offset) {
|
|
assert_msg(0 <= row_offset, "Row offset cannot be negative, got %d.", row_offset);
|
|
assert_msg(0 <= col_offset, "Col offset cannot be negative, got %d.", col_offset);
|
|
assert_msg(M.rows() >= ss.size() + row_offset,
|
|
"Matrix does not have enough rows, expected at least %d = %d + %d, got %d.",
|
|
ss.size()+row_offset, row_offset, ss.size(), M.rows());
|
|
assert_msg(M.cols() >= ss.size(),
|
|
"Matrix does not have enough columns, expected at least %d = %d + %d, got %d.",
|
|
ss.size()+col_offset, col_offset, ss.size(), M.cols());
|
|
assert_msg(rhs.size() >= ss.size() + row_offset,
|
|
"RHS vector does not have enough rows, expected at least %d = %d + %d, got %d.",
|
|
ss.size()+row_offset, row_offset, ss.size(), M.rows());
|
|
#ifndef NDEBUG
|
|
// Warning if the matrix or rhs is not zero initialized.
|
|
if (M.squaredNorm() > 1e-10) {
|
|
std::cerr << "Warning: matrix in implicit operators not initialized to zero!" << std::endl;
|
|
}
|
|
if (rhs.squaredNorm() > 1e-10) {
|
|
std::cerr << "Warning: rhs in implicit operators not initialized to zero!" << std::endl;
|
|
}
|
|
#endif
|
|
}
|
|
|
|
template <class S, class matrix_type, class rhs_type>
|
|
void ImplicitOperators<S, matrix_type, rhs_type>::addToM(int i, int j, scalar_t v) {
|
|
assert_msg(M->rows() >= ss->size(),
|
|
"Matrix does not have enough rows, expected at least %d, got %d.",
|
|
ss->size(), M->rows());
|
|
assert_msg(M->cols() >= ss->size(),
|
|
"Matrix does not have enough columns, expected at least %d, got %d.",
|
|
ss->size(), M->cols());
|
|
assert_msg(v == v, "You tried to set M(%d, %d) to NaN. Check your computed shapes.",
|
|
row_offset+i, col_offset+j);
|
|
M->coeffRef(i+row_offset, j+col_offset) += v;
|
|
}
|
|
|
|
template <class shape_storage_type, class matrix_type, class rhs_type>
|
|
void ImplicitOperators<shape_storage_type, matrix_type, rhs_type>::addToRhs(int i, scalar_t v) {
|
|
assert_msg(rhs->size() >= ss->size(),
|
|
"RHS vector does not have enough rows, expected at least %d, got %d.",
|
|
ss->size(), rhs->size());
|
|
assert_msg(v == v, "You tried to set rhs(%d) to NaN. Check your computed shapes.",
|
|
row_offset+i);
|
|
rhs->operator[](i+row_offset) += v;
|
|
}
|
|
|
|
/// Output basic info about given operators.
|
|
template <typename S, typename M, typename R>
|
|
std::ostream& operator<<(std::ostream& os, const ImplicitOperators<S, M, R>& op) {
|
|
os << "Implicit operators ";
|
|
os << ((op.hasShapes()) ? "with" : "without") << " linked storage ";
|
|
if (!op.hasShapes()) {
|
|
os << "without linked storage";
|
|
} else {
|
|
os << "over storage: " << *op.ss;
|
|
}
|
|
os << " with " << ((op.hasMatrix()) ? "defined" : "undefined") << " problem matrix";
|
|
os << " and with " << ((op.hasRhs()) ? "defined" : "undefined") << " problem rhs.";
|
|
return os;
|
|
}
|
|
|
|
/// @cond
|
|
template <typename Derived, typename vec_t, typename OpFamilies>
|
|
template <typename M, typename R>
|
|
ImplicitOperators<Derived, M, R>
|
|
ShapeStorage<Derived, vec_t, OpFamilies>::implicitOperators(M& matrix, R& rhs) const {
|
|
return {*static_cast<const Derived*>(this), matrix, rhs};
|
|
}
|
|
/// @endcond
|
|
|
|
} // namespace mm
|
|
|
|
#endif // MEDUSA_BITS_OPERATORS_IMPLICITOPERATORS_HPP_
|
|
|