segragator.h

Go to the documentation of this file.

#pragma once
 
#include "../utility/empty_base.h"
#include "../utility/meta.h"
#include "segragator_fwd.h"
#include "type_allocator.h"
 
#include <memory>
#include <type_traits>
 
namespace ktl
{
    template<size_t Threshold, typename P, typename F>
    class segragator
    {
    private:
        static_assert(detail::has_no_value_type_v<P>, "Building on top of typed allocators is not allowed. Use allocators without a type");
        static_assert(detail::has_no_value_type_v<F>, "Building on top of typed allocators is not allowed. Use allocators without a type");
 
    public:
        typedef typename detail::get_size_type_t<P> size_type;
 
        segragator() = default;
 
        template<typename Primary,
            typename = std::enable_if_t<std::is_constructible_v<P, Primary>>>
        explicit segragator(Primary&& primary)
            noexcept(std::is_nothrow_constructible_v<P, Primary> && std::is_nothrow_default_constructible_v<F>) :
            m_Primary(std::forward<Primary>(primary)),
            m_Fallback() {}
 
        template<typename Primary, typename Fallback,
            typename = std::enable_if_t<
            std::is_constructible_v<P, Primary> &&
            std::is_constructible_v<F, Fallback>>>
        explicit segragator(Primary&& primary, Fallback&& fallback)
            noexcept(std::is_nothrow_constructible_v<P, Primary> && std::is_nothrow_constructible_v<F, Fallback>) :
            m_Primary(std::forward<Primary>(primary)),
            m_Fallback(std::forward<Fallback>(fallback)) {}
 
        template<typename... Args,
            typename = std::enable_if_t<
            std::is_constructible_v<P, Args...>>>
        explicit segragator(std::tuple<Args...>&& primary)
            noexcept(std::is_nothrow_constructible_v<P, Args...> && std::is_nothrow_default_constructible_v<F>) :
            m_Primary(std::make_from_tuple<P>(std::forward<std::tuple<Args...>>(primary))),
            m_Fallback() {}
 
        template<typename... ArgsP, typename... ArgsF,
            typename = std::enable_if_t<
            std::is_constructible_v<P, ArgsP...> &&
            std::is_constructible_v<F, ArgsF...>>>
        explicit segragator(std::tuple<ArgsP...>&& primary, std::tuple<ArgsF...>&& fallback)
            noexcept(std::is_nothrow_constructible_v<P, ArgsP...> && std::is_nothrow_constructible_v<F, ArgsF...>) :
            m_Primary(std::make_from_tuple<P>(std::forward<std::tuple<ArgsP...>>(primary))),
            m_Fallback(std::make_from_tuple<F>(std::forward<std::tuple<ArgsF...>>(fallback))) {}
 
        segragator(const segragator&) = default;
 
        segragator(segragator&&) = default;
 
        segragator& operator=(const segragator&) = default;
 
        segragator& operator=(segragator&&) = default;
 
        bool operator==(const segragator& rhs) const
            noexcept(detail::has_nothrow_equal_v<P> && detail::has_nothrow_equal_v<F>)
        {
            return m_Primary == rhs.m_Primary && m_Fallback == rhs.m_Fallback;
        }
 
        bool operator!=(const segragator& rhs) const
            noexcept(detail::has_nothrow_not_equal_v<P>&& detail::has_nothrow_not_equal_v<F>)
        {
            return m_Primary != rhs.m_Primary || m_Fallback != rhs.m_Fallback;
        }
 
#pragma region Allocation
        void* allocate(size_t n)
            noexcept(detail::has_nothrow_allocate_v<P> && detail::has_nothrow_allocate_v<F>)
        {
            if (n <= Threshold)
                return m_Primary.allocate(n);
            else
                return m_Fallback.allocate(n);
        }
 
        void deallocate(void* p, size_t n)
            noexcept(detail::has_nothrow_deallocate_v<P> && detail::has_nothrow_deallocate_v<F>)
        {
            if (n <= Threshold)
                return m_Primary.deallocate(p, n);
            else
                return m_Fallback.deallocate(p, n);
        }
#pragma endregion
 
#pragma region Construction
        template<typename T, typename... Args>
        typename std::enable_if<detail::has_construct_v<P, T*, Args...> || detail::has_construct_v<F, T*, Args...>, void>::type
        construct(T* p, Args&&... args) noexcept(
            (!detail::has_construct_v<P, T*, Args...> || detail::has_nothrow_construct_v<P, T*, Args...>) &&
            (!detail::has_construct_v<F, T*, Args...> || detail::has_nothrow_construct_v<F, T*, Args...>) &&
            std::is_nothrow_constructible_v<T, Args...>)
        {
            bool owned = m_Primary.owns(p);
 
            if constexpr (detail::has_construct_v<P, T*, Args...>)
            {
                if (owned)
                {
                    m_Primary.construct(p, std::forward<Args>(args)...);
                    return;
                }
            }
 
            if constexpr (detail::has_construct_v<F, T*, Args...>)
            {
                if (!owned)
                {
                    m_Fallback.construct(p, std::forward<Args>(args)...);
                    return;
                }
            }
 
            ::new(p) T(std::forward<Args>(args)...);
        }
 
        template<typename T>
        typename std::enable_if<detail::has_destroy_v<P, T*> || detail::has_destroy_v<F, T*>, void>::type
        destroy(T* p) noexcept(
            (!detail::has_destroy_v<P, T*> || detail::has_nothrow_destroy_v<P, T*>) &&
            (!detail::has_destroy_v<F, T*> || detail::has_nothrow_destroy_v<F, T*>) &&
            std::is_nothrow_destructible_v<T>)
        {
            bool owned = m_Primary.owns(p);
 
            if constexpr (detail::has_destroy_v<P, T*>)
            {
                if (owned)
                {
                    m_Primary.destroy(p);
                    return;
                }
            }
 
            if constexpr (detail::has_destroy_v<F, T*>)
            {
                if (!owned)
                {
                    m_Fallback.destroy(p);
                    return;
                }
            }
 
            p->~T();
        }
#pragma endregion
 
#pragma region Utility
        template<typename Primary = P, typename Fallback = F>
        typename std::enable_if<detail::has_max_size_v<Primary> && detail::has_max_size_v<Fallback>, size_type>::type
        max_size() const
            noexcept(detail::has_nothrow_max_size_v<Primary> && detail::has_nothrow_max_size_v<Fallback>)
        {
            return (std::max)(m_Primary.max_size(), m_Fallback.max_size());
        }
 
        template<typename Primary = P, typename Fallback = F>
        typename std::enable_if<detail::has_owns_v<Primary> && detail::has_owns_v<Fallback>, bool>::type
        owns(void* p) const
            noexcept(detail::has_nothrow_owns_v<Primary> && detail::has_nothrow_owns_v<Fallback>)
        {
            if (m_Primary.owns(p))
                return true;
 
            return m_Fallback.owns(p);
        }
#pragma endregion
 
    private:
        KTL_EMPTY_BASE P m_Primary;
        KTL_EMPTY_BASE F m_Fallback;
    };
}