BruteForceFinder.h

Go to the documentation of this file.

#pragma once
 
 
#include "objects/finders/NeighbourFinder.h"
#include "objects/geometry/Box.h"
 
NAMESPACE_SPH_BEGIN
 
class BruteForceFinder : public FinderTemplate<BruteForceFinder> {
protected:
    // no need to implement these for brute force
    virtual void buildImpl(IScheduler& UNUSED(scheduler), ArrayView<const Vector> UNUSED(values)) override {}
 
public:
    template <bool FindAll>
    Size find(const Vector& pos, const Size index, const Float radius, Array<NeighbourRecord>& neighs) const {
        for (Size i = 0; i < this->values.size(); ++i) {
            const Float distSqr = getSqrLength(this->values[i] - pos);
            if (distSqr < sqr(radius) && (FindAll || rank[i] < rank[index])) {
                neighs.push(NeighbourRecord{ i, distSqr });
            }
        }
        return neighs.size();
    }
};
 
NAMESPACE_SPH_END