diff options
Diffstat (limited to 'src/common/virtual_buffer.h')
| -rw-r--r-- | src/common/virtual_buffer.h | 36 |
1 files changed, 30 insertions, 6 deletions
diff --git a/src/common/virtual_buffer.h b/src/common/virtual_buffer.h index 125cb42f0..91d430036 100644 --- a/src/common/virtual_buffer.h +++ b/src/common/virtual_buffer.h | |||
| @@ -4,29 +4,53 @@ | |||
| 4 | 4 | ||
| 5 | #pragma once | 5 | #pragma once |
| 6 | 6 | ||
| 7 | #include "common/common_funcs.h" | 7 | #include <type_traits> |
| 8 | #include <utility> | ||
| 8 | 9 | ||
| 9 | namespace Common { | 10 | namespace Common { |
| 10 | 11 | ||
| 11 | void* AllocateMemoryPages(std::size_t size); | 12 | void* AllocateMemoryPages(std::size_t size) noexcept; |
| 12 | void FreeMemoryPages(void* base, std::size_t size); | 13 | void FreeMemoryPages(void* base, std::size_t size) noexcept; |
| 13 | 14 | ||
| 14 | template <typename T> | 15 | template <typename T> |
| 15 | class VirtualBuffer final : NonCopyable { | 16 | class VirtualBuffer final { |
| 16 | public: | 17 | public: |
| 18 | static_assert( | ||
| 19 | std::is_trivially_constructible_v<T>, | ||
| 20 | "T must be trivially constructible, as non-trivial constructors will not be executed " | ||
| 21 | "with the current allocator"); | ||
| 22 | |||
| 17 | constexpr VirtualBuffer() = default; | 23 | constexpr VirtualBuffer() = default; |
| 18 | explicit VirtualBuffer(std::size_t count) : alloc_size{count * sizeof(T)} { | 24 | explicit VirtualBuffer(std::size_t count) : alloc_size{count * sizeof(T)} { |
| 19 | base_ptr = reinterpret_cast<T*>(AllocateMemoryPages(alloc_size)); | 25 | base_ptr = reinterpret_cast<T*>(AllocateMemoryPages(alloc_size)); |
| 20 | } | 26 | } |
| 21 | 27 | ||
| 22 | ~VirtualBuffer() { | 28 | ~VirtualBuffer() noexcept { |
| 23 | FreeMemoryPages(base_ptr, alloc_size); | 29 | FreeMemoryPages(base_ptr, alloc_size); |
| 24 | } | 30 | } |
| 25 | 31 | ||
| 32 | VirtualBuffer(const VirtualBuffer&) = delete; | ||
| 33 | VirtualBuffer& operator=(const VirtualBuffer&) = delete; | ||
| 34 | |||
| 35 | VirtualBuffer(VirtualBuffer&& other) noexcept | ||
| 36 | : alloc_size{std::exchange(other.alloc_size, 0)}, base_ptr{std::exchange(other.base_ptr), | ||
| 37 | nullptr} {} | ||
| 38 | |||
| 39 | VirtualBuffer& operator=(VirtualBuffer&& other) noexcept { | ||
| 40 | alloc_size = std::exchange(other.alloc_size, 0); | ||
| 41 | base_ptr = std::exchange(other.base_ptr, nullptr); | ||
| 42 | return *this; | ||
| 43 | } | ||
| 44 | |||
| 26 | void resize(std::size_t count) { | 45 | void resize(std::size_t count) { |
| 46 | const auto new_size = count * sizeof(T); | ||
| 47 | if (new_size == alloc_size) { | ||
| 48 | return; | ||
| 49 | } | ||
| 50 | |||
| 27 | FreeMemoryPages(base_ptr, alloc_size); | 51 | FreeMemoryPages(base_ptr, alloc_size); |
| 28 | 52 | ||
| 29 | alloc_size = count * sizeof(T); | 53 | alloc_size = new_size; |
| 30 | base_ptr = reinterpret_cast<T*>(AllocateMemoryPages(alloc_size)); | 54 | base_ptr = reinterpret_cast<T*>(AllocateMemoryPages(alloc_size)); |
| 31 | } | 55 | } |
| 32 | 56 | ||