UniformGrid.h

Go to the documentation of this file.

#pragma once
 
 
#include "objects/containers/LookupMap.h"
#include "objects/finders/NeighbourFinder.h"
 
NAMESPACE_SPH_BEGIN
 
class UniformGridFinder : public FinderTemplate<UniformGridFinder> {
protected:
    LookupMap lut;
 
    Float relativeCellCnt;
 
    virtual void buildImpl(IScheduler& scheduler, ArrayView<const Vector> points) override;
 
public:
    UniformGridFinder(const Float relativeCellCnt = 1);
 
    ~UniformGridFinder();
 
    template <bool FindAll>
    Size find(const Vector& pos, const Size index, const Float radius, Array<NeighbourRecord>& neighs) const;
 
 
    template <typename TFunctor>
    void iterate(const TFunctor& functor) {
        const Size size = lut.getDimensionSize();
        for (Size z = 0; z < size; ++z) {
            for (Size y = 0; y < size; ++y) {
                for (Size x = 0; x < size; ++x) {
                    const Indices idxs = Indices(x, y, z);
                    Array<Size>& particles = lut(idxs);
                    functor(particles);
                }
            }
        }
    }
 
    INLINE Size getDimensionSize() const {
        return lut.getDimensionSize();
    }
};
 
NAMESPACE_SPH_END