MPMFluidSystemData.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_MPM_FLUID_SYSTEM_DATA_HPP
#define CUBBYFLOW_MPM_FLUID_SYSTEM_DATA_HPP
 
#include <Core/Particle/MPM/MPMSystemData.hpp>
 
namespace CubbyFlow
{
template <size_t N>
class MPMFluidSystemData final : public MPMTransferSystemData<N>
{
 public:
    static_assert(N == 2 || N == 3, "MPM supports only 2-D and 3-D.");
 
    using Base = MPMTransferSystemData<N>;
    using MatrixType = Matrix<double, N, N>;
 
    explicit MPMFluidSystemData(
        const Vector<size_t, N>& resolution =
            Vector<size_t, N>::MakeConstant(1),
        const Vector<double, N>& gridSpacing =
            Vector<double, N>::MakeConstant(1.0),
        const Vector<double, N>& gridOrigin = Vector<double, N>{},
        size_t numberOfParticles = 0);
 
    void Resize(size_t newNumberOfParticles) override;
 
    void Deserialize(const std::vector<uint8_t>& buffer) override;
 
    void Set(const ParticleSystemData<N>& other) override;
 
    [[nodiscard]] ConstArrayView1<double> VolumeRatios() const;
 
    [[nodiscard]] ArrayView1<double> VolumeRatios();
 
    [[nodiscard]] ConstArrayView1<MatrixType> VelocityGradients() const;
 
    [[nodiscard]] ArrayView1<MatrixType> VelocityGradients();
 
    void TransferFromGridToParticles(double timeStepInSeconds);
 
 private:
    void ResetFluidState();
 
    Array1<double> m_volumeRatios;
    Array1<MatrixType> m_velocityGradients;
};
 
using MPMFluidSystemData2 = MPMFluidSystemData<2>;
 
using MPMFluidSystemData3 = MPMFluidSystemData<3>;
 
using MPMFluidSystemData2Ptr = std::shared_ptr<MPMFluidSystemData2>;
 
using MPMFluidSystemData3Ptr = std::shared_ptr<MPMFluidSystemData3>;
}  // namespace CubbyFlow
 
#include <Core/Particle/MPM/MPMFluidSystemData-Impl.hpp>
 
#endif