main.cpp

Go to the documentation of this file.

#include <filesystem>
 
#include <CLI/CLI.hpp>
 
#include <h5pp/h5pp.h>
 
#include <polyfem/State.hpp>
#include <polyfem/OptState.hpp>
 
#include <polyfem/utils/JSONUtils.hpp>
#include <polyfem/utils/Logger.hpp>
#include <polyfem/io/YamlToJson.hpp>
 
using namespace polyfem;
using namespace solver;
 
bool has_arg(const CLI::App &command_line, const std::string &value)
{
    const auto *opt = command_line.get_option_no_throw(value.size() == 1 ? ("-" + value) : ("--" + value));
    if (!opt)
        return false;
 
    return opt->count() > 0;
}
 
bool load_json(const std::string &json_file, json &out)
{
    std::ifstream file(json_file);
 
    if (!file.is_open())
        return false;
 
    file >> out;
 
    if (!out.contains("root_path"))
        out["root_path"] = json_file;
 
    return true;
}
 
bool load_yaml(const std::string &yaml_file, json &out)
{
    try
    {
        out = io::yaml_file_to_json(yaml_file);
        if (!out.contains("root_path"))
            out["root_path"] = yaml_file;
    }
    catch (...)
    {
        return false;
    }
    return true;
}
 
int forward_simulation(const CLI::App &command_line,
                       const std::string &hdf5_file,
                       const std::string output_dir,
                       const unsigned max_threads,
                       const bool is_strict,
                       const bool fallback_solver,
                       const spdlog::level::level_enum &log_level,
                       json &in_args);
 
int optimization_simulation(const CLI::App &command_line,
                            const unsigned max_threads,
                            const bool is_strict,
                            const spdlog::level::level_enum &log_level,
                            json &opt_args);
 
int main(int argc, char **argv)
{
    using namespace polyfem;
 
    CLI::App command_line{"polyfem"};
 
    command_line.ignore_case();
    command_line.ignore_underscore();
 
    // Eigen::setNbThreads(1);
    unsigned max_threads = std::numeric_limits<unsigned>::max();
    command_line.add_option("--max_threads", max_threads, "Maximum number of threads");
 
    auto input = command_line.add_option_group("input");
 
    std::string json_file = "";
    input->add_option("-j,--json", json_file, "Simulation JSON file")->check(CLI::ExistingFile);
 
    std::string yaml_file = "";
    input->add_option("-y,--yaml", yaml_file, "Simulation YAML file")->check(CLI::ExistingFile);
 
    std::string hdf5_file = "";
    input->add_option("--hdf5", hdf5_file, "Simulation HDF5 file")->check(CLI::ExistingFile);
 
    input->require_option(1);
 
    std::string output_dir = "";
    command_line.add_option("-o,--output_dir", output_dir, "Directory for output files")->check(CLI::ExistingDirectory | CLI::NonexistentPath);
 
    bool is_strict = true;
    command_line.add_flag("-s,--strict_validation,!--ns,!--no_strict_validation", is_strict, "Disables strict validation of input JSON");
 
    bool fallback_solver = false;
    command_line.add_flag("--enable_overwrite_solver", fallback_solver, "If solver in input is not present, falls back to default.");
 
    const std::vector<std::pair<std::string, spdlog::level::level_enum>>
        SPDLOG_LEVEL_NAMES_TO_LEVELS = {
            {"trace", spdlog::level::trace},
            {"debug", spdlog::level::debug},
            {"info", spdlog::level::info},
            {"warning", spdlog::level::warn},
            {"error", spdlog::level::err},
            {"critical", spdlog::level::critical},
            {"off", spdlog::level::off}};
    spdlog::level::level_enum log_level = spdlog::level::debug;
    command_line.add_option("--log_level", log_level, "Log level")
        ->transform(CLI::CheckedTransformer(SPDLOG_LEVEL_NAMES_TO_LEVELS, CLI::ignore_case));
 
    CLI11_PARSE(command_line, argc, argv);
 
    json in_args = json({});
 
    if (!json_file.empty() || !yaml_file.empty())
    {
        const bool ok = !json_file.empty() ? load_json(json_file, in_args) : load_yaml(yaml_file, in_args);
 
        if (!ok)
            log_and_throw_error(fmt::format("unable to open {} file", json_file));
 
        if (in_args.contains("states"))
            return optimization_simulation(command_line, max_threads, is_strict, log_level, in_args);
        else
            return forward_simulation(command_line, "", output_dir, max_threads,
                                      is_strict, fallback_solver, log_level, in_args);
    }
    else
        return forward_simulation(command_line, hdf5_file, output_dir, max_threads,
                                  is_strict, fallback_solver, log_level, in_args);
}
 
