CUDASPHSystemData2.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_SPH_SYSTEM_DATA2_HPP
#define CUBBYFLOW_CUDA_SPH_SYSTEM_DATA2_HPP

#ifdef CUBBYFLOW_USE_CUDA

#include <Core/CUDA/CUDAParticleSystemData2.hpp>
#include <Core/Utils/Constants.hpp>

namespace CubbyFlow
{
class CUDASPHSystemData2 : public CUDAParticleSystemData2
{
 public:
    CUDASPHSystemData2();

    explicit CUDASPHSystemData2(size_t numberOfParticles);

    ~CUDASPHSystemData2() override = default;

    CUDASPHSystemData2(const CUDASPHSystemData2& other);

    CUDASPHSystemData2(CUDASPHSystemData2&& other) noexcept;

    CUDASPHSystemData2& operator=(const CUDASPHSystemData2& other);

    CUDASPHSystemData2& operator=(CUDASPHSystemData2&& other) noexcept;

    ConstCUDAArrayView1<float> Densities() const;

    CUDAArrayView1<float> Densities();

    ConstCUDAArrayView1<float> Pressures() const;

    CUDAArrayView1<float> Pressures();

    void UpdateDensities();

    float TargetDensity() const;

    void SetTargetDensity(float targetDensity);

    float TargetSpacing() const;

    void SetTargetSpacing(float spacing);

    float RelativeKernelRadius() const;

    void SetRelativeKernelRadius(float relativeRadius);

    float KernelRadius() const;

    void SetKernelRadius(float kernelRadius);

    float Mass() const;

    void BuildNeighborSearcher();

    void BuildNeighborListsAndUpdateDensities();

    void Set(const CUDASPHSystemData2& other);

 private:
    float m_targetDensity = WATER_DENSITY_FLOAT;

    float m_targetSpacing = 0.1f;

    float m_kernelRadiusOverTargetSpacing = 1.8f;

    float m_kernelRadius = 0.1f;

    float m_mass = 1e-3f;

    size_t m_pressureIdx = 0;

    size_t m_densityIdx = 0;

    void ComputeMass();
};

using CUDASPHSystemData2Ptr = std::shared_ptr<CUDASPHSystemData2>;

}  // namespace CubbyFlow

#endif

#endif