source: XIOS/xios_training/hands-on-5/answer/test_tp5.f90_2 @ 2357

PROGRAM test_tp5

  USE XIOS
  IMPLICIT NONE
  INCLUDE "mpif.h"
  INTEGER :: rank
  INTEGER :: size
  INTEGER :: ierr

  INTEGER :: comm
  TYPE(xios_duration) :: dtime

  TYPE(xios_date) :: dorigin
  CHARACTER(len=20) :: dorigin_str

  TYPE(xios_date) :: dstart
  CHARACTER(len=20) :: dstart_str

  INTEGER :: axis_size
  DOUBLE PRECISION, ALLOCATABLE :: axis_value(:)

  CHARACTER(len=20) :: domain_type
  INTEGER :: ni_glo, nj_glo
  DOUBLE PRECISION, ALLOCATABLE :: lonvalue(:), latvalue(:)
  
  INTEGER :: i, j
  
  DOUBLE PRECISION, ALLOCATABLE :: field_A(:,:,:)
  DOUBLE PRECISION, ALLOCATABLE :: field_B(:,:,:)
  DOUBLE PRECISION, ALLOCATABLE :: field_C(:,:,:)
 
  INTEGER :: ts
  
  INTEGER :: ni, nj, ibegin, jbegin
 
  CALL MPI_INIT(ierr)

  CALL xios_initialize("client",return_comm=comm)

  CALL MPI_COMM_RANK(comm,rank,ierr)
  CALL MPI_COMM_SIZE(comm,size,ierr)
  
  print*, "Hello XIOS from proc", rank
  
  CALL xios_context_initialize("test",comm)

  !CALL xios_define_calendar(type="Gregorian") 
  !We define the calendar type in xml

  CALL xios_get_time_origin(dorigin)
  CALL xios_date_convert_to_string(dorigin, dorigin_str)
  if (rank .EQ. 0) print*, "calendar time_origin = ", dorigin_str

  CALL xios_get_start_date(dstart)
  CALL xios_date_convert_to_string(dstart, dstart_str)
  if (rank .EQ. 0) print*, "calendar start_date = ", dstart_str

  dtime%second = 3600
  CALL xios_set_timestep(dtime)

  CALL xios_get_axis_attr("axis_A", n_glo = axis_size)
  ALLOCATE(axis_value(axis_size))
  CALL xios_get_axis_attr("axis_A", value = axis_value)
  if (rank .EQ. 0) print*, "axis value = ", axis_value(:)

  CALL xios_get_domain_attr("domain_A", type = domain_type)
  CALL xios_get_domain_attr("domain_A", ni_glo = ni_glo, nj_glo=nj_glo)
  if(rank.EQ.0) print*, "domain type = ", domain_type
  if(rank.EQ.0) print*, "domain size = ", ni_glo, "*", nj_glo

  ni=ni_glo/size
  nj=nj_glo

  ibegin=ni*rank
  jbegin=0

  CALL xios_set_domain_attr("domain_A", ni=ni, nj=nj, ibegin=ibegin,jbegin=jbegin)
  print*, "Proc ",rank, "occupies subdomain",ibegin,"->",ibegin+ni,"X",jbegin,"->",jbegin+nj


  ALLOCATE(lonvalue(ni))
  ALLOCATE(latvalue(nj))

  DO i=1,ni
    lonvalue(i) = -180 + (rank*ni+i) * 360/ni_glo
  ENDDO

  DO j=1,nj
    latvalue(j) = -90 + j * 180/nj_glo
  ENDDO

  CALL xios_set_domain_attr("domain_A", lonvalue_1d=lonvalue,latvalue_1d=latvalue)
  
  !--- Case 2 ---!
  ! CALL xios_set_domain_attr("domain_A", data_dim=2, data_ibegin=-1, data_ni=ni+2, data_jbegin=-1, data_nj=nj+2)
  
  !--- Case 3 ---!
  !CALL xios_set_domain_attr("domain_A", data_dim=2, data_ibegin=1, data_ni=ni-2, data_jbegin=0, data_nj=nj/2)

  CALL xios_set_domain_attr("domain_read", type="curvilinear", ni_glo=10, nj_glo=10)
  CALL xios_set_domain_attr("domain_read", ni=10, ibegin=0, nj=10,jbegin=0)
 
  CALL xios_close_context_definition()

  !--- Case 1 ---!
  ALLOCATE(field_A(ni, nj, axis_size))
  field_A(:,:,:)=rank

  !--- Case 2 ---!
  ! ALLOCATE(field_A(ni+2, nj+2, axis_size))
  ! field_A(:,:,:)=rank

  !--- Case 3 ---!
  ! ALLOCATE(field_A(ni-2, nj/2, axis_size))
  ! field_A(:,:,:)=rank

  ALLOCATE(field_B(ni, nj, axis_size))
  ALLOCATE(field_C(ni, nj, axis_size))
  field_B(:,:,:)=-1 * rank
  field_C(:,:,:)=rank*rank
  
  DO ts=1,4
    CALL xios_update_calendar(ts)
    CALL xios_send_field("field_A", field_A)
    CALL xios_send_field("field_B", field_B)
    CALL xios_send_field("field_C", field_C)
  ENDDO

  CALL xios_context_finalize()

  DEALLOCATE(axis_value)
  DEALLOCATE(lonvalue, latvalue)
  DEALLOCATE(field_A)
  DEALLOCATE(field_B)
  DEALLOCATE(field_C)

  CALL MPI_COMM_FREE(comm, ierr)

  CALL xios_finalize()

  CALL MPI_FINALIZE(ierr)

END PROGRAM test_tp5