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 "CellList.hpp"
#include "CellNNIteratorM.hpp"
 
struct PV_cl
{
    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 typename CellBase::internal_vector_pos_type & pos,
                     const openfpm::vector<pos_v<typename CellBase::internal_vector_pos_type>> & 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_ */