MODULE write_output_mod
    IMPLICIT NONE
PRIVATE
    
    INTERFACE write_output
      MODULE PROCEDURE write_output_d0,write_output_d1,write_output_d2, &
                       write_output_i0,write_output_i1,write_output_i2, &
                       write_output_l0,write_output_l1,write_output_l2

    END INTERFACE write_output

    PUBLIC write_output
    
CONTAINS

  SUBROUTINE write_output_d0(field_name,title,units,field)
  ! For a surface field
#ifdef CPP_XIOS
  use xios_output_mod, only: send_xios_field
#endif
  IMPLICIT NONE
  include "dimensions.h"
  INTEGER ngrid
  PARAMETER( ngrid = 2+(jjm-1)*iim - 1/jjm   )
  CHARACTER(LEN=*),INTENT(IN)    :: field_name
  CHARACTER(LEN=*),INTENT(IN)    :: title
  CHARACTER(LEN=*),INTENT(IN)    :: units
  REAL,INTENT(IN)                :: field
  
  call writediagfi(ngrid,field_name,title,units,0,field)
#ifdef CPP_XIOS 
  call send_xios_field(field_name,field)
#endif
  
  END SUBROUTINE write_output_d0

  SUBROUTINE write_output_d1(field_name,title,units,field)
  ! For a surface field
#ifdef CPP_XIOS
  use xios_output_mod, only: send_xios_field
#endif
  IMPLICIT NONE
  include "dimensions.h"
  INTEGER ngrid
  PARAMETER( ngrid = 2+(jjm-1)*iim - 1/jjm   )
  CHARACTER(LEN=*),INTENT(IN)    :: field_name
  CHARACTER(LEN=*),INTENT(IN)    :: title
  CHARACTER(LEN=*),INTENT(IN)    :: units
  REAL,INTENT(IN)                :: field(:)
 
  call writediagfi(ngrid,field_name,title,units,2,field)
#ifdef CPP_XIOS 
  call send_xios_field(field_name,field)
#endif
  
  END SUBROUTINE write_output_d1

  SUBROUTINE write_output_d2(field_name,title,units,field)
  ! For a "3D" horizontal-vertical field
#ifdef CPP_XIOS
  use xios_output_mod, only: send_xios_field
#endif
  use comsoil_h, only: nsoilmx
  use writediagsoil_mod, only: writediagsoil
  IMPLICIT NONE
  include "dimensions.h"
  INTEGER ngrid
  PARAMETER( ngrid = 2+(jjm-1)*iim - 1/jjm   )
  CHARACTER(LEN=*),INTENT(IN)    :: field_name
  CHARACTER(LEN=*),INTENT(IN)    :: title
  CHARACTER(LEN=*),INTENT(IN)    :: units
  REAL,INTENT(IN)                :: field(:,:)

  if(size(field(:,:),2).eq.nsoilmx) then
    call writediagsoil(ngrid,field_name,title,units,3,field)
  else 
    call writediagfi(ngrid,field_name,title,units,3,field(:,:))
  endif
#ifdef CPP_XIOS 
  call send_xios_field(field_name,field)
#endif
  
  END SUBROUTINE write_output_d2

  SUBROUTINE write_output_i0(field_name,title,units,field)
  ! For a surface field
#ifdef CPP_XIOS
  use xios_output_mod, only: send_xios_field
#endif
  IMPLICIT NONE
  include "dimensions.h"
  INTEGER ngrid
  PARAMETER( ngrid = 2+(jjm-1)*iim - 1/jjm   )
  CHARACTER(LEN=*),INTENT(IN)    :: field_name
  CHARACTER(LEN=*),INTENT(IN)    :: title
  CHARACTER(LEN=*),INTENT(IN)    :: units
  INTEGER,INTENT(IN)             :: field
  
  call writediagfi(ngrid,field_name,title,units,0,real(field))
#ifdef CPP_XIOS 
  call send_xios_field(field_name,real(field))
#endif
  
  END SUBROUTINE write_output_i0

  SUBROUTINE write_output_i1(field_name,title,units,field)
  ! For a surface field
#ifdef CPP_XIOS
  use xios_output_mod, only: send_xios_field
#endif
  IMPLICIT NONE
  include "dimensions.h"
  INTEGER ngrid
  PARAMETER( ngrid = 2+(jjm-1)*iim - 1/jjm   )
  CHARACTER(LEN=*),INTENT(IN)    :: field_name
  CHARACTER(LEN=*),INTENT(IN)    :: title
  CHARACTER(LEN=*),INTENT(IN)    :: units
  INTEGER,INTENT(IN)             :: field(:)
 
  call writediagfi(ngrid,field_name,title,units,2,real(field))
#ifdef CPP_XIOS 
  call send_xios_field(field_name,real(field))
