source: LMDZ6/trunk/libf/dynphy_lonlat/inigeomphy_mod.f90 @ 5281

Last change on this file since 5281 was 5268, checked in by abarral, 7 weeks ago

.f90 <-> .F90 depending on cpp key use

File size: 9.6 KB
RevLine 
[2588]1!
2! $Id: $
3!
4MODULE inigeomphy_mod
5
6CONTAINS
7
8SUBROUTINE inigeomphy(iim,jjm,nlayer, &
9                     nbp, communicator, &
10                     rlatu,rlatv,rlonu,rlonv,aire,cu,cv)
11  USE mod_grid_phy_lmdz, ONLY: klon_glo,  & ! number of atmospheric columns (on full grid)
12                               regular_lonlat, &  ! regular longitude-latitude grid type
13                               nbp_lon, nbp_lat, nbp_lev
14  USE mod_phys_lmdz_para, ONLY: klon_omp, & ! number of columns (on local omp grid)
15                                klon_omp_begin, & ! start index of local omp subgrid
16                                klon_omp_end, & ! end index of local omp subgrid
17                                klon_mpi_begin ! start indes of columns (on local mpi grid)
18  USE geometry_mod, ONLY : init_geometry
19  USE physics_distribution_mod, ONLY : init_physics_distribution
20  USE regular_lonlat_mod, ONLY : init_regular_lonlat, &
21                                 east, west, north, south, &
22                                 north_east, north_west, &
23                                 south_west, south_east
24  USE mod_interface_dyn_phys, ONLY :  init_interface_dyn_phys
25  USE nrtype, ONLY: pi
[2786]26  USE comvert_mod, ONLY: preff, ap, bp, aps, bps, presnivs, &
[4228]27                         scaleheight, pseudoalt, presinter
[2786]28  USE vertical_layers_mod, ONLY: init_vertical_layers
[2588]29  IMPLICIT NONE
30
31  ! =======================================================================
32  ! Initialisation of the physical constants and some positional and
33  ! geometrical arrays for the physics
34  ! =======================================================================
35
36  include "iniprint.h"
37
38  INTEGER, INTENT (IN) :: nlayer ! number of atmospheric layers
39  INTEGER, INTENT (IN) :: iim ! number of atmospheric columns along longitudes
40  INTEGER, INTENT (IN) :: jjm ! number of atompsheric columns along latitudes
41  INTEGER, INTENT(IN) :: nbp ! number of physics columns for this MPI process
42  INTEGER, INTENT(IN) :: communicator ! MPI communicator
43  REAL, INTENT (IN) :: rlatu(jjm+1) ! latitudes of the physics grid
44  REAL, INTENT (IN) :: rlatv(jjm) ! latitude boundaries of the physics grid
45  REAL, INTENT (IN) :: rlonv(iim+1) ! longitudes of the physics grid
46  REAL, INTENT (IN) :: rlonu(iim+1) ! longitude boundaries of the physics grid
47  REAL, INTENT (IN) :: aire(iim+1,jjm+1) ! area of the dynamics grid (m2)
48  REAL, INTENT (IN) :: cu((iim+1)*(jjm+1)) ! cu coeff. (u_covariant = cu * u)
49  REAL, INTENT (IN) :: cv((iim+1)*jjm) ! cv coeff. (v_covariant = cv * v)
50
51  INTEGER :: ibegin, iend, offset
52  INTEGER :: i,j,k
53  CHARACTER (LEN=20) :: modname = 'inigeomphy'
54  CHARACTER (LEN=80) :: abort_message
55  REAL :: total_area_phy, total_area_dyn
56
57  ! boundaries, on global grid
58  REAL,ALLOCATABLE :: boundslon_reg(:,:)
59  REAL,ALLOCATABLE :: boundslat_reg(:,:)
60
61  ! global array, on full physics grid:
62  REAL,ALLOCATABLE :: latfi_glo(:)
63  REAL,ALLOCATABLE :: lonfi_glo(:)
64  REAL,ALLOCATABLE :: cufi_glo(:)
65  REAL,ALLOCATABLE :: cvfi_glo(:)
66  REAL,ALLOCATABLE :: airefi_glo(:)
67  REAL,ALLOCATABLE :: boundslonfi_glo(:,:)
68  REAL,ALLOCATABLE :: boundslatfi_glo(:,:)
69
70  ! local arrays, on given MPI/OpenMP domain:
71  REAL,ALLOCATABLE,SAVE :: latfi(:)
72  REAL,ALLOCATABLE,SAVE :: lonfi(:)
73  REAL,ALLOCATABLE,SAVE :: cufi(:)
74  REAL,ALLOCATABLE,SAVE :: cvfi(:)
75  REAL,ALLOCATABLE,SAVE :: airefi(:)
76  REAL,ALLOCATABLE,SAVE :: boundslonfi(:,:)
77  REAL,ALLOCATABLE,SAVE :: boundslatfi(:,:)
[3435]78  INTEGER,ALLOCATABLE,SAVE :: ind_cell_glo_fi(:)
79!$OMP THREADPRIVATE (latfi,lonfi,cufi,cvfi,airefi,boundslonfi,boundslatfi,ind_cell_glo_fi)
[2588]80
81  ! Initialize Physics distibution and parameters and interface with dynamics
82  IF (iim*jjm>1) THEN ! general 3D case
83    CALL init_physics_distribution(regular_lonlat,4, &
84                                 nbp,iim,jjm+1,nlayer,communicator)
85  ELSE ! For 1D model
86    CALL init_physics_distribution(regular_lonlat,4, &
87                                 1,1,1,nlayer,communicator)
88  ENDIF
89  CALL init_interface_dyn_phys
90 
91  ! init regular global longitude-latitude grid points and boundaries
92  ALLOCATE(boundslon_reg(iim,2))
93  ALLOCATE(boundslat_reg(jjm+1,2))
94 
[3895]95  ! specific handling of the -180 longitude scalar grid point boundaries
96  boundslon_reg(1,east)=rlonu(1)
97  boundslon_reg(1,west)=rlonu(iim)-2*PI
98  DO i=2,iim
99   boundslon_reg(i,east)=rlonu(i)
100   boundslon_reg(i,west)=rlonu(i-1)
[2588]101  ENDDO
102
103  boundslat_reg(1,north)= PI/2
104  boundslat_reg(1,south)= rlatv(1)
105  DO j=2,jjm
106   boundslat_reg(j,north)=rlatv(j-1)
107   boundslat_reg(j,south)=rlatv(j)
108  ENDDO
109  boundslat_reg(jjm+1,north)= rlatv(jjm)
110  boundslat_reg(jjm+1,south)= -PI/2
111
112  ! Write values in module regular_lonlat_mod
113  CALL init_regular_lonlat(iim,jjm+1, rlonv(1:iim), rlatu, &
114                           boundslon_reg, boundslat_reg)
115
116  ! Generate global arrays on full physics grid
117  ALLOCATE(latfi_glo(klon_glo),lonfi_glo(klon_glo))
118  ALLOCATE(cufi_glo(klon_glo),cvfi_glo(klon_glo))
119  ALLOCATE(airefi_glo(klon_glo))
120  ALLOCATE(boundslonfi_glo(klon_glo,4))
121  ALLOCATE(boundslatfi_glo(klon_glo,4))
122
123  IF (klon_glo>1) THEN ! general case
124    ! North pole
125    latfi_glo(1)=rlatu(1)
126    lonfi_glo(1)=0.
127    cufi_glo(1) = cu(1)
128    cvfi_glo(1) = cv(1)
[3895]129    boundslonfi_glo(1,north_east)=PI
[2588]130    boundslatfi_glo(1,north_east)=PI/2
[3895]131    boundslonfi_glo(1,north_west)=-PI
[2588]132    boundslatfi_glo(1,north_west)=PI/2
[3895]133    boundslonfi_glo(1,south_west)=-PI
[2588]134    boundslatfi_glo(1,south_west)=rlatv(1)
[3895]135    boundslonfi_glo(1,south_east)=PI
[2588]136    boundslatfi_glo(1,south_east)=rlatv(1)
137    DO j=2,jjm
138      DO i=1,iim
139        k=(j-2)*iim+1+i
140        latfi_glo(k)= rlatu(j)
141        lonfi_glo(k)= rlonv(i)
142        cufi_glo(k) = cu((j-1)*(iim+1)+i)
143        cvfi_glo(k) = cv((j-1)*(iim+1)+i)
144        boundslonfi_glo(k,north_east)=rlonu(i)
145        boundslatfi_glo(k,north_east)=rlatv(j-1)
[3895]146        if (i.eq.1) then
147          ! special case for the first longitude's west bound
148          boundslonfi_glo(k,north_west)=rlonu(iim)-2*PI
149          boundslonfi_glo(k,south_west)=rlonu(iim)-2*PI
150        else
151          boundslonfi_glo(k,north_west)=rlonu(i-1)
152          boundslonfi_glo(k,south_west)=rlonu(i-1)
153        endif
[2588]154        boundslatfi_glo(k,north_west)=rlatv(j-1)
155        boundslatfi_glo(k,south_west)=rlatv(j)
156        boundslonfi_glo(k,south_east)=rlonu(i)
157        boundslatfi_glo(k,south_east)=rlatv(j)
158      ENDDO
159    ENDDO
160    ! South pole
161    latfi_glo(klon_glo)= rlatu(jjm+1)
162    lonfi_glo(klon_glo)= 0.
163    cufi_glo(klon_glo) = cu((iim+1)*jjm+1)
164    cvfi_glo(klon_glo) = cv((iim+1)*jjm-iim)
[3895]165    boundslonfi_glo(klon_glo,north_east)= PI
[2588]166    boundslatfi_glo(klon_glo,north_east)= rlatv(jjm)
[3895]167    boundslonfi_glo(klon_glo,north_west)= -PI
[2588]168    boundslatfi_glo(klon_glo,north_west)= rlatv(jjm)
[3895]169    boundslonfi_glo(klon_glo,south_west)= -PI
[2588]170    boundslatfi_glo(klon_glo,south_west)= -PI/2
[3895]171    boundslonfi_glo(klon_glo,south_east)= PI
[2588]172    boundslatfi_glo(klon_glo,south_east)= -Pi/2
173
174    ! build airefi(), mesh area on physics grid
175    CALL gr_dyn_fi(1,iim+1,jjm+1,klon_glo,aire,airefi_glo)
176    ! Poles are single points on physics grid
177    airefi_glo(1)=sum(aire(1:iim,1))
178    airefi_glo(klon_glo)=sum(aire(1:iim,jjm+1))
179
180    ! Sanity check: do total planet area match between physics and dynamics?
181    total_area_dyn=sum(aire(1:iim,1:jjm+1))
182    total_area_phy=sum(airefi_glo(1:klon_glo))
183    IF (total_area_dyn/=total_area_phy) THEN
184      WRITE (lunout, *) 'inigeomphy: planet total surface discrepancy !!!'
185      WRITE (lunout, *) '     in the dynamics total_area_dyn=', total_area_dyn
186      WRITE (lunout, *) '  but in the physics total_area_phy=', total_area_phy
187      IF (abs(total_area_dyn-total_area_phy)>0.00001*total_area_dyn) THEN
188        ! stop here if the relative difference is more than 0.001%
189        abort_message = 'planet total surface discrepancy'
190        CALL abort_gcm(modname, abort_message, 1)
191      ENDIF
192    ENDIF
193  ELSE ! klon_glo==1, running the 1D model
194    ! just copy over input values
195    latfi_glo(1)=rlatu(1)
196    lonfi_glo(1)=rlonv(1)
197    cufi_glo(1)=cu(1)
198    cvfi_glo(1)=cv(1)
199    airefi_glo(1)=aire(1,1)
200    boundslonfi_glo(1,north_east)=rlonu(1)
201    boundslatfi_glo(1,north_east)=PI/2
202    boundslonfi_glo(1,north_west)=rlonu(2)
203    boundslatfi_glo(1,north_west)=PI/2
204    boundslonfi_glo(1,south_west)=rlonu(2)
205    boundslatfi_glo(1,south_west)=rlatv(1)
206    boundslonfi_glo(1,south_east)=rlonu(1)
207    boundslatfi_glo(1,south_east)=rlatv(1)
208  ENDIF ! of IF (klon_glo>1)
209
210!$OMP PARALLEL
211  ! Now generate local lon/lat/cu/cv/area/bounds arrays
212  ALLOCATE(latfi(klon_omp),lonfi(klon_omp),cufi(klon_omp),cvfi(klon_omp))
213  ALLOCATE(airefi(klon_omp))
214  ALLOCATE(boundslonfi(klon_omp,4))
215  ALLOCATE(boundslatfi(klon_omp,4))
[3435]216  ALLOCATE(ind_cell_glo_fi(klon_omp))
[2588]217
218
219  offset = klon_mpi_begin - 1
220  airefi(1:klon_omp) = airefi_glo(offset+klon_omp_begin:offset+klon_omp_end)
221  cufi(1:klon_omp) = cufi_glo(offset+klon_omp_begin:offset+klon_omp_end)
222  cvfi(1:klon_omp) = cvfi_glo(offset+klon_omp_begin:offset+klon_omp_end)
223  lonfi(1:klon_omp) = lonfi_glo(offset+klon_omp_begin:offset+klon_omp_end)
224  latfi(1:klon_omp) = latfi_glo(offset+klon_omp_begin:offset+klon_omp_end)
225  boundslonfi(1:klon_omp,:) = boundslonfi_glo(offset+klon_omp_begin:offset+klon_omp_end,:)
226  boundslatfi(1:klon_omp,:) = boundslatfi_glo(offset+klon_omp_begin:offset+klon_omp_end,:)
[3435]227  ind_cell_glo_fi(1:klon_omp)=(/ (i,i=offset+klon_omp_begin,offset+klon_omp_end) /)
[2588]228
229  ! copy over local grid longitudes and latitudes
230  CALL init_geometry(klon_omp,lonfi,latfi,boundslonfi,boundslatfi, &
[3435]231                     airefi,ind_cell_glo_fi,cufi,cvfi)
[2588]232
[2786]233  ! copy over preff , ap(), bp(), etc
234  CALL init_vertical_layers(nlayer,preff,scaleheight, &
[4228]235                            ap,bp,aps,bps,presnivs,presinter,pseudoalt)
[2786]236
[2588]237!$OMP END PARALLEL
238
239
240END SUBROUTINE inigeomphy
241
242END MODULE inigeomphy_mod
243
Note: See TracBrowser for help on using the repository browser.