InertiaForm.hpp

Go to the documentation of this file.

#pragma once
 
#include "Form.hpp"
 
#include <polyfem/utils/Types.hpp>
#include <polyfem/time_integrator/ImplicitTimeIntegrator.hpp>
 
namespace polyfem
{
    namespace assembler
    {
        class Mass;
        class AssemblyValsCache;
    } // namespace assembler
 
    namespace basis
    {
        class ElementBases;
    }
} // namespace polyfem
 
namespace polyfem::solver
{
    class InertiaForm : public Form
    {
    public:
        InertiaForm(const StiffnessMatrix &mass,
                    const time_integrator::ImplicitTimeIntegrator &time_integrator);
 
        std::string name() const override { return "inertia"; }
 
        static void force_shape_derivative(
            bool is_volume,
            const int n_geom_bases,
            const double t, 
            const std::vector<basis::ElementBases> &bases,
            const std::vector<basis::ElementBases> &geom_bases,
            const assembler::Mass &assembler,
            const assembler::AssemblyValsCache &ass_vals_cache,
            const Eigen::MatrixXd &velocity,
            const Eigen::MatrixXd &adjoint,
            Eigen::VectorXd &term);
 
    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;
 
    private:
        // TODO mass might be time dependent
        const StiffnessMatrix &mass_;                                    
        const time_integrator::ImplicitTimeIntegrator &time_integrator_; 
    };
} // namespace polyfem::solver