我需要实现一个类,使用模板说'MyClass'。
template<class T>
class MyClass
{
public:
T var1;
T1 var2;
};
有两个成员变量var1和var2。如果类模板参数'T'是基本类型(例如:float,double或long double),则变量var1和var2的类型应与模板参数相同。在上例中,这是T1 = T.
但是如果模板参数是std::complex<T>
,我想要
T var1;
std::complex<T> var2;
如何在C ++ 11中实现它?
答案 0 :(得分:7)
可能的解决方案是定义一个简单的类型特征来提取正确的类型
template <typename T>
struct myTypeTraits
{ using type = T; };
template <typename T>
struct myTypeTraits<std::complex<T>>
{ using type = T; };
和MyClass
成为
template <typename T>
class MyClass
{
using T0 = typename myTypeTraits<T>::type;
T0 var1;
T var2;
};
如果你想确定T
是一种基本类型(或者你的意思是算术吗?),那就复杂一点了。
如果类型为true
false
或std::complex
)
template <typename>
struct isComplex : public std::false_type
{ };
template <typename T>
struct isComplex<std::complex<T>> : public std::true_type
{ };
接下来使用声明(无通用定义)和两个默认bool值修改myTypeTraits
template <typename T, bool = std::is_fundamental<T>::value,
bool = isComplex<T>::value>
struct myTypeTraits;
接下来的两个专业,第一个是基础,第二个是复合类型
template <typename T>
struct myTypeTraits<T, true, false>
{ using type = T; };
template <typename T>
struct myTypeTraits<std::complex<T>, false, true>
{ using type = T; };
MyClass
类保持等于但如果您尝试使用(例如)std::string
实例化它,现在会出错。
以下是完整的编译示例
#include <complex>
#include <type_traits>
template <typename>
struct isComplex : public std::false_type
{ };
template <typename T>
struct isComplex<std::complex<T>> : public std::true_type
{ };
template <typename T, bool = std::is_fundamental<T>::value,
bool = isComplex<T>::value>
struct myTypeTraits;
template <typename T>
struct myTypeTraits<T, true, false>
{ using type = T; };
template <typename T>
struct myTypeTraits<std::complex<T>, false, true>
{ using type = T; };
template <typename T>
class MyClass
{
public: // public to check with the static_assert()
using T0 = typename myTypeTraits<T>::type;
private:
T0 var1;
T var2;
};
int main ()
{
MyClass<int> mi; // compile
MyClass<std::complex<float>> mc; // compile
// MyClass<std::string> ms; // compilation error
static_assert( std::is_same<int, decltype(mi)::T0>{}, "!" );
static_assert( std::is_same<float, decltype(mc)::T0>{}, "!" );
}