IO.hpp

Go to the documentation of this file.

#pragma once
 
#include "stats/DataPacket.hpp"
 
#include "utils/Invariant.hpp"
 
#include <nlohmann/json.hpp>
#include <string>
 
namespace ppforest2::io {
  using ErrorHandler = void (*)(bool, std::string const&);

  void check_file_exists(std::string const& path);

  void check_file_not_exists(std::string const& path);

  void check_dir_not_exists(std::string const& path);
}
 
namespace ppforest2::io::json {
  std::string ensure_extension(std::string const& path);

  nlohmann::json read_file(std::string const& path, ErrorHandler on_error = invariant);

  void write_file(nlohmann::json const& data, std::string const& path, ErrorHandler on_error = invariant);
}
 
namespace ppforest2::io::text {
  void write_file(std::string const& content, std::string const& path, ErrorHandler on_error = invariant);
}
 
namespace ppforest2::io::csv {
  stats::DataPacket read(std::string const& filename);

  stats::DataPacket read_sorted(std::string const& filename);

  void write(stats::DataPacket const& data, std::string const& filename);

  struct FeatureSet {
    types::FeatureMatrix x;
    types::Names feature_names;
    std::vector<std::vector<std::string>> raw_rows;
  };

  FeatureSet read_features_from_string(std::string const& content);
}