FMMLevelSetSolver2.h

Go to the documentation of this file.

/*************************************************************************
> File Name: FMMLevelSetSolver2.h
> Project Name: CubbyFlow
> Author: Chan-Ho Chris Ohk
> Purpose: Two-dimensional fast marching method (FMM) implementation.
> Created Time: 2017/09/01
> Copyright (c) 2018, Chan-Ho Chris Ohk
*************************************************************************/
#ifndef CUBBYFLOW_FMM_LEVEL_SET_SOLVER2_H
#define CUBBYFLOW_FMM_LEVEL_SET_SOLVER2_H

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

namespace CubbyFlow
{
    class FMMLevelSetSolver2 final : public LevelSetSolver2
    {
    public:
        FMMLevelSetSolver2();

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

        void Extrapolate(
            const ScalarGrid2& input,
            const ScalarField2& sdf,
            double maxDistance,
            ScalarGrid2* output) override;

        void Extrapolate(
            const CollocatedVectorGrid2& input,
            const ScalarField2& sdf,
            double maxDistance,
            CollocatedVectorGrid2* output) override;

        void Extrapolate(
            const FaceCenteredGrid2& input,
            const ScalarField2& sdf,
            double maxDistance,
            FaceCenteredGrid2* output) override;

    private:
        void Extrapolate(
            const ConstArrayAccessor2<double>& input,
            const ConstArrayAccessor2<double>& sdf,
            const Vector2D& gridSpacing,
            double maxDistance,
            ArrayAccessor2<double> output);
    };

    using FMMLevelSetSolver2Ptr = std::shared_ptr<FMMLevelSetSolver2>;
}

#endif