nc_io_array_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"

//=== ArrayOfMatrix ==========================================================


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

  long* vnrows = new long[nelem];
  long* vncols = new long[nelem];
  aom.resize(nelem);
  nca_get_data_long(ncid, "Matrix_nrows", vnrows);
  nca_get_data_long(ncid, "Matrix_ncols", vncols);
  size_t pos = 0;
  for (Index i = 0; i < nelem; i++) {
    aom[i].resize(vnrows[i], vncols[i]);
    nca_get_dataa_double(ncid,
                         "ArrayOfMatrix",
                         pos,
                         vnrows[i] * vncols[i],
                         aom[i].get_c_array());
    pos += vnrows[i] * vncols[i];
  }

  delete[] vnrows;
  delete[] vncols;
}


void nca_write_to_file(const int ncid,
                       const ArrayOfMatrix& aom,
                       const Verbosity&) {
  int retval;
  int ncdim, varid_nrows, varid_ncols;
  int ncdim_total, varid;
  long nelem_total = 0;
  long* vncols = new long[aom.nelem()];
  long* vnrows = new long[aom.nelem()];
  for (Index i = 0; i < aom.nelem(); i++) {
    vnrows[i] = aom[i].nrows();
    vncols[i] = aom[i].ncols();
    nelem_total += vnrows[i] * vncols[i];
  }

  if ((retval = nc_def_dim(ncid, "nelem", aom.nelem(), &ncdim)))
    nca_error(retval, "nc_def_dim");
  if ((retval = nc_def_dim(ncid, "nelem_total", nelem_total, &ncdim_total)))
    nca_error(retval, "nc_def_dim");

  if ((retval =
           nc_def_var(ncid, "Matrix_nrows", NC_LONG, 1, &ncdim, &varid_nrows)))
    nca_error(retval, "nc_def_var");
  if ((retval =
           nc_def_var(ncid, "Matrix_ncols", NC_LONG, 1, &ncdim, &varid_ncols)))
    nca_error(retval, "nc_def_var");
  if ((retval = nc_def_var(
           ncid, "ArrayOfMatrix", NC_DOUBLE, 1, &ncdim_total, &varid)))
    nca_error(retval, "nc_def_var");

  if ((retval = nc_enddef(ncid))) nca_error(retval, "nc_enddef");

  if ((retval = nc_put_var_long(ncid, varid_nrows, vnrows)))
    nca_error(retval, "nc_put_var");
  if ((retval = nc_put_var_long(ncid, varid_ncols, vncols)))
    nca_error(retval, "nc_put_var");

  size_t pos = 0;
  for (Index i = 0; i < aom.nelem(); i++) {
    size_t count = aom[i].nrows() * aom[i].ncols();
    if ((retval = nc_put_vara_double(
             ncid, varid, &pos, &count, aom[i].get_c_array())))
      nca_error(retval, "nc_put_var");
    pos += count;
  }

  delete[] vnrows;
  delete[] vncols;
}

//=== ArrayOfVector ==========================================================


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

  long* vnelem = new long[nelem];
  aov.resize(nelem);
  nca_get_data_long(ncid, "Vector_nelem", vnelem);
  size_t pos = 0;
  for (Index i = 0; i < nelem; i++) {
    aov[i].resize(vnelem[i]);
    nca_get_dataa_double(
        ncid, "ArrayOfVector", pos, vnelem[i], aov[i].get_c_array());
    pos += vnelem[i];
  }

  delete[] vnelem;
}


void nca_write_to_file(const int ncid,
                       const ArrayOfVector& aov,
                       const Verbosity&) {
  int retval;
  int ncdim, varid_nelem;
  int ncdim_total, varid;
  long nelem_total = 0;
  long* velems = new long[aov.nelem()];
  for (Index i = 0; i < aov.nelem(); i++) {
    velems[i] = aov[i].nelem();
    nelem_total += velems[i];
  }

  if ((retval = nc_def_dim(ncid, "nelem", aov.nelem(), &ncdim)))
    nca_error(retval, "nc_def_dim");
  if ((retval = nc_def_dim(ncid, "nelem_total", nelem_total, &ncdim_total)))
    nca_error(retval, "nc_def_dim");

  if ((retval =
           nc_def_var(ncid, "Vector_nelem", NC_LONG, 1, &ncdim, &varid_nelem)))
    nca_error(retval, "nc_def_var");
  if ((retval = nc_def_var(
           ncid, "ArrayOfVector", NC_DOUBLE, 1, &ncdim_total, &varid)))
    nca_error(retval, "nc_def_var");

  if ((retval = nc_enddef(ncid))) nca_error(retval, "nc_enddef");

  if ((retval = nc_put_var_long(ncid, varid_nelem, velems)))
    nca_error(retval, "nc_put_var");

  size_t pos = 0;
  for (Index i = 0; i < aov.nelem(); i++) {
    size_t count = aov[i].nelem();
    if ((retval = nc_put_vara_double(
             ncid, varid, &pos, &count, aov[i].get_c_array())))
      nca_error(retval, "nc_put_var");
    pos += count;
  }

  delete[] velems;
}

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