Changeset 4517 for palm/trunk/SOURCE/land_surface_model_mod.f90

Timestamp:

May 3, 2020 2:29:30 PM (4 years ago)

Author:

raasch

Message:

added restart with MPI-IO for reading local arrays

File:

• palm/trunk/SOURCE/land_surface_model_mod.f90 (modified) (7 diffs)

palm/trunk/SOURCE/land_surface_model_mod.f90

@@ -25 +25 @@
 ! -----------------
 ! $Id$
+! added restart with MPI-IO for reading local arrays
+! 
+! 4495 2020-04-13 20:11:20Z raasch
 ! restart data handling with MPI-IO added
 ! 
@@ -291 +294 @@
 
     USE restart_data_mpi_io_mod,                                                                   &
-        ONLY:  rd_mpi_io_surface_filetypes, wrd_mpi_io, wrd_mpi_io_surface
+        ONLY:  rd_mpi_io_check_array, rd_mpi_io_surface_filetypes, rrd_mpi_io, rrd_mpi_io_surface, &
+               wrd_mpi_io, wrd_mpi_io_surface
 
     USE statistics,                                                            &
@@ -808 +812 @@
 
     INTERFACE lsm_rrd_local
-       MODULE PROCEDURE lsm_rrd_local
+       MODULE PROCEDURE lsm_rrd_local_ftn
+       MODULE PROCEDURE lsm_rrd_local_mpi
     END INTERFACE lsm_rrd_local
 
@@ -6674 +6679 @@
        CALL wrd_mpi_io( 'lsm_end_index_h',  surf_lsm_h%end_index )
        CALL wrd_mpi_io( 'lsm_global_start_index_h', global_start_index )
-       CALL wrd_mpi_io_surface ( 't_soil_h', t_soil_h%var_2d )
+       CALL wrd_mpi_io_surface( 't_soil_h', t_soil_h%var_2d )
        CALL wrd_mpi_io_surface( 'm_soil_h',  m_soil_h%var_2d )
        CALL wrd_mpi_io_surface( 'm_liq_h', m_liq_h%var_1d )
@@ -6686 +6691 @@
                                              surface_data_to_write, global_start_index )
 
