BodyForm.hpp

Go to the documentation of this file.

#pragma once
 
#include "Form.hpp"
 
#include <polyfem/assembler/RhsAssembler.hpp>
#include <polyfem/mesh/LocalBoundary.hpp>
#include <polyfem/utils/ElasticityUtils.hpp>
 
#include <polyfem/utils/Types.hpp>
 
namespace polyfem::solver
{
    class BodyForm : public Form
    {
    public:
        BodyForm(const int ndof,
                 const int n_pressure_bases,
                 const std::vector<int> &boundary_nodes,
                 const std::vector<mesh::LocalBoundary> &local_boundary,
                 const std::vector<mesh::LocalBoundary> &local_neumann_boundary,
                 const int n_boundary_samples,
                 const Eigen::MatrixXd &rhs,
                 const assembler::RhsAssembler &rhs_assembler,
                 const assembler::Density &density,
                 const bool apply_DBC,
                 const bool is_formulation_mixed,
                 const bool is_time_dependent);
 
        std::string name() const override { return "body"; }
 
    protected:
        double value_unweighted(const Eigen::VectorXd &x) const override;
 
        void first_derivative_unweighted(const Eigen::VectorXd &x, Eigen::VectorXd &gradv) const override;
 
        void second_derivative_unweighted(const Eigen::VectorXd &x, StiffnessMatrix &hessian) const override;
 
    public:
        void update_quantities(const double t, const Eigen::VectorXd &x) override;
 
        bool get_apply_DBC() { return apply_DBC_; }
        void set_apply_DBC(const Eigen::VectorXd &x, const bool val) override
        {
            if (val != apply_DBC_)
            {
                apply_DBC_ = val;
                update_current_rhs(x);
            }
        }
 
        void force_shape_derivative(
            const int n_verts,
            const double t,
            const Eigen::MatrixXd &x,
            const Eigen::MatrixXd &adjoint,
            Eigen::VectorXd &term);
 
        void hessian_wrt_u_prev(
            const Eigen::VectorXd &u_prev,
            const double t,
            StiffnessMatrix &hessian) const;
 
    private:
        double t_;       
        const int ndof_; 
        const int n_pressure_bases_;
 
        Eigen::MatrixXd x_prev_; 
 
        const std::vector<int> &boundary_nodes_;
        const std::vector<mesh::LocalBoundary> &local_boundary_;
        const std::vector<mesh::LocalBoundary> &local_neumann_boundary_;
        const int n_boundary_samples_;
 
        const Eigen::MatrixXd &rhs_;                   
        const assembler::RhsAssembler &rhs_assembler_; 
        const assembler::Density &density_;
 
        bool apply_DBC_;            
        bool is_formulation_mixed_; 
 
        Eigen::MatrixXd current_rhs_; 
 
        void update_current_rhs(const Eigen::VectorXd &x);
    };
} // namespace polyfem::solver