point_parallel_hash_grid_searcher2.h

Go to the documentation of this file.

// Copyright (c) 2018 Doyub Kim
//
// I am making my contributions/submissions to this project solely in my
// personal capacity and am not conveying any rights to any intellectual
// property of any third parties.
 
#ifndef INCLUDE_JET_POINT_PARALLEL_HASH_GRID_SEARCHER2_H_
#define INCLUDE_JET_POINT_PARALLEL_HASH_GRID_SEARCHER2_H_
 
#include <jet/point_neighbor_searcher2.h>
#include <jet/point2.h>
#include <jet/size2.h>
#include <vector>
 
class PointParallelHashGridSearcher2Tests;
 
namespace jet {
 
class PointParallelHashGridSearcher2 final : public PointNeighborSearcher2 {
 public:
    JET_NEIGHBOR_SEARCHER2_TYPE_NAME(PointParallelHashGridSearcher2)
 
    class Builder;
 
    PointParallelHashGridSearcher2(
        const Size2& resolution, double gridSpacing);
 
    PointParallelHashGridSearcher2(
        size_t resolutionX,
        size_t resolutionY,
        double gridSpacing);
 
    PointParallelHashGridSearcher2(const PointParallelHashGridSearcher2& other);
 
    void build(const ConstArrayAccessor1<Vector2D>& points) override;
 
    void forEachNearbyPoint(
        const Vector2D& origin,
        double radius,
        const ForEachNearbyPointFunc& callback) const override;
 
    bool hasNearbyPoint(
        const Vector2D& origin, double radius) const override;
 
    const std::vector<size_t>& keys() const;
 
    const std::vector<size_t>& startIndexTable() const;
 
    const std::vector<size_t>& endIndexTable() const;
 
    const std::vector<size_t>& sortedIndices() const;
 
    size_t getHashKeyFromBucketIndex(const Point2I& bucketIndex) const;
 
    Point2I getBucketIndex(const Vector2D& position) const;
 
    PointNeighborSearcher2Ptr clone() const override;
 
    PointParallelHashGridSearcher2& operator=(
        const PointParallelHashGridSearcher2& other);
 
    void set(const PointParallelHashGridSearcher2& other);
 
    void serialize(std::vector<uint8_t>* buffer) const override;
 
    void deserialize(const std::vector<uint8_t>& buffer) override;
 
    static Builder builder();
 
 private:
    friend class PointParallelHashGridSearcher2Tests;
 
    double _gridSpacing = 1.0;
    Point2I _resolution = Point2I(1, 1);
    std::vector<Vector2D> _points;
    std::vector<size_t> _keys;
    std::vector<size_t> _startIndexTable;
    std::vector<size_t> _endIndexTable;
    std::vector<size_t> _sortedIndices;
 
    size_t getHashKeyFromPosition(const Vector2D& position) const;
 
    void getNearbyKeys(const Vector2D& position, size_t* bucketIndices) const;
};
 
typedef std::shared_ptr<PointParallelHashGridSearcher2>
    PointParallelHashGridSearcher2Ptr;
 
class PointParallelHashGridSearcher2::Builder final
    : public PointNeighborSearcherBuilder2 {
 public:
    Builder& withResolution(const Size2& resolution);
 
    Builder& withGridSpacing(double gridSpacing);
 
    PointParallelHashGridSearcher2 build() const;
 
    PointParallelHashGridSearcher2Ptr makeShared() const;
 
    PointNeighborSearcher2Ptr buildPointNeighborSearcher() const override;
 
 private:
    Size2 _resolution{64, 64};
    double _gridSpacing = 1.0;
};
 
}  // namespace jet
 
#endif  // INCLUDE_JET_POINT_PARALLEL_HASH_GRID_SEARCHER2_H_