-          CALL wrd_mpi_io( 'lsm_start_index_v_' //dum,  surf_lsm_v(l)%start_index )
-          CALL wrd_mpi_io( 'lsm_end_index_v_' //dum,  surf_lsm_v(l)%end_index )
-          CALL wrd_mpi_io( 'lsm_global_start_index_v_' //dum , global_start_index )
+          CALL wrd_mpi_io( 'lsm_start_index_v_' // dum,  surf_lsm_v(l)%start_index )
+          CALL wrd_mpi_io( 'lsm_end_index_v_' // dum,  surf_lsm_v(l)%end_index )
+          CALL wrd_mpi_io( 'lsm_global_start_index_v_' // dum , global_start_index )
 
           IF ( .NOT. surface_data_to_write )  CYCLE
@@ -6708 +6713 @@
 ! Description:
 ! ------------
-!> Soubroutine reads lsm data from restart file(s)
+!> Read module-specific local restart data arrays (Fortran binary format).
 !------------------------------------------------------------------------------!
- SUBROUTINE lsm_rrd_local( k, nxlf, nxlc, nxl_on_file, nxrf, nxrc,              &
-                          nxr_on_file, nynf, nync, nyn_on_file, nysf, nysc,    &
-                          nys_on_file, tmp_2d, found )
+ SUBROUTINE lsm_rrd_local_ftn( k, nxlf, nxlc, nxl_on_file, nxrf, nxrc,              &
+                               nxr_on_file, nynf, nync, nyn_on_file, nysf, nysc,    &
+                               nys_on_file, tmp_2d, found )
 
 
@@ -7305 +7310 @@
 
 
- END SUBROUTINE lsm_rrd_local
+ END SUBROUTINE lsm_rrd_local_ftn
+
+!------------------------------------------------------------------------------!
+!
+! Description:
+! ------------
+!> Read module-specific local restart data arrays (MPI-IO).
+!------------------------------------------------------------------------------!
+ SUBROUTINE lsm_rrd_local_mpi
+
+    USE control_parameters
+
+    USE indices
+
+    USE pegrid
+
+
+    IMPLICIT NONE
+
+    CHARACTER(LEN=1) ::  dum  !< dummy to create correct string for creating variable string
+
+    INTEGER(iwp) ::  l   !< running index surface orientation
+
+    INTEGER(iwp), DIMENSION(nys:nyn,nxl:nxr) ::  end_index
+    INTEGER(iwp), DIMENSION(nys:nyn,nxl:nxr) ::  global_start
+    INTEGER(iwp), DIMENSION(nys:nyn,nxl:nxr) ::  start_index
+
+    LOGICAL      :: array_found
+    LOGICAL      :: ldum
+
+
+    CALL rd_mpi_io_check_array( 'c_liq_av', found = array_found )
+    IF ( array_found )  THEN
+       IF ( .NOT. ALLOCATED( c_liq_av ) )  ALLOCATE( c_liq_av(nysg:nyng,nxlg:nxrg) )
+       CALL rrd_mpi_io( 'c_liq_av',  c_liq_av )
+    ENDIF
+
+    CALL rd_mpi_io_check_array( 'c_soil_av', found = array_found )
+    IF ( array_found )  THEN
+       IF ( .NOT. ALLOCATED( c_soil_av ) )  ALLOCATE( c_soil_av(nysg:nyng,nxlg:nxrg) )
+       CALL rrd_mpi_io( 'c_soil_av',  c_soil_av )
+    ENDIF
+
+    CALL rd_mpi_io_check_array( 'c_veg_av', found = array_found )
+    IF ( array_found )  THEN
+       IF ( .NOT. ALLOCATED( c_veg_av ) )  ALLOCATE( c_veg_av(nysg:nyng,nxlg:nxrg) )
+       CALL rrd_mpi_io( 'c_veg_av',  c_veg_av )
+    ENDIF
+
+    CALL rd_mpi_io_check_array( 'lai_av', found = array_found )
+    IF ( array_found )  THEN
+       IF ( .NOT. ALLOCATED( lai_av ) )  ALLOCATE( lai_av(nysg:nyng,nxlg:nxrg) )
+       CALL rrd_mpi_io( 'lai_av',  lai_av )
+    ENDIF
+
+    CALL rd_mpi_io_check_array( 'm_liq_av', found = array_found )
+    IF ( array_found )  THEN
+       IF ( .NOT. ALLOCATED( m_liq_av ) )  ALLOCATE( m_liq_av(nysg:nyng,nxlg:nxrg) )
+       CALL rrd_mpi_io( 'm_liq_av',  m_liq_av )
+    ENDIF
+
+    CALL rd_mpi_io_check_array( 'm_soil_av', found = array_found )
+    IF ( array_found )  THEN
+       IF ( .NOT. ALLOCATED( m_soil_av ) )  ALLOCATE( m_soil_av(nzb_soil:nzt_soil,nysg:nyng,nxlg:nxrg) )
+       CALL rrd_mpi_io( 'm_soil_av', m_soil_av, nzb_soil, nzt_soil )
+    ENDIF
+
+    CALL rd_mpi_io_check_array( 'qsws_liq_av', found = array_found )
+    IF ( array_found )  THEN
+       IF ( .NOT. ALLOCATED( qsws_liq_av ) )  ALLOCATE( qsws_liq_av(nysg:nyng,nxlg:nxrg) )
+       CALL rrd_mpi_io( 'qsws_liq_av',  qsws_liq_av )
+    ENDIF
+
+    CALL rd_mpi_io_check_array( 'qsws_soil_av', found = array_found )
+    IF ( array_found )  THEN
+       IF ( .NOT. ALLOCATED( qsws_soil_av ) )  ALLOCATE( qsws_soil_av(nysg:nyng,nxlg:nxrg) )
+       CALL rrd_mpi_io( 'qsws_soil_av',  qsws_soil_av )
+    ENDIF
+
+    CALL rd_mpi_io_check_array( 'qsws_veg_av', found = array_found )
+    IF ( array_found )  THEN
+       IF ( .NOT. ALLOCATED( qsws_veg_av ) )  ALLOCATE( qsws_veg_av(nysg:nyng,nxlg:nxrg) )
+       CALL rrd_mpi_io( 'qsws_veg_av',  qsws_veg_av )
+    ENDIF
+
+    CALL rd_mpi_io_check_array( 't_soil_av', found = array_found )
+    IF ( array_found )  THEN
+       IF ( .NOT. ALLOCATED( t_soil_av ) )  ALLOCATE( t_soil_av(nzb_soil:nzt_soil,nysg:nyng,nxlg:nxrg) )
+       CALL rrd_mpi_io( 't_soil_av', t_soil_av, nzb_soil, nzt_soil )
+    ENDIF
+
+    CALL rrd_mpi_io( 'lsm_start_index_h',  surf_lsm_h%start_index )
+    CALL rrd_mpi_io( 'lsm_end_index_h',  surf_lsm_h%end_index )
+    CALL rrd_mpi_io( 'lsm_global_start_index_h', global_start )
+
+    CALL rd_mpi_io_surface_filetypes ( surf_lsm_h%start_index, surf_lsm_h%end_index, ldum,         &
+                                       global_start )
+
+    CALL rrd_mpi_io_surface( 't_soil_h', t_soil_h%var_2d )
+    CALL rrd_mpi_io_surface( 'm_soil_h',  m_soil_h%var_2d )
+    CALL rrd_mpi_io_surface( 'm_liq_h', m_liq_h%var_1d )
+    CALL rrd_mpi_io_surface( 't_surface_h', t_surface_h%var_1d )
+
+
+    DO  l = 0, 3
+
+       WRITE( dum, '(I1)')  l
+
+!kk    In case of nothing to do, the settings of start_index and end_index differ
+!kk    between writing and reading restart file
+!kk
+!kk    Has to be discussed with the developpers
+
+       CALL rrd_mpi_io( 'lsm_start_index_v_' // dum,  start_index )
+       CALL rrd_mpi_io( 'lsm_end_index_v_' // dum,  end_index )
+       CALL rrd_mpi_io( 'lsm_global_start_index_v_' // dum , global_start )
+
+       CALL rd_mpi_io_surface_filetypes( start_index, end_index, ldum, global_start )
+
+       IF ( MAXVAL( end_index ) <= 0 )  CYCLE
+
+       CALL rrd_mpi_io_surface( 't_soil_v(' // dum // ')', t_soil_v(l)%var_2d )
+       CALL rrd_mpi_io_surface( 'm_soil_v(' // dum // ')', m_soil_v(l)%var_2d )
+       CALL rrd_mpi_io_surface( 'm_liq_v(' // dum // ')',  m_liq_v(l)%var_1d )
+       CALL rrd_mpi_io_surface( 't_surface_v(' // dum // ')',  t_surface_v(l)%var_1d )
+
+    ENDDO
+
+ END SUBROUTINE lsm_rrd_local_mpi
 
 !------------------------------------------------------------------------------!