State.hpp

Go to the documentation of this file.

#pragma once
 
#include <polyfem/Common.hpp>
#include <polyfem/io/ResourceIO.hpp>
#include <polyfem/utils/Types.hpp>
 
#include <Eigen/Dense>
 
#include <spdlog/spdlog.h>
 
#include <functional>
#include <limits>
#include <memory>
#include <optional>
#include <string>
#include <vector>
 
namespace GEO
{
    class Mesh;
}
 
namespace polyfem::varform
{
    class VarForm;
}
 
namespace polyfem::io
{
    class CheckpointReader;
}
 
namespace polyfem
{
    class State
    {
    public:
        State();
        ~State() = default;
 
        void init(
            const json &args,
            bool strict_validation,
            bool is_adjoint_optimization = false);
 
        void init(
            const io::CheckpointReader &checkpoint,
            bool strict_validation);
        void init(
            const json &args,
            const io::CheckpointReader &checkpoint,
            bool strict_validation);
 
        void init(
            const json &args,
            const io::ResourceIO &resources,
            bool strict_validation,
            bool is_adjoint_optimization = false);
 
        void set_max_threads(const int max_threads = std::numeric_limits<int>::max());
 
        json args;
 
        std::shared_ptr<varform::VarForm> variational_formulation;
 
        std::function<void(int, int, double, double)> time_callback = nullptr;
 
        //---------------------------------------------------
        //-----------------logger----------------------------
        //---------------------------------------------------
 
        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);
 
    private:
        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;
 
    public:
        void solve(Eigen::MatrixXd &sol);
 
        //---------------------------------------------------
        //-----------------Geometry--------------------------
        //---------------------------------------------------
 
        void load_mesh(bool non_conforming = false);
 
        void set_mesh(GEO::Mesh &meshin, const std::function<int(const size_t, const std::vector<int> &, const RowVectorNd &, bool)> &boundary_marker, bool non_conforming = false, bool skip_boundary_sideset = false);
 
        void set_mesh(const Eigen::MatrixXd &V, const Eigen::MatrixXi &F, bool non_conforming = false);
 
    private:
        std::unique_ptr<const io::ResourceIO> resources_;
        std::optional<std::reference_wrapper<const io::CheckpointReader>> checkpoint_;
    };
 
} // namespace polyfem