GridBackwardEulerDiffusionSolver2.h

Go to the documentation of this file.

/*************************************************************************
> File Name: GridBackwardEulerDiffusionSolver2.h
> Project Name: CubbyFlow
> Author: Chan-Ho Chris Ohk
> Purpose: 2-D grid-based backward Euler diffusion solver.
> Created Time: 2017/08/11
> Copyright (c) 2018, Chan-Ho Chris Ohk
*************************************************************************/
#ifndef CUBBYFLOW_GRID_BACKWARD_EULER_DIFFUSION_SOLVER2_H
#define CUBBYFLOW_GRID_BACKWARD_EULER_DIFFUSION_SOLVER2_H

#include <Core/Solver/FDM/FDMLinearSystemSolver2.h>
#include <Core/Solver/Grid/GridDiffusionSolver2.h>

namespace CubbyFlow
{
    class GridBackwardEulerDiffusionSolver2 final : public GridDiffusionSolver2
    {
    public:
        enum class BoundaryType
        {
            Dirichlet,
            Neumann
        };

        explicit GridBackwardEulerDiffusionSolver2(BoundaryType boundaryType = BoundaryType::Neumann);

        void Solve(
            const ScalarGrid2& source,
            double diffusionCoefficient,
            double timeIntervalInSeconds,
            ScalarGrid2* dest,
            const ScalarField2& boundarySDF = ConstantScalarField2(std::numeric_limits<double>::max()),
            const ScalarField2& fluidSDF = ConstantScalarField2(-std::numeric_limits<double>::max())) override;

        void Solve(
            const CollocatedVectorGrid2& source,
            double diffusionCoefficient,
            double timeIntervalInSeconds,
            CollocatedVectorGrid2* dest,
            const ScalarField2& boundarySDF = ConstantScalarField2(std::numeric_limits<double>::max()),
            const ScalarField2& fluidSDF = ConstantScalarField2(-std::numeric_limits<double>::max())) override;

        void Solve(
            const FaceCenteredGrid2& source,
            double diffusionCoefficient,
            double timeIntervalInSeconds,
            FaceCenteredGrid2* dest,
            const ScalarField2& boundarySDF = ConstantScalarField2(std::numeric_limits<double>::max()),
            const ScalarField2& fluidSDF = ConstantScalarField2(-std::numeric_limits<double>::max())) override;

        void SetLinearSystemSolver(const FDMLinearSystemSolver2Ptr& solver);

    private:
        BoundaryType m_boundaryType;
        FDMLinearSystem2 m_system;
        FDMLinearSystemSolver2Ptr m_systemSolver;
        Array2<char> m_markers;

        void BuildMarkers(
            const Size2& size,
            const std::function<Vector2D(size_t, size_t)>& pos,
            const ScalarField2& boundarySDF,
            const ScalarField2& fluidSDF);

        void BuildMatrix(
            const Size2& size,
            const Vector2D& c);

        void BuildVectors(
            const ConstArrayAccessor2<double>& f,
            const Vector2D& c);

        void BuildVectors(
            const ConstArrayAccessor2<Vector2D>& f,
            const Vector2D& c,
            size_t component);
    };

    using GridBackwardEulerDiffusionSolver2Ptr = std::shared_ptr<GridBackwardEulerDiffusionSolver2>;
}

#endif