ImplicitViscosity.h

Go to the documentation of this file.

#pragma once
#include "ParticleApproximation.h"
 
namespace dyno 
{
    template<typename TDataType>
    class ImplicitViscosity : public ParticleApproximation<TDataType>
    {
        DECLARE_TCLASS(ImplicitViscosity, TDataType)
    public:
        typedef typename TDataType::Real Real;
        typedef typename TDataType::Coord Coord;
 
        ImplicitViscosity();
        ~ImplicitViscosity() override;
        
    public:
        DEF_VAR(Real, Viscosity, 0.05, "");
 
        DEF_VAR(int, InterationNumber, 3, "");
 
        DEF_VAR_IN(Real, TimeStep, "");
 
        DEF_ARRAY_IN(Coord, Position, DeviceType::GPU, "");
 
        DEF_ARRAY_IN(Coord, Velocity, DeviceType::GPU, "");
 
        DEF_ARRAYLIST_IN(int, NeighborIds, DeviceType::GPU, "");
 
    public:
        void compute() override;
 
    private:
        DArray<Coord> mVelOld;
        DArray<Coord> mVelBuf;
    };
 
    IMPLEMENT_TCLASS(ImplicitViscosity, TDataType)
}