IterativeLevelSetSolver3.h

Go to the documentation of this file.

/*************************************************************************
> File Name: IterativeLevelSetSolver3.h
> Project Name: CubbyFlow
> Author: Chan-Ho Chris Ohk
> Purpose: Abstract base class for 3-D PDE-based iterative level set solver.
> Created Time: 2017/08/30
> Copyright (c) 2018, Chan-Ho Chris Ohk
*************************************************************************/
#ifndef CUBBYFLOW_ITERATIVE_LEVEL_SET_SOLVER3_H
#define CUBBYFLOW_ITERATIVE_LEVEL_SET_SOLVER3_H

#include <Core/Solver/LevelSet/LevelSetSolver3.h>

namespace CubbyFlow
{
    class IterativeLevelSetSolver3 : public LevelSetSolver3
    {
    public:
        IterativeLevelSetSolver3();

        virtual ~IterativeLevelSetSolver3();

        void Reinitialize(
            const ScalarGrid3& inputSDF,
            double maxDistance,
            ScalarGrid3* outputSDF) override;

        void Extrapolate(
            const ScalarGrid3& input,
            const ScalarField3& sdf,
            double maxDistance,
            ScalarGrid3* output) override;

        void Extrapolate(
            const CollocatedVectorGrid3& input,
            const ScalarField3& sdf,
            double maxDistance,
            CollocatedVectorGrid3* output) override;

        void Extrapolate(
            const FaceCenteredGrid3& input,
            const ScalarField3& sdf,
            double maxDistance,
            FaceCenteredGrid3* output) override;

        double GetMaxCFL() const;

        void SetMaxCFL(double newMaxCFL);

    protected:
        virtual void GetDerivatives(
            ConstArrayAccessor3<double> grid,
            const Vector3D& gridSpacing,
            size_t i, size_t j, size_t k,
            std::array<double, 2>* dx, std::array<double, 2>* dy, std::array<double, 2>* dz) const = 0;

    private:
        double m_maxCFL = 0.5;

        void Extrapolate(
            const ConstArrayAccessor3<double>& input,
            const ConstArrayAccessor3<double>& sdf,
            const Vector3D& gridSpacing,
            double maxDistance,
            ArrayAccessor3<double> output);

        static unsigned int DistanceToNumberOfIterations(double distance, double dtau);

        static double Sign(
            const ConstArrayAccessor3<double>& sdf,
            const Vector3D& gridSpacing,
            size_t i, size_t j, size_t k);

        double PseudoTimeStep(
            ConstArrayAccessor3<double> sdf,
            const Vector3D& gridSpacing) const;
    };

    using IterativeLevelSetSolver3Ptr = std::shared_ptr<IterativeLevelSetSolver3>;
}

#endif