75 lines
2.2 KiB
C
75 lines
2.2 KiB
C
/*
|
|
Copyright (c) 2006-2015, The Regents of the University of California,
|
|
through Lawrence Berkeley National Laboratory (subject to receipt of any
|
|
required approvals from the U.S. Dept. of Energy) and the Paul Scherrer
|
|
Institut (Switzerland). All rights reserved.
|
|
|
|
License: see file COPYING in top level of source distribution.
|
|
*/
|
|
|
|
#include "H5hut.h"
|
|
|
|
// name of output file
|
|
const char* fname = "example_setview.h5";
|
|
|
|
// H5hut verbosity level
|
|
const h5_int64_t h5_verbosity = H5_VERBOSE_DEFAULT;
|
|
|
|
// we are going to write multiple consecutive blocks
|
|
const h5_int64_t num_blocks = 4;
|
|
const h5_int64_t num_particles_per_block = 32;
|
|
|
|
int
|
|
main (
|
|
int argc,
|
|
char* argv[]
|
|
){
|
|
|
|
// initialize MPI & H5hut
|
|
MPI_Init (&argc, &argv);
|
|
MPI_Comm comm = MPI_COMM_WORLD;
|
|
int comm_size = 1;
|
|
MPI_Comm_size (comm, &comm_size);
|
|
int comm_rank = 0;
|
|
MPI_Comm_rank (comm, &comm_rank);
|
|
H5AbortOnError ();
|
|
H5SetVerbosityLevel (h5_verbosity);
|
|
|
|
// open file and create first step
|
|
h5_file_t file = H5OpenFile (fname, H5_O_WRONLY, H5_PROP_DEFAULT);
|
|
H5SetStep (file, 0);
|
|
|
|
/*
|
|
If we want to write consecutive blocks, the 'view' can be defined
|
|
with H5PartSetview(). Otherwise we have to define the total number
|
|
of particles with H5PartSetNumParticles().
|
|
*/
|
|
const h5_int64_t offset = comm_rank * num_blocks * num_particles_per_block;
|
|
H5PartSetView (
|
|
file,
|
|
offset,
|
|
offset + num_blocks*num_particles_per_block -1);
|
|
|
|
// write multiple consecutive blocks
|
|
for (int i = 0; i < num_blocks; i++) {
|
|
// create fake data
|
|
h5_int32_t data[num_particles_per_block];
|
|
for (int j = 0; j < num_particles_per_block; j++) {
|
|
data[j] = j + i*num_particles_per_block + offset;
|
|
}
|
|
|
|
// set the "view" to select a subset of the dataset
|
|
H5PartSetView (
|
|
file,
|
|
offset + i*num_particles_per_block,
|
|
offset + (i+1)*num_particles_per_block - 1);
|
|
// write data
|
|
H5PartWriteDataInt32 (file, "data", data);
|
|
}
|
|
|
|
// done
|
|
H5CloseFile(file);
|
|
MPI_Finalize();
|
|
return 0;
|
|
}
|