PointParallelHashGridSearcher3.h

Go to the documentation of this file.

/*************************************************************************
> File Name: PointParallelHashGridSearcher3.h
> Project Name: CubbyFlow
> Author: Chan-Ho Chris Ohk
> Purpose: Parallel version of hash grid-based 3-D point searcher.
> Created Time: 2017/05/07
> Copyright (c) 2018, Chan-Ho Chris Ohk
*************************************************************************/
#ifndef CUBBYFLOW_POINT_PARALLEL_HASH_GRID_SEARCHER3_H
#define CUBBYFLOW_POINT_PARALLEL_HASH_GRID_SEARCHER3_H

#include <Core/Searcher/PointNeighborSearcher3.h>
#include <Core/Size/Size3.h>

namespace CubbyFlow
{
    class PointParallelHashGridSearcher3 final : public PointNeighborSearcher3
    {
    public:
        CUBBYFLOW_NEIGHBOR_SEARCHER3_TYPE_NAME(PointParallelHashGridSearcher3)

        class Builder;

        PointParallelHashGridSearcher3(const Size3& resolution, double gridSpacing);

        PointParallelHashGridSearcher3(size_t resolutionX, size_t resolutionY, size_t resolutionZ, double gridSpacing);

        PointParallelHashGridSearcher3(const PointParallelHashGridSearcher3& other);

        void Build(const ConstArrayAccessor1<Vector3D>& points) override;

        void ForEachNearbyPoint(const Vector3D& origin, double radius, const ForEachNearbyPointFunc& callback) const override;

        bool HasNearbyPoint(const Vector3D& 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 Point3I& bucketIndex) const;

        Point3I GetBucketIndex(const Vector3D& position) const;

        PointNeighborSearcher3Ptr Clone() const override;

        PointParallelHashGridSearcher3& operator=(const PointParallelHashGridSearcher3& other);

        void Set(const PointParallelHashGridSearcher3& other);

        void Serialize(std::vector<uint8_t>* buffer) const override;

        void Deserialize(const std::vector<uint8_t>& buffer) override;

        static Builder GetBuilder();

    private:
        double m_gridSpacing = 1.0;
        Point3I m_resolution = Point3I(1, 1, 1);
        std::vector<Vector3D> m_points;
        std::vector<size_t> m_keys;
        std::vector<size_t> m_startIndexTable;
        std::vector<size_t> m_endIndexTable;
        std::vector<size_t> m_sortedIndices;

        size_t GetHashKeyFromPosition(const Vector3D& position) const;

        void GetNearbyKeys(const Vector3D& position, size_t* bucketIndices) const;
    };

    using PointParallelHashGridSearcher3Ptr = std::shared_ptr<PointParallelHashGridSearcher3>;

    class PointParallelHashGridSearcher3::Builder final : public PointNeighborSearcherBuilder3
    {
    public:
        Builder& WithResolution(const Size3& resolution);

        Builder& WithGridSpacing(double gridSpacing);

        PointParallelHashGridSearcher3 Build() const;

        PointParallelHashGridSearcher3Ptr MakeShared() const;

        PointNeighborSearcher3Ptr BuildPointNeighborSearcher() const override;

    private:
        Size3 m_resolution{ 64, 64, 64 };
        double m_gridSpacing = 1.0;
    };
}

#endif