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>;
 
    enum class OutputLocation : std::uint8_t {
        kNodes = 0,  
        kQPs = 1     
    };
 
    Outputs(
        const std::string& output_file, size_t num_nodes,
        OutputLocation location = OutputLocation::kNodes
    );
 
    Outputs(
        const std::string& output_file, const std::string& time_series_file, size_t num_nodes,
        OutputLocation location = OutputLocation::kNodes
    );
 
    [[nodiscard]] std::unique_ptr<util::NodeStateWriter>& GetOutputWriter();
 
    [[nodiscard]] std::unique_ptr<util::TimeSeriesWriter>& GetTimeSeriesWriter();
 
    [[nodiscard]] OutputLocation GetLocation() const;
 
    void WriteNodeOutputsAtTimestep(const HostState<DeviceType>& host_state, size_t timestep);
 
 
    void WriteRotorTimeSeriesAtTimestep(size_t timestep, double azimuth_angle, double rotor_speed);
 
private:
    std::unique_ptr<util::NodeStateWriter> output_writer_;        
    std::unique_ptr<util::TimeSeriesWriter> time_series_writer_;  
    size_t num_nodes_;         
    OutputLocation location_;  
 
    // 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_;
};
 
}  // namespace kynema::interfaces