VariationalApproximateProjection.h

Go to the documentation of this file.

#pragma once
 
#include "Collision/Attribute.h"
 
//ParticleSystem
#include "ParticleApproximation.h"
 
namespace dyno 
{
    template<typename TDataType> class SummationDensity;
    template<typename TDataType> class Reduction;
    template<typename TDataType> class Arithmetic;

    template<typename TDataType>
    class VariationalApproximateProjection : public ParticleApproximation<TDataType>
    {
        DECLARE_TCLASS(VariationalApproximateProjection, TDataType)
    public:
        typedef typename TDataType::Real Real;
        typedef typename TDataType::Coord Coord;
 
        VariationalApproximateProjection();
        ~VariationalApproximateProjection() override;
        
    public:
        DEF_VAR(Real, RestDensity, Real(1000), "");
 
        DEF_VAR_IN(Real, TimeStep, "Time step size");
 
        DEF_ARRAY_IN(Coord, Position, DeviceType::GPU, "");
 
        DEF_ARRAY_IN(Coord, Velocity, DeviceType::GPU, "");
 
        DEF_ARRAY_IN(Coord, Normal, DeviceType::GPU, "");
 
        DEF_ARRAY_IN(Attribute, Attribute, DeviceType::GPU, "");
        
        DEF_ARRAYLIST_IN(int, NeighborIds, DeviceType::GPU, "");
 
 
    protected:
        void compute() override;
 
    private:
        Real mAlphaMax;
        Real mAMax;
        Real mAirPressure = 0.0f;
 
        Real mTangential = 0.1f;
        Real mSeparation = 0.1f;
        //Real mRestDensity = 1000.0f;
 
        //Refer to "A Nonlocal Variational Particle Framework for Incompressible Free Surface Flows" for their exact meanings
        DArray<Real> mAlpha;
        DArray<Real> mAii;
        DArray<Real> mAiiFluid;
        DArray<Real> mAiiTotal;
 
        DArray<Real> mPressure;
        DArray<Real> mDivergence;
 
        //Indicate whether a particle is near the free surface boundary.
        DArray<bool> mIsSurface;
 
        //Internal variables used to solve the linear system of equations with a conjugate gradient method.
        DArray<Real> m_y;
        DArray<Real> m_r;
        DArray<Real> m_p;
 
        Reduction<Real>* m_reduce;
        Arithmetic<Real>* m_arithmetic;
 
        std::shared_ptr<SummationDensity<TDataType>> mDensityCalculator;
    };
 
    IMPLEMENT_TCLASS(VariationalApproximateProjection, TDataType)
}