imstkPbdEdgeEdgeCCDConstraint.h

/*
** This file is part of the Interactive Medical Simulation Toolkit (iMSTK)
** iMSTK is distributed under the Apache License, Version 2.0.
** See accompanying NOTICE for details.
*/

#pragma once

#include "imstkPbdCollisionConstraint.h"

namespace imstk
{
class PbdEdgeEdgeCCDConstraint : public PbdCollisionConstraint
{
public:
    PbdEdgeEdgeCCDConstraint() : PbdCollisionConstraint(2, 2) { }
    ~PbdEdgeEdgeCCDConstraint() override = default;

    IMSTK_TYPE_NAME(PbdEdgeEdgeCCDConstraint)

public:
    void initConstraint(
        Vec3d* prevPtA0, Vec3d* prevPtA1,
        Vec3d* prevPtB0, Vec3d* prevPtB1,
        const PbdParticleId& ptA0, const PbdParticleId& ptA1,
        const PbdParticleId& ptB0, const PbdParticleId& ptB1,
        double stiffnessA, double stiffnessB,
        int ccdSubsteps = 25);

    void projectConstraint(PbdState& bodies,
                           const double dt, const SolverType& type) override;

    bool computeValueAndGradient(PbdState& bodies,
                                 double& c, std::vector<Vec3d>& dcdx) override;

protected:
    // Extra particles used but not solved for
    std::array<Vec3d*, 2> m_prevEdgeA = { nullptr, nullptr };
    std::array<Vec3d*, 2> m_prevEdgeB = { nullptr, nullptr };
    int m_ccdSubsteps = 25;
};
} // namespace imstk