davey.hpp

#ifndef ENTT_TOOLS_DAVEY_HPP
#define ENTT_TOOLS_DAVEY_HPP
 
#include <cstdint>
#include <ios>
#include <sstream>
#include <string>
#include <imgui.h>
#include "../entity/mixin.hpp"
#include "../entity/registry.hpp"
#include "../entity/sparse_set.hpp"
#include "../entity/storage.hpp"
#include "../locator/locator.hpp"
#include "../meta/container.hpp"
#include "../meta/context.hpp"
#include "../meta/meta.hpp"
#include "../meta/pointer.hpp"
#include "../meta/resolve.hpp"
 
namespace entt {

namespace internal {
 
template<typename Entity, typename OnEntity>
static void present_element(const meta_any &obj, OnEntity on_entity) {
    for([[maybe_unused]] const auto [id, data]: obj.type().data()) {
        const auto elem = data.get(obj);
        const char *label = data.name() ? data.name() : std::string{data.type().info().name()}.data();
 
        if(auto type = data.type(); type.info() == type_id<const char *>()) {
            ImGui::Text("%s: %s", label, elem.template cast<const char *>());
        } else if(type.info() == type_id<std::string>()) {
            ImGui::Text("%s: %s", label, elem.template cast<const std::string &>().data());
        } else if(type.info() == type_id<Entity>()) {
            if(const auto entt = elem.template cast<Entity>(); entt == null) {
                ImGui::Text("%s: %s", label, "null");
            } else {
                on_entity(label, entt);
            }
        } else if(type.is_enum()) {
            const char *as_string = nullptr;
 
            for(auto [id, curr]: type.data()) {
                if(curr.get({}) == elem) {
                    as_string = curr.name();
                    break;
                }
            }
 
            if(as_string) {
                ImGui::Text("%s: %s", label, as_string);
            } else {
                ImGui::Text("%s: %zu", label, elem.template allow_cast<std::uint64_t>().template cast<std::uint64_t>());
            }
        } else if(type.is_arithmetic()) {
            if(type.info() == type_id<bool>()) {
                std::stringstream buffer{};
                buffer << std::boolalpha << elem.template cast<bool>();
                ImGui::Text("%s: %s", label, buffer.str().data());
            } else if(type.info() == type_id<char>()) {
                ImGui::Text("%s: %c", label, elem.template cast<char>());
            } else if(type.is_integral()) {
                ImGui::Text("%s: %zu", label, elem.template allow_cast<std::uint64_t>().template cast<std::uint64_t>());
            } else {
                ImGui::Text("%s: %f", label, elem.template allow_cast<double>().template cast<double>());
            }
        } else if(type.is_pointer_like()) {
            if(auto deref = *obj; deref) {
                if(ImGui::TreeNode(label)) {
                    present_element<Entity>(*obj, on_entity);
                    ImGui::TreePop();
                }
            } else {
                ImGui::Text("%s: %s", label, "null");
            }
        } else if(type.is_sequence_container()) {
            if(ImGui::TreeNode(label)) {
                meta_sequence_container view = elem.as_sequence_container();
 
                for(std::size_t pos{}, last = view.size(); pos < last; ++pos) {
                    ImGui::PushID(static_cast<int>(pos));
 
                    if(ImGui::TreeNode(label, "%zu", pos)) {
                        present_element<Entity>(view[pos], on_entity);
                        ImGui::TreePop();
                    }
 
                    ImGui::PopID();
                }
 
                ImGui::TreePop();
            }
        } else if(type.is_associative_container()) {
            if(ImGui::TreeNode(label)) {
                meta_associative_container view = elem.as_associative_container();
                auto it = view.begin();
 
                for(std::size_t pos{}, last = view.size(); pos < last; ++pos, ++it) {
                    ImGui::PushID(static_cast<int>(pos));
 
                    if(ImGui::TreeNode(label, "%zu", pos)) {
                        const auto [key, value] = *it;
 
                        if(ImGui::TreeNode("key")) {
                            present_element<Entity>(key, on_entity);
                            ImGui::TreePop();
                        }
 
                        if(ImGui::TreeNode("value")) {
                            present_element<Entity>(value, on_entity);
                            ImGui::TreePop();
                        }
 
                        ImGui::TreePop();
                    }
 
                    ImGui::PopID();
                }
 
                ImGui::TreePop();
            }
        } else if(type.is_class()) {
            if(ImGui::TreeNode(label)) {
                present_element<Entity>(elem, on_entity);
                ImGui::TreePop();
            }
        } else {
            const std::string underlying_type{data.type().info().name()};
            ImGui::Text("%s: %s", label, underlying_type.data());
        }
    }
}
 
template<typename Entity, typename Allocator>
static void present_storage(const meta_ctx &ctx, const basic_sparse_set<Entity, Allocator> &storage) {
    if(auto type = resolve(ctx, storage.info()); type) {
        for(auto entt: storage) {
            ImGui::PushID(static_cast<int>(to_entity(entt)));
 
            if(ImGui::TreeNode(&storage.info(), "%d [%d/%d]", to_integral(entt), to_entity(entt), to_version(entt))) {
                if(const auto obj = type.from_void(storage.value(entt)); obj) {
                    present_element<typename std::decay_t<decltype(storage)>::entity_type>(obj, [](const char *name, const Entity entt) {
                        ImGui::Text("%s: %d [%d/%d]", name, to_integral(entt), to_entity(entt), to_version(entt));
                    });
                }
 
                ImGui::TreePop();
            }
 
            ImGui::PopID();
        }
    } else {
        for(auto entt: storage) {
            ImGui::Text("%d [%d/%d]", to_integral(entt), to_entity(entt), to_version(entt));
        }
    }
}
 
template<typename Entity, typename It>
static void present_entity(const meta_ctx &ctx, const Entity entt, const It from, const It to) {
    for(auto it = from; it != to; ++it) {
        if(const auto &storage = it->second; storage.contains(entt)) {
            if(auto type = resolve(ctx, storage.info()); type) {
                const char *label = type.name() ? type.name() : std::string{storage.info().name()}.data();
 
                if(ImGui::TreeNode(&storage.info(), "%s", label)) {
                    if(const auto obj = type.from_void(storage.value(entt)); obj) {
                        present_element<Entity>(obj, [&ctx, from, to](const char *name, const Entity other) {
                            if(ImGui::TreeNode(name, "%s: %d [%d/%d]", name, to_integral(other), to_entity(other), to_version(other))) {
                                present_entity<Entity>(ctx, other, from, to);
                                ImGui::TreePop();
                            }
                        });
                    }
 
                    ImGui::TreePop();
                }
            } else {
                const std::string name{storage.info().name()};
                ImGui::Text("%s", name.data());
            }
        }
    }
}
 
template<typename... Get, typename... Exclude, std::size_t... Index>
static void present_view(const meta_ctx &ctx, const basic_view<get_t<Get...>, exclude_t<Exclude...>> &view, std::index_sequence<Index...>) {
    using view_type = basic_view<get_t<Get...>, exclude_t<Exclude...>>;
    const std::array<const typename view_type::common_type *, sizeof...(Index)> range{view.template storage<Index>()...};
 
    for(auto tup: view.each()) {
        const auto entt = std::get<0>(tup);
        ImGui::PushID(static_cast<int>(to_entity(entt)));
 
        if(ImGui::TreeNode(&type_id<typename view_type::entity_type>(), "%d [%d/%d]", to_integral(entt), to_entity(entt), to_version(entt))) {
            for(const auto *storage: range) {
                if(auto type = resolve(ctx, storage->info()); type) {
                    const char *label = type.name() ? type.name() : std::string{storage->info().name()}.data();
 
                    if(ImGui::TreeNode(&storage->info(), "%s", label)) {
                        if(const auto obj = type.from_void(storage->value(entt)); obj) {
                            present_element<typename view_type::entity_type>(obj, [](const char *name, const typename view_type::entity_type entt) {
                                ImGui::Text("%s: %d [%d/%d]", name, to_integral(entt), to_entity(entt), to_version(entt));
                            });
                        }
 
                        ImGui::TreePop();
                    }
                } else {
                    const std::string name{storage->info().name()};
                    ImGui::Text("%s", name.data());
                }
            }
 
            ImGui::TreePop();
        }
 
        ImGui::PopID();
    }
}
 
} // namespace internal

template<typename Type, typename Entity, typename Allocator>
void davey(const meta_ctx &ctx, const basic_storage<Type, Entity, Allocator> &storage) {
    internal::present_storage(ctx, storage);
}

template<typename Type, typename Entity, typename Allocator>
void davey(const basic_storage<Type, Entity, Allocator> &storage) {
    davey(locator<meta_ctx>::value_or(), storage);
}

template<typename... Get, typename... Exclude>
void davey(const meta_ctx &ctx, const basic_view<get_t<Get...>, exclude_t<Exclude...>> &view) {
    internal::present_view(ctx, view, std::index_sequence_for<Get...>{});
}

template<typename... Get, typename... Exclude>
void davey(const basic_view<get_t<Get...>, exclude_t<Exclude...>> &view) {
    davey(locator<meta_ctx>::value_or(), view);
}

template<typename Entity, typename Allocator>
void davey(const meta_ctx &ctx, const basic_registry<Entity, Allocator> &registry) {
    ImGui::BeginTabBar("#tabs");
 
    if(ImGui::BeginTabItem("Entity")) {
        for(const auto [entt]: registry.template storage<Entity>()->each()) {
            ImGui::PushID(static_cast<int>(to_entity(entt)));
 
            if(ImGui::TreeNode(&type_id<Entity>(), "%d [%d/%d]", to_integral(entt), to_entity(entt), to_version(entt))) {
                const auto range = registry.storage();
                internal::present_entity(ctx, entt, range.begin(), range.end());
                ImGui::TreePop();
            }
 
            ImGui::PopID();
        }
 
        ImGui::EndTabItem();
    }
 
    if(ImGui::BeginTabItem("Storage")) {
        for([[maybe_unused]] auto [id, storage]: registry.storage()) {
            const auto type = resolve(ctx, storage.info());
            const char *label = type.name() ? type.name() : std::string{storage.info().name()}.data();
 
            if(ImGui::TreeNode(&storage.info(), "%s (%zu)", label, storage.size())) {
                internal::present_storage(ctx, storage);
                ImGui::TreePop();
            }
        }
 
        ImGui::EndTabItem();
    }
 
    ImGui::EndTabBar();
}

template<typename Entity, typename Allocator>
void davey(const basic_registry<Entity, Allocator> &registry) {
    davey(locator<meta_ctx>::value_or(), registry);
}
 
} // namespace entt
 
#endif