1. 程式人生 > 程式設計 >C++11模板超程式設計-std::enable_if示例詳解

C++11模板超程式設計-std::enable_if示例詳解

C++11中引入了std::enable_if函式,函式原型如下:

template< bool B,class T = void >
struct enable_if;

可能的函式實現:

template<bool B,class T = void>
struct enable_if {};
 
template<class T>
struct enable_if<true,T> { typedef T type; };

由上可知,只有當第一個模板引數為true時,enable_if會包含一個type=T的公有成員,否則沒有該公有成員。

標頭檔案:

#include <type_traits>

std::enable_if使用場景

1、限制模板函式的引數型別

在某些場景下,我們需要實現只有特定型別可以呼叫的模板函式。如下程式碼所示,通過對返回值使用std::enable_if和在模板引數中使用std::enable_if均實現了只允許整形引數呼叫函式的功能。

// enable_if example: two ways of using enable_if
#include <iostream>
#include <type_traits>

// 1. the return type (bool) is only valid if T is an integral type:
template <class T>
typename std::enable_if<std::is_integral<T>::value,bool>::type
 is_odd (T i) {return bool(i%2);}

// 2. the second template argument is only valid if T is an integral type:
template < class T,class = typename std::enable_if<std::is_integral<T>::value>::type>
bool is_even (T i) {return !bool(i%2);}

int main() {

 short int i = 1;  // code does not compile if type of i is not integral

 std::cout << std::boolalpha;
 std::cout << "i is odd: " << is_odd(i) << std::endl;
 std::cout << "i is even: " << is_even(i) << std::endl;

 return 0;
}

當使用float型別引數呼叫函式時,程式會報錯:

error: no matching function for call to 'is_odd(float&)'

2. 模板型別偏特化

在使用模板程式設計時,可以利用std::enable_if的特性根據模板引數的不同特性進行不同的型別選擇。

如下所示,我們可以實現一個檢測變數是否為智慧指標的實現:

#include <iostream>
#include <type_traits>
#include <memory>

template <typename T>
struct is_smart_pointer_helper : public std::false_type {};

template <typename T>
struct is_smart_pointer_helper<std::shared_ptr<T> > : public std::true_type {};

template <typename T>
struct is_smart_pointer_helper<std::unique_ptr<T> > : public std::true_type {};

template <typename T>
struct is_smart_pointer_helper<std::weak_ptr<T> > : public std::true_type {};

template <typename T>
struct is_smart_pointer : public is_smart_pointer_helper<typename std::remove_cv<T>::type> {};

template <typename T>
typename std::enable_if<is_smart_pointer<T>::value,void>::type check_smart_pointer(const T& t)
{
  std::cout << "is smart pointer" << std::endl;
}

template <typename T>
typename std::enable_if<!is_smart_pointer<T>::value,void>::type check_smart_pointer(const T& t)
{
  std::cout << "not smart pointer" << std::endl;
}

int main()
{
  int* p(new int(2));
  std::shared_ptr<int> pp(new int(2));
  std::unique_ptr<int> upp(new int(4));

  check_smart_pointer(p);
  check_smart_pointer(pp);
  check_smart_pointer(upp);
  
  return 0;
}

程式輸出:

not smart pointer
is smart pointer
is smart pointer

參考材料

http://www.cplusplus.com/reference/type_traits/enable_if/

https://en.cppreference.com/w/cpp/types/enable_if

總結

到此這篇關於C++11模板超程式設計-std::enable_if的文章就介紹到這了,更多相關C++11模板超程式設計-std::enable_if內容請搜尋我們以前的文章或繼續瀏覽下面的相關文章希望大家以後多多支援我們!