vector_map_iterator.hpp

/*
 * vector_map_iterator.hpp
 *
 *  Created on: Apr 1, 2017
 *      Author: i-bird
 */
 
#ifndef OPENFPM_DATA_SRC_VECTOR_VECTOR_MAP_ITERATOR_HPP_
#define OPENFPM_DATA_SRC_VECTOR_VECTOR_MAP_ITERATOR_HPP_
 
namespace openfpm
{
    template<unsigned int p, typename vector_type>
    class vector_key_iterator_ele
    {
        const vector_type & vect;
 
        size_t end;
 
    protected:
 
        size_t gk;
 
    public:
 
        vector_key_iterator_ele(const vector_type & vect, size_t end, size_t start = 0)
                : vect(vect),end(end),gk(start)
        {}
 
 
        vector_key_iterator_ele & operator++()
        {
 
            gk++;
 
            return *this;
        }
 
        void set(int d, size_t sz)
        {
            // set the counter dim to sz
 
            gk = sz;
        }
 
        bool isNext() const
        {
            if (gk < end)
            {
 
                return true;
            }
 
            return false;
        }
 
        auto get() const -> decltype(vect.template get<p>(gk))
        {
            return vect.template get<p>(gk);
        }
    };
 
    class vector_key_iterator
    {
        size_t end;
 
    protected:
 
        size_t gk;
 
    public:
 
        vector_key_iterator(size_t end, size_t start = 0)
        : end(end),gk(start)
        {}
 
 
        vector_key_iterator & operator++()
        {
 
            gk++;
 
            return *this;
        }
 
        void set(int d, size_t sz)
        {
            // set the counter dim to sz
 
            gk = sz;
        }
 
        bool isNext() const
        {
            if (gk < end)
            {
 
                return true;
            }
 
            return false;
        }
 
        size_t get() const
        {
            return gk;
        }
    };
 
    template<typename lid>
    class vector_key_iterator_seq
    {
        openfpm::vector<lid> & dp;
 
    protected:
 
        size_t gk;
 
    public:
 
        vector_key_iterator_seq(openfpm::vector<lid> & dp)
        :dp(dp),gk(0)
        {}
 
 
        vector_key_iterator_seq<lid> & operator++()
        {
 
            gk++;
 
            return *this;
        }
 
        void set(int d, size_t sz)
        {
            // set the counter dim to sz
 
            gk = sz;
        }
 
        bool isNext() const
        {
            if (gk < dp.size())
            {
 
                return true;
            }
 
            return false;
        }
 
        size_t get() const
        {
            return dp.get(gk);
        }
    };
}
 
 
#endif /* OPENFPM_DATA_SRC_VECTOR_VECTOR_MAP_ITERATOR_HPP_ */