subroutine user_init(comp, importState, exportState, clock, rc)
type(ESMF_CplComp) :: comp
type(ESMF_State) :: importState, exportState
type(ESMF_Clock) :: clock
integer, intent(out) :: rc
! Local variables
integer :: itemcount, localPet, i
type(ESMF_FieldBundle) :: srcFieldBundle, dstFieldBundle
type(ESMF_VM) :: vm
real(ESMF_KIND_R8), allocatable:: factorList(:)
integer, allocatable:: factorIndexList(:,:)
! Initialize return code
rc = ESMF_SUCCESS
print *, "User Coupler Init starting"
call ESMF_StateGet(importState, itemcount=itemcount, rc=rc)
if (rc/=ESMF_SUCCESS) call ESMF_Finalize(rc=rc, endflag=ESMF_END_ABORT)
print *, "Import State contains ", itemcount, " items."
! Need to reconcile import and export states
call ESMF_CplCompGet(comp, vm=vm, rc=rc)
if (rc/=ESMF_SUCCESS) call ESMF_Finalize(rc=rc, endflag=ESMF_END_ABORT)
call ESMF_StateReconcile(importState, vm=vm, rc=rc)
if (rc/=ESMF_SUCCESS) call ESMF_Finalize(rc=rc, endflag=ESMF_END_ABORT)
call ESMF_StateReconcile(exportState, vm=vm, rc=rc)
if (rc/=ESMF_SUCCESS) call ESMF_Finalize(rc=rc, endflag=ESMF_END_ABORT)
! Get source FieldBundle out of import State
call ESMF_StateGet(importState, "fieldbundle data", srcFieldBundle, rc=rc)
if (rc/=ESMF_SUCCESS) call ESMF_Finalize(rc=rc, endflag=ESMF_END_ABORT)
! Get destination FieldBundle out of export State
call ESMF_StateGet(exportState, "fieldbundle data", dstFieldBundle, rc=rc)
if (rc/=ESMF_SUCCESS) call ESMF_Finalize(rc=rc, endflag=ESMF_END_ABORT)
! get localPet
call ESMF_VMGet(vm, localPet=localPet, rc=rc)
if (rc/=ESMF_SUCCESS) call ESMF_Finalize(rc=rc, endflag=ESMF_END_ABORT)
! Setup identity sparse matrix as a combination of PET 0 and PET 4
! there are 15000 elements on the diagonal, defined as two overlapping
! lists of each 10000 elements (overlapping 5000).
allocate(factorList(10000))
allocate(factorIndexList(2,10000))
if (localPet==0) then
factorIndexList(1,:) = (/(i,i=1,10000)/)
factorIndexList(2,:) = factorIndexList(1,:)
do i=1, 5000
factorList(i) = 1.d0
enddo
do i=5001, 10000
factorList(i) = .253d0
enddo
endif
if (localPet==4) then
factorIndexList(1,:) = (/(i,i=5001,15000)/)
factorIndexList(2,:) = factorIndexList(1,:)
do i=1, 5000
factorList(i) = .747d0
enddo
do i=5001, 10000
factorList(i) = 1.d0
enddo
endif
! Precompute and store an FieldBundleSMM operation
if (localPet==0 .or. localPet==4) then
! only PET 0 and PET 4 provide factors
call ESMF_FieldBundleSMMStore(srcFieldBundle=srcFieldBundle, &
dstFieldBundle=dstFieldBundle, &
routehandle=routehandle, factorList=factorList, &
factorIndexList=factorIndexList, rc=rc)
if (rc/=ESMF_SUCCESS) return ! bail out
else
call ESMF_FieldBundleSMMStore(srcFieldBundle=srcFieldBundle, dstFieldBundle=dstFieldBundle, &
routehandle=routehandle, rc=rc)
if (rc/=ESMF_SUCCESS) call ESMF_Finalize(rc=rc, endflag=ESMF_END_ABORT)
endif
deallocate(factorIndexList)
deallocate(factorList)
print *, "User Coupler Init returning"
end subroutine user_init