Program Listing for File MicropolarModel_Bender2017.h

↰ Return to documentation for file (SPlisHSPlasH/Vorticity/MicropolarModel_Bender2017.h)

#ifndef __MicropolarModel_Bender2017_h__
#define __MicropolarModel_Bender2017_h__

#include "SPlisHSPlasH/Common.h"
#include "SPlisHSPlasH/FluidModel.h"
#include "SPlisHSPlasH/NonPressureForceBase.h"

namespace SPH
{
    class MicropolarModel_Bender2017 : public NonPressureForceBase
    {
    protected:
        std::vector<Vector3r> m_angularAcceleration;
        std::vector<Vector3r> m_omega;

        Real m_vorticityCoeff;
        Real m_viscosityOmega;
        Real m_inertiaInverse;

        virtual void initParameters();

    public:
        static std::string METHOD_NAME;
        static int VORTICITY_COEFFICIENT;
        static int VISCOSITY_OMEGA;
        static int INERTIA_INVERSE;

        MicropolarModel_Bender2017(FluidModel *model);
        virtual ~MicropolarModel_Bender2017(void);

        static NonPressureForceBase* creator(FluidModel* model) { return new MicropolarModel_Bender2017(model); }
        virtual std::string getMethodName() { return METHOD_NAME; }

        virtual void step();
        virtual void reset();

        virtual void performNeighborhoodSearchSort();

        FORCE_INLINE const Vector3r& getAngularAcceleration(const unsigned int i) const
        {
            return m_angularAcceleration[i];
        }

        FORCE_INLINE Vector3r& getAngularAcceleration(const unsigned int i)
        {
            return m_angularAcceleration[i];
        }

        FORCE_INLINE void setAngularAcceleration(const unsigned int i, const Vector3r& val)
        {
            m_angularAcceleration[i] = val;
        }

        FORCE_INLINE const Vector3r& getAngularVelocity(const unsigned int i) const
        {
            return m_omega[i];
        }

        FORCE_INLINE Vector3r& getAngularVelocity(const unsigned int i)
        {
            return m_omega[i];
        }

        FORCE_INLINE void setAngularVelocity(const unsigned int i, const Vector3r& val)
        {
            m_omega[i] = val;
        }
    };
}

#endif