grid_dist_amr_key_iterator.hpp

/*
 * grid_amr_dist_key_iterator.hpp
 *
 *  Created on: Sep 22, 2017
 *      Author: i-bird
 */

#ifndef SRC_AMR_GRID_DIST_AMR_KEY_ITERATOR_HPP_
#define SRC_AMR_GRID_DIST_AMR_KEY_ITERATOR_HPP_

#include "Vector/map_vector.hpp"
#include "Grid/Iterators/grid_dist_id_iterator.hpp"
#include "grid_dist_amr_key.hpp"

template<unsigned int dim, typename device_grid, typename device_sub_it, typename it_type = grid_dist_iterator<dim,device_grid,device_sub_it,FREE>>
class grid_dist_amr_key_iterator
{
    openfpm::vector<it_type> & git;

    struct actual_it
    {
        it_type & it;
    };

    it_type * a_it;

    size_t g_c;



    void selectValidGrid()
    {
        // When the grid has size 0 potentially all the other informations are garbage
        while (g_c < git.size() && git.get(g_c).isNext() == false ) g_c++;

        // get the next grid iterator
        if (g_c < git.size())
        {
            a_it = &git.get(g_c);
        }
    }

public:

    grid_dist_amr_key_iterator(openfpm::vector<it_type> & git)
    :git(git),g_c(0)
    {
        a_it = &git.get(0);

        selectValidGrid();
    }


    ~grid_dist_amr_key_iterator()
    {
    }


    inline grid_dist_amr_key_iterator<dim,device_grid,device_sub_it,it_type> & operator++()
    {
        ++(*a_it);

        // check if a_it is at the end

        if (a_it->isNext() == true)
        {return *this;}
        else
        {
            // switch to the new iterator
            g_c++;

            selectValidGrid();
        }

        return *this;
    }

    inline bool isNext()
    {
        return g_c < git.size();
    }

    inline grid_dist_amr_key<dim> get()
    {
        return grid_dist_amr_key<dim>(g_c,a_it->get());
    }

    inline grid_key_dx<dim> getGKey()
    {
        return git.get(g_c).getGKey(a_it->get());
    }

    inline size_t getLvl() const
    {
        return g_c;
    }
};


#endif /* SRC_AMR_GRID_DIST_AMR_KEY_ITERATOR_HPP_ */