1064 lines
28 KiB
C++
1064 lines
28 KiB
C++
// -*- C++ -*-
|
|
|
|
// Copyright (C) 2007-2017 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 */
|