Context Navigation

wxios.F90 @ 5428

Last change on this file since 5428 was 4477, checked in by lfalletti, 22 months ago
New axis (axis_lat_greordered) that fixes a bug with latitudes being reversed for outputs from CMIP6 workflow (for zonal mean files only)
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.6 KB

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

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

Download in other formats:

Original Format