一个可能的解决方案是定义一个简单的类型特征来提取正确的类型
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
{ };
接下来修改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>{}, "!" );
}