variadic version of assert which can take a parameter pack as its input.

This commit is contained in:
Mehdi Goli 2018-08-01 12:19:14 +01:00
parent edf46bd7a2
commit 3a197a60e6
2 changed files with 26 additions and 0 deletions

View File

@ -1076,4 +1076,28 @@ namespace Eigen {
# endif
#endif
#ifdef EIGEN_HAS_VARIADIC_TEMPLATES
// Provide a variadic version of assert which can take a parameter pack as its input
// The eigen_assert macro used here might have been redefined to use other macros such as EIGEN_THROW, such as used in Eigen's test suite, therefore this needs to be defined after the other macros.
// Note that this does not provide as nice a string to assert as a straight forward call to eigen_assert, so we add a message to the assert.
#if defined(EIGEN_NO_DEBUG)
#define eigen_variadic_assert(x)
#else
namespace Eigen {
namespace internal {
inline void variadic_assert(const char*) {}
template<typename... Bools> inline void variadic_assert(const char* message, bool first, Bools ... others) {
eigen_assert(first && message);
variadic_assert(message, others...);
EIGEN_UNUSED_VARIABLE(first);
}
}
}
#define EIGEN_VARIADIC_ASSERT_MESSAGE(x) EIGEN_MAKESTRING(x) " in " __FILE__ ":" EIGEN_MAKESTRING(__LINE__)
#define eigen_variadic_assert(x) \
do { Eigen::internal::variadic_assert(EIGEN_VARIADIC_ASSERT_MESSAGE(x), x); } while(false);
#endif
#endif
#endif // EIGEN_MACROS_H

View File

@ -150,6 +150,7 @@ template<typename PlainObjectType, int Options_, template <class> class MakePoin
EIGEN_STRONG_INLINE const Scalar& operator()(Index firstIndex, Index secondIndex, IndexTypes... otherIndices) const
{
EIGEN_STATIC_ASSERT(sizeof...(otherIndices) + 2 == NumIndices, YOU_MADE_A_PROGRAMMING_MISTAKE)
eigen_variadic_assert((Eigen::NumTraits<Index>::highest() >= otherIndices)...);
if (PlainObjectType::Options&RowMajor) {
const Index index = m_dimensions.IndexOfRowMajor(array<Index, NumIndices>{{firstIndex, secondIndex, otherIndices...}});
return m_data[index];
@ -237,6 +238,7 @@ template<typename PlainObjectType, int Options_, template <class> class MakePoin
EIGEN_STRONG_INLINE Scalar& operator()(Index firstIndex, Index secondIndex, IndexTypes... otherIndices)
{
static_assert(sizeof...(otherIndices) + 2 == NumIndices || NumIndices == Dynamic, "Number of indices used to access a tensor coefficient must be equal to the rank of the tensor.");
eigen_variadic_assert((Eigen::NumTraits<Index>::highest() >= otherIndices)...);
const std::size_t NumDims = sizeof...(otherIndices) + 2;
if (PlainObjectType::Options&RowMajor) {
const Index index = m_dimensions.IndexOfRowMajor(array<Index, NumDims>{{firstIndex, secondIndex, otherIndices...}});