OptState.hpp

Go to the documentation of this file.

#pragma once
 
#include <polyfem/Common.hpp>
 
#include <polyfem/utils/JSONUtils.hpp>
#include <polyfem/utils/Logger.hpp>
 
#include <polyfem/optimization/DiffCache.hpp>
#include <polyfem/optimization/var2sims/VariableToSimulationGroup.hpp>
#include <polyfem/varforms/diff/DifferentiableVarForm.hpp>
#include <polyfem/io/ResourceIO.hpp>
 
#include <Eigen/Core>
 
#include <polysolve/nonlinear/Criteria.hpp>
 
#include <string>
#include <vector>
#include <memory>
 
namespace polyfem
{
    namespace solver
    {
        class AdjointNLProblem;
    } // namespace solver
 
    class OptState
    {
    public:
        //---------------------------------------------------
        //-----------------initialization--------------------
        //---------------------------------------------------
 
        ~OptState();
        OptState();
 
        void init(const json &args, const bool strict_validation);
        void init(const json &args, const io::ResourceIO &resources, bool strict_validation);
 
        int run(json args, const bool strict_validation);
        int run(json args, const io::ResourceIO &resources, bool strict_validation);
 
        json args;
 
        void init_logger(
            const std::string &log_file,
            const spdlog::level::level_enum log_level,
            const spdlog::level::level_enum file_log_level,
            const bool is_quiet);
 
        void init_logger(std::ostream &os, const spdlog::level::level_enum log_level);
 
        void set_log_level(const spdlog::level::level_enum log_level);
 
        void create_varforms(const int max_threads = -1);
 
        void init_variables();
 
        void create_problem();
 
        void initial_guess(Eigen::VectorXd &x); // shoud be const
 
        double eval(Eigen::VectorXd &x) const;
 
        polysolve::nonlinear::Status solve(Eigen::VectorXd &x);
 
        //---------------------------------------------------
        //-----------------varforms--------------------
        //---------------------------------------------------
 
        std::vector<std::shared_ptr<varform::DifferentiableVarForm>> varforms;
        std::vector<json> state_args;
        std::vector<std::unique_ptr<const io::ResourceIO>> state_resources;
        std::vector<std::shared_ptr<DiffCache>> diff_caches;
 
        std::vector<int> variable_sizes;
        int ndof;
 
        solver::VariableToSimulationGroup variable_to_simulations;
 
        std::unique_ptr<solver::AdjointNLProblem> nl_problem;
 
    private:
        void check_unsupported() const;
 
        void init_logger(const std::vector<spdlog::sink_ptr> &sinks, const spdlog::level::level_enum log_level);
 
        spdlog::sink_ptr console_sink_ = nullptr;
        spdlog::sink_ptr file_sink_ = nullptr;
 
        bool remeshing_requested_ = false;
        bool strict_validation_ = true;
        std::unique_ptr<const io::ResourceIO> resources_;
 
        //---------------------------------------------------
        //-----------------output--------------------
        //---------------------------------------------------
 
    public:
        std::string output_dir;
    };
} // namespace polyfem