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_ */