聲明模板參數是迭代器/指針

Question

我有一個帶指針的模板化函數。

template <typename T>
void foo(const T* bar){}

如何更改foo以確保通過迭代器/指針？ 我認為有一種static_assert或enable_if方式可以做到這一點，但我似乎找不到它。

Answer 1

您可以使用std::iterator_traits來檢查它是否是迭代器（或指針）

template <typename IT>
decltype(std::iterator_traits<IT>::iterator_category{}, void())
foo(IT bar);

Answer 2

要檢查指針，可以使用std::is_pointer 。 要檢查迭代器，可以按照此答案中的說明定義自己的is_iterator特征。 如果將兩者結合起來，將得到：

#include <type_traits>
#include <iterator>

template<typename T, typename = void>
struct is_iterator
{
   static constexpr bool value = false;
};

template<typename T>
struct is_iterator<T, typename std::enable_if<
    !std::is_same<typename std::iterator_traits<T>::value_type, void>::value>::type>
{
   static constexpr bool value = true;
};

template<class T>
void test(T t) {
  static_assert(std::is_pointer<T>::value || is_iterator<T>::value,
                "T must be pointer or iterator");
}