IterativeDensitySolver.h

Go to the documentation of this file.

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

    template<typename TDataType>
    class IterativeDensitySolver : public ParticleApproximation<TDataType>
    {
        DECLARE_TCLASS(IterativeDensitySolver, TDataType)
    public:
        typedef typename TDataType::Real Real;
        typedef typename TDataType::Coord Coord;
 
        IterativeDensitySolver();
        ~IterativeDensitySolver() override;
 
    public:
        DEF_VAR_IN(Real, TimeStep, "Time Step");

        DEF_ARRAY_IN(Coord, Position, DeviceType::GPU, "Input particle position");

        DEF_ARRAY_IN(Coord, Velocity, DeviceType::GPU, "Input particle velocity");

        DEF_ARRAY_IN(Attribute, Attribute, DeviceType::GPU, "Particle attribute");

        DEF_ARRAYLIST_IN(int, NeighborIds, DeviceType::GPU, "Neighboring particles' ids");

        DEF_ARRAY_OUT(Real, Density, DeviceType::GPU, "Final particle density");
 
    public:
        DEF_VAR(int, IterationNumber, 5, "Iteration number of the PBD solver");
 
        DEF_VAR(Real, RestDensity, 1000, "Reference density");
 
        DEF_VAR(Real, Kappa, Real(1), "");
 
    protected:
        void compute() override;
 
    public:
        void takeOneIteration();
 
        void updateVelocity();
 
    private:
        DArray<Real> mLamda;
        DArray<Coord> mDeltaPos;
        DArray<Coord> mPositionOld;
 
 
        Arithmetic<Real>* m_arithmetic;
 
 
    private:
        std::shared_ptr<SummationDensity<TDataType>> mSummation;
    };
 
    IMPLEMENT_TCLASS(IterativeDensitySolver, TDataType)
}