OneDimElasticityModule.h

Go to the documentation of this file.

 
#pragma once
#include "Module/ConstraintModule.h"
 
namespace dyno {
 
    template<typename TDataType>
    class OneDimElasticityModule : public ConstraintModule
    {
    public:
        typedef typename TDataType::Real Real;
        typedef typename TDataType::Coord Coord;
 
        OneDimElasticityModule();
        ~OneDimElasticityModule() override;
        
        void constrain() override;
 
        void solveElasticity();
 
        void setIterationNumber(int num) { m_iterNum.setValue(num); }
        int getIterationNumber() { return m_iterNum.getData(); }
 
        void setMaterialStiffness(Real stiff) { m_lambda.setValue(stiff); }
 
    protected:
        bool initializeImpl() override;
 
        void updateVelocity();
 
    public:
        FVar<Real> m_distance;
 
        DeviceArrayField<Coord> m_position;
        DeviceArrayField<Coord> m_velocity;
 
        DeviceArrayField<Real> m_mass;
 
    protected:
        FVar<Real> m_lambda;
 
        DArray<Coord> m_position_old;
        DArray<Coord> m_position_buf;
 
    private:
        FVar<int> m_iterNum;
    };
}