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