CUDAParticleSystemSolverBase2.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_SOLVER_BASE2_HPP
#define CUBBYFLOW_CUDA_PARTICLE_SYSTEM_SOLVER_BASE2_HPP
 
#ifdef CUBBYFLOW_USE_CUDA
 
#include <Core/Animation/PhysicsAnimation.hpp>
#include <Core/CUDA/CUDAParticleSystemData2.hpp>
#include <Core/Matrix/Matrix.hpp>
#include <Core/Utils/Constants.hpp>
 
namespace CubbyFlow
{
class CUDAParticleSystemSolverBase2 : public PhysicsAnimation
{
 public:
    CUDAParticleSystemSolverBase2();
 
    CUDAParticleSystemSolverBase2(const CUDAParticleSystemSolverBase2&) =
        delete;
 
    CUDAParticleSystemSolverBase2(CUDAParticleSystemSolverBase2&&) noexcept =
        delete;
 
    ~CUDAParticleSystemSolverBase2() override = default;
 
    CUDAParticleSystemSolverBase2& operator=(
        const CUDAParticleSystemSolverBase2&) = delete;
 
    CUDAParticleSystemSolverBase2& operator=(
        CUDAParticleSystemSolverBase2&&) noexcept = delete;
 
    float DragCoefficient() const;
 
    void SetDragCoefficient(float newDragCoefficient);
 
    float RestitutionCoefficient() const;
 
    void SetRestitutionCoefficient(float newRestitutionCoefficient);
 
    const Vector2F& Gravity() const;
 
    void SetGravity(const Vector2F& newGravity);
 
    virtual CUDAParticleSystemData2* ParticleSystemData();
 
    virtual const CUDAParticleSystemData2* ParticleSystemData() const;
 
 protected:
    void OnInitialize() override;
 
    void UpdateCollider(double timeStepInSeconds);
 
    void UpdateEmitter(double timeStepInSeconds);
 
 private:
    float m_dragCoefficient = 1e-4f;
    float m_restitutionCoefficient = 0.0f;
    Vector2F m_gravity{ 0.0f, GRAVITY_FLOAT };
    CUDAParticleSystemData2Ptr m_particleSystemData;
};
 
using CUDAParticleSystemSolverBase2Ptr =
    std::shared_ptr<CUDAParticleSystemSolverBase2>;
 
template <typename DerivedBuilder>
class CUDAParticleSystemSolverBuilderBase2
{
 public:
    DerivedBuilder& WithDragCoefficient(float coeff);
 
    DerivedBuilder& WithRestitutionCoefficient(float coeff);
 
    DerivedBuilder& WithGravity(const Vector2F& gravity);
 
 protected:
    float m_dragCoefficient = 1e-4f;
    float m_restitutionCoefficient = 0.0f;
    Vector2F m_gravity{ 0.0f, GRAVITY_FLOAT };
};
 
template <typename T>
T& CUDAParticleSystemSolverBuilderBase2<T>::WithDragCoefficient(float coeff)
{
    m_dragCoefficient = coeff;
    return static_cast<T&>(*this);
}
 
template <typename T>
T& CUDAParticleSystemSolverBuilderBase2<T>::WithRestitutionCoefficient(
    float coeff)
{
    m_restitutionCoefficient = coeff;
    return static_cast<T&>(*this);
}
 
template <typename T>
T& CUDAParticleSystemSolverBuilderBase2<T>::WithGravity(const Vector2F& gravity)
{
    m_gravity = gravity;
    return static_cast<T&>(*this);
}
}  // namespace CubbyFlow
 
#endif
 
#endif