CellListIterator.hpp

 
/*
 * CellListIt.hpp
 *
 *  Created on: May 18, 2016
 *      Author: Yaroslav Zaluzhnyi
 */
 
#ifndef OPENFPM_DATA_SRC_NN_CELLLIST_CELLLISTITERATOR_HPP_
#define OPENFPM_DATA_SRC_NN_CELLLIST_CELLLISTITERATOR_HPP_
 
template<typename T>
class ParticleIt_CellP
{
private:
 
    T & NN;
 
    size_t cell_count;
 
    long int p_count;
 
 
    inline void fp()
    {
        ++p_count;
 
        const auto & SFCkeys = NN.getCellSFC().getKeys();
 
        if (p_count >= (long int)NN.getNelements(SFCkeys.get(cell_count)))
        {
            p_count = 0;
            ++cell_count;
 
            while (cell_count < SFCkeys.size() && NN.getNelements(SFCkeys.get(cell_count)) == 0)
                ++cell_count;
        }
    }
 
public:
 
    // Constructor
    ParticleIt_CellP(T & NN)
    :NN(NN)
    {
        reset();
    }
 
    // Destructor
    ~ParticleIt_CellP()
    {
    }
 
 
 
    inline ParticleIt_CellP & operator++()
    {
        fp();
        while (isNext() && this->get() >= NN.get_gm())
        {
            fp();
        }
 
        return *this;
    }
 
 
    inline bool isNext()
    {
        if (cell_count >= NN.getCellSFC().getKeys().size())
        {
            return false;
        }
 
        return true;
    }
 
 
    inline size_t get()
    {
        auto cell_id = NN.getCellSFC().getKeys().get(cell_count);
        auto p = NN.get(cell_id,p_count);
 
        return p;
    }
 
    void reset()
    {
        cell_count = 0;
        p_count = -1;
 
        this->operator++();
    }
};
 
 
#endif /* OPENFPM_DATA_SRC_NN_CELLLIST_CELLLISTITERATOR_HPP_ */