prt-fmi.f90

Go to the documentation of this file.

module prtfmimodule
 
  use kindmodule, only: dp, i4b, lgp
  use errorutilmodule, only: pstop
  use constantsmodule, only: dzero, lenauxname, lenpackagename, lenvarname
  use simmodule, only: store_error
  use simvariablesmodule, only: errmsg
  use flowmodelinterfacemodule, only: flowmodelinterfacetype
  use basedismodule, only: disbasetype
  use budgetobjectmodule, only: budgetobjecttype
 
  implicit none
  private
  public :: prtfmitype
  public :: fmi_cr
 
  character(len=LENPACKAGENAME) :: text = '    PRTFMI'
 
  type, extends(flowmodelinterfacetype) :: prtfmitype
    private
    integer(I4B), public :: max_faces !< maximum number of faces for grid cell polygons
    real(dp), allocatable, public :: sourceflows(:) ! cell source flows array
    real(dp), allocatable, public :: sinkflows(:) ! cell sink flows array
    real(dp), allocatable, public :: storageflows(:) ! cell storage flows array
    real(dp), allocatable, public :: boundaryflows(:) ! cell boundary flows array
    integer(I4B), allocatable, public :: boundaryfaces(:) ! bitmask of assigned boundary faces
 
  contains
 
    procedure :: fmi_ad
    procedure :: fmi_df => prtfmi_df
    procedure, private :: accumulate_flows
    procedure :: mark_boundary_face
    procedure :: is_boundary_face
    procedure :: is_net_out_boundary_face
    procedure, private :: iflowface_to_icellface
 
  end type prtfmitype
 
