box.h

Go to the documentation of this file.

#pragma once
 
#include "di/assert/assert_bool.h"
#include "di/container/allocator/allocator.h"
#include "di/container/allocator/fallible_allocator.h"
#include "di/container/allocator/infallible_allocator.h"
#include "di/meta/language.h"
#include "di/platform/prelude.h"
#include "di/types/prelude.h"
#include "di/util/exchange.h"
#include "di/util/std_new.h"
#include "di/vocab/error/result.h"
#include "di/vocab/expected/unexpected.h"
 
namespace di::vocab {
template<typename T>
struct DefaultDelete {
    DefaultDelete() = default;
 
    template<typename U>
    requires(concepts::ConvertibleTo<U*, T*>)
    constexpr DefaultDelete(DefaultDelete<U> const&) {}
 
    constexpr void operator()(T* pointer) const { delete pointer; }
};
 
template<typename T, typename Deleter = DefaultDelete<T>>
class Box {
public:
    using Type = T;
 
    Box()
    requires(concepts::DefaultConstructible<Deleter> && !concepts::Pointer<Deleter>)
    = default;
 
    constexpr Box(nullptr_t)
    requires(concepts::DefaultConstructible<Deleter> && !concepts::Pointer<Deleter>)
    {}
 
    constexpr explicit Box(T* pointer)
    requires(concepts::DefaultConstructible<Deleter> && !concepts::Pointer<Deleter>)
        : m_pointer(pointer) {}
 
    Box(Box const&) = delete;
    constexpr Box(Box&& other)
    requires(concepts::MoveConstructible<Deleter>)
        : m_pointer(other.release()), m_deleter(util::forward<Deleter>(other.get_deleter())) {}
 
    constexpr Box(T* pointer, Deleter const& deleter)
    requires(concepts::CopyConstructible<Deleter>)
        : m_pointer(pointer), m_deleter(deleter) {}
 
    template<concepts::MoveConstructible D = Deleter>
    requires(!concepts::LValueReference<Deleter>)
    constexpr Box(T* pointer, D&& deleter) : m_pointer(pointer, util::forward<D>(deleter)) {}
 
    template<typename D = Deleter>
    requires(concepts::LValueReference<Deleter>)
    Box(T*, meta::RemoveReference<D>&&) = delete;
 
    template<typename U, typename E>
    requires(concepts::ImplicitlyConvertibleTo<U*, T*> && !concepts::LanguageArray<U> &&
             ((concepts::Reference<Deleter> && concepts::SameAs<E, Deleter>) ||
              (!concepts::Reference<Deleter> && concepts::ImplicitlyConvertibleTo<E, Deleter>) ))
    constexpr Box(Box<U, E>&& other) : m_pointer(other.release()), m_deleter(util::forward<E>(other.get_deleter())) {}
 
    constexpr ~Box() { reset(); }
 
    auto operator=(Box const&) -> Box& = delete;
    constexpr auto operator=(Box&& other) -> Box&
    requires(concepts::MoveAssignable<Deleter>)
    {
        reset(other.release());
        m_deleter = util::forward<Deleter>(other.get_deleter());
        return *this;
    }
 
    template<typename U, typename E>
    requires(concepts::ImplicitlyConvertibleTo<U*, T*> && !concepts::LanguageArray<U> &&
             concepts::AssignableFrom<Deleter&, E &&>)
    constexpr auto operator=(Box<U, E>&& other) -> Box& {
        reset(other.release());
        m_deleter = util::forward<Deleter>(other.get_deleter());
        return *this;
    }
 
    constexpr auto operator=(nullptr_t) -> Box& {
        reset();
        return *this;
    }
 
    constexpr auto release() -> T* { return util::exchange(m_pointer, nullptr); }
    constexpr void reset(T* pointer = nullptr) {
        auto* old_pointer = m_pointer;
        m_pointer = pointer;
        if (old_pointer) {
            function::invoke(m_deleter, old_pointer);
        }
    }
 
    constexpr auto get() const -> T* { return m_pointer; }
 
    constexpr auto get_deleter() -> Deleter& { return m_deleter; }
    constexpr auto get_deleter() const -> Deleter const& { return m_deleter; }
 
    constexpr explicit operator bool() const { return !!m_pointer; }
 
    constexpr auto operator*() const -> T& {
        DI_ASSERT(*this);
        return *get();
    }
    constexpr auto operator->() const -> T* {
        DI_ASSERT(*this);
        return get();
    }
 
    constexpr auto begin() const -> T* { return get(); }
    constexpr auto end() const -> T* {
        if (get()) {
            return get() + 1;
        }
        return nullptr;
    }
 
private:
    template<typename U>
    constexpr friend auto operator==(Box const& a, Box<U> const& b) -> bool {
        return const_cast<void*>(static_cast<void const volatile*>(a.get())) ==
               const_cast<void*>(static_cast<void const volatile*>(b.get()));
    }
 
    template<typename U>
    constexpr friend auto operator<=>(Box const& a, Box<U> const& b) {
        return const_cast<void*>(static_cast<void const volatile*>(a.get())) <=>
               const_cast<void*>(static_cast<void const volatile*>(b.get()));
    }
 
    constexpr friend auto operator==(Box const& a, nullptr_t) -> bool { return a.get() == static_cast<T*>(nullptr); }
    constexpr friend auto operator<=>(Box const& a, nullptr_t) {
        return a == nullptr ? di::strong_ordering::equal : di::strong_ordering::greater;
    }
 
    T* m_pointer { nullptr };
    [[no_unique_address]] Deleter m_deleter {};
};
 
namespace detail {
    template<typename T>
    struct MakeBoxFunction {
        template<typename... Args>
        requires(!concepts::LanguageArray<T> && concepts::ConstructibleFrom<T, Args...>)
        constexpr auto operator()(Args&&... args) const -> meta::AllocatorResult<platform::DefaultAllocator, Box<T>> {
            if constexpr (concepts::FallibleAllocator<platform::DefaultAllocator>) {
                auto* result = ::new (std::nothrow) T(util::forward<Args>(args)...);
                if (!result) {
                    return vocab::Unexpected(BasicError::NotEnoughMemory);
                }
                return Box<T>(result);
            } else {
                auto* result = ::new T(util::forward<Args>(args)...);
                DI_ASSERT(result);
                return Box<T>(result);
            }
        }
    };
}
 
template<typename T>
constexpr inline auto make_box = detail::MakeBoxFunction<T> {};
}
 
namespace di {
using vocab::Box;
using vocab::DefaultDelete;
using vocab::make_box;
}