Changeset 2715 for trunk/NEMOGCM/NEMO/OPA_SRC/DYN/dynzdf_imp.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/DYN/dynzdf_imp.F90 (modified) (4 diffs)

trunk/NEMOGCM/NEMO/OPA_SRC/DYN/dynzdf_imp.F90

@@ -1 +1 @@
 MODULE dynzdf_imp
-   !!==============================================================================
+   !!======================================================================
    !!                    ***  MODULE  dynzdf_imp  ***
    !! Ocean dynamics:  vertical component(s) of the momentum mixing trend
-   !!==============================================================================
+   !!======================================================================
    !! History :  OPA  !  1990-10  (B. Blanke)  Original code
    !!            8.0  !  1997-05  (G. Madec)  vertical component of isopycnal
-   !!   NEMO     1.0  !  2002-08  (G. Madec)  F90: Free form and module
+   !!   NEMO     0.5  !  2002-08  (G. Madec)  F90: Free form and module
    !!            3.3  !  2010-04  (M. Leclair, G. Madec)  Forcing averaged over 2 time steps
    !!----------------------------------------------------------------------
 
    !!----------------------------------------------------------------------
-   !!   dyn_zdf_imp  : update the momentum trend with the vertical diffu-
-   !!                  sion using a implicit time-stepping.
+   !!   dyn_zdf_imp  : update the momentum trend with the vertical diffusion using a implicit time-stepping
    !!----------------------------------------------------------------------
    USE oce             ! ocean dynamics and tracers
@@ -20 +19 @@
    USE phycst          ! physical constants
    USE in_out_manager  ! I/O manager
+   USE lib_mpp         ! MPP library
 
    IMPLICIT NONE
@@ -54 +54 @@
       !! ** Action : - Update (ua,va) arrays with the after vertical diffusive mixing trend.
       !!---------------------------------------------------------------------
-      USE oce, ONLY :  zwd   => ta      ! use ta as workspace
-      USE oce, ONLY :  zws   => sa      ! use sa as workspace
-      !!
-      INTEGER , INTENT( in ) ::   kt    ! ocean time-step index
-      REAL(wp), INTENT( in ) ::  p2dt   ! vertical profile of tracer time-step
-      !!
-      INTEGER  ::   ji, jj, jk             ! dummy loop indices
-      REAL(wp) ::   z1_p2dt, zcoef         ! temporary scalars
-      REAL(wp) ::   zzwi, zzws, zrhs       ! temporary scalars
-      REAL(wp), DIMENSION(jpi,jpj,jpk) ::   zwi        ! 3D workspace
+      USE wrk_nemo, ONLY:   wrk_in_use, wrk_not_released
+      USE oce     , ONLY:  zwd  => ta       , zws   => sa   ! (ta,sa) used as 3D workspace
+      USE wrk_nemo, ONLY:   zwi => wrk_3d_3                 ! 3D workspace
+      !!
+      INTEGER , INTENT(in) ::   kt    ! ocean time-step index
+      REAL(wp), INTENT(in) ::  p2dt   ! vertical profile of tracer time-step
+      !!
+      INTEGER  ::   ji, jj, jk   ! dummy loop indices
+      REAL(wp) ::   z1_p2dt, zcoef, zzwi, zzws, zrhs   ! local scalars
       !!----------------------------------------------------------------------
+
+      IF( wrk_in_use(3, 3) ) THEN
+         CALL ctl_stop('dyn_zdf_imp: requested workspace array unavailable')   ;   RETURN
+      END IF
 
       IF( kt == nit000 ) THEN
@@ -253 +256 @@
       END DO
       !
+      IF( wrk_not_released(3, 3) )   CALL ctl_stop('dyn_zdf_imp: failed to release workspace array')
+      !
    END SUBROUTINE dyn_zdf_imp