slice

Author: scivision

src/interface.f90

@@ -105,55 +105,62 @@ module subroutine nc_write_1d(self, dname, value, dims, istart, iend, chunk_size
 character(*), intent(in) :: dname
 class(*), intent(in) :: value(:)
 character(*), intent(in), optional :: dims(1)
-integer, intent(in), dimension(1), optional :: istart, iend, chunk_size
+integer, intent(in), dimension(1), optional :: istart, iend
+integer, intent(in), dimension(1), optional :: chunk_size
 end subroutine
 
 module subroutine nc_write_2d(self, dname, value, dims, istart, iend, chunk_size)
 class(netcdf_file), intent(in) :: self
 character(*), intent(in) :: dname
 class(*), intent(in) :: value(:,:)
 character(*), intent(in), optional :: dims(2)
-integer, intent(in), dimension(2), optional :: istart, iend, chunk_size
+integer, intent(in), dimension(2), optional :: istart, iend
+integer, intent(in), dimension(2), optional :: chunk_size
 end subroutine
 
 module subroutine nc_write_3d(self, dname, value, dims, istart, iend, chunk_size)
 class(netcdf_file), intent(in) :: self
 character(*), intent(in) :: dname
 class(*), intent(in) :: value(:,:,:)
 character(*), intent(in), optional :: dims(3)
-integer, intent(in), dimension(3), optional :: istart, iend, chunk_size
+integer, intent(in), dimension(3), optional :: istart, iend
+integer, intent(in), dimension(3), optional :: chunk_size
 end subroutine
 
 module subroutine nc_write_4d(self, dname, value, dims, istart, iend, chunk_size)
 class(netcdf_file), intent(in) :: self
 character(*), intent(in) :: dname
 class(*), intent(in) :: value(:,:,:,:)
 character(*), intent(in), optional :: dims(4)
-integer, intent(in), dimension(4), optional :: istart, iend, chunk_size
+integer, intent(in), dimension(4), optional :: istart, iend
+integer, intent(in), dimension(4), optional :: chunk_size
 end subroutine
 
 module subroutine nc_write_5d(self, dname, value, dims, istart, iend, chunk_size)
 class(netcdf_file), intent(in) :: self
 character(*), intent(in) :: dname
 class(*), intent(in) :: value(:,:,:,:,:)
 character(*), intent(in), optional :: dims(5)
-integer, intent(in), dimension(5), optional :: istart, iend, chunk_size
+integer, intent(in), dimension(5), optional :: istart, iend
+integer, intent(in), dimension(5), optional :: chunk_size
 end subroutine
 
 module subroutine nc_write_6d(self, dname, value, dims, istart, iend, chunk_size)
 class(netcdf_file), intent(in) :: self
 character(*), intent(in) :: dname
 class(*), intent(in) :: value(:,:,:,:,:,:)
 character(*), intent(in), optional :: dims(6)
-integer, intent(in), dimension(6), optional :: istart, iend, chunk_size
+integer, intent(in), dimension(6), optional :: istart, iend
+integer, intent(in), dimension(6), optional :: chunk_size
 end subroutine
 
 module subroutine nc_write_7d(self, dname, value, dims, istart, iend, chunk_size)
 class(netcdf_file), intent(in) :: self
 character(*), intent(in) :: dname
 class(*), intent(in) :: value(:,:,:,:,:,:,:)
 character(*), intent(in), optional :: dims(7)
-integer, intent(in), dimension(7), optional :: istart, iend, chunk_size
+integer, intent(in), dimension(7), optional :: istart, iend
+integer, intent(in), dimension(7), optional :: chunk_size
 end subroutine
 
 end interface

src/reader.inc

@@ -1,4 +1,4 @@
-integer :: varid, ier, drank
+integer :: varid, ier, drank, i(rank(value))
 
 if(.not.self%is_open) error stop 'nc4fortran:reader file handle not open'
 
@@ -12,17 +12,24 @@ if(drank /= rank(value)) then
   error stop
 endif
 
+if(present(istart) .and. present(iend)) then
+  i = iend - istart + 1
+else
+  i = shape(value)
+endif
+
 select type (value)
 type is (real(real64))
-  ier = nf90_get_var(self%file_id, varid, value)
+  ier = nf90_get_var(self%file_id, varid, value, start=istart, count=i)
 type is (real(real32))
-  ier = nf90_get_var(self%file_id, varid, value)
+  ier = nf90_get_var(self%file_id, varid, value, start=istart, count=i)
 type is (integer(int64))
-  ier = nf90_get_var(self%file_id, varid, value)
+  ier = nf90_get_var(self%file_id, varid, value, start=istart, count=i)
 type is (integer(int32))
-  ier = nf90_get_var(self%file_id, varid, value)
+  ier = nf90_get_var(self%file_id, varid, value, start=istart, count=i)
 class default
   ier = NF90_EBADTYPE
 end select
 
+
 if (check_error(ier, dname)) error stop 'nc4fortran:read read ' // dname // ' in ' // self%filename

src/writer.inc

@@ -1,5 +1,5 @@
 integer :: varid, ier, dtype
-integer, dimension(rank(value)) :: i0, i1
+integer, dimension(rank(value)) :: i
 
 if (self%exist(dname)) then
   ier = nf90_inq_varid(self%file_id, dname, varid)
@@ -20,21 +20,21 @@ else
   call nc_create(self, dname, dtype, dims=shape(value), dim_names=dims, chunk_size=chunk_size, varid=varid)
 endif
 
-i0 = 1
-if(present(istart)) i0 = istart
-
-i1 = shape(value)
-if(present(iend)) i1 = iend
+if(present(istart) .and. present(iend)) then
+  i = iend - istart + 1
+else
+  i = shape(value)
+endif
 
 select type (value)
 type is (real(real32))
-  ier = nf90_put_var(self%file_id, varid, value, start=istart)
+  ier = nf90_put_var(self%file_id, varid, value, start=istart, count=i)
 type is (real(real64))
-  ier = nf90_put_var(self%file_id, varid, value, start=istart)
+  ier = nf90_put_var(self%file_id, varid, value, start=istart, count=i)
 type is (integer(int32))
-  ier = nf90_put_var(self%file_id, varid, value, start=istart)
+  ier = nf90_put_var(self%file_id, varid, value, start=istart, count=i)
 type is (integer(int64))
-  ier = nf90_put_var(self%file_id, varid, value, start=istart)
+  ier = nf90_put_var(self%file_id, varid, value, start=istart, count=i)
 class default
   ier = NF90_EBADTYPE
 end select