PCISPHSolver2.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_PCISPH_SOLVER2_HPP
#define CUBBYFLOW_PCISPH_SOLVER2_HPP

#include <Core/Solver/Particle/SPH/SPHSolver2.hpp>

namespace CubbyFlow
{
class PCISPHSolver2 : public SPHSolver2
{
 public:
    class Builder;

    PCISPHSolver2();

    PCISPHSolver2(double targetDensity, double targetSpacing,
                  double relativeKernelRadius);

    PCISPHSolver2(const PCISPHSolver2&) = delete;

    PCISPHSolver2(PCISPHSolver2&&) noexcept = delete;

    ~PCISPHSolver2() override = default;

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

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

    [[nodiscard]] double GetMaxDensityErrorRatio() const;

    void SetMaxDensityErrorRatio(double ratio);

    [[nodiscard]] unsigned int GetMaxNumberOfIterations() const;

    void SetMaxNumberOfIterations(unsigned int n);

    [[nodiscard]] static Builder GetBuilder();

 protected:
    void AccumulatePressureForce(double timeIntervalInSeconds) override;

    void OnBeginAdvanceTimeStep(double timeStepInSeconds) override;

 private:
    [[nodiscard]] double ComputeDelta(double timeStepInSeconds) const;
    [[nodiscard]] double ComputeBeta(double timeStepInSeconds) const;

    double m_maxDensityErrorRatio = 0.01;
    unsigned int m_maxNumberOfIterations = 5;

    ParticleSystemData2::VectorData m_tempPositions;
    ParticleSystemData2::VectorData m_tempVelocities;
    ParticleSystemData2::VectorData m_pressureForces;
    ParticleSystemData2::ScalarData m_densityErrors;
};

using PCISPHSolver2Ptr = std::shared_ptr<PCISPHSolver2>;

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

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

#endif