source: LMDZ5/branches/LMDZ_tree_FC/libf/phylmd/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
RevLine 
[781]1!
2MODULE surf_land_orchidee_mod
[1146]3#ifndef ORCHIDEE_NOOPENMP
[2571]4#ifndef ORCHIDEE_NOZ0H
[2924]5#ifndef ORCHIDEE_NOFREIN
6!FC la derniere ligne et la fin
[781]7!
[2410]8! This module controles the interface towards the model ORCHIDEE.
[781]9!
[2410]10! Compatibility with ORCHIDIEE :
[2571]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.
[2410]13!
[781]14! Subroutines in this module : surf_land_orchidee
15!                              Init_orchidee_index
16!                              Get_orchidee_communicator
17!                              Init_neighbours
18
19  USE dimphy
[1067]20#ifdef CPP_VEGET
[781]21  USE intersurf     ! module d'ORCHIDEE
[1067]22#endif
[781]23  USE cpl_mod,      ONLY : cpl_send_land_fields
[996]24  USE surface_data, ONLY : type_ocean
[2351]25  USE geometry_mod, ONLY : dx, dy
[781]26  USE mod_grid_phy_lmdz
[2351]27  USE mod_phys_lmdz_para, mpi_root_rank=>mpi_master
[781]28
29  IMPLICIT NONE
30
31  PRIVATE
32  PUBLIC  :: surf_land_orchidee
33
34CONTAINS
35!
36!****************************************************************************************
37
38  SUBROUTINE surf_land_orchidee(itime, dtime, date0, knon, &
[2410]39       knindex, rlon, rlat, yrmu0, pctsrf, &
[781]40       debut, lafin, &
[2240]41       plev,  u1_lay, v1_lay, gustiness, temp_air, spechum, epot_air, ccanopy, &
[781]42       tq_cdrag, petAcoef, peqAcoef, petBcoef, peqBcoef, &
43       precip_rain, precip_snow, lwdown, swnet, swdown, &
[1146]44       ps, q2m, t2m, &
[781]45       evap, fluxsens, fluxlat, &             
[888]46       tsol_rad, tsurf_new, alb1_new, alb2_new, &
[2924]47       emis_new, z0m_new, z0h_new, qsurf, &
48       veget, lai, height )
49!FC
[1279]50
[2924]51
[1279]52    USE mod_surf_para
53    USE mod_synchro_omp
[1454]54    USE carbon_cycle_mod, ONLY : carbon_cycle_cpl
[1785]55    USE indice_sol_mod
[2311]56    USE print_control_mod, ONLY: lunout
[2344]57    USE mod_grid_phy_lmdz, ONLY: nbp_lon, nbp_lat
[2348]58#ifdef CPP_VEGET
59    USE time_phylmdz_mod, ONLY: itau_phy
60#endif
[781]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
[1279]85!   ccanopy      concentration CO2 canopee, correspond au co2_send de
86!                carbon_cycle_mod ou valeur constant co2_ppm
[781]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
[888]107!   alb1_new     albedo in visible SW interval
108!   alb2_new     albedo in near IR interval
[781]109!   emis_new     emissivite
[2571]110!   z0m_new      surface roughness for momentum
111!   z0h_new      surface roughness for heat
[781]112!   qsurf        air moisture at surface
113!
[793]114    INCLUDE "YOMCST.h"
[2924]115!FC
116    INCLUDE "dimpft.h"
117
118
[781]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
[2410]131    REAL, DIMENSION(klon), INTENT(IN)         :: yrmu0 ! cosine of solar zenith angle
[781]132    REAL, DIMENSION(klon), INTENT(IN)         :: plev
[2240]133    REAL, DIMENSION(klon), INTENT(IN)         :: u1_lay, v1_lay, gustiness
[781]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
[1146]141    REAL, DIMENSION(klon), INTENT(IN)         :: q2m, t2m
[781]142
143! Parametres de sortie
144!****************************************************************************************
145    REAL, DIMENSION(klon), INTENT(OUT)        :: evap, fluxsens, fluxlat, qsurf
[888]146    REAL, DIMENSION(klon), INTENT(OUT)        :: tsol_rad, tsurf_new
147    REAL, DIMENSION(klon), INTENT(OUT)        :: alb1_new, alb2_new
[2571]148    REAL, DIMENSION(klon), INTENT(OUT)        :: emis_new, z0m_new, z0h_new
[2924]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
[781]153
[2924]154
[781]155! Local
156!****************************************************************************************
157    INTEGER                                   :: ij, jj, igrid, ireal, index
158    INTEGER                                   :: error
[1146]159    REAL, DIMENSION(klon)                     :: swdown_vrai
[781]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)
[987]218   
219    INTEGER :: orch_omp_rank
220    INTEGER :: orch_omp_size
[781]221!
222! Fin definition
223!****************************************************************************************
224
225    IF (check) WRITE(lunout,*)'Entree ', modname
226 
227! Initialisation
228 
229    IF (debut) THEN
[1067]230! Test of coherence between variable ok_veget and cpp key CPP_VEGET
231#ifndef CPP_VEGET
[2924]232             print* , "nvm_orch FC  ", nvm_orch
233             print* , "nvm_lmdz FC  ", nvm_lmdz
234
[1067]235       abort_message='Pb de coherence: ok_veget = .true. mais CPP_VEGET = .false.'
[2311]236       CALL abort_physic(modname,abort_message,1)
[1067]237#endif
238
[987]239       CALL Init_surf_para(knon)
[781]240       ALLOCATE(ktindex(knon))
241       IF ( .NOT. ALLOCATED(albedo_keep)) THEN
[987]242!ym          ALLOCATE(albedo_keep(klon))
243!ym bizarre que non alloué en knon precedement
244          ALLOCATE(albedo_keep(knon))
[781]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
[2344]256          IF ( indi > nbp_lon) THEN
[781]257             indi = 1
258             indj = indj + 1
259          ENDIF
260          ig(igrid) = indi
261          jg(igrid) = indj
262       ENDDO
263       ig(klon) = 1
[2344]264       jg(klon) = nbp_lat
[781]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'
[2311]270             CALL abort_physic(modname,abort_message,1)
[781]271          ENDIF
272       ENDIF
273       IF ((.NOT. ALLOCATED(lon_scat))) THEN
[2344]274          ALLOCATE(lon_scat(nbp_lon,nbp_lat), stat = error)
[781]275          IF (error /= 0) THEN
276             abort_message='Pb allocation lon_scat'
[2311]277             CALL abort_physic(modname,abort_message,1)
[781]278          ENDIF
279       ENDIF
280       IF ((.NOT. ALLOCATED(lat_scat))) THEN
[2344]281          ALLOCATE(lat_scat(nbp_lon,nbp_lat), stat = error)
[781]282          IF (error /= 0) THEN
283             abort_message='Pb allocation lat_scat'
[2311]284             CALL abort_physic(modname,abort_message,1)
[781]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
[2344]302          DO jj = 2, nbp_lat-1
303             DO ij = 1, nbp_lon
[781]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)
[2344]311          lon_scat(:,nbp_lat) = lon_scat(:,2)
312          lat_scat(:,nbp_lat) = rlat_g(klon_glo)
[781]313       ENDIF
314   
[1023]315       CALL bcast(lon_scat)
316       CALL bcast(lat_scat)
[781]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'
[2311]324             CALL abort_physic(modname,abort_message,1)
[781]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'
[2311]332             CALL abort_physic(modname,abort_message,1)
[781]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'
[2311]350             CALL abort_physic(modname,abort_message,1)
[781]351          ENDIF
352       ENDIF
353       DO igrid = 1, knon
354          ij = knindex(igrid)
[2351]355          resolution(igrid,1) = dx(ij)
356          resolution(igrid,2) = dy(ij)
[781]357       ENDDO
358     
359       ALLOCATE(coastalflow(klon), stat = error)
360       IF (error /= 0) THEN
361          abort_message='Pb allocation coastalflow'
[2311]362          CALL abort_physic(modname,abort_message,1)
[781]363       ENDIF
364       
365       ALLOCATE(riverflow(klon), stat = error)
366       IF (error /= 0) THEN
367          abort_message='Pb allocation riverflow'
[2311]368          CALL abort_physic(modname,abort_message,1)
[781]369       ENDIF
[1279]370!
[1454]371! carbon_cycle_cpl not possible with this interface and version of ORHCHIDEE
[1279]372!
373       IF (carbon_cycle_cpl) THEN
[1454]374          abort_message='carbon_cycle_cpl not yet possible with this interface of ORCHIDEE'
[2311]375          CALL abort_physic(modname,abort_message,1)
[1279]376       END IF
377       
[781]378    ENDIF                          ! (fin debut)
[1279]379 
[781]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
[987]386     
[781]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.05*temp_air(1:knon))*9.80665)
[2011]396!    zlev(1:knon) = (100.*plev(1:knon))/((ps(1:knon)/RD*temp_air(1:knon))*RG)
397     zlev(1:knon) = plev(1:knon)*RD*temp_air(1:knon)/((ps(1:knon)*100.0)*RG)
[781]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
[987]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       
[1023]412       IF (knon > 0) THEN
[1067]413#ifdef CPP_VEGET
[987]414         CALL Init_intersurf(nbp_lon,nbp_lat,knon,ktindex,offset,orch_omp_size,orch_omp_rank,orch_comm)
[1067]415#endif
[987]416       ENDIF
[802]417
[987]418       
419       IF (knon > 0) THEN
420
[1067]421#ifdef CPP_VEGET
[2571]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, &
[802]425               cdrag, petA_orc, peqA_orc, petB_orc, peqB_orc, &
426               precip_rain, precip_snow, lwdown, swnet, swdown, ps, &
[2927]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)
[1067]430#endif         
[781]431       ENDIF
432
[987]433       CALL Synchro_omp
434
[781]435       albedo_keep(1:knon) = (albedo_out(1:knon,1)+albedo_out(1:knon,2))/2.
436
437    ENDIF
438
[987]439   
[781]440!  swdown_vrai(1:knon) = swnet(1:knon)/(1. - albedo_keep(1:knon))
441    swdown_vrai(1:knon) = swdown(1:knon)
442
[987]443    IF (knon > 0) THEN
[1067]444#ifdef CPP_VEGET   
[2924]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
[2571]452       CALL intersurf_main_gathered (itime+itau_phy, nbp_lon, nbp_lat, knon, ktindex, dtime,  &
[802]453            lrestart_read, lrestart_write, lalo, &
454            contfrac, neighbours, resolution, date0, &
[781]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), &
[2571]459            tsol_rad(1:knon), tsurf_new(1:knon), qsurf(1:knon), albedo_out(1:knon,:), emis_new(1:knon), z0m_new(1:knon), &
[2924]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))
[1067]463#endif       
[2924]464!FC on doit mettre nvm_orch la ???
[781]465    ENDIF
466
[987]467    CALL Synchro_omp
468   
[781]469    albedo_keep(1:knon) = (albedo_out(1:knon,1)+albedo_out(1:knon,2))/2.
470
471!* Send to coupler
472!
[996]473    IF (type_ocean=='couple') THEN
[781]474       CALL cpl_send_land_fields(itime, knon, knindex, &
475            riverflow, coastalflow)
476    ENDIF
477
[888]478    alb1_new(1:knon) = albedo_out(1:knon,1)
479    alb2_new(1:knon) = albedo_out(1:knon,2)
[781]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   
[987]489    IF (debut) CALL Finalize_surf_para
[1279]490
[987]491   
[781]492  END SUBROUTINE surf_land_orchidee
493!
494!****************************************************************************************
495!
[987]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)
[781]504   
[987]505    INTEGER               :: ktindex_glo(knon_glo)
506    INTEGER               :: offset_para(0:omp_size*mpi_size-1)
507    INTEGER               :: LastPoint
508    INTEGER               :: task
[781]509   
[987]510    ktindex(1:knon)=knindex(1:knon)+(klon_mpi_begin-1)+(klon_omp_begin-1)+nbp_lon-1
[781]511   
[987]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
[781]522    ENDIF
523   
[987]524    CALL bcast(offset_para)
[781]525   
[987]526    offset=offset_para(omp_size*mpi_rank+omp_rank)
527   
528    ktindex(1:knon)=ktindex(1:knon)-offset
[781]529
[987]530  END SUBROUTINE Init_orchidee_index
531
[781]532!
[987]533!************************* ***************************************************************
[781]534!
[987]535
536  SUBROUTINE Get_orchidee_communicator(orch_comm,orch_omp_size,orch_omp_rank)
537  USE  mod_surf_para
538     
[1001]539#ifdef CPP_MPI
[781]540    INCLUDE 'mpif.h'
541#endif   
542
543    INTEGER,INTENT(OUT) :: orch_comm
[987]544    INTEGER,INTENT(OUT) :: orch_omp_size
545    INTEGER,INTENT(OUT) :: orch_omp_rank
[781]546    INTEGER             :: color
[987]547    INTEGER             :: i,ierr
[802]548!
549! End definition
550!****************************************************************************************
[781]551   
[987]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   
[1001]561#ifdef CPP_MPI   
[987]562      CALL MPI_COMM_SPLIT(COMM_LMDZ_PHY,color,mpi_rank,orch_comm,ierr)
[781]563#endif
[1023]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
[987]577   
[781]578   
579  END SUBROUTINE Get_orchidee_communicator
580!
581!****************************************************************************************
582
[987]583
584  SUBROUTINE Init_neighbours(knon,neighbours,knindex,pctsrf)
585    USE mod_grid_phy_lmdz
586    USE mod_surf_para   
[1785]587    USE indice_sol_mod
[987]588
[1001]589#ifdef CPP_MPI
[781]590    INCLUDE 'mpif.h'
591#endif   
592
[802]593! Input arguments
594!****************************************************************************************
[781]595    INTEGER, INTENT(IN)                     :: knon
[987]596    INTEGER, DIMENSION(klon), INTENT(IN)    :: knindex
[781]597    REAL, DIMENSION(klon), INTENT(IN)       :: pctsrf
598   
[802]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 
[987]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)
[802]614!
615! End definition
616!****************************************************************************************
617
[987]618    ktindex(1:knon)=knindex(1:knon)+(klon_mpi_begin-1)+(klon_omp_begin-1)+nbp_lon-1
[781]619   
[987]620    CALL gather_surf(ktindex(1:knon),ktindex_glo)
621    CALL gather(pctsrf,pctsrf_glo)
[781]622   
[987]623    IF (is_mpi_root .AND. is_omp_root) THEN
624      neighbours_glo(:,:)=-1
[781]625!  Initialisation des offset   
626!
627! offset bord ouest
[987]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
[781]631! offset point normal
[987]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
[781]635! offset bord   est
[987]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
[781]639!
640!
641! Attention aux poles
642!
[987]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
[781]647          correspond(ij,jj) = igrid
648       ENDDO
[2126]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
[781]657       
[987]658       DO igrid = 1, knon_glo
659          iglob = ktindex_glo(igrid)
660         
661          IF (MOD(iglob, nbp_lon) == 1) THEN
[781]662             offset = off_ini(:,1)
[987]663          ELSE IF(MOD(iglob, nbp_lon) == 0) THEN
[781]664             offset = off_ini(:,3)
665          ELSE
666             offset = off_ini(:,2)
667          ENDIF
[987]668         
[781]669          DO i = 1, 8
670             index = iglob + offset(i)
[987]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)
[781]676             ENDIF
677          ENDDO
678       ENDDO
[2126]679       ENDIF !fin knon_glo == 1
[781]680
681    ENDIF
682   
[987]683    DO i = 1, 8
684      CALL scatter_surf(neighbours_glo(:,i),neighbours(1:knon,i))
[781]685    ENDDO
686  END SUBROUTINE Init_neighbours
[987]687
[781]688!
689!****************************************************************************************
690!
[1146]691#endif
[2571]692#endif
[2924]693#endif
[781]694END MODULE surf_land_orchidee_mod
Note: See TracBrowser for help on using the repository browser.