outputs.hpp

Go to the documentation of this file.

#pragma once
 
#include <memory>
#include <string>
#include <vector>
 
#include <Kokkos_Core.hpp>
 
#include "utilities/netcdf/node_state_writer.hpp"
#include "utilities/netcdf/time_series_writer.hpp"
 
namespace kynema::interfaces {
 
template <typename DeviceType>
struct HostState;
 
class Outputs {
public:
    using DeviceType =
        Kokkos::Device<Kokkos::DefaultExecutionSpace, Kokkos::DefaultExecutionSpace::memory_space>;
 
    Outputs(
        const std::string& output_file, size_t num_nodes, const std::string& time_series_file = "",
        const std::vector<std::string>& enabled_state_prefixes = {"x", "u", "v", "a", "f"},
        size_t buffer_size = util::NodeStateWriter::kDefaultBufferSize
    );
 
    Outputs(
        const std::string& output_file, size_t num_nodes, const std::string& time_series_file,
        const std::vector<std::string>& time_series_channel_names,
        const std::vector<std::string>& time_series_channel_units = {},
        const std::vector<std::string>& enabled_state_prefixes = {"x", "u", "v", "a", "f"},
        size_t node_state_buffer_size = util::NodeStateWriter::kDefaultBufferSize,
        size_t time_series_buffer_size = util::TimeSeriesWriter::kDefaultBufferSize
    );
 
    [[nodiscard]] std::unique_ptr<util::NodeStateWriter>& GetOutputWriter();
 
    [[nodiscard]] std::unique_ptr<util::TimeSeriesWriter>& GetTimeSeriesWriter();
 
    void WriteNodeOutputsAtTimestep(const HostState<DeviceType>& host_state, size_t timestep);
 
    void WriteValueAtTimestep(size_t timestep, const std::string& name, double value);
 
    void WriteTimeSeriesRowAtTimestep(size_t timestep, std::span<const double> row);
 
    void Close();
 
    void Open();
 
private:
    std::unique_ptr<util::NodeStateWriter> output_writer_;  
    size_t num_nodes_;  
    std::unique_ptr<util::TimeSeriesWriter> time_series_writer_;  
    std::vector<std::string> enabled_state_prefixes_;  
 
    // Pre-allocated vectors for storing output data
    std::vector<double> x_data_;
    std::vector<double> y_data_;
    std::vector<double> z_data_;
    std::vector<double> i_data_;
    std::vector<double> j_data_;
    std::vector<double> k_data_;
    std::vector<double> w_data_;
 
    [[nodiscard]] bool IsStateComponentEnabled(const std::string& prefix) const;
};
 
}  // namespace kynema::interfaces