MRAllocator.h

Go to the documentation of this file.

#pragma once
 
#include "config.h"
#include "MRPch/MRBindingMacros.h"
#include <memory>
#include <new>

#if !defined( __EMSCRIPTEN__ ) && !defined( MRMESH_NO_MIMALLOC )
#define MR_ALLOCATOR_IS_MIMALLOC
#endif
 
#ifdef MR_ALLOCATOR_IS_MIMALLOC
#include <mimalloc.h>
#if defined( _MSC_VER ) && !defined( MR_USE_CMAKE_CONFIGURE_FILE )
#ifdef _DEBUG
#pragma comment( lib, "mimalloc-debug.dll.lib" )
#else
#pragma comment( lib, "mimalloc.dll.lib" )
#endif
#endif
#endif
 
namespace MR
{
 
 
#ifdef MR_ALLOCATOR_IS_MIMALLOC
template <typename T>
struct MR_BIND_IGNORE AllocatorBase {};
#else
template <typename T>
using AllocatorBase = std::allocator<T>;
#endif

template <typename T>
struct MR_BIND_IGNORE Allocator : AllocatorBase<T>
{
    using value_type = T;
    template <typename U>
    struct rebind { using other = Allocator<U>; };
 
    Allocator() noexcept = default;
    template <typename U>
    Allocator( const Allocator<U> & ) noexcept {}
 
#ifdef MR_ALLOCATOR_IS_MIMALLOC
    using propagate_on_container_move_assignment = std::true_type;
 
    [[nodiscard]] T * allocate( std::size_t n )
    {
        if constexpr ( alignof( T ) > __STDCPP_DEFAULT_NEW_ALIGNMENT__ )
        {
            if ( n > std::size_t( -1 ) / sizeof( T ) )
                throw std::bad_array_new_length();
            return static_cast<T *>( mi_new_aligned( n * sizeof( T ), alignof( T ) ) );
        }
        else
            return static_cast<T *>( mi_new_n( n, sizeof( T ) ) );
    }
    void deallocate( T * p, std::size_t ) noexcept { mi_free( p ); }
#endif
    [[nodiscard]] Allocator select_on_container_copy_construction() const noexcept { return *this; }
};
 
template <typename T, typename U>
[[nodiscard]] inline bool operator ==( const Allocator<T> &, const Allocator<U> & ) noexcept { return true; }
 
}