xref: /llvm-project/libcxx/include/__algorithm/sort_heap.h (revision 580f60484e193b4f7f853ea34cc35f4a3240dfa7)
1134723edSLouis Dionne //===----------------------------------------------------------------------===//
2134723edSLouis Dionne //
3134723edSLouis Dionne // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4134723edSLouis Dionne // See https://llvm.org/LICENSE.txt for license information.
5134723edSLouis Dionne // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6134723edSLouis Dionne //
7134723edSLouis Dionne //===----------------------------------------------------------------------===//
8134723edSLouis Dionne 
9134723edSLouis Dionne #ifndef _LIBCPP___ALGORITHM_SORT_HEAP_H
10134723edSLouis Dionne #define _LIBCPP___ALGORITHM_SORT_HEAP_H
11134723edSLouis Dionne 
12134723edSLouis Dionne #include <__algorithm/comp.h>
13134723edSLouis Dionne #include <__algorithm/comp_ref_type.h>
14a7c3379cSKonstantin Varlamov #include <__algorithm/iterator_operations.h>
15134723edSLouis Dionne #include <__algorithm/pop_heap.h>
164d81a46fSArthur O'Dwyer #include <__config>
177e1ee1e1SDanila Kutenin #include <__debug_utils/strict_weak_ordering_check.h>
18134723edSLouis Dionne #include <__iterator/iterator_traits.h>
19*580f6048SNikolas Klauser #include <__type_traits/is_assignable.h>
20*580f6048SNikolas Klauser #include <__type_traits/is_constructible.h>
21c945bd0dSKonstantin Varlamov #include <__utility/move.h>
22134723edSLouis Dionne 
23134723edSLouis Dionne #if !defined(_LIBCPP_HAS_NO_PRAGMA_SYSTEM_HEADER)
24134723edSLouis Dionne #  pragma GCC system_header
25134723edSLouis Dionne #endif
26134723edSLouis Dionne 
277b462251SLouis Dionne _LIBCPP_PUSH_MACROS
287b462251SLouis Dionne #include <__undef_macros>
297b462251SLouis Dionne 
30134723edSLouis Dionne _LIBCPP_BEGIN_NAMESPACE_STD
31134723edSLouis Dionne 
32a7c3379cSKonstantin Varlamov template <class _AlgPolicy, class _Compare, class _RandomAccessIterator>
339783f28cSLouis Dionne inline _LIBCPP_HIDE_FROM_ABI _LIBCPP_CONSTEXPR_SINCE_CXX14 void
__sort_heap(_RandomAccessIterator __first,_RandomAccessIterator __last,_Compare && __comp)349783f28cSLouis Dionne __sort_heap(_RandomAccessIterator __first, _RandomAccessIterator __last, _Compare&& __comp) {
357e1ee1e1SDanila Kutenin   _RandomAccessIterator __saved_last   = __last;
36ed2d3644SNikolas Klauser   __comp_ref_type<_Compare> __comp_ref = __comp;
37c945bd0dSKonstantin Varlamov 
38c945bd0dSKonstantin Varlamov   using difference_type = typename iterator_traits<_RandomAccessIterator>::difference_type;
39134723edSLouis Dionne   for (difference_type __n = __last - __first; __n > 1; --__last, (void)--__n)
40db7d7959SKonstantin Varlamov     std::__pop_heap<_AlgPolicy>(__first, __last, __comp_ref, __n);
417e1ee1e1SDanila Kutenin   std::__check_strict_weak_ordering_sorted(__first, __saved_last, __comp_ref);
42134723edSLouis Dionne }
43134723edSLouis Dionne 
44134723edSLouis Dionne template <class _RandomAccessIterator, class _Compare>
459783f28cSLouis Dionne inline _LIBCPP_HIDE_FROM_ABI _LIBCPP_CONSTEXPR_SINCE_CXX20 void
sort_heap(_RandomAccessIterator __first,_RandomAccessIterator __last,_Compare __comp)469783f28cSLouis Dionne sort_heap(_RandomAccessIterator __first, _RandomAccessIterator __last, _Compare __comp) {
475dd19adaSvarconst   static_assert(std::is_copy_constructible<_RandomAccessIterator>::value, "Iterators must be copy constructible.");
485dd19adaSvarconst   static_assert(std::is_copy_assignable<_RandomAccessIterator>::value, "Iterators must be copy assignable.");
495dd19adaSvarconst 
50a7c3379cSKonstantin Varlamov   std::__sort_heap<_ClassicAlgPolicy>(std::move(__first), std::move(__last), __comp);
51134723edSLouis Dionne }
52134723edSLouis Dionne 
53134723edSLouis Dionne template <class _RandomAccessIterator>
549783f28cSLouis Dionne inline _LIBCPP_HIDE_FROM_ABI _LIBCPP_CONSTEXPR_SINCE_CXX20 void
sort_heap(_RandomAccessIterator __first,_RandomAccessIterator __last)559783f28cSLouis Dionne sort_heap(_RandomAccessIterator __first, _RandomAccessIterator __last) {
5688632e48SNikolas Klauser   std::sort_heap(std::move(__first), std::move(__last), __less<>());
57134723edSLouis Dionne }
58134723edSLouis Dionne 
59134723edSLouis Dionne _LIBCPP_END_NAMESPACE_STD
60134723edSLouis Dionne 
617b462251SLouis Dionne _LIBCPP_POP_MACROS
627b462251SLouis Dionne 
63134723edSLouis Dionne #endif // _LIBCPP___ALGORITHM_SORT_HEAP_H
64