LBPM/IO/netcdf.h

#ifndef NETCDF_READER
#define NETCDF_READER

#include <string>
#include <vector>

#include "common/Array.h"
#include "common/Communication.h"
#include "common/MPI.h"


namespace netcdf {


//! Enum to hold variable type
enum VariableType { BYTE, SHORT, USHORT, INT, UINT, INT64, UINT64, FLOAT, DOUBLE, STRING, UNKNOWN };

//! Enum to hold variable type
enum FileMode { READ, WRITE, CREATE };


//! Convert the VariableType to a string
std::string VariableTypeName( VariableType type );


/*!
 * @brief  Open netcdf file
 * @details  This function opens a netcdf file
 * @return This function returns a handle to the file
 * @param filename      File to open
 * @param mode          Open the file for reading or writing
 * @param comm          MPI communicator to use (MPI_COMM_WORLD: don't use parallel netcdf)
 */
int open( const std::string &filename, FileMode mode, const Utilities::MPI &comm = MPI_COMM_NULL );


/*!
 * @brief  Close netcdf file
 * @details  This function closes a netcdf file
 * @param fid           Handle to the open file
 */
void close( int fid );


/*!
 * @brief  Read the variable names
 * @details  This function reads a list of the variable names in the file
 * @param fid           Handle to the open file
 */
std::vector<std::string> getVarNames( int fid );


/*!
 * @brief  Read the attribute names
 * @details  This function reads a list of the attribute names in the file
 * @param fid           Handle to the open file
 */
std::vector<std::string> getAttNames( int fid );


/*!
 * @brief  Return the variable type
 * @details  This function returns the type for a variable
 * @param fid           Handle to the open file
 * @param var           Variable to read
 */
VariableType getVarType( int fid, const std::string &var );


/*!
 * @brief  Return the attribute type
 * @details  This function returns the type for an attribute
 * @param fid           Handle to the open file
 * @param att           Attribute to read
 */
VariableType getAttType( int fid, const std::string &att );


/*!
 * @brief  Return the variable dimensions
 * @details  This function returns the die for a variable
 * @param fid           Handle to the open file
 * @param var           Variable to read
 */
std::vector<size_t> getVarDim( int fid, const std::string &var );


/*!
 * @brief  Read a variable
 * @details  This function reads a variable with the given name from the file
 * @param fid           Handle to the open file
 * @param var           Variable to read
 */
template<class TYPE>
Array<TYPE> getVar( int fid, const std::string &var );


/*!
 * @brief  Read a strided variable
 * @details  This function reads a strided variable with the given name from the file
 * @param fid           Handle to the open file
 * @param var           Variable to read
 * @param start         Starting corner for the read
 * @param count         Number of elements to read
 * @param stride        Stride size for the read
 */
template<class TYPE>
Array<TYPE> getVar( int fid, const std::string &var, const std::vector<int> &start,
    const std::vector<int> &count, const std::vector<int> &stride );


/*!
 * @brief  Read an attribute
 * @details  This function reads an attribute with the given name from the file
 * @param fid           Handle to the open file
 * @param att           Attribute to read
 */
template<class TYPE>
Array<TYPE> getAtt( int fid, const std::string &att );


/*!
 * @brief  Write the dimensions
 * @details  This function writes the grid dimensions to netcdf.
 * @param fid           Handle to the open file
 * @param names
 * @param dims
 */
std::vector<int> defDim(
    int fid, const std::vector<std::string> &names, const std::vector<int> &dims );


/*!
 * @brief  Write a variable
 * @details  This function writes a variable to netcdf.
 * @param fid           Handle to the open file
 * @param var           Variable to read
 * @param dimids
 * @param data
 * @param rank_info
 */
template<class TYPE>
void write( int fid, const std::string &var, const std::vector<int> &dimids,
    const Array<TYPE> &data, const RankInfoStruct &rank_info );


} // namespace netcdf


#endif