grid_backward_euler_diffusion_solver3.h

Go to the documentation of this file.

// Copyright (c) 2018 Doyub Kim
//
// I am making my contributions/submissions to this project solely in my
// personal capacity and am not conveying any rights to any intellectual
// property of any third parties.
 
#ifndef INCLUDE_JET_GRID_BACKWARD_EULER_DIFFUSION_SOLVER3_H_
#define INCLUDE_JET_GRID_BACKWARD_EULER_DIFFUSION_SOLVER3_H_
 
#include <jet/constant_scalar_field3.h>
#include <jet/fdm_linear_system_solver3.h>
#include <jet/grid_diffusion_solver3.h>
#include <limits>
#include <memory>
 
namespace jet {
 
class GridBackwardEulerDiffusionSolver3 final : public GridDiffusionSolver3 {
 public:
    enum BoundaryType {
        Dirichlet,
        Neumann
    };
 
    explicit GridBackwardEulerDiffusionSolver3(
        BoundaryType boundaryType = Neumann);
 
    void solve(
        const ScalarGrid3& source,
        double diffusionCoefficient,
        double timeIntervalInSeconds,
        ScalarGrid3* dest,
        const ScalarField3& boundarySdf
            = ConstantScalarField3(kMaxD),
        const ScalarField3& fluidSdf
            = ConstantScalarField3(-kMaxD)) override;
 
    void solve(
        const CollocatedVectorGrid3& source,
        double diffusionCoefficient,
        double timeIntervalInSeconds,
        CollocatedVectorGrid3* dest,
        const ScalarField3& boundarySdf
            = ConstantScalarField3(kMaxD),
        const ScalarField3& fluidSdf
            = ConstantScalarField3(-kMaxD)) override;
 
    void solve(
        const FaceCenteredGrid3& source,
        double diffusionCoefficient,
        double timeIntervalInSeconds,
        FaceCenteredGrid3* dest,
        const ScalarField3& boundarySdf
            = ConstantScalarField3(kMaxD),
        const ScalarField3& fluidSdf
            = ConstantScalarField3(-kMaxD)) override;
 
    void setLinearSystemSolver(const FdmLinearSystemSolver3Ptr& solver);
 
 private:
    BoundaryType _boundaryType;
    FdmLinearSystem3 _system;
    FdmLinearSystemSolver3Ptr _systemSolver;
    Array3<char> _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);
};
 
typedef std::shared_ptr<GridBackwardEulerDiffusionSolver3>
    GridBackwardEulerDiffusionSolver3Ptr;
 
}  // namespace jet
 
#endif  // INCLUDE_JET_GRID_BACKWARD_EULER_DIFFUSION_SOLVER3_H_