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 <map>
#include <string>
#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) {
      cmdio::Output("Failed to output TOML file. file_path : " + file_path,
                    OutputDetail::kSystem);
      return;
    }
 
    ofs.write(res_str.c_str(), res_str.length());  // ファイルに書き込む.
 
    ofs.close();  // ファイルを閉じる.
 
    cmdio::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_