mirror of
https://github.com/JakeHillion/object-introspection.git
synced 2024-09-19 19:19:05 +01:00
181 lines
5.6 KiB
TOML
181 lines
5.6 KiB
TOML
[info]
|
|
type_name = "std::map"
|
|
stub_template_params = [2,3]
|
|
ctype = "STD_MAP_TYPE"
|
|
header = "map"
|
|
|
|
# Old:
|
|
typeName = "std::map<"
|
|
ns = ["namespace std"]
|
|
numTemplateParams = 2
|
|
replaceTemplateParamIndex = [2]
|
|
allocatorIndex = 3
|
|
|
|
[codegen]
|
|
decl = """
|
|
template<class K, class T, class C, class A>
|
|
void getSizeType(const %1%<K, T, C, A> &container, size_t& returnArg);
|
|
"""
|
|
|
|
func = """
|
|
template<class K, class T, class C, class A>
|
|
void getSizeType(const %1%<K, T, C, A> &container, size_t& returnArg)
|
|
{
|
|
constexpr size_t nodeSize = sizeof(typename %1%<K, T, C, A>::node_type);
|
|
size_t numElems = container.size();
|
|
|
|
SAVE_SIZE(sizeof(%1%<K, T, C, A>) + (nodeSize * numElems));
|
|
|
|
SAVE_DATA((uintptr_t)nodeSize);
|
|
SAVE_DATA((uintptr_t)numElems);
|
|
|
|
for (auto const& it : container)
|
|
{
|
|
getSizeType(it.first, returnArg);
|
|
getSizeType(it.second, returnArg);
|
|
}
|
|
}
|
|
"""
|
|
|
|
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::List<DB, types::st::Pair<DB,
|
|
typename TypeHandler<DB, T0>::type,
|
|
typename TypeHandler<DB, T1>::type
|
|
>>>;
|
|
|
|
static types::st::Unit<DB> getSizeType(
|
|
const %1%<T0, T1, T2, T3>& container,
|
|
typename TypeHandler<DB, %1%<T0, T1, T2, T3>>::type returnArg) {
|
|
constexpr size_t nodeSize = sizeof(typename %1%<T0, T1, T2, T3>::node_type);
|
|
|
|
auto tail = returnArg.write(nodeSize).write(container.size());
|
|
|
|
// The double ampersand is needed otherwise this loop doesn't work with
|
|
// vector<bool>
|
|
for (const auto& it : container) {
|
|
tail = tail.delegate([&it](auto ret) {
|
|
return OIInternal::getSizeType<DB>(it.second, ret.delegate([&it](auto ret) {
|
|
return OIInternal::getSizeType<DB>(it.first, ret);
|
|
}));
|
|
});
|
|
}
|
|
|
|
return tail.finish();
|
|
}
|
|
};
|
|
"""
|
|
|
|
traversal_func = """
|
|
auto tail = returnArg
|
|
.write((uintptr_t)&container)
|
|
.write(container.size());
|
|
|
|
for (const auto &entry: container) {
|
|
tail = tail.delegate([&key = entry.first, &value = entry.second](auto ret) {
|
|
auto start = maybeCaptureKey<captureKeys, DB, T0>(key, ret);
|
|
auto next = start.delegate([&key](typename TypeHandler<DB, T0>::type ret) {
|
|
return OIInternal::getSizeType<DB>(key, ret);
|
|
});
|
|
return OIInternal::getSizeType<DB>(value, next);
|
|
});
|
|
}
|
|
|
|
return tail.finish();
|
|
"""
|
|
|
|
[[codegen.processor]]
|
|
type = "types::st::VarInt<DB>"
|
|
func = "el.pointer = std::get<ParsedData::VarInt>(d.val).value;"
|
|
|
|
[[codegen.processor]]
|
|
type = """
|
|
std::conditional_t<captureKeys,
|
|
types::st::List<DB,
|
|
types::st::Pair<DB,
|
|
typename CaptureKeyHandler<DB, T0>::type,
|
|
types::st::Pair<DB,
|
|
typename TypeHandler<DB, T0>::type,
|
|
typename TypeHandler<DB, T1>::type>>>,
|
|
types::st::List<DB,
|
|
types::st::Pair<DB,
|
|
typename TypeHandler<DB, T0>::type,
|
|
typename TypeHandler<DB, T1>::type>>
|
|
>
|
|
"""
|
|
func = """
|
|
#ifdef __GLIBCXX__
|
|
/* We don't have access to the _Rb_tree_node struct, so we manually re-create it
|
|
* to get the effective size of a map entry. Is there a better way to do this?
|
|
*
|
|
* https://gcc.gnu.org/onlinedocs/libstdc++/latest-doxygen/a00716_source.html#l00216
|
|
* From the source of <bits/stl_tree.h>, an _Rb_tree_node has the following members:
|
|
*/
|
|
struct OI_Rb_tree_node {
|
|
using _Rb_tree_color = int; // enum
|
|
using _Base_ptr = std::nullptr_t; // pointer
|
|
using _value_type = std::pair<const T0, T1>; // from https://gcc.gnu.org/onlinedocs/libstdc++/latest-doxygen/a00692_source.html#l00105
|
|
|
|
_Rb_tree_color _M_color; // from parent _Rb_tree_node_base
|
|
_Base_ptr _M_parent; // from parent _Rb_tree_node_base
|
|
_Base_ptr _M_left; // from parent _Rb_tree_node_base
|
|
_Base_ptr _M_right; // from parent _Rb_tree_node_base
|
|
_value_type _M_value;
|
|
};
|
|
|
|
static constexpr size_t element_size = sizeof(OI_Rb_tree_node);
|
|
#elif _LIBCPP_VERSION
|
|
static_assert(false && "libc++ is currently not supported");
|
|
|
|
/* We don't have access to the __tree_node struct, so we manually re-create it
|
|
* to get the effective size of a map entry. Is there a better way to do this?
|
|
*
|
|
* https://github.com/llvm/llvm-project/blob/1b10920164695a487669405223f8bbe93799430c/libcxx/include/__tree#L729-L781
|
|
* From the source of <__tree>, a __tree_node has the following members:
|
|
*/
|
|
struct OI__tree_node {
|
|
using pointer = std::nullptr_t; // pointer
|
|
using __parent_pointer = std::nullptr_t; // pointer
|
|
using _value_pair = std::pair<const T0, T1>; // from https://github.com/llvm/llvm-project/blob/main/libcxx/include/map#L1024
|
|
|
|
pointer __left_; // from parent __tree_end_node
|
|
pointer __right_; // from parent __tree_node_base
|
|
__parent_pointer __parent_; // from parent __tree_node_base
|
|
bool __is_black_; // from parent __tree_node_base
|
|
_value_type __value_;
|
|
};
|
|
|
|
static constexpr size_t element_size = sizeof(OI__tree_node);
|
|
#else
|
|
static_assert(false && "No known element_size for sets. See types/std_map_type.toml");
|
|
#endif
|
|
|
|
static constexpr std::array<inst::Field, 2> element_fields{
|
|
make_field<DB, T0>("key"),
|
|
make_field<DB, T1>("value"),
|
|
};
|
|
|
|
static constexpr auto processors = maybeCaptureKeysProcessor<captureKeys, DB, T0>();
|
|
|
|
static constexpr inst::Field element{
|
|
element_size,
|
|
element_size - sizeof(T0) - sizeof(T1),
|
|
"[]",
|
|
std::array<std::string_view, 0>{},
|
|
element_fields,
|
|
processors,
|
|
};
|
|
|
|
auto list = std::get<ParsedData::List>(d.val);
|
|
el.container_stats.emplace(result::Element::ContainerStats {
|
|
.capacity = list.length,
|
|
.length = list.length,
|
|
});
|
|
|
|
for (size_t i = 0; i < list.length; i++)
|
|
stack_ins(element);
|
|
"""
|