! $Id: wxios.F90 $ #ifdef CPP_XIOS MODULE wxios USE xios USE iaxis USE iaxis_attr USE icontext_attr USE idate USE idomain_attr USE ifield_attr USE ifile_attr USE ixml_tree !Variables disponibles pendant toute l'execution du programme: INTEGER, SAVE :: g_comm CHARACTER(len=100), SAVE :: g_ctx_name TYPE(xios_context), SAVE :: g_ctx !$OMP THREADPRIVATE(g_comm,g_cts_name,g_ctx) LOGICAL, SAVE :: g_flag_xml = .FALSE. CHARACTER(len=100) :: g_field_name = "nofield" !$OMP THREADPRIVATE(g_flag_xml,g_field_name) CONTAINS !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!! ! 36day => 36d etc !!!!!!!!!!!!!!!!!!!! !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!! SUBROUTINE reformadate(odate, ndate) CHARACTER(len=*), INTENT(IN) :: odate CHARACTER(len=100), INTENT(OUT) :: ndate INTEGER :: i = 0 !!!!!!!!!!!!!!!!!! ! Pour XIOS: ! year : y ! month : mo ! day : d ! hour : h ! minute : mi ! second : s !!!!!!!!!!!!!!!!!! i = INDEX(odate, "day") IF (i > 0) THEN ndate = odate(1:i-1)//"d" END IF i = INDEX(odate, "hr") IF (i > 0) THEN ndate = odate(1:i-1)//"h" END IF i = INDEX(odate, "mth") IF (i > 0) THEN ndate = odate(1:i-1)//"mo" END IF !IF (prt_level >= 10) WRITE(lunout,*) "Xios. ", odate, " => ", ndate END SUBROUTINE reformadate !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!! ! ave(X) => average etc !!!!!!!!!!!!!!! !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!! CHARACTER(len=7) FUNCTION reformaop(op) CHARACTER(len=*), INTENT(IN) :: op INTEGER :: i = 0 reformaop = "average" IF (op.EQ."inst(X)") THEN reformaop = "instant" END IF IF (op.EQ."once") THEN reformaop = "once" END IF IF (op.EQ."t_max(X)") THEN reformaop = "maximum" END IF IF (op.EQ."t_min(X)") THEN reformaop = "minimum" END IF !IF (prt_level >= 10) WRITE(lunout,*) "Xios. ", op, " => ", reformaop END FUNCTION reformaop !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!! ! Routine d'initialisation !!!!!!!!!!!!! ! A lancer juste après mpi_init !!!!!!!!!!!!! !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!! SUBROUTINE wxios_init(xios_ctx_name, locom, outcom) IMPLICIT NONE INCLUDE 'iniprint.h' CHARACTER(len=*), INTENT(IN) :: xios_ctx_name INTEGER, INTENT(IN), OPTIONAL :: locom INTEGER, INTENT(OUT), OPTIONAL :: outcom TYPE(xios_context) :: xios_ctx INTEGER :: xios_comm IF (prt_level >= 10) WRITE(lunout,*) "wxios_init: Initialization" IF (PRESENT(locom)) THEN CALL xios_initialize(xios_ctx_name, local_comm = locom, return_comm = xios_comm ) IF (prt_level >= 10) WRITE(lunout,*) "wxios_init: ctx=",xios_ctx_name," local_comm=",locom,", return_comm=",xios_comm ELSE CALL xios_initialize(xios_ctx_name, return_comm = xios_comm ) IF (prt_level >= 10) WRITE(lunout,*) "wxios_init: ctx=",xios_ctx_name," return_comm=",xios_comm END IF IF (PRESENT(outcom)) THEN outcom = xios_comm IF (prt_level >= 10) WRITE(lunout,*) "wxios_init: ctx=",xios_ctx_name," outcom=",outcom END IF !Enregistrement des variables globales: g_comm = xios_comm g_ctx_name = xios_ctx_name CALL wxios_context_init() END SUBROUTINE wxios_init SUBROUTINE wxios_context_init() IMPLICIT NONE INCLUDE 'iniprint.h' TYPE(xios_context) :: xios_ctx !Initialisation du contexte: CALL xios_context_initialize(g_ctx_name, g_comm) CALL xios_get_handle(g_ctx_name, xios_ctx) !Récupération CALL xios_set_current_context(xios_ctx) !Activation g_ctx = xios_ctx IF (prt_level >= 10) THEN WRITE(lunout,*) "wxios_context_init: Current context is ",trim(g_ctx_name) WRITE(lunout,*) " now call xios_solve_inheritance()" ENDIF !Une première analyse des héritages: CALL xios_solve_inheritance() END SUBROUTINE wxios_context_init !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!! ! Routine de paramétrisation !!!!!!!!!!!!!!!!!! !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!! SUBROUTINE wxios_set_cal(pasdetemps, calendrier, annee, mois, jour, heure) IMPLICIT NONE INCLUDE 'iniprint.h' !Paramètres: CHARACTER(len=*), INTENT(IN) :: calendrier INTEGER, INTENT(IN) :: annee, mois, jour REAL, INTENT(IN) :: pasdetemps, heure !Variables: CHARACTER(len=80) :: abort_message CHARACTER(len=19) :: date INTEGER :: njour = 1 !Variables pour xios: TYPE(xios_time) :: mdtime !REAL(kind = 8) :: year=0, month=0, day=0, hour=0, minute=0, second=0 mdtime = xios_time(0, 0, 0, 0, 0, pasdetemps) !Réglage du calendrier: SELECT CASE (calendrier) CASE('earth_360d') CALL xios_set_context_attr_hdl(g_ctx, calendar_type= "D360") IF (prt_level >= 10) WRITE(lunout,*) 'wxios_set_cal: Calendrier terrestre a 360 jours/an' CASE('earth_365d') CALL xios_set_context_attr_hdl(g_ctx, calendar_type= "NoLeap") IF (prt_level >= 10) WRITE(lunout,*) 'wxios_set_cal: Calendrier terrestre a 365 jours/an' CASE('earth_366d') CALL xios_set_context_attr_hdl(g_ctx, calendar_type= "Gregorian") IF (prt_level >= 10) WRITE(lunout,*) 'wxios_set_cal: Calendrier gregorien' CASE DEFAULT abort_message = 'wxios_set_cal: Mauvais choix de calendrier' CALL abort_gcm('Gcm:Xios',abort_message,1) END SELECT !Formatage de la date de départ: WRITE(date, "(i4.4,'-',i2.2,'-',i2.2,' 00:00:00')") annee, mois, jour IF (prt_level >= 10) WRITE(lunout,*) "wxios_set_cal: Initial time: ", date CALL xios_set_context_attr_hdl(g_ctx, start_date= date) !Et enfin,le pas de temps: CALL xios_set_timestep(mdtime) IF (prt_level >= 10) WRITE(lunout,*) "wxios_set_cal: ts=",mdtime END SUBROUTINE wxios_set_cal SUBROUTINE wxios_set_timestep(ts) REAL, INTENT(IN) :: ts TYPE(xios_time) :: mdtime mdtime = xios_time(0, 0, 0, 0, 0, ts) CALL xios_set_timestep(mdtime) END SUBROUTINE wxios_set_timestep !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!! ! Pour initialiser un domaine !!!!!!!!!!!!!!!!!!!! !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!! SUBROUTINE wxios_domain_param(dom_id, is_sequential, ni, nj, ni_glo, nj_glo, & ibegin, iend, ii_begin, ii_end, jbegin, jend, & data_ni, data_ibegin, data_iend, & io_lat, io_lon,is_south_pole,mpi_rank) IMPLICIT NONE INCLUDE 'iniprint.h' CHARACTER(len=*),INTENT(IN) :: dom_id ! domain identifier LOGICAL,INTENT(IN) :: is_sequential ! flag INTEGER,INTENT(IN) :: ni ! local MPI domain number of longitudes INTEGER,INTENT(IN) :: nj ! local MPI domain number of latitudes INTEGER,INTENT(IN) :: ni_glo ! global grid number of longitudes INTEGER,INTENT(IN) :: nj_glo ! global grid number of latitudes INTEGER,INTENT(IN) :: ibegin ! start index, on global grid, of local MPI domain INTEGER,INTENT(IN) :: iend ! end index, on global grid, of local MPI domain INTEGER,INTENT(IN) :: ii_begin ! i index at which local data starts (first row) INTEGER,INTENT(IN) :: ii_end ! i index at which local data ends (last row) INTEGER,INTENT(IN) :: jbegin ! start index, on global grid, of local MPI domain INTEGER,INTENT(IN) :: jend ! end index, on global grid, of local MPI domain INTEGER,INTENT(IN) :: data_ni INTEGER,INTENT(IN) :: data_ibegin INTEGER,INTENT(IN) :: data_iend REAL,INTENT(IN) :: io_lat(:) ! latitudes (of global grid) REAL,INTENT(IN) :: io_lon(:) ! longitudes (of global grid) logical,intent(in) :: is_south_pole ! does this process include the south pole? integer,intent(in) :: mpi_rank ! rank of process TYPE(xios_domain) :: dom LOGICAL :: boool !Masque pour les problèmes de recouvrement MPI: LOGICAL :: mask(ni,nj) !On récupère le handle: CALL xios_get_domain_handle(dom_id, dom) IF (prt_level >= 10) THEN WRITE(lunout,*) "wxios_domain_param: mpirank=",mpi_rank," ni:",ni," ni_glo:", ni_glo, " nj:", nj, " nj_glo:", nj_glo WRITE(lunout,*) "wxios_domain_param: mpirank=",mpi_rank," ibegin:",ibegin," iend:", iend, " jbegin:", jbegin, " jend:", jend WRITE(lunout,*) "wxios_domain_param: mpirank=",mpi_rank," ii_begin:",ii_begin," ii_end:", ii_end WRITE(lunout,*) "wxios_domain_param: mpirank=",mpi_rank," Size io_lon:", SIZE(io_lon(ibegin:iend)), " io_lat:", SIZE(io_lat(jbegin:jend)) ENDIF !On parametrise le domaine: CALL xios_set_domain_attr_hdl(dom, ni_glo=ni_glo, ibegin=ibegin, ni=ni) CALL xios_set_domain_attr_hdl(dom, nj_glo=nj_glo, jbegin=jbegin, nj=nj, data_dim=2) CALL xios_set_domain_attr_hdl(dom, lonvalue=io_lon(ibegin:iend), latvalue=io_lat(jbegin:jend)) IF (.NOT.is_sequential) THEN mask(:,:)=.TRUE. if (ii_begin>1) mask(1:ii_begin-1,1) = .FALSE. if (ii_end= 10) THEN WRITE(lunout,*) "wxios_domain_param: mpirank=",mpi_rank," mask(:,1)=",mask(:,1) WRITE(lunout,*) "wxios_domain_param: mpirank=",mpi_rank," mask(:,nj)=",mask(:,nj) ENDIF CALL xios_set_domain_attr_hdl(dom, mask=mask) END IF CALL xios_is_defined_domain_attr_hdl(dom,ni_glo=boool) !Vérification: IF (xios_is_valid_domain(dom_id)) THEN IF (prt_level >= 10) WRITE(lunout,*) "wxios_domain_param: Domain initialized: ", trim(dom_id), boool ELSE IF (prt_level >= 10) WRITE(lunout,*) "wxios_domain_param: Invalid domain: ", trim(dom_id) END IF END SUBROUTINE wxios_domain_param !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!! ! Pour déclarer un axe vertical !!!!!!!!!!!!!!! !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!! SUBROUTINE wxios_add_vaxis(axisgroup_id, axis_file, axis_size, axis_value) IMPLICIT NONE INCLUDE 'iniprint.h' CHARACTER (len=*), INTENT(IN) :: axisgroup_id, axis_file INTEGER, INTENT(IN) :: axis_size REAL, DIMENSION(axis_size), INTENT(IN) :: axis_value TYPE(xios_axisgroup) :: axgroup TYPE(xios_axis) :: ax CHARACTER(len=50) :: axis_id IF (len_trim(axisgroup_id).gt.len(axis_id)) THEN WRITE(lunout,*) "wxios_add_vaxis: error, size of axis_id too small!!" WRITE(lunout,*) " increase it to at least ",len_trim(axisgroup_id) CALL abort_gcm("wxios_add_vaxis","len(axis_id) too small",1) ENDIF axis_id=trim(axisgroup_id) !On récupère le groupe d'axes qui va bien: CALL xios_get_axisgroup_handle(axisgroup_id, axgroup) !On ajoute l'axe correspondant à ce fichier: CALL xios_add_axis(axgroup, ax, TRIM(ADJUSTL(axis_id))) !Et on le parametrise: CALL xios_set_axis_attr_hdl(ax, size=axis_size, value=axis_value) !Vérification: IF (xios_is_valid_axis(TRIM(ADJUSTL(axis_id)))) THEN IF (prt_level >= 10) WRITE(lunout,*) "wxios_add_vaxis: Axis created: ", TRIM(ADJUSTL(axis_id)) ELSE WRITE(lunout,*) "wxios_add_vaxis: Invalid axis: ", TRIM(ADJUSTL(axis_id)) END IF END SUBROUTINE wxios_add_vaxis !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!! ! Pour déclarer un fichier !!!!!!!!!!!!!!!!!!! !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!! SUBROUTINE wxios_add_file(fname, ffreq, flvl) IMPLICIT NONE INCLUDE 'iniprint.h' CHARACTER(len=*), INTENT(IN) :: fname CHARACTER(len=*), INTENT(IN) :: ffreq INTEGER, INTENT(IN) :: flvl TYPE(xios_file) :: x_file TYPE(xios_filegroup) :: x_fg CHARACTER(len=100) :: nffreq !On regarde si le fichier n'est pas défini par XML: IF (.NOT.xios_is_valid_file(fname)) THEN !On créé le noeud: CALL xios_get_filegroup_handle("defile", x_fg) CALL xios_add_file(x_fg, x_file, fname) !On reformate la fréquence: CALL reformadate(ffreq, nffreq) !On configure: CALL xios_set_file_attr_hdl(x_file, name="X"//fname,& output_freq=TRIM(ADJUSTL(nffreq)), output_level=flvl, enabled=.TRUE.) IF (xios_is_valid_file("X"//fname)) THEN IF (prt_level >= 10) THEN WRITE(lunout,*) "wxios_add_file: New file: ", "X"//fname WRITE(lunout,*) "wxios_add_file: output_freq=",TRIM(ADJUSTL(nffreq)),"; output_lvl=",flvl ENDIF ELSE WRITE(lunout,*) "wxios_add_file: Error, invalid file: ", "X"//trim(fname) WRITE(lunout,*) "wxios_add_file: output_freq=",TRIM(ADJUSTL(nffreq)),"; output_lvl=",flvl END IF ELSE IF (prt_level >= 10) THEN WRITE(lunout,*) "wxios_add_file: File ",trim(fname), " défined using XML." ENDIF ! Ehouarn: add an enable=.true. on top of xml definitions... why??? CALL xios_set_file_attr(fname, enabled=.TRUE.) END IF END SUBROUTINE wxios_add_file !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!! ! Pour créer un champ !!!!!!!!!!!!!!!!!!!! !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!! SUBROUTINE wxios_add_field(fieldname, fieldgroup, fieldlongname, fieldunit) USE netcdf, only: nf90_fill_real IMPLICIT NONE INCLUDE 'iniprint.h' CHARACTER(len=*), INTENT(IN) :: fieldname TYPE(xios_fieldgroup), INTENT(IN) :: fieldgroup CHARACTER(len=*), INTENT(IN) :: fieldlongname CHARACTER(len=*), INTENT(IN) :: fieldunit TYPE(xios_field) :: field CHARACTER(len=10) :: newunit REAL(KIND=8) :: def !La valeur par défaut des champs non définis: def = nf90_fill_real IF (fieldunit .EQ. " ") THEN newunit = "-" ELSE newunit = fieldunit ENDIF !On ajoute le champ: CALL xios_add_field(fieldgroup, field, fieldname) !IF (prt_level >= 10) WRITE(lunout,*) "wxios_add_field: ",fieldname,fieldgroup, fieldlongname, fieldunit !On rentre ses paramètres: CALL xios_set_field_attr_hdl(field, standard_name=fieldlongname, unit=newunit, default_value=def) IF (prt_level >= 10) WRITE(lunout,*) "wxios_add_field: Field ",trim(fieldname), "cree:" IF (prt_level >= 10) WRITE(lunout,*) "wxios_add_field: long_name=",trim(fieldlongname),"; unit=",trim(newunit),"; default_value=",nf90_fill_real END SUBROUTINE wxios_add_field !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!! ! Pour déclarer un champ !!!!!!!!!!!!!!!!! !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!! SUBROUTINE wxios_add_field_to_file(fieldname, fdim, fid, fname, fieldlongname, fieldunit, field_level, op) IMPLICIT NONE INCLUDE 'iniprint.h' CHARACTER(len=*), INTENT(IN) :: fieldname INTEGER, INTENT(IN) :: fdim, fid CHARACTER(len=*), INTENT(IN) :: fname CHARACTER(len=*), INTENT(IN) :: fieldlongname CHARACTER(len=*), INTENT(IN) :: fieldunit INTEGER, INTENT(IN) :: field_level CHARACTER(len=*), INTENT(IN) :: op CHARACTER(len=20) :: axis_id ! Ehouarn: dangerous... CHARACTER(len=100) :: operation TYPE(xios_file) :: f TYPE(xios_field) :: field TYPE(xios_fieldgroup) :: fieldgroup LOGICAL :: bool=.FALSE. INTEGER :: lvl =0 ! Ajout Abd pour NMC: IF (fid.LE.6) THEN axis_id="presnivs" ELSE axis_id="plev" ENDIF !on prépare le nom de l'opération: operation = reformaop(op) !On selectionne le bon groupe de champs: IF (fdim.EQ.2) THEN CALL xios_get_fieldgroup_handle("fields_2D", fieldgroup) ELSE CALL xios_get_fieldgroup_handle("fields_3D", fieldgroup) ENDIF !On regarde si le champ à déjà été créé ou non: IF (xios_is_valid_field(fieldname) .AND. .NOT. g_field_name == fieldname) THEN !Si ce champ existe via XML (ie, dès le premier passage, ie g_field_name != fieldname) alors rien d'autre à faire IF (prt_level >= 10) WRITE(lunout,*) "wxios_add_field_to_file: Field ", trim(fieldname), "exists via XML" g_flag_xml = .TRUE. g_field_name = fieldname ELSE IF (.NOT. g_field_name == fieldname) THEN !Si premier pssage et champ indéfini, alors on le créé IF (prt_level >= 10) WRITE(lunout,*) "wxios_add_field_to_file: Field ", trim(fieldname), "does not exist" !On le créé: CALL wxios_add_field(fieldname, fieldgroup, fieldlongname, fieldunit) IF (xios_is_valid_field(fieldname)) THEN IF (prt_level >= 10) WRITE(lunout,*) "wxios_add_field_to_file: Field ", trim(fieldname), "created" ENDIF g_flag_xml = .FALSE. g_field_name = fieldname END IF IF (.NOT. g_flag_xml) THEN !Champ existe déjà, mais pas XML, alors on l'ajoute !On ajoute le champ: CALL xios_get_file_handle(fname, f) CALL xios_add_fieldtofile(f, field) !L'operation, sa frequence: CALL xios_set_field_attr_hdl(field, field_ref=fieldname, operation=TRIM(ADJUSTL(operation)), freq_op="1ts", prec=4) !On rentre ses paramètres: CALL xios_set_field_attr_hdl(field, level=field_level, enabled=.TRUE.) IF (fdim.EQ.2) THEN !Si c'est un champ 2D: IF (prt_level >= 10) THEN WRITE(lunout,*) "wxios_add_field_to_file: 2D Field ", trim(fieldname), " in ", "X"//trim(fname) ," configured with:" WRITE(lunout,*) "wxios_add_field_to_file: op=", TRIM(ADJUSTL(operation)) WRITE(lunout,*) "wxios_add_field_to_file: freq_op=1ts","; lvl=",field_level ENDIF ELSE !Si 3D : !On ajoute l'axe vertical qui va bien: CALL xios_set_field_attr_hdl(field, axis_ref=TRIM(ADJUSTL(axis_id))) IF (prt_level >= 10) THEN WRITE(lunout,*) "wxios_add_field_to_file: 3D Field",trim(fieldname), " in ", "X"//trim(fname), "configured with:" WRITE(lunout,*) "wxios_add_field_to_file: freq_op=1ts","; lvl=",field_level WRITE(lunout,*) "wxios_add_field_to_file: axis=",TRIM(ADJUSTL(axis_id)) ENDIF END IF ELSE !Sinon on se contente de l'activer: CALL xios_set_field_attr(fieldname, enabled=.TRUE.) ENDIF END SUBROUTINE wxios_add_field_to_file SUBROUTINE wxios_update_calendar(ito) INTEGER, INTENT(IN) :: ito CALL xios_update_calendar(ito) END SUBROUTINE wxios_update_calendar SUBROUTINE wxios_write_2D(fieldname, fdata) CHARACTER(len=*), INTENT(IN) :: fieldname REAL, DIMENSION(:,:), INTENT(IN) :: fdata CALL xios_send_field(fieldname, fdata) END SUBROUTINE wxios_write_2D SUBROUTINE wxios_write_3D(fieldname, fdata) CHARACTER(len=*), INTENT(IN) :: fieldname REAL, DIMENSION(:,:,:), INTENT(IN) :: fdata CALL xios_send_field(fieldname, fdata) END SUBROUTINE wxios_write_3D SUBROUTINE wxios_closedef() CALL xios_close_context_definition() CALL xios_update_calendar(0) END SUBROUTINE wxios_closedef SUBROUTINE wxios_close() CALL xios_context_finalize() CALL xios_finalize() END SUBROUTINE wxios_close END MODULE wxios #endif