mirror of
https://github.com/HDFGroup/hdf5.git
synced 2024-12-03 02:32:04 +08:00
144 lines
4.9 KiB
Fortran
144 lines
4.9 KiB
Fortran
! * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *
|
|
! Copyright by The HDF Group. *
|
|
! Copyright by the Board of Trustees of the University of Illinois. *
|
|
! All rights reserved. *
|
|
! *
|
|
! This file is part of HDF5. The full HDF5 copyright notice, including *
|
|
! terms governing use, modification, and redistribution, is contained in *
|
|
! the files COPYING and Copyright.html. COPYING can be found at the root *
|
|
! of the source code distribution tree; Copyright.html can be found at the *
|
|
! root level of an installed copy of the electronic HDF5 document set and *
|
|
! is linked from the top-level documents page. It can also be found at *
|
|
! http://hdfgroup.org/HDF5/doc/Copyright.html. If you do not have *
|
|
! access to either file, you may request a copy from help@hdfgroup.org. *
|
|
! * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *
|
|
!
|
|
! Fortran parallel example. Copied from Tutorial's example program of
|
|
! dataset.f90.
|
|
|
|
PROGRAM DATASET
|
|
|
|
USE HDF5 ! This module contains all necessary modules
|
|
|
|
IMPLICIT NONE
|
|
|
|
INCLUDE 'mpif.h'
|
|
CHARACTER(LEN=10), PARAMETER :: default_fname = "sds.h5" ! Default name
|
|
CHARACTER(LEN=8), PARAMETER :: dsetname = "IntArray" ! Dataset name
|
|
|
|
CHARACTER(LEN=100) :: filename ! File name
|
|
INTEGER :: fnamelen ! File name length
|
|
INTEGER(HID_T) :: file_id ! File identifier
|
|
INTEGER(HID_T) :: dset_id ! Dataset identifier
|
|
INTEGER(HID_T) :: filespace ! Dataspace identifier in file
|
|
INTEGER(HID_T) :: plist_id ! Property list identifier
|
|
|
|
INTEGER(HSIZE_T), DIMENSION(2) :: dimsf = (/5,8/) ! Dataset dimensions.
|
|
! INTEGER, DIMENSION(7) :: dimsfi = (/5,8,0,0,0,0,0/)
|
|
! INTEGER(HSIZE_T), DIMENSION(2) :: dimsfi = (/5,8/)
|
|
INTEGER(HSIZE_T), DIMENSION(2) :: dimsfi
|
|
|
|
INTEGER, ALLOCATABLE :: data(:,:) ! Data to write
|
|
INTEGER :: rank = 2 ! Dataset rank
|
|
|
|
INTEGER :: error, error_n ! Error flags
|
|
INTEGER :: i, j
|
|
!
|
|
! MPI definitions and calls.
|
|
!
|
|
INTEGER :: mpierror ! MPI error flag
|
|
INTEGER :: comm, info
|
|
INTEGER :: mpi_size, mpi_rank
|
|
comm = MPI_COMM_WORLD
|
|
info = MPI_INFO_NULL
|
|
CALL MPI_INIT(mpierror)
|
|
CALL MPI_COMM_SIZE(comm, mpi_size, mpierror)
|
|
CALL MPI_COMM_RANK(comm, mpi_rank, mpierror)
|
|
!
|
|
! Initialize data buffer with trivial data.
|
|
!
|
|
ALLOCATE ( data(dimsf(1),dimsf(2)))
|
|
do i = 1, dimsf(2)
|
|
do j = 1, dimsf(1)
|
|
data(j,i) = j - 1 + (i-1)*dimsf(1)
|
|
enddo
|
|
enddo
|
|
!
|
|
! Initialize FORTRAN interface
|
|
!
|
|
CALL h5open_f(error)
|
|
|
|
!
|
|
! Setup file access property list with parallel I/O access.
|
|
!
|
|
CALL h5pcreate_f(H5P_FILE_ACCESS_F, plist_id, error)
|
|
CALL h5pset_fapl_mpio_f(plist_id, comm, info, error)
|
|
|
|
!
|
|
! Figure out the filename to use. If your system does not support
|
|
! getenv, comment that statement with this,
|
|
! filename = ""
|
|
CALL getenv("HDF5_PARAPREFIX", filename)
|
|
fnamelen = LEN_TRIM(filename)
|
|
if ( fnamelen == 0 ) then
|
|
filename = default_fname
|
|
else
|
|
filename = filename(1:fnamelen) // "/" // default_fname
|
|
endif
|
|
print *, "Using filename = ", filename
|
|
|
|
!
|
|
! Create the file collectively.
|
|
!
|
|
CALL h5fcreate_f(filename, H5F_ACC_TRUNC_F, file_id, error, access_prp = plist_id)
|
|
CALL h5pclose_f(plist_id, error)
|
|
!
|
|
! Create the data space for the dataset.
|
|
!
|
|
CALL h5screate_simple_f(rank, dimsf, filespace, error)
|
|
|
|
!
|
|
! Create the dataset with default properties.
|
|
!
|
|
CALL h5dcreate_f(file_id, dsetname, H5T_NATIVE_INTEGER, filespace, &
|
|
dset_id, error)
|
|
!
|
|
! Create property list for collective dataset write
|
|
!
|
|
CALL h5pcreate_f(H5P_DATASET_XFER_F, plist_id, error)
|
|
CALL h5pset_dxpl_mpio_f(plist_id, H5FD_MPIO_COLLECTIVE_F, error)
|
|
!
|
|
! For independent write use
|
|
! CALL h5pset_dxpl_mpio_f(plist_id, H5FD_MPIO_INDEPENDENT_F, error)
|
|
!
|
|
|
|
!
|
|
! Write the dataset collectively.
|
|
!
|
|
CALL h5dwrite_f(dset_id, H5T_NATIVE_INTEGER, data, dimsfi, error, &
|
|
xfer_prp = plist_id)
|
|
!
|
|
! Deallocate data buffer.
|
|
!
|
|
DEALLOCATE(data)
|
|
|
|
!
|
|
! Close resources.
|
|
!
|
|
CALL h5sclose_f(filespace, error)
|
|
CALL h5dclose_f(dset_id, error)
|
|
CALL h5pclose_f(plist_id, error)
|
|
CALL h5fclose_f(file_id, error)
|
|
! Attempt to remove the data file. Remove the line if the compiler
|
|
! does not support it.
|
|
!CALL unlink(filename)
|
|
|
|
!
|
|
! Close FORTRAN interface
|
|
!
|
|
CALL h5close_f(error)
|
|
|
|
CALL MPI_FINALIZE(mpierror)
|
|
|
|
END PROGRAM DATASET
|