SurfaceTractionForms.hpp

Go to the documentation of this file.

#pragma once
 
#include <polyfem/utils/Types.hpp>
#include <polyfem/solver/forms/BarrierContactForm.hpp>
#include <polyfem/optimization/DiffCache.hpp>
#include <polyfem/optimization/forms/SpatialIntegralForms.hpp>
 
#include <Eigen/Core>
 
#include <memory>
#include <map>
#include <set>
#include <vector>
#include <utility>
 
namespace polyfem::solver
{
    // class TractionNormForm : public SpatialIntegralForm
    // {
    // public:
    //  TractionNormForm(const VariableToSimulationGroup &variable_to_simulations, const varform::DifferentiableVarForm &varform, const json &args) : SpatialIntegralForm(variable_to_simulations, varform, args)
    //  {
    //      set_integral_type(SpatialIntegralType::Surface);
 
    //      auto tmp_ids = args["surface_selection"].get<std::vector<int>>();
    //      ids_ = std::set(tmp_ids.begin(), tmp_ids.end());
 
    //      if (args["power"] > 0)
    //          in_power_ = args["power"];
    //  }
 
    // protected:
    //  IntegrableFunctional get_integral_functional() const override;
 
    // private:
    //  int in_power_ = 2;
    // };
 
    // class TrueContactForceForm : public StaticForm
    // {
    // public:
    //  TrueContactForceForm(const VariableToSimulationGroup &variable_to_simulations, const varform::DifferentiableVarForm &varform, const json &args) : StaticForm(variable_to_simulations), varform_(varform)
    //  {
    //      auto tmp_ids = args["surface_selection"].get<std::vector<int>>();
    //      ids_ = std::set(tmp_ids.begin(), tmp_ids.end());
 
    //      build_active_nodes();
    //  }
    //  ~TrueContactForceForm() = default;
 
    //  const varform::DifferentiableVarForm &get_varform() { return varform_; }
 
    //  double value_unweighted_step(const int time_step, const Eigen::VectorXd &x) const override;
    //  Eigen::VectorXd compute_adjoint_rhs_step(const int time_step, const Eigen::VectorXd &x, const varform::DifferentiableVarForm &varform) const override;
    //  Eigen::VectorXd compute_adjoint_rhs_step_prev(const int time_step, const Eigen::VectorXd &x, const varform::DifferentiableVarForm &varform) const override;
    //  virtual void compute_partial_gradient_step(const int time_step, const Eigen::VectorXd &x, Eigen::VectorXd &gradv) const override;
 
    //  void build_active_nodes();
 
    // protected:
    //  const varform::DifferentiableVarForm &varform_;
    //  std::set<int> ids_;
    //  Eigen::VectorXi active_nodes_;
    //  StiffnessMatrix active_nodes_mat_;
    //  int dim_;
 
    //  double dhat_, epsv_, friction_coefficient_;
    // };
 
    class ProxyContactForceForm : public StaticForm
    {
    public:
        ProxyContactForceForm(
            const VariableToSimulationGroup &variable_to_simulations,
            std::shared_ptr<const varform::DifferentiableVarForm> varform,
            std::shared_ptr<const DiffCache> diff_cache,
            const double dhat,
            const bool quadratic_potential,
            const json &args);
        ~ProxyContactForceForm() = default;
 
        double value_unweighted_step(const int time_step, const Eigen::VectorXd &x) const override;
        Eigen::VectorXd compute_adjoint_rhs_step(const int time_step, const Eigen::VectorXd &x, const varform::DifferentiableVarForm &varform, const DiffCache &diff_cache) const override;
        void compute_partial_gradient_step(const int time_step, const Eigen::VectorXd &x, Eigen::VectorXd &gradv) const override;
        void solution_changed_step(const int time_step, const Eigen::VectorXd &x) override
        {
            if (curr_x_.size() > 0 && (curr_x_ - x).norm() < 1e-8)
                return;
 
            collision_set_indicator_.setZero();
            build_collision_mesh();
            curr_x_ = x;
        }
 
    protected:
        void build_collision_mesh();
        void build_forward_collision_mesh();
        const ipc::NormalCollisions &get_or_compute_collision_set(const int time_step, const Eigen::MatrixXd &displaced_surface) const;
 
        std::shared_ptr<const varform::DifferentiableVarForm> varform_;
        std::shared_ptr<const DiffCache> diff_cache_;
        std::set<int> boundary_ids_;
        std::map<int, std::set<int>> boundary_ids_to_dof_;
        Eigen::MatrixXi can_collide_cache_;
 
        Eigen::MatrixXd node_positions_;
 
        mutable Eigen::VectorXi collision_set_indicator_;
        std::vector<std::shared_ptr<ipc::NormalCollisions>> collision_sets_;
 
        ipc::CollisionMesh collision_mesh_;
        const double dhat_;
        const double dmin_ = 0;
        ipc::BroadPhaseMethod broad_phase_method_;
 
        Eigen::VectorXd curr_x_;
 
        ipc::BarrierPotential barrier_potential_;
    };
 
} // namespace polyfem::solver