PCGConstraintSolver.h

Go to the documentation of this file.

 
#pragma once
 
#include "Module/ConstraintModule.h"
#include "RigidBody/RigidBodyShared.h"
#include "Topology/DiscreteElements.h"
#include "Collision/Attribute.h"
#include "Collision/CollisionData.h"
 
namespace dyno
{
    template<typename TDataType>
    class PCGConstraintSolver : public ConstraintModule
    {
        DECLARE_TCLASS(PCGConstraintSolver, TDataType)
 
    public:
        typedef typename TDataType::Real Real;
        typedef typename TDataType::Coord Coord;
        typedef typename TDataType::Matrix Matrix;
 
        typedef typename ::dyno::Quat<Real> TQuat;
        typedef typename ::dyno::TContactPair<Real> ContactPair;
        typedef typename ::dyno::TConstraintPair<Real> Constraint;
 
        typedef typename ::dyno::BallAndSocketJoint<Real> BallAndSocketJoint;
        typedef typename ::dyno::SliderJoint<Real> SliderJoint;
        typedef typename ::dyno::HingeJoint<Real> HingeJoint;
        typedef typename ::dyno::FixedJoint<Real> FixedJoint;
        typedef typename ::dyno::PointJoint<Real> PointJoint;
 
        PCGConstraintSolver();
        ~PCGConstraintSolver();
 
    public:
        DEF_VAR(bool, FrictionEnabled, true, "");
 
        DEF_VAR(bool, GravityEnabled, true, "");
 
        DEF_VAR(Real, GravityValue, 9.8, "");
 
        DEF_VAR(Real, FrictionCoefficient, 10, "");
 
        DEF_VAR(Real, Slop, 0.001, "");
 
        DEF_VAR(Real, Frequency, 30, "");
 
        DEF_VAR(Real, DampingRatio, 1.0, "");
 
        DEF_VAR(uint, IterationNumberForVelocitySolverCG, 30, "");
 
        DEF_VAR(uint, IterationNumberForVelocitySolverJacobi, 0, "");
 
        DEF_VAR(Real, LinearDamping, 0.1, "");
 
        DEF_VAR(Real, AngularDamping, 0.1, "");
 
        DEF_VAR(Real, Tolerance, 0.00001, "");
 
    public:
        DEF_VAR_IN(Real, TimeStep, "Time step size");
 
        DEF_ARRAY_IN(Real, Mass, DeviceType::GPU, "Mass of rigid bodies");
 
        DEF_ARRAY_IN(Coord, Center, DeviceType::GPU, "Center of rigid bodies");
 
        DEF_ARRAY_IN(Coord, Velocity, DeviceType::GPU, "Velocity of rigid bodies");
 
        DEF_ARRAY_IN(Coord, AngularVelocity, DeviceType::GPU, "Angular velocity of rigid bodies");
 
        DEF_ARRAY_IN(Matrix, RotationMatrix, DeviceType::GPU, "Rotation matrix of rigid bodies");
 
        DEF_ARRAY_IN(Matrix, Inertia, DeviceType::GPU, "Interial matrix");
 
        DEF_ARRAY_IN(Matrix, InitialInertia, DeviceType::GPU, "Interial matrix");
 
        DEF_ARRAY_IN(TQuat, Quaternion, DeviceType::GPU, "Quaternion");
 
        DEF_ARRAY_IN(ContactPair, Contacts, DeviceType::GPU, "");
 
        DEF_INSTANCE_IN(DiscreteElements<TDataType>, DiscreteElements, "");
 
        DEF_ARRAY_IN(Attribute, Attribute, DeviceType::GPU, "Rigid body attributes");
    
    protected:
        void constrain() override;
 
    private:
        void initializeJacobian(Real dt);
 
    private:
        DArray<Coord> mJ;
        DArray<Coord> mB;
 
        DArray<Coord> mImpulseC;
        DArray<Coord> mImpulseExt;
 
        DArray<Real> mEta;
        DArray<Real> mLambda;
        DArray<Real> mLambdaOldJoint;
 
 
        DArray<ContactPair> mContactsInLocalFrame;
        DArray<Constraint> mVelocityConstraints;
 
        DArray<Real> mResidualOld;
        DArray<Real> mResidual;
        DArray<Real> tmpArray;
        DArray<Real> mP;
        DArray<Real> mAp;
        
        int cnt = 0;
        DArray<Real> mErrors;
 
        std::vector<float> residuals;
 
 
        DArray<int> mContactNumber;
 
        DArray<Real> mK_1;
        DArray<Mat2f> mK_2;
        DArray<Matrix> mK_3;
 
        DArray<Real> mA;
        DArray<Real> mZ;
        DArray<Real> mZold;
 
        DArray<Real> mCFM;
        DArray<Real> mERP;
 
        int mJointSize = -1;
        
    };
}