PointKdTreeSearcher3.h

Go to the documentation of this file.

/*************************************************************************
> File Name: PointKdTreeSearcher3.h
> Project Name: CubbyFlow
> Author: Chan-Ho Chris Ohk
> Purpose: KdTree-based 3-D point searcher.
> Created Time: 2017/12/05
> Copyright (c) 2018, Chan-Ho Chris Ohk
*************************************************************************/
#ifndef CUBBYFLOW_POINT_KDTREE_SEARCHER3_H
#define CUBBYFLOW_POINT_KDTREE_SEARCHER3_H

#include <Core/Geometry/KdTree.h>
#include <Core/Searcher/PointNeighborSearcher3.h>

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

        class Builder;

        PointKdTreeSearcher3();

        PointKdTreeSearcher3(const PointKdTreeSearcher3& 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;

        PointNeighborSearcher3Ptr Clone() const override;

        PointKdTreeSearcher3& operator=(const PointKdTreeSearcher3& other);

        void Set(const PointKdTreeSearcher3& other);

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

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

        static Builder GetBuilder();

    private:
        KdTree<double, 3> m_tree;
    };

    using PointKdTreeSearcher3Ptr = std::shared_ptr<PointKdTreeSearcher3>;

    class PointKdTreeSearcher3::Builder final : public PointNeighborSearcherBuilder3
    {
    public:
        PointKdTreeSearcher3 Build() const;

        PointKdTreeSearcher3Ptr MakeShared() const;

        PointNeighborSearcher3Ptr BuildPointNeighborSearcher() const override;
    };
}

#endif