source: LMDZ5/trunk/libf/bibio/wxios.F90 @ 1838

Last change on this file since 1838 was 1825, checked in by Ehouarn Millour, 11 years ago

Première étape de l'implémentation de XIOS. Modifications isolées dans des flags CPP_XIOS. Sorties opérationnelles (sauf stations et régionalisation) en modes séquentiel et omp, pas mpi.
UG
...........................................
First step of the XIOS implementation. Modifications are confined into CPP_XIOS flags. Output is operationnal (except for stations and regionalization) in sequential and omp modes (not mpi).
UG
File size: 14.4 KB
Line 
1! $Id: wxios.F90 $
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
17    CHARACTER(len=100), SAVE :: g_ctx_name
18    TYPE(xios_context), SAVE :: g_ctx
19!$OMP THREADPRIVATE(g_comm,g_cts_name,g_ctx)
20
21    CONTAINS
22   
23    !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
24    !   str + i   =>   str_i   !!!!!!!!!!!!!!!!!!!!
25    !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
26   
27    SUBROUTINE concat(str, i, str_i) !MAX i = 99
28        CHARACTER(len=*), INTENT(IN) :: str
29        INTEGER, INTENT(IN) :: i
30        CHARACTER(len=100), INTENT(OUT) :: str_i
31       
32       
33        !INT -> CHAR:
34        CHARACTER(len=10) :: num
35        WRITE(num, "(I5)") i
36        str_i = TRIM(ADJUSTL(str//"_"//TRIM(ADJUSTL(num))))
37       
38    END SUBROUTINE concat
39   
40    !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
41    !   36day => 36d etc     !!!!!!!!!!!!!!!!!!!!
42    !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
43   
44    SUBROUTINE reformadate(odate, ndate)
45        CHARACTER(len=*), INTENT(IN) :: odate
46        CHARACTER(len=100), INTENT(OUT) :: ndate
47       
48        INTEGER :: i = 0
49       
50        i = INDEX(odate, "day")
51        IF (i > 0) THEN
52            ndate = odate(1:i-1)//"d"
53        ELSE
54            i = INDEX(odate, "hr")
55            IF (i > 0) THEN
56                ndate = odate(1:i-1)//"h"
57            ELSE
58                ndate = odate
59            END IF
60        END IF
61       
62        !WRITE(*,*) "Xios. ", odate, " => ", ndate
63    END SUBROUTINE reformadate
64   
65    !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
66    !   ave(X) => average etc     !!!!!!!!!!!!!!!
67    !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
68   
69    CHARACTER(len=7) FUNCTION reformaop(op)
70        CHARACTER(len=*), INTENT(IN) :: op
71       
72        INTEGER :: i = 0
73        reformaop = "average"
74       
75        IF (op.EQ."inst(X)") THEN
76            reformaop = "instant"
77        END IF
78       
79        IF (op.EQ."once") THEN
80            reformaop = "once"
81        END IF
82       
83        IF (op.EQ."t_max(X)") THEN
84            reformaop = "maximum"
85        END IF
86       
87        IF (op.EQ."t_min(X)") THEN
88            reformaop = "minimum"
89        END IF
90       
91        !WRITE(*,*) "Xios. ", op, " => ", reformaop
92    END FUNCTION reformaop
93
94    !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
95    ! Routine d'initialisation      !!!!!!!!!!!!!
96    !     A lancer juste après mpi_init !!!!!!!!!!!!!
97    !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
98
99    SUBROUTINE wxios_init(xios_ctx_name)
100      CHARACTER(len=*), INTENT(IN) :: xios_ctx_name
101       
102      INTEGER :: xios_comm
103        TYPE(xios_context) :: xios_ctx
104     
105        WRITE(*,*) "Xios. Initialization"
106
107        !Lancement de xios:
108        CALL xios_initialize(xios_ctx_name, return_comm = xios_comm )
109       
110        !Initialisation du contexte:
111        CALL xios_context_initialize(xios_ctx_name, xios_comm)
112        CALL xios_get_handle(xios_ctx_name, xios_ctx)    !Récupération
113        CALL xios_set_current_context(xios_ctx)            !Activation
114       
115        !Enregistrement des variables globales:
116        g_comm = xios_comm
117        g_ctx_name = xios_ctx_name
118        g_ctx = xios_ctx
119       
120        WRITE(*,*) "Xios. Current context is ", xios_ctx_name
121    END SUBROUTINE wxios_init
122
123    !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
124    ! Routine de paramétrisation !!!!!!!!!!!!!!!!!!
125    !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
126
127    SUBROUTINE wxios_set_cal(pasdetemps, calendrier, annee, mois, jour, heure)
128     !Paramètres:
129     CHARACTER(len=*), INTENT(IN) :: calendrier
130     INTEGER, INTENT(IN) :: annee, mois, jour
131     REAL, INTENT(IN) :: pasdetemps, heure
132     
133     !Variables:
134     CHARACTER(len=80) :: abort_message
135     CHARACTER(len=19) :: date
136     INTEGER :: njour = 1
137     
138     !Variables pour xios:
139     TYPE(xios_time) :: mdtime
140     !REAL(kind = 8) :: year=0, month=0, day=0, hour=0, minute=0, second=0
141     
142        mdtime = xios_time(0, 0, 0, 0, 0, pasdetemps)
143
144        !Réglage du calendrier:
145        SELECT CASE (calendrier)
146            CASE('earth_360d')
147                CALL xios_set_context_attr_hdl(g_ctx, calendar_type= "D360")
148                WRITE(*,*) 'Xios. Calendrier terrestre a 360 jours/an'
149            CASE('earth_365d')
150                CALL xios_set_context_attr_hdl(g_ctx, calendar_type= "NoLeap")
151                WRITE(*,*) 'Xios. Calendrier terrestre a 365 jours/an'
152            CASE('earth_366d')
153                CALL xios_set_context_attr_hdl(g_ctx, calendar_type= "Gregorian")
154                WRITE(*,*) 'Xios. Calendrier gregorien'
155            CASE DEFAULT
156                abort_message = 'Xios. Mauvais choix de calendrier'
157                CALL abort_gcm('Gcm:Xios',abort_message,1)
158        END SELECT
159       
160        !Formatage de la date de départ:
161        WRITE(date, "(i4.4,'-',i2.2,'-',i2.2,' 00:00:00')") annee, mois, jour
162       
163        WRITE(*,*) "Xios. Initial time: ", date
164       
165        CALL xios_set_context_attr_hdl(g_ctx, start_date= date)
166       
167        !Et enfin,le pas de temps:
168        CALL xios_set_timestep(mdtime)
169        WRITE(*,*) "Xios. ts=",mdtime
170    END SUBROUTINE wxios_set_cal
171
172    !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
173    ! Pour initialiser un domaine !!!!!!!!!!!!!!!!!!!!
174    !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
175    SUBROUTINE wxios_domain_param(dom_id, is_sequential, iim, jjm, io_lat, io_lon)
176         
177        CHARACTER (len=*), INTENT(IN) :: dom_id
178        LOGICAL, INTENT(IN) :: is_sequential
179        INTEGER, INTENT(IN) :: iim, jjm
180        REAL, DIMENSION(:) :: io_lat, io_lon
181       
182       
183        TYPE(xios_domain) :: dom
184        INTEGER :: ni, nj, ni_glo, nj_glo, ibegin, iend, jbegin, jend
185        LOGICAl :: boool
186       
187        ni_glo = iim
188        nj_glo = jjm
189        ni = iim
190        nj = jjm
191        ibegin = 1
192        jbegin = 1
193        iend = ibegin + ni - 1
194        jend = jbegin + nj - 1
195       
196        !On récupère le handle:
197        CALL xios_get_domain_handle(dom_id, dom)
198       
199        WRITE(*,*) "Xios. ni:",iim," ni_glo:", iim, " nj:", jjm, " nj_glo:", jjm
200        WRITE(*,*) "Xios. Size lon:", SIZE(io_lon), " lat:", SIZE(io_lat)
201       
202        !On parametrise le domaine:
203        !IF (is_sequential) THEN
204            CALL xios_set_domain_attr_hdl(dom, ni_glo=iim, ibegin=1, ni=iim,&
205            & nj_glo=jjm, jbegin=1,nj=jjm,&
206            & lonvalue=io_lon(ibegin:iend), latvalue=io_lat(jbegin:jend))
207        !END IF
208         CALL xios_is_defined_domain_attr_hdl(dom,ni_glo=boool)
209        !Vérification:
210        IF (xios_is_valid_domain(dom_id)) THEN
211            WRITE(*,*) "Xios. Domain initialized: ", dom_id, boool
212        ELSE
213            WRITE(*,*) "Xios. Invalid domain: ", dom_id
214        END IF
215    END SUBROUTINE wxios_domain_param
216   
217    !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
218    ! Pour déclarer un axe vertical !!!!!!!!!!!!!!!
219    !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
220    SUBROUTINE wxios_add_vaxis(axisgroup_id, axis_file_id, axis_size, axis_value)
221        CHARACTER (len=*), INTENT(IN) :: axisgroup_id
222        INTEGER, INTENT(IN) :: axis_file_id, axis_size
223        REAL, DIMENSION(axis_size), INTENT(IN) :: axis_value
224       
225        TYPE(xios_axisgroup) :: axgroup
226        TYPE(xios_axis) :: ax
227        CHARACTER(len=100) :: axis_id
228       
229       
230        !Préparation du nom de l'axe:
231        CALL concat(axisgroup_id, axis_file_id, axis_id)
232       
233        !On récupère le groupe d'axes qui va bien:
234        CALL xios_get_axisgroup_handle(axisgroup_id, axgroup)
235       
236        !On ajoute l'axe correspondant à ce fichier:
237        CALL xios_add_axis(axgroup, ax, axis_id)
238       
239        !Et on le parametrise:
240        CALL xios_set_axis_attr_hdl(ax, size=axis_size, value=axis_value)
241       
242        !Vérification:
243        IF (xios_is_valid_axis(axis_id)) THEN
244            WRITE(*,*) "Xios. Axis created: ", axis_id
245        ELSE
246            WRITE(*,*) "Xios. Invalid axis: ", axis_id
247        END IF
248
249    END SUBROUTINE wxios_add_vaxis
250   
251   
252    !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
253    ! Pour déclarer un fichier  !!!!!!!!!!!!!!!!!!!
254    !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
255    SUBROUTINE wxios_add_file(fname, ffreq, flvl)
256        CHARACTER(len=*), INTENT(IN) :: fname
257        CHARACTER(len=*), INTENT(IN) :: ffreq
258        INTEGER, INTENT(IN) :: flvl
259       
260        TYPE(xios_file) :: x_file
261        TYPE(xios_filegroup) :: x_fg
262        CHARACTER(len=100) :: nffreq
263       
264        !On créé le noeud:
265        CALL xios_get_filegroup_handle("defile", x_fg)
266        CALL xios_add_file(x_fg, x_file, "X"//fname)
267       
268        !On reformate la fréquence:
269        CALL reformadate(ffreq, nffreq)
270       
271        !On configure:
272        CALL xios_set_file_attr_hdl(x_file, name="X"//fname,&
273                output_freq=TRIM(ADJUSTL(nffreq)), output_level=flvl, enabled=.TRUE.)
274       
275        IF (xios_is_valid_file("X"//fname)) THEN
276            WRITE(*,*) "Xios. New file: ", "X"//fname
277            WRITE(*,*) "Xios. output_freq=",TRIM(ADJUSTL(nffreq)),"; output_lvl=",flvl
278        ELSE
279            WRITE(*,*) "Xios. Error, invalid file: ", "X"//fname
280            WRITE(*,*) "Xios. output_freq=",TRIM(ADJUSTL(nffreq)),"; output_lvl=",flvl
281        END IF
282    END SUBROUTINE wxios_add_file
283   
284    !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
285    ! Pour créer un champ      !!!!!!!!!!!!!!!
286    !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
287    SUBROUTINE wxios_add_field(fieldname, fieldgroup, fieldlongname, fieldunit)
288        USE netcdf
289       
290        CHARACTER(len=*), INTENT(IN) :: fieldname
291        TYPE(xios_fieldgroup), INTENT(IN) :: fieldgroup
292        CHARACTER(len=*), INTENT(IN) :: fieldlongname
293        CHARACTER(len=*), INTENT(IN) :: fieldunit
294       
295        TYPE(xios_field) :: field
296        CHARACTER(len=10) :: newunit
297        REAL(KIND=8) :: def
298       
299        !La valeur par défaut des champs non définis:
300        def = nf90_fill_real
301       
302        IF (fieldunit .EQ. " ") THEN
303            newunit = "-"
304        ELSE
305            newunit = fieldunit
306        ENDIF
307       
308        !On ajoute le champ:
309        CALL xios_add_field(fieldgroup, field, fieldname)
310        !WRITE(*,*) "Xios. ",fieldname,fieldgroup, fieldlongname, fieldunit
311       
312        !On rentre ses paramètres:
313        CALL xios_set_field_attr_hdl(field, standard_name=fieldlongname, unit=newunit, default_value=def)
314        WRITE(*,*) "Xios. Champ ", fieldname, "cree:"
315        WRITE(*,*) "Xios. long_name=",fieldlongname,"; unit=",newunit,";  default_value=",nf90_fill_real
316
317    END SUBROUTINE wxios_add_field
318   
319    !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
320    ! Pour déclarer un champ      !!!!!!!!!!!!!!!
321    !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
322    SUBROUTINE wxios_add_field_to_file(fieldname, fdim, fid, fname, fieldlongname, fieldunit, field_level, op)
323        CHARACTER(len=*), INTENT(IN) :: fieldname
324        INTEGER, INTENT(IN)          :: fdim, fid
325        CHARACTER(len=*), INTENT(IN) :: fname
326        CHARACTER(len=*), INTENT(IN) :: fieldlongname
327        CHARACTER(len=*), INTENT(IN) :: fieldunit
328        INTEGER, INTENT(IN)          :: field_level
329        CHARACTER(len=*), INTENT(IN) :: op
330       
331        CHARACTER(len=100) :: axis_id
332        CHARACTER(len=100) :: operation
333        TYPE(xios_file) :: f
334        TYPE(xios_field) :: field
335        TYPE(xios_fieldgroup) :: fieldgroup
336       
337       
338        !Préparation du nom de l'axe:
339        CALL concat("presnivs", fid, axis_id)
340       
341        !on prépare le nom de l'opération:
342        operation = reformaop(op)
343       
344       
345       
346        !On selectionne le bon groupe de champs:
347        IF (fdim.EQ.2) THEN
348            CALL xios_get_fieldgroup_handle("fields_2D", fieldgroup)
349        ELSE
350          CALL xios_get_fieldgroup_handle("fields_3D", fieldgroup)
351        ENDIF
352       
353        !On regarde si le champ à déjà été créé ou non:
354        IF (xios_is_valid_field(fieldname)) THEN
355            WRITE(*,*) "Xios. Champ ", fieldname, "existe"
356        ELSE
357            WRITE(*,*) "Xios. Champ ", fieldname, "nexiste pas"
358           
359            !On le créé:
360            CALL wxios_add_field(fieldname,  fieldgroup, fieldlongname, fieldunit)
361            IF (xios_is_valid_field(fieldname)) THEN
362                WRITE(*,*) "Xios. Champ ", fieldname, "cree"
363            ENDIF
364        ENDIF
365       
366        !On ajoute le champ:
367        CALL xios_get_file_handle("X"//fname, f)
368        CALL xios_add_fieldtofile(f, field)
369       
370       
371        !L'operation, sa frequence:
372        CALL xios_set_field_attr_hdl(field, field_ref=fieldname, operation=TRIM(ADJUSTL(operation)), freq_op="1ts", prec=4)
373
374       
375        !On rentre ses paramètres:
376        CALL xios_set_field_attr_hdl(field, level=field_level, enabled=.TRUE.)
377       
378        IF (fdim.EQ.2) THEN
379            !Si c'est un champ 2D:
380            WRITE(*,*) "Xios. Champ 2D ", fieldname, " de ", "X"//fname ," configure:"
381            WRITE (*,*) "Xios. op=", TRIM(ADJUSTL(operation))
382            WRITE(*,*) "Xios. freq_op=1ts","; lvl=",field_level
383        ELSE
384            !Si 3D :
385            !On ajoute l'axe vertical qui va bien:
386            CALL xios_set_field_attr_hdl(field, axis_ref=axis_id)
387           
388            WRITE(*,*) "Xios. Champ 3D ", fieldname, " de ", "X"//fname, "configure:"
389            WRITE(*,*) "Xios. freq_op=1ts","; lvl=",field_level
390            WRITE(*,*) "Xios. axe=",axis_id
391        END IF
392       
393    END SUBROUTINE wxios_add_field_to_file
394   
395    SUBROUTINE wxios_update_calendar(ito)
396        INTEGER, INTENT(IN) :: ito
397        CALL xios_update_calendar(ito)
398    END SUBROUTINE wxios_update_calendar
399   
400    SUBROUTINE wxios_write_2D(fieldname, fdata)
401        CHARACTER(len=*), INTENT(IN) :: fieldname
402        REAL, DIMENSION(:,:), INTENT(IN) :: fdata
403       
404        CALL xios_send_field(fieldname, fdata)
405    END SUBROUTINE wxios_write_2D
406   
407    SUBROUTINE wxios_write_3D(fieldname, fdata)
408        CHARACTER(len=*), INTENT(IN) :: fieldname
409        REAL, DIMENSION(:,:,:), INTENT(IN) :: fdata
410       
411        CALL xios_send_field(fieldname, fdata)
412    END SUBROUTINE wxios_write_3D
413   
414    SUBROUTINE wxios_closedef()
415        CALL xios_close_context_definition()
416        CALL xios_update_calendar(0)
417    END SUBROUTINE wxios_closedef
418   
419    SUBROUTINE wxios_close()
420        CALL xios_context_finalize()
421         CALL xios_finalize()
422     END SUBROUTINE wxios_close
423END MODULE wxios
424#endif
Note: See TracBrowser for help on using the repository browser.