PointIterator.hpp

/*
 * PointIterator.hpp
 *
 *  Created on: Jan 3, 2017
 *      Author: i-bird
 */
 
#ifndef OPENFPM_NUMERICS_SRC_DRAW_POINTITERATOR_HPP_
#define OPENFPM_NUMERICS_SRC_DRAW_POINTITERATOR_HPP_
 
 
template<unsigned int dim, typename T, typename Decomposition>
class PointIterator: protected grid_dist_id_iterator_dec<Decomposition>
{
    Point<dim,T> ap;
 
    Box<dim,T> sub_domain;
 
    Box<dim,T> domain;
 
    T sp[dim];
 
    static grid_key_dx<dim> getStart(size_t (& gs)[dim], const Box<dim,T> & dom, Box<dim,T> & sub_dom, T (& sp)[dim])
    {
        for (size_t i = 0 ; i < dim ; i++)
            sp[i] = (dom.getHigh(i) - dom.getLow(i)) / (gs[i] -1);
 
        grid_key_dx<dim> pkey;
 
        for (size_t i = 0 ; i < dim ; i++)
            pkey.set_d(i,std::ceil( (sub_dom.getLow(i) - dom.getLow(i)) / sp[i]));
 
        return pkey;
    }
 
    static grid_key_dx<dim> getStop(size_t (& gs)[dim], const Box<dim,T> & dom, Box<dim,T> & sub_dom, T (& sp)[dim])
    {
        for (size_t i = 0 ; i < dim ; i++)
            sp[i] = (dom.getHigh(i) - dom.getLow(i)) / (gs[i] - 1);
 
        grid_key_dx<dim> pkey;
 
        for (size_t i = 0 ; i < dim ; i++)
            pkey.set_d(i,std::floor( (sub_dom.getHigh(i) - dom.getLow(i)) / sp[i]));
 
        return pkey;
    }
 
    void calculateAp()
    {
        if (grid_dist_id_iterator_dec<Decomposition>::isNext() == false)
            return;
 
        grid_key_dx<dim> key = grid_dist_id_iterator_dec<Decomposition>::get();
 
        for (size_t i = 0 ; i < dim ; i++)
            ap.get(i) = key.get(i) * sp[i] + domain.getLow(i);
    }
 
public:
 
    PointIterator( Decomposition & dec, size_t (& sz)[dim], const Box<dim,T> & domain, Box<dim,T> & sub_domain)
    :grid_dist_id_iterator_dec<Decomposition>(dec, sz, getStart(sz,domain,sub_domain,sp), getStop(sz,domain,sub_domain,sp)),sub_domain(sub_domain),domain(domain)
    {
        calculateAp();
    }
 
    Point<dim,T> & get()
    {
        return ap;
    }
 
    PointIterator & operator++()
    {
        grid_dist_id_iterator_dec<Decomposition>::operator++();
 
        calculateAp();
 
        return *this;
    }
 
    bool isNext()
    {
        return grid_dist_id_iterator_dec<Decomposition>::isNext();
    }
 
    Box<dim,T> getBoxMargins()
    {
        Box<dim,T> box;
 
        grid_key_dx<dim> start = grid_dist_id_iterator_dec<Decomposition>::getStart();
        grid_key_dx<dim> stop = grid_dist_id_iterator_dec<Decomposition>::getStop();
 
        for (size_t i = 0 ; i < dim ; i++)
        {
            box.setLow(i, start.get(i)*sp[i] + domain.getLow(i));
            box.setHigh(i, stop.get(i)*sp[i] + domain.getLow(i));
        }
 
        return box;
    }
};
 
 
#endif /* OPENFPM_NUMERICS_SRC_DRAW_POINTITERATOR_HPP_ */