GridSmokeSolver2.hpp

Go to the documentation of this file.

// This code is based on Jet framework.
// Copyright (c) 2018 Doyub Kim
// CubbyFlow is voxel-based fluid simulation engine for computer games.
// Copyright (c) 2020 CubbyFlow Team
// Core Part: Chris Ohk, Junwoo Hwang, Jihong Sin, Seungwoo Yoo
// AI Part: Dongheon Cho, Minseo Kim
// We are making my contributions/submissions to this project solely in our
// personal capacity and are not conveying any rights to any intellectual
// property of any third parties.

#ifndef CUBBYFLOW_GRID_SMOKE_SOLVER2_HPP
#define CUBBYFLOW_GRID_SMOKE_SOLVER2_HPP

#include <Core/Solver/Grid/GridFluidSolver2.hpp>

namespace CubbyFlow
{
class GridSmokeSolver2 : public GridFluidSolver2
{
 public:
    class Builder;

    GridSmokeSolver2();

    GridSmokeSolver2(const Vector2UZ& resolution, const Vector2D& gridSpacing,
                     const Vector2D& gridOrigin);

    GridSmokeSolver2(const GridSmokeSolver2&) = delete;

    GridSmokeSolver2(GridSmokeSolver2&&) noexcept = delete;

    ~GridSmokeSolver2() override = default;

    GridSmokeSolver2& operator=(const GridSmokeSolver2&) = delete;

    GridSmokeSolver2& operator=(GridSmokeSolver2&&) noexcept = delete;

    [[nodiscard]] double GetSmokeDiffusionCoefficient() const;

    void SetSmokeDiffusionCoefficient(double newValue);

    [[nodiscard]] double GetTemperatureDiffusionCoefficient() const;

    void SetTemperatureDiffusionCoefficient(double newValue);

    [[nodiscard]] double GetBuoyancySmokeDensityFactor() const;

    void SetBuoyancySmokeDensityFactor(double newValue);

    [[nodiscard]] double GetBuoyancyTemperatureFactor() const;

    void SetBuoyancyTemperatureFactor(double newValue);

    [[nodiscard]] double GetSmokeDecayFactor() const;

    void SetSmokeDecayFactor(double newValue);

    [[nodiscard]] double GetTemperatureDecayFactor() const;

    void SetTemperatureDecayFactor(double newValue);

    [[nodiscard]] ScalarGrid2Ptr GetSmokeDensity() const;

    [[nodiscard]] ScalarGrid2Ptr GetTemperature() const;

    [[nodiscard]] static Builder GetBuilder();

 protected:
    void OnEndAdvanceTimeStep(double timeIntervalInSeconds) override;

    void ComputeExternalForces(double timeIntervalInSeconds) override;

 private:
    void ComputeDiffusion(double timeIntervalInSeconds);

    void ComputeBuoyancyForce(double timeIntervalInSeconds);

    size_t m_smokeDensityDataID = 0;
    size_t m_temperatureDataID = 0;
    double m_smokeDiffusionCoefficient = 0.0;
    double m_temperatureDiffusionCoefficient = 0.0;
    double m_buoyancySmokeDensityFactor = -0.000625;
    double m_buoyancyTemperatureFactor = 5.0;
    double m_smokeDecayFactor = 0.001;
    double m_temperatureDecayFactor = 0.001;
};

using GridSmokeSolver2Ptr = std::shared_ptr<GridSmokeSolver2>;

class GridSmokeSolver2::Builder final
    : public GridFluidSolverBuilderBase2<Builder>
{
 public:
    [[nodiscard]] GridSmokeSolver2 Build() const;

    [[nodiscard]] GridSmokeSolver2Ptr MakeShared() const;
};
}  // namespace CubbyFlow

#endif