DriverPrologue.hpp

#pragma once
 
#include <wmtk/utils/Logger.hpp>
#include <wmtk/utils/resolve_path.hpp>
 
#include <jse/jse.h>
#include <spdlog/common.h>
#include <nlohmann/json.hpp>
 
#include <filesystem>
#include <string>
#include <vector>
 
namespace wmtk::utils {
 
inline std::filesystem::path verify_and_setup_logger(
    nlohmann::json& json_params,
    const nlohmann::json& spec,
    const bool strict = false)
{
    {
        jse::JSE spec_engine;
        spec_engine.strict = strict;
        if (!spec_engine.verify_json(json_params, spec)) {
            log_and_throw_error(spec_engine.log2str());
        }
        json_params = spec_engine.inject_defaults(json_params, spec);
    }
 
    const std::filesystem::path root = json_params["input_dir"];
 
    {
        std::string log_file_name = json_params["log_file"];
        if (!log_file_name.empty()) {
            log_file_name = resolve_path(root, log_file_name).string();
            wmtk::set_file_logger(log_file_name);
            logger().flush_on(spdlog::level::info);
        }
    }
 
    return root;
}
 
inline std::vector<std::string> resolve_input_paths(
    const nlohmann::json& json_params,
    const std::filesystem::path& root)
{
    std::vector<std::string> input_paths = json_params["input"];
    for (std::string& p : input_paths) {
        p = resolve_path(root, p).string();
    }
    return input_paths;
}
 
} // namespace wmtk::utils