mirror of
https://github.com/Unidata/netcdf-c.git
synced 2024-12-03 08:01:25 +08:00
1498 lines
44 KiB
C
1498 lines
44 KiB
C
/*********************************************************************
|
|
* Copyright 1993, UCAR/Unidata
|
|
* 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"
|
|
|
|
/* Must follow netcdf.h */
|
|
#include <pnetcdf.h>
|
|
|
|
#define NCP_MODE_DATA 0x0001
|
|
#define NCP_MODE_INDEP 0x0002
|
|
|
|
typedef struct NCP_INFO
|
|
{
|
|
/* pnetcdf_access_mode keeps track of whether independpent or collective
|
|
* mode is set currently.
|
|
*/
|
|
int pnetcdf_access_mode;
|
|
} NCP_INFO;
|
|
|
|
/* Define accessors for the dispatchdata */
|
|
#define NCP_DATA(nc) ((NCP_INFO*)(nc)->dispatchdata)
|
|
#define NCP_DATA_SET(nc,data) ((nc)->dispatchdata = (void*)(data))
|
|
|
|
/* Cannot have NC_MPIPOSIX flag, ignore NC_MPIIO as PnetCDF use MPIIO */
|
|
static const int LEGAL_CREATE_FLAGS = (NC_NOCLOBBER | NC_64BIT_OFFSET | NC_CLASSIC_MODEL | NC_SHARE | NC_LOCK | NC_64BIT_DATA | NC_MPIIO);
|
|
|
|
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);
|
|
|
|
|
|
/**************************************************/
|
|
|
|
static int
|
|
NCP_create(const char *path, int cmode,
|
|
size_t initialsz, int basepe, size_t *chunksizehintp,
|
|
int use_parallel, void* mpidata,
|
|
struct NC_Dispatch* table, NC* nc)
|
|
{
|
|
int res, default_format;
|
|
NCP_INFO* nc5;
|
|
MPI_Comm comm;
|
|
MPI_Info info;
|
|
|
|
/* Check the cmode for only valid flags*/
|
|
if(cmode & ~LEGAL_CREATE_FLAGS)
|
|
{res = NC_EINVAL; goto done;}
|
|
|
|
/* No MPI environment initialized */
|
|
if (mpidata == NULL)
|
|
{res = NC_ENOPAR; goto done;}
|
|
|
|
comm = ((NC_MPI_INFO *)mpidata)->comm;
|
|
info = ((NC_MPI_INFO *)mpidata)->info;
|
|
|
|
/* Create our specific NCP_INFO instance */
|
|
|
|
nc5 = (NCP_INFO*)calloc(1,sizeof(NCP_INFO));
|
|
if(nc5 == NULL)
|
|
{res = NC_ENOMEM; goto done;}
|
|
|
|
/* Link nc5 and nc */
|
|
NCP_DATA_SET(nc,nc5);
|
|
|
|
res = ncmpi_create(comm, path, cmode, info, &(nc->int_ncid));
|
|
|
|
|
|
/* Default to independent access, like netCDF-4/HDF5 files. */
|
|
fSet(nc5->pnetcdf_access_mode, NCP_MODE_INDEP);
|
|
|
|
if(res && nc5 != NULL) free(nc5); /* reclaim allocated space */
|
|
|
|
done:
|
|
return res;
|
|
}
|
|
|
|
static int
|
|
NCP_open(const char *path, int cmode,
|
|
int basepe, size_t *chunksizehintp,
|
|
int use_parallel, void* mpidata,
|
|
struct NC_Dispatch* table, NC* nc)
|
|
{
|
|
int res;
|
|
NCP_INFO* nc5;
|
|
MPI_Comm comm;
|
|
MPI_Info info;
|
|
|
|
/* Check the cmode for only valid flags*/
|
|
if(cmode & ~LEGAL_OPEN_FLAGS)
|
|
{res = NC_EINVAL; goto done;}
|
|
|
|
/* No MPI environment initialized */
|
|
if (mpidata == NULL)
|
|
{res = NC_ENOPAR; goto done;}
|
|
|
|
comm = ((NC_MPI_INFO *)mpidata)->comm;
|
|
info = ((NC_MPI_INFO *)mpidata)->info;
|
|
|
|
/* Create our specific NCP_INFO instance */
|
|
nc5 = (NCP_INFO*)calloc(1,sizeof(NCP_INFO));
|
|
if(nc5 == NULL) {res = NC_ENOMEM; goto done;}
|
|
|
|
/* file open automatically enters data mode */
|
|
fSet(nc5->pnetcdf_access_mode, NCP_MODE_DATA);
|
|
|
|
/* Link nc5 and nc */
|
|
NCP_DATA_SET(nc,nc5);
|
|
|
|
res = ncmpi_open(comm, path, cmode, info, &(nc->int_ncid));
|
|
|
|
/* Default to independent access, like netCDF-4/HDF5 files. */
|
|
if (res == NC_NOERR) {
|
|
res = ncmpi_begin_indep_data(nc->int_ncid);
|
|
fSet(nc5->pnetcdf_access_mode, NCP_MODE_INDEP);
|
|
}
|
|
done:
|
|
return res;
|
|
}
|
|
|
|
static int
|
|
NCP_redef(int ncid)
|
|
{
|
|
NC* nc;
|
|
NCP_INFO* nc5;
|
|
|
|
int status = NC_check_id(ncid, &nc);
|
|
if(status != NC_NOERR) return status;
|
|
|
|
nc5 = NCP_DATA(nc);
|
|
fClr(nc5->pnetcdf_access_mode, NCP_MODE_DATA);
|
|
|
|
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)
|
|
{
|
|
int status;
|
|
NC* nc;
|
|
NCP_INFO* nc5;
|
|
MPI_Offset mpi_h_minfree = h_minfree;
|
|
MPI_Offset mpi_v_align = v_align;
|
|
MPI_Offset mpi_v_minfree = v_minfree;
|
|
MPI_Offset mpi_r_align = r_align;
|
|
|
|
status = NC_check_id(ncid, &nc);
|
|
if(status != NC_NOERR)
|
|
return status;
|
|
|
|
nc5 = NCP_DATA(nc);
|
|
assert(nc5);
|
|
|
|
#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_h_minfree, mpi_v_align,
|
|
mpi_v_minfree, mpi_r_align);
|
|
#else
|
|
status = ncmpi_enddef(nc->int_ncid);
|
|
#endif
|
|
|
|
if(!status) {
|
|
fSet(nc5->pnetcdf_access_mode, NCP_MODE_DATA);
|
|
if (fIsSet(nc5->pnetcdf_access_mode, NCP_MODE_INDEP))
|
|
status = ncmpi_begin_indep_data(nc->int_ncid);
|
|
}
|
|
return status;
|
|
}
|
|
|
|
static int
|
|
NCP_sync(int ncid)
|
|
{
|
|
NC* nc;
|
|
int status = NC_check_id(ncid, &nc);
|
|
if(status != NC_NOERR) return status;
|
|
return ncmpi_sync(nc->int_ncid);
|
|
}
|
|
|
|
static int
|
|
NCP_abort(int ncid)
|
|
{
|
|
NC* nc;
|
|
NCP_INFO* nc5;
|
|
int status = NC_check_id(ncid, &nc);
|
|
if(status != NC_NOERR) goto done;
|
|
|
|
status = ncmpi_abort(nc->int_ncid);
|
|
|
|
done:
|
|
nc5 = NCP_DATA(nc);
|
|
if(nc5 != NULL) free(nc5); /* reclaim allocated space */
|
|
return status;
|
|
}
|
|
|
|
|
|
static int
|
|
NCP_close(int ncid, void* ignored)
|
|
{
|
|
NC* nc;
|
|
NCP_INFO* nc5;
|
|
int status = NC_check_id(ncid, &nc);
|
|
if(status != NC_NOERR) goto done;
|
|
|
|
status = ncmpi_close(nc->int_ncid);
|
|
|
|
done:
|
|
nc5 = NCP_DATA(nc);
|
|
if(nc5 != NULL) free(nc5); /* reclaim allocated space */
|
|
return status;
|
|
}
|
|
|
|
static int
|
|
NCP_set_fill(int ncid, int fillmode, int *old_mode_ptr)
|
|
{
|
|
NC* nc;
|
|
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
|
|
}
|
|
|
|
static int
|
|
NCP_inq_base_pe(int ncid, int* pep)
|
|
{
|
|
if(pep) *pep = 0;
|
|
return NC_NOERR;
|
|
}
|
|
|
|
static int
|
|
NCP_set_base_pe(int ncid, int pe)
|
|
{
|
|
return NC_NOERR;
|
|
}
|
|
|
|
static int
|
|
NCP_inq_format(int ncid, int* formatp)
|
|
{
|
|
NC* nc;
|
|
int status = NC_check_id(ncid, &nc);
|
|
if(status != NC_NOERR) return status;
|
|
status = ncmpi_inq_format(nc->int_ncid,formatp);
|
|
return status;
|
|
}
|
|
|
|
static int
|
|
NCP_inq_format_extended(int ncid, int* formatp, int *modep)
|
|
{
|
|
NC* nc;
|
|
int status = NC_check_id(ncid, &nc);
|
|
if(status != NC_NOERR) return status;
|
|
if(modep) *modep = nc->mode;
|
|
/* Note that we do not use NC_FORMAT_CDF5 because PNETCDF has a dispatch table */
|
|
if(formatp) *formatp = NC_FORMATX_PNETCDF;
|
|
return NC_NOERR;
|
|
}
|
|
|
|
static int
|
|
NCP_inq(int ncid,
|
|
int *ndimsp,
|
|
int *nvarsp,
|
|
int *nattsp,
|
|
int *unlimp)
|
|
{
|
|
NC* nc;
|
|
int status = NC_check_id(ncid, &nc);
|
|
if(status != NC_NOERR) return status;
|
|
return ncmpi_inq(nc->int_ncid,ndimsp,nvarsp,nattsp,unlimp);
|
|
}
|
|
|
|
|
|
static int
|
|
NCP_inq_type(int ncid, nc_type typeid, char* name, size_t* size)
|
|
{
|
|
/* Assert mode & NC_FORMAT_CDF5 */
|
|
if (typeid < NC_BYTE || typeid >= NC_STRING)
|
|
return NC_EBADTYPE;
|
|
if(name)
|
|
strcpy(name, NC_atomictypename(typeid));
|
|
if(size)
|
|
*size = NC_atomictypelen(typeid);
|
|
return NC_NOERR;
|
|
}
|
|
|
|
static int
|
|
NCP_def_dim(int ncid, const char* name, size_t len, int* idp)
|
|
{
|
|
int status;
|
|
NCP_INFO* nc5;
|
|
NC* nc;
|
|
|
|
status = NC_check_id(ncid, &nc);
|
|
if(status != NC_NOERR)
|
|
return status;
|
|
|
|
nc5 = NCP_DATA(nc);
|
|
assert(nc5);
|
|
|
|
return ncmpi_def_dim(nc->int_ncid, name, len, idp);
|
|
}
|
|
|
|
static int
|
|
NCP_inq_dimid(int ncid, const char *name, int *idp)
|
|
{
|
|
NC* nc;
|
|
int status = NC_check_id(ncid, &nc);
|
|
if(status != NC_NOERR) return status;
|
|
return ncmpi_inq_dimid(nc->int_ncid,name,idp);
|
|
}
|
|
|
|
static int
|
|
NCP_inq_dim(int ncid, int dimid, char *name, size_t* lenp)
|
|
{
|
|
int status;
|
|
NC* nc;
|
|
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(lenp) *lenp = mpilen;
|
|
return status;
|
|
}
|
|
|
|
static int
|
|
NCP_inq_unlimdim(int ncid, int *unlimdimidp)
|
|
{
|
|
NC* nc;
|
|
int status = NC_check_id(ncid, &nc);
|
|
if(status != NC_NOERR) return status;
|
|
return ncmpi_inq_unlimdim(nc->int_ncid,unlimdimidp);
|
|
}
|
|
|
|
static int
|
|
NCP_rename_dim(int ncid, int dimid, const char* newname)
|
|
{
|
|
NC* nc;
|
|
int status = NC_check_id(ncid, &nc);
|
|
if(status != NC_NOERR) return status;
|
|
return ncmpi_rename_dim(nc->int_ncid,dimid,newname);
|
|
}
|
|
|
|
static int
|
|
NCP_inq_att(int ncid, int varid, const char* name, nc_type* xtypep, size_t* lenp)
|
|
{
|
|
NC* nc;
|
|
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) return status;
|
|
if(lenp) *lenp = mpilen;
|
|
return status;
|
|
}
|
|
|
|
static int
|
|
NCP_inq_attid(int ncid, int varid, const char *name, int *idp)
|
|
{
|
|
NC* nc;
|
|
int status = NC_check_id(ncid, &nc);
|
|
if(status != NC_NOERR) return status;
|
|
return ncmpi_inq_attid(nc->int_ncid,varid,name,idp);
|
|
}
|
|
|
|
static int
|
|
NCP_inq_attname(int ncid, int varid, int attnum, char *name)
|
|
{
|
|
NC* nc;
|
|
int status = NC_check_id(ncid, &nc);
|
|
if(status != NC_NOERR) return status;
|
|
return ncmpi_inq_attname(nc->int_ncid,varid,attnum,name);
|
|
|
|
}
|
|
|
|
static int
|
|
NCP_rename_att(int ncid, int varid, const char *name,
|
|
const char *newname)
|
|
{
|
|
NC* nc;
|
|
int status = NC_check_id(ncid, &nc);
|
|
if(status != NC_NOERR) return status;
|
|
return ncmpi_rename_att(nc->int_ncid,varid,name,newname);
|
|
}
|
|
|
|
static int
|
|
NCP_del_att(int ncid, int varid, const char *name)
|
|
{
|
|
NC* nc;
|
|
int status = NC_check_id(ncid, &nc);
|
|
if(status != NC_NOERR) return status;
|
|
return ncmpi_del_att(nc->int_ncid,varid,name);
|
|
}
|
|
|
|
int
|
|
NCP_get_att(
|
|
int ncid,
|
|
int varid,
|
|
const char *name,
|
|
void *ip,
|
|
nc_type memtype)
|
|
{
|
|
NC* nc;
|
|
int status;
|
|
|
|
status = NC_check_id(ncid, &nc);
|
|
if(status != NC_NOERR) return status;
|
|
|
|
if (memtype == NC_NAT) {
|
|
status = NCP_inq_att(ncid,varid,name,&memtype,NULL);
|
|
if (status != NC_NOERR) return status;
|
|
}
|
|
|
|
switch (memtype) {
|
|
case NC_CHAR:
|
|
return ncmpi_get_att_text(nc->int_ncid, varid, name, (char*)ip);
|
|
case NC_BYTE:
|
|
return ncmpi_get_att_schar(nc->int_ncid, varid, name, (signed char*)ip);
|
|
case NC_SHORT:
|
|
return ncmpi_get_att_short(nc->int_ncid, varid, name, (short*)ip);
|
|
case NC_INT:
|
|
return ncmpi_get_att_int(nc->int_ncid, varid, name, (int*)ip);
|
|
case NC_FLOAT:
|
|
return ncmpi_get_att_float(nc->int_ncid, varid, name, (float*)ip);
|
|
case NC_DOUBLE:
|
|
return ncmpi_get_att_double(nc->int_ncid, varid, name, (double*)ip);
|
|
case NC_UBYTE:
|
|
return ncmpi_get_att_uchar(nc->int_ncid, varid, name, (unsigned char*)ip);
|
|
case NC_USHORT:
|
|
return ncmpi_get_att_ushort(nc->int_ncid, varid, name, (unsigned short*)ip);
|
|
case NC_UINT:
|
|
return ncmpi_get_att_uint(nc->int_ncid, varid, name, (unsigned int*)ip);
|
|
case NC_INT64:
|
|
return ncmpi_get_att_longlong(nc->int_ncid, varid, name, (long long*)ip);
|
|
case NC_UINT64:
|
|
return ncmpi_get_att_ulonglong(nc->int_ncid, varid, name, (unsigned long long*)ip);
|
|
default:
|
|
break;
|
|
}
|
|
return NC_EBADTYPE;
|
|
}
|
|
|
|
int
|
|
NCP_put_att(
|
|
int ncid,
|
|
int varid,
|
|
const char *name,
|
|
nc_type xtype,
|
|
size_t len,
|
|
const void *ip,
|
|
nc_type memtype)
|
|
{
|
|
NC* nc;
|
|
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;
|
|
|
|
if (!name || (strlen(name) > NC_MAX_NAME))
|
|
return NC_EBADNAME;
|
|
|
|
/* 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;
|
|
|
|
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:
|
|
break;
|
|
}
|
|
return NC_EBADTYPE;
|
|
}
|
|
|
|
static int
|
|
NCP_def_var(int ncid, const char *name, nc_type xtype,
|
|
int ndims, const int *dimidsp, int *varidp)
|
|
{
|
|
NC* nc;
|
|
NCP_INFO* nc5;
|
|
int status;
|
|
|
|
status = NC_check_id(ncid, &nc);
|
|
if(status != NC_NOERR) return status;
|
|
nc5 = NCP_DATA(nc);
|
|
assert(nc5);
|
|
|
|
status = ncmpi_def_var(nc->int_ncid,name,xtype,ndims,dimidsp,varidp);
|
|
return status;
|
|
}
|
|
|
|
static int
|
|
NCP_inq_varid(int ncid, const char *name, int *varidp)
|
|
{
|
|
NC* nc;
|
|
int status = NC_check_id(ncid, &nc);
|
|
if(status != NC_NOERR) return status;
|
|
return ncmpi_inq_varid(nc->int_ncid,name,varidp);
|
|
}
|
|
|
|
static int
|
|
NCP_rename_var(int ncid, int varid, const char *name)
|
|
{
|
|
NC* nc;
|
|
int status = NC_check_id(ncid, &nc);
|
|
if(status != NC_NOERR) return status;
|
|
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* ip,
|
|
nc_type memtype)
|
|
{
|
|
NC* nc;
|
|
NCP_INFO* nc5;
|
|
int status;
|
|
MPI_Offset mpi_start[NC_MAX_VAR_DIMS], mpi_count[NC_MAX_VAR_DIMS];
|
|
int d;
|
|
int rank = 0;
|
|
|
|
status = NC_check_id(ncid, &nc);
|
|
if(status != NC_NOERR) return status;
|
|
|
|
nc5 = NCP_DATA(nc);
|
|
assert(nc5);
|
|
|
|
/* get variable's rank */
|
|
status= ncmpi_inq_varndims(nc->int_ncid, varid, &rank);
|
|
if(status) return status;
|
|
|
|
/* We must convert the start and count arrays to MPI_Offset type. */
|
|
for (d = 0; d < rank; d++) {
|
|
mpi_start[d] = startp[d];
|
|
mpi_count[d] = countp[d];
|
|
}
|
|
|
|
if (memtype == NC_NAT) {
|
|
status = ncmpi_inq_vartype(nc->int_ncid, varid, &memtype);
|
|
if (status) return status;
|
|
}
|
|
|
|
if (fIsSet(nc5->pnetcdf_access_mode, NCP_MODE_INDEP)) {
|
|
switch(memtype) {
|
|
case NC_BYTE:
|
|
status=ncmpi_get_vara_schar(nc->int_ncid, varid, mpi_start, mpi_count, ip); break;
|
|
case NC_CHAR:
|
|
status=ncmpi_get_vara_text(nc->int_ncid, varid, mpi_start, mpi_count, ip); break;
|
|
case NC_SHORT:
|
|
status=ncmpi_get_vara_short(nc->int_ncid, varid, mpi_start, mpi_count, ip); break;
|
|
case NC_INT:
|
|
status=ncmpi_get_vara_int(nc->int_ncid, varid, mpi_start, mpi_count, ip); break;
|
|
case NC_FLOAT:
|
|
status=ncmpi_get_vara_float(nc->int_ncid, varid, mpi_start, mpi_count, ip); break;
|
|
case NC_DOUBLE:
|
|
status=ncmpi_get_vara_double(nc->int_ncid, varid, mpi_start, mpi_count, ip); break;
|
|
case NC_UBYTE:
|
|
status=ncmpi_get_vara_uchar(nc->int_ncid, varid, mpi_start, mpi_count, ip); break;
|
|
case NC_USHORT:
|
|
status=ncmpi_get_vara_ushort(nc->int_ncid, varid, mpi_start, mpi_count, ip); break;
|
|
case NC_UINT:
|
|
status=ncmpi_get_vara_uint(nc->int_ncid, varid, mpi_start, mpi_count, ip); break;
|
|
case NC_INT64:
|
|
status=ncmpi_get_vara_longlong(nc->int_ncid, varid, mpi_start, mpi_count, ip); break;
|
|
case NC_UINT64:
|
|
status=ncmpi_get_vara_ulonglong(nc->int_ncid, varid, mpi_start, mpi_count, ip); break;
|
|
default:
|
|
status = NC_EBADTYPE;
|
|
}
|
|
} else {
|
|
switch(memtype) {
|
|
case NC_BYTE:
|
|
status=ncmpi_get_vara_schar_all(nc->int_ncid, varid, mpi_start, mpi_count, ip); break;
|
|
case NC_CHAR:
|
|
status=ncmpi_get_vara_text_all(nc->int_ncid, varid, mpi_start, mpi_count, ip); break;
|
|
case NC_SHORT:
|
|
status=ncmpi_get_vara_short_all(nc->int_ncid, varid, mpi_start, mpi_count, ip); break;
|
|
case NC_INT:
|
|
status=ncmpi_get_vara_int_all(nc->int_ncid, varid, mpi_start, mpi_count, ip); break;
|
|
case NC_FLOAT:
|
|
status=ncmpi_get_vara_float_all(nc->int_ncid, varid, mpi_start, mpi_count, ip); break;
|
|
case NC_DOUBLE:
|
|
status=ncmpi_get_vara_double_all(nc->int_ncid, varid, mpi_start, mpi_count, ip); break;
|
|
case NC_UBYTE:
|
|
status=ncmpi_get_vara_uchar_all(nc->int_ncid, varid, mpi_start, mpi_count, ip); break;
|
|
case NC_USHORT:
|
|
status=ncmpi_get_vara_ushort_all(nc->int_ncid, varid, mpi_start, mpi_count, ip); break;
|
|
case NC_UINT:
|
|
status=ncmpi_get_vara_uint_all(nc->int_ncid, varid, mpi_start, mpi_count, ip); break;
|
|
case NC_INT64:
|
|
status=ncmpi_get_vara_longlong_all(nc->int_ncid, varid, mpi_start, mpi_count, ip); break;
|
|
case NC_UINT64:
|
|
status=ncmpi_get_vara_ulonglong_all(nc->int_ncid, varid, mpi_start, mpi_count, ip); break;
|
|
default:
|
|
status = NC_EBADTYPE;
|
|
}
|
|
}
|
|
return status;
|
|
}
|
|
|
|
static int
|
|
NCP_put_vara(int ncid,
|
|
int varid,
|
|
const size_t* startp,
|
|
const size_t* countp,
|
|
const void*ip,
|
|
nc_type memtype)
|
|
{
|
|
NC* nc;
|
|
NCP_INFO* nc5;
|
|
int status;
|
|
MPI_Offset mpi_start[NC_MAX_VAR_DIMS], mpi_count[NC_MAX_VAR_DIMS];
|
|
int d;
|
|
int rank;
|
|
|
|
status = NC_check_id(ncid, &nc);
|
|
if(status != NC_NOERR) return status;
|
|
|
|
nc5 = NCP_DATA(nc);
|
|
assert(nc5);
|
|
|
|
/* get variable's rank */
|
|
status = ncmpi_inq_varndims(nc->int_ncid, varid, &rank);
|
|
if(status) return status;
|
|
|
|
/* We must convert the start and count arrays to MPI_Offset type. */
|
|
for (d = 0; d < rank; d++) {
|
|
mpi_start[d] = startp[d];
|
|
mpi_count[d] = countp[d];
|
|
}
|
|
|
|
if (memtype == NC_NAT) {
|
|
status = ncmpi_inq_vartype(nc->int_ncid, varid, &memtype);
|
|
if (status) return status;
|
|
}
|
|
|
|
if (fIsSet(nc5->pnetcdf_access_mode, NCP_MODE_INDEP)) {
|
|
switch(memtype) {
|
|
case NC_BYTE:
|
|
status = ncmpi_put_vara_schar(nc->int_ncid, varid, mpi_start, mpi_count, ip); break;
|
|
case NC_CHAR:
|
|
status = ncmpi_put_vara_text(nc->int_ncid, varid, mpi_start, mpi_count, ip); break;
|
|
case NC_SHORT:
|
|
status = ncmpi_put_vara_short(nc->int_ncid, varid, mpi_start, mpi_count, ip); break;
|
|
case NC_INT:
|
|
status = ncmpi_put_vara_int(nc->int_ncid, varid, mpi_start, mpi_count, ip); break;
|
|
case NC_FLOAT:
|
|
status = ncmpi_put_vara_float(nc->int_ncid, varid, mpi_start, mpi_count, ip); break;
|
|
case NC_DOUBLE:
|
|
status = ncmpi_put_vara_double(nc->int_ncid, varid, mpi_start, mpi_count, ip); break;
|
|
case NC_UBYTE:
|
|
status = ncmpi_put_vara_uchar(nc->int_ncid, varid, mpi_start, mpi_count, ip); break;
|
|
case NC_USHORT:
|
|
status = ncmpi_put_vara_ushort(nc->int_ncid, varid, mpi_start, mpi_count, ip); break;
|
|
case NC_UINT:
|
|
status = ncmpi_put_vara_uint(nc->int_ncid, varid, mpi_start, mpi_count, ip); break;
|
|
case NC_INT64:
|
|
status = ncmpi_put_vara_longlong(nc->int_ncid, varid, mpi_start, mpi_count, ip); break;
|
|
case NC_UINT64:
|
|
status = ncmpi_put_vara_ulonglong(nc->int_ncid, varid, mpi_start, mpi_count, ip); break;
|
|
default:
|
|
status = NC_EBADTYPE;
|
|
}
|
|
} else {
|
|
switch(memtype) {
|
|
case NC_BYTE:
|
|
status = ncmpi_put_vara_schar_all(nc->int_ncid, varid, mpi_start, mpi_count, ip); break;
|
|
case NC_CHAR:
|
|
status = ncmpi_put_vara_text_all(nc->int_ncid, varid, mpi_start, mpi_count, ip); break;
|
|
case NC_SHORT:
|
|
status = ncmpi_put_vara_short_all(nc->int_ncid, varid, mpi_start, mpi_count, ip); break;
|
|
case NC_INT:
|
|
status = ncmpi_put_vara_int_all(nc->int_ncid, varid, mpi_start, mpi_count, ip); break;
|
|
case NC_FLOAT:
|
|
status = ncmpi_put_vara_float_all(nc->int_ncid, varid, mpi_start, mpi_count, ip); break;
|
|
case NC_DOUBLE:
|
|
status = ncmpi_put_vara_double_all(nc->int_ncid, varid, mpi_start, mpi_count, ip); break;
|
|
case NC_UBYTE:
|
|
status = ncmpi_put_vara_uchar_all(nc->int_ncid, varid, mpi_start, mpi_count, ip); break;
|
|
case NC_USHORT:
|
|
status = ncmpi_put_vara_ushort_all(nc->int_ncid, varid, mpi_start, mpi_count, ip); break;
|
|
case NC_UINT:
|
|
status = ncmpi_put_vara_uint_all(nc->int_ncid, varid, mpi_start, mpi_count, ip); break;
|
|
case NC_INT64:
|
|
status = ncmpi_put_vara_longlong_all(nc->int_ncid, varid, mpi_start, mpi_count, ip); break;
|
|
case NC_UINT64:
|
|
status = ncmpi_put_vara_ulonglong_all(nc->int_ncid, varid, mpi_start, mpi_count, ip); break;
|
|
default:
|
|
status = NC_EBADTYPE;
|
|
}
|
|
}
|
|
return status;
|
|
}
|
|
|
|
static int
|
|
NCP_get_vars(int ncid,
|
|
int varid,
|
|
const size_t* startp,
|
|
const size_t* countp,
|
|
const ptrdiff_t* stridep,
|
|
void* ip,
|
|
nc_type memtype)
|
|
{
|
|
NC* nc;
|
|
NCP_INFO* nc5;
|
|
int status;
|
|
MPI_Offset mpi_start[NC_MAX_VAR_DIMS], mpi_count[NC_MAX_VAR_DIMS], mpi_stride[NC_MAX_VAR_DIMS];
|
|
int d;
|
|
int rank = 0;
|
|
|
|
status = NC_check_id(ncid, &nc);
|
|
if(status != NC_NOERR) return status;
|
|
|
|
nc5 = NCP_DATA(nc);
|
|
assert(nc5);
|
|
|
|
/* get variable's rank */
|
|
status= ncmpi_inq_varndims(nc->int_ncid, varid, &rank);
|
|
if(status) return status;
|
|
|
|
/* We must convert the start, count, and stride arrays to MPI_Offset type. */
|
|
for (d = 0; d < rank; d++) {
|
|
mpi_start[d] = startp[d];
|
|
mpi_count[d] = countp[d];
|
|
mpi_stride[d] = stridep[d];
|
|
}
|
|
|
|
if (memtype == NC_NAT) {
|
|
status = ncmpi_inq_vartype(nc->int_ncid, varid, &memtype);
|
|
if (status) return status;
|
|
}
|
|
|
|
if (fIsSet(nc5->pnetcdf_access_mode, NCP_MODE_INDEP)) {
|
|
switch(memtype) {
|
|
case NC_BYTE:
|
|
status=ncmpi_get_vars_schar(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, ip); break;
|
|
case NC_CHAR:
|
|
status=ncmpi_get_vars_text(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, ip); break;
|
|
case NC_SHORT:
|
|
status=ncmpi_get_vars_short(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, ip); break;
|
|
case NC_INT:
|
|
status=ncmpi_get_vars_int(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, ip); break;
|
|
case NC_FLOAT:
|
|
status=ncmpi_get_vars_float(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, ip); break;
|
|
case NC_DOUBLE:
|
|
status=ncmpi_get_vars_double(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, ip); break;
|
|
case NC_UBYTE:
|
|
status=ncmpi_get_vars_uchar(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, ip); break;
|
|
case NC_USHORT:
|
|
status=ncmpi_get_vars_ushort(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, ip); break;
|
|
case NC_UINT:
|
|
status=ncmpi_get_vars_uint(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, ip); break;
|
|
case NC_INT64:
|
|
status=ncmpi_get_vars_longlong(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, ip); break;
|
|
case NC_UINT64:
|
|
status=ncmpi_get_vars_ulonglong(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, ip); break;
|
|
default:
|
|
status = NC_EBADTYPE;
|
|
}
|
|
} else {
|
|
switch(memtype) {
|
|
case NC_BYTE:
|
|
status=ncmpi_get_vars_schar_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, ip); break;
|
|
case NC_CHAR:
|
|
status=ncmpi_get_vars_text_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, ip); break;
|
|
case NC_SHORT:
|
|
status=ncmpi_get_vars_short_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, ip); break;
|
|
case NC_INT:
|
|
status=ncmpi_get_vars_int_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, ip); break;
|
|
case NC_FLOAT:
|
|
status=ncmpi_get_vars_float_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, ip); break;
|
|
case NC_DOUBLE:
|
|
status=ncmpi_get_vars_double_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, ip); break;
|
|
case NC_UBYTE:
|
|
status=ncmpi_get_vars_uchar_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, ip); break;
|
|
case NC_USHORT:
|
|
status=ncmpi_get_vars_ushort_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, ip); break;
|
|
case NC_UINT:
|
|
status=ncmpi_get_vars_uint_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, ip); break;
|
|
case NC_INT64:
|
|
status=ncmpi_get_vars_longlong_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, ip); break;
|
|
case NC_UINT64:
|
|
status=ncmpi_get_vars_ulonglong_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, ip); break;
|
|
default:
|
|
status = NC_EBADTYPE;
|
|
}
|
|
}
|
|
return status;
|
|
}
|
|
|
|
static int
|
|
NCP_put_vars(int ncid,
|
|
int varid,
|
|
const size_t* startp,
|
|
const size_t* countp,
|
|
const ptrdiff_t* stridep,
|
|
const void*ip,
|
|
nc_type memtype)
|
|
{
|
|
NC* nc;
|
|
NCP_INFO* nc5;
|
|
int status;
|
|
MPI_Offset mpi_start[NC_MAX_VAR_DIMS], mpi_count[NC_MAX_VAR_DIMS], mpi_stride[NC_MAX_VAR_DIMS];
|
|
int d;
|
|
int rank;
|
|
|
|
status = NC_check_id(ncid, &nc);
|
|
if(status != NC_NOERR) return status;
|
|
|
|
nc5 = NCP_DATA(nc);
|
|
assert(nc5);
|
|
|
|
/* get variable's rank */
|
|
status = ncmpi_inq_varndims(nc->int_ncid, varid, &rank);
|
|
if(status) return status;
|
|
|
|
/* We must convert the start, count, and stride arrays to MPI_Offset type. */
|
|
for (d = 0; d < rank; d++) {
|
|
mpi_start[d] = startp[d];
|
|
mpi_count[d] = countp[d];
|
|
mpi_stride[d] = stridep[d];
|
|
}
|
|
|
|
if (memtype == NC_NAT) {
|
|
status = ncmpi_inq_vartype(nc->int_ncid, varid, &memtype);
|
|
if (status) return status;
|
|
}
|
|
|
|
if (fIsSet(nc5->pnetcdf_access_mode, NCP_MODE_INDEP)) {
|
|
switch(memtype) {
|
|
case NC_BYTE:
|
|
status = ncmpi_put_vars_schar(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, ip); break;
|
|
case NC_CHAR:
|
|
status = ncmpi_put_vars_text(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, ip); break;
|
|
case NC_SHORT:
|
|
status = ncmpi_put_vars_short(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, ip); break;
|
|
case NC_INT:
|
|
status = ncmpi_put_vars_int(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, ip); break;
|
|
case NC_FLOAT:
|
|
status = ncmpi_put_vars_float(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, ip); break;
|
|
case NC_DOUBLE:
|
|
status = ncmpi_put_vars_double(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, ip); break;
|
|
case NC_UBYTE:
|
|
status = ncmpi_put_vars_uchar(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, ip); break;
|
|
case NC_USHORT:
|
|
status = ncmpi_put_vars_ushort(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, ip); break;
|
|
case NC_UINT:
|
|
status = ncmpi_put_vars_uint(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, ip); break;
|
|
case NC_INT64:
|
|
status = ncmpi_put_vars_longlong(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, ip); break;
|
|
case NC_UINT64:
|
|
status = ncmpi_put_vars_ulonglong(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, ip); break;
|
|
default:
|
|
status = NC_EBADTYPE;
|
|
}
|
|
} else {
|
|
switch(memtype) {
|
|
case NC_BYTE:
|
|
status = ncmpi_put_vars_schar_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, ip); break;
|
|
case NC_CHAR:
|
|
status = ncmpi_put_vars_text_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, ip); break;
|
|
case NC_SHORT:
|
|
status = ncmpi_put_vars_short_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, ip); break;
|
|
case NC_INT:
|
|
status = ncmpi_put_vars_int_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, ip); break;
|
|
case NC_FLOAT:
|
|
status = ncmpi_put_vars_float_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, ip); break;
|
|
case NC_DOUBLE:
|
|
status = ncmpi_put_vars_double_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, ip); break;
|
|
case NC_UBYTE:
|
|
status = ncmpi_put_vars_uchar_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, ip); break;
|
|
case NC_USHORT:
|
|
status = ncmpi_put_vars_ushort_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, ip); break;
|
|
case NC_UINT:
|
|
status = ncmpi_put_vars_uint_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, ip); break;
|
|
case NC_INT64:
|
|
status = ncmpi_put_vars_longlong_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, ip); break;
|
|
case NC_UINT64:
|
|
status = ncmpi_put_vars_ulonglong_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, ip); break;
|
|
default:
|
|
status = NC_EBADTYPE;
|
|
}
|
|
}
|
|
return status;
|
|
}
|
|
|
|
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)
|
|
{
|
|
NC* nc;
|
|
NCP_INFO* nc5;
|
|
int status;
|
|
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];
|
|
int d;
|
|
int rank = 0;
|
|
|
|
status = NC_check_id(ncid, &nc);
|
|
if(status != NC_NOERR) return status;
|
|
|
|
nc5 = NCP_DATA(nc);
|
|
assert(nc5);
|
|
|
|
/* get variable's rank */
|
|
status= ncmpi_inq_varndims(nc->int_ncid, varid, &rank);
|
|
if(status) return status;
|
|
|
|
/* We must convert the start, count, stride, and imap arrays to MPI_Offset type. */
|
|
for (d = 0; d < rank; d++) {
|
|
mpi_start[d] = startp[d];
|
|
mpi_count[d] = countp[d];
|
|
mpi_stride[d] = stridep[d];
|
|
mpi_imap[d] = imapp[d];
|
|
}
|
|
|
|
if (memtype == NC_NAT) {
|
|
status = ncmpi_inq_vartype(nc->int_ncid, varid, &memtype);
|
|
if (status) return status;
|
|
}
|
|
|
|
if (fIsSet(nc5->pnetcdf_access_mode, NCP_MODE_INDEP)) {
|
|
switch(memtype) {
|
|
case NC_BYTE:
|
|
status=ncmpi_get_varm_schar(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, ip); break;
|
|
case NC_CHAR:
|
|
status=ncmpi_get_varm_text(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, ip); break;
|
|
case NC_SHORT:
|
|
status=ncmpi_get_varm_short(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, ip); break;
|
|
case NC_INT:
|
|
status=ncmpi_get_varm_int(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, ip); break;
|
|
case NC_FLOAT:
|
|
status=ncmpi_get_varm_float(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, ip); break;
|
|
case NC_DOUBLE:
|
|
status=ncmpi_get_varm_double(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, ip); break;
|
|
case NC_UBYTE:
|
|
status=ncmpi_get_varm_uchar(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, ip); break;
|
|
case NC_USHORT:
|
|
status=ncmpi_get_varm_ushort(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, ip); break;
|
|
case NC_UINT:
|
|
status=ncmpi_get_varm_uint(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, ip); break;
|
|
case NC_INT64:
|
|
status=ncmpi_get_varm_longlong(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, ip); break;
|
|
case NC_UINT64:
|
|
status=ncmpi_get_varm_ulonglong(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, ip); break;
|
|
default:
|
|
status = NC_EBADTYPE;
|
|
}
|
|
} else {
|
|
switch(memtype) {
|
|
case NC_BYTE:
|
|
status=ncmpi_get_varm_schar_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, ip); break;
|
|
case NC_CHAR:
|
|
status=ncmpi_get_varm_text_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, ip); break;
|
|
case NC_SHORT:
|
|
status=ncmpi_get_varm_short_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, ip); break;
|
|
case NC_INT:
|
|
status=ncmpi_get_varm_int_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, ip); break;
|
|
case NC_FLOAT:
|
|
status=ncmpi_get_varm_float_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, ip); break;
|
|
case NC_DOUBLE:
|
|
status=ncmpi_get_varm_double_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, ip); break;
|
|
case NC_UBYTE:
|
|
status=ncmpi_get_varm_uchar_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, ip); break;
|
|
case NC_USHORT:
|
|
status=ncmpi_get_varm_ushort_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, ip); break;
|
|
case NC_UINT:
|
|
status=ncmpi_get_varm_uint_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, ip); break;
|
|
case NC_INT64:
|
|
status=ncmpi_get_varm_longlong_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, ip); break;
|
|
case NC_UINT64:
|
|
status=ncmpi_get_varm_ulonglong_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, ip); break;
|
|
default:
|
|
status = NC_EBADTYPE;
|
|
}
|
|
}
|
|
return status;
|
|
}
|
|
|
|
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*ip,
|
|
nc_type memtype)
|
|
{
|
|
NC* nc;
|
|
NCP_INFO* nc5;
|
|
int status;
|
|
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];
|
|
int d;
|
|
int rank;
|
|
|
|
status = NC_check_id(ncid, &nc);
|
|
if(status != NC_NOERR) return status;
|
|
|
|
nc5 = NCP_DATA(nc);
|
|
assert(nc5);
|
|
|
|
/* get variable's rank */
|
|
status = ncmpi_inq_varndims(nc->int_ncid, varid, &rank);
|
|
if(status) return status;
|
|
|
|
/* We must convert the start, count, stride, and imap arrays to MPI_Offset type. */
|
|
for (d = 0; d < rank; d++) {
|
|
mpi_start[d] = startp[d];
|
|
mpi_count[d] = countp[d];
|
|
mpi_stride[d] = stridep[d];
|
|
mpi_imap[d] = imapp[d];
|
|
}
|
|
|
|
if (memtype == NC_NAT) {
|
|
status = ncmpi_inq_vartype(nc->int_ncid, varid, &memtype);
|
|
if (status) return status;
|
|
}
|
|
|
|
if (fIsSet(nc5->pnetcdf_access_mode, NCP_MODE_INDEP)) {
|
|
switch(memtype) {
|
|
case NC_BYTE:
|
|
status = ncmpi_put_varm_schar(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, ip); break;
|
|
case NC_CHAR:
|
|
status = ncmpi_put_varm_text(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, ip); break;
|
|
case NC_SHORT:
|
|
status = ncmpi_put_varm_short(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, ip); break;
|
|
case NC_INT:
|
|
status = ncmpi_put_varm_int(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, ip); break;
|
|
case NC_FLOAT:
|
|
status = ncmpi_put_varm_float(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, ip); break;
|
|
case NC_DOUBLE:
|
|
status = ncmpi_put_varm_double(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, ip); break;
|
|
case NC_UBYTE:
|
|
status = ncmpi_put_varm_uchar(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, ip); break;
|
|
case NC_USHORT:
|
|
status = ncmpi_put_varm_ushort(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, ip); break;
|
|
case NC_UINT:
|
|
status = ncmpi_put_varm_uint(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, ip); break;
|
|
case NC_INT64:
|
|
status = ncmpi_put_varm_longlong(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, ip); break;
|
|
case NC_UINT64:
|
|
status = ncmpi_put_varm_ulonglong(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, ip); break;
|
|
default:
|
|
status = NC_EBADTYPE;
|
|
}
|
|
} else {
|
|
switch(memtype) {
|
|
case NC_BYTE:
|
|
status = ncmpi_put_varm_schar_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, ip); break;
|
|
case NC_CHAR:
|
|
status = ncmpi_put_varm_text_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, ip); break;
|
|
case NC_SHORT:
|
|
status = ncmpi_put_varm_short_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, ip); break;
|
|
case NC_INT:
|
|
status = ncmpi_put_varm_int_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, ip); break;
|
|
case NC_FLOAT:
|
|
status = ncmpi_put_varm_float_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, ip); break;
|
|
case NC_DOUBLE:
|
|
status = ncmpi_put_varm_double_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, ip); break;
|
|
case NC_UBYTE:
|
|
status = ncmpi_put_varm_uchar_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, ip); break;
|
|
case NC_USHORT:
|
|
status = ncmpi_put_varm_ushort_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, ip); break;
|
|
case NC_UINT:
|
|
status = ncmpi_put_varm_uint_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, ip); break;
|
|
case NC_INT64:
|
|
status = ncmpi_put_varm_longlong_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, ip); break;
|
|
case NC_UINT64:
|
|
status = ncmpi_put_varm_ulonglong_all(nc->int_ncid, varid, mpi_start, mpi_count, mpi_stride, mpi_imap, ip); break;
|
|
default:
|
|
status = NC_EBADTYPE;
|
|
}
|
|
}
|
|
return status;
|
|
}
|
|
|
|
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)
|
|
{
|
|
int status;
|
|
NC* nc;
|
|
|
|
status = NC_check_id(ncid, &nc);
|
|
if(status != NC_NOERR) return status;
|
|
|
|
status = ncmpi_inq_var(nc->int_ncid, varid, name, xtypep, ndimsp, dimidsp, nattsp);
|
|
if(status) 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) ncmpi_inq_var_fill(nc->int_ncid, varid, no_fill, fill_valuep);
|
|
#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;
|
|
return NC_NOERR;
|
|
}
|
|
|
|
static int
|
|
NCP_def_var_fill(int ncid, int varid, int no_fill, const void *fill_value)
|
|
{
|
|
NC* nc;
|
|
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 */
|
|
return ncmpi_def_var_fill(nc->int_ncid, varid, no_fill, fill_value);
|
|
#else
|
|
return NC_EPNETCDF;
|
|
#endif
|
|
}
|
|
|
|
static int
|
|
NCP_var_par_access(int ncid, int varid, int par_access)
|
|
{
|
|
NC *nc;
|
|
NCP_INFO* nc5;
|
|
int status;
|
|
|
|
if (par_access != NC_INDEPENDENT && par_access != NC_COLLECTIVE)
|
|
return NC_EINVAL;
|
|
|
|
#ifdef _DO_NOT_IGNORE_VARID_
|
|
if (varid != NC_GLOBAL) /* PnetCDF cannot do per-veriable mode change */
|
|
return NC_EINVAL;
|
|
#endif
|
|
|
|
status = NC_check_id(ncid, &nc);
|
|
if(status != NC_NOERR) return status;
|
|
|
|
nc5 = NCP_DATA(nc);
|
|
assert(nc5);
|
|
|
|
/* 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;
|
|
}
|
|
|
|
#ifdef USE_NETCDF4
|
|
|
|
static int
|
|
NCP_show_metadata(int ncid)
|
|
{
|
|
return NC_NOERR;
|
|
}
|
|
|
|
static int
|
|
NCP_inq_unlimdims(int ncid, int *ndimsp, int *unlimdimidsp)
|
|
{
|
|
int retval;
|
|
int unlimid;
|
|
|
|
if((retval = NCP_inq_unlimdim(ncid, &unlimid)))
|
|
return retval;
|
|
if(unlimid != -1) {
|
|
if(ndimsp) *ndimsp = 1;
|
|
if(unlimdimidsp)
|
|
unlimdimidsp[0] = unlimid;
|
|
} else
|
|
if(ndimsp) *ndimsp = 0;
|
|
return NC_NOERR;
|
|
}
|
|
|
|
static int
|
|
NCP_inq_type_equal(int ncid1, nc_type typeid1, int ncid2, nc_type typeid2, int* equalp)
|
|
{
|
|
/* Check input. */
|
|
if(equalp == NULL) return NC_NOERR;
|
|
|
|
if (typeid1 <= NC_NAT || typeid2 <= NC_NAT)
|
|
return NC_EINVAL;
|
|
|
|
*equalp = 0; /* assume */
|
|
|
|
/* 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;
|
|
}
|
|
|
|
/* If both are atomic types, the answer is easy. */
|
|
if (typeid1 <= ATOMICTYPEMAX5) {
|
|
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)
|
|
{
|
|
return NC_ENOTNC4;
|
|
}
|
|
|
|
static int
|
|
NCP_rename_grp(int ncid, const char *name)
|
|
{
|
|
return NC_ENOTNC4;
|
|
}
|
|
|
|
static int
|
|
NCP_inq_ncid(int ncid, const char *name, int *grp_ncid)
|
|
{
|
|
if(grp_ncid) *grp_ncid = ncid;
|
|
return NC_NOERR;
|
|
}
|
|
|
|
static int
|
|
NCP_inq_grps(int ncid, int *numgrps, int *ncids)
|
|
{
|
|
if(numgrps)
|
|
*numgrps = 0;
|
|
return NC_NOERR;
|
|
}
|
|
|
|
static int
|
|
NCP_inq_grpname(int ncid, char *name)
|
|
{
|
|
if(name)
|
|
strcpy(name, "/");
|
|
return NC_NOERR;
|
|
}
|
|
|
|
static int
|
|
NCP_inq_grpname_full(int ncid, size_t *lenp, char *full_name)
|
|
{
|
|
if(full_name)
|
|
strcpy(full_name, "/");
|
|
if(lenp) *lenp = 1;
|
|
return NC_NOERR;
|
|
}
|
|
|
|
static int
|
|
NCP_inq_grp_parent(int ncid, int *parent_ncid)
|
|
{
|
|
return NC_ENOGRP;
|
|
}
|
|
|
|
static int
|
|
NCP_inq_grp_full_ncid(int ncid, const char *full_name, int *grp_ncid)
|
|
{
|
|
return NC_ENOGRP;
|
|
}
|
|
|
|
static int
|
|
NCP_inq_varids(int ncid, int *nvarsp, int *varids)
|
|
{
|
|
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)))
|
|
return retval;
|
|
if(nvarsp) *nvarsp = nvars;
|
|
if(varids)
|
|
for (v = 0; v < nvars; v++)
|
|
varids[v] = v;
|
|
return NC_NOERR;
|
|
}
|
|
|
|
static int
|
|
NCP_inq_dimids(int ncid, int *ndimsp, int *dimids, int include_parents)
|
|
{
|
|
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)))
|
|
return retval;
|
|
if(ndimsp) *ndimsp = ndims;
|
|
if(dimids)
|
|
for (d = 0; d < ndims; d++)
|
|
dimids[d] = d;
|
|
return NC_NOERR;
|
|
}
|
|
|
|
static int
|
|
NCP_inq_typeid(int ncid, const char *name, nc_type *typeidp)
|
|
{
|
|
int i;
|
|
for (i = 0; i <= ATOMICTYPEMAX5; i++)
|
|
if(!strcmp(name, NC_atomictypename(i))) {
|
|
if(typeidp) *typeidp = i;
|
|
return NC_NOERR;
|
|
}
|
|
return NC_ENOTNC4;
|
|
}
|
|
|
|
static int
|
|
NCP_inq_typeids(int ncid, int *ntypes, int *typeids)
|
|
{
|
|
if(ntypes) *ntypes = 0;
|
|
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)
|
|
{
|
|
return NC_ENOTNC4;
|
|
}
|
|
|
|
#endif /*USE_NETCDF4*/
|
|
|
|
/**************************************************/
|
|
/* Pnetcdf Dispatch table */
|
|
|
|
NC_Dispatch NCP_dispatcher = {
|
|
|
|
NC_FORMATX_PNETCDF,
|
|
|
|
NCP_create,
|
|
NCP_open,
|
|
|
|
NCP_redef,
|
|
NCP__enddef,
|
|
NCP_sync,
|
|
NCP_abort,
|
|
NCP_close,
|
|
NCP_set_fill,
|
|
NCP_inq_base_pe,
|
|
NCP_set_base_pe,
|
|
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,
|
|
NCP_get_vars,
|
|
NCP_put_vars,
|
|
NCP_get_varm,
|
|
NCP_put_varm,
|
|
|
|
NCP_inq_var_all,
|
|
|
|
NCP_var_par_access,
|
|
NCP_def_var_fill,
|
|
|
|
#ifdef USE_NETCDF4
|
|
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,
|
|
|
|
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,
|
|
#endif /*USE_NETCDF4*/
|
|
|
|
};
|
|
|
|
NC_Dispatch* NCP_dispatch_table = NULL; /* moved here from ddispatch.c */
|
|
|
|
int
|
|
NCP_initialize(void)
|
|
{
|
|
NCP_dispatch_table = &NCP_dispatcher;
|
|
return NC_NOERR;
|
|
}
|
|
|
|
int
|
|
NCP_finalize(void)
|
|
{
|
|
return NC_NOERR;
|
|
}
|