SmoothingForms.hpp

Go to the documentation of this file.

#pragma once
 
#include <polyfem/optimization/forms/AdjointForm.hpp>
 
#include <Eigen/Core>
 
#include <memory>
#include <set>
#include <utility>
#include <vector>
 
namespace polyfem::solver
{
    class BoundarySmoothingForm : public AdjointForm
    {
    public:
        BoundarySmoothingForm(
            const VariableToSimulationGroup &variable_to_simulations,
            std::shared_ptr<const varform::DifferentiableVarForm> varform,
            const bool scale_invariant,
            const int power,
            const std::vector<int> &surface_selections,
            const std::vector<int> &active_dims);
 
        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_;
        const bool scale_invariant_;
        const int power_; // only if scale_invariant_ is true
        Eigen::SparseMatrix<bool, Eigen::RowMajor> adj;
        Eigen::SparseMatrix<double, Eigen::RowMajor> L;
        std::set<int> surface_ids_;
        std::vector<int> active_dims_;
    };
 
    class ElasticMaterialSmoothingForm : public AdjointForm
    {
    public:
        ElasticMaterialSmoothingForm(
            const VariableToSimulationGroup &variable_to_simulations,
            std::shared_ptr<const varform::DifferentiableVarForm> varform,
            const std::vector<int> &volume_selections);
 
        std::string name() const override { return "elastic_material_smoothing"; }
 
        double value_unweighted(const Eigen::VectorXd &x) const override;
        void compute_partial_gradient(const Eigen::VectorXd &x, Eigen::VectorXd &gradv) const override;
 
    private:
        Eigen::VectorXd lame_parameters(const Eigen::VectorXd &x) const;
 
        std::shared_ptr<const varform::DifferentiableVarForm> varform_;
        std::vector<std::pair<int, int>> adjacent_elements_;
    };
} // namespace polyfem::solver