2019-02-03 22:43:56 +08:00
|
|
|
/* Copyright 2003-2019, University Corporation for Atmospheric
|
2018-05-15 20:51:44 +08:00
|
|
|
* Research. See the COPYRIGHT file for copying and redistribution
|
|
|
|
* conditions. */
|
|
|
|
/**
|
|
|
|
* @file @internal This file is part of netcdf-4, a netCDF-like
|
|
|
|
* interface for HDF5, or a HDF5 backend for netCDF, depending on your
|
|
|
|
* point of view.
|
|
|
|
*
|
|
|
|
* This file includes the HDF5 code to deal with dimensions.
|
|
|
|
*
|
|
|
|
* @author Ed Hartnett
|
|
|
|
*/
|
|
|
|
|
2018-07-16 18:50:15 +08:00
|
|
|
#include "config.h"
|
2018-07-12 21:05:21 +08:00
|
|
|
#include "hdf5internal.h"
|
2018-05-15 20:51:44 +08:00
|
|
|
|
|
|
|
/**
|
|
|
|
* @internal Dimensions are defined in attributes attached to the
|
|
|
|
* appropriate group in the data file.
|
|
|
|
*
|
|
|
|
* @param ncid File and group ID.
|
|
|
|
* @param name Name of the new dimension.
|
|
|
|
* @param len Length of the new dimension.
|
|
|
|
* @param idp Pointer that gets the ID of the new dimension. Ignored
|
|
|
|
* if NULL.
|
|
|
|
*
|
|
|
|
* @return ::NC_NOERR No error.
|
|
|
|
* @return ::NC_EBADID Bad ncid.
|
|
|
|
* @return ::NC_EMAXNAME Name is too long.
|
|
|
|
* @return ::NC_EBADNAME Name breaks netCDF name rules.
|
|
|
|
* @return ::NC_EINVAL Invalid input.
|
|
|
|
* @return ::NC_EPERM Read-only file.
|
|
|
|
* @return ::NC_EUNLIMIT Only one unlimited dim for classic model.
|
|
|
|
* @return ::NC_ENOTINDEFINE Not in define mode.
|
|
|
|
* @return ::NC_EDIMSIZE Dim length too large.
|
|
|
|
* @return ::NC_ENAMEINUSE Name already in use in group.
|
|
|
|
* @return ::NC_ENOMEM Out of memory.
|
|
|
|
* @author Ed Hartnett
|
|
|
|
*/
|
|
|
|
int
|
2020-08-18 09:15:47 +08:00
|
|
|
HDF5_def_dim(int ncid, const char *name, size_t len, int *idp)
|
2018-05-15 20:51:44 +08:00
|
|
|
{
|
2019-02-19 20:18:02 +08:00
|
|
|
NC *nc;
|
|
|
|
NC_GRP_INFO_T *grp;
|
|
|
|
NC_FILE_INFO_T *h5;
|
|
|
|
NC_DIM_INFO_T *dim;
|
|
|
|
char norm_name[NC_MAX_NAME + 1];
|
|
|
|
int retval = NC_NOERR;
|
|
|
|
int i;
|
|
|
|
|
|
|
|
LOG((2, "%s: ncid 0x%x name %s len %d", __func__, ncid, name,
|
|
|
|
(int)len));
|
|
|
|
|
|
|
|
/* Find our global metadata structure. */
|
|
|
|
if ((retval = nc4_find_nc_grp_h5(ncid, &nc, &grp, &h5)))
|
|
|
|
return retval;
|
|
|
|
assert(h5 && nc && grp);
|
|
|
|
|
|
|
|
/* If the file is read-only, return an error. */
|
|
|
|
if (h5->no_write)
|
|
|
|
return NC_EPERM;
|
|
|
|
|
|
|
|
/* Check some stuff if strict nc3 rules are in effect. */
|
|
|
|
if (h5->cmode & NC_CLASSIC_MODEL)
|
|
|
|
{
|
|
|
|
/* Only one limited dimenson for strict nc3. */
|
|
|
|
if (len == NC_UNLIMITED) {
|
|
|
|
for(i=0;i<ncindexsize(grp->dim);i++) {
|
|
|
|
dim = (NC_DIM_INFO_T*)ncindexith(grp->dim,i);
|
|
|
|
if(dim == NULL) continue;
|
|
|
|
if (dim->unlimited)
|
|
|
|
return NC_EUNLIMIT;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
/* Must be in define mode for stict nc3. */
|
|
|
|
if (!(h5->flags & NC_INDEF))
|
|
|
|
return NC_ENOTINDEFINE;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Make sure this is a valid netcdf name. */
|
|
|
|
if ((retval = nc4_check_name(name, norm_name)))
|
|
|
|
return retval;
|
|
|
|
|
|
|
|
/* For classic model: dim length has to fit in a 32-bit unsigned
|
|
|
|
* int, as permitted for 64-bit offset format. */
|
|
|
|
if (h5->cmode & NC_CLASSIC_MODEL)
|
|
|
|
if(len > X_UINT_MAX) /* Backward compat */
|
|
|
|
return NC_EDIMSIZE;
|
|
|
|
|
|
|
|
/* Make sure the name is not already in use. */
|
|
|
|
dim = (NC_DIM_INFO_T*)ncindexlookup(grp->dim,norm_name);
|
|
|
|
if(dim != NULL)
|
|
|
|
return NC_ENAMEINUSE;
|
|
|
|
|
|
|
|
/* If it's not in define mode, enter define mode. Do this only
|
|
|
|
* after checking all input data, so we only enter define mode if
|
|
|
|
* input is good. */
|
|
|
|
if (!(h5->flags & NC_INDEF))
|
|
|
|
if ((retval = NC4_redef(ncid)))
|
|
|
|
return retval;
|
|
|
|
|
|
|
|
/* Add a dimension to the list. The ID must come from the file
|
|
|
|
* information, since dimids are visible in more than one group. */
|
|
|
|
if ((retval = nc4_dim_list_add(grp, norm_name, len, -1, &dim)))
|
|
|
|
return retval;
|
|
|
|
|
|
|
|
/* Create struct for HDF5-specific dim info. */
|
|
|
|
if (!(dim->format_dim_info = calloc(1, sizeof(NC_HDF5_DIM_INFO_T))))
|
|
|
|
return NC_ENOMEM;
|
|
|
|
|
|
|
|
/* Pass back the dimid. */
|
|
|
|
if (idp)
|
|
|
|
*idp = dim->hdr.id;
|
|
|
|
|
|
|
|
return retval;
|
2018-05-15 20:51:44 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* @internal Find out name and len of a dim. For an unlimited
|
|
|
|
* dimension, the length is the largest length so far written. If the
|
|
|
|
* name of lenp pointers are NULL, they will be ignored.
|
|
|
|
*
|
|
|
|
* @param ncid File and group ID.
|
|
|
|
* @param dimid Dimension ID.
|
|
|
|
* @param name Pointer that gets name of the dimension.
|
|
|
|
* @param lenp Pointer that gets length of dimension.
|
|
|
|
*
|
|
|
|
* @return ::NC_NOERR No error.
|
|
|
|
* @return ::NC_EBADID Bad ncid.
|
|
|
|
* @return ::NC_EDIMSIZE Dimension length too large.
|
|
|
|
* @return ::NC_EBADDIM Dimension not found.
|
|
|
|
* @author Ed Hartnett
|
|
|
|
*/
|
|
|
|
int
|
2020-08-18 09:15:47 +08:00
|
|
|
HDF5_inq_dim(int ncid, int dimid, char *name, size_t *lenp)
|
2018-05-15 20:51:44 +08:00
|
|
|
{
|
2019-02-19 20:18:02 +08:00
|
|
|
NC *nc;
|
|
|
|
NC_FILE_INFO_T *h5;
|
|
|
|
NC_GRP_INFO_T *grp, *dim_grp;
|
|
|
|
NC_DIM_INFO_T *dim;
|
|
|
|
int ret = NC_NOERR;
|
|
|
|
|
|
|
|
LOG((2, "%s: ncid 0x%x dimid %d", __func__, ncid, dimid));
|
|
|
|
|
|
|
|
/* Find our global metadata structure. */
|
|
|
|
if ((ret = nc4_find_nc_grp_h5(ncid, &nc, &grp, &h5)))
|
|
|
|
return ret;
|
|
|
|
assert(h5 && nc && grp);
|
|
|
|
|
|
|
|
/* Find the dimension and its home group. */
|
|
|
|
if ((ret = nc4_find_dim(grp, dimid, &dim, &dim_grp)))
|
|
|
|
return ret;
|
|
|
|
assert(dim);
|
|
|
|
|
|
|
|
/* Return the dimension name, if the caller wants it. */
|
|
|
|
if (name && dim->hdr.name)
|
|
|
|
strcpy(name, dim->hdr.name);
|
|
|
|
|
|
|
|
/* Return the dimension length, if the caller wants it. */
|
|
|
|
if (lenp)
|
|
|
|
{
|
|
|
|
if (dim->unlimited)
|
|
|
|
{
|
2022-04-26 21:54:47 +08:00
|
|
|
*lenp = 0;
|
|
|
|
|
2019-02-19 20:18:02 +08:00
|
|
|
/* Since this is an unlimited dimension, go to the file
|
|
|
|
and see how many records there are. Take the max number
|
|
|
|
of records from all the vars that share this
|
|
|
|
dimension. */
|
2022-04-26 21:54:47 +08:00
|
|
|
if (*lenp == 0)
|
|
|
|
{
|
Proof-of-Concept: Avoid N^2 behavior in NC4_inq_dim
The current library seems to have some behavior which is N^2 in the number of vars in a file.
The `NC4_inq_dim` routine calls down to `nc4_find_dim_len` which iterates through each `var` in the file/group and calls `find_var_dim_max_length` on each var and finds the largest length of the dim on each of those vars. This is done only for unlimited vars.
I have a file with 129 dim and 1630 vars. The unlimited dimension is of length 41. In my test program, I am reading data from 4 files which have the same dim and var count and reading every 4th time step (unlimited dimension). If I run a profile, I see that 98.2% of the program time is in the `nc_get_vara_float` call tree and most of that is in `find_var_dim_max_length` (94.8%).
There are 66,142 calls to `nc_get_vara_float` resulting in 107,307,290 calls to `find_var_dim_max_length` with twice that number of calls to `malloc/free` and calls to 5 HDF5 routines. All of this, at least in my case, to return the same `41` each time.
The proof of concept patch here will check whether the file is read-only (or no_write) and if so, it will cache the value of the dim length the first time it is calculated. With this change, my example run is sped up by a factor of 60. The time for `NC4_inq_dim` and below drops from 97.2% down to 2.7%.
I'm not sure whether this is the correct fix, or if there is some behavior that I am overlooking, but my users would definitely like a 10 second run compared to a 10 minute run...
This is on current Netcdf master branch.
I will try to attach some valgrind/callgrind profiles.
2020-05-01 01:01:10 +08:00
|
|
|
if ((ret = nc4_find_dim_len(dim_grp, dimid, &lenp)))
|
|
|
|
return ret;
|
|
|
|
dim->len = *lenp;
|
|
|
|
}
|
2019-02-19 20:18:02 +08:00
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
if (dim->too_long)
|
|
|
|
{
|
|
|
|
ret = NC_EDIMSIZE;
|
|
|
|
*lenp = NC_MAX_UINT;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
*lenp = dim->len;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return ret;
|
2018-05-15 20:51:44 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* @internal Rename a dimension, for those who like to prevaricate.
|
|
|
|
*
|
2019-02-25 21:36:39 +08:00
|
|
|
* @note If we're not in define mode, new name must be of equal or
|
|
|
|
* less size, if strict nc3 rules are in effect for this file. But we
|
|
|
|
* don't check this because reproducing the exact classic behavior
|
|
|
|
* would be too difficult. See github issue #1340.
|
|
|
|
*
|
2018-05-15 20:51:44 +08:00
|
|
|
* @param ncid File and group ID.
|
|
|
|
* @param dimid Dimension ID.
|
|
|
|
* @param name New dimension name.
|
|
|
|
*
|
|
|
|
* @return ::NC_NOERR No error.
|
|
|
|
* @return ::NC_EBADID Bad ncid.
|
|
|
|
* @return ::NC_EHDFERR HDF5 returned error.
|
|
|
|
* @return ::NC_ENOMEM Out of memory.
|
|
|
|
* @return ::NC_EINVAL Name must be provided.
|
|
|
|
* @return ::NC_ENAMEINUSE Name is already in use in group.
|
|
|
|
* @return ::NC_EMAXNAME Name is too long.
|
|
|
|
* @return ::NC_EBADDIM Dimension not found.
|
|
|
|
* @return ::NC_EBADNAME Name breaks netCDF name rules.
|
|
|
|
* @return ::NC_EDIMMETA Unable to delete HDF5 dataset.
|
|
|
|
* @author Ed Hartnett
|
|
|
|
*/
|
|
|
|
int
|
2020-08-18 09:15:47 +08:00
|
|
|
HDF5_rename_dim(int ncid, int dimid, const char *name)
|
2018-05-15 20:51:44 +08:00
|
|
|
{
|
2019-02-19 20:18:02 +08:00
|
|
|
NC_GRP_INFO_T *grp;
|
|
|
|
NC_DIM_INFO_T *dim;
|
|
|
|
NC_HDF5_DIM_INFO_T *hdf5_dim;
|
|
|
|
NC_FILE_INFO_T *h5;
|
|
|
|
char norm_name[NC_MAX_NAME + 1];
|
|
|
|
int retval;
|
|
|
|
|
|
|
|
/* Note: name is new name */
|
|
|
|
if (!name)
|
|
|
|
return NC_EINVAL;
|
|
|
|
|
|
|
|
LOG((2, "%s: ncid 0x%x dimid %d name %s", __func__, ncid,
|
|
|
|
dimid, name));
|
|
|
|
|
|
|
|
/* Find info for this file and group, and set pointer to each. */
|
|
|
|
if ((retval = nc4_find_grp_h5(ncid, &grp, &h5)))
|
|
|
|
return retval;
|
|
|
|
assert(h5 && grp);
|
|
|
|
|
|
|
|
/* Trying to write to a read-only file? No way, Jose! */
|
|
|
|
if (h5->no_write)
|
|
|
|
return NC_EPERM;
|
|
|
|
|
|
|
|
/* Make sure this is a valid netcdf name. */
|
|
|
|
if ((retval = nc4_check_name(name, norm_name)))
|
|
|
|
return retval;
|
|
|
|
|
|
|
|
/* Get the original dim. */
|
|
|
|
if ((retval = nc4_find_dim(grp, dimid, &dim, NULL)))
|
|
|
|
return retval;
|
|
|
|
assert(dim && dim->format_dim_info);
|
|
|
|
hdf5_dim = (NC_HDF5_DIM_INFO_T *)dim->format_dim_info;
|
|
|
|
|
|
|
|
/* Check if new name is in use. */
|
|
|
|
if (ncindexlookup(grp->dim, norm_name))
|
|
|
|
return NC_ENAMEINUSE;
|
|
|
|
|
|
|
|
/* Check for renaming dimension w/o variable. */
|
|
|
|
if (hdf5_dim->hdf_dimscaleid)
|
|
|
|
{
|
|
|
|
assert(!dim->coord_var);
|
|
|
|
LOG((3, "dim %s is a dim without variable", dim->hdr.name));
|
|
|
|
|
|
|
|
/* Delete the dimscale-only dataset. */
|
|
|
|
if ((retval = delete_dimscale_dataset(grp, dimid, dim)))
|
|
|
|
return retval;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Give the dimension its new name in metadata. UTF8 normalization
|
|
|
|
* has been done. */
|
|
|
|
assert(dim->hdr.name);
|
|
|
|
free(dim->hdr.name);
|
|
|
|
if (!(dim->hdr.name = strdup(norm_name)))
|
|
|
|
return NC_ENOMEM;
|
|
|
|
LOG((3, "dim is now named %s", dim->hdr.name));
|
|
|
|
|
2020-11-20 08:01:04 +08:00
|
|
|
/* rebuild index. */
|
2019-02-19 20:18:02 +08:00
|
|
|
if (!ncindexrebuild(grp->dim))
|
|
|
|
return NC_EINTERNAL;
|
|
|
|
|
|
|
|
/* Check if dimension was a coordinate variable, but names are
|
|
|
|
* different now. */
|
|
|
|
if (dim->coord_var && strcmp(dim->hdr.name, dim->coord_var->hdr.name))
|
|
|
|
{
|
|
|
|
/* Break up the coordinate variable. */
|
|
|
|
if ((retval = nc4_break_coord_var(grp, dim->coord_var, dim)))
|
2018-05-15 20:51:44 +08:00
|
|
|
return retval;
|
2019-02-19 20:18:02 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
/* Check if dimension should become a coordinate variable. */
|
|
|
|
if (!dim->coord_var)
|
|
|
|
{
|
|
|
|
NC_VAR_INFO_T *var;
|
|
|
|
|
|
|
|
/* Attempt to find a variable with the same name as the
|
|
|
|
* dimension in the current group. */
|
|
|
|
if ((retval = nc4_find_var(grp, dim->hdr.name, &var)))
|
|
|
|
return retval;
|
|
|
|
|
|
|
|
/* Check if we found a variable and the variable has the
|
|
|
|
* dimension in index 0. */
|
|
|
|
if (var && var->dim[0] == dim)
|
|
|
|
{
|
|
|
|
/* Sanity check */
|
|
|
|
assert(var->dimids[0] == dim->hdr.id);
|
|
|
|
|
|
|
|
/* Reform the coordinate variable. */
|
|
|
|
if ((retval = nc4_reform_coord_var(grp, var, dim)))
|
|
|
|
return retval;
|
|
|
|
}
|
|
|
|
}
|
2018-05-15 20:51:44 +08:00
|
|
|
|
2019-02-19 20:18:02 +08:00
|
|
|
return NC_NOERR;
|
2018-05-15 20:51:44 +08:00
|
|
|
}
|