iterator_for.hpp

#ifndef MIGRAPH_GUARD_RTGLIB_ITERATOR_FOR_HPP
#define MIGRAPH_GUARD_RTGLIB_ITERATOR_FOR_HPP

namespace migraph {

template <class T> 
    struct iterator_for_range {
    T* base;
    using base_iterator = decltype(base->begin());

    struct iterator {
        base_iterator i;
        base_iterator operator * () { return i; }
        base_iterator operator ++ () { return ++i; }
        bool operator != (const iterator& rhs) { return i != rhs.i; }
    };

    iterator begin() { return {base->begin()}; }
    iterator end() { return {base->end()}; }
};
template <class T>
iterator_for_range<T> iterator_for(T& x) 
{
    return {&x};
}

} // namespace migraph

#endif