nc_io_basic_types.cc

Go to the documentation of this file.

/* Copyright (C) 2003-2012 Oliver Lemke <olemke@core-dump.info>

   This program 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 2, or (at your option) any
   later version.

   This program 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 this program; if not, write to the Free Software
   Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307,
   USA. */

//   File description

#include "arts.h"

#ifdef ENABLE_NETCDF

#include "nc_io.h"
#include "nc_io_types.h"

//=== Matrix ==========================================================


void nca_read_from_file(const int ncid, Matrix& m, const Verbosity&) {
  Index nrows, ncols;
  nrows = nc_get_dim(ncid, "nrows");
  ncols = nc_get_dim(ncid, "ncols");

  m.resize(nrows, ncols);
  nca_get_data_double(ncid, "Matrix", m.get_c_array());
}


void nca_write_to_file(const int ncid, const Matrix& m, const Verbosity&) {
  int retval;
  int ncdims[2], varid;
  if ((retval = nc_def_dim(ncid, "nrows", m.nrows(), &ncdims[0])))
    nca_error(retval, "nc_def_dim");
  if ((retval = nc_def_dim(ncid, "ncols", m.ncols(), &ncdims[1])))
    nca_error(retval, "nc_def_dim");
  if ((retval = nc_def_var(ncid, "Matrix", NC_DOUBLE, 2, &ncdims[0], &varid)))
    nca_error(retval, "nc_def_var");
  if ((retval = nc_enddef(ncid))) nca_error(retval, "nc_enddef");
  if ((retval = nc_put_var_double(ncid, varid, m.get_c_array())))
    nca_error(retval, "nc_put_var");
}

//=== Tensor3 ==========================================================


void nca_read_from_file(const int ncid, Tensor3& t, const Verbosity&) {
  Index npages, nrows, ncols;
  npages = nc_get_dim(ncid, "npages");
  nrows = nc_get_dim(ncid, "nrows");
  ncols = nc_get_dim(ncid, "ncols");

  t.resize(npages, nrows, ncols);
  nca_get_data_double(ncid, "Tensor3", t.get_c_array());
}


void nca_write_to_file(const int ncid, const Tensor3& t, const Verbosity&) {
  int retval;
  int ncdims[3], varid;
  if ((retval = nc_def_dim(ncid, "npages", t.npages(), &ncdims[0])))
    nca_error(retval, "nc_def_dim");
  if ((retval = nc_def_dim(ncid, "nrows", t.nrows(), &ncdims[1])))
    nca_error(retval, "nc_def_dim");
  if ((retval = nc_def_dim(ncid, "ncols", t.ncols(), &ncdims[2])))
    nca_error(retval, "nc_def_dim");
  if ((retval = nc_def_var(ncid, "Tensor3", NC_DOUBLE, 3, &ncdims[0], &varid)))
    nca_error(retval, "nc_def_var");
  if ((retval = nc_enddef(ncid))) nca_error(retval, "nc_enddef");
  if ((retval = nc_put_var_double(ncid, varid, t.get_c_array())))
    nca_error(retval, "nc_put_var");
}

//=== Tensor4 ==========================================================


void nca_read_from_file(const int ncid, Tensor4& t, const Verbosity&) {
  Index nbooks, npages, nrows, ncols;
  nbooks = nc_get_dim(ncid, "nbooks");
  npages = nc_get_dim(ncid, "npages");
  nrows = nc_get_dim(ncid, "nrows");
  ncols = nc_get_dim(ncid, "ncols");

  t.resize(nbooks, npages, nrows, ncols);
  nca_get_data_double(ncid, "Tensor4", t.get_c_array());
}


