mirror of
https://github.com/boostorg/unordered.git
synced 2025-07-30 11:27:15 +02:00
Write pmr memory resource 'counted_new_delete_resource'
This commit is contained in:
55
test/helpers/pmr.hpp
Normal file
55
test/helpers/pmr.hpp
Normal file
@ -0,0 +1,55 @@
|
|||||||
|
|
||||||
|
// Copyright 2024 Braden Ganetsky.
|
||||||
|
// Distributed under the Boost Software License, Version 1.0. (See accompanying
|
||||||
|
// file LICENSE_1_0.txt or move at http://www.boost.org/LICENSE_1_0.txt)
|
||||||
|
|
||||||
|
#ifndef BOOST_UNORDERED_TEST_PMR_HEADER
|
||||||
|
#define BOOST_UNORDERED_TEST_PMR_HEADER
|
||||||
|
|
||||||
|
#include <boost/config.hpp>
|
||||||
|
|
||||||
|
#ifndef BOOST_NO_CXX17_HDR_MEMORY_RESOURCE
|
||||||
|
|
||||||
|
#include <memory_resource>
|
||||||
|
|
||||||
|
namespace test {
|
||||||
|
class counted_new_delete_resource : public std::pmr::memory_resource
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
using std::pmr::memory_resource::memory_resource;
|
||||||
|
~counted_new_delete_resource() override {}
|
||||||
|
|
||||||
|
std::size_t count() const { return _count; }
|
||||||
|
|
||||||
|
private:
|
||||||
|
void* do_allocate(std::size_t bytes, std::size_t alignment) override
|
||||||
|
{
|
||||||
|
_count += bytes;
|
||||||
|
return ::operator new(bytes, std::align_val_t(alignment));
|
||||||
|
}
|
||||||
|
|
||||||
|
void do_deallocate(
|
||||||
|
void* p, std::size_t bytes, std::size_t alignment) override
|
||||||
|
{
|
||||||
|
_count -= bytes;
|
||||||
|
#if __cpp_sized_deallocation
|
||||||
|
::operator delete(p, bytes, std::align_val_t(alignment));
|
||||||
|
#else
|
||||||
|
::operator delete(p, std::align_val_t(alignment));
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
bool do_is_equal(
|
||||||
|
const std::pmr::memory_resource& other) const noexcept override
|
||||||
|
{
|
||||||
|
return this == &other;
|
||||||
|
}
|
||||||
|
|
||||||
|
private:
|
||||||
|
std::size_t _count = 0;
|
||||||
|
};
|
||||||
|
} // namespace test
|
||||||
|
|
||||||
|
#endif // !defined(BOOST_NO_CXX17_HDR_MEMORY_RESOURCE)
|
||||||
|
|
||||||
|
#endif // !defined(BOOST_UNORDERED_TEST_PMR_HEADER)
|
Reference in New Issue
Block a user