vector_dist_iterator.hpp

/*
 * vector_dist_iterator.hpp
 *
 *  Created on: Mar 10, 2015
 *      Author: Pietro Incardona
 */

#ifndef VECTOR_DIST_ITERATOR_HPP_
#define VECTOR_DIST_ITERATOR_HPP_

#include "Vector/vector_dist_key.hpp"
#include "VCluster/VCluster.hpp"

class vector_dist_iterator
{
    size_t v_it;

    size_t stop;

    public:

    vector_dist_iterator(size_t start, size_t stop)
    :v_it(start),stop(stop)
    {
    }

    // Destructor
    ~vector_dist_iterator()
    {
    }

    vector_dist_iterator & operator++()
    {
        ++v_it;

        return *this;
    }

    bool isNext()
    {
        // If there are no other grid stop

        if (v_it >= stop)
            return false;

        return true;
    }

    inline vect_dist_key_dx get()
    {
        vect_dist_key_dx v;
        v.setKey(v_it);
        return v;
    }

    inline vect_dist_key_dx getOrig()
    {
        vect_dist_key_dx v;
        v.setKey(v_it);
        return v;
    }

    void reset()
    {
        v_it = 0;
    }
};


class vector_dist_iterator_subset
{
    size_t v_it;

    size_t stop;

    const openfpm::vector<aggregate<int>> & pid;

    public:

    vector_dist_iterator_subset(size_t start, size_t stop, const openfpm::vector<aggregate<int>> & pid)
    :v_it(start),stop(stop),pid(pid)
    {
    }

    // Destructor
    ~vector_dist_iterator_subset()
    {
    }

    vector_dist_iterator_subset & operator++()
    {
        ++v_it;

        return *this;
    }

    bool isNext()
    {
        // If there are no other grid stop

        if (v_it >= stop)
            return false;

        return true;
    }

    inline vect_dist_key_dx get()
    {
        vect_dist_key_dx v;
        v.setKey(v_it);
        return v;
    }

    inline vect_dist_key_dx getOrig()
    {
        vect_dist_key_dx v;
        v.setKey(pid.template get<0>(v_it));
        return v;
    }

    void reset()
    {
        v_it = 0;
    }
};

#endif /* VECTOR_DIST_ITERATOR_HPP_ */