/*
- * Copyright 2016 Facebook, Inc.
+ * Copyright 2017 Facebook, Inc.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
#pragma once
-#include <stdexcept>
-#include <cstdlib>
-#include <type_traits>
#include <algorithm>
-#include <iterator>
#include <cassert>
+#include <cstdlib>
+#include <cstring>
+#include <iterator>
+#include <stdexcept>
+#include <type_traits>
+#include <utility>
-#include <boost/operators.hpp>
-#include <boost/type_traits.hpp>
-#include <boost/mpl/if.hpp>
+#include <boost/mpl/count.hpp>
+#include <boost/mpl/empty.hpp>
#include <boost/mpl/eval_if.hpp>
-#include <boost/mpl/vector.hpp>
-#include <boost/mpl/front.hpp>
#include <boost/mpl/filter_view.hpp>
+#include <boost/mpl/front.hpp>
#include <boost/mpl/identity.hpp>
+#include <boost/mpl/if.hpp>
#include <boost/mpl/placeholders.hpp>
-#include <boost/mpl/empty.hpp>
#include <boost/mpl/size.hpp>
-#include <boost/mpl/count.hpp>
+#include <boost/mpl/vector.hpp>
+#include <boost/operators.hpp>
+#include <boost/type_traits.hpp>
+#include <folly/ConstexprMath.h>
#include <folly/FormatTraits.h>
-#include <folly/Malloc.h>
#include <folly/Portability.h>
#include <folly/SmallLocks.h>
-#include <folly/portability/Constexpr.h>
+#include <folly/Traits.h>
+#include <folly/lang/Assume.h>
+#include <folly/memory/Malloc.h>
+#include <folly/portability/BitsFunctexcept.h>
#include <folly/portability/Malloc.h>
#include <folly/portability/TypeTraits.h>
// Ignore shadowing warnings within this file, so includers can use -Wshadow.
-#pragma GCC diagnostic push
-#pragma GCC diagnostic ignored "-Wshadow"
+FOLLY_PUSH_WARNING
+FOLLY_GCC_DISABLE_WARNING("-Wshadow")
namespace folly {
//////////////////////////////////////////////////////////////////////
-} // small_vector_policy
+} // namespace small_vector_policy
//////////////////////////////////////////////////////////////////////
-template<class T, std::size_t M, class A, class B, class C>
+template <class T, std::size_t M, class A, class B, class C>
class small_vector;
//////////////////////////////////////////////////////////////////////
namespace detail {
+/*
+ * Move objects in memory to the right into some uninitialized
+ * memory, where the region overlaps. This doesn't just use
+ * std::move_backward because move_backward only works if all the
+ * memory is initialized to type T already.
+ */
+template <class T>
+typename std::enable_if<!FOLLY_IS_TRIVIALLY_COPYABLE(T)>::type
+moveObjectsRight(T* first, T* lastConstructed, T* realLast) {
+ if (lastConstructed == realLast) {
+ return;
+ }
+
+ T* end = first - 1; // Past the end going backwards.
+ T* out = realLast - 1;
+ T* in = lastConstructed - 1;
+ try {
+ for (; in != end && out >= lastConstructed; --in, --out) {
+ new (out) T(std::move(*in));
+ }
+ for (; in != end; --in, --out) {
+ *out = std::move(*in);
+ }
+ for (; out >= lastConstructed; --out) {
+ new (out) T();
+ }
+ } catch (...) {
+ // We want to make sure the same stuff is uninitialized memory
+ // if we exit via an exception (this is to make sure we provide
+ // the basic exception safety guarantee for insert functions).
+ if (out < lastConstructed) {
+ out = lastConstructed - 1;
+ }
+ for (auto it = out + 1; it != realLast; ++it) {
+ it->~T();
+ }
+ throw;
+ }
+}
+
+// Specialization for trivially copyable types. The call to
+// std::move_backward here will just turn into a memmove. (TODO:
+// change to std::is_trivially_copyable when that works.)
+template <class T>
+typename std::enable_if<FOLLY_IS_TRIVIALLY_COPYABLE(T)>::type
+moveObjectsRight(T* first, T* lastConstructed, T* realLast) {
+ std::move_backward(first, lastConstructed, realLast);
+}
+
+/*
+ * Populate a region of memory using `op' to construct elements. If
+ * anything throws, undo what we did.
+ */
+template <class T, class Function>
+void populateMemForward(T* mem, std::size_t n, Function const& op) {
+ std::size_t idx = 0;
+ try {
+ for (size_t i = 0; i < n; ++i) {
+ op(&mem[idx]);
+ ++idx;
+ }
+ } catch (...) {
+ for (std::size_t i = 0; i < idx; ++i) {
+ mem[i].~T();
+ }
+ throw;
+ }
+}
+
+template <class SizeType, bool ShouldUseHeap>
+struct IntegralSizePolicyBase {
+ typedef SizeType InternalSizeType;
+
+ IntegralSizePolicyBase() : size_(0) {}
+
+ protected:
+ static constexpr std::size_t policyMaxSize() {
+ return SizeType(~kExternMask);
+ }
+
+ std::size_t doSize() const {
+ return size_ & ~kExternMask;
+ }
+
+ std::size_t isExtern() const {
+ return kExternMask & size_;
+ }
+
+ void setExtern(bool b) {
+ if (b) {
+ size_ |= kExternMask;
+ } else {
+ size_ &= ~kExternMask;
+ }
+ }
+
+ void setSize(std::size_t sz) {
+ assert(sz <= policyMaxSize());
+ size_ = (kExternMask & size_) | SizeType(sz);
+ }
+
+ void swapSizePolicy(IntegralSizePolicyBase& o) {
+ std::swap(size_, o.size_);
+ }
+
+ protected:
+ static bool constexpr kShouldUseHeap = ShouldUseHeap;
+
+ private:
+ static SizeType constexpr kExternMask =
+ kShouldUseHeap ? SizeType(1) << (sizeof(SizeType) * 8 - 1) : 0;
+
+ SizeType size_;
+};
+
+template <class SizeType, bool ShouldUseHeap>
+struct IntegralSizePolicy;
+
+template <class SizeType>
+struct IntegralSizePolicy<SizeType, true>
+ : public IntegralSizePolicyBase<SizeType, true> {
+ public:
/*
* Move a range to a range of uninitialized memory. Assumes the
* ranges don't overlap.
*/
- template<class T>
- typename std::enable_if<
- !FOLLY_IS_TRIVIALLY_COPYABLE(T)
- >::type
+ template <class T>
+ typename std::enable_if<!FOLLY_IS_TRIVIALLY_COPYABLE(T)>::type
moveToUninitialized(T* first, T* last, T* out) {
std::size_t idx = 0;
try {
}
// Specialization for trivially copyable types.
- template<class T>
- typename std::enable_if<
- FOLLY_IS_TRIVIALLY_COPYABLE(T)
- >::type
+ template <class T>
+ typename std::enable_if<FOLLY_IS_TRIVIALLY_COPYABLE(T)>::type
moveToUninitialized(T* first, T* last, T* out) {
std::memmove(out, first, (last - first) * sizeof *first);
}
/*
- * Move objects in memory to the right into some uninitialized
- * memory, where the region overlaps. This doesn't just use
- * std::move_backward because move_backward only works if all the
- * memory is initialized to type T already.
+ * Move a range to a range of uninitialized memory. Assumes the
+ * ranges don't overlap. Inserts an element at out + pos using
+ * emplaceFunc(). out will contain (end - begin) + 1 elements on success and
+ * none on failure. If emplaceFunc() throws [begin, end) is unmodified.
*/
- template<class T>
- typename std::enable_if<
- !FOLLY_IS_TRIVIALLY_COPYABLE(T)
- >::type
- moveObjectsRight(T* first, T* lastConstructed, T* realLast) {
- if (lastConstructed == realLast) {
- return;
- }
-
- T* end = first - 1; // Past the end going backwards.
- T* out = realLast - 1;
- T* in = lastConstructed - 1;
+ template <class T, class EmplaceFunc>
+ void moveToUninitializedEmplace(
+ T* begin,
+ T* end,
+ T* out,
+ SizeType pos,
+ EmplaceFunc&& emplaceFunc) {
+ // Must be called first so that if it throws [begin, end) is unmodified.
+ // We have to support the strong exception guarantee for emplace_back().
+ emplaceFunc(out + pos);
+ // move old elements to the left of the new one
try {
- for (; in != end && out >= lastConstructed; --in, --out) {
- new (out) T(std::move(*in));
- }
- for (; in != end; --in, --out) {
- *out = std::move(*in);
- }
- for (; out >= lastConstructed; --out) {
- new (out) T();
- }
+ this->moveToUninitialized(begin, begin + pos, out);
} catch (...) {
- // We want to make sure the same stuff is uninitialized memory
- // if we exit via an exception (this is to make sure we provide
- // the basic exception safety guarantee for insert functions).
- if (out < lastConstructed) {
- out = lastConstructed - 1;
- }
- for (auto it = out + 1; it != realLast; ++it) {
- it->~T();
- }
+ out[pos].~T();
throw;
}
- }
-
- // Specialization for trivially copyable types. The call to
- // std::move_backward here will just turn into a memmove. (TODO:
- // change to std::is_trivially_copyable when that works.)
- template<class T>
- typename std::enable_if<
- FOLLY_IS_TRIVIALLY_COPYABLE(T)
- >::type
- moveObjectsRight(T* first, T* lastConstructed, T* realLast) {
- std::move_backward(first, lastConstructed, realLast);
- }
-
- /*
- * Populate a region of memory using `op' to construct elements. If
- * anything throws, undo what we did.
- */
- template<class T, class Function>
- void populateMemForward(T* mem, std::size_t n, Function const& op) {
- std::size_t idx = 0;
+ // move old elements to the right of the new one
try {
- for (size_t i = 0; i < n; ++i) {
- op(&mem[idx]);
- ++idx;
+ if (begin + pos < end) {
+ this->moveToUninitialized(begin + pos, end, out + pos + 1);
}
} catch (...) {
- for (std::size_t i = 0; i < idx; ++i) {
- mem[i].~T();
+ for (SizeType i = 0; i <= pos; ++i) {
+ out[i].~T();
}
throw;
}
}
+};
- template<class SizeType, bool ShouldUseHeap>
- struct IntegralSizePolicy {
- typedef SizeType InternalSizeType;
-
- IntegralSizePolicy() : size_(0) {}
-
- protected:
- static constexpr std::size_t policyMaxSize() {
- return SizeType(~kExternMask);
- }
-
- std::size_t doSize() const {
- return size_ & ~kExternMask;
- }
-
- std::size_t isExtern() const {
- return kExternMask & size_;
- }
-
- void setExtern(bool b) {
- if (b) {
- size_ |= kExternMask;
- } else {
- size_ &= ~kExternMask;
- }
- }
-
- void setSize(std::size_t sz) {
- assert(sz <= policyMaxSize());
- size_ = (kExternMask & size_) | SizeType(sz);
- }
-
- void swapSizePolicy(IntegralSizePolicy& o) {
- std::swap(size_, o.size_);
- }
-
- protected:
- static bool const kShouldUseHeap = ShouldUseHeap;
-
- private:
- static SizeType const kExternMask =
- kShouldUseHeap ? SizeType(1) << (sizeof(SizeType) * 8 - 1)
- : 0;
+template <class SizeType>
+struct IntegralSizePolicy<SizeType, false>
+ : public IntegralSizePolicyBase<SizeType, false> {
+ public:
+ template <class T>
+ void moveToUninitialized(T* /*first*/, T* /*last*/, T* /*out*/) {
+ assume_unreachable();
+ }
+ template <class T, class EmplaceFunc>
+ void moveToUninitializedEmplace(
+ T* /* begin */,
+ T* /* end */,
+ T* /* out */,
+ SizeType /* pos */,
+ EmplaceFunc&& /* emplaceFunc */) {
+ assume_unreachable();
+ }
+};
- SizeType size_;
- };
+/*
+ * If you're just trying to use this class, ignore everything about
+ * this next small_vector_base class thing.
+ *
+ * The purpose of this junk is to minimize sizeof(small_vector<>)
+ * and allow specifying the template parameters in whatever order is
+ * convenient for the user. There's a few extra steps here to try
+ * to keep the error messages at least semi-reasonable.
+ *
+ * Apologies for all the black magic.
+ */
+namespace mpl = boost::mpl;
+template <
+ class Value,
+ std::size_t RequestedMaxInline,
+ class InPolicyA,
+ class InPolicyB,
+ class InPolicyC>
+struct small_vector_base {
+ typedef mpl::vector<InPolicyA, InPolicyB, InPolicyC> PolicyList;
/*
- * If you're just trying to use this class, ignore everything about
- * this next small_vector_base class thing.
- *
- * The purpose of this junk is to minimize sizeof(small_vector<>)
- * and allow specifying the template parameters in whatever order is
- * convenient for the user. There's a few extra steps here to try
- * to keep the error messages at least semi-reasonable.
- *
- * Apologies for all the black magic.
+ * Determine the size type
*/
- namespace mpl = boost::mpl;
- template<class Value,
- std::size_t RequestedMaxInline,
- class InPolicyA,
- class InPolicyB,
- class InPolicyC>
- struct small_vector_base {
- typedef mpl::vector<InPolicyA,InPolicyB,InPolicyC> PolicyList;
-
- /*
- * Determine the size type
- */
- typedef typename mpl::filter_view<
+ typedef typename mpl::filter_view<
PolicyList,
- boost::is_integral<mpl::placeholders::_1>
- >::type Integrals;
- typedef typename mpl::eval_if<
+ boost::is_integral<mpl::placeholders::_1>>::type Integrals;
+ typedef typename mpl::eval_if<
mpl::empty<Integrals>,
mpl::identity<std::size_t>,
- mpl::front<Integrals>
- >::type SizeType;
+ mpl::front<Integrals>>::type SizeType;
- static_assert(std::is_unsigned<SizeType>::value,
- "Size type should be an unsigned integral type");
- static_assert(mpl::size<Integrals>::value == 0 ||
- mpl::size<Integrals>::value == 1,
- "Multiple size types specified in small_vector<>");
+ static_assert(
+ std::is_unsigned<SizeType>::value,
+ "Size type should be an unsigned integral type");
+ static_assert(
+ mpl::size<Integrals>::value == 0 || mpl::size<Integrals>::value == 1,
+ "Multiple size types specified in small_vector<>");
- /*
- * Determine whether we should allow spilling to the heap or not.
- */
- typedef typename mpl::count<
- PolicyList,small_vector_policy::NoHeap
- >::type HasNoHeap;
+ /*
+ * Determine whether we should allow spilling to the heap or not.
+ */
+ typedef typename mpl::count<PolicyList, small_vector_policy::NoHeap>::type
+ HasNoHeap;
- static_assert(HasNoHeap::value == 0 || HasNoHeap::value == 1,
- "Multiple copies of small_vector_policy::NoHeap "
- "supplied; this is probably a mistake");
+ static_assert(
+ HasNoHeap::value == 0 || HasNoHeap::value == 1,
+ "Multiple copies of small_vector_policy::NoHeap "
+ "supplied; this is probably a mistake");
- /*
- * Make the real policy base classes.
- */
- typedef IntegralSizePolicy<SizeType,!HasNoHeap::value>
- ActualSizePolicy;
+ /*
+ * Make the real policy base classes.
+ */
+ typedef IntegralSizePolicy<SizeType, !HasNoHeap::value> ActualSizePolicy;
- /*
- * Now inherit from them all. This is done in such a convoluted
- * way to make sure we get the empty base optimizaton on all these
- * types to keep sizeof(small_vector<>) minimal.
- */
- typedef boost::totally_ordered1<
- small_vector<Value,RequestedMaxInline,InPolicyA,InPolicyB,InPolicyC>,
- ActualSizePolicy
- > type;
- };
+ /*
+ * Now inherit from them all. This is done in such a convoluted
+ * way to make sure we get the empty base optimizaton on all these
+ * types to keep sizeof(small_vector<>) minimal.
+ */
+ typedef boost::totally_ordered1<
+ small_vector<Value, RequestedMaxInline, InPolicyA, InPolicyB, InPolicyC>,
+ ActualSizePolicy>
+ type;
+};
- template <class T>
- T* pointerFlagSet(T* p) {
- return reinterpret_cast<T*>(reinterpret_cast<uintptr_t>(p) | 1);
- }
- template <class T>
- bool pointerFlagGet(T* p) {
- return reinterpret_cast<uintptr_t>(p) & 1;
- }
- template <class T>
- T* pointerFlagClear(T* p) {
- return reinterpret_cast<T*>(
- reinterpret_cast<uintptr_t>(p) & ~uintptr_t(1));
- }
- inline void* shiftPointer(void* p, size_t sizeBytes) {
- return static_cast<char*>(p) + sizeBytes;
- }
+template <class T>
+T* pointerFlagSet(T* p) {
+ return reinterpret_cast<T*>(reinterpret_cast<uintptr_t>(p) | 1);
}
+template <class T>
+bool pointerFlagGet(T* p) {
+ return reinterpret_cast<uintptr_t>(p) & 1;
+}
+template <class T>
+T* pointerFlagClear(T* p) {
+ return reinterpret_cast<T*>(reinterpret_cast<uintptr_t>(p) & ~uintptr_t(1));
+}
+inline void* shiftPointer(void* p, size_t sizeBytes) {
+ return static_cast<char*>(p) + sizeBytes;
+}
+} // namespace detail
//////////////////////////////////////////////////////////////////////
FOLLY_PACK_PUSH
-template<class Value,
- std::size_t RequestedMaxInline = 1,
- class PolicyA = void,
- class PolicyB = void,
- class PolicyC = void>
-class small_vector
- : public detail::small_vector_base<
- Value,RequestedMaxInline,PolicyA,PolicyB,PolicyC
- >::type
-{
- typedef typename detail::small_vector_base<
- Value,RequestedMaxInline,PolicyA,PolicyB,PolicyC
- >::type BaseType;
+template <
+ class Value,
+ std::size_t RequestedMaxInline = 1,
+ class PolicyA = void,
+ class PolicyB = void,
+ class PolicyC = void>
+class small_vector : public detail::small_vector_base<
+ Value,
+ RequestedMaxInline,
+ PolicyA,
+ PolicyB,
+ PolicyC>::type {
+ typedef typename detail::
+ small_vector_base<Value, RequestedMaxInline, PolicyA, PolicyB, PolicyC>::
+ type BaseType;
typedef typename BaseType::InternalSizeType InternalSizeType;
/*
* the user asks for less inlined elements than we can fit unioned
* into our value_type*, we will inline more than they asked.)
*/
- enum {
- MaxInline =
- constexpr_max(sizeof(Value*) / sizeof(Value), RequestedMaxInline),
- };
-
-public:
- typedef std::size_t size_type;
- typedef Value value_type;
- typedef value_type& reference;
- typedef value_type const& const_reference;
- typedef value_type* iterator;
- typedef value_type const* const_iterator;
- typedef std::ptrdiff_t difference_type;
-
- typedef std::reverse_iterator<iterator> reverse_iterator;
+ static constexpr std::size_t MaxInline{
+ constexpr_max(sizeof(Value*) / sizeof(Value), RequestedMaxInline)};
+
+ public:
+ typedef std::size_t size_type;
+ typedef Value value_type;
+ typedef value_type& reference;
+ typedef value_type const& const_reference;
+ typedef value_type* iterator;
+ typedef value_type* pointer;
+ typedef value_type const* const_iterator;
+ typedef std::ptrdiff_t difference_type;
+
+ typedef std::reverse_iterator<iterator> reverse_iterator;
typedef std::reverse_iterator<const_iterator> const_reverse_iterator;
- explicit small_vector() = default;
+ small_vector() = default;
+ // Allocator is unused here. It is taken in for compatibility with std::vector
+ // interface, but it will be ignored.
+ small_vector(const std::allocator<Value>&) {}
small_vector(small_vector const& o) {
auto n = o.size();
this->setSize(n);
}
- small_vector(small_vector&& o)
- noexcept(std::is_nothrow_move_constructible<Value>::value) {
+ small_vector(small_vector&& o) noexcept(
+ std::is_nothrow_move_constructible<Value>::value) {
if (o.isExtern()) {
swap(o);
} else {
- std::uninitialized_copy(std::make_move_iterator(o.begin()),
- std::make_move_iterator(o.end()),
- begin());
+ std::uninitialized_copy(
+ std::make_move_iterator(o.begin()),
+ std::make_move_iterator(o.end()),
+ begin());
this->setSize(o.size());
}
}
constructImpl(il.begin(), il.end(), std::false_type());
}
- explicit small_vector(size_type n, value_type const& t = value_type()) {
- doConstruct(n, t);
+ explicit small_vector(size_type n) {
+ doConstruct(n, [&](void* p) { new (p) value_type(); });
}
- template<class Arg>
- explicit small_vector(Arg arg1, Arg arg2) {
+ small_vector(size_type n, value_type const& t) {
+ doConstruct(n, [&](void* p) { new (p) value_type(t); });
+ }
+
+ template <class Arg>
+ explicit small_vector(Arg arg1, Arg arg2) {
// Forward using std::is_arithmetic to get to the proper
// implementation; this disambiguates between the iterators and
// (size_t, value_type) meaning for this constructor.
small_vector& operator=(small_vector&& o) {
// TODO: optimization:
// if both are internal, use move assignment where possible
- if (this == &o) return *this;
+ if (this == &o) {
+ return *this;
+ }
clear();
swap(o);
return *this;
}
static constexpr size_type max_size() {
- return !BaseType::kShouldUseHeap ? MaxInline
+ return !BaseType::kShouldUseHeap ? static_cast<size_type>(MaxInline)
: BaseType::policyMaxSize();
}
- size_type size() const { return this->doSize(); }
- bool empty() const { return !size(); }
+ size_type size() const {
+ return this->doSize();
+ }
+ bool empty() const {
+ return !size();
+ }
- iterator begin() { return data(); }
- iterator end() { return data() + size(); }
- const_iterator begin() const { return data(); }
- const_iterator end() const { return data() + size(); }
- const_iterator cbegin() const { return begin(); }
- const_iterator cend() const { return end(); }
+ iterator begin() {
+ return data();
+ }
+ iterator end() {
+ return data() + size();
+ }
+ const_iterator begin() const {
+ return data();
+ }
+ const_iterator end() const {
+ return data() + size();
+ }
+ const_iterator cbegin() const {
+ return begin();
+ }
+ const_iterator cend() const {
+ return end();
+ }
- reverse_iterator rbegin() { return reverse_iterator(end()); }
- reverse_iterator rend() { return reverse_iterator(begin()); }
+ reverse_iterator rbegin() {
+ return reverse_iterator(end());
+ }
+ reverse_iterator rend() {
+ return reverse_iterator(begin());
+ }
const_reverse_iterator rbegin() const {
return const_reverse_iterator(end());
return const_reverse_iterator(begin());
}
- const_reverse_iterator crbegin() const { return rbegin(); }
- const_reverse_iterator crend() const { return rend(); }
+ const_reverse_iterator crbegin() const {
+ return rbegin();
+ }
+ const_reverse_iterator crend() const {
+ return rend();
+ }
/*
* Usually one of the simplest functions in a Container-like class
auto thisCapacity = this->capacity();
auto oCapacity = o.capacity();
- std::swap(unpackHack(&u.pdata_.heap_), unpackHack(&o.u.pdata_.heap_));
+ auto* tmp = u.pdata_.heap_;
+ u.pdata_.heap_ = o.u.pdata_.heap_;
+ o.u.pdata_.heap_ = tmp;
this->setCapacity(oCapacity);
o.setCapacity(thisCapacity);
auto& oldIntern = o.isExtern() ? *this : o;
auto oldExternCapacity = oldExtern.capacity();
- auto oldExternHeap = oldExtern.u.pdata_.heap_;
+ auto oldExternHeap = oldExtern.u.pdata_.heap_;
auto buff = oldExtern.u.buffer();
size_type i = 0;
return;
}
makeSize(sz);
- detail::populateMemForward(begin() + size(), sz - size(),
- [&] (void* p) { new (p) value_type(); }
- );
+ detail::populateMemForward(
+ begin() + size(), sz - size(), [&](void* p) { new (p) value_type(); });
this->setSize(sz);
}
return;
}
makeSize(sz);
- detail::populateMemForward(begin() + size(), sz - size(),
- [&] (void* p) { new (p) value_type(v); }
- );
+ detail::populateMemForward(
+ begin() + size(), sz - size(), [&](void* p) { new (p) value_type(v); });
this->setSize(sz);
}
return this->isExtern() ? u.heap() : u.buffer();
}
- template<class ...Args>
+ template <class... Args>
iterator emplace(const_iterator p, Args&&... args) {
if (p == cend()) {
emplace_back(std::forward<Args>(args)...);
size_type capacity() const {
if (this->isExtern()) {
if (u.hasCapacity()) {
- return *u.getCapacity();
+ return u.getCapacity();
}
return malloc_usable_size(u.pdata_.heap_) / sizeof(value_type);
}
tmp.swap(*this);
}
- template<class ...Args>
+ template <class... Args>
void emplace_back(Args&&... args) {
- // call helper function for static dispatch of special cases
- emplaceBack(std::forward<Args>(args)...);
- }
-
- void emplace_back(const value_type& t) {
- push_back(t);
- }
- void emplace_back(value_type& t) {
- push_back(t);
- }
-
- void emplace_back(value_type&& t) {
- push_back(std::move(t));
- }
-
- void push_back(value_type&& t) {
if (capacity() == size()) {
- makeSize(std::max(size_type(2), 3 * size() / 2), &t, size());
+ // Any of args may be references into the vector.
+ // When we are reallocating, we have to be careful to construct the new
+ // element before modifying the data in the old buffer.
+ makeSize(
+ size() + 1,
+ [&](void* p) { new (p) value_type(std::forward<Args>(args)...); },
+ size());
} else {
- new (end()) value_type(std::move(t));
+ new (end()) value_type(std::forward<Args>(args)...);
}
this->setSize(size() + 1);
}
+ void push_back(value_type&& t) {
+ return emplace_back(std::move(t));
+ }
+
void push_back(value_type const& t) {
- // TODO: we'd like to make use of makeSize (it can be optimized better,
- // because it manipulates the internals)
- // unfortunately the current implementation only supports moving from
- // a supplied rvalue, and doing an extra move just to reuse it is a perf
- // net loss
- if (size() == capacity()) {// && isInside(&t)) {
- value_type tmp(t);
- emplaceBack(std::move(tmp));
- } else {
- emplaceBack(t);
- }
+ emplace_back(t);
}
void pop_back() {
auto offset = p - begin();
if (capacity() == size()) {
- makeSize(size() + 1, &t, offset);
+ makeSize(
+ size() + 1,
+ [&t](void* ptr) { new (ptr) value_type(std::move(t)); },
+ offset);
this->setSize(this->size() + 1);
} else {
- makeSize(size() + 1);
- detail::moveObjectsRight(data() + offset,
- data() + size(),
- data() + size() + 1);
+ detail::moveObjectsRight(
+ data() + offset, data() + size(), data() + size() + 1);
this->setSize(size() + 1);
data()[offset] = std::move(t);
}
return begin() + offset;
-
}
iterator insert(const_iterator p, value_type const& t) {
iterator insert(const_iterator pos, size_type n, value_type const& val) {
auto offset = pos - begin();
makeSize(size() + n);
- detail::moveObjectsRight(data() + offset,
- data() + size(),
- data() + size() + n);
+ detail::moveObjectsRight(
+ data() + offset, data() + size(), data() + size() + n);
this->setSize(size() + n);
std::generate_n(begin() + offset, n, [&] { return val; });
return begin() + offset;
}
- template<class Arg>
+ template <class Arg>
iterator insert(const_iterator p, Arg arg1, Arg arg2) {
// Forward using std::is_arithmetic to get to the proper
// implementation; this disambiguates between the iterators and
}
iterator erase(const_iterator q1, const_iterator q2) {
- if (q1 == q2) return unconst(q1);
+ if (q1 == q2) {
+ return unconst(q1);
+ }
std::move(unconst(q2), end(), unconst(q1));
for (auto it = (end() - std::distance(q1, q2)); it != end(); ++it) {
it->~value_type();
erase(begin(), end());
}
- template<class Arg>
+ template <class Arg>
void assign(Arg first, Arg last) {
clear();
insert(end(), first, last);
insert(end(), n, t);
}
- reference front() { assert(!empty()); return *begin(); }
- reference back() { assert(!empty()); return *(end() - 1); }
- const_reference front() const { assert(!empty()); return *begin(); }
- const_reference back() const { assert(!empty()); return *(end() - 1); }
+ reference front() {
+ assert(!empty());
+ return *begin();
+ }
+ reference back() {
+ assert(!empty());
+ return *(end() - 1);
+ }
+ const_reference front() const {
+ assert(!empty());
+ return *begin();
+ }
+ const_reference back() const {
+ assert(!empty());
+ return *(end() - 1);
+ }
reference operator[](size_type i) {
assert(i < size());
reference at(size_type i) {
if (i >= size()) {
- throw std::out_of_range("index out of range");
+ std::__throw_out_of_range("index out of range");
}
return (*this)[i];
}
const_reference at(size_type i) const {
if (i >= size()) {
- throw std::out_of_range("index out of range");
+ std::__throw_out_of_range("index out of range");
}
return (*this)[i];
}
-private:
-
- /*
- * This is doing the same like emplace_back, but we need this helper
- * to catch the special case - see the next overload function..
- */
- template<class ...Args>
- void emplaceBack(Args&&... args) {
- makeSize(size() + 1);
- new (end()) value_type(std::forward<Args>(args)...);
- this->setSize(size() + 1);
- }
-
+ private:
static iterator unconst(const_iterator it) {
return const_cast<iterator>(it);
}
- /*
- * g++ doesn't allow you to bind a non-const reference to a member
- * of a packed structure, presumably because it would make it too
- * easy to accidentally make an unaligned memory access?
- */
- template<class T> static T& unpackHack(T* p) {
- return *p;
- }
-
// The std::false_type argument is part of disambiguating the
// iterator insert functions from integral types (see insert().)
- template<class It>
+ template <class It>
iterator insertImpl(iterator pos, It first, It last, std::false_type) {
typedef typename std::iterator_traits<It>::iterator_category categ;
- if (std::is_same<categ,std::input_iterator_tag>::value) {
+ if (std::is_same<categ, std::input_iterator_tag>::value) {
auto offset = pos - begin();
while (first != last) {
pos = insert(pos, *first++);
auto distance = std::distance(first, last);
auto offset = pos - begin();
makeSize(size() + distance);
- detail::moveObjectsRight(data() + offset,
- data() + size(),
- data() + size() + distance);
+ detail::moveObjectsRight(
+ data() + offset, data() + size(), data() + size() + distance);
this->setSize(size() + distance);
std::copy_n(first, distance, begin() + offset);
return begin() + offset;
}
- iterator insertImpl(iterator pos, size_type n, const value_type& val,
- std::true_type) {
+ iterator
+ insertImpl(iterator pos, size_type n, const value_type& val, std::true_type) {
// The true_type means this should call the size_t,value_type
// overload. (See insert().)
return insert(pos, n, val);
// The std::false_type argument came from std::is_arithmetic as part
// of disambiguating an overload (see the comment in the
// constructor).
- template<class It>
+ template <class It>
void constructImpl(It first, It last, std::false_type) {
typedef typename std::iterator_traits<It>::iterator_category categ;
- if (std::is_same<categ,std::input_iterator_tag>::value) {
+ if (std::is_same<categ, std::input_iterator_tag>::value) {
// With iterators that only allow a single pass, we can't really
// do anything sane here.
while (first != last) {
makeSize(distance);
this->setSize(distance);
try {
- detail::populateMemForward(data(), distance,
- [&] (void* p) { new (p) value_type(*first++); }
- );
+ detail::populateMemForward(
+ data(), distance, [&](void* p) { new (p) value_type(*first++); });
} catch (...) {
if (this->isExtern()) {
u.freeHeap();
}
}
- void doConstruct(size_type n, value_type const& val) {
+ template <typename InitFunc>
+ void doConstruct(size_type n, InitFunc&& func) {
makeSize(n);
this->setSize(n);
try {
- detail::populateMemForward(data(), n,
- [&] (void* p) { new (p) value_type(val); }
- );
+ detail::populateMemForward(data(), n, std::forward<InitFunc>(func));
} catch (...) {
if (this->isExtern()) {
u.freeHeap();
// The true_type means we should forward to the size_t,value_type
// overload.
void constructImpl(size_type n, value_type const& val, std::true_type) {
- doConstruct(n, val);
+ doConstruct(n, [&](void* p) { new (p) value_type(val); });
}
- void makeSize(size_type size, value_type* v = nullptr) {
- makeSize(size, v, size - 1);
+ /*
+ * Compute the size after growth.
+ */
+ size_type computeNewSize() const {
+ return std::min((3 * capacity()) / 2 + 1, max_size());
+ }
+
+ void makeSize(size_type newSize) {
+ makeSizeInternal(newSize, false, [](void*) { assume_unreachable(); }, 0);
+ }
+
+ template <typename EmplaceFunc>
+ void makeSize(size_type newSize, EmplaceFunc&& emplaceFunc, size_type pos) {
+ assert(size() == capacity());
+ makeSizeInternal(
+ newSize, true, std::forward<EmplaceFunc>(emplaceFunc), pos);
}
/*
- * Ensure we have a large enough memory region to be size `size'.
+ * Ensure we have a large enough memory region to be size `newSize'.
* Will move/copy elements if we are spilling to heap_ or needed to
* allocate a new region, but if resized in place doesn't initialize
* anything in the new region. In any case doesn't change size().
* Supports insertion of new element during reallocation by given
* pointer to new element and position of new element.
- * NOTE: If reallocation is not needed, and new element should be
- * inserted in the middle of vector (not at the end), do the move
- * objects and insertion outside the function, otherwise exception is thrown.
+ * NOTE: If reallocation is not needed, insert must be false,
+ * because we only know how to emplace elements into new memory.
*/
- void makeSize(size_type size, value_type* v, size_type pos) {
- if (size > this->max_size()) {
+ template <typename EmplaceFunc>
+ void makeSizeInternal(
+ size_type newSize,
+ bool insert,
+ EmplaceFunc&& emplaceFunc,
+ size_type pos) {
+ if (newSize > max_size()) {
throw std::length_error("max_size exceeded in small_vector");
}
- if (size <= this->capacity()) {
+ if (newSize <= capacity()) {
+ assert(!insert);
return;
}
- auto needBytes = size * sizeof(value_type);
+ assert(this->kShouldUseHeap);
+ // This branch isn't needed for correctness, but allows the optimizer to
+ // skip generating code for the rest of this function in NoHeap
+ // small_vectors.
+ if (!this->kShouldUseHeap) {
+ return;
+ }
+
+ newSize = std::max(newSize, computeNewSize());
+
+ auto needBytes = newSize * sizeof(value_type);
// If the capacity isn't explicitly stored inline, but the heap
// allocation is grown to over some threshold, we should store
// a capacity at the front of the heap allocation.
bool heapifyCapacity =
- !kHasInlineCapacity && needBytes > kHeapifyCapacityThreshold;
+ !kHasInlineCapacity && needBytes > kHeapifyCapacityThreshold;
if (heapifyCapacity) {
needBytes += kHeapifyCapacitySize;
}
assert(!detail::pointerFlagGet(newh));
value_type* newp = static_cast<value_type*>(
- heapifyCapacity ?
- detail::shiftPointer(newh, kHeapifyCapacitySize) :
- newh);
-
- if (v != nullptr) {
- // move new element
- try {
- new (&newp[pos]) value_type(std::move(*v));
- } catch (...) {
- free(newh);
- throw;
- }
+ heapifyCapacity ? detail::shiftPointer(newh, kHeapifyCapacitySize)
+ : newh);
- // move old elements to the left of the new one
- try {
- detail::moveToUninitialized(begin(), begin() + pos, newp);
- } catch (...) {
- newp[pos].~value_type();
- free(newh);
- throw;
- }
-
- // move old elements to the right of the new one
- try {
- if (pos < size-1) {
- detail::moveToUninitialized(begin() + pos, end(), newp + pos + 1);
- }
- } catch (...) {
- for (size_type i = 0; i <= pos; ++i) {
- newp[i].~value_type();
- }
- free(newh);
- throw;
- }
- } else {
- // move without inserting new element
- try {
- detail::moveToUninitialized(begin(), end(), newp);
- } catch (...) {
- free(newh);
- throw;
+ try {
+ if (insert) {
+ // move and insert the new element
+ this->moveToUninitializedEmplace(
+ begin(), end(), newp, pos, std::forward<EmplaceFunc>(emplaceFunc));
+ } else {
+ // move without inserting new element
+ this->moveToUninitialized(begin(), end(), newp);
}
+ } catch (...) {
+ free(newh);
+ throw;
}
for (auto& val : *this) {
val.~value_type();
assert(this->isExtern());
if (u.hasCapacity()) {
assert(newCapacity < std::numeric_limits<InternalSizeType>::max());
- *u.getCapacity() = InternalSizeType(newCapacity);
+ u.setCapacity(newCapacity);
}
}
-private:
+ private:
struct HeapPtrWithCapacity {
void* heap_;
InternalSizeType capacity_;
- InternalSizeType* getCapacity() {
- return &capacity_;
+ InternalSizeType getCapacity() const {
+ return capacity_;
+ }
+ void setCapacity(InternalSizeType c) {
+ capacity_ = c;
}
} FOLLY_PACK_ATTR;
// stored at the front of the heap allocation.
void* heap_;
- InternalSizeType* getCapacity() {
+ InternalSizeType getCapacity() const {
assert(detail::pointerFlagGet(heap_));
- return static_cast<InternalSizeType*>(
- detail::pointerFlagClear(heap_));
+ return *static_cast<InternalSizeType*>(detail::pointerFlagClear(heap_));
+ }
+ void setCapacity(InternalSizeType c) {
+ *static_cast<InternalSizeType*>(detail::pointerFlagClear(heap_)) = c;
}
} FOLLY_PACK_ATTR;
typedef unsigned char InlineStorageDataType[sizeof(value_type) * MaxInline];
#else
typedef typename std::aligned_storage<
- sizeof(value_type) * MaxInline,
- alignof(value_type)
- >::type InlineStorageDataType;
+ sizeof(value_type) * MaxInline,
+ alignof(value_type)>::type InlineStorageDataType;
#endif
typedef typename std::conditional<
- sizeof(value_type) * MaxInline != 0,
- InlineStorageDataType,
- void*
- >::type InlineStorageType;
+ sizeof(value_type) * MaxInline != 0,
+ InlineStorageDataType,
+ void*>::type InlineStorageType;
- static bool const kHasInlineCapacity =
- sizeof(HeapPtrWithCapacity) < sizeof(InlineStorageType);
+ static bool constexpr kHasInlineCapacity =
+ sizeof(HeapPtrWithCapacity) < sizeof(InlineStorageType);
// This value should we multiple of word size.
- static size_t const kHeapifyCapacitySize = sizeof(
- typename std::aligned_storage<
- sizeof(InternalSizeType),
- alignof(value_type)
- >::type);
+ static size_t constexpr kHeapifyCapacitySize = sizeof(
+ typename std::
+ aligned_storage<sizeof(InternalSizeType), alignof(value_type)>::type);
+
// Threshold to control capacity heapifying.
- static size_t const kHeapifyCapacityThreshold =
- 100 * kHeapifyCapacitySize;
+ static size_t constexpr kHeapifyCapacityThreshold =
+ 100 * kHeapifyCapacitySize;
- typedef typename std::conditional<
- kHasInlineCapacity,
- HeapPtrWithCapacity,
- HeapPtr
- >::type PointerType;
+ typedef typename std::
+ conditional<kHasInlineCapacity, HeapPtrWithCapacity, HeapPtr>::type
+ PointerType;
union Data {
- explicit Data() { pdata_.heap_ = 0; }
+ explicit Data() {
+ pdata_.heap_ = nullptr;
+ }
PointerType pdata_;
InlineStorageType storage_;
value_type* heap() noexcept {
if (kHasInlineCapacity || !detail::pointerFlagGet(pdata_.heap_)) {
return static_cast<value_type*>(pdata_.heap_);
+ } else {
+ return static_cast<value_type*>(detail::shiftPointer(
+ detail::pointerFlagClear(pdata_.heap_), kHeapifyCapacitySize));
}
- return static_cast<value_type*>(
- detail::shiftPointer(
- detail::pointerFlagClear(pdata_.heap_), kHeapifyCapacitySize));
}
value_type const* heap() const noexcept {
return const_cast<Data*>(this)->heap();
bool hasCapacity() const {
return kHasInlineCapacity || detail::pointerFlagGet(pdata_.heap_);
}
- InternalSizeType* getCapacity() {
+ InternalSizeType getCapacity() const {
return pdata_.getCapacity();
}
- InternalSizeType* getCapacity() const {
- return const_cast<Data*>(this)->getCapacity();
+ void setCapacity(InternalSizeType c) {
+ pdata_.setCapacity(c);
}
void freeHeap() {
// Basic guarantee only, or provides the nothrow guarantee iff T has a
// nothrow move or copy constructor.
-template<class T, std::size_t MaxInline, class A, class B, class C>
-void swap(small_vector<T,MaxInline,A,B,C>& a,
- small_vector<T,MaxInline,A,B,C>& b) {
+template <class T, std::size_t MaxInline, class A, class B, class C>
+void swap(
+ small_vector<T, MaxInline, A, B, C>& a,
+ small_vector<T, MaxInline, A, B, C>& b) {
a.swap(b);
}
// Format support.
template <class T, size_t M, class A, class B, class C>
struct IndexableTraits<small_vector<T, M, A, B, C>>
- : public IndexableTraitsSeq<small_vector<T, M, A, B, C>> {
-};
+ : public IndexableTraitsSeq<small_vector<T, M, A, B, C>> {};
-} // namespace detail
+} // namespace detail
-} // namespace folly
+} // namespace folly
-#pragma GCC diagnostic pop
+FOLLY_POP_WARNING