FullNLProblem.hpp

Go to the documentation of this file.

#pragma once
 
#include <polyfem/solver/forms/Form.hpp>
#include <polysolve/nonlinear/Problem.hpp>
 
#include <memory>
#include <vector>
 
namespace polyfem::solver
{
    class FullNLProblem : public polysolve::nonlinear::Problem
    {
    public:
        FullNLProblem(const std::vector<std::shared_ptr<Form>> &forms);
        virtual ~FullNLProblem() = default;
        virtual void init(const TVector &x0) override;
 
        virtual double value(const TVector &x) override;
        virtual void gradient(const TVector &x, TVector &gradv) override;
        virtual void hessian(const TVector &x, THessian &hessian) override;
 
        virtual bool is_step_valid(const TVector &x0, const TVector &x1) override;
        virtual bool is_step_collision_free(const TVector &x0, const TVector &x1);
        virtual double max_step_size(const TVector &x0, const TVector &x1) override;
 
        virtual void line_search_begin(const TVector &x0, const TVector &x1) override;
        virtual void line_search_end() override;
        virtual void post_step(const polysolve::nonlinear::PostStepData &data) override;
 
        virtual void set_project_to_psd(bool val) override;
 
        virtual void solution_changed(const TVector &new_x) override;
 
        virtual void init_lagging(const TVector &x);
        virtual void update_lagging(const TVector &x, const int iter_num);
        int max_lagging_iterations() const;
        bool uses_lagging() const;
 
        std::vector<std::shared_ptr<Form>> &forms() { return forms_; }
 
        virtual bool stop(const TVector &x) override { return false; }
 
        void finish()
        {
            for (auto &form : forms_)
                form->finish();
        }
 
    protected:
        std::vector<std::shared_ptr<Form>> forms_;
    };
} // namespace polyfem::solver