CellListM.hpp

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

#ifndef OPENFPM_DATA_SRC_NN_CELLLIST_CELLLISTM_HPP_
#define OPENFPM_DATA_SRC_NN_CELLLIST_CELLLISTM_HPP_

#include "CellNNIteratorM.hpp"
#include "CellList.hpp"

struct PV_cl
{
    size_t ele;
    size_t v;
};

template<unsigned int dim, typename T, unsigned int sh_byte, typename CellBase=CellList<dim,T,Mem_fast<>,shift<dim, T>> >
class CellListM : public CellBase
{
    typedef boost::high_bit_mask_t<sh_byte>  mask_high;

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

public:

    typedef CellNNIteratorSymM<dim,CellListM<dim,T,sh_byte,CellBase>,sh_byte,RUNTIME,NO_CHECK> SymNNIterator;

    CellListM()
    {};

    CellListM(const CellListM<dim,T,sh_byte,CellBase> & cell)
    {
        this->operator=(cell);
    }

    CellListM(CellListM<dim,T,sh_byte,CellBase> && cell)
    {
        this->operator=(cell);
    }


    CellListM(Box<dim,T> & box, const size_t (&div)[dim], Matrix<dim,T> mat, const size_t pad = 1, size_t slot=STARTING_NSLOT)
    :CellBase(box,div,mat,pad,slot)
    {}

    CellListM(Box<dim,T> & box, const size_t (&div)[dim], const size_t pad = 1, size_t slot=STARTING_NSLOT)
    :CellBase(box,div,pad,slot)
    {}


    ~CellListM()
    {}

    inline void addCell(size_t cell_id, size_t ele, size_t v_id)
    {
        size_t ele_k = ele | (v_id << (sizeof(size_t)*8-sh_byte));

        CellBase::addCell(cell_id,ele_k);
    }

    inline void add(const T (& pos)[dim], size_t ele, size_t v_id)
    {
        // calculate the Cell id

        size_t cell_id = this->getCell(pos);
        size_t ele_k = ele | (v_id << (sizeof(size_t)*8-sh_byte));

        // add the element to the cell

        CellBase::addCell(cell_id,ele_k);
    }

    inline void add(const Point<dim,T> & pos, size_t ele, size_t v_id)
    {
        // calculate the Cell id

        size_t cell_id = this->getCell(pos);
        size_t ele_k = ele | (v_id << (sizeof(size_t)*8-sh_byte));

        // add the element to the cell

        CellBase::addCell(cell_id,ele_k);
    }

    static inline size_t getP(size_t ele)
    {
        return ele & mask_low::sig_bits_fast;
    }

    static inline size_t getV(size_t ele)
    {
        return ele >> (sizeof(size_t)*8-sh_byte);
    }

    inline size_t getP(size_t cell, size_t ele)
    {
        return CellBase::get(cell,ele) & mask_low::sig_bits_fast;
    }

    inline size_t getV(size_t cell, size_t ele)
    {
        return (CellBase::get(cell,ele)) >> (sizeof(size_t)*8-sh_byte);
    }

    inline void swap(CellListM<dim,T,sh_byte,CellBase> & cl)
    {
        CellBase::swap(*this);
    }

    CellIterator<CellListM<dim,T,sh_byte,CellBase>> getIterator(size_t cell)
    {
        return CellBase::getIterator(cell);
    }

    template<unsigned int impl=NO_CHECK> inline CellNNIteratorM<dim,CellListM<dim,T,sh_byte,CellBase>,sh_byte,FULL,impl> getNNIterator(size_t cell)
    {
        CellNNIteratorM<dim,CellListM<dim,T,sh_byte,CellBase>,sh_byte,FULL,impl> cln(cell,CellListM<dim,T,sh_byte,CellBase>::NNc_full,*this);

        return cln;
    }


    template<unsigned int impl> inline CellNNIteratorSymM<dim,CellListM<dim,T,sh_byte,CellBase>,sh_byte,SYM,impl>
    getNNIteratorSym(size_t cell,
                     size_t pp,
                     size_t p,
                     const openfpm::vector<Point<dim,typename CellBase::stype>> & pos,
                     const openfpm::vector<pos_v<dim,typename CellBase::stype>> & v)
    {
        CellNNIteratorSymM<dim,CellListM<dim,T,sh_byte,CellBase>,sh_byte,SYM,impl> cln(cell,pp,p,CellListM<dim,T,sh_byte,CellBase>::NNc_sym,*this,pos,v);

        return cln;
    }


    CellListM<dim,T,sh_byte,CellBase> & operator=(CellListM<dim,T,sh_byte,CellBase> && clm)
    {
        CellBase::swap(clm);

        return *this;
    }

    CellListM<dim,T,sh_byte,CellBase> & operator=(const CellListM<dim,T,sh_byte,CellBase> & clm)
    {
        static_cast<CellBase *>(this)->operator=(*static_cast<const CellBase *>(&clm));

        return *this;
    }
};

#endif /* OPENFPM_DATA_SRC_NN_CELLLIST_CELLLISTM_HPP_ */