Minor cleanup

This commit is contained in:
Victor Zverovich
2025-08-16 07:42:37 -07:00
parent e9ddc97b9a
commit 7ffc3ca15b
3 changed files with 35 additions and 52 deletions

View File

@@ -752,12 +752,11 @@ template <typename T> struct allocator : private std::decay<void> {
void deallocate(T* p, size_t) { std::free(p); }
FMT_CONSTEXPR20 friend bool operator==(allocator, allocator) noexcept {
constexpr friend auto operator==(allocator, allocator) noexcept -> bool {
return true; // All instances of this allocator are equivalent.
}
FMT_CONSTEXPR20 friend bool operator!=(allocator a, allocator b) noexcept {
return !(a == b);
constexpr friend auto operator!=(allocator, allocator) noexcept -> bool {
return false;
}
};
@@ -834,42 +833,32 @@ class basic_memory_buffer : public detail::buffer<T> {
FMT_CONSTEXPR20 ~basic_memory_buffer() { deallocate(); }
private:
template <typename Alloc = Allocator>
FMT_CONSTEXPR20
typename std::enable_if<std::allocator_traits<Alloc>::
propagate_on_container_move_assignment::value,
bool>::type
allocator_move_impl(basic_memory_buffer& other) {
template <typename Alloc = Allocator,
FMT_ENABLE_IF(std::allocator_traits<Alloc>::
propagate_on_container_move_assignment::value)>
FMT_CONSTEXPR20 auto move_alloc(basic_memory_buffer& other) -> bool {
alloc_ = std::move(other.alloc_);
return true;
}
// If the allocator does not propagate,
// then copy the content from source buffer.
template <typename Alloc = Allocator>
FMT_CONSTEXPR20
typename std::enable_if<!std::allocator_traits<Alloc>::
propagate_on_container_move_assignment::value,
bool>::type
allocator_move_impl(basic_memory_buffer& other) {
// If the allocator does not propagate then copy the data from other.
template <typename Alloc = Allocator,
FMT_ENABLE_IF(!std::allocator_traits<Alloc>::
propagate_on_container_move_assignment::value)>
FMT_CONSTEXPR20 auto move_alloc(basic_memory_buffer& other) -> bool {
T* data = other.data();
if (alloc_ != other.alloc_ && data != other.store_) {
size_t size = other.size();
// Perform copy operation, allocators are different
this->resize(size);
detail::copy<T>(data, data + size, this->data());
return false;
}
return true;
if (alloc_ == other.alloc_ || data == other.store_) return true;
size_t size = other.size();
// Perform copy operation, allocators are different.
this->resize(size);
detail::copy<T>(data, data + size, this->data());
return false;
}
// Move data from other to this buffer.
FMT_CONSTEXPR20 void move(basic_memory_buffer& other) {
T* data = other.data();
size_t size = other.size(), capacity = other.capacity();
// Replicate the behaviour of std library containers
if (!allocator_move_impl(other)) {
return;
}
if (!move_alloc(other)) return;
if (data == other.store_) {
this->set(store_, capacity);
detail::copy<T>(other.store_, other.store_ + size, store_);

View File

@@ -307,7 +307,7 @@ TEST(memory_buffer_test, ctor) {
EXPECT_EQ(123u, buffer.capacity());
}
using std_allocator = allocator_ref<std::allocator<char>, true>;
using std_allocator = allocator_ref<std::allocator<char>>;
TEST(memory_buffer_test, move_ctor_inline_buffer) {
auto check_move_buffer =
@@ -351,15 +351,15 @@ TEST(memory_buffer_test, move_ctor_dynamic_buffer) {
EXPECT_GT(buffer2.capacity(), 4u);
}
using std_allocator_n = allocator_ref<std::allocator<char>, false>;
using std_allocator_noprop = allocator_ref<std::allocator<char>, false>;
TEST(memory_buffer_test, move_ctor_inline_buffer_non_propagating) {
auto check_move_buffer =
[](const char* str,
basic_memory_buffer<char, 5, std_allocator_n>& buffer) {
basic_memory_buffer<char, 5, std_allocator_noprop>& buffer) {
std::allocator<char>* original_alloc_ptr = buffer.get_allocator().get();
const char* original_data_ptr = &buffer[0];
basic_memory_buffer<char, 5, std_allocator_n> buffer2(
basic_memory_buffer<char, 5, std_allocator_noprop> buffer2(
std::move(buffer));
const char* new_data_ptr = &buffer2[0];
EXPECT_NE(original_data_ptr, new_data_ptr);
@@ -373,8 +373,8 @@ TEST(memory_buffer_test, move_ctor_inline_buffer_non_propagating) {
EXPECT_NE(original_alloc_ptr, buffer2.get_allocator().get());
};
auto alloc = std::allocator<char>();
basic_memory_buffer<char, 5, std_allocator_n> buffer(
(std_allocator_n(&alloc)));
basic_memory_buffer<char, 5, std_allocator_noprop> buffer(
(std_allocator_noprop(&alloc)));
const char test[] = "test";
buffer.append(string_view(test, 4));
check_move_buffer("test", buffer);
@@ -384,15 +384,15 @@ TEST(memory_buffer_test, move_ctor_inline_buffer_non_propagating) {
TEST(memory_buffer_test, move_ctor_dynamic_buffer_non_propagating) {
auto alloc = std::allocator<char>();
basic_memory_buffer<char, 4, std_allocator_n> buffer(
(std_allocator_n(&alloc)));
basic_memory_buffer<char, 4, std_allocator_noprop> buffer(
(std_allocator_noprop(&alloc)));
const char test[] = "test";
buffer.append(test, test + 4);
const char* inline_buffer_ptr = &buffer[0];
buffer.push_back('a');
EXPECT_NE(&buffer[0], inline_buffer_ptr);
std::allocator<char>* original_alloc_ptr = buffer.get_allocator().get();
basic_memory_buffer<char, 4, std_allocator_n> buffer2;
basic_memory_buffer<char, 4, std_allocator_noprop> buffer2;
buffer2 = std::move(buffer);
EXPECT_EQ(std::string(&buffer2[0], buffer2.size()), "testa");
EXPECT_GT(buffer2.capacity(), 4u);

View File

@@ -37,8 +37,7 @@ template <typename T> class mock_allocator {
MOCK_METHOD(void, deallocate, (T*, size_t));
};
template <typename Allocator, bool PropagateOnMove = false>
class allocator_ref {
template <typename Allocator, bool PropagateOnMove = true> class allocator_ref {
private:
Allocator* alloc_;
@@ -50,8 +49,7 @@ class allocator_ref {
public:
using value_type = typename Allocator::value_type;
using propagate_on_container_move_assignment =
typename std::conditional<PropagateOnMove, std::true_type,
std::false_type>::type;
fmt::bool_constant<PropagateOnMove>;
explicit allocator_ref(Allocator* alloc = nullptr) : alloc_(alloc) {}
@@ -70,23 +68,19 @@ class allocator_ref {
}
public:
Allocator* get() const { return alloc_; }
auto get() const -> Allocator* { return alloc_; }
value_type* allocate(size_t n) {
auto allocate(size_t n) -> value_type* {
return std::allocator_traits<Allocator>::allocate(*alloc_, n);
}
void deallocate(value_type* p, size_t n) { alloc_->deallocate(p, n); }
FMT_CONSTEXPR20 friend bool operator==(const allocator_ref& a,
const allocator_ref& b) noexcept {
friend auto operator==(allocator_ref a, allocator_ref b) noexcept -> bool {
if (a.alloc_ == b.alloc_) return true;
if (a.alloc_ == nullptr || b.alloc_ == nullptr) return false;
return *a.alloc_ == *b.alloc_;
return a.alloc_ && b.alloc_ && *a.alloc_ == *b.alloc_;
}
FMT_CONSTEXPR20 friend bool operator!=(const allocator_ref& a,
const allocator_ref& b) noexcept {
friend auto operator!=(allocator_ref a, allocator_ref b) noexcept -> bool {
return !(a == b);
}
};