SurfaceEnergyForce.h

Go to the documentation of this file.

#pragma once
#include "ParticleApproximation.h"
 
namespace dyno {

    template<typename TDataType>
    class SurfaceEnergyForce : public ParticleApproximation<TDataType>
    {
        DECLARE_TCLASS(SurfaceEnergyForce, TDataType)
    public:
        typedef typename TDataType::Real Real;
        typedef typename TDataType::Coord Coord;
 
        SurfaceEnergyForce();
        ~SurfaceEnergyForce() override;
        
        DEF_VAR(Real, Kappa, Real(1), "Surface tension coefficient");
 
        DEF_VAR(Real, RestDensity, Real(1000), "Rest density");
 
    public:
        DEF_VAR_IN(Real, TimeStep, "Time step size!");
 
        DEF_ARRAY_IN(Coord, Position, DeviceType::GPU, "");
 
        DEF_ARRAY_IN(Coord, Velocity, DeviceType::GPU, "");

        DEF_ARRAYLIST_IN(int, NeighborIds, DeviceType::GPU, "Neighboring particles' ids");
 
    public:
        void compute() override;
 
    private:
        DArray<Real> mFreeSurfaceEnergy;
    };
 
    IMPLEMENT_TCLASS(SurfaceEnergyForce, TDataType);
}