Changeset 3923 for trunk/LMDZ.MARS/libf/phymars/writediagfi.F

Timestamp:

Oct 6, 2025, 11:24:25 AM (3 months ago)

Author:

jmauxion

Message:

Mars PCM:
Adding a slow_diagfi flag to the run.def file for 1D model only. When False, the netcdf
file is opened/closed once, thus saving significant computing time. When true,
the opening frequency is at output frequency (required in debug mode).
JM

File:

• trunk/LMDZ.MARS/libf/phymars/writediagfi.F (modified) (8 diffs)

trunk/LMDZ.MARS/libf/phymars/writediagfi.F

@@ -41 +41 @@
       use surfdat_h, only: phisfi
       use geometry_mod, only: cell_area
-      use time_phylmdz_mod, only: steps_per_sol, outputs_per_sol
-      use time_phylmdz_mod, only: day_ini
+      use time_phylmdz_mod, only: steps_per_sol, outputs_per_sol,
+     &                            day_end, day_ini, slow_diagfi
       USE mod_phys_lmdz_para, only : is_parallel, is_mpi_root,
      &                               is_master, gather
@@ -79 +79 @@
 
       integer,save :: zitau=0
+      integer,save :: lastzitau=0
       character(len=27),save :: firstnom='1234567890'
+      character(len=27),save :: prevnom='1234567890'
+      character(len=27),save :: lastnom='1234567890'
 !$OMP THREADPRIVATE(zitau,firstnom)
 
@@ -86 +89 @@
 !$OMP THREADPRIVATE(ntime)
       integer :: idim,varid
-      integer :: nid
+      integer, save :: nid
       character(len=*),parameter :: fichnom="diagfi.nc"
       integer, dimension(4) :: id
@@ -130 +133 @@
          ! Compute the output rate
          isample=steps_per_sol/outputs_per_sol
-
+         lastzitau=(day_end-day_ini)*steps_per_sol-1
 !$OMP MASTER
   !      Open diagfi.def definition file if there is one:
@@ -254 +257 @@
       endif ! if (firstnom.eq.'1234567890')
 
-! Increment time index 'zitau' if it is the "fist call" (at given time level)
+      ! Find lastnom
+      if (lastnom.eq.'1234567890') then
+         if (nom.eq.firstnom) then
+            if (prevnom.ne.'1234567890') then
+               lastnom=prevnom
+            endif
+         endif
+         prevnom=nom
+      endif
+
+! Increment time index 'zitau' if it is the "first call" (at given time level)
 ! to writediagfi
 !------------------------------------------------------------------------
@@ -264 +277 @@
 ! Write the variables to output file if it's time to do so
 !--------------------------------------------------------
-
       if ( MOD(zitau+1,isample) .eq.0.) then
       
@@ -273 +285 @@
         if (is_master) then
           ! only the master is required to do this
-
-          ! Time to write data, open NETCDF file
-          ierr=NF_OPEN(fichnom,NF_WRITE,nid)
-
-          if (nom.eq.firstnom) then
+          
+          ! 1D and slow_diagfi=.false. => open/close once
+          if ((klon_glo.eq.1).and.(.not.slow_diagfi)) then
+            ! if the very first time to write, open
+            if ((nom.eq.firstnom).and.((zitau+1)/isample.eq.1)) then
+              write(*,*) "Open NETCDF file for firstnom=", firstnom, " and zitau=", zitau
+              ierr=NF_OPEN(fichnom,NF_WRITE,nid) ! open once in all simu
+            endif
+          else ! 3D or slow_diagfi=.true. => open/close at output frequency
+            ierr=NF_OPEN(fichnom,NF_WRITE,nid)            
+          endif
+          
+          if(nom.eq.firstnom) then
             ! We have identified a "first call" (at given date)
             ntime=ntime+1 ! increment # of stored time steps
@@ -613 +633 @@
         endif ! of if (dim.eq.3) elseif(dim.eq.2)...
 
+        ! Only the master do it
         if (is_master) then
-          ! Close until next variable to dump or next dump iteration
-          ierr= NF_CLOSE(nid)
+          ! 1D and slow_diagfi=.false. => open/close once
+          if ((klon_glo.eq.1).and.(.not.slow_diagfi)) then
+            ! if the very last time to write, close
+            if ((nom.eq.lastnom).and.(zitau.eq.lastzitau)) then
+              write(*,*) "Close NETCDF file for lastnom=",lastnom, "and zitau=", zitau
+              ierr = NF_CLOSE(nid) ! close once in all simu
+            endif
+          else ! 3D or slow_diagfi=.true. => open/close at output frequency
+            ierr = NF_CLOSE(nid)
+          endif
         endif