nc_io.cc

Go to the documentation of this file.

/* Copyright (C) 2008 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"
#include "nc_io.h"
#include "nc_io_types.h"
#include "nc_io_instantiation.h"
#include "file.h"
#include "messages.h"
#include "exceptions.h"


//   Default file name


void
filename_nc (      String&  filename,
             const String&  varname )
{
  if ("" == filename)
    {
      extern const String out_basename;
      filename = out_basename + "." + varname + ".nc";
    }
}




void
filename_nc_with_index (
                    String&  filename,
              const Index&   file_index,
              const String&  varname )
{
  Index dummy = file_index;
  dummy += 1;

  if ("" == filename)
    {
      extern const String out_basename;
      ostringstream os;
      os << out_basename << "." << varname << "." << file_index << ".nc";
      filename = os.str();
    }
  else
    {
      ostringstream os;
      os << filename << "." << file_index << ".nc";
      filename = os.str();
    }
}


template<typename T> void
nc_read_from_file (const String& filename,
                         T&      type)
{
  out2 << "  Reading " << filename << '\n';

  int ncid;
  if (nc_open (filename.c_str(), NC_NOWRITE, &ncid))
    {
      ostringstream os;
      os << "Error reading file: " << filename << endl;
      throw runtime_error (os.str());
    }

  nc_read_from_file (ncid, type);

  nc_close (ncid);
}


template<typename T> void
nc_write_to_file (const String&  filename,
                  const      T&  type)
{
  out2 << "  Writing " << filename << '\n';

  int ncid;
  if (nc_create (filename.c_str(), NC_CLOBBER | NC_FORMAT_NETCDF4, &ncid))
    {
      ostringstream os;
      os << "Error writing file: " << filename << endl;
      throw runtime_error (os.str());
    }

  nc_write_to_file (ncid, type);

  nc_close (ncid);
}


void nc_get_data_int (const int ncid, const String &name, int *data)
{
  int retval, varid;
  if ((retval = nc_inq_varid (ncid, name.c_str(), &varid)))
    ncerror (retval, "nc_inq_varid("+name+")");
  if ((retval = nc_get_var_int (ncid, varid, data)))
    ncerror (retval, "nc_get_var("+name+")");
}


void nc_get_data_long (const int ncid, const String &name, long *data)
{
  int retval, varid;
  if ((retval = nc_inq_varid (ncid, name.c_str(), &varid)))
    ncerror (retval, "nc_inq_varid("+name+")");
  if ((retval = nc_get_var_long (ncid, varid, data)))
    ncerror (retval, "nc_get_var("+name+")");
}


void nc_get_data_double (const int ncid, const String &name, Numeric *data)
{
  int retval, varid;
  if ((retval = nc_inq_varid (ncid, name.c_str(), &varid)))
    ncerror (retval, "nc_inq_varid("+name+")");
  if ((retval = nc_get_var_double (ncid, varid, data)))
    ncerror (retval, "nc_get_var("+name+")");
}


void nc_get_dataa_double (const int ncid, const String &name,
                          size_t start, size_t count, Numeric *data)
{
  int retval, varid;
  if ((retval = nc_inq_varid (ncid, name.c_str(), &varid)))
    ncerror (retval, "nc_inq_varid("+name+")");
  if ((retval = nc_get_vara_double (ncid, varid, &start, &count, data)))
    ncerror (retval, "nc_get_var("+name+")");
}


Index nc_get_dim (const int ncid, const String &name)
{
  int retval, dimid;
  size_t ndim;
  if ((retval = nc_inq_dimid (ncid, name.c_str(), &dimid)))
    ncerror (retval, "nc_inq_ndims("+name+")");
  if ((retval = nc_inq_dimlen (ncid, dimid, &ndim)))
    ncerror (retval, "nc_inq_dimlen("+name+")");

  return (Index)ndim;
}


void ncerror (const int e, const String s)
{
  ostringstream os;
  os << "NetCDF error: " << s << ", " << e;
  throw runtime_error (os.str());
}

---