SummationDensity.h

Go to the documentation of this file.

#pragma once
#include "ParticleApproximation.h"
 
namespace dyno {
    template<typename TDataType>
    class SummationDensity : public virtual ParticleApproximation<TDataType>
    {
        DECLARE_TCLASS(SummationDensity, TDataType)
    public:
        typedef typename TDataType::Real Real;
        typedef typename TDataType::Coord Coord;
 
        SummationDensity();
        ~SummationDensity() override {};
 
        void compute() override;
 
    public:
        void compute(
            DArray<Real>& rho,
            DArray<Coord>& pos,
            DArrayList<int>& neighbors,
            Real smoothingLength,
            Real mass);
 
        void compute(
            DArray<Real>& rho,
            DArray<Coord>& pos,
            DArray<Coord>& posQueried,
            DArrayList<int>& neighbors,
            Real smoothingLength,
            Real mass);
 
    public:
        DEF_VAR(Real, RestDensity, 1000, "Rest Density");


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

        DEF_ARRAY_IN(Coord, Other, DeviceType::GPU, "Particle position");

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


        DEF_ARRAY_OUT(Real, Density, DeviceType::GPU, "Return the particle density");
 
        Real getParticleMass() {
            return m_particle_mass;
        }
 
    private:
        void calculateParticleMass();
 
        Real m_particle_mass;
        Real m_factor;
    };
 
    IMPLEMENT_TCLASS(SummationDensity, TDataType)
}