CellNNIteratorM.hpp

/*
 * CellNNIteratorM.hpp
 *
 *  Created on: Jun 23, 2016
 *      Author: i-bird
 */

#ifndef OPENFPM_DATA_SRC_NN_CELLLIST_CELLNNITERATORM_HPP_
#define OPENFPM_DATA_SRC_NN_CELLLIST_CELLNNITERATORM_HPP_


#include "util/mathutil.hpp"
#include <boost/integer/integer_mask.hpp>
#include "CellList_util.hpp"
#include "CellNNIterator.hpp"

template<unsigned int dim, typename Cell, unsigned int sh_byte, int NNc_size, unsigned int impl>
class CellNNIteratorSymM : public CellNNIterator<dim,Cell,NNc_size,impl>
{
    typedef boost::low_bits_mask_t<sizeof(size_t)*8-sh_byte>  mask_low;

    size_t pp;

    size_t p;

    const typename Cell::internal_vector_pos_type & pos;

    const typename openfpm::vector<pos_v<typename Cell::internal_vector_pos_type>> & ps;

    inline void selectValid()
    {
        if (this->NNc[this->NNc_id] == 0)
        {
            while (this->start_id < this->stop_id)
            {
                size_t q = this->cl.get_lin(this->start_id);
                for (long int i = dim-1 ; i >= 0 ; i--)
                {
                    if (pos.template get<0>(p)[i] < ps.get(q >> (sizeof(size_t)*8-sh_byte)).pos.template get<0>(q & mask_low::sig_bits_fast)[i])
                        return;
                    else if (pos.template get<0>(p)[i] > ps.get(q >> (sizeof(size_t)*8-sh_byte)).pos.template get<0>(q & mask_low::sig_bits_fast)[i])
                        goto next;
                }
                if (q >> (sizeof(size_t)*8-sh_byte) != pp)  return;
                if ((q & mask_low::sig_bits_fast) >= p) return;
next:
                this->start_id++;
            }

            CellNNIterator<dim,Cell,NNc_size,impl>::selectValid();
        }
        else
        {
            CellNNIterator<dim,Cell,NNc_size,impl>::selectValid();
        }
    }

public:

    CellNNIteratorSymM(size_t cell,
                       size_t pp,
                       size_t p,
                       const NNc_array<dim,NNc_size> & NNc,
                       Cell & cl,
                       const typename Cell::internal_vector_pos_type & pos,
                       const typename openfpm::vector<pos_v<typename Cell::internal_vector_pos_type>> & ps)
    :CellNNIterator<dim,Cell,NNc_size,impl>(cell,NNc,cl),pp(pp),p(p),pos(pos),ps(ps)
    {
        selectValid();
    }


    inline size_t getP()
    {
        return CellNNIterator<dim,Cell,NNc_size,impl>::get() & mask_low::sig_bits_fast;
    }

    inline size_t getV()
    {
        return (CellNNIterator<dim,Cell,NNc_size,impl>::get()) >> (sizeof(size_t)*8-sh_byte);
    }

    inline CellNNIteratorSymM<dim,Cell,sh_byte,NNc_size,impl> & operator++()
    {
        this->start_id++;

        selectValid();

        return *this;
    }
};

template<unsigned int dim, typename Cell, unsigned int sh_byte, int NNc_size, unsigned int impl> class CellNNIteratorM : public CellNNIterator<dim,Cell,NNc_size,impl>
{
    typedef boost::low_bits_mask_t<sizeof(size_t)*8-sh_byte>  mask_low;

public:

    CellNNIteratorM(size_t cell, const NNc_array<dim,NNc_size> &NNc, Cell & cl)
    :CellNNIterator<dim,Cell,NNc_size,impl>(cell,NNc,cl)
    {}


    inline size_t getP()
    {
        return CellNNIterator<dim,Cell,NNc_size,impl>::get() & mask_low::sig_bits_fast;
    }

    inline size_t getV()
    {
        return (CellNNIterator<dim,Cell,NNc_size,impl>::get()) >> (sizeof(size_t)*8-sh_byte);
    }
};

template<typename Cell, unsigned int sh_byte> class CellIteratorM : public CellIterator<Cell>
{

    typedef boost::low_bits_mask_t<sizeof(size_t)*8-sh_byte>  mask_low;

public:

    CellIteratorM(const size_t cell, Cell & cl)
    :CellIterator<Cell>(cell,cl)
    {}

    inline size_t getP()
    {
        return CellIterator<Cell>::get() & mask_low::sig_bits_fast;
    }

    inline size_t getV()
    {
        return (CellIterator<Cell>::get()) >> (sizeof(size_t)*8-sh_byte);
    }
};

#include "CellNNIteratorRuntimeM.hpp"

#endif /* OPENFPM_DATA_SRC_NN_CELLLIST_CELLNNITERATORM_HPP_ */