Fixed indexing of id dimension coordinate

MintonGroup · Dec 10, 2022 · 5d72cf4 · 5d72cf4
1 parent d383507
commit 5d72cf4
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 6 deletions.
diff --git a/src/encounter/encounter_io.f90 b/src/encounter/encounter_io.f90
@@ -84,7 +84,7 @@ module subroutine encounter_io_initialize(self, param)
          ! Dimensions
          call check( nf90_def_dim(nc%id, nc%time_dimname, nc%time_dimsize, nc%time_dimid), "encounter_io_initialize nf90_def_dim time_dimid" ) ! Simulation time dimension
          call check( nf90_def_dim(nc%id, nc%space_dimname, NDIM , nc%space_dimid), "encounter_io_initialize nf90_def_dim space_dimid" )           ! 3D space dimension
-         call check( nf90_def_dim(nc%id, nc%id_dimname, param%maxid, nc%id_dimid), "encounter_io_initialize nf90_def_dim id_dimid" )       ! dimension to store particle id numbers
+         call check( nf90_def_dim(nc%id, nc%id_dimname, param%maxid+1, nc%id_dimid), "encounter_io_initialize nf90_def_dim id_dimid" )       ! dimension to store particle id numbers
          call check( nf90_def_dim(nc%id, nc%str_dimname, NAMELEN, nc%str_dimid), "encounter_io_initialize nf90_def_dim str_dimid"  )          ! Dimension for string variables (aka character arrays)
 
          ! Dimension coordinates
@@ -129,7 +129,7 @@ module subroutine encounter_io_initialize(self, param)
          call check( nf90_put_var(nc%id, nc%space_varid, nc%space_coords, start=[1], count=[NDIM]), "encounter_io_initialize nf90_put_var space"  )
 
          ! Pre-fill name slots with ids 
-         call check( nf90_put_var(nc%id, nc%id_varid, [(-1,i=1,param%maxid)], start=[1], count=[param%maxid]), "encounter_io_initialize nf90_put_var pl id_varid"  )
+         call check( nf90_put_var(nc%id, nc%id_varid, [(-1,i=1,param%maxid+1)], start=[1], count=[param%maxid+1]), "encounter_io_initialize nf90_put_var pl id_varid"  )
       end associate
 
       return

diff --git a/src/fraggle/fraggle_io.f90 b/src/fraggle/fraggle_io.f90
@@ -54,7 +54,7 @@ module subroutine fraggle_io_initialize_output(self, param)
          ! Dimensions
          call check( nf90_def_dim(nc%id, nc%event_dimname, NF90_UNLIMITED, nc%event_dimid), "fraggle_io_initialize nf90_def_dim event_dimid"  ) ! Dimension to store individual collision events
          call check( nf90_def_dim(nc%id, nc%space_dimname, NDIM,           nc%space_dimid), "fraggle_io_initialize nf90_def_dim space_dimid" )  ! 3D space dimension
-         call check( nf90_def_dim(nc%id, nc%id_dimname,    param%maxid,    nc%id_dimid),    "fraggle_io_initialize nf90_def_dim id_dimid" )     ! Dimension to store particle id numbers
+         call check( nf90_def_dim(nc%id, nc%id_dimname,    param%maxid+1,  nc%id_dimid),    "fraggle_io_initialize nf90_def_dim id_dimid" )     ! Dimension to store particle id numbers
          call check( nf90_def_dim(nc%id, nc%str_dimname,   NAMELEN,        nc%str_dimid),   "fraggle_io_initialize nf90_def_dim str_dimid"  )   ! Dimension for string variables (aka character arrays)
          call check( nf90_def_dim(nc%id, nc%stage_dimname, 2,              nc%stage_dimid), "fraggle_io_initialize nf90_def_dim stage_dimid"  ) ! Dimension for stage variables (aka "before" vs. "after"
 
@@ -138,7 +138,7 @@ module subroutine fraggle_io_initialize_output(self, param)
          call check( nf90_put_var(nc%id, nc%stage_varid, nc%stage_coords(2), start=[1,2], count=[len(nc%stage_coords(2)),1]), "fraggle_io_initialize nf90_put_var stage 2"  )
 
          ! Pre-fill id slots with ids
-         call check( nf90_put_var(nc%id, nc%id_varid, [(-1,i=1,param%maxid)], start=[1], count=[param%maxid]), "fraggle_io_initialize nf90_put_varid_varid"  )
+         call check( nf90_put_var(nc%id, nc%id_varid, [(-1,i=1,param%maxid+1)], start=[1], count=[param%maxid+1]), "fraggle_io_initialize nf90_put_varid_varid"  )
       end associate
 
       return
@@ -208,8 +208,6 @@ module subroutine fraggle_io_write_frame(self, nc, param)
             call check( nf90_put_var(nc%id, nc%L_orb_varid,  fragments%Lorbit_after(:),  start=[1, 2, eslot], count=[NDIM, 1, eslot]), "fraggle_io_write_frame nf90_put_var L_orb_varid after" )
             call check( nf90_put_var(nc%id, nc%L_spin_varid, fragments%Lspin_before(:),  start=[1, 1, eslot], count=[NDIM, 1, eslot]), "fraggle_io_write_frame nf90_put_var L_spin_varid before" )
             call check( nf90_put_var(nc%id, nc%L_spin_varid, fragments%Lspin_after(:),   start=[1, 2, eslot], count=[NDIM, 1, eslot]), "fraggle_io_write_frame nf90_put_var L_spin_varid after" )
-
-
 
             call check( nf90_set_fill(nc%id, old_mode, old_mode) )
          end select