#endif
  
  END SUBROUTINE write_output_i1

  SUBROUTINE write_output_i2(field_name,title,units,field)
  ! For a "3D" horizontal-vertical field
#ifdef CPP_XIOS
  use xios_output_mod, only: send_xios_field
#endif
  use comsoil_h, only: nsoilmx
  use writediagsoil_mod, only: writediagsoil
  IMPLICIT NONE
  include "dimensions.h"
  INTEGER ngrid
  PARAMETER( ngrid = 2+(jjm-1)*iim - 1/jjm   )
  CHARACTER(LEN=*),INTENT(IN)    :: field_name
  CHARACTER(LEN=*),INTENT(IN)    :: title
  CHARACTER(LEN=*),INTENT(IN)    :: units
  INTEGER,INTENT(IN)             :: field(:,:)

  if(size(field(:,:),2).eq.nsoilmx) then
    call writediagsoil(ngrid,field_name,title,units,3,real(field))
  else 
    call writediagfi(ngrid,field_name,title,units,3,real(field(:,:)))
  endif
#ifdef CPP_XIOS 
  call send_xios_field(field_name,real(field))
#endif
  
  END SUBROUTINE write_output_i2

  SUBROUTINE write_output_l0(field_name,title,units,field)
  ! For a surface field
#ifdef CPP_XIOS
  use xios_output_mod, only: send_xios_field
#endif
  IMPLICIT NONE
  include "dimensions.h"
  INTEGER ngrid
  PARAMETER( ngrid = 2+(jjm-1)*iim - 1/jjm   )
  CHARACTER(LEN=*),INTENT(IN)    :: field_name
  CHARACTER(LEN=*),INTENT(IN)    :: title
  CHARACTER(LEN=*),INTENT(IN)    :: units
  LOGICAL,INTENT(IN)             :: field
! Local argument used to convert logical to real
  REAL                           :: field_real

  field_real=0
  if(field) field_real=1
  
  call writediagfi(ngrid,field_name,title,units,0,field_real)
#ifdef CPP_XIOS 
  call send_xios_field(field_name,field_real)
#endif
  
  END SUBROUTINE write_output_l0

  SUBROUTINE write_output_l1(field_name,title,units,field)
  ! For a surface field
#ifdef CPP_XIOS
  use xios_output_mod, only: send_xios_field
#endif
  IMPLICIT NONE
  include "dimensions.h"
  INTEGER ngrid
  PARAMETER( ngrid = 2+(jjm-1)*iim - 1/jjm   )
  CHARACTER(LEN=*),INTENT(IN)    :: field_name
  CHARACTER(LEN=*),INTENT(IN)    :: title
  CHARACTER(LEN=*),INTENT(IN)    :: units
  LOGICAL,INTENT(IN)             :: field(:)
! Local argument used to convert logical to real
  REAL                           :: field_real(ngrid)
  INTEGER                        :: i

  field_real(:)=0.
  DO i=1,ngrid
    if(field(i)) field_real(i)=1.
  ENDDO
 
  call writediagfi(ngrid,field_name,title,units,2,field_real(:))
#ifdef CPP_XIOS 
  call send_xios_field(field_name,field_real)
#endif
  
  END SUBROUTINE write_output_l1

  SUBROUTINE write_output_l2(field_name,title,units,field)
  ! For a "3D" horizontal-vertical field
#ifdef CPP_XIOS
  use xios_output_mod, only: send_xios_field
#endif
  use comsoil_h, only: nsoilmx
  use writediagsoil_mod, only: writediagsoil
  IMPLICIT NONE
  include "dimensions.h"
  INTEGER ngrid
  PARAMETER( ngrid = 2+(jjm-1)*iim - 1/jjm   )
  CHARACTER(LEN=*),INTENT(IN)    :: field_name
  CHARACTER(LEN=*),INTENT(IN)    :: title
  CHARACTER(LEN=*),INTENT(IN)    :: units
  LOGICAL,INTENT(IN)             :: field(:,:)
! Local argument used to convert logical to real
  REAL,allocatable               :: field_real(:,:)
  INTEGER                        :: i,j

  allocate(field_real(size(field,1),size(field,2)))

  field_real(:,:)=0.
  DO i=1,size(field,1)
    DO j=1,size(field,2)
      if(field(i,j)) field_real(i,j)=1.
    ENDDO
  ENDDO

  if(size(field(:,:),2).eq.nsoilmx) then
    call writediagsoil(ngrid,field_name,title,units,3,field_real)
  else 
    call writediagfi(ngrid,field_name,title,units,3,field_real(:,:))
  endif

#ifdef CPP_XIOS 
  call send_xios_field(field_name,field_real)
#endif
  
  deallocate(field_real)

  END SUBROUTINE write_output_l2

END MODULE write_output_mod