netcdf_file.hpp

Go to the documentation of this file.

#pragma once
 
#include <span>
#include <string>
#include <vector>
 
#include <netcdf.h>
 
namespace kynema::util {
 
class NetCDFFile {
public:
    explicit NetCDFFile(const std::string& file_path, bool create = true);
 
    // Prevent copying and moving (since we don't want copies made of output file)
    NetCDFFile(const NetCDFFile&) = delete;
    NetCDFFile& operator=(const NetCDFFile&) = delete;
    NetCDFFile(NetCDFFile&&) = delete;
    NetCDFFile& operator=(NetCDFFile&&) = delete;
 
    ~NetCDFFile();
 
    //--------------------------------------------------------------------------
    // Setter/Write methods
    //--------------------------------------------------------------------------
 
    [[nodiscard]] int AddDimension(const std::string& name, size_t length) const;
 
    template <typename T>
    [[nodiscard]] int AddVariable(const std::string& name, std::span<const int> dim_ids) const;
 
    void AddAttribute(const std::string& var_name, const std::string& attr_name, float value) const;
    void AddAttribute(const std::string& var_name, const std::string& attr_name, double value) const;
    void AddAttribute(const std::string& var_name, const std::string& attr_name, int value) const;
    void AddAttribute(
        const std::string& var_name, const std::string& attr_name, const std::string& value
    ) const;
 
    void WriteVariable(const std::string& name, std::span<const float> data) const;
    void WriteVariable(const std::string& name, std::span<const double> data) const;
    void WriteVariable(const std::string& name, std::span<const int> data) const;
    void WriteVariable(const std::string& name, std::span<const std::string> data) const;
 
    void WriteVariableAt(
        const std::string& name, std::span<const size_t> start, std::span<const size_t> count,
        std::span<const float> data
    ) const;
    void WriteVariableAt(
        const std::string& name, std::span<const size_t> start, std::span<const size_t> count,
        std::span<const double> data
    ) const;
    void WriteVariableAt(
        const std::string& name, std::span<const size_t> start, std::span<const size_t> count,
        std::span<const int> data
    ) const;
    void WriteVariableAt(
        const std::string& name, std::span<const size_t> start, std::span<const size_t> count,
        std::span<const std::string> data
    ) const;
 
    void Sync() const;
 
    //--------------------------------------------------------------------------
    // Getter/Read methods
    //--------------------------------------------------------------------------
 
    [[nodiscard]] int GetNetCDFId() const;
 
    [[nodiscard]] int GetDimensionId(const std::string& name) const;
 
    [[nodiscard]] int GetVariableId(const std::string& name) const;
 
    [[nodiscard]] size_t GetNumberOfDimensions(const std::string& var_name) const;
 
    [[nodiscard]] size_t GetDimensionLength(int dim_id) const;
 
    [[nodiscard]] size_t GetDimensionLength(const std::string& name) const;
 
    [[nodiscard]] std::vector<size_t> GetShape(const std::string& var_name) const;
 
    void ReadVariable(const std::string& name, float* data) const;
    void ReadVariable(const std::string& name, double* data) const;
    void ReadVariable(const std::string& name, int* data) const;
 
    void ReadVariableAt(
        const std::string& name, std::span<const size_t> start, std::span<const size_t> count,
        float* data
    ) const;
    void ReadVariableAt(
        const std::string& name, std::span<const size_t> start, std::span<const size_t> count,
        double* data
    ) const;
    void ReadVariableAt(
        const std::string& name, std::span<const size_t> start, std::span<const size_t> count,
        int* data
    ) const;
 
    void ReadVariableWithStride(
        const std::string& name, std::span<const size_t> start, std::span<const size_t> count,
        std::span<const ptrdiff_t> stride, float* data
    ) const;
    void ReadVariableWithStride(
        const std::string& name, std::span<const size_t> start, std::span<const size_t> count,
        std::span<const ptrdiff_t> stride, double* data
    ) const;
    void ReadVariableWithStride(
        const std::string& name, std::span<const size_t> start, std::span<const size_t> count,
        std::span<const ptrdiff_t> stride, int* data
    ) const;
 
private:
    int netcdf_id_{-1};
};
 
}  // namespace kynema::util