DifferentiableLinearElasticVarForm.hpp

Go to the documentation of this file.

#pragma once
 
#include <polyfem/varforms/LinearElasticVarForm.hpp>
#include <polyfem/varforms/diff/DifferentiableVarForm.hpp>
 
namespace polyfem::varform
{
    class DifferentiableLinearElasticVarForm final : public LinearElasticVarForm, public DifferentiableVarForm
    {
    public:
        using LinearElasticVarForm::LinearElasticVarForm;
        std::string name() const override;
        void solve(
            Eigen::MatrixXd &solution,
            const InitialConditionOverride *initial_condition_override,
            const ForwardStepCallback &post_step,
            bool differentiable) override;
        void prepare() override;
        void save_vtu(const std::string &path, const Eigen::MatrixXd &solution, double time, double dt) const override;
 
        json &get_args() override;
        const json &get_args() const override;
        const mesh::Mesh &get_mesh() const override;
        assembler::Problem &get_problem() override;
        const assembler::Problem &get_problem() const override;
        std::string output_file_path(const std::string &path) const override;
        const Units &get_units() const override;
        bool is_contact_enabled() const override;
 
        const FESpace &primary_space() const override;
        const VarFormBoundaryState &boundary_state() const override;
        const assembler::Assembler &primary_assembler() const override;
        const assembler::Mass &mass_assembler() const override;
        const assembler::AssemblyValsCache &assembly_cache() const override;
        const assembler::AssemblyValsCache &mass_assembly_cache() const override;
        const StiffnessMatrix &mass_matrix() const override;
        solver::SolveData *solve_data() override;
        const solver::SolveData *solve_data() const override;
        void initial_solution(Eigen::MatrixXd &solution, const InitialConditionOverride *override = nullptr) const override;
        void initial_velocity(Eigen::MatrixXd &velocity, const InitialConditionOverride *override = nullptr) const override;
        void initial_acceleration(Eigen::MatrixXd &acceleration, const InitialConditionOverride *override = nullptr) const override;
 
    protected:
        mesh::Mesh &mutable_mesh() override;
        void invalidate_after_geometry_update() override;
        void invalidate_after_parameter_update() override;
        QuadratureOrders boundary_samples(int discr_order, int discr_orderq, int geometry_discr_order) const override;
    };
} // namespace polyfem::varform