AssembleGlobalMatrix.h

#pragma once
 
#include <sofa/simulation/MechanicalVisitor.h>
 
namespace SofaCaribou::visitor {
 
class AssembleGlobalMatrix : public sofa::simulation::MechanicalVisitor {
    using Base = sofa::simulation::MechanicalVisitor;
    using MechanicalParams = sofa::core::MechanicalParams;
    using MultiMatrixAccessor = sofa::core::behavior::MultiMatrixAccessor;
public:
    // Constructor
    AssembleGlobalMatrix(const MechanicalParams* mparams, const MultiMatrixAccessor* matrix )
    : Base(mparams), p_multi_matrix(matrix) {}
 
    Result fwdForceField(sofa::simulation::Node* node, sofa::core::behavior::BaseForceField* ff) override;
 
    bool stopAtMechanicalMapping(sofa::simulation::Node* node, sofa::core::BaseMapping* map) override;
 
    const char* getClassName() const override { return "AssembleGlobalMatrix"; }
private:
    const sofa::core::behavior::MultiMatrixAccessor * p_multi_matrix;
};
 
} // namespace SofaCaribou::visitor