PointCloud.h

Go to the documentation of this file.

#pragma once
 
#include "objects/geometry/Box.h"
#include "objects/geometry/Sphere.h"
#include <unordered_map>
 
NAMESPACE_SPH_BEGIN
 
class PointCloud : public Noncopyable {
private:
    using Cell = Array<Vector>;
    using HashMap = std::unordered_map<Indices, Cell, std::hash<Indices>, IndicesEqual>;
 
    HashMap map;
    Size count = 0;
    Float cellSize;
 
public:
    class Handle {
    private:
        Indices data;
 
    public:
        Handle() = default;
 
        Handle(const Indices& idxs, const Size index, Badge<PointCloud>)
            : data(idxs) {
            data[3] = index;
        }
 
        Indices coords() const {
            return data;
        }
 
        Size index() const {
            return data[3];
        }
 
        bool operator==(const Handle& other) const {
            return all(coords() == other.coords()) && index() == other.index();
        }
 
        bool operator!=(const Handle& other) const {
            return !(*this == other);
        }
    };
 
    explicit PointCloud(const Float cellSize)
        : cellSize(cellSize) {}
 
    Handle push(const Vector& p);
 
    void push(ArrayView<const Vector> points);
 
    Vector point(const Handle& handle) const;
 
    Array<Vector> array() const;
 
    Size size() const;
 
    Size getClosePointsCount(const Vector& center, const Float radius) const;
 
    void findClosePoints(const Vector& center, const Float radius, Array<Handle>& handles) const;
 
    void findClosePoints(const Vector& center, const Float radius, Array<Vector>& neighs) const;
 
private:
    template <typename TAdd>
    void findClosePoints(const Vector& center, const Float radius, const TAdd& add) const;
 
    INLINE Box cellBox(const Indices& idxs) const {
        return Box(Vector(idxs) * cellSize, Vector(idxs + Indices(1, 1, 1)) * cellSize);
    }
};
 
NAMESPACE_SPH_END