grid_skin_iterator.hpp

/*
 * grid_skin_iterator.hpp
 *
 *  Created on: Jun 24, 2016
 *      Author: i-bird
 */

#ifndef OPENFPM_DATA_SRC_GRID_GRID_SKIN_ITERATOR_HPP_
#define OPENFPM_DATA_SRC_GRID_GRID_SKIN_ITERATOR_HPP_

#include "grid_key_dx_iterator_sub_bc.hpp"

template<unsigned int dim>
class grid_skin_iterator_bc
{
protected:
    grid_key_dx_iterator_sub_bc<dim> sub_it[2*dim];

private:

    size_t act;

public:

    template <typename T> grid_skin_iterator_bc(const grid_sm<dim,T> & g_sm, const Box<dim,size_t> & A, const Box<dim,size_t> & B, const size_t (& bc)[dim])
    :act(0)
    {
        for (size_t i = 0 ; i < dim ; i++)
        {
            grid_key_dx<dim> k1;
            grid_key_dx<dim> k2;

            for (size_t j = 0 ; j < dim ; j++)
            {
                if (j == i)
                {
                    k1.set_d(j,A.getHigh(j));
                    k2.set_d(j,B.getHigh(j));
                }
                else
                {
                    if ( j < i )
                    {
                        k1.set_d(j,A.getLow(j)+1);
                        k2.set_d(j,A.getHigh(j)-1);
                    }
                    else
                    {
                        k1.set_d(j,B.getLow(j));
                        k2.set_d(j,B.getHigh(j));
                    }
                }
            }

            // Initialize a box from the keys and check if it is a valid box
            Box<dim,long int> br(k1,k2);
            if (br.isValid() == true)
                sub_it[2*i].Initialize(g_sm,k1,k2,bc);

            for (size_t j = 0 ; j < dim ; j++)
            {
                if (j == i)
                {
                    k1.set_d(j,B.getLow(j));
                    k2.set_d(j,A.getLow(j));
                }
                else
                {
                    /* coverity[dead_error_line] */
                    if ( j < i )
                    {
                        k1.set_d(j,A.getLow(j)+1);
                        k2.set_d(j,A.getHigh(j)-1);
                    }
                    else
                    {
                        k1.set_d(j,B.getLow(j));
                        k2.set_d(j,B.getHigh(j));
                    }
                }
            }

            // Initialize a box from the keys and check if it is a valid box
            Box<dim,long int> bl(k1,k2);
            if (bl.isValid() == true && bl != br)
                sub_it[2*i+1].Initialize(g_sm,k1,k2,bc);
        }

        while (sub_it[act].isNext() == false)
            act++;
    }

    grid_skin_iterator_bc<dim> & operator++()
    {
        ++sub_it[act];

        while (act < 2*dim && sub_it[act].isNext() == false)
            act++;

        return *this;
    }

    bool isNext()
    {
        if (act < 2*dim)
        {

            return true;
        }

        return false;
    }

    inline grid_key_dx<dim> get() const
    {
        return sub_it[act].get();
    }

    void reset()
    {
        act = 0;

        for (size_t i = 0 ; i < 2*dim ; i++)
        {sub_it[i].reset();}
    }
};


#endif /* OPENFPM_DATA_SRC_GRID_GRID_SKIN_ITERATOR_HPP_ */