eigen_compat.hpp

Go to the documentation of this file.

#pragma once

 
#ifdef ENABLE_EIGEN_SUPPORT
#include <Eigen/Dense>
 
// Foward Declarations
namespace sigma {
template<typename T>
class Affine;
 
template<typename T>
class ThresholdedAffine;
 
} // namespace sigma

#define EIGEN_NUMTRAITS(float_type)                                       \
                   \
    template<>                                                            \
    struct NumTraits<sigma::Affine<float_type>> : NumTraits<float_type> { \
                                            \
        using Affine = sigma::Affine<float_type>;                         \
                                \
        using Real = Affine;                                              \
                         \
        using NonInteger = Affine;                                        \
                             \
        using Literal = Affine;                                           \
                              \
        using Nested = Affine;                                            \
        enum {                                                            \
            IsComplex             = 0,                                    \
            IsInteger             = 0,                                    \
            IsSigned              = 1,                                    \
            RequireInitialization = 1,                                    \
            ReadCost              = 1,                                    \
            AddCost               = 3,                                    \
            MulCost               = 3                                     \
        };                                                                \
    };                                                                    \
        \
    template<>                                                            \
    struct NumTraits<sigma::ThresholdedAffine<float_type>>                \
      : NumTraits<float_type> {                                           \
                                            \
        using Affine = sigma::ThresholdedAffine<float_type>;              \
                                \
        using Real = Affine;                                              \
                         \
        using NonInteger = Affine;                                        \
                             \
        using Literal = Affine;                                           \
                              \
        using Nested = Affine;                                            \
        enum {                                                            \
            IsComplex             = 0,                                    \
            IsInteger             = 0,                                    \
            IsSigned              = 1,                                    \
            RequireInitialization = 1,                                    \
            ReadCost              = 1,                                    \
            AddCost               = 3,                                    \
            MulCost               = 3                                     \
        };                                                                \
    }

namespace Eigen {
 
EIGEN_NUMTRAITS(float);
EIGEN_NUMTRAITS(double);
 
} // namespace Eigen
 
#undef EIGEN_NUMTRAITS
#endif // ENABLE_EIGEN_SUPPORT