CUDAParticleSystemSolver2.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_CUDA_PARTICLE_SYSTEM_SOLVER2_HPP
#define CUBBYFLOW_CUDA_PARTICLE_SYSTEM_SOLVER2_HPP

#ifdef CUBBYFLOW_USE_CUDA

#include <Core/CUDA/CUDAParticleSystemSolverBase2.hpp>

namespace CubbyFlow
{
class CUDAParticleSystemSolver2 : public CUDAParticleSystemSolverBase2
{
 public:
    class Builder;

    CUDAParticleSystemSolver2();

    CUDAParticleSystemSolver2(float radius, float mass);

    CUDAParticleSystemSolver2(const CUDAParticleSystemSolver2&) = delete;

    CUDAParticleSystemSolver2(CUDAParticleSystemSolver2&&) noexcept = delete;

    ~CUDAParticleSystemSolver2() override = default;

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

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

    float Radius() const;

    void SetRadius(float newRadius);

    float Mass() const;

    void SetMass(float newMass);

    static Builder GetBuilder();

 protected:
    void OnAdvanceTimeStep(double timeStepInSeconds) override;

 private:
    float m_radius = 1e-3f;
    float m_mass = 1e-3f;
};

using CUDAParticleSystemSolver2Ptr = std::shared_ptr<CUDAParticleSystemSolver2>;

class CUDAParticleSystemSolver2::Builder final
    : public CUDAParticleSystemSolverBuilderBase2<Builder>
{
 public:
    Builder& WithRadius(float radius);

    Builder& WithMass(float mass);

    CUDAParticleSystemSolver2 Build() const;

    CUDAParticleSystemSolver2Ptr MakeShared() const;

 private:
    float m_radius = 1e-3f;
    float m_mass = 1e-3f;
};
}  // namespace CubbyFlow

#endif

#endif