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