CellListNNIteratorRadius.hpp

/*
 * CellListNNIteratorRadius.hpp
 *
 *  Created on: Aug 17, 2016
 *      Author: i-bird
 */

#ifndef OPENFPM_DATA_SRC_NN_CELLLIST_CELLLISTNNITERATORRADIUS_HPP_
#define OPENFPM_DATA_SRC_NN_CELLLIST_CELLLISTNNITERATORRADIUS_HPP_

#include "Vector/map_vector.hpp"

template<unsigned int dim, typename Cell, unsigned int impl> class CellNNIteratorRadius
{
    // Cell list
    Cell & cl;

    // Actual NNc_id;
    size_t NNc_id;

    // actual cell id = NNc[NNc_id]+cell stored for performance reason
    size_t cell_id;

    // actual element id
    size_t ele_id;

    // NN number of neighborhood cells
    const openfpm::vector<long int> & NNc;

    // Center cell, or cell for witch we are searching the NN-cell
    const long int cell;

    inline void selectValid()
    {
        while (ele_id >= cl.getNelements(cell_id))
        {
            NNc_id++;

            // No more Cell
            if (NNc_id >= NNc.size()) return;

            cell_id = NNc.get(NNc_id) + cell;

            ele_id = 0;
        }
    }

public:

    inline CellNNIteratorRadius(size_t cell, const openfpm::vector<long int> &NNc, Cell & cl)
    :cl(cl),NNc_id(0),cell_id(NNc.get(NNc_id) + cell),ele_id(0),NNc(NNc),cell(cell)
    {
#ifdef SE_CLASS1
        if (cell_id < 0)
            std::cerr << "Error " << __FILE__ ":" << __LINE__ << " cell_id is negative, please check the the padding is chosen correctly." <<
                                                                  "Remember, if you choose a radius that span N neighborhood cell-list, padding must be one" << std::endl;
#endif

        selectValid();
    }

    inline bool isNext()
    {
        if (NNc_id >= NNc.size())
            return false;
        return true;
    }

    inline CellNNIteratorRadius & operator++()
    {
        ele_id++;

        selectValid();

        return *this;
    }

    inline typename Cell::value_type & get()
    {
        return cl.get(cell_id,ele_id);
    }
};


#endif /* OPENFPM_DATA_SRC_NN_CELLLIST_CELLLISTNNITERATORRADIUS_HPP_ */