mirror of
https://github.com/Unidata/netcdf-c.git
synced 2024-12-15 08:30:11 +08:00
100 lines
2.5 KiB
C
100 lines
2.5 KiB
C
#include <stdlib.h>
|
|
#include <stdio.h>
|
|
#include <unistd.h>
|
|
#include <sys/resource.h>
|
|
#if defined(__APPLE__) && defined(__MACH__)
|
|
#include <mach/mach.h>
|
|
#endif
|
|
|
|
#include "netcdf.h"
|
|
|
|
#define filename "file://tmp_cachetest.zarr#mode=zarr,file"
|
|
#define chunkSize ((size_t)(1<<17)) /* 128k */
|
|
#define numCells ((size_t)(50 * chunkSize))
|
|
#define numSteps ((size_t)360)
|
|
|
|
static float var[numCells];
|
|
|
|
size_t getPeakRSS(void)
|
|
{
|
|
struct rusage rusage;
|
|
getrusage( RUSAGE_SELF, &rusage );
|
|
#if defined(__APPLE__) && defined(__MACH__)
|
|
return (size_t)rusage.ru_maxrss;
|
|
#else
|
|
return (size_t)(rusage.ru_maxrss * 1024L);
|
|
#endif
|
|
}
|
|
|
|
static void
|
|
nce(int istat)
|
|
{
|
|
if (istat != NC_NOERR)
|
|
{
|
|
fprintf(stderr, "%s\n", nc_strerror(istat));
|
|
exit(-1);
|
|
}
|
|
}
|
|
|
|
int
|
|
main(void)
|
|
{
|
|
printf("write: chunkSize=%zu, numCells=%zu, numSteps=%zu, filename=%s\n", chunkSize, numCells, numSteps, filename);
|
|
|
|
int ncId;
|
|
nce(nc_create(filename, NC_CLOBBER | NC_NETCDF4, &ncId));
|
|
int oldfill;
|
|
nce(nc_set_fill(ncId, NC_NOFILL, &oldfill));
|
|
|
|
int dims[2];
|
|
nce(nc_def_dim(ncId, "time", numSteps, &dims[0]));
|
|
nce(nc_def_dim(ncId, "cells", numCells, &dims[1]));
|
|
|
|
int varId;
|
|
nce(nc_def_var(ncId, "var", NC_FLOAT, 2, dims, &varId));
|
|
|
|
size_t chunks[2] = {1, chunkSize};
|
|
nc_def_var_chunking(ncId, varId, NC_CHUNKED, chunks);
|
|
|
|
int shuffle = 0, deflate = 1, level = 3;
|
|
nce(nc_def_var_deflate(ncId, varId, shuffle, deflate, level));
|
|
|
|
size_t size, nelems;
|
|
float preemption;
|
|
nce(nc_get_var_chunk_cache(ncId, varId, &size, &nelems, &preemption));
|
|
printf("default chunk cache: size=%zu, nelems=%zu, preemption=%g\n", size, nelems, preemption);
|
|
size = 4 * numCells; // one float field at one time step
|
|
nelems = 1000;
|
|
preemption = 0.5;
|
|
nce(nc_set_var_chunk_cache(ncId, varId, size, nelems, preemption));
|
|
printf("set chunk cache: size=%zu, nelems=%zu, preemption=%g\n", size, nelems, preemption);
|
|
|
|
nce(nc_enddef(ncId));
|
|
|
|
{
|
|
for (size_t i = 0; i < numCells; ++i) var[i] = 0.0f;
|
|
for (size_t i = 0; i < numSteps; ++i)
|
|
{
|
|
size_t start[2], count[2];
|
|
start[0] = i; start[1] = 0;
|
|
count[0] = 1; count[1] = numCells;
|
|
nce(nc_put_vara_float(ncId, varId, start, count, var));
|
|
}
|
|
}
|
|
|
|
nce(nc_close(ncId));
|
|
|
|
{
|
|
size_t mbused = getPeakRSS() / (1024 * 1024);
|
|
printf("Max mem: %zu MB\n", mbused);
|
|
if(mbused > 128) {
|
|
printf("*** Failed: used: %luMB expected: < 128\n",mbused);
|
|
return (1);
|
|
} else {
|
|
printf("*** Passed: used: %luMB (< 128)\n",mbused);
|
|
return 0;
|
|
}
|
|
}
|
|
|
|
}
|