program Arb2ArbFldReDist
! ESMF Framework module
use ESMF
use ESMF_TestMod
implicit none
! Local variables
integer :: status
integer :: i, j, j1, i1, add
integer :: counts(2), localCounts(2), miscount, localCount
integer :: npets, localPet
integer, dimension(:,:), allocatable :: myIndices1, myIndices2
logical :: match
real(ESMF_KIND_R8) :: min(2), max(2), compval
real(ESMF_KIND_R8) :: pi = 3.1416d0
real(ESMF_KIND_R8), dimension(:), pointer :: coordX, coordY
real(ESMF_KIND_R8), dimension(:), pointer :: srcdata, resdata
type(ESMF_ArraySpec) :: arrayspec1, arrayspec2
type(ESMF_Field) :: humidity1, humidity2, humidity3
type(ESMF_grid) :: grid1, grid2
type(ESMF_RouteHandle) :: rh12, rh23
type(ESMF_VM) :: vm
! cumulative result: count failures; no failures equals "all pass"
integer :: result = 0
! individual test name
character(ESMF_MAXSTR) :: testname
! individual test failure message and final rc msg
character(ESMF_MAXSTR) :: failMsg, finalMsg
!-------------------------------------------------------------------------
!-------------------------------------------------------------------------
print *, "System Test FieldRedistArb2Arb."
!
!-------------------------------------------------------------------------
!-------------------------------------------------------------------------
! Create section
!-------------------------------------------------------------------------
!-------------------------------------------------------------------------
!
! Initialize the framework and get back the default global VM
call ESMF_Initialize(vm=vm, defaultlogfilename="FieldRedistArb2ArbSTest.Log", &
logkindflag=ESMF_LOGKIND_MULTI, rc=status)
if (status .ne. ESMF_SUCCESS) goto 20
call ESMF_LogSet (flush=.true.)
! Get the PET count and our PET number
call ESMF_VMGet(vm, localPet=localPet, petCount=npets, rc=status)
if (status .ne. ESMF_SUCCESS) goto 20
miscount = 0
print *, "Create section finished"
!
!-------------------------------------------------------------------------
!-------------------------------------------------------------------------
! Init section
!-------------------------------------------------------------------------
!-------------------------------------------------------------------------
!
! Create the grids and corresponding Fields
counts(1) = 60
counts(2) = 40
min(1) = 0.0
max(1) = 60.0
min(2) = 0.0
max(2) = 50.0
! make two identical grids, both are distributed arbitrarily
! but with different styles
!
! Arbitrary case
! allocate myIndices to maximum number of points on any DE in the first
! dimension and 2 in the second dimension.
localCount = int((counts(1)*counts(2) + npets -1)/npets)
allocate (myIndices1(localCount,2))
! calculate myIndices based on DE number
! for now, start at point (1,1+localPet) and go up in the j-direction first
! to create a semi-regular distribution of points
i1 = 1 + localPet
add = 0
do j = 1,counts(2)
do i = i1,counts(1),npets
add = add + 1
myIndices1(add,1) = i
myIndices1(add,2) = j
enddo
i1 = i - counts(1)
enddo
grid1 = ESMF_GridCreateNoPeriDim(coordTypeKind=ESMF_TYPEKIND_R8, &
minIndex=(/1,1/), maxIndex=counts, &
arbIndexList=myIndices1,arbIndexCount=localCount, &
name="source grid", rc=status)
if (status .ne. ESMF_SUCCESS) goto 20
call ESMF_GridAddCoord(grid1, rc=status)
if (status .ne. ESMF_SUCCESS) goto 20
! allocate myIndices to maximum number of points on any DE in the first
! dimension and 2 in the second dimension.
localCount = int((counts(1)*counts(2) + npets -1)/npets)
allocate (myIndices2(localCount,2))
! calculate myIndices based on DE number
! for now, start at point (1,1+localPet) and go up in the j-direction first
! to create a semi-regular distribution of points
j1 = 1 + localPet
add = 0
do i = 1,counts(1)
do j = j1,counts(2),npets
add = add + 1
myIndices2(add,1) = i
myIndices2(add,2) = j
enddo
j1 = j - counts(2)
enddo
grid2 = ESMF_GridCreateNoPeriDim(coordTypeKind=ESMF_TYPEKIND_R8, &
minIndex=(/1,1/), maxIndex=counts, &
arbIndexList=myIndices2,arbIndexCount=localCount, &
name="dest grid", rc=status)
if (status .ne. ESMF_SUCCESS) goto 20
! Set up a 1D (for the arbitrarily distributed Field) and a 2D real array
call ESMF_ArraySpecSet(arrayspec1, rank=1, &
typekind=ESMF_TYPEKIND_R8)
if (status .ne. ESMF_SUCCESS) goto 20
call ESMF_ArraySpecSet(arrayspec2, rank=1, &
typekind=ESMF_TYPEKIND_R8)
if (status .ne. ESMF_SUCCESS) goto 20
! Create the field and have it create the array internally for each grid
humidity1 = ESMF_FieldCreate(grid1, arrayspec1, &
name="humidity1", rc=status)
if (status .ne. ESMF_SUCCESS) goto 20
humidity2 = ESMF_FieldCreate(grid2, arrayspec2, &
name="humidity2", rc=status)
if (status .ne. ESMF_SUCCESS) goto 20
humidity3 = ESMF_FieldCreate(grid1, arrayspec1, &
name="humidity3", rc=status)
if (status .ne. ESMF_SUCCESS) goto 20
! precompute communication patterns, the first from the 1st arbitrarily
! distributed Field to the 2nd arbitrarily distributed Field; and
! the second from the 2nd arbitrarily distributed Field back to
! the 1st distributed Field
call ESMF_FieldRedistStore(humidity1, humidity2, &
routehandle=rh12, rc=status)
call ESMF_FieldRedistStore(humidity2, humidity3, &
routehandle=rh23, rc=status)
! get coordinate arrays available for setting the source data array
call ESMF_GridGetCoord(grid1, localDE=0, coordDim=1, &
farrayPtr=coordX, totalCount=localCounts, rc=status)
if (status .ne. ESMF_SUCCESS) goto 20
call ESMF_GridGetCoord(grid1, localDE=0, coordDim=2, &
farrayPtr=coordY, rc=status)
if (status .ne. ESMF_SUCCESS) goto 20
! Get pointers to the data and set it up
call ESMF_FieldGet(humidity1, farrayPtr=srcdata, rc=status)
if (status .ne. ESMF_SUCCESS) goto 20
call ESMF_FieldGet(humidity3, farrayPtr=resdata, rc=status)
if (status .ne. ESMF_SUCCESS) goto 20
! initialize data arrays
srcdata = 0.0
resdata = 0.0
! set data array to a function of coordinates (in the computational part
! of the array only, not the halo region)
do i = 1,localCounts(1)
coordX(i) = ((max(1)-min(1))*i)/localCounts(1)
coordY(i) = ((max(2)-min(2))*i)/localCounts(1)
srcdata(i) = 10.0 + 5.0*sin(coordX(i)/60.0*pi) &
+ 2.0*sin(coordY(i)/50.0*pi)
enddo
print *, "Initial data, before Redistribution:"
! No deallocate() is needed for array data, it will be freed when the
! Array is destroyed.
print *, "Init section finished"
!
!-------------------------------------------------------------------------
!-------------------------------------------------------------------------
! Run section
!-------------------------------------------------------------------------
!-------------------------------------------------------------------------
!
! Call redistribution method here, output ends up in humidity2
call ESMF_FieldRedist(humidity1, humidity2, rh12, rc=status)
if (status .ne. ESMF_SUCCESS) goto 20
print *, "Array contents after Transpose:"
! Redistribute back so we can compare contents
! output ends up in humidity3
call ESMF_FieldRedist(humidity2, humidity3, rh23, rc=status)
if (status .ne. ESMF_SUCCESS) goto 20
print *, "Array contents after second Redistribution, should match original:"
print *, "Run section finished"
!
!-------------------------------------------------------------------------
!-------------------------------------------------------------------------
! Finalize section
!-------------------------------------------------------------------------
!-------------------------------------------------------------------------
! Print result
print *, "-----------------------------------------------------------------"
print *, "-----------------------------------------------------------------"
print *, "Result from PET number ", localPet
print *, "-----------------------------------------------------------------"
print *, "-----------------------------------------------------------------"
! check and make sure the original data and the data that has been
! distributed to the 1D Field and back again are the same
match = .true.
miscount = 0
do i = 1,localCounts(1)
compval = 10.0 + 5.0*sin(coordX(i)/60.0*pi) &
+ 2.0*sin(coordY(i)/50.0*pi)
if ((srcdata(i) .ne. resdata(i)) .OR. &
(abs(resdata(i)-compval).ge.1.0d-12)) then
print *, "array contents do not match at: (", i , ") on DE ", &
localPet, ". src=", srcdata(i), "dst=", &
resdata(i), "realval=", compval
match = .false.
miscount = miscount + 1
if (miscount .gt. 40) then
print *, "more than 40 mismatches, skipping rest of loop"
goto 10
endif
endif
enddo
if (match) print *, "Array contents matched correctly!! PET = ", localPet
10 continue
print *, "Finalize section finished"
!
!-------------------------------------------------------------------------
!-------------------------------------------------------------------------
! Destroy section
!-------------------------------------------------------------------------
!-------------------------------------------------------------------------
! Clean up
deallocate(myIndices1, myIndices2)
call ESMF_FieldRedistRelease(rh12, rc=status)
if (status .ne. ESMF_SUCCESS) goto 20
call ESMF_FieldRedistRelease(rh23, rc=status)
if (status .ne. ESMF_SUCCESS) goto 20
call ESMF_FieldDestroy(humidity1, rc=status)
if (status .ne. ESMF_SUCCESS) goto 20
call ESMF_FieldDestroy(humidity2, rc=status)
if (status .ne. ESMF_SUCCESS) goto 20
call ESMF_FieldDestroy(humidity3, rc=status)
if (status .ne. ESMF_SUCCESS) goto 20
call ESMF_GridDestroy(grid1, rc=status)
if (status .ne. ESMF_SUCCESS) goto 20
call ESMF_GridDestroy(grid2, rc=status)
if (status .ne. ESMF_SUCCESS) goto 20
print *, "All Destroy routines done"
!-------------------------------------------------------------------------
!-------------------------------------------------------------------------
20 print *, "System Test FieldRedistArb2Arb complete."
write(failMsg, *) "Redistribution back not same as original"
write(testname, *) "System Test FieldRedistArb2Arb: Field Redistribute"
if (status .ne. ESMF_SUCCESS) then
! Separate message to console, for quick confirmation of success/failure
if ((miscount.eq.0) .and. (status .eq. ESMF_SUCCESS)) then
write(finalMsg, *) "SUCCESS: Data redistributed twice same as original."
else
write(finalMsg, *) "System Test did not succeed. ", &
"Data redistribution does not match expected values, or error code", &
" set ",status
endif
write(0, *) ""
write(0, *) trim(testname)
write(0, *) trim(finalMsg)
write(0, *) ""
endif
! IMPORTANT: ESMF_STest() prints the PASS string and the # of processors
! into the Log file that the scripts grep for.
call ESMF_STest((status.eq.ESMF_SUCCESS), testname, failMsg, result, &
__FILE__, &
__LINE__)
call ESMF_Finalize(rc=status)
end program Arb2ArbFldReDist