SPHSystemData2.h

Go to the documentation of this file.

/*************************************************************************
> File Name: SPHSystemData2.h
> Project Name: CubbyFlow
> Author: Chan-Ho Chris Ohk
> Purpose: 2-D SPH particle system data.
> Created Time: 2017/05/31
> Copyright (c) 2018, Chan-Ho Chris Ohk
*************************************************************************/
#ifndef CUBBYFLOW_SPH_SYSTEM_DATA2_H
#define CUBBYFLOW_SPH_SYSTEM_DATA2_H

#include <Core/Particle/ParticleSystemData2.h>

namespace CubbyFlow
{
    class SPHSystemData2 : public ParticleSystemData2
    {
    public:
        SPHSystemData2();

        explicit SPHSystemData2(size_t numberOfParticles);

        SPHSystemData2(const SPHSystemData2& other);

        virtual ~SPHSystemData2();

        void SetRadius(double newRadius) override;

        void SetMass(double newMass) override;

        ConstArrayAccessor1<double> GetDensities() const;

        ArrayAccessor1<double> GetDensities();

        ConstArrayAccessor1<double> GetPressures() const;

        ArrayAccessor1<double> GetPressures();

        void UpdateDensities();

        void SetTargetDensity(double targetDensity);

        double GetTargetDensity() const;

        void SetTargetSpacing(double spacing);

        double GetTargetSpacing() const;

        void SetRelativeKernelRadius(double relativeRadius);

        double GetRelativeKernelRadius() const;

        void SetKernelRadius(double kernelRadius);

        double GetKernelRadius() const;

        double SumOfKernelNearby(const Vector2D& position) const;

        double Interpolate(const Vector2D& origin, const ConstArrayAccessor1<double>& values) const;

        Vector2D Interpolate(const Vector2D& origin, const ConstArrayAccessor1<Vector2D>& values) const;

        Vector2D GradientAt(size_t i, const ConstArrayAccessor1<double>& values) const;

        double LaplacianAt(size_t i, const ConstArrayAccessor1<double>& values) const;

        Vector2D LaplacianAt(size_t i, const ConstArrayAccessor1<Vector2D>& values) const;

        void BuildNeighborSearcher();

        void BuildNeighborLists();

        void Serialize(std::vector<uint8_t>* buffer) const override;

        void Deserialize(const std::vector<uint8_t>& buffer) override;

        void Set(const SPHSystemData2& other);

        SPHSystemData2& operator=(const SPHSystemData2& other);

    private:
        double m_targetDensity = WATER_DENSITY;

        double m_targetSpacing = 0.1;

        double m_kernelRadiusOverTargetSpacing = 1.8;

        double m_kernelRadius;

        size_t m_pressureIdx;

        size_t m_densityIdx;

        void ComputeMass();
    };

    using SPHSystemData2Ptr = std::shared_ptr<SPHSystemData2>;
}

#endif