user_init Subroutine

public subroutine user_init(comp, importState, exportState, clock, rc)

Arguments

Type IntentOptional Attributes Name
type(ESMF_CplComp) :: comp
type(ESMF_State) :: importState
type(ESMF_State) :: exportState
type(ESMF_Clock) :: clock
integer, intent(out) :: rc

Calls

proc~~user_init~86~~CallsGraph proc~user_init~86 user_init esmf_stateget esmf_stateget proc~user_init~86->esmf_stateget interface~esmf_arraybundlesmmstore ESMF_ArrayBundleSMMStore proc~user_init~86->interface~esmf_arraybundlesmmstore interface~esmf_vmget ESMF_VMGet proc~user_init~86->interface~esmf_vmget proc~esmf_cplcompget ESMF_CplCompGet proc~user_init~86->proc~esmf_cplcompget proc~esmf_statereconcile ESMF_StateReconcile proc~user_init~86->proc~esmf_statereconcile proc~esmf_arraybundlesmmstorei4 ESMF_ArrayBundleSMMStoreI4 interface~esmf_arraybundlesmmstore->proc~esmf_arraybundlesmmstorei4 proc~esmf_arraybundlesmmstorei8 ESMF_ArrayBundleSMMStoreI8 interface~esmf_arraybundlesmmstore->proc~esmf_arraybundlesmmstorei8 proc~esmf_arraybundlesmmstorenf ESMF_ArrayBundleSMMStoreNF interface~esmf_arraybundlesmmstore->proc~esmf_arraybundlesmmstorenf proc~esmf_arraybundlesmmstorer4 ESMF_ArrayBundleSMMStoreR4 interface~esmf_arraybundlesmmstore->proc~esmf_arraybundlesmmstorer4 proc~esmf_arraybundlesmmstorer8 ESMF_ArrayBundleSMMStoreR8 interface~esmf_arraybundlesmmstore->proc~esmf_arraybundlesmmstorer8 proc~esmf_vmgetdefault ESMF_VMGetDefault interface~esmf_vmget->proc~esmf_vmgetdefault proc~esmf_vmgetpetspecific ESMF_VMGetPetSpecific interface~esmf_vmget->proc~esmf_vmgetpetspecific proc~esmf_compget ESMF_CompGet proc~esmf_cplcompget->proc~esmf_compget proc~esmf_compstatusget ESMF_CompStatusGet proc~esmf_cplcompget->proc~esmf_compstatusget proc~esmf_cplcompgetinit ESMF_CplCompGetInit proc~esmf_cplcompget->proc~esmf_cplcompgetinit proc~esmf_imerr ESMF_IMErr proc~esmf_cplcompget->proc~esmf_imerr proc~esmf_logfounderror ESMF_LogFoundError proc~esmf_cplcompget->proc~esmf_logfounderror proc~esmf_statereconcile->interface~esmf_vmget destroy destroy proc~esmf_statereconcile->destroy initialize initialize proc~esmf_statereconcile->initialize interface~esmf_vmbroadcast ESMF_VMBroadcast proc~esmf_statereconcile->interface~esmf_vmbroadcast proc~esmf_statereconcile->proc~esmf_imerr proc~esmf_infodump ESMF_InfoDump proc~esmf_statereconcile->proc~esmf_infodump proc~esmf_statereconcile->proc~esmf_logfounderror proc~esmf_logseterror ESMF_LogSetError proc~esmf_statereconcile->proc~esmf_logseterror proc~esmf_stategetinit ESMF_StateGetInit proc~esmf_statereconcile->proc~esmf_stategetinit proc~esmf_statereconcile_driver ESMF_StateReconcile_driver proc~esmf_statereconcile->proc~esmf_statereconcile_driver proc~esmf_statereconcileisnoop ESMF_StateReconcileIsNoop proc~esmf_statereconcile->proc~esmf_statereconcileisnoop proc~esmf_traceregionenter ESMF_TraceRegionEnter proc~esmf_statereconcile->proc~esmf_traceregionenter proc~esmf_traceregionexit ESMF_TraceRegionExit proc~esmf_statereconcile->proc~esmf_traceregionexit proc~esmf_utilstringdiffmatch ESMF_UtilStringDiffMatch proc~esmf_statereconcile->proc~esmf_utilstringdiffmatch proc~esmf_vmgetcurrent ESMF_VMGetCurrent proc~esmf_statereconcile->proc~esmf_vmgetcurrent proc~esmf_vmgetinit ESMF_VMGetInit proc~esmf_statereconcile->proc~esmf_vmgetinit update update proc~esmf_statereconcile->update

Source Code

  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
    type(ESMF_ArrayBundle):: srcArraybundle, dstArraybundle
    type(ESMF_VM) :: vm
    real(ESMF_KIND_R8):: factorList(10000)
    integer:: i, factorIndexList(2,10000)

    ! Initialize return code
    rc = ESMF_SUCCESS

    print *, "User Coupler Init starting"

    call ESMF_StateGet(importState, itemcount=itemcount, rc=rc)
    if (rc/=ESMF_SUCCESS) return ! bail out
    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) return ! bail out
    call ESMF_StateReconcile(importState, vm=vm, rc=rc)
    if (rc/=ESMF_SUCCESS) return ! bail out
    call ESMF_StateReconcile(exportState, vm=vm, rc=rc)
    if (rc/=ESMF_SUCCESS) return ! bail out

    ! Get source ArrayBundle out of import state
    call ESMF_StateGet(importState, "srcAryBndl", srcArraybundle, rc=rc)
    if (rc/=ESMF_SUCCESS) return ! bail out

    ! Get destination ArrayBundle out of export state
    call ESMF_StateGet(exportState, "dstAryBndl", dstArraybundle, rc=rc)
    if (rc/=ESMF_SUCCESS) return ! bail out

    ! get localPet
    call ESMF_VMGet(vm, localPet=localPet, rc=rc)
    if (rc/=ESMF_SUCCESS) return ! bail out

    ! 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).
    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 ArrayBundleSMM operation
    if (localPet==0 .or. localPet==4) then
      ! only PET 0 and PET 4 provide factors
      call ESMF_ArrayBundleSMMStore(&
        srcArrayBundle=srcArraybundle, dstArrayBundle=dstArraybundle, &
        routehandle=routehandle, factorList=factorList, &
        factorIndexList=factorIndexList, rc=rc)
      if (rc/=ESMF_SUCCESS) return ! bail out
    else
      call ESMF_ArrayBundleSMMStore(&
        srcArrayBundle=srcArraybundle, dstArrayBundle=dstArraybundle, &
        routehandle=routehandle, rc=rc)
      if (rc/=ESMF_SUCCESS) return ! bail out
    endif
    
    print *, "User Coupler Init returning"
   
  end subroutine user_init