FastSweepingMethodGPU.h

Go to the documentation of this file.

#pragma once
#include "Topology/LevelSet.h"
#include "Topology/TriangleSet.h"
 
#include "Module/ComputeModule.h"
 
namespace dyno {
    typedef Vector<unsigned int, 3> Vec3ui;
    typedef Vector<int, 3> Vec3i;
 
    typedef CArray3D<unsigned int> CArray3ui;
    typedef CArray3D<float> CArray3f;
    typedef CArray3D<int> CArray3i;

    template<typename TDataType>
    class FastSweepingMethodGPU : public ComputeModule
    {
        DECLARE_TCLASS(FastSweepingMethodGPU, TDataType)
    public:
        typedef typename TDataType::Real Real;
        typedef typename TDataType::Coord Coord;
 
        FastSweepingMethodGPU();
        ~FastSweepingMethodGPU() override;
 
        DEF_VAR(Real, Spacing, 0.05f, "");
 
        DEF_VAR(uint, Padding, 10, "");
 
        DEF_VAR(uint, PassNumber, 2, "");
 
    public:
        DEF_INSTANCE_IN(TriangleSet<TDataType>, TriangleSet, "");
 
        DEF_INSTANCE_OUT(LevelSet<TDataType>, LevelSet, "");
 
    protected:
        void compute() override;
 
    private:
        void makeLevelSet();
 
        int ni;
        int nj;
        int nk;
        Vec3f origin;
        Vec3f maxPoint;
    };
}