如何在编译时检查模板参数是否具有特定的枚举,如果为真,则获取该枚举值。
我正在为模板参数T
寻找类似的东西,它可能(或可能不)具有枚举类型NeedsToAlign
:
compile_time_if (T has NeedToAlign) {
// only then compile the following
EIGEN_MAKE_ALIGNED_OPERATOR_IF(typename T::NeedToAlign)
}
请继续阅读以查看我的实际问题:
所以我有一个名为Payload
的课程,它定义了一个枚举NeedsToAlign
,如下所示:
struct Payload<int N> {
typedef Eigen::Matrix<float, N, 1> DataType;
DataType data;
enum {
NeedToAlign = (sizeof(DataType)%16)==0, ///< Check for alignement need
};
EIGEN_MAKE_ALIGNED_OPERATOR_NEW_IF(EigenNeedsToAlign)
};
虽然与我的实际问题无关,但对于某些固定大小EIGEN_MAKE_ALIGNED_OPERATOR_NEW_IF()
使用Eigen library to generates aligned pointers需要DataType
宏。
现在说我有另一个模板类,它只存储一个T类型的变量,它可能是Payload类型或更简单的类型,如int / float等。
template <typename T>
class A {
T value;
// My goal:
// if T is of type Payload we need to enable the following:
// EIGEN_MAKE_ALIGNED_OPERATOR_IF(T::NeedToAlign)
};
注意我不能简单地使用EIGEN_MAKE_ALIGNED_OPERATOR_IF(typename T::NeedToAlign)
,因为我希望class A
也可以用T = int进行编译。
这些是否有Boost MPL解决方案?这是我迄今为止所能想到的:
typedef typename boost::mpl::if_<
has_EigenNeedsToAlign<Payload>,
typename Payload::NeedToAlign,
boost::mpl::int_< 0 >
>::type result;
EIGEN_MAKE_ALIGNED_OPERATOR_NEW_IF(result::value);
但当然上述T = int
类型无效。
我可以看到解决方案的一种方法是对Payload类型或具有T :: NeedToAlign的T
类型进行A类的部分特化。但是有一种Boost MPL方式吗?
正如你可能已经想到的那样,我对此非常陌生。
UPDATE1:
我已经找到了一种方法,即通过推导A类如下:
template <typename T>
class A : public EigenAllocatorHelper<T>{
T value;
....
};
其中EigenAllocatorHelper
通过部分特化(如下所示)注入适当的行为:
template<typename T, class Enable = void>
class EigenAllocatorHelper {
};
BOOST_MPL_HAS_XXX_TRAIT_DEF(NeedsToAlignMPL)
template< typename T >
class EigenAllocatorHelper< T, typename boost::enable_if< has_NeedsToAlignMPL<T> >::type > {
public:
EIGEN_MAKE_ALIGNED_OPERATOR_NEW_IF(typename T::NeedsToAlign);
};
此外,我必须使用typedef boost::mpl::int_<(sizeof(DataType)%16)==0> NeedToAlignMPL;
,因为当我使用简单的枚举时,mpl has_xxx
无效。
这个解决方案对我来说没问题,事实上我也可以在其他地方重用这个EigenAllocatorHelper
类。但如果有更好的MPL魔法不需要这个,请告诉我。
答案 0 :(得分:2)
我目前正在使用以下解决方案:
#include <boost/utility/enable_if.hpp>
#include <boost/mpl/has_xxx.hpp>
#include <boost/mpl/and.hpp>
//! The macro is used for enable_if if T is an Eigen Type
#define ENABLE_IF_EIGEN_TYPE(T)\
typename boost::enable_if< is_eigen_type<T> >::type
/** The macro enables Eigen new() when required. T can be any type
*
* Example Usage:
* EIGEN_MAKE_ALIGNED_OPERATOR_NEW_IF_REQUIRED(Eigen::Vector2d) will enable Eigen's new()
* EIGEN_MAKE_ALIGNED_OPERATOR_NEW_IF_REQUIRED(Eigen::Vector3d) will NOT
* EIGEN_MAKE_ALIGNED_OPERATOR_NEW_IF_REQUIRED(int) will NOT
*/
#define EIGEN_MAKE_ALIGNED_OPERATOR_NEW_IF_REQUIRED(T)\
EIGEN_MAKE_ALIGNED_OPERATOR_NEW_IF(requires_eigen_new_allign<T>::value)
namespace detail {
BOOST_MPL_HAS_XXX_TRAIT_DEF(Scalar)
BOOST_MPL_HAS_XXX_TRAIT_DEF(Index)
BOOST_MPL_HAS_XXX_TRAIT_DEF(StorageKind)
}
/**
* Traits for checking if T is indeed an Eigen Type
* @tparam T any Type
*
* Example Usage:
* is_eigen_type<int>::value // evaluates to false
* is_eigen_type<int>::type // evaluates to false_type
* is_eigen_type<Eigen::Vector2d>::value // evaluates to true
* is_eigen_type<Eigen::Vector2d>::type // true_type
*/
template<typename T>
struct is_eigen_type:
boost::mpl::and_<
detail::has_Scalar<T>,
detail::has_Index<T>,
detail::has_StorageKind<T> > {
};
template<class T, class Enable = void>
struct requires_eigen_new_allign {
static const bool value = false;
};
template<class T>
struct requires_eigen_new_allign<T, ENABLE_IF_EIGEN_TYPE(T)> {
typedef typename T::Scalar Scalar;
static const bool value = (((T::SizeAtCompileTime) != Eigen::Dynamic)
&& ((sizeof(Scalar) * (T::SizeAtCompileTime)) % 16 == 0));
};