PoissonDiskSampler.h

Go to the documentation of this file.

 
#pragma once
#include "Node.h"
 
#include "Primitive/Primitive3D.h"
#include "Topology/PointSet.h"
#include "Samplers/Sampler.h"
#include "Field/FilePath.h"
#include "Topology/DistanceField3D.h"
#include <Volume/VolumeOctree.h>
#include "SdfSampler.h"
namespace dyno
{
    struct GridIndex
    {
        int i, j, k;
    };
 
    template<typename TDataType>
    class PoissonDiskSampler : public SdfSampler<TDataType>
    {
        DECLARE_TCLASS(PoissonDiskSampler, TDataType);
 
    public:
        typedef typename TDataType::Real Real;
        typedef typename TDataType::Coord Coord;
 
        PoissonDiskSampler();
        ~PoissonDiskSampler();
 
        void resetStates() override;
 
    private:
 
        //std::shared_ptr<DistanceField3D<TDataType>> loadSdf();
 
        //.SDF file
        //DEF_VAR(FilePath, SdfFileName, "", "");
 
        std::shared_ptr<DistanceField3D<TDataType>>  getSDF() {   return m_inputSDF;    }
 
        void ConstructGrid();
 
        bool collisionJudge2D(Coord point);
 
        bool collisionJudge(Coord point);
 
        Real lerp(Real a, Real b, Real alpha);
 
        Real getDistanceFromSDF(const Coord& p, Coord& normal);
 
        GridIndex searchGrid(Coord point);
 
        int indexTransform(int i, int j, int k);
 
        int pointNumberRecommend();
 
        Coord getOnePointInsideSDF();
 
        std::vector<int> m_grid;
 
        std::vector<Coord> m_points;
 
        std::shared_ptr<DistanceField3D<TDataType>> m_inputSDF;
 
        CArray3D<Real> host_dist;
 
        int nx, ny, nz;       //grid number 
 
        unsigned int m_attempted_times = 10;
 
        Real m_grid_dx;        //resolution of grid 
 
        Coord area_a, area_b;  //box 
 
    };
 
    IMPLEMENT_TCLASS(PoissonDiskSampler, TDataType);
}