LinearElasticitySolver.h

Go to the documentation of this file.

#pragma once
#include "Module/ConstraintModule.h"
 
#include "Peridynamics/Bond.h"
 
namespace dyno {

    template<typename TDataType>
    class LinearElasticitySolver : public ConstraintModule
    {
        DECLARE_TCLASS(LinearElasticitySolver, TDataType)
 
    public:
        typedef typename TDataType::Real Real;
        typedef typename TDataType::Coord Coord;
        typedef typename TDataType::Matrix Matrix;
        typedef typename ::dyno::TBond<TDataType> Bond;
 
        LinearElasticitySolver();
        ~LinearElasticitySolver() override;
 
    protected:
        void constrain() override;
 
        virtual void solveElasticity();
 
        void preprocess() override;

        virtual void enforceElasticity();
        virtual void computeMaterialStiffness();
 
        void updateVelocity();
        void computeInverseK();
 
    public:
        DEF_VAR_IN(Real, Horizon, "");
 
        DEF_VAR_IN(Real, TimeStep, "");

        DEF_ARRAY_IN(Coord, X, DeviceType::GPU, "Rest Pos");

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

        DEF_ARRAY_IN(Coord, Velocity, DeviceType::GPU, "Particle velocity");
 
        
        DEF_ARRAYLIST_IN(Bond, Bonds, DeviceType::GPU, "Peridynamic bonds");
 
    public:
        DEF_VAR(Real, Mu, 0.001, "Lame parameters: mu");
 
        DEF_VAR(Real, Lambda, 0.01, "Lame parameters: lambda");
 
        DEF_VAR(uint, IterationNumber, 30, "Iteration number");
 
    protected:
        DArray<Real> mBulkStiffness;
        DArray<Real> mWeights;
 
        DArray<Coord> mDisplacement;
        DArray<Coord> mPosBuf;
 
        DArray<Matrix> mF;
        DArray<Matrix> mInvK;
    };
}