Unordered: Simpler erase implementation.

[SVN r81207]
This commit is contained in:
Daniel James
2012-11-05 18:32:59 +00:00
parent 38d8d052d1
commit ccc3d1c83d
3 changed files with 104 additions and 227 deletions

View File

@ -545,9 +545,7 @@ namespace boost { namespace unordered { namespace detail {
std::size_t key_hash = this->hash(k); std::size_t key_hash = this->hash(k);
std::size_t bucket_index = std::size_t bucket_index =
policy::to_bucket(this->bucket_count_, key_hash); policy::to_bucket(this->bucket_count_, key_hash);
bucket_pointer this_bucket = this->get_bucket(bucket_index); link_pointer prev = this->get_previous_start(bucket_index);
link_pointer prev = this_bucket->next_;
if (!prev) return 0; if (!prev) return 0;
for (;;) for (;;)
@ -565,13 +563,12 @@ namespace boost { namespace unordered { namespace detail {
prev = static_cast<node_pointer>(prev->next_)->group_prev_; prev = static_cast<node_pointer>(prev->next_)->group_prev_;
} }
node_pointer pos = static_cast<node_pointer>(prev->next_); node_pointer first_node = static_cast<node_pointer>(prev->next_);
link_pointer end1 = link_pointer end = static_cast<node_pointer>(first_node->group_prev_)->next_;
static_cast<node_pointer>(pos->group_prev_)->next_;
node_pointer end = static_cast<node_pointer>(end1); std::size_t count = this->delete_nodes(prev, end);
prev->next_ = end1; this->fix_bucket(bucket_index, prev);
this->fix_buckets(this_bucket, prev, end); return count;
return this->delete_nodes(c_iterator(pos), c_iterator(end));
} }
iterator erase(c_iterator r) iterator erase(c_iterator r)
@ -579,126 +576,72 @@ namespace boost { namespace unordered { namespace detail {
BOOST_ASSERT(r.node_); BOOST_ASSERT(r.node_);
iterator next(r.node_); iterator next(r.node_);
++next; ++next;
erase_nodes(r.node_, next.node_);
bucket_pointer this_bucket = this->get_bucket(
policy::to_bucket(this->bucket_count_, r.node_->hash_));
link_pointer prev = unlink_node(*this_bucket, r.node_);
this->fix_buckets(this_bucket, prev, next.node_);
this->delete_node(r);
return next; return next;
} }
iterator erase_range(c_iterator r1, c_iterator r2) iterator erase_range(c_iterator r1, c_iterator r2)
{ {
if (r1 == r2) return iterator(r2.node_); if (r1 == r2) return iterator(r2.node_);
erase_nodes(r1.node_, r2.node_);
std::size_t bucket_index =
policy::to_bucket(this->bucket_count_, r1.node_->hash_);
link_pointer prev = unlink_nodes(
*this->get_bucket(bucket_index), r1.node_, r2.node_);
this->fix_buckets_range(bucket_index, prev, r1.node_, r2.node_);
this->delete_nodes(r1, r2);
return iterator(r2.node_); return iterator(r2.node_);
} }
static link_pointer unlink_node(bucket& b, node_pointer n) link_pointer erase_nodes(node_pointer begin, node_pointer end)
{ {
node_pointer next = static_cast<node_pointer>(n->next_); std::size_t bucket_index =
link_pointer prev = n->group_prev_; policy::to_bucket(this->bucket_count_, begin->hash_);
if(prev->next_ != n) { // Split the groups containing 'begin' and 'end'.
// The node is at the beginning of a group. // And get the pointer to the node before begin while
// we're at it.
link_pointer prev = split_groups(begin, end);
// Find the previous node pointer: // If we don't have a 'prev' it means that begin is at the
prev = b.next_; // beginning of a block, so search through the blocks in the
while(prev->next_ != n) { // same bucket.
prev = static_cast<node_pointer>(prev->next_)->group_prev_; if (!prev) {
} prev = this->get_previous_start(bucket_index);
// Remove from group
if (next && next->group_prev_ == n)
{
next->group_prev_ = n->group_prev_;
}
}
else if (next && next->group_prev_ == n)
{
// The deleted node is not at the end of the group, so
// change the link from the next node.
next->group_prev_ = n->group_prev_;
}
else {
// The deleted node is at the end of the group, so the
// first node in the group is pointing to it.
// Find that to change its pointer.
node_pointer x = static_cast<node_pointer>(n->group_prev_);
while (x->group_prev_ != n) {
x = static_cast<node_pointer>(x->group_prev_);
}
x->group_prev_ = n->group_prev_;
}
prev->next_ = next;
return prev;
}
static link_pointer unlink_nodes(bucket& b,
node_pointer begin, node_pointer end)
{
link_pointer prev = begin->group_prev_;
if (prev->next_ != begin) {
// The node is at the beginning of a group.
// Find the previous node pointer:
prev = b.next_;
while (prev->next_ != begin) while (prev->next_ != begin)
prev = static_cast<node_pointer>(prev->next_)->group_prev_; prev = static_cast<node_pointer>(prev->next_)->group_prev_;
if (end) split_group(end);
}
else {
node_pointer group1 = split_group(begin);
if (end) {
node_pointer group2 = split_group(end);
if(begin == group2) {
link_pointer end1 = group1->group_prev_;
link_pointer end2 = end->group_prev_;
group1->group_prev_ = end2;
end->group_prev_ = end1;
}
}
} }
prev->next_ = end; // Delete the nodes.
do {
link_pointer group_end =
static_cast<node_pointer>(
static_cast<node_pointer>(prev->next_)->group_prev_)->next_;
this->delete_nodes(prev, group_end);
bucket_index = this->fix_bucket(bucket_index, prev);
} while(prev->next_ != end);
return prev; return prev;
} }
// Break a ciruclar list into two, with split as the beginning static link_pointer split_groups(node_pointer begin, node_pointer end)
// of the second group (if split is at the beginning then don't
// split).
static node_pointer split_group(node_pointer split)
{ {
// Find first node in group. node_pointer prev = static_cast<node_pointer>(begin->group_prev_);
node_pointer first = split; if (prev->next_ != begin) prev = node_pointer();
while (static_cast<node_pointer>(first->group_prev_)->next_ ==
first)
first = static_cast<node_pointer>(first->group_prev_);
if(first == split) return split; if (end) {
node_pointer first = end;
while (first != begin && static_cast<node_pointer>(first->group_prev_)->next_ == first) {
first = static_cast<node_pointer>(first->group_prev_);
}
link_pointer last = first->group_prev_; boost::swap(first->group_prev_, end->group_prev_);
first->group_prev_ = split->group_prev_; if (first == begin) return prev;
split->group_prev_ = last; }
return first; if (prev) {
node_pointer first = prev;
while (static_cast<node_pointer>(first->group_prev_)->next_ == first) {
first = static_cast<node_pointer>(first->group_prev_);
}
boost::swap(first->group_prev_, begin->group_prev_);
}
return prev;
} }
//////////////////////////////////////////////////////////////////////// ////////////////////////////////////////////////////////////////////////

View File

@ -498,26 +498,29 @@ namespace boost { namespace unordered { namespace detail {
delete_buckets(); delete_buckets();
} }
void delete_node(c_iterator n) void delete_node(link_pointer prev)
{ {
node_pointer n = static_cast<node_pointer>(prev->next_);
prev->next_ = n->next_;
boost::unordered::detail::destroy_value_impl(node_alloc(), boost::unordered::detail::destroy_value_impl(node_alloc(),
n.node_->value_ptr()); n->value_ptr());
node_allocator_traits::destroy(node_alloc(), node_allocator_traits::destroy(node_alloc(),
boost::addressof(*n.node_)); boost::addressof(*n));
node_allocator_traits::deallocate(node_alloc(), n.node_, 1); node_allocator_traits::deallocate(node_alloc(), n, 1);
--size_; --size_;
} }
std::size_t delete_nodes(c_iterator begin, c_iterator end) std::size_t delete_nodes(link_pointer prev, link_pointer end)
{ {
BOOST_ASSERT(prev->next_ != end);
std::size_t count = 0; std::size_t count = 0;
while(begin != end) { do {
c_iterator n = begin; delete_node(prev);
++begin;
delete_node(n);
++count; ++count;
} } while (prev->next_ != end);
return count; return count;
} }
@ -525,7 +528,7 @@ namespace boost { namespace unordered { namespace detail {
void delete_buckets() void delete_buckets()
{ {
if(buckets_) { if(buckets_) {
delete_nodes(begin(), iterator()); if (size_) delete_nodes(get_previous_start(), link_pointer());
if (bucket::extra_node) { if (bucket::extra_node) {
node_pointer n = static_cast<node_pointer>( node_pointer n = static_cast<node_pointer>(
@ -545,10 +548,9 @@ namespace boost { namespace unordered { namespace detail {
void clear() void clear()
{ {
if(!size_) return; if (!size_) return;
delete_nodes(begin(), iterator()); delete_nodes(get_previous_start(), link_pointer());
get_previous_start()->next_ = link_pointer();
clear_buckets(); clear_buckets();
BOOST_ASSERT(!size_); BOOST_ASSERT(!size_);
@ -577,86 +579,33 @@ namespace boost { namespace unordered { namespace detail {
} }
//////////////////////////////////////////////////////////////////////// ////////////////////////////////////////////////////////////////////////
// Fix buckets after erase // Fix buckets after delete
//
// This is called after erasing a node or group of nodes to fix up std::size_t fix_bucket(std::size_t bucket_index, link_pointer prev)
// the bucket pointers.
void fix_buckets(bucket_pointer this_bucket,
link_pointer prev, node_pointer next)
{ {
if (!next) link_pointer end = prev->next_;
std::size_t bucket_index2 = bucket_index;
if (end)
{ {
if (this_bucket->next_ == prev) bucket_index2 = policy::to_bucket(bucket_count_,
this_bucket->next_ = node_pointer(); static_cast<node_pointer>(end)->hash_);
}
else
{
bucket_pointer next_bucket = get_bucket(
policy::to_bucket(bucket_count_, next->hash_));
if (next_bucket != this_bucket) // If begin and end are in the same bucket, then
{ // there's nothing to do.
next_bucket->next_ = prev; if (bucket_index == bucket_index2) return bucket_index2;
if (this_bucket->next_ == prev)
this_bucket->next_ = node_pointer();
}
}
}
// This is called after erasing a range of nodes to fix any bucket // Update the bucket containing end.
// pointers into that range. get_bucket(bucket_index2)->next_ = prev;
void fix_buckets_range(std::size_t bucket_index,
link_pointer prev, node_pointer begin, node_pointer end)
{
node_pointer n = begin;
// If we're not at the start of the current bucket, then
// go to the start of the next bucket.
if (get_bucket(bucket_index)->next_ != prev)
{
for(;;) {
n = static_cast<node_pointer>(n->next_);
if (n == end) {
if (n) {
std::size_t new_bucket_index =
policy::to_bucket(bucket_count_, n->hash_);
if (bucket_index != new_bucket_index) {
get_bucket(new_bucket_index)->next_ = prev;
}
}
return;
}
std::size_t new_bucket_index =
policy::to_bucket(bucket_count_, n->hash_);
if (bucket_index != new_bucket_index) {
bucket_index = new_bucket_index;
break;
}
}
} }
// Iterate through the remaining nodes, clearing out the bucket // Check if this bucket is now empty.
// pointers. bucket_pointer this_bucket = get_bucket(bucket_index);
get_bucket(bucket_index)->next_ = link_pointer(); if (this_bucket->next_ == prev)
for(;;) { this_bucket->next_ = link_pointer();
n = static_cast<node_pointer>(n->next_);
if (n == end) break;
std::size_t new_bucket_index = return bucket_index2;
policy::to_bucket(bucket_count_, n->hash_);
if (bucket_index != new_bucket_index) {
bucket_index = new_bucket_index;
get_bucket(bucket_index)->next_ = link_pointer();
}
};
// Finally fix the bucket containing the trailing node.
if (n) {
get_bucket(
policy::to_bucket(bucket_count_, n->hash_))->next_
= prev;
}
} }
//////////////////////////////////////////////////////////////////////// ////////////////////////////////////////////////////////////////////////

View File

@ -519,9 +519,7 @@ namespace boost { namespace unordered { namespace detail {
std::size_t key_hash = this->hash(k); std::size_t key_hash = this->hash(k);
std::size_t bucket_index = std::size_t bucket_index =
policy::to_bucket(this->bucket_count_, key_hash); policy::to_bucket(this->bucket_count_, key_hash);
bucket_pointer this_bucket = this->get_bucket(bucket_index); link_pointer prev = this->get_previous_start(bucket_index);
link_pointer prev = this_bucket->next_;
if (!prev) return 0; if (!prev) return 0;
for (;;) for (;;)
@ -539,11 +537,11 @@ namespace boost { namespace unordered { namespace detail {
prev = prev->next_; prev = prev->next_;
} }
node_pointer pos = static_cast<node_pointer>(prev->next_); link_pointer end = static_cast<node_pointer>(prev->next_)->next_;
node_pointer end = static_cast<node_pointer>(pos->next_);
prev->next_ = pos->next_; std::size_t count = this->delete_nodes(prev, end);
this->fix_buckets(this_bucket, prev, end); this->fix_bucket(bucket_index, prev);
return this->delete_nodes(c_iterator(pos), c_iterator(end)); return count;
} }
iterator erase(c_iterator r) iterator erase(c_iterator r)
@ -551,44 +549,31 @@ namespace boost { namespace unordered { namespace detail {
BOOST_ASSERT(r.node_); BOOST_ASSERT(r.node_);
iterator next(r.node_); iterator next(r.node_);
++next; ++next;
erase_nodes(r.node_, next.node_);
bucket_pointer this_bucket = this->get_bucket(
policy::to_bucket(this->bucket_count_, r.node_->hash_));
link_pointer prev = unlink_node(*this_bucket, r.node_);
this->fix_buckets(this_bucket, prev, next.node_);
this->delete_node(r);
return next; return next;
} }
iterator erase_range(c_iterator r1, c_iterator r2) iterator erase_range(c_iterator r1, c_iterator r2)
{ {
if (r1 == r2) return iterator(r2.node_); if (r1 == r2) return iterator(r2.node_);
erase_nodes(r1.node_, r2.node_);
std::size_t bucket_index =
policy::to_bucket(this->bucket_count_, r1.node_->hash_);
link_pointer prev = unlink_nodes(
*this->get_bucket(bucket_index), r1.node_, r2.node_);
this->fix_buckets_range(bucket_index, prev, r1.node_, r2.node_);
this->delete_nodes(r1, r2);
return iterator(r2.node_); return iterator(r2.node_);
} }
static link_pointer unlink_node(bucket& b, node_pointer n) void erase_nodes(node_pointer begin, node_pointer end)
{ {
return unlink_nodes(b, n, static_cast<node_pointer>(n->next_)); std::size_t bucket_index =
} policy::to_bucket(this->bucket_count_, begin->hash_);
static link_pointer unlink_nodes(bucket& b, // Find the node before begin.
node_pointer begin, node_pointer end) link_pointer prev = this->get_previous_start(bucket_index);
{ while(prev->next_ != begin) prev = prev->next_;
link_pointer prev = b.next_;
while (prev->next_ != begin) prev = prev->next_; // Delete the nodes.
prev->next_ = end; do {
return prev; this->delete_node(prev);
bucket_index = this->fix_bucket(bucket_index, prev);
} while (prev->next_ != end);
} }
//////////////////////////////////////////////////////////////////////// ////////////////////////////////////////////////////////////////////////