How to find the index of current object in range-based for loop?

by

Yes you can, it just take some massaging 😉

The trick is to use composition: instead of iterating over the container directly, you “zip” it with an index along the way.

Specialized zipper code:

template <typename T>
struct iterator_extractor { typedef typename T::iterator type; };

template <typename T>
struct iterator_extractor<T const> { typedef typename T::const_iterator type; };


template <typename T>
class Indexer {
public:
    class iterator {
        typedef typename iterator_extractor<T>::type inner_iterator;

        typedef typename std::iterator_traits<inner_iterator>::reference inner_reference;
    public:
        typedef std::pair<size_t, inner_reference> reference;

        iterator(inner_iterator it): _pos(0), _it(it) {}

        reference operator*() const { return reference(_pos, *_it); }

        iterator& operator++() { ++_pos; ++_it; return *this; }
        iterator operator++(int) { iterator tmp(*this); ++*this; return tmp; }

        bool operator==(iterator const& it) const { return _it == it._it; }
        bool operator!=(iterator const& it) const { return !(*this == it); }

    private:
        size_t _pos;
        inner_iterator _it;
    };

    Indexer(T& t): _container(t) {}

    iterator begin() const { return iterator(_container.begin()); }
    iterator end() const { return iterator(_container.end()); }

private:
    T& _container;
}; // class Indexer

template <typename T>
Indexer<T> index(T& t) { return Indexer<T>(t); }

And using it:

#include <iostream>
#include <iterator>
#include <limits>
#include <vector>

// Zipper code here

int main() {
    std::vector<int> v{1, 2, 3, 4, 5, 6, 7, 8, 9};

    for (auto p: index(v)) {
        std::cout << p.first << ": " << p.second << "\n";
    }
}

You can see it at ideone, though it lacks the for-range loop support so it’s less pretty.

EDIT:

Just remembered that I should check Boost.Range more often. Unfortunately no zip range, but I did found a pearl: boost::adaptors::indexed. However it requires access to the iterator to pull of the index. Shame 😡

Otherwise with the counting_range and a generic zip I am sure it could be possible to do something interesting…

In the ideal world I would imagine:

int main() {
    std::vector<int> v{1, 2, 3, 4, 5, 6, 7, 8, 9};

    for (auto tuple: zip(iota(0), v)) {
        std::cout << tuple.at<0>() << ": " << tuple.at<1>() << "\n";
    }
}

With zip automatically creating a view as a range of tuples of references and iota(0) simply creating a “false” range that starts from 0 and just counts toward infinity (or well, the maximum of its type…).

More Related Contents:

  1. C++ – Find numbers between first and last negative numbers in Array
  2. c++ removing whitespace fails by using iterators
  3. Why use iterators instead of array indices?
  4. Deleting elements from std::set while iterating
  5. What’s the best way to iterate over two or more containers simultaneously
  6. Flattening iterator
  7. iterate vector, remove certain items as I go
  8. comparing iterators from different containers
  9. How are iterators and pointers related?
  10. What’s faster, iterating an STL vector with vector::iterator or with at()?
  11. Can I convert a reverse iterator to a forward iterator?
  12. Does moving a vector invalidate iterators?
  13. What is an iterator’s default value?
  14. Why is “!=” used with iterators instead of “
  15. What is an iterator in general?
  16. Why does a push_back on an std::list change a reverse iterator initialized with rbegin?
  17. What happens if you increment an iterator that is equal to the end iterator of an STL container
  18. How might I wrap the FindXFile-style APIs to the STL-style Iterator Pattern in C++?
  19. Creating my own Iterators
  20. Need iterator when using ranged-based for loops
  21. What is iterator invalidation?
  22. Are const_iterators faster?
  23. Is an iterator in C++ a pointer?
  24. What is the reason behind cbegin/cend?
  25. How to check that the passed Iterator is a random access iterator?
  26. Chaining iterators for C++
  27. iterator adapter to iterate just the values in a map?
  28. How to increment an iterator by 2?
  29. How to erase & delete pointers to objects stored in a vector?
  30. How can I copy one map into another using std::copy?

Leave a Comment