#156: footprinting: support vector of non-serializable elements

DARMA-tasking · Nov 19, 2020 · 98a67ab · 98a67ab
1 parent 56f3104
commit 98a67ab
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 0 deletions.
diff --git a/src/checkpoint/container/vector_serialize.h b/src/checkpoint/container/vector_serialize.h
@@ -102,6 +102,22 @@ void serialize(Serializer& s, std::vector<bool, VectorAllocator>& vec) {
   }
 }
 
+template <
+  typename SerializerT,
+  typename T,
+  typename VectorAllocator,
+  typename = std::enable_if_t<
+    std::is_same< // is_footprintable
+      size_t,
+      decltype(checkpoint::getMemoryFootprint(std::declval<T>()))
+    >::value
+  >
+>
+void serialize(SerializerT& s, std::vector<T, VectorAllocator>& vec) {
+  s.countBytes(vec);
+  s.addBytes(vec.capacity() * sizeof(T));
+}
+
 } /* end namespace checkpoint */
 
 #endif /*INCLUDED_CHECKPOINT_CONTAINER_VECTOR_SERIALIZE_H*/
diff --git a/tests/unit/test_footprinter.cc b/tests/unit/test_footprinter.cc
@@ -586,4 +586,17 @@ TEST_F(TestFootprinter, test_ompi) {
   }
 }
 
+struct TestNoSerialize {
+  double d;
+  int i;
+};
+
+TEST_F(TestFootprinter, test_no_serialize) {
+  std::vector<TestNoSerialize> v(7);
+
+  EXPECT_EQ(
+    checkpoint::getMemoryFootprint(v),
+    sizeof(v) + v.capacity() * sizeof(TestNoSerialize)
+  );
+}
 }}} // end namespace checkpoint::tests::unit