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;
    }
 
    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(pid.template get<0>(v_it));
        return v;
    }
 
    inline vect_dist_key_dx getSubset()
    {
        vect_dist_key_dx v;
        v.setKey(v_it);
        return v;
    }
 
    void reset()
    {
        v_it = 0;
    }
};
 
#endif /* VECTOR_DIST_ITERATOR_HPP_ */