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/forms/VariableToSimulation.hpp>
 
#include <Eigen/Core>
 
#include <string>
#include <vector>
#include <memory>
 
namespace polyfem
{
    class State;
 
    namespace solver
    {
        class AdjointNLProblem;
    } // namespace solver
 
    class OptState
    {
    public:
        //---------------------------------------------------
        //-----------------initialization--------------------
        //---------------------------------------------------
 
        ~OptState();
        OptState();
 
        void init(const json &args, const 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_states(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;
 
        void solve(Eigen::VectorXd &x);
 
    private:
        inline std::string root_path() const
        {
            if (utils::is_param_valid(args, "root_path"))
                return args["root_path"].get<std::string>();
            return "";
        }
 
        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;
 
        //---------------------------------------------------
        //-----------------state--------------------
        //---------------------------------------------------
 
        std::vector<std::shared_ptr<State>> states;
        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;
 
    public:
        std::string output_dir;
    };
} // namespace polyfem