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 */