node_state_writer.hpp

Go to the documentation of this file.

#pragma once
 
#include <algorithm>
#include <array>
#include <string>
#include <vector>
 
#include "netcdf_file.hpp"
 
namespace kynema::util {
 
class NodeStateWriter {
public:
    NodeStateWriter(const std::string& file_path, bool create, size_t num_nodes);
 
    void WriteStateDataAtTimestep(
        size_t timestep, const std::string& component_prefix, const std::vector<double>& x,
        const std::vector<double>& y, const std::vector<double>& z, const std::vector<double>& i,
        const std::vector<double>& j, const std::vector<double>& k,
        const std::vector<double>& w = std::vector<double>()
    ) const;
 
    void WriteDeformationDataAtTimestep(
        size_t timestep, const std::vector<double>& x, const std::vector<double>& y,
        const std::vector<double>& z
    ) const;
 
    [[nodiscard]] const NetCDFFile& GetFile() const;
 
    [[nodiscard]] size_t GetNumNodes() const;
 
private:
    NetCDFFile file_;
    size_t num_nodes_;
 
    void DefineStateVariables(
        const std::string& prefix, const std::vector<int>& dimensions, bool has_w
    );
};
 
}  // namespace kynema::util