mirror of
https://github.com/Unidata/netcdf-c.git
synced 2025-02-17 16:50:18 +08:00
148 lines
5.7 KiB
C
148 lines
5.7 KiB
C
/* This is part of the netCDF package.
|
|
Copyright 2021 University Corporation for Atmospheric Research/Unidata
|
|
See COPYRIGHT file for conditions of use.
|
|
|
|
Test quantization of netcdf-4 variables. Quantization is the
|
|
zeroing-out of bits in float or double data beyond a desired
|
|
precision.
|
|
|
|
Ed Hartnett, 8/19/21
|
|
*/
|
|
|
|
#include <nc_tests.h>
|
|
#include "err_macros.h"
|
|
#include "netcdf.h"
|
|
|
|
#define FILE_NAME "tst_quantize.nc"
|
|
#define NDIMS1 1
|
|
#define DIM_NAME_1 "meters_along_canal"
|
|
#define DIM_LEN_1 3
|
|
#define VAR_NAME_1 "Amsterdam_houseboat_location"
|
|
#define VAR_NAME_2 "Amsterdam_street_noise_decibels"
|
|
#define NSD_1 3
|
|
#define NSD_2 9
|
|
|
|
int
|
|
main(int argc, char **argv)
|
|
{
|
|
printf("\n*** Testing netcdf-4 variable quantization functions.\n");
|
|
printf("**** testing quantization setting and error conditions...");
|
|
{
|
|
int ncid, dimid, varid1, varid2;
|
|
int quantize_mode_in, nsd_in;
|
|
|
|
/* Create a netcdf classic file with one var. Attempt
|
|
* quantization. It will not work. */
|
|
if (nc_create(FILE_NAME, NC_CLOBBER, &ncid)) ERR;
|
|
if (nc_def_dim(ncid, DIM_NAME_1, DIM_LEN_1, &dimid)) ERR;
|
|
if (nc_def_var(ncid, VAR_NAME_1, NC_FLOAT, NDIMS1, &dimid, &varid1)) ERR;
|
|
if (nc_def_var_quantize(ncid, varid1, NC_QUANTIZE_BITGROOM, NSD_1) != NC_ENOTNC4) ERR;
|
|
if (nc_inq_var_quantize(ncid, varid1, &quantize_mode_in, &nsd_in) != NC_ENOTNC4) ERR;
|
|
if (nc_close(ncid)) ERR;
|
|
|
|
/* Create a netcdf-4 file with two vars. Attempt
|
|
* quantization. It will work, eventually... */
|
|
if (nc_create(FILE_NAME, NC_NETCDF4|NC_CLOBBER, &ncid)) ERR;
|
|
if (nc_def_dim(ncid, DIM_NAME_1, DIM_LEN_1, &dimid)) ERR;
|
|
if (nc_def_var(ncid, VAR_NAME_1, NC_FLOAT, NDIMS1, &dimid, &varid1)) ERR;
|
|
if (nc_def_var(ncid, VAR_NAME_2, NC_DOUBLE, NDIMS1, &dimid, &varid2)) ERR;
|
|
|
|
/* Bad varid. */
|
|
if (nc_def_var_quantize(ncid, NC_GLOBAL, NC_QUANTIZE_BITGROOM, NSD_1) != NC_EGLOBAL) ERR;
|
|
if (nc_def_var_quantize(ncid, varid2 + 1, NC_QUANTIZE_BITGROOM, NSD_1) != NC_ENOTVAR) ERR;
|
|
/* Invalid values. */
|
|
if (nc_def_var_quantize(ncid, varid1, NC_QUANTIZE_BITGROOM + 1, NSD_1) != NC_EINVAL) ERR;
|
|
if (nc_def_var_quantize(ncid, varid1, NC_QUANTIZE_BITGROOM, -1) != NC_EINVAL) ERR;
|
|
if (nc_def_var_quantize(ncid, varid1, NC_QUANTIZE_BITGROOM, NC_QUANTIZE_MAX_FLOAT_NSD + 1) != NC_EINVAL) ERR;
|
|
if (nc_def_var_quantize(ncid, varid2, NC_QUANTIZE_BITGROOM + 1, 3) != NC_EINVAL) ERR;
|
|
if (nc_def_var_quantize(ncid, varid2, NC_QUANTIZE_BITGROOM, -1) != NC_EINVAL) ERR;
|
|
if (nc_def_var_quantize(ncid, varid2, NC_QUANTIZE_BITGROOM, NC_QUANTIZE_MAX_DOUBLE_NSD + 1) != NC_EINVAL) ERR;
|
|
|
|
/* This will work. */
|
|
if (nc_def_var_quantize(ncid, varid1, NC_QUANTIZE_BITGROOM, NSD_1)) ERR;
|
|
if (nc_inq_var_quantize(ncid, varid1, &quantize_mode_in, &nsd_in)) ERR;
|
|
if (quantize_mode_in != NC_QUANTIZE_BITGROOM) ERR;
|
|
if (nsd_in != NSD_1) ERR;
|
|
|
|
/* Wait, I changed my mind! Let's turn off quantization. */
|
|
if (nc_def_var_quantize(ncid, varid1, NC_NOQUANTIZE, 0)) ERR;
|
|
if (nc_inq_var_quantize(ncid, varid1, &quantize_mode_in, &nsd_in)) ERR;
|
|
if (quantize_mode_in != NC_NOQUANTIZE) ERR;
|
|
if (nsd_in != 0) ERR;
|
|
|
|
/* Changed my mind again, turn it on. */
|
|
if (nc_def_var_quantize(ncid, varid1, NC_QUANTIZE_BITGROOM, NSD_1)) ERR;
|
|
|
|
/* I changed my mind again! Turn it off! */
|
|
if (nc_def_var_quantize(ncid, varid1, NC_QUANTIZE_BITGROOM, 0)) ERR;
|
|
if (nc_inq_var_quantize(ncid, varid1, &quantize_mode_in, &nsd_in)) ERR;
|
|
if (quantize_mode_in != NC_NOQUANTIZE) ERR;
|
|
if (nsd_in != 0) ERR;
|
|
|
|
/* Changed my mind again, turn it on. */
|
|
if (nc_def_var_quantize(ncid, varid1, NC_QUANTIZE_BITGROOM, NSD_1)) ERR;
|
|
|
|
/* This also will work for double. */
|
|
if (nc_def_var_quantize(ncid, varid2, NC_QUANTIZE_BITGROOM, NSD_2)) ERR;
|
|
if (nc_inq_var_quantize(ncid, varid2, &quantize_mode_in, &nsd_in)) ERR;
|
|
if (quantize_mode_in != NC_QUANTIZE_BITGROOM) ERR;
|
|
if (nsd_in != NSD_2) ERR;
|
|
|
|
/* End define mode. */
|
|
if (nc_enddef(ncid)) ERR;
|
|
|
|
/* This will not work, it's too late! */
|
|
if (nc_def_var_quantize(ncid, varid1, NC_QUANTIZE_BITGROOM, NSD_1) != NC_ELATEDEF) ERR;
|
|
|
|
/* Close the file. */
|
|
if (nc_close(ncid)) ERR;
|
|
|
|
/* Open the file and check. */
|
|
if (nc_open(FILE_NAME, NC_WRITE, &ncid)) ERR;
|
|
if (nc_inq_var_quantize(ncid, 0, &quantize_mode_in, &nsd_in)) ERR;
|
|
if (quantize_mode_in != NC_QUANTIZE_BITGROOM) ERR;
|
|
if (nsd_in != NSD_1) ERR;
|
|
if (nc_inq_var_quantize(ncid, 1, &quantize_mode_in, &nsd_in)) ERR;
|
|
if (quantize_mode_in != NC_QUANTIZE_BITGROOM) ERR;
|
|
if (nsd_in != NSD_2) ERR;
|
|
if (nc_close(ncid)) ERR;
|
|
}
|
|
SUMMARIZE_ERR;
|
|
|
|
printf("**** testing quantization values...");
|
|
{
|
|
int ncid, dimid, varid1, varid2;
|
|
int quantize_mode_in, nsd_in;
|
|
float float_data[DIM_LEN_1] = {1.0, 1.11111, 1.1111111};
|
|
double double_data[DIM_LEN_1] = {1.0, 1.111111111, 1.111111111111};
|
|
|
|
/* Create a netcdf-4 file with two vars. Attempt
|
|
* quantization. */
|
|
if (nc_create(FILE_NAME, NC_NETCDF4|NC_CLOBBER, &ncid)) ERR;
|
|
if (nc_def_dim(ncid, DIM_NAME_1, DIM_LEN_1, &dimid)) ERR;
|
|
if (nc_def_var(ncid, VAR_NAME_1, NC_FLOAT, NDIMS1, &dimid, &varid1)) ERR;
|
|
if (nc_def_var(ncid, VAR_NAME_2, NC_DOUBLE, NDIMS1, &dimid, &varid2)) ERR;
|
|
|
|
/* This will work. */
|
|
if (nc_def_var_quantize(ncid, varid1, NC_QUANTIZE_BITGROOM, NSD_1)) ERR;
|
|
if (nc_def_var_quantize(ncid, varid2, NC_QUANTIZE_BITGROOM, NSD_2)) ERR;
|
|
|
|
/* Write some data. */
|
|
if (nc_put_var_float(ncid, varid1, float_data)) ERR;
|
|
if (nc_put_var_double(ncid, varid2, double_data)) ERR;
|
|
|
|
/* Close the file. */
|
|
if (nc_close(ncid)) ERR;
|
|
|
|
/* Open the file and check. */
|
|
if (nc_open(FILE_NAME, NC_WRITE, &ncid)) ERR;
|
|
if (nc_inq_var_quantize(ncid, 0, &quantize_mode_in, &nsd_in)) ERR;
|
|
if (quantize_mode_in != NC_QUANTIZE_BITGROOM || nsd_in != NSD_1) ERR;
|
|
if (nc_inq_var_quantize(ncid, 1, &quantize_mode_in, &nsd_in)) ERR;
|
|
if (quantize_mode_in != NC_QUANTIZE_BITGROOM || nsd_in != NSD_2) ERR;
|
|
if (nc_close(ncid)) ERR;
|
|
}
|
|
SUMMARIZE_ERR;
|
|
FINAL_RESULTS;
|
|
}
|