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_ */