模板类型派生

时间:2017-11-16 16:28:15

标签: c++ c++11 templates template-specialization template-deduction

我需要实现一个类,使用模板说'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中实现它?

1 个答案:

答案 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

,可能的方法是定义类型特征(falsestd::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>{}, "!" );
 }