AMIPSForm.hpp

Go to the documentation of this file.

#pragma once
 
#include <polyfem/optimization/var2sims/VariableToSimulationGroup.hpp>
#include <polyfem/optimization/forms/AdjointForm.hpp>
#include <polyfem/assembler/AssemblyValsCache.hpp>
#include <polyfem/basis/ElementBases.hpp>
#include <memory>
#include <utility>
 
namespace polyfem
{
    namespace assembler
    {
        class Assembler;
    }
} // namespace polyfem
 
namespace polyfem::solver
{
    class MinJacobianForm : public AdjointForm
    {
    public:
        MinJacobianForm(const VariableToSimulationGroup &variable_to_simulation, std::shared_ptr<const varform::DifferentiableVarForm> varform)
            : AdjointForm(variable_to_simulation),
              varform_(std::move(varform))
        {
        }
 
        virtual std::string name() const override { return "min-jacobian"; }
 
        double value_unweighted(const Eigen::VectorXd &x) const override;
        void compute_partial_gradient(const Eigen::VectorXd &x, Eigen::VectorXd &gradv) const override;
 
    private:
        std::shared_ptr<const varform::DifferentiableVarForm> varform_;
    };
 
    class AMIPSForm : public AdjointForm
    {
    public:
        AMIPSForm(const VariableToSimulationGroup &variable_to_simulation, std::shared_ptr<const varform::DifferentiableVarForm> varform);
 
        virtual std::string name() const override { return "AMIPS"; }
 
        double value_unweighted(const Eigen::VectorXd &x) const override;
        void compute_partial_gradient(const Eigen::VectorXd &x, Eigen::VectorXd &gradv) const override;
        bool is_step_valid(const Eigen::VectorXd &x0, const Eigen::VectorXd &x1) const override;
 
    private:
        Eigen::VectorXd get_updated_mesh_nodes(const Eigen::VectorXd &x) const
        {
            Eigen::VectorXd X = X_rest;
            variable_to_simulations_.compute_state_variable(ParameterType::Shape, *varform_, x, X);
            return X;
        }
 
        std::shared_ptr<const varform::DifferentiableVarForm> varform_;
 
        Eigen::VectorXd X_rest;
        Eigen::MatrixXi F;
        std::vector<polyfem::basis::ElementBases> init_geom_bases_;
        assembler::AssemblyValsCache init_ass_vals_cache_;
        // amips_energy_ is actually owned exclusively by this class.
        std::shared_ptr<assembler::Assembler> amips_energy_;
    };
} // namespace polyfem::solver