grid_dist_id_iterator_util.hpp

/*
 * grid_dist_id_iterator_util.hpp
 *
 *  Created on: Jan 6, 2017
 *      Author: i-bird
 */
 
#ifndef SRC_GRID_ITERATORS_GRID_DIST_ID_ITERATOR_UTIL_HPP_
#define SRC_GRID_ITERATORS_GRID_DIST_ID_ITERATOR_UTIL_HPP_
 
 
template<typename Decomposition>
static inline bool compute_subset_domain(const openfpm::vector<GBoxes<Decomposition::dims>> & gdb_ext,
                                  size_t g_c,
                                  grid_key_dx<Decomposition::dims> & start,
                                  grid_key_dx<Decomposition::dims> & stop,
                                  grid_key_dx<Decomposition::dims> & start_c,
                                  grid_key_dx<Decomposition::dims> & stop_c)
{
    if (gdb_ext.get(g_c).Dbox.isValid() == false)
    {return false;}
 
    // Intersect the grid keys
 
    for (size_t i = 0 ; i < Decomposition::dims ; i++)
    {
        long int start_p = gdb_ext.get(g_c).Dbox.getP1().get(i) + gdb_ext.get(g_c).origin.get(i);
        long int stop_p = gdb_ext.get(g_c).Dbox.getP2().get(i) + gdb_ext.get(g_c).origin.get(i);
        if (start.get(i) <= start_p)
        {start_c.set_d(i,gdb_ext.get(g_c).Dbox.getP1().get(i));}
        else if (start.get(i) <= stop_p)
        {start_c.set_d(i,start.get(i) - gdb_ext.get(g_c).origin.get(i));}
        else
        {return false;}
 
        if (stop.get(i) >= stop_p)
        {stop_c.set_d(i,gdb_ext.get(g_c).Dbox.getP2().get(i));}
        else if (stop.get(i) >= start_p)
        {stop_c.set_d(i,stop.get(i) - gdb_ext.get(g_c).origin.get(i));}
        else
        {return false;}
    }
 
    return true;
}
 
template<typename Decomposition>
static inline bool compute_subset_ghost(const openfpm::vector<GBoxes<Decomposition::dims>> & gdb_ext,
                                        size_t g_c,
                                        grid_key_dx<Decomposition::dims> & start,
                                        grid_key_dx<Decomposition::dims> & stop,
                                        grid_key_dx<Decomposition::dims> & start_c,
                                        grid_key_dx<Decomposition::dims> & stop_c)
{
    // Intersect the grid keys
 
    for (size_t i = 0 ; i < Decomposition::dims ; i++)
    {
        long int start_p = gdb_ext.get(g_c).GDbox.getP1().get(i) + gdb_ext.get(g_c).origin.get(i);
        long int stop_p = gdb_ext.get(g_c).GDbox.getP2().get(i) + gdb_ext.get(g_c).origin.get(i);
        if (start.get(i) < start_p)
        {start_c.set_d(i,gdb_ext.get(g_c).GDbox.getP1().get(i));}
        else if (start.get(i) <= stop_p)
        {start_c.set_d(i,start.get(i) - gdb_ext.get(g_c).origin.get(i));}
        else
        {return false;}
 
        if (stop.get(i) > stop_p)
        {stop_c.set_d(i,gdb_ext.get(g_c).GDbox.getP2().get(i));}
        else if (stop.get(i) >= start_p)
        {stop_c.set_d(i,stop.get(i) - gdb_ext.get(g_c).origin.get(i));}
        else
        {return false;}
    }
 
    return true;
}
 
template<typename Decomposition, bool ghost_or_domain>
static inline bool compute_subset(const openfpm::vector<GBoxes<Decomposition::dims>> & gdb_ext,
        size_t g_c,
        grid_key_dx<Decomposition::dims> & start,
        grid_key_dx<Decomposition::dims> & stop,
        grid_key_dx<Decomposition::dims> & start_c,
        grid_key_dx<Decomposition::dims> & stop_c)
{
    if (ghost_or_domain == false)
    {return compute_subset_domain<Decomposition>(gdb_ext,g_c,start,stop,start_c,stop_c);}
 
    return compute_subset_ghost<Decomposition>(gdb_ext,g_c,start,stop,start_c,stop_c);
}
 
#endif /* SRC_GRID_ITERATORS_GRID_DIST_ID_ITERATOR_UTIL_HPP_ */