模板類：檢查在編譯時是否複雜

Question

我有一個A類A模板與Scalar可以是實數或複數值。它有一個方法realPart它應該返回數字的實際部分。如果Scalar是實數值，則它應該返回原始標量，如果它是複雜類型，則返回.real()。

當寫

#include <complex> 
#include <iostream> 

template<class Scalar> 
class A { 
    public: 
    A (const Scalar z): 
     z_(z) 
    { 
    } 

    Scalar realPart() 
    { 
     return z_.real(); 
    } 

    private: 
    Scalar z_; 
}; 

int main() { 
    A<std::complex<double>> z0((1.0, 2.0)); 
    std::cout << z0.realPart() << std::endl; // check 

    A<double> z1(1.0); 
    std::cout << z1.realPart() << std::endl; // mööp 
} 

編譯器會在z1.realPart()抱怨，因爲double不知道.real()。

在編譯時如何防範.real()？

Answer 1

這可以用一個簡單的is_complex特質和SFINAE來完成：

template<class T> struct is_complex : std::false_type {}; 
template<class T> struct is_complex<std::complex<T>> : std::true_type {}; 

template<class Scalar> 
class A { 
public: 
    A(const Scalar z) : z_(z) 
    { } 

    template<class S = Scalar, std::enable_if_t<is_complex<S>{}>* = nullptr> 
    Scalar realPart() 
    { 
     return z_.real(); 
    } 

    template<class S = Scalar, std::enable_if_t<!is_complex<S>{}>* = nullptr> 
    Scalar realPart() 
    { 
     return z_; 
    } 
private: 
    Scalar z_; 
};