46 lines
1.2 KiB
C
46 lines
1.2 KiB
C
#include <stdlib.h>
|
|
#include <assert.h>
|
|
#include <mpi.h>
|
|
#include <H5.h>
|
|
#include <H5Part.h>
|
|
|
|
#define DATASIZE 32
|
|
|
|
int
|
|
main (
|
|
int argc,
|
|
char** argv
|
|
) {
|
|
int rank, nprocs;
|
|
h5_float64_t x[DATASIZE];
|
|
h5_float64_t y[DATASIZE];
|
|
h5_float64_t z[DATASIZE];
|
|
h5_float64_t px[DATASIZE];
|
|
h5_float64_t py[DATASIZE];
|
|
h5_float64_t pz[DATASIZE];
|
|
h5_int64_t nparticles = DATASIZE;
|
|
h5_file_t file;
|
|
|
|
// initialize MPI
|
|
MPI_Init (&argc, &argv);
|
|
MPI_Comm_rank (MPI_COMM_WORLD, &rank);
|
|
MPI_Comm_size (MPI_COMM_WORLD, &nprocs);
|
|
|
|
H5SetVerbosityLevel(4);
|
|
|
|
file = H5OpenFile("particles.h5", H5_O_WRONLY, MPI_COMM_WORLD);
|
|
H5SetStep(file, 0);
|
|
H5PartSetNumParticles(file, nparticles);
|
|
H5PartWriteDataFloat64(file, "x", x);
|
|
H5PartWriteDataFloat64(file, "y", y);
|
|
H5PartWriteDataFloat64(file, "z", z);
|
|
H5PartWriteDataFloat64(file, "px", px);
|
|
H5PartWriteDataFloat64(file, "py", py);
|
|
H5PartWriteDataFloat64(file, "pz", pz);
|
|
H5CloseFile(file);
|
|
|
|
MPI_Finalize();
|
|
return H5_SUCCESS;
|
|
}
|
|
|