mirror of
https://github.com/fmtlib/fmt.git
synced 2025-09-28 23:50:54 +02:00
Minor cleanup
This commit is contained in:
@@ -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_);
|
||||
|
@@ -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);
|
||||
|
@@ -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);
|
||||
}
|
||||
};
|
||||
|
Reference in New Issue
Block a user