source: LMDZ6/branches/Ocean_skin/libf/misc/wxios.F90 @ 4110

Last change on this file since 4110 was 3605, checked in by lguez, 5 years ago

Merge revisions 3427:3600 of trunk into branch Ocean_skin

  • 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
File size: 25.2 KB
RevLine 
[2509]1! $Id$
[1825]2#ifdef CPP_XIOS
3MODULE wxios
4    USE xios
5    USE iaxis
6    USE iaxis_attr
7    USE icontext_attr
8    USE idate
9    USE idomain_attr
10    USE ifield_attr
11    USE ifile_attr
12    USE ixml_tree
13
14    !Variables disponibles pendant toute l'execution du programme:
15   
16    INTEGER, SAVE :: g_comm
[3605]17    CHARACTER(len=100), SAVE :: g_ctx_name ="LMDZ"
[1825]18    TYPE(xios_context), SAVE :: g_ctx
19!$OMP THREADPRIVATE(g_comm,g_cts_name,g_ctx)
[1852]20    LOGICAL, SAVE :: g_flag_xml = .FALSE.
21    CHARACTER(len=100) :: g_field_name = "nofield"
22!$OMP THREADPRIVATE(g_flag_xml,g_field_name)
[2271]23    REAL :: missing_val_omp
24    REAL :: missing_val
25!$OMP THREADPRIVATE(missing_val)
[1825]26
[2741]27#ifdef XIOS1
28#error "XIOS v1 no longer supported, use XIOS v2."
29#endif
30
[1825]31    CONTAINS
32   
33    !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
34    !   36day => 36d etc     !!!!!!!!!!!!!!!!!!!!
35    !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
36   
37    SUBROUTINE reformadate(odate, ndate)
38        CHARACTER(len=*), INTENT(IN) :: odate
[2509]39        TYPE(xios_duration) :: ndate
[1825]40       
41        INTEGER :: i = 0
[1852]42         !!!!!!!!!!!!!!!!!!
43         ! Pour XIOS:
44         !  year : y
45         !  month : mo
46         !  day : d
47         !  hour : h
48         !  minute : mi
49         !  second : s
50         !!!!!!!!!!!!!!!!!!
51
[1825]52        i = INDEX(odate, "day")
53        IF (i > 0) THEN
[2509]54            read(odate(1:i-1),*) ndate%day
[1825]55        END IF
[1852]56
57        i = INDEX(odate, "hr")
58        IF (i > 0) THEN
[2509]59            read(odate(1:i-1),*) ndate%hour
[1852]60        END IF
61
62        i = INDEX(odate, "mth")
63        IF (i > 0) THEN
[2509]64            read(odate(1:i-1),*) ndate%month
[1852]65        END IF
[1825]66       
[1852]67        !IF (prt_level >= 10) WRITE(lunout,*) "Xios. ", odate, " => ", ndate
[1825]68    END SUBROUTINE reformadate
69   
70    !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
71    !   ave(X) => average etc     !!!!!!!!!!!!!!!
72    !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
73   
74    CHARACTER(len=7) FUNCTION reformaop(op)
75        CHARACTER(len=*), INTENT(IN) :: op
76       
77        INTEGER :: i = 0
78        reformaop = "average"
79       
80        IF (op.EQ."inst(X)") THEN
81            reformaop = "instant"
82        END IF
83       
84        IF (op.EQ."once") THEN
85            reformaop = "once"
86        END IF
87       
88        IF (op.EQ."t_max(X)") THEN
89            reformaop = "maximum"
90        END IF
91       
92        IF (op.EQ."t_min(X)") THEN
93            reformaop = "minimum"
94        END IF
95       
[1852]96        !IF (prt_level >= 10) WRITE(lunout,*) "Xios. ", op, " => ", reformaop
[1825]97    END FUNCTION reformaop
98
99    !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
100    ! Routine d'initialisation      !!!!!!!!!!!!!
101    !     A lancer juste après mpi_init !!!!!!!!!!!!!
102    !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
103
[2055]104    SUBROUTINE wxios_init(xios_ctx_name, locom, outcom, type_ocean)
[2509]105        USE print_control_mod, ONLY : prt_level, lunout
[1852]106        IMPLICIT NONE
107
[1825]108      CHARACTER(len=*), INTENT(IN) :: xios_ctx_name
[1852]109      INTEGER, INTENT(IN), OPTIONAL :: locom
110      INTEGER, INTENT(OUT), OPTIONAL :: outcom
[2055]111      CHARACTER(len=6), INTENT(IN), OPTIONAL :: type_ocean
[1852]112
113   
[1825]114        TYPE(xios_context) :: xios_ctx
[1852]115        INTEGER :: xios_comm
[1825]116
[1897]117        IF (prt_level >= 10) WRITE(lunout,*) "wxios_init: Initialization"
[1852]118
119
120
121        IF (PRESENT(locom)) THEN
[1897]122          CALL xios_initialize(xios_ctx_name, local_comm = locom, return_comm = xios_comm )
123          IF (prt_level >= 10) WRITE(lunout,*) "wxios_init: ctx=",xios_ctx_name," local_comm=",locom,", return_comm=",xios_comm
[1852]124        ELSE
[1897]125          CALL xios_initialize(xios_ctx_name, return_comm = xios_comm )
126          IF (prt_level >= 10) WRITE(lunout,*) "wxios_init: ctx=",xios_ctx_name," return_comm=",xios_comm
[1852]127        END IF
[1825]128       
[1852]129        IF (PRESENT(outcom)) THEN
[1897]130          outcom = xios_comm
131          IF (prt_level >= 10) WRITE(lunout,*) "wxios_init: ctx=",xios_ctx_name," outcom=",outcom
[1852]132        END IF
[1825]133       
134        !Enregistrement des variables globales:
135        g_comm = xios_comm
136        g_ctx_name = xios_ctx_name
137       
[3605]138!        ! Si couple alors init fait dans cpl_init
139!        IF (.not. PRESENT(type_ocean)) THEN
140!            CALL wxios_context_init()
141!        ENDIF
[2054]142
[1825]143    END SUBROUTINE wxios_init
144
[1852]145    SUBROUTINE wxios_context_init()
[2509]146        USE print_control_mod, ONLY : prt_level, lunout
[3605]147        USE mod_phys_lmdz_mpi_data, ONLY : COMM_LMDZ_PHY
[1852]148        IMPLICIT NONE
149
150        TYPE(xios_context) :: xios_ctx
151
[2509]152!$OMP MASTER
[1852]153        !Initialisation du contexte:
[3605]154        !!CALL xios_context_initialize(g_ctx_name, g_comm)
155        CALL xios_context_initialize(g_ctx_name, COMM_LMDZ_PHY)
[1852]156        CALL xios_get_handle(g_ctx_name, xios_ctx)    !Récupération
157        CALL xios_set_current_context(xios_ctx)            !Activation
158        g_ctx = xios_ctx
159
[2001]160        IF (prt_level >= 10) THEN
161          WRITE(lunout,*) "wxios_context_init: Current context is ",trim(g_ctx_name)
162          WRITE(lunout,*) "     now call xios_solve_inheritance()"
163        ENDIF
[1852]164        !Une première analyse des héritages:
165        CALL xios_solve_inheritance()
[2509]166!$OMP END MASTER
[1852]167    END SUBROUTINE wxios_context_init
168
[3605]169
170    SUBROUTINE wxios_set_context()
171        IMPLICIT NONE
172        TYPE(xios_context) :: xios_ctx
173
174       !$OMP MASTER
175        CALL xios_get_handle(g_ctx_name, xios_ctx)    !Récupération
176        CALL xios_set_current_context(xios_ctx)            !Activation
177       !$OMP END MASTER
178
179    END SUBROUTINE wxios_set_context
180
[1825]181    !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
182    ! Routine de paramétrisation !!!!!!!!!!!!!!!!!!
183    !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
184
[2095]185    SUBROUTINE wxios_set_cal(pasdetemps, calendrier, annee, mois, jour, heure, ini_an, ini_mois, ini_jour, ini_heure)
[2509]186        USE print_control_mod, ONLY : prt_level, lunout
[1852]187        IMPLICIT NONE
188
[1825]189     !Paramètres:
190     CHARACTER(len=*), INTENT(IN) :: calendrier
[2095]191     INTEGER, INTENT(IN) :: annee, mois, jour, ini_an, ini_mois, ini_jour
192     REAL, INTENT(IN) :: pasdetemps, heure, ini_heure
[1825]193     
194     !Variables:
195     CHARACTER(len=80) :: abort_message
196     CHARACTER(len=19) :: date
197     INTEGER :: njour = 1
198     
199     !Variables pour xios:
[2509]200     TYPE(xios_duration) :: mdtime
[1825]201     !REAL(kind = 8) :: year=0, month=0, day=0, hour=0, minute=0, second=0
202     
[2509]203        mdtime%second=pasdetemps
[1825]204
205        !Réglage du calendrier:
206        SELECT CASE (calendrier)
207            CASE('earth_360d')
[2509]208                CALL xios_define_calendar("D360")
209                IF (prt_level >= 10) WRITE(lunout,*) 'wxios_set_cal: Calendrier terrestre a 360 jours/an'
210            CASE('earth_365d')
211                CALL xios_define_calendar("NoLeap")
212                IF (prt_level >= 10) WRITE(lunout,*) 'wxios_set_cal: Calendrier terrestre a 365 jours/an'
213            CASE('gregorian')
214                CALL xios_define_calendar("Gregorian")
215                IF (prt_level >= 10) WRITE(lunout,*) 'wxios_set_cal: Calendrier gregorien'
216            CASE DEFAULT
217                abort_message = 'wxios_set_cal: Mauvais choix de calendrier'
[3605]218                CALL abort_physic('Gcm:Xios',abort_message,1)
[2509]219        END SELECT
[1825]220       
[2095]221        !Formatage de la date d'origine:
[2509]222        WRITE(date, "(i4.4,'-',i2.2,'-',i2.2,' ',i2.2,':00:00')") annee, mois, jour, int(heure)
[1825]223       
[2095]224        IF (prt_level >= 10) WRITE(lunout,*) "wxios_set_cal: Time origin: ", date
[2509]225        CALL xios_set_time_origin(xios_date(annee,mois,jour,int(heure),0,0))
[2095]226
227        !Formatage de la date de debut:
228
229        WRITE(date, "(i4.4,'-',i2.2,'-',i2.2,' ',i2.2,':00:00')") ini_an, ini_mois, ini_jour, int(ini_heure)
[1825]230       
[2095]231        IF (prt_level >= 10) WRITE(lunout,*) "wxios_set_cal: Start date: ", date
232       
[2509]233        CALL xios_set_start_date(xios_date(ini_an,ini_mois,ini_jour,int(ini_heure),0,0))
[2095]234       
[1825]235        !Et enfin,le pas de temps:
236        CALL xios_set_timestep(mdtime)
[1897]237        IF (prt_level >= 10) WRITE(lunout,*) "wxios_set_cal: ts=",mdtime
[1825]238    END SUBROUTINE wxios_set_cal
239
[1852]240    SUBROUTINE wxios_set_timestep(ts)
241        REAL, INTENT(IN) :: ts
[2509]242        TYPE(xios_duration) :: mdtime     
[1852]243
[2509]244        mdtime%timestep = ts
245
[1852]246        CALL xios_set_timestep(mdtime)
247    END SUBROUTINE wxios_set_timestep
248
[1825]249    !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
250    ! Pour initialiser un domaine !!!!!!!!!!!!!!!!!!!!
251    !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
[3605]252    SUBROUTINE wxios_domain_param(dom_id)
253       USE dimphy, only: klon
254       USE mod_phys_lmdz_transfert_para, ONLY: gather, bcast
255       USE mod_phys_lmdz_para, only: jj_nb, jj_begin, jj_end, ii_begin, ii_end, &
256                                     mpi_size, mpi_rank, klon_mpi, &
257                                     is_sequential, is_south_pole_dyn
258       USE mod_grid_phy_lmdz, only: nbp_lon, nbp_lat, klon_glo         
259       USE print_control_mod, ONLY : prt_level, lunout
260       USE geometry_mod
[1897]261
[3605]262       IMPLICIT NONE
263        CHARACTER(len=*),INTENT(IN) :: dom_id ! domain identifier
[1852]264
[3605]265        REAL   :: rlat_glo(klon_glo)
266        REAL   :: rlon_glo(klon_glo)
267        REAL   :: io_lat(nbp_lat)
268        REAL   :: io_lon(nbp_lon)
269        LOGICAL :: mask(nbp_lon,jj_nb) !Masque pour les problèmes de recouvrement MPI
[1825]270        TYPE(xios_domain) :: dom
[3605]271        INTEGER :: i
[1852]272        LOGICAL :: boool
[1825]273       
[3605]274
275
276        CALL gather(latitude_deg,rlat_glo)
277        CALL bcast(rlat_glo)
278        CALL gather(longitude_deg,rlon_glo)
279        CALL bcast(rlon_glo)
280   
281  !$OMP MASTER 
282        io_lat(1)=rlat_glo(1)
283        io_lat(nbp_lat)=rlat_glo(klon_glo)
284        IF ((nbp_lon*nbp_lat) > 1) then
285          DO i=2,nbp_lat-1
286            io_lat(i)=rlat_glo(2+(i-2)*nbp_lon)
287          ENDDO
288        ENDIF
289
290        IF (klon_glo == 1) THEN
291          io_lon(1)=rlon_glo(1)
292        ELSE
293          io_lon(1:nbp_lon)=rlon_glo(2:nbp_lon+1)
294        ENDIF
295
[1825]296       
297        !On récupère le handle:
298        CALL xios_get_domain_handle(dom_id, dom)
299       
300        !On parametrise le domaine:
[3605]301        CALL xios_set_domain_attr_hdl(dom, ni_glo=nbp_lon, ibegin=0, ni=nbp_lon, type="rectilinear")
302        CALL xios_set_domain_attr_hdl(dom, nj_glo=nbp_lat, jbegin=jj_begin-1, nj=jj_nb, data_dim=2)
303        CALL xios_set_domain_attr_hdl(dom, lonvalue_1d=io_lon(1:nbp_lon), latvalue_1d=io_lat(jj_begin:jj_end))
304        CALL xios_set_domain_attr("dom_out", domain_ref=dom_id)
305
[3165]306        !On definit un axe de latitudes pour les moyennes zonales
307        IF (xios_is_valid_axis("axis_lat")) THEN
[3605]308           CALL xios_set_axis_attr( "axis_lat", n_glo=nbp_lat, n=jj_nb, begin=jj_begin-1, value=io_lat(jj_begin:jj_end))
[3165]309        ENDIF
310
[1852]311        IF (.NOT.is_sequential) THEN
312            mask(:,:)=.TRUE.
[1897]313            if (ii_begin>1) mask(1:ii_begin-1,1) = .FALSE.
[3605]314            if (ii_end<nbp_lon) mask(ii_end+1:nbp_lon,jj_nb) = .FALSE.
[1897]315            ! special case for south pole
[3605]316            if ((ii_end==1).and.(is_south_pole_dyn)) mask(1:nbp_lon,jj_nb)=.true.
[1897]317            IF (prt_level >= 10) THEN
318              WRITE(lunout,*) "wxios_domain_param: mpirank=",mpi_rank," mask(:,1)=",mask(:,1)
[3605]319              WRITE(lunout,*) "wxios_domain_param: mpirank=",mpi_rank," mask(:,jj_nb)=",mask(:,jj_nb)
[1897]320            ENDIF
[2509]321            CALL xios_set_domain_attr_hdl(dom, mask_2d=mask)
[1852]322        END IF
323
[1825]324         CALL xios_is_defined_domain_attr_hdl(dom,ni_glo=boool)
325        !Vérification:
326        IF (xios_is_valid_domain(dom_id)) THEN
[1897]327            IF (prt_level >= 10) WRITE(lunout,*) "wxios_domain_param: Domain initialized: ", trim(dom_id), boool
[1825]328        ELSE
[1897]329            IF (prt_level >= 10) WRITE(lunout,*) "wxios_domain_param: Invalid domain: ", trim(dom_id)
[1825]330        END IF
[3605]331!$OMP END MASTER
332       
[1825]333    END SUBROUTINE wxios_domain_param
334   
[3605]335
336    SUBROUTINE wxios_domain_param_unstructured(dom_id)
337        USE geometry_mod, ONLY : longitude, latitude, boundslon, boundslat,ind_cell_glo
338        USE mod_grid_phy_lmdz, ONLY : nvertex, klon_glo
339        USE mod_phys_lmdz_para
340        USE nrtype, ONLY : PI
341        USE ioipsl_getin_p_mod, ONLY : getin_p
342        IMPLICIT NONE
343        CHARACTER(len=*),INTENT(IN) :: dom_id ! domain identifier
344        REAL :: lon_mpi(klon_mpi)
345        REAL :: lat_mpi(klon_mpi)
346        REAL :: boundslon_mpi(klon_mpi,nvertex)
347        REAL :: boundslat_mpi(klon_mpi,nvertex)
348        INTEGER :: ind_cell_glo_mpi(klon_mpi)
349        TYPE(xios_domain) :: dom
350       
351        LOGICAL :: remap_output
352
353        CALL gather_omp(longitude*180/PI,lon_mpi)
354        CALL gather_omp(latitude*180/PI,lat_mpi)
355        CALL gather_omp(boundslon*180/PI,boundslon_mpi)
356        CALL gather_omp(boundslat*180/PI,boundslat_mpi)
357        CALL gather_omp(ind_cell_glo,ind_cell_glo_mpi)
358       
359        remap_output=.TRUE.
360        CALL getin_p("remap_output",remap_output)
361
362!$OMP MASTER
363        CALL xios_get_domain_handle(dom_id, dom)
364       
365        !On parametrise le domaine:
366        CALL xios_set_attr(dom, ni_glo=klon_glo, ibegin=ij_begin-1, ni=ij_nb, type="unstructured")
367        CALL xios_set_attr(dom, nvertex=nvertex, lonvalue_1d=lon_mpi, latvalue_1d=lat_mpi, &
368                           bounds_lon_1d=TRANSPOSE(boundslon_mpi), bounds_lat_1d=TRANSPOSE(boundslat_mpi) )
369        CALL xios_set_attr(dom, i_index=ind_cell_glo_mpi(:)-1)
370        IF (remap_output) THEN
371          CALL xios_set_domain_attr("dom_out", domain_ref="dom_regular")
372          CALL xios_set_fieldgroup_attr("remap_expr", expr="@this_ref")
373          CALL xios_set_fieldgroup_attr("remap_1800s", freq_op=xios_duration_convert_from_string("1800s"))
374          CALL xios_set_fieldgroup_attr("remap_1h",    freq_op=xios_duration_convert_from_string("1h"))
375          CALL xios_set_fieldgroup_attr("remap_3h",    freq_op=xios_duration_convert_from_string("3h"))
376          CALL xios_set_fieldgroup_attr("remap_6h",    freq_op=xios_duration_convert_from_string("6h"))
377          CALL xios_set_fieldgroup_attr("remap_1d",    freq_op=xios_duration_convert_from_string("1d"))
378          CALL xios_set_fieldgroup_attr("remap_1mo",   freq_op=xios_duration_convert_from_string("1mo"))
379        ENDIF
380!$OMP END MASTER
381
382    END SUBROUTINE wxios_domain_param_unstructured
383
384
385
386
[1825]387    !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
388    ! Pour déclarer un axe vertical !!!!!!!!!!!!!!!
389    !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
[3003]390    SUBROUTINE wxios_add_vaxis(axis_id, axis_size, axis_value,     &
391                               positif, bnds)
[2509]392        USE print_control_mod, ONLY : prt_level, lunout
[1852]393        IMPLICIT NONE
394
[2002]395        CHARACTER (len=*), INTENT(IN) :: axis_id
[1852]396        INTEGER, INTENT(IN) :: axis_size
[1825]397        REAL, DIMENSION(axis_size), INTENT(IN) :: axis_value
[3003]398        CHARACTER (len=*), INTENT(IN), OPTIONAL :: positif
399        REAL, DIMENSION(axis_size, 2), INTENT(IN), OPTIONAL :: bnds
[1825]400       
[2002]401!        TYPE(xios_axisgroup) :: axgroup
402!        TYPE(xios_axis) :: ax
403!        CHARACTER(len=50) :: axis_id
[1825]404       
[2002]405!        IF (len_trim(axisgroup_id).gt.len(axis_id)) THEN
406!          WRITE(lunout,*) "wxios_add_vaxis: error, size of axis_id too small!!"
407!          WRITE(lunout,*) "     increase it to at least ",len_trim(axisgroup_id)
408!          CALL abort_gcm("wxios_add_vaxis","len(axis_id) too small",1)
409!        ENDIF
410!        axis_id=trim(axisgroup_id)
[1825]411       
412        !On récupère le groupe d'axes qui va bien:
[2002]413        !CALL xios_get_axisgroup_handle(axisgroup_id, axgroup)
[1825]414       
415        !On ajoute l'axe correspondant à ce fichier:
[2002]416        !CALL xios_add_axis(axgroup, ax, TRIM(ADJUSTL(axis_id)))
[1825]417       
418        !Et on le parametrise:
[2002]419        !CALL xios_set_axis_attr_hdl(ax, size=axis_size, value=axis_value)
[1825]420       
[2002]421        ! Ehouarn: New way to declare axis, without axis_group:
[3003]422        if (PRESENT(positif) .AND. PRESENT(bnds)) then
423          CALL xios_set_axis_attr(trim(axis_id),n_glo=axis_size,value=axis_value, &
424                                  positive=positif, bounds=bnds)
425        else if (PRESENT(positif)) then
426          CALL xios_set_axis_attr(trim(axis_id),n_glo=axis_size,value=axis_value, &
427                                  positive=positif)
428        else if (PRESENT(bnds)) then
429          CALL xios_set_axis_attr(trim(axis_id),n_glo=axis_size,value=axis_value, &
430                                  bounds=bnds)
431        else
432          CALL xios_set_axis_attr(trim(axis_id),n_glo=axis_size,value=axis_value)
433        endif
[2741]434
[1825]435        !Vérification:
[1852]436        IF (xios_is_valid_axis(TRIM(ADJUSTL(axis_id)))) THEN
[1897]437            IF (prt_level >= 10) WRITE(lunout,*) "wxios_add_vaxis: Axis created: ", TRIM(ADJUSTL(axis_id))
[1825]438        ELSE
[2001]439            WRITE(lunout,*) "wxios_add_vaxis: Invalid axis: ", TRIM(ADJUSTL(axis_id))
[1825]440        END IF
441
442    END SUBROUTINE wxios_add_vaxis
443   
444   
445    !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
446    ! Pour déclarer un fichier  !!!!!!!!!!!!!!!!!!!
447    !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
448    SUBROUTINE wxios_add_file(fname, ffreq, flvl)
[2509]449        USE print_control_mod, ONLY : prt_level, lunout
[1852]450        IMPLICIT NONE
451
[1825]452        CHARACTER(len=*), INTENT(IN) :: fname
453        CHARACTER(len=*), INTENT(IN) :: ffreq
454        INTEGER, INTENT(IN) :: flvl
455       
456        TYPE(xios_file) :: x_file
457        TYPE(xios_filegroup) :: x_fg
[2509]458        TYPE(xios_duration) :: nffreq
[1825]459       
[1852]460        !On regarde si le fichier n'est pas défini par XML:
461        IF (.NOT.xios_is_valid_file(fname)) THEN
462            !On créé le noeud:
463            CALL xios_get_filegroup_handle("defile", x_fg)
464            CALL xios_add_file(x_fg, x_file, fname)
[1825]465       
[1852]466            !On reformate la fréquence:
467            CALL reformadate(ffreq, nffreq)
[1825]468       
[1852]469            !On configure:
470            CALL xios_set_file_attr_hdl(x_file, name="X"//fname,&
[2509]471                output_freq=nffreq, output_level=flvl, enabled=.TRUE.)
472       
[1852]473            IF (xios_is_valid_file("X"//fname)) THEN
[2001]474                IF (prt_level >= 10) THEN
475                  WRITE(lunout,*) "wxios_add_file: New file: ", "X"//fname
[2509]476                  WRITE(lunout,*) "wxios_add_file: output_freq=",nffreq,"; output_lvl=",flvl
[2001]477                ENDIF
[1852]478            ELSE
[2001]479                WRITE(lunout,*) "wxios_add_file: Error, invalid file: ", "X"//trim(fname)
[2509]480                WRITE(lunout,*) "wxios_add_file: output_freq=",nffreq,"; output_lvl=",flvl
[1852]481            END IF
[1825]482        ELSE
[2001]483            IF (prt_level >= 10) THEN
484              WRITE(lunout,*) "wxios_add_file: File ",trim(fname), " défined using XML."
485            ENDIF
486            ! Ehouarn: add an enable=.true. on top of xml definitions... why???
487            CALL xios_set_file_attr(fname, enabled=.TRUE.)
[1825]488        END IF
489    END SUBROUTINE wxios_add_file
490   
491    !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
[1852]492    ! Pour créer un champ      !!!!!!!!!!!!!!!!!!!!
[1825]493    !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
494    SUBROUTINE wxios_add_field(fieldname, fieldgroup, fieldlongname, fieldunit)
[1897]495        USE netcdf, only: nf90_fill_real
[1852]496
497        IMPLICIT NONE
498        INCLUDE 'iniprint.h'
[1825]499       
500        CHARACTER(len=*), INTENT(IN) :: fieldname
501        TYPE(xios_fieldgroup), INTENT(IN) :: fieldgroup
502        CHARACTER(len=*), INTENT(IN) :: fieldlongname
503        CHARACTER(len=*), INTENT(IN) :: fieldunit
504       
505        TYPE(xios_field) :: field
506        CHARACTER(len=10) :: newunit
507        REAL(KIND=8) :: def
508       
509        !La valeur par défaut des champs non définis:
510        def = nf90_fill_real
511       
512        IF (fieldunit .EQ. " ") THEN
513            newunit = "-"
514        ELSE
515            newunit = fieldunit
516        ENDIF
517       
518        !On ajoute le champ:
519        CALL xios_add_field(fieldgroup, field, fieldname)
[1897]520        !IF (prt_level >= 10) WRITE(lunout,*) "wxios_add_field: ",fieldname,fieldgroup, fieldlongname, fieldunit
[1825]521       
522        !On rentre ses paramètres:
523        CALL xios_set_field_attr_hdl(field, standard_name=fieldlongname, unit=newunit, default_value=def)
[1897]524        IF (prt_level >= 10) WRITE(lunout,*) "wxios_add_field: Field ",trim(fieldname), "cree:"
525        IF (prt_level >= 10) WRITE(lunout,*) "wxios_add_field: long_name=",trim(fieldlongname),"; unit=",trim(newunit),";  default_value=",nf90_fill_real
[1825]526
527    END SUBROUTINE wxios_add_field
528   
529    !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
[1852]530    ! Pour déclarer un champ      !!!!!!!!!!!!!!!!!
[1825]531    !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
[2137]532    SUBROUTINE wxios_add_field_to_file(fieldname, fdim, fid, fname, fieldlongname, fieldunit, field_level, op, nam_axvert)
[2509]533        USE print_control_mod, ONLY : prt_level, lunout
[1852]534        IMPLICIT NONE
535
[1825]536        CHARACTER(len=*), INTENT(IN) :: fieldname
537        INTEGER, INTENT(IN)          :: fdim, fid
538        CHARACTER(len=*), INTENT(IN) :: fname
539        CHARACTER(len=*), INTENT(IN) :: fieldlongname
540        CHARACTER(len=*), INTENT(IN) :: fieldunit
541        INTEGER, INTENT(IN)          :: field_level
542        CHARACTER(len=*), INTENT(IN) :: op
543       
[2001]544        CHARACTER(len=20) :: axis_id ! Ehouarn: dangerous...
[2137]545        CHARACTER(len=20), INTENT(IN), OPTIONAL :: nam_axvert
[1825]546        CHARACTER(len=100) :: operation
547        TYPE(xios_file) :: f
548        TYPE(xios_field) :: field
549        TYPE(xios_fieldgroup) :: fieldgroup
[2509]550        TYPE(xios_duration) :: freq_op
[2741]551
[1852]552        LOGICAL :: bool=.FALSE.
553        INTEGER :: lvl =0
[1825]554       
555       
[2001]556        ! Ajout Abd pour NMC:
557        IF (fid.LE.6) THEN
558          axis_id="presnivs"
559        ELSE
560          axis_id="plev"
561        ENDIF
[2137]562 
563        IF (PRESENT(nam_axvert)) THEN
564           axis_id=nam_axvert
565           print*,'nam_axvert=',axis_id
566        ENDIF
[1825]567       
568        !on prépare le nom de l'opération:
569        operation = reformaop(op)
570       
571       
572        !On selectionne le bon groupe de champs:
573        IF (fdim.EQ.2) THEN
[2001]574          CALL xios_get_fieldgroup_handle("fields_2D", fieldgroup)
[1825]575        ELSE
576          CALL xios_get_fieldgroup_handle("fields_3D", fieldgroup)
577        ENDIF
578       
579        !On regarde si le champ à déjà été créé ou non:
[1852]580        IF (xios_is_valid_field(fieldname) .AND. .NOT. g_field_name == fieldname) THEN
581            !Si ce champ existe via XML (ie, dès le premier passage, ie g_field_name != fieldname) alors rien d'autre à faire
[1897]582            IF (prt_level >= 10) WRITE(lunout,*) "wxios_add_field_to_file: Field ", trim(fieldname), "exists via XML"
[1852]583            g_flag_xml = .TRUE.
584            g_field_name = fieldname
585
586        ELSE IF (.NOT. g_field_name == fieldname) THEN
587            !Si premier pssage et champ indéfini, alors on le créé
588
[1897]589            IF (prt_level >= 10) WRITE(lunout,*) "wxios_add_field_to_file: Field ", trim(fieldname), "does not exist"
[1825]590           
591            !On le créé:
592            CALL wxios_add_field(fieldname,  fieldgroup, fieldlongname, fieldunit)
593            IF (xios_is_valid_field(fieldname)) THEN
[1897]594                IF (prt_level >= 10) WRITE(lunout,*) "wxios_add_field_to_file: Field ", trim(fieldname), "created"
[1825]595            ENDIF
596
[1852]597            g_flag_xml = .FALSE.
598            g_field_name = fieldname
599
600        END IF
601
602        IF (.NOT. g_flag_xml) THEN
603            !Champ existe déjà, mais pas XML, alors on l'ajoute
604            !On ajoute le champ:
605            CALL xios_get_file_handle(fname, f)
606            CALL xios_add_fieldtofile(f, field)
607           
608           
609            !L'operation, sa frequence:
[2509]610            freq_op%timestep=1
611            CALL xios_set_field_attr_hdl(field, field_ref=fieldname, operation=TRIM(ADJUSTL(operation)), freq_op=freq_op, prec=4)
[1852]612
613           
614            !On rentre ses paramètres:
615            CALL xios_set_field_attr_hdl(field, level=field_level, enabled=.TRUE.)
616           
617            IF (fdim.EQ.2) THEN
618                !Si c'est un champ 2D:
[1897]619                IF (prt_level >= 10) THEN
620                  WRITE(lunout,*) "wxios_add_field_to_file: 2D Field ", trim(fieldname), " in ", "X"//trim(fname) ," configured with:"
621                  WRITE(lunout,*) "wxios_add_field_to_file: op=", TRIM(ADJUSTL(operation))
622                  WRITE(lunout,*) "wxios_add_field_to_file: freq_op=1ts","; lvl=",field_level
623                ENDIF
[1852]624            ELSE
625                !Si 3D :
626                !On ajoute l'axe vertical qui va bien:
627                CALL xios_set_field_attr_hdl(field, axis_ref=TRIM(ADJUSTL(axis_id)))
628               
[1897]629                IF (prt_level >= 10) THEN
630                  WRITE(lunout,*) "wxios_add_field_to_file: 3D Field",trim(fieldname), " in ", "X"//trim(fname), "configured with:"
631                  WRITE(lunout,*) "wxios_add_field_to_file: freq_op=1ts","; lvl=",field_level
632                  WRITE(lunout,*) "wxios_add_field_to_file: axis=",TRIM(ADJUSTL(axis_id))
633                ENDIF
[1852]634            END IF
[1825]635       
636        ELSE
[1852]637            !Sinon on se contente de l'activer:
638            CALL xios_set_field_attr(fieldname, enabled=.TRUE.)
[2002]639            !NB: This will override an enable=.false. set by a user in the xml file;
640            !   then the only way to not output the field is by changing its
641            !   output level
[1852]642        ENDIF       
[1825]643       
644    END SUBROUTINE wxios_add_field_to_file
645   
[2002]646!    SUBROUTINE wxios_update_calendar(ito)
647!        INTEGER, INTENT(IN) :: ito
648!        CALL xios_update_calendar(ito)
649!    END SUBROUTINE wxios_update_calendar
650!   
651!    SUBROUTINE wxios_write_2D(fieldname, fdata)
652!        CHARACTER(len=*), INTENT(IN) :: fieldname
653!        REAL, DIMENSION(:,:), INTENT(IN) :: fdata
654!
655!        CALL xios_send_field(fieldname, fdata)
656!    END SUBROUTINE wxios_write_2D
[1825]657   
[2002]658!    SUBROUTINE wxios_write_3D(fieldname, fdata)
659!        CHARACTER(len=*), INTENT(IN) :: fieldname
660!        REAL, DIMENSION(:,:,:), INTENT(IN) :: fdata
661!       
662!        CALL xios_send_field(fieldname, fdata)
663!    END SUBROUTINE wxios_write_3D
[1825]664   
665    SUBROUTINE wxios_closedef()
666        CALL xios_close_context_definition()
[2095]667!        CALL xios_update_calendar(0)
[1825]668    END SUBROUTINE wxios_closedef
669   
670    SUBROUTINE wxios_close()
671        CALL xios_context_finalize()
672         CALL xios_finalize()
673     END SUBROUTINE wxios_close
674END MODULE wxios
675#endif
Note: See TracBrowser for help on using the repository browser.