adding another test for index code (#618)

* adding new test

* adding test

* new test

* new test

tests/CMakeLists.txt

@@ -121,6 +121,7 @@ if(FTP_TEST_FILES)
     create_test(test_getgb2ir ${kind})
     create_test(test_file_blend ${kind})
     create_test(test_aqm ${kind})
+    create_test(test_create_index_gdas ${kind})
     # This test depends on gdt2gds(), which is not present if the GRIB1
     # library is not also included in the build.
     if (BUILD_WITH_W3EMC)

tests/test_create_index_gdas.F90

@@ -0,0 +1,180 @@
+! This is a test program for NCEPLIBS-g2.
+!
+! This program tests index file functionality with g2_create_index()
+! for one of the files optionally downloaded via FTP during testing:
+! gdas.t12z.pgrb2.1p00.anl.grib2
+!
+! Ed Hartnett 2/17/24
+program test_create_index_gdas
+  use grib_mod
+  implicit none
+
+  ! These are the test files we will use.
+  character(*) :: TEST_FILE_GDAS
+  parameter (TEST_FILE_GDAS = 'gdas.t12z.pgrb2.1p00.anl.grib2')
+  character(*) :: TEST_FILE_GDAS_INDEX
+  parameter (TEST_FILE_GDAS_INDEX = 'test_index_gdas.grb2index')
+  character(len=1), pointer, dimension(:) :: cbuf(:)
+  integer :: idxver, myidxver, nlen, nnum, lugi = 31, lugb = 11
+  integer :: j, jdisc, jpdtn, jgdtn
+  integer :: jids(13), jpdt(100), jgdt(250)
+  integer :: i
+  integer :: k, lpos, iret
+  type(gribfield) :: gfld
+  integer :: expected_idsect(13) = (/ 7, 0, 2, 1, 0, 2022, 11, 6, 12, 0, 0, 0, 0 /)
+  integer :: expected_igdtmpl(19) = (/ 6, 0, 0, 0, 0, 0, 0, 360, 181, 0, -1, &
+       90000000, 0, 48, -90000000, 359000000, 1000000, 1000000, 0 /)
+
+  ! These are the PDT templates of the 19 messages in the test file,
+  ! verified with degrib2.
+  integer :: expected_ipdtmpl(15, 2) = reshape((/ &
+       2, 10, 0, 0, 81, 0, 0, 1, 0, 100, 0, 80000, 255, 0, 0, &
+       2, 10, 0, 0, 81, 0, 0, 1, 0, 100, 0, 85000, 255, 0, 0 /), &
+       shape(expected_ipdtmpl))
+
+  ! These are the DRT templates of the 19 messages in the test file,
+  ! verified with degrib2.
+  integer :: expected_idrtmpl(18, 2) = reshape((/ &
+       -1000579487, 0, 6, 12, 0, 1, 0, 1649987994, -1, 2178, 0, 4, 1, 1, 104, 8, 2, 2, &
+       -997588251, 0, 6, 12, 0, 1, 0, 1649987994, -1, 2159, 0, 4, 1, 1, 104, 7, 2, 2 /), &
+        shape(expected_idrtmpl))
+
+  integer :: ios
+
+  interface
+     subroutine getg2i2(lugi, cbuf, idxver, nlen, nnum, iret)
+       integer, intent(in) :: lugi
+       character(len=1), pointer, dimension(:) :: cbuf
+       integer, intent(out) :: idxver, nlen, nnum, iret
+     end subroutine getg2i2
+     subroutine g2_create_index(lugb, lugi, idxver, filename, iret)
+       integer, intent(in) :: lugb, lugi, idxver
+       character*(*) :: filename
+       integer, intent(out) :: iret
+     end subroutine g2_create_index
+  end interface
+
+  print *, 'Testing index creation and reading.'
+  do idxver = 1, 2
+     print *, 'testing g2_create_index version ', idxver
+
+     ! Open GRIB2 file for reading.
+     call baopenr(lugb, 'data/' // TEST_FILE_GDAS, ios)
+     if (ios .ne. 0) stop 2
+
+     ! Open output file where index will be written.
+     call baopen(lugi, TEST_FILE_GDAS_INDEX, ios)
+     if (ios .ne. 0) stop 3
+
+     call g2_create_index(lugb, lugi, idxver, TEST_FILE_GDAS, iret)
+     if (iret .ne. 0) stop 10
+
+     call baclose(lugb, ios)
+     if (ios .ne. 0) stop 11
+     call baclose(lugi, ios)
+     if (ios .ne. 0) stop 12
+
+     print *, '   OK!'
+     print *, '   testing that index file can be read with getg2i2()...'
+
+     ! Open the index file.
+     call baopen(lugi, TEST_FILE_GDAS_INDEX, iret)
+     if (iret .ne. 0) stop 20
+
+     ! Read the index file.
+     call getg2i2(lugi, cbuf, myidxver, nlen, nnum, iret)
+     print *, myidxver, nlen, nnum, iret
+     if (myidxver .ne. idxver) stop 79
+     if (idxver .eq. 1) then
+        if (nlen .ne. 452) stop 80
+     else
+        if (nlen .ne. 460) stop 80
+     endif
+     if (nnum .ne. 2 .or. iret .ne. 0) stop 81
+
+     ! Close the index file.
+     call baclose(lugi, iret)
+     if (iret .ne. 0) stop 100
+
+     print *, '   OK!'
+
+     ! Parse the index info in cbuf, and fill gfld with the info about
+     ! the first message.
+     jdisc = -1
+     do i = 1, 13
+        jids(i) = -9999
+     end do
+     jpdtn = -1
+     do i = 1, 100
+        jpdt(i) = -9999
+     end do
+     jgdtn = -1
+     do i = 1, 250
+        jgdt(i) = -9999
+     end do
+     do j = 1, nnum
+        print *, '   testing unpacking index buffer with getgb2s2() for message', j
+        call getgb2s2(cbuf, idxver, nlen, nnum, j - 1, jdisc, jids, jpdtn, jpdt, jgdtn, &
+             jgdt, k, gfld, lpos, iret)
+        if (iret .ne. 0) stop 101
+
+        ! print *, gfld%version, gfld%discipline, gfld%idsectlen, gfld%ifldnum, gfld%griddef
+        ! print *, gfld%ngrdpts, gfld%numoct_opt, gfld%interp_opt, gfld%num_opt
+        ! print *, gfld%igdtnum, gfld%igdtlen
+        ! print *, gfld%ipdtnum, gfld%ipdtlen, gfld%num_coord
+        ! print *, gfld%unpacked, gfld%ibmap
+        ! print *, 'sec1:', gfld%idsect
+        ! print *, 'gdt:', gfld%igdtmpl
+
+        ! Check that the information is correct for the first record.
+        if (gfld%version .ne. 2) stop 102
+        if (j .lt. 5) then
+           if (gfld%discipline .ne. 0) stop 103
+        else
+           if (gfld%discipline .ne. 10) stop 104
+        end if
+        if (gfld%idsectlen .ne. 13) stop 110
+        if (gfld%ifldnum .ne. 1) stop 111
+        if (gfld%griddef .ne. 0) stop 112
+        if (gfld%ngrdpts .ne. 65160) stop 120
+        if (gfld%numoct_opt .ne. 0 .or. gfld%interp_opt .ne. 0 .or. gfld%num_opt .ne. 0) stop 122
+        if (gfld%igdtnum .ne. 0 .or. gfld%igdtlen .ne. 19) stop 123
+        if (gfld%ipdtnum .ne. 0 .or. gfld%ipdtlen .ne. 15 .or. gfld%num_coord .ne. 0) stop 130
+        if (gfld%unpacked .neqv. .FALSE.) stop 131
+        if (gfld%ibmap .ne. 255) stop 132
+        do i = 1, gfld%idsectlen
+           if (gfld%idsect(i) .ne. expected_idsect(i)) stop 200
+        end do
+        do i = 1, gfld%igdtlen
+           if (gfld%igdtmpl(i) .ne. expected_igdtmpl(i)) stop 210
+        end do
+        do i = 1, gfld%ipdtlen
+           if (gfld%ipdtmpl(i) .ne. expected_ipdtmpl(i, j)) then
+              print *, i, gfld%ipdtmpl(i), expected_ipdtmpl(i, j)
+              print *, 'gfld%ipdtmpl', gfld%ipdtmpl
+              print *, 'expected_ipdtmpl', expected_ipdtmpl
+              stop 220
+           endif
+        end do
+        do i = 1, gfld%idrtlen
+           if (gfld%idrtmpl(i) .ne. expected_idrtmpl(i, j)) then
+              print *, i, gfld%idrtmpl(i), expected_idrtmpl(i, j)
+              print *, 'gfld%idrtmpl', gfld%idrtmpl
+              print *, 'expected_idrtmpl', expected_idrtmpl
+              stop 230
+           endif
+        end do
+
+        ! Free memory.
+        call gf_free(gfld)
+        print *, '   OK!'
+     end do
+
+     ! Clean up.
+     deallocate(cbuf)
+     call gf_finalize(iret)
+     if (iret .ne. 0) stop 200
+     print *, 'OK!'
+  end do
+  print *, 'SUCCESS!...'
+end program test_create_index_gdas