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