ProcKeys.hpp

/*
 * ProcKeys.hpp
 *
 *  Created on: Mar 14, 2017
 *      Author: i-bird
 */
 
#ifndef OPENFPM_DATA_SRC_NN_CELLLIST_PROCKEYS_HPP_
#define OPENFPM_DATA_SRC_NN_CELLLIST_PROCKEYS_HPP_
 
extern "C"
{
#include "hilbertKey.h"
}
 
/* !Brief Class for a linear (1D-like) order processing of cell keys for CellList_gen implementation
 *
 * \tparam dim Dimansionality of the space
 */
template<unsigned int dim, typename CellList>
class Process_keys_lin
{
    openfpm::vector<size_t> keys;
 
public:
 
    typedef ParticleIt_CellP<CellList> Pit;
 
    inline const openfpm::vector<size_t> & getKeys() const
    {
        return keys;
    }
 
    template<typename S> void get_hkey(S & obj, grid_key_dx<dim> gk, size_t m)
    {
        size_t point[dim];
 
        for (size_t i = 0; i < dim; i++)
        {
            point[i] = gk.get(i) + obj.getPadding(i);
        }
 
        keys.add(obj.getGrid().LinIdPtr(static_cast<size_t *>(point)));
    }
 
    template<typename S> void linearize_hkeys(S & obj, size_t m)
    {
        return;
    }
};
 
template<unsigned int dim, typename CellList>
class Process_keys_hilb
{
    openfpm::vector<size_t> keys;
 
    openfpm::vector<size_t> p_keys;
 
    size_t m = 0;
 
public:
 
    typedef ParticleIt_CellP<CellList> Pit;
 
    inline const openfpm::vector<size_t> & getKeys() const
    {
        return keys;
    }
 
    template<typename S> inline void get_hkey(S & obj, grid_key_dx<dim> gk, size_t m)
    {
        //An integer to handle errors
        int err;
 
        uint64_t point[dim];
 
        for (size_t i = 0; i < dim; i++)
        {
            point[i] = gk.get(i);
        }
 
        size_t hkey = getHKeyFromIntCoord(m, dim, point, &err);
 
        keys.add(hkey);
    }
 
    template<typename S> inline void linearize_hkeys(S & obj, size_t m)
    {
        //An integer to handle errors
        int err;
 
        //Array to handle output
        uint64_t coord[dim];
        size_t coord2[dim];
 
        keys.sort();
 
        openfpm::vector<size_t> keys_new;
 
        for(size_t i = 0; i < obj.getKeys().size(); i++)
        {
            getIntCoordFromHKey(coord, m, dim, obj.getKeys().get(i), &err);
 
            for (size_t j = 0 ; j < dim ; j++)  {coord2[j] = coord[j] + obj.getPadding(j);}
 
            keys_new.add(obj.getGrid().LinIdPtr(static_cast<size_t *>(coord2)));
        }
 
        keys.swap(keys_new);
    }
};
 
template<unsigned int dim, typename CellList>
class Process_keys_grid
{
    openfpm::vector<size_t> keys;
 
    openfpm::vector<size_t> p_keys;
 
    size_t m = 0;
 
public:
 
    typedef ParticleIt_CellP<CellList> Pit;
 
    inline const openfpm::vector<size_t> & getKeys() const
    {
        return keys;
    }
 
    template<typename S> inline void get_hkey(S & obj, grid_key_dx<dim> gk, size_t m)
    {
        uint64_t point[dim];
 
        for (size_t i = 0; i < dim; i++)
        {
            point[i] = gk.get(i);
        }
 
        size_t hkey = obj.getGrid().LinId(gk);
 
        keys.add(hkey);
    }
 
    template<typename S> inline void linearize_hkeys(S & obj, size_t m)
    {
    }
};
 
#endif /* OPENFPM_DATA_SRC_NN_CELLLIST_PROCKEYS_HPP_ */