ParticleIt_Cells.hpp

/*
 * ParticleIt_Cells.hpp
 *
 *  Created on: Mar 5, 2017
 *      Author: i-bird
 */

#ifndef OPENFPM_DATA_SRC_NN_CELLLIST_PARTICLEIT_CELLS_HPP_
#define OPENFPM_DATA_SRC_NN_CELLLIST_PARTICLEIT_CELLS_HPP_

template<unsigned int dim,typename CellListType> class ParticleIt_Cells
{
private:

    const size_t * start;

    const size_t * stop;

    size_t cid;

    const openfpm::vector<size_t> &dom_cell;

    CellListType & cli;

    size_t g_m;

    void selectValid()
    {
        while (start == stop)
        {
            cid++;

            size_t s_cell;
            if (cid >= dom_cell.size())
                return;
            else
                s_cell = dom_cell.get(cid);

            // Get the starting particle
            start = &cli.getStartId(s_cell);

            // Get the stop particle
            stop = &cli.getStopId(s_cell);
        }
    }

public:

    ParticleIt_Cells(CellListType & cli,
                     const openfpm::vector<size_t> & dom_cell,
                     size_t g_m)
    :cid(0),dom_cell(dom_cell),cli(cli),g_m(g_m)
    {
        size_t s_cell;
        if (dom_cell.size() != 0)
            s_cell = dom_cell.get(0);
        else
        {
            cid = 1;
            start = NULL;
            stop = NULL;

            return;
        }

        // Get the starting particle
        start = &cli.getStartId(s_cell);

        // Get the stop particle
        stop = &cli.getStopId(s_cell);

        selectValid();

        while (*start >= g_m)
        {
            ++start;
            selectValid();
        }
    }

    ParticleIt_Cells & operator++()
    {
        ++start;
        selectValid();

        while (*start >= g_m)
        {
            ++start;
            selectValid();
        }

        return *this;
    }

    bool isNext()
    {
        return cid < dom_cell.size();
    }

    size_t get()
    {
        return *start;
    }
};


#endif /* OPENFPM_DATA_SRC_NN_CELLLIST_PARTICLEIT_CELLS_HPP_ */