Context Navigation

calfis_p.F @ 3881

Last change on this file since 3881 was 3836, checked in by ymipsl, 10 years ago
Update physic call interface. YM
File size: 29.6 KB

Line
1	!
2	! $Id: calfis_p.F 2239 2015-03-23 07:27:30Z emillour $
3	!
4	C
5	C
6	SUBROUTINE calfis_p(lafin,
7	$ jD_cur, jH_cur,
8	$ pucov,
9	$ pvcov,
10	$ pteta,
11	$ pq,
12	$ pmasse,
13	$ pps,
14	$ pp,
15	$ ppk,
16	$ pphis,
17	$ pphi,
18	$ pducov,
19	$ pdvcov,
20	$ pdteta,
21	$ pdq,
22	$ flxw,
23	$ pdufi,
24	$ pdvfi,
25	$ pdhfi,
26	$ pdqfi,
27	$ pdpsfi)
28	#ifdef CPP_PHYS
29	! If using physics
30	USE dimphy
31	USE mod_phys_lmdz_mpi_data, mpi_root_xx=>mpi_master
32	USE mod_phys_lmdz_omp_data, ONLY: klon_omp, klon_omp_begin
33	USE mod_const_mpi, ONLY: COMM_LMDZ
34	USE mod_interface_dyn_phys
35	USE IOPHY
36	#endif
37	#ifdef CPP_PARA
38	USE parallel_lmdz, ONLY : omp_chunk, using_mpi
39	USE Write_Field
40	Use Write_field_p
41	USE Times
42	#endif
43	USE infotrac, ONLY: nqtot, niadv, tname
44	USE control_mod, ONLY: planet_type, nsplit_phys
45
46	IMPLICIT NONE
47	c=======================================================================
48	c
49	c 1. rearrangement des tableaux et transformation
50	c variables dynamiques > variables physiques
51	c 2. calcul des termes physiques
52	c 3. retransformation des tendances physiques en tendances dynamiques
53	c
54	c remarques:
55	c ----------
56	c
57	c - les vents sont donnes dans la physique par leurs composantes
58	c naturelles.
59	c - la variable thermodynamique de la physique est une variable
60	c intensive : T
61	c pour la dynamique on prend T * ( preff / p(l) ) **kappa
62	c - les deux seules variables dependant de la geometrie necessaires
63	c pour la physique sont la latitude pour le rayonnement et
64	c l'aire de la maille quand on veut integrer une grandeur
65	c horizontalement.
66	c - les points de la physique sont les points scalaires de la
67	c la dynamique; numerotation:
68	c 1 pour le pole nord
69	c (jjm-1)*iim pour l'interieur du domaine
70	c ngridmx pour le pole sud
71	c ---> ngridmx=2+(jjm-1)*iim
72	c
73	c Input :
74	c -------
75	c ecritphy frequence d'ecriture (en jours)de histphy
76	c pucov covariant zonal velocity
77	c pvcov covariant meridional velocity
78	c pteta potential temperature
79	c pps surface pressure
80	c pmasse masse d'air dans chaque maille
81	c pts surface temperature (K)
82	c callrad clef d'appel au rayonnement
83	c
84	c Output :
85	c --------
86	c pdufi tendency for the natural zonal velocity (ms-1)
87	c pdvfi tendency for the natural meridional velocity
88	c pdhfi tendency for the potential temperature
89	c pdtsfi tendency for the surface temperature
90	c
91	c pdtrad radiative tendencies \ both input
92	c pfluxrad radiative fluxes / and output
93	c
94	c=======================================================================
95	c
96	c-----------------------------------------------------------------------
97	c
98	c 0. Declarations :
99	c ------------------
100
101	#include "dimensions.h"
102	#include "paramet.h"
103	#include "temps.h"
104
105	INTEGER ngridmx
106	PARAMETER( ngridmx = 2+(jjm-1)*iim - 1/jjm )
107
108	#include "comconst.h"
109	#include "comvert.h"
110	#include "comgeom2.h"
111	#include "iniprint.h"
112	#ifdef CPP_MPI
113	include 'mpif.h'
114	#endif
115	c Arguments :
116	c -----------
117	LOGICAL,INTENT(IN) :: lafin ! .true. for the very last call to physics
118	REAL,INTENT(IN) :: jD_cur, jH_cur
119	REAL,INTENT(IN) :: pvcov(iip1,jjm,llm) ! covariant meridional velocity
120	REAL,INTENT(IN) :: pucov(iip1,jjp1,llm) ! covariant zonal velocity
121	REAL,INTENT(IN) :: pteta(iip1,jjp1,llm) ! potential temperature
122	REAL,INTENT(IN) :: pmasse(iip1,jjp1,llm) ! mass in each cell ! not used
123	REAL,INTENT(IN) :: pq(iip1,jjp1,llm,nqtot) ! tracers
124	REAL,INTENT(IN) :: pphis(iip1,jjp1) ! surface geopotential
125	REAL,INTENT(IN) :: pphi(iip1,jjp1,llm) ! geopotential
126
127	REAL,INTENT(IN) :: pdvcov(iip1,jjm,llm) ! dynamical tendency on vcov ! not used
128	REAL,INTENT(IN) :: pducov(iip1,jjp1,llm) ! dynamical tendency on ucov
129	REAL,INTENT(IN) :: pdteta(iip1,jjp1,llm) ! dynamical tendency on teta
130	! NB: pdteta is used only to compute pcvgt which is in fact not used...
131	REAL,INTENT(IN) :: pdq(iip1,jjp1,llm,nqtot) ! dynamical tendency on tracers
132	! NB: pdq is only used to compute pcvgq which is in fact not used...
133
134	REAL,INTENT(IN) :: pps(iip1,jjp1) ! surface pressure (Pa)
135	REAL,INTENT(IN) :: pp(iip1,jjp1,llmp1) ! pressure at mesh interfaces (Pa)
136	REAL,INTENT(IN) :: ppk(iip1,jjp1,llm) ! Exner at mid-layer
137	REAL,INTENT(IN) :: flxw(iip1,jjp1,llm) ! Vertical mass flux on dynamics grid
138
139	! tendencies (in */s) from the physics
140	REAL,INTENT(OUT) :: pdvfi(iip1,jjm,llm) ! tendency on covariant meridional wind
141	REAL,INTENT(OUT) :: pdufi(iip1,jjp1,llm) ! tendency on covariant zonal wind
142	REAL,INTENT(OUT) :: pdhfi(iip1,jjp1,llm) ! tendency on potential temperature (K/s)
143	REAL,INTENT(OUT) :: pdqfi(iip1,jjp1,llm,nqtot) ! tendency on tracers
144	REAL,INTENT(OUT) :: pdpsfi(iip1,jjp1) ! tendency on surface pressure (Pa/s)
145
146	#ifdef CPP_PARA
147	#ifdef CPP_PHYS
148	! Ehouarn: for now calfis_p needs some informations from physics to compile
149	c Local variables :
150	c -----------------
151
152	INTEGER i,j,l,ig0,ig,iq,iiq
153	REAL,ALLOCATABLE,SAVE :: zpsrf(:)
154	REAL,ALLOCATABLE,SAVE :: zplev(:,:),zplay(:,:)
155	REAL,ALLOCATABLE,SAVE :: zphi(:,:),zphis(:)
156	c
157	REAL,ALLOCATABLE,SAVE :: zufi(:,:), zvfi(:,:)
158	REAL,ALLOCATABLE,SAVE :: ztfi(:,:),zqfi(:,:,:)
159	c
160	REAL,ALLOCATABLE,SAVE :: pcvgu(:,:), pcvgv(:,:)
161	REAL,ALLOCATABLE,SAVE :: pcvgt(:,:), pcvgq(:,:,:)
162	c
163	REAL,ALLOCATABLE,SAVE :: zdufi(:,:),zdvfi(:,:)
164	REAL,ALLOCATABLE,SAVE :: zdtfi(:,:),zdqfi(:,:,:)
165	REAL,ALLOCATABLE,SAVE :: zdpsrf(:)
166	REAL,SAVE,ALLOCATABLE :: flxwfi(:,:) ! Flux de masse verticale sur la grille physiq
167
168	c
169	REAL,ALLOCATABLE,SAVE :: zplev_omp(:,:)
170	REAL,ALLOCATABLE,SAVE :: zplay_omp(:,:)
171	REAL,ALLOCATABLE,SAVE :: zphi_omp(:,:)
172	REAL,ALLOCATABLE,SAVE :: zphis_omp(:)
173	REAL,ALLOCATABLE,SAVE :: presnivs_omp(:)
174	REAL,ALLOCATABLE,SAVE :: zufi_omp(:,:)
175	REAL,ALLOCATABLE,SAVE :: zvfi_omp(:,:)
176	REAL,ALLOCATABLE,SAVE :: ztfi_omp(:,:)
177	REAL,ALLOCATABLE,SAVE :: zqfi_omp(:,:,:)
178	REAL,ALLOCATABLE,SAVE :: zdufi_omp(:,:)
179	REAL,ALLOCATABLE,SAVE :: zdvfi_omp(:,:)
180	REAL,ALLOCATABLE,SAVE :: zdtfi_omp(:,:)
181	REAL,ALLOCATABLE,SAVE :: zdqfi_omp(:,:,:)
182	REAL,ALLOCATABLE,SAVE :: zdpsrf_omp(:)
183	REAL,SAVE,ALLOCATABLE :: flxwfi_omp(:,:) ! Flux de masse verticale sur la grille physiq
184
185	!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
186	! Introduction du splitting (FH)
187	! Question pour Yann :
188	! J'ai été surpris au début que les tableaux zufi_omp, zdufi_omp n'co soitent
189	! en SAVE. Je crois comprendre que c'est parce que tu voulais qu'il
190	! soit allocatable (plutot par exemple que de passer une dimension
191	! dépendant du process en argument des routines) et que, du coup,
192	! le SAVE évite d'avoir à refaire l'allocation à chaque appel.
193	! Tu confirmes ?
194	! J'ai suivi le même principe pour les zdufic_omp
195	! Mais c'est surement bien que tu controles.
196	!
197
198	REAL,ALLOCATABLE,SAVE :: zdufic_omp(:,:)
199	REAL,ALLOCATABLE,SAVE :: zdvfic_omp(:,:)
200	REAL,ALLOCATABLE,SAVE :: zdtfic_omp(:,:)
201	REAL,ALLOCATABLE,SAVE :: zdqfic_omp(:,:,:)
202	REAL jH_cur_split,zdt_split
203	LOGICAL debut_split,lafin_split
204	INTEGER isplit
205	!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
206
207	c$OMP THREADPRIVATE(zplev_omp,zplay_omp,zphi_omp,zphis_omp,
208	c$OMP+ presnivs_omp,zufi_omp,zvfi_omp,ztfi_omp,
209	c$OMP+ zqfi_omp,zdufi_omp,zdvfi_omp,
210	c$OMP+ zdtfi_omp,zdqfi_omp,zdpsrf_omp,flxwfi_omp,
211	c$OMP+ zdufic_omp,zdvfic_omp,zdtfic_omp,zdqfic_omp)
212
213	LOGICAL,SAVE :: first_omp=.true.
214	c$OMP THREADPRIVATE(first_omp)
215
216	REAL zsin(iim),zcos(iim),z1(iim)
217	REAL zsinbis(iim),zcosbis(iim),z1bis(iim)
218	REAL unskap, pksurcp
219	c
220	REAL SSUM
221
222	LOGICAL,SAVE :: firstcal=.true., debut=.true.
223	c$OMP THREADPRIVATE(firstcal,debut)
224
225	REAL,SAVE,dimension(1:iim,1:llm):: du_send,du_recv,dv_send,dv_recv
226	INTEGER :: ierr
227	#ifdef CPP_MPI
228	INTEGER,dimension(MPI_STATUS_SIZE,4) :: Status
229	#else
230	INTEGER,dimension(1,4) :: Status
231	#endif
232	INTEGER, dimension(4) :: Req
233	REAL,ALLOCATABLE,SAVE:: zdufi2(:,:),zdvfi2(:,:)
234	integer :: k,kstart,kend
235	INTEGER :: offset
236
237	LOGICAL tracerdyn
238	c
239	c-----------------------------------------------------------------------
240	c
241	c 1. Initialisations :
242	c --------------------
243	c
244
245	klon=klon_mpi
246
247	c
248	IF ( firstcal ) THEN
249	debut = .TRUE.
250	IF (ngridmx.NE.2+(jjm-1)*iim) THEN
251	write(lunout,*) 'STOP dans calfis'
252	write(lunout,*)
253	& 'La dimension ngridmx doit etre egale a 2 + (jjm-1)*iim'
254	write(lunout,*) ' ngridmx jjm iim '
255	write(lunout,*) ngridmx,jjm,iim
256	STOP
257	ENDIF
258	c$OMP MASTER
259	ALLOCATE(zpsrf(klon))
260	ALLOCATE(zplev(klon,llm+1),zplay(klon,llm))
261	ALLOCATE(zphi(klon,llm),zphis(klon))
262	ALLOCATE(zufi(klon,llm), zvfi(klon,llm))
263	ALLOCATE(ztfi(klon,llm),zqfi(klon,llm,nqtot))
264	ALLOCATE(pcvgu(klon,llm), pcvgv(klon,llm))
265	ALLOCATE(pcvgt(klon,llm), pcvgq(klon,llm,2))
266	ALLOCATE(zdufi(klon,llm),zdvfi(klon,llm))
267	ALLOCATE(zdtfi(klon,llm),zdqfi(klon,llm,nqtot))
268	ALLOCATE(zdpsrf(klon))
269	ALLOCATE(zdufi2(klon+iim,llm),zdvfi2(klon+iim,llm))
270	ALLOCATE(flxwfi(klon,llm))
271	c$OMP END MASTER
272	c$OMP BARRIER
273	ELSE
274	debut = .FALSE.
275	ENDIF
276
277	c
278	c
279	c-----------------------------------------------------------------------
280	c 40. transformation des variables dynamiques en variables physiques:
281	c ---------------------------------------------------------------
282
283	c 41. pressions au sol (en Pascals)
284	c ----------------------------------
285
286	c$OMP MASTER
287	call start_timer(timer_physic)
288	c$OMP END MASTER
289
290	c$OMP MASTER
291	!CDIR ON_ADB(index_i)
292	!CDIR ON_ADB(index_j)
293	do ig0=1,klon
294	i=index_i(ig0)
295	j=index_j(ig0)
296	zpsrf(ig0)=pps(i,j)
297	enddo
298	c$OMP END MASTER
299
300
301	c 42. pression intercouches :
302	c
303	c -----------------------------------------------------------------
304	c .... zplev definis aux (llm +1) interfaces des couches ....
305	c .... zplay definis aux ( llm ) milieux des couches ....
306	c -----------------------------------------------------------------
307
308	c ... Exner = cp * ( p(l) / preff ) ** kappa ....
309	c
310	unskap = 1./ kappa
311	c
312	c print *,omp_rank,'klon--->',klon
313	c$OMP DO SCHEDULE(STATIC,OMP_CHUNK)
314	DO l = 1, llmp1
315	!CDIR ON_ADB(index_i)
316	!CDIR ON_ADB(index_j)
317	do ig0=1,klon
318	i=index_i(ig0)
319	j=index_j(ig0)
320	zplev( ig0,l ) = pp(i,j,l)
321	enddo
322	ENDDO
323	c$OMP END DO NOWAIT
324	c
325	c
326
327	c 43. temperature naturelle (en K) et pressions milieux couches .
328	c ---------------------------------------------------------------
329	c$OMP DO SCHEDULE(STATIC,OMP_CHUNK)
330	DO l=1,llm
331	!CDIR ON_ADB(index_i)
332	!CDIR ON_ADB(index_j)
333	do ig0=1,klon
334	i=index_i(ig0)
335	j=index_j(ig0)
336	pksurcp = ppk(i,j,l) / cpp
337	zplay(ig0,l) = preff * pksurcp ** unskap
338	ztfi(ig0,l) = pteta(i,j,l) * pksurcp
339	enddo
340
341	ENDDO
342	c$OMP END DO NOWAIT
343
344	c 43.bis traceurs
345	c ---------------
346	c
347
348	DO iq=1,nqtot
349	iiq=niadv(iq)
350	c$OMP DO SCHEDULE(STATIC,OMP_CHUNK)
351	DO l=1,llm
352	!CDIR ON_ADB(index_i)
353	!CDIR ON_ADB(index_j)
354	do ig0=1,klon
355	i=index_i(ig0)
356	j=index_j(ig0)
357	zqfi(ig0,l,iq) = pq(i,j,l,iiq)
358	enddo
359	ENDDO
360	c$OMP END DO NOWAIT
361	ENDDO
362
363
364	c Geopotentiel calcule par rapport a la surface locale:
365	c -----------------------------------------------------
366
367	CALL gr_dyn_fi_p(llm,iip1,jjp1,klon,pphi,zphi)
368
369	CALL gr_dyn_fi_p(1,iip1,jjp1,klon,pphis,zphis)
370
371	c$OMP BARRIER
372
373	c$OMP DO SCHEDULE(STATIC,OMP_CHUNK)
374	DO l=1,llm
375	DO ig=1,klon
376	zphi(ig,l)=zphi(ig,l)-zphis(ig)
377	ENDDO
378	ENDDO
379	c$OMP END DO NOWAIT
380
381
382	c
383	c 45. champ u:
384	c ------------
385
386	kstart=1
387	kend=klon
388
389	if (is_north_pole) kstart=2
390	if (is_south_pole) kend=klon-1
391
392	c$OMP DO SCHEDULE(STATIC,OMP_CHUNK)
393	DO l=1,llm
394	!CDIR ON_ADB(index_i)
395	!CDIR ON_ADB(index_j)
396	!CDIR SPARSE
397	do ig0=kstart,kend
398	i=index_i(ig0)
399	j=index_j(ig0)
400	if (i==1) then
401	zufi(ig0,l)= 0.5 *( pucov(iim,j,l)/cu(iim,j)
402	$ + pucov(1,j,l)/cu(1,j) )
403	else
404	zufi(ig0,l)= 0.5*( pucov(i-1,j,l)/cu(i-1,j)
405	$ + pucov(i,j,l)/cu(i,j) )
406	endif
407	enddo
408	ENDDO
409	c$OMP END DO NOWAIT
410
411	c 46.champ v:
412	c -----------
413
414	c$OMP DO SCHEDULE(STATIC,OMP_CHUNK)
415	DO l=1,llm
416	!CDIR ON_ADB(index_i)
417	!CDIR ON_ADB(index_j)
418	DO ig0=kstart,kend
419	i=index_i(ig0)
420	j=index_j(ig0)
421	zvfi(ig0,l)= 0.5 *( pvcov(i,j-1,l)/cv(i,j-1)
422	$ + pvcov(i,j,l)/cv(i,j) )
423
424	ENDDO
425	ENDDO
426	c$OMP END DO NOWAIT
427
428	c 47. champs de vents aux pole nord
429	c ------------------------------
430	c U = 1 / pi * integrale [ v * cos(long) * d long ]
431	c V = 1 / pi * integrale [ v * sin(long) * d long ]
432
433	if (is_north_pole) then
434	c$OMP DO SCHEDULE(STATIC,OMP_CHUNK)
435	DO l=1,llm
436
437	z1(1) =(rlonu(1)-rlonu(iim)+2.pi)pvcov(1,1,l)/cv(1,1)
438	DO i=2,iim
439	z1(i) =(rlonu(i)-rlonu(i-1))*pvcov(i,1,l)/cv(i,1)
440	ENDDO
441
442	DO i=1,iim
443	zcos(i) = COS(rlonv(i))*z1(i)
444	zsin(i) = SIN(rlonv(i))*z1(i)
445	ENDDO
446
447	zufi(1,l) = SSUM(iim,zcos,1)/pi
448	zvfi(1,l) = SSUM(iim,zsin,1)/pi
449
450	ENDDO
451	c$OMP END DO NOWAIT
452	endif
453
454
455	c 48. champs de vents aux pole sud:
456	c ---------------------------------
457	c U = 1 / pi * integrale [ v * cos(long) * d long ]
458	c V = 1 / pi * integrale [ v * sin(long) * d long ]
459
460	if (is_south_pole) then
461	c$OMP DO SCHEDULE(STATIC,OMP_CHUNK)
462	DO l=1,llm
463
464	z1(1) =(rlonu(1)-rlonu(iim)+2.pi)pvcov(1,jjm,l)/cv(1,jjm)
465	DO i=2,iim
466	z1(i) =(rlonu(i)-rlonu(i-1))*pvcov(i,jjm,l)/cv(i,jjm)
467	ENDDO
468
469	DO i=1,iim
470	zcos(i) = COS(rlonv(i))*z1(i)
471	zsin(i) = SIN(rlonv(i))*z1(i)
472	ENDDO
473
474	zufi(klon,l) = SSUM(iim,zcos,1)/pi
475	zvfi(klon,l) = SSUM(iim,zsin,1)/pi
476	ENDDO
477	c$OMP END DO NOWAIT
478	endif
479
480	c On change de grille, dynamique vers physiq, pour le flux de masse verticale
481	CALL gr_dyn_fi_p(llm,iip1,jjp1,klon,flxw,flxwfi)
482
483	c-----------------------------------------------------------------------
484	c Appel de la physique:
485	c ---------------------
486
487
488	c$OMP BARRIER
489	if (first_omp) then
490	klon=klon_omp
491
492	allocate(zplev_omp(klon,llm+1))
493	allocate(zplay_omp(klon,llm))
494	allocate(zphi_omp(klon,llm))
495	allocate(zphis_omp(klon))
496	allocate(presnivs_omp(llm))
497	allocate(zufi_omp(klon,llm))
498	allocate(zvfi_omp(klon,llm))
499	allocate(ztfi_omp(klon,llm))
500	allocate(zqfi_omp(klon,llm,nqtot))
501	allocate(zdufi_omp(klon,llm))
502	allocate(zdvfi_omp(klon,llm))
503	allocate(zdtfi_omp(klon,llm))
504	allocate(zdqfi_omp(klon,llm,nqtot))
505	allocate(zdufic_omp(klon,llm))
506	allocate(zdvfic_omp(klon,llm))
507	allocate(zdtfic_omp(klon,llm))
508	allocate(zdqfic_omp(klon,llm,nqtot))
509	allocate(zdpsrf_omp(klon))
510	allocate(flxwfi_omp(klon,llm))
511	first_omp=.false.
512	endif
513
514
515	klon=klon_omp
516	offset=klon_omp_begin-1
517
518	do l=1,llm+1
519	do i=1,klon
520	zplev_omp(i,l)=zplev(offset+i,l)
521	enddo
522	enddo
523
524	do l=1,llm
525	do i=1,klon
526	zplay_omp(i,l)=zplay(offset+i,l)
527	enddo
528	enddo
529
530	do l=1,llm
531	do i=1,klon
532	zphi_omp(i,l)=zphi(offset+i,l)
533	enddo
534	enddo
535
536	do i=1,klon
537	zphis_omp(i)=zphis(offset+i)
538	enddo
539
540
541	do l=1,llm
542	presnivs_omp(l)=presnivs(l)
543	enddo
544
545	do l=1,llm
546	do i=1,klon
547	zufi_omp(i,l)=zufi(offset+i,l)
548	enddo
549	enddo
550
551	do l=1,llm
552	do i=1,klon
553	zvfi_omp(i,l)=zvfi(offset+i,l)
554	enddo
555	enddo
556
557	do l=1,llm
558	do i=1,klon
559	ztfi_omp(i,l)=ztfi(offset+i,l)
560	enddo
561	enddo
562
563	do iq=1,nqtot
564	do l=1,llm
565	do i=1,klon
566	zqfi_omp(i,l,iq)=zqfi(offset+i,l,iq)
567	enddo
568	enddo
569	enddo
570
571	do l=1,llm
572	do i=1,klon
573	zdufi_omp(i,l)=zdufi(offset+i,l)
574	enddo
575	enddo
576
577	do l=1,llm
578	do i=1,klon
579	zdvfi_omp(i,l)=zdvfi(offset+i,l)
580	enddo
581	enddo
582
583	do l=1,llm
584	do i=1,klon
585	zdtfi_omp(i,l)=zdtfi(offset+i,l)
586	enddo
587	enddo
588
589	do iq=1,nqtot
590	do l=1,llm
591	do i=1,klon
592	zdqfi_omp(i,l,iq)=zdqfi(offset+i,l,iq)
593	enddo
594	enddo
595	enddo
596
597	do i=1,klon
598	zdpsrf_omp(i)=zdpsrf(offset+i)
599	enddo
600
601	do l=1,llm
602	do i=1,klon
603	flxwfi_omp(i,l)=flxwfi(offset+i,l)
604	enddo
605	enddo
606
607	c$OMP BARRIER
608
609	!$OMP MASTER
610	! write(lunout,*) 'PHYSIQUE AVEC NSPLIT_PHYS=',nsplit_phys
611	!$OMP END MASTER
612	zdt_split=dtphys/nsplit_phys
613	zdufic_omp(:,:)=0.
614	zdvfic_omp(:,:)=0.
615	zdtfic_omp(:,:)=0.
616	zdqfic_omp(:,:,:)=0.
617
618	#ifdef CPP_PHYS
619	do isplit=1,nsplit_phys
620
621	jH_cur_split=jH_cur+(isplit-1) * dtvr / (daysec *nsplit_phys)
622	debut_split=debut.and.isplit==1
623	lafin_split=lafin.and.isplit==nsplit_phys
624
625	if (planet_type=="earth") then
626
627	CALL physiq (klon,
628	. llm,
629	. debut_split,
630	. lafin_split,
631	. zdt_split,
632	. zplev_omp,
633	. zplay_omp,
634	. zphi_omp,
635	. zphis_omp,
636	. presnivs_omp,
637	. zufi_omp,
638	. zvfi_omp,
639	. ztfi_omp,
640	. zqfi_omp,
641	. flxwfi_omp,
642	. zdufi_omp,
643	. zdvfi_omp,
644	. zdtfi_omp,
645	. zdqfi_omp,
646	. zdpsrf_omp)
647
648	else if ( planet_type=="generic" ) then
649
650	CALL physiq (klon, !! ngrid
651	. llm, !! nlayer
652	. nqtot, !! nq
653	. tname, !! tracer names from dynamical core (given in infotrac)
654	. debut_split, !! firstcall
655	. lafin_split, !! lastcall
656	. jD_cur, !! pday. see leapfrog_p
657	. jH_cur_split, !! ptime "fraction of day"
658	. zdt_split, !! ptimestep
659	. zplev_omp, !! pplev
660	. zplay_omp, !! pplay
661	. zphi_omp, !! pphi
662	. zufi_omp, !! pu
663	. zvfi_omp, !! pv
664	. ztfi_omp, !! pt
665	. zqfi_omp, !! pq
666	. flxwfi_omp, !! pw !! or 0. anyway this is for diagnostic. not used in physiq.
667	. zdufi_omp, !! pdu
668	. zdvfi_omp, !! pdv
669	. zdtfi_omp, !! pdt
670	. zdqfi_omp, !! pdq
671	. zdpsrf_omp, !! pdpsrf
672	. tracerdyn) !! tracerdyn <-- utilite ???
673
674	endif ! of if (planet_type=="earth")
675
676
677	zufi_omp(:,:)=zufi_omp(:,:)+zdufi_omp(:,:)*zdt_split
678	zvfi_omp(:,:)=zvfi_omp(:,:)+zdvfi_omp(:,:)*zdt_split
679	ztfi_omp(:,:)=ztfi_omp(:,:)+zdtfi_omp(:,:)*zdt_split
680	zqfi_omp(:,:,:)=zqfi_omp(:,:,:)+zdqfi_omp(:,:,:)*zdt_split
681
682	zdufic_omp(:,:)=zdufic_omp(:,:)+zdufi_omp(:,:)
683	zdvfic_omp(:,:)=zdvfic_omp(:,:)+zdvfi_omp(:,:)
684	zdtfic_omp(:,:)=zdtfic_omp(:,:)+zdtfi_omp(:,:)
685	zdqfic_omp(:,:,:)=zdqfic_omp(:,:,:)+zdqfi_omp(:,:,:)
686
687	enddo
688
689	#endif
690	! of #ifdef CPP_PHYS
691
692	zdufi_omp(:,:)=zdufic_omp(:,:)/nsplit_phys
693	zdvfi_omp(:,:)=zdvfic_omp(:,:)/nsplit_phys
694	zdtfi_omp(:,:)=zdtfic_omp(:,:)/nsplit_phys
695	zdqfi_omp(:,:,:)=zdqfic_omp(:,:,:)/nsplit_phys
696	c$OMP BARRIER
697
698	do l=1,llm+1
699	do i=1,klon
700	zplev(offset+i,l)=zplev_omp(i,l)
701	enddo
702	enddo
703
704	do l=1,llm
705	do i=1,klon
706	zplay(offset+i,l)=zplay_omp(i,l)
707	enddo
708	enddo
709
710	do l=1,llm
711	do i=1,klon
712	zphi(offset+i,l)=zphi_omp(i,l)
713	enddo
714	enddo
715
716
717	do i=1,klon
718	zphis(offset+i)=zphis_omp(i)
719	enddo
720
721
722	do l=1,llm
723	presnivs(l)=presnivs_omp(l)
724	enddo
725
726	do l=1,llm
727	do i=1,klon
728	zufi(offset+i,l)=zufi_omp(i,l)
729	enddo
730	enddo
731
732	do l=1,llm
733	do i=1,klon
734	zvfi(offset+i,l)=zvfi_omp(i,l)
735	enddo
736	enddo
737
738	do l=1,llm
739	do i=1,klon
740	ztfi(offset+i,l)=ztfi_omp(i,l)
741	enddo
742	enddo
743
744	do iq=1,nqtot
745	do l=1,llm
746	do i=1,klon
747	zqfi(offset+i,l,iq)=zqfi_omp(i,l,iq)
748	enddo
749	enddo
750	enddo
751
752	do l=1,llm
753	do i=1,klon
754	zdufi(offset+i,l)=zdufi_omp(i,l)
755	enddo
756	enddo
757
758	do l=1,llm
759	do i=1,klon
760	zdvfi(offset+i,l)=zdvfi_omp(i,l)
761	enddo
762	enddo
763
764	do l=1,llm
765	do i=1,klon
766	zdtfi(offset+i,l)=zdtfi_omp(i,l)
767	enddo
768	enddo
769
770	do iq=1,nqtot
771	do l=1,llm
772	do i=1,klon
773	zdqfi(offset+i,l,iq)=zdqfi_omp(i,l,iq)
774	enddo
775	enddo
776	enddo
777
778	do i=1,klon
779	zdpsrf(offset+i)=zdpsrf_omp(i)
780	enddo
781
782
783	klon=klon_mpi
784	500 CONTINUE
785	c$OMP BARRIER
786
787	c$OMP MASTER
788	call stop_timer(timer_physic)
789	c$OMP END MASTER
790
791	IF (using_mpi) THEN
792
793	if (MPI_rank>0) then
794
795	c$OMP DO SCHEDULE(STATIC,OMP_CHUNK)
796	DO l=1,llm
797	du_send(1:iim,l)=zdufi(1:iim,l)
798	dv_send(1:iim,l)=zdvfi(1:iim,l)
799	ENDDO
800	c$OMP END DO NOWAIT
801
802	c$OMP BARRIER
803	#ifdef CPP_MPI
804	c$OMP MASTER
805	!$OMP CRITICAL (MPI)
806	call MPI_ISSEND(du_send,iim*llm,MPI_REAL8,MPI_Rank-1,401,
807	& COMM_LMDZ,Req(1),ierr)
808	call MPI_ISSEND(dv_send,iim*llm,MPI_REAL8,MPI_Rank-1,402,
809	& COMM_LMDZ,Req(2),ierr)
810	!$OMP END CRITICAL (MPI)
811	c$OMP END MASTER
812	#endif
813	c$OMP BARRIER
814
815	endif
816
817	if (MPI_rank<MPI_Size-1) then
818	c$OMP BARRIER
819	#ifdef CPP_MPI
820	c$OMP MASTER
821	!$OMP CRITICAL (MPI)
822	call MPI_IRECV(du_recv,iim*llm,MPI_REAL8,MPI_Rank+1,401,
823	& COMM_LMDZ,Req(3),ierr)
824	call MPI_IRECV(dv_recv,iim*llm,MPI_REAL8,MPI_Rank+1,402,
825	& COMM_LMDZ,Req(4),ierr)
826	!$OMP END CRITICAL (MPI)
827	c$OMP END MASTER
828	#endif
829	endif
830
831	c$OMP BARRIER
832
833
834	#ifdef CPP_MPI
835	c$OMP MASTER
836	!$OMP CRITICAL (MPI)
837	if (MPI_rank>0 .and. MPI_rank< MPI_Size-1) then
838	call MPI_WAITALL(4,Req(1),Status,ierr)
839	else if (MPI_rank>0) then
840	call MPI_WAITALL(2,Req(1),Status,ierr)
841	else if (MPI_rank <MPI_Size-1) then
842	call MPI_WAITALL(2,Req(3),Status,ierr)
843	endif
844	!$OMP END CRITICAL (MPI)
845	c$OMP END MASTER
846	#endif
847
848	c$OMP BARRIER
849
850	ENDIF ! using_mpi
851
852
853	c$OMP DO SCHEDULE(STATIC,OMP_CHUNK)
854	DO l=1,llm
855
856	zdufi2(1:klon,l)=zdufi(1:klon,l)
857	zdufi2(klon+1:klon+iim,l)=du_recv(1:iim,l)
858
859	zdvfi2(1:klon,l)=zdvfi(1:klon,l)
860	zdvfi2(klon+1:klon+iim,l)=dv_recv(1:iim,l)
861
862	pdhfi(:,jj_begin,l)=0
863	pdqfi(:,jj_begin,l,:)=0
864	pdufi(:,jj_begin,l)=0
865	pdvfi(:,jj_begin,l)=0
866
867	if (.not. is_south_pole) then
868	pdhfi(:,jj_end,l)=0
869	pdqfi(:,jj_end,l,:)=0
870	pdufi(:,jj_end,l)=0
871	pdvfi(:,jj_end,l)=0
872	endif
873
874	ENDDO
875	c$OMP END DO NOWAIT
876
877	c$OMP MASTER
878	pdpsfi(:,jj_begin)=0
879	if (.not. is_south_pole) then
880	pdpsfi(:,jj_end)=0
881	endif
882	c$OMP END MASTER
883	c-----------------------------------------------------------------------
884	c transformation des tendances physiques en tendances dynamiques:
885	c ---------------------------------------------------------------
886
887	c tendance sur la pression :
888	c -----------------------------------
889	CALL gr_fi_dyn_p(1,klon,iip1,jjp1,zdpsrf,pdpsfi)
890	c
891	c 62. enthalpie potentielle
892	c ---------------------
893
894	kstart=1
895	kend=klon
896
897	if (is_north_pole) kstart=2
898	if (is_south_pole) kend=klon-1
899
900	c$OMP DO SCHEDULE(STATIC,OMP_CHUNK)
901	DO l=1,llm
902
903	!CDIR ON_ADB(index_i)
904	!CDIR ON_ADB(index_j)
905	!cdir NODEP
906	do ig0=kstart,kend
907	i=index_i(ig0)
908	j=index_j(ig0)
909	pdhfi(i,j,l) = cpp * zdtfi(ig0,l) / ppk(i,j,l)
910	if (i==1) pdhfi(iip1,j,l) = cpp * zdtfi(ig0,l) / ppk(i,j,l)
911	enddo
912
913	if (is_north_pole) then
914	DO i=1,iip1
915	pdhfi(i,1,l) = cpp * zdtfi(1,l) / ppk(i, 1 ,l)
916	enddo
917	endif
918
919	if (is_south_pole) then
920	DO i=1,iip1
921	pdhfi(i,jjp1,l) = cpp * zdtfi(klon,l)/ ppk(i,jjp1,l)
922	ENDDO
923	endif
924	ENDDO
925	c$OMP END DO NOWAIT
926
927	c 62. humidite specifique
928	c ---------------------
929	! Ehouarn: removed this useless bit: was overwritten at step 63 anyways
930	! DO iq=1,nqtot
931	!c$OMP DO SCHEDULE(STATIC,OMP_CHUNK)
932	! DO l=1,llm
933	!!!cdir NODEP
934	! do ig0=kstart,kend
935	! i=index_i(ig0)
936	! j=index_j(ig0)
937	! pdqfi(i,j,l,iq) = zdqfi(ig0,l,iq)
938	! if (i==1) pdqfi(iip1,j,l,iq) = zdqfi(ig0,l,iq)
939	! enddo
940	!
941	! if (is_north_pole) then
942	! do i=1,iip1
943	! pdqfi(i,1,l,iq) = zdqfi(1,l,iq)
944	! enddo
945	! endif
946	!
947	! if (is_south_pole) then
948	! do i=1,iip1
949	! pdqfi(i,jjp1,l,iq) = zdqfi(klon,l,iq)
950	! enddo
951	! endif
952	! ENDDO
953	!c$OMP END DO NOWAIT
954	! ENDDO
955
956	c 63. traceurs
957	c ------------
958	C initialisation des tendances
959
960	c$OMP DO SCHEDULE(STATIC,OMP_CHUNK)
961	DO l=1,llm
962	pdqfi(:,:,l,:)=0.
963	ENDDO
964	c$OMP END DO NOWAIT
965
966	C
967	!cdir NODEP
968	DO iq=1,nqtot
969	iiq=niadv(iq)
970	c$OMP DO SCHEDULE(STATIC,OMP_CHUNK)
971	DO l=1,llm
972	!CDIR ON_ADB(index_i)
973	!CDIR ON_ADB(index_j)
974	!cdir NODEP
975	DO ig0=kstart,kend
976	i=index_i(ig0)
977	j=index_j(ig0)
978	pdqfi(i,j,l,iiq) = zdqfi(ig0,l,iq)
979	if (i==1) pdqfi(iip1,j,l,iiq) = zdqfi(ig0,l,iq)
980	ENDDO
981
982	IF (is_north_pole) then
983	DO i=1,iip1
984	pdqfi(i,1,l,iiq) = zdqfi(1,l,iq)
985	ENDDO
986	ENDIF
987
988	IF (is_south_pole) then
989	DO i=1,iip1
990	pdqfi(i,jjp1,l,iiq) = zdqfi(klon,l,iq)
991	ENDDO
992	ENDIF
993
994	ENDDO
995	c$OMP END DO NOWAIT
996	ENDDO
997
998	c 65. champ u:
999	c ------------
1000	c$OMP DO SCHEDULE(STATIC,OMP_CHUNK)
1001	DO l=1,llm
1002	!CDIR ON_ADB(index_i)
1003	!CDIR ON_ADB(index_j)
1004	!cdir NODEP
1005	do ig0=kstart,kend
1006	i=index_i(ig0)
1007	j=index_j(ig0)
1008
1009	if (i/=iim) then
1010	pdufi(i,j,l)=0.5(zdufi2(ig0,l)+zdufi2(ig0+1,l))cu(i,j)
1011	endif
1012
1013	if (i==1) then
1014	pdufi(iim,j,l)=0.5*( zdufi2(ig0,l)
1015	$ + zdufi2(ig0+iim-1,l))*cu(iim,j)
1016	pdufi(iip1,j,l)=0.5(zdufi2(ig0,l)+zdufi2(ig0+1,l))cu(i,j)
1017	endif
1018
1019	enddo
1020
1021	if (is_north_pole) then
1022	DO i=1,iip1
1023	pdufi(i,1,l) = 0.
1024	ENDDO
1025	endif
1026
1027	if (is_south_pole) then
1028	DO i=1,iip1
1029	pdufi(i,jjp1,l) = 0.
1030	ENDDO
1031	endif
1032
1033	ENDDO
1034	c$OMP END DO NOWAIT
1035
1036	c 67. champ v:
1037	c ------------
1038
1039	kstart=1
1040	kend=klon
1041
1042	if (is_north_pole) kstart=2
1043	if (is_south_pole) kend=klon-1-iim
1044
1045	c$OMP DO SCHEDULE(STATIC,OMP_CHUNK)
1046	DO l=1,llm
1047	!CDIR ON_ADB(index_i)
1048	!CDIR ON_ADB(index_j)
1049	!cdir NODEP
1050	do ig0=kstart,kend
1051	i=index_i(ig0)
1052	j=index_j(ig0)
1053	pdvfi(i,j,l)=0.5(zdvfi2(ig0,l)+zdvfi2(ig0+iim,l))cv(i,j)
1054	if (i==1) pdvfi(iip1,j,l) = 0.5*(zdvfi2(ig0,l)+
1055	$ zdvfi2(ig0+iim,l))
1056	$ *cv(i,j)
1057	enddo
1058
1059	ENDDO
1060	c$OMP END DO NOWAIT
1061
1062
1063	c 68. champ v pres des poles:
1064	c ---------------------------
1065	c v = U * cos(long) + V * SIN(long)
1066
1067	if (is_north_pole) then
1068
1069	c$OMP DO SCHEDULE(STATIC,OMP_CHUNK)
1070	DO l=1,llm
1071
1072	DO i=1,iim
1073	pdvfi(i,1,l)=
1074	$ zdufi(1,l)COS(rlonv(i))+zdvfi(1,l)SIN(rlonv(i))
1075
1076	pdvfi(i,1,l)=
1077	$ 0.5(pdvfi(i,1,l)+zdvfi(i+1,l))cv(i,1)
1078	ENDDO
1079
1080	pdvfi(iip1,1,l) = pdvfi(1,1,l)
1081
1082	ENDDO
1083	c$OMP END DO NOWAIT
1084
1085	endif
1086
1087	if (is_south_pole) then
1088
1089	c$OMP DO SCHEDULE(STATIC,OMP_CHUNK)
1090	DO l=1,llm
1091
1092	DO i=1,iim
1093	pdvfi(i,jjm,l)=zdufi(klon,l)*COS(rlonv(i))
1094	$ +zdvfi(klon,l)*SIN(rlonv(i))
1095
1096	pdvfi(i,jjm,l)=
1097	$ 0.5(pdvfi(i,jjm,l)+zdvfi(klon-iip1+i,l))cv(i,jjm)
1098	ENDDO
1099
1100	pdvfi(iip1,jjm,l)= pdvfi(1,jjm,l)
1101
1102	ENDDO
1103	c$OMP END DO NOWAIT
1104
1105	endif
1106	c-----------------------------------------------------------------------
1107
1108	700 CONTINUE
1109
1110	firstcal = .FALSE.
1111
1112	#else
1113	write(lunout,*)
1114	& "calfis_p: for now can only work with parallel physics"
1115	stop
1116	#endif
1117	! of #ifdef CPP_PHYS
1118	#endif
1119	! of #ifdef CPP_PARA
1120	END

Note: See TracBrowser for help on using the repository browser.

Download in other formats:

Original Format