Changeset 2715 for trunk/NEMOGCM/NEMO/OPA_SRC/TRD/trdmld_oce.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/TRD/trdmld_oce.F90 (modified) (6 diffs)

trunk/NEMOGCM/NEMO/OPA_SRC/TRD/trdmld_oce.F90

@@ -4 +4 @@
    !! Ocean trends :   set tracer and momentum trend variables
    !!======================================================================
-   !! History :  9.0  !  04-08  (C. Talandier)  New trends organization
+   !! History :  1.0  ! 2004-08  (C. Talandier)  New trends organization
    !!----------------------------------------------------------------------
-   USE par_oce         ! ocean parameters
+   USE par_oce        ! ocean parameters
 
    IMPLICIT NONE
    PRIVATE
+
+   PUBLIC   trdmld_oce_alloc    ! Called in trdmld.F90
 
 #if defined key_trdmld
@@ -17 +19 @@
 #endif
    !!* mixed layer trends indices
-   INTEGER, PARAMETER, PUBLIC ::   jpltrd = 11    !: number of mixed-layer trends arrays
-   INTEGER, PUBLIC   & 
-#if !defined key_agrif
-      , PARAMETER  &
-#endif
-::   jpktrd = jpk   !: max level for mixed-layer trends diag.
+   INTEGER, PARAMETER, PUBLIC ::   jpltrd = 11      !: number of mixed-layer trends arrays
+   INTEGER, PUBLIC            ::   jpktrd           !: max level for mixed-layer trends diag.
    !
    INTEGER, PUBLIC, PARAMETER ::   jpmld_xad =  1   !:  zonal      
@@ -46 +44 @@
    CHARACTER(LEN=80) , PUBLIC :: clname, ctrd(jpltrd+1,2)
 
-   INTEGER , PUBLIC, DIMENSION(jpi,jpj)     ::   nmld   !: mixed layer depth indexes 
-   INTEGER , PUBLIC, DIMENSION(jpi,jpj)     ::   nbol   !: mixed-layer depth indexes when read from file
+   INTEGER , PUBLIC, ALLOCATABLE, SAVE, DIMENSION(:,:)   ::   nmld   !: mixed layer depth indexes 
+   INTEGER , PUBLIC, ALLOCATABLE, SAVE, DIMENSION(:,:)   ::   nbol   !: mixed-layer depth indexes when read from file
 
-   REAL(wp), PUBLIC, DIMENSION(jpi,jpj,jpk) ::   wkx    !:
+   REAL(wp), PUBLIC, ALLOCATABLE, SAVE, DIMENSION(:,:,:) ::   wkx    !:
 
-   REAL(wp), PUBLIC, DIMENSION(jpi,jpj) ::  &
+   REAL(wp), PUBLIC, ALLOCATABLE, SAVE, DIMENSION(:,:)   ::  &
       rmld   ,                      & !: mld depth (m) corresponding to nmld
       tml    , sml  ,               & !: \ "now" mixed layer temperature/salinity
@@ -66 +64 @@
       rmld_sum, rmldbn                !: needed to compute the leap-frog time mean of the ML depth
 
-   REAL(wp), PUBLIC, DIMENSION(jpi,jpj) ::  &
+   REAL(wp), PUBLIC, ALLOCATABLE, SAVE, DIMENSION(:,:) ::  &
       tmlatfb, tmlatfn ,            & !: "before" Asselin contribution at begining of the averaging
       smlatfb, smlatfn,             & !: period (i.e. last contrib. from previous such period) and 
@@ -72 +70 @@
       tmlatfm, smlatfm                !: accumulator for Asselin trends (needed for storage only)
 
-   REAL(wp), PUBLIC, DIMENSION(jpi,jpj,jpltrd) ::  &
+   REAL(wp), PUBLIC, ALLOCATABLE, DIMENSION(:,:,:) ::  &
       tmltrd,                       & !: \ physical contributions to the total trend (for T/S),
       smltrd,                       & !: / cumulated over the current analysis window
@@ -83 +81 @@
 #endif
    !!----------------------------------------------------------------------
-   !! NEMO/OPA 3.3 , NEMO Consortium (2010)
+   !! NEMO/OPA 4.0 , NEMO Consortium (2011)
    !! $Id$ 
-   !! Software governed by the CeCILL licence (NEMOGCM/NEMO_CeCILL.txt)
+   !! Software governed by the CeCILL licence     (NEMOGCM/NEMO_CeCILL.txt)
+   !!----------------------------------------------------------------------
+CONTAINS
+
+  INTEGER FUNCTION trdmld_oce_alloc()
+     !!----------------------------------------------------------------------
+     !!                 ***  FUNCTION trdmld_oce_alloc   ***
+     !!----------------------------------------------------------------------
+     USE lib_mpp
+     INTEGER :: ierr(5)
+     !!----------------------------------------------------------------------
+
+     ! Initialise jpktrd here as can no longer do it in MODULE body since
+     ! jpk is now a variable.
+     jpktrd = jpk   !: max level for mixed-layer trends diag.
+
+     ierr(:) = 0
+
+#if   defined  key_trdmld   ||   defined key_esopa
+     ALLOCATE( nmld(jpi,jpj), nbol(jpi,jpj),       &
+        &      wkx(jpi,jpj,jpk), rmld(jpi,jpj),    & 
+        &      tml(jpi,jpj)    , sml(jpi,jpj),     & 
+        &      tmlb(jpi,jpj)   , smlb(jpi,jpj) ,   &
+        &      tmlbb(jpi,jpj)  , smlbb(jpi,jpj), STAT = ierr(1) )
+
+     ALLOCATE( tmlbn(jpi,jpj)  , smlbn(jpi,jpj),   &
+        &      tmltrdm(jpi,jpj), smltrdm(jpi,jpj), &
+        &      tml_sum(jpi,jpj), tml_sumb(jpi,jpj),&
+        &      tmltrd_atf_sumb(jpi,jpj)           , STAT=ierr(2) )
+
+     ALLOCATE( sml_sum(jpi,jpj), sml_sumb(jpi,jpj), &
+        &      smltrd_atf_sumb(jpi,jpj),            &
+        &      rmld_sum(jpi,jpj), rmldbn(jpi,jpj),  &
+        &      tmlatfb(jpi,jpj), tmlatfn(jpi,jpj), STAT = ierr(3) )
+
+     ALLOCATE( smlatfb(jpi,jpj), smlatfn(jpi,jpj), & 
+        &      tmlatfm(jpi,jpj), smlatfm(jpi,jpj), &
+        &      tmltrd(jpi,jpj,jpltrd),   smltrd(jpi,jpj,jpltrd), STAT=ierr(4))
+
+     ALLOCATE( tmltrd_sum(jpi,jpj,jpltrd),tmltrd_csum_ln(jpi,jpj,jpltrd),      &
+        &      tmltrd_csum_ub(jpi,jpj,jpltrd), smltrd_sum(jpi,jpj,jpltrd),     &
+        &      smltrd_csum_ln(jpi,jpj,jpltrd), smltrd_csum_ub(jpi,jpj,jpltrd), STAT=ierr(5) )
+#endif
+      !
+      trdmld_oce_alloc = MAXVAL( ierr )
+      IF( lk_mpp                )   CALL mpp_sum ( trdmld_oce_alloc )
+      IF( trdmld_oce_alloc /= 0 )   CALL ctl_warn('trdmld_oce_alloc: failed to allocate arrays')
+      !
+   END FUNCTION trdmld_oce_alloc
+
    !!======================================================================
 END MODULE trdmld_oce