Context Navigation

surf_land_orchidee_mod.F90 @ 2927

Last change on this file since 2927 was 2927, checked in by fcheruy, 7 years ago
changed the position of argument nvm to be consistent with optional argument in orchidee and allow retro-compatibility
Property copyright set to Name of program: LMDZ Creation date: 1984 Version: LMDZ5 License: CeCILL version 2 Holder: Laboratoire de m\'et\'eorologie dynamique, CNRS, UMR 8539 See the license file in the root directory Property svn:eol-style set to `native` Property svn:executable set to ``* Property svn:keywords set to `Author Date Id Revision`
File size: 23.3 KB

Line
1	!
2	MODULE surf_land_orchidee_mod
3	#ifndef ORCHIDEE_NOOPENMP
4	#ifndef ORCHIDEE_NOZ0H
5	#ifndef ORCHIDEE_NOFREIN
6	!FC la derniere ligne et la fin
7	!
8	! This module controles the interface towards the model ORCHIDEE.
9	!
10	! Compatibility with ORCHIDIEE :
11	! The current version can be used with ORCHIDEE/trunk from revision 3525.
12	! This interface is used if none of the cpp keys ORCHIDEE_NOOPENMP or ORCHIDEE_NOZ0H is set.
13	!
14	! Subroutines in this module : surf_land_orchidee
15	! Init_orchidee_index
16	! Get_orchidee_communicator
17	! Init_neighbours
18
19	USE dimphy
20	#ifdef CPP_VEGET
21	USE intersurf ! module d'ORCHIDEE
22	#endif
23	USE cpl_mod, ONLY : cpl_send_land_fields
24	USE surface_data, ONLY : type_ocean
25	USE geometry_mod, ONLY : dx, dy
26	USE mod_grid_phy_lmdz
27	USE mod_phys_lmdz_para, mpi_root_rank=>mpi_master
28
29	IMPLICIT NONE
30
31	PRIVATE
32	PUBLIC :: surf_land_orchidee
33
34	CONTAINS
35	!
36	!****************************************************************************************
37	!
38	SUBROUTINE surf_land_orchidee(itime, dtime, date0, knon, &
39	knindex, rlon, rlat, yrmu0, pctsrf, &
40	debut, lafin, &
41	plev, u1_lay, v1_lay, gustiness, temp_air, spechum, epot_air, ccanopy, &
42	tq_cdrag, petAcoef, peqAcoef, petBcoef, peqBcoef, &
43	precip_rain, precip_snow, lwdown, swnet, swdown, &
44	ps, q2m, t2m, &
45	evap, fluxsens, fluxlat, &
46	tsol_rad, tsurf_new, alb1_new, alb2_new, &
47	emis_new, z0m_new, z0h_new, qsurf, &
48	veget, lai, height )
49	!FC
50
51
52	USE mod_surf_para
53	USE mod_synchro_omp
54	USE carbon_cycle_mod, ONLY : carbon_cycle_cpl
55	USE indice_sol_mod
56	USE print_control_mod, ONLY: lunout
57	USE mod_grid_phy_lmdz, ONLY: nbp_lon, nbp_lat
58	#ifdef CPP_VEGET
59	USE time_phylmdz_mod, ONLY: itau_phy
60	#endif
61	!
62	! Cette routine sert d'interface entre le modele atmospherique et le
63	! modele de sol continental. Appel a sechiba
64	!
65	! L. Fairhead 02/2000
66	!
67	! input:
68	! itime numero du pas de temps
69	! dtime pas de temps de la physique (en s)
70	! nisurf index de la surface a traiter (1 = sol continental)
71	! knon nombre de points de la surface a traiter
72	! knindex index des points de la surface a traiter
73	! rlon longitudes de la grille entiere
74	! rlat latitudes de la grille entiere
75	! pctsrf tableau des fractions de surface de chaque maille
76	! debut logical: 1er appel a la physique (lire les restart)
77	! lafin logical: dernier appel a la physique (ecrire les restart)
78	! (si false calcul simplifie des fluxs sur les continents)
79	! plev hauteur de la premiere couche (Pa)
80	! u1_lay vitesse u 1ere couche
81	! v1_lay vitesse v 1ere couche
82	! temp_air temperature de l'air 1ere couche
83	! spechum humidite specifique 1ere couche
84	! epot_air temp pot de l'air
85	! ccanopy concentration CO2 canopee, correspond au co2_send de
86	! carbon_cycle_mod ou valeur constant co2_ppm
87	! tq_cdrag cdrag
88	! petAcoef coeff. A de la resolution de la CL pour t
89	! peqAcoef coeff. A de la resolution de la CL pour q
90	! petBcoef coeff. B de la resolution de la CL pour t
91	! peqBcoef coeff. B de la resolution de la CL pour q
92	! precip_rain precipitation liquide
93	! precip_snow precipitation solide
94	! lwdown flux IR descendant a la surface
95	! swnet flux solaire net
96	! swdown flux solaire entrant a la surface
97	! ps pression au sol
98	! radsol rayonnement net aus sol (LW + SW)
99	!
100	!
101	! output:
102	! evap evaporation totale
103	! fluxsens flux de chaleur sensible
104	! fluxlat flux de chaleur latente
105	! tsol_rad
106	! tsurf_new temperature au sol
107	! alb1_new albedo in visible SW interval
108	! alb2_new albedo in near IR interval
109	! emis_new emissivite
110	! z0m_new surface roughness for momentum
111	! z0h_new surface roughness for heat
112	! qsurf air moisture at surface
113	!
114	INCLUDE "YOMCST.h"
115	!FC
116	INCLUDE "dimpft.h"
117
118
119
120	!
121	! Parametres d'entree
122	!****************************************************************************************
123	INTEGER, INTENT(IN) :: itime
124	REAL, INTENT(IN) :: dtime
125	REAL, INTENT(IN) :: date0
126	INTEGER, INTENT(IN) :: knon
127	INTEGER, DIMENSION(klon), INTENT(IN) :: knindex
128	LOGICAL, INTENT(IN) :: debut, lafin
129	REAL, DIMENSION(klon,nbsrf), INTENT(IN) :: pctsrf
130	REAL, DIMENSION(klon), INTENT(IN) :: rlon, rlat
131	REAL, DIMENSION(klon), INTENT(IN) :: yrmu0 ! cosine of solar zenith angle
132	REAL, DIMENSION(klon), INTENT(IN) :: plev
133	REAL, DIMENSION(klon), INTENT(IN) :: u1_lay, v1_lay, gustiness
134	REAL, DIMENSION(klon), INTENT(IN) :: temp_air, spechum
135	REAL, DIMENSION(klon), INTENT(IN) :: epot_air, ccanopy
136	REAL, DIMENSION(klon), INTENT(IN) :: tq_cdrag
137	REAL, DIMENSION(klon), INTENT(IN) :: petAcoef, peqAcoef
138	REAL, DIMENSION(klon), INTENT(IN) :: petBcoef, peqBcoef
139	REAL, DIMENSION(klon), INTENT(IN) :: precip_rain, precip_snow
140	REAL, DIMENSION(klon), INTENT(IN) :: lwdown, swnet, swdown, ps
141	REAL, DIMENSION(klon), INTENT(IN) :: q2m, t2m
142
143	! Parametres de sortie
144	!****************************************************************************************
145	REAL, DIMENSION(klon), INTENT(OUT) :: evap, fluxsens, fluxlat, qsurf
146	REAL, DIMENSION(klon), INTENT(OUT) :: tsol_rad, tsurf_new
147	REAL, DIMENSION(klon), INTENT(OUT) :: alb1_new, alb2_new
148	REAL, DIMENSION(klon), INTENT(OUT) :: emis_new, z0m_new, z0h_new
149	!FC
150	REAL, DIMENSION (klon,nvm_lmdz),INTENT(OUT) :: veget
151	REAL, DIMENSION (klon,nvm_lmdz),INTENT(OUT) :: lai
152	REAL, DIMENSION (klon,nvm_lmdz) ,INTENT(OUT) :: height
153
154
155	! Local
156	!****************************************************************************************
157	INTEGER :: ij, jj, igrid, ireal, index
158	INTEGER :: error
159	REAL, DIMENSION(klon) :: swdown_vrai
160	CHARACTER (len = 20) :: modname = 'surf_land_orchidee'
161	CHARACTER (len = 80) :: abort_message
162	LOGICAL,SAVE :: check = .FALSE.
163	!$OMP THREADPRIVATE(check)
164
165	! type de couplage dans sechiba
166	! character (len=10) :: coupling = 'implicit'
167	! drapeaux controlant les appels dans SECHIBA
168	! type(control_type), save :: control_in
169	! Preserved albedo
170	REAL, ALLOCATABLE, DIMENSION(:), SAVE :: albedo_keep, zlev
171	!$OMP THREADPRIVATE(albedo_keep,zlev)
172	! coordonnees geographiques
173	REAL, ALLOCATABLE, DIMENSION(:,:), SAVE :: lalo
174	!$OMP THREADPRIVATE(lalo)
175	! pts voisins
176	INTEGER,ALLOCATABLE, DIMENSION(:,:), SAVE :: neighbours
177	!$OMP THREADPRIVATE(neighbours)
178	! fractions continents
179	REAL,ALLOCATABLE, DIMENSION(:), SAVE :: contfrac
180	!$OMP THREADPRIVATE(contfrac)
181	! resolution de la grille
182	REAL, ALLOCATABLE, DIMENSION (:,:), SAVE :: resolution
183	!$OMP THREADPRIVATE(resolution)
184
185	REAL, ALLOCATABLE, DIMENSION (:,:), SAVE :: lon_scat, lat_scat
186	!$OMP THREADPRIVATE(lon_scat,lat_scat)
187
188	LOGICAL, SAVE :: lrestart_read = .TRUE.
189	!$OMP THREADPRIVATE(lrestart_read)
190	LOGICAL, SAVE :: lrestart_write = .FALSE.
191	!$OMP THREADPRIVATE(lrestart_write)
192
193	REAL, DIMENSION(knon,2) :: albedo_out
194
195	! Pb de nomenclature
196	REAL, DIMENSION(klon) :: petA_orc, peqA_orc
197	REAL, DIMENSION(klon) :: petB_orc, peqB_orc
198	! Pb de correspondances de grilles
199	INTEGER, DIMENSION(:), SAVE, ALLOCATABLE :: ig, jg
200	!$OMP THREADPRIVATE(ig,jg)
201	INTEGER :: indi, indj
202	INTEGER, SAVE, ALLOCATABLE,DIMENSION(:) :: ktindex
203	!$OMP THREADPRIVATE(ktindex)
204
205	! Essai cdrag
206	REAL, DIMENSION(klon) :: cdrag
207	INTEGER,SAVE :: offset
208	!$OMP THREADPRIVATE(offset)
209
210	REAL, DIMENSION(klon_glo) :: rlon_g,rlat_g
211	INTEGER, SAVE :: orch_comm
212	!$OMP THREADPRIVATE(orch_comm)
213
214	REAL, ALLOCATABLE, DIMENSION(:), SAVE :: coastalflow
215	!$OMP THREADPRIVATE(coastalflow)
216	REAL, ALLOCATABLE, DIMENSION(:), SAVE :: riverflow
217	!$OMP THREADPRIVATE(riverflow)
218
219	INTEGER :: orch_omp_rank
220	INTEGER :: orch_omp_size
221	!
222	! Fin definition
223	!****************************************************************************************
224
225	IF (check) WRITE(lunout,*)'Entree ', modname
226
227	! Initialisation
228
229	IF (debut) THEN
230	! Test of coherence between variable ok_veget and cpp key CPP_VEGET
231	#ifndef CPP_VEGET
232	print* , "nvm_orch FC ", nvm_orch
233	print* , "nvm_lmdz FC ", nvm_lmdz
234
235	abort_message='Pb de coherence: ok_veget = .true. mais CPP_VEGET = .false.'
236	CALL abort_physic(modname,abort_message,1)
237	#endif
238
239	CALL Init_surf_para(knon)
240	ALLOCATE(ktindex(knon))
241	IF ( .NOT. ALLOCATED(albedo_keep)) THEN
242	!ym ALLOCATE(albedo_keep(klon))
243	!ym bizarre que non alloué en knon precedement
244	ALLOCATE(albedo_keep(knon))
245	ALLOCATE(zlev(knon))
246	ENDIF
247	! Pb de correspondances de grilles
248	ALLOCATE(ig(klon))
249	ALLOCATE(jg(klon))
250	ig(1) = 1
251	jg(1) = 1
252	indi = 0
253	indj = 2
254	DO igrid = 2, klon - 1
255	indi = indi + 1
256	IF ( indi > nbp_lon) THEN
257	indi = 1
258	indj = indj + 1
259	ENDIF
260	ig(igrid) = indi
261	jg(igrid) = indj
262	ENDDO
263	ig(klon) = 1
264	jg(klon) = nbp_lat
265
266	IF ((.NOT. ALLOCATED(lalo))) THEN
267	ALLOCATE(lalo(knon,2), stat = error)
268	IF (error /= 0) THEN
269	abort_message='Pb allocation lalo'
270	CALL abort_physic(modname,abort_message,1)
271	ENDIF
272	ENDIF
273	IF ((.NOT. ALLOCATED(lon_scat))) THEN
274	ALLOCATE(lon_scat(nbp_lon,nbp_lat), stat = error)
275	IF (error /= 0) THEN
276	abort_message='Pb allocation lon_scat'
277	CALL abort_physic(modname,abort_message,1)
278	ENDIF
279	ENDIF
280	IF ((.NOT. ALLOCATED(lat_scat))) THEN
281	ALLOCATE(lat_scat(nbp_lon,nbp_lat), stat = error)
282	IF (error /= 0) THEN
283	abort_message='Pb allocation lat_scat'
284	CALL abort_physic(modname,abort_message,1)
285	ENDIF
286	ENDIF
287	lon_scat = 0.
288	lat_scat = 0.
289	DO igrid = 1, knon
290	index = knindex(igrid)
291	lalo(igrid,2) = rlon(index)
292	lalo(igrid,1) = rlat(index)
293	ENDDO
294
295
296
297	CALL Gather(rlon,rlon_g)
298	CALL Gather(rlat,rlat_g)
299
300	IF (is_mpi_root) THEN
301	index = 1
302	DO jj = 2, nbp_lat-1
303	DO ij = 1, nbp_lon
304	index = index + 1
305	lon_scat(ij,jj) = rlon_g(index)
306	lat_scat(ij,jj) = rlat_g(index)
307	ENDDO
308	ENDDO
309	lon_scat(:,1) = lon_scat(:,2)
310	lat_scat(:,1) = rlat_g(1)
311	lon_scat(:,nbp_lat) = lon_scat(:,2)
312	lat_scat(:,nbp_lat) = rlat_g(klon_glo)
313	ENDIF
314
315	CALL bcast(lon_scat)
316	CALL bcast(lat_scat)
317	!
318	! Allouer et initialiser le tableau des voisins et des fraction de continents
319	!
320	IF ( (.NOT.ALLOCATED(neighbours))) THEN
321	ALLOCATE(neighbours(knon,8), stat = error)
322	IF (error /= 0) THEN
323	abort_message='Pb allocation neighbours'
324	CALL abort_physic(modname,abort_message,1)
325	ENDIF
326	ENDIF
327	neighbours = -1.
328	IF (( .NOT. ALLOCATED(contfrac))) THEN
329	ALLOCATE(contfrac(knon), stat = error)
330	IF (error /= 0) THEN
331	abort_message='Pb allocation contfrac'
332	CALL abort_physic(modname,abort_message,1)
333	ENDIF
334	ENDIF
335
336	DO igrid = 1, knon
337	ireal = knindex(igrid)
338	contfrac(igrid) = pctsrf(ireal,is_ter)
339	ENDDO
340
341
342	CALL Init_neighbours(knon,neighbours,knindex,pctsrf(:,is_ter))
343
344	!
345	! Allocation et calcul resolutions
346	IF ( (.NOT.ALLOCATED(resolution))) THEN
347	ALLOCATE(resolution(knon,2), stat = error)
348	IF (error /= 0) THEN
349	abort_message='Pb allocation resolution'
350	CALL abort_physic(modname,abort_message,1)
351	ENDIF
352	ENDIF
353	DO igrid = 1, knon
354	ij = knindex(igrid)
355	resolution(igrid,1) = dx(ij)
356	resolution(igrid,2) = dy(ij)
357	ENDDO
358
359	ALLOCATE(coastalflow(klon), stat = error)
360	IF (error /= 0) THEN
361	abort_message='Pb allocation coastalflow'
362	CALL abort_physic(modname,abort_message,1)
363	ENDIF
364
365	ALLOCATE(riverflow(klon), stat = error)
366	IF (error /= 0) THEN
367	abort_message='Pb allocation riverflow'
368	CALL abort_physic(modname,abort_message,1)
369	ENDIF
370	!
371	! carbon_cycle_cpl not possible with this interface and version of ORHCHIDEE
372	!
373	IF (carbon_cycle_cpl) THEN
374	abort_message='carbon_cycle_cpl not yet possible with this interface of ORCHIDEE'
375	CALL abort_physic(modname,abort_message,1)
376	END IF
377
378	ENDIF ! (fin debut)
379
380
381	!
382	! Appel a la routine sols continentaux
383	!
384	IF (lafin) lrestart_write = .TRUE.
385	IF (check) WRITE(lunout,*)'lafin ',lafin,lrestart_write
386
387	petA_orc(1:knon) = petBcoef(1:knon) * dtime
388	petB_orc(1:knon) = petAcoef(1:knon)
389	peqA_orc(1:knon) = peqBcoef(1:knon) * dtime
390	peqB_orc(1:knon) = peqAcoef(1:knon)
391
392	cdrag = 0.
393	cdrag(1:knon) = tq_cdrag(1:knon)
394
395	! zlev(1:knon) = (100.plev(1:knon))/((ps(1:knon)/287.05temp_air(1:knon))*9.80665)
396	! zlev(1:knon) = (100.plev(1:knon))/((ps(1:knon)/RDtemp_air(1:knon))*RG)
397	zlev(1:knon) = plev(1:knon)RDtemp_air(1:knon)/((ps(1:knon)100.0)RG)
398
399
400	! PF et PASB
401	! where(cdrag > 0.01)
402	! cdrag = 0.01
403	! endwhere
404	! write(,)'Cdrag = ',minval(cdrag),maxval(cdrag)
405
406
407	IF (debut) THEN
408	CALL Init_orchidee_index(knon,knindex,offset,ktindex)
409	CALL Get_orchidee_communicator(orch_comm,orch_omp_size,orch_omp_rank)
410	CALL Init_synchro_omp
411
412	IF (knon > 0) THEN
413	#ifdef CPP_VEGET
414	CALL Init_intersurf(nbp_lon,nbp_lat,knon,ktindex,offset,orch_omp_size,orch_omp_rank,orch_comm)
415	#endif
416	ENDIF
417
418
419	IF (knon > 0) THEN
420
421	#ifdef CPP_VEGET
422	CALL intersurf_initialize_gathered (itime+itau_phy-1, nbp_lon, nbp_lat, knon, ktindex, dtime, &
423	lrestart_read, lrestart_write, lalo, contfrac, neighbours, resolution, date0, &
424	zlev, u1_lay, v1_lay, spechum, temp_air, epot_air, &
425	cdrag, petA_orc, peqA_orc, petB_orc, peqB_orc, &
426	precip_rain, precip_snow, lwdown, swnet, swdown, ps, &
427	evap, fluxsens, fluxlat, coastalflow, riverflow, &
428	tsol_rad, tsurf_new, qsurf, albedo_out, emis_new, z0m_new, &
429	lon_scat, lat_scat, q2m, t2m, z0h_new, nvm_orch)
430	#endif
431	ENDIF
432
433	CALL Synchro_omp
434
435	albedo_keep(1:knon) = (albedo_out(1:knon,1)+albedo_out(1:knon,2))/2.
436
437	ENDIF
438
439
440	! swdown_vrai(1:knon) = swnet(1:knon)/(1. - albedo_keep(1:knon))
441	swdown_vrai(1:knon) = swdown(1:knon)
442
443	IF (knon > 0) THEN
444	#ifdef CPP_VEGET
445	!FC
446	if(nvm_orch .ne.nvm_lmdz ) then
447	abort_message='Pb de dimensiosn PFT: nvm_orch et nvm_lmdz differents.'
448	CALL abort_physic(modname,abort_message,1)
449	endif
450	!FC
451
452	CALL intersurf_main_gathered (itime+itau_phy, nbp_lon, nbp_lat, knon, ktindex, dtime, &
453	lrestart_read, lrestart_write, lalo, &
454	contfrac, neighbours, resolution, date0, &
455	zlev, u1_lay(1:knon), v1_lay(1:knon), spechum(1:knon), temp_air(1:knon), epot_air(1:knon), ccanopy(1:knon), &
456	cdrag(1:knon), petA_orc(1:knon), peqA_orc(1:knon), petB_orc(1:knon), peqB_orc(1:knon), &
457	precip_rain(1:knon), precip_snow(1:knon), lwdown(1:knon), swnet(1:knon), swdown_vrai(1:knon), ps(1:knon), &
458	evap(1:knon), fluxsens(1:knon), fluxlat(1:knon), coastalflow(1:knon), riverflow(1:knon), &
459	tsol_rad(1:knon), tsurf_new(1:knon), qsurf(1:knon), albedo_out(1:knon,:), emis_new(1:knon), z0m_new(1:knon), &
460	lon_scat, lat_scat, q2m, t2m, z0h_new(1:knon),&
461	veget(1:knon,:),lai(1:knon,:),height(1:knon,:),&
462	coszang=yrmu0(1:knon))
463	#endif
464	!FC on doit mettre nvm_orch la ???
465	ENDIF
466
467	CALL Synchro_omp
468
469	albedo_keep(1:knon) = (albedo_out(1:knon,1)+albedo_out(1:knon,2))/2.
470
471	!* Send to coupler
472	!
473	IF (type_ocean=='couple') THEN
474	CALL cpl_send_land_fields(itime, knon, knindex, &
475	riverflow, coastalflow)
476	ENDIF
477
478	alb1_new(1:knon) = albedo_out(1:knon,1)
479	alb2_new(1:knon) = albedo_out(1:knon,2)
480
481	! Convention orchidee: positif vers le haut
482	fluxsens(1:knon) = -1. * fluxsens(1:knon)
483	fluxlat(1:knon) = -1. * fluxlat(1:knon)
484
485	! evap = -1. * evap
486
487	IF (debut) lrestart_read = .FALSE.
488
489	IF (debut) CALL Finalize_surf_para
490
491
492	END SUBROUTINE surf_land_orchidee
493	!
494	!****************************************************************************************
495	!
496	SUBROUTINE Init_orchidee_index(knon,knindex,offset,ktindex)
497	USE mod_surf_para
498	USE mod_grid_phy_lmdz
499
500	INTEGER,INTENT(IN) :: knon
501	INTEGER,INTENT(IN) :: knindex(klon)
502	INTEGER,INTENT(OUT) :: offset
503	INTEGER,INTENT(OUT) :: ktindex(klon)
504
505	INTEGER :: ktindex_glo(knon_glo)
506	INTEGER :: offset_para(0:omp_size*mpi_size-1)
507	INTEGER :: LastPoint
508	INTEGER :: task
509
510	ktindex(1:knon)=knindex(1:knon)+(klon_mpi_begin-1)+(klon_omp_begin-1)+nbp_lon-1
511
512	CALL gather_surf(ktindex(1:knon),ktindex_glo)
513
514	IF (is_mpi_root .AND. is_omp_root) THEN
515	LastPoint=0
516	DO Task=0,mpi_size*omp_size-1
517	IF (knon_glo_para(Task)>0) THEN
518	offset_para(task)= LastPoint-MOD(LastPoint,nbp_lon)
519	LastPoint=ktindex_glo(knon_glo_end_para(task))
520	ENDIF
521	ENDDO
522	ENDIF
523
524	CALL bcast(offset_para)
525
526	offset=offset_para(omp_size*mpi_rank+omp_rank)
527
528	ktindex(1:knon)=ktindex(1:knon)-offset
529
530	END SUBROUTINE Init_orchidee_index
531
532	!
533	!*********************** *************************************************************
534	!
535
536	SUBROUTINE Get_orchidee_communicator(orch_comm,orch_omp_size,orch_omp_rank)
537	USE mod_surf_para
538
539	#ifdef CPP_MPI
540	INCLUDE 'mpif.h'
541	#endif
542
543	INTEGER,INTENT(OUT) :: orch_comm
544	INTEGER,INTENT(OUT) :: orch_omp_size
545	INTEGER,INTENT(OUT) :: orch_omp_rank
546	INTEGER :: color
547	INTEGER :: i,ierr
548	!
549	! End definition
550	!****************************************************************************************
551
552
553	IF (is_omp_root) THEN
554
555	IF (knon_mpi==0) THEN
556	color = 0
557	ELSE
558	color = 1
559	ENDIF
560
561	#ifdef CPP_MPI
562	CALL MPI_COMM_SPLIT(COMM_LMDZ_PHY,color,mpi_rank,orch_comm,ierr)
563	#endif
564
565	ENDIF
566	CALL bcast_omp(orch_comm)
567
568	IF (knon_mpi /= 0) THEN
569	orch_omp_size=0
570	DO i=0,omp_size-1
571	IF (knon_omp_para(i) /=0) THEN
572	orch_omp_size=orch_omp_size+1
573	IF (i==omp_rank) orch_omp_rank=orch_omp_size-1
574	ENDIF
575	ENDDO
576	ENDIF
577
578
579	END SUBROUTINE Get_orchidee_communicator
580	!
581	!****************************************************************************************
582	!
583
584	SUBROUTINE Init_neighbours(knon,neighbours,knindex,pctsrf)
585	USE mod_grid_phy_lmdz
586	USE mod_surf_para
587	USE indice_sol_mod
588
589	#ifdef CPP_MPI
590	INCLUDE 'mpif.h'
591	#endif
592
593	! Input arguments
594	!****************************************************************************************
595	INTEGER, INTENT(IN) :: knon
596	INTEGER, DIMENSION(klon), INTENT(IN) :: knindex
597	REAL, DIMENSION(klon), INTENT(IN) :: pctsrf
598
599	! Output arguments
600	!****************************************************************************************
601	INTEGER, DIMENSION(knon,8), INTENT(OUT) :: neighbours
602
603	! Local variables
604	!****************************************************************************************
605	INTEGER :: i, igrid, jj, ij, iglob
606	INTEGER :: ierr, ireal, index
607	INTEGER, DIMENSION(8,3) :: off_ini
608	INTEGER, DIMENSION(8) :: offset
609	INTEGER, DIMENSION(nbp_lon,nbp_lat) :: correspond
610	INTEGER, DIMENSION(knon_glo) :: ktindex_glo
611	INTEGER, DIMENSION(knon_glo,8) :: neighbours_glo
612	REAL, DIMENSION(klon_glo) :: pctsrf_glo
613	INTEGER :: ktindex(klon)
614	!
615	! End definition
616	!****************************************************************************************
617
618	ktindex(1:knon)=knindex(1:knon)+(klon_mpi_begin-1)+(klon_omp_begin-1)+nbp_lon-1
619
620	CALL gather_surf(ktindex(1:knon),ktindex_glo)
621	CALL gather(pctsrf,pctsrf_glo)
622
623	IF (is_mpi_root .AND. is_omp_root) THEN
624	neighbours_glo(:,:)=-1
625	! Initialisation des offset
626	!
627	! offset bord ouest
628	off_ini(1,1) = - nbp_lon ; off_ini(2,1) = - nbp_lon + 1 ; off_ini(3,1) = 1
629	off_ini(4,1) = nbp_lon + 1 ; off_ini(5,1) = nbp_lon ; off_ini(6,1) = 2 * nbp_lon - 1
630	off_ini(7,1) = nbp_lon -1 ; off_ini(8,1) = - 1
631	! offset point normal
632	off_ini(1,2) = - nbp_lon ; off_ini(2,2) = - nbp_lon + 1 ; off_ini(3,2) = 1
633	off_ini(4,2) = nbp_lon + 1 ; off_ini(5,2) = nbp_lon ; off_ini(6,2) = nbp_lon - 1
634	off_ini(7,2) = -1 ; off_ini(8,2) = - nbp_lon - 1
635	! offset bord est
636	off_ini(1,3) = - nbp_lon ; off_ini(2,3) = - 2 * nbp_lon + 1 ; off_ini(3,3) = - nbp_lon + 1
637	off_ini(4,3) = 1 ; off_ini(5,3) = nbp_lon ; off_ini(6,3) = nbp_lon - 1
638	off_ini(7,3) = -1 ; off_ini(8,3) = - nbp_lon - 1
639	!
640	!
641	! Attention aux poles
642	!
643	DO igrid = 1, knon_glo
644	index = ktindex_glo(igrid)
645	jj = INT((index - 1)/nbp_lon) + 1
646	ij = index - (jj - 1) * nbp_lon
647	correspond(ij,jj) = igrid
648	ENDDO
649	!sonia : Les mailles des voisines doivent etre toutes egales (pour couplage orchidee)
650	IF (knon_glo == 1) THEN
651	igrid = 1
652	DO i = 1,8
653	neighbours_glo(igrid, i) = igrid
654	ENDDO
655	ELSE
656	print*,'sonia : knon_glo,ij,jj', knon_glo, ij,jj
657
658	DO igrid = 1, knon_glo
659	iglob = ktindex_glo(igrid)
660
661	IF (MOD(iglob, nbp_lon) == 1) THEN
662	offset = off_ini(:,1)
663	ELSE IF(MOD(iglob, nbp_lon) == 0) THEN
664	offset = off_ini(:,3)
665	ELSE
666	offset = off_ini(:,2)
667	ENDIF
668
669	DO i = 1, 8
670	index = iglob + offset(i)
671	ireal = (MIN(MAX(1, index - nbp_lon + 1), klon_glo))
672	IF (pctsrf_glo(ireal) > EPSFRA) THEN
673	jj = INT((index - 1)/nbp_lon) + 1
674	ij = index - (jj - 1) * nbp_lon
675	neighbours_glo(igrid, i) = correspond(ij, jj)
676	ENDIF
677	ENDDO
678	ENDDO
679	ENDIF !fin knon_glo == 1
680
681	ENDIF
682
683	DO i = 1, 8
684	CALL scatter_surf(neighbours_glo(:,i),neighbours(1:knon,i))
685	ENDDO
686	END SUBROUTINE Init_neighbours
687
688	!
689	!****************************************************************************************
690	!
691	#endif
692	#endif
693	#endif
694	END MODULE surf_land_orchidee_mod

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

Download in other formats:

Original Format