Implement Container V2 for F14NodeSet

test/integration/folly_f14_node_set.toml

@@ -0,0 +1,83 @@
+includes = ["folly/container/F14Set.h"]
+definitions = '''
+  struct Bar {
+    float a, b;
+  };
+
+  inline bool operator==(const Bar &lhs, const Bar &rhs) noexcept {
+    return lhs.a == rhs.a && lhs.b == rhs.b;
+  }
+
+  struct BarHasher {
+    inline size_t operator()(const Bar &bar) const {
+      return folly::hash::hash_combine(bar.a, bar.b);
+    }
+  };
+
+  struct Foo {
+    folly::F14NodeSet<int> m1;
+    folly::F14NodeSet<Bar, BarHasher> m2;
+    folly::F14NodeSet<int> m3;
+    folly::F14NodeSet<int> m4;
+  };
+'''
+
+[cases]
+  [cases.a]
+    param_types = ["const Foo&"]
+    setup = '''
+      Foo foo;
+
+      foo.m1.reserve(3);
+      for (int i = 0; i < 3; i++) {
+        foo.m1.emplace(i);
+      }
+
+      foo.m2.reserve(5);
+      for (int i = 0; i < 5; i++) {
+        foo.m2.emplace(Bar{(float)i, 0.0f});
+      }
+
+      foo.m3.reserve(7);
+      for (int i = 0; i < 7; i++) {
+        foo.m3.emplace(i);
+      }
+
+      foo.m4.reserve(9);
+      for (int i = 0; i < 9; i++) {
+        foo.m4.emplace(i);
+      }
+
+      return {foo};
+    '''
+    expect_json = '''[{
+      "staticSize":96,
+      "dynamicSize":404,
+      "members":[
+        {"name":"m1", "staticSize":24, "dynamicSize":60, "length":3, "capacity":3, "elementStaticSize":4},
+        {
+          "name":"m2",
+          "staticSize":24,
+          "dynamicSize":104,
+          "length":5,
+          "capacity":5,
+          "elementStaticSize":8,
+          "members":[
+            {"staticSize":8, "members":[{"name":"a"}, {"name": "b"}]},
+            {"staticSize":8, "members":[{"name":"a"}, {"name": "b"}]},
+            {"staticSize":8, "members":[{"name":"a"}, {"name": "b"}]},
+            {"staticSize":8, "members":[{"name":"a"}, {"name": "b"}]},
+            {"staticSize":8, "members":[{"name":"a"}, {"name": "b"}]}
+        ]},
+        {"name":"m3", "staticSize":24, "dynamicSize":108, "length":7, "capacity":7, "elementStaticSize":4},
+        {"name":"m4", "staticSize":24, "dynamicSize":132, "length":9, "capacity":9, "elementStaticSize":4}
+      ]}]'''
+    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}
+      ]}]'''

types/f14_node_set.toml

@@ -2,7 +2,7 @@
 type_name = "folly::F14NodeSet"
 stub_template_params = [1,2,3]
 ctype = "F14_SET"
-header = "folly/container/detail/F14SetFallback.h"
+header = "folly/container/F14Set.h"
 
 # Old:
 typeName = "folly::F14NodeSet<"
@@ -34,3 +34,70 @@ void getSizeType(const %1%<Key, Hasher, KeyEqual, Alloc> &container, size_t& ret
     }
 }
 """
+
+handler = """
+template <typename DB, typename T0, typename T1, typename T2, typename T3>
+struct TypeHandler<DB, %1%<T0, T1, T2, T3>> {
+  using type = types::st::Pair<DB,
+    types::st::VarInt<DB>,
+    types::st::Pair<DB,
+      types::st::VarInt<DB>,
+      types::st::List<DB,
+        typename TypeHandler<DB, T0>::type>>>;
+
+  static types::st::Unit<DB> getSizeType(
+      const %1%<T0, T1, T2, T3>& container,
+      typename TypeHandler<DB, %1%<T0, T1, T2, T3>>::type returnArg) {
+    size_t memorySize = container.getAllocatedMemorySize();
+    auto tail = returnArg
+      .write(memorySize)
+      .write(container.bucket_count())
+      .write(container.size());
+
+    for (auto &&entry: container) {
+      tail = tail.delegate([&entry](auto ret) {
+        return OIInternal::getSizeType<DB>(entry, ret);
+      });
+    }
+
+    return tail.finish();
+  }
+};
+"""
+
+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(container.size());
+
+for (auto &&entry: container) {
+  tail = tail.delegate([&entry](auto ret) {
+    return OIInternal::getSizeType<DB>(entry, ret);
+  });
+}
+
+return tail.finish();
+"""
+
+[[codegen.processor]]
+type = "types::st::VarInt<DB>"
+func = "el.pointer = std::get<ParsedData::VarInt>(d.val).value;"
+
+[[codegen.processor]]
+type = """
+types::st::List<DB, typename TypeHandler<DB, T0>::type>
+"""
+func = """
+auto list = std::get<ParsedData::List>(d.val);
+el.container_stats.emplace(result::Element::ContainerStats {
+  .capacity = list.length,
+  .length = list.length,
+});
+
+static constexpr auto childField = make_field<DB, T0>("[]");
+for (size_t i = 0; i < list.length; i++)
+  ins.emplace(childField);
+"""