source: LMDZ6/trunk/libf/phylmd/iophy.F90 @ 3082

Last change on this file since 3082 was 3082, checked in by oboucher, 7 years ago

Adding new diagnostics for clean (no aerosol) clear sky SW radiative fluxes

  • 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:keywords set to Author Date Id Revision
File size: 48.4 KB
Line 
1!
2! $Id: iophy.F90 3082 2017-11-21 16:18:23Z oboucher $
3!
4MODULE iophy
5
6! abd  REAL,private,allocatable,DIMENSION(:),save :: io_lat
7! abd  REAL,private,allocatable,DIMENSION(:),save :: io_lon
8  REAL,ALLOCATABLE,DIMENSION(:),SAVE :: io_lat
9  REAL,ALLOCATABLE,DIMENSION(:),SAVE :: io_lon
10  INTEGER, SAVE :: phys_domain_id
11  INTEGER, SAVE :: npstn
12  INTEGER, ALLOCATABLE, DIMENSION(:), SAVE :: nptabij
13  INTEGER, SAVE :: itau_iophy
14
15!$OMP THREADPRIVATE(itau_iophy)
16
17#ifdef CPP_XIOS
18  INTERFACE histwrite_phy
19!#ifdef CPP_XIOSnew
20    MODULE PROCEDURE histwrite2d_phy,histwrite3d_phy,histwrite2d_phy_old,histwrite3d_phy_old,histwrite2d_xios,histwrite3d_xios,histwrite0d_xios
21!#else
22!    MODULE PROCEDURE histwrite2d_phy,histwrite3d_phy,histwrite2d_phy_old,histwrite3d_phy_old,histwrite2d_xios,histwrite3d_xios
23!#endif
24
25  END INTERFACE
26#else
27  INTERFACE histwrite_phy
28    MODULE PROCEDURE histwrite2d_phy,histwrite3d_phy,histwrite2d_phy_old,histwrite3d_phy_old
29  END INTERFACE
30#endif
31
32  INTERFACE histbeg_phy_all
33    MODULE PROCEDURE histbeg_phy,histbeg_phyxios,histbeg_phy_points
34  END INTERFACE
35
36
37CONTAINS
38
39! ug Routine pour définir itau_iophy depuis phys_output_write_mod:
40  SUBROUTINE set_itau_iophy(ito)
41      IMPLICIT NONE
42      INTEGER, INTENT(IN) :: ito
43      itau_iophy = ito
44  END SUBROUTINE
45
46  SUBROUTINE init_iophy_new(rlat,rlon)
47  USE dimphy, ONLY: klon
48  USE mod_phys_lmdz_para, ONLY: gather, bcast, &
49                                jj_nb, jj_begin, jj_end, ii_begin, ii_end, &
50                                mpi_size, mpi_rank, klon_mpi, &
51                                is_sequential, is_south_pole_dyn
52  USE mod_grid_phy_lmdz, ONLY: nbp_lon, nbp_lat, klon_glo
53  USE print_control_mod, ONLY: prt_level,lunout
54#ifdef CPP_IOIPSL
55  USE ioipsl, ONLY: flio_dom_set
56#endif
57#ifdef CPP_XIOS
58  USE wxios, ONLY: wxios_domain_param
59#endif
60  IMPLICIT NONE
61    REAL,DIMENSION(klon),INTENT(IN) :: rlon
62    REAL,DIMENSION(klon),INTENT(IN) :: rlat
63
64    REAL,DIMENSION(klon_glo)        :: rlat_glo
65    REAL,DIMENSION(klon_glo)        :: rlon_glo
66   
67    INTEGER,DIMENSION(2) :: ddid
68    INTEGER,DIMENSION(2) :: dsg
69    INTEGER,DIMENSION(2) :: dsl
70    INTEGER,DIMENSION(2) :: dpf
71    INTEGER,DIMENSION(2) :: dpl
72    INTEGER,DIMENSION(2) :: dhs
73    INTEGER,DIMENSION(2) :: dhe
74    INTEGER :: i   
75    INTEGER :: data_ibegin, data_iend
76
77    CALL gather(rlat,rlat_glo)
78    CALL bcast(rlat_glo)
79    CALL gather(rlon,rlon_glo)
80    CALL bcast(rlon_glo)
81   
82!$OMP MASTER 
83    ALLOCATE(io_lat(nbp_lat))
84    IF (klon_glo == 1) THEN
85      io_lat(1)=rlat_glo(1)
86    ELSE
87      io_lat(1)=rlat_glo(1)
88      io_lat(nbp_lat)=rlat_glo(klon_glo)
89      DO i=2,nbp_lat-1
90        io_lat(i)=rlat_glo(2+(i-2)*nbp_lon)
91      ENDDO
92    ENDIF
93
94    ALLOCATE(io_lon(nbp_lon))
95    IF (klon_glo == 1) THEN
96      io_lon(1)=rlon_glo(1)
97    ELSE
98      io_lon(1:nbp_lon)=rlon_glo(2:nbp_lon+1)
99    ENDIF
100
101!! (I) dtnb   : total number of domains
102!! (I) dnb    : domain number
103!! (I) did(:) : distributed dimensions identifiers
104!!              (up to 5 dimensions are supported)
105!! (I) dsg(:) : total number of points for each dimension
106!! (I) dsl(:) : local number of points for each dimension
107!! (I) dpf(:) : position of first local point for each dimension
108!! (I) dpl(:) : position of last local point for each dimension
109!! (I) dhs(:) : start halo size for each dimension
110!! (I) dhe(:) : end halo size for each dimension
111!! (C) cdnm   : Model domain definition name.
112!!              The names actually supported are :
113!!              "BOX", "APPLE", "ORANGE".
114!!              These names are case insensitive.
115
116    ddid=(/ 1,2 /)
117    dsg=(/ nbp_lon, nbp_lat /)
118    dsl=(/ nbp_lon, jj_nb /)
119    dpf=(/ 1,jj_begin /)
120    dpl=(/ nbp_lon, jj_end /)
121    dhs=(/ ii_begin-1,0 /)
122    IF (mpi_rank==mpi_size-1) THEN
123      dhe=(/0,0/)
124    ELSE
125      dhe=(/ nbp_lon-ii_end,0 /) 
126    ENDIF
127
128#ifndef CPP_IOIPSL_NO_OUTPUT   
129    CALL flio_dom_set(mpi_size,mpi_rank,ddid,dsg,dsl,dpf,dpl,dhs,dhe, &
130                      'APPLE',phys_domain_id)
131#endif
132#ifdef CPP_XIOS
133    ! Set values for the mask:
134    IF (mpi_rank == 0) THEN
135        data_ibegin = 0
136    ELSE
137        data_ibegin = ii_begin - 1
138    ENDIF
139
140    IF (mpi_rank == mpi_size-1) THEN
141        data_iend = nbp_lon
142    ELSE
143        data_iend = ii_end + 1
144    ENDIF
145
146    IF (prt_level>=10) THEN
147      write(lunout,*) "init_iophy_new: mpirank=",mpi_rank," iibegin=",ii_begin , " ii_end=",ii_end," jjbegin=",jj_begin," jj_nb=",jj_nb," jj_end=",jj_end
148      write(lunout,*) "init_iophy_new: mpirank=",mpi_rank," nbp_lon=",nbp_lon," nbp_lat=",nbp_lat
149      write(lunout,*) "init_iophy_new: mpirank=",mpi_rank," data_ibegin=",data_ibegin," data_iend=",data_iend
150      write(lunout,*) "init_iophy_new: mpirank=",mpi_rank," data_ibegin=",data_ibegin," data_iend=",data_iend
151      write(lunout,*) "init_iophy_new: mpirank=",mpi_rank," is_south_pole=",is_south_pole_dyn
152    ENDIF
153
154    ! Initialize the XIOS domain coreesponding to this process:
155    CALL wxios_domain_param("dom_glo", is_sequential, nbp_lon, jj_nb, nbp_lon, nbp_lat, &
156                            1, nbp_lon, ii_begin, ii_end, jj_begin, jj_end,             &
157                            klon_mpi+2*(nbp_lon-1), data_ibegin, data_iend,             &
158                            io_lat, io_lon,is_south_pole_dyn,mpi_rank)
159#endif
160!$OMP END MASTER
161     
162  END SUBROUTINE init_iophy_new
163
164  SUBROUTINE init_iophy(lat,lon)
165  USE mod_phys_lmdz_para, ONLY: jj_begin, jj_end, ii_begin, ii_end, jj_nb, &
166                                mpi_size, mpi_rank
167  USE ioipsl, ONLY: flio_dom_set
168  USE mod_grid_phy_lmdz, ONLY : nbp_lon, nbp_lat
169  IMPLICIT NONE
170    REAL,DIMENSION(nbp_lon),INTENT(IN) :: lon
171    REAL,DIMENSION(nbp_lat),INTENT(IN) :: lat
172
173    INTEGER,DIMENSION(2) :: ddid
174    INTEGER,DIMENSION(2) :: dsg
175    INTEGER,DIMENSION(2) :: dsl
176    INTEGER,DIMENSION(2) :: dpf
177    INTEGER,DIMENSION(2) :: dpl
178    INTEGER,DIMENSION(2) :: dhs
179    INTEGER,DIMENSION(2) :: dhe
180
181!$OMP MASTER 
182    ALLOCATE(io_lat(nbp_lat))
183    io_lat(:)=lat(:)
184    ALLOCATE(io_lon(nbp_lon))
185    io_lon(:)=lon(:)
186   
187    ddid=(/ 1,2 /)
188    dsg=(/ nbp_lon, nbp_lat /)
189    dsl=(/ nbp_lon, jj_nb /)
190    dpf=(/ 1,jj_begin /)
191    dpl=(/ nbp_lon, jj_end /)
192    dhs=(/ ii_begin-1,0 /)
193    IF (mpi_rank==mpi_size-1) THEN
194      dhe=(/0,0/)
195    ELSE
196      dhe=(/ nbp_lon-ii_end,0 /) 
197    ENDIF
198   
199#ifndef CPP_IOIPSL_NO_OUTPUT
200    call flio_dom_set(mpi_size,mpi_rank,ddid,dsg,dsl,dpf,dpl,dhs,dhe, &
201                      'APPLE',phys_domain_id)
202#endif
203!$OMP END MASTER
204     
205  END SUBROUTINE init_iophy
206
207 SUBROUTINE histbeg_phyxios(name,itau0,zjulian,dtime,ffreq,lev,nhori,nid_day)
208!  USE dimphy
209  USE mod_phys_lmdz_para, ONLY: is_sequential, is_using_mpi, is_mpi_root, &
210                                jj_begin, jj_end, jj_nb
211  USE mod_grid_phy_lmdz, ONLY : nbp_lon, nbp_lat
212  USE ioipsl, ONLY: histbeg
213#ifdef CPP_XIOS
214  USE wxios, ONLY: wxios_add_file
215#endif
216  IMPLICIT NONE
217  include 'clesphys.h'
218   
219    CHARACTER*(*), INTENT(IN) :: name
220    INTEGER, INTENT(IN) :: itau0
221    REAL,INTENT(IN) :: zjulian
222    REAL,INTENT(IN) :: dtime
223    CHARACTER(LEN=*), INTENT(IN) :: ffreq
224    INTEGER,INTENT(IN) :: lev
225    INTEGER,INTENT(OUT) :: nhori
226    INTEGER,INTENT(OUT) :: nid_day
227
228!$OMP MASTER   
229    IF (is_sequential) THEN
230      call histbeg(name,nbp_lon,io_lon, jj_nb,io_lat(jj_begin:jj_end), &
231                   1,nbp_lon,1,jj_nb,itau0, zjulian, dtime, nhori, nid_day)
232    ELSE
233      call histbeg(name,nbp_lon,io_lon, jj_nb,io_lat(jj_begin:jj_end), &
234                   1,nbp_lon,1,jj_nb,itau0, zjulian, dtime, nhori, nid_day,phys_domain_id)
235    ENDIF
236
237#ifdef CPP_XIOS
238    ! ug OMP en chantier...
239    IF((.NOT. is_using_mpi) .OR. is_mpi_root) THEN
240        ! ug Création du fichier
241      IF (.not. ok_all_xml) THEN
242        CALL wxios_add_file(name, ffreq, lev)
243      ENDIF
244    ENDIF
245#endif
246!$OMP END MASTER
247 
248  END SUBROUTINE histbeg_phyxios
249 
250  SUBROUTINE histbeg_phy(name,itau0,zjulian,dtime,nhori,nid_day)
251
252  USE mod_phys_lmdz_para, ONLY: jj_begin, jj_end, jj_nb, is_sequential
253  USE mod_grid_phy_lmdz, ONLY : nbp_lon, nbp_lat
254  USE ioipsl, ONLY: histbeg
255
256  IMPLICIT NONE
257   
258    CHARACTER*(*), INTENT(IN) :: name
259    INTEGER, INTENT(IN) :: itau0
260    REAL,INTENT(IN) :: zjulian
261    REAL,INTENT(IN) :: dtime
262    INTEGER,INTENT(OUT) :: nhori
263    INTEGER,INTENT(OUT) :: nid_day
264
265!$OMP MASTER   
266#ifndef CPP_IOIPSL_NO_OUTPUT
267    IF (is_sequential) THEN
268      call histbeg(name,nbp_lon,io_lon, jj_nb,io_lat(jj_begin:jj_end), &
269                   1,nbp_lon,1,jj_nb,itau0, zjulian, dtime, nhori, nid_day)
270    ELSE
271      call histbeg(name,nbp_lon,io_lon, jj_nb,io_lat(jj_begin:jj_end), &
272                   1,nbp_lon,1,jj_nb,itau0, zjulian, dtime, nhori, nid_day,phys_domain_id)
273    ENDIF
274#endif
275!$OMP END MASTER
276 
277  END SUBROUTINE histbeg_phy
278
279
280  SUBROUTINE histbeg_phy_points(rlon,rlat,pim,tabij,ipt,jpt, &
281             plon,plat,plon_bounds,plat_bounds, &
282             nname,itau0,zjulian,dtime,nnhori,nnid_day)
283  USE dimphy, ONLY: klon
284  USE mod_phys_lmdz_para, ONLY: gather, bcast, &
285                                is_sequential, klon_mpi_begin, klon_mpi_end, &
286                                mpi_rank
287  USE mod_grid_phy_lmdz, ONLY: klon_glo, nbp_lon, nbp_lat
288  USE ioipsl, ONLY: histbeg
289
290  IMPLICIT NONE
291
292    REAL,DIMENSION(klon),INTENT(IN) :: rlon
293    REAL,DIMENSION(klon),INTENT(IN) :: rlat
294    INTEGER, INTENT(IN) :: itau0
295    REAL,INTENT(IN) :: zjulian
296    REAL,INTENT(IN) :: dtime
297    INTEGER, INTENT(IN) :: pim
298    INTEGER, intent(out) :: nnhori
299    CHARACTER(len=20), INTENT(IN) :: nname
300    INTEGER, INTENT(OUT) :: nnid_day
301    INTEGER :: i
302    REAL,DIMENSION(klon_glo)        :: rlat_glo
303    REAL,DIMENSION(klon_glo)        :: rlon_glo
304    INTEGER, DIMENSION(pim), INTENT(IN)  :: tabij
305    REAL,DIMENSION(pim), INTENT(IN) :: plat, plon
306    INTEGER,DIMENSION(pim), INTENT(IN) :: ipt, jpt
307    REAL,DIMENSION(pim,2), intent(out) :: plat_bounds, plon_bounds
308
309    INTEGER, SAVE :: tabprocbeg, tabprocend
310!$OMP THREADPRIVATE(tabprocbeg, tabprocend)
311    INTEGER :: ip
312    INTEGER, PARAMETER :: nip=1
313    INTEGER :: npproc
314    REAL, allocatable, DIMENSION(:) :: npplat, npplon
315    REAL, allocatable, DIMENSION(:,:) :: npplat_bounds, npplon_bounds
316    REAL, DIMENSION(nbp_lon,nbp_lat) :: zx_lon, zx_lat
317
318    CALL gather(rlat,rlat_glo)
319    CALL bcast(rlat_glo)
320    CALL gather(rlon,rlon_glo)
321    CALL bcast(rlon_glo)
322
323!$OMP MASTER
324    DO i=1,pim
325
326!    print*,'CFMIP_iophy i tabij lon lat',i,tabij(i),plon(i),plat(i)
327
328     plon_bounds(i,1)=rlon_glo(tabij(i)-1)
329     plon_bounds(i,2)=rlon_glo(tabij(i)+1)
330     IF (plon_bounds(i,2).LE.0..AND.plon_bounds(i,1).GE.0.) THEN
331      IF (rlon_glo(tabij(i)).GE.0.) THEN
332       plon_bounds(i,2)=-1*plon_bounds(i,2)
333      ENDIF
334     ENDIF
335     IF (plon_bounds(i,2).GE.0..AND.plon_bounds(i,1).LE.0.) THEN
336      IF (rlon_glo(tabij(i)).LE.0.) THEN
337       plon_bounds(i,2)=-1*plon_bounds(i,2)
338      ENDIF
339     ENDIF
340!
341     IF ( tabij(i).LE.nbp_lon) THEN
342      plat_bounds(i,1)=rlat_glo(tabij(i))
343     ELSE
344      plat_bounds(i,1)=rlat_glo(tabij(i)-nbp_lon)
345     ENDIF
346     plat_bounds(i,2)=rlat_glo(tabij(i)+nbp_lon)
347!
348!    print*,'CFMIP_iophy point i lon lon_bds',i,plon_bounds(i,1),rlon_glo(tabij(i)),plon_bounds(i,2)
349!    print*,'CFMIP_iophy point i lat lat_bds',i,plat_bounds(i,1),rlat_glo(tabij(i)),plat_bounds(i,2)
350!
351    ENDDO
352    if (is_sequential) then
353
354     npstn=pim
355     IF(.NOT. ALLOCATED(nptabij)) THEN
356      ALLOCATE(nptabij(pim))
357     ENDIF
358     DO i=1,pim
359      nptabij(i)=tabij(i)
360     ENDDO
361
362       CALL gr_fi_ecrit(1,klon,nbp_lon,nbp_lat,rlon_glo,zx_lon)
363       IF ((nbp_lon*nbp_lat).GT.1) THEN
364       DO i = 1, nbp_lon
365         zx_lon(i,1) = rlon_glo(i+1)
366         zx_lon(i,nbp_lat) = rlon_glo(i+1)
367       ENDDO
368       ENDIF
369       CALL gr_fi_ecrit(1,klon,nbp_lon,nbp_lat,rlat_glo,zx_lat)
370
371    DO i=1,pim
372!    print*,'CFMIP_iophy i tabij lon lat',i,tabij(i),plon(i),plat(i)
373
374     plon_bounds(i,1)=zx_lon(ipt(i)-1,jpt(i))
375     plon_bounds(i,2)=zx_lon(ipt(i)+1,jpt(i))
376
377     IF (ipt(i).EQ.1) THEN
378      plon_bounds(i,1)=zx_lon(nbp_lon,jpt(i))
379      plon_bounds(i,2)=360.+zx_lon(ipt(i)+1,jpt(i))
380     ENDIF
381 
382     IF (ipt(i).EQ.nbp_lon) THEN
383      plon_bounds(i,2)=360.+zx_lon(1,jpt(i))
384     ENDIF
385
386     plat_bounds(i,1)=zx_lat(ipt(i),jpt(i)-1)
387     plat_bounds(i,2)=zx_lat(ipt(i),jpt(i)+1)
388
389     IF (jpt(i).EQ.1) THEN
390      plat_bounds(i,1)=zx_lat(ipt(i),1)+0.001
391      plat_bounds(i,2)=zx_lat(ipt(i),1)-0.001
392     ENDIF
393 
394     IF (jpt(i).EQ.nbp_lat) THEN
395      plat_bounds(i,1)=zx_lat(ipt(i),nbp_lat)+0.001
396      plat_bounds(i,2)=zx_lat(ipt(i),nbp_lat)-0.001
397     ENDIF
398!
399!    print*,'CFMIP_iophy point i lon lon_bds',i,plon_bounds(i,1),rlon(tabij(i)),plon_bounds(i,2)
400!    print*,'CFMIP_iophy point i lat lat_bds',i,plat_bounds(i,1),rlat(tabij(i)),plat_bounds(i,2)
401!
402    ENDDO
403
404#ifndef CPP_IOIPSL_NO_OUTPUT
405     call histbeg(nname,pim,plon,plon_bounds, &
406                           plat,plat_bounds, &
407                           itau0, zjulian, dtime, nnhori, nnid_day)
408#endif
409    ELSE
410     npproc=0
411     DO ip=1, pim
412      tabprocbeg=klon_mpi_begin
413      tabprocend=klon_mpi_end
414      IF(tabij(ip).GE.tabprocbeg.AND.tabij(ip).LE.tabprocend) THEN
415       npproc=npproc+1
416       npstn=npproc
417      ENDIF
418     ENDDO
419!    print*,'CFMIP_iophy mpi_rank npstn',mpi_rank,npstn
420     IF(.NOT. ALLOCATED(nptabij)) THEN
421      ALLOCATE(nptabij(npstn))
422      ALLOCATE(npplon(npstn), npplat(npstn))
423      ALLOCATE(npplon_bounds(npstn,2), npplat_bounds(npstn,2))
424     ENDIF
425     npproc=0
426     DO ip=1, pim
427      IF(tabij(ip).GE.tabprocbeg.AND.tabij(ip).LE.tabprocend) THEN
428       npproc=npproc+1
429       nptabij(npproc)=tabij(ip)
430!      print*,'mpi_rank npproc ip plon plat tabij=',mpi_rank,npproc,ip, &
431!      plon(ip),plat(ip),tabij(ip)
432       npplon(npproc)=plon(ip)
433       npplat(npproc)=plat(ip)
434       npplon_bounds(npproc,1)=plon_bounds(ip,1)
435       npplon_bounds(npproc,2)=plon_bounds(ip,2)
436       npplat_bounds(npproc,1)=plat_bounds(ip,1)
437       npplat_bounds(npproc,2)=plat_bounds(ip,2)
438!!!
439!!! print qui sert a reordonner les points stations selon l'ordre CFMIP
440!!! ne pas enlever
441        print*,'iophy_mpi rank ip lon lat',mpi_rank,ip,plon(ip),plat(ip)
442!!!
443      ENDIF
444     ENDDO
445#ifndef CPP_IOIPSL_NO_OUTPUT
446     call histbeg(nname,npstn,npplon,npplon_bounds, &
447                            npplat,npplat_bounds, &
448                            itau0,zjulian,dtime,nnhori,nnid_day,phys_domain_id)
449#endif
450    ENDIF
451!$OMP END MASTER
452
453  END SUBROUTINE histbeg_phy_points
454
455
456  SUBROUTINE histdef2d_old (iff,lpoint,flag_var,nomvar,titrevar,unitvar)
457
458    USE ioipsl, ONLY: histdef
459    USE mod_phys_lmdz_para, ONLY: jj_nb
460    USE phys_output_var_mod, ONLY: type_ecri, zoutm, zdtime_moy, lev_files, &
461                                   nid_files, nhorim, swaero_diag, dryaod_diag, nfiles, &
462                                   ok_4xCO2atm
463    USE mod_grid_phy_lmdz, ONLY : nbp_lon, nbp_lat
464    USE aero_mod, ONLY : naero_tot, name_aero_tau
465
466    IMPLICIT NONE
467
468    INCLUDE "clesphys.h"
469
470    INTEGER                          :: iff
471    INTEGER                          :: naero
472    LOGICAL                          :: lpoint
473    INTEGER, DIMENSION(nfiles)       :: flag_var
474    CHARACTER(LEN=20)                :: nomvar
475    CHARACTER(LEN=*)                 :: titrevar
476    CHARACTER(LEN=*)                 :: unitvar
477
478    REAL zstophym
479
480    IF (type_ecri(iff)=='inst(X)'.OR.type_ecri(iff)=='once') THEN
481       zstophym=zoutm(iff)
482    ELSE
483       zstophym=zdtime_moy
484    ENDIF
485
486    ! Appel a la lecture des noms et niveau d'ecriture des variables dans output.def
487    CALL conf_physoutputs(nomvar,flag_var)
488
489    IF(.NOT.lpoint) THEN 
490       IF ( flag_var(iff)<=lev_files(iff) ) THEN
491          CALL histdef (nid_files(iff),nomvar,titrevar,unitvar, &
492               nbp_lon,jj_nb,nhorim(iff), 1,1,1, -99, 32, &
493               type_ecri(iff), zstophym,zoutm(iff))               
494       ENDIF
495    ELSE
496       IF ( flag_var(iff)<=lev_files(iff) ) THEN
497          CALL histdef (nid_files(iff),nomvar,titrevar,unitvar, &
498               npstn,1,nhorim(iff), 1,1,1, -99, 32, &
499               type_ecri(iff), zstophym,zoutm(iff))               
500       ENDIF
501    ENDIF
502
503    ! Set swaero_diag=true if at least one of the concerned variables are defined
504    IF (nomvar=='topswad' .OR. nomvar=='topswad0' .OR. nomvar=='solswad' .OR. nomvar=='solswad0' .OR. &
505        nomvar=='toplwad' .OR. nomvar=='toplwad0' .OR. nomvar=='sollwad' .OR. nomvar=='sollwad0' .OR. &
506        nomvar=='topswai' .OR. nomvar=='solswai' ) THEN
507       IF  ( flag_var(iff)<=lev_files(iff) ) swaero_diag=.TRUE.
508    ENDIF
509
510    ! Set dryaod_diag=true if at least one of the concerned variables are defined
511    IF (nomvar=='dryod550aer') THEN
512      IF  ( flag_var(iff)<=lev_files(iff) ) dryaod_diag=.TRUE.
513    ENDIF
514    DO naero = 1, naero_tot-1
515      IF (nomvar=='dryod550_'//name_aero_tau(naero)) THEN
516        IF  ( flag_var(iff)<=lev_files(iff) ) dryaod_diag=.TRUE.
517      ENDIF
518    ENDDO
519
520    ! Set ok_4xCO2atm=true if at least one of the concerned variables are
521    ! defined
522    IF (nomvar=='rsut4co2'.OR.nomvar=='rlut4co2'.OR.nomvar=='rsutcs4co2' &
523        .OR. nomvar=='rlutcs4co2'.OR.nomvar=='rsu4co2'.OR.nomvar=='rsucs4co2' &
524        .OR.nomvar=='rsu4co2'.OR.nomvar=='rsucs4co2'.OR.nomvar=='rsd4co2'.OR. &
525        nomvar=='rsdcs4co2'.OR.nomvar=='rlu4co2'.OR.nomvar=='rlucs4co2'.OR.&
526        nomvar=='rld4co2'.OR.nomvar=='rldcs4co2') THEN
527        IF ( flag_var(iff)<=lev_files(iff) ) ok_4xCO2atm=.TRUE.
528    ENDIF
529  END SUBROUTINE histdef2d_old
530
531  SUBROUTINE histdef3d_old (iff,lpoint,flag_var,nomvar,titrevar,unitvar)
532
533    USE ioipsl, ONLY: histdef
534    USE dimphy, ONLY: klev
535    USE mod_phys_lmdz_para, ONLY: jj_nb
536    USE phys_output_var_mod, ONLY: type_ecri, zoutm, lev_files, nid_files, &
537                                   nhorim, zdtime_moy, levmin, levmax, &
538                                   nvertm, nfiles
539    USE mod_grid_phy_lmdz, ONLY : nbp_lon, nbp_lat
540    IMPLICIT NONE
541
542    INCLUDE "clesphys.h"
543
544    INTEGER                          :: iff
545    LOGICAL                          :: lpoint
546    INTEGER, DIMENSION(nfiles)       :: flag_var
547    CHARACTER(LEN=20)                 :: nomvar
548    CHARACTER(LEN=*)                 :: titrevar
549    CHARACTER(LEN=*)                 :: unitvar
550
551    REAL zstophym
552
553    ! Appel a la lecture des noms et niveau d'ecriture des variables dans output.def
554    CALL conf_physoutputs(nomvar,flag_var)
555
556    IF (type_ecri(iff)=='inst(X)'.OR.type_ecri(iff)=='once') THEN
557       zstophym=zoutm(iff)
558    ELSE
559       zstophym=zdtime_moy
560    ENDIF
561
562    IF(.NOT.lpoint) THEN
563       IF ( flag_var(iff)<=lev_files(iff) ) THEN
564          CALL histdef (nid_files(iff), nomvar, titrevar, unitvar, &
565               nbp_lon, jj_nb, nhorim(iff), klev, levmin(iff), &
566               levmax(iff)-levmin(iff)+1, nvertm(iff), 32, type_ecri(iff), &
567               zstophym, zoutm(iff))
568       ENDIF
569    ELSE
570       IF ( flag_var(iff)<=lev_files(iff) ) THEN
571          CALL histdef (nid_files(iff), nomvar, titrevar, unitvar, &
572               npstn,1,nhorim(iff), klev, levmin(iff), &
573               levmax(iff)-levmin(iff)+1, nvertm(iff), 32, &
574               type_ecri(iff), zstophym,zoutm(iff))
575       ENDIF
576    ENDIF
577  END SUBROUTINE histdef3d_old
578
579  SUBROUTINE histdef2d (iff,var)
580
581    USE ioipsl, ONLY: histdef
582    USE mod_phys_lmdz_para, ONLY: jj_nb
583    USE phys_output_var_mod, ONLY: ctrl_out, type_ecri_files, zoutm, zdtime_moy, &
584                                   clef_stations, phys_out_filenames, lev_files, &
585                                   nid_files, nhorim, swaerofree_diag, swaero_diag, dryaod_diag,&
586                                   ok_4xCO2atm
587    USE print_control_mod, ONLY: prt_level,lunout
588    USE mod_grid_phy_lmdz, ONLY : nbp_lon, nbp_lat
589    USE aero_mod, ONLY : naero_tot, name_aero_tau
590#ifdef CPP_XIOS
591    USE wxios, ONLY: wxios_add_field_to_file
592#endif
593    IMPLICIT NONE
594
595    INCLUDE "clesphys.h"
596
597    INTEGER                          :: iff
598    INTEGER                          :: naero
599    TYPE(ctrl_out)                   :: var
600
601    REAL zstophym
602    CHARACTER(LEN=20) :: typeecrit
603
604    ! ug On récupère le type écrit de la structure:
605    !       Assez moche, à refaire si meilleure méthode...
606    IF (INDEX(var%type_ecrit(iff), "once") > 0) THEN
607       typeecrit = 'once'
608    ELSE IF(INDEX(var%type_ecrit(iff), "t_min") > 0) THEN
609       typeecrit = 't_min(X)'
610    ELSE IF(INDEX(var%type_ecrit(iff), "t_max") > 0) THEN
611       typeecrit = 't_max(X)'
612    ELSE IF(INDEX(var%type_ecrit(iff), "inst") > 0) THEN
613       typeecrit = 'inst(X)'
614    ELSE
615       typeecrit = type_ecri_files(iff)
616    ENDIF
617
618    IF (typeecrit=='inst(X)'.OR.typeecrit=='once') THEN
619       zstophym=zoutm(iff)
620    ELSE
621       zstophym=zdtime_moy
622    ENDIF
623
624    ! Appel a la lecture des noms et niveau d'ecriture des variables dans output.def
625    CALL conf_physoutputs(var%name, var%flag)
626
627    IF(.NOT.clef_stations(iff)) THEN 
628
629#ifdef CPP_XIOS
630      IF (.not. ok_all_xml) THEN
631        IF ( var%flag(iff)<=lev_files(iff) ) THEN
632          CALL wxios_add_field_to_file(var%name, 2, iff, phys_out_filenames(iff), &
633          var%description, var%unit, var%flag(iff), typeecrit)
634          IF (prt_level >= 10) THEN
635            WRITE(lunout,*) 'histdef2d: call wxios_add_field_to_file var%name iff: ', &
636                            trim(var%name),iff
637          ENDIF
638        ENDIF
639      ENDIF
640#endif
641#ifndef CPP_IOIPSL_NO_OUTPUT
642
643       IF ( var%flag(iff)<=lev_files(iff) ) THEN
644          CALL histdef (nid_files(iff), var%name, var%description, var%unit, &
645               nbp_lon,jj_nb,nhorim(iff), 1,1,1, -99, 32, &
646               typeecrit, zstophym,zoutm(iff))               
647       ENDIF
648    ELSE
649       IF ( var%flag(iff)<=lev_files(iff)) THEN
650          CALL histdef (nid_files(iff), var%name, var%description, var%unit, &
651               npstn,1,nhorim(iff), 1,1,1, -99, 32, &
652               typeecrit, zstophym,zoutm(iff))               
653       ENDIF
654#endif
655    ENDIF
656
657    ! Set swaero_diag=true if at least one of the concerned variables are defined
658    !--OB 30/05/2016 use wider set of variables
659    IF ( var%name=='topswad' .OR. var%name=='topswad0' .OR. var%name=='solswad' .OR. var%name=='solswad0' .OR. &
660         var%name=='topswai' .OR. var%name=='solswai'  .OR. ( iflag_rrtm==1 .AND. (                            &
661         var%name=='toplwad' .OR. var%name=='toplwad0' .OR. var%name=='sollwad' .OR. var%name=='sollwad0' .OR. &
662         var%name=='toplwai' .OR. var%name=='sollwai'  ) ) ) THEN
663       IF  ( var%flag(iff)<=lev_files(iff) ) swaero_diag=.TRUE.
664    ENDIF
665
666    ! Set swaerofree_diag=true if at least one of the concerned variables are defined
667    IF (var%name=='SWupTOAcleanclr' .OR. var%name=='SWupSFCcleanclr' .OR. var%name=='SWdnSFCcleanclr' ) THEN
668       IF  ( var%flag(iff)<=lev_files(iff) ) swaerofree_diag=.TRUE.
669    ENDIF
670
671    ! set dryaod_dry=true if at least one of the concerned variables are defined
672    IF (var%name=='dryod550aer') THEN
673      IF  ( var%flag(iff)<=lev_files(iff) ) dryaod_diag=.TRUE.
674    ENDIF
675    !
676    DO naero = 1, naero_tot-1
677      IF (var%name=='dryod550_'//name_aero_tau(naero)) THEN
678        IF  ( var%flag(iff)<=lev_files(iff) ) dryaod_diag=.TRUE.
679      ENDIF
680    ENDDO
681    ! Set ok_4xCO2atm=true if at least one of the concerned variables are
682    ! defined
683    IF (var%name=='rsut4co2'.OR.var%name=='rlut4co2'.OR.var%name=='rsutcs4co2' &
684        .OR. var%name=='rlutcs4co2'.OR.var%name=='rsu4co2'.OR.var%name=='rsucs4co2' &
685        .OR.var%name=='rsu4co2'.OR.var%name=='rsucs4co2'.OR.var%name=='rsd4co2'.OR. &
686        var%name=='rsdcs4co2'.OR.var%name=='rlu4co2'.OR.var%name=='rlucs4co2'.OR.&
687        var%name=='rld4co2'.OR.var%name=='rldcs4co2') THEN
688        IF ( var%flag(iff)<=lev_files(iff) ) ok_4xCO2atm=.TRUE.
689    ENDIF
690  END SUBROUTINE histdef2d
691
692  SUBROUTINE histdef3d (iff,var)
693
694    USE ioipsl, ONLY: histdef
695    USE dimphy, ONLY: klev
696    USE mod_phys_lmdz_para, ONLY: jj_nb
697    USE phys_output_var_mod, ONLY: ctrl_out, type_ecri_files, zoutm, zdtime_moy, &
698                                   clef_stations, phys_out_filenames, lev_files, &
699                                   nid_files, nhorim, swaerofree_diag, swaero_diag, dryaod_diag, levmin, &
700                                   levmax, nvertm
701    USE print_control_mod, ONLY: prt_level,lunout
702    USE mod_grid_phy_lmdz, ONLY : nbp_lon, nbp_lat
703#ifdef CPP_XIOS
704    USE wxios, ONLY: wxios_add_field_to_file
705#endif
706    IMPLICIT NONE
707
708    INCLUDE "clesphys.h"
709
710    INTEGER                          :: iff
711    TYPE(ctrl_out)                   :: var
712
713    REAL zstophym
714    CHARACTER(LEN=20) :: typeecrit
715
716    ! ug On récupère le type écrit de la structure:
717    !       Assez moche, à refaire si meilleure méthode...
718    IF (INDEX(var%type_ecrit(iff), "once") > 0) THEN
719       typeecrit = 'once'
720    ELSE IF(INDEX(var%type_ecrit(iff), "t_min") > 0) THEN
721       typeecrit = 't_min(X)'
722    ELSE IF(INDEX(var%type_ecrit(iff), "t_max") > 0) THEN
723       typeecrit = 't_max(X)'
724    ELSE IF(INDEX(var%type_ecrit(iff), "inst") > 0) THEN
725       typeecrit = 'inst(X)'
726    ELSE
727       typeecrit = type_ecri_files(iff)
728    ENDIF
729
730
731    ! Appel a la lecture des noms et niveau d'ecriture des variables dans output.def
732    CALL conf_physoutputs(var%name,var%flag)
733
734    IF (typeecrit=='inst(X)'.OR.typeecrit=='once') THEN
735       zstophym=zoutm(iff)
736    ELSE
737       zstophym=zdtime_moy
738    ENDIF
739
740    IF(.NOT.clef_stations(iff)) THEN
741
742#ifdef CPP_XIOS
743       IF (.not. ok_all_xml) THEN
744         IF ( var%flag(iff)<=lev_files(iff) ) THEN
745         CALL wxios_add_field_to_file(var%name, 3, iff, phys_out_filenames(iff), &
746         var%description, var%unit, var%flag(iff), typeecrit)
747           IF (prt_level >= 10) THEN
748             WRITE(lunout,*) 'histdef3d: call wxios_add_field_to_file var%name iff: ', &
749                             trim(var%name),iff
750           ENDIF
751         ENDIF
752       ENDIF
753#endif
754#ifndef CPP_IOIPSL_NO_OUTPUT
755
756       IF ( var%flag(iff)<=lev_files(iff) ) THEN
757          CALL histdef (nid_files(iff), var%name, var%description, var%unit, &
758               nbp_lon, jj_nb, nhorim(iff), klev, levmin(iff), &
759               levmax(iff)-levmin(iff)+1, nvertm(iff), 32, typeecrit, &
760               zstophym, zoutm(iff))
761       ENDIF
762    ELSE
763       IF ( var%flag(iff)<=lev_files(iff)) THEN
764          CALL histdef (nid_files(iff), var%name, var%description, var%unit, &
765               npstn,1,nhorim(iff), klev, levmin(iff), &
766               levmax(iff)-levmin(iff)+1, nvertm(iff), 32, &
767               typeecrit, zstophym,zoutm(iff))
768       ENDIF
769#endif
770    ENDIF
771
772    ! Set swaerofree_diag=true if at least one of the concerned variables are defined
773    IF (var%name=='rsucsaf' .OR. var%name=='rsdcsaf') THEN
774       IF  ( var%flag(iff)<=lev_files(iff) ) swaerofree_diag=.TRUE.
775    ENDIF
776
777  END SUBROUTINE histdef3d
778
779  SUBROUTINE conf_physoutputs(nam_var,flag_var)
780!!! Lecture des noms et niveau de sortie des variables dans output.def
781    !   en utilisant les routines getin de IOIPSL 
782    USE ioipsl, ONLY: getin
783    USE phys_output_var_mod, ONLY: nfiles
784    USE print_control_mod, ONLY: prt_level,lunout
785    IMPLICIT NONE
786
787    CHARACTER(LEN=20)                :: nam_var
788    INTEGER, DIMENSION(nfiles)      :: flag_var
789
790    IF(prt_level>10) WRITE(lunout,*)'Avant getin: nam_var flag_var ',nam_var,flag_var(:)
791    CALL getin('flag_'//nam_var,flag_var)
792    CALL getin('name_'//nam_var,nam_var)
793    IF(prt_level>10) WRITE(lunout,*)'Apres getin: nam_var flag_var ',nam_var,flag_var(:)
794
795  END SUBROUTINE conf_physoutputs
796
797 
798  SUBROUTINE histwrite2d_phy_old(nid,lpoint,name,itau,field)
799  USE dimphy, ONLY: klon
800  USE mod_phys_lmdz_para, ONLY: Gather_omp, grid1Dto2D_mpi, &
801                                is_sequential, klon_mpi_begin, klon_mpi_end, &
802                                jj_nb, klon_mpi
803  USE ioipsl, ONLY: histwrite
804  USE print_control_mod, ONLY: prt_level,lunout
805  USE mod_grid_phy_lmdz, ONLY : nbp_lon, nbp_lat
806  IMPLICIT NONE
807   
808    INTEGER,INTENT(IN) :: nid
809    LOGICAL,INTENT(IN) :: lpoint
810    CHARACTER*(*), INTENT(IN) :: name
811    INTEGER, INTENT(IN) :: itau
812    REAL,DIMENSION(:),INTENT(IN) :: field
813    REAL,DIMENSION(klon_mpi) :: buffer_omp
814    INTEGER, allocatable, DIMENSION(:) :: index2d
815    REAL :: Field2d(nbp_lon,jj_nb)
816
817    INTEGER :: ip
818    REAL,ALLOCATABLE,DIMENSION(:) :: fieldok
819
820    IF (size(field)/=klon) CALL abort_physic('iophy::histwrite2d','Field first DIMENSION not equal to klon',1)
821   
822    CALL Gather_omp(field,buffer_omp)   
823!$OMP MASTER
824    CALL grid1Dto2D_mpi(buffer_omp,Field2d)
825    if(.NOT.lpoint) THEN
826     ALLOCATE(index2d(nbp_lon*jj_nb))
827     ALLOCATE(fieldok(nbp_lon*jj_nb))
828     IF (prt_level >= 10) write(lunout,*)'Sending ',name,' to IOIPSL'
829     CALL histwrite(nid,name,itau,Field2d,nbp_lon*jj_nb,index2d)
830     IF (prt_level >= 10) write(lunout,*)'Finished sending ',name,' to IOIPSL'
831    ELSE
832     ALLOCATE(fieldok(npstn))
833     ALLOCATE(index2d(npstn))
834
835     IF (is_sequential) THEN
836!     klon_mpi_begin=1
837!     klon_mpi_end=klon
838      DO ip=1, npstn
839       fieldok(ip)=buffer_omp(nptabij(ip))
840      ENDDO
841     ELSE
842      DO ip=1, npstn
843!     print*,'histwrite2d is_sequential npstn ip name nptabij',npstn,ip,name,nptabij(ip)
844       IF(nptabij(ip).GE.klon_mpi_begin.AND. &
845          nptabij(ip).LE.klon_mpi_end) THEN
846         fieldok(ip)=buffer_omp(nptabij(ip)-klon_mpi_begin+1)
847       ENDIF
848      ENDDO
849     ENDIF
850     IF (prt_level >= 10) write(lunout,*)'Sending ',name,' to IOIPSL'
851     CALL histwrite(nid,name,itau,fieldok,npstn,index2d)
852     IF (prt_level >= 10) write(lunout,*)'Finished sending ',name,' to IOIPSL'
853!
854    ENDIF
855    DEALLOCATE(index2d)
856    DEALLOCATE(fieldok)
857!$OMP END MASTER   
858
859 
860  END SUBROUTINE histwrite2d_phy_old
861
862  SUBROUTINE histwrite3d_phy_old(nid,lpoint,name,itau,field)
863  USE dimphy, ONLY: klon
864  USE mod_phys_lmdz_para, ONLY: Gather_omp, grid1Dto2D_mpi, &
865                                is_sequential, klon_mpi_begin, klon_mpi_end, &
866                                jj_nb, klon_mpi
867  USE mod_grid_phy_lmdz, ONLY : nbp_lon, nbp_lat
868  USE ioipsl, ONLY: histwrite
869  USE print_control_mod, ONLY: prt_level,lunout
870  IMPLICIT NONE
871   
872    INTEGER,INTENT(IN) :: nid
873    LOGICAL,INTENT(IN) :: lpoint
874    CHARACTER*(*), INTENT(IN) :: name
875    INTEGER, INTENT(IN) :: itau
876    REAL,DIMENSION(:,:),INTENT(IN) :: field  ! --> field(klon,:)
877    REAL,DIMENSION(klon_mpi,size(field,2)) :: buffer_omp
878    REAL :: Field3d(nbp_lon,jj_nb,size(field,2))
879    INTEGER :: ip, n, nlev
880    INTEGER, ALLOCATABLE, DIMENSION(:) :: index3d
881    REAL,allocatable, DIMENSION(:,:) :: fieldok
882
883
884    IF (size(field,1)/=klon) CALL abort_physic('iophy::histwrite3d','Field first DIMENSION not equal to klon',1)
885    nlev=size(field,2)
886
887    CALL Gather_omp(field,buffer_omp)
888!$OMP MASTER
889    CALL grid1Dto2D_mpi(buffer_omp,field3d)
890    IF (.NOT.lpoint) THEN
891     ALLOCATE(index3d(nbp_lon*jj_nb*nlev))
892     ALLOCATE(fieldok(nbp_lon*jj_nb,nlev))
893     IF (prt_level >= 10) write(lunout,*)'Sending ',name,' to IOIPSL'
894     CALL histwrite(nid,name,itau,Field3d,nbp_lon*jj_nb*nlev,index3d)
895     IF (prt_level >= 10) write(lunout,*)'Finished sending ',name,' to IOIPSL'
896   ELSE
897      nlev=size(field,2)
898      ALLOCATE(index3d(npstn*nlev))
899      ALLOCATE(fieldok(npstn,nlev))
900
901      IF (is_sequential) THEN
902!      klon_mpi_begin=1
903!      klon_mpi_end=klon
904       DO n=1, nlev
905       DO ip=1, npstn
906        fieldok(ip,n)=buffer_omp(nptabij(ip),n)
907       ENDDO
908       ENDDO
909      ELSE
910       DO n=1, nlev
911       DO ip=1, npstn
912        IF(nptabij(ip).GE.klon_mpi_begin.AND. &
913         nptabij(ip).LE.klon_mpi_end) THEN
914         fieldok(ip,n)=buffer_omp(nptabij(ip)-klon_mpi_begin+1,n)
915        ENDIF
916       ENDDO
917       ENDDO
918      ENDIF
919      IF (prt_level >= 10) write(lunout,*)'Sending ',name,' to IOIPSL'
920      CALL histwrite(nid,name,itau,fieldok,npstn*nlev,index3d)
921      IF (prt_level >= 10) write(lunout,*)'Finished sending ',name,' to IOIPSL'
922    ENDIF
923  DEALLOCATE(index3d)
924  DEALLOCATE(fieldok)
925!$OMP END MASTER   
926
927  END SUBROUTINE histwrite3d_phy_old
928
929
930
931
932! ug NOUVELLE VERSION DES WRITE AVEC LA BOUCLE DO RENTREE
933  SUBROUTINE histwrite2d_phy(var,field, STD_iff)
934  USE dimphy, ONLY: klon, klev
935  USE mod_phys_lmdz_para, ONLY: gather_omp, grid1dto2d_mpi, &
936                                jj_nb, klon_mpi, klon_mpi_begin, &
937                                klon_mpi_end, is_sequential
938  USE ioipsl, ONLY: histwrite
939  USE phys_output_var_mod, ONLY: ctrl_out, clef_files, lev_files, &
940                                 nfiles, vars_defined, clef_stations, &
941                                 nid_files
942  USE print_control_mod, ONLY: prt_level,lunout
943  USE mod_grid_phy_lmdz, ONLY : nbp_lon, nbp_lat
944#ifdef CPP_XIOS
945  USE xios, ONLY: xios_send_field
946#endif
947
948
949  IMPLICIT NONE
950  include 'clesphys.h'
951
952    TYPE(ctrl_out), INTENT(IN) :: var
953    REAL, DIMENSION(:), INTENT(IN) :: field
954    INTEGER, INTENT(IN), OPTIONAL :: STD_iff ! ug RUSTINE POUR LES STD LEVS.....
955     
956    INTEGER :: iff, iff_beg, iff_end
957    LOGICAL, SAVE  :: firstx
958!$OMP THREADPRIVATE(firstx)
959
960    REAL,DIMENSION(klon_mpi) :: buffer_omp
961    INTEGER, allocatable, DIMENSION(:) :: index2d
962    REAL :: Field2d(nbp_lon,jj_nb)
963
964    INTEGER :: ip
965    REAL, ALLOCATABLE, DIMENSION(:) :: fieldok
966
967    IF (prt_level >= 10) THEN
968      WRITE(lunout,*)'Begin histwrite2d_phy for ',trim(var%name)
969    ENDIF
970! ug RUSTINE POUR LES STD LEVS.....
971      IF (PRESENT(STD_iff)) THEN
972            iff_beg = STD_iff
973            iff_end = STD_iff
974      ELSE
975            iff_beg = 1
976            iff_end = nfiles
977      ENDIF
978
979  ! On regarde si on est dans la phase de définition ou d'écriture:
980  IF (.NOT.vars_defined) THEN
981!$OMP MASTER
982      !Si phase de définition.... on définit
983      IF (.not. ok_all_xml) THEN
984      IF (prt_level >= 10) THEN
985      WRITE (lunout,*)"histwrite2d_phy: .not.vars_defined ; time to define ", trim(var%name)
986      ENDIF
987      DO iff=iff_beg, iff_end
988         IF (clef_files(iff)) THEN
989            CALL histdef2d(iff, var)
990         ENDIF
991      ENDDO
992      ENDIF
993!$OMP END MASTER
994  ELSE
995
996    !Et sinon on.... écrit
997    IF (SIZE(field)/=klon .AND. SIZE(field)/=klev) CALL abort_physic('iophy::histwrite2d_phy','Field first DIMENSION not equal to klon/klev',1)   
998    IF (prt_level >= 10) THEn
999      WRITE (lunout,*)"histwrite2d_phy: .not.vars_defined ; time to gather and write ", trim(var%name)
1000    ENDIF
1001   
1002   
1003    IF (SIZE(field) == klon) then
1004        CALL Gather_omp(field,buffer_omp)
1005    ELSE
1006        buffer_omp(:)=0.
1007    ENDIF
1008!$OMP MASTER
1009    CALL grid1Dto2D_mpi(buffer_omp,Field2d)
1010
1011! La boucle sur les fichiers:
1012      firstx=.true.
1013
1014      IF (ok_all_xml) THEN
1015#ifdef CPP_XIOS
1016          IF (prt_level >= 10) THEN
1017             write(lunout,*)'Dans iophy histwrite2D,var%name ', trim(var%name)                       
1018          ENDIF
1019          IF (SIZE(field) == klon) then
1020              CALL xios_send_field(var%name, Field2d)
1021          ELSE
1022             CALL xios_send_field(var%name, field)
1023          ENDIF
1024          IF (prt_level >= 10) THEN
1025             WRITE (lunout,*)'Dans iophy histwrite2D,var%name apres xios_send ', trim(var%name)                       
1026          ENDIF
1027#else
1028        CALL abort_physic ('iophy','cannot have ok_all_xml = .T. without CPP_XIOS defined' ,1)
1029#endif
1030      ELSE 
1031        DO iff=iff_beg, iff_end
1032            IF (var%flag(iff) <= lev_files(iff) .AND. clef_files(iff)) THEN
1033
1034#ifdef CPP_XIOS
1035               IF (firstx) THEN
1036                  IF (prt_level >= 10) THEN
1037                     WRITE (lunout,*)'Dans iophy histwrite2D,iff,var%name ', iff,trim(var%name)                       
1038                     WRITE (lunout,*)"histwrite2d_phy:.NOT.clef_stations(iff)and iff==iff_beg, call xios_send_field"
1039                  ENDIF
1040                  IF (SIZE(field) == klon) then
1041                     CALL xios_send_field(var%name, Field2d)
1042                  ELSE
1043                     CALL xios_send_field(var%name, field)
1044                  ENDIF
1045                  firstx=.false.
1046               ENDIF
1047#endif
1048
1049                  IF (.NOT.clef_stations(iff)) THEN
1050                        ALLOCATE(index2d(nbp_lon*jj_nb))
1051                        ALLOCATE(fieldok(nbp_lon*jj_nb))
1052#ifndef CPP_IOIPSL_NO_OUTPUT
1053                        CALL histwrite(nid_files(iff),var%name,itau_iophy,Field2d,nbp_lon*jj_nb,index2d)
1054#endif
1055!#ifdef CPP_XIOS
1056!                        IF (iff == iff_beg) THEN
1057!                          if (prt_level >= 10) then
1058!                            write(lunout,*)"histwrite2d_phy: .NOT.clef_stations(iff) and iff==iff_beg, call xios_send_field"
1059!                          endif
1060!                          CALL xios_send_field(var%name, Field2d)
1061!                        ENDIF
1062!#endif
1063                  ELSE
1064                        ALLOCATE(fieldok(npstn))
1065                        ALLOCATE(index2d(npstn))
1066
1067                        IF (is_sequential) THEN
1068                          DO ip=1, npstn
1069                            fieldok(ip)=buffer_omp(nptabij(ip))
1070                          ENDDO
1071                        ELSE
1072                              DO ip=1, npstn
1073                                write(lunout,*)'histwrite2d_phy is_sequential npstn ip namenptabij',npstn,ip,var%name,nptabij(ip)
1074                                     IF(nptabij(ip).GE.klon_mpi_begin.AND. &
1075                                        nptabij(ip).LE.klon_mpi_end) THEN
1076                                       fieldok(ip)=buffer_omp(nptabij(ip)-klon_mpi_begin+1)
1077                                     ENDIF
1078                              ENDDO
1079                       ENDIF ! of IF (is_sequential)
1080#ifndef CPP_IOIPSL_NO_OUTPUT
1081                       IF (prt_level >= 10) THEn
1082                         write(lunout,*)"histwrite2d_phy: clef_stations(iff) and iff==iff_beg, call wxios_write_2D"
1083                       ENDIF
1084                       CALL histwrite(nid_files(iff),var%name,itau_iophy,fieldok,npstn,index2d)
1085#endif
1086                  ENDIF ! of IF(.NOT.clef_stations(iff))
1087                 
1088                DEALLOCATE(index2d)
1089                DEALLOCATE(fieldok)
1090            ENDIF !levfiles
1091        ENDDO ! of DO iff=iff_beg, iff_end
1092      ENDIF
1093!$OMP END MASTER   
1094  ENDIF ! vars_defined
1095  IF (prt_level >= 10) WRITE(lunout,*)'End histwrite2d_phy ',trim(var%name)
1096  END SUBROUTINE histwrite2d_phy
1097
1098
1099! ug NOUVELLE VERSION DES WRITE AVEC LA BOUCLE DO RENTREE
1100  SUBROUTINE histwrite3d_phy(var, field, STD_iff)
1101  USE dimphy, ONLY: klon, klev
1102  USE mod_phys_lmdz_para, ONLY: gather_omp, grid1dto2d_mpi, &
1103                                jj_nb, klon_mpi, klon_mpi_begin, &
1104                                klon_mpi_end, is_sequential
1105  USE ioipsl, ONLY: histwrite
1106  USE phys_output_var_mod, ONLY: ctrl_out, clef_files, lev_files, &
1107                                 nfiles, vars_defined, clef_stations, &
1108                                 nid_files
1109  USE mod_grid_phy_lmdz, ONLY : nbp_lon, nbp_lat
1110#ifdef CPP_XIOS
1111  USE xios, ONLY: xios_send_field
1112#endif
1113  USE print_control_mod, ONLY: prt_level,lunout
1114
1115  IMPLICIT NONE
1116  include 'clesphys.h'
1117
1118    TYPE(ctrl_out), INTENT(IN) :: var
1119    REAL, DIMENSION(:,:), INTENT(IN) :: field ! --> field(klon,:)
1120    INTEGER, INTENT(IN), OPTIONAL :: STD_iff ! ug RUSTINE POUR LES STD LEVS.....
1121     
1122    INTEGER :: iff, iff_beg, iff_end
1123    LOGICAL, SAVE  :: firstx
1124!$OMP THREADPRIVATE(firstx)
1125    REAL,DIMENSION(klon_mpi,SIZE(field,2)) :: buffer_omp
1126    REAL :: Field3d(nbp_lon,jj_nb,SIZE(field,2))
1127    INTEGER :: ip, n, nlev, nlevx
1128    INTEGER, ALLOCATABLE, DIMENSION(:) :: index3d
1129    REAL,ALLOCATABLE, DIMENSION(:,:) :: fieldok
1130
1131  IF (prt_level >= 10) write(lunout,*)'Begin histrwrite3d ',var%name
1132
1133! ug RUSTINE POUR LES STD LEVS.....
1134      IF (PRESENT(STD_iff)) THEN
1135            iff_beg = STD_iff
1136            iff_end = STD_iff
1137      ELSE
1138            iff_beg = 1
1139            iff_end = nfiles
1140      ENDIF
1141
1142  ! On regarde si on est dans la phase de définition ou d'écriture:
1143  IF(.NOT.vars_defined) THEN
1144      !Si phase de définition.... on définit
1145!$OMP MASTER
1146      DO iff=iff_beg, iff_end
1147        IF (clef_files(iff)) THEN
1148          CALL histdef3d(iff, var)
1149        ENDIF
1150      ENDDO
1151!$OMP END MASTER
1152  ELSE
1153    !Et sinon on.... écrit
1154
1155    IF (SIZE(field,1)/=klon .AND. SIZE(field,1)/=klev) CALL abort_physic('iophy::histwrite3d_xios','Field first DIMENSION not equal to klon/klev',1)
1156
1157    nlev=SIZE(field,2)
1158    IF (nlev.EQ.klev+1) THEN
1159        nlevx=klev
1160    ELSE
1161        nlevx=nlev
1162    ENDIF
1163
1164    IF (SIZE(field,1) == klon) then
1165        CALL Gather_omp(field,buffer_omp)
1166    ELSE
1167        buffer_omp(:,:)=0.
1168    ENDIF
1169!$OMP MASTER
1170    CALL grid1Dto2D_mpi(buffer_omp,field3d)
1171
1172
1173! BOUCLE SUR LES FICHIERS
1174     firstx=.true.
1175
1176      IF (ok_all_xml) THEN
1177#ifdef CPP_XIOS
1178          IF (prt_level >= 10) THEN
1179             write(lunout,*)'Dans iophy histwrite3D,var%name ',&
1180                             trim(var%name)                       
1181          ENDIF
1182          IF (SIZE(field,1) == klon) then
1183             CALL xios_send_field(var%name, Field3d(:,:,1:nlevx))
1184          ELSE
1185             CALL xios_send_field(var%name, field)
1186          ENDIF
1187#else
1188        CALL abort_physic ('iophy','cannot have ok_all_xml = .T. without CPP_XIOS defined' ,1)
1189#endif
1190      ELSE 
1191
1192
1193     DO iff=iff_beg, iff_end
1194            IF (var%flag(iff) <= lev_files(iff) .AND. clef_files(iff)) THEN
1195#ifdef CPP_XIOS
1196              IF (firstx) THEN
1197                IF (prt_level >= 10) THEn
1198                  WRITE (lunout,*)'Dans iophy, histwrite3D iff nlev klev firstx', &
1199                                  iff,nlev,klev, firstx                       
1200                  WRITE (lunout,*)'histwrite3d_phy: call xios_send_field for ', &
1201                                  trim(var%name), ' with iim jjm nlevx = ', &
1202                                  nbp_lon,jj_nb,nlevx
1203                ENDIF
1204                IF (SIZE(field,1) == klon) then
1205                    CALL xios_send_field(var%name, Field3d(:,:,1:nlevx))
1206                ELSE
1207                     CALL xios_send_field(var%name, field)
1208                ENDIF
1209                firstx=.false.
1210              ENDIF
1211#endif
1212                IF (.NOT.clef_stations(iff)) THEN
1213                        ALLOCATE(index3d(nbp_lon*jj_nb*nlev))
1214                        ALLOCATE(fieldok(nbp_lon*jj_nb,nlev))
1215
1216#ifndef CPP_IOIPSL_NO_OUTPUT
1217                        CALL histwrite(nid_files(iff),var%name,itau_iophy,Field3d,nbp_lon*jj_nb*nlev,index3d)
1218#endif
1219
1220!#ifdef CPP_XIOS
1221!                        IF (iff == 1) THEN
1222!                              CALL xios_send_field(var%name, Field3d(:,:,1:klev))
1223!                        ENDIF
1224!#endif
1225!                       
1226                ELSE
1227                        nlev=size(field,2)
1228                        ALLOCATE(index3d(npstn*nlev))
1229                        ALLOCATE(fieldok(npstn,nlev))
1230
1231                        IF (is_sequential) THEN
1232                              DO n=1, nlev
1233                                    DO ip=1, npstn
1234                                          fieldok(ip,n)=buffer_omp(nptabij(ip),n)
1235                                    ENDDO
1236                              ENDDO
1237                        ELSE
1238                              DO n=1, nlev
1239                                    DO ip=1, npstn
1240                                                IF(nptabij(ip).GE.klon_mpi_begin.AND. &
1241                                                      nptabij(ip).LE.klon_mpi_end) THEN
1242                                                fieldok(ip,n)=buffer_omp(nptabij(ip)-klon_mpi_begin+1,n)
1243                                          ENDIF
1244                                    ENDDO
1245                              ENDDO
1246                        ENDIF
1247#ifndef CPP_IOIPSL_NO_OUTPUT
1248                        CALL histwrite(nid_files(iff),var%name,itau_iophy,fieldok,npstn*nlev,index3d)
1249#endif
1250                  ENDIF
1251                  DEALLOCATE(index3d)
1252                  DEALLOCATE(fieldok)
1253            ENDIF
1254      ENDDO
1255      ENDIF
1256!$OMP END MASTER   
1257  ENDIF ! vars_defined
1258  IF (prt_level >= 10) write(lunout,*)'End histrwrite3d ',var%name
1259  END SUBROUTINE histwrite3d_phy
1260 
1261
1262! VERSION DES HISTWRITE DEDIEES AU TOUT-XIOS-XML DEJA UTILISEE DANS PHYDEV
1263#ifdef CPP_XIOS
1264  SUBROUTINE histwrite2d_xios(field_name,field)
1265  USE dimphy, ONLY: klon, klev
1266  USE mod_phys_lmdz_para, ONLY: gather_omp, grid1Dto2D_mpi, &
1267                                is_sequential, klon_mpi_begin, klon_mpi_end, &
1268                                jj_nb, klon_mpi
1269  USE mod_grid_phy_lmdz, ONLY : nbp_lon, nbp_lat
1270  USE xios, ONLY: xios_send_field
1271  USE print_control_mod, ONLY: prt_level,lunout
1272
1273  IMPLICIT NONE
1274
1275    CHARACTER(LEN=*), INTENT(IN) :: field_name
1276    REAL, DIMENSION(:), INTENT(IN) :: field
1277     
1278    REAL,DIMENSION(klon_mpi) :: buffer_omp
1279    INTEGER, allocatable, DIMENSION(:) :: index2d
1280    REAL :: Field2d(nbp_lon,jj_nb)
1281
1282    INTEGER :: ip
1283    REAL, ALLOCATABLE, DIMENSION(:) :: fieldok
1284
1285    IF (prt_level >= 10) WRITE(lunout,*)'Begin histrwrite2d_xios ',field_name
1286
1287    !Et sinon on.... écrit
1288    IF (SIZE(field)/=klon .AND. SIZE(field)/=klev) CALL abort_physic('iophy::histwrite2d_xios','Field first DIMENSION not equal to klon/klev',1)
1289   
1290    IF (SIZE(field) == klev) then
1291!$OMP MASTER
1292        CALL xios_send_field(field_name,field)
1293!$OMP END MASTER   
1294    ELSE
1295        CALL Gather_omp(field,buffer_omp)   
1296!$OMP MASTER
1297        CALL grid1Dto2D_mpi(buffer_omp,Field2d)
1298   
1299!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
1300!ATTENTION, STATIONS PAS GEREES !
1301!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
1302    !IF(.NOT.clef_stations(iff)) THEN
1303        IF (.TRUE.) THEN
1304            ALLOCATE(index2d(nbp_lon*jj_nb))
1305            ALLOCATE(fieldok(nbp_lon*jj_nb))
1306   
1307   
1308            CALL xios_send_field(field_name, Field2d)
1309   
1310        ELSE
1311            ALLOCATE(fieldok(npstn))
1312            ALLOCATE(index2d(npstn))
1313   
1314            IF (is_sequential) THEN
1315                DO ip=1, npstn
1316                    fieldok(ip)=buffer_omp(nptabij(ip))
1317                ENDDO
1318            ELSE
1319                DO ip=1, npstn
1320                    PRINT*,'histwrite2d_xios is_sequential npstn ip namenptabij',npstn,ip,field_name,nptabij(ip)
1321                    IF(nptabij(ip).GE.klon_mpi_begin.AND. &
1322                    nptabij(ip).LE.klon_mpi_end) THEN
1323                        fieldok(ip)=buffer_omp(nptabij(ip)-klon_mpi_begin+1)
1324                    ENDIF
1325                ENDDO
1326            ENDIF
1327   
1328        ENDIF
1329                 
1330        DEALLOCATE(index2d)
1331        DEALLOCATE(fieldok)
1332!$OMP END MASTER   
1333    ENDIF
1334
1335  IF (prt_level >= 10) WRITE(lunout,*)'End histrwrite2d_xios ',field_name
1336  END SUBROUTINE histwrite2d_xios
1337
1338
1339! ug NOUVELLE VERSION DES WRITE AVEC LA BOUCLE DO RENTREE
1340  SUBROUTINE histwrite3d_xios(field_name, field)
1341  USE dimphy, ONLY: klon, klev
1342  USE mod_phys_lmdz_para, ONLY: gather_omp, grid1Dto2D_mpi, &
1343                                is_sequential, klon_mpi_begin, klon_mpi_end, &
1344                                jj_nb, klon_mpi
1345  USE xios, ONLY: xios_send_field
1346  USE mod_grid_phy_lmdz, ONLY : nbp_lon, nbp_lat
1347  USE print_control_mod, ONLY: prt_level,lunout
1348
1349  IMPLICIT NONE
1350
1351    CHARACTER(LEN=*), INTENT(IN) :: field_name
1352    REAL, DIMENSION(:,:), INTENT(IN) :: field ! --> field(klon,:)
1353
1354    REAL,DIMENSION(klon_mpi,SIZE(field,2)) :: buffer_omp
1355    REAL :: Field3d(nbp_lon,jj_nb,SIZE(field,2))
1356    INTEGER :: ip, n, nlev
1357    INTEGER, ALLOCATABLE, DIMENSION(:) :: index3d
1358    REAL,ALLOCATABLE, DIMENSION(:,:) :: fieldok
1359
1360  IF (prt_level >= 10) write(lunout,*)'Begin histrwrite3d_xios ',field_name
1361
1362    !Et on.... écrit
1363    IF (SIZE(field,1)/=klon .AND. SIZE(field,1)/=klev) CALL abort_physic('iophy::histwrite3d_xios','Field first DIMENSION not equal to klon/klev',1)
1364   
1365    IF (SIZE(field,1) == klev) then
1366!$OMP MASTER
1367        CALL xios_send_field(field_name,field)
1368!$OMP END MASTER   
1369    ELSE
1370        nlev=SIZE(field,2)
1371
1372
1373        CALL Gather_omp(field,buffer_omp)
1374!$OMP MASTER
1375        CALL grid1Dto2D_mpi(buffer_omp,field3d)
1376
1377!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
1378!ATTENTION, STATIONS PAS GEREES !
1379!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
1380    !IF (.NOT.clef_stations(iff)) THEN
1381        IF(.TRUE.)THEN
1382            ALLOCATE(index3d(nbp_lon*jj_nb*nlev))
1383            ALLOCATE(fieldok(nbp_lon*jj_nb,nlev))
1384            CALL xios_send_field(field_name, Field3d(:,:,1:nlev))
1385                           
1386        ELSE
1387            nlev=size(field,2)
1388            ALLOCATE(index3d(npstn*nlev))
1389            ALLOCATE(fieldok(npstn,nlev))
1390   
1391            IF (is_sequential) THEN
1392                DO n=1, nlev
1393                    DO ip=1, npstn
1394                        fieldok(ip,n)=buffer_omp(nptabij(ip),n)
1395                    ENDDO
1396                ENDDO
1397            ELSE
1398                DO n=1, nlev
1399                    DO ip=1, npstn
1400                        IF(nptabij(ip).GE.klon_mpi_begin.AND. &
1401                        nptabij(ip).LE.klon_mpi_end) THEN
1402                            fieldok(ip,n)=buffer_omp(nptabij(ip)-klon_mpi_begin+1,n)
1403                        ENDIF
1404                    ENDDO
1405                ENDDO
1406            ENDIF
1407        ENDIF
1408        DEALLOCATE(index3d)
1409        DEALLOCATE(fieldok)
1410!$OMP END MASTER   
1411    ENDIF
1412
1413  IF (prt_level >= 10) write(lunout,*)'End histrwrite3d_xios ',field_name
1414  END SUBROUTINE histwrite3d_xios
1415
1416#ifdef CPP_XIOS
1417  SUBROUTINE histwrite0d_xios(field_name, field)
1418  USE xios, ONLY: xios_send_field
1419  IMPLICIT NONE
1420
1421    CHARACTER(LEN=*), INTENT(IN) :: field_name
1422    REAL, INTENT(IN) :: field ! --> scalar
1423
1424!$OMP MASTER
1425   CALL xios_send_field(field_name, field)
1426!$OMP END MASTER
1427
1428  END SUBROUTINE histwrite0d_xios
1429#endif
1430
1431#endif
1432end module iophy
Note: See TracBrowser for help on using the repository browser.