vn-verdnaturachat/ios/Pods/Flipper-Folly/folly/container/F14Map.h

1448 lines
42 KiB
C
Raw Normal View History

/*
* Copyright (c) Facebook, Inc. and its affiliates.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#pragma once
/**
* F14NodeMap, F14ValueMap, and F14VectorMap
*
* F14FastMap conditionally works like F14ValueMap or F14VectorMap
*
* See F14.md
*
* @author Nathan Bronson <ngbronson@fb.com>
* @author Xiao Shi <xshi@fb.com>
*/
#include <cstddef>
#include <initializer_list>
#include <stdexcept>
#include <tuple>
#include <folly/Range.h>
#include <folly/Traits.h>
#include <folly/lang/Exception.h>
#include <folly/lang/SafeAssert.h>
#include <folly/container/F14Map-fwd.h>
#include <folly/container/detail/F14Policy.h>
#include <folly/container/detail/F14Table.h>
#include <folly/container/detail/Util.h>
#if FOLLY_F14_VECTOR_INTRINSICS_AVAILABLE
//////// Common case for supported platforms
namespace folly {
namespace f14 {
namespace detail {
template <typename Policy>
class F14BasicMap {
template <typename K, typename T>
using EnableHeterogeneousFind = std::enable_if_t<
EligibleForHeterogeneousFind<
typename Policy::Key,
typename Policy::Hasher,
typename Policy::KeyEqual,
K>::value,
T>;
template <typename K, typename T>
using EnableHeterogeneousInsert = std::enable_if_t<
EligibleForHeterogeneousInsert<
typename Policy::Key,
typename Policy::Hasher,
typename Policy::KeyEqual,
K>::value,
T>;
template <typename K, typename T>
using EnableHeterogeneousErase = std::enable_if_t<
EligibleForHeterogeneousFind<
typename Policy::Value,
typename Policy::Hasher,
typename Policy::KeyEqual,
K>::value &&
!std::is_same<typename Policy::Iter, remove_cvref_t<K>>::value &&
!std::is_same<typename Policy::ConstIter, remove_cvref_t<K>>::value,
T>;
public:
//// PUBLIC - Member types
using key_type = typename Policy::Key;
using mapped_type = typename Policy::Mapped;
using value_type = typename Policy::Value;
using size_type = std::size_t;
using difference_type = std::ptrdiff_t;
using hasher = typename Policy::Hasher;
using key_equal = typename Policy::KeyEqual;
using allocator_type = typename Policy::Alloc;
using reference = value_type&;
using const_reference = value_type const&;
using pointer = typename Policy::AllocTraits::pointer;
using const_pointer = typename Policy::AllocTraits::const_pointer;
using iterator = typename Policy::Iter;
using const_iterator = typename Policy::ConstIter;
private:
using ItemIter = typename Policy::ItemIter;
public:
//// PUBLIC - Member functions
F14BasicMap() noexcept(Policy::kDefaultConstructIsNoexcept)
: F14BasicMap(0) {}
explicit F14BasicMap(
std::size_t initialCapacity,
hasher const& hash = hasher{},
key_equal const& eq = key_equal{},
allocator_type const& alloc = allocator_type{})
: table_{initialCapacity, hash, eq, alloc} {}
explicit F14BasicMap(std::size_t initialCapacity, allocator_type const& alloc)
: F14BasicMap(initialCapacity, hasher{}, key_equal{}, alloc) {}
explicit F14BasicMap(
std::size_t initialCapacity,
hasher const& hash,
allocator_type const& alloc)
: F14BasicMap(initialCapacity, hash, key_equal{}, alloc) {}
explicit F14BasicMap(allocator_type const& alloc)
: F14BasicMap(0, hasher{}, key_equal{}, alloc) {}
template <typename InputIt>
F14BasicMap(
InputIt first,
InputIt last,
std::size_t initialCapacity = 0,
hasher const& hash = hasher{},
key_equal const& eq = key_equal{},
allocator_type const& alloc = allocator_type{})
: table_{initialCapacity, hash, eq, alloc} {
initialInsert(first, last, initialCapacity);
}
template <typename InputIt>
F14BasicMap(
InputIt first,
InputIt last,
std::size_t initialCapacity,
allocator_type const& alloc)
: table_{initialCapacity, hasher{}, key_equal{}, alloc} {
initialInsert(first, last, initialCapacity);
}
template <typename InputIt>
F14BasicMap(
InputIt first,
InputIt last,
std::size_t initialCapacity,
hasher const& hash,
allocator_type const& alloc)
: table_{initialCapacity, hash, key_equal{}, alloc} {
initialInsert(first, last, initialCapacity);
}
F14BasicMap(F14BasicMap const& rhs) = default;
F14BasicMap(F14BasicMap const& rhs, allocator_type const& alloc)
: table_{rhs.table_, alloc} {}
F14BasicMap(F14BasicMap&& rhs) = default;
F14BasicMap(F14BasicMap&& rhs, allocator_type const& alloc) noexcept(
Policy::kAllocIsAlwaysEqual)
: table_{std::move(rhs.table_), alloc} {}
F14BasicMap(
std::initializer_list<value_type> init,
std::size_t initialCapacity = 0,
hasher const& hash = hasher{},
key_equal const& eq = key_equal{},
allocator_type const& alloc = allocator_type{})
: table_{initialCapacity, hash, eq, alloc} {
initialInsert(init.begin(), init.end(), initialCapacity);
}
F14BasicMap(
std::initializer_list<value_type> init,
std::size_t initialCapacity,
allocator_type const& alloc)
: table_{initialCapacity, hasher{}, key_equal{}, alloc} {
initialInsert(init.begin(), init.end(), initialCapacity);
}
F14BasicMap(
std::initializer_list<value_type> init,
std::size_t initialCapacity,
hasher const& hash,
allocator_type const& alloc)
: table_{initialCapacity, hash, key_equal{}, alloc} {
initialInsert(init.begin(), init.end(), initialCapacity);
}
F14BasicMap& operator=(F14BasicMap const&) = default;
F14BasicMap& operator=(F14BasicMap&&) = default;
F14BasicMap& operator=(std::initializer_list<value_type> ilist) {
clear();
bulkInsert(ilist.begin(), ilist.end(), false);
return *this;
}
allocator_type get_allocator() const noexcept {
return table_.alloc();
}
//// PUBLIC - Iterators
iterator begin() noexcept {
return table_.makeIter(table_.begin());
}
const_iterator begin() const noexcept {
return cbegin();
}
const_iterator cbegin() const noexcept {
return table_.makeConstIter(table_.begin());
}
iterator end() noexcept {
return table_.makeIter(table_.end());
}
const_iterator end() const noexcept {
return cend();
}
const_iterator cend() const noexcept {
return table_.makeConstIter(table_.end());
}
//// PUBLIC - Capacity
bool empty() const noexcept {
return table_.empty();
}
std::size_t size() const noexcept {
return table_.size();
}
std::size_t max_size() const noexcept {
return table_.max_size();
}
//// PUBLIC - Modifiers
void clear() noexcept {
table_.clear();
}
std::pair<iterator, bool> insert(value_type const& value) {
return emplace(value);
}
template <typename P>
std::enable_if_t<
std::is_constructible<value_type, P&&>::value,
std::pair<iterator, bool>>
insert(P&& value) {
return emplace(std::forward<P>(value));
}
// TODO(T31574848): Work around libstdc++ versions (e.g., GCC < 6) with no
// implementation of N4387 ("perfect initialization" for pairs and tuples).
template <typename U1, typename U2>
std::enable_if_t<
std::is_constructible<key_type, U1 const&>::value &&
std::is_constructible<mapped_type, U2 const&>::value,
std::pair<iterator, bool>>
insert(std::pair<U1, U2> const& value) {
return emplace(value);
}
// TODO(T31574848)
template <typename U1, typename U2>
std::enable_if_t<
std::is_constructible<key_type, U1&&>::value &&
std::is_constructible<mapped_type, U2&&>::value,
std::pair<iterator, bool>>
insert(std::pair<U1, U2>&& value) {
return emplace(std::move(value));
}
std::pair<iterator, bool> insert(value_type&& value) {
return emplace(std::move(value));
}
// std::unordered_map's hinted insertion API is misleading. No
// implementation I've seen actually uses the hint. Code restructuring
// by the caller to use the hinted API is at best unnecessary, and at
// worst a pessimization. It is used, however, so we provide it.
iterator insert(const_iterator /*hint*/, value_type const& value) {
return insert(value).first;
}
template <typename P>
std::enable_if_t<std::is_constructible<value_type, P&&>::value, iterator>
insert(const_iterator /*hint*/, P&& value) {
return insert(std::forward<P>(value)).first;
}
iterator insert(const_iterator /*hint*/, value_type&& value) {
return insert(std::move(value)).first;
}
template <class... Args>
iterator emplace_hint(const_iterator /*hint*/, Args&&... args) {
return emplace(std::forward<Args>(args)...).first;
}
private:
template <class InputIt>
FOLLY_ALWAYS_INLINE void
bulkInsert(InputIt first, InputIt last, bool autoReserve) {
if (autoReserve) {
auto n = std::distance(first, last);
if (n == 0) {
return;
}
table_.reserveForInsert(n);
}
while (first != last) {
insert(*first);
++first;
}
}
template <class InputIt>
void initialInsert(InputIt first, InputIt last, std::size_t initialCapacity) {
FOLLY_SAFE_DCHECK(empty() && bucket_count() >= initialCapacity, "");
// It's possible that there are a lot of duplicates in first..last and
// so we will oversize ourself. The common case, however, is that
// we can avoid a lot of rehashing if we pre-expand. The behavior
// is easy to disable at a particular call site by asking for an
// initialCapacity of 1.
bool autoReserve =
std::is_same<
typename std::iterator_traits<InputIt>::iterator_category,
std::random_access_iterator_tag>::value &&
initialCapacity == 0;
bulkInsert(first, last, autoReserve);
}
public:
template <class InputIt>
void insert(InputIt first, InputIt last) {
// Bulk reserve is a heuristic choice, so it can backfire. We restrict
// ourself to situations that mimic bulk construction without an
// explicit initialCapacity.
bool autoReserve =
std::is_same<
typename std::iterator_traits<InputIt>::iterator_category,
std::random_access_iterator_tag>::value &&
bucket_count() == 0;
bulkInsert(first, last, autoReserve);
}
void insert(std::initializer_list<value_type> ilist) {
insert(ilist.begin(), ilist.end());
}
template <typename M>
std::pair<iterator, bool> insert_or_assign(key_type const& key, M&& obj) {
auto rv = try_emplace(key, std::forward<M>(obj));
if (!rv.second) {
rv.first->second = std::forward<M>(obj);
}
return rv;
}
template <typename M>
std::pair<iterator, bool> insert_or_assign(key_type&& key, M&& obj) {
auto rv = try_emplace(std::move(key), std::forward<M>(obj));
if (!rv.second) {
rv.first->second = std::forward<M>(obj);
}
return rv;
}
template <typename M>
iterator
insert_or_assign(const_iterator /*hint*/, key_type const& key, M&& obj) {
return insert_or_assign(key, std::move(obj)).first;
}
template <typename M>
iterator insert_or_assign(const_iterator /*hint*/, key_type&& key, M&& obj) {
return insert_or_assign(std::move(key), std::move(obj)).first;
}
template <typename K, typename M>
EnableHeterogeneousInsert<K, std::pair<iterator, bool>> insert_or_assign(
K&& key,
M&& obj) {
auto rv = try_emplace(std::forward<K>(key), std::forward<M>(obj));
if (!rv.second) {
rv.first->second = std::forward<M>(obj);
}
return rv;
}
private:
template <typename Arg>
using UsableAsKey =
EligibleForHeterogeneousFind<key_type, hasher, key_equal, Arg>;
public:
template <typename... Args>
std::pair<iterator, bool> emplace(Args&&... args) {
auto rv = folly::detail::callWithExtractedKey<key_type, UsableAsKey>(
table_.alloc(),
[&](auto&&... inner) {
return table_.tryEmplaceValue(
std::forward<decltype(inner)>(inner)...);
},
std::forward<Args>(args)...);
return std::make_pair(table_.makeIter(rv.first), rv.second);
}
template <typename... Args>
std::pair<iterator, bool> try_emplace(key_type const& key, Args&&... args) {
auto rv = table_.tryEmplaceValue(
key,
std::piecewise_construct,
std::forward_as_tuple(key),
std::forward_as_tuple(std::forward<Args>(args)...));
return std::make_pair(table_.makeIter(rv.first), rv.second);
}
template <typename... Args>
std::pair<iterator, bool> try_emplace(key_type&& key, Args&&... args) {
auto rv = table_.tryEmplaceValue(
key,
std::piecewise_construct,
std::forward_as_tuple(std::move(key)),
std::forward_as_tuple(std::forward<Args>(args)...));
return std::make_pair(table_.makeIter(rv.first), rv.second);
}
template <typename... Args>
iterator
try_emplace(const_iterator /*hint*/, key_type const& key, Args&&... args) {
auto rv = table_.tryEmplaceValue(
key,
std::piecewise_construct,
std::forward_as_tuple(key),
std::forward_as_tuple(std::forward<Args>(args)...));
return table_.makeIter(rv.first);
}
template <typename... Args>
iterator
try_emplace(const_iterator /*hint*/, key_type&& key, Args&&... args) {
auto rv = table_.tryEmplaceValue(
key,
std::piecewise_construct,
std::forward_as_tuple(std::move(key)),
std::forward_as_tuple(std::forward<Args>(args)...));
return table_.makeIter(rv.first);
}
template <typename K, typename... Args>
EnableHeterogeneousInsert<K, std::pair<iterator, bool>> try_emplace(
K&& key,
Args&&... args) {
auto rv = table_.tryEmplaceValue(
key,
std::piecewise_construct,
std::forward_as_tuple(std::forward<K>(key)),
std::forward_as_tuple(std::forward<Args>(args)...));
return std::make_pair(table_.makeIter(rv.first), rv.second);
}
FOLLY_ALWAYS_INLINE iterator erase(const_iterator pos) {
return eraseInto(pos, [](key_type&&, mapped_type&&) {});
}
// This form avoids ambiguity when key_type has a templated constructor
// that accepts const_iterator
FOLLY_ALWAYS_INLINE iterator erase(iterator pos) {
return eraseInto(pos, [](key_type&&, mapped_type&&) {});
}
iterator erase(const_iterator first, const_iterator last) {
return eraseInto(first, last, [](key_type&&, mapped_type&&) {});
}
size_type erase(key_type const& key) {
return eraseInto(key, [](key_type&&, mapped_type&&) {});
}
template <typename K>
EnableHeterogeneousErase<K, size_type> erase(K const& key) {
return eraseInto(key, [](key_type&&, mapped_type&&) {});
}
protected:
template <typename BeforeDestroy>
FOLLY_ALWAYS_INLINE void tableEraseIterInto(
ItemIter pos,
BeforeDestroy&& beforeDestroy) {
table_.eraseIterInto(pos, [&](value_type&& v) {
auto p = Policy::moveValue(v);
beforeDestroy(std::move(p.first), std::move(p.second));
});
}
template <typename K, typename BeforeDestroy>
FOLLY_ALWAYS_INLINE std::size_t tableEraseKeyInto(
K const& key,
BeforeDestroy&& beforeDestroy) {
return table_.eraseKeyInto(key, [&](value_type&& v) {
auto p = Policy::moveValue(v);
beforeDestroy(std::move(p.first), std::move(p.second));
});
}
public:
// eraseInto contains the same overloads as erase but provides
// an additional callback argument which is called with an rvalue
// reference (not const) to the key and an rvalue reference to the
// mapped value directly before it is destroyed. This can be used
// to extract an entry out of a F14Map while avoiding a copy.
template <typename BeforeDestroy>
FOLLY_ALWAYS_INLINE iterator
eraseInto(const_iterator pos, BeforeDestroy&& beforeDestroy) {
// If we are inlined then gcc and clang can optimize away all of the
// work of itemPos.advance() if our return value is discarded.
auto itemPos = table_.unwrapIter(pos);
tableEraseIterInto(itemPos, beforeDestroy);
itemPos.advanceLikelyDead();
return table_.makeIter(itemPos);
}
// This form avoids ambiguity when key_type has a templated constructor
// that accepts const_iterator
template <typename BeforeDestroy>
FOLLY_ALWAYS_INLINE iterator
eraseInto(iterator pos, BeforeDestroy&& beforeDestroy) {
const_iterator cpos{pos};
return eraseInto(cpos, beforeDestroy);
}
template <typename BeforeDestroy>
iterator eraseInto(
const_iterator first,
const_iterator last,
BeforeDestroy&& beforeDestroy) {
auto itemFirst = table_.unwrapIter(first);
auto itemLast = table_.unwrapIter(last);
while (itemFirst != itemLast) {
tableEraseIterInto(itemFirst, beforeDestroy);
itemFirst.advance();
}
return table_.makeIter(itemFirst);
}
template <typename BeforeDestroy>
size_type eraseInto(key_type const& key, BeforeDestroy&& beforeDestroy) {
return tableEraseKeyInto(key, beforeDestroy);
}
template <typename K, typename BeforeDestroy>
EnableHeterogeneousErase<K, size_type> eraseInto(
K const& key,
BeforeDestroy&& beforeDestroy) {
return tableEraseKeyInto(key, beforeDestroy);
}
//// PUBLIC - Lookup
FOLLY_ALWAYS_INLINE mapped_type& at(key_type const& key) {
return at(*this, key);
}
FOLLY_ALWAYS_INLINE mapped_type const& at(key_type const& key) const {
return at(*this, key);
}
template <typename K>
EnableHeterogeneousFind<K, mapped_type&> at(K const& key) {
return at(*this, key);
}
template <typename K>
EnableHeterogeneousFind<K, mapped_type const&> at(K const& key) const {
return at(*this, key);
}
mapped_type& operator[](key_type const& key) {
return try_emplace(key).first->second;
}
mapped_type& operator[](key_type&& key) {
return try_emplace(std::move(key)).first->second;
}
template <typename K>
EnableHeterogeneousInsert<K, mapped_type&> operator[](K&& key) {
return try_emplace(std::forward<K>(key)).first->second;
}
FOLLY_ALWAYS_INLINE size_type count(key_type const& key) const {
return contains(key) ? 1 : 0;
}
template <typename K>
FOLLY_ALWAYS_INLINE EnableHeterogeneousFind<K, size_type> count(
K const& key) const {
return contains(key) ? 1 : 0;
}
// prehash(key) does the work of evaluating hash_function()(key)
// (including additional bit-mixing for non-avalanching hash functions),
// wraps the result of that work in a token for later reuse, and
// begins prefetching the first steps of looking for key into the
// local CPU cache.
//
// The returned token may be used at any time, may be used more than
// once, and may be used in other F14 sets and maps. Tokens are
// transferrable between any F14 containers (maps and sets) with the
// same key_type and equal hash_function()s.
//
// Hash tokens are not hints -- it is a bug to call any method on this
// class with a token t and key k where t isn't the result of a call
// to prehash(k2) with k2 == k.
F14HashToken prehash(key_type const& key) const {
return table_.prehash(key);
}
template <typename K>
EnableHeterogeneousFind<K, F14HashToken> prehash(K const& key) const {
return table_.prehash(key);
}
FOLLY_ALWAYS_INLINE iterator find(key_type const& key) {
return table_.makeIter(table_.find(key));
}
FOLLY_ALWAYS_INLINE const_iterator find(key_type const& key) const {
return table_.makeConstIter(table_.find(key));
}
FOLLY_ALWAYS_INLINE iterator
find(F14HashToken const& token, key_type const& key) {
return table_.makeIter(table_.find(token, key));
}
FOLLY_ALWAYS_INLINE const_iterator
find(F14HashToken const& token, key_type const& key) const {
return table_.makeConstIter(table_.find(token, key));
}
template <typename K>
FOLLY_ALWAYS_INLINE EnableHeterogeneousFind<K, iterator> find(K const& key) {
return table_.makeIter(table_.find(key));
}
template <typename K>
FOLLY_ALWAYS_INLINE EnableHeterogeneousFind<K, const_iterator> find(
K const& key) const {
return table_.makeConstIter(table_.find(key));
}
template <typename K>
FOLLY_ALWAYS_INLINE EnableHeterogeneousFind<K, iterator> find(
F14HashToken const& token,
K const& key) {
return table_.makeIter(table_.find(token, key));
}
template <typename K>
FOLLY_ALWAYS_INLINE EnableHeterogeneousFind<K, const_iterator> find(
F14HashToken const& token,
K const& key) const {
return table_.makeConstIter(table_.find(token, key));
}
FOLLY_ALWAYS_INLINE bool contains(key_type const& key) const {
return !table_.find(key).atEnd();
}
template <typename K>
FOLLY_ALWAYS_INLINE EnableHeterogeneousFind<K, bool> contains(
K const& key) const {
return !table_.find(key).atEnd();
}
FOLLY_ALWAYS_INLINE bool contains(
F14HashToken const& token,
key_type const& key) const {
return !table_.find(token, key).atEnd();
}
template <typename K>
FOLLY_ALWAYS_INLINE EnableHeterogeneousFind<K, bool> contains(
F14HashToken const& token,
K const& key) const {
return !table_.find(token, key).atEnd();
}
std::pair<iterator, iterator> equal_range(key_type const& key) {
return equal_range(*this, key);
}
std::pair<const_iterator, const_iterator> equal_range(
key_type const& key) const {
return equal_range(*this, key);
}
template <typename K>
EnableHeterogeneousFind<K, std::pair<iterator, iterator>> equal_range(
K const& key) {
return equal_range(*this, key);
}
template <typename K>
EnableHeterogeneousFind<K, std::pair<const_iterator, const_iterator>>
equal_range(K const& key) const {
return equal_range(*this, key);
}
//// PUBLIC - Bucket interface
std::size_t bucket_count() const noexcept {
return table_.bucket_count();
}
std::size_t max_bucket_count() const noexcept {
return table_.max_bucket_count();
}
//// PUBLIC - Hash policy
float load_factor() const noexcept {
return table_.load_factor();
}
float max_load_factor() const noexcept {
return table_.max_load_factor();
}
void max_load_factor(float v) {
table_.max_load_factor(v);
}
void rehash(std::size_t bucketCapacity) {
// The standard's rehash() requires understanding the max load factor,
// which is easy to get wrong. Since we don't actually allow adjustment
// of max_load_factor there is no difference.
reserve(bucketCapacity);
}
void reserve(std::size_t capacity) {
table_.reserve(capacity);
}
//// PUBLIC - Observers
hasher hash_function() const {
return table_.hasher();
}
key_equal key_eq() const {
return table_.keyEqual();
}
//// PUBLIC - F14 Extensions
// containsEqualValue returns true iff there is an element in the map
// that compares equal to value using operator==. It is undefined
// behavior to call this function if operator== on key_type can ever
// return true when the same keys passed to key_eq() would return false
// (the opposite is allowed).
bool containsEqualValue(value_type const& value) const {
auto it = table_.findMatching(
value.first, [&](auto& key) { return value.first == key; });
return !it.atEnd() && value.second == table_.valueAtItem(it.citem()).second;
}
// Get memory footprint, not including sizeof(*this).
std::size_t getAllocatedMemorySize() const {
return table_.getAllocatedMemorySize();
}
// Enumerates classes of allocated memory blocks currently owned
// by this table, calling visitor(allocationSize, allocationCount).
// This can be used to get a more accurate indication of memory footprint
// than getAllocatedMemorySize() if you have some way of computing the
// internal fragmentation of the allocator, such as JEMalloc's nallocx.
// The visitor might be called twice with the same allocationSize. The
// visitor's computation should produce the same result for visitor(8,
// 2) as for two calls to visitor(8, 1), for example. The visitor may
// be called with a zero allocationCount.
template <typename V>
void visitAllocationClasses(V&& visitor) const {
return table_.visitAllocationClasses(visitor);
}
// Calls visitor with two value_type const*, b and e, such that every
// entry in the table is included in exactly one of the ranges [b,e).
// This can be used to efficiently iterate elements in bulk when crossing
// an API boundary that supports contiguous blocks of items.
template <typename V>
void visitContiguousRanges(V&& visitor) const;
F14TableStats computeStats() const noexcept {
return table_.computeStats();
}
private:
template <typename Self, typename K>
FOLLY_ALWAYS_INLINE static auto& at(Self& self, K const& key) {
auto iter = self.find(key);
if (iter == self.end()) {
throw_exception<std::out_of_range>("at() did not find key");
}
return iter->second;
}
template <typename Self, typename K>
static auto equal_range(Self& self, K const& key) {
auto first = self.find(key);
auto last = first;
if (last != self.end()) {
++last;
}
return std::make_pair(first, last);
}
protected:
F14Table<Policy> table_;
};
} // namespace detail
} // namespace f14
template <
typename Key,
typename Mapped,
typename Hasher,
typename KeyEqual,
typename Alloc>
class F14ValueMap
: public f14::detail::F14BasicMap<f14::detail::MapPolicyWithDefaults<
f14::detail::ValueContainerPolicy,
Key,
Mapped,
Hasher,
KeyEqual,
Alloc>> {
protected:
using Policy = f14::detail::MapPolicyWithDefaults<
f14::detail::ValueContainerPolicy,
Key,
Mapped,
Hasher,
KeyEqual,
Alloc>;
private:
using Super = f14::detail::F14BasicMap<Policy>;
public:
using typename Super::value_type;
F14ValueMap() = default;
using Super::Super;
F14ValueMap& operator=(std::initializer_list<value_type> ilist) {
Super::operator=(ilist);
return *this;
}
void swap(F14ValueMap& rhs) noexcept(Policy::kSwapIsNoexcept) {
this->table_.swap(rhs.table_);
}
template <typename V>
void visitContiguousRanges(V&& visitor) const {
this->table_.visitContiguousItemRanges(visitor);
}
};
template <
typename Key,
typename Mapped,
typename Hasher,
typename KeyEqual,
typename Alloc>
class F14NodeMap
: public f14::detail::F14BasicMap<f14::detail::MapPolicyWithDefaults<
f14::detail::NodeContainerPolicy,
Key,
Mapped,
Hasher,
KeyEqual,
Alloc>> {
protected:
using Policy = f14::detail::MapPolicyWithDefaults<
f14::detail::NodeContainerPolicy,
Key,
Mapped,
Hasher,
KeyEqual,
Alloc>;
private:
using Super = f14::detail::F14BasicMap<Policy>;
public:
using typename Super::value_type;
F14NodeMap() = default;
using Super::Super;
F14NodeMap& operator=(std::initializer_list<value_type> ilist) {
Super::operator=(ilist);
return *this;
}
void swap(F14NodeMap& rhs) noexcept(Policy::kSwapIsNoexcept) {
this->table_.swap(rhs.table_);
}
template <typename V>
void visitContiguousRanges(V&& visitor) const {
this->table_.visitItems([&](typename Policy::Item ptr) {
value_type const* b = std::addressof(*ptr);
visitor(b, b + 1);
});
}
// TODO extract and node_handle insert
};
namespace f14 {
namespace detail {
template <
typename Key,
typename Mapped,
typename Hasher,
typename KeyEqual,
typename Alloc,
typename EligibleForPerturbedInsertionOrder>
class F14VectorMapImpl : public F14BasicMap<MapPolicyWithDefaults<
VectorContainerPolicy,
Key,
Mapped,
Hasher,
KeyEqual,
Alloc,
EligibleForPerturbedInsertionOrder>> {
protected:
using Policy = MapPolicyWithDefaults<
VectorContainerPolicy,
Key,
Mapped,
Hasher,
KeyEqual,
Alloc,
EligibleForPerturbedInsertionOrder>;
private:
using Super = F14BasicMap<Policy>;
template <typename K, typename T>
using EnableHeterogeneousVectorErase = std::enable_if_t<
EligibleForHeterogeneousFind<
typename Policy::Value,
typename Policy::Hasher,
typename Policy::KeyEqual,
K>::value &&
!std::is_same<typename Policy::Iter, remove_cvref_t<K>>::value &&
!std::is_same<typename Policy::ConstIter, remove_cvref_t<K>>::value &&
!std::is_same<typename Policy::ReverseIter, remove_cvref_t<K>>::
value &&
!std::is_same<typename Policy::ConstReverseIter, remove_cvref_t<K>>::
value,
T>;
public:
using typename Super::const_iterator;
using typename Super::iterator;
using typename Super::key_type;
using typename Super::mapped_type;
using typename Super::value_type;
F14VectorMapImpl() = default;
// inherit constructors
using Super::Super;
F14VectorMapImpl& operator=(std::initializer_list<value_type> ilist) {
Super::operator=(ilist);
return *this;
}
iterator begin() {
return this->table_.linearBegin(this->size());
}
const_iterator begin() const {
return cbegin();
}
const_iterator cbegin() const {
return this->table_.linearBegin(this->size());
}
iterator end() {
return this->table_.linearEnd();
}
const_iterator end() const {
return cend();
}
const_iterator cend() const {
return this->table_.linearEnd();
}
private:
template <typename BeforeDestroy>
void eraseUnderlying(
typename Policy::ItemIter underlying,
BeforeDestroy&& beforeDestroy) {
Alloc& a = this->table_.alloc();
auto values = this->table_.values_;
// Remove the ptr from the base table and destroy the value.
auto index = underlying.item();
// The item still needs to be hashable during this call, so we must destroy
// the value _afterwards_.
this->tableEraseIterInto(underlying, beforeDestroy);
Policy::AllocTraits::destroy(a, std::addressof(values[index]));
// move the last element in values_ down and fix up the inbound index
auto tailIndex = this->size();
if (tailIndex != index) {
auto tail = this->table_.find(
VectorContainerIndexSearch{static_cast<uint32_t>(tailIndex)});
tail.item() = index;
auto p = std::addressof(values[index]);
assume(p != nullptr);
this->table_.transfer(a, std::addressof(values[tailIndex]), p, 1);
}
}
template <typename K, typename BeforeDestroy>
std::size_t eraseUnderlyingKey(K const& key, BeforeDestroy&& beforeDestroy) {
auto underlying = this->table_.find(key);
if (underlying.atEnd()) {
return 0;
} else {
eraseUnderlying(underlying, beforeDestroy);
return 1;
}
}
public:
FOLLY_ALWAYS_INLINE iterator erase(const_iterator pos) {
return eraseInto(pos, [](key_type&&, mapped_type&&) {});
}
// This form avoids ambiguity when key_type has a templated constructor
// that accepts const_iterator
FOLLY_ALWAYS_INLINE iterator erase(iterator pos) {
return eraseInto(pos, [](key_type&&, mapped_type&&) {});
}
iterator erase(const_iterator first, const_iterator last) {
return eraseInto(first, last, [](key_type&&, mapped_type&&) {});
}
std::size_t erase(key_type const& key) {
return eraseInto(key, [](key_type&&, mapped_type&&) {});
}
template <typename K>
EnableHeterogeneousVectorErase<K, std::size_t> erase(K const& key) {
return eraseInto(key, [](key_type&&, mapped_type&&) {});
}
template <typename BeforeDestroy>
FOLLY_ALWAYS_INLINE iterator
eraseInto(const_iterator pos, BeforeDestroy&& beforeDestroy) {
auto index = this->table_.iterToIndex(pos);
auto underlying = this->table_.find(VectorContainerIndexSearch{index});
eraseUnderlying(underlying, beforeDestroy);
return index == 0 ? end() : this->table_.indexToIter(index - 1);
}
// This form avoids ambiguity when key_type has a templated constructor
// that accepts const_iterator
template <typename BeforeDestroy>
FOLLY_ALWAYS_INLINE iterator
eraseInto(iterator pos, BeforeDestroy&& beforeDestroy) {
const_iterator cpos{pos};
return eraseInto(cpos, beforeDestroy);
}
template <typename BeforeDestroy>
iterator eraseInto(
const_iterator first,
const_iterator last,
BeforeDestroy&& beforeDestroy) {
while (first != last) {
first = eraseInto(first, beforeDestroy);
}
auto index = this->table_.iterToIndex(first);
return index == 0 ? end() : this->table_.indexToIter(index - 1);
}
template <typename BeforeDestroy>
std::size_t eraseInto(key_type const& key, BeforeDestroy&& beforeDestroy) {
return eraseUnderlyingKey(key, beforeDestroy);
}
template <typename K, typename BeforeDestroy>
EnableHeterogeneousVectorErase<K, std::size_t> eraseInto(
K const& key,
BeforeDestroy&& beforeDestroy) {
return eraseUnderlyingKey(key, beforeDestroy);
}
template <typename V>
void visitContiguousRanges(V&& visitor) const {
auto n = this->table_.size();
if (n > 0) {
value_type const* b = std::addressof(this->table_.values_[0]);
visitor(b, b + n);
}
}
};
} // namespace detail
} // namespace f14
template <
typename Key,
typename Mapped,
typename Hasher,
typename KeyEqual,
typename Alloc>
class F14VectorMap : public f14::detail::F14VectorMapImpl<
Key,
Mapped,
Hasher,
KeyEqual,
Alloc,
std::false_type> {
using Super = f14::detail::
F14VectorMapImpl<Key, Mapped, Hasher, KeyEqual, Alloc, std::false_type>;
public:
using typename Super::const_iterator;
using typename Super::iterator;
using typename Super::value_type;
using reverse_iterator = typename Super::Policy::ReverseIter;
using const_reverse_iterator = typename Super::Policy::ConstReverseIter;
F14VectorMap() = default;
// inherit constructors
using Super::Super;
F14VectorMap& operator=(std::initializer_list<value_type> ilist) {
Super::operator=(ilist);
return *this;
}
void swap(F14VectorMap& rhs) noexcept(Super::Policy::kSwapIsNoexcept) {
this->table_.swap(rhs.table_);
}
// ITERATION ORDER
//
// Deterministic iteration order for insert-only workloads is part of
// F14VectorMap's supported API: iterator is LIFO and reverse_iterator
// is FIFO.
//
// If there have been no calls to erase() then iterator and
// const_iterator enumerate entries in the opposite of insertion order.
// begin()->first is the key most recently inserted. reverse_iterator
// and reverse_const_iterator, therefore, enumerate in LIFO (insertion)
// order for insert-only workloads. Deterministic iteration order is
// only guaranteed if no keys were removed since the last time the
// map was empty. Iteration order is preserved across rehashes and
// F14VectorMap copies and moves.
//
// iterator uses LIFO order so that erasing while iterating with begin()
// and end() is safe using the erase(it++) idiom, which is supported
// by std::map and std::unordered_map. erase(iter) invalidates iter
// and all iterators before iter in the non-reverse iteration order.
// Every successful erase invalidates all reverse iterators.
//
// No erase is provided for reverse_iterator or const_reverse_iterator
// to make it harder to shoot yourself in the foot by erasing while
// reverse-iterating. You can write that as map.erase(map.iter(riter))
// if you really need it.
reverse_iterator rbegin() {
return this->table_.values_;
}
const_reverse_iterator rbegin() const {
return crbegin();
}
const_reverse_iterator crbegin() const {
return this->table_.values_;
}
reverse_iterator rend() {
return this->table_.values_ + this->table_.size();
}
const_reverse_iterator rend() const {
return crend();
}
const_reverse_iterator crend() const {
return this->table_.values_ + this->table_.size();
}
// explicit conversions between iterator and reverse_iterator
iterator iter(reverse_iterator riter) {
return this->table_.iter(riter);
}
const_iterator iter(const_reverse_iterator riter) const {
return this->table_.iter(riter);
}
reverse_iterator riter(iterator it) {
return this->table_.riter(it);
}
const_reverse_iterator riter(const_iterator it) const {
return this->table_.riter(it);
}
};
template <typename K, typename M, typename H, typename E, typename A>
Range<typename F14VectorMap<K, M, H, E, A>::const_reverse_iterator>
order_preserving_reinsertion_view(const F14VectorMap<K, M, H, E, A>& c) {
return {c.rbegin(), c.rend()};
}
template <
typename Key,
typename Mapped,
typename Hasher,
typename KeyEqual,
typename Alloc>
class F14FastMap : public std::conditional_t<
sizeof(std::pair<Key const, Mapped>) < 24,
F14ValueMap<Key, Mapped, Hasher, KeyEqual, Alloc>,
f14::detail::F14VectorMapImpl<
Key,
Mapped,
Hasher,
KeyEqual,
Alloc,
std::true_type>> {
using Super = std::conditional_t<
sizeof(std::pair<Key const, Mapped>) < 24,
F14ValueMap<Key, Mapped, Hasher, KeyEqual, Alloc>,
f14::detail::F14VectorMapImpl<
Key,
Mapped,
Hasher,
KeyEqual,
Alloc,
std::true_type>>;
public:
using typename Super::value_type;
F14FastMap() = default;
using Super::Super;
F14FastMap& operator=(std::initializer_list<value_type> ilist) {
Super::operator=(ilist);
return *this;
}
void swap(F14FastMap& rhs) noexcept(Super::Policy::kSwapIsNoexcept) {
this->table_.swap(rhs.table_);
}
};
} // namespace folly
#else // !if FOLLY_F14_VECTOR_INTRINSICS_AVAILABLE
//////// Compatibility for unsupported platforms (not x86_64 and not aarch64)
#include <folly/container/detail/F14MapFallback.h>
#endif // if FOLLY_F14_VECTOR_INTRINSICS_AVAILABLE else
namespace folly {
namespace f14 {
namespace detail {
template <typename M>
bool mapsEqual(M const& lhs, M const& rhs) {
if (lhs.size() != rhs.size()) {
return false;
}
for (auto& kv : lhs) {
if (!rhs.containsEqualValue(kv)) {
return false;
}
}
return true;
}
} // namespace detail
} // namespace f14
template <typename K, typename M, typename H, typename E, typename A>
bool operator==(
F14ValueMap<K, M, H, E, A> const& lhs,
F14ValueMap<K, M, H, E, A> const& rhs) {
return mapsEqual(lhs, rhs);
}
template <typename K, typename M, typename H, typename E, typename A>
bool operator!=(
F14ValueMap<K, M, H, E, A> const& lhs,
F14ValueMap<K, M, H, E, A> const& rhs) {
return !(lhs == rhs);
}
template <typename K, typename M, typename H, typename E, typename A>
bool operator==(
F14NodeMap<K, M, H, E, A> const& lhs,
F14NodeMap<K, M, H, E, A> const& rhs) {
return mapsEqual(lhs, rhs);
}
template <typename K, typename M, typename H, typename E, typename A>
bool operator!=(
F14NodeMap<K, M, H, E, A> const& lhs,
F14NodeMap<K, M, H, E, A> const& rhs) {
return !(lhs == rhs);
}
template <typename K, typename M, typename H, typename E, typename A>
bool operator==(
F14VectorMap<K, M, H, E, A> const& lhs,
F14VectorMap<K, M, H, E, A> const& rhs) {
return mapsEqual(lhs, rhs);
}
template <typename K, typename M, typename H, typename E, typename A>
bool operator!=(
F14VectorMap<K, M, H, E, A> const& lhs,
F14VectorMap<K, M, H, E, A> const& rhs) {
return !(lhs == rhs);
}
template <typename K, typename M, typename H, typename E, typename A>
bool operator==(
F14FastMap<K, M, H, E, A> const& lhs,
F14FastMap<K, M, H, E, A> const& rhs) {
return mapsEqual(lhs, rhs);
}
template <typename K, typename M, typename H, typename E, typename A>
bool operator!=(
F14FastMap<K, M, H, E, A> const& lhs,
F14FastMap<K, M, H, E, A> const& rhs) {
return !(lhs == rhs);
}
template <typename K, typename M, typename H, typename E, typename A>
void swap(
F14ValueMap<K, M, H, E, A>& lhs,
F14ValueMap<K, M, H, E, A>& rhs) noexcept(noexcept(lhs.swap(rhs))) {
lhs.swap(rhs);
}
template <typename K, typename M, typename H, typename E, typename A>
void swap(
F14NodeMap<K, M, H, E, A>& lhs,
F14NodeMap<K, M, H, E, A>& rhs) noexcept(noexcept(lhs.swap(rhs))) {
lhs.swap(rhs);
}
template <typename K, typename M, typename H, typename E, typename A>
void swap(
F14VectorMap<K, M, H, E, A>& lhs,
F14VectorMap<K, M, H, E, A>& rhs) noexcept(noexcept(lhs.swap(rhs))) {
lhs.swap(rhs);
}
template <typename K, typename M, typename H, typename E, typename A>
void swap(
F14FastMap<K, M, H, E, A>& lhs,
F14FastMap<K, M, H, E, A>& rhs) noexcept(noexcept(lhs.swap(rhs))) {
lhs.swap(rhs);
}
template <
typename K,
typename M,
typename H,
typename E,
typename A,
typename Pred>
void erase_if(F14ValueMap<K, M, H, E, A>& c, Pred pred) {
f14::detail::erase_if_impl(c, pred);
}
template <
typename K,
typename M,
typename H,
typename E,
typename A,
typename Pred>
void erase_if(F14NodeMap<K, M, H, E, A>& c, Pred pred) {
f14::detail::erase_if_impl(c, pred);
}
template <
typename K,
typename M,
typename H,
typename E,
typename A,
typename Pred>
void erase_if(F14VectorMap<K, M, H, E, A>& c, Pred pred) {
f14::detail::erase_if_impl(c, pred);
}
template <
typename K,
typename M,
typename H,
typename E,
typename A,
typename Pred>
void erase_if(F14FastMap<K, M, H, E, A>& c, Pred pred) {
f14::detail::erase_if_impl(c, pred);
}
} // namespace folly