grid_dist_id_iterator.hpp

/*
 * grid_dist_id_iterator_sub.hpp
 *
 *  Created on: Feb 4, 2015
 *      Author: Pietro Incardona
 */

#ifndef GRID_DIST_ID_ITERATOR_HPP_
#define GRID_DIST_ID_ITERATOR_HPP_

#define FREE 1
#define FIXED 2

#include "Grid/grid_dist_key.hpp"
#include "VCluster/VCluster.hpp"
#include "util/GBoxes.hpp"


template<unsigned int dim, typename device_grid, int impl,typename stencil = no_stencil >
class grid_dist_iterator
{

};


template<unsigned int dim, typename device_grid, typename stencil>
class grid_dist_iterator<dim,device_grid,FREE,stencil>
{
    size_t g_c;

    openfpm::vector<device_grid> & gList;

    const openfpm::vector<GBoxes<device_grid::dims>> & gdb_ext;

    grid_key_dx_iterator_sub<dim,stencil> a_it;

    grid_key_dx<dim> stop;

    void selectValidGrid()
    {
        // When the grid has size 0 potentially all the other informations are garbage
        while (g_c < gList.size() && (gList.get(g_c).size() == 0 || gdb_ext.get(g_c).Dbox.isValid() == false ) )
        {g_c++;}

        // get the next grid iterator
        if (g_c < gList.size())
        {
            a_it.reinitialize(gList.get(g_c).getIterator(gdb_ext.get(g_c).Dbox.getKP1(),gdb_ext.get(g_c).Dbox.getKP2()));
        }
    }

    public:

    grid_dist_iterator(openfpm::vector<device_grid> & gk, const openfpm::vector<GBoxes<device_grid::dims>> & gdb_ext, const grid_key_dx<dim> & stop)
    :g_c(0),gList(gk),gdb_ext(gdb_ext),stop(stop)
    {
        // Initialize the current iterator
        // with the first grid
        selectValidGrid();
    }

    grid_dist_iterator(openfpm::vector<device_grid> & gk,
                       const openfpm::vector<GBoxes<device_grid::dims>> & gdb_ext,
                       const grid_key_dx<dim> & stop,
                       const grid_key_dx<dim> (& stencil_pnt)[stencil::nsp])
    :g_c(0),gList(gk),gdb_ext(gdb_ext),a_it(stencil_pnt),stop(stop)
    {
        // Initialize the current iterator
        // with the first grid
        selectValidGrid();
    }

    // Destructor
    ~grid_dist_iterator()
    {
    }

    inline grid_dist_iterator<dim,device_grid,FREE,stencil> & operator++()
    {
        ++a_it;

        // check if a_it is at the end

        if (a_it.isNext() == true)
            return *this;
        else
        {
            // switch to the new grid
            g_c++;

            selectValidGrid();
        }

        return *this;
    }

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

        if (g_c >= gList.size())
        {return false;}

        return true;
    }

    inline grid_dist_key_dx<dim> get()
    {
        return grid_dist_key_dx<dim>(g_c,a_it.get());
    }

    inline grid_key_dx<dim> getStop() const
    {
        return stop;
    }

    inline grid_key_dx<dim> getStart() const
    {
        grid_key_dx<dim> start;

        start.zero();

        return start;
    }

    inline const openfpm::vector<GBoxes<device_grid::dims>> & getGBoxes()
    {
        return gdb_ext;
    }

    inline grid_key_dx<dim> getGKey(const grid_dist_key_dx<dim> & k)
    {
        // Get the sub-domain id
        size_t sub_id = k.getSub();

        grid_key_dx<dim> k_glob = k.getKey();

        // shift
        k_glob = k_glob + gdb_ext.get(sub_id).origin;

        return k_glob;
    }

    template<unsigned int id> inline grid_dist_lin_dx getStencil()
    {
        return grid_dist_lin_dx(g_c,a_it.template getStencil<id>());
    }
};


template<unsigned int dim, typename device_grid,typename stencil>
class grid_dist_iterator<dim,device_grid,FIXED,stencil>
{
    size_t g_c;

    const openfpm::vector<device_grid> & gList;

    const openfpm::vector<GBoxes<device_grid::dims>> & gdb_ext;

    grid_key_dx_iterator<dim,stencil> a_it;

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

        // get the next grid iterator
        if (g_c < gList.size())
        {
            a_it.reinitialize(gList.get(g_c).getIterator(gdb_ext.get(g_c).Dbox.getKP1(),gdb_ext.get(g_c).Dbox.getKP2()));
        }
    }

    public:

    grid_dist_iterator<dim,device_grid,FIXED> & operator=(const grid_dist_iterator<dim,device_grid,FIXED> & tmp)
    {
        g_c = tmp.g_c;
        gList = tmp.gList;
        gdb_ext = tmp.gdb_ext;
        a_it.reinitialize(tmp.a_it);

        return *this;
    }

    grid_dist_iterator(const openfpm::vector<device_grid> & gk, const openfpm::vector<GBoxes<device_grid::dims>> & gdb_ext)
    :g_c(0),gList(gk),gdb_ext(gdb_ext)
    {
        // Initialize the current iterator
        // with the first grid
        selectValidGrid();
    }

    // Destructor
    ~grid_dist_iterator()
    {
    }

    grid_dist_iterator<dim,device_grid,FIXED> &  operator++()
    {
        ++a_it;

        // check if a_it is at the end

        if (a_it.isNext() == true)
            return *this;
        else
        {
            // switch to the new grid
            g_c++;
            selectValidGrid();
        }

        return *this;
    }

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

        if (g_c >= gList.size())
            return false;

        return true;
    }

    grid_dist_key_dx<dim> get()
    {
        return grid_dist_key_dx<dim>(g_c,a_it.get());
    }

    inline const openfpm::vector<GBoxes<device_grid::dims>> & getGBoxes()
    {
        return gdb_ext;
    }

    inline grid_key_dx<dim> getGKey(const grid_dist_key_dx<dim> & k)
    {
        // Get the sub-domain id
        size_t sub_id = k.getSub();

        grid_key_dx<dim> k_glob = k.getKey();

        // shift
        k_glob = k_glob + gdb_ext.get(sub_id).origin;

        return k_glob;
    }

    template<unsigned int id> inline grid_dist_lin_dx getStencil()
    {
        return grid_dist_lin_dx(g_c,a_it.template getStencil<id>());
    }
};

#endif /* GRID_DIST_ID_ITERATOR_SUB_HPP_ */