Marc Howison
75 lines
1.8 KiB
Fortran
75 lines
1.8 KiB
Fortran
program H5PartTest
|
|
implicit none
|
|
|
|
#ifdef PARALLEL_IO
|
|
include 'mpif.h'
|
|
#endif
|
|
include 'H5PartF.h'
|
|
|
|
#ifdef PARALLEL_IO
|
|
integer :: comm, ierr, i
|
|
integer*8 :: file_id, status, npoints
|
|
#endif
|
|
real*8, allocatable :: x(:),y(:),z(:),px(:),py(:),pz(:)
|
|
integer*8, allocatable :: id(:)
|
|
|
|
#ifdef PARALLEL_IO
|
|
call MPI_INIT(ierr)
|
|
comm = MPI_COMM_WORLD
|
|
#endif
|
|
|
|
! this enables level 3 ("info") messages to be
|
|
! printed by the H5Part library
|
|
! (3_8 is the literal for an integer*8 with value 3)
|
|
status = h5pt_set_verbosity_level (3_8)
|
|
|
|
! open the a file called 'test.h5' in parallel for writing
|
|
#ifdef PARALLEL_IO
|
|
file_id = h5pt_openw_par ('test.h5', comm)
|
|
#else
|
|
file_id = h5pt_openw ('test.h5')
|
|
#endif
|
|
|
|
! 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.')
|
|
|
|
! set the size of the data array
|
|
npoints = 99
|
|
status = h5pt_setnpoints (file_id, npoints)
|
|
|
|
! create fake data
|
|
allocate( x(npoints), y(npoints), z(npoints) )
|
|
allocate( px(npoints), py(npoints), pz(npoints) )
|
|
allocate( id(npoints) )
|
|
do i=1,int(npoints)
|
|
x(i) = 0.0 + real(i)
|
|
y(i) = 0.1 + real(i)
|
|
z(i) = 0.2 + real(i)
|
|
px(i) = 0.3 + real(i)
|
|
py(i) = 0.4 + real(i)
|
|
pz(i) = 0.5 + real(i)
|
|
id(i) = i
|
|
enddo
|
|
|
|
! write the data
|
|
status = h5pt_writedata_r8 (file_id, "x", x)
|
|
status = h5pt_writedata_r8 (file_id, "y", y)
|
|
status = h5pt_writedata_r8 (file_id, "z", z)
|
|
status = h5pt_writedata_r8 (file_id, "px", px)
|
|
status = h5pt_writedata_r8 (file_id, "py", py)
|
|
status = h5pt_writedata_r8 (file_id, "pz", pz)
|
|
status = h5pt_writedata_i8 (file_id, "id", id)
|
|
|
|
! close the file
|
|
status = h5pt_close (file_id)
|
|
|
|
#ifdef PARALLEL_IO
|
|
call MPI_FINALIZE(ierr)
|
|
#endif
|
|
|
|
end program H5PartTest
|
|
|