New URL for NEMO forge!   http://forge.nemo-ocean.eu

Since March 2022 along with NEMO 4.2 release, the code development moved to a self-hosted GitLab.
This present forge is now archived and remained online for history.
Changeset 13409 for NEMO/branches – NEMO

Changeset 13409 for NEMO/branches


Ignore:
Timestamp:
2020-08-17T15:28:54+02:00 (4 years ago)
Author:
hadcv
Message:

Remaining changes prior to trunk merge

Location:
NEMO/branches/UKMO/dev_r12866_HPC-02_Daley_Tiling_trial_extra_halo/src/OCE
Files:
36 edited

Legend:

Unmodified
Added
Removed

  • NEMO/branches/UKMO/dev_r12866_HPC-02_Daley_Tiling_trial_extra_halo/src/OCE/ASM/asminc.F90

    r12958 r13409  
    518518      INTEGER  :: it 
    519519      REAL(wp) :: zincwgt  ! IAU weight for current time step 
    520       REAL (wp), DIMENSION(jpi,jpj,jpk) :: fzptnz ! 3d freezing point values 
     520      REAL(wp), DIMENSION(A2D,jpk) :: fzptnz ! 3d freezing point values 
    521521      !!---------------------------------------------------------------------- 
    522522      ! 
    523523      ! freezing point calculation taken from oc_fz_pt (but calculated for all depths)  
    524524      ! used to prevent the applied increments taking the temperature below the local freezing point  
    525       DO jk = 1, jpkm1 
    526         CALL eos_fzp( pts(:,:,jk,jp_sal,Kmm), fzptnz(:,:,jk), gdept(:,:,jk,Kmm) ) 
    527       END DO 
     525      ! TODO: NOT TESTED- logical is forced to False 
     526      IF( ln_temnofreeze ) THEN 
     527         DO jk = 1, jpkm1 
     528           CALL eos_fzp( pts(:,:,jk,jp_sal,Kmm), fzptnz(:,:,jk), gdept(:,:,jk,Kmm) ) 
     529         END DO 
     530      ENDIF 
    528531         ! 
    529532         !                             !-------------------------------------- 
    … …  
    536539            zincwgt = wgtiau(it) / rn_Dt   ! IAU weight for the current time step 
    537540            ! 
    538             IF(lwp) THEN 
    539                WRITE(numout,*)  
    540                WRITE(numout,*) 'tra_asm_inc : Tracer IAU at time step = ', kt,' with IAU weight = ', wgtiau(it) 
    541                WRITE(numout,*) '~~~~~~~~~~~~' 
     541            IF( ntile == 0 .OR. ntile == 1 )  THEN                       ! Do only on the first tile 
     542               IF(lwp) THEN 
     543                  WRITE(numout,*) 
     544                  WRITE(numout,*) 'tra_asm_inc : Tracer IAU at time step = ', kt,' with IAU weight = ', wgtiau(it) 
     545                  WRITE(numout,*) '~~~~~~~~~~~~' 
     546               ENDIF 
    542547            ENDIF 
    543548            ! 
    544549            ! Update the tracer tendencies 
     550            ! TODO: NOT TESTED- logical is forced to False 
    545551            DO jk = 1, jpkm1 
    546552               IF (ln_temnofreeze) THEN 
    547553                  ! Do not apply negative increments if the temperature will fall below freezing 
    548                   WHERE(t_bkginc(:,:,jk) > 0.0_wp .OR. & 
    549                      &   pts(:,:,jk,jp_tem,Kmm) + pts(:,:,jk,jp_tem,Krhs) + t_bkginc(:,:,jk) * wgtiau(it) > fzptnz(:,:,jk) )  
    550                      pts(:,:,jk,jp_tem,Krhs) = pts(:,:,jk,jp_tem,Krhs) + t_bkginc(:,:,jk) * zincwgt   
     554                  WHERE(t_bkginc(A2D0,jk) > 0.0_wp .OR. & 
     555                     &   pts(A2D0,jk,jp_tem,Kmm) + pts(A2D0,jk,jp_tem,Krhs) + t_bkginc(A2D0,jk) * wgtiau(it) > fzptnz(:,:,jk) ) 
     556                     pts(A2D0,jk,jp_tem,Krhs) = pts(A2D0,jk,jp_tem,Krhs) + t_bkginc(A2D0,jk) * zincwgt 
    551557                  END WHERE 
    552558               ELSE 
    553                   pts(:,:,jk,jp_tem,Krhs) = pts(:,:,jk,jp_tem,Krhs) + t_bkginc(:,:,jk) * zincwgt   
     559                  DO_2D_00_00 
     560                     pts(ji,jj,jk,jp_tem,Krhs) = pts(ji,jj,jk,jp_tem,Krhs) + t_bkginc(ji,jj,jk) * zincwgt 
     561                  END_2D 
    554562               ENDIF 
    555563               IF (ln_salfix) THEN 
    556564                  ! Do not apply negative increments if the salinity will fall below a specified 
    557565                  ! minimum value salfixmin 
    558                   WHERE(s_bkginc(:,:,jk) > 0.0_wp .OR. & 
    559                      &   pts(:,:,jk,jp_sal,Kmm) + pts(:,:,jk,jp_sal,Krhs) + s_bkginc(:,:,jk) * wgtiau(it) > salfixmin )  
    560                      pts(:,:,jk,jp_sal,Krhs) = pts(:,:,jk,jp_sal,Krhs) + s_bkginc(:,:,jk) * zincwgt 
     566                  WHERE(s_bkginc(A2D0,jk) > 0.0_wp .OR. & 
     567                     &   pts(A2D0,jk,jp_sal,Kmm) + pts(A2D0,jk,jp_sal,Krhs) + s_bkginc(A2D0,jk) * wgtiau(it) > salfixmin ) 
     568                     pts(A2D0,jk,jp_sal,Krhs) = pts(A2D0,jk,jp_sal,Krhs) + s_bkginc(A2D0,jk) * zincwgt 
    561569                  END WHERE 
    562570               ELSE 
    563                   pts(:,:,jk,jp_sal,Krhs) = pts(:,:,jk,jp_sal,Krhs) + s_bkginc(:,:,jk) * zincwgt 
     571                  DO_2D_00_00 
     572                     pts(ji,jj,jk,jp_sal,Krhs) = pts(ji,jj,jk,jp_sal,Krhs) + s_bkginc(ji,jj,jk) * zincwgt 
     573                  END_2D 
    564574               ENDIF 
    565575            END DO 
    … …  
    567577         ENDIF 
    568578         ! 
    569          IF ( kt == nitiaufin_r + 1  ) THEN   ! For bias crcn to work 
    570             DEALLOCATE( t_bkginc ) 
    571             DEALLOCATE( s_bkginc ) 
     579         IF( ntile == 0 .OR. ntile == nijtile )  THEN                ! Do only on the last tile 
     580            IF ( kt == nitiaufin_r + 1  ) THEN   ! For bias crcn to work 
     581               DEALLOCATE( t_bkginc ) 
     582               DEALLOCATE( s_bkginc ) 
     583            ENDIF 
    572584         ENDIF 
    573585         !                             !-------------------------------------- 
    … …  
    580592            ! 
    581593            ! Initialize the now fields with the background + increment 
     594            ! TODO: NOT TESTED- logical is forced to False 
    582595            IF (ln_temnofreeze) THEN 
    583596               ! Do not apply negative increments if the temperature will fall below freezing 
    584                WHERE( t_bkginc(:,:,:) > 0.0_wp .OR. pts(:,:,:,jp_tem,Kmm) + t_bkginc(:,:,:) > fzptnz(:,:,:) )  
    585                   pts(:,:,:,jp_tem,Kmm) = t_bkg(:,:,:) + t_bkginc(:,:,:)    
     597               WHERE( t_bkginc(A2D0,:) > 0.0_wp .OR. pts(A2D0,:,jp_tem,Kmm) + t_bkginc(A2D0,:) > fzptnz(:,:,:) ) 
     598                  pts(A2D0,:,jp_tem,Kmm) = t_bkg(A2D0,:) + t_bkginc(A2D0,:) 
    586599               END WHERE 
    587600            ELSE 
    588                pts(:,:,:,jp_tem,Kmm) = t_bkg(:,:,:) + t_bkginc(:,:,:)    
     601               DO_3D_00_00( 1, jpk ) 
     602                  pts(ji,jj,jk,jp_tem,Kmm) = t_bkg(ji,jj,jk) + t_bkginc(ji,jj,jk) 
     603               END_3D 
    589604            ENDIF 
    590605            IF (ln_salfix) THEN 
    591606               ! Do not apply negative increments if the salinity will fall below a specified 
    592607               ! minimum value salfixmin 
    593                WHERE( s_bkginc(:,:,:) > 0.0_wp .OR. pts(:,:,:,jp_sal,Kmm) + s_bkginc(:,:,:) > salfixmin )  
    594                   pts(:,:,:,jp_sal,Kmm) = s_bkg(:,:,:) + s_bkginc(:,:,:)    
     608               WHERE( s_bkginc(A2D0,:) > 0.0_wp .OR. pts(A2D0,:,jp_sal,Kmm) + s_bkginc(A2D0,:) > salfixmin ) 
     609                  pts(A2D0,:,jp_sal,Kmm) = s_bkg(A2D0,:) + s_bkginc(A2D0,:) 
    595610               END WHERE 
    596611            ELSE 
    597                pts(:,:,:,jp_sal,Kmm) = s_bkg(:,:,:) + s_bkginc(:,:,:)    
    598             ENDIF 
    599  
    600             pts(:,:,:,:,Kbb) = pts(:,:,:,:,Kmm)                 ! Update before fields 
     612               DO_3D_00_00( 1, jpk ) 
     613                  pts(ji,jj,jk,jp_sal,Kmm) = s_bkg(ji,jj,jk) + s_bkginc(ji,jj,jk) 
     614               END_3D 
     615            ENDIF 
     616 
     617            DO_3D_00_00( 1, jpk ) 
     618               pts(ji,jj,jk,:,Kbb) = pts(ji,jj,jk,:,Kmm)             ! Update before fields 
     619            END_3D 
    601620 
    602621            CALL eos( pts(:,:,:,:,Kbb), rhd, rhop, gdept_0(:,:,:) )  ! Before potential and in situ densities 
    … …  
    605624!!gm 
    606625 
    607             IF( ln_zps .AND. .NOT. lk_c1d .AND. .NOT. ln_isfcav)           & 
    608                &  CALL zps_hde    ( kt, Kmm, jpts, pts(:,:,:,:,Kbb), gtsu, gtsv,        &  ! Partial steps: before horizontal gradient 
    609                &                              rhd, gru , grv               )  ! of t, s, rd at the last ocean level 
    610             IF( ln_zps .AND. .NOT. lk_c1d .AND.       ln_isfcav)                       & 
    611                &  CALL zps_hde_isf( nit000, Kmm, jpts, pts(:,:,:,:,Kbb), gtsu, gtsv, gtui, gtvi,    &  ! Partial steps for top cell (ISF) 
    612                &                                  rhd, gru , grv , grui, grvi          )  ! of t, s, rd at the last ocean level 
    613  
    614             DEALLOCATE( t_bkginc ) 
    615             DEALLOCATE( s_bkginc ) 
    616             DEALLOCATE( t_bkg    ) 
    617             DEALLOCATE( s_bkg    ) 
     626            ! TEMP: This change not necessary after extra haloes development (lbc_lnk removed from zps_hde*) 
     627            IF( ntile == 0 .OR. ntile == nijtile )  THEN                ! Do only for the full domain 
     628               IF( ln_zps .AND. .NOT. lk_c1d .AND. .NOT. ln_isfcav)           & 
     629                  &  CALL zps_hde    ( kt, Kmm, jpts, pts(:,:,:,:,Kbb), gtsu, gtsv,        &  ! Partial steps: before horizontal gradient 
     630                  &                              rhd, gru , grv               )  ! of t, s, rd at the last ocean level 
     631               IF( ln_zps .AND. .NOT. lk_c1d .AND.       ln_isfcav)                       & 
     632                  &  CALL zps_hde_isf( nit000, Kmm, jpts, pts(:,:,:,:,Kbb), gtsu, gtsv, gtui, gtvi,    &  ! Partial steps for top cell (ISF) 
     633                  &                                  rhd, gru , grv , grui, grvi          )  ! of t, s, rd at the last ocean level 
     634            ENDIF 
     635 
     636            IF( ntile == 0 .OR. ntile == nijtile )  THEN                ! Do only on the last tile 
     637               DEALLOCATE( t_bkginc ) 
     638               DEALLOCATE( s_bkginc ) 
     639               DEALLOCATE( t_bkg    ) 
     640               DEALLOCATE( s_bkg    ) 
     641            ENDIF 
     642         ! 
    618643         ENDIF 
    619644         !   
    620645      ENDIF 
     646      ! TODO: NOT TESTED- logical is forced to False 
    621647      ! Perhaps the following call should be in step 
    622648      IF ( ln_seaiceinc  )   CALL seaice_asm_inc ( kt )   ! apply sea ice concentration increment 
    … …  
    825851      INTEGER, INTENT(in), OPTIONAL ::   kindic   ! flag for disabling the deallocation 
    826852      ! 
     853      INTEGER  ::   ji, jj 
    827854      INTEGER  ::   it 
    828855      REAL(wp) ::   zincwgt   ! IAU weight for current time step 
    829856#if defined key_si3 
    830       REAL(wp), DIMENSION(jpi,jpj) ::   zofrld, zohicif, zseaicendg, zhicifinc 
     857      REAL(wp), DIMENSION(A2D) ::   zofrld, zohicif, zseaicendg, zhicifinc 
    831858      REAL(wp) ::   zhicifmin = 0.5_wp      ! ice minimum depth in metres 
    832859#endif 
    … …  
    843870            ! note this is not a tendency so should not be divided by rn_Dt (as with the tracer and other increments) 
    844871            ! 
    845             IF(lwp) THEN 
    846                WRITE(numout,*)  
    847                WRITE(numout,*) 'seaice_asm_inc : sea ice conc IAU at time step = ', kt,' with IAU weight = ', wgtiau(it) 
    848                WRITE(numout,*) '~~~~~~~~~~~~' 
     872            IF( ntile == 0 .OR. ntile == 1 )  THEN                       ! Do only on the first tile 
     873               IF(lwp) THEN 
     874                  WRITE(numout,*) 
     875                  WRITE(numout,*) 'seaice_asm_inc : sea ice conc IAU at time step = ', kt,' with IAU weight = ', wgtiau(it) 
     876                  WRITE(numout,*) '~~~~~~~~~~~~' 
     877               ENDIF 
    849878            ENDIF 
    850879            ! 
    … …  
    852881            ! 
    853882#if defined key_si3 
    854             zofrld (:,:) = 1._wp - at_i(:,:) 
    855             zohicif(:,:) = hm_i(:,:) 
    856             ! 
    857             at_i  (:,:) = 1. - MIN( MAX( 1.-at_i  (:,:) - seaice_bkginc(:,:) * zincwgt, 0.0_wp), 1.0_wp) 
    858             at_i_b(:,:) = 1. - MIN( MAX( 1.-at_i_b(:,:) - seaice_bkginc(:,:) * zincwgt, 0.0_wp), 1.0_wp) 
    859             fr_i(:,:) = at_i(:,:)        ! adjust ice fraction 
    860             ! 
    861             zseaicendg(:,:) = zofrld(:,:) - (1. - at_i(:,:))   ! find out actual sea ice nudge applied 
     883            DO_2D_00_00 
     884               zofrld (ji,jj) = 1._wp - at_i(ji,jj) 
     885               zohicif(ji,jj) = hm_i(ji,jj) 
     886               ! 
     887               at_i  (ji,jj) = 1. - MIN( MAX( 1.-at_i  (ji,jj) - seaice_bkginc(ji,jj) * zincwgt, 0.0_wp), 1.0_wp) 
     888               at_i_b(ji,jj) = 1. - MIN( MAX( 1.-at_i_b(ji,jj) - seaice_bkginc(ji,jj) * zincwgt, 0.0_wp), 1.0_wp) 
     889               fr_i(ji,jj) = at_i(ji,jj)        ! adjust ice fraction 
     890               ! 
     891               zseaicendg(ji,jj) = zofrld(ji,jj) - (1. - at_i(ji,jj))   ! find out actual sea ice nudge applied 
     892            END_2D 
    862893            ! 
    863894            ! Nudge sea ice depth to bring it up to a required minimum depth 
    864             WHERE( zseaicendg(:,:) > 0.0_wp .AND. hm_i(:,:) < zhicifmin )  
    865                zhicifinc(:,:) = (zhicifmin - hm_i(:,:)) * zincwgt     
     895            WHERE( zseaicendg(:,:) > 0.0_wp .AND. hm_i(A2D0) < zhicifmin ) 
     896               zhicifinc(:,:) = (zhicifmin - hm_i(A2D0)) * zincwgt 
    866897            ELSEWHERE 
    867898               zhicifinc(:,:) = 0.0_wp 
    … …  
    869900            ! 
    870901            ! nudge ice depth 
    871             hm_i (:,:) = hm_i (:,:) + zhicifinc(:,:) 
     902            DO_2D_00_00 
     903               hm_i (ji,jj) = hm_i (ji,jj) + zhicifinc(ji,jj) 
     904            END_2D 
    872905            ! 
    873906            ! seaice salinity balancing (to add) 
    … …  
    876909#if defined key_cice && defined key_asminc 
    877910            ! Sea-ice : CICE case. Pass ice increment tendency into CICE 
    878             ndaice_da(:,:) = seaice_bkginc(:,:) * zincwgt / rn_Dt 
    879 #endif 
    880             ! 
    881             IF ( kt == nitiaufin_r ) THEN 
    882                DEALLOCATE( seaice_bkginc ) 
     911            DO_2D_00_00 
     912               ndaice_da(ji,jj) = seaice_bkginc(ji,jj) * zincwgt / rn_Dt 
     913            END_2D 
     914#endif 
     915            ! 
     916            IF( ntile == 0 .OR. ntile == nijtile )  THEN                ! Do only on the last tile 
     917               IF ( kt == nitiaufin_r ) THEN 
     918                  DEALLOCATE( seaice_bkginc ) 
     919               ENDIF 
    883920            ENDIF 
    884921            ! 
    … …  
    886923            ! 
    887924#if defined key_cice && defined key_asminc 
    888             ndaice_da(:,:) = 0._wp        ! Sea-ice : CICE case. Zero ice increment tendency into CICE 
     925            DO_2D_00_00 
     926               ndaice_da(ji,jj) = 0._wp        ! Sea-ice : CICE case. Zero ice increment tendency into CICE 
     927            END_2D 
    889928#endif 
    890929            ! 
    … …  
    901940            ! 
    902941#if defined key_si3 
    903             zofrld (:,:) = 1._wp - at_i(:,:) 
    904             zohicif(:,:) = hm_i(:,:) 
    905             !  
    906             ! Initialize the now fields the background + increment 
    907             at_i(:,:) = 1. - MIN( MAX( 1.-at_i(:,:) - seaice_bkginc(:,:), 0.0_wp), 1.0_wp) 
    908             at_i_b(:,:) = at_i(:,:)  
    909             fr_i(:,:) = at_i(:,:)        ! adjust ice fraction 
    910             ! 
    911             zseaicendg(:,:) = zofrld(:,:) - (1. - at_i(:,:))   ! find out actual sea ice nudge applied 
     942            DO_2D_00_00 
     943               zofrld (ji,jj) = 1._wp - at_i(ji,jj) 
     944               zohicif(ji,jj) = hm_i(ji,jj) 
     945               ! 
     946               ! Initialize the now fields the background + increment 
     947               at_i(ji,jj) = 1. - MIN( MAX( 1.-at_i(ji,jj) - seaice_bkginc(ji,jj), 0.0_wp), 1.0_wp) 
     948               at_i_b(ji,jj) = at_i(ji,jj) 
     949               fr_i(ji,jj) = at_i(ji,jj)        ! adjust ice fraction 
     950               ! 
     951               zseaicendg(ji,jj) = zofrld(ji,jj) - (1. - at_i(ji,jj))   ! find out actual sea ice nudge applied 
     952            END_2D 
    912953            ! 
    913954            ! Nudge sea ice depth to bring it up to a required minimum depth 
    914             WHERE( zseaicendg(:,:) > 0.0_wp .AND. hm_i(:,:) < zhicifmin )  
    915                zhicifinc(:,:) = zhicifmin - hm_i(:,:) 
     955            WHERE( zseaicendg(:,:) > 0.0_wp .AND. hm_i(A2D0) < zhicifmin ) 
     956               zhicifinc(:,:) = zhicifmin - hm_i(A2D0) 
    916957            ELSEWHERE 
    917958               zhicifinc(:,:) = 0.0_wp 
    … …  
    919960            ! 
    920961            ! nudge ice depth 
    921             hm_i (:,:) = hm_i (:,:) + zhicifinc(:,:) 
     962            DO_2D_00_00 
     963               hm_i(ji,jj) = hm_i (ji,jj) + zhicifinc(ji,jj) 
     964            END_2D 
    922965            ! 
    923966            ! seaice salinity balancing (to add) 
    … …  
    926969#if defined key_cice && defined key_asminc 
    927970            ! Sea-ice : CICE case. Pass ice increment tendency into CICE 
    928            ndaice_da(:,:) = seaice_bkginc(:,:) / rn_Dt 
    929 #endif 
    930             IF ( .NOT. PRESENT(kindic) ) THEN 
    931                DEALLOCATE( seaice_bkginc ) 
    932             END IF 
     971            DO_2D_00_00 
     972               ndaice_da(ji,jj) = seaice_bkginc(ji,jj) / rn_Dt 
     973            END_2D 
     974#endif 
     975            IF( ntile == 0 .OR. ntile == nijtile )  THEN                ! Do only on the last tile 
     976               IF ( .NOT. PRESENT(kindic) ) THEN 
     977                  DEALLOCATE( seaice_bkginc ) 
     978               END IF 
     979            ENDIF 
    933980            ! 
    934981         ELSE 
    935982            ! 
    936983#if defined key_cice && defined key_asminc 
    937             ndaice_da(:,:) = 0._wp     ! Sea-ice : CICE case. Zero ice increment tendency into CICE 
     984            DO_2D_00_00 
     985               ndaice_da(ji,jj) = 0._wp     ! Sea-ice : CICE case. Zero ice increment tendency into CICE 
     986            END_2D 
    938987#endif 
    939988            ! 

  • NEMO/branches/UKMO/dev_r12866_HPC-02_Daley_Tiling_trial_extra_halo/src/OCE/BDY/bdytra.F90

    r12377 r13409  
    1313   !!---------------------------------------------------------------------- 
    1414   USE oce            ! ocean dynamics and tracers variables 
    15    USE dom_oce        ! ocean space and time domain variables  
     15   USE dom_oce        ! ocean space and time domain variables 
    1616   USE bdy_oce        ! ocean open boundary conditions 
    1717   USE bdylib         ! for orlanski library routines 
    … …  
    157157      INTEGER  ::   ib_bdy         ! Loop index 
    158158      !!---------------------------------------------------------------------- 
     159      ! TODO: TO BE TILED 
     160      ! TODO: NOT TESTED- requires bdy 
     161      ! NOTE: Tiling these BDY loops is nontrivial; IF statements to check whether a point is in the current tile won't work (will be for every ib, every tile). The idx_bdy structure might require modifying to include a %nblen and list of ib indices for the current tile. 
     162      IF( ntile /= 0 .AND. ntile /= 1 ) RETURN                        ! Do only for the full domain 
    159163      ! 
    160164      IF( ln_timing )   CALL timing_start('bdy_tra_dmp') 

  • NEMO/branches/UKMO/dev_r12866_HPC-02_Daley_Tiling_trial_extra_halo/src/OCE/DIA/diaar5.F90

    r13054 r13409  
    303303   END SUBROUTINE dia_ar5 
    304304 
    305    ! TODO: These changes and lbc_lnk not necessary if using XIOS (subdomain support, will not output haloes) 
     305   ! TEMP: These changes and lbc_lnk not necessary if using XIOS (subdomain support, will not output haloes) 
    306306   SUBROUTINE dia_ar5_hst( ktra, cptr, puflx, pvflx ) 
    307307      !!---------------------------------------------------------------------- 

  • NEMO/branches/UKMO/dev_r12866_HPC-02_Daley_Tiling_trial_extra_halo/src/OCE/DIA/diaptr.F90

    r13054 r13409  
    2222   USE oce              ! ocean dynamics and active tracers 
    2323   USE dom_oce          ! ocean space and time domain 
     24   ! TEMP: Possibly not necessary if using XIOS (if cumulative axis operations are possible) 
    2425   USE domain, ONLY : dom_tile 
    2526   USE phycst           ! physical constants 
    … …  
    7475CONTAINS 
    7576 
    76    ! TODO: Most changes and some code in this module not necessary if using XIOS (subdomain support, axis operations) 
     77   ! TEMP: Most changes and some code in this module not necessary if using XIOS (subdomain support, axis operations) 
    7778   SUBROUTINE dia_ptr( kt, Kmm, pvtr ) 
    7879      !!---------------------------------------------------------------------- 
    … …  
    8182      INTEGER                         , INTENT(in)           ::   kt     ! ocean time-step index      
    8283      INTEGER                         , INTENT(in)           ::   Kmm    ! time level index 
    83       REAL(wp), DIMENSION(jpi,jpj,jpk), INTENT(in), OPTIONAL ::   pvtr   ! j-effective transport 
     84      REAL(wp), DIMENSION(A2D,jpk)    , INTENT(in), OPTIONAL ::   pvtr   ! j-effective transport 
     85      !!---------------------------------------------------------------------- 
     86      ! 
     87      IF( ln_timing )   CALL timing_start('dia_ptr') 
     88 
     89      IF( kt == nit000 .AND. ll_init )   CALL dia_ptr_init 
     90      ! 
     91      IF( l_diaptr ) THEN 
     92         ! Calculate zonal integrals 
     93         IF( PRESENT( pvtr ) ) THEN 
     94            CALL dia_ptr_zint( Kmm, pvtr ) 
     95         ELSE 
     96            CALL dia_ptr_zint( Kmm ) 
     97         ENDIF 
     98 
     99         ! Calculate diagnostics only when zonal integrals have finished 
     100         IF( ntile == 0 .OR. ntile == nijtile ) CALL dia_ptr_iom(kt, Kmm, pvtr) 
     101      ENDIF 
     102 
     103      IF( ln_timing )   CALL timing_stop('dia_ptr') 
     104      ! 
     105   END SUBROUTINE dia_ptr 
     106 
     107 
     108   SUBROUTINE dia_ptr_iom( kt, Kmm, pvtr ) 
     109      !!---------------------------------------------------------------------- 
     110      !!                  ***  ROUTINE dia_ptr_iom  *** 
     111      !!---------------------------------------------------------------------- 
     112      !! ** Purpose : Calculate diagnostics and send to XIOS 
     113      !!---------------------------------------------------------------------- 
     114      INTEGER                         , INTENT(in)           ::   kt     ! ocean time-step index 
     115      INTEGER                         , INTENT(in)           ::   Kmm    ! time level index 
     116      REAL(wp), DIMENSION(A2D,jpk)    , INTENT(in), OPTIONAL ::   pvtr   ! j-effective transport 
    84117      ! 
    85118      INTEGER  ::   ji, jj, jk, jn   ! dummy loop indices 
    86       INTEGER  ::   itile 
    87119      REAL(wp), DIMENSION(jpi,jpj)     ::  z2d   ! 2D workspace 
    88120      REAL(wp), DIMENSION(jpj)      ::  zvsum, ztsum, zssum   ! 1D workspace 
    … …  
    95127      REAL(wp), DIMENSION(jpi,jpj,nptr)      :: z3dtr ! i-mean T and S, j-Stream-Function 
    96128      !!---------------------------------------------------------------------- 
    97       ! 
    98       IF( ln_timing )   CALL timing_start('dia_ptr') 
    99  
    100       IF( kt == nit000 .AND. ll_init )   CALL dia_ptr_init 
    101       ! 
    102       IF( .NOT. l_diaptr )   RETURN 
    103  
    104       ! Calculate zonal integrals 
    105       IF( PRESENT( pvtr ) ) THEN 
    106          CALL dia_ptr_zint( Kmm, pvtr ) 
    107       ELSE 
    108          CALL dia_ptr_zint( Kmm ) 
    109       ENDIF 
    110  
    111       ! Calculate diagnostics only when zonal integrals have finished 
    112       IF( ntile /= 0 .AND. ntile /= nijtile )  RETURN 
    113129 
    114130      IF( PRESENT( pvtr ) ) THEN 
    … …  
    155171 
    156172         IF( iom_use( 'sopstbtr' ) .OR. iom_use( 'sophtbtr' ) ) THEN 
    157             ! Calculate barotropic heat and salt transport here  
     173            ! Calculate barotropic heat and salt transport here 
    158174            DO jn = 1, nptr 
    159175               sjk(:,1,jn) = SUM( pvtr_int(:,:,jp_msk,jn), 2 ) 
    … …  
    182198            ENDDO 
    183199            CALL iom_put( 'sopstbtr', z3dtr ) 
    184          ENDIF  
     200         ENDIF 
    185201         ! 
    186202         hstr_ove(:,:,:) = 0._wp       ! Zero before next timestep 
    … …  
    217233         ! 
    218234         !                                ! Advective and diffusive heat and salt transport 
    219          IF( iom_use( 'sophtadv' ) .OR. iom_use( 'sopstadv' ) ) THEN   
    220             !  
     235         IF( iom_use( 'sophtadv' ) .OR. iom_use( 'sopstadv' ) ) THEN 
     236            ! 
    221237            DO jn = 1, nptr 
    222238               z3dtr(1,:,jn) = hstr_adv(:,jp_tem,jn) * rc_pwatt  !  (conversion in PW) 
    … …  
    235251         ENDIF 
    236252         ! 
    237          IF( iom_use( 'sophtldf' ) .OR. iom_use( 'sopstldf' ) ) THEN   
    238             !  
     253         IF( iom_use( 'sophtldf' ) .OR. iom_use( 'sopstldf' ) ) THEN 
     254            ! 
    239255            DO jn = 1, nptr 
    240256               z3dtr(1,:,jn) = hstr_ldf(:,jp_tem,jn) * rc_pwatt  !  (conversion in PW) 
    … …  
    253269         ENDIF 
    254270         ! 
    255          IF( iom_use( 'sophteiv' ) .OR. iom_use( 'sopsteiv' ) ) THEN   
    256             !  
     271         IF( iom_use( 'sophteiv' ) .OR. iom_use( 'sopsteiv' ) ) THEN 
     272            ! 
    257273            DO jn = 1, nptr 
    258274               z3dtr(1,:,jn) = hstr_eiv(:,jp_tem,jn) * rc_pwatt  !  (conversion in PW) 
    … …  
    288304         ENDIF 
    289305         ! 
    290          ! TODO: Possibly not necessary if using XIOS (if cumulative axis operations are possible) 
     306         ! TEMP: Possibly not necessary if using XIOS (if cumulative axis operations are possible) 
     307         ! TODO: NOT TESTED- hangs on iom_get_var 
    291308         IF( iom_use( 'uocetr_vsum_cumul' ) ) THEN 
    292             itile = ntile 
    293             CALL dom_tile( ntsi, ntsj, ntei, ntej, ktile = 0 )                            ! Use full domain 
     309            IF( ln_tile ) CALL dom_tile( ntsi, ntsj, ntei, ntej, ktile = 0 )         ! Use full domain 
    294310            CALL iom_get_var(  'uocetr_vsum_op', z2d ) ! get uocetr_vsum_op from xml 
    295             z2d(:,:) = ptr_ci_2d( z2d(:,:) )   
     311            z2d(:,:) = ptr_ci_2d( z2d(:,:) ) 
    296312            CALL iom_put( 'uocetr_vsum_cumul', z2d ) 
    297             IF( ntile /= itile ) CALL dom_tile( ntsi, ntsj, ntei, ntej, ktile = itile )   ! Revert to tile domain 
     313            IF( ln_tile ) CALL dom_tile( ntsi, ntsj, ntei, ntej, ktile = nijtile )   ! Revert to tile domain 
    298314         ENDIF 
    299315         ! 
    … …  
    304320         pzon_int(:,:,:,:) = 0._wp 
    305321      ENDIF 
    306       ! 
    307       IF( ln_timing )   CALL timing_stop('dia_ptr') 
    308       ! 
    309    END SUBROUTINE dia_ptr 
     322   END SUBROUTINE dia_ptr_iom 
    310323 
    311324 
    … …  
    330343      REAL(wp)                                           :: zsfc, zvfc   ! i-k surface area 
    331344      INTEGER  ::   ji, jj, jk, jn                                       ! dummy loop indices 
     345      !!---------------------------------------------------------------------- 
    332346 
    333347      IF( PRESENT( pvtr ) ) THEN 
    … …  
    442456         IF( lk_mpp )   CALL mpp_ini_znl( numout )     ! Define MPI communicator for zonal sum 
    443457 
    444          btmsk(:,:,1) = tmask_i(:,:)                  
     458         btmsk(:,:,:) = 0._wp 
     459         btmsk(:,:,1) = tmask_i(:,:) 
    445460         CALL iom_open( 'subbasins', inum,  ldstop = .FALSE.  ) 
    446461         CALL iom_get( inum, jpdom_global, 'atlmsk', btmsk(:,:,2) )   ! Atlantic basin 

  • NEMO/branches/UKMO/dev_r12866_HPC-02_Daley_Tiling_trial_extra_halo/src/OCE/DOM/domain.F90

    r12979 r13409  
    295295      !!              - nijtile        : total number of tiles 
    296296      !!---------------------------------------------------------------------- 
    297       INTEGER, INTENT(out) :: ktsi, ktsj, ktei, ktej      ! Tile domain indices 
     297      TYPE(tiledom), INTENT(out) :: ktsi, ktsj, ktei, ktej      ! Tile domain indices 
    298298      INTEGER, INTENT(in), OPTIONAL :: ktile              ! Tile number 
    299299      INTEGER ::   jt                                     ! dummy loop argument 
    … …  
    302302      IF( PRESENT(ktile) .AND. ln_tile ) THEN 
    303303         ntile = ktile                 ! Set domain indices for tile 
    304          ktsi = ntsi_a(ktile) 
    305          ktsj = ntsj_a(ktile) 
    306          ktei = ntei_a(ktile) 
    307          ktej = ntej_a(ktile) 
     304         ktsi%h0 = ntsi_a(ktile) 
     305         ktsj%h0 = ntsj_a(ktile) 
     306         ktei%h0 = ntei_a(ktile) 
     307         ktej%h0 = ntej_a(ktile) 
    308308      ELSE 
    309309         ntile = 0                     ! Initialise to full domain 
    310310         nijtile = 1 
    311          ktsi = Nis0 
    312          ktsj = Njs0 
    313          ktei = Nie0 
    314          ktej = Nje0 
     311         ktsi%h0 = Nis0 
     312         ktsj%h0 = Njs0 
     313         ktei%h0 = Nie0 
     314         ktej%h0 = Nje0 
    315315 
    316316         IF( ln_tile ) THEN            ! Calculate tile domain indices 
    … …  
    323323            ALLOCATE( ntsi_a(0:nijtile), ntsj_a(0:nijtile), ntei_a(0:nijtile), ntej_a(0:nijtile) ) 
    324324 
    325             ntsi_a(0) = ktsi                 ! Full domain 
    326             ntsj_a(0) = ktsj 
    327             ntei_a(0) = ktei 
    328             ntej_a(0) = ktej 
     325            ntsi_a(0) = ktsi%h0                 ! Full domain 
     326            ntsj_a(0) = ktsj%h0 
     327            ntei_a(0) = ktei%h0 
     328            ntej_a(0) = ktej%h0 
    329329 
    330330            DO jt = 1, nijtile               ! Tile domains 
    … …  
    353353            ELSE 
    354354               WRITE(numout,*) 'No domain tiling' 
    355                WRITE(numout,*) '    i indices =', ktsi, ':', ktei 
    356                WRITE(numout,*) '    j indices =', ktsj, ':', ktej 
     355               WRITE(numout,*) '    i indices =', ktsi%h0, ':', ktei%h0 
     356               WRITE(numout,*) '    j indices =', ktsj%h0, ':', ktej%h0 
    357357            ENDIF 
    358358         ENDIF 
    359359      ENDIF 
     360 
     361      ktsi%h1 = ktsi%h0 - 1 
     362      ktsj%h1 = ktsj%h0 - 1 
     363      ktei%h1 = ktei%h0 + 1 
     364      ktej%h1 = ktej%h0 + 1 
     365      ktsi%h2 = ktsi%h0 - nn_hls 
     366      ktsj%h2 = ktsj%h0 - nn_hls 
     367      ktei%h2 = ktei%h0 + nn_hls 
     368      ktej%h2 = ktej%h0 + nn_hls 
    360369   END SUBROUTINE dom_tile 
    361370 

  • NEMO/branches/UKMO/dev_r12866_HPC-02_Daley_Tiling_trial_extra_halo/src/OCE/DOM/domutl.F90

    r12807 r13409  
    2121   PRIVATE 
    2222 
     23   INTERFACE is_tile 
     24      MODULE PROCEDURE is_tile_2d, is_tile_3d, is_tile_4d 
     25   END INTERFACE is_tile 
     26 
    2327   PUBLIC dom_ngb    ! routine called in iom.F90 module 
    2428   PUBLIC dom_uniq   ! Called by dommsk and domwri 
     29   PUBLIC is_tile 
    2530 
    2631   !!---------------------------------------------------------------------- 
    … …  
    115120      ! 
    116121   END SUBROUTINE dom_uniq 
    117     
     122 
     123 
     124   PURE FUNCTION is_tile_2d( pt ) 
     125      !! 
     126      REAL(wp), DIMENSION(:,:), INTENT(in) ::   pt 
     127      INTEGER :: is_tile_2d 
     128      !! 
     129      IF( ln_tile .AND. SIZE(pt, 1) < jpi ) THEN 
     130         is_tile_2d = 1 
     131      ELSE 
     132         is_tile_2d = 0 
     133      ENDIF 
     134   END FUNCTION is_tile_2d 
     135 
     136 
     137   PURE FUNCTION is_tile_3d( pt ) 
     138      !! 
     139      REAL(wp), DIMENSION(:,:,:), INTENT(in) ::   pt 
     140      INTEGER :: is_tile_3d 
     141      !! 
     142      IF( ln_tile .AND. SIZE(pt, 1) < jpi ) THEN 
     143         is_tile_3d = 1 
     144      ELSE 
     145         is_tile_3d = 0 
     146      ENDIF 
     147   END FUNCTION is_tile_3d 
     148 
     149 
     150   PURE FUNCTION is_tile_4d( pt ) 
     151      !! 
     152      REAL(wp), DIMENSION(:,:,:,:), INTENT(in) ::   pt 
     153      INTEGER :: is_tile_4d 
     154      !! 
     155      IF( ln_tile .AND. SIZE(pt, 1) < jpi ) THEN 
     156         is_tile_4d = 1 
     157      ELSE 
     158         is_tile_4d = 0 
     159      ENDIF 
     160   END FUNCTION is_tile_4d 
     161 
    118162   !!====================================================================== 
    119163END MODULE domutl 

  • NEMO/branches/UKMO/dev_r12866_HPC-02_Daley_Tiling_trial_extra_halo/src/OCE/DOM/dtatsd.F90

    r12962 r13409  
    1818   USE phycst          ! physical constants 
    1919   USE dom_oce         ! ocean space and time domain 
     20   USE domain, ONLY : dom_tile 
    2021   USE fldread         ! read input fields 
    2122   ! 
    … …  
    135136      !! ** Action  :   ptsd   T-S data on medl mesh and interpolated at time-step kt 
    136137      !!---------------------------------------------------------------------- 
    137       INTEGER                              , INTENT(in   ) ::   kt     ! ocean time-step 
    138       REAL(wp), DIMENSION(jpi,jpj,jpk,jpts), INTENT(  out) ::   ptsd   ! T & S data 
     138      INTEGER                          , INTENT(in   ) ::   kt     ! ocean time-step 
     139      REAL(wp), DIMENSION(A2D,jpk,jpts), INTENT(  out) ::   ptsd   ! T & S data 
    139140      ! 
    140141      INTEGER ::   ji, jj, jk, jl, jkk   ! dummy loop indicies 
    … …  
    144145      !!---------------------------------------------------------------------- 
    145146      ! 
    146       CALL fld_read( kt, 1, sf_tsd )      !==   read T & S data at kt time step   ==! 
    147       ! 
    148       ! 
     147      ! NOTE: We can read only part of the global field (kstart, kcount in iom_get & some other mods for tiling) but the lbc_lnk in iom_get can't be overcome. Opening & reading from the file multiple times is also inefficient. Therefore, we call fld_read once for the full domain (since sf_tsd is never deallocated) and do the remaining steps for the tile domain. 
     148      IF( ntile == 0 .OR. ntile == 1 )  THEN                                         ! Do only for the full domain 
     149         IF( ln_tile ) CALL dom_tile( ntsi, ntsj, ntei, ntej, ktile = 0 )            ! Use full domain 
     150            CALL fld_read( kt, 1, sf_tsd )   !==   read T & S data at kt time step   ==! 
     151      ! 
     152      ! 
     153! NOTE: Do this for the full domain too, since it is modifying sf_tsd in-place and only for ORCA2 
    149154!!gm  This should be removed from the code   ===>>>>  T & S files has to be changed 
    150       ! 
    151       !                                   !==   ORCA_R2 configuration and T & S damping   ==!  
    152       IF( cn_cfg == "orca" .OR. cn_cfg == "ORCA" ) THEN 
    153          IF( nn_cfg == 2 .AND. ln_tsd_dmp ) THEN    ! some hand made alterations 
    154             ! 
    155             ij0 = 101 + nn_hls       ;   ij1 = 109 + nn_hls                       ! Reduced T & S in the Alboran Sea 
    156             ii0 = 141 + nn_hls - 1   ;   ii1 = 155 + nn_hls - 1 
    157             DO jj = mj0(ij0), mj1(ij1) 
    158                DO ji = mi0(ii0), mi1(ii1) 
    159                   sf_tsd(jp_tem)%fnow(ji,jj,13:13) = sf_tsd(jp_tem)%fnow(ji,jj,13:13) - 0.20_wp 
    160                   sf_tsd(jp_tem)%fnow(ji,jj,14:15) = sf_tsd(jp_tem)%fnow(ji,jj,14:15) - 0.35_wp 
    161                   sf_tsd(jp_tem)%fnow(ji,jj,16:25) = sf_tsd(jp_tem)%fnow(ji,jj,16:25) - 0.40_wp 
    162                   ! 
    163                   sf_tsd(jp_sal)%fnow(ji,jj,13:13) = sf_tsd(jp_sal)%fnow(ji,jj,13:13) - 0.15_wp 
    164                   sf_tsd(jp_sal)%fnow(ji,jj,14:15) = sf_tsd(jp_sal)%fnow(ji,jj,14:15) - 0.25_wp 
    165                   sf_tsd(jp_sal)%fnow(ji,jj,16:17) = sf_tsd(jp_sal)%fnow(ji,jj,16:17) - 0.30_wp 
    166                   sf_tsd(jp_sal)%fnow(ji,jj,18:25) = sf_tsd(jp_sal)%fnow(ji,jj,18:25) - 0.35_wp 
     155         ! 
     156         !                                   !==   ORCA_R2 configuration and T & S damping   ==! 
     157         ! TODO: NOT TESTED- requires orca2 
     158         IF( cn_cfg == "orca" .OR. cn_cfg == "ORCA" ) THEN 
     159            IF( nn_cfg == 2 .AND. ln_tsd_dmp ) THEN    ! some hand made alterations 
     160               ! 
     161               ij0 = 101 + nn_hls       ;   ij1 = 109 + nn_hls                       ! Reduced T & S in the Alboran Sea 
     162               ii0 = 141 + nn_hls - 1   ;   ii1 = 155 + nn_hls - 1 
     163               DO jj = mj0(ij0), mj1(ij1) 
     164                  DO ji = mi0(ii0), mi1(ii1) 
     165                     sf_tsd(jp_tem)%fnow(ji,jj,13:13) = sf_tsd(jp_tem)%fnow(ji,jj,13:13) - 0.20_wp 
     166                     sf_tsd(jp_tem)%fnow(ji,jj,14:15) = sf_tsd(jp_tem)%fnow(ji,jj,14:15) - 0.35_wp 
     167                     sf_tsd(jp_tem)%fnow(ji,jj,16:25) = sf_tsd(jp_tem)%fnow(ji,jj,16:25) - 0.40_wp 
     168                     ! 
     169                     sf_tsd(jp_sal)%fnow(ji,jj,13:13) = sf_tsd(jp_sal)%fnow(ji,jj,13:13) - 0.15_wp 
     170                     sf_tsd(jp_sal)%fnow(ji,jj,14:15) = sf_tsd(jp_sal)%fnow(ji,jj,14:15) - 0.25_wp 
     171                     sf_tsd(jp_sal)%fnow(ji,jj,16:17) = sf_tsd(jp_sal)%fnow(ji,jj,16:17) - 0.30_wp 
     172                     sf_tsd(jp_sal)%fnow(ji,jj,18:25) = sf_tsd(jp_sal)%fnow(ji,jj,18:25) - 0.35_wp 
     173                  END DO 
    167174               END DO 
    168             END DO 
    169             ij0 =  87 + nn_hls       ;   ij1 =  96 + nn_hls                       ! Reduced temperature in Red Sea 
    170             ii0 = 148 + nn_hls - 1   ;   ii1 = 160 + nn_hls - 1 
    171             sf_tsd(jp_tem)%fnow( mi0(ii0):mi1(ii1) , mj0(ij0):mj1(ij1) ,  4:10 ) = 7.0_wp 
    172             sf_tsd(jp_tem)%fnow( mi0(ii0):mi1(ii1) , mj0(ij0):mj1(ij1) , 11:13 ) = 6.5_wp 
    173             sf_tsd(jp_tem)%fnow( mi0(ii0):mi1(ii1) , mj0(ij0):mj1(ij1) , 14:20 ) = 6.0_wp 
    174          ENDIF 
    175       ENDIF 
     175               ij0 =  87 + nn_hls       ;   ij1 =  96 + nn_hls                       ! Reduced temperature in Red Sea 
     176               ii0 = 148 + nn_hls - 1   ;   ii1 = 160 + nn_hls - 1 
     177               sf_tsd(jp_tem)%fnow( mi0(ii0):mi1(ii1) , mj0(ij0):mj1(ij1) ,  4:10 ) = 7.0_wp 
     178               sf_tsd(jp_tem)%fnow( mi0(ii0):mi1(ii1) , mj0(ij0):mj1(ij1) , 11:13 ) = 6.5_wp 
     179               sf_tsd(jp_tem)%fnow( mi0(ii0):mi1(ii1) , mj0(ij0):mj1(ij1) , 14:20 ) = 6.0_wp 
     180            ENDIF 
     181         ENDIF 
    176182!!gm end 
    177       ! 
    178       ptsd(:,:,:,jp_tem) = sf_tsd(jp_tem)%fnow(:,:,:)    ! NO mask 
    179       ptsd(:,:,:,jp_sal) = sf_tsd(jp_sal)%fnow(:,:,:)  
     183         IF( ln_tile ) CALL dom_tile( ntsi, ntsj, ntei, ntej, ktile = 1 )            ! Revert to tile domain 
     184      ENDIF 
     185      ! 
     186      DO_3D_00_00( 1, jpk ) 
     187         ptsd(ji,jj,jk,jp_tem) = sf_tsd(jp_tem)%fnow(ji,jj,jk)    ! NO mask 
     188         ptsd(ji,jj,jk,jp_sal) = sf_tsd(jp_sal)%fnow(ji,jj,jk) 
     189      END_3D 
    180190      ! 
    181191      IF( ln_sco ) THEN                   !==   s- or mixed s-zps-coordinate   ==! 
    182192         ! 
    183          IF( kt == nit000 .AND. lwp )THEN 
    184             WRITE(numout,*) 
    185             WRITE(numout,*) 'dta_tsd: interpolates T & S data onto the s- or mixed s-z-coordinate mesh' 
     193         IF( ntile == 0 .OR. ntile == 1 )  THEN                       ! Do only on the first tile 
     194            IF( kt == nit000 .AND. lwp )THEN 
     195               WRITE(numout,*) 
     196               WRITE(numout,*) 'dta_tsd: interpolates T & S data onto the s- or mixed s-z-coordinate mesh' 
     197            ENDIF 
    186198         ENDIF 
    187199         ! 
    … …  
    215227      ELSE                                !==   z- or zps- coordinate   ==! 
    216228         !                              
    217          ptsd(:,:,:,jp_tem) = ptsd(:,:,:,jp_tem) * tmask(:,:,:)    ! Mask 
    218          ptsd(:,:,:,jp_sal) = ptsd(:,:,:,jp_sal) * tmask(:,:,:) 
    219          ! 
     229         DO_3D_00_00( 1, jpk ) 
     230            ptsd(ji,jj,jk,jp_tem) = ptsd(ji,jj,jk,jp_tem) * tmask(ji,jj,jk)    ! Mask 
     231            ptsd(ji,jj,jk,jp_sal) = ptsd(ji,jj,jk,jp_sal) * tmask(ji,jj,jk) 
     232         END_3D 
     233         ! 
     234         ! TODO: NOT TESTED- requires zps 
    220235         IF( ln_zps ) THEN                      ! zps-coordinate (partial steps) interpolation at the last ocean level 
    221236            DO_2D_11_11 

  • NEMO/branches/UKMO/dev_r12866_HPC-02_Daley_Tiling_trial_extra_halo/src/OCE/DYN/dynhpg.F90

    r12377 r13409  
    300300      INTEGER  ::   iku, ikv                         ! temporary integers 
    301301      REAL(wp) ::   zcoef0, zcoef1, zcoef2, zcoef3   ! temporary scalars 
    302       REAL(wp), DIMENSION(jpi,jpj,jpk) ::  zhpi, zhpj 
    303       REAL(wp), DIMENSION(jpi,jpj) :: zgtsu, zgtsv, zgru, zgrv 
     302      REAL(wp), DIMENSION(jpi,jpj,jpk) :: zhpi, zhpj 
     303      REAL(wp), DIMENSION(jpi,jpj,1)   :: zgtsu, zgtsv 
     304      REAL(wp), DIMENSION(jpi,jpj)     :: zgru, zgrv 
    304305      !!---------------------------------------------------------------------- 
    305306      ! 

  • NEMO/branches/UKMO/dev_r12866_HPC-02_Daley_Tiling_trial_extra_halo/src/OCE/IOM/prtctl.F90

    r12942 r13409  
    4242 
    4343   SUBROUTINE prt_ctl (tab2d_1, tab3d_1, mask1, clinfo1, tab2d_2, tab3d_2,   & 
    44       &                                  mask2, clinfo2, kdim, clinfo3 ) 
     44      &                                  mask2, clinfo2, kdim, clinfo3, psum1, psum2 ) 
    4545      !!---------------------------------------------------------------------- 
    4646      !!                     ***  ROUTINE prt_ctl  *** 
    … …  
    7575      !!                    clinfo2 : information about the tab[23]d_2 array 
    7676      !!                    kdim    : k- direction for 3D arrays  
    77       !!                    clinfo3 : additional information  
     77      !!                    clinfo3 : additional information 
     78      !!                    psum1   : sum of tab[23]d_1 over tiles (required if ln_tile=T) 
     79      !!                    psum2   : sum of tab[23]d_2 over tiles (required if ln_tile=T) 
    7880      !!---------------------------------------------------------------------- 
    7981      REAL(wp), DIMENSION(:,:)  , INTENT(in), OPTIONAL ::   tab2d_1 
    … …  
    8789      INTEGER                   , INTENT(in), OPTIONAL ::   kdim 
    8890      CHARACTER (len=*)         , INTENT(in), OPTIONAL ::   clinfo3 
     91      REAL(wp)                  , INTENT(inout), OPTIONAL :: psum1, psum2 
    8992      ! 
    9093      CHARACTER (len=15) :: cl2 
    9194      INTEGER ::  jn, sind, eind, kdir,j_id 
     95      INTEGER :: iictls, iictle, ijctls, ijctle 
    9296      REAL(wp) :: zsum1, zsum2, zvctl1, zvctl2 
    9397      REAL(wp), DIMENSION(jpi,jpj)     :: ztab2d_1, ztab2d_2 
    9498      REAL(wp), DIMENSION(jpi,jpj,jpk) :: zmask1, zmask2, ztab3d_1, ztab3d_2 
    9599      !!---------------------------------------------------------------------- 
    96       ! TODO: TO BE TILED 
    97       IF( ntile /= 0 .AND. ntile /= nijtile ) RETURN 
    98100 
    99101      ! Arrays, scalars initialization  
    … …  
    120122      IF( PRESENT(mask1)   )   zmask1  (:,:,:)      = mask1  (:,:,:) 
    121123      IF( PRESENT(mask2)   )   zmask2  (:,:,:)      = mask2  (:,:,:) 
     124 
     125      IF( ntile == 0 .OR. ntile == 1 ) THEN   ! Do only on the first tile 
     126         IF( PRESENT(psum1) ) psum1 = 0.e0 
     127         IF( PRESENT(psum2) ) psum2 = 0.e0 
     128      ENDIF 
     129 
     130      ! psum1 and psum2 must be used to sum over tiles 
     131      IF( ntile /= 0 ) THEN 
     132         IF( (PRESENT(tab2d_1) .OR. PRESENT(tab3d_1)) .AND. .NOT. PRESENT(psum1) ) & 
     133            & CALL ctl_stop('prt_ctl must be called with psum1 when ln_tile = T (clinfo1 = "'//TRIM(clinfo1)//'")') 
     134         IF( (PRESENT(tab2d_2) .OR. PRESENT(tab3d_2)) .AND. .NOT. PRESENT(psum2) ) & 
     135            & CALL ctl_stop('prt_ctl must be called with psum2 when ln_tile = T (clinfo2 = "'//TRIM(cl2)//'")') 
     136      ENDIF 
    122137 
    123138      IF( lk_mpp .AND. jpnij > 1 ) THEN       ! processor number 
    … …  
    158173         ENDIF 
    159174 
    160          IF( PRESENT(clinfo3)) THEN 
    161             IF ( clinfo3 == 'tra' )  THEN 
    162                zvctl1 = t_ctll(jn) 
    163                zvctl2 = s_ctll(jn) 
    164             ELSEIF ( clinfo3 == 'dyn' )   THEN 
    165                zvctl1 = u_ctll(jn) 
    166                zvctl2 = v_ctll(jn) 
     175         IF( ntile == 0 .OR. ntile == nijtile ) THEN                       ! Do only on the last tile 
     176            IF( PRESENT(clinfo3)) THEN 
     177               IF ( clinfo3 == 'tra' )  THEN 
     178                  zvctl1 = t_ctll(jn) 
     179                  zvctl2 = s_ctll(jn) 
     180               ELSEIF ( clinfo3 == 'dyn' )   THEN 
     181                  zvctl1 = u_ctll(jn) 
     182                  zvctl2 = v_ctll(jn) 
     183               ENDIF 
    167184            ENDIF 
    168185         ENDIF 
    169186 
    170          ! Compute the sum control 
    171          ! 2D arrays 
    172          IF( PRESENT(tab2d_1) )   THEN 
    173             zsum1 = SUM( ztab2d_1(nictls:nictle,njctls:njctle)*zmask1(nictls:nictle,njctls:njctle,1) ) 
    174             zsum2 = SUM( ztab2d_2(nictls:nictle,njctls:njctle)*zmask2(nictls:nictle,njctls:njctle,1) ) 
     187         iictls = MAX( nictls, ntsi%h0 ) 
     188         iictle = MIN( nictle, ntei%h0 ) 
     189         ijctls = MAX( njctls, ntsj%h0 ) 
     190         ijctle = MIN( njctle, ntej%h0 ) 
     191 
     192         ! Compute the sum control only where the tile domain and control print area overlap 
     193         IF( iictle >= iictls .AND. ijctle >= ijctls ) THEN 
     194            ! 2D arrays 
     195            IF( PRESENT(tab2d_1) )   THEN 
     196               zsum1 = SUM( ztab2d_1(iictls:iictle,ijctls:ijctle)*zmask1(iictls:iictle,ijctls:ijctle,1) ) 
     197               zsum2 = SUM( ztab2d_2(iictls:iictle,ijctls:ijctle)*zmask2(iictls:iictle,ijctls:ijctle,1) ) 
     198            ENDIF 
     199 
     200            ! 3D arrays 
     201            IF( PRESENT(tab3d_1) )   THEN 
     202               zsum1 = SUM( ztab3d_1(iictls:iictle,ijctls:ijctle,1:kdir)*zmask1(iictls:iictle,ijctls:ijctle,1:kdir) ) 
     203               zsum2 = SUM( ztab3d_2(iictls:iictle,ijctls:ijctle,1:kdir)*zmask2(iictls:iictle,ijctls:ijctle,1:kdir) ) 
     204            ENDIF 
     205 
     206            ! Sum over tiles 
     207            IF( ntile == 0 .OR. ntile == nijtile ) THEN                    ! Do only on the last tile 
     208               IF( PRESENT(psum1) ) zsum1 = psum1 + zsum1 
     209               IF( PRESENT(psum2) ) zsum2 = psum2 + zsum2 
     210            ELSE 
     211               IF( PRESENT(psum1) ) psum1 = psum1 + zsum1 
     212               IF( PRESENT(psum2) ) psum2 = psum2 + zsum2 
     213            ENDIF 
    175214         ENDIF 
    176215 
    177          ! 3D arrays 
    178          IF( PRESENT(tab3d_1) )   THEN 
    179             zsum1 = SUM( ztab3d_1(nictls:nictle,njctls:njctle,1:kdir)*zmask1(nictls:nictle,njctls:njctle,1:kdir) ) 
    180             zsum2 = SUM( ztab3d_2(nictls:nictle,njctls:njctle,1:kdir)*zmask2(nictls:nictle,njctls:njctle,1:kdir) ) 
    181          ENDIF 
    182  
    183216         ! Print the result 
    184          IF( PRESENT(clinfo3) )   THEN 
    185             WRITE(j_id,FMT='(a,D23.16,3x,a,D23.16)')clinfo1, zsum1-zvctl1, cl2, zsum2-zvctl2 
    186             SELECT CASE( clinfo3 ) 
    187             CASE ( 'tra-ta' )  
    188                t_ctll(jn) = zsum1 
    189             CASE ( 'tra' )  
    190                 t_ctll(jn) = zsum1 
    191                 s_ctll(jn) = zsum2 
    192             CASE ( 'dyn' )  
    193                 u_ctll(jn) = zsum1 
    194                 v_ctll(jn) = zsum2  
    195             END SELECT 
    196          ELSEIF ( PRESENT(clinfo2) .OR. PRESENT(tab2d_2) .OR. PRESENT(tab3d_2) )   THEN 
    197             WRITE(j_id,FMT='(a,D23.16,3x,a,D23.16)')clinfo1, zsum1, cl2, zsum2 
    198          ELSE 
    199             WRITE(j_id,FMT='(a,D23.16)')clinfo1, zsum1 
     217         IF( ntile == 0 .OR. ntile == nijtile ) THEN                       ! Do only on the last tile 
     218            IF( PRESENT(clinfo3) )   THEN 
     219               WRITE(j_id,FMT='(a,D23.16,3x,a,D23.16)')clinfo1, zsum1-zvctl1, cl2, zsum2-zvctl2 
     220               SELECT CASE( clinfo3 ) 
     221               CASE ( 'tra-ta' ) 
     222                  t_ctll(jn) = zsum1 
     223               CASE ( 'tra' ) 
     224                   t_ctll(jn) = zsum1 
     225                   s_ctll(jn) = zsum2 
     226               CASE ( 'dyn' ) 
     227                   u_ctll(jn) = zsum1 
     228                   v_ctll(jn) = zsum2 
     229               END SELECT 
     230            ELSEIF ( PRESENT(clinfo2) .OR. PRESENT(tab2d_2) .OR. PRESENT(tab3d_2) )   THEN 
     231               WRITE(j_id,FMT='(a,D23.16,3x,a,D23.16)')clinfo1, zsum1, cl2, zsum2 
     232            ELSE 
     233               WRITE(j_id,FMT='(a,D23.16)')clinfo1, zsum1 
     234            ENDIF 
    200235         ENDIF 
    201236 

  • NEMO/branches/UKMO/dev_r12866_HPC-02_Daley_Tiling_trial_extra_halo/src/OCE/LDF/ldftra.F90

    r12738 r13409  
    725725      !! ** Action  : pu, pv increased by the eiv transport 
    726726      !!---------------------------------------------------------------------- 
    727       INTEGER                         , INTENT(in   ) ::   kt        ! ocean time-step index 
    728       INTEGER                         , INTENT(in   ) ::   kit000    ! first time step index 
    729       INTEGER                         , INTENT(in   ) ::   Kmm, Krhs ! ocean time level indices 
    730       CHARACTER(len=3)                , INTENT(in   ) ::   cdtype    ! =TRA or TRC (tracer indicator) 
    731       REAL(wp), DIMENSION(jpi,jpj,jpk), INTENT(inout) ::   pu      ! in : 3 ocean transport components   [m3/s] 
    732       REAL(wp), DIMENSION(jpi,jpj,jpk), INTENT(inout) ::   pv      ! out: 3 ocean transport components   [m3/s] 
    733       REAL(wp), DIMENSION(jpi,jpj,jpk), INTENT(inout) ::   pw      ! increased by the eiv                [m3/s] 
     727      INTEGER                     , INTENT(in   ) ::   kt        ! ocean time-step index 
     728      INTEGER                     , INTENT(in   ) ::   kit000    ! first time step index 
     729      INTEGER                     , INTENT(in   ) ::   Kmm, Krhs ! ocean time level indices 
     730      CHARACTER(len=3)            , INTENT(in   ) ::   cdtype    ! =TRA or TRC (tracer indicator) 
     731      REAL(wp), DIMENSION(A2D,jpk), INTENT(inout) ::   pu        ! in : 3 ocean transport components   [m3/s] 
     732      REAL(wp), DIMENSION(A2D,jpk), INTENT(inout) ::   pv        ! out: 3 ocean transport components   [m3/s] 
     733      REAL(wp), DIMENSION(A2D,jpk), INTENT(inout) ::   pw        ! increased by the eiv                [m3/s] 
    734734      !! 
    735735      INTEGER  ::   ji, jj, jk                 ! dummy loop indices 
    736736      REAL(wp) ::   zuwk, zuwk1, zuwi, zuwi1   ! local scalars 
    737737      REAL(wp) ::   zvwk, zvwk1, zvwj, zvwj1   !   -      - 
    738       REAL(wp), DIMENSION(jpi,jpj,jpk) ::   zpsi_uw, zpsi_vw 
    739       !!---------------------------------------------------------------------- 
    740       ! 
    741       IF( kt == kit000 )  THEN 
    742          IF(lwp) WRITE(numout,*) 
    743          IF(lwp) WRITE(numout,*) 'ldf_eiv_trp : eddy induced advection on ', cdtype,' :' 
    744          IF(lwp) WRITE(numout,*) '~~~~~~~~~~~   add to velocity fields the eiv component' 
     738      REAL(wp), DIMENSION(A2D,jpk) ::   zpsi_uw, zpsi_vw 
     739      !!---------------------------------------------------------------------- 
     740      ! 
     741      IF( ntile == 0 .OR. ntile == 1 )  THEN                       ! Do only on the first tile 
     742         IF( kt == kit000 )  THEN 
     743            IF(lwp) WRITE(numout,*) 
     744            IF(lwp) WRITE(numout,*) 'ldf_eiv_trp : eddy induced advection on ', cdtype,' :' 
     745            IF(lwp) WRITE(numout,*) '~~~~~~~~~~~   add to velocity fields the eiv component' 
     746         ENDIF 
    745747      ENDIF 
    746748 
    … …  
    781783      !! 
    782784      !!---------------------------------------------------------------------- 
    783       REAL(wp), DIMENSION(jpi,jpj,jpk), INTENT(inout) ::   psi_uw, psi_vw   ! streamfunction   [m3/s] 
    784       INTEGER                         , INTENT(in   ) ::   Kmm   ! ocean time level indices 
     785      REAL(wp), DIMENSION(A2D,jpk), INTENT(inout) ::   psi_uw, psi_vw   ! streamfunction   [m3/s] 
     786      INTEGER                     , INTENT(in   ) ::   Kmm   ! ocean time level indices 
    785787      ! 
    786788      INTEGER  ::   ji, jj, jk    ! dummy loop indices 
    787789      REAL(wp) ::   zztmp   ! local scalar 
    788       REAL(wp), DIMENSION(jpi,jpj)     ::   zw2d   ! 2D workspace 
    789       REAL(wp), DIMENSION(jpi,jpj,jpk) ::   zw3d   ! 3D workspace 
     790      REAL(wp), DIMENSION(A2D)     ::   zw2d   ! 2D workspace 
     791      REAL(wp), DIMENSION(A2D,jpk) ::   zw3d   ! 3D workspace 
    790792      !!---------------------------------------------------------------------- 
    791793      ! 
    … …  
    802804      zw3d(:,:,jpk) = 0._wp                                    ! bottom value always 0 
    803805      ! 
    804       DO jk = 1, jpkm1                                         ! e2u e3u u_eiv = -dk[psi_uw] 
    805          zw3d(:,:,jk) = ( psi_uw(:,:,jk+1) - psi_uw(:,:,jk) ) / ( e2u(:,:) * e3u(:,:,jk,Kmm) ) 
    806       END DO 
     806      DO_3D_11_11( 1, jpkm1 )                                  ! e2u e3u u_eiv = -dk[psi_uw] 
     807         zw3d(ji,jj,jk) = ( psi_uw(ji,jj,jk+1) - psi_uw(ji,jj,jk) ) / ( e2u(ji,jj) * e3u(ji,jj,jk,Kmm) ) 
     808      END_3D 
    807809      CALL iom_put( "uoce_eiv", zw3d ) 
    808810      ! 
    809       DO jk = 1, jpkm1                                         ! e1v e3v v_eiv = -dk[psi_vw] 
    810          zw3d(:,:,jk) = ( psi_vw(:,:,jk+1) - psi_vw(:,:,jk) ) / ( e1v(:,:) * e3v(:,:,jk,Kmm) ) 
    811       END DO 
     811      DO_3D_11_11( 1, jpkm1 )                                  ! e1v e3v v_eiv = -dk[psi_vw] 
     812         zw3d(ji,jj,jk) = ( psi_vw(ji,jj,jk+1) - psi_vw(ji,jj,jk) ) / ( e1v(ji,jj) * e3v(ji,jj,jk,Kmm) ) 
     813      END_3D 
    812814      CALL iom_put( "voce_eiv", zw3d ) 
    813815      ! 
    … …  
    820822      ! 
    821823      IF( iom_use('weiv_masstr') ) THEN   ! vertical mass transport & its square value 
    822          zw2d(:,:) = rho0 * e1e2t(:,:) 
     824         DO_2D_11_11 
     825            zw2d(ji,jj) = rho0 * e1e2t(ji,jj) 
     826         END_2D 
    823827         DO jk = 1, jpk 
    824828            zw3d(:,:,jk) = zw3d(:,:,jk) * zw2d(:,:) 
    … …  
    866870      END_3D 
    867871      CALL lbc_lnk( 'ldftra', zw2d, 'V', -1. ) 
    868       CALL iom_put( "veiv_heattr", zztmp * zw2d )                  !  heat transport in j-direction 
    869       CALL iom_put( "veiv_heattr", zztmp * zw3d )                  !  heat transport in j-direction 
     872      CALL lbc_lnk( 'ldftra', zw3d, 'V', -1. ) 
     873      CALL iom_put( "veiv_heattr"  , zztmp * zw2d )                  !  heat transport in j-direction 
     874      CALL iom_put( "veiv_heattr3d", zztmp * zw3d )                  !  heat transport in j-direction 
    870875      ! 
    871876      IF( iom_use( 'sophteiv' ) )   CALL dia_ptr_hst( jp_tem, 'eiv', 0.5 * zw3d ) 
    … …  
    893898      END_3D 
    894899      CALL lbc_lnk( 'ldftra', zw2d, 'V', -1. ) 
    895       CALL iom_put( "veiv_salttr", zztmp * zw2d )                  !  salt transport in j-direction 
    896       CALL iom_put( "veiv_salttr", zztmp * zw3d )                  !  salt transport in j-direction 
     900      CALL lbc_lnk( 'ldftra', zw3d, 'V', -1. ) 
     901      CALL iom_put( "veiv_salttr"  , zztmp * zw2d )                  !  salt transport in j-direction 
     902      CALL iom_put( "veiv_salttr3d", zztmp * zw3d )                  !  salt transport in j-direction 
    897903      ! 
    898904      IF( iom_use( 'sopsteiv' ) ) CALL dia_ptr_hst( jp_sal, 'eiv', 0.5 * zw3d ) 

  • NEMO/branches/UKMO/dev_r12866_HPC-02_Daley_Tiling_trial_extra_halo/src/OCE/TRA/eosbn2.F90

    r12489 r13409  
    3939   !!---------------------------------------------------------------------- 
    4040   USE dom_oce        ! ocean space and time domain 
     41   USE domutl, ONLY : is_tile 
    4142   USE phycst         ! physical constants 
    4243   USE stopar         ! Stochastic T/S fluctuations 
    … …  
    186187   !!---------------------------------------------------------------------- 
    187188CONTAINS 
    188  
    189189   SUBROUTINE eos_insitu( pts, prd, pdep ) 
     190      !! 
     191      REAL(wp), DIMENSION(:,:,:,:), INTENT(in   ) ::   pts   ! 1 : potential temperature  [Celsius] 
     192      !                                                      ! 2 : salinity               [psu] 
     193      REAL(wp), DIMENSION(:,:,:)  , INTENT(  out) ::   prd   ! in situ density            [-] 
     194      REAL(wp), DIMENSION(:,:,:)  , INTENT(in   ) ::   pdep  ! depth                      [m] 
     195      !! 
     196      CALL eos_insitu_t( pts, is_tile(pts), prd, is_tile(prd), pdep, is_tile(pdep) ) 
     197   END SUBROUTINE eos_insitu 
     198 
     199   SUBROUTINE eos_insitu_t( pts, ktts, prd, ktrd, pdep, ktdep ) 
    190200      !!---------------------------------------------------------------------- 
    191201      !!                   ***  ROUTINE eos_insitu  *** 
    … …  
    221231      !!                TEOS-10 Manual, 2010 
    222232      !!---------------------------------------------------------------------- 
    223       REAL(wp), DIMENSION(jpi,jpj,jpk,jpts), INTENT(in   ) ::   pts   ! 1 : potential temperature  [Celsius] 
    224       !                                                               ! 2 : salinity               [psu] 
    225       REAL(wp), DIMENSION(jpi,jpj,jpk     ), INTENT(  out) ::   prd   ! in situ density            [-] 
    226       REAL(wp), DIMENSION(jpi,jpj,jpk     ), INTENT(in   ) ::   pdep  ! depth                      [m] 
     233      INTEGER                                 , INTENT(in   ) ::   ktts, ktrd, ktdep 
     234      REAL(wp), DIMENSION(T2D(ktts) ,jpk,jpts), INTENT(in   ) ::   pts   ! 1 : potential temperature  [Celsius] 
     235      !                                                                  ! 2 : salinity               [psu] 
     236      REAL(wp), DIMENSION(T2D(ktrd) ,jpk     ), INTENT(  out) ::   prd   ! in situ density            [-] 
     237      REAL(wp), DIMENSION(T2D(ktdep),jpk     ), INTENT(in   ) ::   pdep  ! depth                      [m] 
    227238      ! 
    228239      INTEGER  ::   ji, jj, jk                ! dummy loop indices 
    229240      REAL(wp) ::   zt , zh , zs , ztm        ! local scalars 
    230241      REAL(wp) ::   zn , zn0, zn1, zn2, zn3   !   -      - 
     242      REAL(wp), SAVE :: zsum1 
    231243      !!---------------------------------------------------------------------- 
    232244      ! 
    … …  
    288300      END SELECT 
    289301      ! 
    290       IF(sn_cfctl%l_prtctl)   CALL prt_ctl( tab3d_1=prd, clinfo1=' eos-insitu  : ', kdim=jpk ) 
     302      IF(sn_cfctl%l_prtctl)   CALL prt_ctl( tab3d_1=prd, clinfo1=' eos-insitu  : ', kdim=jpk, psum1=zsum1 ) 
    291303      ! 
    292304      IF( ln_timing )   CALL timing_stop('eos-insitu') 
    293305      ! 
    294    END SUBROUTINE eos_insitu 
     306   END SUBROUTINE eos_insitu_t 
    295307 
    296308 
    297309   SUBROUTINE eos_insitu_pot( pts, prd, prhop, pdep ) 
     310      !! 
     311      REAL(wp), DIMENSION(:,:,:,:), INTENT(in   ) ::   pts    ! 1 : potential temperature  [Celsius] 
     312      !                                                       ! 2 : salinity               [psu] 
     313      REAL(wp), DIMENSION(:,:,:)  , INTENT(  out) ::   prd    ! in situ density            [-] 
     314      REAL(wp), DIMENSION(:,:,:)  , INTENT(  out) ::   prhop  ! potential density (surface referenced) 
     315      REAL(wp), DIMENSION(:,:,:)  , INTENT(in   ) ::   pdep   ! depth                      [m] 
     316      !! 
     317      CALL eos_insitu_pot_t( pts, is_tile(pts), prd, is_tile(prd), prhop, is_tile(prhop), pdep, is_tile(pdep) ) 
     318   END SUBROUTINE eos_insitu_pot 
     319 
     320 
     321   SUBROUTINE eos_insitu_pot_t( pts, ktts, prd, ktrd, prhop, ktrhop, pdep, ktdep ) 
    298322      !!---------------------------------------------------------------------- 
    299323      !!                  ***  ROUTINE eos_insitu_pot  *** 
    … …  
    308332      !! 
    309333      !!---------------------------------------------------------------------- 
    310       REAL(wp), DIMENSION(jpi,jpj,jpk,jpts), INTENT(in   ) ::   pts    ! 1 : potential temperature  [Celsius] 
    311       !                                                                ! 2 : salinity               [psu] 
    312       REAL(wp), DIMENSION(jpi,jpj,jpk     ), INTENT(  out) ::   prd    ! in situ density            [-] 
    313       REAL(wp), DIMENSION(jpi,jpj,jpk     ), INTENT(  out) ::   prhop  ! potential density (surface referenced) 
    314       REAL(wp), DIMENSION(jpi,jpj,jpk     ), INTENT(in   ) ::   pdep   ! depth                      [m] 
     334      INTEGER                                  , INTENT(in   ) ::   ktts, ktrd, ktrhop, ktdep 
     335      REAL(wp), DIMENSION(T2D(ktts)  ,jpk,jpts), INTENT(in   ) ::   pts    ! 1 : potential temperature  [Celsius] 
     336      !                                                                    ! 2 : salinity               [psu] 
     337      REAL(wp), DIMENSION(T2D(ktrd)  ,jpk     ), INTENT(  out) ::   prd    ! in situ density            [-] 
     338      REAL(wp), DIMENSION(T2D(ktrhop),jpk     ), INTENT(  out) ::   prhop  ! potential density (surface referenced) 
     339      REAL(wp), DIMENSION(T2D(ktdep) ,jpk     ), INTENT(in   ) ::   pdep   ! depth                      [m] 
    315340      ! 
    316341      INTEGER  ::   ji, jj, jk, jsmp             ! dummy loop indices 
    … …  
    318343      REAL(wp) ::   zt , zh , zstemp, zs , ztm   ! local scalars 
    319344      REAL(wp) ::   zn , zn0, zn1, zn2, zn3      !   -      - 
     345      REAL(wp), SAVE :: zsum1, zsum2 
    320346      REAL(wp), DIMENSION(:), ALLOCATABLE :: zn0_sto, zn_sto, zsign    ! local vectors 
    321347      !!---------------------------------------------------------------------- 
    … …  
    443469      END SELECT 
    444470      ! 
    445       IF(sn_cfctl%l_prtctl)   CALL prt_ctl( tab3d_1=prd, clinfo1=' eos-pot: ', tab3d_2=prhop, clinfo2=' pot : ', kdim=jpk ) 
     471      IF(sn_cfctl%l_prtctl)   CALL prt_ctl( tab3d_1=prd, clinfo1=' eos-pot: ', psum1=zsum1, & 
     472         &                                  tab3d_2=prhop, clinfo2=' pot : ', psum2=zsum2, kdim=jpk ) 
    446473      ! 
    447474      IF( ln_timing )   CALL timing_stop('eos-pot') 
    448475      ! 
    449    END SUBROUTINE eos_insitu_pot 
     476   END SUBROUTINE eos_insitu_pot_t 
    450477 
    451478 
    452479   SUBROUTINE eos_insitu_2d( pts, pdep, prd ) 
     480      !! 
     481      REAL(wp), DIMENSION(:,:,:), INTENT(in   ) ::   pts   ! 1 : potential temperature  [Celsius] 
     482      !                                                    ! 2 : salinity               [psu] 
     483      REAL(wp), DIMENSION(:,:)  , INTENT(in   ) ::   pdep  ! depth                      [m] 
     484      REAL(wp), DIMENSION(:,:)  , INTENT(  out) ::   prd   ! in situ density 
     485      !! 
     486      CALL eos_insitu_2d_t( pts, is_tile(pts), pdep, is_tile(pdep), prd, is_tile(prd) ) 
     487   END SUBROUTINE eos_insitu_2d 
     488 
     489 
     490   SUBROUTINE eos_insitu_2d_t( pts, ktts, pdep, ktdep, prd, ktrd ) 
    453491      !!---------------------------------------------------------------------- 
    454492      !!                  ***  ROUTINE eos_insitu_2d  *** 
    … …  
    461499      !! 
    462500      !!---------------------------------------------------------------------- 
    463       REAL(wp), DIMENSION(jpi,jpj,jpts), INTENT(in   ) ::   pts   ! 1 : potential temperature  [Celsius] 
    464       !                                                           ! 2 : salinity               [psu] 
    465       REAL(wp), DIMENSION(jpi,jpj)     , INTENT(in   ) ::   pdep  ! depth                      [m] 
    466       REAL(wp), DIMENSION(jpi,jpj)     , INTENT(  out) ::   prd   ! in situ density 
     501      INTEGER                            , INTENT(in   ) ::   ktts, ktdep, ktrd 
     502      REAL(wp), DIMENSION(T2D(ktts),jpts), INTENT(in   ) ::   pts   ! 1 : potential temperature  [Celsius] 
     503      !                                                             ! 2 : salinity               [psu] 
     504      REAL(wp), DIMENSION(T2D(ktdep)    ), INTENT(in   ) ::   pdep  ! depth                      [m] 
     505      REAL(wp), DIMENSION(T2D(ktrd)     ), INTENT(  out) ::   prd   ! in situ density 
    467506      ! 
    468507      INTEGER  ::   ji, jj, jk                ! dummy loop indices 
    469508      REAL(wp) ::   zt , zh , zs              ! local scalars 
    470509      REAL(wp) ::   zn , zn0, zn1, zn2, zn3   !   -      - 
     510      REAL(wp), SAVE :: zsum1 
    471511      !!---------------------------------------------------------------------- 
    472512      ! 
    … …  
    530570      END SELECT 
    531571      ! 
    532       IF(sn_cfctl%l_prtctl)   CALL prt_ctl( tab2d_1=prd, clinfo1=' eos2d: ' ) 
     572      IF(sn_cfctl%l_prtctl)   CALL prt_ctl( tab2d_1=prd, clinfo1=' eos2d: ', psum1=zsum1 ) 
    533573      ! 
    534574      IF( ln_timing )   CALL timing_stop('eos2d') 
    535575      ! 
    536    END SUBROUTINE eos_insitu_2d 
     576   END SUBROUTINE eos_insitu_2d_t 
    537577 
    538578 
    539579   SUBROUTINE rab_3d( pts, pab, Kmm ) 
     580      !! 
     581      INTEGER                     , INTENT(in   ) ::   Kmm   ! time level index 
     582      REAL(wp), DIMENSION(:,:,:,:), INTENT(in   ) ::   pts   ! pot. temperature & salinity 
     583      REAL(wp), DIMENSION(:,:,:,:), INTENT(  out) ::   pab   ! thermal/haline expansion ratio 
     584      !! 
     585      CALL rab_3d_t( pts, is_tile(pts), pab, is_tile(pab), Kmm ) 
     586   END SUBROUTINE rab_3d 
     587 
     588 
     589   SUBROUTINE rab_3d_t( pts, ktts, pab, ktab, Kmm ) 
    540590      !!---------------------------------------------------------------------- 
    541591      !!                 ***  ROUTINE rab_3d  *** 
    … …  
    547597      !! ** Action  : - pab     : thermal/haline expansion ratio at T-points 
    548598      !!---------------------------------------------------------------------- 
    549       INTEGER                              , INTENT(in   ) ::   Kmm   ! time level index 
    550       REAL(wp), DIMENSION(jpi,jpj,jpk,jpts), INTENT(in   ) ::   pts   ! pot. temperature & salinity 
    551       REAL(wp), DIMENSION(jpi,jpj,jpk,jpts), INTENT(  out) ::   pab   ! thermal/haline expansion ratio 
     599      INTEGER                                , INTENT(in   ) ::   Kmm   ! time level index 
     600      INTEGER                                , INTENT(in   ) ::   ktts, ktab 
     601      REAL(wp), DIMENSION(T2D(ktts),jpk,jpts), INTENT(in   ) ::   pts   ! pot. temperature & salinity 
     602      REAL(wp), DIMENSION(T2D(ktab),jpk,jpts), INTENT(  out) ::   pab   ! thermal/haline expansion ratio 
    552603      ! 
    553604      INTEGER  ::   ji, jj, jk                ! dummy loop indices 
    554605      REAL(wp) ::   zt , zh , zs , ztm        ! local scalars 
    555606      REAL(wp) ::   zn , zn0, zn1, zn2, zn3   !   -      - 
     607      REAL(wp), SAVE :: zsum1, zsum2 
    556608      !!---------------------------------------------------------------------- 
    557609      ! 
    … …  
    635687      END SELECT 
    636688      ! 
    637       IF(sn_cfctl%l_prtctl)   CALL prt_ctl( tab3d_1=pab(:,:,:,jp_tem), clinfo1=' rab_3d_t: ', & 
    638          &                                  tab3d_2=pab(:,:,:,jp_sal), clinfo2=' rab_3d_s : ', kdim=jpk ) 
     689      IF(sn_cfctl%l_prtctl)   CALL prt_ctl( tab3d_1=pab(:,:,:,jp_tem), clinfo1=' rab_3d_t: ', psum1=zsum1, & 
     690         &                                  tab3d_2=pab(:,:,:,jp_sal), clinfo2=' rab_3d_s : ', psum2=zsum2, kdim=jpk ) 
    639691      ! 
    640692      IF( ln_timing )   CALL timing_stop('rab_3d') 
    641693      ! 
    642    END SUBROUTINE rab_3d 
     694   END SUBROUTINE rab_3d_t 
    643695 
    644696 
    645697   SUBROUTINE rab_2d( pts, pdep, pab, Kmm ) 
     698      !! 
     699      INTEGER                   , INTENT(in   ) ::   Kmm   ! time level index 
     700      REAL(wp), DIMENSION(:,:,:), INTENT(in   ) ::   pts    ! pot. temperature & salinity 
     701      REAL(wp), DIMENSION(:,:)  , INTENT(in   ) ::   pdep   ! depth                  [m] 
     702      REAL(wp), DIMENSION(:,:,:), INTENT(  out) ::   pab    ! thermal/haline expansion ratio 
     703      !! 
     704      CALL rab_2d_t(pts, is_tile(pts), pdep, is_tile(pdep), pab, is_tile(pab), Kmm) 
     705   END SUBROUTINE rab_2d 
     706 
     707 
     708   SUBROUTINE rab_2d_t( pts, ktts, pdep, ktdep, pab, ktab, Kmm ) 
    646709      !!---------------------------------------------------------------------- 
    647710      !!                 ***  ROUTINE rab_2d  *** 
    … …  
    651714      !! ** Action  : - pab     : thermal/haline expansion ratio at T-points 
    652715      !!---------------------------------------------------------------------- 
    653       INTEGER                              , INTENT(in   ) ::   Kmm   ! time level index 
    654       REAL(wp), DIMENSION(jpi,jpj,jpts)    , INTENT(in   ) ::   pts    ! pot. temperature & salinity 
    655       REAL(wp), DIMENSION(jpi,jpj)         , INTENT(in   ) ::   pdep   ! depth                  [m] 
    656       REAL(wp), DIMENSION(jpi,jpj,jpts)    , INTENT(  out) ::   pab    ! thermal/haline expansion ratio 
     716      INTEGER                            , INTENT(in   ) ::   Kmm   ! time level index 
     717      INTEGER                            , INTENT(in   ) ::   ktts, ktdep, ktab 
     718      REAL(wp), DIMENSION(T2D(ktts),jpts), INTENT(in   ) ::   pts    ! pot. temperature & salinity 
     719      REAL(wp), DIMENSION(T2D(ktdep)    ), INTENT(in   ) ::   pdep   ! depth                  [m] 
     720      REAL(wp), DIMENSION(T2D(ktab),jpts), INTENT(  out) ::   pab    ! thermal/haline expansion ratio 
    657721      ! 
    658722      INTEGER  ::   ji, jj, jk                ! dummy loop indices 
    659723      REAL(wp) ::   zt , zh , zs              ! local scalars 
    660724      REAL(wp) ::   zn , zn0, zn1, zn2, zn3   !   -      - 
     725      REAL(wp), SAVE :: zsum1, zsum2 
    661726      !!---------------------------------------------------------------------- 
    662727      ! 
    … …  
    742807      END SELECT 
    743808      ! 
    744       IF(sn_cfctl%l_prtctl)   CALL prt_ctl( tab2d_1=pab(:,:,jp_tem), clinfo1=' rab_2d_t: ', & 
    745          &                                  tab2d_2=pab(:,:,jp_sal), clinfo2=' rab_2d_s : ' ) 
     809      IF(sn_cfctl%l_prtctl)   CALL prt_ctl( tab2d_1=pab(:,:,jp_tem), clinfo1=' rab_2d_t: ', psum1=zsum1, & 
     810         &                                  tab2d_2=pab(:,:,jp_sal), clinfo2=' rab_2d_s : ', psum2=zsum2 ) 
    746811      ! 
    747812      IF( ln_timing )   CALL timing_stop('rab_2d') 
    748813      ! 
    749    END SUBROUTINE rab_2d 
     814   END SUBROUTINE rab_2d_t 
    750815 
    751816 
    … …  
    848913 
    849914   SUBROUTINE bn2( pts, pab, pn2, Kmm ) 
     915      !! 
     916      INTEGER                              , INTENT(in   ) ::  Kmm   ! time level index 
     917      REAL(wp), DIMENSION(jpi,jpj,jpk,jpts), INTENT(in   ) ::  pts   ! pot. temperature and salinity   [Celsius,psu] 
     918      REAL(wp), DIMENSION(:,:,:,:)         , INTENT(in   ) ::  pab   ! thermal/haline expansion coef.  [Celsius-1,psu-1] 
     919      REAL(wp), DIMENSION(:,:,:)           , INTENT(  out) ::  pn2   ! Brunt-Vaisala frequency squared [1/s^2] 
     920      !! 
     921      CALL bn2_t( pts, pab, is_tile(pab), pn2, is_tile(pn2), Kmm ) 
     922   END SUBROUTINE bn2 
     923 
     924 
     925   SUBROUTINE bn2_t( pts, pab, ktab, pn2, ktn2, Kmm ) 
    850926      !!---------------------------------------------------------------------- 
    851927      !!                  ***  ROUTINE bn2  *** 
    … …  
    861937      !! 
    862938      !!---------------------------------------------------------------------- 
    863       INTEGER                              , INTENT(in   ) ::   Kmm   ! time level index 
    864       REAL(wp), DIMENSION(jpi,jpj,jpk,jpts), INTENT(in   ) ::  pts   ! pot. temperature and salinity   [Celsius,psu] 
    865       REAL(wp), DIMENSION(jpi,jpj,jpk,jpts), INTENT(in   ) ::  pab   ! thermal/haline expansion coef.  [Celsius-1,psu-1] 
    866       REAL(wp), DIMENSION(jpi,jpj,jpk     ), INTENT(  out) ::  pn2   ! Brunt-Vaisala frequency squared [1/s^2] 
     939      INTEGER                                , INTENT(in   ) ::  Kmm   ! time level index 
     940      INTEGER                                , INTENT(in   ) ::  ktab, ktn2 
     941      REAL(wp), DIMENSION(jpi,jpj,  jpk,jpts), INTENT(in   ) ::  pts   ! pot. temperature and salinity   [Celsius,psu] 
     942      REAL(wp), DIMENSION(T2D(ktab),jpk,jpts), INTENT(in   ) ::  pab   ! thermal/haline expansion coef.  [Celsius-1,psu-1] 
     943      REAL(wp), DIMENSION(T2D(ktn2),jpk     ), INTENT(  out) ::  pn2   ! Brunt-Vaisala frequency squared [1/s^2] 
    867944      ! 
    868945      INTEGER  ::   ji, jj, jk      ! dummy loop indices 
    869946      REAL(wp) ::   zaw, zbw, zrw   ! local scalars 
     947      REAL(wp), SAVE :: zsum1 
    870948      !!---------------------------------------------------------------------- 
    871949      ! 
    … …  
    884962      END_3D 
    885963      ! 
    886       IF(sn_cfctl%l_prtctl)   CALL prt_ctl( tab3d_1=pn2, clinfo1=' bn2  : ', kdim=jpk ) 
     964      IF(sn_cfctl%l_prtctl)   CALL prt_ctl( tab3d_1=pn2, clinfo1=' bn2  : ', kdim=jpk, psum1=zsum1 ) 
    887965      ! 
    888966      IF( ln_timing )   CALL timing_stop('bn2') 
    889967      ! 
    890    END SUBROUTINE bn2 
     968   END SUBROUTINE bn2_t 
    891969 
    892970 
    … …  
    9481026 
    9491027 
    950    SUBROUTINE  eos_fzp_2d( psal, ptf, pdep ) 
     1028   SUBROUTINE eos_fzp_2d( psal, ptf, pdep ) 
     1029      !! 
     1030      REAL(wp), DIMENSION(jpi,jpj), INTENT(in   )           ::   psal   ! salinity   [psu] 
     1031      REAL(wp), DIMENSION(jpi,jpj), INTENT(in   ), OPTIONAL ::   pdep   ! depth      [m] 
     1032      REAL(wp), DIMENSION(:,:)    , INTENT(out  )           ::   ptf    ! freezing temperature [Celsius] 
     1033      !! 
     1034      CALL eos_fzp_2d_t( psal, ptf, is_tile(ptf), pdep ) 
     1035   END SUBROUTINE eos_fzp_2d 
     1036 
     1037 
     1038   SUBROUTINE  eos_fzp_2d_t( psal, ptf, kttf, pdep ) 
    9511039      !!---------------------------------------------------------------------- 
    9521040      !!                 ***  ROUTINE eos_fzp  *** 
    … …  
    9601048      !! Reference  :   UNESCO tech. papers in the marine science no. 28. 1978 
    9611049      !!---------------------------------------------------------------------- 
    962       REAL(wp), DIMENSION(jpi,jpj), INTENT(in   )           ::   psal   ! salinity   [psu] 
    963       REAL(wp), DIMENSION(jpi,jpj), INTENT(in   ), OPTIONAL ::   pdep   ! depth      [m] 
    964       REAL(wp), DIMENSION(jpi,jpj), INTENT(out  )           ::   ptf    ! freezing temperature [Celsius] 
     1050      INTEGER                       , INTENT(in   )           ::   kttf 
     1051      REAL(wp), DIMENSION(jpi,jpj)  , INTENT(in   )           ::   psal   ! salinity   [psu] 
     1052      REAL(wp), DIMENSION(jpi,jpj)  , INTENT(in   ), OPTIONAL ::   pdep   ! depth      [m] 
     1053      REAL(wp), DIMENSION(T2D(kttf)), INTENT(out  )           ::   ptf    ! freezing temperature [Celsius] 
    9651054      ! 
    9661055      INTEGER  ::   ji, jj          ! dummy loop indices 
    … …  
    9951084      END SELECT       
    9961085      ! 
    997   END SUBROUTINE eos_fzp_2d 
     1086  END SUBROUTINE eos_fzp_2d_t 
    9981087 
    9991088 

  • NEMO/branches/UKMO/dev_r12866_HPC-02_Daley_Tiling_trial_extra_halo/src/OCE/TRA/traadv.F90

    r12810 r13409  
    1818   USE oce            ! ocean dynamics and active tracers 
    1919   USE dom_oce        ! ocean space and time domain 
     20   ! TEMP: This change not necessary after trd_tra is tiled and extended haloes development 
     21   USE domain, ONLY : dom_tile 
    2022   USE domvvl         ! variable vertical scale factors 
    2123   USE sbcwave        ! wave module 
    … …  
    6567   INTEGER, PARAMETER ::   np_UBS     = 4   ! 3rd order Upstream Biased Scheme 
    6668   INTEGER, PARAMETER ::   np_QCK     = 5   ! QUICK scheme 
    67     
     69 
     70   !! * Substitutions 
     71#  include "do_loop_substitute.h90" 
    6872   !!---------------------------------------------------------------------- 
    6973   !! NEMO/OCE 4.0 , NEMO Consortium (2018) 
    … …  
    8589      REAL(wp), DIMENSION(jpi,jpj,jpk,jpts,jpt), INTENT(inout) :: pts            ! active tracers and RHS of tracer equation 
    8690      ! 
    87       INTEGER ::   jk   ! dummy loop index 
    88       REAL(wp), DIMENSION(jpi,jpj,jpk)        :: zuu, zvv, zww   ! 3D workspace 
    89       REAL(wp), DIMENSION(:,:,:), ALLOCATABLE ::   ztrdt, ztrds 
     91      ! TEMP: This change not necessary after trd_tra is tiled 
     92      INTEGER ::   itile 
     93      INTEGER ::   ji, jj, jk   ! dummy loop index 
     94      REAL(wp), SAVE :: zsum1, zsum2 
     95      ! TEMP: This change not necessary and can be A2D if using XIOS (subdomain support) 
     96      REAL(wp), DIMENSION(:,:,:), ALLOCATABLE, SAVE :: zuu, zvv, zww   ! 3D workspace 
     97      ! TEMP: This change not necessary after trd_tra is tiled 
     98      REAL(wp), DIMENSION(:,:,:), ALLOCATABLE, SAVE :: ztrdt, ztrds 
     99      ! TEMP: This change not necessary after extra haloes development 
     100      LOGICAL :: lskip 
    90101      !!---------------------------------------------------------------------- 
    91102      ! 
    92103      IF( ln_timing )   CALL timing_start('tra_adv') 
    93104      ! 
    94       !                                         !==  effective transport  ==! 
    95       zuu(:,:,jpk) = 0._wp 
    96       zvv(:,:,jpk) = 0._wp 
    97       zww(:,:,jpk) = 0._wp 
    98       IF( ln_wave .AND. ln_sdw )  THEN 
    99          DO jk = 1, jpkm1                                                       ! eulerian transport + Stokes Drift 
    100             zuu(:,:,jk) = e2u  (:,:) * e3u(:,:,jk,Kmm) * ( uu(:,:,jk,Kmm) + usd(:,:,jk) ) 
    101             zvv(:,:,jk) = e1v  (:,:) * e3v(:,:,jk,Kmm) * ( vv(:,:,jk,Kmm) + vsd(:,:,jk) ) 
    102             zww(:,:,jk) = e1e2t(:,:)                 * ( ww(:,:,jk) + wsd(:,:,jk) ) 
    103          END DO 
    104       ELSE 
    105          DO jk = 1, jpkm1 
    106             zuu(:,:,jk) = e2u  (:,:) * e3u(:,:,jk,Kmm) * uu(:,:,jk,Kmm)               ! eulerian transport only 
    107             zvv(:,:,jk) = e1v  (:,:) * e3v(:,:,jk,Kmm) * vv(:,:,jk,Kmm) 
    108             zww(:,:,jk) = e1e2t(:,:)                 * ww(:,:,jk) 
    109          END DO 
    110       ENDIF 
    111       ! 
    112       IF( ln_vvl_ztilde .OR. ln_vvl_layer ) THEN                                ! add z-tilde and/or vvl corrections 
    113          zuu(:,:,:) = zuu(:,:,:) + un_td(:,:,:) 
    114          zvv(:,:,:) = zvv(:,:,:) + vn_td(:,:,:) 
    115       ENDIF 
    116       ! 
    117       zuu(:,:,jpk) = 0._wp                                                      ! no transport trough the bottom 
    118       zvv(:,:,jpk) = 0._wp 
    119       zww(:,:,jpk) = 0._wp 
    120       ! 
    121       IF( ln_ldfeiv .AND. .NOT. ln_traldf_triad )   & 
    122          &              CALL ldf_eiv_trp( kt, nit000, zuu, zvv, zww, 'TRA', Kmm, Krhs )   ! add the eiv transport (if necessary) 
    123       ! 
    124       IF( ln_mle    )   CALL tra_mle_trp( kt, nit000, zuu, zvv, zww, 'TRA', Kmm       )   ! add the mle transport (if necessary) 
    125       ! 
    126       CALL iom_put( "uocetr_eff", zuu )                                        ! output effective transport       
    127       CALL iom_put( "vocetr_eff", zvv ) 
    128       CALL iom_put( "wocetr_eff", zww ) 
    129       ! 
    130 !!gm ??? 
    131       CALL dia_ptr( kt, Kmm, zvv )                                    ! diagnose the effective MSF  
    132 !!gm ??? 
    133       ! 
    134  
    135       IF( l_trdtra )   THEN                    !* Save ta and sa trends 
    136          ALLOCATE( ztrdt(jpi,jpj,jpk), ztrds(jpi,jpj,jpk) ) 
    137          ztrdt(:,:,:) = pts(:,:,:,jp_tem,Krhs) 
    138          ztrds(:,:,:) = pts(:,:,:,jp_sal,Krhs) 
    139       ENDIF 
    140       ! 
    141       SELECT CASE ( nadv )                      !==  compute advection trend and add it to general trend  ==! 
    142       ! 
    143       CASE ( np_CEN )                                 ! Centered scheme : 2nd / 4th order 
    144          CALL tra_adv_cen    ( kt, nit000, 'TRA',         zuu, zvv, zww, Kmm, pts, jpts, Krhs, nn_cen_h, nn_cen_v ) 
    145       CASE ( np_FCT )                                 ! FCT scheme      : 2nd / 4th order 
    146          CALL tra_adv_fct    ( kt, nit000, 'TRA', rDt, zuu, zvv, zww, Kbb, Kmm, pts, jpts, Krhs, nn_fct_h, nn_fct_v ) 
    147       CASE ( np_MUS )                                 ! MUSCL 
    148          CALL tra_adv_mus    ( kt, nit000, 'TRA', rDt, zuu, zvv, zww, Kbb, Kmm, pts, jpts, Krhs, ln_mus_ups )  
    149       CASE ( np_UBS )                                 ! UBS 
    150          CALL tra_adv_ubs    ( kt, nit000, 'TRA', rDt, zuu, zvv, zww, Kbb, Kmm, pts, jpts, Krhs, nn_ubs_v   ) 
    151       CASE ( np_QCK )                                 ! QUICKEST 
    152          CALL tra_adv_qck    ( kt, nit000, 'TRA', rDt, zuu, zvv, zww, Kbb, Kmm, pts, jpts, Krhs ) 
    153       ! 
    154       END SELECT 
    155       ! 
    156       IF( l_trdtra )   THEN                      ! save the advective trends for further diagnostics 
    157          DO jk = 1, jpkm1 
    158             ztrdt(:,:,jk) = pts(:,:,jk,jp_tem,Krhs) - ztrdt(:,:,jk) 
    159             ztrds(:,:,jk) = pts(:,:,jk,jp_sal,Krhs) - ztrds(:,:,jk) 
    160          END DO 
    161          CALL trd_tra( kt, Kmm, Krhs, 'TRA', jp_tem, jptra_totad, ztrdt ) 
    162          CALL trd_tra( kt, Kmm, Krhs, 'TRA', jp_sal, jptra_totad, ztrds ) 
    163          DEALLOCATE( ztrdt, ztrds ) 
     105      lskip = .FALSE. 
     106 
     107      ! TEMP: These changes not necessary if using XIOS (subdomain support) 
     108      IF( ntile == 0 .OR. ntile == 1 )  THEN                       ! Do only on the first tile 
     109         ALLOCATE( zuu(jpi,jpj,jpk), zvv(jpi,jpj,jpk), zww(jpi,jpj,jpk) ) 
     110         ! TEMP: This can be A2D after trd_tra is tiled 
     111         IF( l_trdtra ) ALLOCATE( ztrdt(jpi,jpj,jpk), ztrds(jpi,jpj,jpk) ) 
     112      ENDIF 
     113 
     114      ! TEMP: These changes not necessary after extra haloes development (lbc_lnk removed from tra_adv_*, ldf_eiv_dia) 
     115      IF( nadv /= np_CEN .OR. (nadv == np_CEN .AND. nn_cen_h == 4) .OR. ln_ldfeiv_dia )  THEN 
     116         IF( ln_tile ) THEN 
     117            IF( ntile == 1 ) THEN 
     118               CALL dom_tile( ntsi, ntsj, ntei, ntej, ktile = 0 ) 
     119            ELSE 
     120               lskip = .TRUE. 
     121            ENDIF 
     122         ENDIF 
     123      ENDIF 
     124      IF( .NOT. lskip ) THEN 
     125 
     126         ! TEMP: This change not necessary after trd_tra is tiled 
     127         itile = ntile 
     128         !                                         !==  effective transport  ==! 
     129         ! TODO: NOT TESTED- requires waves 
     130         IF( ln_wave .AND. ln_sdw )  THEN 
     131            DO_3D_11_11( 1, jpkm1 ) 
     132               zuu(ji,jj,jk) = e2u  (ji,jj) * e3u(ji,jj,jk,Kmm) * ( uu(ji,jj,jk,Kmm) + usd(ji,jj,jk) ) 
     133               zvv(ji,jj,jk) = e1v  (ji,jj) * e3v(ji,jj,jk,Kmm) * ( vv(ji,jj,jk,Kmm) + vsd(ji,jj,jk) ) 
     134               zww(ji,jj,jk) = e1e2t(ji,jj)                     * ( ww(ji,jj,jk)     + wsd(ji,jj,jk) ) 
     135            END_3D 
     136         ELSE 
     137            DO_3D_11_11( 1, jpkm1 ) 
     138               zuu(ji,jj,jk) = e2u  (ji,jj) * e3u(ji,jj,jk,Kmm) * uu(ji,jj,jk,Kmm)               ! eulerian transport only 
     139               zvv(ji,jj,jk) = e1v  (ji,jj) * e3v(ji,jj,jk,Kmm) * vv(ji,jj,jk,Kmm) 
     140               zww(ji,jj,jk) = e1e2t(ji,jj)                     * ww(ji,jj,jk) 
     141            END_3D 
     142         ENDIF 
     143         ! 
     144         ! TODO: NOT TESTED- requires ztilde 
     145         IF( ln_vvl_ztilde .OR. ln_vvl_layer ) THEN                                ! add z-tilde and/or vvl corrections 
     146            DO_3D_11_11( 1, jpkm1 ) 
     147               zuu(ji,jj,jk) = zuu(ji,jj,jk) + un_td(ji,jj,jk) 
     148               zvv(ji,jj,jk) = zvv(ji,jj,jk) + vn_td(ji,jj,jk) 
     149            END_3D 
     150         ENDIF 
     151         ! 
     152         DO_2D_11_11 
     153            zuu(ji,jj,jpk) = 0._wp                                                      ! no transport trough the bottom 
     154            zvv(ji,jj,jpk) = 0._wp 
     155            zww(ji,jj,jpk) = 0._wp 
     156         END_2D 
     157         ! 
     158         ! TEMP: These changes not necessary if using XIOS (subdomain support) 
     159         IF( ln_ldfeiv .AND. .NOT. ln_traldf_triad )   & 
     160            &              CALL ldf_eiv_trp( kt, nit000, zuu(A2D,:), zvv(A2D,:), zww(A2D,:), & 
     161            &                                'TRA', Kmm, Krhs )   ! add the eiv transport (if necessary) 
     162         ! 
     163         IF( ln_mle    )   CALL tra_mle_trp( kt, nit000, zuu(A2D,:), zvv(A2D,:), zww(A2D,:), & 
     164            &                                'TRA', Kmm       )   ! add the mle transport (if necessary) 
     165         ! 
     166         ! TEMP: This change not necessary if using XIOS (subdomain support) 
     167         IF( ntile == 0 .OR. ntile == nijtile )  THEN                ! Do only on the last tile 
     168            CALL iom_put( "uocetr_eff", zuu )                                        ! output effective transport 
     169            CALL iom_put( "vocetr_eff", zvv ) 
     170            CALL iom_put( "wocetr_eff", zww ) 
     171         ENDIF 
     172         ! 
     173   !!gm ??? 
     174         ! TEMP: This change not necessary if using XIOS (subdomain support) 
     175         CALL dia_ptr( kt, Kmm, zvv(A2D,:) )                                    ! diagnose the effective MSF 
     176   !!gm ??? 
     177         ! 
     178 
     179         IF( l_trdtra )   THEN                    !* Save ta and sa trends 
     180            DO_3D_00_00( 1, jpkm1 ) 
     181               ztrdt(ji,jj,jk) = pts(ji,jj,jk,jp_tem,Krhs) 
     182               ztrds(ji,jj,jk) = pts(ji,jj,jk,jp_sal,Krhs) 
     183            END_3D 
     184         ENDIF 
     185         ! 
     186         SELECT CASE ( nadv )                      !==  compute advection trend and add it to general trend  ==! 
     187         ! 
     188         CASE ( np_CEN )                                 ! Centered scheme : 2nd / 4th order 
     189            CALL tra_adv_cen    ( kt, nit000, 'TRA',         zuu, zvv, zww, Kmm, pts, jpts, Krhs, nn_cen_h, nn_cen_v ) 
     190         CASE ( np_FCT )                                 ! FCT scheme      : 2nd / 4th order 
     191            CALL tra_adv_fct    ( kt, nit000, 'TRA', rDt, zuu, zvv, zww, Kbb, Kmm, pts, jpts, Krhs, nn_fct_h, nn_fct_v ) 
     192         CASE ( np_MUS )                                 ! MUSCL 
     193            CALL tra_adv_mus    ( kt, nit000, 'TRA', rDt, zuu, zvv, zww, Kbb, Kmm, pts, jpts, Krhs, ln_mus_ups ) 
     194         CASE ( np_UBS )                                 ! UBS 
     195            CALL tra_adv_ubs    ( kt, nit000, 'TRA', rDt, zuu, zvv, zww, Kbb, Kmm, pts, jpts, Krhs, nn_ubs_v   ) 
     196         CASE ( np_QCK )                                 ! QUICKEST 
     197            CALL tra_adv_qck    ( kt, nit000, 'TRA', rDt, zuu, zvv, zww, Kbb, Kmm, pts, jpts, Krhs ) 
     198         ! 
     199         END SELECT 
     200         ! 
     201         ! TEMP: These changes not necessary after trd_tra is tiled 
     202         IF( l_trdtra )   THEN                      ! save the advective trends for further diagnostics 
     203            DO_3D_00_00( 1, jpkm1 ) 
     204               ztrdt(ji,jj,jk) = pts(ji,jj,jk,jp_tem,Krhs) - ztrdt(ji,jj,jk) 
     205               ztrds(ji,jj,jk) = pts(ji,jj,jk,jp_sal,Krhs) - ztrds(ji,jj,jk) 
     206            END_3D 
     207 
     208            IF( ntile == 0 .OR. ntile == nijtile )  THEN                ! Do only for the full domain 
     209               IF( ln_tile ) CALL dom_tile( ntsi, ntsj, ntei, ntej, ktile = 0 )         ! Use full domain 
     210 
     211               ! TODO: TO BE TILED- trd_tra 
     212               CALL trd_tra( kt, Kmm, Krhs, 'TRA', jp_tem, jptra_totad, ztrdt ) 
     213               CALL trd_tra( kt, Kmm, Krhs, 'TRA', jp_sal, jptra_totad, ztrds ) 
     214               DEALLOCATE( ztrdt, ztrds ) 
     215 
     216               IF( ln_tile ) CALL dom_tile( ntsi, ntsj, ntei, ntej, ktile = itile )     ! Revert to tile domain 
     217            ENDIF 
     218         ENDIF 
     219 
     220         ! TEMP: This change not necessary after extra haloes development (lbc_lnk removed from tra_adv_*, ldf_eiv_dia) 
     221         IF( ln_tile .AND. ntile == 0 ) CALL dom_tile( ntsi, ntsj, ntei, ntej, ktile = 1 ) 
     222 
    164223      ENDIF 
    165224      !                                              ! print mean trends (used for debugging) 
    166       IF(sn_cfctl%l_prtctl)   CALL prt_ctl( tab3d_1=pts(:,:,:,jp_tem,Krhs), clinfo1=' adv  - Ta: ', mask1=tmask,               & 
    167          &                                  tab3d_2=pts(:,:,:,jp_sal,Krhs), clinfo2=       ' Sa: ', mask2=tmask, clinfo3='tra' ) 
     225      IF(sn_cfctl%l_prtctl)   CALL prt_ctl( tab3d_1=pts(:,:,:,jp_tem,Krhs), clinfo1=' adv  - Ta: ', mask1=tmask, psum1=zsum1, & 
     226         &                                  tab3d_2=pts(:,:,:,jp_sal,Krhs), clinfo2=       ' Sa: ', mask2=tmask, psum2=zsum2, & 
     227         &                                  clinfo3='tra' ) 
     228 
     229      ! TEMP: This change not necessary if using XIOS (subdomain support) 
     230      IF( ntile == 0 .OR. ntile == nijtile )  THEN                ! Do only for the full domain 
     231         DEALLOCATE( zuu, zvv, zww ) 
     232      ENDIF 
    168233      ! 
    169234      IF( ln_timing )   CALL timing_stop( 'tra_adv' ) 

  • NEMO/branches/UKMO/dev_r12866_HPC-02_Daley_Tiling_trial_extra_halo/src/OCE/TRA/traadv_cen.F90

    r12377 r13409  
    1212   !!---------------------------------------------------------------------- 
    1313   USE dom_oce        ! ocean space and time domain 
     14   ! TEMP: This change not necessary after trd_tra is tiled 
     15   USE domain, ONLY : dom_tile 
    1416   USE eosbn2         ! equation of state 
    1517   USE traadv_fct     ! acces to routine interp_4th_cpt  
    … …  
    7072      INTEGER                                  , INTENT(in   ) ::   kn_cen_h        ! =2/4 (2nd or 4th order scheme) 
    7173      INTEGER                                  , INTENT(in   ) ::   kn_cen_v        ! =2/4 (2nd or 4th order scheme) 
     74      ! TEMP: This can be A2D after trd_tra is tiled 
    7275      REAL(wp), DIMENSION(jpi,jpj,jpk         ), INTENT(in   ) ::   pU, pV, pW      ! 3 ocean volume flux components 
    7376      REAL(wp), DIMENSION(jpi,jpj,jpk,kjpt,jpt), INTENT(inout) ::   pt              ! tracers and RHS of tracer equation 
    … …  
    7578      INTEGER  ::   ji, jj, jk, jn   ! dummy loop indices 
    7679      INTEGER  ::   ierr             ! local integer 
    77       REAL(wp) ::   zC2t_u, zC4t_u   ! local scalars 
    78       REAL(wp) ::   zC2t_v, zC4t_v   !   -      - 
    79       REAL(wp), DIMENSION(jpi,jpj,jpk) ::   zwx, zwy, zwz, ztu, ztv, ztw 
     80      ! TEMP: This change not necessary after trd_tra is tiled 
     81      INTEGER  ::   itile 
     82      REAL(wp) ::   zC2t_u, zC2t_v   ! local scalars 
     83      REAL(wp), DIMENSION(A2D,jpk) ::   zwx, zwy, zwz, ztu, ztv, ztw, zltu, zltv 
     84      ! TEMP: This change not necessary after trd_tra is tiled 
     85      REAL(wp), DIMENSION(:,:,:), SAVE, ALLOCATABLE ::   ztrdx, ztrdy, ztrdz 
    8086      !!---------------------------------------------------------------------- 
    81       ! 
    82       IF( kt == kit000 )  THEN 
    83          IF(lwp) WRITE(numout,*) 
    84          IF(lwp) WRITE(numout,*) 'tra_adv_cen : centered advection scheme on ', cdtype, ' order h/v =', kn_cen_h,'/', kn_cen_v 
    85          IF(lwp) WRITE(numout,*) '~~~~~~~~~~~~ ' 
     87      ! TEMP: This change not necessary after trd_tra is tiled 
     88      itile = ntile 
     89      ! 
     90      IF( ntile == 0 .OR. ntile == 1 )  THEN                       ! Do only on the first tile 
     91         IF( kt == kit000 )  THEN 
     92            IF(lwp) WRITE(numout,*) 
     93            IF(lwp) WRITE(numout,*) 'tra_adv_cen : centered advection scheme on ', cdtype, ' order h/v =', kn_cen_h,'/', kn_cen_v 
     94            IF(lwp) WRITE(numout,*) '~~~~~~~~~~~~ ' 
     95         ENDIF 
     96         !                          ! set local switches 
     97         l_trd = .FALSE. 
     98         l_hst = .FALSE. 
     99         l_ptr = .FALSE. 
     100         IF( ( cdtype == 'TRA' .AND. l_trdtra ) .OR. ( cdtype == 'TRC' .AND. l_trdtrc ) )       l_trd = .TRUE. 
     101         IF(   cdtype == 'TRA' .AND. ( iom_use( 'sophtadv' ) .OR. iom_use( 'sophtadv' ) )  )    l_ptr = .TRUE. 
     102         IF(   cdtype == 'TRA' .AND. ( iom_use("uadv_heattr") .OR. iom_use("vadv_heattr") .OR. & 
     103            &                          iom_use("uadv_salttr") .OR. iom_use("vadv_salttr")  ) )  l_hst = .TRUE. 
     104 
     105         ! TEMP: This can be A2D after trd_tra is tiled 
     106         IF( kt == kit000 .AND. l_trd ) THEN 
     107            ALLOCATE( ztrdx(jpi,jpj,jpk), ztrdy(jpi,jpj,jpk), ztrdz(jpi,jpj,jpk) ) 
     108         ENDIF 
    86109      ENDIF 
    87       !                          ! set local switches 
    88       l_trd = .FALSE. 
    89       l_hst = .FALSE. 
    90       l_ptr = .FALSE. 
    91       IF( ( cdtype == 'TRA' .AND. l_trdtra ) .OR. ( cdtype == 'TRC' .AND. l_trdtrc ) )       l_trd = .TRUE. 
    92       IF(   cdtype == 'TRA' .AND. ( iom_use( 'sophtadv' ) .OR. iom_use( 'sophtadv' ) )  )    l_ptr = .TRUE.  
    93       IF(   cdtype == 'TRA' .AND. ( iom_use("uadv_heattr") .OR. iom_use("vadv_heattr") .OR. & 
    94          &                          iom_use("uadv_salttr") .OR. iom_use("vadv_salttr")  ) )  l_hst = .TRUE. 
    95110      ! 
    96111      !                     
    … …  
    109124            ! 
    110125         CASE(  4  )                         !* 4th order centered 
    111             ztu(:,:,jpk) = 0._wp                   ! Bottom value : flux set to zero 
    112             ztv(:,:,jpk) = 0._wp 
    113             DO_3D_00_00( 1, jpkm1 ) 
    114                ztu(ji,jj,jk) = ( pt(ji+1,jj  ,jk,jn,Kmm) - pt(ji,jj,jk,jn,Kmm) ) * umask(ji,jj,jk) 
    115                ztv(ji,jj,jk) = ( pt(ji  ,jj+1,jk,jn,Kmm) - pt(ji,jj,jk,jn,Kmm) ) * vmask(ji,jj,jk) 
    116             END_3D 
    117             CALL lbc_lnk_multi( 'traadv_cen', ztu, 'U', -1. , ztv, 'V', -1. )   ! Lateral boundary cond. 
    118             ! 
    119             DO_3D_00_10( 1, jpkm1 ) 
     126            zltu(:,:,jpk) = 0._wp            ! Bottom value : flux set to zero 
     127            zltv(:,:,jpk) = 0._wp 
     128            DO jk = 1, jpkm1                 ! Laplacian 
     129               DO_2D_10_10 
     130                  ztu(ji,jj,jk) = ( pt(ji+1,jj  ,jk,jn,Kmm) - pt(ji,jj,jk,jn,Kmm) ) * umask(ji,jj,jk) 
     131                  ztv(ji,jj,jk) = ( pt(ji  ,jj+1,jk,jn,Kmm) - pt(ji,jj,jk,jn,Kmm) ) * vmask(ji,jj,jk) 
     132               END_2D 
     133               DO_2D_00_00 
     134                  zltu(ji,jj,jk) = ztu(ji,jj,jk) + ztu(ji-1,jj,jk) 
     135                  zltv(ji,jj,jk) = ztv(ji,jj,jk) + ztv(ji,jj-1,jk) 
     136               END_2D 
     137            END DO 
     138            CALL lbc_lnk_multi( 'traadv_cen', zltu, 'T', 1. , zltv, 'T', 1. )   ! Lateral boundary cond. (unchanged sgn) 
     139            ! 
     140            DO_3D_10_10( 1, jpkm1 ) 
    120141               zC2t_u = pt(ji,jj,jk,jn,Kmm) + pt(ji+1,jj  ,jk,jn,Kmm)   ! C2 interpolation of T at u- & v-points (x2) 
    121142               zC2t_v = pt(ji,jj,jk,jn,Kmm) + pt(ji  ,jj+1,jk,jn,Kmm) 
    122                !                                                  ! C4 interpolation of T at u- & v-points (x2) 
    123                zC4t_u =  zC2t_u + r1_6 * ( ztu(ji-1,jj,jk) - ztu(ji+1,jj,jk) ) 
    124                zC4t_v =  zC2t_v + r1_6 * ( ztv(ji,jj-1,jk) - ztv(ji,jj+1,jk) ) 
    125143               !                                                  ! C4 fluxes 
    126                zwx(ji,jj,jk) =  0.5_wp * pU(ji,jj,jk) * zC4t_u 
    127                zwy(ji,jj,jk) =  0.5_wp * pV(ji,jj,jk) * zC4t_v 
     144               zwx(ji,jj,jk) =  0.5_wp * pU(ji,jj,jk) * ( zC2t_u + r1_6 * (zltu(ji,jj,jk) - zltu(ji+1,jj,jk)) ) 
     145               zwy(ji,jj,jk) =  0.5_wp * pV(ji,jj,jk) * ( zC2t_v + r1_6 * (zltv(ji,jj,jk) - zltv(ji,jj+1,jk)) ) 
    128146            END_3D 
    129147            ! 
    … …  
    148166         ! 
    149167         IF( ln_linssh ) THEN                !* top value   (linear free surf. only as zwz is multiplied by wmask) 
     168            ! TODO: NOT TESTED- requires isf 
    150169            IF( ln_isfcav ) THEN                  ! ice-shelf cavities (top of the ocean) 
    151170               DO_2D_11_11 
    … …  
    153172               END_2D 
    154173            ELSE                                   ! no ice-shelf cavities (only ocean surface) 
    155                zwz(:,:,1) = pW(:,:,1) * pt(:,:,1,jn,Kmm) 
     174               DO_2D_11_11 
     175                  zwz(ji,jj,1) = pW(ji,jj,1) * pt(ji,jj,1,jn,Kmm) 
     176               END_2D 
    156177            ENDIF 
    157178         ENDIF 
    … …  
    164185         END_3D 
    165186         !                             ! trend diagnostics 
     187         ! TEMP: These changes not necessary after trd_tra is tiled 
    166188         IF( l_trd ) THEN 
    167             CALL trd_tra( kt, Kmm, Krhs, cdtype, jn, jptra_xad, zwx, pU, pt(:,:,:,jn,Kmm) ) 
    168             CALL trd_tra( kt, Kmm, Krhs, cdtype, jn, jptra_yad, zwy, pV, pt(:,:,:,jn,Kmm) ) 
    169             CALL trd_tra( kt, Kmm, Krhs, cdtype, jn, jptra_zad, zwz, pW, pt(:,:,:,jn,Kmm) ) 
    170          END IF 
    171          !                                 ! "Poleward" heat and salt transports  
     189            DO_3D_11_11( 1, jpk ) 
     190               ztrdx(ji,jj,jk) = zwx(ji,jj,jk) 
     191               ztrdy(ji,jj,jk) = zwy(ji,jj,jk) 
     192               ztrdz(ji,jj,jk) = zwz(ji,jj,jk) 
     193            END_3D 
     194 
     195            IF( ntile == 0 .OR. ntile == nijtile )  THEN                ! Do only for the full domain 
     196               IF( ln_tile ) CALL dom_tile( ntsi, ntsj, ntei, ntej, ktile = 0 )         ! Use full domain 
     197 
     198               ! TODO: TO BE TILED- trd_tra 
     199               CALL trd_tra( kt, Kmm, Krhs, cdtype, jn, jptra_xad, ztrdx, pU, pt(:,:,:,jn,Kmm) ) 
     200               CALL trd_tra( kt, Kmm, Krhs, cdtype, jn, jptra_yad, ztrdy, pV, pt(:,:,:,jn,Kmm) ) 
     201               CALL trd_tra( kt, Kmm, Krhs, cdtype, jn, jptra_zad, ztrdz, pW, pt(:,:,:,jn,Kmm) ) 
     202 
     203               IF( ln_tile ) CALL dom_tile( ntsi, ntsj, ntei, ntej, ktile = itile )     ! Revert to tile domain 
     204            ENDIF 
     205         ENDIF 
     206         !                                 ! "Poleward" heat and salt transports 
    172207         IF( l_ptr )   CALL dia_ptr_hst( jn, 'adv', zwy(:,:,:) ) 
    173208         !                                 !  heat and salt transport 

  • NEMO/branches/UKMO/dev_r12866_HPC-02_Daley_Tiling_trial_extra_halo/src/OCE/TRA/traadv_fct.F90

    r12866 r13409  
    1515   USE oce            ! ocean dynamics and active tracers 
    1616   USE dom_oce        ! ocean space and time domain 
     17   ! TEMP: This change not necessary after trd_tra is tiled 
     18   USE domain, ONLY : dom_tile 
    1719   USE trc_oce        ! share passive tracers/Ocean variables 
    1820   USE trd_oce        ! trends: ocean variables 
    … …  
    7880      INTEGER                                  , INTENT(in   ) ::   kn_fct_v        ! order of the FCT scheme (=2 or 4) 
    7981      REAL(wp)                                 , INTENT(in   ) ::   p2dt            ! tracer time-step 
     82      ! TEMP: This can be A2D after trd_tra is tiled 
    8083      REAL(wp), DIMENSION(jpi,jpj,jpk         ), INTENT(in   ) ::   pU, pV, pW      ! 3 ocean volume flux components 
    8184      REAL(wp), DIMENSION(jpi,jpj,jpk,kjpt,jpt), INTENT(inout) ::   pt              ! tracers and RHS of tracer equation 
    8285      ! 
    8386      INTEGER  ::   ji, jj, jk, jn                           ! dummy loop indices   
     87      ! TEMP: This change not necessary after trd_tra is tiled 
     88      INTEGER  ::   itile 
    8489      REAL(wp) ::   ztra                                     ! local scalar 
    8590      REAL(wp) ::   zfp_ui, zfp_vj, zfp_wk, zC2t_u, zC4t_u   !   -      - 
    8691      REAL(wp) ::   zfm_ui, zfm_vj, zfm_wk, zC2t_v, zC4t_v   !   -      - 
    87       REAL(wp), DIMENSION(jpi,jpj,jpk)        ::   zwi, zwx, zwy, zwz, ztu, ztv, zltu, zltv, ztw 
    88       REAL(wp), DIMENSION(:,:,:), ALLOCATABLE ::   ztrdx, ztrdy, ztrdz, zptry 
    89       REAL(wp), DIMENSION(:,:,:), ALLOCATABLE ::   zwinf, zwdia, zwsup 
     92      REAL(wp), DIMENSION(A2D,jpk)        ::   zwi, zwx, zwy, zwz, ztu, ztv, zltu, zltv, ztw 
     93      ! TEMP: This change not necessary after trd_tra is tiled 
     94      REAL(wp), DIMENSION(:,:,:), SAVE, ALLOCATABLE ::   ztrdx, ztrdy, ztrdz 
     95      REAL(wp), DIMENSION(:,:,:)      , ALLOCATABLE ::   zptry 
     96      REAL(wp), DIMENSION(:,:,:)      , ALLOCATABLE ::   zwinf, zwdia, zwsup 
    9097      LOGICAL  ::   ll_zAimp                                 ! flag to apply adaptive implicit vertical advection 
    9198      !!---------------------------------------------------------------------- 
    92       ! 
    93       IF( kt == kit000 )  THEN 
    94          IF(lwp) WRITE(numout,*) 
    95          IF(lwp) WRITE(numout,*) 'tra_adv_fct : FCT advection scheme on ', cdtype 
    96          IF(lwp) WRITE(numout,*) '~~~~~~~~~~~' 
     99      ! TEMP: This change not necessary after trd_tra is tiled 
     100      itile = ntile 
     101      ! 
     102      IF( ntile == 0 .OR. ntile == 1 )  THEN                       ! Do only on the first tile 
     103         IF( kt == kit000 )  THEN 
     104            IF(lwp) WRITE(numout,*) 
     105            IF(lwp) WRITE(numout,*) 'tra_adv_fct : FCT advection scheme on ', cdtype 
     106            IF(lwp) WRITE(numout,*) '~~~~~~~~~~~' 
     107         ENDIF 
     108         ! 
     109         l_trd = .FALSE.            ! set local switches 
     110         l_hst = .FALSE. 
     111         l_ptr = .FALSE. 
     112         ll_zAimp = .FALSE. 
     113         IF( ( cdtype == 'TRA' .AND. l_trdtra  ) .OR. ( cdtype =='TRC' .AND. l_trdtrc ) )      l_trd = .TRUE. 
     114         IF(   cdtype == 'TRA' .AND. ( iom_use( 'sophtadv' ) .OR. iom_use( 'sophtadv' ) ) )    l_ptr = .TRUE. 
     115         IF(   cdtype == 'TRA' .AND. ( iom_use("uadv_heattr") .OR. iom_use("vadv_heattr") .OR.  & 
     116            &                         iom_use("uadv_salttr") .OR. iom_use("vadv_salttr")  ) )  l_hst = .TRUE. 
     117         ! 
     118         ! TEMP: This can be A2D after trd_tra is tiled 
     119         IF( kt == kit000 .AND. (l_trd .OR. l_hst) )  THEN 
     120            ALLOCATE( ztrdx(jpi,jpj,jpk), ztrdy(jpi,jpj,jpk), ztrdz(jpi,jpj,jpk) ) 
     121         ENDIF 
    97122      ENDIF 
    98123      ! 
    99       l_trd = .FALSE.            ! set local switches 
    100       l_hst = .FALSE. 
    101       l_ptr = .FALSE. 
    102       ll_zAimp = .FALSE. 
    103       IF( ( cdtype == 'TRA' .AND. l_trdtra  ) .OR. ( cdtype =='TRC' .AND. l_trdtrc ) )      l_trd = .TRUE. 
    104       IF(   cdtype == 'TRA' .AND. ( iom_use( 'sophtadv' ) .OR. iom_use( 'sophtadv' ) ) )    l_ptr = .TRUE.  
    105       IF(   cdtype == 'TRA' .AND. ( iom_use("uadv_heattr") .OR. iom_use("vadv_heattr") .OR.  & 
    106          &                         iom_use("uadv_salttr") .OR. iom_use("vadv_salttr")  ) )  l_hst = .TRUE. 
    107       ! 
    108       IF( l_trd .OR. l_hst )  THEN 
    109          ALLOCATE( ztrdx(jpi,jpj,jpk), ztrdy(jpi,jpj,jpk), ztrdz(jpi,jpj,jpk) ) 
    110          ztrdx(:,:,:) = 0._wp   ;    ztrdy(:,:,:) = 0._wp   ;   ztrdz(:,:,:) = 0._wp 
    111       ENDIF 
    112       ! 
    113124      IF( l_ptr ) THEN   
    114          ALLOCATE( zptry(jpi,jpj,jpk) ) 
     125         ALLOCATE( zptry(A2D,jpk) ) 
    115126         zptry(:,:,:) = 0._wp 
    116127      ENDIF 
    … …  
    123134      ! If adaptive vertical advection, check if it is needed on this PE at this time 
    124135      IF( ln_zad_Aimp ) THEN 
    125          IF( MAXVAL( ABS( wi(:,:,:) ) ) > 0._wp ) ll_zAimp = .TRUE. 
     136         IF( MAXVAL( ABS( wi(A2D,:) ) ) > 0._wp ) ll_zAimp = .TRUE. 
    126137      END IF 
    127138      ! If active adaptive vertical advection, build tridiagonal matrix 
    128139      IF( ll_zAimp ) THEN 
    129          ALLOCATE(zwdia(jpi,jpj,jpk), zwinf(jpi,jpj,jpk),zwsup(jpi,jpj,jpk)) 
     140         ALLOCATE(zwdia(A2D,jpk), zwinf(A2D,jpk), zwsup(A2D,jpk)) 
    130141         DO_3D_00_00( 1, jpkm1 ) 
    131142            zwdia(ji,jj,jk) =  1._wp + p2dt * ( MAX( wi(ji,jj,jk  ) , 0._wp ) - MIN( wi(ji,jj,jk+1) , 0._wp ) ) / e3t(ji,jj,jk,Krhs) 
    … …  
    155166         END_3D 
    156167         IF( ln_linssh ) THEN    ! top ocean value (only in linear free surface as zwz has been w-masked) 
     168            ! TODO: NOT TESTED- requires isf 
    157169            IF( ln_isfcav ) THEN             ! top of the ice-shelf cavities and at the ocean surface 
    158170               DO_2D_11_11 
    … …  
    193205         END IF 
    194206         !                 
     207         ! TEMP: This change not necessary after trd_tra is tiled 
    195208         IF( l_trd .OR. l_hst )  THEN             ! trend diagnostics (contribution of upstream fluxes) 
    196             ztrdx(:,:,:) = zwx(:,:,:)   ;   ztrdy(:,:,:) = zwy(:,:,:)   ;   ztrdz(:,:,:) = zwz(:,:,:) 
     209            DO_3D_11_11( 1, jpk ) 
     210               ztrdx(ji,jj,jk) = zwx(ji,jj,jk)   ;   ztrdy(ji,jj,jk) = zwy(ji,jj,jk)   ;   ztrdz(ji,jj,jk) = zwz(ji,jj,jk) 
     211            END_3D 
    197212         END IF 
    198213         !                             ! "Poleward" heat and salt transports (contribution of upstream fluxes) 
    … …  
    322337         END IF          
    323338         ! 
     339         ! TEMP: These changes not necessary after trd_tra is tiled 
    324340         IF( l_trd .OR. l_hst ) THEN   ! trend diagnostics // heat/salt transport 
    325             ztrdx(:,:,:) = ztrdx(:,:,:) + zwx(:,:,:)  ! <<< add anti-diffusive fluxes  
    326             ztrdy(:,:,:) = ztrdy(:,:,:) + zwy(:,:,:)  !     to upstream fluxes 
    327             ztrdz(:,:,:) = ztrdz(:,:,:) + zwz(:,:,:)  ! 
    328             ! 
    329             IF( l_trd ) THEN              ! trend diagnostics 
    330                CALL trd_tra( kt, Kmm, Krhs, cdtype, jn, jptra_xad, ztrdx, pU, pt(:,:,:,jn,Kmm) ) 
    331                CALL trd_tra( kt, Kmm, Krhs, cdtype, jn, jptra_yad, ztrdy, pV, pt(:,:,:,jn,Kmm) ) 
    332                CALL trd_tra( kt, Kmm, Krhs, cdtype, jn, jptra_zad, ztrdz, pW, pt(:,:,:,jn,Kmm) ) 
     341            DO_3D_11_11( 1, jpk ) 
     342               ztrdx(ji,jj,jk) = ztrdx(ji,jj,jk) + zwx(ji,jj,jk)  ! <<< add anti-diffusive fluxes 
     343               ztrdy(ji,jj,jk) = ztrdy(ji,jj,jk) + zwy(ji,jj,jk)  !     to upstream fluxes 
     344               ztrdz(ji,jj,jk) = ztrdz(ji,jj,jk) + zwz(ji,jj,jk)  ! 
     345            END_3D 
     346            ! 
     347            IF( ntile == 0 .OR. ntile == nijtile )  THEN                ! Do only for the full domain 
     348               IF( l_trd ) THEN              ! trend diagnostics 
     349                  IF( ln_tile ) CALL dom_tile( ntsi, ntsj, ntei, ntej, ktile = 0 )         ! Use full domain 
     350 
     351                  ! TODO: TO BE TILED- trd_tra 
     352                  CALL trd_tra( kt, Kmm, Krhs, cdtype, jn, jptra_xad, ztrdx, pU, pt(:,:,:,jn,Kmm) ) 
     353                  CALL trd_tra( kt, Kmm, Krhs, cdtype, jn, jptra_yad, ztrdy, pV, pt(:,:,:,jn,Kmm) ) 
     354                  CALL trd_tra( kt, Kmm, Krhs, cdtype, jn, jptra_zad, ztrdz, pW, pt(:,:,:,jn,Kmm) ) 
     355 
     356                  IF( ln_tile ) CALL dom_tile( ntsi, ntsj, ntei, ntej, ktile = itile )     ! Revert to tile domain 
     357               ENDIF 
    333358            ENDIF 
    334359            !                             ! heat/salt transport 
    335             IF( l_hst )   CALL dia_ar5_hst( jn, 'adv', ztrdx(:,:,:), ztrdy(:,:,:) ) 
     360            IF( l_hst )   CALL dia_ar5_hst( jn, 'adv', ztrdx(A2D,:), ztrdy(A2D,:) ) 
    336361            ! 
    337362         ENDIF 
    … …  
    346371         DEALLOCATE( zwdia, zwinf, zwsup ) 
    347372      ENDIF 
    348       IF( l_trd .OR. l_hst ) THEN  
    349          DEALLOCATE( ztrdx, ztrdy, ztrdz ) 
    350       ENDIF 
     373      ! TEMP: These changes not necessary after trd_tra is tiled 
     374!      IF( l_trd .OR. l_hst ) THEN 
     375!         DEALLOCATE( ztrdx, ztrdy, ztrdz ) 
     376!      ENDIF 
    351377      IF( l_ptr ) THEN  
    352378         DEALLOCATE( zptry ) 
    … …  
    369395      !!       in-space based differencing for fluid 
    370396      !!---------------------------------------------------------------------- 
    371       INTEGER                          , INTENT(in   ) ::   Kmm             ! time level index  
    372       REAL(wp)                         , INTENT(in   ) ::   p2dt            ! tracer time-step 
    373       REAL(wp), DIMENSION (jpi,jpj,jpk), INTENT(in   ) ::   pbef, paft      ! before & after field 
    374       REAL(wp), DIMENSION (jpi,jpj,jpk), INTENT(inout) ::   paa, pbb, pcc   ! monotonic fluxes in the 3 directions 
     397      INTEGER                         , INTENT(in   ) ::   Kmm             ! time level index 
     398      REAL(wp)                        , INTENT(in   ) ::   p2dt            ! tracer time-step 
     399      REAL(wp), DIMENSION(jpi,jpj,jpk), INTENT(in   ) ::   pbef            ! before field 
     400      REAL(wp), DIMENSION(A2D    ,jpk), INTENT(in   ) ::   paft            ! after field 
     401      REAL(wp), DIMENSION(A2D    ,jpk), INTENT(inout) ::   paa, pbb, pcc   ! monotonic fluxes in the 3 directions 
    375402      ! 
    376403      INTEGER  ::   ji, jj, jk   ! dummy loop indices 
    … …  
    378405      REAL(wp) ::   zpos, zneg, zbt, za, zb, zc, zbig, zrtrn    ! local scalars 
    379406      REAL(wp) ::   zau, zbu, zcu, zav, zbv, zcv, zup, zdo            !   -      - 
    380       REAL(wp), DIMENSION(jpi,jpj,jpk) :: zbetup, zbetdo, zbup, zbdo 
     407      REAL(wp), DIMENSION(A2D,jpk) :: zbetup, zbetdo, zbup, zbdo 
    381408      !!---------------------------------------------------------------------- 
    382409      ! 
    … …  
    388415      ! -------------------- 
    389416      ! max/min of pbef & paft with large negative/positive value (-/+zbig) inside land 
    390       zbup = MAX( pbef * tmask - zbig * ( 1._wp - tmask ),   & 
    391          &        paft * tmask - zbig * ( 1._wp - tmask )  ) 
    392       zbdo = MIN( pbef * tmask + zbig * ( 1._wp - tmask ),   & 
    393          &        paft * tmask + zbig * ( 1._wp - tmask )  ) 
     417      DO_3D_11_11( 1, jpk ) 
     418         zbup(ji,jj,jk) = MAX( pbef(ji,jj,jk) * tmask(ji,jj,jk) - zbig * ( 1._wp - tmask(ji,jj,jk) ),   & 
     419            &                  paft(ji,jj,jk) * tmask(ji,jj,jk) - zbig * ( 1._wp - tmask(ji,jj,jk) )  ) 
     420         zbdo(ji,jj,jk) = MIN( pbef(ji,jj,jk) * tmask(ji,jj,jk) + zbig * ( 1._wp - tmask(ji,jj,jk) ),   & 
     421            &                  paft(ji,jj,jk) * tmask(ji,jj,jk) + zbig * ( 1._wp - tmask(ji,jj,jk) )  ) 
     422      END_3D 
    394423 
    395424      DO jk = 1, jpkm1 
    … …  
    523552      !!---------------------------------------------------------------------- 
    524553      REAL(wp),DIMENSION(jpi,jpj,jpk), INTENT(in   ) ::   pt_in    ! field at t-point 
    525       REAL(wp),DIMENSION(jpi,jpj,jpk), INTENT(  out) ::   pt_out   ! field interpolated at w-point 
     554      REAL(wp),DIMENSION(A2D    ,jpk), INTENT(  out) ::   pt_out   ! field interpolated at w-point 
    526555      ! 
    527556      INTEGER ::   ji, jj, jk   ! dummy loop integers 
    528557      INTEGER ::   ikt, ikb     ! local integers 
    529       REAL(wp),DIMENSION(jpi,jpj,jpk) :: zwd, zwi, zws, zwrm, zwt 
     558      REAL(wp),DIMENSION(A2D,jpk) :: zwd, zwi, zws, zwrm, zwt 
    530559      !!---------------------------------------------------------------------- 
    531560      ! 
    … …  
    547576!!gm   
    548577      ! 
     578      ! TODO: NOT TESTED- requires isf 
    549579      IF ( ln_isfcav ) THEN            ! set level two values which may not be set in ISF case 
    550580         zwd(:,:,2) = 1._wp  ;  zwi(:,:,2) = 0._wp  ;  zws(:,:,2) = 0._wp  ;  zwrm(:,:,2) = 0._wp 
    … …  
    612642      !!        The 3d array zwt is used as a work space array. 
    613643      !!---------------------------------------------------------------------- 
    614       REAL(wp),DIMENSION(:,:,:), INTENT(in   ) ::   pD, pU, PL    ! 3-diagonal matrix 
    615       REAL(wp),DIMENSION(:,:,:), INTENT(in   ) ::   pRHS          ! Right-Hand-Side 
    616       REAL(wp),DIMENSION(:,:,:), INTENT(  out) ::   pt_out        !!gm field at level=F(klev) 
    617       INTEGER                  , INTENT(in   ) ::   klev          ! =1 pt_out at w-level  
    618       !                                                           ! =0 pt at t-level 
     644      REAL(wp),DIMENSION(A2D,jpk), INTENT(in   ) ::   pD, pU, PL    ! 3-diagonal matrix 
     645      REAL(wp),DIMENSION(A2D,jpk), INTENT(in   ) ::   pRHS          ! Right-Hand-Side 
     646      REAL(wp),DIMENSION(A2D,jpk), INTENT(  out) ::   pt_out        !!gm field at level=F(klev) 
     647      INTEGER                    , INTENT(in   ) ::   klev          ! =1 pt_out at w-level 
     648      !                                                             ! =0 pt at t-level 
    619649      INTEGER ::   ji, jj, jk   ! dummy loop integers 
    620650      INTEGER ::   kstart       ! local indices 
    621       REAL(wp),DIMENSION(jpi,jpj,jpk) ::   zwt   ! 3D work array 
     651      REAL(wp),DIMENSION(A2D,jpk) ::   zwt   ! 3D work array 
    622652      !!---------------------------------------------------------------------- 
    623653      ! 

  • NEMO/branches/UKMO/dev_r12866_HPC-02_Daley_Tiling_trial_extra_halo/src/OCE/TRA/traadv_mus.F90

    r12810 r13409  
    1919   USE trc_oce        ! share passive tracers/Ocean variables 
    2020   USE dom_oce        ! ocean space and time domain 
     21   ! TEMP: This change not necessary after trd_tra is tiled 
     22   USE domain, ONLY : dom_tile 
    2123   USE trd_oce        ! trends: ocean variables 
    2224   USE trdtra         ! tracers trends manager 
    … …  
    8082      LOGICAL                                  , INTENT(in   ) ::   ld_msc_ups      ! use upstream scheme within muscl 
    8183      REAL(wp)                                 , INTENT(in   ) ::   p2dt            ! tracer time-step 
     84      ! TEMP: This can be A2D after trd_tra is tiled 
    8285      REAL(wp), DIMENSION(jpi,jpj,jpk         ), INTENT(in   ) ::   pU, pV, pW      ! 3 ocean volume flux components 
    8386      REAL(wp), DIMENSION(jpi,jpj,jpk,kjpt,jpt), INTENT(inout) ::   pt              ! tracers and RHS of tracer equation 
    8487      ! 
     88      ! TEMP: This change not necessary after trd_tra is tiled 
     89      INTEGER  ::   itile 
    8590      INTEGER  ::   ji, jj, jk, jn   ! dummy loop indices 
    8691      INTEGER  ::   ierr             ! local integer 
    8792      REAL(wp) ::   zu, z0u, zzwx, zw , zalpha   ! local scalars 
    8893      REAL(wp) ::   zv, z0v, zzwy, z0w           !   -      - 
    89       REAL(wp), DIMENSION(jpi,jpj,jpk) ::   zwx, zslpx   ! 3D workspace 
    90       REAL(wp), DIMENSION(jpi,jpj,jpk) ::   zwy, zslpy   ! -      -  
     94      REAL(wp), DIMENSION(A2D,jpk) ::   zwx, zslpx   ! 3D workspace 
     95      REAL(wp), DIMENSION(A2D,jpk) ::   zwy, zslpy   ! -      - 
     96      ! TEMP: This change not necessary after trd_tra is tiled 
     97      REAL(wp), DIMENSION(:,:,:), SAVE, ALLOCATABLE ::   ztrdx, ztrdy, ztrdz 
    9198      !!---------------------------------------------------------------------- 
    92       ! 
    93       IF( kt == kit000 )  THEN 
    94          IF(lwp) WRITE(numout,*) 
    95          IF(lwp) WRITE(numout,*) 'tra_adv : MUSCL advection scheme on ', cdtype 
    96          IF(lwp) WRITE(numout,*) '        : mixed up-stream           ', ld_msc_ups 
    97          IF(lwp) WRITE(numout,*) '~~~~~~~' 
    98          IF(lwp) WRITE(numout,*) 
    99          ! 
    100          ! Upstream / MUSCL scheme indicator 
    101          ! 
    102          ALLOCATE( xind(jpi,jpj,jpk), STAT=ierr ) 
    103          xind(:,:,:) = 1._wp              ! set equal to 1 where up-stream is not needed 
    104          ! 
    105          IF( ld_msc_ups ) THEN            ! define the upstream indicator (if asked) 
    106             ALLOCATE( upsmsk(jpi,jpj), STAT=ierr ) 
    107             upsmsk(:,:) = 0._wp                             ! not upstream by default 
    108             ! 
    109             DO jk = 1, jpkm1 
    110                xind(:,:,jk) = 1._wp                              &                 ! =>1 where up-stream is not needed 
    111                   &         - MAX ( rnfmsk(:,:) * rnfmsk_z(jk),  &                 ! =>0 near runoff mouths (& closed sea outflows) 
    112                   &                 upsmsk(:,:)                ) * tmask(:,:,jk)   ! =>0 in some user defined area 
    113             END DO 
    114          ENDIF  
    115          ! 
    116       ENDIF  
    117       !       
    118       l_trd = .FALSE. 
    119       l_hst = .FALSE. 
    120       l_ptr = .FALSE. 
    121       IF( ( cdtype == 'TRA' .AND. l_trdtra ) .OR. ( cdtype == 'TRC' .AND. l_trdtrc ) )      l_trd = .TRUE. 
    122       IF(   cdtype == 'TRA' .AND. ( iom_use( 'sophtadv' ) .OR. iom_use( 'sophtadv' ) )  )   l_ptr = .TRUE.  
    123       IF(   cdtype == 'TRA' .AND. ( iom_use("uadv_heattr") .OR. iom_use("vadv_heattr") .OR. & 
    124          &                          iom_use("uadv_salttr") .OR. iom_use("vadv_salttr")  ) ) l_hst = .TRUE. 
     99      ! TEMP: This change not necessary after trd_tra is tiled 
     100      itile = ntile 
     101      ! 
     102      IF( ntile == 0 .OR. ntile == 1 )  THEN                       ! Do only on the first tile 
     103         IF( kt == kit000 )  THEN 
     104            IF(lwp) WRITE(numout,*) 
     105            IF(lwp) WRITE(numout,*) 'tra_adv : MUSCL advection scheme on ', cdtype 
     106            IF(lwp) WRITE(numout,*) '        : mixed up-stream           ', ld_msc_ups 
     107            IF(lwp) WRITE(numout,*) '~~~~~~~' 
     108            IF(lwp) WRITE(numout,*) 
     109            ! 
     110            ! Upstream / MUSCL scheme indicator 
     111            ! 
     112            ALLOCATE( xind(jpi,jpj,jpk), STAT=ierr ) 
     113            xind(:,:,:) = 1._wp              ! set equal to 1 where up-stream is not needed 
     114            ! 
     115            IF( ld_msc_ups ) THEN            ! define the upstream indicator (if asked) 
     116               ALLOCATE( upsmsk(jpi,jpj), STAT=ierr ) 
     117               upsmsk(:,:) = 0._wp                             ! not upstream by default 
     118               ! 
     119               DO jk = 1, jpkm1 
     120                  xind(:,:,jk) = 1._wp                              &                 ! =>1 where up-stream is not needed 
     121                     &         - MAX ( rnfmsk(:,:) * rnfmsk_z(jk),  &                 ! =>0 near runoff mouths (& closed sea outflows) 
     122                     &                 upsmsk(:,:)                ) * tmask(:,:,jk)   ! =>0 in some user defined area 
     123               END DO 
     124            ENDIF 
     125            ! 
     126         ENDIF 
     127         ! 
     128         l_trd = .FALSE. 
     129         l_hst = .FALSE. 
     130         l_ptr = .FALSE. 
     131         IF( ( cdtype == 'TRA' .AND. l_trdtra ) .OR. ( cdtype == 'TRC' .AND. l_trdtrc ) )      l_trd = .TRUE. 
     132         IF(   cdtype == 'TRA' .AND. ( iom_use( 'sophtadv' ) .OR. iom_use( 'sophtadv' ) )  )   l_ptr = .TRUE. 
     133         IF(   cdtype == 'TRA' .AND. ( iom_use("uadv_heattr") .OR. iom_use("vadv_heattr") .OR. & 
     134            &                          iom_use("uadv_salttr") .OR. iom_use("vadv_salttr")  ) ) l_hst = .TRUE. 
     135 
     136         ! TEMP: This can be A2D after trd_tra is tiled 
     137         IF( kt == kit000 .AND. l_trd ) THEN 
     138            ALLOCATE( ztrdx(jpi,jpj,jpk), ztrdy(jpi,jpj,jpk), ztrdz(jpi,jpj,jpk) ) 
     139         ENDIF 
     140      ENDIF 
    125141      ! 
    126142      DO jn = 1, kjpt            !==  loop over the tracers  ==! 
    … …  
    180196         END_3D 
    181197         !                                ! trend diagnostics 
     198         ! TEMP: These changes not necessary after trd_tra is tiled 
    182199         IF( l_trd )  THEN 
    183             CALL trd_tra( kt, Kmm, Krhs, cdtype, jn, jptra_xad, zwx, pU, pt(:,:,:,jn,Kbb) ) 
    184             CALL trd_tra( kt, Kmm, Krhs, cdtype, jn, jptra_yad, zwy, pV, pt(:,:,:,jn,Kbb) ) 
     200            DO_3D_11_11( 1, jpk ) 
     201               ztrdx(ji,jj,jk) = zwx(ji,jj,jk) 
     202               ztrdy(ji,jj,jk) = zwy(ji,jj,jk) 
     203            END_3D 
     204 
     205            IF( ntile == 0 .OR. ntile == nijtile )  THEN                ! Do only for the full domain 
     206               IF( ln_tile ) CALL dom_tile( ntsi, ntsj, ntei, ntej, ktile = 0 )         ! Use full domain 
     207 
     208               ! TODO: TO BE TILED- trd_tra 
     209               CALL trd_tra( kt, Kmm, Krhs, cdtype, jn, jptra_xad, ztrdx, pU, pt(:,:,:,jn,Kbb) ) 
     210               CALL trd_tra( kt, Kmm, Krhs, cdtype, jn, jptra_yad, ztrdy, pV, pt(:,:,:,jn,Kbb) ) 
     211 
     212               IF( ln_tile ) CALL dom_tile( ntsi, ntsj, ntei, ntej, ktile = nijtile )   ! Revert to tile domain 
     213            ENDIF 
    185214         END IF 
    186215         !                                 ! "Poleward" heat and salt transports  
    … …  
    194223         zwx(:,:, 1 ) = 0._wp                   ! surface & bottom boundary conditions 
    195224         zwx(:,:,jpk) = 0._wp 
    196          DO jk = 2, jpkm1                       ! interior values 
    197             zwx(:,:,jk) = tmask(:,:,jk) * ( pt(:,:,jk-1,jn,Kbb) - pt(:,:,jk,jn,Kbb) ) 
    198          END DO 
     225         DO_3D_11_11( 2, jpkm1 )                ! interior values 
     226            zwx(ji,jj,jk) = tmask(ji,jj,jk) * ( pt(ji,jj,jk-1,jn,Kbb) - pt(ji,jj,jk,jn,Kbb) ) 
     227         END_3D 
    199228         !                                !-- Slopes of tracer 
    200229         zslpx(:,:,1) = 0._wp                   ! surface values 
    … …  
    217246         END_3D 
    218247         IF( ln_linssh ) THEN                   ! top values, linear free surface only 
     248            ! TODO: NOT TESTED- requires isf 
    219249            IF( ln_isfcav ) THEN                      ! ice-shelf cavities (top of the ocean) 
    220250               DO_2D_11_11 
    … …  
    222252               END_2D 
    223253            ELSE                                      ! no cavities: only at the ocean surface 
    224                zwx(:,:,1) = pW(:,:,1) * pt(:,:,1,jn,Kbb) 
     254               DO_2D_11_11 
     255                  zwx(ji,jj,1) = pW(ji,jj,1) * pt(ji,jj,1,jn,Kbb) 
     256               END_2D 
    225257            ENDIF 
    226258         ENDIF 
    … …  
    230262         END_3D 
    231263         !                                ! send trends for diagnostic 
    232          IF( l_trd )  CALL trd_tra( kt, Kmm, Krhs, cdtype, jn, jptra_zad, zwx, pW, pt(:,:,:,jn,Kbb) ) 
     264         ! TEMP: These changes not necessary after trd_tra is tiled 
     265         IF( l_trd ) THEN 
     266            DO_3D_11_11( 1, jpk ) 
     267               ztrdz(ji,jj,jk) = zwx(ji,jj,jk) 
     268            END_3D 
     269 
     270            IF( ntile == 0 .OR. ntile == nijtile )  THEN                ! Do only for the full domain 
     271               IF( ln_tile ) CALL dom_tile( ntsi, ntsj, ntei, ntej, ktile = 0 )         ! Use full domain 
     272 
     273               ! TODO: TO BE TILED- trd_tra 
     274               CALL trd_tra( kt, Kmm, Krhs, cdtype, jn, jptra_zad, ztrdz, pW, pt(:,:,:,jn,Kbb) ) 
     275 
     276               IF( ln_tile ) CALL dom_tile( ntsi, ntsj, ntei, ntej, ktile = itile )     ! Revert to tile domain 
     277            ENDIF 
     278         ENDIF 
    233279         ! 
    234280      END DO                     ! end of tracer loop 

  • NEMO/branches/UKMO/dev_r12866_HPC-02_Daley_Tiling_trial_extra_halo/src/OCE/TRA/traadv_qck.F90

    r12377 r13409  
    1717   USE oce             ! ocean dynamics and active tracers 
    1818   USE dom_oce         ! ocean space and time domain 
     19   ! TEMP: This change not necessary after trd_tra is tiled 
     20   USE domain, ONLY : dom_tile 
    1921   USE trc_oce         ! share passive tracers/Ocean variables 
    2022   USE trd_oce         ! trends: ocean variables 
    … …  
    9092      INTEGER                                  , INTENT(in   ) ::   kjpt            ! number of tracers 
    9193      REAL(wp)                                 , INTENT(in   ) ::   p2dt            ! tracer time-step 
     94      ! TEMP: This can be A2D after trd_tra is tiled 
    9295      REAL(wp), DIMENSION(jpi,jpj,jpk         ), INTENT(in   ) ::   pU, pV, pW      ! 3 ocean volume transport components 
    9396      REAL(wp), DIMENSION(jpi,jpj,jpk,kjpt,jpt), INTENT(inout) ::   pt              ! tracers and RHS of tracer equation 
    9497      !!---------------------------------------------------------------------- 
    9598      ! 
    96       IF( kt == kit000 )  THEN 
    97          IF(lwp) WRITE(numout,*) 
    98          IF(lwp) WRITE(numout,*) 'tra_adv_qck : 3rd order quickest advection scheme on ', cdtype 
    99          IF(lwp) WRITE(numout,*) '~~~~~~~~~~~~' 
    100          IF(lwp) WRITE(numout,*) 
     99      IF( ntile == 0 .OR. ntile == 1 )  THEN                       ! Do only on the first tile 
     100         IF( kt == kit000 )  THEN 
     101            IF(lwp) WRITE(numout,*) 
     102            IF(lwp) WRITE(numout,*) 'tra_adv_qck : 3rd order quickest advection scheme on ', cdtype 
     103            IF(lwp) WRITE(numout,*) '~~~~~~~~~~~~' 
     104            IF(lwp) WRITE(numout,*) 
     105         ENDIF 
     106         ! 
     107         l_trd = .FALSE. 
     108         l_ptr = .FALSE. 
     109         IF( ( cdtype == 'TRA' .AND. l_trdtra ) .OR. ( cdtype == 'TRC' .AND. l_trdtrc ) )   l_trd = .TRUE. 
     110         IF(   cdtype == 'TRA' .AND. ( iom_use( 'sophtadv' ) .OR. iom_use( 'sophtadv' ) ) ) l_ptr = .TRUE. 
    101111      ENDIF 
    102       ! 
    103       l_trd = .FALSE. 
    104       l_ptr = .FALSE. 
    105       IF( ( cdtype == 'TRA' .AND. l_trdtra ) .OR. ( cdtype == 'TRC' .AND. l_trdtrc ) )   l_trd = .TRUE. 
    106       IF(   cdtype == 'TRA' .AND. ( iom_use( 'sophtadv' ) .OR. iom_use( 'sophtadv' ) ) ) l_ptr = .TRUE.  
    107112      ! 
    108113      ! 
    … …  
    126131      INTEGER                                  , INTENT(in   ) ::   kjpt       ! number of tracers 
    127132      REAL(wp)                                 , INTENT(in   ) ::   p2dt       ! tracer time-step 
     133      ! TEMP: This can be A2D after trd_tra is tiled 
    128134      REAL(wp), DIMENSION(jpi,jpj,jpk         ), INTENT(in   ) ::   pU        ! i-velocity components 
    129135      REAL(wp), DIMENSION(jpi,jpj,jpk,kjpt,jpt), INTENT(inout) ::   pt              ! active tracers and RHS of tracer equation 
    130136      !! 
     137      ! TEMP: This change not necessary after trd_tra is tiled 
     138      INTEGER  ::   itile 
    131139      INTEGER  ::   ji, jj, jk, jn   ! dummy loop indices 
    132140      REAL(wp) ::   ztra, zbtr, zdir, zdx, zmsk   ! local scalars 
    133       REAL(wp), DIMENSION(jpi,jpj,jpk) ::   zwx, zfu, zfc, zfd 
     141      REAL(wp), DIMENSION(A2D,jpk) ::   zwx, zfu, zfc, zfd 
     142      ! TEMP: This change not necessary after trd_tra is tiled 
     143      REAL(wp), DIMENSION(:,:,:), SAVE, ALLOCATABLE ::   ztrdx 
    134144      !---------------------------------------------------------------------- 
    135       ! 
     145      ! TEMP: This change not necessary after trd_tra is tiled 
     146      itile = ntile 
     147      ! 
     148      ! TEMP: This can be A2D after trd_tra is tiled 
     149      IF( ntile == 0 .OR. ntile == 1 )  THEN                       ! Do only on the first tile 
     150         IF( kt == nit000 .AND. l_trd ) THEN 
     151            ALLOCATE( ztrdx(jpi,jpj,jpk) ) 
     152         ENDIF 
     153      ENDIF 
    136154      !                                                          ! =========== 
    137155      DO jn = 1, kjpt                                            ! tracer loop 
    … …  
    199217         END_3D 
    200218         !                                 ! trend diagnostics 
    201          IF( l_trd )   CALL trd_tra( kt, Kmm, Krhs, cdtype, jn, jptra_xad, zwx, pU, pt(:,:,:,jn,Kmm) ) 
     219         ! TEMP: These changes not necessary after trd_tra is tiled 
     220         IF( l_trd )  THEN 
     221            DO_3D_11_11( 1, jpk ) 
     222               ztrdx(ji,jj,jk) = zwx(ji,jj,jk) 
     223            END_3D 
     224 
     225            IF( ntile == 0 .OR. ntile == nijtile )  THEN                ! Do only for the full domain 
     226               IF( ln_tile ) CALL dom_tile( ntsi, ntsj, ntei, ntej, ktile = 0 )         ! Use full domain 
     227 
     228               ! TODO: TO BE TILED- trd_tra 
     229               CALL trd_tra( kt, Kmm, Krhs, cdtype, jn, jptra_xad, ztrdx, pU, pt(:,:,:,jn,Kmm) ) 
     230 
     231               IF( ln_tile ) CALL dom_tile( ntsi, ntsj, ntei, ntej, ktile = itile )     ! Revert to tile domain 
     232            ENDIF 
     233         ENDIF 
    202234         ! 
    203235      END DO 
    … …  
    215247      INTEGER                                  , INTENT(in   ) ::   kjpt       ! number of tracers 
    216248      REAL(wp)                                 , INTENT(in   ) ::   p2dt       ! tracer time-step 
     249      ! TEMP: This can be A2D after trd_tra is tiled 
    217250      REAL(wp), DIMENSION(jpi,jpj,jpk         ), INTENT(in   ) ::   pV        ! j-velocity components 
    218251      REAL(wp), DIMENSION(jpi,jpj,jpk,kjpt,jpt), INTENT(inout) ::   pt              ! active tracers and RHS of tracer equation 
    219252      !! 
     253      ! TEMP: This change not necessary after trd_tra is tiled 
     254      INTEGER  ::   itile 
    220255      INTEGER  :: ji, jj, jk, jn                ! dummy loop indices 
    221256      REAL(wp) :: ztra, zbtr, zdir, zdx, zmsk   ! local scalars 
    222       REAL(wp), DIMENSION(jpi,jpj,jpk) ::   zwy, zfu, zfc, zfd   ! 3D workspace 
     257      REAL(wp), DIMENSION(A2D,jpk) ::   zwy, zfu, zfc, zfd   ! 3D workspace 
     258      ! TEMP: This change not necessary after trd_tra is tiled 
     259      REAL(wp), DIMENSION(:,:,:), SAVE, ALLOCATABLE ::   ztrdy 
    223260      !---------------------------------------------------------------------- 
    224       ! 
     261      ! TEMP: This change not necessary after trd_tra is tiled 
     262      itile = ntile 
     263      ! 
     264      ! TEMP: This can be A2D after trd_tra is tiled 
     265      IF( ntile == 0 .OR. ntile == 1 )  THEN                       ! Do only on the first tile 
     266         IF( kt == nit000 .AND. l_trd ) THEN 
     267            ALLOCATE( ztrdy(jpi,jpj,jpk) ) 
     268         ENDIF 
     269      ENDIF 
    225270      !                                                          ! =========== 
    226271      DO jn = 1, kjpt                                            ! tracer loop 
    … …  
    295340         END_3D 
    296341         !                                 ! trend diagnostics 
    297          IF( l_trd )   CALL trd_tra( kt, Kmm, Krhs, cdtype, jn, jptra_yad, zwy, pV, pt(:,:,:,jn,Kmm) ) 
     342         ! TEMP: These changes not necessary after trd_tra is tiled 
     343         IF( l_trd )  THEN 
     344            DO_3D_11_11( 1, jpk ) 
     345               ztrdy(ji,jj,jk) = zwy(ji,jj,jk) 
     346            END_3D 
     347 
     348            IF( ntile == 0 .OR. ntile == nijtile )  THEN                ! Do only for the full domain 
     349               IF( ln_tile ) CALL dom_tile( ntsi, ntsj, ntei, ntej, ktile = 0 )         ! Use full domain 
     350 
     351               ! TODO: TO BE TILED- trd_tra 
     352               CALL trd_tra( kt, Kmm, Krhs, cdtype, jn, jptra_yad, ztrdy, pV, pt(:,:,:,jn,Kmm) ) 
     353 
     354               IF( ln_tile ) CALL dom_tile( ntsi, ntsj, ntei, ntej, ktile = itile )     ! Revert to tile domain 
     355            ENDIF 
     356         ENDIF 
    298357         !                                 ! "Poleward" heat and salt transports (contribution of upstream fluxes) 
    299358         IF( l_ptr )   CALL dia_ptr_hst( jn, 'adv', zwy(:,:,:) ) 
    … …  
    312371      CHARACTER(len=3)                         , INTENT(in   ) ::   cdtype   ! =TRA or TRC (tracer indicator) 
    313372      INTEGER                                  , INTENT(in   ) ::   kjpt     ! number of tracers 
    314       REAL(wp), DIMENSION(jpi,jpj,jpk         ), INTENT(in   ) ::   pW      ! vertical velocity  
     373      ! TEMP: This can be A2D after trd_tra is tiled 
     374      REAL(wp), DIMENSION(jpi,jpj,jpk         ), INTENT(in   ) ::   pW      ! vertical velocity 
    315375      REAL(wp), DIMENSION(jpi,jpj,jpk,kjpt,jpt), INTENT(inout) ::   pt              ! active tracers and RHS of tracer equation 
    316376      ! 
     377      ! TEMP: This change not necessary after trd_tra is tiled 
     378      INTEGER  ::   itile 
    317379      INTEGER  ::   ji, jj, jk, jn   ! dummy loop indices 
    318       REAL(wp), DIMENSION(jpi,jpj,jpk) ::   zwz   ! 3D workspace 
    319       !!---------------------------------------------------------------------- 
    320       ! 
     380      REAL(wp), DIMENSION(A2D,jpk) ::   zwz   ! 3D workspace 
     381      ! TEMP: This change not necessary after trd_tra is tiled 
     382      REAL(wp), DIMENSION(:,:,:), SAVE, ALLOCATABLE ::   ztrdz 
     383      !!---------------------------------------------------------------------- 
     384      ! TEMP: This change not necessary after trd_tra is tiled 
     385      itile = ntile 
     386      ! 
     387      ! TEMP: This can be A2D after trd_tra is tiled 
     388      IF( ntile == 0 .OR. ntile == 1 )  THEN                       ! Do only on the first tile 
     389         IF( kt == nit000 .AND. l_trd ) THEN 
     390            ALLOCATE( ztrdz(jpi,jpj,jpk) ) 
     391         ENDIF 
     392      ENDIF 
     393 
    321394      zwz(:,:, 1 ) = 0._wp       ! surface & bottom values set to zero for all tracers 
    322395      zwz(:,:,jpk) = 0._wp 
    … …  
    330403         END_3D 
    331404         IF( ln_linssh ) THEN                !* top value   (only in linear free surf. as zwz is multiplied by wmask) 
     405            ! TODO: NOT TESTED- requires isf 
    332406            IF( ln_isfcav ) THEN                  ! ice-shelf cavities (top of the ocean) 
    333407               DO_2D_11_11 
    … …  
    335409               END_2D 
    336410            ELSE                                   ! no ocean cavities (only ocean surface) 
    337                zwz(:,:,1) = pW(:,:,1) * pt(:,:,1,jn,Kmm) 
     411               DO_2D_11_11 
     412                  zwz(ji,jj,1) = pW(ji,jj,1) * pt(ji,jj,1,jn,Kmm) 
     413               END_2D 
    338414            ENDIF 
    339415         ENDIF 
    … …  
    344420         END_3D 
    345421         !                                 ! Send trends for diagnostic 
    346          IF( l_trd )  CALL trd_tra( kt, Kmm, Krhs, cdtype, jn, jptra_zad, zwz, pW, pt(:,:,:,jn,Kmm) ) 
     422         ! TEMP: These changes not necessary after trd_tra is tiled 
     423         IF( l_trd )  THEN 
     424            DO_3D_11_11( 1, jpk ) 
     425               ztrdz(ji,jj,jk) = zwz(ji,jj,jk) 
     426            END_3D 
     427 
     428            IF( ntile == 0 .OR. ntile == nijtile )  THEN                ! Do only for the full domain 
     429               IF( ln_tile ) CALL dom_tile( ntsi, ntsj, ntei, ntej, ktile = 0 )         ! Use full domain 
     430 
     431               ! TODO: TO BE TILED- trd_tra 
     432               CALL trd_tra( kt, Kmm, Krhs, cdtype, jn, jptra_zad, ztrdz, pW, pt(:,:,:,jn,Kmm) ) 
     433 
     434               IF( ln_tile ) CALL dom_tile( ntsi, ntsj, ntei, ntej, ktile = itile )     ! Revert to tile domain 
     435            ENDIF 
     436         ENDIF 
    347437         ! 
    348438      END DO 
    … …  
    358448      !! ** Method :    
    359449      !!---------------------------------------------------------------------- 
    360       REAL(wp), DIMENSION(jpi,jpj,jpk), INTENT(in   ) ::   pfu   ! second upwind point 
    361       REAL(wp), DIMENSION(jpi,jpj,jpk), INTENT(in   ) ::   pfd   ! first douwning point 
    362       REAL(wp), DIMENSION(jpi,jpj,jpk), INTENT(in   ) ::   pfc   ! the central point (or the first upwind point) 
    363       REAL(wp), DIMENSION(jpi,jpj,jpk), INTENT(inout) ::   puc   ! input as Courant number ; output as flux 
     450      REAL(wp), DIMENSION(A2D,jpk), INTENT(in   ) ::   pfu   ! second upwind point 
     451      REAL(wp), DIMENSION(A2D,jpk), INTENT(in   ) ::   pfd   ! first douwning point 
     452      REAL(wp), DIMENSION(A2D,jpk), INTENT(in   ) ::   pfc   ! the central point (or the first upwind point) 
     453      REAL(wp), DIMENSION(A2D,jpk), INTENT(inout) ::   puc   ! input as Courant number ; output as flux 
    364454      !! 
    365455      INTEGER  ::  ji, jj, jk               ! dummy loop indices  

  • NEMO/branches/UKMO/dev_r12866_HPC-02_Daley_Tiling_trial_extra_halo/src/OCE/TRA/traadv_ubs.F90

    r12377 r13409  
    1414   USE oce            ! ocean dynamics and active tracers 
    1515   USE dom_oce        ! ocean space and time domain 
     16   ! TEMP: This change not necessary after trd_tra is tiled 
     17   USE domain, ONLY : dom_tile 
    1618   USE trc_oce        ! share passive tracers/Ocean variables 
    1719   USE trd_oce        ! trends: ocean variables 
    … …  
    9193      INTEGER                                  , INTENT(in   ) ::   kn_ubs_v        ! number of tracers 
    9294      REAL(wp)                                 , INTENT(in   ) ::   p2dt            ! tracer time-step 
     95      ! TEMP: This can be A2D after trd_tra is tiled 
    9396      REAL(wp), DIMENSION(jpi,jpj,jpk         ), INTENT(in   ) ::   pU, pV, pW      ! 3 ocean volume transport components 
    9497      REAL(wp), DIMENSION(jpi,jpj,jpk,kjpt,jpt), INTENT(inout) ::   pt              ! tracers and RHS of tracer equation 
    9598      ! 
     99      ! TEMP: This change not necessary after trd_tra is tiled 
     100      INTEGER  ::   itile 
    96101      INTEGER  ::   ji, jj, jk, jn   ! dummy loop indices 
    97102      REAL(wp) ::   ztra, zbtr, zcoef                       ! local scalars 
    98103      REAL(wp) ::   zfp_ui, zfm_ui, zcenut, ztak, zfp_wk, zfm_wk   !   -      - 
    99104      REAL(wp) ::   zfp_vj, zfm_vj, zcenvt, zeeu, zeev, z_hdivn    !   -      - 
    100       REAL(wp), DIMENSION(jpi,jpj,jpk) ::   ztu, ztv, zltu, zltv, zti, ztw   ! 3D workspace 
    101       !!---------------------------------------------------------------------- 
    102       ! 
    103       IF( kt == kit000 )  THEN 
    104          IF(lwp) WRITE(numout,*) 
    105          IF(lwp) WRITE(numout,*) 'tra_adv_ubs :  horizontal UBS advection scheme on ', cdtype 
    106          IF(lwp) WRITE(numout,*) '~~~~~~~~~~~~' 
     105      REAL(wp), DIMENSION(A2D,jpk) ::   ztu, ztv, zltu, zltv, zti, ztw     ! 3D workspace 
     106      ! TEMP: This change not necessary after trd_tra is tiled 
     107      REAL(wp), DIMENSION(:,:,:), SAVE, ALLOCATABLE ::   ztrdx, ztrdy, ztrdz 
     108      !!---------------------------------------------------------------------- 
     109      ! TEMP: This change not necessary after trd_tra is tiled 
     110      itile = ntile 
     111      ! 
     112      IF( ntile == 0 .OR. ntile == 1 )  THEN                       ! Do only on the first tile 
     113         IF( kt == kit000 )  THEN 
     114            IF(lwp) WRITE(numout,*) 
     115            IF(lwp) WRITE(numout,*) 'tra_adv_ubs :  horizontal UBS advection scheme on ', cdtype 
     116            IF(lwp) WRITE(numout,*) '~~~~~~~~~~~~' 
     117         ENDIF 
     118         ! 
     119         l_trd = .FALSE. 
     120         l_hst = .FALSE. 
     121         l_ptr = .FALSE. 
     122         IF( ( cdtype == 'TRA' .AND. l_trdtra ) .OR. ( cdtype == 'TRC' .AND. l_trdtrc ) )      l_trd = .TRUE. 
     123         IF(   cdtype == 'TRA' .AND. ( iom_use( 'sophtadv' ) .OR. iom_use( 'sophtadv' ) )  )   l_ptr = .TRUE. 
     124         IF(   cdtype == 'TRA' .AND. ( iom_use("uadv_heattr") .OR. iom_use("vadv_heattr") .OR. & 
     125            &                          iom_use("uadv_salttr") .OR. iom_use("vadv_salttr")  ) ) l_hst = .TRUE. 
     126 
     127         ! TEMP: This can be A2D after trd_tra is tiled 
     128         IF( kt == kit000 .AND. l_trd ) THEN 
     129            ALLOCATE( ztrdx(jpi,jpj,jpk), ztrdy(jpi,jpj,jpk), ztrdz(jpi,jpj,jpk) ) 
     130         ENDIF 
    107131      ENDIF 
    108       ! 
    109       l_trd = .FALSE. 
    110       l_hst = .FALSE. 
    111       l_ptr = .FALSE. 
    112       IF( ( cdtype == 'TRA' .AND. l_trdtra ) .OR. ( cdtype == 'TRC' .AND. l_trdtrc ) )      l_trd = .TRUE. 
    113       IF(   cdtype == 'TRA' .AND. ( iom_use( 'sophtadv' ) .OR. iom_use( 'sophtadv' ) )  )   l_ptr = .TRUE.  
    114       IF(   cdtype == 'TRA' .AND. ( iom_use("uadv_heattr") .OR. iom_use("vadv_heattr") .OR. & 
    115          &                          iom_use("uadv_salttr") .OR. iom_use("vadv_salttr")  ) ) l_hst = .TRUE. 
    116132      ! 
    117133      ztw (:,:, 1 ) = 0._wp      ! surface & bottom value : set to zero for all tracers 
    … …  
    152168         END_3D 
    153169         ! 
    154          zltu(:,:,:) = pt(:,:,:,jn,Krhs)      ! store the initial trends before its update 
     170         DO_3D_11_11( 1, jpk ) 
     171            zltu(ji,jj,jk) = pt(ji,jj,jk,jn,Krhs)      ! store the initial trends before its update 
     172         END_3D 
    155173         ! 
    156174         DO jk = 1, jpkm1        !==  add the horizontal advective trend  ==! 
    … …  
    163181         END DO 
    164182         ! 
    165          zltu(:,:,:) = pt(:,:,:,jn,Krhs) - zltu(:,:,:)    ! Horizontal advective trend used in vertical 2nd order FCT case 
    166          !                                            ! and/or in trend diagnostic (l_trd=T)  
    167          !                 
     183         DO_3D_11_11( 1, jpk ) 
     184            zltu(ji,jj,jk) = pt(ji,jj,jk,jn,Krhs) - zltu(ji,jj,jk)  ! Horizontal advective trend used in vertical 2nd order FCT case 
     185         END_3D                                                     ! and/or in trend diagnostic (l_trd=T) 
     186         ! 
     187         ! TEMP: These changes not necessary after trd_tra is tiled 
    168188         IF( l_trd ) THEN                  ! trend diagnostics 
    169              CALL trd_tra( kt, Kmm, Krhs, cdtype, jn, jptra_xad, ztu, pU, pt(:,:,:,jn,Kmm) ) 
    170              CALL trd_tra( kt, Kmm, Krhs, cdtype, jn, jptra_yad, ztv, pV, pt(:,:,:,jn,Kmm) ) 
     189            DO_3D_11_11( 1, jpk ) 
     190               ztrdx(ji,jj,jk) = ztu(ji,jj,jk) 
     191               ztrdy(ji,jj,jk) = ztv(ji,jj,jk) 
     192            END_3D 
     193 
     194            IF( ntile == 0 .OR. ntile == nijtile )  THEN                ! Do only for the full domain 
     195               IF( ln_tile ) CALL dom_tile( ntsi, ntsj, ntei, ntej, ktile = 0 )         ! Use full domain 
     196 
     197               ! TODO: TO BE TILED- trd_tra 
     198               CALL trd_tra( kt, Kmm, Krhs, cdtype, jn, jptra_xad, ztrdx, pU, pt(:,:,:,jn,Kmm) ) 
     199               CALL trd_tra( kt, Kmm, Krhs, cdtype, jn, jptra_yad, ztrdy, pV, pt(:,:,:,jn,Kmm) ) 
     200 
     201               IF( ln_tile ) CALL dom_tile( ntsi, ntsj, ntei, ntej, ktile = itile )     ! Revert to tile domain 
     202            ENDIF 
    171203         END IF 
    172204         !      
    … …  
    183215         CASE(  2  )                   ! 2nd order FCT  
    184216            !          
    185             IF( l_trd )   zltv(:,:,:) = pt(:,:,:,jn,Krhs)          ! store pt(:,:,:,:,Krhs) if trend diag. 
     217            IF( l_trd ) THEN 
     218               DO_3D_11_11( 1, jpk ) 
     219                  zltv(ji,jj,jk) = pt(ji,jj,jk,jn,Krhs)          ! store pt(:,:,:,:,Krhs) if trend diag. 
     220               END_3D 
     221            ENDIF 
    186222            ! 
    187223            !                          !*  upstream advection with initial mass fluxes & intermediate update  ==! 
    … …  
    192228            END_3D 
    193229            IF( ln_linssh ) THEN             ! top ocean value (only in linear free surface as ztw has been w-masked) 
     230               ! TODO: NOT TESTED- requires isf 
    194231               IF( ln_isfcav ) THEN                ! top of the ice-shelf cavities and at the ocean surface 
    195232                  DO_2D_11_11 
    … …  
    197234                  END_2D 
    198235               ELSE                                ! no cavities: only at the ocean surface 
    199                   ztw(:,:,1) = pW(:,:,1) * pt(:,:,1,jn,Kbb) 
     236                  DO_2D_11_11 
     237                     ztw(ji,jj,1) = pW(ji,jj,1) * pt(ji,jj,1,jn,Kbb) 
     238                  END_2D 
    200239               ENDIF 
    201240            ENDIF 
    … …  
    223262               ztw(ji,jj,jk) = pW(ji,jj,jk) * ztw(ji,jj,jk) * wmask(ji,jj,jk) 
    224263            END_3D 
    225             IF( ln_linssh )   ztw(:,:, 1 ) = pW(:,:,1) * pt(:,:,1,jn,Kmm)     !!gm ISF & 4th COMPACT doesn't work 
     264            IF( ln_linssh ) THEN 
     265               DO_2D_11_11 
     266                  ztw(ji,jj,1) = pW(ji,jj,1) * pt(ji,jj,1,jn,Kmm)     !!gm ISF & 4th COMPACT doesn't work 
     267               END_2D 
     268            ENDIF 
    226269            ! 
    227270         END SELECT 
    … …  
    231274         END_3D 
    232275         ! 
     276         ! TEMP: These changes not necessary after trd_tra is tiled 
    233277         IF( l_trd )  THEN       ! vertical advective trend diagnostics 
    234278            DO_3D_00_00( 1, jpkm1 ) 
    235                zltv(ji,jj,jk) = pt(ji,jj,jk,jn,Krhs) - zltv(ji,jj,jk)                          & 
     279               ztrdz(ji,jj,jk) = pt(ji,jj,jk,jn,Krhs) - zltv(ji,jj,jk)                          & 
    236280                  &           + pt(ji,jj,jk,jn,Kmm) * (  pW(ji,jj,jk) - pW(ji,jj,jk+1)  )   & 
    237281                  &                              * r1_e1e2t(ji,jj) / e3t(ji,jj,jk,Kmm) 
    238282            END_3D 
    239             CALL trd_tra( kt, Kmm, Krhs, cdtype, jn, jptra_zad, zltv ) 
     283 
     284            IF( ntile == 0 .OR. ntile == nijtile )  THEN                ! Do only for the full domain 
     285               IF( ln_tile ) CALL dom_tile( ntsi, ntsj, ntei, ntej, ktile = 0 )         ! Use full domain 
     286 
     287               ! TODO: TO BE TILED- trd_tra 
     288               CALL trd_tra( kt, Kmm, Krhs, cdtype, jn, jptra_zad, ztrdz ) 
     289 
     290               IF( ln_tile ) CALL dom_tile( ntsi, ntsj, ntei, ntej, ktile = nijtile )   ! Revert to tile domain 
     291            ENDIF 
    240292         ENDIF 
    241293         ! 
    … …  
    258310      !!       in-space based differencing for fluid 
    259311      !!---------------------------------------------------------------------- 
    260       INTEGER , INTENT(in   )                          ::   Kmm    ! time level index 
    261       REAL(wp), INTENT(in   )                          ::   p2dt   ! tracer time-step 
    262       REAL(wp),                DIMENSION (jpi,jpj,jpk) ::   pbef   ! before field 
    263       REAL(wp), INTENT(inout), DIMENSION (jpi,jpj,jpk) ::   paft   ! after field 
    264       REAL(wp), INTENT(inout), DIMENSION (jpi,jpj,jpk) ::   pcc    ! monotonic flux in the k direction 
     312      INTEGER , INTENT(in   )                         ::   Kmm    ! time level index 
     313      REAL(wp), INTENT(in   )                         ::   p2dt   ! tracer time-step 
     314      REAL(wp),                DIMENSION(jpi,jpj,jpk) ::   pbef   ! before field 
     315      REAL(wp), INTENT(inout), DIMENSION(A2D    ,jpk) ::   paft   ! after field 
     316      REAL(wp), INTENT(inout), DIMENSION(A2D    ,jpk) ::   pcc    ! monotonic flux in the k direction 
    265317      ! 
    266318      INTEGER  ::   ji, jj, jk   ! dummy loop indices 
    267319      INTEGER  ::   ikm1         ! local integer 
    268320      REAL(wp) ::   zpos, zneg, zbt, za, zb, zc, zbig, zrtrn   ! local scalars 
    269       REAL(wp), DIMENSION(jpi,jpj,jpk) ::   zbetup, zbetdo     ! 3D workspace 
     321      REAL(wp), DIMENSION(A2D,jpk) ::   zbetup, zbetdo         ! 3D workspace 
    270322      !!---------------------------------------------------------------------- 
    271323      ! 
    … …  
    277329      ! -------------------- 
    278330      !                    ! large negative value (-zbig) inside land 
    279       pbef(:,:,:) = pbef(:,:,:) * tmask(:,:,:) - zbig * ( 1.e0 - tmask(:,:,:) ) 
    280       paft(:,:,:) = paft(:,:,:) * tmask(:,:,:) - zbig * ( 1.e0 - tmask(:,:,:) ) 
     331      DO_3D_00_00( 1, jpk ) 
     332         pbef(ji,jj,jk) = pbef(ji,jj,jk) * tmask(ji,jj,jk) - zbig * ( 1.e0 - tmask(ji,jj,jk) ) 
     333         paft(ji,jj,jk) = paft(ji,jj,jk) * tmask(ji,jj,jk) - zbig * ( 1.e0 - tmask(ji,jj,jk) ) 
     334      END_3D 
    281335      ! 
    282336      DO jk = 1, jpkm1     ! search maximum in neighbourhood 
    … …  
    289343      END DO 
    290344      !                    ! large positive value (+zbig) inside land 
    291       pbef(:,:,:) = pbef(:,:,:) * tmask(:,:,:) + zbig * ( 1.e0 - tmask(:,:,:) ) 
    292       paft(:,:,:) = paft(:,:,:) * tmask(:,:,:) + zbig * ( 1.e0 - tmask(:,:,:) ) 
     345      DO_3D_00_00( 1, jpk ) 
     346         pbef(ji,jj,jk) = pbef(ji,jj,jk) * tmask(ji,jj,jk) + zbig * ( 1.e0 - tmask(ji,jj,jk) ) 
     347         paft(ji,jj,jk) = paft(ji,jj,jk) * tmask(ji,jj,jk) + zbig * ( 1.e0 - tmask(ji,jj,jk) ) 
     348      END_3D 
    293349      ! 
    294350      DO jk = 1, jpkm1     ! search minimum in neighbourhood 
    … …  
    301357      END DO 
    302358      !                    ! restore masked values to zero 
    303       pbef(:,:,:) = pbef(:,:,:) * tmask(:,:,:) 
    304       paft(:,:,:) = paft(:,:,:) * tmask(:,:,:) 
     359      DO_3D_00_00( 1, jpk ) 
     360         pbef(ji,jj,jk) = pbef(ji,jj,jk) * tmask(ji,jj,jk) 
     361         paft(ji,jj,jk) = paft(ji,jj,jk) * tmask(ji,jj,jk) 
     362      END_3D 
    305363      ! 
    306364      ! Positive and negative part of fluxes and beta terms 

  • NEMO/branches/UKMO/dev_r12866_HPC-02_Daley_Tiling_trial_extra_halo/src/OCE/TRA/trabbc.F90

    r12489 r13409  
    1717   USE oce            ! ocean variables 
    1818   USE dom_oce        ! domain: ocean 
     19   ! TEMP: This change not necessary after trd_tra is tiled 
     20   USE domain, ONLY : dom_tile 
    1921   USE phycst         ! physical constants 
    2022   USE trd_oce        ! trends: ocean variables 
    … …  
    7981      REAL(wp), DIMENSION(jpi,jpj,jpk,jpts,jpt), INTENT(inout) :: pts        ! active tracers and RHS of tracer equation 
    8082      ! 
    81       INTEGER  ::   ji, jj    ! dummy loop indices 
    82       REAL(wp), ALLOCATABLE, DIMENSION(:,:,:) ::   ztrdt   ! 3D workspace 
     83      INTEGER  ::   ji, jj, jk    ! dummy loop indices 
     84      REAL(wp), SAVE :: zsum1 
     85      ! TEMP: This change not necessary after trd_tra is tiled 
     86      REAL(wp), ALLOCATABLE, SAVE, DIMENSION(:,:,:) ::   ztrdt   ! 3D workspace 
    8387      !!---------------------------------------------------------------------- 
    8488      ! 
    8589      IF( ln_timing )   CALL timing_start('tra_bbc') 
    8690      ! 
    87       IF( l_trdtra )   THEN         ! Save the input temperature trend 
    88          ALLOCATE( ztrdt(jpi,jpj,jpk) ) 
    89          ztrdt(:,:,:) = pts(:,:,:,jp_tem,Krhs) 
     91      IF( l_trdtra ) THEN           ! Save the input temperature trend 
     92         IF( ntile == 0 .OR. ntile == 1 )  THEN                       ! Do only on the first tile 
     93            ! TEMP: This can be A2D after trd_tra is tiled 
     94            ALLOCATE( ztrdt(jpi,jpj,jpk) ) 
     95         ENDIF 
     96 
     97         DO_3D_00_00( 1, jpkm1 ) 
     98            ztrdt(ji,jj,jk) = pts(ji,jj,jk,jp_tem,Krhs) 
     99         END_3D 
    90100      ENDIF 
    91101      !                             !  Add the geothermal trend on temperature 
    … …  
    94104      END_2D 
    95105      ! 
    96       CALL lbc_lnk( 'trabbc', pts(:,:,:,jp_tem,Krhs) , 'T', 1. ) 
    97       ! 
    98       IF( l_trdtra ) THEN        ! Send the trend for diagnostics 
    99          ztrdt(:,:,:) = pts(:,:,:,jp_tem,Krhs) - ztrdt(:,:,:) 
    100          CALL trd_tra( kt, Kmm, Krhs, 'TRA', jp_tem, jptra_bbc, ztrdt ) 
    101          DEALLOCATE( ztrdt ) 
    102       ENDIF 
    103       ! 
    104       CALL iom_put ( "hfgeou" , rho0_rcp * qgh_trd0(:,:) ) 
    105       IF(sn_cfctl%l_prtctl)   CALL prt_ctl( tab3d_1=pts(:,:,:,jp_tem,Krhs), clinfo1=' bbc  - Ta: ', mask1=tmask, clinfo3='tra-ta' ) 
     106      ! TEMP: These changes not necessary after trd_tra is tiled, lbc_lnk not necessary if using XIOS (subdomain support, will not output haloes) 
     107      IF( l_trdtra ) THEN 
     108         DO_3D_00_00( 1, jpkm1 ) 
     109            ztrdt(ji,jj,jk) = pts(ji,jj,jk,jp_tem,Krhs) - ztrdt(ji,jj,jk) 
     110         END_3D 
     111      ENDIF 
     112 
     113      IF( ntile == 0 .OR. ntile == nijtile )  THEN                ! Do only for the full domain 
     114         ! NOTE: I don't think pts needs to be the input here? 
     115         CALL lbc_lnk( 'trabbc', ztrdt(:,:,:) , 'T', 1. ) 
     116         ! 
     117         IF( l_trdtra ) THEN        ! Send the trend for diagnostics 
     118            IF( ln_tile ) CALL dom_tile( ntsi, ntsj, ntei, ntej, ktile = 0 )         ! Use full domain 
     119 
     120            ! TODO: TO BE TILED- trd_tra 
     121            CALL trd_tra( kt, Kmm, Krhs, 'TRA', jp_tem, jptra_bbc, ztrdt ) 
     122            DEALLOCATE( ztrdt ) 
     123 
     124            IF( ln_tile ) CALL dom_tile( ntsi, ntsj, ntei, ntej, ktile = nijtile )   ! Revert to tile domain 
     125         ENDIF 
     126         ! 
     127         CALL iom_put ( "hfgeou" , rho0_rcp * qgh_trd0(:,:) ) 
     128      ENDIF 
     129 
     130      IF(sn_cfctl%l_prtctl)   CALL prt_ctl( tab3d_1=pts(:,:,:,jp_tem,Krhs), clinfo1=' bbc  - Ta: ', mask1=tmask, psum1=zsum1, & 
     131         &                                  clinfo3='tra-ta' ) 
    106132      ! 
    107133      IF( ln_timing )   CALL timing_stop('tra_bbc') 

  • NEMO/branches/UKMO/dev_r12866_HPC-02_Daley_Tiling_trial_extra_halo/src/OCE/TRA/trabbl.F90

    r12377 r13409  
    2626   USE oce            ! ocean dynamics and active tracers 
    2727   USE dom_oce        ! ocean space and time domain 
     28   ! TEMP: This change not necessary after trd_tra is tiled 
     29   USE domain, ONLY : dom_tile 
    2830   USE phycst         ! physical constant 
    2931   USE eosbn2         ! equation of state 
    … …  
    105107      REAL(wp), DIMENSION(jpi,jpj,jpk,jpts,jpt), INTENT(inout) :: pts             ! active tracers and RHS of tracer equation 
    106108      ! 
    107       REAL(wp), ALLOCATABLE, DIMENSION(:,:,:) ::   ztrdt, ztrds 
     109      INTEGER  ::   ji, jj, jk   ! Dummy loop indices 
     110      REAL(wp), SAVE :: zsum1, zsum2, zsum3, zsum4 
     111      ! TEMP: This change not necessary after trd_tra is tiled 
     112      REAL(wp), ALLOCATABLE, SAVE, DIMENSION(:,:,:) ::   ztrdt, ztrds 
    108113      !!---------------------------------------------------------------------- 
    109114      ! 
    … …  
    111116      ! 
    112117      IF( l_trdtra )   THEN                         !* Save the T-S input trends 
    113          ALLOCATE( ztrdt(jpi,jpj,jpk) , ztrds(jpi,jpj,jpk) ) 
    114          ztrdt(:,:,:) = pts(:,:,:,jp_tem,Krhs) 
    115          ztrds(:,:,:) = pts(:,:,:,jp_sal,Krhs) 
     118         IF( ntile == 0 .OR. ntile == 1 )  THEN                       ! Do only on the first tile 
     119            ! TEMP: This can be A2D after trd_tra is tiled 
     120            ALLOCATE( ztrdt(jpi,jpj,jpk), ztrds(jpi,jpj,jpk) ) 
     121         ENDIF 
     122 
     123         DO_3D_00_00( 1, jpkm1 ) 
     124            ztrdt(ji,jj,jk) = pts(ji,jj,jk,jp_tem,Krhs) 
     125            ztrds(ji,jj,jk) = pts(ji,jj,jk,jp_sal,Krhs) 
     126         END_3D 
    116127      ENDIF 
    117128 
    … …  
    122133         CALL tra_bbl_dif( pts(:,:,:,:,Kbb), pts(:,:,:,:,Krhs), jpts, Kmm ) 
    123134         IF( sn_cfctl%l_prtctl )  & 
    124          CALL prt_ctl( tab3d_1=pts(:,:,:,jp_tem,Krhs), clinfo1=' bbl_ldf  - Ta: ', mask1=tmask, & 
    125             &          tab3d_2=pts(:,:,:,jp_sal,Krhs), clinfo2=           ' Sa: ', mask2=tmask, clinfo3='tra' ) 
    126          ! lateral boundary conditions ; just need for outputs 
    127          CALL lbc_lnk_multi( 'trabbl', ahu_bbl, 'U', 1. , ahv_bbl, 'V', 1. ) 
    128          CALL iom_put( "ahu_bbl", ahu_bbl )   ! bbl diffusive flux i-coef 
    129          CALL iom_put( "ahv_bbl", ahv_bbl )   ! bbl diffusive flux j-coef 
     135         CALL prt_ctl( tab3d_1=pts(:,:,:,jp_tem,Krhs), clinfo1=' bbl_ldf  - Ta: ', mask1=tmask, psum1=zsum1, & 
     136            &          tab3d_2=pts(:,:,:,jp_sal,Krhs), clinfo2=           ' Sa: ', mask2=tmask, psum2=zsum2, & 
     137            &          clinfo3='tra' ) 
     138         IF( ntile == 0 .OR. ntile == nijtile ) THEN                       ! Do only on the last tile 
     139            ! lateral boundary conditions ; just need for outputs 
     140            CALL lbc_lnk_multi( 'trabbl', ahu_bbl, 'U', 1. , ahv_bbl, 'V', 1. ) 
     141            CALL iom_put( "ahu_bbl", ahu_bbl )   ! bbl diffusive flux i-coef 
     142            CALL iom_put( "ahv_bbl", ahv_bbl )   ! bbl diffusive flux j-coef 
     143         ENDIF 
    130144         ! 
    131145      ENDIF 
    … …  
    135149         CALL tra_bbl_adv( pts(:,:,:,:,Kbb), pts(:,:,:,:,Krhs), jpts, Kmm ) 
    136150         IF(sn_cfctl%l_prtctl)   & 
    137          CALL prt_ctl( tab3d_1=pts(:,:,:,jp_tem,Krhs), clinfo1=' bbl_adv  - Ta: ', mask1=tmask,   & 
    138             &          tab3d_2=pts(:,:,:,jp_sal,Krhs), clinfo2=           ' Sa: ', mask2=tmask, clinfo3='tra' ) 
    139          ! lateral boundary conditions ; just need for outputs 
    140          CALL lbc_lnk_multi( 'trabbl', utr_bbl, 'U', 1. , vtr_bbl, 'V', 1. ) 
    141          CALL iom_put( "uoce_bbl", utr_bbl )  ! bbl i-transport 
    142          CALL iom_put( "voce_bbl", vtr_bbl )  ! bbl j-transport 
    143          ! 
    144       ENDIF 
    145  
     151         CALL prt_ctl( tab3d_1=pts(:,:,:,jp_tem,Krhs), clinfo1=' bbl_adv  - Ta: ', mask1=tmask, psum1=zsum3, & 
     152            &          tab3d_2=pts(:,:,:,jp_sal,Krhs), clinfo2=           ' Sa: ', mask2=tmask, psum2=zsum4, & 
     153            &          clinfo3='tra' ) 
     154         IF( ntile == 0 .OR. ntile == nijtile ) THEN                       ! Do only on the last tile 
     155            ! lateral boundary conditions ; just need for outputs 
     156            CALL lbc_lnk_multi( 'trabbl', utr_bbl, 'U', 1. , vtr_bbl, 'V', 1. ) 
     157            CALL iom_put( "uoce_bbl", utr_bbl )  ! bbl i-transport 
     158            CALL iom_put( "voce_bbl", vtr_bbl )  ! bbl j-transport 
     159         ENDIF 
     160         ! 
     161      ENDIF 
     162 
     163      ! TEMP: These changes not necessary after trd_tra is tiled 
    146164      IF( l_trdtra )   THEN                      ! send the trends for further diagnostics 
    147          ztrdt(:,:,:) = pts(:,:,:,jp_tem,Krhs) - ztrdt(:,:,:) 
    148          ztrds(:,:,:) = pts(:,:,:,jp_sal,Krhs) - ztrds(:,:,:) 
    149          CALL trd_tra( kt, Kmm, Krhs, 'TRA', jp_tem, jptra_bbl, ztrdt ) 
    150          CALL trd_tra( kt, Kmm, Krhs, 'TRA', jp_sal, jptra_bbl, ztrds ) 
    151          DEALLOCATE( ztrdt, ztrds ) 
     165         DO_3D_00_00( 1, jpkm1 ) 
     166            ztrdt(ji,jj,jk) = pts(ji,jj,jk,jp_tem,Krhs) - ztrdt(ji,jj,jk) 
     167            ztrds(ji,jj,jk) = pts(ji,jj,jk,jp_sal,Krhs) - ztrds(ji,jj,jk) 
     168         END_3D 
     169 
     170         IF( ntile == 0 .OR. ntile == nijtile )  THEN                ! Do only for the full domain 
     171            IF( ln_tile ) CALL dom_tile( ntsi, ntsj, ntei, ntej, ktile = 0 )         ! Use full domain 
     172 
     173            ! TODO: TO BE TILED- trd_tra 
     174            CALL trd_tra( kt, Kmm, Krhs, 'TRA', jp_tem, jptra_bbl, ztrdt ) 
     175            CALL trd_tra( kt, Kmm, Krhs, 'TRA', jp_sal, jptra_bbl, ztrds ) 
     176            DEALLOCATE( ztrdt, ztrds ) 
     177 
     178            IF( ln_tile ) CALL dom_tile( ntsi, ntsj, ntei, ntej, ktile = nijtile )   ! Revert to tile domain 
     179         ENDIF 
    152180      ENDIF 
    153181      ! 
    … …  
    186214      INTEGER  ::   ik           ! local integers 
    187215      REAL(wp) ::   zbtr         ! local scalars 
    188       REAL(wp), DIMENSION(jpi,jpj) ::   zptb   ! workspace 
     216      REAL(wp), DIMENSION(A2D) ::   zptb   ! workspace 
    189217      !!---------------------------------------------------------------------- 
    190218      ! 
    … …  
    241269      DO jn = 1, kjpt                                            ! tracer loop 
    242270         !                                                       ! =========== 
    243          DO jj = 1, jpjm1 
    244             DO ji = 1, jpim1            ! CAUTION start from i=1 to update i=2 when cyclic east-west 
    245                IF( utr_bbl(ji,jj) /= 0.e0 ) THEN            ! non-zero i-direction bbl advection 
    246                   ! down-slope i/k-indices (deep)      &   up-slope i/k indices (shelf) 
    247                   iid  = ji + MAX( 0, mgrhu(ji,jj) )   ;   iis  = ji + 1 - MAX( 0, mgrhu(ji,jj) ) 
    248                   ikud = mbku_d(ji,jj)                 ;   ikus = mbku(ji,jj) 
    249                   zu_bbl = ABS( utr_bbl(ji,jj) ) 
    250                   ! 
    251                   !                                               ! up  -slope T-point (shelf bottom point) 
    252                   zbtr = r1_e1e2t(iis,jj) / e3t(iis,jj,ikus,Kmm) 
    253                   ztra = zu_bbl * ( pt(iid,jj,ikus,jn) - pt(iis,jj,ikus,jn) ) * zbtr 
    254                   pt_rhs(iis,jj,ikus,jn) = pt_rhs(iis,jj,ikus,jn) + ztra 
    255                   ! 
    256                   DO jk = ikus, ikud-1                            ! down-slope upper to down T-point (deep column) 
    257                      zbtr = r1_e1e2t(iid,jj) / e3t(iid,jj,jk,Kmm) 
    258                      ztra = zu_bbl * ( pt(iid,jj,jk+1,jn) - pt(iid,jj,jk,jn) ) * zbtr 
    259                      pt_rhs(iid,jj,jk,jn) = pt_rhs(iid,jj,jk,jn) + ztra 
    260                   END DO 
    261                   ! 
    262                   zbtr = r1_e1e2t(iid,jj) / e3t(iid,jj,ikud,Kmm) 
    263                   ztra = zu_bbl * ( pt(iis,jj,ikus,jn) - pt(iid,jj,ikud,jn) ) * zbtr 
    264                   pt_rhs(iid,jj,ikud,jn) = pt_rhs(iid,jj,ikud,jn) + ztra 
    265                ENDIF 
    266                ! 
    267                IF( vtr_bbl(ji,jj) /= 0.e0 ) THEN            ! non-zero j-direction bbl advection 
    268                   ! down-slope j/k-indices (deep)        &   up-slope j/k indices (shelf) 
    269                   ijd  = jj + MAX( 0, mgrhv(ji,jj) )     ;   ijs  = jj + 1 - MAX( 0, mgrhv(ji,jj) ) 
    270                   ikvd = mbkv_d(ji,jj)                   ;   ikvs = mbkv(ji,jj) 
    271                   zv_bbl = ABS( vtr_bbl(ji,jj) ) 
    272                   ! 
    273                   ! up  -slope T-point (shelf bottom point) 
    274                   zbtr = r1_e1e2t(ji,ijs) / e3t(ji,ijs,ikvs,Kmm) 
    275                   ztra = zv_bbl * ( pt(ji,ijd,ikvs,jn) - pt(ji,ijs,ikvs,jn) ) * zbtr 
    276                   pt_rhs(ji,ijs,ikvs,jn) = pt_rhs(ji,ijs,ikvs,jn) + ztra 
    277                   ! 
    278                   DO jk = ikvs, ikvd-1                            ! down-slope upper to down T-point (deep column) 
    279                      zbtr = r1_e1e2t(ji,ijd) / e3t(ji,ijd,jk,Kmm) 
    280                      ztra = zv_bbl * ( pt(ji,ijd,jk+1,jn) - pt(ji,ijd,jk,jn) ) * zbtr 
    281                      pt_rhs(ji,ijd,jk,jn) = pt_rhs(ji,ijd,jk,jn)  + ztra 
    282                   END DO 
    283                   !                                               ! down-slope T-point (deep bottom point) 
    284                   zbtr = r1_e1e2t(ji,ijd) / e3t(ji,ijd,ikvd,Kmm) 
    285                   ztra = zv_bbl * ( pt(ji,ijs,ikvs,jn) - pt(ji,ijd,ikvd,jn) ) * zbtr 
    286                   pt_rhs(ji,ijd,ikvd,jn) = pt_rhs(ji,ijd,ikvd,jn) + ztra 
    287                ENDIF 
    288             END DO 
     271         DO_2D_10_10            ! CAUTION start from i=1 to update i=2 when cyclic east-west 
     272            IF( utr_bbl(ji,jj) /= 0.e0 ) THEN            ! non-zero i-direction bbl advection 
     273               ! down-slope i/k-indices (deep)      &   up-slope i/k indices (shelf) 
     274               iid  = ji + MAX( 0, mgrhu(ji,jj) )   ;   iis  = ji + 1 - MAX( 0, mgrhu(ji,jj) ) 
     275               ikud = mbku_d(ji,jj)                 ;   ikus = mbku(ji,jj) 
     276               zu_bbl = ABS( utr_bbl(ji,jj) ) 
     277               ! 
     278               !                                               ! up  -slope T-point (shelf bottom point) 
     279               zbtr = r1_e1e2t(iis,jj) / e3t(iis,jj,ikus,Kmm) 
     280               ztra = zu_bbl * ( pt(iid,jj,ikus,jn) - pt(iis,jj,ikus,jn) ) * zbtr 
     281               pt_rhs(iis,jj,ikus,jn) = pt_rhs(iis,jj,ikus,jn) + ztra 
     282               ! 
     283               DO jk = ikus, ikud-1                            ! down-slope upper to down T-point (deep column) 
     284                  zbtr = r1_e1e2t(iid,jj) / e3t(iid,jj,jk,Kmm) 
     285                  ztra = zu_bbl * ( pt(iid,jj,jk+1,jn) - pt(iid,jj,jk,jn) ) * zbtr 
     286                  pt_rhs(iid,jj,jk,jn) = pt_rhs(iid,jj,jk,jn) + ztra 
     287               END DO 
     288               ! 
     289               zbtr = r1_e1e2t(iid,jj) / e3t(iid,jj,ikud,Kmm) 
     290               ztra = zu_bbl * ( pt(iis,jj,ikus,jn) - pt(iid,jj,ikud,jn) ) * zbtr 
     291               pt_rhs(iid,jj,ikud,jn) = pt_rhs(iid,jj,ikud,jn) + ztra 
     292            ENDIF 
    289293            ! 
    290          END DO 
    291          !                                                  ! =========== 
    292       END DO                                                ! end tracer 
    293       !                                                     ! =========== 
     294            IF( vtr_bbl(ji,jj) /= 0.e0 ) THEN            ! non-zero j-direction bbl advection 
     295               ! down-slope j/k-indices (deep)        &   up-slope j/k indices (shelf) 
     296               ijd  = jj + MAX( 0, mgrhv(ji,jj) )     ;   ijs  = jj + 1 - MAX( 0, mgrhv(ji,jj) ) 
     297               ikvd = mbkv_d(ji,jj)                   ;   ikvs = mbkv(ji,jj) 
     298               zv_bbl = ABS( vtr_bbl(ji,jj) ) 
     299               ! 
     300               ! up  -slope T-point (shelf bottom point) 
     301               zbtr = r1_e1e2t(ji,ijs) / e3t(ji,ijs,ikvs,Kmm) 
     302               ztra = zv_bbl * ( pt(ji,ijd,ikvs,jn) - pt(ji,ijs,ikvs,jn) ) * zbtr 
     303               pt_rhs(ji,ijs,ikvs,jn) = pt_rhs(ji,ijs,ikvs,jn) + ztra 
     304               ! 
     305               DO jk = ikvs, ikvd-1                            ! down-slope upper to down T-point (deep column) 
     306                  zbtr = r1_e1e2t(ji,ijd) / e3t(ji,ijd,jk,Kmm) 
     307                  ztra = zv_bbl * ( pt(ji,ijd,jk+1,jn) - pt(ji,ijd,jk,jn) ) * zbtr 
     308                  pt_rhs(ji,ijd,jk,jn) = pt_rhs(ji,ijd,jk,jn)  + ztra 
     309               END DO 
     310               !                                               ! down-slope T-point (deep bottom point) 
     311               zbtr = r1_e1e2t(ji,ijd) / e3t(ji,ijd,ikvd,Kmm) 
     312               ztra = zv_bbl * ( pt(ji,ijs,ikvs,jn) - pt(ji,ijd,ikvd,jn) ) * zbtr 
     313               pt_rhs(ji,ijd,ikvd,jn) = pt_rhs(ji,ijd,ikvd,jn) + ztra 
     314            ENDIF 
     315         END_2D 
     316         !                                                       ! =========== 
     317      END DO                                                     ! end tracer 
     318      !                                                          ! =========== 
    294319   END SUBROUTINE tra_bbl_adv 
    295320 
    … …  
    332357      REAL(wp) ::   za, zb, zgdrho            ! local scalars 
    333358      REAL(wp) ::   zsign, zsigna, zgbbl      !   -      - 
    334       REAL(wp), DIMENSION(jpi,jpj,jpts)   :: zts, zab         ! 3D workspace 
    335       REAL(wp), DIMENSION(jpi,jpj)        :: zub, zvb, zdep   ! 2D workspace 
    336       !!---------------------------------------------------------------------- 
    337       ! 
    338       IF( kt == kit000 )  THEN 
    339          IF(lwp)  WRITE(numout,*) 
    340          IF(lwp)  WRITE(numout,*) 'trabbl:bbl : Compute bbl velocities and diffusive coefficients in ', cdtype 
    341          IF(lwp)  WRITE(numout,*) '~~~~~~~~~~' 
     359      REAL(wp), DIMENSION(A2D,jpts)   :: zts, zab         ! 3D workspace 
     360      REAL(wp), DIMENSION(A2D)        :: zub, zvb, zdep   ! 2D workspace 
     361      !!---------------------------------------------------------------------- 
     362      ! 
     363      IF( ntile == 0 .OR. ntile == 1 )  THEN                       ! Do only on the first tile 
     364         IF( kt == kit000 )  THEN 
     365            IF(lwp)  WRITE(numout,*) 
     366            IF(lwp)  WRITE(numout,*) 'trabbl:bbl : Compute bbl velocities and diffusive coefficients in ', cdtype 
     367            IF(lwp)  WRITE(numout,*) '~~~~~~~~~~' 
     368         ENDIF 
    342369      ENDIF 
    343370      !                                        !* bottom variables (T, S, alpha, beta, depth, velocity) 

  • NEMO/branches/UKMO/dev_r12866_HPC-02_Daley_Tiling_trial_extra_halo/src/OCE/TRA/tradmp.F90

    r12738 r13409  
    2424   USE oce            ! ocean: variables 
    2525   USE dom_oce        ! ocean: domain variables 
     26   ! TEMP: This change not necessary after trd_tra is tiled 
     27   USE domain, ONLY : dom_tile 
    2628   USE c1d            ! 1D vertical configuration 
    2729   USE trd_oce        ! trends: ocean variables 
    … …  
    9597      ! 
    9698      INTEGER ::   ji, jj, jk, jn   ! dummy loop indices 
    97       REAL(wp), DIMENSION(jpi,jpj,jpk,jpts)     ::  zts_dta 
    98       REAL(wp), DIMENSION(:,:,:,:), ALLOCATABLE ::  ztrdts 
     99      REAL(wp), DIMENSION(A2D,jpk,jpts)     ::  zts_dta 
     100      REAL(wp), SAVE :: zsum1, zsum2 
     101      ! TEMP: This change not necessary after trd_tra is tiled 
     102      REAL(wp), DIMENSION(:,:,:,:), ALLOCATABLE, SAVE ::  ztrdts 
    99103      !!---------------------------------------------------------------------- 
    100104      ! 
    … …  
    102106      ! 
    103107      IF( l_trdtra )   THEN                    !* Save ta and sa trends 
    104          ALLOCATE( ztrdts(jpi,jpj,jpk,jpts) )  
    105          ztrdts(:,:,:,:) = pts(:,:,:,:,Krhs)  
     108         IF( ntile == 0 .OR. ntile == 1 )  THEN                       ! Do only on the first tile 
     109            ! TEMP: This can be A2D after trd_tra is tiled 
     110            ALLOCATE( ztrdts(jpi,jpj,jpk,jpts) ) 
     111         ENDIF 
     112 
     113         DO_3D_00_00( 1, jpkm1 ) 
     114            ztrdts(ji,jj,jk,:) = pts(ji,jj,jk,:,Krhs) 
     115         END_3D 
    106116      ENDIF 
    107117      !                           !==  input T-S data at kt  ==! 
    … …  
    140150      END SELECT 
    141151      ! 
     152      ! TEMP: These changes not necessary after trd_tra is tiled 
    142153      IF( l_trdtra )   THEN       ! trend diagnostic 
    143          ztrdts(:,:,:,:) = pts(:,:,:,:,Krhs) - ztrdts(:,:,:,:) 
    144          CALL trd_tra( kt, Kmm, Krhs, 'TRA', jp_tem, jptra_dmp, ztrdts(:,:,:,jp_tem) ) 
    145          CALL trd_tra( kt, Kmm, Krhs, 'TRA', jp_sal, jptra_dmp, ztrdts(:,:,:,jp_sal) ) 
    146          DEALLOCATE( ztrdts )  
     154         DO_3D_00_00( 1, jpkm1 ) 
     155            ztrdts(ji,jj,jk,:) = pts(ji,jj,jk,:,Krhs) - ztrdts(ji,jj,jk,:) 
     156         END_3D 
     157 
     158         IF( ntile == 0 .OR. ntile == nijtile )  THEN                ! Do only for the full domain 
     159            IF( ln_tile ) CALL dom_tile( ntsi, ntsj, ntei, ntej, ktile = 0 )         ! Use full domain 
     160 
     161            ! TODO: TO BE TILED- trd_tra 
     162            CALL trd_tra( kt, Kmm, Krhs, 'TRA', jp_tem, jptra_dmp, ztrdts(:,:,:,jp_tem) ) 
     163            CALL trd_tra( kt, Kmm, Krhs, 'TRA', jp_sal, jptra_dmp, ztrdts(:,:,:,jp_sal) ) 
     164            DEALLOCATE( ztrdts ) 
     165 
     166            IF( ln_tile ) CALL dom_tile( ntsi, ntsj, ntei, ntej, ktile = nijtile )   ! Revert to tile domain 
     167         ENDIF 
    147168      ENDIF 
    148169      !                           ! Control print 
    149       IF(sn_cfctl%l_prtctl)   CALL prt_ctl( tab3d_1=pts(:,:,:,jp_tem,Krhs), clinfo1=' dmp  - Ta: ', mask1=tmask,   & 
    150          &                                  tab3d_2=pts(:,:,:,jp_sal,Krhs), clinfo2=       ' Sa: ', mask2=tmask, clinfo3='tra' ) 
     170      IF(sn_cfctl%l_prtctl)   CALL prt_ctl( tab3d_1=pts(:,:,:,jp_tem,Krhs), clinfo1=' dmp  - Ta: ', mask1=tmask, psum1=zsum1, & 
     171         &                                  tab3d_2=pts(:,:,:,jp_sal,Krhs), clinfo2=       ' Sa: ', mask2=tmask, psum2=zsum2, & 
     172         &                                  clinfo3='tra' ) 
    151173      ! 
    152174      IF( ln_timing )   CALL timing_stop('tra_dmp') 

  • NEMO/branches/UKMO/dev_r12866_HPC-02_Daley_Tiling_trial_extra_halo/src/OCE/TRA/traisf.F90

    r12377 r13409  
    1111   !!---------------------------------------------------------------------- 
    1212   USE isf_oce                                     ! Ice shelf variables 
     13   USE par_oce , ONLY : nijtile, ntile, ntsi, ntei, ntsj, ntej 
    1314   USE dom_oce , ONLY : e3t, r1_e1e2t            ! ocean space domain variables 
    1415   USE isfutils, ONLY : debug                      ! debug option 
    … …  
    3031CONTAINS 
    3132 
     33   ! TODO: NOT TESTED- requires isf 
    3234   SUBROUTINE tra_isf ( kt, Kmm, pts, Krhs ) 
    3335      !!---------------------------------------------------------------------- 
    … …  
    4547      IF( ln_timing )   CALL timing_start('tra_isf') 
    4648      ! 
    47       IF( kt == nit000 ) THEN 
    48          IF(lwp) WRITE(numout,*) 
    49          IF(lwp) WRITE(numout,*) 'tra_isf : Ice shelf heat fluxes' 
    50          IF(lwp) WRITE(numout,*) '~~~~~~~ ' 
     49      IF( ntile == 0 .OR. ntile == 1 )  THEN                       ! Do only on the first tile 
     50         IF( kt == nit000 ) THEN 
     51            IF(lwp) WRITE(numout,*) 
     52            IF(lwp) WRITE(numout,*) 'tra_isf : Ice shelf heat fluxes' 
     53            IF(lwp) WRITE(numout,*) '~~~~~~~ ' 
     54         ENDIF 
    5155      ENDIF 
    5256      ! 
    … …  
    7579      ! 
    7680      IF ( ln_isfdebug ) THEN 
    77          CALL debug('tra_isf: pts(:,:,:,:,Krhs) T', pts(:,:,:,1,Krhs)) 
    78          CALL debug('tra_isf: pts(:,:,:,:,Krhs) S', pts(:,:,:,2,Krhs)) 
     81         IF( ntile == 0 .OR. ntile == nijtile ) THEN                       ! Do only for the full domain 
     82            CALL debug('tra_isf: pts(:,:,:,:,Krhs) T', pts(:,:,:,1,Krhs)) 
     83            CALL debug('tra_isf: pts(:,:,:,:,Krhs) S', pts(:,:,:,2,Krhs)) 
     84         ENDIF 
    7985      END IF 
    8086      ! 
    … …  
    8389   END SUBROUTINE tra_isf 
    8490   ! 
     91   ! TODO: NOT TESTED- requires isf 
    8592   SUBROUTINE tra_isf_mlt(ktop, kbot, phtbl, pfrac, ptsc, ptsc_b, pts) 
    8693      !!---------------------------------------------------------------------- 
    … …  
    100107      INTEGER                      :: ji,jj,jk  ! loop index    
    101108      INTEGER                      :: ikt, ikb  ! top and bottom level of the tbl 
    102       REAL(wp), DIMENSION(jpi,jpj) :: ztc       ! total ice shelf tracer trend 
     109      REAL(wp), DIMENSION(A2D)    :: ztc       ! total ice shelf tracer trend 
    103110      !!---------------------------------------------------------------------- 
    104111      ! 
    105112      ! compute 2d total trend due to isf 
    106       ztc(:,:) = 0.5_wp * ( ptsc(:,:,jp_tem) + ptsc_b(:,:,jp_tem) ) / phtbl(:,:) 
     113      DO_2D_11_11 
     114         ztc(ji,jj) = 0.5_wp * ( ptsc(ji,jj,jp_tem) + ptsc_b(ji,jj,jp_tem) ) / phtbl(ji,jj) 
     115      END_2D 
    107116      ! 
    108117      ! update pts(:,:,:,:,Krhs) 
    … …  
    124133   END SUBROUTINE tra_isf_mlt 
    125134   ! 
     135   ! TODO: NOT TESTED- requires isf 
    126136   SUBROUTINE tra_isf_cpl( Kmm, ptsc, ptsa ) 
    127137      !!---------------------------------------------------------------------- 
    … …  
    136146      REAL(wp), DIMENSION(jpi,jpj,jpk,jpts), INTENT(in   ) :: ptsc 
    137147      !!---------------------------------------------------------------------- 
    138       INTEGER :: jk 
     148      INTEGER :: ji, jj, jk 
    139149      !!---------------------------------------------------------------------- 
    140150      ! 
    141       DO jk = 1,jpk 
    142          ptsa(:,:,jk,jp_tem) = ptsa(:,:,jk,jp_tem) + ptsc(:,:,jk,jp_tem) * r1_e1e2t(:,:) / e3t(:,:,jk,Kmm) 
    143          ptsa(:,:,jk,jp_sal) = ptsa(:,:,jk,jp_sal) + ptsc(:,:,jk,jp_sal) * r1_e1e2t(:,:) / e3t(:,:,jk,Kmm) 
    144       END DO 
     151      DO_3D_00_00( 1, jpk ) 
     152         ptsa(ji,jj,jk,jp_tem) = ptsa(ji,jj,jk,jp_tem) + ptsc(ji,jj,jk,jp_tem) * r1_e1e2t(ji,jj) / e3t(ji,jj,jk,Kmm) 
     153         ptsa(ji,jj,jk,jp_sal) = ptsa(ji,jj,jk,jp_sal) + ptsc(ji,jj,jk,jp_sal) * r1_e1e2t(ji,jj) / e3t(ji,jj,jk,Kmm) 
     154      END_3D 
    145155      ! 
    146156   END SUBROUTINE tra_isf_cpl 

  • NEMO/branches/UKMO/dev_r12866_HPC-02_Daley_Tiling_trial_extra_halo/src/OCE/TRA/traldf.F90

    r12942 r13409  
    1717   USE oce            ! ocean dynamics and tracers 
    1818   USE dom_oce        ! ocean space and time domain 
     19   ! TEMP: This change not necessary after trd_tra is tiled 
     20   USE domain, ONLY : dom_tile 
    1921   USE phycst         ! physical constants 
    2022   USE ldftra         ! lateral diffusion: eddy diffusivity & EIV coeff. 
    … …  
    3739   PUBLIC   tra_ldf        ! called by step.F90  
    3840   PUBLIC   tra_ldf_init   ! called by nemogcm.F90  
    39     
     41 
     42   !! * Substitutions 
     43#  include "do_loop_substitute.h90" 
    4044   !!---------------------------------------------------------------------- 
    4145   !! NEMO/OCE 4.0 , NEMO Consortium (2018) 
    … …  
    5559      REAL(wp), DIMENSION(jpi,jpj,jpk,jpts,jpt), INTENT(inout) :: pts             ! active tracers and RHS of tracer equation 
    5660      !! 
    57       REAL(wp), ALLOCATABLE, DIMENSION(:,:,:) ::   ztrdt, ztrds 
     61      ! TEMP: This change not necessary after trd_tra is tiled 
     62      INTEGER ::   itile 
     63      INTEGER ::   ji, jj, jk    ! dummy loop indices 
     64      REAL(wp), SAVE :: zsum1, zsum2 
     65      ! TEMP: This change not necessary after trd_tra is tiled 
     66      REAL(wp), ALLOCATABLE, SAVE, DIMENSION(:,:,:) ::   ztrdt, ztrds 
     67      ! TEMP: This change not necessary after extra haloes development 
     68      LOGICAL :: lskip 
    5869      !!---------------------------------------------------------------------- 
    5970      ! 
    6071      IF( ln_timing )   CALL timing_start('tra_ldf') 
    6172      ! 
    62       IF( ntile == 0 .OR. ntile == nijtile )  THEN                ! Do only after all tiles finish 
    63          IF( l_trdtra )   THEN                    !* Save ta and sa trends 
    64             ! TODO: TO BE TILED 
    65             ALLOCATE( ztrdt(jpi,jpj,jpk) , ztrds(jpi,jpj,jpk) ) 
    66             ztrdt(:,:,:) = pts(:,:,:,jp_tem,Krhs) 
    67             ztrds(:,:,:) = pts(:,:,:,jp_sal,Krhs) 
     73      lskip = .FALSE. 
     74 
     75      IF( l_trdtra ) THEN 
     76         IF( ntile == 0 .OR. ntile == 1 )  THEN                       ! Do only on the first tile 
     77            ! TEMP: This can be A2D after trd_tra is tiled 
     78            ALLOCATE( ztrdt(jpi,jpj,jpk), ztrds(jpi,jpj,jpk) ) 
    6879         ENDIF 
    6980      ENDIF 
    70       ! 
    71       SELECT CASE ( nldf_tra )                 !* compute lateral mixing trend and add it to the general trend 
    72       CASE ( np_lap   )                                  ! laplacian: iso-level operator 
    73          CALL tra_ldf_lap  ( kt, Kmm, nit000,'TRA', ahtu, ahtv, gtsu, gtsv, gtui, gtvi, pts(:,:,:,:,Kbb), pts(:,:,:,:,Krhs),                   jpts,  1 ) 
    74       CASE ( np_lap_i )                                  ! laplacian: standard iso-neutral operator (Madec) 
    75          CALL tra_ldf_iso  ( kt, Kmm, nit000,'TRA', ahtu, ahtv, gtsu, gtsv, gtui, gtvi, pts(:,:,:,:,Kbb), pts(:,:,:,:,Kbb), pts(:,:,:,:,Krhs), jpts,  1 ) 
    76       CASE ( np_lap_it )                                 ! laplacian: triad iso-neutral operator (griffies) 
    77          CALL tra_ldf_triad( kt, Kmm, nit000,'TRA', ahtu, ahtv, gtsu, gtsv, gtui, gtvi, pts(:,:,:,:,Kbb), pts(:,:,:,:,Kbb), pts(:,:,:,:,Krhs), jpts,  1 ) 
    78       CASE ( np_blp , np_blp_i , np_blp_it )             ! bilaplacian: iso-level & iso-neutral operators 
    79          CALL tra_ldf_blp  ( kt, Kmm, nit000,'TRA', ahtu, ahtv, gtsu, gtsv, gtui, gtvi, pts(:,:,:,:,Kbb), pts(:,:,:,:,Krhs),             jpts, nldf_tra ) 
    80       END SELECT 
    81       ! 
    82       IF( ntile == 0 .OR. ntile == nijtile )  THEN                ! Do only after all tiles finish 
    83          IF( l_trdtra )   THEN                    !* save the horizontal diffusive trends for further diagnostics 
    84             ! TODO: TO BE TILED 
    85             ztrdt(:,:,:) = pts(:,:,:,jp_tem,Krhs) - ztrdt(:,:,:) 
    86             ztrds(:,:,:) = pts(:,:,:,jp_sal,Krhs) - ztrds(:,:,:) 
    87             ! TODO: TO BE TILED 
    88             CALL trd_tra( kt, Kmm, Krhs, 'TRA', jp_tem, jptra_ldf, ztrdt ) 
    89             CALL trd_tra( kt, Kmm, Krhs, 'TRA', jp_sal, jptra_ldf, ztrds ) 
    90             DEALLOCATE( ztrdt, ztrds ) 
     81 
     82      ! TEMP: These changes not necessary after extra haloes development (lbc_lnk removed from tra_ldf_blp, zps_hde*) 
     83      IF( nldf_tra == np_blp .OR. nldf_tra == np_blp_i .OR. nldf_tra == np_blp_it )  THEN 
     84         IF( ln_tile ) THEN 
     85            IF( ntile == 1 ) THEN 
     86               CALL dom_tile( ntsi, ntsj, ntei, ntej, ktile = 0 ) 
     87            ELSE 
     88               lskip = .TRUE. 
     89            ENDIF 
    9190         ENDIF 
    9291      ENDIF 
     92      IF( .NOT. lskip ) THEN 
     93 
     94         ! TEMP: This change not necessary after trd_tra is tiled 
     95         itile = ntile 
     96 
     97         IF( l_trdtra )   THEN                    !* Save ta and sa trends 
     98            DO_3D_00_00( 1, jpkm1 ) 
     99               ztrdt(ji,jj,jk) = pts(ji,jj,jk,jp_tem,Krhs) 
     100               ztrds(ji,jj,jk) = pts(ji,jj,jk,jp_sal,Krhs) 
     101            END_3D 
     102         ENDIF 
     103         ! 
     104         SELECT CASE ( nldf_tra )                 !* compute lateral mixing trend and add it to the general trend 
     105         CASE ( np_lap   )                                  ! laplacian: iso-level operator 
     106            CALL tra_ldf_lap  ( kt, Kmm, nit000,'TRA', ahtu, ahtv, gtsu, gtsv, gtui, gtvi, pts(:,:,:,:,Kbb), pts(:,:,:,:,Krhs),                   jpts,  1 ) 
     107         CASE ( np_lap_i )                                  ! laplacian: standard iso-neutral operator (Madec) 
     108            CALL tra_ldf_iso  ( kt, Kmm, nit000,'TRA', ahtu, ahtv, gtsu, gtsv, gtui, gtvi, pts(:,:,:,:,Kbb), pts(:,:,:,:,Kbb), pts(:,:,:,:,Krhs), jpts,  1 ) 
     109         CASE ( np_lap_it )                                 ! laplacian: triad iso-neutral operator (griffies) 
     110            CALL tra_ldf_triad( kt, Kmm, nit000,'TRA', ahtu, ahtv, gtsu, gtsv, gtui, gtvi, pts(:,:,:,:,Kbb), pts(:,:,:,:,Kbb), pts(:,:,:,:,Krhs), jpts,  1 ) 
     111         CASE ( np_blp , np_blp_i , np_blp_it )             ! bilaplacian: iso-level & iso-neutral operators 
     112            CALL tra_ldf_blp  ( kt, Kmm, nit000,'TRA', ahtu, ahtv, gtsu, gtsv, gtui, gtvi, pts(:,:,:,:,Kbb), pts(:,:,:,:,Krhs),             jpts, nldf_tra ) 
     113         END SELECT 
     114         ! 
     115         ! TEMP: These changes not necessary after trd_tra is tiled 
     116         IF( l_trdtra )   THEN                    !* save the horizontal diffusive trends for further diagnostics 
     117            DO_3D_00_00( 1, jpkm1 ) 
     118               ztrdt(ji,jj,jk) = pts(ji,jj,jk,jp_tem,Krhs) - ztrdt(ji,jj,jk) 
     119               ztrds(ji,jj,jk) = pts(ji,jj,jk,jp_sal,Krhs) - ztrds(ji,jj,jk) 
     120            END_3D 
     121 
     122            IF( ntile == 0 .OR. ntile == nijtile )  THEN                ! Do only for the full domain 
     123               IF( ln_tile ) CALL dom_tile( ntsi, ntsj, ntei, ntej, ktile = 0 )         ! Use full domain 
     124 
     125               ! TODO: TO BE TILED- trd_tra 
     126               CALL trd_tra( kt, Kmm, Krhs, 'TRA', jp_tem, jptra_ldf, ztrdt ) 
     127               CALL trd_tra( kt, Kmm, Krhs, 'TRA', jp_sal, jptra_ldf, ztrds ) 
     128               DEALLOCATE( ztrdt, ztrds ) 
     129 
     130               IF( ln_tile ) CALL dom_tile( ntsi, ntsj, ntei, ntej, ktile = itile )     ! Revert to tile domain 
     131            ENDIF 
     132         ENDIF 
     133 
     134         ! TEMP: This change not necessary after extra haloes development (lbc_lnk removed from tra_ldf_blp, zps_hde*) 
     135         IF( ln_tile .AND. ntile == 0 ) CALL dom_tile( ntsi, ntsj, ntei, ntej, ktile = 1 ) 
     136 
     137      ENDIF 
    93138      !                                        !* print mean trends (used for debugging) 
    94       IF(sn_cfctl%l_prtctl)   CALL prt_ctl( tab3d_1=pts(:,:,:,jp_tem,Krhs), clinfo1=' ldf  - Ta: ', mask1=tmask,               & 
    95          &                                  tab3d_2=pts(:,:,:,jp_sal,Krhs), clinfo2=       ' Sa: ', mask2=tmask, clinfo3='tra' ) 
     139      IF(sn_cfctl%l_prtctl)   CALL prt_ctl( tab3d_1=pts(:,:,:,jp_tem,Krhs), clinfo1=' ldf  - Ta: ', mask1=tmask, psum1=zsum1, & 
     140         &                                  tab3d_2=pts(:,:,:,jp_sal,Krhs), clinfo2=       ' Sa: ', mask2=tmask, psum2=zsum2, & 
     141         &                                  clinfo3='tra' ) 
    96142      ! 
    97143      IF( ln_timing )   CALL timing_stop('tra_ldf') 

  • NEMO/branches/UKMO/dev_r12866_HPC-02_Daley_Tiling_trial_extra_halo/src/OCE/TRA/traldf_iso.F90

    r12942 r13409  
    1919   USE oce            ! ocean dynamics and active tracers 
    2020   USE dom_oce        ! ocean space and time domain 
     21   USE domutl, ONLY : is_tile 
    2122   USE trc_oce        ! share passive tracers/Ocean variables 
    2223   USE zdf_oce        ! ocean vertical physics 
    … …  
    3637   PUBLIC   tra_ldf_iso   ! routine called by step.F90 
    3738 
     39   LOGICAL  ::   l_ptr   ! flag to compute poleward transport 
     40   LOGICAL  ::   l_hst   ! flag to compute heat transport 
     41 
    3842   !! * Substitutions 
    3943#  include "do_loop_substitute.h90" 
    … …  
    4549CONTAINS 
    4650 
    47   SUBROUTINE tra_ldf_iso( kt, Kmm, kit000, cdtype, pahu, pahv,                    & 
    48       &                                            pgu , pgv    ,   pgui, pgvi,   & 
    49       &                                       pt , pt2 , pt_rhs , kjpt  , kpass ) 
     51   SUBROUTINE tra_ldf_iso( kt, Kmm, kit000, cdtype, pahu, pahv,             & 
     52      &                                             pgu , pgv , pgui, pgvi, & 
     53      &                                             pt, pt2, pt_rhs, kjpt, kpass ) 
     54      !! 
     55      INTEGER                     , INTENT(in   ) ::   kt         ! ocean time-step index 
     56      INTEGER                     , INTENT(in   ) ::   kit000     ! first time step index 
     57      CHARACTER(len=3)            , INTENT(in   ) ::   cdtype     ! =TRA or TRC (tracer indicator) 
     58      INTEGER                     , INTENT(in   ) ::   kjpt       ! number of tracers 
     59      INTEGER                     , INTENT(in   ) ::   kpass      ! =1/2 first or second passage 
     60      INTEGER                     , INTENT(in   ) ::   Kmm        ! ocean time level index 
     61      REAL(wp), DIMENSION(:,:,:)  , INTENT(in   ) ::   pahu, pahv ! eddy diffusivity at u- and v-points  [m2/s] 
     62      REAL(wp), DIMENSION(:,:,:)  , INTENT(in   ) ::   pgu, pgv   ! tracer gradient at pstep levels 
     63      REAL(wp), DIMENSION(:,:,:)  , INTENT(in   ) ::   pgui, pgvi ! tracer gradient at top   levels 
     64      REAL(wp), DIMENSION(:,:,:,:), INTENT(in   ) ::   pt         ! tracer (kpass=1) or laplacian of tracer (kpass=2) 
     65      REAL(wp), DIMENSION(:,:,:,:), INTENT(in   ) ::   pt2        ! tracer (only used in kpass=2) 
     66      REAL(wp), DIMENSION(:,:,:,:), INTENT(inout) ::   pt_rhs     ! tracer trend 
     67      !! 
     68      CALL tra_ldf_iso_t( kt, Kmm, kit000, cdtype, pahu, pahv, is_tile(pahu),                             & 
     69         &                                         pgu , pgv , is_tile(pgu) , pgui, pgvi, is_tile(pgui),  & 
     70         &                                         pt, is_tile(pt), pt2, is_tile(pt2), pt_rhs, is_tile(pt_rhs), kjpt, kpass ) 
     71   END SUBROUTINE tra_ldf_iso 
     72 
     73 
     74  SUBROUTINE tra_ldf_iso_t( kt, Kmm, kit000, cdtype, pahu, pahv, ktah,                    & 
     75      &                                              pgu , pgv , ktg , pgui, pgvi, ktgi,  & 
     76      &                                              pt, ktt, pt2, ktt2, pt_rhs, ktt_rhs, kjpt, kpass ) 
    5077      !!---------------------------------------------------------------------- 
    5178      !!                  ***  ROUTINE tra_ldf_iso  *** 
    … …  
    88115      !! ** Action :   Update pt_rhs arrays with the before rotated diffusion 
    89116      !!---------------------------------------------------------------------- 
    90       INTEGER                              , INTENT(in   ) ::   kt         ! ocean time-step index 
    91       INTEGER                              , INTENT(in   ) ::   kit000     ! first time step index 
    92       CHARACTER(len=3)                     , INTENT(in   ) ::   cdtype     ! =TRA or TRC (tracer indicator) 
    93       INTEGER                              , INTENT(in   ) ::   kjpt       ! number of tracers 
    94       INTEGER                              , INTENT(in   ) ::   kpass      ! =1/2 first or second passage 
    95       INTEGER                              , INTENT(in   ) ::   Kmm        ! ocean time level index 
    96       REAL(wp), DIMENSION(jpi,jpj,jpk)     , INTENT(in   ) ::   pahu, pahv ! eddy diffusivity at u- and v-points  [m2/s] 
    97       REAL(wp), DIMENSION(jpi,jpj    ,kjpt), INTENT(in   ) ::   pgu, pgv   ! tracer gradient at pstep levels 
    98       REAL(wp), DIMENSION(jpi,jpj,    kjpt), INTENT(in   ) ::   pgui, pgvi ! tracer gradient at top   levels 
    99       REAL(wp), DIMENSION(jpi,jpj,jpk,kjpt), INTENT(in   ) ::   pt         ! tracer (kpass=1) or laplacian of tracer (kpass=2) 
    100       REAL(wp), DIMENSION(jpi,jpj,jpk,kjpt), INTENT(in   ) ::   pt2        ! tracer (only used in kpass=2) 
    101       REAL(wp), DIMENSION(jpi,jpj,jpk,kjpt), INTENT(inout) ::   pt_rhs     ! tracer trend 
    102       ! 
    103       LOGICAL  ::  l_ptr                                 ! flag to compute poleward transport 
    104       LOGICAL  ::  l_hst                                 ! flag to compute heat transport 
     117      INTEGER                                   , INTENT(in   ) ::   kt         ! ocean time-step index 
     118      INTEGER                                   , INTENT(in   ) ::   kit000     ! first time step index 
     119      CHARACTER(len=3)                          , INTENT(in   ) ::   cdtype     ! =TRA or TRC (tracer indicator) 
     120      INTEGER                                   , INTENT(in   ) ::   kjpt       ! number of tracers 
     121      INTEGER                                   , INTENT(in   ) ::   kpass      ! =1/2 first or second passage 
     122      INTEGER                                   , INTENT(in   ) ::   Kmm        ! ocean time level index 
     123      INTEGER                                   , INTENT(in   ) ::   ktah, ktg, ktgi, ktt, ktt2, ktt_rhs 
     124      REAL(wp), DIMENSION(T2D(ktah)   ,jpk)     , INTENT(in   ) ::   pahu, pahv ! eddy diffusivity at u- and v-points  [m2/s] 
     125      REAL(wp), DIMENSION(T2D(ktg)        ,kjpt), INTENT(in   ) ::   pgu, pgv   ! tracer gradient at pstep levels 
     126      REAL(wp), DIMENSION(T2D(ktgi)       ,kjpt), INTENT(in   ) ::   pgui, pgvi ! tracer gradient at top   levels 
     127      REAL(wp), DIMENSION(T2D(ktt)    ,jpk,kjpt), INTENT(in   ) ::   pt         ! tracer (kpass=1) or laplacian of tracer (kpass=2) 
     128      REAL(wp), DIMENSION(T2D(ktt2)   ,jpk,kjpt), INTENT(in   ) ::   pt2        ! tracer (only used in kpass=2) 
     129      REAL(wp), DIMENSION(T2D(ktt_rhs),jpk,kjpt), INTENT(inout) ::   pt_rhs     ! tracer trend 
     130      ! 
    105131      INTEGER  ::  ji, jj, jk, jn   ! dummy loop indices 
    106132      INTEGER  ::  ikt 
    … …  
    115141      IF( kpass == 1 .AND. kt == kit000 )  THEN 
    116142         IF( ntile == 0 .OR. ntile == 1 )  THEN                       ! Do only on the first tile 
    117             ! TODO: TO BE TILED 
    118143            IF(lwp) WRITE(numout,*) 
    119144            IF(lwp) WRITE(numout,*) 'tra_ldf_iso : rotated laplacian diffusion operator on ', cdtype 
    … …  
    121146         ENDIF 
    122147         ! 
    123          DO_3D_11_11( 1, jpk ) 
     148         DO_3D_00_00( 1, jpk ) 
    124149            akz     (ji,jj,jk) = 0._wp 
    125150            ah_wslp2(ji,jj,jk) = 0._wp 
    126151         END_3D 
    127152      ENDIF 
    128       !    
    129       l_hst = .FALSE. 
    130       l_ptr = .FALSE. 
    131       IF( cdtype == 'TRA' .AND. ( iom_use( 'sophtldf' ) .OR. iom_use( 'sopstldf' ) ) )     l_ptr = .TRUE.  
    132       IF( cdtype == 'TRA' .AND. ( iom_use("uadv_heattr") .OR. iom_use("vadv_heattr") .OR. & 
    133          &                        iom_use("uadv_salttr") .OR. iom_use("vadv_salttr")  ) )   l_hst = .TRUE. 
     153      ! 
     154      IF( ntile == 0 .OR. ntile == 1 )  THEN                           ! Do only on the first tile 
     155         l_hst = .FALSE. 
     156         l_ptr = .FALSE. 
     157         IF( cdtype == 'TRA' .AND. ( iom_use( 'sophtldf' ) .OR. iom_use( 'sopstldf' ) ) )     l_ptr = .TRUE. 
     158         IF( cdtype == 'TRA' .AND. ( iom_use("uadv_heattr") .OR. iom_use("vadv_heattr") .OR. & 
     159            &                        iom_use("uadv_salttr") .OR. iom_use("vadv_salttr")  ) )   l_hst = .TRUE. 
     160      ENDIF 
    134161      ! 
    135162      ! 
    … …  
    170197            ! 
    171198            IF( ln_traldf_blp ) THEN                ! bilaplacian operator 
    172                DO_3D_10_10( 2, jpkm1 ) 
     199               DO_3D_00_00( 2, jpkm1 ) 
    173200                  akz(ji,jj,jk) = 16._wp * ah_wslp2(ji,jj,jk)   & 
    174201                     &          * (  akz(ji,jj,jk) + ah_wslp2(ji,jj,jk) / ( e3w(ji,jj,jk,Kmm) * e3w(ji,jj,jk,Kmm) )  ) 
    175202               END_3D 
    176203            ELSEIF( ln_traldf_lap ) THEN              ! laplacian operator 
    177                DO_3D_10_10( 2, jpkm1 ) 
     204               DO_3D_00_00( 2, jpkm1 ) 
    178205                  ze3w_2 = e3w(ji,jj,jk,Kmm) * e3w(ji,jj,jk,Kmm) 
    179206                  zcoef0 = rDt * (  akz(ji,jj,jk) + ah_wslp2(ji,jj,jk) / ze3w_2  ) 
    … …  
    183210           ! 
    184211         ELSE                                    ! 33 flux set to zero with akz=ah_wslp2 ==>> computed in full implicit 
    185             DO_3D_11_11( 1, jpk ) 
     212            DO_3D_00_00( 1, jpk ) 
    186213               akz(ji,jj,jk) = ah_wslp2(ji,jj,jk) 
    187214            END_3D 
    … …  
    206233            zdjt(ji,jj,jk) = ( pt(ji  ,jj+1,jk,jn) - pt(ji,jj,jk,jn) ) * vmask(ji,jj,jk) 
    207234         END_3D 
     235         ! TODO: NOT TESTED- requires zps 
    208236         IF( ln_zps ) THEN      ! botton and surface ocean correction of the horizontal gradient 
    209237            DO_2D_10_10 
    … …  
    211239               zdjt(ji,jj,mbkv(ji,jj)) = pgv(ji,jj,jn) 
    212240            END_2D 
     241            ! TODO: NOT TESTED- requires isf 
    213242            IF( ln_isfcav ) THEN      ! first wet level beneath a cavity 
    214243               DO_2D_10_10 
    … …  
    326355            !                             ! "Poleward" diffusive heat or salt transports (T-S case only) 
    327356               ! note sign is reversed to give down-gradient diffusive transports ) 
    328                ! TODO: TO BE TILED 
    329357            IF( l_ptr )  CALL dia_ptr_hst( jn, 'ldf', -zftv(:,:,:)  ) 
    330358            !                          ! Diffusive heat transports 
    331                ! TODO: TO BE TILED 
    332359            IF( l_hst )  CALL dia_ar5_hst( jn, 'ldf', -zftu(:,:,:), -zftv(:,:,:) ) 
    333360            ! 
    … …  
    337364      END DO                                                      ! end tracer loop 
    338365      ! 
    339    END SUBROUTINE tra_ldf_iso 
     366   END SUBROUTINE tra_ldf_iso_t 
    340367 
    341368   !!============================================================================== 

  • NEMO/branches/UKMO/dev_r12866_HPC-02_Daley_Tiling_trial_extra_halo/src/OCE/TRA/traldf_lap_blp.F90

    r12377 r13409  
    1313   USE oce            ! ocean dynamics and active tracers 
    1414   USE dom_oce        ! ocean space and time domain 
     15   USE domutl, ONLY : is_tile 
    1516   USE ldftra         ! lateral physics: eddy diffusivity 
    1617   USE traldf_iso     ! iso-neutral lateral diffusion (standard operator)     (tra_ldf_iso   routine) 
    … …  
    4546CONTAINS 
    4647 
    47    SUBROUTINE tra_ldf_lap( kt, Kmm, kit000, cdtype, pahu, pahv  ,               & 
    48       &                                             pgu , pgv   , pgui, pgvi,   & 
    49       &                                             pt  , pt_rhs, kjpt, kpass )  
     48   SUBROUTINE tra_ldf_lap( kt, Kmm, kit000, cdtype, pahu, pahv,             & 
     49      &                                             pgu , pgv , pgui, pgvi, & 
     50      &                                             pt, pt_rhs, kjpt, kpass ) 
     51      !! 
     52      INTEGER                     , INTENT(in   ) ::   kt         ! ocean time-step index 
     53      INTEGER                     , INTENT(in   ) ::   kit000     ! first time step index 
     54      CHARACTER(len=3)            , INTENT(in   ) ::   cdtype     ! =TRA or TRC (tracer indicator) 
     55      INTEGER                     , INTENT(in   ) ::   kjpt       ! number of tracers 
     56      INTEGER                     , INTENT(in   ) ::   kpass      ! =1/2 first or second passage 
     57      INTEGER                     , INTENT(in   ) ::   Kmm        ! ocean time level index 
     58      REAL(wp), DIMENSION(:,:,:)  , INTENT(in   ) ::   pahu, pahv ! eddy diffusivity at u- and v-points  [m2/s] 
     59      REAL(wp), DIMENSION(:,:,:)  , INTENT(in   ) ::   pgu, pgv   ! tracer gradient at pstep levels 
     60      REAL(wp), DIMENSION(:,:,:)  , INTENT(in   ) ::   pgui, pgvi ! tracer gradient at top   levels 
     61      REAL(wp), DIMENSION(:,:,:,:), INTENT(in   ) ::   pt         ! before tracer fields 
     62      REAL(wp), DIMENSION(:,:,:,:), INTENT(inout) ::   pt_rhs     ! tracer trend 
     63      !! 
     64      CALL tra_ldf_lap_t( kt, Kmm, kit000, cdtype, pahu, pahv, is_tile(pahu),                            & 
     65      &                                            pgu , pgv , is_tile(pgu) , pgui, pgvi, is_tile(pgui), & 
     66      &                                            pt, is_tile(pt), pt_rhs, is_tile(pt_rhs), kjpt, kpass ) 
     67   END SUBROUTINE tra_ldf_lap 
     68 
     69 
     70   SUBROUTINE tra_ldf_lap_t( kt, Kmm, kit000, cdtype, pahu, pahv, ktah,                   & 
     71      &                                               pgu , pgv , ktg , pgui, pgvi, ktgi, & 
     72      &                                               pt, ktt, pt_rhs, ktt_rhs, kjpt, kpass ) 
    5073      !!---------------------------------------------------------------------- 
    5174      !!                  ***  ROUTINE tra_ldf_lap  *** 
    … …  
    7194      INTEGER                              , INTENT(in   ) ::   kpass      ! =1/2 first or second passage 
    7295      INTEGER                              , INTENT(in   ) ::   Kmm        ! ocean time level index 
    73       REAL(wp), DIMENSION(jpi,jpj,jpk)     , INTENT(in   ) ::   pahu, pahv ! eddy diffusivity at u- and v-points  [m2/s] 
    74       REAL(wp), DIMENSION(jpi,jpj    ,kjpt), INTENT(in   ) ::   pgu, pgv   ! tracer gradient at pstep levels 
    75       REAL(wp), DIMENSION(jpi,jpj,    kjpt), INTENT(in   ) ::   pgui, pgvi ! tracer gradient at top   levels 
    76       REAL(wp), DIMENSION(jpi,jpj,jpk,kjpt), INTENT(in   ) ::   pt         ! before tracer fields 
    77       REAL(wp), DIMENSION(jpi,jpj,jpk,kjpt), INTENT(inout) ::   pt_rhs     ! tracer trend  
     96      INTEGER                              , INTENT(in   ) ::   ktah, ktg, ktgi, ktt, ktt_rhs 
     97      REAL(wp), DIMENSION(T2D(ktah),   jpk)     , INTENT(in   ) ::   pahu, pahv ! eddy diffusivity at u- and v-points  [m2/s] 
     98      REAL(wp), DIMENSION(T2D(ktg),        kjpt), INTENT(in   ) ::   pgu, pgv   ! tracer gradient at pstep levels 
     99      REAL(wp), DIMENSION(T2D(ktgi),       kjpt), INTENT(in   ) ::   pgui, pgvi ! tracer gradient at top   levels 
     100      REAL(wp), DIMENSION(T2D(ktt),    jpk,kjpt), INTENT(in   ) ::   pt         ! before tracer fields 
     101      REAL(wp), DIMENSION(T2D(ktt_rhs),jpk,kjpt), INTENT(inout) ::   pt_rhs     ! tracer trend 
    78102      ! 
    79103      INTEGER  ::   ji, jj, jk, jn   ! dummy loop indices 
    80104      REAL(wp) ::   zsign            ! local scalars 
    81       REAL(wp), DIMENSION(jpi,jpj,jpk) ::   ztu, ztv, zaheeu, zaheev 
    82       !!---------------------------------------------------------------------- 
    83       ! 
    84       IF( kt == nit000 .AND. lwp )  THEN 
    85          WRITE(numout,*) 
    86          WRITE(numout,*) 'tra_ldf_lap : iso-level laplacian diffusion on ', cdtype, ', pass=', kpass 
    87          WRITE(numout,*) '~~~~~~~~~~~ ' 
    88       ENDIF 
    89       ! 
    90       l_hst = .FALSE. 
    91       l_ptr = .FALSE. 
    92       IF( cdtype == 'TRA' .AND. ( iom_use( 'sophtldf' ) .OR. iom_use( 'sopstldf' ) ) )     l_ptr = .TRUE. 
    93       IF( cdtype == 'TRA' .AND. ( iom_use("uadv_heattr") .OR. iom_use("vadv_heattr") .OR. & 
    94          &                        iom_use("uadv_salttr") .OR. iom_use("vadv_salttr")  ) )  l_hst = .TRUE. 
     105      REAL(wp), DIMENSION(A2D,jpk) ::   ztu, ztv, zaheeu, zaheev 
     106      !!---------------------------------------------------------------------- 
     107      ! 
     108      IF( ntile == 0 .OR. ntile == 1 )  THEN                       ! Do only on the first tile 
     109         IF( kt == nit000 .AND. lwp )  THEN 
     110            WRITE(numout,*) 
     111            WRITE(numout,*) 'tra_ldf_lap : iso-level laplacian diffusion on ', cdtype, ', pass=', kpass 
     112            WRITE(numout,*) '~~~~~~~~~~~ ' 
     113         ENDIF 
     114         ! 
     115         l_hst = .FALSE. 
     116         l_ptr = .FALSE. 
     117         IF( cdtype == 'TRA' .AND. ( iom_use( 'sophtldf' ) .OR. iom_use( 'sopstldf' ) ) )     l_ptr = .TRUE. 
     118         IF( cdtype == 'TRA' .AND. ( iom_use("uadv_heattr") .OR. iom_use("vadv_heattr") .OR. & 
     119            &                        iom_use("uadv_salttr") .OR. iom_use("vadv_salttr")  ) )  l_hst = .TRUE. 
     120      ENDIF 
    95121      ! 
    96122      !                                !==  Initialization of metric arrays used for all tracers  ==! 
    … …  
    111137            ztv(ji,jj,jk) = zaheev(ji,jj,jk) * ( pt(ji  ,jj+1,jk,jn) - pt(ji,jj,jk,jn) ) 
    112138         END_3D 
     139         ! TODO: NOT TESTED- requires zps 
    113140         IF( ln_zps ) THEN                ! set gradient at bottom/top ocean level 
    114141            DO_2D_10_10 
    … …  
    116143               ztv(ji,jj,mbkv(ji,jj)) = zaheev(ji,jj,mbkv(ji,jj)) * pgv(ji,jj,jn) 
    117144            END_2D 
     145            ! TODO: NOT TESTED- requires isf 
    118146            IF( ln_isfcav ) THEN                ! top in ocean cavities only 
    119147               DO_2D_10_10 
    … …  
    141169      !                             ! ================== 
    142170      ! 
    143    END SUBROUTINE tra_ldf_lap 
     171   END SUBROUTINE tra_ldf_lap_t 
    144172    
    145173 
    … …  
    172200      ! 
    173201      INTEGER ::   ji, jj, jk, jn   ! dummy loop indices 
    174       REAL(wp), DIMENSION(jpi,jpj,jpk,kjpt) :: zlap         ! laplacian at t-point 
    175       REAL(wp), DIMENSION(jpi,jpj,    kjpt) :: zglu, zglv   ! bottom GRADh of the laplacian (u- and v-points) 
    176       REAL(wp), DIMENSION(jpi,jpj,    kjpt) :: zgui, zgvi   ! top    GRADh of the laplacian (u- and v-points) 
     202      REAL(wp), DIMENSION(A2D,jpk,kjpt) :: zlap         ! laplacian at t-point 
     203      REAL(wp), DIMENSION(A2D,    kjpt) :: zglu, zglv   ! bottom GRADh of the laplacian (u- and v-points) 
     204      REAL(wp), DIMENSION(A2D,    kjpt) :: zgui, zgvi   ! top    GRADh of the laplacian (u- and v-points) 
    177205      !!--------------------------------------------------------------------- 
    178206      ! 
    179       IF( kt == kit000 .AND. lwp )  THEN 
    180          WRITE(numout,*) 
    181          SELECT CASE ( kldf ) 
    182          CASE ( np_blp    )   ;   WRITE(numout,*) 'tra_ldf_blp : iso-level   bilaplacian operator on ', cdtype 
    183          CASE ( np_blp_i  )   ;   WRITE(numout,*) 'tra_ldf_blp : iso-neutral bilaplacian operator on ', cdtype, ' (Standard)' 
    184          CASE ( np_blp_it )   ;   WRITE(numout,*) 'tra_ldf_blp : iso-neutral bilaplacian operator on ', cdtype, ' (triad)' 
    185          END SELECT 
    186          WRITE(numout,*) '~~~~~~~~~~~' 
     207      IF( ntile == 0 .OR. ntile == 1 )  THEN                       ! Do only on the first tile 
     208         IF( kt == kit000 .AND. lwp )  THEN 
     209            WRITE(numout,*) 
     210            SELECT CASE ( kldf ) 
     211            CASE ( np_blp    )   ;   WRITE(numout,*) 'tra_ldf_blp : iso-level   bilaplacian operator on ', cdtype 
     212            CASE ( np_blp_i  )   ;   WRITE(numout,*) 'tra_ldf_blp : iso-neutral bilaplacian operator on ', cdtype, ' (Standard)' 
     213            CASE ( np_blp_it )   ;   WRITE(numout,*) 'tra_ldf_blp : iso-neutral bilaplacian operator on ', cdtype, ' (triad)' 
     214            END SELECT 
     215            WRITE(numout,*) '~~~~~~~~~~~' 
     216         ENDIF 
    187217      ENDIF 
    188218 
    … …  
    201231      CALL lbc_lnk( 'traldf_lap_blp', zlap(:,:,:,:) , 'T', 1. )     ! Lateral boundary conditions (unchanged sign) 
    202232      !                                               ! Partial top/bottom cell: GRADh( zlap )   
     233      ! TODO: NOT TESTED- requires zps and isf 
    203234      IF( ln_isfcav .AND. ln_zps ) THEN   ;   CALL zps_hde_isf( kt, Kmm, kjpt, zlap, zglu, zglv, zgui, zgvi )  ! both top & bottom 
    204235      ELSEIF(             ln_zps ) THEN   ;   CALL zps_hde    ( kt, Kmm, kjpt, zlap, zglu, zglv )              ! only bottom  

  • NEMO/branches/UKMO/dev_r12866_HPC-02_Daley_Tiling_trial_extra_halo/src/OCE/TRA/traldf_triad.F90

    r12489 r13409  
    1313   USE oce            ! ocean dynamics and active tracers 
    1414   USE dom_oce        ! ocean space and time domain 
     15   ! TEMP: This change not necessary if lbc_lnk is removed from ldf_eiv_dia and XIOS has subdomain support 
     16   USE domain, ONLY : dom_tile 
     17   USE domutl, ONLY : is_tile 
    1518   USE phycst         ! physical constants 
    1619   USE trc_oce        ! share passive tracers/Ocean variables 
    … …  
    3336   PUBLIC   tra_ldf_triad   ! routine called by traldf.F90 
    3437 
    35    REAL(wp), DIMENSION(:,:,:), ALLOCATABLE, SAVE ::   zdkt3d   !: vertical tracer gradient at 2 levels 
    36  
    3738   LOGICAL  ::   l_ptr   ! flag to compute poleward transport 
    3839   LOGICAL  ::   l_hst   ! flag to compute heat transport 
    … …  
    4849CONTAINS 
    4950 
    50   SUBROUTINE tra_ldf_triad( kt, Kmm, kit000, cdtype, pahu, pahv,               & 
    51       &                                              pgu , pgv  , pgui, pgvi , & 
    52       &                                         pt , pt2, pt_rhs, kjpt, kpass ) 
     51   SUBROUTINE tra_ldf_triad( kt, Kmm, kit000, cdtype, pahu, pahv,             & 
     52      &                                               pgu , pgv , pgui, pgvi, & 
     53      &                                               pt, pt2, pt_rhs, kjpt, kpass ) 
     54      !! 
     55      INTEGER                     , INTENT(in   ) ::   kt         ! ocean time-step index 
     56      INTEGER                     , INTENT(in   ) ::   kit000     ! first time step index 
     57      CHARACTER(len=3)            , INTENT(in   ) ::   cdtype     ! =TRA or TRC (tracer indicator) 
     58      INTEGER                     , INTENT(in   ) ::   kjpt       ! number of tracers 
     59      INTEGER                     , INTENT(in   ) ::   kpass      ! =1/2 first or second passage 
     60      INTEGER                     , INTENT(in   ) ::   Kmm        ! ocean time level indices 
     61      REAL(wp), DIMENSION(:,:,:)  , INTENT(in   ) ::   pahu, pahv ! eddy diffusivity at u- and v-points  [m2/s] 
     62      REAL(wp), DIMENSION(:,:,:)  , INTENT(in   ) ::   pgu , pgv  ! tracer gradient at pstep levels 
     63      REAL(wp), DIMENSION(:,:,:)  , INTENT(in   ) ::   pgui, pgvi ! tracer gradient at top   levels 
     64      REAL(wp), DIMENSION(:,:,:,:), INTENT(in   ) ::   pt         ! tracer (kpass=1) or laplacian of tracer (kpass=2) 
     65      REAL(wp), DIMENSION(:,:,:,:), INTENT(in   ) ::   pt2        ! tracer (only used in kpass=2) 
     66      REAL(wp), DIMENSION(:,:,:,:), INTENT(inout) ::   pt_rhs     ! tracer trend 
     67      !! 
     68      CALL tra_ldf_triad_t( kt, Kmm, kit000, cdtype, pahu, pahv, is_tile(pahu),                            & 
     69      &                                              pgu , pgv , is_tile(pgu) , pgui, pgvi, is_tile(pgui), & 
     70      &                                              pt, is_tile(pt), pt2, is_tile(pt2), pt_rhs, is_tile(pt_rhs), kjpt, kpass ) 
     71   END SUBROUTINE tra_ldf_triad 
     72 
     73 
     74  SUBROUTINE tra_ldf_triad_t( kt, Kmm, kit000, cdtype, pahu, pahv, ktah,                   & 
     75      &                                                pgu , pgv , ktg , pgui, pgvi, ktgi, & 
     76      &                                                pt, ktt, pt2, ktt2, pt_rhs, ktt_rhs, kjpt, kpass ) 
    5377      !!---------------------------------------------------------------------- 
    5478      !!                  ***  ROUTINE tra_ldf_triad  *** 
    … …  
    76100      INTEGER                              , INTENT(in   ) ::   kpass      ! =1/2 first or second passage 
    77101      INTEGER                              , INTENT(in)    ::   Kmm        ! ocean time level indices 
    78       REAL(wp), DIMENSION(jpi,jpj,jpk)     , INTENT(in   ) ::   pahu, pahv ! eddy diffusivity at u- and v-points  [m2/s] 
    79       REAL(wp), DIMENSION(jpi,jpj    ,kjpt), INTENT(in   ) ::   pgu , pgv  ! tracer gradient at pstep levels 
    80       REAL(wp), DIMENSION(jpi,jpj,    kjpt), INTENT(in   ) ::   pgui, pgvi ! tracer gradient at top   levels 
    81       REAL(wp), DIMENSION(jpi,jpj,jpk,kjpt), INTENT(in   ) ::   pt         ! tracer (kpass=1) or laplacian of tracer (kpass=2) 
    82       REAL(wp), DIMENSION(jpi,jpj,jpk,kjpt), INTENT(in   ) ::   pt2        ! tracer (only used in kpass=2) 
    83       REAL(wp), DIMENSION(jpi,jpj,jpk,kjpt), INTENT(inout) ::   pt_rhs     ! tracer trend 
     102      INTEGER                              , INTENT(in   ) ::   ktah, ktg, ktgi, ktt, ktt2, ktt_rhs 
     103      REAL(wp), DIMENSION(T2D(ktah),   jpk)     , INTENT(in   ) ::   pahu, pahv ! eddy diffusivity at u- and v-points  [m2/s] 
     104      REAL(wp), DIMENSION(T2D(ktg),        kjpt), INTENT(in   ) ::   pgu , pgv  ! tracer gradient at pstep levels 
     105      REAL(wp), DIMENSION(T2D(ktgi),       kjpt), INTENT(in   ) ::   pgui, pgvi ! tracer gradient at top   levels 
     106      REAL(wp), DIMENSION(T2D(ktt),    jpk,kjpt), INTENT(in   ) ::   pt         ! tracer (kpass=1) or laplacian of tracer (kpass=2) 
     107      REAL(wp), DIMENSION(T2D(ktt2),   jpk,kjpt), INTENT(in   ) ::   pt2        ! tracer (only used in kpass=2) 
     108      REAL(wp), DIMENSION(T2D(ktt_rhs),jpk,kjpt), INTENT(inout) ::   pt_rhs     ! tracer trend 
    84109      ! 
    85110      INTEGER  ::  ji, jj, jk, jn   ! dummy loop indices 
    … …  
    93118      REAL(wp) ::   ze1ur, ze2vr, ze3wr, zdxt, zdyt, zdzt 
    94119      REAL(wp) ::   zah, zah_slp, zaei_slp 
    95       REAL(wp), DIMENSION(jpi,jpj    ) ::   z2d                                              ! 2D workspace 
    96       REAL(wp), DIMENSION(jpi,jpj,jpk) ::   zdit, zdjt, zftu, zftv, ztfw, zpsi_uw, zpsi_vw   ! 3D     - 
     120      REAL(wp), DIMENSION(A2D,0:1)     ::   zdkt3d                         ! vertical tracer gradient at 2 levels 
     121      REAL(wp), DIMENSION(A2D        ) ::   z2d                            ! 2D workspace 
     122      REAL(wp), DIMENSION(A2D    ,jpk) ::   zdit, zdjt, zftu, zftv, ztfw   ! 3D     - 
     123      ! TEMP: This can be A2D if lbc_lnk is removed from ldf_eiv_dia and XIOS has subdomain support 
     124      REAL(wp), DIMENSION(jpi,jpj,jpk) ::   zpsi_uw, zpsi_vw 
    97125      !!---------------------------------------------------------------------- 
    98126      ! 
    99       IF( .NOT.ALLOCATED(zdkt3d) )  THEN 
    100          ALLOCATE( zdkt3d(jpi,jpj,0:1) , STAT=ierr ) 
    101          CALL mpp_sum ( 'traldf_triad', ierr ) 
    102          IF( ierr > 0 )   CALL ctl_stop('STOP', 'tra_ldf_triad: unable to allocate arrays') 
    103       ENDIF 
    104      ! 
    105       IF( kpass == 1 .AND. kt == kit000 )  THEN 
    106          IF(lwp) WRITE(numout,*) 
    107          IF(lwp) WRITE(numout,*) 'tra_ldf_triad : rotated laplacian diffusion operator on ', cdtype 
    108          IF(lwp) WRITE(numout,*) '~~~~~~~~~~~~~' 
    109       ENDIF 
    110       !    
    111       l_hst = .FALSE. 
    112       l_ptr = .FALSE. 
    113       IF( cdtype == 'TRA' ) THEN 
    114          IF( iom_use( 'sophtldf' ) .OR. iom_use( 'sopstldf') )      l_ptr = .TRUE.  
    115          IF( iom_use("uadv_heattr") .OR. iom_use("vadv_heattr") .OR.                   & 
    116          &   iom_use("uadv_salttr") .OR. iom_use("vadv_salttr")  )   l_hst = .TRUE. 
     127      IF( ntile == 0 .OR. ntile == 1 )  THEN                       ! Do only on the first tile 
     128         IF( kpass == 1 .AND. kt == kit000 )  THEN 
     129            IF(lwp) WRITE(numout,*) 
     130            IF(lwp) WRITE(numout,*) 'tra_ldf_triad : rotated laplacian diffusion operator on ', cdtype 
     131            IF(lwp) WRITE(numout,*) '~~~~~~~~~~~~~' 
     132         ENDIF 
     133         ! 
     134         l_hst = .FALSE. 
     135         l_ptr = .FALSE. 
     136         IF( cdtype == 'TRA' ) THEN 
     137            IF( iom_use( 'sophtldf' ) .OR. iom_use( 'sopstldf') )      l_ptr = .TRUE. 
     138            IF( iom_use("uadv_heattr") .OR. iom_use("vadv_heattr") .OR.                   & 
     139            &   iom_use("uadv_salttr") .OR. iom_use("vadv_salttr")  )   l_hst = .TRUE. 
     140         ENDIF 
    117141      ENDIF 
    118142      ! 
    … …  
    127151      IF( kpass == 1 ) THEN         !==  first pass only  and whatever the tracer is  ==! 
    128152         ! 
    129          akz     (:,:,:) = 0._wp       
    130          ah_wslp2(:,:,:) = 0._wp 
    131          IF( ln_ldfeiv_dia ) THEN 
    132             zpsi_uw(:,:,:) = 0._wp 
    133             zpsi_vw(:,:,:) = 0._wp 
    134          ENDIF 
     153         DO_3D_00_00( 1, jpk ) 
     154            akz     (ji,jj,jk) = 0._wp 
     155            ah_wslp2(ji,jj,jk) = 0._wp 
     156         END_3D 
    135157         ! 
    136158         DO ip = 0, 1                            ! i-k triads 
    137159            DO kp = 0, 1 
    138                DO_3D_10_10( 1, jpkm1 ) 
    139                   ze3wr = 1._wp / e3w(ji+ip,jj,jk+kp,Kmm) 
    140                   zbu   = e1e2u(ji,jj) * e3u(ji,jj,jk,Kmm) 
    141                   zah   = 0.25_wp * pahu(ji,jj,jk) 
    142                   zslope_skew = triadi_g(ji+ip,jj,jk,1-ip,kp) 
     160               DO_3D_00_00( 1, jpkm1 ) 
     161                  ze3wr = 1._wp / e3w(ji,jj,jk+kp,Kmm) 
     162                  zbu   = e1e2u(ji-ip,jj) * e3u(ji-ip,jj,jk,Kmm) 
     163                  zah   = 0.25_wp * pahu(ji-ip,jj,jk) 
     164                  zslope_skew = triadi_g(ji,jj,jk,1-ip,kp) 
    143165                  ! Subtract s-coordinate slope at t-points to give slope rel to s-surfaces (do this by *adding* gradient of depth) 
    144                   zslope2 = zslope_skew + ( gdept(ji+1,jj,jk,Kmm) - gdept(ji,jj,jk,Kmm) ) * r1_e1u(ji,jj) * umask(ji,jj,jk+kp) 
     166                  zslope2 = zslope_skew + ( gdept(ji-ip+1,jj,jk,Kmm) - gdept(ji-ip,jj,jk,Kmm) ) * r1_e1u(ji-ip,jj) * umask(ji-ip,jj,jk+kp) 
    145167                  zslope2 = zslope2 *zslope2 
    146                   ah_wslp2(ji+ip,jj,jk+kp) = ah_wslp2(ji+ip,jj,jk+kp) + zah * zbu * ze3wr * r1_e1e2t(ji+ip,jj) * zslope2 
    147                   akz     (ji+ip,jj,jk+kp) = akz     (ji+ip,jj,jk+kp) + zah * r1_e1u(ji,jj)       & 
    148                      &                                                      * r1_e1u(ji,jj) * umask(ji,jj,jk+kp) 
     168                  ah_wslp2(ji,jj,jk+kp) = ah_wslp2(ji,jj,jk+kp) + zah * zbu * ze3wr * r1_e1e2t(ji,jj) * zslope2 
     169                  akz     (ji,jj,jk+kp) = akz     (ji,jj,jk+kp) + zah * r1_e1u(ji-ip,jj)       & 
     170                     &                                                      * r1_e1u(ji-ip,jj) * umask(ji-ip,jj,jk+kp) 
    149171                     ! 
    150                  IF( ln_ldfeiv_dia )   zpsi_uw(ji,jj,jk+kp) = zpsi_uw(ji,jj,jk+kp)   & 
    151                      &                                       + 0.25_wp * aeiu(ji,jj,jk) * e2u(ji,jj) * zslope_skew 
    152172               END_3D 
    153173            END DO 
    … …  
    156176         DO jp = 0, 1                            ! j-k triads  
    157177            DO kp = 0, 1 
    158                DO_3D_10_10( 1, jpkm1 ) 
    159                   ze3wr = 1.0_wp / e3w(ji,jj+jp,jk+kp,Kmm) 
    160                   zbv   = e1e2v(ji,jj) * e3v(ji,jj,jk,Kmm) 
    161                   zah   = 0.25_wp * pahv(ji,jj,jk) 
    162                   zslope_skew = triadj_g(ji,jj+jp,jk,1-jp,kp) 
     178               DO_3D_00_00( 1, jpkm1 ) 
     179                  ze3wr = 1.0_wp / e3w(ji,jj,jk+kp,Kmm) 
     180                  zbv   = e1e2v(ji,jj-jp) * e3v(ji,jj-jp,jk,Kmm) 
     181                  zah   = 0.25_wp * pahv(ji,jj-jp,jk) 
     182                  zslope_skew = triadj_g(ji,jj,jk,1-jp,kp) 
    163183                  ! Subtract s-coordinate slope at t-points to give slope rel to s surfaces 
    164184                  !    (do this by *adding* gradient of depth) 
    165                   zslope2 = zslope_skew + ( gdept(ji,jj+1,jk,Kmm) - gdept(ji,jj,jk,Kmm) ) * r1_e2v(ji,jj) * vmask(ji,jj,jk+kp) 
     185                  zslope2 = zslope_skew + ( gdept(ji,jj-jp+1,jk,Kmm) - gdept(ji,jj-jp,jk,Kmm) ) * r1_e2v(ji,jj-jp) * vmask(ji,jj-jp,jk+kp) 
    166186                  zslope2 = zslope2 * zslope2 
    167                   ah_wslp2(ji,jj+jp,jk+kp) = ah_wslp2(ji,jj+jp,jk+kp) + zah * zbv * ze3wr * r1_e1e2t(ji,jj+jp) * zslope2 
    168                   akz     (ji,jj+jp,jk+kp) = akz     (ji,jj+jp,jk+kp) + zah * r1_e2v(ji,jj)     & 
    169                      &                                                      * r1_e2v(ji,jj) * vmask(ji,jj,jk+kp) 
     187                  ah_wslp2(ji,jj,jk+kp) = ah_wslp2(ji,jj,jk+kp) + zah * zbv * ze3wr * r1_e1e2t(ji,jj) * zslope2 
     188                  akz     (ji,jj,jk+kp) = akz     (ji,jj,jk+kp) + zah * r1_e2v(ji,jj-jp)     & 
     189                     &                                                      * r1_e2v(ji,jj-jp) * vmask(ji,jj-jp,jk+kp) 
    170190                  ! 
    171                   IF( ln_ldfeiv_dia )   zpsi_vw(ji,jj,jk+kp) = zpsi_vw(ji,jj,jk+kp)   & 
    172                      &                                       + 0.25 * aeiv(ji,jj,jk) * e1v(ji,jj) * zslope_skew 
    173191               END_3D 
    174192            END DO 
    … …  
    178196            ! 
    179197            IF( ln_traldf_blp ) THEN                ! bilaplacian operator 
    180                DO_3D_10_10( 2, jpkm1 ) 
     198               DO_3D_00_00( 2, jpkm1 ) 
    181199                  akz(ji,jj,jk) = 16._wp * ah_wslp2(ji,jj,jk)   & 
    182200                     &          * (  akz(ji,jj,jk) + ah_wslp2(ji,jj,jk) / ( e3w(ji,jj,jk,Kmm) * e3w(ji,jj,jk,Kmm) )  ) 
    183201               END_3D 
    184202            ELSEIF( ln_traldf_lap ) THEN              ! laplacian operator 
    185                DO_3D_10_10( 2, jpkm1 ) 
     203               DO_3D_00_00( 2, jpkm1 ) 
    186204                  ze3w_2 = e3w(ji,jj,jk,Kmm) * e3w(ji,jj,jk,Kmm) 
    187205                  zcoef0 = rDt * (  akz(ji,jj,jk) + ah_wslp2(ji,jj,jk) / ze3w_2  ) 
    … …  
    191209           ! 
    192210         ELSE                                    ! 33 flux set to zero with akz=ah_wslp2 ==>> computed in full implicit 
    193             akz(:,:,:) = ah_wslp2(:,:,:)       
    194          ENDIF 
    195          ! 
    196          IF( ln_ldfeiv_dia .AND. cdtype == 'TRA' )   CALL ldf_eiv_dia( zpsi_uw, zpsi_vw, Kmm ) 
     211            DO_3D_00_00( 1, jpk ) 
     212               akz(ji,jj,jk) = ah_wslp2(ji,jj,jk) 
     213            END_3D 
     214         ENDIF 
     215         ! 
     216         ! TEMP: These changes not necessary if lbc_lnk is removed from ldf_eiv_dia and XIOS has subdomain support 
     217         IF( ntile == 0 .OR. ntile == nijtile )  THEN                ! Do only for the full domain 
     218            IF( ln_ldfeiv_dia .AND. cdtype == 'TRA' ) THEN 
     219               IF( ln_tile ) CALL dom_tile( ntsi, ntsj, ntei, ntej, ktile = 0 ) 
     220 
     221               zpsi_uw(:,:,:) = 0._wp 
     222               zpsi_vw(:,:,:) = 0._wp 
     223 
     224               DO jp = 0, 1 
     225                  DO kp = 0, 1 
     226                     DO_3D_10_10( 1, jpkm1 ) 
     227                        zpsi_uw(ji,jj,jk+kp) = zpsi_uw(ji,jj,jk+kp) & 
     228                           & + 0.25_wp * aeiu(ji,jj,jk) * e2u(ji,jj) * triadi_g(ji+jp,jj,jk,1-jp,kp) 
     229                        zpsi_vw(ji,jj,jk+kp) = zpsi_vw(ji,jj,jk+kp) & 
     230                           & + 0.25_wp * aeiv(ji,jj,jk) * e1v(ji,jj) * triadj_g(ji,jj+jp,jk,1-jp,kp) 
     231                     END_3D 
     232                  END DO 
     233               END DO 
     234               CALL ldf_eiv_dia( zpsi_uw, zpsi_vw, Kmm ) 
     235 
     236               IF( ln_tile ) CALL dom_tile( ntsi, ntsj, ntei, ntej, ktile = nijtile ) 
     237            ENDIF 
     238         ENDIF 
    197239         ! 
    198240      ENDIF                                  !==  end 1st pass only  ==! 
    … …  
    211253            zdjt(ji,jj,jk) = ( pt(ji  ,jj+1,jk,jn) - pt(ji,jj,jk,jn) ) * vmask(ji,jj,jk) 
    212254         END_3D 
     255         ! TODO: NOT TESTED- requires zps 
    213256         IF( ln_zps .AND. l_grad_zps ) THEN    ! partial steps: correction at top/bottom ocean level 
    214257            DO_2D_10_10 
    … …  
    216259               zdjt(ji,jj,mbkv(ji,jj)) = pgv(ji,jj,jn) 
    217260            END_2D 
     261            ! TODO: NOT TESTED- requires isf 
    218262            IF( ln_isfcav ) THEN                   ! top level (ocean cavities only) 
    219263               DO_2D_10_10 
    … …  
    230274         DO jk = 1, jpkm1 
    231275            !                    !==  Vertical tracer gradient at level jk and jk+1 
    232             zdkt3d(:,:,1) = ( pt(:,:,jk,jn) - pt(:,:,jk+1,jn) ) * tmask(:,:,jk+1) 
     276            DO_2D_11_11 
     277               zdkt3d(ji,jj,1) = ( pt(ji,jj,jk,jn) - pt(ji,jj,jk+1,jn) ) * tmask(ji,jj,jk+1) 
     278            END_2D 
    233279            ! 
    234280            !                    ! surface boundary condition: zdkt3d(jk=0)=zdkt3d(jk=1) 
    235281            IF( jk == 1 ) THEN   ;   zdkt3d(:,:,0) = zdkt3d(:,:,1) 
    236             ELSE                 ;   zdkt3d(:,:,0) = ( pt(:,:,jk-1,jn) - pt(:,:,jk,jn) ) * tmask(:,:,jk) 
     282            ELSE 
     283               DO_2D_11_11 
     284                  zdkt3d(ji,jj,0) = ( pt(ji,jj,jk-1,jn) - pt(ji,jj,jk,jn) ) * tmask(ji,jj,jk) 
     285               END_2D 
    237286            ENDIF 
    238287            ! 
    … …  
    374423      END DO                                                      ! end tracer loop 
    375424      !                                                           ! =============== 
    376    END SUBROUTINE tra_ldf_triad 
     425   END SUBROUTINE tra_ldf_triad_t 
    377426 
    378427   !!============================================================================== 

  • NEMO/branches/UKMO/dev_r12866_HPC-02_Daley_Tiling_trial_extra_halo/src/OCE/TRA/tramle.F90

    r12489 r13409  
    7878      !!             Fox-Kemper and Ferrari, JPO, 38, 1166-1179, 2008 
    7979      !!---------------------------------------------------------------------- 
    80       INTEGER                         , INTENT(in   ) ::   kt         ! ocean time-step index 
    81       INTEGER                         , INTENT(in   ) ::   kit000     ! first time step index 
    82       INTEGER                         , INTENT(in   ) ::   Kmm        ! ocean time level index 
    83       CHARACTER(len=3)                , INTENT(in   ) ::   cdtype     ! =TRA or TRC (tracer indicator) 
    84       REAL(wp), DIMENSION(jpi,jpj,jpk), INTENT(inout) ::   pu         ! in : 3 ocean transport components 
    85       REAL(wp), DIMENSION(jpi,jpj,jpk), INTENT(inout) ::   pv         ! out: same 3  transport components 
    86       REAL(wp), DIMENSION(jpi,jpj,jpk), INTENT(inout) ::   pw         !   increased by the MLE induced transport 
     80      INTEGER                     , INTENT(in   ) ::   kt         ! ocean time-step index 
     81      INTEGER                     , INTENT(in   ) ::   kit000     ! first time step index 
     82      INTEGER                     , INTENT(in   ) ::   Kmm        ! ocean time level index 
     83      CHARACTER(len=3)            , INTENT(in   ) ::   cdtype     ! =TRA or TRC (tracer indicator) 
     84      REAL(wp), DIMENSION(A2D,jpk), INTENT(inout) ::   pu         ! in : 3 ocean transport components 
     85      REAL(wp), DIMENSION(A2D,jpk), INTENT(inout) ::   pv         ! out: same 3  transport components 
     86      REAL(wp), DIMENSION(A2D,jpk), INTENT(inout) ::   pw         !   increased by the MLE induced transport 
    8787      ! 
    8888      INTEGER  ::   ji, jj, jk          ! dummy loop indices 
    … …  
    9090      REAL(wp) ::   zcuw, zmuw, zc      ! local scalar 
    9191      REAL(wp) ::   zcvw, zmvw          !   -      - 
    92       INTEGER , DIMENSION(jpi,jpj)     :: inml_mle 
    93       REAL(wp), DIMENSION(jpi,jpj)     :: zpsim_u, zpsim_v, zmld, zbm, zhu, zhv, zn2, zLf_NH, zLf_MH 
    94       REAL(wp), DIMENSION(jpi,jpj,jpk) :: zpsi_uw, zpsi_vw 
     92      INTEGER , DIMENSION(A2D)     :: inml_mle 
     93      REAL(wp), DIMENSION(A2D)     :: zpsim_u, zpsim_v, zmld, zbm, zhu, zhv, zn2, zLf_MH 
     94      REAL(wp), DIMENSION(A2D,jpk) :: zpsi_uw, zpsi_vw 
     95      ! TEMP: These changes not necessary if using XIOS (subdomain support) 
     96      REAL(wp), DIMENSION(:,:),   ALLOCATABLE, SAVE :: zLf_NH 
     97      REAL(wp), DIMENSION(:,:,:), ALLOCATABLE, SAVE :: zpsiu_mle, zpsiv_mle 
    9598      !!---------------------------------------------------------------------- 
    9699      ! 
    97100      !                                      !==  MLD used for MLE  ==! 
    98101      !                                                ! compute from the 10m density to deal with the diurnal cycle 
    99       inml_mle(:,:) = mbkt(:,:) + 1                    ! init. to number of ocean w-level (T-level + 1) 
     102      DO_2D_11_11 
     103         inml_mle(ji,jj) = mbkt(ji,jj) + 1                    ! init. to number of ocean w-level (T-level + 1) 
     104      END_2D 
    100105      IF ( nla10 > 0 ) THEN                            ! avoid case where first level is thicker than 10m 
    101106         DO_3DS_11_11( jpkm1, nlb10, -1 ) 
    … …  
    134139      END SELECT 
    135140      !                                                ! convert density into buoyancy 
    136       zbm(:,:) = + grav * zbm(:,:) / MAX( e3t(:,:,1,Kmm), zmld(:,:) ) 
     141      DO_2D_11_11 
     142         zbm(ji,jj) = + grav * zbm(ji,jj) / MAX( e3t(ji,jj,1,Kmm), zmld(ji,jj) ) 
     143      END_2D 
    137144      ! 
    138145      ! 
    … …  
    205212      END DO 
    206213 
     214      ! TEMP: These changes not necessary if using XIOS (subdomain support) 
    207215      IF( cdtype == 'TRA') THEN              !==  outputs  ==! 
    208          ! 
    209          zLf_NH(:,:) = SQRT( rb_c * zmld(:,:) ) * r1_ft(:,:)      ! Lf = N H / f 
    210          CALL iom_put( "Lf_NHpf" , zLf_NH  )    ! Lf = N H / f 
     216         IF( kt == nit000 .AND. (ntile == 0 .OR. ntile == 1) )  THEN             ! Do only on the first tile and timestep 
     217            ALLOCATE( zLf_NH(jpi,jpj), zpsiu_mle(jpi,jpj,jpk), zpsiv_mle(jpi,jpj,jpk) ) 
     218         ENDIF 
     219         ! 
     220         DO_2D_11_11 
     221            zLf_NH(ji,jj) = SQRT( rb_c * zmld(ji,jj) ) * r1_ft(ji,jj)      ! Lf = N H / f 
     222         END_2D 
    211223         ! 
    212224         ! divide by cross distance to give streamfunction with dimensions m^2/s 
    213          DO jk = 1, ikmax+1 
    214             zpsi_uw(:,:,jk) = zpsi_uw(:,:,jk) * r1_e2u(:,:) 
    215             zpsi_vw(:,:,jk) = zpsi_vw(:,:,jk) * r1_e1v(:,:) 
    216          END DO 
    217          CALL iom_put( "psiu_mle", zpsi_uw )    ! i-mle streamfunction 
    218          CALL iom_put( "psiv_mle", zpsi_vw )    ! j-mle streamfunction 
     225         DO_3D_11_11( 1, ikmax+1 ) 
     226            zpsiu_mle(ji,jj,jk) = zpsi_uw(ji,jj,jk) * r1_e2u(ji,jj) 
     227            zpsiv_mle(ji,jj,jk) = zpsi_vw(ji,jj,jk) * r1_e1v(ji,jj) 
     228         END_3D 
     229 
     230         IF( ntile == 0 .OR. ntile == nijtile ) THEN                       ! Do only on the last tile 
     231            CALL iom_put( "Lf_NHpf" , zLf_NH  )    ! Lf = N H / f 
     232            CALL iom_put( "psiu_mle", zpsiu_mle )    ! i-mle streamfunction 
     233            CALL iom_put( "psiv_mle", zpsiv_mle )    ! j-mle streamfunction 
     234         ENDIF 
    219235      ENDIF 
    220236      ! 

  • NEMO/branches/UKMO/dev_r12866_HPC-02_Daley_Tiling_trial_extra_halo/src/OCE/TRA/tranpc.F90

    r12489 r13409  
    1717   USE oce            ! ocean dynamics and active tracers 
    1818   USE dom_oce        ! ocean space and time domain 
     19   ! TEMP: This change not necessary after trd_tra is tiled and extra haloes development (lbc_lnk removed) 
     20   USE domain, ONLY : dom_tile 
    1921   USE phycst         ! physical constants 
    2022   USE zdf_oce        ! ocean vertical physics 
    … …  
    3335   PUBLIC   tra_npc    ! routine called by step.F90 
    3436 
     37   INTEGER  ::   nnpcc        ! number of statically instable water column 
     38    
    3539   !! * Substitutions 
    3640#  include "do_loop_substitute.h90" 
    … …  
    6367      ! 
    6468      INTEGER  ::   ji, jj, jk   ! dummy loop indices 
    65       INTEGER  ::   inpcc        ! number of statically instable water column 
    6669      INTEGER  ::   jiter, ikbot, ikp, ikup, ikdown, ilayer, ik_low   ! local integers 
    6770      LOGICAL  ::   l_bottom_reached, l_column_treated 
    … …  
    6972      REAL(wp) ::   zsa, zbeta, zsum_sali, zsum_beta, zbw, zrw, z1_rDt 
    7073      REAL(wp), PARAMETER ::   zn2_zero = 1.e-14_wp             ! acceptance criteria for neutrality (N2==0) 
    71       REAL(wp), DIMENSION(        jpk     )   ::   zvn2         ! vertical profile of N2 at 1 given point... 
    72       REAL(wp), DIMENSION(        jpk,jpts)   ::   zvts, zvab   ! vertical profile of T & S , and  alpha & betaat 1 given point 
    73       REAL(wp), DIMENSION(jpi,jpj,jpk     )   ::   zn2          ! N^2  
    74       REAL(wp), DIMENSION(jpi,jpj,jpk,jpts)   ::   zab          ! alpha and beta 
    75       REAL(wp), ALLOCATABLE, DIMENSION(:,:,:) ::   ztrdt, ztrds ! 3D workspace 
     74      REAL(wp), DIMENSION(    jpk     )   ::   zvn2             ! vertical profile of N2 at 1 given point... 
     75      REAL(wp), DIMENSION(    jpk,jpts)   ::   zvts, zvab       ! vertical profile of T & S , and  alpha & betaat 1 given point 
     76      REAL(wp), DIMENSION(A2D,jpk     )   ::   zn2              ! N^2 
     77      REAL(wp), DIMENSION(A2D,jpk,jpts)   ::   zab              ! alpha and beta 
     78      ! TEMP: This change not necessary after trd_tra is tiled 
     79      REAL(wp), ALLOCATABLE, SAVE, DIMENSION(:,:,:) ::   ztrdt, ztrds ! 3D workspace 
    7680      ! 
    7781      LOGICAL, PARAMETER :: l_LB_debug = .FALSE. ! set to true if you want to follow what is 
    … …  
    8185      ! 
    8286      IF( ln_timing )   CALL timing_start('tra_npc') 
     87 
     88      IF( l_trdtra )   THEN 
     89         IF( ntile == 0 .OR. ntile == 1 )  THEN                       ! Do only on the first tile 
     90            ! TEMP: This can be A2D after trd_tra is tiled 
     91            ALLOCATE( ztrdt(jpi,jpj,jpk), ztrds(jpi,jpj,jpk) ) 
     92         ENDIF 
     93      ENDIF 
    8394      ! 
    8495      IF( MOD( kt, nn_npc ) == 0 ) THEN 
    8596         ! 
    8697         IF( l_trdtra )   THEN                    !* Save initial after fields 
    87             ALLOCATE( ztrdt(jpi,jpj,jpk) , ztrds(jpi,jpj,jpk) ) 
    88             ztrdt(:,:,:) = pts(:,:,:,jp_tem,Kaa)  
    89             ztrds(:,:,:) = pts(:,:,:,jp_sal,Kaa) 
    90          ENDIF 
    91          ! 
     98            DO_3D_00_00( 1, jpkm1 ) 
     99               ztrdt(ji,jj,jk) = pts(ji,jj,jk,jp_tem,Kaa) 
     100               ztrds(ji,jj,jk) = pts(ji,jj,jk,jp_sal,Kaa) 
     101            END_3D 
     102         ENDIF 
     103         ! 
     104         ! TODO: NOT TESTED- requires ORCA2 
    92105         IF( l_LB_debug ) THEN 
    93106            ! Location of 1 known convection site to follow what's happening in the water column 
    … …  
    100113         CALL bn2    ( pts(:,:,:,:,Kaa), zab, zn2, Kmm )    ! after Brunt-Vaisala  (given on W-points) 
    101114         ! 
    102          inpcc = 0 
     115         IF( ntile == 0 .OR. ntile == 1 ) nnpcc = 0         ! Do only on the first tile 
    103116         ! 
    104117         DO_2D_00_00 
    … …  
    159172                        ENDIF 
    160173                        ! 
    161                         IF( jiter == 1 )   inpcc = inpcc + 1  
     174                        IF( jiter == 1 )   nnpcc = nnpcc + 1  
    162175                        ! 
    163176                        IF( lp_monitor_point )   WRITE(numout, *) 'Negative N2 at ikp =',ikp,' for layer #', ilayer 
    … …  
    300313         END_2D 
    301314         ! 
    302          IF( l_trdtra ) THEN         ! send the Non penetrative mixing trends for diagnostic 
     315         ! TEMP: These changes not necessary after trd_tra is tiled and extra haloes development (lbc_lnk removed) 
     316         IF( l_trdtra ) THEN 
    303317            z1_rDt = 1._wp / (2._wp * rn_Dt) 
    304             ztrdt(:,:,:) = ( pts(:,:,:,jp_tem,Kaa) - ztrdt(:,:,:) ) * z1_rDt 
    305             ztrds(:,:,:) = ( pts(:,:,:,jp_sal,Kaa) - ztrds(:,:,:) ) * z1_rDt 
    306             CALL trd_tra( kt, Kmm, Krhs, 'TRA', jp_tem, jptra_npc, ztrdt ) 
    307             CALL trd_tra( kt, Kmm, Krhs, 'TRA', jp_sal, jptra_npc, ztrds ) 
    308             DEALLOCATE( ztrdt, ztrds ) 
    309          ENDIF 
    310          ! 
    311          CALL lbc_lnk_multi( 'tranpc', pts(:,:,:,jp_tem,Kaa), 'T', 1., pts(:,:,:,jp_sal,Kaa), 'T', 1. ) 
    312          ! 
    313          IF( lwp .AND. l_LB_debug ) THEN 
    314             WRITE(numout,*) 'Exiting tra_npc , kt = ',kt,', => numb. of statically instable water-columns: ', inpcc 
    315             WRITE(numout,*) 
     318 
     319            DO_3D_00_00( 1, jpkm1 ) 
     320               ztrdt(ji,jj,jk) = ( pts(ji,jj,jk,jp_tem,Kaa) - ztrdt(ji,jj,jk) ) * z1_rDt 
     321               ztrds(ji,jj,jk) = ( pts(ji,jj,jk,jp_sal,Kaa) - ztrds(ji,jj,jk) ) * z1_rDt 
     322            END_3D 
     323         ENDIF 
     324 
     325         IF( ntile == 0 .OR. ntile == nijtile )  THEN                ! Do only for the full domain 
     326            IF( ln_tile ) CALL dom_tile( ntsi, ntsj, ntei, ntej, ktile = 0 )         ! Use full domain 
     327 
     328            IF( l_trdtra ) THEN         ! send the Non penetrative mixing trends for diagnostic 
     329               ! TODO: TO BE TILED- trd_tra 
     330               CALL trd_tra( kt, Kmm, Krhs, 'TRA', jp_tem, jptra_npc, ztrdt ) 
     331               CALL trd_tra( kt, Kmm, Krhs, 'TRA', jp_sal, jptra_npc, ztrds ) 
     332               DEALLOCATE( ztrdt, ztrds ) 
     333            ENDIF 
     334 
     335            IF( ln_tile ) CALL dom_tile( ntsi, ntsj, ntei, ntej, ktile = nijtile )   ! Revert to tile domain 
     336            ! 
     337            CALL lbc_lnk_multi( 'tranpc', pts(:,:,:,jp_tem,Kaa), 'T', 1., pts(:,:,:,jp_sal,Kaa), 'T', 1. ) 
     338            ! 
     339            IF( lwp .AND. l_LB_debug ) THEN 
     340               WRITE(numout,*) 'Exiting tra_npc , kt = ',kt,', => numb. of statically instable water-columns: ', nnpcc 
     341               WRITE(numout,*) 
     342            ENDIF 
    316343         ENDIF 
    317344         ! 

  • NEMO/branches/UKMO/dev_r12866_HPC-02_Daley_Tiling_trial_extra_halo/src/OCE/TRA/traqsr.F90

    r12738 r13409  
    2222   USE phycst         ! physical constants 
    2323   USE dom_oce        ! ocean space and time domain 
     24   USE domain, ONLY : dom_tile 
    2425   USE sbc_oce        ! surface boundary condition: ocean 
    2526   USE trc_oce        ! share SMS/Ocean variables 
    … …  
    112113      REAL(wp) ::   zz0 , zz1                !    -         - 
    113114      REAL(wp) ::   zCb, zCmax, zze, zpsi, zpsimax, zdelpsi, zCtot, zCze 
    114       REAL(wp) ::   zlogc, zlogc2, zlogc3  
     115      REAL(wp) ::   zlogc, zlogc2, zlogc3 
     116      REAL(wp), SAVE :: zsum1 
    115117      REAL(wp), ALLOCATABLE, DIMENSION(:,:)   :: zekb, zekg, zekr 
    116       REAL(wp), ALLOCATABLE, DIMENSION(:,:,:) :: ze0, ze1, ze2, ze3, zea, ztrdt 
     118      ! TEMP: These changes not necessary after trd_tra is tiled 
     119      REAL(wp), ALLOCATABLE, SAVE, DIMENSION(:,:,:) ::   ztrdt 
     120      REAL(wp), ALLOCATABLE, DIMENSION(:,:,:) :: ze0, ze1, ze2, ze3, zea 
    117121      REAL(wp), ALLOCATABLE, DIMENSION(:,:,:) :: zetot, zchl3d 
    118122      !!---------------------------------------------------------------------- 
    … …  
    120124      IF( ln_timing )   CALL timing_start('tra_qsr') 
    121125      ! 
    122       IF( kt == nit000 ) THEN 
    123          IF(lwp) WRITE(numout,*) 
    124          IF(lwp) WRITE(numout,*) 'tra_qsr : penetration of the surface solar radiation' 
    125          IF(lwp) WRITE(numout,*) '~~~~~~~' 
     126      IF( ntile == 0 .OR. ntile == 1 )  THEN                       ! Do only on the first tile 
     127         IF( kt == nit000 ) THEN 
     128            IF(lwp) WRITE(numout,*) 
     129            IF(lwp) WRITE(numout,*) 'tra_qsr : penetration of the surface solar radiation' 
     130            IF(lwp) WRITE(numout,*) '~~~~~~~' 
     131         ENDIF 
    126132      ENDIF 
    127133      ! 
    128134      IF( l_trdtra ) THEN      ! trends diagnostic: save the input temperature trend 
    129          ALLOCATE( ztrdt(jpi,jpj,jpk) )  
    130          ztrdt(:,:,:) = pts(:,:,:,jp_tem,Krhs) 
     135         IF( ntile == 0 .OR. ntile == 1 )  THEN                       ! Do only on the first tile 
     136            ! TEMP: This can be A2D after trd_tra is tiled 
     137            ALLOCATE( ztrdt(jpi,jpj,jpk) ) 
     138         ENDIF 
     139 
     140         DO_3D_00_00( 1, jpkm1 ) 
     141            ztrdt(ji,jj,jk) = pts(ji,jj,jk,jp_tem,Krhs) 
     142         END_3D 
    131143      ENDIF 
    132144      ! 
    … …  
    136148      IF( kt == nit000 ) THEN          !==  1st time step  ==! 
    137149         IF( ln_rstart .AND. iom_varid( numror, 'qsr_hc_b', ldstop = .FALSE. ) > 0  .AND. .NOT.l_1st_euler ) THEN    ! read in restart 
    138             IF(lwp) WRITE(numout,*) '          nit000-1 qsr tracer content forcing field read in the restart file' 
    139150            z1_2 = 0.5_wp 
    140             CALL iom_get( numror, jpdom_auto, 'qsr_hc_b', qsr_hc_b, ldxios = lrxios )   ! before heat content trend due to Qsr flux 
     151            IF( ntile == 0 .OR. ntile == 1 )  THEN                       ! Do only on the first tile 
     152               IF(lwp) WRITE(numout,*) '          nit000-1 qsr tracer content forcing field read in the restart file' 
     153               CALL iom_get( numror, jpdom_auto, 'qsr_hc_b', qsr_hc_b, ldxios = lrxios )   ! before heat content trend due to Qsr flux 
     154            ENDIF 
    141155         ELSE                                           ! No restart or restart not found: Euler forward time stepping 
    142156            z1_2 = 1._wp 
    143             qsr_hc_b(:,:,:) = 0._wp 
     157            DO_3D_00_00( 1, jpk ) 
     158               qsr_hc_b(ji,jj,jk) = 0._wp 
     159            END_3D 
    144160         ENDIF 
    145161      ELSE                             !==  Swap of qsr heat content  ==! 
    146162         z1_2 = 0.5_wp 
    147          qsr_hc_b(:,:,:) = qsr_hc(:,:,:) 
     163         DO_3D_00_00( 1, jpk ) 
     164            qsr_hc_b(ji,jj,jk) = qsr_hc(ji,jj,jk) 
     165         END_3D 
    148166      ENDIF 
    149167      ! 
    … …  
    154172      CASE( np_BIO )                   !==  bio-model fluxes  ==! 
    155173         ! 
    156          DO jk = 1, nksr 
    157             qsr_hc(:,:,jk) = r1_rho0_rcp * ( etot3(:,:,jk) - etot3(:,:,jk+1) ) 
    158          END DO 
     174         DO_3D_00_00( 1, nksr ) 
     175            qsr_hc(ji,jj,jk) = r1_rho0_rcp * ( etot3(ji,jj,jk) - etot3(ji,jj,jk+1) ) 
     176         END_3D 
    159177         ! 
    160178      CASE( np_RGB , np_RGBc )         !==  R-G-B fluxes  ==! 
    161179         ! 
    162          ALLOCATE( zekb(jpi,jpj)     , zekg(jpi,jpj)     , zekr  (jpi,jpj)     , & 
    163             &      ze0 (jpi,jpj,jpk) , ze1 (jpi,jpj,jpk) , ze2   (jpi,jpj,jpk) , & 
    164             &      ze3 (jpi,jpj,jpk) , zea (jpi,jpj,jpk) , zchl3d(jpi,jpj,jpk)   )  
     180         ALLOCATE( zekb(A2D)     , zekg(A2D)     , zekr  (A2D)     , & 
     181            &      ze0 (A2D,jpk) , ze1 (A2D,jpk) , ze2   (A2D,jpk) , & 
     182            &      ze3 (A2D,jpk) , zea (A2D,jpk) , zchl3d(A2D,jpk)   ) 
    165183         ! 
    166184         IF( nqsr == np_RGBc ) THEN          !*  Variable Chlorophyll 
    167             CALL fld_read( kt, 1, sf_chl )         ! Read Chl data and provides it at the current time step 
    168             DO jk = 1, nksr + 1 
    169                DO jj = 2, jpjm1                       ! Separation in R-G-B depending of the surface Chl 
    170                   DO ji = 2, jpim1 
    171                      zchl    = MIN( 10. , MAX( 0.03, sf_chl(1)%fnow(ji,jj,1) ) ) 
    172                      zCtot   = 40.6  * zchl**0.459 
    173                      zze     = 568.2 * zCtot**(-0.746) 
    174                      IF( zze > 102. ) zze = 200.0 * zCtot**(-0.293) 
    175                      zpsi    = gdepw(ji,jj,jk,Kmm) / zze 
    176                      ! 
    177                      zlogc   = LOG( zchl ) 
    178                      zlogc2  = zlogc * zlogc 
    179                      zlogc3  = zlogc * zlogc * zlogc 
    180                      zCb     = 0.768 + 0.087 * zlogc - 0.179 * zlogc2 - 0.025 * zlogc3 
    181                      zCmax   = 0.299 - 0.289 * zlogc + 0.579 * zlogc2 
    182                      zpsimax = 0.6   - 0.640 * zlogc + 0.021 * zlogc2 + 0.115 * zlogc3 
    183                      zdelpsi = 0.710 + 0.159 * zlogc + 0.021 * zlogc2 
    184                      zCze    = 1.12  * (zchl)**0.803  
    185                      ! 
    186                      zchl3d(ji,jj,jk) = zCze * ( zCb + zCmax * EXP( -( (zpsi - zpsimax) / zdelpsi )**2 ) ) 
    187                   END DO 
    188                   ! 
    189                END DO 
    190             END DO 
     185            IF( ntile == 0 .OR. ntile == 1 )  THEN                                         ! Do only for the full domain 
     186               IF( ln_tile ) CALL dom_tile( ntsi, ntsj, ntei, ntej, ktile = 0 )            ! Use full domain 
     187               CALL fld_read( kt, 1, sf_chl )         ! Read Chl data and provides it at the current time step 
     188               IF( ln_tile ) CALL dom_tile( ntsi, ntsj, ntei, ntej, ktile = 1 )            ! Revert to tile domain 
     189            ENDIF 
     190            DO_3D_00_00( 1, nksr + 1 )                ! Separation in R-G-B depending of the surface Chl 
     191               zchl    = MIN( 10. , MAX( 0.03, sf_chl(1)%fnow(ji,jj,1) ) ) 
     192               zCtot   = 40.6  * zchl**0.459 
     193               zze     = 568.2 * zCtot**(-0.746) 
     194               IF( zze > 102. ) zze = 200.0 * zCtot**(-0.293) 
     195               zpsi    = gdepw(ji,jj,jk,Kmm) / zze 
     196               ! 
     197               zlogc   = LOG( zchl ) 
     198               zlogc2  = zlogc * zlogc 
     199               zlogc3  = zlogc * zlogc * zlogc 
     200               zCb     = 0.768 + 0.087 * zlogc - 0.179 * zlogc2 - 0.025 * zlogc3 
     201               zCmax   = 0.299 - 0.289 * zlogc + 0.579 * zlogc2 
     202               zpsimax = 0.6   - 0.640 * zlogc + 0.021 * zlogc2 + 0.115 * zlogc3 
     203               zdelpsi = 0.710 + 0.159 * zlogc + 0.021 * zlogc2 
     204               zCze    = 1.12  * (zchl)**0.803 
     205               ! 
     206               zchl3d(ji,jj,jk) = zCze * ( zCb + zCmax * EXP( -( (zpsi - zpsimax) / zdelpsi )**2 ) ) 
     207            END_3D 
    191208         ELSE                                !* constant chrlorophyll 
    192            DO jk = 1, nksr + 1 
    193               zchl3d(:,:,jk) = 0.05  
    194             ENDDO 
     209            DO_3D_11_11( 1, nksr + 1 ) 
     210               zchl3d(ji,jj,jk) = 0.05 
     211            END_3D 
    195212         ENDIF 
    196213         ! 
    … …  
    256273         ENDIF 
    257274      END_2D 
    258       CALL lbc_lnk( 'traqsr', fraqsr_1lev(:,:), 'T', 1._wp ) 
    259       ! 
    260       IF( iom_use('qsr3d') ) THEN      ! output the shortwave Radiation distribution 
    261          ALLOCATE( zetot(jpi,jpj,jpk) ) 
    262          zetot(:,:,nksr+1:jpk) = 0._wp     ! below ~400m set to zero 
    263          DO jk = nksr, 1, -1 
    264             zetot(:,:,jk) = zetot(:,:,jk+1) + qsr_hc(:,:,jk) * rho0_rcp 
    265          END DO          
    266          CALL iom_put( 'qsr3d', zetot )   ! 3D distribution of shortwave Radiation 
    267          DEALLOCATE( zetot )  
    268       ENDIF 
    269       ! 
    270       IF( lrst_oce ) THEN     ! write in the ocean restart file 
    271          IF( lwxios ) CALL iom_swap(      cwxios_context          ) 
    272          CALL iom_rstput( kt, nitrst, numrow, 'qsr_hc_b'   , qsr_hc     , ldxios = lwxios ) 
    273          CALL iom_rstput( kt, nitrst, numrow, 'fraqsr_1lev', fraqsr_1lev, ldxios = lwxios )  
    274          IF( lwxios ) CALL iom_swap(      cxios_context          ) 
    275       ENDIF 
    276       ! 
     275      ! TEMP: This change not necessary after extra haloes development (lbc_lnk removed) 
     276      IF( ntile == 0 .OR. ntile == nijtile ) THEN                       ! Do only on the last tile 
     277         CALL lbc_lnk( 'traqsr', fraqsr_1lev(:,:), 'T', 1._wp ) 
     278      ENDIF 
     279      ! 
     280      ! TEMP: This change not necessary and working array can use A2D if using XIOS (subdomain support) 
     281      IF( ntile == 0 .OR. ntile == nijtile )  THEN                ! Do only for the full domain 
     282         IF( iom_use('qsr3d') ) THEN      ! output the shortwave Radiation distribution 
     283            ALLOCATE( zetot(jpi,jpj,jpk) ) 
     284            zetot(:,:,nksr+1:jpk) = 0._wp     ! below ~400m set to zero 
     285            DO jk = nksr, 1, -1 
     286               zetot(:,:,jk) = zetot(:,:,jk+1) + qsr_hc(:,:,jk) * rho0_rcp 
     287            END DO 
     288            CALL iom_put( 'qsr3d', zetot )   ! 3D distribution of shortwave Radiation 
     289            DEALLOCATE( zetot ) 
     290         ENDIF 
     291      ENDIF 
     292      ! 
     293      IF( ntile == 0 .OR. ntile == nijtile )  THEN                ! Do only on the last tile 
     294         IF( lrst_oce ) THEN     ! write in the ocean restart file 
     295            IF( lwxios ) CALL iom_swap(      cwxios_context          ) 
     296            CALL iom_rstput( kt, nitrst, numrow, 'qsr_hc_b'   , qsr_hc     , ldxios = lwxios ) 
     297            CALL iom_rstput( kt, nitrst, numrow, 'fraqsr_1lev', fraqsr_1lev, ldxios = lwxios ) 
     298            IF( lwxios ) CALL iom_swap(      cxios_context          ) 
     299         ENDIF 
     300      ENDIF 
     301      ! 
     302      ! TEMP: These changes not necessary after trd_tra is tiled 
    277303      IF( l_trdtra ) THEN     ! qsr tracers trends saved for diagnostics 
    278          ztrdt(:,:,:) = pts(:,:,:,jp_tem,Krhs) - ztrdt(:,:,:) 
    279          CALL trd_tra( kt, Kmm, Krhs, 'TRA', jp_tem, jptra_qsr, ztrdt ) 
    280          DEALLOCATE( ztrdt )  
     304         DO_3D_00_00( 1, jpkm1 ) 
     305            ztrdt(ji,jj,jk) = pts(ji,jj,jk,jp_tem,Krhs) - ztrdt(ji,jj,jk) 
     306         END_3D 
     307 
     308         IF( ntile == 0 .OR. ntile == nijtile )  THEN                ! Do only for the full domain 
     309            IF( ln_tile ) CALL dom_tile( ntsi, ntsj, ntei, ntej, ktile = 0 )         ! Use full domain 
     310 
     311            ! TODO: TO BE TILED- trd_tra 
     312            CALL trd_tra( kt, Kmm, Krhs, 'TRA', jp_tem, jptra_qsr, ztrdt ) 
     313            DEALLOCATE( ztrdt ) 
     314 
     315            IF( ln_tile ) CALL dom_tile( ntsi, ntsj, ntei, ntej, ktile = nijtile )   ! Revert to tile domain 
     316         ENDIF 
    281317      ENDIF 
    282318      !                       ! print mean trends (used for debugging) 
    283       IF(sn_cfctl%l_prtctl)   CALL prt_ctl( tab3d_1=pts(:,:,:,jp_tem,Krhs), clinfo1=' qsr  - Ta: ', mask1=tmask, clinfo3='tra-ta' ) 
     319      IF(sn_cfctl%l_prtctl)   CALL prt_ctl( tab3d_1=pts(:,:,:,jp_tem,Krhs), clinfo1=' qsr  - Ta: ', mask1=tmask, psum1=zsum1, & 
     320         &                                  clinfo3='tra-ta' ) 
    284321      ! 
    285322      IF( ln_timing )   CALL timing_stop('tra_qsr') 

  • NEMO/branches/UKMO/dev_r12866_HPC-02_Daley_Tiling_trial_extra_halo/src/OCE/TRA/trasbc.F90

    r12738 r13409  
    1919   USE sbc_oce        ! surface boundary condition: ocean 
    2020   USE dom_oce        ! ocean space domain variables 
     21   ! TEMP: This change not necessary after trd_tra is tiled 
     22   USE domain, ONLY : dom_tile 
    2123   USE phycst         ! physical constant 
    2224   USE eosbn2         ! Equation Of State 
    … …  
    7880      INTEGER  ::   ikt, ikb                    ! local integers 
    7981      REAL(wp) ::   zfact, z1_e3t, zdep, ztim   ! local scalar 
    80       REAL(wp), ALLOCATABLE, DIMENSION(:,:,:) ::  ztrdt, ztrds 
     82      REAL(wp), SAVE :: zsum1, zsum2 
     83      ! TEMP: This change not necessary after trd_tra is tiled 
     84      REAL(wp), ALLOCATABLE, SAVE, DIMENSION(:,:,:) ::  ztrdt, ztrds 
    8185      !!---------------------------------------------------------------------- 
    8286      ! 
    8387      IF( ln_timing )   CALL timing_start('tra_sbc') 
    8488      ! 
    85       IF( kt == nit000 ) THEN 
    86          IF(lwp) WRITE(numout,*) 
    87          IF(lwp) WRITE(numout,*) 'tra_sbc : TRAcer Surface Boundary Condition' 
    88          IF(lwp) WRITE(numout,*) '~~~~~~~ ' 
     89      IF( ntile == 0 .OR. ntile == 1 )  THEN                       ! Do only on the first tile 
     90         IF( kt == nit000 ) THEN 
     91            IF(lwp) WRITE(numout,*) 
     92            IF(lwp) WRITE(numout,*) 'tra_sbc : TRAcer Surface Boundary Condition' 
     93            IF(lwp) WRITE(numout,*) '~~~~~~~ ' 
     94         ENDIF 
    8995      ENDIF 
    9096      ! 
    9197      IF( l_trdtra ) THEN                    !* Save ta and sa trends 
    92          ALLOCATE( ztrdt(jpi,jpj,jpk) , ztrds(jpi,jpj,jpk) )  
    93          ztrdt(:,:,:) = pts(:,:,:,jp_tem,Krhs) 
    94          ztrds(:,:,:) = pts(:,:,:,jp_sal,Krhs) 
     98         IF( ntile == 0 .OR. ntile == 1 )  THEN                       ! Do only on the first tile 
     99            ! TEMP: This can be A2D after trd_tra is tiled 
     100            ALLOCATE( ztrdt(jpi,jpj,jpk), ztrds(jpi,jpj,jpk) ) 
     101         ENDIF 
     102 
     103         DO_3D_00_00( 1, jpkm1 ) 
     104            ztrdt(ji,jj,jk) = pts(ji,jj,jk,jp_tem,Krhs) 
     105            ztrds(ji,jj,jk) = pts(ji,jj,jk,jp_sal,Krhs) 
     106         END_3D 
    95107      ENDIF 
    96108      ! 
    97109!!gm  This should be moved into sbcmod.F90 module ? (especially now that ln_traqsr is read in namsbc namelist) 
    98110      IF( .NOT.ln_traqsr ) THEN     ! no solar radiation penetration 
    99          qns(:,:) = qns(:,:) + qsr(:,:)      ! total heat flux in qns 
    100          qsr(:,:) = 0._wp                     ! qsr set to zero 
     111         DO_2D_00_00 
     112            qns(ji,jj) = qns(ji,jj) + qsr(ji,jj)      ! total heat flux in qns 
     113            qsr(ji,jj) = 0._wp                        ! qsr set to zero 
     114         END_2D 
    101115      ENDIF 
    102116 
    … …  
    108122         IF( ln_rstart .AND.    &               ! Restart: read in restart file 
    109123              & iom_varid( numror, 'sbc_hc_b', ldstop = .FALSE. ) > 0 ) THEN 
    110             IF(lwp) WRITE(numout,*) '          nit000-1 sbc tracer content field read in the restart file' 
    111124            zfact = 0.5_wp 
    112             sbc_tsc(:,:,:) = 0._wp 
    113             CALL iom_get( numror, jpdom_auto, 'sbc_hc_b', sbc_tsc_b(:,:,jp_tem), ldxios = lrxios )   ! before heat content sbc trend 
    114             CALL iom_get( numror, jpdom_auto, 'sbc_sc_b', sbc_tsc_b(:,:,jp_sal), ldxios = lrxios )   ! before salt content sbc trend 
     125            IF( ntile == 0 .OR. ntile == 1 )  THEN                       ! Do only on the first tile 
     126               IF(lwp) WRITE(numout,*) '          nit000-1 sbc tracer content field read in the restart file' 
     127               sbc_tsc(:,:,:) = 0._wp 
     128               CALL iom_get( numror, jpdom_auto, 'sbc_hc_b', sbc_tsc_b(:,:,jp_tem), ldxios = lrxios )   ! before heat content sbc trend 
     129               CALL iom_get( numror, jpdom_auto, 'sbc_sc_b', sbc_tsc_b(:,:,jp_sal), ldxios = lrxios )   ! before salt content sbc trend 
     130            ENDIF 
    115131         ELSE                                   ! No restart or restart not found: Euler forward time stepping 
    116132            zfact = 1._wp 
    117             sbc_tsc(:,:,:) = 0._wp 
    118             sbc_tsc_b(:,:,:) = 0._wp 
     133            DO_2D_00_00 
     134               sbc_tsc(ji,jj,:) = 0._wp 
     135               sbc_tsc_b(ji,jj,:) = 0._wp 
     136            END_2D 
    119137         ENDIF 
    120138      ELSE                                !* other time-steps: swap of forcing fields 
    121139         zfact = 0.5_wp 
    122          sbc_tsc_b(:,:,:) = sbc_tsc(:,:,:) 
     140         DO_2D_00_00 
     141            sbc_tsc_b(ji,jj,:) = sbc_tsc(ji,jj,:) 
     142         END_2D 
    123143      ENDIF 
    124144      !                             !==  Now sbc tracer content fields  ==! 
    125       DO_2D_01_00 
     145      DO_2D_00_00 
    126146         sbc_tsc(ji,jj,jp_tem) = r1_rho0_rcp * qns(ji,jj)   ! non solar heat flux 
    127147         sbc_tsc(ji,jj,jp_sal) = r1_rho0     * sfx(ji,jj)   ! salt flux due to freezing/melting 
    128148      END_2D 
    129149      IF( ln_linssh ) THEN                !* linear free surface   
    130          DO_2D_01_00 
     150         DO_2D_00_00 
    131151            sbc_tsc(ji,jj,jp_tem) = sbc_tsc(ji,jj,jp_tem) + r1_rho0 * emp(ji,jj) * pts(ji,jj,1,jp_tem,Kmm) 
    132152            sbc_tsc(ji,jj,jp_sal) = sbc_tsc(ji,jj,jp_sal) + r1_rho0 * emp(ji,jj) * pts(ji,jj,1,jp_sal,Kmm) 
    133153         END_2D 
    134          IF( iom_use('emp_x_sst') )   CALL iom_put( "emp_x_sst", emp (:,:) * pts(:,:,1,jp_tem,Kmm) ) 
    135          IF( iom_use('emp_x_sss') )   CALL iom_put( "emp_x_sss", emp (:,:) * pts(:,:,1,jp_sal,Kmm) ) 
     154         IF( ntile == 0 .OR. ntile == nijtile )  THEN             ! Do only on the last tile 
     155            IF( iom_use('emp_x_sst') )   CALL iom_put( "emp_x_sst", emp (:,:) * pts(:,:,1,jp_tem,Kmm) ) 
     156            IF( iom_use('emp_x_sss') )   CALL iom_put( "emp_x_sss", emp (:,:) * pts(:,:,1,jp_sal,Kmm) ) 
     157         ENDIF 
    136158      ENDIF 
    137159      ! 
    138160      DO jn = 1, jpts               !==  update tracer trend  ==! 
    139          DO_2D_01_00 
     161         DO_2D_00_00 
    140162            pts(ji,jj,1,jn,Krhs) = pts(ji,jj,1,jn,Krhs) + zfact * ( sbc_tsc_b(ji,jj,jn) + sbc_tsc(ji,jj,jn) ) / e3t(ji,jj,1,Kmm) 
    141163         END_2D 
    142164      END DO 
    143165      !                   
    144       IF( lrst_oce ) THEN           !==  write sbc_tsc in the ocean restart file  ==! 
    145          IF( lwxios ) CALL iom_swap(      cwxios_context          ) 
    146          CALL iom_rstput( kt, nitrst, numrow, 'sbc_hc_b', sbc_tsc(:,:,jp_tem), ldxios = lwxios ) 
    147          CALL iom_rstput( kt, nitrst, numrow, 'sbc_sc_b', sbc_tsc(:,:,jp_sal), ldxios = lwxios ) 
    148          IF( lwxios ) CALL iom_swap(      cxios_context          ) 
     166      IF( ntile == 0 .OR. ntile == nijtile )  THEN                ! Do only on the last tile 
     167         IF( lrst_oce ) THEN           !==  write sbc_tsc in the ocean restart file  ==! 
     168            IF( lwxios ) CALL iom_swap(      cwxios_context          ) 
     169            CALL iom_rstput( kt, nitrst, numrow, 'sbc_hc_b', sbc_tsc(:,:,jp_tem), ldxios = lwxios ) 
     170            CALL iom_rstput( kt, nitrst, numrow, 'sbc_sc_b', sbc_tsc(:,:,jp_sal), ldxios = lwxios ) 
     171            IF( lwxios ) CALL iom_swap(      cxios_context          ) 
     172         ENDIF 
    149173      ENDIF 
    150174      ! 
    … …  
    155179      IF( ln_rnf ) THEN         ! input of heat and salt due to river runoff  
    156180         zfact = 0.5_wp 
    157          DO_2D_01_00 
     181         DO_2D_00_00 
    158182            IF( rnf(ji,jj) /= 0._wp ) THEN 
    159183               zdep = zfact / h_rnf(ji,jj) 
    … …  
    168192      ENDIF 
    169193 
    170       IF( iom_use('rnf_x_sst') )   CALL iom_put( "rnf_x_sst", rnf*pts(:,:,1,jp_tem,Kmm) )   ! runoff term on sst 
    171       IF( iom_use('rnf_x_sss') )   CALL iom_put( "rnf_x_sss", rnf*pts(:,:,1,jp_sal,Kmm) )   ! runoff term on sss 
     194      IF( ntile == 0 .OR. ntile == 1 )  THEN                       ! Do only on the first tile 
     195         IF( iom_use('rnf_x_sst') )   CALL iom_put( "rnf_x_sst", rnf*pts(:,:,1,jp_tem,Kmm) )   ! runoff term on sst 
     196         IF( iom_use('rnf_x_sss') )   CALL iom_put( "rnf_x_sss", rnf*pts(:,:,1,jp_sal,Kmm) )   ! runoff term on sss 
     197      ENDIF 
    172198 
    173199#if defined key_asminc 
    … …  
    180206          ! 
    181207         IF( ln_linssh ) THEN  
    182             DO_2D_01_00 
     208            DO_2D_00_00 
    183209               ztim = ssh_iau(ji,jj) / e3t(ji,jj,1,Kmm) 
    184210               pts(ji,jj,1,jp_tem,Krhs) = pts(ji,jj,1,jp_tem,Krhs) + pts(ji,jj,1,jp_tem,Kmm) * ztim 
    … …  
    186212            END_2D 
    187213         ELSE 
    188             DO_2D_01_00 
     214            DO_2D_00_00 
    189215               ztim = ssh_iau(ji,jj) / ( ht(ji,jj) + 1. - ssmask(ji, jj) ) 
    190216               pts(ji,jj,:,jp_tem,Krhs) = pts(ji,jj,:,jp_tem,Krhs) + pts(ji,jj,:,jp_tem,Kmm) * ztim 
    … …  
    197223#endif 
    198224      ! 
     225      ! TEMP: These changes not necessary after trd_tra is tiled 
    199226      IF( l_trdtra )   THEN                      ! save the horizontal diffusive trends for further diagnostics 
    200          ztrdt(:,:,:) = pts(:,:,:,jp_tem,Krhs) - ztrdt(:,:,:) 
    201          ztrds(:,:,:) = pts(:,:,:,jp_sal,Krhs) - ztrds(:,:,:) 
    202          CALL trd_tra( kt, Kmm, Krhs, 'TRA', jp_tem, jptra_nsr, ztrdt ) 
    203          CALL trd_tra( kt, Kmm, Krhs, 'TRA', jp_sal, jptra_nsr, ztrds ) 
    204          DEALLOCATE( ztrdt , ztrds )  
    205       ENDIF 
    206       ! 
    207       IF(sn_cfctl%l_prtctl)   CALL prt_ctl( tab3d_1=pts(:,:,:,jp_tem,Krhs), clinfo1=' sbc  - Ta: ', mask1=tmask,   & 
    208          &                                  tab3d_2=pts(:,:,:,jp_sal,Krhs), clinfo2=       ' Sa: ', mask2=tmask, clinfo3='tra' ) 
     227         DO_3D_00_00( 1, jpkm1 ) 
     228            ztrdt(ji,jj,jk) = pts(ji,jj,jk,jp_tem,Krhs) - ztrdt(ji,jj,jk) 
     229            ztrds(ji,jj,jk) = pts(ji,jj,jk,jp_sal,Krhs) - ztrds(ji,jj,jk) 
     230         END_3D 
     231 
     232         IF( ntile == 0 .OR. ntile == nijtile )  THEN                ! Do only for the full domain 
     233            IF( ln_tile ) CALL dom_tile( ntsi, ntsj, ntei, ntej, ktile = 0 )         ! Use full domain 
     234 
     235            ! TODO: TO BE TILED- trd_tra 
     236            CALL trd_tra( kt, Kmm, Krhs, 'TRA', jp_tem, jptra_nsr, ztrdt ) 
     237            CALL trd_tra( kt, Kmm, Krhs, 'TRA', jp_sal, jptra_nsr, ztrds ) 
     238            DEALLOCATE( ztrdt , ztrds ) 
     239 
     240            IF( ln_tile ) CALL dom_tile( ntsi, ntsj, ntei, ntej, ktile = nijtile )   ! Revert to tile domain 
     241         ENDIF 
     242      ENDIF 
     243      ! 
     244      IF(sn_cfctl%l_prtctl)   CALL prt_ctl( tab3d_1=pts(:,:,:,jp_tem,Krhs), clinfo1=' sbc  - Ta: ', mask1=tmask, psum1=zsum1, & 
     245         &                                  tab3d_2=pts(:,:,:,jp_sal,Krhs), clinfo2=       ' Sa: ', mask2=tmask, psum2=zsum2, & 
     246         &                                  clinfo3='tra' ) 
    209247      ! 
    210248      IF( ln_timing )   CALL timing_stop('tra_sbc') 

  • NEMO/branches/UKMO/dev_r12866_HPC-02_Daley_Tiling_trial_extra_halo/src/OCE/TRA/trazdf.F90

    r12489 r13409  
    1414   USE oce            ! ocean dynamics and tracers variables 
    1515   USE dom_oce        ! ocean space and time domain variables  
     16   ! TEMP: This change not necessary after trd_tra is tiled 
     17   USE domain, ONLY : dom_tile 
    1618   USE domvvl         ! variable volume 
    1719   USE phycst         ! physical constant 
    … …  
    5456      REAL(wp), DIMENSION(jpi,jpj,jpk,jpts,jpt), INTENT(inout) :: pts                 ! active tracers and RHS of tracer equation 
    5557      ! 
    56       INTEGER  ::   jk   ! Dummy loop indices 
    57       REAL(wp), DIMENSION(:,:,:), ALLOCATABLE ::   ztrdt, ztrds   ! 3D workspace 
     58      INTEGER  ::   ji, jj, jk   ! Dummy loop indices 
     59      REAL(wp), SAVE :: zsum1, zsum2 
     60      ! TEMP: This change not necessary after trd_tra is tiled 
     61      REAL(wp), DIMENSION(:,:,:), ALLOCATABLE, SAVE ::   ztrdt, ztrds   ! 3D workspace 
    5862      !!--------------------------------------------------------------------- 
    5963      ! 
    … …  
    6165      ! 
    6266      IF( kt == nit000 )  THEN 
    63          IF(lwp)WRITE(numout,*) 
    64          IF(lwp)WRITE(numout,*) 'tra_zdf : implicit vertical mixing on T & S' 
    65          IF(lwp)WRITE(numout,*) '~~~~~~~ ' 
     67         IF( ntile == 0 .OR. ntile == 1 )  THEN                   ! Do only on the first tile 
     68            IF(lwp)WRITE(numout,*) 
     69            IF(lwp)WRITE(numout,*) 'tra_zdf : implicit vertical mixing on T & S' 
     70            IF(lwp)WRITE(numout,*) '~~~~~~~ ' 
     71         ENDIF 
    6672      ENDIF 
    6773      ! 
    6874      IF( l_trdtra )   THEN                  !* Save ta and sa trends 
    69          ALLOCATE( ztrdt(jpi,jpj,jpk) , ztrds(jpi,jpj,jpk) ) 
    70          ztrdt(:,:,:) = pts(:,:,:,jp_tem,Kaa) 
    71          ztrds(:,:,:) = pts(:,:,:,jp_sal,Kaa) 
     75         IF( ntile == 0 .OR. ntile == 1 )  THEN                       ! Do only on the first tile 
     76            ! TEMP: This can be A2D after trd_tra is tiled 
     77            ALLOCATE( ztrdt(jpi,jpj,jpk), ztrds(jpi,jpj,jpk) ) 
     78         ENDIF 
     79 
     80         DO_3D_00_00( 1, jpkm1 ) 
     81            ztrdt(ji,jj,jk) = pts(ji,jj,jk,jp_tem,Kaa) 
     82            ztrds(ji,jj,jk) = pts(ji,jj,jk,jp_sal,Kaa) 
     83         END_3D 
    7284      ENDIF 
    7385      ! 
    … …  
    7991      ! JMM avoid negative salinities near river outlet ! Ugly fix 
    8092      ! JMM : restore negative salinities to small salinities: 
    81       WHERE( pts(:,:,:,jp_sal,Kaa) < 0._wp )   pts(:,:,:,jp_sal,Kaa) = 0.1_wp 
     93      WHERE( pts(A2D0,:,jp_sal,Kaa) < 0._wp )   pts(A2D0,:,jp_sal,Kaa) = 0.1_wp 
    8294!!gm 
    8395 
     96      ! TEMP: These changes not necessary after trd_tra is tiled 
    8497      IF( l_trdtra )   THEN                      ! save the vertical diffusive trends for further diagnostics 
    85          DO jk = 1, jpkm1 
    86             ztrdt(:,:,jk) = ( ( pts(:,:,jk,jp_tem,Kaa)*e3t(:,:,jk,Kaa) - pts(:,:,jk,jp_tem,Kbb)*e3t(:,:,jk,Kbb) ) & 
    87                &          / (e3t(:,:,jk,Kmm)*rDt) ) - ztrdt(:,:,jk) 
    88             ztrds(:,:,jk) = ( ( pts(:,:,jk,jp_sal,Kaa)*e3t(:,:,jk,Kaa) - pts(:,:,jk,jp_sal,Kbb)*e3t(:,:,jk,Kbb) ) & 
    89               &           / (e3t(:,:,jk,Kmm)*rDt) ) - ztrds(:,:,jk) 
    90          END DO 
     98         DO_3D_00_00( 1, jpkm1 ) 
     99            ztrdt(ji,jj,jk) = ( ( pts(ji,jj,jk,jp_tem,Kaa)*e3t(ji,jj,jk,Kaa) - pts(ji,jj,jk,jp_tem,Kbb)*e3t(ji,jj,jk,Kbb) ) & 
     100               &          / (e3t(ji,jj,jk,Kmm)*rDt) ) - ztrdt(ji,jj,jk) 
     101            ztrds(ji,jj,jk) = ( ( pts(ji,jj,jk,jp_sal,Kaa)*e3t(ji,jj,jk,Kaa) - pts(ji,jj,jk,jp_sal,Kbb)*e3t(ji,jj,jk,Kbb) ) & 
     102              &           / (e3t(ji,jj,jk,Kmm)*rDt) ) - ztrds(ji,jj,jk) 
     103         END_3D 
     104 
     105         IF( ntile == 0 .OR. ntile == nijtile )  THEN                ! Do only for the full domain 
    91106!!gm this should be moved in trdtra.F90 and done on all trends 
    92          CALL lbc_lnk_multi( 'trazdf', ztrdt, 'T', 1. , ztrds, 'T', 1. ) 
     107            CALL lbc_lnk_multi( 'trazdf', ztrdt, 'T', 1. , ztrds, 'T', 1. ) 
    93108!!gm 
    94          CALL trd_tra( kt, Kmm, Krhs, 'TRA', jp_tem, jptra_zdf, ztrdt ) 
    95          CALL trd_tra( kt, Kmm, Krhs, 'TRA', jp_sal, jptra_zdf, ztrds ) 
    96          DEALLOCATE( ztrdt , ztrds ) 
     109            IF( ln_tile ) CALL dom_tile( ntsi, ntsj, ntei, ntej, ktile = 0 )         ! Use full domain 
     110 
     111            ! TODO: TO BE TILED- trd_tra 
     112            CALL trd_tra( kt, Kmm, Krhs, 'TRA', jp_tem, jptra_zdf, ztrdt ) 
     113            CALL trd_tra( kt, Kmm, Krhs, 'TRA', jp_sal, jptra_zdf, ztrds ) 
     114            DEALLOCATE( ztrdt , ztrds ) 
     115 
     116            IF( ln_tile ) CALL dom_tile( ntsi, ntsj, ntei, ntej, ktile = nijtile )   ! Revert to tile domain 
     117         ENDIF 
    97118      ENDIF 
    98119      !                                          ! print mean trends (used for debugging) 
    99       IF(sn_cfctl%l_prtctl)   CALL prt_ctl( tab3d_1=pts(:,:,:,jp_tem,Kaa), clinfo1=' zdf  - Ta: ', mask1=tmask,               & 
    100          &                                  tab3d_2=pts(:,:,:,jp_sal,Kaa), clinfo2=       ' Sa: ', mask2=tmask, clinfo3='tra' ) 
     120      IF(sn_cfctl%l_prtctl)   CALL prt_ctl( tab3d_1=pts(:,:,:,jp_tem,Kaa), clinfo1=' zdf  - Ta: ', mask1=tmask, psum1=zsum1, & 
     121         &                                  tab3d_2=pts(:,:,:,jp_sal,Kaa), clinfo2=       ' Sa: ', mask2=tmask, psum2=zsum2, & 
     122         &                                  clinfo3='tra' ) 
    101123      ! 
    102124      IF( ln_timing )   CALL timing_stop('tra_zdf') 
    … …  
    135157      INTEGER  ::  ji, jj, jk, jn   ! dummy loop indices 
    136158      REAL(wp) ::  zrhs, zzwi, zzws ! local scalars 
    137       REAL(wp), DIMENSION(jpi,jpj,jpk) ::  zwi, zwt, zwd, zws 
     159      REAL(wp), DIMENSION(A2D,jpk) ::  zwi, zwt, zwd, zws 
    138160      !!--------------------------------------------------------------------- 
    139161      ! 
    … …  
    149171            ! 
    150172            ! vertical mixing coef.: avt for temperature, avs for salinity and passive tracers 
    151             IF( cdtype == 'TRA' .AND. jn == jp_tem ) THEN   ;   zwt(:,:,2:jpk) = avt(:,:,2:jpk) 
    152             ELSE                                            ;   zwt(:,:,2:jpk) = avs(:,:,2:jpk) 
     173            IF( cdtype == 'TRA' .AND. jn == jp_tem ) THEN 
     174               DO_3D_11_11( 2, jpk ) 
     175                  zwt(ji,jj,jk) = avt(ji,jj,jk) 
     176               END_3D 
     177            ELSE 
     178               DO_3D_11_11( 2, jpk ) 
     179                  zwt(ji,jj,jk) = avs(ji,jj,jk) 
     180               END_3D 
    153181            ENDIF 
    154182            zwt(:,:,1) = 0._wp 

  • NEMO/branches/UKMO/dev_r12866_HPC-02_Daley_Tiling_trial_extra_halo/src/OCE/TRA/zpshde.F90

    r12377 r13409  
    1717   USE oce             ! ocean: dynamics and tracers variables 
    1818   USE dom_oce         ! domain: ocean variables 
     19   USE domutl, ONLY : is_tile 
    1920   USE phycst          ! physical constants 
    2021   USE eosbn2          ! ocean equation of state 
    … …  
    3940CONTAINS 
    4041 
    41    SUBROUTINE zps_hde( kt, Kmm, kjpt, pta, pgtu, pgtv,   & 
    42       &                          prd, pgru, pgrv    ) 
     42   ! TODO: NOT TESTED- requires zps 
     43   SUBROUTINE zps_hde( kt, Kmm, kjpt, pta, pgtu, pgtv,  & 
     44      &                               prd, pgru, pgrv ) 
     45      !! 
     46      INTEGER                     , INTENT(in   )           ::  kt          ! ocean time-step index 
     47      INTEGER                     , INTENT(in   )           ::  Kmm         ! ocean time level index 
     48      INTEGER                     , INTENT(in   )           ::  kjpt        ! number of tracers 
     49      REAL(wp), DIMENSION(:,:,:,:), INTENT(in   )           ::  pta         ! 4D tracers fields 
     50      REAL(wp), DIMENSION(:,:,:)  , INTENT(  out)           ::  pgtu, pgtv  ! hor. grad. of ptra at u- & v-pts 
     51      REAL(wp), DIMENSION(:,:,:)  , INTENT(in   ), OPTIONAL ::  prd         ! 3D density anomaly fields 
     52      REAL(wp), DIMENSION(:,:)    , INTENT(  out), OPTIONAL ::  pgru, pgrv  ! hor. grad of prd at u- & v-pts (bottom) 
     53      ! 
     54      INTEGER :: itrd, itgr 
     55      !! 
     56      IF( PRESENT(prd)  ) THEN ; itrd = is_tile(prd)  ; ELSE ; itrd = 0 ; ENDIF 
     57      IF( PRESENT(pgru) ) THEN ; itgr = is_tile(pgru) ; ELSE ; itgr = 0 ; ENDIF 
     58 
     59      CALL zps_hde_t( kt, Kmm, kjpt, pta, is_tile(pta), pgtu, pgtv, is_tile(pgtu), & 
     60         &                           prd, itrd,         pgru, pgrv, itgr ) 
     61   END SUBROUTINE zps_hde 
     62 
     63 
     64   SUBROUTINE zps_hde_t( kt, Kmm, kjpt, pta, ktta, pgtu, pgtv, ktgt,   & 
     65      &                                 prd, ktrd, pgru, pgrv, ktgr ) 
    4366      !!---------------------------------------------------------------------- 
    4467      !!                     ***  ROUTINE zps_hde  *** 
    … …  
    84107      !!              - pgru, pgrv: horizontal gradient of rho (if present) at u- & v-points 
    85108      !!---------------------------------------------------------------------- 
    86       INTEGER                              , INTENT(in   )           ::  kt          ! ocean time-step index 
    87       INTEGER                              , INTENT(in   )           ::  Kmm         ! ocean time level index 
    88       INTEGER                              , INTENT(in   )           ::  kjpt        ! number of tracers 
    89       REAL(wp), DIMENSION(jpi,jpj,jpk,kjpt), INTENT(in   )           ::  pta         ! 4D tracers fields 
    90       REAL(wp), DIMENSION(jpi,jpj,    kjpt), INTENT(  out)           ::  pgtu, pgtv  ! hor. grad. of ptra at u- & v-pts  
    91       REAL(wp), DIMENSION(jpi,jpj,jpk     ), INTENT(in   ), OPTIONAL ::  prd         ! 3D density anomaly fields 
    92       REAL(wp), DIMENSION(jpi,jpj         ), INTENT(  out), OPTIONAL ::  pgru, pgrv  ! hor. grad of prd at u- & v-pts (bottom) 
     109      INTEGER                                , INTENT(in   )           ::  kt          ! ocean time-step index 
     110      INTEGER                                , INTENT(in   )           ::  Kmm         ! ocean time level index 
     111      INTEGER                                , INTENT(in   )           ::  kjpt        ! number of tracers 
     112      INTEGER                                , INTENT(in   )           ::  ktta, ktgt, ktrd, ktgr 
     113      REAL(wp), DIMENSION(T2D(ktta),jpk,kjpt), INTENT(in   )           ::  pta         ! 4D tracers fields 
     114      REAL(wp), DIMENSION(T2D(ktgt)    ,kjpt), INTENT(  out)           ::  pgtu, pgtv  ! hor. grad. of ptra at u- & v-pts 
     115      REAL(wp), DIMENSION(T2D(ktrd),jpk     ), INTENT(in   ), OPTIONAL ::  prd         ! 3D density anomaly fields 
     116      REAL(wp), DIMENSION(T2D(ktgr)         ), INTENT(  out), OPTIONAL ::  pgru, pgrv  ! hor. grad of prd at u- & v-pts (bottom) 
    93117      ! 
    94118      INTEGER  ::   ji, jj, jn                  ! Dummy loop indices 
    95119      INTEGER  ::   iku, ikv, ikum1, ikvm1      ! partial step level (ocean bottom level) at u- and v-points 
    96120      REAL(wp) ::   ze3wu, ze3wv, zmaxu, zmaxv  ! local scalars 
    97       REAL(wp), DIMENSION(jpi,jpj)      ::   zri, zrj, zhi, zhj   ! NB: 3rd dim=1 to use eos 
    98       REAL(wp), DIMENSION(jpi,jpj,kjpt) ::   zti, ztj             !  
     121      REAL(wp), DIMENSION(A2D)      ::   zri, zrj, zhi, zhj   ! NB: 3rd dim=1 to use eos 
     122      REAL(wp), DIMENSION(A2D,kjpt) ::   zti, ztj             ! 
    99123      !!---------------------------------------------------------------------- 
    100124      ! 
    … …  
    184208      IF( ln_timing )   CALL timing_stop( 'zps_hde') 
    185209      ! 
    186    END SUBROUTINE zps_hde 
    187  
    188  
     210   END SUBROUTINE zps_hde_t 
     211 
     212 
     213   ! TODO: NOT TESTED- requires zps 
    189214   SUBROUTINE zps_hde_isf( kt, Kmm, kjpt, pta, pgtu, pgtv, pgtui, pgtvi,  & 
    190       &                          prd, pgru, pgrv, pgrui, pgrvi ) 
     215      &                                   prd, pgru, pgrv, pgrui, pgrvi ) 
     216      !! 
     217      INTEGER                     , INTENT(in   )           ::  kt           ! ocean time-step index 
     218      INTEGER                     , INTENT(in   )           ::  Kmm          ! ocean time level index 
     219      INTEGER                     , INTENT(in   )           ::  kjpt         ! number of tracers 
     220      REAL(wp), DIMENSION(:,:,:,:), INTENT(in   )           ::  pta          ! 4D tracers fields 
     221      REAL(wp), DIMENSION(:,:,:)  , INTENT(  out)           ::  pgtu, pgtv   ! hor. grad. of ptra at u- & v-pts 
     222      REAL(wp), DIMENSION(:,:,:)  , INTENT(  out)           ::  pgtui, pgtvi ! hor. grad. of stra at u- & v-pts (ISF) 
     223      REAL(wp), DIMENSION(:,:,:)  , INTENT(in   ), OPTIONAL ::  prd          ! 3D density anomaly fields 
     224      REAL(wp), DIMENSION(:,:)    , INTENT(  out), OPTIONAL ::  pgru, pgrv   ! hor. grad of prd at u- & v-pts (bottom) 
     225      REAL(wp), DIMENSION(:,:)    , INTENT(  out), OPTIONAL ::  pgrui, pgrvi ! hor. grad of prd at u- & v-pts (top) 
     226      ! 
     227      INTEGER :: itrd, itgr, itgri 
     228      !! 
     229      IF( PRESENT(prd)   ) THEN ; itrd  = is_tile(prd)   ; ELSE ; itrd  = 0 ; ENDIF 
     230      IF( PRESENT(pgru)  ) THEN ; itgr  = is_tile(pgru)  ; ELSE ; itgr  = 0 ; ENDIF 
     231      IF( PRESENT(pgrui) ) THEN ; itgri = is_tile(pgrui) ; ELSE ; itgri = 0 ; ENDIF 
     232 
     233      CALL zps_hde_isf_t( kt, Kmm, kjpt, pta, is_tile(pta), pgtu, pgtv, is_tile(pgtu), pgtui, pgtvi, is_tile(pgtui),  & 
     234      &                                  prd, itrd,         pgru, pgrv, itgr,          pgrui, pgrvi, itgri ) 
     235   END SUBROUTINE zps_hde_isf 
     236 
     237 
     238   SUBROUTINE zps_hde_isf_t( kt, Kmm, kjpt, pta, ktta, pgtu, pgtv, ktgt, pgtui, pgtvi, ktgti,  & 
     239      &                                     prd, ktrd, pgru, pgrv, ktgr, pgrui, pgrvi, ktgri ) 
    191240      !!---------------------------------------------------------------------- 
    192241      !!                     ***  ROUTINE zps_hde_isf  *** 
    … …  
    235284      !!              - pgru, pgrv, pgrui, pgtvi: horizontal gradient of rho (if present) at u- & v-points 
    236285      !!---------------------------------------------------------------------- 
    237       INTEGER                              , INTENT(in   )           ::  kt           ! ocean time-step index 
    238       INTEGER                              , INTENT(in   )           ::  Kmm          ! ocean time level index 
    239       INTEGER                              , INTENT(in   )           ::  kjpt         ! number of tracers 
    240       REAL(wp), DIMENSION(jpi,jpj,jpk,kjpt), INTENT(in   )           ::  pta          ! 4D tracers fields 
    241       REAL(wp), DIMENSION(jpi,jpj,    kjpt), INTENT(  out)           ::  pgtu, pgtv   ! hor. grad. of ptra at u- & v-pts  
    242       REAL(wp), DIMENSION(jpi,jpj,    kjpt), INTENT(  out)           ::  pgtui, pgtvi ! hor. grad. of stra at u- & v-pts (ISF) 
    243       REAL(wp), DIMENSION(jpi,jpj,jpk     ), INTENT(in   ), OPTIONAL ::  prd          ! 3D density anomaly fields 
    244       REAL(wp), DIMENSION(jpi,jpj         ), INTENT(  out), OPTIONAL ::  pgru, pgrv   ! hor. grad of prd at u- & v-pts (bottom) 
    245       REAL(wp), DIMENSION(jpi,jpj         ), INTENT(  out), OPTIONAL ::  pgrui, pgrvi ! hor. grad of prd at u- & v-pts (top) 
     286      INTEGER                                , INTENT(in   )           ::  kt           ! ocean time-step index 
     287      INTEGER                                , INTENT(in   )           ::  Kmm          ! ocean time level index 
     288      INTEGER                                , INTENT(in   )           ::  kjpt         ! number of tracers 
     289      INTEGER                                , INTENT(in   )           ::  ktta, ktgt, ktgti, ktrd, ktgr, ktgri 
     290      REAL(wp), DIMENSION(T2D(ktta),jpk,kjpt), INTENT(in   )           ::  pta          ! 4D tracers fields 
     291      REAL(wp), DIMENSION(T2D(ktgt)    ,kjpt), INTENT(  out)           ::  pgtu, pgtv   ! hor. grad. of ptra at u- & v-pts 
     292      REAL(wp), DIMENSION(T2D(ktgti)   ,kjpt), INTENT(  out)           ::  pgtui, pgtvi ! hor. grad. of stra at u- & v-pts (ISF) 
     293      REAL(wp), DIMENSION(T2D(ktrd),jpk     ), INTENT(in   ), OPTIONAL ::  prd          ! 3D density anomaly fields 
     294      REAL(wp), DIMENSION(T2D(ktgr)         ), INTENT(  out), OPTIONAL ::  pgru, pgrv   ! hor. grad of prd at u- & v-pts (bottom) 
     295      REAL(wp), DIMENSION(T2D(ktgri)        ), INTENT(  out), OPTIONAL ::  pgrui, pgrvi ! hor. grad of prd at u- & v-pts (top) 
    246296      ! 
    247297      INTEGER  ::   ji, jj, jn      ! Dummy loop indices 
    248298      INTEGER  ::   iku, ikv, ikum1, ikvm1,ikup1, ikvp1   ! partial step level (ocean bottom level) at u- and v-points 
    249299      REAL(wp) ::  ze3wu, ze3wv, zmaxu, zmaxv             ! temporary scalars 
    250       REAL(wp), DIMENSION(jpi,jpj)      ::  zri, zrj, zhi, zhj   ! NB: 3rd dim=1 to use eos 
    251       REAL(wp), DIMENSION(jpi,jpj,kjpt) ::  zti, ztj             !  
     300      REAL(wp), DIMENSION(A2D)      ::  zri, zrj, zhi, zhj   ! NB: 3rd dim=1 to use eos 
     301      REAL(wp), DIMENSION(A2D,kjpt) ::  zti, ztj             ! 
    252302      !!---------------------------------------------------------------------- 
    253303      ! 
    … …  
    439489      IF( ln_timing )   CALL timing_stop( 'zps_hde_isf') 
    440490      ! 
    441    END SUBROUTINE zps_hde_isf 
     491   END SUBROUTINE zps_hde_isf_t 
    442492 
    443493   !!====================================================================== 

  • NEMO/branches/UKMO/dev_r12866_HPC-02_Daley_Tiling_trial_extra_halo/src/OCE/ZDF/zdfosm.F90

    r12738 r13409  
    4444                      ! uses ww from previous time step (which is now wb) to calculate hbl 
    4545   USE dom_oce        ! ocean space and time domain 
     46   ! TEMP: This change not necessary after trd_tra is tiled 
     47   USE domain, ONLY : dom_tile 
    4648   USE zdf_oce        ! ocean vertical physics 
    4749   USE sbc_oce        ! surface boundary condition: ocean 
    … …  
    14831485     !!----------------------------------------------------------------------------- 
    14841486     IF( TRIM(cdrw) == 'WRITE') THEN     !* Write hbli into the restart file, then return 
     1487        IF( ntile /= 0 .AND. ntile /= nijtile ) RETURN        ! Do only on the last tile 
     1488 
    14851489        IF(lwp) WRITE(numout,*) '---- osm-rst ----' 
    14861490         CALL iom_rstput( kt, nitrst, numrow, 'wn'     , ww  , ldxios = lwxios ) 
    … …  
    15281532      !! ** Method  :   ??? 
    15291533      !!---------------------------------------------------------------------- 
    1530       REAL(wp), DIMENSION(:,:,:), ALLOCATABLE ::   ztrdt, ztrds   ! 3D workspace 
     1534      ! TEMP: This change not necessary after trd_tra is tiled 
     1535      REAL(wp), DIMENSION(:,:,:), ALLOCATABLE, SAVE ::   ztrdt, ztrds   ! 3D workspace 
    15311536      !!---------------------------------------------------------------------- 
    15321537      INTEGER                                  , INTENT(in)    :: kt        ! time step index 
    … …  
    15341539      REAL(wp), DIMENSION(jpi,jpj,jpk,jpts,jpt), INTENT(inout) :: pts       ! active tracers and RHS of tracer equation 
    15351540      ! 
     1541      REAL(wp), SAVE :: zsum1, zsum2 
    15361542      INTEGER :: ji, jj, jk 
    15371543      ! 
    15381544      IF( kt == nit000 ) THEN 
    1539          IF(lwp) WRITE(numout,*) 
    1540          IF(lwp) WRITE(numout,*) 'tra_osm : OSM non-local tracer fluxes' 
    1541          IF(lwp) WRITE(numout,*) '~~~~~~~   ' 
     1545         IF( ntile == 0 .OR. ntile == 1 ) THEN                    ! Do only on the first tile 
     1546            IF(lwp) WRITE(numout,*) 
     1547            IF(lwp) WRITE(numout,*) 'tra_osm : OSM non-local tracer fluxes' 
     1548            IF(lwp) WRITE(numout,*) '~~~~~~~   ' 
     1549         ENDIF 
    15421550      ENDIF 
    15431551 
    15441552      IF( l_trdtra )   THEN                    !* Save ta and sa trends 
    1545          ALLOCATE( ztrdt(jpi,jpj,jpk) )   ;    ztrdt(:,:,:) = pts(:,:,:,jp_tem,Krhs) 
    1546          ALLOCATE( ztrds(jpi,jpj,jpk) )   ;    ztrds(:,:,:) = pts(:,:,:,jp_sal,Krhs) 
     1553         IF( ntile == 0 .OR. ntile == 1 )  THEN                       ! Do only on the first tile 
     1554            ! TEMP: This can be A2D after trd_tra is tiled 
     1555            ALLOCATE( ztrdt(jpi,jpj,jpk), ztrds(jpi,jpj,jpk) ) 
     1556         ENDIF 
     1557 
     1558         DO_3D_00_00( 1, jpkm1 ) 
     1559            ztrdt(ji,jj,jk) = pts(ji,jj,jk,jp_tem,Krhs) 
     1560            ztrds(ji,jj,jk) = pts(ji,jj,jk,jp_sal,Krhs) 
     1561         END_3D 
    15471562      ENDIF 
    15481563 
    … …  
    15571572      END_3D 
    15581573 
    1559  
     1574      ! TEMP: These changes not necessary after trd_tra is tiled 
    15601575      ! save the non-local tracer flux trends for diagnostic 
    15611576      IF( l_trdtra )   THEN 
    1562          ztrdt(:,:,:) = pts(:,:,:,jp_tem,Krhs) - ztrdt(:,:,:) 
    1563          ztrds(:,:,:) = pts(:,:,:,jp_sal,Krhs) - ztrds(:,:,:) 
     1577         DO_3D_00_00( 1, jpkm1 ) 
     1578            ztrdt(ji,jj,jk) = pts(ji,jj,jk,jp_tem,Krhs) - ztrdt(ji,jj,jk) 
     1579            ztrds(ji,jj,jk) = pts(ji,jj,jk,jp_sal,Krhs) - ztrds(ji,jj,jk) 
     1580         END_3D 
     1581 
     1582         IF( ntile == 0 .OR. ntile == nijtile )  THEN                ! Do only for the full domain 
     1583            IF( ln_tile ) CALL dom_tile( ntsi, ntsj, ntei, ntej, ktile = 0 )         ! Use full domain 
     1584 
    15641585!!bug gm jpttdzdf ==> jpttosm 
    1565          CALL trd_tra( kt, Kmm, Krhs, 'TRA', jp_tem, jptra_zdf, ztrdt ) 
    1566          CALL trd_tra( kt, Kmm, Krhs, 'TRA', jp_sal, jptra_zdf, ztrds ) 
    1567          DEALLOCATE( ztrdt )      ;     DEALLOCATE( ztrds ) 
     1586            ! TODO: TO BE TILED- trd_tra 
     1587            CALL trd_tra( kt, Kmm, Krhs, 'TRA', jp_tem, jptra_zdf, ztrdt ) 
     1588            CALL trd_tra( kt, Kmm, Krhs, 'TRA', jp_sal, jptra_zdf, ztrds ) 
     1589            DEALLOCATE( ztrdt )      ;     DEALLOCATE( ztrds ) 
     1590 
     1591            IF( ln_tile ) CALL dom_tile( ntsi, ntsj, ntei, ntej, ktile = nijtile )   ! Revert to tile domain 
     1592         ENDIF 
    15681593      ENDIF 
    15691594 
    15701595      IF(sn_cfctl%l_prtctl) THEN 
    1571          CALL prt_ctl( tab3d_1=pts(:,:,:,jp_tem,Krhs), clinfo1=' osm  - Ta: ', mask1=tmask,   & 
    1572          &             tab3d_2=pts(:,:,:,jp_sal,Krhs), clinfo2=       ' Sa: ', mask2=tmask, clinfo3='tra' ) 
     1596         CALL prt_ctl( tab3d_1=pts(:,:,:,jp_tem,Krhs), clinfo1=' osm  - Ta: ', mask1=tmask, psum1=zsum1, & 
     1597            &          tab3d_2=pts(:,:,:,jp_sal,Krhs), clinfo2=       ' Sa: ', mask2=tmask, psum2=zsum2, & 
     1598            &          clinfo3='tra' ) 
    15731599      ENDIF 
    15741600      ! 

  • NEMO/branches/UKMO/dev_r12866_HPC-02_Daley_Tiling_trial_extra_halo/src/OCE/do_loop_substitute.h90

    r13054 r13409  
    102102 
    103103! 2D loops with 1 
    104 #define Ntis0      ntsi 
    105 #define Ntjs0      ntsj 
    106 #define Ntis1      Ntis0 - 1 
    107 #define Ntjs1      Ntjs0 - 1 
    108 #define Ntis2      Ntis0 - nn_hls 
    109 #define Ntjs2      Ntjs0 - nn_hls 
    110 #define Ntie0      ntei 
    111 #define Ntje0      ntej 
    112 #define Ntie1      Ntie0 + 1 
    113 #define Ntje1      Ntje0 + 1 
    114 #define Ntie2      Ntie0 + nn_hls 
    115 #define Ntje2      Ntje0 + nn_hls 
     104#define Ntis0      ntsi%h0 
     105#define Ntjs0      ntsj%h0 
     106#define Ntis1      ntsi%h1 
     107#define Ntjs1      ntsj%h1 
     108#define Ntis2      ntsi%h2 
     109#define Ntjs2      ntsj%h2 
     110#define Ntie0      ntei%h0 
     111#define Ntje0      ntej%h0 
     112#define Ntie1      ntei%h1 
     113#define Ntje1      ntej%h1 
     114#define Ntie2      ntei%h2 
     115#define Ntje2      ntej%h2 
     116#define A1Di0      Ntis0:Ntie0 
     117#define A1Dj0      Ntjs0:Ntje0 
     118#define A2D0       A1Di0,A1Dj0 
    116119#define A1Di       Ntis2:Ntie2 
    117120#define A1Dj       Ntjs2:Ntje2 
    118121#define A2D        A1Di,A1Dj 
     122#define T1Di(a)    (Ntis2-1)*a+1:(Ntie2-jpi)*a+jpi 
     123#define T1Dj(a)    (Ntjs2-1)*a+1:(Ntje2-jpj)*a+jpj 
     124#define T2D(a)     T1Di(a),T1Dj(a) 
    119125 
    120126#define DO_2D_00_00   DO jj = Ntjs0, Ntje0   ;   DO ji = Ntis0, Ntie0 

  • NEMO/branches/UKMO/dev_r12866_HPC-02_Daley_Tiling_trial_extra_halo/src/OCE/par_oce.F90

    r12979 r13409  
    6363 
    6464   ! Domain tiling 
     65   TYPE tiledom 
     66      INTEGER :: h0, h1, h2 
     67   END TYPE tiledom 
     68 
    6569   INTEGER, PUBLIC ::   nijtile    !: number of tiles in total 
    6670   INTEGER, PUBLIC ::   ntile      !: current tile number 
    67    INTEGER, PUBLIC ::   ntsi       !: start of internal part of tile domain 
    68    INTEGER, PUBLIC ::   ntsj       ! 
    69    INTEGER, PUBLIC ::   ntei       !: end of internal part of tile domain 
    70    INTEGER, PUBLIC ::   ntej       ! 
     71   TYPE(tiledom), PUBLIC ::   ntsi       !: start of internal part of tile domain 
     72   TYPE(tiledom), PUBLIC ::   ntsj       ! 
     73   TYPE(tiledom), PUBLIC ::   ntei       !: end of internal part of tile domain 
     74   TYPE(tiledom), PUBLIC ::   ntej       ! 
    7175 
    7276   !!--------------------------------------------------------------------- 

  • NEMO/branches/UKMO/dev_r12866_HPC-02_Daley_Tiling_trial_extra_halo/src/OCE/step.F90

    r12979 r13409  
    5151   INTEGER, PUBLIC :: Nbb, Nnn, Naa, Nrhs          !! used by nemo_init 
    5252 
     53   !! * Substitutions 
     54#  include "do_loop_substitute.h90" 
    5355   !!---------------------------------------------------------------------- 
    5456   !! NEMO/OCE 4.0 , NEMO Consortium (2018) 
    … …  
    242244      ! Active tracers                               
    243245      !<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<< 
    244                          ts(:,:,:,:,Nrhs) = 0._wp         ! set tracer trends to zero 
    245  
    246       IF(  lk_asminc .AND. ln_asmiau .AND. & 
    247          & ln_trainc )   CALL tra_asm_inc( kstp, Nbb, Nnn, ts, Nrhs )  ! apply tracer assimilation increment 
    248                          CALL tra_sbc    ( kstp,      Nnn, ts, Nrhs )  ! surface boundary condition 
    249       IF( ln_traqsr  )   CALL tra_qsr    ( kstp,      Nnn, ts, Nrhs )  ! penetrative solar radiation qsr 
    250       IF( ln_isf     )   CALL tra_isf    ( kstp,      Nnn, ts, Nrhs )  ! ice shelf heat flux 
    251       IF( ln_trabbc  )   CALL tra_bbc    ( kstp,      Nnn, ts, Nrhs )  ! bottom heat flux 
    252       IF( ln_trabbl  )   CALL tra_bbl    ( kstp, Nbb, Nnn, ts, Nrhs )  ! advective (and/or diffusive) bottom boundary layer scheme 
    253       IF( ln_tradmp  )   CALL tra_dmp    ( kstp, Nbb, Nnn, ts, Nrhs )  ! internal damping trends 
    254       IF( ln_bdy     )   CALL bdy_tra_dmp( kstp, Nbb,      ts, Nrhs )  ! bdy damping trends 
    255 #if defined key_agrif 
    256       IF(.NOT. Agrif_Root())  &  
    257                &         CALL Agrif_Sponge_tra        ! tracers sponge 
    258 #endif 
    259                          CALL tra_adv    ( kstp, Nbb, Nnn, ts, Nrhs )  ! hor. + vert. advection ==> RHS 
    260       IF( ln_zdfosm  )   CALL tra_osm    ( kstp,      Nnn, ts, Nrhs )  ! OSMOSIS non-local tracer fluxes ==> RHS 
    261       IF( lrst_oce .AND. ln_zdfosm ) & 
    262            &             CALL osm_rst    ( kstp,      Nnn, 'WRITE'  )  ! write OSMOSIS outputs + ww (so must do here) to restarts 
    263  
    264246      ! Loop over tile domains 
    265247      DO jtile = 1, nijtile 
    266248         IF( ln_tile ) CALL dom_tile( ntsi, ntsj, ntei, ntej, ktile = jtile ) 
    267                          CALL tra_ldf    ( kstp, Nbb, Nnn, ts, Nrhs )  ! lateral mixing 
     249 
     250         DO_3D_00_00( 1, jpk ) 
     251            ts(ji,jj,jk,:,Nrhs) = 0._wp                                         ! set tracer trends to zero 
     252         END_3D 
     253 
     254         IF(  lk_asminc .AND. ln_asmiau .AND. & 
     255            & ln_trainc )   CALL tra_asm_inc( kstp, Nbb, Nnn, ts, Nrhs )        ! apply tracer assimilation increment 
     256                            CALL tra_sbc    ( kstp,      Nnn, ts, Nrhs )        ! surface boundary condition 
     257         IF( ln_traqsr  )   CALL tra_qsr    ( kstp,      Nnn, ts, Nrhs )        ! penetrative solar radiation qsr 
     258         IF( ln_isf     )   CALL tra_isf    ( kstp,      Nnn, ts, Nrhs )        ! ice shelf heat flux 
     259         IF( ln_trabbc  )   CALL tra_bbc    ( kstp,      Nnn, ts, Nrhs )        ! bottom heat flux 
     260         IF( ln_trabbl  )   CALL tra_bbl    ( kstp, Nbb, Nnn, ts, Nrhs )        ! advective (and/or diffusive) bottom boundary layer scheme 
     261         IF( ln_tradmp  )   CALL tra_dmp    ( kstp, Nbb, Nnn, ts, Nrhs )        ! internal damping trends 
     262         IF( ln_bdy     )   CALL bdy_tra_dmp( kstp, Nbb,      ts, Nrhs )        ! bdy damping trends 
     263#if defined key_agrif 
     264         ! TODO: TO BE TILED- I don't know what this does, or whether it can just be run for ntile == nijtile 
     265         IF(.NOT. Agrif_Root())  & 
     266              &             CALL Agrif_Sponge_tra                               ! tracers sponge 
     267#endif 
    268268      END DO 
     269 
     270      ! TEMP: Loop over tile domains (seperate due to tra_adv workarounds for tiling) 
     271      DO jtile = 1, nijtile 
     272         IF( ln_tile ) CALL dom_tile( ntsi, ntsj, ntei, ntej, ktile = jtile ) 
     273                            CALL tra_adv    ( kstp, Nbb, Nnn, ts, Nrhs )        ! hor. + vert. advection ==> RHS 
     274         IF( ln_zdfosm  )   CALL tra_osm    ( kstp,      Nnn, ts, Nrhs )        ! OSMOSIS non-local tracer fluxes ==> RHS 
     275         IF( lrst_oce .AND. ln_zdfosm ) & 
     276              &             CALL osm_rst    ( kstp,      Nnn, 'WRITE'  )        ! write OSMOSIS outputs + ww (so must do here) to restarts 
     277                            CALL tra_ldf    ( kstp, Nbb, Nnn, ts, Nrhs )        ! lateral mixing 
     278                            CALL tra_zdf    ( kstp, Nbb, Nnn, Nrhs, ts, Naa  )  ! vertical mixing and after tracer fields 
     279         IF( ln_zdfnpc  )   CALL tra_npc    ( kstp,      Nnn, Nrhs, ts, Naa  )  ! update after fields by non-penetrative convection 
     280      END DO 
     281 
    269282      IF( ln_tile ) CALL dom_tile( ntsi, ntsj, ntei, ntej, ktile = 0 ) ! Revert to tile over full domain 
    270                          CALL tra_zdf    ( kstp, Nbb, Nnn, Nrhs, ts, Naa  )  ! vertical mixing and after tracer fields 
    271       IF( ln_zdfnpc  )   CALL tra_npc    ( kstp,      Nnn, Nrhs, ts, Naa  )  ! update after fields by non-penetrative convection 
    272283 
    273284      !>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>> 

  • NEMO/branches/UKMO/dev_r12866_HPC-02_Daley_Tiling_trial_extra_halo/src/OCE/timing.F90

    r12942 r13409  
    9393      !!---------------------------------------------------------------------- 
    9494      CHARACTER(len=*), INTENT(in) :: cdinfo 
    95       ! TODO: TO BE TILED 
    96       IF( ntile /= 0 .AND. ntile /= 1 ) RETURN 
    9795      ! 
    9896       IF(ASSOCIATED(s_timer) ) s_timer_old => s_timer 
    … …  
    111109 
    112110      s_timer%l_tdone = .FALSE. 
    113       s_timer%niter = s_timer%niter + 1 
     111      IF( ntile == 0 .OR. ntile == 1 ) s_timer%niter = s_timer%niter + 1      ! All tiles count as one iteration 
    114112      s_timer%t_cpu = 0. 
    115113      s_timer%t_clock = 0. 
    … …  
    140138      INTEGER  :: ifinal_count, iperiods     
    141139      REAL(wp) :: zcpu_end, zmpitime,zcpu_raw,zclock_raw 
    142       ! TODO: TO BE TILED 
    143       IF( ntile /= 0 .AND. ntile /= nijtile ) RETURN 
    144140      ! 
    145141      s_wrk => NULL() 
Note: See TracChangeset for help on using the changeset viewer.