State.hpp

Go to the documentation of this file.

#pragma once
 
#include <polyfem/Common.hpp>
#include <polyfem/utils/Types.hpp>
 
#include <Eigen/Dense>
 
#include <spdlog/spdlog.h>
 
#include <functional>
#include <limits>
#include <memory>
#include <string>
#include <vector>
 
namespace GEO
{
    class Mesh;
}
 
namespace polyfem::varform
{
    class VarForm;
}
 
namespace polyfem
{
    class State
    {
    public:
        State();
        ~State() = default;
 
        void init(
            const json &args,
            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,
                       const std::vector<std::string> &names = std::vector<std::string>(),
                       const std::vector<Eigen::MatrixXi> &cells = std::vector<Eigen::MatrixXi>(),
                       const std::vector<Eigen::MatrixXd> &vertices = std::vector<Eigen::MatrixXd>());
 
        void load_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 load_mesh(const Eigen::MatrixXd &V, const Eigen::MatrixXi &F, bool non_conforming = false);
    };
 
} // namespace polyfem