- Refurbish block implementation to support nested block
- Add missing extractors
This commit is contained in:
parent
3fff326db9
commit
93a1404d9a
24 changed files with 1205 additions and 630 deletions
|
|
@ -9,50 +9,21 @@
|
|||
#include <rotgen/impl/block.hpp>
|
||||
#include <rotgen/impl/payload.hpp>
|
||||
#include <Eigen/Dense>
|
||||
#include <variant>
|
||||
|
||||
namespace rotgen
|
||||
{
|
||||
#define SIZE 64
|
||||
#define TYPE double
|
||||
#define STORAGE_ORDER Eigen::ColMajor
|
||||
#define USE_CONST
|
||||
#define CONST const
|
||||
#define BASENAME block_const_impl
|
||||
#include "block_indirect.cpp"
|
||||
#undef BASENAME
|
||||
#undef USE_CONST
|
||||
#undef CONST
|
||||
|
||||
#define CLASSNAME ROTGEN_MATRIX_NAME(block_impl,SIZE,_col)
|
||||
#define SOURCENAME ROTGEN_MATRIX_NAME(matrix_impl,SIZE,_col)
|
||||
#include "block_model.cpp"
|
||||
#undef CLASSNAME
|
||||
#undef SOURCENAME
|
||||
#undef STORAGE_ORDER
|
||||
|
||||
#define STORAGE_ORDER Eigen::RowMajor
|
||||
#define CLASSNAME ROTGEN_MATRIX_NAME(block_impl,SIZE,_row)
|
||||
#define SOURCENAME ROTGEN_MATRIX_NAME(matrix_impl,SIZE,_row)
|
||||
#include "block_model.cpp"
|
||||
#undef CLASSNAME
|
||||
#undef SOURCENAME
|
||||
#undef STORAGE_ORDER
|
||||
|
||||
#undef SIZE
|
||||
#undef TYPE
|
||||
|
||||
#define SIZE 32
|
||||
#define TYPE float
|
||||
#define STORAGE_ORDER Eigen::ColMajor
|
||||
|
||||
#define CLASSNAME ROTGEN_MATRIX_NAME(block_impl,SIZE,_col)
|
||||
#define SOURCENAME ROTGEN_MATRIX_NAME(matrix_impl,SIZE,_col)
|
||||
#include "block_model.cpp"
|
||||
#undef CLASSNAME
|
||||
#undef SOURCENAME
|
||||
#undef STORAGE_ORDER
|
||||
|
||||
#define STORAGE_ORDER Eigen::RowMajor
|
||||
#define CLASSNAME ROTGEN_MATRIX_NAME(block_impl,SIZE,_row)
|
||||
#define SOURCENAME ROTGEN_MATRIX_NAME(matrix_impl,SIZE,_row)
|
||||
#include "block_model.cpp"
|
||||
#undef CLASSNAME
|
||||
#undef SOURCENAME
|
||||
#undef STORAGE_ORDER
|
||||
|
||||
#undef SIZE
|
||||
#undef TYPE
|
||||
#define CONST
|
||||
#define BASENAME block_impl
|
||||
#include "block_indirect.cpp"
|
||||
#undef BASENAME
|
||||
#undef CONST
|
||||
}
|
||||
|
|
|
|||
51
src/block_indirect.cpp
Normal file
51
src/block_indirect.cpp
Normal file
|
|
@ -0,0 +1,51 @@
|
|||
#define SIZE 64
|
||||
#define TYPE double
|
||||
#define STORAGE_ORDER Eigen::ColMajor
|
||||
|
||||
#define CLASSNAME ROTGEN_MATRIX_NAME(BASENAME,SIZE,_col)
|
||||
#define SOURCENAME ROTGEN_MATRIX_NAME(matrix_impl,SIZE,_col)
|
||||
#define MAPNAME ROTGEN_MATRIX_NAME(map_impl,SIZE,_col)
|
||||
#include "block_model.cpp"
|
||||
#undef CLASSNAME
|
||||
#undef SOURCENAME
|
||||
#undef MAPNAME
|
||||
#undef STORAGE_ORDER
|
||||
|
||||
#define STORAGE_ORDER Eigen::RowMajor
|
||||
#define CLASSNAME ROTGEN_MATRIX_NAME(BASENAME,SIZE,_row)
|
||||
#define SOURCENAME ROTGEN_MATRIX_NAME(matrix_impl,SIZE,_row)
|
||||
#define MAPNAME ROTGEN_MATRIX_NAME(map_impl,SIZE,_row)
|
||||
#include "block_model.cpp"
|
||||
#undef CLASSNAME
|
||||
#undef SOURCENAME
|
||||
#undef MAPNAME
|
||||
#undef STORAGE_ORDER
|
||||
|
||||
#undef SIZE
|
||||
#undef TYPE
|
||||
|
||||
#define SIZE 32
|
||||
#define TYPE float
|
||||
#define STORAGE_ORDER Eigen::ColMajor
|
||||
|
||||
#define CLASSNAME ROTGEN_MATRIX_NAME(BASENAME,SIZE,_col)
|
||||
#define SOURCENAME ROTGEN_MATRIX_NAME(matrix_impl,SIZE,_col)
|
||||
#define MAPNAME ROTGEN_MATRIX_NAME(map_impl,SIZE,_col)
|
||||
#include "block_model.cpp"
|
||||
#undef CLASSNAME
|
||||
#undef SOURCENAME
|
||||
#undef MAPNAME
|
||||
#undef STORAGE_ORDER
|
||||
|
||||
#define STORAGE_ORDER Eigen::RowMajor
|
||||
#define CLASSNAME ROTGEN_MATRIX_NAME(BASENAME,SIZE,_row)
|
||||
#define SOURCENAME ROTGEN_MATRIX_NAME(matrix_impl,SIZE,_row)
|
||||
#define MAPNAME ROTGEN_MATRIX_NAME(map_impl,SIZE,_row)
|
||||
#include "block_model.cpp"
|
||||
#undef CLASSNAME
|
||||
#undef SOURCENAME
|
||||
#undef MAPNAME
|
||||
#undef STORAGE_ORDER
|
||||
|
||||
#undef SIZE
|
||||
#undef TYPE
|
||||
|
|
@ -11,204 +11,476 @@
|
|||
This file is a X-File to generate various block_impl_* definitions variant
|
||||
*/
|
||||
//==================================================================================================
|
||||
#define STR(text) STR_I(text)
|
||||
#define STR_I(...) #__VA_ARGS__
|
||||
|
||||
//==================================================================================================
|
||||
// Internal payload
|
||||
//==================================================================================================
|
||||
struct CLASSNAME::payload
|
||||
{
|
||||
using base_type = Eigen::Matrix<TYPE,Eigen::Dynamic,Eigen::Dynamic,STORAGE_ORDER>;
|
||||
using data_type = Eigen::Block<base_type>;
|
||||
using stride_type = Eigen::Stride<Eigen::Dynamic,Eigen::Dynamic>;
|
||||
using matrix_type = Eigen::Matrix<TYPE,Eigen::Dynamic,Eigen::Dynamic,STORAGE_ORDER>;
|
||||
using matrix_block_type = Eigen::Block<matrix_type CONST>;
|
||||
using matrix_storage_t = std::pair<matrix_block_type, matrix_type CONST*>;
|
||||
using map_type = Eigen::Map<matrix_type,Eigen::Unaligned,stride_type>;
|
||||
using map_block_type = Eigen::Block<map_type CONST>;
|
||||
using map_storage_t = std::pair<map_block_type, map_type CONST*>;
|
||||
using data_type = std::variant<matrix_storage_t, map_storage_t>;
|
||||
|
||||
data_type data;
|
||||
data_type data;
|
||||
rotgen::Index abs_i0 = 0; // absolute start row in original matrix
|
||||
rotgen::Index abs_j0 = 0; // absolute start col in original matrix
|
||||
rotgen::Index rel_i0 = 0; // relative start row in original matrix
|
||||
rotgen::Index rel_j0 = 0; // relative start col in original matrix
|
||||
|
||||
template <typename Func>
|
||||
void apply(Func f) { std::visit([&](auto& blk) { return f(blk.first); }, data); }
|
||||
|
||||
template <typename Func>
|
||||
void apply(Func f) const { std::visit([&](auto const& blk) { return f(blk.first); }, data); }
|
||||
|
||||
payload (data_type const& o) : data(o) {}
|
||||
|
||||
payload (base_type& r, std::size_t i0, std::size_t j0, std::size_t ni, std::size_t nj)
|
||||
: data(r,i0,j0,ni,nj)
|
||||
payload (SOURCENAME CONST& o, rotgen::Index i0, rotgen::Index j0, rotgen::Index ni, rotgen::Index nj)
|
||||
: data(matrix_storage_t{matrix_block_type{o.storage()->data,i0,j0,ni,nj},&o.storage()->data})
|
||||
, abs_i0(i0), abs_j0(j0), rel_i0(i0), rel_j0(j0)
|
||||
{}
|
||||
|
||||
payload (MAPNAME CONST& o, rotgen::Index i0, rotgen::Index j0, rotgen::Index ni, rotgen::Index nj)
|
||||
: data(map_storage_t{map_block_type{o.storage()->data,i0,j0,ni,nj},&o.storage()->data})
|
||||
, abs_i0(i0), abs_j0(j0), rel_i0(i0), rel_j0(j0)
|
||||
{}
|
||||
};
|
||||
|
||||
//==================================================================================================
|
||||
// Constructors & Special Members
|
||||
//==================================================================================================
|
||||
CLASSNAME::CLASSNAME(SOURCENAME& r, std::size_t i0, std::size_t j0, std::size_t ni, std::size_t nj)
|
||||
: storage_(std::make_unique<payload>(r.storage()->data,i0,j0,ni,nj))
|
||||
CLASSNAME::CLASSNAME(SOURCENAME CONST& r, Index i0, Index j0, Index ni, Index nj)
|
||||
: storage_(std::make_unique<payload>(r, i0,j0,ni,nj))
|
||||
{}
|
||||
|
||||
CLASSNAME::CLASSNAME(CLASSNAME const& o) : storage_(std::make_unique<payload>(o.storage_->data))
|
||||
CLASSNAME::CLASSNAME(MAPNAME CONST& r, Index i0, Index j0, Index ni, Index nj)
|
||||
: storage_(std::make_unique<payload>(r, i0,j0,ni,nj))
|
||||
{}
|
||||
|
||||
// We're building a block from a block - So we have to dig around the internals
|
||||
CLASSNAME::CLASSNAME(payload const& o, Index i0, Index j0, Index ni, Index nj)
|
||||
{
|
||||
// Compute absolute indices from the parent stored block
|
||||
Index abs_i0 = 0;
|
||||
Index abs_j0 = 0;
|
||||
std::visit( [&](auto const& blk)
|
||||
{
|
||||
abs_i0 = blk.first.startRow() + i0;
|
||||
abs_j0 = blk.first.startCol() + j0;
|
||||
}
|
||||
, o.data
|
||||
);
|
||||
|
||||
// Build a payload::data_type that holds the child block using absolute indices
|
||||
payload::data_type new_data =
|
||||
std::visit([&]<typename T>(T const& blk) -> payload::data_type
|
||||
{
|
||||
auto& [parent_block, ref] = blk;
|
||||
using block_type = typename T::first_type;
|
||||
return T{ block_type{ *ref, abs_i0, abs_j0, ni, nj }, ref };
|
||||
}, o.data);
|
||||
|
||||
storage_ = std::make_unique<payload>(new_data);
|
||||
storage_->abs_i0 = abs_i0;
|
||||
storage_->abs_j0 = abs_j0;
|
||||
storage_->rel_i0 = i0;
|
||||
storage_->rel_j0 = j0;
|
||||
}
|
||||
|
||||
CLASSNAME::CLASSNAME(CLASSNAME const& o)
|
||||
: storage_(std::make_unique<payload>(o.storage_->data))
|
||||
{}
|
||||
|
||||
CLASSNAME::CLASSNAME(CLASSNAME&&) noexcept = default;
|
||||
|
||||
#if !defined(USE_CONST)
|
||||
CLASSNAME& CLASSNAME::operator=(CLASSNAME const& o)
|
||||
{
|
||||
if (this != &o) storage_->data = o.storage_->data;
|
||||
if (this != &o) { storage_->data = o.storage_->data; }
|
||||
return *this;
|
||||
}
|
||||
|
||||
CLASSNAME::CLASSNAME(CLASSNAME&&) noexcept = default;
|
||||
CLASSNAME& CLASSNAME::operator=(CLASSNAME&&) noexcept = default;
|
||||
#endif
|
||||
|
||||
|
||||
CLASSNAME::~CLASSNAME() = default;
|
||||
|
||||
#if !defined(USE_CONST)
|
||||
void CLASSNAME::assign(SOURCENAME const& m)
|
||||
{
|
||||
storage_->data = m.storage()->data;
|
||||
storage_->apply([&](auto& blk)
|
||||
{
|
||||
blk = m.storage()->data;
|
||||
});
|
||||
}
|
||||
#endif
|
||||
|
||||
//==================================================================================================
|
||||
// Matrix API
|
||||
//==================================================================================================
|
||||
rotgen::Index CLASSNAME::rows() const { return storage_->data.rows(); }
|
||||
rotgen::Index CLASSNAME::cols() const { return storage_->data.cols(); }
|
||||
rotgen::Index CLASSNAME::size() const { return storage_->data.size(); }
|
||||
rotgen::Index CLASSNAME::innerStride() const { return storage_->data.innerStride(); }
|
||||
rotgen::Index CLASSNAME::outerStride() const { return storage_->data.outerStride(); }
|
||||
rotgen::Index CLASSNAME::rows() const
|
||||
{
|
||||
rotgen::Index that;
|
||||
storage_->apply([&](const auto& blk) { that = blk.rows(); });
|
||||
return that;
|
||||
}
|
||||
|
||||
TYPE& CLASSNAME::operator()(std::size_t i, std::size_t j) { return storage_->data(i,j); }
|
||||
TYPE const& CLASSNAME::operator()(std::size_t i, std::size_t j) const { return storage_->data(i,j); }
|
||||
rotgen::Index CLASSNAME::cols() const
|
||||
{
|
||||
rotgen::Index that;
|
||||
storage_->apply([&](const auto& blk) { that = blk.cols(); });
|
||||
return that;
|
||||
}
|
||||
|
||||
TYPE& CLASSNAME::operator()(std::size_t index) { return storage_->data.data()[index]; }
|
||||
TYPE const& CLASSNAME::operator()(std::size_t index) const { return storage_->data.data()[index]; }
|
||||
rotgen::Index CLASSNAME::size() const
|
||||
{
|
||||
rotgen::Index that;
|
||||
storage_->apply([&](const auto& blk) { that = blk.size(); });
|
||||
return that;
|
||||
}
|
||||
|
||||
TYPE* CLASSNAME::data() { return storage_->data.data(); }
|
||||
const TYPE* CLASSNAME::data() const { return storage_->data.data(); }
|
||||
rotgen::Index CLASSNAME::innerStride() const
|
||||
{
|
||||
rotgen::Index that;
|
||||
storage_->apply([&](const auto& blk) { that = blk.innerStride(); });
|
||||
return that;
|
||||
}
|
||||
|
||||
rotgen::Index CLASSNAME::outerStride() const
|
||||
{
|
||||
rotgen::Index that;
|
||||
storage_->apply([&](const auto& blk) { that = blk.outerStride(); });
|
||||
return that;
|
||||
}
|
||||
|
||||
rotgen::Index CLASSNAME::startRow() const
|
||||
{
|
||||
return storage_->rel_i0;
|
||||
}
|
||||
|
||||
rotgen::Index CLASSNAME::startCol() const
|
||||
{
|
||||
return storage_->rel_j0;
|
||||
}
|
||||
|
||||
// Element access by (i,j)
|
||||
#if !defined(USE_CONST)
|
||||
TYPE& CLASSNAME::operator()(Index i, Index j)
|
||||
{
|
||||
TYPE* ptr = nullptr;
|
||||
storage_->apply([&](auto& blk) { ptr = &blk(i,j); });
|
||||
return *ptr;
|
||||
}
|
||||
#endif
|
||||
|
||||
TYPE CLASSNAME::operator()(Index i, Index j) const
|
||||
{
|
||||
TYPE ptr;
|
||||
storage_->apply([&](auto const& blk) { ptr = blk(i,j); });
|
||||
return ptr;
|
||||
}
|
||||
|
||||
// Element access by linear index
|
||||
#if !defined(USE_CONST)
|
||||
TYPE& CLASSNAME::operator()(Index index)
|
||||
{
|
||||
TYPE* ptr = nullptr;
|
||||
storage_->apply([&](auto& blk) { ptr = blk.data() + index; });
|
||||
return *ptr;
|
||||
}
|
||||
#endif
|
||||
|
||||
TYPE CLASSNAME::operator()(Index index) const
|
||||
{
|
||||
TYPE ptr;
|
||||
storage_->apply([&](auto const& blk) { ptr = *(blk.data() + index); });
|
||||
return ptr;
|
||||
}
|
||||
|
||||
// Raw pointer access
|
||||
#if !defined(USE_CONST)
|
||||
TYPE* CLASSNAME::data()
|
||||
{
|
||||
TYPE* ptr = nullptr;
|
||||
storage_->apply([&](auto& blk) { ptr = blk.data(); });
|
||||
return ptr;
|
||||
}
|
||||
#endif
|
||||
|
||||
TYPE const* CLASSNAME::data() const
|
||||
{
|
||||
TYPE const* ptr = nullptr;
|
||||
storage_->apply([&](auto const& blk) { ptr = blk.data(); });
|
||||
return ptr;
|
||||
}
|
||||
|
||||
//==================================================================================================
|
||||
// Matrix operations
|
||||
//==================================================================================================
|
||||
SOURCENAME CLASSNAME::transpose() const
|
||||
{
|
||||
SOURCENAME result;
|
||||
result.storage()->assign(storage_->data.transpose().eval());
|
||||
storage_->apply([&](const auto& blk) { result.storage()->assign(blk.transpose().eval()); });
|
||||
return result;
|
||||
}
|
||||
|
||||
SOURCENAME CLASSNAME::conjugate() const {
|
||||
SOURCENAME CLASSNAME::conjugate() const
|
||||
{
|
||||
SOURCENAME result;
|
||||
result.storage()->assign(storage_->data.conjugate().eval());
|
||||
storage_->apply([&](const auto& blk) { result.storage()->assign(blk.conjugate().eval()); });
|
||||
return result;
|
||||
}
|
||||
|
||||
SOURCENAME CLASSNAME::adjoint() const {
|
||||
SOURCENAME CLASSNAME::adjoint() const
|
||||
{
|
||||
SOURCENAME result;
|
||||
result.storage()->assign(storage_->data.adjoint().eval());
|
||||
storage_->apply([&](const auto& blk) { result.storage()->assign(blk.adjoint().eval()); });
|
||||
return result;
|
||||
}
|
||||
|
||||
#if !defined(USE_CONST)
|
||||
void CLASSNAME::transposeInPlace()
|
||||
{
|
||||
storage_->data.transposeInPlace();
|
||||
storage_->apply([](auto& blk) { blk.transposeInPlace(); });
|
||||
}
|
||||
|
||||
void CLASSNAME::adjointInPlace()
|
||||
{
|
||||
storage_->data.adjointInPlace();
|
||||
storage_->apply([](auto& blk) { blk.adjointInPlace(); });
|
||||
}
|
||||
#endif
|
||||
|
||||
//==================================================================================================
|
||||
// Reductions
|
||||
//==================================================================================================
|
||||
TYPE CLASSNAME::sum() const
|
||||
{
|
||||
TYPE val{};
|
||||
storage_->apply([&](const auto& blk) { val = blk.sum(); });
|
||||
return val;
|
||||
}
|
||||
|
||||
TYPE CLASSNAME::sum() const { return storage_->data.sum(); }
|
||||
TYPE CLASSNAME::prod() const { return storage_->data.prod(); }
|
||||
TYPE CLASSNAME::mean() const { return storage_->data.mean(); }
|
||||
TYPE CLASSNAME::trace() const { return storage_->data.trace(); }
|
||||
TYPE CLASSNAME::prod() const
|
||||
{
|
||||
TYPE val{};
|
||||
storage_->apply([&](const auto& blk) { val = blk.prod(); });
|
||||
return val;
|
||||
}
|
||||
|
||||
TYPE CLASSNAME::minCoeff() const { return storage_->data.minCoeff(); }
|
||||
TYPE CLASSNAME::maxCoeff() const { return storage_->data.maxCoeff(); }
|
||||
TYPE CLASSNAME::mean() const
|
||||
{
|
||||
TYPE val{};
|
||||
storage_->apply([&](const auto& blk) { val = blk.mean(); });
|
||||
return val;
|
||||
}
|
||||
|
||||
TYPE CLASSNAME::minCoeff(std::ptrdiff_t* row, std::ptrdiff_t* col) const { return storage_->data.minCoeff(row, col); }
|
||||
TYPE CLASSNAME::maxCoeff(std::ptrdiff_t* row, std::ptrdiff_t* col) const { return storage_->data.maxCoeff(row, col); }
|
||||
TYPE CLASSNAME::trace() const
|
||||
{
|
||||
TYPE val{};
|
||||
storage_->apply([&](const auto& blk) { val = blk.trace(); });
|
||||
return val;
|
||||
}
|
||||
|
||||
TYPE CLASSNAME::squaredNorm() const { return storage_->data.squaredNorm(); }
|
||||
TYPE CLASSNAME::norm() const { return storage_->data.norm(); }
|
||||
TYPE CLASSNAME::minCoeff() const
|
||||
{
|
||||
TYPE val{};
|
||||
storage_->apply([&](const auto& blk) { val = blk.minCoeff(); });
|
||||
return val;
|
||||
}
|
||||
|
||||
TYPE CLASSNAME::maxCoeff() const
|
||||
{
|
||||
TYPE val{};
|
||||
storage_->apply([&](const auto& blk) { val = blk.maxCoeff(); });
|
||||
return val;
|
||||
}
|
||||
|
||||
TYPE CLASSNAME::minCoeff(std::ptrdiff_t* row, std::ptrdiff_t* col) const
|
||||
{
|
||||
TYPE val{};
|
||||
storage_->apply([&](const auto& blk) { val = blk.minCoeff(row, col); });
|
||||
return val;
|
||||
}
|
||||
|
||||
TYPE CLASSNAME::maxCoeff(std::ptrdiff_t* row, std::ptrdiff_t* col) const
|
||||
{
|
||||
TYPE val{};
|
||||
storage_->apply([&](const auto& blk) { val = blk.maxCoeff(row, col); });
|
||||
return val;
|
||||
}
|
||||
|
||||
TYPE CLASSNAME::squaredNorm() const
|
||||
{
|
||||
TYPE val{};
|
||||
storage_->apply([&](const auto& blk) { val = blk.squaredNorm(); });
|
||||
return val;
|
||||
}
|
||||
|
||||
TYPE CLASSNAME::norm() const
|
||||
{
|
||||
TYPE val{};
|
||||
storage_->apply([&](const auto& blk) { val = blk.norm(); });
|
||||
return val;
|
||||
}
|
||||
|
||||
TYPE CLASSNAME::lpNorm(int p) const
|
||||
{
|
||||
if (p == 1) return storage_->data.lpNorm<1>();
|
||||
else if (p == 2) return storage_->data.lpNorm<2>();
|
||||
else return storage_->data.lpNorm<Eigen::Infinity>();
|
||||
TYPE val{};
|
||||
storage_->apply([&](const auto& blk)
|
||||
{
|
||||
if (p == 1) val = blk.template lpNorm<1>();
|
||||
else if (p == 2) val = blk.template lpNorm<2>();
|
||||
else val = blk.template lpNorm<Eigen::Infinity>();
|
||||
});
|
||||
return val;
|
||||
}
|
||||
|
||||
//==================================================================================================
|
||||
// Operators
|
||||
//==================================================================================================
|
||||
ROTGEN_EXPORT std::ostream& operator<<(std::ostream& os,CLASSNAME const& m)
|
||||
ROTGEN_EXPORT std::ostream& operator<<(std::ostream& os, CLASSNAME const& m)
|
||||
{
|
||||
return os << m.storage_->data;
|
||||
m.storage_->apply([&](const auto& blk) { os << blk; });
|
||||
return os;
|
||||
}
|
||||
|
||||
ROTGEN_EXPORT bool operator==(CLASSNAME const& lhs, CLASSNAME const& rhs)
|
||||
{
|
||||
return lhs.storage_->data == rhs.storage_->data;
|
||||
return std::visit ( [](auto const& lhs_blk, auto const& rhs_blk) { return lhs_blk.first == rhs_blk.first; }
|
||||
, lhs.storage_->data
|
||||
, rhs.storage_->data
|
||||
);
|
||||
}
|
||||
|
||||
ROTGEN_EXPORT bool operator!=(CLASSNAME const& lhs, CLASSNAME const& rhs)
|
||||
{
|
||||
return lhs.storage_->data != rhs.storage_->data;
|
||||
return std::visit ( [](auto const& lhs_blk, auto const& rhs_blk) { return lhs_blk.first != rhs_blk.first; }
|
||||
, lhs.storage_->data
|
||||
, rhs.storage_->data
|
||||
);
|
||||
}
|
||||
|
||||
|
||||
SOURCENAME CLASSNAME::operator-() const
|
||||
{
|
||||
SOURCENAME result;
|
||||
storage_->apply([&](const auto& blk) { result.storage()->assign(blk); });
|
||||
return -result;
|
||||
}
|
||||
|
||||
#if !defined(USE_CONST)
|
||||
CLASSNAME& CLASSNAME::operator+=(CLASSNAME const& rhs)
|
||||
{
|
||||
storage_->data += rhs.storage_->data;
|
||||
std::visit( [](auto& lhs_blk, auto const& rhs_blk) { lhs_blk.first += rhs_blk.first; }
|
||||
, storage_->data
|
||||
, rhs.storage_->data
|
||||
);
|
||||
return *this;
|
||||
}
|
||||
|
||||
CLASSNAME& CLASSNAME::operator-=(CLASSNAME const& rhs)
|
||||
{
|
||||
storage_->data -= rhs.storage_->data;
|
||||
std::visit( [](auto& lhs_blk, auto const& rhs_blk) { lhs_blk.first -= rhs_blk.first; }
|
||||
, storage_->data
|
||||
, rhs.storage_->data
|
||||
);
|
||||
return *this;
|
||||
}
|
||||
|
||||
SOURCENAME CLASSNAME::operator-() const
|
||||
{
|
||||
SOURCENAME result;
|
||||
result.storage()->assign(storage_->data);
|
||||
return -result;
|
||||
}
|
||||
|
||||
CLASSNAME& CLASSNAME::operator*=(CLASSNAME const& rhs)
|
||||
{
|
||||
storage_->data *= rhs.storage_->data;
|
||||
std::visit( [](auto& lhs_blk, auto const& rhs_blk) { lhs_blk.first *= rhs_blk.first; }
|
||||
, storage_->data
|
||||
, rhs.storage_->data
|
||||
);
|
||||
return *this;
|
||||
}
|
||||
|
||||
CLASSNAME& CLASSNAME::operator*=(TYPE s)
|
||||
{
|
||||
storage_->data *= s;
|
||||
storage_->apply([&](auto& blk) { blk *= s; });
|
||||
return *this;
|
||||
}
|
||||
|
||||
CLASSNAME& CLASSNAME::operator/=(TYPE s)
|
||||
{
|
||||
storage_->data /= s;
|
||||
storage_->apply([&](auto& blk) { blk /= s; });
|
||||
return *this;
|
||||
}
|
||||
#endif
|
||||
|
||||
SOURCENAME CLASSNAME::add(CLASSNAME const& rhs) const
|
||||
{
|
||||
SOURCENAME result;
|
||||
result.storage()->assign(storage_->data + rhs.storage_->data);
|
||||
result.storage()->assign( std::visit( [](auto const& lhs_blk, auto const& rhs_blk)
|
||||
{
|
||||
return (lhs_blk.first + rhs_blk.first).eval();
|
||||
}
|
||||
, storage_->data
|
||||
, rhs.storage_->data
|
||||
)
|
||||
);
|
||||
return result;
|
||||
}
|
||||
|
||||
SOURCENAME CLASSNAME::sub(CLASSNAME const& rhs) const
|
||||
{
|
||||
SOURCENAME result;
|
||||
result.storage()->assign(storage_->data - rhs.storage_->data);
|
||||
result.storage()->assign( std::visit( [](auto const& lhs_blk, auto const& rhs_blk)
|
||||
{
|
||||
return (lhs_blk.first - rhs_blk.first).eval();
|
||||
}
|
||||
, storage_->data
|
||||
, rhs.storage_->data
|
||||
)
|
||||
);
|
||||
return result;
|
||||
}
|
||||
|
||||
SOURCENAME CLASSNAME::mul(CLASSNAME const& rhs) const
|
||||
{
|
||||
SOURCENAME result;
|
||||
result.storage()->assign(storage_->data * rhs.storage_->data);
|
||||
result.storage()->assign( std::visit( [](auto const& lhs_blk, auto const& rhs_blk)
|
||||
{
|
||||
return (lhs_blk.first * rhs_blk.first).eval();
|
||||
}
|
||||
, storage_->data
|
||||
, rhs.storage_->data
|
||||
)
|
||||
);
|
||||
return result;
|
||||
}
|
||||
|
||||
SOURCENAME CLASSNAME::mul(TYPE s) const
|
||||
{
|
||||
SOURCENAME result;
|
||||
result.storage()->assign(storage_->data * s);
|
||||
result.storage()->assign( std::visit( [s](auto const& lhs_blk)
|
||||
{
|
||||
return (lhs_blk.first * s).eval();
|
||||
}
|
||||
, storage_->data
|
||||
)
|
||||
);
|
||||
return result;
|
||||
}
|
||||
|
||||
SOURCENAME CLASSNAME::div(TYPE s) const
|
||||
{
|
||||
SOURCENAME result;
|
||||
result.storage()->assign(storage_->data / s);
|
||||
result.storage()->assign( std::visit( [s](auto const& lhs_blk)
|
||||
{
|
||||
return (lhs_blk.first / s).eval();
|
||||
}
|
||||
, storage_->data
|
||||
)
|
||||
);
|
||||
return result;
|
||||
}
|
||||
}
|
||||
|
|
|
|||
96
src/map.cpp
96
src/map.cpp
|
|
@ -14,97 +14,15 @@ namespace rotgen
|
|||
{
|
||||
#define USE_CONST
|
||||
#define CONST const
|
||||
|
||||
#define SIZE 64
|
||||
#define TYPE double
|
||||
#define STORAGE_ORDER Eigen::ColMajor
|
||||
|
||||
#define CLASSNAME ROTGEN_MATRIX_NAME(map_const_impl,SIZE,_col)
|
||||
#define SOURCENAME ROTGEN_MATRIX_NAME(matrix_impl,SIZE,_col)
|
||||
#include "map_model.cpp"
|
||||
#undef CLASSNAME
|
||||
#undef SOURCENAME
|
||||
#undef STORAGE_ORDER
|
||||
|
||||
#define STORAGE_ORDER Eigen::RowMajor
|
||||
#define CLASSNAME ROTGEN_MATRIX_NAME(map_const_impl,SIZE,_row)
|
||||
#define SOURCENAME ROTGEN_MATRIX_NAME(matrix_impl,SIZE,_row)
|
||||
#include "map_model.cpp"
|
||||
#undef CLASSNAME
|
||||
#undef SOURCENAME
|
||||
#undef STORAGE_ORDER
|
||||
|
||||
#undef SIZE
|
||||
#undef TYPE
|
||||
|
||||
#define SIZE 32
|
||||
#define TYPE float
|
||||
#define STORAGE_ORDER Eigen::ColMajor
|
||||
|
||||
#define CLASSNAME ROTGEN_MATRIX_NAME(map_const_impl,SIZE,_col)
|
||||
#define SOURCENAME ROTGEN_MATRIX_NAME(matrix_impl,SIZE,_col)
|
||||
#include "map_model.cpp"
|
||||
#undef CLASSNAME
|
||||
#undef SOURCENAME
|
||||
#undef STORAGE_ORDER
|
||||
|
||||
#define STORAGE_ORDER Eigen::RowMajor
|
||||
#define CLASSNAME ROTGEN_MATRIX_NAME(map_const_impl,SIZE,_row)
|
||||
#define SOURCENAME ROTGEN_MATRIX_NAME(matrix_impl,SIZE,_row)
|
||||
#include "map_model.cpp"
|
||||
#undef CLASSNAME
|
||||
#undef SOURCENAME
|
||||
#undef STORAGE_ORDER
|
||||
|
||||
#undef SIZE
|
||||
#undef TYPE
|
||||
|
||||
#define BASENAME map_const_impl
|
||||
#include "map_indirect.cpp"
|
||||
#undef BASENAME
|
||||
#undef USE_CONST
|
||||
#undef CONST
|
||||
|
||||
#define CONST
|
||||
|
||||
#define SIZE 64
|
||||
#define TYPE double
|
||||
#define STORAGE_ORDER Eigen::ColMajor
|
||||
|
||||
#define CLASSNAME ROTGEN_MATRIX_NAME(map_impl,SIZE,_col)
|
||||
#define SOURCENAME ROTGEN_MATRIX_NAME(matrix_impl,SIZE,_col)
|
||||
#include "map_model.cpp"
|
||||
#undef CLASSNAME
|
||||
#undef SOURCENAME
|
||||
#undef STORAGE_ORDER
|
||||
|
||||
#define STORAGE_ORDER Eigen::RowMajor
|
||||
#define CLASSNAME ROTGEN_MATRIX_NAME(map_impl,SIZE,_row)
|
||||
#define SOURCENAME ROTGEN_MATRIX_NAME(matrix_impl,SIZE,_row)
|
||||
#include "map_model.cpp"
|
||||
#undef CLASSNAME
|
||||
#undef SOURCENAME
|
||||
#undef STORAGE_ORDER
|
||||
|
||||
#undef SIZE
|
||||
#undef TYPE
|
||||
|
||||
#define SIZE 32
|
||||
#define TYPE float
|
||||
#define STORAGE_ORDER Eigen::ColMajor
|
||||
|
||||
#define CLASSNAME ROTGEN_MATRIX_NAME(map_impl,SIZE,_col)
|
||||
#define SOURCENAME ROTGEN_MATRIX_NAME(matrix_impl,SIZE,_col)
|
||||
#include "map_model.cpp"
|
||||
#undef CLASSNAME
|
||||
#undef SOURCENAME
|
||||
#undef STORAGE_ORDER
|
||||
|
||||
#define STORAGE_ORDER Eigen::RowMajor
|
||||
#define CLASSNAME ROTGEN_MATRIX_NAME(map_impl,SIZE,_row)
|
||||
#define SOURCENAME ROTGEN_MATRIX_NAME(matrix_impl,SIZE,_row)
|
||||
#include "map_model.cpp"
|
||||
#undef CLASSNAME
|
||||
#undef SOURCENAME
|
||||
#undef STORAGE_ORDER
|
||||
|
||||
#undef SIZE
|
||||
#undef TYPE
|
||||
#define BASENAME map_impl
|
||||
#include "map_indirect.cpp"
|
||||
#undef BASENAME
|
||||
#undef CONST
|
||||
}
|
||||
|
|
|
|||
43
src/map_indirect.cpp
Normal file
43
src/map_indirect.cpp
Normal file
|
|
@ -0,0 +1,43 @@
|
|||
#define SIZE 64
|
||||
#define TYPE double
|
||||
#define STORAGE_ORDER Eigen::ColMajor
|
||||
|
||||
#define CLASSNAME ROTGEN_MATRIX_NAME(BASENAME,SIZE,_col)
|
||||
#define SOURCENAME ROTGEN_MATRIX_NAME(matrix_impl,SIZE,_col)
|
||||
#include "map_model.cpp"
|
||||
#undef CLASSNAME
|
||||
#undef SOURCENAME
|
||||
#undef STORAGE_ORDER
|
||||
|
||||
#define STORAGE_ORDER Eigen::RowMajor
|
||||
#define CLASSNAME ROTGEN_MATRIX_NAME(BASENAME,SIZE,_row)
|
||||
#define SOURCENAME ROTGEN_MATRIX_NAME(matrix_impl,SIZE,_row)
|
||||
#include "map_model.cpp"
|
||||
#undef CLASSNAME
|
||||
#undef SOURCENAME
|
||||
#undef STORAGE_ORDER
|
||||
|
||||
#undef SIZE
|
||||
#undef TYPE
|
||||
|
||||
#define SIZE 32
|
||||
#define TYPE float
|
||||
#define STORAGE_ORDER Eigen::ColMajor
|
||||
|
||||
#define CLASSNAME ROTGEN_MATRIX_NAME(BASENAME,SIZE,_col)
|
||||
#define SOURCENAME ROTGEN_MATRIX_NAME(matrix_impl,SIZE,_col)
|
||||
#include "map_model.cpp"
|
||||
#undef CLASSNAME
|
||||
#undef SOURCENAME
|
||||
#undef STORAGE_ORDER
|
||||
|
||||
#define STORAGE_ORDER Eigen::RowMajor
|
||||
#define CLASSNAME ROTGEN_MATRIX_NAME(BASENAME,SIZE,_row)
|
||||
#define SOURCENAME ROTGEN_MATRIX_NAME(matrix_impl,SIZE,_row)
|
||||
#include "map_model.cpp"
|
||||
#undef CLASSNAME
|
||||
#undef SOURCENAME
|
||||
#undef STORAGE_ORDER
|
||||
|
||||
#undef SIZE
|
||||
#undef TYPE
|
||||
|
|
@ -48,7 +48,6 @@
|
|||
rotgen::Index CLASSNAME::rows() const { return storage_->data.rows(); }
|
||||
rotgen::Index CLASSNAME::cols() const { return storage_->data.cols(); }
|
||||
rotgen::Index CLASSNAME::size() const { return storage_->data.size(); }
|
||||
|
||||
rotgen::Index CLASSNAME::innerStride() const { return storage_->data.innerStride(); }
|
||||
rotgen::Index CLASSNAME::outerStride() const { return storage_->data.outerStride(); }
|
||||
|
||||
|
|
|
|||
|
|
@ -51,9 +51,9 @@ CLASSNAME::~CLASSNAME() = default;
|
|||
//==================================================================================================
|
||||
// Matrix API
|
||||
//==================================================================================================
|
||||
rotgen::Index CLASSNAME::rows() const { return storage_->data.rows(); }
|
||||
rotgen::Index CLASSNAME::cols() const { return storage_->data.cols(); }
|
||||
rotgen::Index CLASSNAME::size() const { return storage_->data.size(); }
|
||||
rotgen::Index CLASSNAME::rows() const { return storage_->data.rows(); }
|
||||
rotgen::Index CLASSNAME::cols() const { return storage_->data.cols(); }
|
||||
rotgen::Index CLASSNAME::size() const { return storage_->data.size(); }
|
||||
|
||||
void CLASSNAME::resize(std::size_t new_rows, std::size_t new_cols)
|
||||
{
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue