NCFile.hh

Go to the documentation of this file.

// Copyright (C) 2012, 2013, 2014, 2015, 2016, 2017, 2019, 2020, 2021, 2023 PISM Authors
//
// This file is part of PISM.
//
// PISM is free software; you can redistribute it and/or modify it under the
// terms of the GNU General Public License as published by the Free Software
// Foundation; either version 3 of the License, or (at your option) any later
// version.
//
// PISM is distributed in the hope that it will be useful, but WITHOUT ANY
// WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
// FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
// details.
//
// You should have received a copy of the GNU General Public License
// along with PISM; if not, write to the Free Software
// Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
 
#ifndef PISM_NCFILE_H
#define PISM_NCFILE_H
 
#include <memory>
#include <string>
#include <vector>
 
#include <mpi.h>
 
namespace pism {
 
class Grid;
 
namespace io {
enum Mode : int;
enum Type : int;
}
 
//! Input and output code (NetCDF wrappers, etc)
namespace io {
 
//! \brief The PISM wrapper for a subset of the NetCDF C API.
/*!
 * The goal of this class is to hide the fact that we need to communicate data
 * to and from the processor zero. Using this wrapper we should be able to
 * write code that looks good and works both on 1-processor and
 * multi-processor systems.
 *
 * Moreover, this way we can switch underlying I/O implementations.
 *
 * Notes:
 * - It uses C++ STL strings instead of C character arrays
 * - It hides NetCDF ncid, dimid and varid and uses strings to reference
 *   dimensions and variables instead.
 * - This class does not and should not use any PETSc API calls.
 * - This wrapper provides access to a very small portion of the NetCDF C API.
 *   (Only calls used in PISM.) This is intentional.
 * - Methods of this class should do what corresponding NetCDF C API calls do,
 *   no more and no less.
 */
class NCFile {
public:
  typedef std::shared_ptr<NCFile> Ptr;
 
  NCFile(MPI_Comm com);
  virtual ~NCFile() = default;
 
  // open/create/close
  void open(const std::string &filename, io::Mode mode);
 
  void create(const std::string &filename);
 
  void sync() const;
 
  void close();
 
  // redef/enddef
  void enddef() const;
 
  void redef() const;
 
  // dim
  void def_dim(const std::string &name, size_t length) const;
 
  void inq_dimid(const std::string &dimension_name, bool &exists) const;
 
  void inq_dimlen(const std::string &dimension_name, unsigned int &result) const;
 
  void inq_unlimdim(std::string &result) const;
 
  // var
  void def_var(const std::string &name, io::Type nctype,
               const std::vector<std::string> &dims) const;
 
  void def_var_chunking(const std::string &name, std::vector<size_t> &dimensions) const;
 
  void get_vara_double(const std::string &variable_name, const std::vector<unsigned int> &start,
                       const std::vector<unsigned int> &count, double *ip) const;
 
  void put_vara_double(const std::string &variable_name, const std::vector<unsigned int> &start,
                       const std::vector<unsigned int> &count, const double *op) const;
 
  void write_darray(const std::string &variable_name, const Grid &grid, unsigned int z_count,
                    bool time_dependent, unsigned int record, const double *input);
 
  void get_varm_double(const std::string &variable_name, const std::vector<unsigned int> &start,
                       const std::vector<unsigned int> &count,
                       const std::vector<unsigned int> &imap, double *ip) const;
 
  void inq_nvars(int &result) const;
 
  void inq_vardimid(const std::string &variable_name, std::vector<std::string> &result) const;
 
  void inq_varnatts(const std::string &variable_name, int &result) const;
 
  void inq_varid(const std::string &variable_name, bool &result) const;
 
  void inq_varname(unsigned int j, std::string &result) const;
 
  void set_compression_level(int level) const;
 
  // att
  void get_att_double(const std::string &variable_name, const std::string &att_name,
                      std::vector<double> &result) const;
 
  void get_att_text(const std::string &variable_name, const std::string &att_name,
                    std::string &result) const;
 
  void put_att_double(const std::string &variable_name, const std::string &att_name, io::Type xtype,
                      const std::vector<double> &data) const;
 
  void put_att_text(const std::string &variable_name, const std::string &att_name,
                    const std::string &value) const;
 
  void inq_attname(const std::string &variable_name, unsigned int n, std::string &result) const;
 
  void inq_atttype(const std::string &variable_name, const std::string &att_name,
                   io::Type &result) const;
 
  // misc
  void set_fill(int fillmode, int &old_modep) const;
 
  std::string filename() const;
 
  void del_att(const std::string &variable_name, const std::string &att_name) const;
 
protected:
  // implementations:
 
  // open/create/close
  virtual void open_impl(const std::string &filename, io::Mode mode) = 0;
  virtual void create_impl(const std::string &filename)              = 0;
  virtual void sync_impl() const                                     = 0;
  virtual void close_impl()                                          = 0;
 
  // redef/enddef
  virtual void enddef_impl() const = 0;
 
  virtual void redef_impl() const = 0;
 
  // dim
  virtual void def_dim_impl(const std::string &name, size_t length) const = 0;
 
  virtual void inq_dimid_impl(const std::string &dimension_name, bool &exists) const = 0;
 
  virtual void inq_dimlen_impl(const std::string &dimension_name, unsigned int &result) const = 0;
 
  virtual void inq_unlimdim_impl(std::string &result) const = 0;
 
  // var
  virtual void def_var_impl(const std::string &name, io::Type nctype,
                            const std::vector<std::string> &dims) const = 0;
 
  virtual void def_var_chunking_impl(const std::string &name,
                                     std::vector<size_t> &dimensions) const;
 
  virtual void get_vara_double_impl(const std::string &variable_name,
                                    const std::vector<unsigned int> &start,
                                    const std::vector<unsigned int> &count, double *ip) const = 0;
 
  virtual void put_vara_double_impl(const std::string &variable_name,
                                    const std::vector<unsigned int> &start,
                                    const std::vector<unsigned int> &count,
                                    const double *op) const = 0;
 
  virtual void write_darray_impl(const std::string &variable_name, const Grid &grid,
                                 unsigned int z_count, bool time_dependent, unsigned int record,
                                 const double *input);
 
  virtual void get_varm_double_impl(const std::string &variable_name,
                                    const std::vector<unsigned int> &start,
                                    const std::vector<unsigned int> &count,
                                    const std::vector<unsigned int> &imap, double *ip) const = 0;
 
  virtual void inq_nvars_impl(int &result) const = 0;
 
  virtual void inq_vardimid_impl(const std::string &variable_name,
                                 std::vector<std::string> &result) const = 0;
 
  virtual void inq_varnatts_impl(const std::string &variable_name, int &result) const = 0;
 
  virtual void inq_varid_impl(const std::string &variable_name, bool &exists) const = 0;
 
  virtual void inq_varname_impl(unsigned int j, std::string &result) const = 0;
 
  virtual void set_compression_level_impl(int level) const = 0;
 
  // att
  virtual void get_att_double_impl(const std::string &variable_name, const std::string &att_name,
                                   std::vector<double> &result) const = 0;
 
  virtual void get_att_text_impl(const std::string &variable_name, const std::string &att_name,
                                 std::string &result) const = 0;
 
  virtual void put_att_double_impl(const std::string &variable_name, const std::string &att_name,
                                   io::Type xtype, const std::vector<double> &data) const = 0;
 
  virtual void put_att_text_impl(const std::string &variable_name, const std::string &att_name,
                                 const std::string &value) const = 0;
 
  virtual void inq_attname_impl(const std::string &variable_name, unsigned int n,
                                std::string &result) const = 0;
 
  virtual void inq_atttype_impl(const std::string &variable_name, const std::string &att_name,
                                io::Type &result) const = 0;
 
  // misc
  virtual void set_fill_impl(int fillmode, int &old_modep) const = 0;
 
  virtual void del_att_impl(const std::string &variable_name, const std::string &att_name) const = 0;
 
protected:                      // data members
 
  MPI_Comm m_com;
  int m_file_id;
  std::string m_filename;
private:
  mutable bool m_define_mode;
};
 
} // end of namespace io
} // end of namespace pism
 
#endif /* PISM_NCFILE_H */