int forward_simulation(const CLI::App &command_line,
                       const std::string &hdf5_file,
                       const std::string output_dir,
                       const unsigned max_threads,
                       const bool is_strict,
                       const bool fallback_solver,
                       const spdlog::level::level_enum &log_level,
                       json &in_args)
{
    std::vector<std::string> names;
    std::vector<Eigen::MatrixXi> cells;
    std::vector<Eigen::MatrixXd> vertices;
 
    if (in_args.empty() && hdf5_file.empty())
    {
        logger().error("No input file specified!");
        return command_line.exit(CLI::RequiredError("--json or --hdf5"));
    }
 
    if (in_args.empty() && !hdf5_file.empty())
    {
        using MatrixXl = Eigen::Matrix<int64_t, Eigen::Dynamic, Eigen::Dynamic>;
 
        h5pp::File file(hdf5_file, h5pp::FileAccess::READONLY);
        std::string json_string = file.readDataset<std::string>("json");
 
        in_args = json::parse(json_string);
        in_args["root_path"] = hdf5_file;
 
        names = file.findGroups("", "/meshes");
        cells.resize(names.size());
        vertices.resize(names.size());
 
        for (int i = 0; i < names.size(); ++i)
        {
            const std::string &name = names[i];
            cells[i] = file.readDataset<MatrixXl>("/meshes/" + name + "/c").cast<int>();
            vertices[i] = file.readDataset<Eigen::MatrixXd>("/meshes/" + name + "/v");
        }
    }
 
    json tmp = json::object();
    if (has_arg(command_line, "log_level"))
        tmp["/output/log/level"_json_pointer] = int(log_level);
    if (has_arg(command_line, "max_threads"))
        tmp["/solver/max_threads"_json_pointer] = max_threads;
    if (has_arg(command_line, "output_dir"))
        tmp["/output/directory"_json_pointer] = std::filesystem::absolute(output_dir);
    if (has_arg(command_line, "enable_overwrite_solver"))
        tmp["/solver/linear/enable_overwrite_solver"_json_pointer] = fallback_solver;
    assert(tmp.is_object());
    in_args.merge_patch(tmp);
 
    State state;
    state.init(in_args, is_strict);
    state.load_mesh(/*non_conforming=*/false, names, cells, vertices);
 
    // Mesh was not loaded successfully; load_mesh() logged the error.
    if (state.mesh == nullptr)
    {
        // Cannot proceed without a mesh.
        return EXIT_FAILURE;
    }
 
    state.stats.compute_mesh_stats(*state.mesh);
 
    state.build_basis();
 
    state.assemble_rhs();
    state.assemble_mass_mat();
 
    Eigen::MatrixXd sol;
    Eigen::MatrixXd pressure;
 
    state.solve_problem(sol, pressure);
 
    state.compute_errors(sol);
 
    logger().info("total time: {}s", state.timings.total_time());
 
    state.save_json(sol);
    state.export_data(sol, pressure);
 
    return EXIT_SUCCESS;
}
 
int optimization_simulation(const CLI::App &command_line,
                            const unsigned max_threads,
                            const bool is_strict,
                            const spdlog::level::level_enum &log_level,
                            json &opt_args)
{
    json tmp = json::object();
    if (has_arg(command_line, "log_level"))
        tmp["/output/log/level"_json_pointer] = int(log_level);
    if (has_arg(command_line, "max_threads"))
        tmp["/solver/max_threads"_json_pointer] = max_threads;
    opt_args.merge_patch(tmp);
 
    OptState opt_state;
    opt_state.init(opt_args, is_strict);
 
    opt_state.create_states(opt_state.args["compute_objective"].get<bool>() ? polyfem::solver::CacheLevel::Solution : polyfem::solver::CacheLevel::Derivatives, opt_state.args["solver"]["max_threads"].get<int>());
    opt_state.init_variables();
    opt_state.create_problem();
 
    Eigen::VectorXd x;
    opt_state.initial_guess(x);
 
    if (opt_state.args["compute_objective"].get<bool>())
    {
        logger().info("Objective is {}", opt_state.eval(x));
        return EXIT_SUCCESS;
    }
 
    opt_state.solve(x);
    return EXIT_SUCCESS;
}