|
- // -*- C++ -*-
-
- // Copyright (C) 2007-2020 Free Software Foundation, Inc.
- //
- // This file is part of the GNU ISO C++ Library. This library is free
- // software; you can redistribute it and/or modify it under the terms
- // of the GNU General Public License as published by the Free Software
- // Foundation; either version 3, or (at your option) any later
- // version.
-
- // This library is distributed in the hope that it will be useful, but
- // WITHOUT ANY WARRANTY; without even the implied warranty of
- // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
- // General Public License for more details.
-
- // Under Section 7 of GPL version 3, you are granted additional
- // permissions described in the GCC Runtime Library Exception, version
- // 3.1, as published by the Free Software Foundation.
-
- // You should have received a copy of the GNU General Public License and
- // a copy of the GCC Runtime Library Exception along with this program;
- // see the files COPYING3 and COPYING.RUNTIME respectively. If not, see
- // <http://www.gnu.org/licenses/>.
-
- /** @file parallel/losertree.h
- * @brief Many generic loser tree variants.
- * This file is a GNU parallel extension to the Standard C++ Library.
- */
-
- // Written by Johannes Singler.
-
- #ifndef _GLIBCXX_PARALLEL_LOSERTREE_H
- #define _GLIBCXX_PARALLEL_LOSERTREE_H 1
-
- #include <bits/stl_algobase.h>
- #include <bits/stl_function.h>
- #include <parallel/features.h>
- #include <parallel/base.h>
-
- namespace __gnu_parallel
- {
- /**
- * @brief Guarded loser/tournament tree.
- *
- * The smallest element is at the top.
- *
- * Guarding is done explicitly through one flag _M_sup per element,
- * inf is not needed due to a better initialization routine. This
- * is a well-performing variant.
- *
- * @param _Tp the element type
- * @param _Compare the comparator to use, defaults to std::less<_Tp>
- */
- template<typename _Tp, typename _Compare>
- class _LoserTreeBase
- {
- protected:
- /** @brief Internal representation of a _LoserTree element. */
- struct _Loser
- {
- /** @brief flag, true iff this is a "maximum" __sentinel. */
- bool _M_sup;
- /** @brief __index of the __source __sequence. */
- int _M_source;
- /** @brief _M_key of the element in the _LoserTree. */
- _Tp _M_key;
- };
-
- unsigned int _M_ik, _M_k, _M_offset;
-
- /** log_2{_M_k} */
- unsigned int _M_log_k;
-
- /** @brief _LoserTree __elements. */
- _Loser* _M_losers;
-
- /** @brief _Compare to use. */
- _Compare _M_comp;
-
- /**
- * @brief State flag that determines whether the _LoserTree is empty.
- *
- * Only used for building the _LoserTree.
- */
- bool _M_first_insert;
-
- public:
- /**
- * @brief The constructor.
- *
- * @param __k The number of sequences to merge.
- * @param __comp The comparator to use.
- */
- _LoserTreeBase(unsigned int __k, _Compare __comp)
- : _M_comp(__comp)
- {
- _M_ik = __k;
-
- // Compute log_2{_M_k} for the _Loser Tree
- _M_log_k = __rd_log2(_M_ik - 1) + 1;
-
- // Next greater power of 2.
- _M_k = 1 << _M_log_k;
- _M_offset = _M_k;
-
- // Avoid default-constructing _M_losers[]._M_key
- _M_losers = static_cast<_Loser*>(::operator new(2 * _M_k
- * sizeof(_Loser)));
- for (unsigned int __i = _M_ik - 1; __i < _M_k; ++__i)
- _M_losers[__i + _M_k]._M_sup = true;
-
- _M_first_insert = true;
- }
-
- /**
- * @brief The destructor.
- */
- ~_LoserTreeBase()
- {
- for (unsigned int __i = 0; __i < (2 * _M_k); ++__i)
- _M_losers[__i].~_Loser();
- ::operator delete(_M_losers);
- }
-
- /**
- * @brief Initializes the sequence "_M_source" with the element "__key".
- *
- * @param __key the element to insert
- * @param __source __index of the __source __sequence
- * @param __sup flag that determines whether the value to insert is an
- * explicit __supremum.
- */
- void
- __insert_start(const _Tp& __key, int __source, bool __sup)
- {
- unsigned int __pos = _M_k + __source;
-
- if (_M_first_insert)
- {
- // Construct all keys, so we can easily destruct them.
- for (unsigned int __i = 0; __i < (2 * _M_k); ++__i)
- ::new(&(_M_losers[__i]._M_key)) _Tp(__key);
- _M_first_insert = false;
- }
- else
- _M_losers[__pos]._M_key = __key;
-
- _M_losers[__pos]._M_sup = __sup;
- _M_losers[__pos]._M_source = __source;
- }
-
- /**
- * @return the index of the sequence with the smallest element.
- */
- int __get_min_source()
- { return _M_losers[0]._M_source; }
- };
-
- /**
- * @brief Stable _LoserTree variant.
- *
- * Provides the stable implementations of insert_start, __init_winner,
- * __init and __delete_min_insert.
- *
- * Unstable variant is done using partial specialisation below.
- */
- template<bool __stable/* default == true */, typename _Tp,
- typename _Compare>
- class _LoserTree
- : public _LoserTreeBase<_Tp, _Compare>
- {
- typedef _LoserTreeBase<_Tp, _Compare> _Base;
- using _Base::_M_k;
- using _Base::_M_comp;
- using _Base::_M_losers;
- using _Base::_M_first_insert;
-
- public:
- _LoserTree(unsigned int __k, _Compare __comp)
- : _Base::_LoserTreeBase(__k, __comp)
- { }
-
- unsigned int
- __init_winner(unsigned int __root)
- {
- if (__root >= _M_k)
- return __root;
- else
- {
- unsigned int __left = __init_winner(2 * __root);
- unsigned int __right = __init_winner(2 * __root + 1);
- if (_M_losers[__right]._M_sup
- || (!_M_losers[__left]._M_sup
- && !_M_comp(_M_losers[__right]._M_key,
- _M_losers[__left]._M_key)))
- {
- // Left one is less or equal.
- _M_losers[__root] = _M_losers[__right];
- return __left;
- }
- else
- {
- // Right one is less.
- _M_losers[__root] = _M_losers[__left];
- return __right;
- }
- }
- }
-
- void __init()
- { _M_losers[0] = _M_losers[__init_winner(1)]; }
-
- /**
- * @brief Delete the smallest element and insert a new element from
- * the previously smallest element's sequence.
- *
- * This implementation is stable.
- */
- // Do not pass a const reference since __key will be used as
- // local variable.
- void
- __delete_min_insert(_Tp __key, bool __sup)
- {
- using std::swap;
- #if _GLIBCXX_PARALLEL_ASSERTIONS
- // no dummy sequence can ever be at the top!
- _GLIBCXX_PARALLEL_ASSERT(_M_losers[0]._M_source != -1);
- #endif
-
- int __source = _M_losers[0]._M_source;
- for (unsigned int __pos = (_M_k + __source) / 2; __pos > 0;
- __pos /= 2)
- {
- // The smaller one gets promoted, ties are broken by _M_source.
- if ((__sup && (!_M_losers[__pos]._M_sup
- || _M_losers[__pos]._M_source < __source))
- || (!__sup && !_M_losers[__pos]._M_sup
- && ((_M_comp(_M_losers[__pos]._M_key, __key))
- || (!_M_comp(__key, _M_losers[__pos]._M_key)
- && _M_losers[__pos]._M_source < __source))))
- {
- // The other one is smaller.
- std::swap(_M_losers[__pos]._M_sup, __sup);
- std::swap(_M_losers[__pos]._M_source, __source);
- swap(_M_losers[__pos]._M_key, __key);
- }
- }
-
- _M_losers[0]._M_sup = __sup;
- _M_losers[0]._M_source = __source;
- _M_losers[0]._M_key = __key;
- }
- };
-
- /**
- * @brief Unstable _LoserTree variant.
- *
- * Stability (non-stable here) is selected with partial specialization.
- */
- template<typename _Tp, typename _Compare>
- class _LoserTree</* __stable == */false, _Tp, _Compare>
- : public _LoserTreeBase<_Tp, _Compare>
- {
- typedef _LoserTreeBase<_Tp, _Compare> _Base;
- using _Base::_M_log_k;
- using _Base::_M_k;
- using _Base::_M_comp;
- using _Base::_M_losers;
- using _Base::_M_first_insert;
-
- public:
- _LoserTree(unsigned int __k, _Compare __comp)
- : _Base::_LoserTreeBase(__k, __comp)
- { }
-
- /**
- * Computes the winner of the competition at position "__root".
- *
- * Called recursively (starting at 0) to build the initial tree.
- *
- * @param __root __index of the "game" to start.
- */
- unsigned int
- __init_winner(unsigned int __root)
- {
- if (__root >= _M_k)
- return __root;
- else
- {
- unsigned int __left = __init_winner(2 * __root);
- unsigned int __right = __init_winner(2 * __root + 1);
- if (_M_losers[__right]._M_sup
- || (!_M_losers[__left]._M_sup
- && !_M_comp(_M_losers[__right]._M_key,
- _M_losers[__left]._M_key)))
- {
- // Left one is less or equal.
- _M_losers[__root] = _M_losers[__right];
- return __left;
- }
- else
- {
- // Right one is less.
- _M_losers[__root] = _M_losers[__left];
- return __right;
- }
- }
- }
-
- void
- __init()
- { _M_losers[0] = _M_losers[__init_winner(1)]; }
-
- /**
- * Delete the _M_key smallest element and insert the element __key
- * instead.
- *
- * @param __key the _M_key to insert
- * @param __sup true iff __key is an explicitly marked supremum
- */
- // Do not pass a const reference since __key will be used as local
- // variable.
- void
- __delete_min_insert(_Tp __key, bool __sup)
- {
- using std::swap;
- #if _GLIBCXX_PARALLEL_ASSERTIONS
- // no dummy sequence can ever be at the top!
- _GLIBCXX_PARALLEL_ASSERT(_M_losers[0]._M_source != -1);
- #endif
-
- int __source = _M_losers[0]._M_source;
- for (unsigned int __pos = (_M_k + __source) / 2; __pos > 0;
- __pos /= 2)
- {
- // The smaller one gets promoted.
- if (__sup || (!_M_losers[__pos]._M_sup
- && _M_comp(_M_losers[__pos]._M_key, __key)))
- {
- // The other one is smaller.
- std::swap(_M_losers[__pos]._M_sup, __sup);
- std::swap(_M_losers[__pos]._M_source, __source);
- swap(_M_losers[__pos]._M_key, __key);
- }
- }
-
- _M_losers[0]._M_sup = __sup;
- _M_losers[0]._M_source = __source;
- _M_losers[0]._M_key = __key;
- }
- };
-
- /**
- * @brief Base class of _Loser Tree implementation using pointers.
- */
- template<typename _Tp, typename _Compare>
- class _LoserTreePointerBase
- {
- protected:
- /** @brief Internal representation of _LoserTree __elements. */
- struct _Loser
- {
- bool _M_sup;
- int _M_source;
- const _Tp* _M_keyp;
- };
-
- unsigned int _M_ik, _M_k, _M_offset;
- _Loser* _M_losers;
- _Compare _M_comp;
-
- public:
- _LoserTreePointerBase(unsigned int __k,
- _Compare __comp = std::less<_Tp>())
- : _M_comp(__comp)
- {
- _M_ik = __k;
-
- // Next greater power of 2.
- _M_k = 1 << (__rd_log2(_M_ik - 1) + 1);
- _M_offset = _M_k;
- _M_losers = new _Loser[_M_k * 2];
- for (unsigned int __i = _M_ik - 1; __i < _M_k; __i++)
- _M_losers[__i + _M_k]._M_sup = true;
- }
-
- ~_LoserTreePointerBase()
- { delete[] _M_losers; }
-
- int __get_min_source()
- { return _M_losers[0]._M_source; }
-
- void __insert_start(const _Tp& __key, int __source, bool __sup)
- {
- unsigned int __pos = _M_k + __source;
-
- _M_losers[__pos]._M_sup = __sup;
- _M_losers[__pos]._M_source = __source;
- _M_losers[__pos]._M_keyp = &__key;
- }
- };
-
- /**
- * @brief Stable _LoserTree implementation.
- *
- * The unstable variant is implemented using partial instantiation below.
- */
- template<bool __stable/* default == true */, typename _Tp, typename _Compare>
- class _LoserTreePointer
- : public _LoserTreePointerBase<_Tp, _Compare>
- {
- typedef _LoserTreePointerBase<_Tp, _Compare> _Base;
- using _Base::_M_k;
- using _Base::_M_comp;
- using _Base::_M_losers;
-
- public:
- _LoserTreePointer(unsigned int __k, _Compare __comp = std::less<_Tp>())
- : _Base::_LoserTreePointerBase(__k, __comp)
- { }
-
- unsigned int
- __init_winner(unsigned int __root)
- {
- if (__root >= _M_k)
- return __root;
- else
- {
- unsigned int __left = __init_winner(2 * __root);
- unsigned int __right = __init_winner(2 * __root + 1);
- if (_M_losers[__right]._M_sup
- || (!_M_losers[__left]._M_sup
- && !_M_comp(*_M_losers[__right]._M_keyp,
- *_M_losers[__left]._M_keyp)))
- {
- // Left one is less or equal.
- _M_losers[__root] = _M_losers[__right];
- return __left;
- }
- else
- {
- // Right one is less.
- _M_losers[__root] = _M_losers[__left];
- return __right;
- }
- }
- }
-
- void __init()
- { _M_losers[0] = _M_losers[__init_winner(1)]; }
-
- void __delete_min_insert(const _Tp& __key, bool __sup)
- {
- #if _GLIBCXX_PARALLEL_ASSERTIONS
- // no dummy sequence can ever be at the top!
- _GLIBCXX_PARALLEL_ASSERT(_M_losers[0]._M_source != -1);
- #endif
-
- const _Tp* __keyp = &__key;
- int __source = _M_losers[0]._M_source;
- for (unsigned int __pos = (_M_k + __source) / 2; __pos > 0;
- __pos /= 2)
- {
- // The smaller one gets promoted, ties are broken by __source.
- if ((__sup && (!_M_losers[__pos]._M_sup
- || _M_losers[__pos]._M_source < __source))
- || (!__sup && !_M_losers[__pos]._M_sup &&
- ((_M_comp(*_M_losers[__pos]._M_keyp, *__keyp))
- || (!_M_comp(*__keyp, *_M_losers[__pos]._M_keyp)
- && _M_losers[__pos]._M_source < __source))))
- {
- // The other one is smaller.
- std::swap(_M_losers[__pos]._M_sup, __sup);
- std::swap(_M_losers[__pos]._M_source, __source);
- std::swap(_M_losers[__pos]._M_keyp, __keyp);
- }
- }
-
- _M_losers[0]._M_sup = __sup;
- _M_losers[0]._M_source = __source;
- _M_losers[0]._M_keyp = __keyp;
- }
- };
-
- /**
- * @brief Unstable _LoserTree implementation.
- *
- * The stable variant is above.
- */
- template<typename _Tp, typename _Compare>
- class _LoserTreePointer</* __stable == */false, _Tp, _Compare>
- : public _LoserTreePointerBase<_Tp, _Compare>
- {
- typedef _LoserTreePointerBase<_Tp, _Compare> _Base;
- using _Base::_M_k;
- using _Base::_M_comp;
- using _Base::_M_losers;
-
- public:
- _LoserTreePointer(unsigned int __k, _Compare __comp = std::less<_Tp>())
- : _Base::_LoserTreePointerBase(__k, __comp)
- { }
-
- unsigned int
- __init_winner(unsigned int __root)
- {
- if (__root >= _M_k)
- return __root;
- else
- {
- unsigned int __left = __init_winner(2 * __root);
- unsigned int __right = __init_winner(2 * __root + 1);
- if (_M_losers[__right]._M_sup
- || (!_M_losers[__left]._M_sup
- && !_M_comp(*_M_losers[__right]._M_keyp,
- *_M_losers[__left]._M_keyp)))
- {
- // Left one is less or equal.
- _M_losers[__root] = _M_losers[__right];
- return __left;
- }
- else
- {
- // Right one is less.
- _M_losers[__root] = _M_losers[__left];
- return __right;
- }
- }
- }
-
- void __init()
- { _M_losers[0] = _M_losers[__init_winner(1)]; }
-
- void __delete_min_insert(const _Tp& __key, bool __sup)
- {
- #if _GLIBCXX_PARALLEL_ASSERTIONS
- // no dummy sequence can ever be at the top!
- _GLIBCXX_PARALLEL_ASSERT(_M_losers[0]._M_source != -1);
- #endif
-
- const _Tp* __keyp = &__key;
- int __source = _M_losers[0]._M_source;
- for (unsigned int __pos = (_M_k + __source) / 2; __pos > 0;
- __pos /= 2)
- {
- // The smaller one gets promoted.
- if (__sup || (!_M_losers[__pos]._M_sup
- && _M_comp(*_M_losers[__pos]._M_keyp, *__keyp)))
- {
- // The other one is smaller.
- std::swap(_M_losers[__pos]._M_sup, __sup);
- std::swap(_M_losers[__pos]._M_source, __source);
- std::swap(_M_losers[__pos]._M_keyp, __keyp);
- }
- }
-
- _M_losers[0]._M_sup = __sup;
- _M_losers[0]._M_source = __source;
- _M_losers[0]._M_keyp = __keyp;
- }
- };
-
- /** @brief Base class for unguarded _LoserTree implementation.
- *
- * The whole element is copied into the tree structure.
- *
- * No guarding is done, therefore not a single input sequence must
- * run empty. Unused __sequence heads are marked with a sentinel which
- * is > all elements that are to be merged.
- *
- * This is a very fast variant.
- */
- template<typename _Tp, typename _Compare>
- class _LoserTreeUnguardedBase
- {
- protected:
- struct _Loser
- {
- int _M_source;
- _Tp _M_key;
- };
-
- unsigned int _M_ik, _M_k, _M_offset;
- _Loser* _M_losers;
- _Compare _M_comp;
-
- public:
- _LoserTreeUnguardedBase(unsigned int __k, const _Tp& __sentinel,
- _Compare __comp = std::less<_Tp>())
- : _M_comp(__comp)
- {
- _M_ik = __k;
-
- // Next greater power of 2.
- _M_k = 1 << (__rd_log2(_M_ik - 1) + 1);
- _M_offset = _M_k;
- // Avoid default-constructing _M_losers[]._M_key
- _M_losers = static_cast<_Loser*>(::operator new(2 * _M_k
- * sizeof(_Loser)));
-
- for (unsigned int __i = 0; __i < _M_k; ++__i)
- {
- ::new(&(_M_losers[__i]._M_key)) _Tp(__sentinel);
- _M_losers[__i]._M_source = -1;
- }
- for (unsigned int __i = _M_k + _M_ik - 1; __i < (2 * _M_k); ++__i)
- {
- ::new(&(_M_losers[__i]._M_key)) _Tp(__sentinel);
- _M_losers[__i]._M_source = -1;
- }
- }
-
- ~_LoserTreeUnguardedBase()
- {
- for (unsigned int __i = 0; __i < (2 * _M_k); ++__i)
- _M_losers[__i].~_Loser();
- ::operator delete(_M_losers);
- }
-
- int
- __get_min_source()
- {
- #if _GLIBCXX_PARALLEL_ASSERTIONS
- // no dummy sequence can ever be at the top!
- _GLIBCXX_PARALLEL_ASSERT(_M_losers[0]._M_source != -1);
- #endif
- return _M_losers[0]._M_source;
- }
-
- void
- __insert_start(const _Tp& __key, int __source, bool)
- {
- unsigned int __pos = _M_k + __source;
-
- ::new(&(_M_losers[__pos]._M_key)) _Tp(__key);
- _M_losers[__pos]._M_source = __source;
- }
- };
-
- /**
- * @brief Stable implementation of unguarded _LoserTree.
- *
- * Unstable variant is selected below with partial specialization.
- */
- template<bool __stable/* default == true */, typename _Tp, typename _Compare>
- class _LoserTreeUnguarded
- : public _LoserTreeUnguardedBase<_Tp, _Compare>
- {
- typedef _LoserTreeUnguardedBase<_Tp, _Compare> _Base;
- using _Base::_M_k;
- using _Base::_M_comp;
- using _Base::_M_losers;
-
- public:
- _LoserTreeUnguarded(unsigned int __k, const _Tp& __sentinel,
- _Compare __comp = std::less<_Tp>())
- : _Base::_LoserTreeUnguardedBase(__k, __sentinel, __comp)
- { }
-
- unsigned int
- __init_winner(unsigned int __root)
- {
- if (__root >= _M_k)
- return __root;
- else
- {
- unsigned int __left = __init_winner(2 * __root);
- unsigned int __right = __init_winner(2 * __root + 1);
- if (!_M_comp(_M_losers[__right]._M_key,
- _M_losers[__left]._M_key))
- {
- // Left one is less or equal.
- _M_losers[__root] = _M_losers[__right];
- return __left;
- }
- else
- {
- // Right one is less.
- _M_losers[__root] = _M_losers[__left];
- return __right;
- }
- }
- }
-
- void
- __init()
- {
- _M_losers[0] = _M_losers[__init_winner(1)];
-
- #if _GLIBCXX_PARALLEL_ASSERTIONS
- // no dummy sequence can ever be at the top at the beginning
- // (0 sequences!)
- _GLIBCXX_PARALLEL_ASSERT(_M_losers[0]._M_source != -1);
- #endif
- }
-
- // Do not pass a const reference since __key will be used as
- // local variable.
- void
- __delete_min_insert(_Tp __key, bool)
- {
- using std::swap;
- #if _GLIBCXX_PARALLEL_ASSERTIONS
- // no dummy sequence can ever be at the top!
- _GLIBCXX_PARALLEL_ASSERT(_M_losers[0]._M_source != -1);
- #endif
-
- int __source = _M_losers[0]._M_source;
- for (unsigned int __pos = (_M_k + __source) / 2; __pos > 0;
- __pos /= 2)
- {
- // The smaller one gets promoted, ties are broken by _M_source.
- if (_M_comp(_M_losers[__pos]._M_key, __key)
- || (!_M_comp(__key, _M_losers[__pos]._M_key)
- && _M_losers[__pos]._M_source < __source))
- {
- // The other one is smaller.
- std::swap(_M_losers[__pos]._M_source, __source);
- swap(_M_losers[__pos]._M_key, __key);
- }
- }
-
- _M_losers[0]._M_source = __source;
- _M_losers[0]._M_key = __key;
- }
- };
-
- /**
- * @brief Non-Stable implementation of unguarded _LoserTree.
- *
- * Stable implementation is above.
- */
- template<typename _Tp, typename _Compare>
- class _LoserTreeUnguarded</* __stable == */false, _Tp, _Compare>
- : public _LoserTreeUnguardedBase<_Tp, _Compare>
- {
- typedef _LoserTreeUnguardedBase<_Tp, _Compare> _Base;
- using _Base::_M_k;
- using _Base::_M_comp;
- using _Base::_M_losers;
-
- public:
- _LoserTreeUnguarded(unsigned int __k, const _Tp& __sentinel,
- _Compare __comp = std::less<_Tp>())
- : _Base::_LoserTreeUnguardedBase(__k, __sentinel, __comp)
- { }
-
- unsigned int
- __init_winner(unsigned int __root)
- {
- if (__root >= _M_k)
- return __root;
- else
- {
- unsigned int __left = __init_winner(2 * __root);
- unsigned int __right = __init_winner(2 * __root + 1);
-
- #if _GLIBCXX_PARALLEL_ASSERTIONS
- // If __left one is sentinel then __right one must be, too.
- if (_M_losers[__left]._M_source == -1)
- _GLIBCXX_PARALLEL_ASSERT(_M_losers[__right]._M_source == -1);
- #endif
-
- if (!_M_comp(_M_losers[__right]._M_key,
- _M_losers[__left]._M_key))
- {
- // Left one is less or equal.
- _M_losers[__root] = _M_losers[__right];
- return __left;
- }
- else
- {
- // Right one is less.
- _M_losers[__root] = _M_losers[__left];
- return __right;
- }
- }
- }
-
- void
- __init()
- {
- _M_losers[0] = _M_losers[__init_winner(1)];
-
- #if _GLIBCXX_PARALLEL_ASSERTIONS
- // no dummy sequence can ever be at the top at the beginning
- // (0 sequences!)
- _GLIBCXX_PARALLEL_ASSERT(_M_losers[0]._M_source != -1);
- #endif
- }
-
- // Do not pass a const reference since __key will be used as
- // local variable.
- void
- __delete_min_insert(_Tp __key, bool)
- {
- using std::swap;
- #if _GLIBCXX_PARALLEL_ASSERTIONS
- // no dummy sequence can ever be at the top!
- _GLIBCXX_PARALLEL_ASSERT(_M_losers[0]._M_source != -1);
- #endif
-
- int __source = _M_losers[0]._M_source;
- for (unsigned int __pos = (_M_k + __source) / 2; __pos > 0;
- __pos /= 2)
- {
- // The smaller one gets promoted.
- if (_M_comp(_M_losers[__pos]._M_key, __key))
- {
- // The other one is smaller.
- std::swap(_M_losers[__pos]._M_source, __source);
- swap(_M_losers[__pos]._M_key, __key);
- }
- }
-
- _M_losers[0]._M_source = __source;
- _M_losers[0]._M_key = __key;
- }
- };
-
- /** @brief Unguarded loser tree, keeping only pointers to the
- * elements in the tree structure.
- *
- * No guarding is done, therefore not a single input sequence must
- * run empty. This is a very fast variant.
- */
- template<typename _Tp, typename _Compare>
- class _LoserTreePointerUnguardedBase
- {
- protected:
- struct _Loser
- {
- int _M_source;
- const _Tp* _M_keyp;
- };
-
- unsigned int _M_ik, _M_k, _M_offset;
- _Loser* _M_losers;
- _Compare _M_comp;
-
- public:
-
- _LoserTreePointerUnguardedBase(unsigned int __k, const _Tp& __sentinel,
- _Compare __comp = std::less<_Tp>())
- : _M_comp(__comp)
- {
- _M_ik = __k;
-
- // Next greater power of 2.
- _M_k = 1 << (__rd_log2(_M_ik - 1) + 1);
- _M_offset = _M_k;
- // Avoid default-constructing _M_losers[]._M_key
- _M_losers = new _Loser[2 * _M_k];
-
- for (unsigned int __i = _M_k + _M_ik - 1; __i < (2 * _M_k); ++__i)
- {
- _M_losers[__i]._M_keyp = &__sentinel;
- _M_losers[__i]._M_source = -1;
- }
- }
-
- ~_LoserTreePointerUnguardedBase()
- { delete[] _M_losers; }
-
- int
- __get_min_source()
- {
- #if _GLIBCXX_PARALLEL_ASSERTIONS
- // no dummy sequence can ever be at the top!
- _GLIBCXX_PARALLEL_ASSERT(_M_losers[0]._M_source != -1);
- #endif
- return _M_losers[0]._M_source;
- }
-
- void
- __insert_start(const _Tp& __key, int __source, bool)
- {
- unsigned int __pos = _M_k + __source;
-
- _M_losers[__pos]._M_keyp = &__key;
- _M_losers[__pos]._M_source = __source;
- }
- };
-
- /**
- * @brief Stable unguarded _LoserTree variant storing pointers.
- *
- * Unstable variant is implemented below using partial specialization.
- */
- template<bool __stable/* default == true */, typename _Tp, typename _Compare>
- class _LoserTreePointerUnguarded
- : public _LoserTreePointerUnguardedBase<_Tp, _Compare>
- {
- typedef _LoserTreePointerUnguardedBase<_Tp, _Compare> _Base;
- using _Base::_M_k;
- using _Base::_M_comp;
- using _Base::_M_losers;
-
- public:
- _LoserTreePointerUnguarded(unsigned int __k, const _Tp& __sentinel,
- _Compare __comp = std::less<_Tp>())
- : _Base::_LoserTreePointerUnguardedBase(__k, __sentinel, __comp)
- { }
-
- unsigned int
- __init_winner(unsigned int __root)
- {
- if (__root >= _M_k)
- return __root;
- else
- {
- unsigned int __left = __init_winner(2 * __root);
- unsigned int __right = __init_winner(2 * __root + 1);
- if (!_M_comp(*_M_losers[__right]._M_keyp,
- *_M_losers[__left]._M_keyp))
- {
- // Left one is less or equal.
- _M_losers[__root] = _M_losers[__right];
- return __left;
- }
- else
- {
- // Right one is less.
- _M_losers[__root] = _M_losers[__left];
- return __right;
- }
- }
- }
-
- void
- __init()
- {
- _M_losers[0] = _M_losers[__init_winner(1)];
-
- #if _GLIBCXX_PARALLEL_ASSERTIONS
- // no dummy sequence can ever be at the top at the beginning
- // (0 sequences!)
- _GLIBCXX_PARALLEL_ASSERT(_M_losers[0]._M_source != -1);
- #endif
- }
-
- void
- __delete_min_insert(const _Tp& __key, bool __sup)
- {
- #if _GLIBCXX_PARALLEL_ASSERTIONS
- // no dummy sequence can ever be at the top!
- _GLIBCXX_PARALLEL_ASSERT(_M_losers[0]._M_source != -1);
- #endif
-
- const _Tp* __keyp = &__key;
- int __source = _M_losers[0]._M_source;
- for (unsigned int __pos = (_M_k + __source) / 2; __pos > 0;
- __pos /= 2)
- {
- // The smaller one gets promoted, ties are broken by _M_source.
- if (_M_comp(*_M_losers[__pos]._M_keyp, *__keyp)
- || (!_M_comp(*__keyp, *_M_losers[__pos]._M_keyp)
- && _M_losers[__pos]._M_source < __source))
- {
- // The other one is smaller.
- std::swap(_M_losers[__pos]._M_source, __source);
- std::swap(_M_losers[__pos]._M_keyp, __keyp);
- }
- }
-
- _M_losers[0]._M_source = __source;
- _M_losers[0]._M_keyp = __keyp;
- }
- };
-
- /**
- * @brief Unstable unguarded _LoserTree variant storing pointers.
- *
- * Stable variant is above.
- */
- template<typename _Tp, typename _Compare>
- class _LoserTreePointerUnguarded</* __stable == */false, _Tp, _Compare>
- : public _LoserTreePointerUnguardedBase<_Tp, _Compare>
- {
- typedef _LoserTreePointerUnguardedBase<_Tp, _Compare> _Base;
- using _Base::_M_k;
- using _Base::_M_comp;
- using _Base::_M_losers;
-
- public:
- _LoserTreePointerUnguarded(unsigned int __k, const _Tp& __sentinel,
- _Compare __comp = std::less<_Tp>())
- : _Base::_LoserTreePointerUnguardedBase(__k, __sentinel, __comp)
- { }
-
- unsigned int
- __init_winner(unsigned int __root)
- {
- if (__root >= _M_k)
- return __root;
- else
- {
- unsigned int __left = __init_winner(2 * __root);
- unsigned int __right = __init_winner(2 * __root + 1);
-
- #if _GLIBCXX_PARALLEL_ASSERTIONS
- // If __left one is sentinel then __right one must be, too.
- if (_M_losers[__left]._M_source == -1)
- _GLIBCXX_PARALLEL_ASSERT(_M_losers[__right]._M_source == -1);
- #endif
-
- if (!_M_comp(*_M_losers[__right]._M_keyp,
- *_M_losers[__left]._M_keyp))
- {
- // Left one is less or equal.
- _M_losers[__root] = _M_losers[__right];
- return __left;
- }
- else
- {
- // Right one is less.
- _M_losers[__root] = _M_losers[__left];
- return __right;
- }
- }
- }
-
- void
- __init()
- {
- _M_losers[0] = _M_losers[__init_winner(1)];
-
- #if _GLIBCXX_PARALLEL_ASSERTIONS
- // no dummy sequence can ever be at the top at the beginning
- // (0 sequences!)
- _GLIBCXX_PARALLEL_ASSERT(_M_losers[0]._M_source != -1);
- #endif
- }
-
- void
- __delete_min_insert(const _Tp& __key, bool __sup)
- {
- #if _GLIBCXX_PARALLEL_ASSERTIONS
- // no dummy sequence can ever be at the top!
- _GLIBCXX_PARALLEL_ASSERT(_M_losers[0]._M_source != -1);
- #endif
-
- const _Tp* __keyp = &__key;
- int __source = _M_losers[0]._M_source;
- for (unsigned int __pos = (_M_k + __source) / 2; __pos > 0;
- __pos /= 2)
- {
- // The smaller one gets promoted.
- if (_M_comp(*(_M_losers[__pos]._M_keyp), *__keyp))
- {
- // The other one is smaller.
- std::swap(_M_losers[__pos]._M_source, __source);
- std::swap(_M_losers[__pos]._M_keyp, __keyp);
- }
- }
-
- _M_losers[0]._M_source = __source;
- _M_losers[0]._M_keyp = __keyp;
- }
- };
- } // namespace __gnu_parallel
-
- #endif /* _GLIBCXX_PARALLEL_LOSERTREE_H */
|