gpt4 book ai didi

c++ - 基于模板参数的条件编译时类型映射

转载 作者:行者123 更新时间:2023-11-30 04:05:11 25 4
gpt4 key购买 nike

我如何在编译时检查模板参数是否具有特定枚举,如果是,则获取该枚举值。

我正在为模板参数寻找类似这样的东西 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 类型的变量,它可能是一个有效载荷类型或更简单的类型,如 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,上述当然失败了类型。

我能看到解决方案的一种方法是对有效载荷类型或 T 的类 A 进行某种部分特化。具有 T::NeedToAlign 的类型。但是有没有一种 Boost MPL 方法可以做到这一点?

您可能已经发现,我对此很陌生。


更新1:

我想出了一种方法,即按如下方式派生 A 类:

template <typename T>
class A : public EigenAllocatorHelper<T>{
T value;
....
};

哪里EigenAllocatorHelper通过部分特化注入(inject)适当的行为(如下所示):

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 魔法不需要这个,请告诉我。

最佳答案

我目前正在使用以下解决方案:

#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));
};

关于c++ - 基于模板参数的条件编译时类型映射,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/23401223/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com