Feature: loadtxt skiprows and max_rows

src/stdlib_io.fypp

@@ -81,7 +81,7 @@ module stdlib_io
 contains
 
  #:for k1, t1 in KINDS_TYPES
- subroutine loadtxt_${t1[0]}$${k1}$(filename, d)
+ subroutine loadtxt_${t1[0]}$${k1}$(filename, d, skiprows, max_rows)
  !! version: experimental
  !!
  !! Loads a 2D array from a text file.
@@ -93,6 +93,10 @@ contains
  character(len=*), intent(in) :: filename
  !! The array 'd' will be automatically allocated with the correct dimensions
  ${t1}$, allocatable, intent(out) :: d(:,:)
+ !! Skip the first `skiprows` lines. If skipping more rows than present, a 0-sized array will be returned. The default is 0.
+ integer, intent(in), optional :: skiprows
+ !! Read max_rows lines of content after skiprows lines. The default is -1, to read all the lines.
+ integer, intent(in), optional :: max_rows
  !!
  !! Example
  !! -------
@@ -111,7 +115,10 @@ contains
  !! ...
  !!
  integer :: s
- integer :: nrow, ncol, i
+ integer :: nrow, ncol, i, skiprows_, max_rows_
+
+ skiprows_ = max(optval(skiprows, 0), 0)
+ max_rows_ = optval(max_rows, -1)
 
  s = open(filename)
 
@@ -123,9 +130,16 @@ contains
 
  ! determine number or rows
  nrow = number_of_rows(s)
+ skiprows_ = min(skiprows_, nrow)
+ if ( max_rows_ < 0 .or. max_rows_ > (nrow - skiprows_)) max_rows_ = nrow - skiprows_
+
+ allocate(d(max_rows_, ncol))
+
+ do i = 1, skiprows_
+ read(s, *)
+ end do
 
- allocate(d(nrow, ncol))
- do i = 1, nrow
+ do i = 1, max_rows_
  #:if 'real' in t1
  read(s, "(*"//FMT_REAL_${k1}$(1:len(FMT_REAL_${k1}$)-1)//",1x))") d(i, :)
  #:elif 'complex' in t1

src/tests/io/test_loadtxt.f90

@@ -19,6 +19,7 @@ subroutine collect_loadtxt(testsuite)
  new_unittest("loadtxt_sp_huge", test_loadtxt_sp_huge), &
  new_unittest("loadtxt_sp_tiny", test_loadtxt_sp_tiny), &
  new_unittest("loadtxt_dp", test_loadtxt_dp), &
+ new_unittest("test_loadtxt_dp_max_skip", test_loadtxt_dp_max_skip), &
  new_unittest("loadtxt_dp_huge", test_loadtxt_dp_huge), &
  new_unittest("loadtxt_dp_tiny", test_loadtxt_dp_tiny), &
  new_unittest("loadtxt_complex", test_loadtxt_complex) &
@@ -134,6 +135,29 @@ subroutine test_loadtxt_dp(error)
  end subroutine test_loadtxt_dp
 
 
+ subroutine test_loadtxt_dp_max_skip(error)
+ !> Error handling
+ type(error_type), allocatable, intent(out) :: error
+ real(dp), allocatable :: input(:,:), expected(:,:)
+ integer :: n, m
+
+ allocate(input(10,10))
+
+ do m = 0, 5
+ do n = 1, 11
+ call random_number(input)
+ input = input - 0.5
+ call savetxt('test_dp_max_skip.txt', input)
+ call loadtxt('test_dp_max_skip.txt', expected, skiprows=m, max_rows=n)
+ call check(error, all(input(m+1:min(n+m,10),:) == expected))
+ deallocate(expected)
+ if (allocated(error)) return
+ end do
+ end do
+
+ end subroutine test_loadtxt_dp_max_skip
+
+
  subroutine test_loadtxt_dp_huge(error)
  !> Error handling
  type(error_type), allocatable, intent(out) :: error