CsrUtils.f90

Go to the documentation of this file.

module csrutilsmodule
 
  implicit none
  private
 
  public :: getcsrindex
 
contains
 
  !> @brief Return index for element i,j in CSR storage,
  !< returns -1 when not there
  function getcsrindex(i, j, ia, ja) result(csrIndex)
    use kindmodule, only: i4b
    integer(I4B), intent(in) :: i !< the row index
    integer(I4B), intent(in) :: j !< the column index
    integer(I4B), dimension(:), intent(in) :: ia !< CSR ia array
    integer(I4B), dimension(:), intent(in) :: ja !< CSR ja array
    integer(I4B) :: csrindex !< the CSR ndex of element i,j
    ! local
    integer(I4B) :: idx
 
    csrindex = -1
    do idx = ia(i), ia(i + 1) - 1
      if (ja(idx) == j) then
        csrindex = idx
        return
      end if
    end do
 
  end function
 
end module