Changeset 14033

Timestamp:

2020-12-03T11:30:33+01:00 (4 years ago)

Author:

epico

Message:

changes in module_example

File:

• NEMO/trunk/src/OCE/module_example (modified) (2 diffs)

NEMO/trunk/src/OCE/module_example

@@ -49 +49 @@
 
    !! * Substitutions
+   ! for DO macro
+#  include "do_loop_substitute.h90"
+   !for other substitutions
 #  include "exampl_substitute.h90"
    !!----------------------------------------------------------------------
@@ -95 +98 @@
       REAL(wp) ::   zmlmin, zbbrho   ! temporary scalars     (DOCTOR : start with z)
       REAL(wp) ::   zfact1, zfact2   ! do not use continuation lines in declaration
-      REAL(wp), DIMENSION(jpi,jpj) ::   zwrk_2d   ! 2D workspace
+      REAL(wp), DIMENSION(A2D(nn_hls))     ::   zwrk_2d   ! 2D workspace
+      REAL(wp), DIMENSION(A2D(nn_hls),jpk) ::   zwrk_3d   ! 3D workspace
       !!--------------------------------------------------------------------
       !
-      IF( kt == nit000  )   CALL exa_mpl_init    ! Initialization (first time-step only)
+      IF( ntile == 0 .OR. ntile == 1 )  THEN                       ! Do only on the first tile
+         IF( kt == nit000  )   CALL exa_mpl_init    ! Initialization (first time-step only)
 
-      zmlmin = 1.e-8                             ! Local constant initialization
-      zbbrho =  .5 * ebb / rho0
-      zfact1 = -.5 * rdt * efave
-      zfact2 = 1.5 * rdt * ediss
-
+         zmlmin = 1.e-8                             ! Local constant initialization
+         zbbrho =  .5 * ebb / rho0
+         zfact1 = -.5 * rdt * efave
+         zfact2 = 1.5 * rdt * ediss
+      ENDIF
+     
       SELECT CASE ( npdl )                       ! short description of the action
       !
       CASE ( 0 )                                      ! describe case 1
-         DO jk = 2, jpkm1
-            DO jj = 2, jpjm1
-               DO ji = fs_2, fs_jpim1   ! vector opt.
-                  avm(ji,jj,jk) = ....
-               END DO
-            END DO
-         END DO
+         DO_3D( nn_hls-1, nn_hls-1, nn_hls-1, nn_hls-1, 1, jpkm1 )
+            avm(ji,jj,jk) = ....
+         END_3D
          !
       CASE ( 1 )                                      ! describe case 2
-         DO jk = 2, jpkm1
-            DO jj = 2, jpjm1
-               DO ji = fs_2, fs_jpim1   ! vector opt.
-                  avm(ji,jj,jk) = ...
-               END DO
-            END DO
-         END DO
+         DO_3D( nn_hls-1, nn_hls-1, nn_hls-1, nn_hls-1, 1, jpkm1 )
+            avm(ji,jj,jk) = ....
+         END_3D
          !
       END SELECT
       !
-      CALL lbc_lnk( 'module_example', avm, 'T', 1. )              ! Lateral boundary conditions (unchanged sign)
+      CALL lbc_lnk( 'module_example', avm, 'T', 1., ncsten=true )     ! Lateral boundary conditions (unchanged sign)
+      !                                                                ! ncsten=false for 5-points stencil communication
+      !                                                                ! ncsten=true (default)  for 9-points stencil communication
       !
    END SUBROUTINE exa_mpl