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