Marc Howison
62 lines
1.7 KiB
Fortran
62 lines
1.7 KiB
Fortran
program H5PartTest
|
|
implicit none
|
|
|
|
include 'mpif.h'
|
|
include 'H5PartF.h'
|
|
|
|
integer :: comm, rank, ierr
|
|
integer*8 :: file_id, status, npoints, i
|
|
real*8, allocatable :: particles(:)
|
|
integer*8, allocatable :: id(:)
|
|
|
|
comm = MPI_COMM_WORLD
|
|
call mpi_init(ierr)
|
|
call mpi_comm_rank(comm, rank, ierr)
|
|
|
|
! open the a file for parallel writing
|
|
file_id = h5pt_openw_par('test.h5', comm)
|
|
|
|
! in the Fortran API, steps start at 1
|
|
status = h5pt_setstep(file_id, 1_8)
|
|
|
|
! write an attribute to the file
|
|
status = h5pt_writefileattrib_string(file_id, 'desc', 'This is a test.')
|
|
|
|
! create fake data
|
|
npoints = 99
|
|
allocate(particles(6*npoints), id(npoints))
|
|
do i=0,npoints-1
|
|
particles(6*i+1) = 0.0 + real(i+npoints*rank)
|
|
particles(6*i+2) = 0.1 + real(i+npoints*rank)
|
|
particles(6*i+3) = 0.2 + real(i+npoints*rank)
|
|
particles(6*i+4) = 0.3 + real(i+npoints*rank)
|
|
particles(6*i+5) = 0.4 + real(i+npoints*rank)
|
|
particles(6*i+6) = 0.5 + real(i+npoints*rank)
|
|
id(i+1) = i+npoints*rank
|
|
enddo
|
|
|
|
! set the striding to 6
|
|
status = h5pt_setnpoints_strided(file_id, npoints, 6_8)
|
|
|
|
! write the particles
|
|
status = h5pt_writedata_r8(file_id, "x", particles(1))
|
|
status = h5pt_writedata_r8(file_id, "y", particles(2))
|
|
status = h5pt_writedata_r8(file_id, "z", particles(3))
|
|
status = h5pt_writedata_r8(file_id, "px", particles(4))
|
|
status = h5pt_writedata_r8(file_id, "py", particles(5))
|
|
status = h5pt_writedata_r8(file_id, "pz", particles(6))
|
|
|
|
! disable the striding to write the ids
|
|
status = h5pt_setnpoints(file_id, npoints)
|
|
status = h5pt_writedata_i8(file_id, "id", id)
|
|
|
|
! close the file
|
|
status = h5pt_close(file_id)
|
|
|
|
deallocate(particles, id)
|
|
|
|
call mpi_finalize(ierr)
|
|
|
|
end program H5PartTest
|
|
|