Fix exclusive_size for F14FastMap

Author: ttreyer

test/integration/folly_f14_fast_map.toml

@@ -8,7 +8,7 @@ definitions = '''
     folly::F14FastMap<int, int> m1;
     folly::F14FastMap<int, Bar> m2;
     folly::F14FastMap<int, int> m3;
-    folly::F14FastMap<int, int> m4;
+    folly::F14FastMap<int, long> m4;
   };
 '''
 
@@ -42,7 +42,7 @@ definitions = '''
     '''
     expect_json = '''[{
       "staticSize":96,
-      "dynamicSize":304,
+      "dynamicSize":368,
       "members":[
         {"name":"m1", "staticSize":24, "dynamicSize":48, "length":3, "capacity":3, "elementStaticSize":8},
         {
@@ -65,14 +65,31 @@ definitions = '''
             {"staticSize":8, "members":[{"name":"a"}, {"name": "b"}]}
         ]},
         {"name":"m3", "staticSize":24, "dynamicSize":80, "length":7, "capacity":7, "elementStaticSize":8},
-        {"name":"m4", "staticSize":24, "dynamicSize":96, "length":9, "capacity":9, "elementStaticSize":8}
+        {"name":"m4", "staticSize":24, "dynamicSize":160, "length":9, "capacity":9, "elementStaticSize":12}
       ]}]'''
     expect_json_v2 = '''[{
       "staticSize":96,
       "exclusiveSize": 0,
       "members":[
-        {"name":"m1", "staticSize":24, "exclusiveSize": 24, "length": 3, "capacity": 3},
-        {"name":"m2", "staticSize":24, "exclusiveSize": 24, "length": 5, "capacity": 5},
-        {"name":"m3", "staticSize":24, "exclusiveSize": 24, "length": 7, "capacity": 7},
-        {"name":"m4", "staticSize":24, "exclusiveSize": 24, "length": 9, "capacity": 9}
-      ]}]'''
+        {"name":"m1", "staticSize":24, "exclusiveSize": 48, "length": 3, "capacity": 3},
+        {"name":"m2", "staticSize":24, "exclusiveSize": 44, "length": 5, "capacity": 5},
+        {"name":"m3", "staticSize":24, "exclusiveSize": 48, "length": 7, "capacity": 7},
+        {
+          "name":"m4",
+          "staticSize":24,
+          "exclusiveSize": 40,
+          "length": 9,
+          "capacity": 9,
+          "members":[
+            {"staticSize":16, "exclusiveSize": 4},
+            {"staticSize":16, "exclusiveSize": 4},
+            {"staticSize":16, "exclusiveSize": 4},
+            {"staticSize":16, "exclusiveSize": 4},
+            {"staticSize":16, "exclusiveSize": 4},
+            {"staticSize":16, "exclusiveSize": 4},
+            {"staticSize":16, "exclusiveSize": 4},
+            {"staticSize":16, "exclusiveSize": 4},
+            {"staticSize":16, "exclusiveSize": 4}
+          ]
+        }]
+      }]'''

types/f14_fast_map.toml

@@ -71,11 +71,9 @@ struct TypeHandler<DB, %1%<T0, T1, T2, T3, T4>> {
 """
 
 traversal_func = """
-// TODO: This implementation enables the traversal of the container,
-//       but doesn't report the memory footprint accurately.
-//       Revisit this implementation and fix memory footprint reporting.
 auto tail = returnArg
-  .write((uintptr_t)&container)
+  .write((uintptr_t)container.getAllocatedMemorySize())
+  .write((uintptr_t)container.bucket_count())
   .write(container.size());
 
 for (auto &&entry: container) {
@@ -94,26 +92,40 @@ return tail.finish();
 type = "types::st::VarInt<DB>"
 func = "el.pointer = std::get<ParsedData::VarInt>(d.val).value;"
 
+[[codegen.processor]]
+type = "types::st::VarInt<DB>"
+func = """
+el.container_stats.emplace(result::Element::ContainerStats {
+  .capacity = std::get<ParsedData::VarInt>(d.val).value,
+});
+"""
+
 [[codegen.processor]]
 type = """
 types::st::List<DB, types::st::Pair<DB,
   typename TypeHandler<DB, T0>::type,
   typename TypeHandler<DB, T1>::type>>
 """
 func = """
+static constexpr size_t element_size = sizeof(typename container_type::value_type);
+
+auto allocationSize = el.pointer.value();
+el.pointer.reset();
+
 auto list = std::get<ParsedData::List>(d.val);
-el.container_stats.emplace(result::Element::ContainerStats {
-  .capacity = list.length,
-  .length = list.length,
-});
+el.container_stats->length = list.length;
+
+el.exclusive_size += allocationSize - list.length * element_size;
 
 static constexpr std::array<inst::Field, 2> element_fields{
   make_field<DB, T0>("key"),
   make_field<DB, T1>("value"),
 };
 
 static constexpr inst::Field element{
-  0, 0, "[]",
+  element_size,
+  element_size - sizeof(T0) - sizeof(T1),
+  "[]",
   std::array<std::string_view, 0>{},
   element_fields,
   std::array<inst::ProcessorInst, 0>{},