如何推断最嵌套的迭代器类型?

时间:2019-01-18 11:35:25

标签: c++ c++11 iterator containers typetraits

我想写一个类型特征,给定一个ContainerType,它能够推断出嵌套最多的IteratorType,这意味着给定一个std::vector<int>或{{1 }}或总是std::vector<std::vector<int>>的{​​{1}},就好像它是std::vector<std::vector<std::vector<int>>>一样。

1 个答案:

答案 0 :(得分:5)

在这里,我写了一个特征和一个小小的演示它是如何工作的:

#include <type_traits>
#include <iterator>
#include <vector>

template<class ...>
using void_t = void;    

template<class T, class = void>
struct is_container : std::false_type{};

template<class T>
struct is_container<T, void_t<decltype(std::begin(std::declval<T>())), decltype(std::end(std::declval<T>()))>> : std::true_type{};

template<class T, class = void>
struct drill_iterator {
    using type =  typename T::iterator;
};

template<class T>
struct drill_iterator<T, typename std::enable_if<is_container<typename T::value_type>::value>::type > {
    using type = typename drill_iterator<typename T::value_type>::type;
};


int main(){
    drill_iterator<std::vector<std::vector<int>>>::type a;
    drill_iterator<std::vector<int>>::type b;

    if(std::is_same<decltype(b), std::vector<int>::iterator>::value
    && std::is_same<decltype(a), std::vector<int>::iterator>::value)
        return 0;

    return 1;
}

Online demo