partial_exporter.hpp

#ifndef __CORE_PARTIAL_EXPORTER_HPP__
#define __CORE_PARTIAL_EXPORTER_HPP__
 
#include <common/execinfo.hpp>
#include <core/post_process.hpp>
#include <cstddef>
#include <cstdint>
#include <dataexporter/data_exporter.hpp>
#include <optional>
#include <span>
#include <string>
#include <string_view>
#include <unordered_map>
#include <vector>
 
namespace Core
{
  class PartialExporter final : public DataExporter
  {
  public:
    PartialExporter(const ExecInfo& info,
                    std::string_view _filename,
                    std::optional<export_metadata_t> user_description
                    = std::nullopt);

    void init_fields(uint64_t n_iter, uint64_t n_compartments);

    void write_particle_data(PostProcessing::BonceBuffer&& bonce,
                             const std::string& ds_name,
                             bool compress_data);
    void write_number_particle(const std::vector<size_t>& distribution);

    void write_probe(const std::string& probe_name,
                     std::span<const double> data);

    void write_tally(std::span<const std::size_t> data);
 
  private:
    // uint64_t probe_counter_n_element; /**< Counter for the number of probe
    //                                      elements. */
    std::unordered_map<std::string, uint64_t> probe_counter_n_element;
  };
 
}; // namespace Core
 
#endif