reset hdf5 every round on multiple hdf5s

salilab · Oct 31, 2023 · 36a8ac8 · 36a8ac8
1 parent 11de2cc
commit 36a8ac8
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 6 deletions.
diff --git a/include/Statistics.h b/include/Statistics.h
@@ -108,7 +108,6 @@ class IMPNPCTRANSPORTEXPORT Statistics : public Object {
     t_sparse_3d_matrix;
   typedef IMP_KERNEL_LARGE_UNORDERED_MAP< core::ParticleType,
     t_sparse_3d_matrix >
-    //    std::vector< std::vector< std::vector<int> > > >
     ParticleTypeXYZDistributionMap;
   ParticleTypeXYZDistributionMap particle_type_xyz_distribution_map_;
   struct t_size_3d_matrix{

diff --git a/src/Statistics.cpp b/src/Statistics.cpp
@@ -943,11 +943,6 @@ void Statistics::update
     }
   }
 
-  // TODO: disable this for now
-  //  ::npctransport_proto::Conformation *conformation =
-  //   output.mutable_conformation();
-  //    save_pb_conformation(get_beads(), sites_, conformation);
-
   // save RMF for future restarts
   if(!no_save_rmf_to_output){
     RMF::BufferHandle buf;
@@ -963,7 +958,14 @@ void Statistics::update
   std::ofstream outf(output_file_name_.c_str(), std::ios::binary);
   output.SerializeToOstream(&outf);
   outf.flush();
+
+  // reset HDF5 stats if having multiple HDF5s
+  if(get_sd()->get_is_multiple_hdf5s())
+  {
+    particle_type_zr_distribution_map_.clear();
+    particle_type_xyz_distribution_map_.clear();
   }
+}
 
 void Statistics::reset_statistics_optimizer_states()
 {