diff options
Diffstat (limited to 'HDF5Examples/FORTRAN/H5PAR/ph5_f90_hyperslab_by_chunk.F90')
-rw-r--r-- | HDF5Examples/FORTRAN/H5PAR/ph5_f90_hyperslab_by_chunk.F90 | 170 |
1 files changed, 170 insertions, 0 deletions
diff --git a/HDF5Examples/FORTRAN/H5PAR/ph5_f90_hyperslab_by_chunk.F90 b/HDF5Examples/FORTRAN/H5PAR/ph5_f90_hyperslab_by_chunk.F90 new file mode 100644 index 0000000..3a707af --- /dev/null +++ b/HDF5Examples/FORTRAN/H5PAR/ph5_f90_hyperslab_by_chunk.F90 @@ -0,0 +1,170 @@ +! +! Number of processes is assumed to be 4 +! + PROGRAM DATASET_BY_CHUNK + + USE HDF5 ! This module contains all necessary modules +! USE MPI + + IMPLICIT NONE + + include 'mpif.h' + CHARACTER(LEN=11), PARAMETER :: filename = "sds_chnk.h5" ! File name + CHARACTER(LEN=8), PARAMETER :: dsetname = "IntArray" ! Dataset name + + 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) :: memspace ! Dataspace identifier in memory + INTEGER(HID_T) :: plist_id ! Property list identifier + + INTEGER(HSIZE_T), DIMENSION(2) :: dimsf = (/4,8/) ! Dataset dimensions + ! in the file. +! INTEGER, DIMENSION(7) :: dimsfi = (/4,8,0,0,0,0,0/) + INTEGER(HSIZE_T), DIMENSION (2) :: dimsfi = (/4,8/) + INTEGER(HSIZE_T), DIMENSION(2) :: chunk_dims = (/2,4/) ! Chunks dimensions + + INTEGER(HSIZE_T), DIMENSION(2) :: count + INTEGER(HSSIZE_T), DIMENSION(2) :: offset + INTEGER(HSIZE_T), DIMENSION(2) :: stride + INTEGER(HSIZE_T), DIMENSION(2) :: block + + INTEGER, ALLOCATABLE :: data (:,:) ! Data to write + INTEGER :: rank = 2 ! Dataset rank + + INTEGER :: error, error_n ! Error flags + ! + ! 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) + ! Quit if mpi_size is not 4 + if (mpi_size .NE. 4) then + write(*,*) 'This example is set up to use only 4 processes' + write(*,*) 'Quitting....' + goto 100 + endif + + ! + ! Initialize HDF5 library and Fortran interfaces. + ! + 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) + + ! + ! 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) + CALL h5screate_simple_f(rank, chunk_dims, memspace, error) + + ! + ! Create chunked dataset. + ! + CALL h5pcreate_f(H5P_DATASET_CREATE_F, plist_id, error) + CALL h5pset_chunk_f(plist_id, rank, chunk_dims, error) + CALL h5dcreate_f(file_id, dsetname, H5T_NATIVE_INTEGER, filespace, & + dset_id, error, plist_id) + CALL h5sclose_f(filespace, error) + ! + ! Each process defines dataset in memory and writes it to the hyperslab + ! in the file. + ! + stride(1) = 1 + stride(2) = 1 + count(1) = 1 + count(2) = 1 + block(1) = chunk_dims(1) + block(2) = chunk_dims(2) + if (mpi_rank .EQ. 0) then + offset(1) = 0 + offset(2) = 0 + endif + if (mpi_rank .EQ. 1) then + offset(1) = chunk_dims(1) + offset(2) = 0 + endif + if (mpi_rank .EQ. 2) then + offset(1) = 0 + offset(2) = chunk_dims(2) + endif + if (mpi_rank .EQ. 3) then + offset(1) = chunk_dims(1) + offset(2) = chunk_dims(2) + endif + ! + ! Select hyperslab in the file. + ! + CALL h5dget_space_f(dset_id, filespace, error) + CALL h5sselect_hyperslab_f (filespace, H5S_SELECT_SET_F, offset, count, error, & + stride, block) + ! + ! Initialize data buffer with trivial data. + ! + ALLOCATE (data(chunk_dims(1),chunk_dims(2))) + data = mpi_rank + 1 + ! + ! 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) + + ! + ! Write the dataset collectively. + ! + CALL h5dwrite_f(dset_id, H5T_NATIVE_INTEGER, data, dimsfi, error, & + file_space_id = filespace, mem_space_id = memspace, xfer_prp = plist_id) + ! + ! Write the dataset independently. + ! +! CALL h5dwrite_f(dset_id, H5T_NATIVE_INTEGER, data, dimsfi,error, & +! file_space_id = filespace, mem_space_id = memspace) + ! + ! Deallocate data buffer. + ! + DEALLOCATE(data) + + ! + ! Close dataspaces. + ! + CALL h5sclose_f(filespace, error) + CALL h5sclose_f(memspace, error) + ! + ! Close the dataset. + ! + CALL h5dclose_f(dset_id, error) + ! + ! Close the property list. + ! + CALL h5pclose_f(plist_id, error) + ! + ! Close the file. + ! + CALL h5fclose_f(file_id, error) + + ! + ! Close FORTRAN interfaces and HDF5 library. + ! + CALL h5close_f(error) + +100 continue + CALL MPI_FINALIZE(mpierror) + + END PROGRAM DATASET_BY_CHUNK |