source: codes/icosagcm/branches/SATURN_DYNAMICO/LMDZ.COMMON/libf/phystd/xios_output.F90 @ 310

MODULE xios_output_mod


 INTEGER :: time_it
 CHARACTER(LEN=*), PARAMETER :: context_id= "lmdz_physics"
 
 PRIVATE :: time_it

 INTERFACE write_xios_field
    MODULE PROCEDURE write_xios_field_2d,write_xios_field_3d
  END INTERFACE
 

CONTAINS

  SUBROUTINE initialize_xios_output
  USE comgeomphy
  USE xios
  USE dimphy
  USE mod_phys_lmdz_para
  USE mod_grid_phy_lmdz
  USE planete_mod, ONLY: ap,bp,preff
  
  IMPLICIT NONE
    INCLUDE "comcstfi.h" 
      
    TYPE(xios_context) :: ctx_hdl
    TYPE(xios_time)      :: dtime
    REAL :: lev_value(klev)

    REAL :: lon_mpi(klon_mpi)
    REAL :: lat_mpi(klon_mpi)
    REAL :: bounds_lon_mpi(klon_mpi,nvertex)
    REAL :: bounds_lat_mpi(klon_mpi,nvertex)
    INTEGER :: l
    REAL :: presnivs(klev)

    CALL gather_omp(rlond,lon_mpi)
    CALL gather_omp(rlatd,lat_mpi)
    CALL gather_omp(rbounds_lon,bounds_lon_mpi)
    CALL gather_omp(rbounds_lat,bounds_lat_mpi)
    
!$OMP BARRIER    
!$OMP MASTER
    CALL xios_context_initialize(context_id,comm_lmdz_phy)
    CALL xios_get_handle(context_id, ctx_hdl)
    CALL xios_set_current_context(ctx_hdl)
   
!    lev_value(:) = (/ (l,l=1,klev) /)     
!    CALL xios_set_axis_attr("altitude",size=klev ,value=lev_value)
    
    ! build presnivs "pseudo-pressure" at mid-layers
    DO l=1,klev
      presnivs(l)=0.5*(ap(l)+bp(l)*preff+ap(l+1)+bp(l+1)*preff)
    ENDDO
    CALL xios_set_axis_attr("presnivs",size=klev,value=presnivs,unit="Pa",positive="down")
  
    CALL xios_set_domain_attr("horiz_domain",ni_glo=klon_glo, ibegin=klon_mpi_begin, ni=klon_mpi)
    CALL xios_set_domain_attr("horiz_domain", data_dim=1, type='unstructured' , nvertex=6)
    CALL xios_set_domain_attr("horiz_domain", lonvalue=lon_mpi*180/Pi, latvalue=lat_mpi*180/Pi)
    CALL xios_set_domain_attr("horiz_domain", bounds_lon=reshape(bounds_lon_mpi,(/nvertex,klon_mpi/),order=(/2,1/))*180./Pi )
    CALL xios_set_domain_attr("horiz_domain", bounds_lat=reshape(bounds_lat_mpi,(/nvertex,klon_mpi/),order=(/2,1/))*180./Pi )
  
    dtime%second=dtphys
    CALL xios_set_timestep(dtime)
    time_it = 0  
    CALL xios_close_context_definition()
!$OMP END MASTER    
!$OMP BARRIER    
    
  END SUBROUTINE initialize_xios_output

  SUBROUTINE finalize_xios_output
  USE xios

!$OMP BARRIER    
!$OMP MASTER
    CALL xios_context_finalize
!$OMP END MASTER    
!$OMP BARRIER    
  
  END SUBROUTINE finalize_xios_output


  SUBROUTINE update_xios_timestep
  USE xios
  IMPLICIT NONE
    CALL set_xios_context
!$OMP MASTER
    time_it=time_it+1
    CALL xios_update_calendar(time_it)
!$OMP END MASTER    
  END SUBROUTINE update_xios_timestep


  SUBROUTINE set_xios_context
  USE XIOS
  IMPLICIT NONE
    TYPE(xios_context) :: ctx_hdl

!$OMP MASTER
    CALL xios_get_handle(context_id,ctx_hdl)
    CALL xios_set_current_context(ctx_hdl)
!$OMP END MASTER    
  END SUBROUTINE set_xios_context


  
  SUBROUTINE write_xios_field_2d(field_id,field)
  USE xios
  USE mod_phys_lmdz_para
  IMPLICIT NONE
    CHARACTER(LEN=*) :: field_id
    REAL,INTENT(IN)  :: field(:)
    REAL             :: field_mpi(klon_mpi)
  
    CALL gather_omp(field,field_mpi)

!$OMP MASTER
    CALL xios_send_field(field_id,field_mpi)
!$OMP END MASTER   
    
  END SUBROUTINE write_xios_field_2d

  SUBROUTINE write_xios_field_3d(field_id,field)
  USE xios
  USE mod_phys_lmdz_para
  IMPLICIT NONE
    CHARACTER(LEN=*) :: field_id
    REAL,INTENT(IN)  :: field(:,:)
    REAL             :: field_mpi(klon_mpi,SIZE(field,2))
  
    CALL gather_omp(field,field_mpi)

!$OMP MASTER
    CALL xios_send_field(field_id,field_mpi)
!$OMP END MASTER   
    
  END SUBROUTINE write_xios_field_3d
  
END MODULE xios_output_mod