toml_file_exporter.h

[詳解]

 
 
// Copyright(c) 2023-2025 Design Engineering Laboratory, Saitama University
// Released under the MIT license
// https://opensource.org/licenses/mit-license.php
 
#ifndef DESIGNLAB_TOML_FILE_EXPORTER_H_
#define DESIGNLAB_TOML_FILE_EXPORTER_H_
 
#include <filesystem>
#include <fstream>
#include <string>
#include <map>
#include <vector>
 
#include "cmdio_util.h"
#include "implicit_metafunction_for_toml11.h"
#include "string_util.h"
#include "toml11_define.h"
 
 
namespace designlab
{
 
template <typename T>
concept HasIntoToml = impl::has_into_toml<T>::value && std::is_default_constructible_v<T>;
 
 
template <HasIntoToml T>
class TomlFileExporter final
{
public:
    void Export(const std::string& file_path, const T& data)
    {
        const toml::basic_value<toml::preserve_comments, std::map> value(data);
        std::string res_str = toml::format(value);  // 設定を文字列に変換．
 
        InsertNewLine(&res_str);  // #をみたら，改行を挿入する．
 
        IndentTable(&res_str);  // Tableの中身をインデントする．
 
        std::ofstream ofs;
        ofs.open(file_path);
 
        // ファイルが開けなかったら何もしない．
        if (!ofs)
        {
            CmdIOUtil::Output("Failed to output TOML file. file_path : " + file_path, OutputDetail::kSystem);
            return;
        }
 
        ofs.write(res_str.c_str(), res_str.length());  // ファイルに書き込む．
 
        ofs.close();  // ファイルを閉じる．
 
        CmdIOUtil::Output("TOML files are output. file_path : " + file_path, OutputDetail::kSystem);
    }
 
private:
    void InsertNewLine(std::string* str)
    {
        if (str == nullptr)
        {
            return;
        }
 
        std::vector<std::string> split_str = string_util::Split((*str), "\n");
 
        std::string res_str;
        char past_first_char = ' ';
 
        for (const auto& s : split_str)
        {
            if (s.size() != 0 &&
                s[0] == '#' &&
                past_first_char != '#' &&
                past_first_char != ' ')
            {
                res_str += "\n";
            }
 
            past_first_char = s.size() != 0 ? s[0] : ' ';
 
            res_str += s + "\n";
        }
 
        *str = res_str;
    }
 
    void IndentTable(std::string* str)
    {
        if (str == nullptr)
        {
            return;
        }
 
        std::vector<std::string> split_str = string_util::Split((*str), "\n");
 
        std::string res_str;
        bool do_indent = false;
 
        for (size_t i = 0; i < split_str.size(); i++)
        {
            std::string past_s = i != 0 ? split_str[i - 1] : "";
            std::string next_s = i != split_str.size() - 1 ? split_str[i + 1] : "";
            std::string s = split_str[i];
 
            if (past_s.size() != 0 && past_s[0] == '[')
            {
                do_indent = true;
            }
 
            if (next_s.size() != 0 && next_s[0] == '[')
            {
                do_indent = false;
            }
 
            if (do_indent)
            {
                res_str += "    ";
            }
 
            res_str += s + "\n";
        }
 
        *str = res_str;
    }
};
 
}  // namespace designlab
 
 
#endif  // DESIGNLAB_TOML_FILE_EXPORTER_H_