contains
 
  !> @brief Create a new PrtFmi object
  subroutine fmi_cr(fmiobj, name_model, input_mempath, inunit, iout)
    ! dummy
    type(prtfmitype), pointer :: fmiobj
    character(len=*), intent(in) :: name_model
    character(len=*), intent(in) :: input_mempath
    integer(I4B), intent(inout) :: inunit
    integer(I4B), intent(in) :: iout
 
    ! Create the object
    allocate (fmiobj)
 
    ! create name and memory path
    call fmiobj%set_names(1, name_model, 'FMI', 'FMI', input_mempath)
    fmiobj%text = text
 
    ! Allocate scalars
    call fmiobj%allocate_scalars()
 
    ! Set variables
    fmiobj%inunit = inunit
    fmiobj%iout = iout
 
    ! Assign dependent variable label
    fmiobj%depvartype = 'TRACKS          '
 
  end subroutine fmi_cr
 
  !> @brief Time step advance
  subroutine fmi_ad(this)
    ! modules
    use constantsmodule, only: dhdry
    ! dummy
    class(prtfmitype) :: this
    ! local
    integer(I4B) :: n
    character(len=15) :: nodestr
    character(len=*), parameter :: fmtdry = &
     &"(/1X,'WARNING: DRY CELL ENCOUNTERED AT ',a,';  RESET AS INACTIVE')"
    character(len=*), parameter :: fmtrewet = &
     &"(/1X,'DRY CELL REACTIVATED AT ', a)"
 
    ! Set flag to indicated that flows are being updated.  For the case where
    ! flows may be reused (only when flows are read from a file) then set
    ! the flag to zero to indicated that flows were not updated
    this%iflowsupdated = 1
 
    ! If reading flows from a budget file, read the next set of records
    if (this%iubud /= 0) call this%advance_bfr()
 
    ! If reading heads from a head file, read the next set of records
    if (this%iuhds /= 0) call this%advance_hfr()
 
    ! If mover flows are being read from file, read the next set of records
    if (this%iumvr /= 0) &
      call this%mvrbudobj%bfr_advance(this%dis, this%iout)
 
    ! Accumulate flows
    call this%accumulate_flows()
 
    ! if flow cell is dry, then set this%ibound = 0
    do n = 1, this%dis%nodes
      ! Calculate the ibound-like array that has 0 if saturation
      ! is zero and 1 otherwise
      if (this%gwfsat(n) > dzero) then
        this%ibdgwfsat0(n) = 1
      else
        this%ibdgwfsat0(n) = 0
      end if
 
      ! Check if active model cell is inactive for flow
      if (this%ibound(n) > 0) then
        if (this%gwfhead(n) == dhdry) then
          ! cell should be made inactive
          this%ibound(n) = 0
          call this%dis%noder_to_string(n, nodestr)
          write (this%iout, fmtdry) trim(nodestr)
        end if
      end if
 
      ! Convert dry model cell to active if flow has rewet
      if (this%ibound(n) == 0) then
        if (this%gwfhead(n) /= dhdry) then
          ! cell is now wet
          this%ibound(n) = 1
          call this%dis%noder_to_string(n, nodestr)
          write (this%iout, fmtrewet) trim(nodestr)
        end if
      end if
    end do
 
  end subroutine fmi_ad
 
  !> @brief Define the flow model interface
  subroutine prtfmi_df(this, dis, idryinactive)
    class(prtfmitype) :: this
    class(disbasetype), pointer, intent(in) :: dis
    integer(I4B), intent(in) :: idryinactive
 
    call this%FlowModelInterfaceType%fmi_df(dis, idryinactive)
 
    this%max_faces = this%dis%get_max_npolyverts() + 2
    allocate (this%StorageFlows(this%dis%nodes))
    allocate (this%SourceFlows(this%dis%nodes))
    allocate (this%SinkFlows(this%dis%nodes))
    allocate (this%BoundaryFlows(this%dis%nodes * this%max_faces))
    allocate (this%BoundaryFaces(this%dis%nodes))
 
  end subroutine prtfmi_df
 
  !> @brief Accumulate flows
  subroutine accumulate_flows(this)
    ! dummy
    class(prtfmitype) :: this
    ! local
    integer(I4B) :: j, i, ip, ib
    integer(I4B) :: ioffset, iflowface, iauxiflowface, icellface
    real(DP) :: qbnd
    character(len=LENAUXNAME) :: auxname
    integer(I4B) :: naux
 
    this%StorageFlows = dzero
    if (this%igwfstrgss /= 0) &
      this%StorageFlows = this%StorageFlows + this%gwfstrgss
    if (this%igwfstrgsy /= 0) &
      this%StorageFlows = this%StorageFlows + this%gwfstrgsy
 
    this%SourceFlows = dzero
    this%SinkFlows = dzero
    this%BoundaryFlows = dzero
    this%BoundaryFaces = 0
    do ip = 1, this%nflowpack
      iauxiflowface = 0
      naux = this%gwfpackages(ip)%naux
      if (naux > 0) then
        do j = 1, naux
          auxname = this%gwfpackages(ip)%auxname(j)
          if (trim(adjustl(auxname)) == "IFLOWFACE") then
            iauxiflowface = j
            exit
          end if
        end do
      end if
      do ib = 1, this%gwfpackages(ip)%nbound
        i = this%gwfpackages(ip)%nodelist(ib)
        if (i <= 0) cycle
        if (this%ibound(i) <= 0) cycle
        qbnd = this%gwfpackages(ip)%get_flow(ib)
        ! todo, after initial release: default iflowface values for different packages
        iflowface = 0
        icellface = 0
        if (iauxiflowface > 0) then
          iflowface = nint(this%gwfpackages(ip)%auxvar(iauxiflowface, ib))
          icellface = this%iflowface_to_icellface(iflowface)
        end if
        if (icellface > 0) then
          call this%mark_boundary_face(i, icellface)
          ioffset = (i - 1) * this%max_faces
          this%BoundaryFlows(ioffset + icellface) = &
            this%BoundaryFlows(ioffset + icellface) + qbnd
        else if (qbnd .gt. dzero) then
          this%SourceFlows(i) = this%SourceFlows(i) + qbnd
        else if (qbnd .lt. dzero) then
          this%SinkFlows(i) = this%SinkFlows(i) + qbnd
        end if
      end do
    end do
 
  end subroutine accumulate_flows
 
  !> @brief Mark a face as a boundary face.
  subroutine mark_boundary_face(this, ic, iface)
    class(prtfmitype) :: this
    integer(I4B), intent(in) :: ic !< node number (reduced)
    integer(I4B), intent(in) :: iface !< cell face number
    ! local
    integer(I4B) :: bit_pos
 
    if (ic <= 0 .or. ic > this%dis%nodes) then
      print *, 'Invalid cell number: ', ic
      print *, 'Expected a value in range [1, ', this%dis%nodes, ']'
      call pstop(1)
    end if
    if (iface <= 0) then
      print *, 'Invalid face number: ', iface
      print *, 'Expected a value in range [1, ', this%max_faces, ']'
      call pstop(1)
    end if
    bit_pos = iface - 1 ! bit position 0-based
    if (bit_pos < 0 .or. bit_pos > 31) then
      print *, 'Invalid bitmask position: ', iface
      print *, 'Expected a value in range [0, 31]'
      call pstop(1)
    end if
    this%BoundaryFaces(ic) = ibset(this%BoundaryFaces(ic), bit_pos)
  end subroutine mark_boundary_face
 
  !> @brief Check if a face is assigned to a boundary package.
  function is_boundary_face(this, ic, iface) result(is_boundary)
    class(prtfmitype) :: this
    integer(I4B), intent(in) :: ic !< node number (reduced)
    integer(I4B), intent(in) :: iface !< cell face number
    logical(LGP) :: is_boundary
    ! local
    integer(I4B) :: bit_pos
 
    is_boundary = .false.
    if (ic <= 0 .or. ic > this%dis%nodes) then
      print *, 'Invalid cell number: ', ic
      print *, 'Expected a value in range [1, ', this%dis%nodes, ']'
      call pstop(1)
    end if
    if (iface <= 0) then
      print *, 'Invalid face number: ', iface
      print *, 'Expected a value in range [1, ', this%max_faces, ']'
      call pstop(1)
    end if
    bit_pos = iface - 1 ! bit position 0-based
    if (bit_pos < 0 .or. bit_pos > 31) then
      print *, 'Invalid bitmask position: ', iface
      print *, 'Expected a value in range [0, 31]'
      call pstop(1)
    end if
    is_boundary = btest(this%BoundaryFaces(ic), bit_pos)
  end function is_boundary_face
 
  !> @brief Check if a face is an assigned boundary with net outflow.
  function is_net_out_boundary_face(this, ic, iface) result(is_net_out_boundary)
    class(prtfmitype) :: this
    integer(I4B), intent(in) :: ic !< node number (reduced)
    integer(I4B), intent(in) :: iface !< cell face number
    logical(LGP) :: is_net_out_boundary
    ! local
    integer(I4B) :: ioffset
 
    is_net_out_boundary = .false.
    if (.not. this%is_boundary_face(ic, iface)) return
    ioffset = (ic - 1) * this%max_faces
    if (this%BoundaryFlows(ioffset + iface) < dzero) is_net_out_boundary = .true.
  end function is_net_out_boundary_face
 
  !> @brief Convert an iflowface number to a cell face number.
  !! Maps bottom (-2) -> max_faces - 1, top (-1) -> max_faces.
  function iflowface_to_icellface(this, iflowface) result(iface)
    class(prtfmitype), intent(inout) :: this
    integer(I4B), intent(in) :: iflowface
    integer(I4B) :: iface
 
    iface = iflowface
    if (iface < 0) iface = iface + this%max_faces + 1
  end function iflowface_to_icellface
 
end module prtfmimodule