Changeset 3078 for trunk/LMDZ.COMMON/libf

Timestamp:

Oct 10, 2023, 11:41:27 AM (15 months ago)

Author:

emillour

Message:

Common physics/dynamics:
Add, for lon-lat outputs, a cell_area_for_lonlat_outputs()
field where polar mesh areas are adapted to match those
of the dynamics grid (where polar mesh points are replicated
along longitudes).

Mars PCM:
Enable outputing cell area with XIOS and make sure the polar grid point
is correctly handled (i.e. on lon-lat grid outputs polar mesh area must
be adjusted to account for the replicated polar meshes).

EM

Location:

trunk/LMDZ.COMMON/libf

Files:

• dynphy_lonlat/inigeomphy_mod.F90 (modified) (9 diffs)
• phy_common/geometry_mod.F90 (modified) (2 diffs)

trunk/LMDZ.COMMON/libf/dynphy_lonlat/inigeomphy_mod.F90

@@ -12 +12 @@
                                 klon_omp_end, & ! end index of local omp subgrid
                                 klon_mpi_begin ! start indes of columns (on local mpi grid)
-  USE geometry_mod, ONLY : init_geometry
+  USE geometry_mod, ONLY : init_geometry, init_geometry_cell_area_for_outputs
   USE physics_distribution_mod, ONLY : init_physics_distribution
   USE regular_lonlat_mod, ONLY : init_regular_lonlat, &
@@ -61 +61 @@
   REAL,ALLOCATABLE :: cvfi_glo(:)
   REAL,ALLOCATABLE :: airefi_glo(:)
+  REAL,ALLOCATABLE :: aire_glo(:)
   REAL,ALLOCATABLE :: boundslonfi_glo(:,:)
   REAL,ALLOCATABLE :: boundslatfi_glo(:,:)
@@ -70 +71 @@
   REAL,ALLOCATABLE,SAVE :: cvfi(:)
   REAL,ALLOCATABLE,SAVE :: airefi(:)
+  REAL,ALLOCATABLE,SAVE :: airefi_for_outputs(:)
   REAL,ALLOCATABLE,SAVE :: boundslonfi(:,:)
   REAL,ALLOCATABLE,SAVE :: boundslatfi(:,:)
   INTEGER,ALLOCATABLE,SAVE :: ind_cell_glo_fi(:)
-!$OMP THREADPRIVATE (latfi,lonfi,cufi,cvfi,airefi,boundslonfi,boundslatfi,ind_cell_glo_fi)
+!$OMP THREADPRIVATE (latfi,lonfi,cufi,cvfi,airefi,airefi_for_outputs)
+!$OMP THREADPRIVATE (boundslonfi,boundslatfi,ind_cell_glo_fi)
 
   ! Initialize Physics distibution and parameters and interface with dynamics
@@ -114 +117 @@
   ALLOCATE(cufi_glo(klon_glo),cvfi_glo(klon_glo))
   ALLOCATE(airefi_glo(klon_glo))
+  ALLOCATE(aire_glo(klon_glo))
   ALLOCATE(boundslonfi_glo(klon_glo,4))
   ALLOCATE(boundslatfi_glo(klon_glo,4))
@@ -173 +177 @@
     airefi_glo(1)=sum(aire(1:iim,1))
     airefi_glo(klon_glo)=sum(aire(1:iim,jjm+1))
+    ! but we also want to store mesh area with poles as on dyn grid
+    ! (for lon-lat outputs)
+    CALL gr_dyn_fi(1,iim+1,jjm+1,klon_glo,aire,aire_glo)
 
     ! Sanity check: do total planet area match between physics and dynamics?
@@ -194 +201 @@
     cvfi_glo(1)=cv(1)
     airefi_glo(1)=aire(1,1)
+    aire_glo(1)=aire(1,1)
     boundslonfi_glo(1,north_east)=rlonu(1)
     boundslatfi_glo(1,north_east)=PI/2
@@ -208 +216 @@
   ALLOCATE(latfi(klon_omp),lonfi(klon_omp),cufi(klon_omp),cvfi(klon_omp))
   ALLOCATE(airefi(klon_omp))
+  ALLOCATE(airefi_for_outputs(klon_omp))
   ALLOCATE(boundslonfi(klon_omp,4))
   ALLOCATE(boundslatfi(klon_omp,4))
@@ -215 +224 @@
   offset = klon_mpi_begin - 1
   airefi(1:klon_omp) = airefi_glo(offset+klon_omp_begin:offset+klon_omp_end)
+  airefi_for_outputs(1:klon_omp) = &
+                         aire_glo(offset+klon_omp_begin:offset+klon_omp_end)
   cufi(1:klon_omp) = cufi_glo(offset+klon_omp_begin:offset+klon_omp_end)
   cvfi(1:klon_omp) = cvfi_glo(offset+klon_omp_begin:offset+klon_omp_end)
@@ -227 +238 @@
                      airefi,ind_cell_glo_fi,cufi,cvfi)
 
+  ! copy over lon-lat mesh area for outputs (with polar "correction")
+  CALL init_geometry_cell_area_for_outputs(klon_omp,airefi_for_outputs)
+
   ! copy over preff , ap(), bp(), etc 
   CALL init_vertical_layers(nlayer,preff,scaleheight, &

trunk/LMDZ.COMMON/libf/phy_common/geometry_mod.F90

@@ -27 +27 @@
 !$OMP THREADPRIVATE(dy)
 
-  REAL,SAVE,ALLOCATABLE :: cell_area(:)      ! area of the cell
+  REAL,SAVE,ALLOCATABLE :: cell_area(:) ! area of the cell (m2)
 !$OMP THREADPRIVATE(cell_area)
+
+  REAL,SAVE,ALLOCATABLE :: cell_area_for_lonlat_outputs(:) 
+                           ! for lon-lat outputs only: area of the cell
+                           ! but with polar values as in dyn grid
+                           ! (with replicated polar grid point)
+!$OMP THREADPRIVATE(cell_area_for_lonlat_outputs)
 
   INTEGER,SAVE,ALLOCATABLE :: ind_cell_glo(:)      ! global index of a local cell
@@ -76 +82 @@
 
 
+  SUBROUTINE init_geometry_cell_area_for_outputs(klon, &
+                                                 cell_area_for_lonlat_outputs_)
+  IMPLICIT NONE
+  INTEGER,INTENT(IN) :: klon ! number of columns for this MPI/OpenMP domain
+  REAL,INTENT(IN) :: cell_area_for_lonlat_outputs_(klon) ! tweaked lon-lat mesh
+                     ! cell areas where polar values are as on dyn lon-lat grid
+                     ! for outputs
+  
+  ALLOCATE(cell_area_for_lonlat_outputs(klon))
+  cell_area_for_lonlat_outputs(1:klon)=cell_area_for_lonlat_outputs_(1:klon)
+  
+  END SUBROUTINE init_geometry_cell_area_for_outputs
+
 END MODULE geometry_mod