netcdf-c/libsrcp/ncpdispatch.c

1487 lines
48 KiB
C
Raw Normal View History

2015-08-17 11:44:18 +08:00
/*********************************************************************
2018-12-07 05:56:42 +08:00
* Copyright 2018, UCAR/Unidata
2013-03-16 04:31:07 +08:00
* See netcdf/COPYRIGHT file for copying and redistribution conditions.
*********************************************************************/
/* WARNING: Order of mpi.h, nc.h, and pnetcdf.h is important */
#include "config.h"
#include <stdlib.h>
#include <mpi.h>
#include "nc.h"
#include "ncdispatch.h"
#include "fbits.h"
2013-03-16 04:31:07 +08:00
/* Must follow netcdf.h */
#include <pnetcdf.h>
#define NCP_MODE_DATA 0x0001
#define NCP_MODE_INDEP 0x0002
typedef struct NCP_INFO
2013-03-16 04:31:07 +08:00
{
/* pnetcdf_access_mode keeps track of whether independent or collective
* mode is set currently and whether the file is in define or data mode.
*/
2013-03-16 04:31:07 +08:00
int pnetcdf_access_mode;
} NCP_INFO;
2013-03-16 04:31:07 +08:00
/* Define accessors for the dispatchdata */
#define NCP_DATA(nc) ((NCP_INFO*)(nc)->dispatchdata)
#define NCP_DATA_SET(nc,data) ((nc)->dispatchdata = (void*)(data))
2013-03-16 04:31:07 +08:00
2018-09-23 09:22:34 +08:00
/* NC_MPIIO and NC_MPIPOSIX are deprecated and hence ignored */
static const int LEGAL_CREATE_FLAGS = (NC_WRITE | NC_NOCLOBBER | NC_64BIT_OFFSET | NC_CLASSIC_MODEL | NC_SHARE | NC_LOCK | NC_64BIT_DATA | NC_MPIIO | NC_MPIPOSIX);
2013-03-16 04:31:07 +08:00
2018-09-23 09:22:34 +08:00
static const int LEGAL_OPEN_FLAGS = (NC_WRITE | NC_NOCLOBBER | NC_SHARE | NC_LOCK | NC_CLASSIC_MODEL | NC_64BIT_OFFSET | NC_64BIT_DATA | NC_MPIIO | NC_MPIPOSIX);
2013-03-16 04:31:07 +08:00
/**************************************************/
static int
NCP_create(const char *path,
int cmode,
size_t initialsz,
int basepe,
size_t *chunksizehintp,
void *mpidata,
const struct NC_Dispatch *table,
2019-08-02 04:30:20 +08:00
int ncid)
2013-03-16 04:31:07 +08:00
{
int status;
NCP_INFO *nc5;
NC *nc;
/* Find pointer to NC for this file. */
2019-08-02 04:30:20 +08:00
status = NC_check_id(ncid, &nc);
if (status != NC_NOERR) return status;
2013-03-16 04:31:07 +08:00
/* Check the cmode for only valid flags */
if (cmode & ~LEGAL_CREATE_FLAGS) return NC_EINVAL;
2013-03-16 04:31:07 +08:00
2015-08-17 11:44:18 +08:00
/* No MPI environment initialized */
if (mpidata == NULL) return NC_ENOPAR;
/* Create NCP_INFO instance */
2015-08-17 11:44:18 +08:00
nc5 = (NCP_INFO*)calloc(1,sizeof(NCP_INFO));
if (nc5 == NULL) return NC_ENOMEM;
2013-03-16 04:31:07 +08:00
2015-08-17 11:44:18 +08:00
/* Link nc5 and nc */
NCP_DATA_SET(nc,nc5);
2013-03-16 04:31:07 +08:00
status = ncmpi_create(((NC_MPI_INFO *)mpidata)->comm, path, cmode,
((NC_MPI_INFO *)mpidata)->info, &nc->int_ncid);
if (status == NC_NOERR)
/* Default to independent access, like netCDF-4/HDF5 files. */
fSet(nc5->pnetcdf_access_mode, NCP_MODE_INDEP);
else
free(nc5); /* reclaim allocated space */
return status;
2013-03-16 04:31:07 +08:00
}
static int
NCP_open(const char *path,
int omode,
int basepe,
size_t *chunksizehintp,
void *mpidata,
const struct NC_Dispatch *table,
2019-08-02 04:30:20 +08:00
int ncid)
2013-03-16 04:31:07 +08:00
{
int status;
NCP_INFO *nc5;
NC *nc;
/* Find pointer to NC for this file. */
2019-08-02 04:30:20 +08:00
status = NC_check_id(ncid, &nc);
if (status != NC_NOERR) return status;
2013-03-16 04:31:07 +08:00
/* Check the omode for only valid flags */
if (omode & ~LEGAL_OPEN_FLAGS) return NC_EINVAL;
2013-03-16 04:31:07 +08:00
/* No MPI environment initialized */
if (mpidata == NULL) return NC_ENOPAR;
2013-03-16 04:31:07 +08:00
/* Create NCP_INFO instance */
2015-08-17 11:44:18 +08:00
nc5 = (NCP_INFO*)calloc(1,sizeof(NCP_INFO));
if (nc5 == NULL) return NC_ENOMEM;
2013-03-16 04:31:07 +08:00
/* file open automatically enters data mode */
fSet(nc5->pnetcdf_access_mode, NCP_MODE_DATA);
2015-08-17 11:44:18 +08:00
/* Link nc5 and nc */
NCP_DATA_SET(nc,nc5);
2013-03-16 04:31:07 +08:00
status = ncmpi_open(((NC_MPI_INFO *)mpidata)->comm, path, omode,
((NC_MPI_INFO *)mpidata)->info, &(nc->int_ncid));
if (status == NC_NOERR) {
/* Default to independent access, like netCDF-4/HDF5 files. */
status = ncmpi_begin_indep_data(nc->int_ncid);
fSet(nc5->pnetcdf_access_mode, NCP_MODE_INDEP);
2013-03-16 04:31:07 +08:00
}
else
free(nc5); /* reclaim allocated space */
return status;
2013-03-16 04:31:07 +08:00
}
static int
NCP_redef(int ncid)
2013-03-16 04:31:07 +08:00
{
NC *nc;
NCP_INFO *nc5;
2013-03-16 04:31:07 +08:00
int status = NC_check_id(ncid, &nc);
if (status != NC_NOERR) return status;
nc5 = NCP_DATA(nc);
/* the file now enters define mode */
fClr(nc5->pnetcdf_access_mode, NCP_MODE_DATA);
2013-03-16 04:31:07 +08:00
return ncmpi_redef(nc->int_ncid);
}
static int
NCP__enddef(int ncid,
size_t h_minfree,
size_t v_align,
size_t v_minfree,
size_t r_align)
2013-03-16 04:31:07 +08:00
{
int status;
NC *nc;
NCP_INFO *nc5;
2013-03-16 04:31:07 +08:00
status = NC_check_id(ncid, &nc);
if (status != NC_NOERR) return status;
2013-03-16 04:31:07 +08:00
2015-08-17 11:44:18 +08:00
nc5 = NCP_DATA(nc);
assert(nc5);
2013-03-16 04:31:07 +08:00
#if (PNETCDF_VERSION_MAJOR*10000 + PNETCDF_VERSION_MINOR*100 + PNETCDF_VERSION_SUB >= 10500)
/* ncmpi__enddef() was first implemented in PnetCDF v1.5.0 */
status = ncmpi__enddef(nc->int_ncid,
(MPI_Offset)h_minfree,
(MPI_Offset)v_align,
(MPI_Offset)v_minfree,
(MPI_Offset)r_align);
2015-08-17 11:44:18 +08:00
#else
2013-03-16 04:31:07 +08:00
status = ncmpi_enddef(nc->int_ncid);
2015-08-17 11:44:18 +08:00
#endif
if (!status) {
/* the file now enters data mode */
fSet(nc5->pnetcdf_access_mode, NCP_MODE_DATA);
/* default is independent data mode */
if (fIsSet(nc5->pnetcdf_access_mode, NCP_MODE_INDEP))
status = ncmpi_begin_indep_data(nc->int_ncid);
2013-03-16 04:31:07 +08:00
}
return status;
}
static int
NCP_sync(int ncid)
2013-03-16 04:31:07 +08:00
{
NC *nc;
2013-03-16 04:31:07 +08:00
int status = NC_check_id(ncid, &nc);
if (status != NC_NOERR) return status;
2013-03-16 04:31:07 +08:00
return ncmpi_sync(nc->int_ncid);
}
static int
NCP_abort(int ncid)
2013-03-16 04:31:07 +08:00
{
NC *nc;
NCP_INFO *nc5;
2013-03-16 04:31:07 +08:00
int status = NC_check_id(ncid, &nc);
if (status != NC_NOERR) return status;
status = ncmpi_abort(nc->int_ncid);
2015-08-17 11:44:18 +08:00
nc5 = NCP_DATA(nc);
if (nc5 != NULL) free(nc5); /* reclaim allocated space */
return status;
2013-03-16 04:31:07 +08:00
}
static int
NCP_close(int ncid, void *ignored)
2013-03-16 04:31:07 +08:00
{
NC *nc;
NCP_INFO *nc5;
2013-03-16 04:31:07 +08:00
int status = NC_check_id(ncid, &nc);
if (status != NC_NOERR) return status;
status = ncmpi_close(nc->int_ncid);
2015-08-17 11:44:18 +08:00
nc5 = NCP_DATA(nc);
if (nc5 != NULL) free(nc5); /* reclaim allocated space */
return status;
2013-03-16 04:31:07 +08:00
}
static int
NCP_set_fill(int ncid, int fillmode, int *old_mode_ptr)
2013-03-16 04:31:07 +08:00
{
NC *nc;
2013-03-16 04:31:07 +08:00
int status = NC_check_id(ncid, &nc);
if (status != NC_NOERR) return status;
#if (PNETCDF_VERSION_MAJOR*10000 + PNETCDF_VERSION_MINOR*100 + PNETCDF_VERSION_SUB >= 10601)
/* ncmpi_set_fill was first implemented in PnetCDF 1.6.1 */
return ncmpi_set_fill(nc->int_ncid, fillmode, old_mode_ptr);
#else
return NC_EPNETCDF;
#endif
2013-03-16 04:31:07 +08:00
}
static int
NCP_inq_format(int ncid, int *formatp)
2013-03-16 04:31:07 +08:00
{
NC *nc;
2013-03-16 04:31:07 +08:00
int status = NC_check_id(ncid, &nc);
if (status != NC_NOERR) return status;
return ncmpi_inq_format(nc->int_ncid, formatp);
2013-03-16 04:31:07 +08:00
}
static int
NCP_inq_format_extended(int ncid, int *formatp, int *modep)
{
/* Note PnetCDF understands classic, 64bit-offset, and CDF-5 formats */
NC *nc;
int status = NC_check_id(ncid, &nc);
if (status != NC_NOERR) return status;
if (modep) *modep = nc->mode;
if (formatp) *formatp = NC_FORMATX_PNETCDF;
return NC_NOERR;
}
2013-03-16 04:31:07 +08:00
static int
NCP_inq(int ncid,
int *ndimsp,
int *nvarsp,
int *nattsp,
int *unlimp)
2013-03-16 04:31:07 +08:00
{
NC *nc;
2013-03-16 04:31:07 +08:00
int status = NC_check_id(ncid, &nc);
if (status != NC_NOERR) return status;
return ncmpi_inq(nc->int_ncid, ndimsp, nvarsp, nattsp, unlimp);
2013-03-16 04:31:07 +08:00
}
static int
NCP_inq_type(int ncid, nc_type typeid, char *name, size_t *size)
2013-03-16 04:31:07 +08:00
{
/* for non-netcdf4 files, no user defined type is allowed */
2015-08-17 11:44:18 +08:00
if (typeid < NC_BYTE || typeid >= NC_STRING)
return NC_EBADTYPE;
if (name)
strcpy(name, NC_atomictypename(typeid));
if (size)
2015-08-17 11:44:18 +08:00
*size = NC_atomictypelen(typeid);
return NC_NOERR;
2013-03-16 04:31:07 +08:00
}
static int
NCP_def_dim(int ncid, const char *name, size_t len, int *idp)
2013-03-16 04:31:07 +08:00
{
NC *nc;
int status = NC_check_id(ncid, &nc);
if (status != NC_NOERR) return status;
2013-03-16 04:31:07 +08:00
return ncmpi_def_dim(nc->int_ncid, name, len, idp);
}
static int
NCP_inq_dimid(int ncid, const char *name, int *idp)
2013-03-16 04:31:07 +08:00
{
NC *nc;
2013-03-16 04:31:07 +08:00
int status = NC_check_id(ncid, &nc);
if (status != NC_NOERR) return status;
return ncmpi_inq_dimid(nc->int_ncid, name, idp);
2013-03-16 04:31:07 +08:00
}
static int
NCP_inq_dim(int ncid, int dimid, char *name, size_t *lenp)
2013-03-16 04:31:07 +08:00
{
int status;
NC *nc;
2013-03-16 04:31:07 +08:00
MPI_Offset mpilen;
status = NC_check_id(ncid, &nc);
if (status != NC_NOERR) return status;
status = ncmpi_inq_dim(nc->int_ncid, dimid, name, &mpilen);
if (status == NC_NOERR && lenp) *lenp = mpilen;
2013-03-16 04:31:07 +08:00
return status;
}
static int
NCP_inq_unlimdim(int ncid, int *unlimdimidp)
2013-03-16 04:31:07 +08:00
{
NC *nc;
2013-03-16 04:31:07 +08:00
int status = NC_check_id(ncid, &nc);
if (status != NC_NOERR) return status;
return ncmpi_inq_unlimdim(nc->int_ncid, unlimdimidp);
2013-03-16 04:31:07 +08:00
}
static int
NCP_rename_dim(int ncid, int dimid, const char *newname)
2013-03-16 04:31:07 +08:00
{
NC *nc;
2013-03-16 04:31:07 +08:00
int status = NC_check_id(ncid, &nc);
if (status != NC_NOERR) return status;
return ncmpi_rename_dim(nc->int_ncid, dimid, newname);
2013-03-16 04:31:07 +08:00
}
static int
NCP_inq_att(int ncid,
int varid,
const char *name,
nc_type *xtypep,
size_t *lenp)
2013-03-16 04:31:07 +08:00
{
NC *nc;
2013-03-16 04:31:07 +08:00
MPI_Offset mpilen;
int status = NC_check_id(ncid, &nc);
if (status != NC_NOERR) return status;
status = ncmpi_inq_att(nc->int_ncid, varid, name, xtypep, &mpilen);
if (status == NC_NOERR && lenp) *lenp = mpilen;
2013-03-16 04:31:07 +08:00
return status;
}
static int
NCP_inq_attid(int ncid, int varid, const char *name, int *idp)
2013-03-16 04:31:07 +08:00
{
NC *nc;
2013-03-16 04:31:07 +08:00
int status = NC_check_id(ncid, &nc);
if (status != NC_NOERR) return status;
return ncmpi_inq_attid(nc->int_ncid,varid, name, idp);
2013-03-16 04:31:07 +08:00
}
static int
NCP_inq_attname(int ncid, int varid, int attnum, char *name)
2013-03-16 04:31:07 +08:00
{
NC *nc;
2013-03-16 04:31:07 +08:00
int status = NC_check_id(ncid, &nc);
if (status != NC_NOERR) return status;
return ncmpi_inq_attname(nc->int_ncid, varid, attnum, name);
2013-03-16 04:31:07 +08:00
}
static int
NCP_rename_att(int ncid, int varid, const char *name,
const char *newname)
2013-03-16 04:31:07 +08:00
{
NC *nc;
2013-03-16 04:31:07 +08:00
int status = NC_check_id(ncid, &nc);
if (status != NC_NOERR) return status;
return ncmpi_rename_att(nc->int_ncid, varid, name, newname);
2013-03-16 04:31:07 +08:00
}
static int
NCP_del_att(int ncid, int varid, const char *name)
2013-03-16 04:31:07 +08:00
{
NC *nc;
2013-03-16 04:31:07 +08:00
int status = NC_check_id(ncid, &nc);
if (status != NC_NOERR) return status;
return ncmpi_del_att(nc->int_ncid, varid, name);
2013-03-16 04:31:07 +08:00
}
static int
NCP_get_att(int ncid,
int varid,
const char *name,
void *op,
nc_type memtype)
2013-03-16 04:31:07 +08:00
{
NC *nc;
2013-03-16 04:31:07 +08:00
int status;
status = NC_check_id(ncid, &nc);
if (status != NC_NOERR) return status;
2013-03-16 04:31:07 +08:00
if (memtype == NC_NAT) {
status = ncmpi_inq_att(nc->int_ncid, varid, name, &memtype, NULL);
if (status != NC_NOERR) return status;
}
2013-03-16 04:31:07 +08:00
switch (memtype) {
case NC_CHAR:
return ncmpi_get_att_text(nc->int_ncid, varid, name, (char*)op);
case NC_BYTE:
return ncmpi_get_att_schar(nc->int_ncid, varid, name, (signed char*)op);
case NC_SHORT:
return ncmpi_get_att_short(nc->int_ncid, varid, name, (short*)op);
case NC_INT:
return ncmpi_get_att_int(nc->int_ncid, varid, name, (int*)op);
case NC_FLOAT:
return ncmpi_get_att_float(nc->int_ncid, varid, name, (float*)op);
case NC_DOUBLE:
return ncmpi_get_att_double(nc->int_ncid, varid, name, (double*)op);
case NC_UBYTE:
return ncmpi_get_att_uchar(nc->int_ncid, varid, name, (unsigned char*)op);
case NC_USHORT:
return ncmpi_get_att_ushort(nc->int_ncid, varid, name, (unsigned short*)op);
case NC_UINT:
return ncmpi_get_att_uint(nc->int_ncid, varid, name, (unsigned int*)op);
case NC_INT64:
return ncmpi_get_att_longlong(nc->int_ncid, varid, name, (long long*)op);
case NC_UINT64:
return ncmpi_get_att_ulonglong(nc->int_ncid, varid, name, (unsigned long long*)op);
default:
return NC_EBADTYPE;
2013-03-16 04:31:07 +08:00
}
}
static int
NCP_put_att(int ncid,
int varid,
const char *name,
nc_type xtype,
size_t len,
const void *ip,
nc_type memtype)
2013-03-16 04:31:07 +08:00
{
NC *nc;
2013-03-16 04:31:07 +08:00
int status;
MPI_Offset mpilen;
/* check if ncid is valid */
status = NC_check_id(ncid, &nc);
if (status != NC_NOERR) return status;
/* check if varid is valid */
status = ncmpi_inq_varnatts(nc->int_ncid, varid, NULL);
if (status != NC_NOERR) return status;
2013-03-16 04:31:07 +08:00
if (!name || (strlen(name) > NC_MAX_NAME))
return NC_EBADNAME;
2013-03-16 04:31:07 +08:00
/* The length needs to be positive (cast needed for braindead
systems with signed size_t). */
if (((unsigned long) len) > X_INT_MAX)
return NC_EINVAL;
2013-03-16 04:31:07 +08:00
mpilen = len;
switch (memtype) {
case NC_CHAR:
return ncmpi_put_att_text(nc->int_ncid, varid, name, mpilen, (char*)ip);
case NC_BYTE:
return ncmpi_put_att_schar(nc->int_ncid, varid, name, xtype, mpilen, (signed char*)ip);
case NC_SHORT:
return ncmpi_put_att_short(nc->int_ncid, varid, name, xtype, mpilen, (short*)ip);
case NC_INT:
return ncmpi_put_att_int(nc->int_ncid, varid, name, xtype, mpilen, (int*)ip);
case NC_FLOAT:
return ncmpi_put_att_float(nc->int_ncid, varid, name, xtype, mpilen, (float*)ip);
case NC_DOUBLE:
return ncmpi_put_att_double(nc->int_ncid, varid, name, xtype, mpilen, (double*)ip);
case NC_UBYTE:
return ncmpi_put_att_uchar(nc->int_ncid, varid, name, xtype, mpilen, (unsigned char*)ip);
case NC_USHORT:
return ncmpi_put_att_ushort(nc->int_ncid, varid, name, xtype, mpilen, (unsigned short*)ip);
case NC_UINT:
return ncmpi_put_att_uint(nc->int_ncid, varid, name, xtype, mpilen, (unsigned int*)ip);
case NC_INT64:
return ncmpi_put_att_longlong(nc->int_ncid, varid, name, xtype, mpilen, (long long*)ip);
case NC_UINT64:
return ncmpi_put_att_ulonglong(nc->int_ncid, varid, name, xtype, mpilen, (unsigned long long*)ip);
default:
return NC_EBADTYPE;
2013-03-16 04:31:07 +08:00
}
}
static int
NCP_def_var(int ncid, const char *name, nc_type xtype,
int ndims, const int *dimidsp, int *varidp)
2013-03-16 04:31:07 +08:00
{
NC *nc;
int status = NC_check_id(ncid, &nc);
if (status != NC_NOERR) return status;
return ncmpi_def_var(nc->int_ncid,name,xtype,ndims,dimidsp,varidp);
2013-03-16 04:31:07 +08:00
}
static int
NCP_inq_varid(int ncid, const char *name, int *varidp)
2013-03-16 04:31:07 +08:00
{
NC *nc;
2013-03-16 04:31:07 +08:00
int status = NC_check_id(ncid, &nc);
if (status != NC_NOERR) return status;
2013-03-16 04:31:07 +08:00
return ncmpi_inq_varid(nc->int_ncid,name,varidp);
}
static int
NCP_rename_var(int ncid, int varid, const char *name)
2013-03-16 04:31:07 +08:00
{
NC *nc;
2013-03-16 04:31:07 +08:00
int status = NC_check_id(ncid, &nc);
if (status != NC_NOERR) return status;
2013-03-16 04:31:07 +08:00
return ncmpi_rename_var(nc->int_ncid,varid,name);
}
static int
NCP_get_vara(int ncid,
int varid,
const size_t *startp,
const size_t *countp,
void *op,
nc_type memtype)
2013-03-16 04:31:07 +08:00
{
NC *nc;
NCP_INFO *nc5;
int d, ndims, status;
2013-03-16 04:31:07 +08:00
MPI_Offset mpi_start[NC_MAX_VAR_DIMS], mpi_count[NC_MAX_VAR_DIMS];
status = NC_check_id(ncid, &nc);
if (status != NC_NOERR) return status;
2015-08-17 11:44:18 +08:00
nc5 = NCP_DATA(nc);
assert(nc5);
2013-03-16 04:31:07 +08:00
/* get variable's ndims */
status = ncmpi_inq_varndims(nc->int_ncid, varid, &ndims);
if (status != NC_NOERR) return status;
2015-08-17 11:44:18 +08:00
/* We must convert the start and count arrays to MPI_Offset type. */
for (d=0; d<ndims; d++) {
mpi_start[d] = startp[d];
mpi_count[d] = countp[d];
2013-03-16 04:31:07 +08:00
}
2015-08-17 11:44:18 +08:00
if (memtype == NC_NAT) {
status = ncmpi_inq_vartype(nc->int_ncid, varid, &memtype);
if (status != NC_NOERR) return status;
2015-08-17 11:44:18 +08:00
}
if (fIsSet(nc5->pnetcdf_access_mode, NCP_MODE_INDEP)) {
switch(memtype) {
case NC_BYTE:
return ncmpi_get_vara_schar(nc->int_ncid, varid, mpi_start, mpi_count, op);
case NC_CHAR:
return ncmpi_get_vara_text(nc->int_ncid, varid, mpi_start, mpi_count, op);
case NC_SHORT:
return ncmpi_get_vara_short(nc->int_ncid, varid, mpi_start, mpi_count, op);
case NC_INT:
return ncmpi_get_vara_int(nc->int_ncid, varid, mpi_start, mpi_count, op);
case NC_FLOAT:
return ncmpi_get_vara_float(nc->int_ncid, varid, mpi_start, mpi_count, op);
case NC_DOUBLE:
return ncmpi_get_vara_double(nc->int_ncid, varid, mpi_start, mpi_count, op);
case NC_UBYTE:
return ncmpi_get_vara_uchar(nc->int_ncid, varid, mpi_start, mpi_count, op);
case NC_USHORT:
return ncmpi_get_vara_ushort(nc->int_ncid, varid, mpi_start, mpi_count, op);
case NC_UINT:
return ncmpi_get_vara_uint(nc->int_ncid, varid, mpi_start, mpi_count, op);
case NC_INT64:
return ncmpi_get_vara_longlong(nc->int_ncid, varid, mpi_start, mpi_count, op);
case NC_UINT64:
return ncmpi_get_vara_ulonglong(nc->int_ncid, varid, mpi_start, mpi_count, op);
default:
return NC_EBADTYPE;
}
} else {
switch(memtype) {
case NC_BYTE:
return ncmpi_get_vara_schar_all(nc->int_ncid, varid, mpi_start, mpi_count, op);
case NC_CHAR:
return ncmpi_get_vara_text_all(nc->int_ncid, varid, mpi_start, mpi_count, op);
case NC_SHORT:
return ncmpi_get_vara_short_all(nc->int_ncid, varid, mpi_start, mpi_count, op);
case NC_INT:
return ncmpi_get_vara_int_all(nc->int_ncid, varid, mpi_start, mpi_count, op);
case NC_FLOAT:
return ncmpi_get_vara_float_all(nc->int_ncid, varid, mpi_start, mpi_count, op);
case NC_DOUBLE:
return ncmpi_get_vara_double_all(nc->int_ncid, varid, mpi_start, mpi_count, op);
case NC_UBYTE:
return ncmpi_get_vara_uchar_all(nc->int_ncid, varid, mpi_start, mpi_count, op);
case NC_USHORT:
return ncmpi_get_vara_ushort_all(nc->int_ncid, varid, mpi_start, mpi_count, op);
case NC_UINT:
return ncmpi_get_vara_uint_all(nc->int_ncid, varid, mpi_start, mpi_count, op);
case NC_INT64:
return ncmpi_get_vara_longlong_all(nc->int_ncid, varid, mpi_start, mpi_count, op);
case NC_UINT64:
return ncmpi_get_vara_ulonglong_all(nc->int_ncid, varid, mpi_start, mpi_count, op);
default:
return NC_EBADTYPE;
}
2013-03-16 04:31:07 +08:00
}
}
static int
NCP_put_vara(int ncid,
int varid,
const size_t *startp,
const size_t *countp,
const void *ip,
nc_type memtype)
2013-03-16 04:31:07 +08:00
{
NC *nc;
NCP_INFO *nc5;
int d, ndims, status;
2013-03-16 04:31:07 +08:00
MPI_Offset mpi_start[NC_MAX_VAR_DIMS], mpi_count[NC_MAX_VAR_DIMS];
status = NC_check_id(ncid, &nc);
if (status != NC_NOERR) return status;
2015-08-17 11:44:18 +08:00
nc5 = NCP_DATA(nc);
assert(nc5);
2013-03-16 04:31:07 +08:00
/* get variable's ndims */
status = ncmpi_inq_varndims(nc->int_ncid, varid, &ndims);
if (status != NC_NOERR) return status;
2015-08-17 11:44:18 +08:00
/* We must convert the start and count arrays to MPI_Offset type. */
for (d=0; d<ndims; d++) {
mpi_start[d] = startp[d];
mpi_count[d] = countp[d];
2013-03-16 04:31:07 +08:00
}
2015-08-17 11:44:18 +08:00
if (memtype == NC_NAT) {
status = ncmpi_inq_vartype(nc->int_ncid, varid, &memtype);
if (status != NC_NOERR) return status;
2015-08-17 11:44:18 +08:00
}
if (fIsSet(nc5->pnetcdf_access_mode, NCP_MODE_INDEP)) {
switch(memtype) {
case NC_BYTE:
return ncmpi_put_vara_schar(nc->int_ncid, varid, mpi_start, mpi_count, ip);
case NC_CHAR:
return ncmpi_put_vara_text(nc->int_ncid, varid, mpi_start, mpi_count, ip);
case NC_SHORT:
return ncmpi_put_vara_short(nc->int_ncid, varid, mpi_start, mpi_count, ip);
case NC_INT:
return ncmpi_put_vara_int(nc->int_ncid, varid, mpi_start, mpi_count, ip);
case NC_FLOAT:
return ncmpi_put_vara_float(nc->int_ncid, varid, mpi_start, mpi_count, ip);
case NC_DOUBLE:
return ncmpi_put_vara_double(nc->int_ncid, varid, mpi_start, mpi_count, ip);
case NC_UBYTE:
return ncmpi_put_vara_uchar(nc->int_ncid, varid, mpi_start, mpi_count, ip);
case NC_USHORT:
return ncmpi_put_vara_ushort(nc->int_ncid, varid, mpi_start, mpi_count, ip);
case NC_UINT:
return ncmpi_put_vara_uint(nc->int_ncid, varid, mpi_start, mpi_count, ip);
case NC_INT64:
return ncmpi_put_vara_longlong(nc->int_ncid, varid, mpi_start, mpi_count, ip);
case NC_UINT64:
return ncmpi_put_vara_ulonglong(nc->int_ncid, varid, mpi_start, mpi_count, ip);
default:
return NC_EBADTYPE;
}
} else {
switch(memtype) {
case NC_BYTE:
return ncmpi_put_vara_schar_all(nc->int_ncid, varid, mpi_start, mpi_count, ip);
case NC_CHAR:
return ncmpi_put_vara_text_all(nc->int_ncid, varid, mpi_start, mpi_count, ip);
case NC_SHORT:
return ncmpi_put_vara_short_all(nc->int_ncid, varid, mpi_start, mpi_count, ip);
case NC_INT:
return ncmpi_put_vara_int_all(nc->int_ncid, varid, mpi_start, mpi_count, ip);
case NC_FLOAT:
return ncmpi_put_vara_float_all(nc->int_ncid, varid, mpi_start, mpi_count, ip);
case NC_DOUBLE:
return ncmpi_put_vara_double_all(nc->int_ncid, varid, mpi_start, mpi_count, ip);
case NC_UBYTE:
return ncmpi_put_vara_uchar_all(nc->int_ncid, varid, mpi_start, mpi_count, ip);
case NC_USHORT:
return ncmpi_put_vara_ushort_all(nc->int_ncid, varid, mpi_start, mpi_count, ip);
case NC_UINT:
return ncmpi_put_vara_uint_all(nc->int_ncid, varid, mpi_start, mpi_count, ip);
case NC_INT64:
return ncmpi_put_vara_longlong_all(nc->int_ncid, varid, mpi_start, mpi_count, ip);
case NC_UINT64:
return ncmpi_put_vara_ulonglong_all(nc->int_ncid, varid, mpi_start, mpi_count, ip);
default:
return NC_EBADTYPE;
}
2015-08-17 11:44:18 +08:00
}
}
static int
NCP_get_vars(int ncid,
int varid,
const size_t *startp,
const size_t *countp,
const ptrdiff_t *stridep,
void *op,
nc_type memtype)
2015-08-17 11:44:18 +08:00
{
NC *nc;
NCP_INFO *nc5;
int d, ndims, status;
2015-08-17 11:44:18 +08:00
MPI_Offset mpi_start[NC_MAX_VAR_DIMS], mpi_count[NC_MAX_VAR_DIMS], mpi_stride[NC_MAX_VAR_DIMS];
if (stridep == NULL)
return NCP_get_vara(ncid, varid, startp, countp, op, memtype);
2015-08-17 11:44:18 +08:00
status = NC_check_id(ncid, &nc);
if (status != NC_NOERR) return status;
2015-08-17 11:44:18 +08:00
nc5 = NCP_DATA(nc);
assert(nc5);
/* get variable's ndims */
status = ncmpi_inq_varndims(nc->int_ncid, varid, &ndims);
if (status != NC_NOERR) return status;
2015-08-17 11:44:18 +08:00
/* We must convert the start, count, and stride arrays to MPI_Offset type. */
for (d=0; d<ndims; d++) {
mpi_start[d] = startp[d];
mpi_count[d] = countp[d];
mpi_stride[d] = stridep[d];
2015-08-17 11:44:18 +08:00
}
if (memtype == NC_NAT) {
status = ncmpi_inq_vartype(nc->int_ncid, varid, &memtype);
if (status != NC_NOERR) return status;
2015-08-17 11:44:18 +08:00
}
if (fIsSet(nc5->pnetcdf_access_mode, NCP_MODE_INDEP)) {
switch(memtype) {
case NC_BYTE:
return ncmpi_get_vars_schar(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, op);
case NC_CHAR:
return ncmpi_get_vars_text(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, op);
case NC_SHORT:
return ncmpi_get_vars_short(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, op);
case NC_INT:
return ncmpi_get_vars_int(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, op);
case NC_FLOAT:
return ncmpi_get_vars_float(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, op);
case NC_DOUBLE:
return ncmpi_get_vars_double(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, op);
case NC_UBYTE:
return ncmpi_get_vars_uchar(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, op);
case NC_USHORT:
return ncmpi_get_vars_ushort(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, op);
case NC_UINT:
return ncmpi_get_vars_uint(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, op);
case NC_INT64:
return ncmpi_get_vars_longlong(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, op);
case NC_UINT64:
return ncmpi_get_vars_ulonglong(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, op);
default:
return NC_EBADTYPE;
}
} else {
switch(memtype) {
case NC_BYTE:
return ncmpi_get_vars_schar_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, op);
case NC_CHAR:
return ncmpi_get_vars_text_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, op);
case NC_SHORT:
return ncmpi_get_vars_short_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, op);
case NC_INT:
return ncmpi_get_vars_int_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, op);
case NC_FLOAT:
return ncmpi_get_vars_float_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, op);
case NC_DOUBLE:
return ncmpi_get_vars_double_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, op);
case NC_UBYTE:
return ncmpi_get_vars_uchar_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, op);
case NC_USHORT:
return ncmpi_get_vars_ushort_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, op);
case NC_UINT:
return ncmpi_get_vars_uint_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, op);
case NC_INT64:
return ncmpi_get_vars_longlong_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, op);
case NC_UINT64:
return ncmpi_get_vars_ulonglong_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, op);
default:
return NC_EBADTYPE;
}
2015-08-17 11:44:18 +08:00
}
}
static int
NCP_put_vars(int ncid,
int varid,
const size_t *startp,
const size_t *countp,
const ptrdiff_t *stridep,
const void *op,
nc_type memtype)
2015-08-17 11:44:18 +08:00
{
NC *nc;
NCP_INFO *nc5;
int d, ndims, status;
2015-08-17 11:44:18 +08:00
MPI_Offset mpi_start[NC_MAX_VAR_DIMS], mpi_count[NC_MAX_VAR_DIMS], mpi_stride[NC_MAX_VAR_DIMS];
if (stridep == NULL)
return NCP_put_vara(ncid, varid, startp, countp, op, memtype);
2015-08-17 11:44:18 +08:00
status = NC_check_id(ncid, &nc);
if (status != NC_NOERR) return status;
2015-08-17 11:44:18 +08:00
nc5 = NCP_DATA(nc);
assert(nc5);
/* get variable's ndims */
status = ncmpi_inq_varndims(nc->int_ncid, varid, &ndims);
if (status != NC_NOERR) return status;
2015-08-17 11:44:18 +08:00
/* We must convert the start, count, and stride arrays to MPI_Offset type. */
for (d=0; d<ndims; d++) {
mpi_start[d] = startp[d];
mpi_count[d] = countp[d];
mpi_stride[d] = stridep[d];
2015-08-17 11:44:18 +08:00
}
if (memtype == NC_NAT) {
status = ncmpi_inq_vartype(nc->int_ncid, varid, &memtype);
if (status != NC_NOERR) return status;
2015-08-17 11:44:18 +08:00
}
if (fIsSet(nc5->pnetcdf_access_mode, NCP_MODE_INDEP)) {
switch(memtype) {
case NC_BYTE:
return ncmpi_put_vars_schar(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, op);
case NC_CHAR:
return ncmpi_put_vars_text(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, op);
case NC_SHORT:
return ncmpi_put_vars_short(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, op);
case NC_INT:
return ncmpi_put_vars_int(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, op);
case NC_FLOAT:
return ncmpi_put_vars_float(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, op);
case NC_DOUBLE:
return ncmpi_put_vars_double(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, op);
case NC_UBYTE:
return ncmpi_put_vars_uchar(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, op);
case NC_USHORT:
return ncmpi_put_vars_ushort(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, op);
case NC_UINT:
return ncmpi_put_vars_uint(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, op);
case NC_INT64:
return ncmpi_put_vars_longlong(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, op);
case NC_UINT64:
return ncmpi_put_vars_ulonglong(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, op);
default:
return NC_EBADTYPE;
}
} else {
switch(memtype) {
case NC_BYTE:
return ncmpi_put_vars_schar_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, op);
case NC_CHAR:
return ncmpi_put_vars_text_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, op);
case NC_SHORT:
return ncmpi_put_vars_short_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, op);
case NC_INT:
return ncmpi_put_vars_int_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, op);
case NC_FLOAT:
return ncmpi_put_vars_float_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, op);
case NC_DOUBLE:
return ncmpi_put_vars_double_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, op);
case NC_UBYTE:
return ncmpi_put_vars_uchar_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, op);
case NC_USHORT:
return ncmpi_put_vars_ushort_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, op);
case NC_UINT:
return ncmpi_put_vars_uint_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, op);
case NC_INT64:
return ncmpi_put_vars_longlong_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, op);
case NC_UINT64:
return ncmpi_put_vars_ulonglong_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, op);
default:
return NC_EBADTYPE;
}
2015-08-17 11:44:18 +08:00
}
}
static int
NCP_get_varm(int ncid,
int varid,
const size_t *startp,
const size_t *countp,
const ptrdiff_t *stridep,
const ptrdiff_t *imapp,
void *ip,
nc_type memtype)
2015-08-17 11:44:18 +08:00
{
NC *nc;
NCP_INFO *nc5;
int d, ndims, status;
2015-08-17 11:44:18 +08:00
MPI_Offset mpi_start[NC_MAX_VAR_DIMS], mpi_count[NC_MAX_VAR_DIMS], mpi_stride[NC_MAX_VAR_DIMS], mpi_imap[NC_MAX_VAR_DIMS];
if (imapp == NULL) {
if (stridep == NULL)
return NCP_get_vara(ncid, varid, startp, countp, ip, memtype);
else
return NCP_get_vars(ncid, varid, startp, countp, stridep, ip, memtype);
}
2015-08-17 11:44:18 +08:00
status = NC_check_id(ncid, &nc);
if (status != NC_NOERR) return status;
2015-08-17 11:44:18 +08:00
nc5 = NCP_DATA(nc);
assert(nc5);
/* get variable's ndims */
status = ncmpi_inq_varndims(nc->int_ncid, varid, &ndims);
if (status != NC_NOERR) return status;
2015-08-17 11:44:18 +08:00
/* We must convert the start, count, stride, and imap arrays to MPI_Offset type. */
for (d=0; d<ndims; d++) {
mpi_start[d] = startp[d];
mpi_count[d] = countp[d];
mpi_stride[d] = (stridep == NULL) ? 1 : stridep[d];
mpi_imap[d] = imapp[d];
2015-08-17 11:44:18 +08:00
}
if (memtype == NC_NAT) {
status = ncmpi_inq_vartype(nc->int_ncid, varid, &memtype);
if (status != NC_NOERR) return status;
2015-08-17 11:44:18 +08:00
}
if (fIsSet(nc5->pnetcdf_access_mode, NCP_MODE_INDEP)) {
switch(memtype) {
case NC_BYTE:
return ncmpi_get_varm_schar(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, ip);
case NC_CHAR:
return ncmpi_get_varm_text(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, ip);
case NC_SHORT:
return ncmpi_get_varm_short(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, ip);
case NC_INT:
return ncmpi_get_varm_int(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, ip);
case NC_FLOAT:
return ncmpi_get_varm_float(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, ip);
case NC_DOUBLE:
return ncmpi_get_varm_double(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, ip);
case NC_UBYTE:
return ncmpi_get_varm_uchar(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, ip);
case NC_USHORT:
return ncmpi_get_varm_ushort(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, ip);
case NC_UINT:
return ncmpi_get_varm_uint(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, ip);
case NC_INT64:
return ncmpi_get_varm_longlong(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, ip);
case NC_UINT64:
return ncmpi_get_varm_ulonglong(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, ip);
default:
return NC_EBADTYPE;
}
2015-08-17 11:44:18 +08:00
} else {
switch(memtype) {
case NC_BYTE:
return ncmpi_get_varm_schar_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, ip);
case NC_CHAR:
return ncmpi_get_varm_text_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, ip);
case NC_SHORT:
return ncmpi_get_varm_short_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, ip);
case NC_INT:
return ncmpi_get_varm_int_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, ip);
case NC_FLOAT:
return ncmpi_get_varm_float_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, ip);
case NC_DOUBLE:
return ncmpi_get_varm_double_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, ip);
case NC_UBYTE:
return ncmpi_get_varm_uchar_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, ip);
case NC_USHORT:
return ncmpi_get_varm_ushort_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, ip);
case NC_UINT:
return ncmpi_get_varm_uint_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, ip);
case NC_INT64:
return ncmpi_get_varm_longlong_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, ip);
case NC_UINT64:
return ncmpi_get_varm_ulonglong_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, ip);
default:
return NC_EBADTYPE;
}
2015-08-17 11:44:18 +08:00
}
}
static int
NCP_put_varm(int ncid,
int varid,
const size_t *startp,
const size_t *countp,
const ptrdiff_t *stridep,
const ptrdiff_t *imapp,
const void *op,
nc_type memtype)
2015-08-17 11:44:18 +08:00
{
NC *nc;
NCP_INFO *nc5;
int d, ndims, status;
2015-08-17 11:44:18 +08:00
MPI_Offset mpi_start[NC_MAX_VAR_DIMS], mpi_count[NC_MAX_VAR_DIMS], mpi_stride[NC_MAX_VAR_DIMS], mpi_imap[NC_MAX_VAR_DIMS];
if (imapp == NULL) {
if (stridep == NULL)
return NCP_put_vara(ncid, varid, startp, countp, op, memtype);
else
return NCP_put_vars(ncid, varid, startp, countp, stridep, op, memtype);
}
2015-08-17 11:44:18 +08:00
status = NC_check_id(ncid, &nc);
if (status != NC_NOERR) return status;
2015-08-17 11:44:18 +08:00
nc5 = NCP_DATA(nc);
assert(nc5);
/* get variable's ndims */
status = ncmpi_inq_varndims(nc->int_ncid, varid, &ndims);
if (status != NC_NOERR) return status;
2015-08-17 11:44:18 +08:00
/* We must convert the start, count, stride, and imap arrays to MPI_Offset type. */
for (d=0; d<ndims; d++) {
mpi_start[d] = startp[d];
mpi_count[d] = countp[d];
mpi_stride[d] = (stridep == NULL) ? 1 : stridep[d];
mpi_imap[d] = imapp[d];
2015-08-17 11:44:18 +08:00
}
if (memtype == NC_NAT) {
status = ncmpi_inq_vartype(nc->int_ncid, varid, &memtype);
if (status != NC_NOERR) return status;
2015-08-17 11:44:18 +08:00
}
if (fIsSet(nc5->pnetcdf_access_mode, NCP_MODE_INDEP)) {
switch(memtype) {
case NC_BYTE:
return ncmpi_put_varm_schar(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, op);
case NC_CHAR:
return ncmpi_put_varm_text(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, op);
case NC_SHORT:
return ncmpi_put_varm_short(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, op);
case NC_INT:
return ncmpi_put_varm_int(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, op);
case NC_FLOAT:
return ncmpi_put_varm_float(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, op);
case NC_DOUBLE:
return ncmpi_put_varm_double(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, op);
case NC_UBYTE:
return ncmpi_put_varm_uchar(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, op);
case NC_USHORT:
return ncmpi_put_varm_ushort(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, op);
case NC_UINT:
return ncmpi_put_varm_uint(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, op);
case NC_INT64:
return ncmpi_put_varm_longlong(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, op);
case NC_UINT64:
return ncmpi_put_varm_ulonglong(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, op);
default:
return NC_EBADTYPE;
}
} else {
switch(memtype) {
case NC_BYTE:
return ncmpi_put_varm_schar_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, op);
case NC_CHAR:
return ncmpi_put_varm_text_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, op);
case NC_SHORT:
return ncmpi_put_varm_short_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, op);
case NC_INT:
return ncmpi_put_varm_int_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, op);
case NC_FLOAT:
return ncmpi_put_varm_float_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, op);
case NC_DOUBLE:
return ncmpi_put_varm_double_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, op);
case NC_UBYTE:
return ncmpi_put_varm_uchar_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, op);
case NC_USHORT:
return ncmpi_put_varm_ushort_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, op);
case NC_UINT:
return ncmpi_put_varm_uint_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, op);
case NC_INT64:
return ncmpi_put_varm_longlong_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, op);
case NC_UINT64:
return ncmpi_put_varm_ulonglong_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, op);
default:
return NC_EBADTYPE;
}
2013-03-16 04:31:07 +08:00
}
}
static int
NCP_inq_var_all(int ncid, int varid, char *name, nc_type *xtypep,
int *ndimsp, int *dimidsp, int *nattsp,
int *shufflep, int *deflatep, int *deflate_levelp,
int *fletcher32p, int *contiguousp, size_t *chunksizesp,
int *no_fill, void *fill_valuep, int *endiannessp,
unsigned int *idp, size_t *nparamsp, unsigned int *params)
2013-03-16 04:31:07 +08:00
{
int status;
NC *nc;
2013-03-16 04:31:07 +08:00
status = NC_check_id(ncid, &nc);
if (status != NC_NOERR) return status;
2013-03-16 04:31:07 +08:00
status = ncmpi_inq_var(nc->int_ncid, varid, name, xtypep, ndimsp, dimidsp, nattsp);
if (status != NC_NOERR) return status;
if (shufflep) *shufflep = 0;
if (deflatep) *deflatep = 0;
if (fletcher32p) *fletcher32p = 0;
if (contiguousp) *contiguousp = NC_CONTIGUOUS;
#if (PNETCDF_VERSION_MAJOR*10000 + PNETCDF_VERSION_MINOR*100 + PNETCDF_VERSION_SUB >= 10601)
/* ncmpi_inq_var_fill was first implemented in PnetCDF 1.6.1 */
if (no_fill) {
status = ncmpi_inq_var_fill(nc->int_ncid, varid, no_fill, fill_valuep);
if (status != NC_NOERR) return status;
}
#else
/* PnetCDF 1.6.0 and priors support NC_NOFILL only */
if (no_fill) *no_fill = 1;
#endif
if (endiannessp) return NC_ENOTNC4;
if (idp) return NC_ENOTNC4;
if (nparamsp) return NC_ENOTNC4;
if (params) return NC_ENOTNC4;
2013-03-16 04:31:07 +08:00
return NC_NOERR;
}
2017-12-21 10:53:30 +08:00
static int
NCP_def_var_fill(int ncid, int varid, int no_fill, const void *fill_value)
{
NC *nc;
2017-12-21 10:53:30 +08:00
int status = NC_check_id(ncid, &nc);
if (status != NC_NOERR) return status;
#if (PNETCDF_VERSION_MAJOR*10000 + PNETCDF_VERSION_MINOR*100 + PNETCDF_VERSION_SUB >= 10601)
/* ncmpi_def_var_fill was first implemented in PnetCDF 1.6.1 */
2017-12-21 10:53:30 +08:00
return ncmpi_def_var_fill(nc->int_ncid, varid, no_fill, fill_value);
#else
return NC_EPNETCDF;
#endif
2017-12-21 10:53:30 +08:00
}
2013-03-16 04:31:07 +08:00
static int
NCP_var_par_access(int ncid, int varid, int par_access)
2013-03-16 04:31:07 +08:00
{
NC *nc;
NCP_INFO *nc5;
2013-03-16 04:31:07 +08:00
int status;
if (par_access != NC_INDEPENDENT && par_access != NC_COLLECTIVE)
return NC_EINVAL;
2013-03-16 04:31:07 +08:00
#ifdef _DO_NOT_IGNORE_VARID_
if (varid != NC_GLOBAL) /* PnetCDF cannot do per-variable mode change */
return NC_EINVAL;
#endif
2013-03-16 04:31:07 +08:00
status = NC_check_id(ncid, &nc);
if (status != NC_NOERR) return status;
2013-03-16 04:31:07 +08:00
2015-08-17 11:44:18 +08:00
nc5 = NCP_DATA(nc);
assert(nc5);
2013-03-16 04:31:07 +08:00
/* if currently in data mode */
if (fIsSet(nc5->pnetcdf_access_mode, NCP_MODE_DATA)) {
if (par_access == NC_INDEPENDENT) {
if (fIsSet(nc5->pnetcdf_access_mode, NCP_MODE_INDEP))
return NC_NOERR;
else { /* currently in collective data mode */
fSet(nc5->pnetcdf_access_mode, NCP_MODE_INDEP);
return ncmpi_begin_indep_data(nc->int_ncid);
}
}
else { /* want to enter collective data mode */
if (fIsSet(nc5->pnetcdf_access_mode, NCP_MODE_INDEP)) {
fClr(nc5->pnetcdf_access_mode, NCP_MODE_INDEP);
return ncmpi_end_indep_data(nc->int_ncid);
}
else
return NC_NOERR;
}
}
else { /* currently in define mode */
if (par_access == NC_INDEPENDENT)
fSet(nc5->pnetcdf_access_mode, NCP_MODE_INDEP);
else
fClr(nc5->pnetcdf_access_mode, NCP_MODE_INDEP);
}
return NC_NOERR;
2013-03-16 04:31:07 +08:00
}
2013-03-16 04:31:07 +08:00
static int
NCP_show_metadata(int ncid)
2013-03-16 04:31:07 +08:00
{
return NC_NOERR;
}
static int
NCP_inq_unlimdims(int ncid, int *ndimsp, int *unlimdimidsp)
2013-03-16 04:31:07 +08:00
{
int retval;
int unlimid;
if ((retval = NCP_inq_unlimdim(ncid, &unlimid)))
2013-03-16 04:31:07 +08:00
return retval;
if (unlimid != -1) {
if (ndimsp) *ndimsp = 1;
if (unlimdimidsp)
2013-03-16 04:31:07 +08:00
unlimdimidsp[0] = unlimid;
} else
if (ndimsp) *ndimsp = 0;
2013-03-16 04:31:07 +08:00
return NC_NOERR;
}
static int
NCP_inq_type_equal(int ncid1,
nc_type typeid1,
int ncid2,
nc_type typeid2,
int *equalp)
2013-03-16 04:31:07 +08:00
{
/* Check input. */
if (equalp == NULL) return NC_NOERR;
2013-03-16 04:31:07 +08:00
if (typeid1 <= NC_NAT || typeid2 <= NC_NAT)
return NC_EINVAL;
2013-03-16 04:31:07 +08:00
*equalp = 0; /* assume */
2013-03-16 04:31:07 +08:00
/* If one is atomic, and the other user-defined, the types are not equal */
if ((typeid1 <= NC_STRING && typeid2 > NC_STRING) ||
(typeid2 <= NC_STRING && typeid1 > NC_STRING)) {
if (equalp) *equalp = 0;
return NC_NOERR;
}
2013-03-16 04:31:07 +08:00
/* If both are atomic types, the answer is easy. */
2015-08-17 11:44:18 +08:00
if (typeid1 <= ATOMICTYPEMAX5) {
2013-03-16 04:31:07 +08:00
if (equalp) {
if (typeid1 == typeid2)
*equalp = 1;
else
*equalp = 0;
}
return NC_NOERR;
}
return NC_NOERR;
}
static int
NCP_def_grp(int parent_ncid, const char *name, int *new_ncid)
2013-03-16 04:31:07 +08:00
{
return NC_ENOTNC4;
}
static int
NCP_rename_grp(int ncid, const char *name)
{
return NC_ENOTNC4;
}
2013-03-16 04:31:07 +08:00
static int
NCP_inq_ncid(int ncid, const char *name, int *grp_ncid)
2013-03-16 04:31:07 +08:00
{
if (grp_ncid) *grp_ncid = ncid;
2013-03-16 04:31:07 +08:00
return NC_NOERR;
}
static int
NCP_inq_grps(int ncid, int *numgrps, int *ncids)
2013-03-16 04:31:07 +08:00
{
if (numgrps)
2013-03-16 04:31:07 +08:00
*numgrps = 0;
return NC_NOERR;
}
static int
NCP_inq_grpname(int ncid, char *name)
2013-03-16 04:31:07 +08:00
{
if (name)
2013-03-16 04:31:07 +08:00
strcpy(name, "/");
return NC_NOERR;
}
static int
NCP_inq_grpname_full(int ncid, size_t *lenp, char *full_name)
2013-03-16 04:31:07 +08:00
{
if (full_name)
2013-03-16 04:31:07 +08:00
strcpy(full_name, "/");
if (lenp) *lenp = 1;
2013-03-16 04:31:07 +08:00
return NC_NOERR;
}
static int
NCP_inq_grp_parent(int ncid, int *parent_ncid)
2013-03-16 04:31:07 +08:00
{
return NC_ENOGRP;
}
static int
NCP_inq_grp_full_ncid(int ncid, const char *full_name, int *grp_ncid)
2013-03-16 04:31:07 +08:00
{
return NC_ENOGRP;
}
static int
NCP_inq_varids(int ncid, int *nvarsp, int *varids)
2013-03-16 04:31:07 +08:00
{
int retval,v,nvars;
/* This is, effectively, a netcdf-3 file, there is only one group, the root
group, and its vars have ids 0 thru nvars - 1. */
if ((retval = NCP_inq(ncid, NULL, &nvars, NULL, NULL)))
2013-03-16 04:31:07 +08:00
return retval;
if (nvarsp) *nvarsp = nvars;
if (varids) {
for (v=0; v<nvars; v++)
2013-03-16 04:31:07 +08:00
varids[v] = v;
}
2013-03-16 04:31:07 +08:00
return NC_NOERR;
}
static int
NCP_inq_dimids(int ncid, int *ndimsp, int *dimids, int include_parents)
2013-03-16 04:31:07 +08:00
{
int retval,d,ndims;
/* If this is like a netcdf-3 file, then the dimids are going to be 0
thru ndims-1, so just provide them. */
if ((retval = NCP_inq(ncid, &ndims, NULL, NULL, NULL)))
2013-03-16 04:31:07 +08:00
return retval;
if (ndimsp) *ndimsp = ndims;
if (dimids) {
for (d=0; d<ndims; d++)
2013-03-16 04:31:07 +08:00
dimids[d] = d;
}
2013-03-16 04:31:07 +08:00
return NC_NOERR;
}
static int
NCP_inq_typeid(int ncid, const char *name, nc_type *typeidp)
2013-03-16 04:31:07 +08:00
{
int i;
for (i=0; i<=ATOMICTYPEMAX5; i++)
if (!strcmp(name, NC_atomictypename(i))) {
if (typeidp) *typeidp = i;
2013-03-16 04:31:07 +08:00
return NC_NOERR;
}
return NC_ENOTNC4;
}
static int
NCP_inq_typeids(int ncid, int *ntypes, int *typeids)
2013-03-16 04:31:07 +08:00
{
if (ntypes) *ntypes = 0;
2013-03-16 04:31:07 +08:00
return NC_NOERR;
}
static int
NCP_inq_user_type(int ncid, nc_type typeid, char *name, size_t *size,
nc_type *base_nc_typep, size_t *nfieldsp, int *classp)
2013-03-16 04:31:07 +08:00
{
return NC_ENOTNC4;
}
/**************************************************/
/* Pnetcdf Dispatch table */
static const NC_Dispatch NCP_dispatcher = {
2015-08-17 11:44:18 +08:00
NC_FORMATX_PNETCDF,
2020-02-28 05:06:45 +08:00
NC_DISPATCH_VERSION,
NCP_create,
NCP_open,
NCP_redef,
NCP__enddef,
NCP_sync,
NCP_abort,
NCP_close,
NCP_set_fill,
NCP_inq_format,
NCP_inq_format_extended,
NCP_inq,
NCP_inq_type,
NCP_def_dim,
NCP_inq_dimid,
NCP_inq_dim,
NCP_inq_unlimdim,
NCP_rename_dim,
NCP_inq_att,
NCP_inq_attid,
NCP_inq_attname,
NCP_rename_att,
NCP_del_att,
NCP_get_att,
NCP_put_att,
NCP_def_var,
NCP_inq_varid,
NCP_rename_var,
NCP_get_vara,
NCP_put_vara,
2015-08-17 11:44:18 +08:00
NCP_get_vars,
NCP_put_vars,
NCP_get_varm,
NCP_put_varm,
2013-03-16 04:31:07 +08:00
NCP_inq_var_all,
2013-03-16 04:31:07 +08:00
NCP_var_par_access,
2017-12-21 10:53:30 +08:00
NCP_def_var_fill,
2013-03-16 04:31:07 +08:00
NCP_show_metadata,
NCP_inq_unlimdims,
NCP_inq_ncid,
NCP_inq_grps,
NCP_inq_grpname,
NCP_inq_grpname_full,
NCP_inq_grp_parent,
NCP_inq_grp_full_ncid,
NCP_inq_varids,
NCP_inq_dimids,
NCP_inq_typeids,
NCP_inq_type_equal,
NCP_def_grp,
NCP_rename_grp,
NCP_inq_user_type,
NCP_inq_typeid,
2018-04-05 04:11:44 +08:00
NC_NOTNC4_def_compound,
NC_NOTNC4_insert_compound,
NC_NOTNC4_insert_array_compound,
NC_NOTNC4_inq_compound_field,
NC_NOTNC4_inq_compound_fieldindex,
NC_NOTNC4_def_vlen,
NC_NOTNC4_put_vlen_element,
NC_NOTNC4_get_vlen_element,
NC_NOTNC4_def_enum,
NC_NOTNC4_insert_enum,
NC_NOTNC4_inq_enum_member,
NC_NOTNC4_inq_enum_ident,
NC_NOTNC4_def_opaque,
NC_NOTNC4_def_var_deflate,
NC_NOTNC4_def_var_fletcher32,
NC_NOTNC4_def_var_chunking,
NC_NOTNC4_def_var_endian,
NC_NOTNC4_def_var_filter,
NC_NOTNC4_set_var_chunk_cache,
NC_NOTNC4_get_var_chunk_cache,
2013-03-16 04:31:07 +08:00
Mostly revert the filter code to reduce its complexity of use. re: https://github.com/Unidata/netcdf-c/issues/1836 Revert the internal filter code to simplify it. From the user's point of view, the only visible changes should be: 1. The functions that convert text to filter specs have had their signature reverted and have been moved to netcdf_aux.h 2. Some filter API functions now return NC_ENOFILTER when inquiry is made about some filter. Internally,the dispatch table has been modified to get rid of the filter_actions entry and associated complex structures. It has been replaced with inq_var_filter_ids and inq_var_filter_info entries and the dispatch table version has been bumped to 3. Corresponding NOOP and NOTNC4 functions were added to libdispatch/dnotnc4.c. Also, the filter_action entries in dispatch tables were replaced for all dispatch code bases (HDF5, DAP2, etc). This should only impact UDF users. In the process, it became clear that the form of the filters field in NC_VAR_INFO_T was format dependent, so I converted it to be of type void* and pushed its management into the various dispatch code bases. Specifically libhdf5 and libnczarr now manage the filters field in their own way. The auxilliary functions for parsing textual filter specifications were moved to netcdf_aux.h and were renamed to the following: * ncaux_h5filterspec_parse * ncaux_h5filterspec_parselist * ncaux_h5filterspec_free * ncaux_h5filter_fix8 Misc. Other Changes: 1. Document NUG/filters.md updated to reflect the changes above. 2. All the old data types (structs and enums) used by filter_actions actions were deleted. The exception is the NC_H5_Filterspec because it is needed by ncaux_h5filterspec_parselist. 3. Clientside filters were removed -- another enhancement for which no-one ever asked. 4. The ability to remove filters was itself removed. 5. Some functionality needed by nczarr was moved from libhdf5 to libsrc4 e.g. nc4_find_default_chunksizes 6. All the filterx code was removed 7. ncfilter.h and nc4filter.c no longer used Misc. Unrelated Changes: 1. The nczarr_test makefile clean was leaving some directories; so add clean-local to take care of them.
2020-09-28 02:43:46 +08:00
NC_NOOP_inq_var_filter_ids,
NC_NOOP_inq_var_filter_info,
NC_NOTNC4_def_var_quantize,
NC_NOTNC4_inq_var_quantize,
Enhance/Fix filter support re: Discussion https://github.com/Unidata/netcdf-c/discussions/2214 The primary change is to support so-called "standard filters". A standard filter is one that is defined by the following netcdf-c API: ```` int nc_def_var_XXX(int ncid, int varid, size_t nparams, unsigned* params); int nc_inq_var_XXXX(int ncid, int varid, int* usefilterp, unsigned* params); ```` So for example, zstandard would be a standard filter by defining the functions *nc_def_var_zstandard* and *nc_inq_var_zstandard*. In order to define these functions, we need a new dispatch function: ```` int nc_inq_filter_avail(int ncid, unsigned filterid); ```` This function, combined with the existing filter API can be used to implement arbitrary standard filters using a simple code pattern. Note that I would have preferred that this function return a list of all available filters, but HDF5 does not support that functionality. So this PR implements the dispatch function and implements the following standard functions: + bzip2 + zstandard + blosc Specific test cases are also provided for HDF5 and NCZarr. Over time, other specific standard filters will be defined. ## Primary Changes * Add nc_inq_filter_avail() to netcdf-c API. * Add standard filter implementations to test use of *nc_inq_filter_avail*. * Bump the dispatch table version number and add to all the relevant dispatch tables (libsrc, libsrcp, etc). * Create a program to invoke nc_inq_filter_avail so that it is accessible to shell scripts. * Cleanup szip support to properly support szip when HDF5 is disabled. This involves detecting libsz separately from testing if HDF5 supports szip. * Integrate shuffle and fletcher32 into the existing filter API. This means that, for example, nc_def_var_fletcher32 is now a wrapper around nc_def_var_filter. * Extend the Codec defaulting to allow multiple default shared libraries. ## Misc. Changes * Modify configure.ac/CMakeLists.txt to look for the relevant libraries implementing standard filters. * Modify libnetcdf.settings to list available standard filters (including deflate and szip). * Add CMake test modules to locate libbz2 and libzstd. * Cleanup the HDF5 memory manager function use in the plugins. * remove unused file include//ncfilter.h * remove tests for the HDF5 memory operations e.g. H5allocate_memory. * Add flag to ncdump to force use of _Filter instead of _Deflate or _Shuffle or _Fletcher32. Used for testing.
2022-03-15 02:39:37 +08:00
NC_NOOP_inq_filter_avail,
2013-03-16 04:31:07 +08:00
};
const NC_Dispatch *NCP_dispatch_table = NULL; /* moved here from ddispatch.c */
2013-03-16 04:31:07 +08:00
int
NCP_initialize(void)
2013-03-16 04:31:07 +08:00
{
NCP_dispatch_table = &NCP_dispatcher;
2013-03-16 04:31:07 +08:00
return NC_NOERR;
}
2015-08-17 11:44:18 +08:00
int
NCP_finalize(void)
{
return NC_NOERR;
}