GridBackwardEulerDiffusionSolver3.h

Go to the documentation of this file.

/*************************************************************************
> File Name: GridBackwardEulerDiffusionSolver3.h
> Project Name: CubbyFlow
> Author: Chan-Ho Chris Ohk
> Purpose: 3-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_SOLVER3_H
#define CUBBYFLOW_GRID_BACKWARD_EULER_DIFFUSION_SOLVER3_H

#include <Core/Solver/FDM/FDMLinearSystemSolver3.h>
#include <Core/Solver/Grid/GridDiffusionSolver3.h>

namespace CubbyFlow
{
    class GridBackwardEulerDiffusionSolver3 final : public GridDiffusionSolver3
    {
    public:
        enum class BoundaryType
        {
            Dirichlet,
            Neumann
        };

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

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

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

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

        void SetLinearSystemSolver(const FDMLinearSystemSolver3Ptr& solver);

    private:
        BoundaryType m_boundaryType;
        FDMLinearSystem3 m_system;
        FDMLinearSystemSolver3Ptr m_systemSolver;
        Array3<char> m_markers;

        void BuildMarkers(
            const Size3& size,
            const std::function<Vector3D(size_t, size_t, size_t)>& pos,
            const ScalarField3& boundarySDF,
            const ScalarField3& fluidSDF);

        void BuildMatrix(
            const Size3& size,
            const Vector3D& c);

        void BuildVectors(
            const ConstArrayAccessor3<double>& f,
            const Vector3D& c);

        void BuildVectors(
            const ConstArrayAccessor3<Vector3D>& f,
            const Vector3D& c,
            size_t component);
    };

    using GridBackwardEulerDiffusionSolver3Ptr = std::shared_ptr<GridBackwardEulerDiffusionSolver3>;
}

#endif