SpatialHash.h

Go to the documentation of this file.

/*
 * SpatialHash.h
 *
 *  Created on: Nov 27, 2019
 *      Author: Nikhil
 */
 
#ifndef SRC_SPATIALHASH_H_
#define SRC_SPATIALHASH_H_
 
#include <vector>
#include <iostream>
#include <map>
#include <vector>
#include "range.h"
#include "typedef.h"
 
class Triplet: public Vector3i
{
public:
    Triplet()=default;
    Triplet(int i,int j,int k): Vector3i(i,j,k) {}
    Triplet(const Vector3i& base) : Vector3i(base){}
 
    // Define < operator to use std::map<Triplet,std::vector<int>>
    bool operator <(const Triplet& rhs)  const
    {
        if (this->operator()(0) < rhs(0))   return true;
        if (rhs(0)   < this->operator()(0)) return false;
        if (this->operator()(1) < rhs(1))   return true;
        if (rhs(1)   < this->operator()(1)) return false;
        if (this->operator()(2) < rhs(2))   return true;
        return false;
    }
 
    std::vector<Triplet> neighborList()
    {
        std::vector<Triplet> tripletList;
        for (auto i : range<int>(-1,2))
            for (auto j : range<int>(-1,2))
                for (auto k : range<int>(-1,2))
                {
                    Triplet s= static_cast<Vector3i>(*this + Triplet(i,j,k));
                    tripletList.push_back(s);
                }
        return tripletList;
    }
 
    virtual ~Triplet(){}
};
 
 
template<bool flag, typename T, typename U>
struct Select
{
    typedef T Result;
};
 
template<typename T, typename U>
struct Select<false, T, U> { typedef U Result; };
 
template<bool isConst>
class SpatialHash {
public:
    typedef typename Select<isConst, const MatrixXd, MatrixXd>::Result A;
    typedef typename Select<isConst, const Vector3d, Vector3d>::Result B;
    typedef typename Select<isConst, const std::vector<Vector3d>, std::vector<Vector3d>>::Result C;
 
    Vector3d origin;
    Vector3d step;
    std::vector<Eigen::Map<B>> coordinates;
    std::map<Triplet,std::vector<int>> hashTable;
 
    SpatialHash();
    SpatialHash(Vector3d origin,
                Vector3d step,
                A& coordinates): origin(origin),step(step)
    {
        int size= coordinates.rows();
        for(int i_point=0; i_point<size; i_point++)
        {
            Eigen::Map<B> vector(&coordinates(i_point,0),3);
            this->coordinates.push_back(vector);
            hashTable[hashFunction(i_point)].push_back(i_point);
        }
    }
    SpatialHash(Vector3d origin,
                Vector3d step,
                C& coordinates) : origin(origin),step(step)
    {
        int size= coordinates.size();
        for(int i_point=0; i_point<size; i_point++)
        {
            Eigen::Map<B> vector(&coordinates[i_point](0),3);
            this->coordinates.push_back(vector);
            hashTable[hashFunction(i_point)].push_back(i_point);
        }
 
    }
 
    Triplet hashFunction(const int& i) const
    {
        assert(!(i<0));
        Triplet triple;
        Vector3i& base= triple;
        Vector3d coordinate= coordinates[i];
        coordinate= coordinate-origin;
        coordinate= (coordinate.array()/step.array()).floor().matrix();
        base= coordinate.template cast<int>();
        return triple;
    }
 
    virtual ~SpatialHash(){}
 
};
typedef SpatialHash<true> ConstSpatialHash;
 
class BoxPoints : public SpatialHash<false>
{
public:
    BoxPoints(Vector3d origin,
              Vector3d step,
              MatrixXd& coordinates):SpatialHash<false>(origin,step,coordinates){}
    BoxPoints(Vector3d origin,
              Vector3d step,
              std::vector<Vector3d>& coordinates) : SpatialHash<false>(origin,step,coordinates) {}
 
    virtual ~BoxPoints(){}
 
    void fold(const Vector3i& pbc)
    {
        int size= coordinates.size();
        for(auto i_point : range<int>(0,size))
        {
            if (hashFunction(i_point) != Triplet(0,0,0))
            {
                std::cout << "Folding point " << i_point << ": "
                          << coordinates[i_point] << " ---> ";
                Vector3d shift;
                shift= ((hashFunction(i_point).template cast<double>()).array()*step.array()).matrix();
                shift= (shift.array()* pbc.template cast<double>().array()).matrix();
                coordinates[i_point]= coordinates[i_point] - shift;
                std::cout << coordinates[i_point] << std::endl;
            }
        }
    }
};
 
#endif /* SRC_SPATIALHASH_H_ */