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_ */