pocketpy/include/pybind11/stl.h

#include "pybind11.h"

#include <array>
#include <vector>
#include <list>
#include <deque>
#include <forward_list>

#include <map>
#include <unordered_map>

namespace pybind11::impl {

template <typename T, std::size_t N>
struct type_caster<std::array<T, N>> {

    struct wrapper {
        std::array<T, N> container = {};

        operator std::array<T, N>&& () { return std::move(container); }
    };

    wrapper value;

    bool load(const handle& src, bool convert) {
        if(!isinstance<list>(src)) { return false; }
        auto list = src.cast<pybind11::list>();

        if(list.size() != N) { return false; }

        for(std::size_t i = 0; i < N; ++i) {
            type_caster<T> caster;
            if(!caster.load(list[i], convert)) { return false; }
            value.container[i] = caster.value;
        }

        return true;
    }

    template <typename U>
    static handle cast(U&& src, return_value_policy policy, handle parent) {
        auto list = pybind11::list();
        for(auto& item: src) {
            list.append(pybind11::cast(item, policy, parent));
        }
        return list;
    }
};

template <typename T>
constexpr bool is_py_list_like_v = false;

template <typename T, typename Allocator>
constexpr bool is_py_list_like_v<std::vector<T, Allocator>> = true;

template <typename T, typename Allocator>
constexpr bool is_py_list_like_v<std::list<T, Allocator>> = true;

template <typename T, typename Allocator>
constexpr bool is_py_list_like_v<std::deque<T, Allocator>> = true;

template <typename T>
struct type_caster<T, std::enable_if_t<is_py_list_like_v<T>>> {

    struct wrapper {
        T container;

        operator T&& () { return std::move(container); }
    };

    wrapper value;

    bool load(const handle& src, bool convert) {
        if(!isinstance<list>(src)) { return false; }
        auto list = src.cast<pybind11::list>();

        for(auto item: list) {
            type_caster<typename T::value_type> caster;
            if(!caster.load(item, convert)) { return false; }
            value.container.push_back(caster.value);
        }

        return true;
    }

    template <typename U>
    static handle cast(U&& src, return_value_policy policy, handle parent) {
        auto list = pybind11::list();
        if constexpr(std::is_same_v<T, std::vector<bool>>) {
            for(auto item: src) {
                list.append(pybind11::cast(bool(item), policy, parent));
            }
        } else {
            for(auto& item: src) {
                list.append(pybind11::cast(item, policy, parent));
            }
        }
        return list;
    }
};

template <typename T>
constexpr bool is_py_map_like_v = false;

template <typename Key, typename T, typename Compare, typename Allocator>
constexpr bool is_py_map_like_v<std::map<Key, T, Compare, Allocator>> = true;

template <typename Key, typename T, typename Hash, typename KeyEqual, typename Allocator>
constexpr bool is_py_map_like_v<std::unordered_map<Key, T, Hash, KeyEqual, Allocator>> = true;

template <typename T>
struct type_caster<T, std::enable_if_t<is_py_map_like_v<T>>> {

    struct wrapper {
        T container;

        operator T&& () { return std::move(container); }
    };

    wrapper value;

    bool load(const handle& src, bool convert) {
        if(!isinstance<dict>(src)) { return false; }
        auto dict = src.cast<pybind11::dict>();

        for(auto item: dict) {
            type_caster<typename T::key_type> key_caster;
            if(!key_caster.load(item.first, convert)) { return false; }

            type_caster<typename T::mapped_type> value_caster;
            if(!value_caster.load(item.second, convert)) { return false; }

            value.container.try_emplace(key_caster.value, value_caster.value);
        }

        return true;
    }

    template <typename U>
    static handle cast(U&& src, return_value_policy policy, handle parent) {
        auto dict = pybind11::dict();
        for(auto& [key, value]: src) {
            dict[pybind11::cast(key, policy, parent)] = pybind11::cast(value, policy, parent);
        }
        return dict;
    }
};

}  // namespace pybind11::impl