CellList_cpu_ker.cuh

/*
 * CellList_gpu_ker.cuh
 *
 *  Created on: Jul 30, 2018
 *      Author: i-bird
 */
 
#ifndef CELLLIST_CPU_KER_CUH_
#define CELLLIST_CPU_KER_CUH_
 
#include "Cuda_cell_list_util_func.hpp"
 
template<unsigned int dim, typename T, typename Mem_type, typename transform_type>
class CellList_cpu_ker: Mem_type
{
    typedef typename Mem_type::local_index_type ids_type;
 
    openfpm::array<T,dim> spacing_c;
 
    openfpm::array<ids_type,dim> div_c;
 
    openfpm::array<ids_type,dim> off;
 
    transform_type pointTransform;
 
    grid_sm<dim,void> cellListGrid;
 
    Point<dim,long int> cellShift;
 
    Box<dim,T> cellListSpaceBox;
 
public:
 
    CellList_cpu_ker(const Mem_type & mt,
                    openfpm::array<T,dim> & spacing_c,
                    openfpm::array<ids_type,dim> & div_c,
                    openfpm::array<ids_type,dim> & off,
                    grid_sm<dim,void> & cellListGrid,
                    Point<dim,long int> & cellShift,
                    Box<dim,T> & cellListSpaceBox,
                    const transform_type & pointTransform)
    : Mem_type(mt),
    spacing_c(spacing_c),
    div_c(div_c),
    off(off),
    pointTransform(pointTransform),
    cellListGrid(cellListGrid),
    cellShift(cellShift),
    cellListSpaceBox(cellListSpaceBox)
    {}
 
    inline __device__ unsigned int getCell(const Point<dim,T> & xp) const
    {
        return cid_<dim,ids_type,transform_type>::get_cid(div_c,spacing_c,off,pointTransform,xp);
    }
 
    const __device__ grid_sm<dim,void> & getGrid() const
    {
        return cellListGrid;
    }
 
    inline __device__ grid_key_dx<dim> getCellGrid(const Point<dim,T> & pos) const
    {
        grid_key_dx<dim> key;
        key.set_d(0,ConvertToID(pos,0));
 
        for (size_t s = 1 ; s < dim ; s++)
        {
            key.set_d(s,ConvertToID(pos,s));
        }
 
        return key;
    }
 
    inline __device__ grid_key_dx<dim> getCellGrid(const T (& pos)[dim]) const
    {
        grid_key_dx<dim> key;
        key.set_d(0,ConvertToID(pos,0));
 
        for (size_t s = 1 ; s < dim ; s++)
        {
            key.set_d(s,ConvertToID(pos,s));
        }
 
        return key;
    }
 
    inline __device__ size_t ConvertToID(const T (&x)[dim], size_t s) const
    {
        size_t id = openfpm::math::size_t_floor(pointTransform.transform(x,s) / cellListSpaceBox.getHigh(s)) + off[s];
        id = (id >= cellListGrid.size(s))?(cellListGrid.size(s)-1-cellShift.get(s)):id-cellShift.get(s);
        return id;
    }
 
    inline __device__ size_t ConvertToID(const Point<dim,T> & x, size_t s, size_t sc = 0) const
    {
        size_t id = openfpm::math::size_t_floor(pointTransform.transform(x,s) / cellListSpaceBox.getHigh(s)) + off[s];
        id = (id >= cellListGrid.size(s))?(cellListGrid.size(s)-1-cellShift.get(s)):id-cellShift.get(s);
        return id;
    }
 
    inline __device__ int getNelements(unsigned int cell) const
    {
        return Mem_type::getNelements(cell);
    }
 
    inline __device__  unsigned int get(unsigned int cell, unsigned int ele)
    {
        return Mem_type::get(cell,ele);
    }
 
};
 
 
#endif /* CELLLIST_GPU_KER_CUH_ */