imstkNeighborSearch.h

/*
** This file is part of the Interactive Medical Simulation Toolkit (iMSTK)
** iMSTK is distributed under the Apache License, Version 2.0.
** See accompanying NOTICE for details.
*/

#pragma once

#include "imstkMath.h"
#include "imstkVecDataArray.h"

namespace imstk
{
class GridBasedNeighborSearch;
class SpatialHashTableSeparateChaining;

class NeighborSearch
{
public:
    enum class Method
    {
        UniformGridBasedSearch,
        SpatialHashing
    };

    NeighborSearch(Method searchMethod, double searchRadius = 0.0);

    void setSearchRadius(const double searchRadius);

    double getSearchRadius() const { return m_SearchRadius; }

    std::vector<std::vector<size_t>> getNeighbors(const VecDataArray<double, 3>& points);

    void getNeighbors(std::vector<std::vector<size_t>>& result, const VecDataArray<double, 3>& points);

    void getNeighbors(std::vector<std::vector<size_t>>& result, const VecDataArray<double, 3>& setA, const VecDataArray<double, 3>& setB);

private:
    Method m_Method;
    double m_SearchRadius = 0.0;

    std::shared_ptr<GridBasedNeighborSearch> m_GridBasedSearcher;
    std::shared_ptr<SpatialHashTableSeparateChaining> m_SpatialHashSearcher;
};
} // namespace imstk