Implement Container V2 for F14VectorSet

This commit is contained in:
Thierry Treyer 2023-09-07 09:44:55 -07:00 committed by Thierry Treyer
parent f8a28b41ef
commit 65fecf314a
2 changed files with 150 additions and 0 deletions

View File

@ -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::F14VectorSet<int> m1;
folly::F14VectorSet<Bar, BarHasher> m2;
folly::F14VectorSet<int> m3;
folly::F14VectorSet<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":304,
"members":[
{"name":"m1", "staticSize":24, "dynamicSize":48, "length":3, "capacity":3, "elementStaticSize":4},
{
"name":"m2",
"staticSize":24,
"dynamicSize":80,
"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":80, "length":7, "capacity":7, "elementStaticSize":4},
{"name":"m4", "staticSize":24, "dynamicSize":96, "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}
]}]'''

View File

@ -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);
"""