Program Listing for File SimulationDataDFSPH.h

↰ Return to documentation for file (SPlisHSPlasH/DFSPH/SimulationDataDFSPH.h)

#ifndef __SimulationDataDFSPH_h__
#define __SimulationDataDFSPH_h__

#include "SPlisHSPlasH/Common.h"
#include <vector>
#include "SPlisHSPlasH/FluidModel.h"

namespace SPH
{
    class SimulationDataDFSPH
    {
        public:
            SimulationDataDFSPH();
            virtual ~SimulationDataDFSPH();

        protected:
            std::vector<std::vector<Real>> m_factor;
            std::vector<std::vector<Real>> m_density_adv;

            std::vector<std::vector<Real>> m_pressure_rho2;
            std::vector<std::vector<Real>> m_pressure_rho2_V;
            std::vector<std::vector<Vector3r>> m_pressureAccel;

        public:

            virtual void init();

            virtual void cleanup();

            virtual void reset();

            void performNeighborhoodSearchSort();
            void emittedParticles(FluidModel *model, const unsigned int startIndex);

            std::vector<std::vector<Real>>& getPressureRho2Data() { return m_pressure_rho2; }
            std::vector<std::vector<Real>>& getPressureRho2VData() { return m_pressure_rho2_V; }

            FORCE_INLINE const Real getFactor(const unsigned int fluidIndex, const unsigned int i) const
            {
                return m_factor[fluidIndex][i];
            }

            FORCE_INLINE Real& getFactor(const unsigned int fluidIndex, const unsigned int i)
            {
                return m_factor[fluidIndex][i];
            }

            FORCE_INLINE void setFactor(const unsigned int fluidIndex, const unsigned int i, const Real p)
            {
                m_factor[fluidIndex][i] = p;
            }

            FORCE_INLINE const Real getDensityAdv(const unsigned int fluidIndex, const unsigned int i) const
            {
                return m_density_adv[fluidIndex][i];
            }

            FORCE_INLINE Real& getDensityAdv(const unsigned int fluidIndex, const unsigned int i)
            {
                return m_density_adv[fluidIndex][i];
            }

            FORCE_INLINE void setDensityAdv(const unsigned int fluidIndex, const unsigned int i, const Real d)
            {
                m_density_adv[fluidIndex][i] = d;
            }

            FORCE_INLINE const Real getPressureRho2(const unsigned int fluidIndex, const unsigned int i) const
            {
                return m_pressure_rho2[fluidIndex][i];
            }

            FORCE_INLINE Real& getPressureRho2(const unsigned int fluidIndex, const unsigned int i)
            {
                return m_pressure_rho2[fluidIndex][i];
            }

            FORCE_INLINE void setPressureRho2(const unsigned int fluidIndex, const unsigned int i, const Real p)
            {
                m_pressure_rho2[fluidIndex][i] = p;
            }

            FORCE_INLINE const Real getPressureRho2_V(const unsigned int fluidIndex, const unsigned int i) const
            {
                return m_pressure_rho2_V[fluidIndex][i];
            }

            FORCE_INLINE Real& getPressureRho2_V(const unsigned int fluidIndex, const unsigned int i)
            {
                return m_pressure_rho2_V[fluidIndex][i];
            }

            FORCE_INLINE void setPressureRho2_V(const unsigned int fluidIndex, const unsigned int i, const Real p)
            {
                m_pressure_rho2_V[fluidIndex][i] = p;
            }

            FORCE_INLINE Vector3r& getPressureAccel(const unsigned int fluidIndex, const unsigned int i)
            {
                return m_pressureAccel[fluidIndex][i];
            }

            FORCE_INLINE const Vector3r& getPressureAccel(const unsigned int fluidIndex, const unsigned int i) const
            {
                return m_pressureAccel[fluidIndex][i];
            }

            FORCE_INLINE void setPressureAccel(const unsigned int fluidIndex, const unsigned int i, const Vector3r& val)
            {
                m_pressureAccel[fluidIndex][i] = val;
            }
    };
}

#endif