void nca_write_to_file(const int ncid, const Tensor4& t, const Verbosity&) {
  int retval;
  int ncdims[4], varid;
  if ((retval = nc_def_dim(ncid, "nbooks", t.nbooks(), &ncdims[0])))
    nca_error(retval, "nc_def_dim");
  if ((retval = nc_def_dim(ncid, "npages", t.npages(), &ncdims[1])))
    nca_error(retval, "nc_def_dim");
  if ((retval = nc_def_dim(ncid, "nrows", t.nrows(), &ncdims[2])))
    nca_error(retval, "nc_def_dim");
  if ((retval = nc_def_dim(ncid, "ncols", t.ncols(), &ncdims[3])))
    nca_error(retval, "nc_def_dim");
  if ((retval = nc_def_var(ncid, "Tensor4", NC_DOUBLE, 4, &ncdims[0], &varid)))
    nca_error(retval, "nc_def_var");
  if ((retval = nc_enddef(ncid))) nca_error(retval, "nc_enddef");
  if ((retval = nc_put_var_double(ncid, varid, t.get_c_array())))
    nca_error(retval, "nc_put_var");
}

//=== Tensor5 ==========================================================


void nca_read_from_file(const int ncid, Tensor5& t, const Verbosity&) {
  Index nshelves, nbooks, npages, nrows, ncols;
  nshelves = nc_get_dim(ncid, "nshelves");
  nbooks = nc_get_dim(ncid, "nbooks");
  npages = nc_get_dim(ncid, "npages");
  nrows = nc_get_dim(ncid, "nrows");
  ncols = nc_get_dim(ncid, "ncols");

  t.resize(nshelves, nbooks, npages, nrows, ncols);
  nca_get_data_double(ncid, "Tensor5", t.get_c_array());
}


void nca_write_to_file(const int ncid, const Tensor5& t, const Verbosity&) {
  int retval;
  int ncdims[5], varid;
  if ((retval = nc_def_dim(ncid, "nshelves", t.nshelves(), &ncdims[0])))
    nca_error(retval, "nc_def_dim");
  if ((retval = nc_def_dim(ncid, "nbooks", t.nbooks(), &ncdims[1])))
    nca_error(retval, "nc_def_dim");
  if ((retval = nc_def_dim(ncid, "npages", t.npages(), &ncdims[2])))
    nca_error(retval, "nc_def_dim");
  if ((retval = nc_def_dim(ncid, "nrows", t.nrows(), &ncdims[3])))
    nca_error(retval, "nc_def_dim");
  if ((retval = nc_def_dim(ncid, "ncols", t.ncols(), &ncdims[4])))
    nca_error(retval, "nc_def_dim");
  if ((retval = nc_def_var(ncid, "Tensor5", NC_DOUBLE, 5, &ncdims[0], &varid)))
    nca_error(retval, "nc_def_var");
  if ((retval = nc_enddef(ncid))) nca_error(retval, "nc_enddef");
  if ((retval = nc_put_var_double(ncid, varid, t.get_c_array())))
    nca_error(retval, "nc_put_var");
}

//=== Vector ==========================================================


void nca_read_from_file(const int ncid, Vector& v, const Verbosity&) {
  Index nelem;
  nelem = nc_get_dim(ncid, "nelem");

  v.resize(nelem);
  nca_get_data_double(ncid, "Vector", v.get_c_array());
}


void nca_write_to_file(const int ncid, const Vector& v, const Verbosity&) {
  int retval;
  int ncdim, varid;
  if ((retval = nc_def_dim(ncid, "nelem", v.nelem(), &ncdim)))
    nca_error(retval, "nc_def_dim");
  if ((retval = nc_def_var(ncid, "Vector", NC_DOUBLE, 1, &ncdim, &varid)))
    nca_error(retval, "nc_def_var");
  if ((retval = nc_enddef(ncid))) nca_error(retval, "nc_enddef");
  if ((retval = nc_put_var_double(ncid, varid, v.get_c_array())))
    nca_error(retval, "nc_put_var");
}

//   Dummy funtion for groups for which
//   IO function have not yet been implemented

#define TMPL_NC_READ_WRITE_FILE_DUMMY(what)                                   \
  void nca_write_to_file(const int, const what&, const Verbosity&) {          \
    throw runtime_error("NetCDF support not yet implemented for this type!"); \
  }                                                                           \
  void nca_read_from_file(const int, what&, const Verbosity&) {               \
    throw runtime_error("NetCDF support not yet implemented for this type!"); \
  }

//==========================================================================

// Undefine the macro to avoid it being used anywhere else
#undef TMPL_NC_READ_WRITE_FILE_DUMMY

#endif /* ENABLE_NETCDF */