Changeset 2715 for trunk/NEMOGCM/NEMO/OPA_SRC/TRA/traldf.F90

Timestamp:

2011-03-30T17:58:35+02:00 (13 years ago)

Author:

rblod

Message:

First attempt to put dynamic allocation on the trunk

File:

• trunk/NEMOGCM/NEMO/OPA_SRC/TRA/traldf.F90 (modified) (5 diffs)

trunk/NEMOGCM/NEMO/OPA_SRC/TRA/traldf.F90

@@ -37 +37 @@
    !
    INTEGER ::   nldf = 0   ! type of lateral diffusion used defined from ln_traldf_... namlist logicals)
-#if defined key_traldf_ano
-   REAL, DIMENSION(jpi,jpj,jpk) ::   t0_ldf, s0_ldf   ! lateral diffusion trends of T & S for a constant profile
-#endif
+
+   REAL, SAVE, ALLOCATABLE, DIMENSION(:,:,:) ::   t0_ldf, s0_ldf   !: lateral diffusion trends of T & S for a cst profile
+   !                                                               !  (key_traldf_ano only)
 
    !! * Substitutions
@@ -130 +130 @@
       !!----------------------------------------------------------------------
       INTEGER ::   ioptio, ierr         ! temporary integers 
-!      
       !!----------------------------------------------------------------------
 
@@ -238 +237 @@
       !! ** Purpose :   initializations of 
       !!----------------------------------------------------------------------
+      USE wrk_nemo, ONLY:   wrk_in_use, wrk_not_released
+      USE wrk_nemo, ONLY:   zt_ref => wrk_3d_1, ztb => wrk_3d_2, zavt => wrk_3d_3   ! 3D workspaces
+      USE wrk_nemo, ONLY:   zs_ref => wrk_3d_4, zsb => wrk_3d_5                     ! 3D workspaces
+      !
       USE zdf_oce         ! vertical mixing
       USE trazdf          ! vertical mixing: double diffusion
       USE zdfddm          ! vertical mixing: double diffusion
-      !!
+      !
       INTEGER  ::   jk              ! Dummy loop indice
-      LOGICAL  ::   llsave          !
-      REAL(wp) ::   zt0, zs0, z12   ! temporary scalar
-      REAL(wp), DIMENSION(jpi,jpj,jpk) ::   zt_ref, ztb, zavt   ! 3D workspace
-      REAL(wp), DIMENSION(jpi,jpj,jpk) ::   zs_ref, zsb         ! 3D workspace
-      !!----------------------------------------------------------------------
+      INTEGER  ::   ierr            ! local integer
+      LOGICAL  ::   llsave          ! local logical
+      REAL(wp) ::   zt0, zs0, z12   ! local scalar
+      !!----------------------------------------------------------------------
+
+      IF( wrk_in_use(3, 1,2,3,4,5) ) THEN
+         CALL ctl_stop('ldf_ano : requested workspace arrays unavailable')   ;   RETURN
+      ENDIF
 
       IF(lwp) THEN
@@ -254 +260 @@
          WRITE(numout,*) '~~~~~~~~~~~'
       ENDIF
+
+      !                              ! allocate trabbl arrays
+      ALLOCATE( t0_ldf(jpi,jpj,jpk) , s0_ldf(jpi,jpj,jpk) , STAT=ierr )
+      IF( lk_mpp    )   CALL mpp_sum( ierr )
+      IF( ierr /= 0 )   CALL ctl_stop( 'STOP', 'ldf_ano: unable to allocate arrays' )
 
       ! defined the T & S reference profiles
@@ -309 +320 @@
       avt(:,:,:)        = zavt(:,:,:)
       !
+      IF( wrk_not_released(3, 1,2,3,4,5) )   CALL ctl_stop('ldf_ano: failed to release workspace arrays')
+      !
    END SUBROUTINE ldf_ano